{
 "cells": [
  {
   "cell_type": "code",
   "execution_count": 1,
   "metadata": {},
   "outputs": [],
   "source": [
    "from sentence_transformers import SentenceTransformer, util\n",
    "import pandas as pd\n",
    "import re\n",
    "from OpenAITools.JRCTTools import split_target_English, WordProcessor"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 2,
   "metadata": {},
   "outputs": [],
   "source": [
    "exclusive_words = [\"triple negative\", \"double positive\"]\n",
    "\n",
    "    # インスタンス作成\n",
    "processor = WordProcessor(exclusive_words)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 15,
   "metadata": {},
   "outputs": [],
   "source": [
    "basedf = pd.read_csv('../ClinicalTrialCSV/JRCT20241215Cancer.csv', index_col=0)\n",
    "basedf = basedf.dropna(subset=['試験等のフェーズ'])\n",
    "# Target列を分割してTargetWord列を追加\n",
    "basedf['TargetWord'] = basedf['TargetEnglish'].apply(split_target_English)\n",
    "# 各行のTargetWord列に処理を適用\n",
    "# NaN や None の場合の処理を追加\n",
    "basedf['TargetWord'] = basedf['TargetWord'].apply(lambda x: processor(x) if isinstance(x, list) else x)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 5,
   "metadata": {},
   "outputs": [
    {
     "name": "stderr",
     "output_type": "stream",
     "text": [
      "/Users/satoc/miniforge3/envs/gradio/lib/python3.12/site-packages/huggingface_hub/file_download.py:1132: FutureWarning: `resume_download` is deprecated and will be removed in version 1.0.0. Downloads always resume when possible. If you want to force a new download, use `force_download=True`.\n",
      "  warnings.warn(\n"
     ]
    },
    {
     "data": {
      "application/vnd.jupyter.widget-view+json": {
       "model_id": "0a239a616af243eeb2a08511e21b3a49",
       "version_major": 2,
       "version_minor": 0
      },
      "text/plain": [
       "pytorch_model.bin:   0%|          | 0.00/438M [00:00<?, ?B/s]"
      ]
     },
     "metadata": {},
     "output_type": "display_data"
    },
    {
     "data": {
      "application/vnd.jupyter.widget-view+json": {
       "model_id": "705ebb2de82e45ffa2aecdf02ef97810",
       "version_major": 2,
       "version_minor": 0
      },
      "text/plain": [
       "tokenizer_config.json:   0%|          | 0.00/388 [00:00<?, ?B/s]"
      ]
     },
     "metadata": {},
     "output_type": "display_data"
    },
    {
     "data": {
      "application/vnd.jupyter.widget-view+json": {
       "model_id": "835c0af2167247298a44e3ba542b4e05",
       "version_major": 2,
       "version_minor": 0
      },
      "text/plain": [
       "vocab.txt:   0%|          | 0.00/226k [00:00<?, ?B/s]"
      ]
     },
     "metadata": {},
     "output_type": "display_data"
    },
    {
     "data": {
      "application/vnd.jupyter.widget-view+json": {
       "model_id": "c35026c9786a4c678d430fa06f2d24fe",
       "version_major": 2,
       "version_minor": 0
      },
      "text/plain": [
       "tokenizer.json:   0%|          | 0.00/461k [00:00<?, ?B/s]"
      ]
     },
     "metadata": {},
     "output_type": "display_data"
    },
    {
     "data": {
      "application/vnd.jupyter.widget-view+json": {
       "model_id": "5125278abc8a49a1b4d9f966fd09ee15",
       "version_major": 2,
       "version_minor": 0
      },
      "text/plain": [
       "special_tokens_map.json:   0%|          | 0.00/112 [00:00<?, ?B/s]"
      ]
     },
     "metadata": {},
     "output_type": "display_data"
    },
    {
     "data": {
      "application/vnd.jupyter.widget-view+json": {
       "model_id": "7143cbff16da475dac2cf4727a4939ce",
       "version_major": 2,
       "version_minor": 0
      },
      "text/plain": [
       "1_Pooling/config.json:   0%|          | 0.00/190 [00:00<?, ?B/s]"
      ]
     },
     "metadata": {},
     "output_type": "display_data"
    }
   ],
   "source": [
    "# モデルのロード\n",
    "model = SentenceTransformer('pritamdeka/S-PubMedBert-MS-MARCO')"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 11,
   "metadata": {},
   "outputs": [],
   "source": [
    "# クエリ\n",
    "#query = \"Triple negative breast cancer\"\n",
    "query = \"breast cancer\"\n",
    "#query = \"glioma\"\n",
    "threshold = 0.925\n",
    "# クエリをベクトル化\n",
    "query_vec = model.encode(query, convert_to_tensor=True)\n"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 19,
   "metadata": {},
   "outputs": [],
   "source": [
    "target_vecs_list = []\n",
    "for idx, target_words in enumerate(basedf['TargetWord']):\n",
    "    target_vecs = model.encode(target_words, convert_to_tensor=True).cpu()\n",
    "    target_vecs_list.append(target_vecs)\n",
    "\n",
    "# Seriesを作成してから代入（dtype=objectを明示）\n",
    "basedf['TargetVec'] = pd.Series(target_vecs_list, index=basedf.index, dtype=object)\n"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 20,
   "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>JRCT ID</th>\n",
       "      <th>NCT No</th>\n",
       "      <th>JapicCTI No</th>\n",
       "      <th>Title</th>\n",
       "      <th>TargetJ</th>\n",
       "      <th>Target</th>\n",
       "      <th>TargetEnglish</th>\n",
       "      <th>研究・治験の目的</th>\n",
       "      <th>試験等のフェーズ</th>\n",
       "      <th>試験の種類</th>\n",
       "      <th>...</th>\n",
       "      <th>Inclusion Criteria</th>\n",
       "      <th>Exclusion Criteria</th>\n",
       "      <th>Age Minimum</th>\n",
       "      <th>Age Maximum</th>\n",
       "      <th>Gender</th>\n",
       "      <th>Discontinuation Criteria</th>\n",
       "      <th>Keyword</th>\n",
       "      <th>Intervention(s)</th>\n",
       "      <th>TargetWord</th>\n",
       "      <th>TargetVec</th>\n",
       "    </tr>\n",
       "  </thead>\n",
       "  <tbody>\n",
       "    <tr>\n",
       "      <th>0</th>\n",
       "      <td>jRCT2031240552</td>\n",
       "      <td>NCT06599502</td>\n",
       "      <td>NaN</td>\n",
       "      <td>KRAS G12D 変異陽性がん患者を対象としてAZD0022 の単剤療法及び抗がん剤との併...</td>\n",
       "      <td>以下の進行固形がん患者\\n・非小細胞肺癌\\n・膵管腺癌\\n・結腸・直腸癌</td>\n",
       "      <td>以下の進行固形がん患者 ・非小細胞肺癌 ・膵管腺癌 ・結腸・直腸癌</td>\n",
       "      <td>Non-Small Cell Lung Cancer (NSCLC) Pancreatic ...</td>\n",
       "      <td>Treatment</td>\n",
       "      <td>1-2</td>\n",
       "      <td>NaN</td>\n",
       "      <td>...</td>\n",
       "      <td>1. Participant must be 18 years or more, or th...</td>\n",
       "      <td>1. Any significant laboratory finding or any s...</td>\n",
       "      <td>18age old over</td>\n",
       "      <td>No limit</td>\n",
       "      <td>Both</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>AZD0022</td>\n",
       "      <td>[Non-Small Cell Lung Cancer (Nsclc) Pancreatic...</td>\n",
       "      <td>[[tensor(-0.1260), tensor(-0.7142), tensor(-0....</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>3</th>\n",
       "      <td>jRCT2031240547</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>FGFR2b陽性切除不能進行・再発の胃または食道胃接合部腺癌を対象としたBemarituzu...</td>\n",
       "      <td>切除不能進行・再発の胃または食道胃接合部腺癌</td>\n",
       "      <td>切除不能進行・再発の胃または食道胃接合部腺癌</td>\n",
       "      <td>Advanced Gastric or Gastroesophageal Junction ...</td>\n",
       "      <td>フッ化ピリミジン系薬剤に不応または不耐となったFGFR2b陽性の切除不能進行・再発の胃腺癌又...</td>\n",
       "      <td>2</td>\n",
       "      <td>NaN</td>\n",
       "      <td>...</td>\n",
       "      <td>1) Histologically documented gastric or GEJ ad...</td>\n",
       "      <td>1) Prior treatment with taxanes 2) Prior treat...</td>\n",
       "      <td>18age old over</td>\n",
       "      <td>No limit</td>\n",
       "      <td>Both</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>- Bemarituzumab (15mg/kg, intravenous, Day1, 1...</td>\n",
       "      <td>[Advanced Gastric, Gastroesophageal Junction C...</td>\n",
       "      <td>[[tensor(-0.4843), tensor(-0.6084), tensor(-0....</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>5</th>\n",
       "      <td>jRCT2031240542</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>自家造血幹細胞移植非適応の初発多発性骨髄腫患者を対象としてベランタマブ マホドチンとレナリド...</td>\n",
       "      <td>多発性骨髄腫</td>\n",
       "      <td>多発性骨髄腫</td>\n",
       "      <td>Multiple Myeloma</td>\n",
       "      <td>自家造血幹細胞移植非適応の初発多発性骨髄腫を対象にGSK2857916、レナリドミド 及び ...</td>\n",
       "      <td>3</td>\n",
       "      <td>NaN</td>\n",
       "      <td>...</td>\n",
       "      <td>1. Is at least 18 or the legal age of consent ...</td>\n",
       "      <td>Participants are excluded from the study if an...</td>\n",
       "      <td>18age old over</td>\n",
       "      <td>No limit</td>\n",
       "      <td>Both</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>Arm A: Belantamab Mafodotin is intraveniously ...</td>\n",
       "      <td>[Multiple Myeloma]</td>\n",
       "      <td>[[tensor(-0.6175), tensor(-0.4283), tensor(-0....</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>7</th>\n",
       "      <td>jRCT2031240536</td>\n",
       "      <td>NCT06003231</td>\n",
       "      <td>NaN</td>\n",
       "      <td>既治療の局所進行切除不能又は転移性HER2発現固形癌を有する成人患者を対象としたDisita...</td>\n",
       "      <td>頭頸部扁平上皮癌、非小細胞肺癌、卵巣癌、子宮内膜癌</td>\n",
       "      <td>頭頸部扁平上皮癌、非小細胞肺癌、卵巣癌、子宮内膜癌</td>\n",
       "      <td>Head and neck squamous cell carcinoma/Non-smal...</td>\n",
       "      <td>既治療の局所進行切除不能又は転移性(LA/m)ヒト上皮増殖因子受容体2型(HER2)発現固形...</td>\n",
       "      <td>2</td>\n",
       "      <td>NaN</td>\n",
       "      <td>...</td>\n",
       "      <td>Cohort 1: Head and neck squamous cell carcinom...</td>\n",
       "      <td>- Prior treatment with a monomethyl auristatin...</td>\n",
       "      <td>18age old over</td>\n",
       "      <td>No limit</td>\n",
       "      <td>Both</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>Experimental Arm - Disitamab vedotin 1.5 mg/kg...</td>\n",
       "      <td>[Head, Neck Squamous Cell Carcinoma/Non-Small ...</td>\n",
       "      <td>[[tensor(-0.2330), tensor(-0.4411), tensor(-0....</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>9</th>\n",
       "      <td>jRCT2031240530</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>HER2遺伝子増幅あるいはHER2遺伝子変異を有するII-III期非小細胞肺癌に対するTra...</td>\n",
       "      <td>HER2遺伝子増幅あるいはHER2遺伝子変異を有するII/IIIA/IIIB(T3-4N2)...</td>\n",
       "      <td>HER2遺伝子増幅あるいはHER2遺伝子変異を有するII/IIIA/IIIB(T3-4N2)...</td>\n",
       "      <td>Stage II-III HER2-Amplified or HER2-Mutant Non...</td>\n",
       "      <td>HER2遺伝子増幅あるいはHER2遺伝子変異を有するII/IIIA/IIIB(T3-4N2)...</td>\n",
       "      <td>2</td>\n",
       "      <td>NaN</td>\n",
       "      <td>...</td>\n",
       "      <td>1. Signed informed consent form 2. Age &gt;= 18 y...</td>\n",
       "      <td>1. NSCLC that is clinically T4 by virtue of me...</td>\n",
       "      <td>18age old over</td>\n",
       "      <td>No limit</td>\n",
       "      <td>Both</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>The study drug will be administered as an IV i...</td>\n",
       "      <td>[Stage Ii-Iii Her2-Amplified, Her2-Mutant Non-...</td>\n",
       "      <td>[[tensor(-0.2839), tensor(-0.7334), tensor(-0....</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>...</th>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>729</th>\n",
       "      <td>jRCT2031200057</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>治療抵抗性乳がんを対象としたTDM-812の腫瘍内投与法の安全性評価を目的とした第I相試験</td>\n",
       "      <td>乳がん</td>\n",
       "      <td>乳がん</td>\n",
       "      <td>Breast cancer</td>\n",
       "      <td>TDM-812の腫瘍内投与の安全性および忍容性を評価し、今後の臨床評価に用いる腫瘍内投与にお...</td>\n",
       "      <td>1</td>\n",
       "      <td>NaN</td>\n",
       "      <td>...</td>\n",
       "      <td>&lt;Disease Characteristics&gt; 1)Patients with inop...</td>\n",
       "      <td>1)Patients correspond to either of the followi...</td>\n",
       "      <td>20age old over</td>\n",
       "      <td>No limit</td>\n",
       "      <td>Female</td>\n",
       "      <td>NaN</td>\n",
       "      <td>Breast cancer</td>\n",
       "      <td>For the target tumor, the dose settled at each...</td>\n",
       "      <td>[Breast Cancer]</td>\n",
       "      <td>[[tensor(-0.0986), tensor(-0.6295), tensor(-0....</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>739</th>\n",
       "      <td>jRCT2073200004</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>切除不能膵癌を対象とした腫瘍溶解性ウイルスの腫瘍内局所投与の臨床第Ⅰ/Ⅱ相試験</td>\n",
       "      <td>膵癌</td>\n",
       "      <td>膵癌</td>\n",
       "      <td>Pancreatic cancer</td>\n",
       "      <td>切除不能膵癌患者に対するSurv.m-CRA-1の腫瘍内投与の安全性及び有効性を検討する。</td>\n",
       "      <td>1-2</td>\n",
       "      <td>NaN</td>\n",
       "      <td>...</td>\n",
       "      <td>The patients with pancreatic tumors must meet ...</td>\n",
       "      <td>1) Patients with the following complications: ...</td>\n",
       "      <td>20age old over</td>\n",
       "      <td>No limit</td>\n",
       "      <td>Both</td>\n",
       "      <td>NaN</td>\n",
       "      <td>Pancreatic ductal carcinoma, unresectable</td>\n",
       "      <td>Intratumoral administration of oncolytic virus</td>\n",
       "      <td>[Pancreatic Cancer]</td>\n",
       "      <td>[[tensor(-0.0844), tensor(-0.9954), tensor(-0....</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>760</th>\n",
       "      <td>jRCT2033190086</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>悪性黒色腫患者を対象としたインターロイキン12発現型遺伝子組換え単純ヘルペスウイルス１型の第...</td>\n",
       "      <td>悪性黒色腫</td>\n",
       "      <td>悪性黒色腫</td>\n",
       "      <td>malignant melanoma</td>\n",
       "      <td>第I相パート:進行期の悪性黒色腫患者を対象として、ヒトIL-12発現型遺伝子組換え単純ヘルペ...</td>\n",
       "      <td>1-2</td>\n",
       "      <td>NaN</td>\n",
       "      <td>...</td>\n",
       "      <td>(Phase 1) 1) Histologically confirmed malignan...</td>\n",
       "      <td>(Phase 1 and 2) 1) Patients who have brain and...</td>\n",
       "      <td>20age old over</td>\n",
       "      <td>No limit</td>\n",
       "      <td>Both</td>\n",
       "      <td>NaN</td>\n",
       "      <td>malignamt melanoma, advanced stage</td>\n",
       "      <td>Test drug (T-hIL12) will be administered into ...</td>\n",
       "      <td>[Malignant Melanoma]</td>\n",
       "      <td>[[tensor(-0.5701), tensor(-0.7062), tensor(-0....</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>762</th>\n",
       "      <td>jRCT2031190072</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>切除不能な進行肝細胞癌患者を対象としたCYT001の第I相臨床試験</td>\n",
       "      <td>肝細胞癌</td>\n",
       "      <td>肝細胞癌</td>\n",
       "      <td>Hepatocellular carcinoma</td>\n",
       "      <td>切除不能な進行肝細胞癌患者を対象としたCYT001の第I相臨床試験</td>\n",
       "      <td>1</td>\n",
       "      <td>NaN</td>\n",
       "      <td>...</td>\n",
       "      <td>1) Histologically or cytologically confirmed h...</td>\n",
       "      <td>1) CNS metastases that are symptomatic or requ...</td>\n",
       "      <td>20age 0month 0week old over</td>\n",
       "      <td>No limit</td>\n",
       "      <td>Both</td>\n",
       "      <td>NaN</td>\n",
       "      <td>Cancer peptide vaccine</td>\n",
       "      <td>Reconstitute two kinds of cancer peptide vacci...</td>\n",
       "      <td>[Hepatocellular Carcinoma]</td>\n",
       "      <td>[[tensor(-0.1773), tensor(-0.3860), tensor(-0....</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>772</th>\n",
       "      <td>jRCT2051190009</td>\n",
       "      <td>NCT03818893</td>\n",
       "      <td>JapicCTI-194618</td>\n",
       "      <td>進行性悪性黒色腫患者を対象としたGEN0101の皮内投与と、ペムブロリズマブ（抗PD-1抗体...</td>\n",
       "      <td>進行性悪性黒色腫</td>\n",
       "      <td>進行性悪性黒色腫</td>\n",
       "      <td>advanced melanoma</td>\n",
       "      <td>抗PD-1抗体治療でSDが確定した、またはPDとなった患者を対象とし、非盲検下でGEN010...</td>\n",
       "      <td>1-2</td>\n",
       "      <td>NaN</td>\n",
       "      <td>...</td>\n",
       "      <td>Inclusion criteria Patients meeting all of the...</td>\n",
       "      <td>Exclusion criteria Patients meeting any of the...</td>\n",
       "      <td>20age old over</td>\n",
       "      <td>86age old not</td>\n",
       "      <td>Both</td>\n",
       "      <td>NaN</td>\n",
       "      <td>advanced melanoma</td>\n",
       "      <td>combination therapy of intracutaneous GEN0101 ...</td>\n",
       "      <td>[Advanced Melanoma]</td>\n",
       "      <td>[[tensor(-0.5761), tensor(-0.6570), tensor(-0....</td>\n",
       "    </tr>\n",
       "  </tbody>\n",
       "</table>\n",
       "<p>427 rows × 41 columns</p>\n",
       "</div>"
      ],
      "text/plain": [
       "            JRCT ID       NCT No      JapicCTI No  \\\n",
       "0    jRCT2031240552  NCT06599502              NaN   \n",
       "3    jRCT2031240547          NaN              NaN   \n",
       "5    jRCT2031240542          NaN              NaN   \n",
       "7    jRCT2031240536  NCT06003231              NaN   \n",
       "9    jRCT2031240530          NaN              NaN   \n",
       "..              ...          ...              ...   \n",
       "729  jRCT2031200057          NaN              NaN   \n",
       "739  jRCT2073200004          NaN              NaN   \n",
       "760  jRCT2033190086          NaN              NaN   \n",
       "762  jRCT2031190072          NaN              NaN   \n",
       "772  jRCT2051190009  NCT03818893  JapicCTI-194618   \n",
       "\n",
       "                                                 Title  \\\n",
       "0    KRAS G12D 変異陽性がん患者を対象としてAZD0022 の単剤療法及び抗がん剤との併...   \n",
       "3    FGFR2b陽性切除不能進行・再発の胃または食道胃接合部腺癌を対象としたBemarituzu...   \n",
       "5    自家造血幹細胞移植非適応の初発多発性骨髄腫患者を対象としてベランタマブ マホドチンとレナリド...   \n",
       "7    既治療の局所進行切除不能又は転移性HER2発現固形癌を有する成人患者を対象としたDisita...   \n",
       "9    HER2遺伝子増幅あるいはHER2遺伝子変異を有するII-III期非小細胞肺癌に対するTra...   \n",
       "..                                                 ...   \n",
       "729      治療抵抗性乳がんを対象としたTDM-812の腫瘍内投与法の安全性評価を目的とした第I相試験   \n",
       "739            切除不能膵癌を対象とした腫瘍溶解性ウイルスの腫瘍内局所投与の臨床第Ⅰ/Ⅱ相試験   \n",
       "760  悪性黒色腫患者を対象としたインターロイキン12発現型遺伝子組換え単純ヘルペスウイルス１型の第...   \n",
       "762                  切除不能な進行肝細胞癌患者を対象としたCYT001の第I相臨床試験   \n",
       "772  進行性悪性黒色腫患者を対象としたGEN0101の皮内投与と、ペムブロリズマブ（抗PD-1抗体...   \n",
       "\n",
       "                                               TargetJ  \\\n",
       "0                 以下の進行固形がん患者\\n・非小細胞肺癌\\n・膵管腺癌\\n・結腸・直腸癌   \n",
       "3                               切除不能進行・再発の胃または食道胃接合部腺癌   \n",
       "5                                               多発性骨髄腫   \n",
       "7                            頭頸部扁平上皮癌、非小細胞肺癌、卵巣癌、子宮内膜癌   \n",
       "9    HER2遺伝子増幅あるいはHER2遺伝子変異を有するII/IIIA/IIIB(T3-4N2)...   \n",
       "..                                                 ...   \n",
       "729                                                乳がん   \n",
       "739                                                 膵癌   \n",
       "760                                              悪性黒色腫   \n",
       "762                                               肝細胞癌   \n",
       "772                                           進行性悪性黒色腫   \n",
       "\n",
       "                                                Target  \\\n",
       "0                    以下の進行固形がん患者 ・非小細胞肺癌 ・膵管腺癌 ・結腸・直腸癌   \n",
       "3                               切除不能進行・再発の胃または食道胃接合部腺癌   \n",
       "5                                               多発性骨髄腫   \n",
       "7                            頭頸部扁平上皮癌、非小細胞肺癌、卵巣癌、子宮内膜癌   \n",
       "9    HER2遺伝子増幅あるいはHER2遺伝子変異を有するII/IIIA/IIIB(T3-4N2)...   \n",
       "..                                                 ...   \n",
       "729                                                乳がん   \n",
       "739                                                 膵癌   \n",
       "760                                              悪性黒色腫   \n",
       "762                                               肝細胞癌   \n",
       "772                                           進行性悪性黒色腫   \n",
       "\n",
       "                                         TargetEnglish  \\\n",
       "0    Non-Small Cell Lung Cancer (NSCLC) Pancreatic ...   \n",
       "3    Advanced Gastric or Gastroesophageal Junction ...   \n",
       "5                                     Multiple Myeloma   \n",
       "7    Head and neck squamous cell carcinoma/Non-smal...   \n",
       "9    Stage II-III HER2-Amplified or HER2-Mutant Non...   \n",
       "..                                                 ...   \n",
       "729                                      Breast cancer   \n",
       "739                                  Pancreatic cancer   \n",
       "760                                 malignant melanoma   \n",
       "762                           Hepatocellular carcinoma   \n",
       "772                                  advanced melanoma   \n",
       "\n",
       "                                              研究・治験の目的 試験等のフェーズ 試験の種類  ...  \\\n",
       "0                                            Treatment      1-2   NaN  ...   \n",
       "3    フッ化ピリミジン系薬剤に不応または不耐となったFGFR2b陽性の切除不能進行・再発の胃腺癌又...        2   NaN  ...   \n",
       "5    自家造血幹細胞移植非適応の初発多発性骨髄腫を対象にGSK2857916、レナリドミド 及び ...        3   NaN  ...   \n",
       "7    既治療の局所進行切除不能又は転移性(LA/m)ヒト上皮増殖因子受容体2型(HER2)発現固形...        2   NaN  ...   \n",
       "9    HER2遺伝子増幅あるいはHER2遺伝子変異を有するII/IIIA/IIIB(T3-4N2)...        2   NaN  ...   \n",
       "..                                                 ...      ...   ...  ...   \n",
       "729  TDM-812の腫瘍内投与の安全性および忍容性を評価し、今後の臨床評価に用いる腫瘍内投与にお...        1   NaN  ...   \n",
       "739      切除不能膵癌患者に対するSurv.m-CRA-1の腫瘍内投与の安全性及び有効性を検討する。      1-2   NaN  ...   \n",
       "760  第I相パート:進行期の悪性黒色腫患者を対象として、ヒトIL-12発現型遺伝子組換え単純ヘルペ...      1-2   NaN  ...   \n",
       "762                  切除不能な進行肝細胞癌患者を対象としたCYT001の第I相臨床試験        1   NaN  ...   \n",
       "772  抗PD-1抗体治療でSDが確定した、またはPDとなった患者を対象とし、非盲検下でGEN010...      1-2   NaN  ...   \n",
       "\n",
       "                                    Inclusion Criteria  \\\n",
       "0    1. Participant must be 18 years or more, or th...   \n",
       "3    1) Histologically documented gastric or GEJ ad...   \n",
       "5    1. Is at least 18 or the legal age of consent ...   \n",
       "7    Cohort 1: Head and neck squamous cell carcinom...   \n",
       "9    1. Signed informed consent form 2. Age >= 18 y...   \n",
       "..                                                 ...   \n",
       "729  <Disease Characteristics> 1)Patients with inop...   \n",
       "739  The patients with pancreatic tumors must meet ...   \n",
       "760  (Phase 1) 1) Histologically confirmed malignan...   \n",
       "762  1) Histologically or cytologically confirmed h...   \n",
       "772  Inclusion criteria Patients meeting all of the...   \n",
       "\n",
       "                                    Exclusion Criteria  \\\n",
       "0    1. Any significant laboratory finding or any s...   \n",
       "3    1) Prior treatment with taxanes 2) Prior treat...   \n",
       "5    Participants are excluded from the study if an...   \n",
       "7    - Prior treatment with a monomethyl auristatin...   \n",
       "9    1. NSCLC that is clinically T4 by virtue of me...   \n",
       "..                                                 ...   \n",
       "729  1)Patients correspond to either of the followi...   \n",
       "739  1) Patients with the following complications: ...   \n",
       "760  (Phase 1 and 2) 1) Patients who have brain and...   \n",
       "762  1) CNS metastases that are symptomatic or requ...   \n",
       "772  Exclusion criteria Patients meeting any of the...   \n",
       "\n",
       "                     Age Minimum    Age Maximum  Gender  \\\n",
       "0                 18age old over       No limit    Both   \n",
       "3                 18age old over       No limit    Both   \n",
       "5                 18age old over       No limit    Both   \n",
       "7                 18age old over       No limit    Both   \n",
       "9                 18age old over       No limit    Both   \n",
       "..                           ...            ...     ...   \n",
       "729               20age old over       No limit  Female   \n",
       "739               20age old over       No limit    Both   \n",
       "760               20age old over       No limit    Both   \n",
       "762  20age 0month 0week old over       No limit    Both   \n",
       "772               20age old over  86age old not    Both   \n",
       "\n",
       "    Discontinuation Criteria                                    Keyword  \\\n",
       "0                        NaN                                        NaN   \n",
       "3                        NaN                                        NaN   \n",
       "5                        NaN                                        NaN   \n",
       "7                        NaN                                        NaN   \n",
       "9                        NaN                                        NaN   \n",
       "..                       ...                                        ...   \n",
       "729                      NaN                              Breast cancer   \n",
       "739                      NaN  Pancreatic ductal carcinoma, unresectable   \n",
       "760                      NaN         malignamt melanoma, advanced stage   \n",
       "762                      NaN                     Cancer peptide vaccine   \n",
       "772                      NaN                          advanced melanoma   \n",
       "\n",
       "                                       Intervention(s)  \\\n",
       "0                                              AZD0022   \n",
       "3    - Bemarituzumab (15mg/kg, intravenous, Day1, 1...   \n",
       "5    Arm A: Belantamab Mafodotin is intraveniously ...   \n",
       "7    Experimental Arm - Disitamab vedotin 1.5 mg/kg...   \n",
       "9    The study drug will be administered as an IV i...   \n",
       "..                                                 ...   \n",
       "729  For the target tumor, the dose settled at each...   \n",
       "739     Intratumoral administration of oncolytic virus   \n",
       "760  Test drug (T-hIL12) will be administered into ...   \n",
       "762  Reconstitute two kinds of cancer peptide vacci...   \n",
       "772  combination therapy of intracutaneous GEN0101 ...   \n",
       "\n",
       "                                            TargetWord  \\\n",
       "0    [Non-Small Cell Lung Cancer (Nsclc) Pancreatic...   \n",
       "3    [Advanced Gastric, Gastroesophageal Junction C...   \n",
       "5                                   [Multiple Myeloma]   \n",
       "7    [Head, Neck Squamous Cell Carcinoma/Non-Small ...   \n",
       "9    [Stage Ii-Iii Her2-Amplified, Her2-Mutant Non-...   \n",
       "..                                                 ...   \n",
       "729                                    [Breast Cancer]   \n",
       "739                                [Pancreatic Cancer]   \n",
       "760                               [Malignant Melanoma]   \n",
       "762                         [Hepatocellular Carcinoma]   \n",
       "772                                [Advanced Melanoma]   \n",
       "\n",
       "                                             TargetVec  \n",
       "0    [[tensor(-0.1260), tensor(-0.7142), tensor(-0....  \n",
       "3    [[tensor(-0.4843), tensor(-0.6084), tensor(-0....  \n",
       "5    [[tensor(-0.6175), tensor(-0.4283), tensor(-0....  \n",
       "7    [[tensor(-0.2330), tensor(-0.4411), tensor(-0....  \n",
       "9    [[tensor(-0.2839), tensor(-0.7334), tensor(-0....  \n",
       "..                                                 ...  \n",
       "729  [[tensor(-0.0986), tensor(-0.6295), tensor(-0....  \n",
       "739  [[tensor(-0.0844), tensor(-0.9954), tensor(-0....  \n",
       "760  [[tensor(-0.5701), tensor(-0.7062), tensor(-0....  \n",
       "762  [[tensor(-0.1773), tensor(-0.3860), tensor(-0....  \n",
       "772  [[tensor(-0.5761), tensor(-0.6570), tensor(-0....  \n",
       "\n",
       "[427 rows x 41 columns]"
      ]
     },
     "execution_count": 20,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "basedf"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 23,
   "metadata": {},
   "outputs": [],
   "source": [
    "from sentence_transformers import util\n",
    "import torch\n",
    "\n",
    "def get_matched_df(basedf, query, model, threshold=0.5):\n",
    "    # queryをベクトル化（テンソル化）しCPUへ移動\n",
    "    query_vec = model.encode(query, convert_to_tensor=True).cpu()\n",
    "\n",
    "    matched_indices = []\n",
    "    for idx, target_vecs in enumerate(basedf['TargetVec']):\n",
    "        # target_vecsは既にCPU上にあるか確認\n",
    "        # もしGPU上にある場合は .cpu() をつける\n",
    "        if target_vecs.is_cuda:\n",
    "            target_vecs = target_vecs.cpu()\n",
    "\n",
    "        # コサイン類似度を計算\n",
    "        cosine_scores = util.cos_sim(query_vec, target_vecs).squeeze()\n",
    "\n",
    "        # thresholdを超えるスコアが1つでもあればマッチと判断\n",
    "        if (cosine_scores >= threshold).any():\n",
    "            matched_indices.append(idx)\n",
    "\n",
    "    # 条件を満たした行を抽出\n",
    "    matched_df = basedf.iloc[matched_indices]\n",
    "    return matched_df\n"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 24,
   "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>JRCT ID</th>\n",
       "      <th>NCT No</th>\n",
       "      <th>JapicCTI No</th>\n",
       "      <th>Title</th>\n",
       "      <th>TargetJ</th>\n",
       "      <th>Target</th>\n",
       "      <th>TargetEnglish</th>\n",
       "      <th>研究・治験の目的</th>\n",
       "      <th>試験等のフェーズ</th>\n",
       "      <th>試験の種類</th>\n",
       "      <th>...</th>\n",
       "      <th>Inclusion Criteria</th>\n",
       "      <th>Exclusion Criteria</th>\n",
       "      <th>Age Minimum</th>\n",
       "      <th>Age Maximum</th>\n",
       "      <th>Gender</th>\n",
       "      <th>Discontinuation Criteria</th>\n",
       "      <th>Keyword</th>\n",
       "      <th>Intervention(s)</th>\n",
       "      <th>TargetWord</th>\n",
       "      <th>TargetVec</th>\n",
       "    </tr>\n",
       "  </thead>\n",
       "  <tbody>\n",
       "    <tr>\n",
       "      <th>31</th>\n",
       "      <td>jRCT2061240081</td>\n",
       "      <td>NCT06393374</td>\n",
       "      <td>NaN</td>\n",
       "      <td>術前薬物療法後の手術時に病理学的完全奏効を達成していないトリプルネガティブ乳癌患者を対象に、...</td>\n",
       "      <td>トリプルネガティブ乳癌</td>\n",
       "      <td>トリプルネガティブ乳癌</td>\n",
       "      <td>Triple negative breast cancer</td>\n",
       "      <td>術前薬物療法後の手術時に病理学的完全奏効を達成していないトリプルネガティブ乳癌患者を対象に、...</td>\n",
       "      <td>3</td>\n",
       "      <td>NaN</td>\n",
       "      <td>...</td>\n",
       "      <td>-Has centrally confirmed TNBC, as defined by t...</td>\n",
       "      <td>-Has a known germline breast cancer gene (BRCA...</td>\n",
       "      <td>18age old over</td>\n",
       "      <td>No limit</td>\n",
       "      <td>Both</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>-Arm 1: MK-2870 4mg/kg intravenous (IV) every ...</td>\n",
       "      <td>[Triple Negative, Breast Cancer]</td>\n",
       "      <td>[[tensor(-0.2188), tensor(-0.4577), tensor(-0....</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>32</th>\n",
       "      <td>jRCT2031240476</td>\n",
       "      <td>NCT06312176</td>\n",
       "      <td>NaN</td>\n",
       "      <td>ホルモン受容体陽性・ヒト上皮成長因子受容体2陰性（HR+/HER2-）の切除不能な局所進行又...</td>\n",
       "      <td>HR+/HER2-の切除不能局所進行又は転移性乳癌の患者</td>\n",
       "      <td>HR+/HER2-の切除不能局所進行又は転移性乳癌の患者</td>\n",
       "      <td>HR+/HER2 unresectable locally advanced or meta...</td>\n",
       "      <td>HR+/HER2-の切除不能な局所進行又は転移性乳癌患者を対象にMK-2870の単剤及びペム...</td>\n",
       "      <td>3</td>\n",
       "      <td>NaN</td>\n",
       "      <td>...</td>\n",
       "      <td>-Has unresectable locally advanced or metastat...</td>\n",
       "      <td>-Has breast cancer amenable to treatment with ...</td>\n",
       "      <td>18age old over</td>\n",
       "      <td>No limit</td>\n",
       "      <td>Both</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>Arm A:MK-2870 Participants receive 4 mg/kg of ...</td>\n",
       "      <td>[Hr+/Her2 Unresectable Locally Advanced, Metas...</td>\n",
       "      <td>[[tensor(0.0489), tensor(-0.7584), tensor(-0.4...</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>157</th>\n",
       "      <td>jRCT2052240059</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>遺伝子HSD17B4高メチル化を有するHER2陽性ER陰性乳癌における非手術療法の有用性を評...</td>\n",
       "      <td>乳がん</td>\n",
       "      <td>乳がん</td>\n",
       "      <td>Breast cancer</td>\n",
       "      <td>HSD17B4高メチル化(HSD17B4 hypermethylation:HH)を有するH...</td>\n",
       "      <td>2</td>\n",
       "      <td>NaN</td>\n",
       "      <td>...</td>\n",
       "      <td>1. Histologically confirmed invasive breast ca...</td>\n",
       "      <td>1. History of other malignancy within the last...</td>\n",
       "      <td>20age old over</td>\n",
       "      <td>No limit</td>\n",
       "      <td>Female</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>Omitting breast surgery after preoperative che...</td>\n",
       "      <td>[Breast Cancer]</td>\n",
       "      <td>[[tensor(-0.0986), tensor(-0.6295), tensor(-0....</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>170</th>\n",
       "      <td>jRCT2031240096</td>\n",
       "      <td>NCT06380751</td>\n",
       "      <td>NaN</td>\n",
       "      <td>本治験の主要目的は、BRCA1、BRCA2 又は PALB2 変異を有し、HR 陽性、HER...</td>\n",
       "      <td>進行乳癌</td>\n",
       "      <td>進行乳癌</td>\n",
       "      <td>Advanced Breast Cancer</td>\n",
       "      <td>Treatment</td>\n",
       "      <td>3</td>\n",
       "      <td>NaN</td>\n",
       "      <td>...</td>\n",
       "      <td>- Adult females, pre/peri-menopausal and/or po...</td>\n",
       "      <td>- Participants with history of MDS/AML or with...</td>\n",
       "      <td>18age old over</td>\n",
       "      <td>No limit</td>\n",
       "      <td>Both</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>Experimental: Arm 1: saruparib (AZD5305) plus ...</td>\n",
       "      <td>[Advanced Breast Cancer]</td>\n",
       "      <td>[[tensor(-0.2116), tensor(-0.5020), tensor(-0....</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>209</th>\n",
       "      <td>jRCT2031230750</td>\n",
       "      <td>NCT06188559</td>\n",
       "      <td>NaN</td>\n",
       "      <td>治療歴のある HER2 陽性又は HER2 低発現の切除不能又は転移性乳癌の被験者を対象に，...</td>\n",
       "      <td>HER2 陽性又は HER2 低発現の切除不能又は転移性乳癌</td>\n",
       "      <td>HER2 陽性又は HER2 低発現の切除不能又は転移性乳癌</td>\n",
       "      <td>HER2-positive or HER2-low unresectable or meta...</td>\n",
       "      <td>治療</td>\n",
       "      <td>2</td>\n",
       "      <td>NaN</td>\n",
       "      <td>...</td>\n",
       "      <td>(1) Male or female, aged &gt;=18 years at the tim...</td>\n",
       "      <td>(1) Presence of brain or subdural metastases, ...</td>\n",
       "      <td>18age old over</td>\n",
       "      <td>No limit</td>\n",
       "      <td>Both</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>Generic Name:NA Study Treatment in dose optimi...</td>\n",
       "      <td>[Her2-Positive, Her2-Low Unresectable, Metasta...</td>\n",
       "      <td>[[tensor(-0.0889), tensor(-0.8979), tensor(-0....</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>215</th>\n",
       "      <td>jRCT2031230723</td>\n",
       "      <td>NCT06112379</td>\n",
       "      <td>NaN</td>\n",
       "      <td>未治療のトリプルネガティブ又はホルモン受容体低発現／HER2陰性乳癌の成人患者を対象として、...</td>\n",
       "      <td>乳癌</td>\n",
       "      <td>乳癌</td>\n",
       "      <td>Breast Cancer</td>\n",
       "      <td>未治療のトリプルネガティブ又はホルモン受容体低発現/HER2陰性乳癌の成人患者を対象として、...</td>\n",
       "      <td>3</td>\n",
       "      <td>NaN</td>\n",
       "      <td>...</td>\n",
       "      <td>1. Participant must be &gt;= 18 years at the time...</td>\n",
       "      <td>1. As judged by the investigator, any evidence...</td>\n",
       "      <td>18age old over</td>\n",
       "      <td>No limit</td>\n",
       "      <td>Both</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>- Experimental arm: Dato-DXd plus durvalumab n...</td>\n",
       "      <td>[Breast Cancer]</td>\n",
       "      <td>[[tensor(-0.0986), tensor(-0.6295), tensor(-0....</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>224</th>\n",
       "      <td>jRCT2051230196</td>\n",
       "      <td>NCT06105632</td>\n",
       "      <td>NaN</td>\n",
       "      <td>CDK4/6 阻害薬による前治療で進行したホルモン受容体陽性HER2 陰性の進行または転移乳...</td>\n",
       "      <td>ホルモン受容体（HR）陽性ヒト上皮増殖因子受容体2（HER2）陰性の進行または転移乳癌</td>\n",
       "      <td>ホルモン受容体(HR)陽性ヒト上皮増殖因子受容体2(HER2)陰性の進行または転移乳癌</td>\n",
       "      <td>HR-positive, HER2-negative Advanced or Metasta...</td>\n",
       "      <td>CDK4/6阻害薬による前治療で進行したHR陽性HER2陰性の進行または転移乳癌患者を対象に...</td>\n",
       "      <td>3</td>\n",
       "      <td>NaN</td>\n",
       "      <td>...</td>\n",
       "      <td>* Histological confirmation of breast cancer w...</td>\n",
       "      <td>* Any medical or psychiatric condition that ma...</td>\n",
       "      <td>18age old over</td>\n",
       "      <td>No limit</td>\n",
       "      <td>Both</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>Arm A (Experimental Arm) PF-07220060 plus Fulv...</td>\n",
       "      <td>[Hr-Positive, Her2-Negative Advanced, Metastat...</td>\n",
       "      <td>[[tensor(-0.0832), tensor(-0.7572), tensor(-0....</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>234</th>\n",
       "      <td>jRCT2031230677</td>\n",
       "      <td>NCT06206837</td>\n",
       "      <td>NaN</td>\n",
       "      <td>18歳以上のER陽性／HER2陰性の進行または転移乳癌患者を対象として，経口タンパク質分解誘...</td>\n",
       "      <td>エストロゲン受容体陽性／ヒト上皮増殖因子受容体 2 陰性の進行または転移乳癌</td>\n",
       "      <td>エストロゲン受容体陽性/ヒト上皮増殖因子受容体 2 陰性の進行または転移乳癌</td>\n",
       "      <td>ER+/HER2- advanced or metastatic breast cancer</td>\n",
       "      <td>進行乳癌患者を対象として,vepdegestrantとPF-07220060を併用投与したと...</td>\n",
       "      <td>1-2</td>\n",
       "      <td>NaN</td>\n",
       "      <td>...</td>\n",
       "      <td>Inclusion Criteria: * Histological or cytologi...</td>\n",
       "      <td>Exclusion Criteria: * visceral crisis at risk ...</td>\n",
       "      <td>18age old over</td>\n",
       "      <td>No limit</td>\n",
       "      <td>Both</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>Drug: vepdegestrant Daily oral dosages of vepd...</td>\n",
       "      <td>[Er+/Her2- Advanced, Metastatic Breast Cancer]</td>\n",
       "      <td>[[tensor(-0.1549), tensor(-0.3944), tensor(-0....</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>240</th>\n",
       "      <td>jRCT2061230102</td>\n",
       "      <td>NCT06103864</td>\n",
       "      <td>NaN</td>\n",
       "      <td>Programmed death-ligand（PD-L1）陽性の局所再発手術不能又は転移性...</td>\n",
       "      <td>乳癌</td>\n",
       "      <td>乳癌</td>\n",
       "      <td>Breast Cancer</td>\n",
       "      <td>PD-L1陽性の局所再発手術不能または転移性TNBC患者を対象に、デュルバルマブ併用または非...</td>\n",
       "      <td>3</td>\n",
       "      <td>NaN</td>\n",
       "      <td>...</td>\n",
       "      <td>Histologically or cytologically documented loc...</td>\n",
       "      <td>As judged by investigator, severe or uncontrol...</td>\n",
       "      <td>18age old over</td>\n",
       "      <td>No limit</td>\n",
       "      <td>Both</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>Arm 1: Dato-DXd + durvalumab Arm 2: Investigat...</td>\n",
       "      <td>[Breast Cancer]</td>\n",
       "      <td>[[tensor(-0.0986), tensor(-0.6295), tensor(-0....</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>307</th>\n",
       "      <td>jRCT2061230074</td>\n",
       "      <td>NCT05952557</td>\n",
       "      <td>NaN</td>\n",
       "      <td>根治的局所治療（化学療法の併用または非併用）を受けて疾患の兆候のない、再発リスクが中間～高リ...</td>\n",
       "      <td>乳がん、早期乳がん</td>\n",
       "      <td>乳がん、早期乳がん</td>\n",
       "      <td>Breast Cancer, Early Breast Cancer</td>\n",
       "      <td>Treatment</td>\n",
       "      <td>3</td>\n",
       "      <td>NaN</td>\n",
       "      <td>...</td>\n",
       "      <td>- Women and Men; 18 years or more at the time ...</td>\n",
       "      <td>- Inoperable locally advanced or metastatic br...</td>\n",
       "      <td>18age old over</td>\n",
       "      <td>No limit</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>arm A: continue with SoC ET as directed by inv...</td>\n",
       "      <td>[Breast Cancer, Early Breast Cancer]</td>\n",
       "      <td>[[tensor(-0.0986), tensor(-0.6295), tensor(-0....</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>342</th>\n",
       "      <td>jRCT2051230094</td>\n",
       "      <td>NCT05794906</td>\n",
       "      <td>NaN</td>\n",
       "      <td>前立腺癌の高リスク生化学的再発（BCR）患者を対象としたアンドロゲン遮断療法（ADT）を併用...</td>\n",
       "      <td>前立腺癌</td>\n",
       "      <td>前立腺癌</td>\n",
       "      <td>prostate cancer</td>\n",
       "      <td>BAY 1841788(darolutamide)とADTの24ヵ月間併用投与によって、プラ...</td>\n",
       "      <td>3</td>\n",
       "      <td>NaN</td>\n",
       "      <td>...</td>\n",
       "      <td>- Histologically or cytologically confirmed ad...</td>\n",
       "      <td>- Pathological finding consistent with small c...</td>\n",
       "      <td>18age old over</td>\n",
       "      <td>No limit</td>\n",
       "      <td>Male</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>Drug: Darolutamide (BAY1841788, Nubeqa) Coated...</td>\n",
       "      <td>[Prostate Cancer]</td>\n",
       "      <td>[[tensor(0.0202), tensor(-0.7650), tensor(-0.5...</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>351</th>\n",
       "      <td>jRCT2031230279</td>\n",
       "      <td>NCT05862285</td>\n",
       "      <td>NaN</td>\n",
       "      <td>GENENTECH社及び／又はF.HOFFMANN-LA ROCHE LTDが依頼した試験に...</td>\n",
       "      <td>癌</td>\n",
       "      <td>癌</td>\n",
       "      <td>Cancer</td>\n",
       "      <td>本継続投与試験の目的は,親治験から移行する時点でまだ治験治療を受けており,その地域でその治療...</td>\n",
       "      <td>3</td>\n",
       "      <td>NaN</td>\n",
       "      <td>...</td>\n",
       "      <td>- Eligible for continuing Roche IMP-based ther...</td>\n",
       "      <td>- Meet any of the study treatment discontinuat...</td>\n",
       "      <td>18age old over</td>\n",
       "      <td>No limit</td>\n",
       "      <td>Both</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>Ipatasertib: Ipatasertib will be administered ...</td>\n",
       "      <td>[Cancer]</td>\n",
       "      <td>[[tensor(-0.2145), tensor(-0.7045), tensor(-0....</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>385</th>\n",
       "      <td>jRCT2031230121</td>\n",
       "      <td>NCT04557449</td>\n",
       "      <td>NaN</td>\n",
       "      <td>進行固形がん患者を対象として，PF-07220060 の単剤投与および併用投与の安全性，忍容...</td>\n",
       "      <td>進行または転移性の乳癌，前立腺癌，およびその他の固形癌</td>\n",
       "      <td>進行または転移性の乳癌,前立腺癌,およびその他の固形癌</td>\n",
       "      <td>advanced or metastatic BC, prostate cancer or ...</td>\n",
       "      <td>パート1(用量漸増,食事の影響,DDIの評価パート) 1A: PF-07220060のMTD...</td>\n",
       "      <td>1</td>\n",
       "      <td>NaN</td>\n",
       "      <td>...</td>\n",
       "      <td>Disease * Part 1A/1D/1E: HR+HER2-BC, HR+HER2+B...</td>\n",
       "      <td>* Part 1D: participants who have had a gastrec...</td>\n",
       "      <td>18age old over</td>\n",
       "      <td>No limit</td>\n",
       "      <td>Both</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>* Part 1A/1D/1E: PF-07220060 (CDK4 inhibitor) ...</td>\n",
       "      <td>[Advanced, Metastatic Bc, Prostate Cancer, Oth...</td>\n",
       "      <td>[[tensor(-0.4654), tensor(-0.4115), tensor(-0....</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>388</th>\n",
       "      <td>jRCT2031230109</td>\n",
       "      <td>NCT05514054</td>\n",
       "      <td>NaN</td>\n",
       "      <td>EMBER-4：2～5年間の術後内分泌療法による前治療歴を有する再発高リスクのER+、HER...</td>\n",
       "      <td>乳癌</td>\n",
       "      <td>乳癌</td>\n",
       "      <td>Breast Neoplasms</td>\n",
       "      <td>早期乳癌患者を対象としたimlunestrantと標準的な内分泌療法の比較試験</td>\n",
       "      <td>3</td>\n",
       "      <td>NaN</td>\n",
       "      <td>...</td>\n",
       "      <td>-Have a diagnosis of ER+, HER2- early-stage, r...</td>\n",
       "      <td>-Have any evidence of metastatic disease (incl...</td>\n",
       "      <td>18age old over</td>\n",
       "      <td>No limit</td>\n",
       "      <td>Both</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>-Drug: Imlunestrant Administered orally. Other...</td>\n",
       "      <td>[Breast Neoplasms]</td>\n",
       "      <td>[[tensor(-0.3342), tensor(-0.6973), tensor(-0....</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>391</th>\n",
       "      <td>jRCT2031230090</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>MELK阻害剤OTS167POにおける転移性・進行性乳がん患者を対象とした安全性、忍容性およ...</td>\n",
       "      <td>再発・難治性の局所進行性・転移性乳がん及びトリプルネガティブ乳がん</td>\n",
       "      <td>再発・難治性の局所進行性・転移性乳がん及びトリプルネガティブ乳がん</td>\n",
       "      <td>Relapsed/Refractory Locally Advanced or Metast...</td>\n",
       "      <td>再発・難治性の局所進行性または転移性乳がん患者に対して、OTS167を経口カプセルで投与する...</td>\n",
       "      <td>1</td>\n",
       "      <td>NaN</td>\n",
       "      <td>...</td>\n",
       "      <td>Dose Escalation and Dose Expansion Cohorts 1. ...</td>\n",
       "      <td>Dose Escalation and Expansion Cohorts 1. Women...</td>\n",
       "      <td>18age old over</td>\n",
       "      <td>No limit</td>\n",
       "      <td>Female</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>This is a Phase I dose escalation/expansion mu...</td>\n",
       "      <td>[Relapsed/Refract, Y Locally Advanced, Metasta...</td>\n",
       "      <td>[[tensor(-0.0153), tensor(-0.4994), tensor(-0....</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>392</th>\n",
       "      <td>jRCT2031230096</td>\n",
       "      <td>NCT05774951</td>\n",
       "      <td>NaN</td>\n",
       "      <td>根治的局所領域療法（化学療法の併用または非併用）および標準補助内分泌療法（ET）を少なくとも...</td>\n",
       "      <td>乳がん、早期乳がん</td>\n",
       "      <td>乳がん、早期乳がん</td>\n",
       "      <td>Breast Cancer, Early Breast Cancer</td>\n",
       "      <td>Treatment</td>\n",
       "      <td>3</td>\n",
       "      <td>NaN</td>\n",
       "      <td>...</td>\n",
       "      <td>- Women and Men, greater than or equal to 18 y...</td>\n",
       "      <td>- Inoperable locally advanced or metastatic br...</td>\n",
       "      <td>18age old over</td>\n",
       "      <td>No limit</td>\n",
       "      <td>Both</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>arm A: continue with SoC ET as directed by inv...</td>\n",
       "      <td>[Breast Cancer, Early Breast Cancer]</td>\n",
       "      <td>[[tensor(-0.0986), tensor(-0.6295), tensor(-0....</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>402</th>\n",
       "      <td>jRCT2061230009</td>\n",
       "      <td>NCT05485766</td>\n",
       "      <td>NaN</td>\n",
       "      <td>gBRCA1/2遺伝子変異を有するトリプルネガティブ原発乳がんに対するプラチナ製剤、PARP...</td>\n",
       "      <td>トリプルネガティブ乳がん</td>\n",
       "      <td>トリプルネガティブ乳がん</td>\n",
       "      <td>Triple Negative Breast Neoplasms</td>\n",
       "      <td>術前療法としてgBRCA変異陽性手術可能または局所進行TNBCに対してペムブロリズマブ+パク...</td>\n",
       "      <td>2</td>\n",
       "      <td>NaN</td>\n",
       "      <td>...</td>\n",
       "      <td>1)Male/female subjects who are at least 18 yea...</td>\n",
       "      <td>1) Subjects who has a positive urine pregnancy...</td>\n",
       "      <td>18age old exceed</td>\n",
       "      <td>No limit</td>\n",
       "      <td>Both</td>\n",
       "      <td>NaN</td>\n",
       "      <td>Triple Negative Breast Cancer, Breast Neoplasm...</td>\n",
       "      <td>Drug: Pembrolizumab 200 mg fixed dose, IV, eve...</td>\n",
       "      <td>[Triple Negative, Breast Neoplasms]</td>\n",
       "      <td>[[tensor(-0.2188), tensor(-0.4577), tensor(-0....</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>463</th>\n",
       "      <td>jRCT2061220087</td>\n",
       "      <td>NCT05629585</td>\n",
       "      <td>NaN</td>\n",
       "      <td>術前薬物療法後の外科的切除時に乳房及び／又は腋窩リンパ節に浸潤性残存病変を有するステージI～...</td>\n",
       "      <td>乳癌</td>\n",
       "      <td>乳癌</td>\n",
       "      <td>Breast Cancer</td>\n",
       "      <td>術前薬物療法後の外科的切除時に乳房および/または腋窩リンパ節に浸潤性残存病変を有するI~II...</td>\n",
       "      <td>3</td>\n",
       "      <td>NaN</td>\n",
       "      <td>...</td>\n",
       "      <td>Participant must be &gt;= 18 years at the time of...</td>\n",
       "      <td>Stage IV (metastatic) TNBC. History of prior i...</td>\n",
       "      <td>18age old over</td>\n",
       "      <td>130age old under</td>\n",
       "      <td>Both</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>Arm 1: Dato-DXd 6 mg/kg IV Q3W x 8 cycles + Du...</td>\n",
       "      <td>[Breast Cancer]</td>\n",
       "      <td>[[tensor(-0.0986), tensor(-0.6295), tensor(-0....</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>515</th>\n",
       "      <td>jRCT2031220276</td>\n",
       "      <td>NCT05307705</td>\n",
       "      <td>NaN</td>\n",
       "      <td>PIK3CA H1047R変異を有する進行乳がん患者及びその他の固形がん患者を対象としたLO...</td>\n",
       "      <td>乳がん</td>\n",
       "      <td>乳がん</td>\n",
       "      <td>Breast Cancer</td>\n",
       "      <td>LOXO-783 の単独投与及び他の抗がん剤との併用投与における 第 2 相試験の推奨用量、...</td>\n",
       "      <td>1</td>\n",
       "      <td>NaN</td>\n",
       "      <td>...</td>\n",
       "      <td>-Have advanced breast cancer or another solid ...</td>\n",
       "      <td>-Medical Conditions -Colorectal cancer -Endome...</td>\n",
       "      <td>18age old over</td>\n",
       "      <td>No limit</td>\n",
       "      <td>Both</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>-Drug: LOXO-783 Oral Other Name: LY3849524 -Dr...</td>\n",
       "      <td>[Breast Cancer]</td>\n",
       "      <td>[[tensor(-0.0986), tensor(-0.6295), tensor(-0....</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>555</th>\n",
       "      <td>jRCT2031220034</td>\n",
       "      <td>NCT04862663</td>\n",
       "      <td>NaN</td>\n",
       "      <td>局所進行（切除不能）又は転移性のホルモン受容体陽性／ヒト上皮細胞成長因子受容体2 陰性乳癌を...</td>\n",
       "      <td>局所進行（切除不能）又は転移性乳癌</td>\n",
       "      <td>局所進行(切除不能)又は転移性乳癌</td>\n",
       "      <td>Locally Advanced (Inoperable) or Metastatic Br...</td>\n",
       "      <td>Treatment</td>\n",
       "      <td>3</td>\n",
       "      <td>NaN</td>\n",
       "      <td>...</td>\n",
       "      <td>Key inclusion criteria for both phases: 1. Adu...</td>\n",
       "      <td>Key exclusion criteria for both phases: 1. His...</td>\n",
       "      <td>18age old over</td>\n",
       "      <td>99age old under</td>\n",
       "      <td>Both</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>Capivasertib: PO BD 4 days on /3 days off per ...</td>\n",
       "      <td>[Locally Advanced (Inoperable), Metastatic Bre...</td>\n",
       "      <td>[[tensor(-0.1555), tensor(-0.8327), tensor(-0....</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>590</th>\n",
       "      <td>jRCT2031210585</td>\n",
       "      <td>NCT05123482</td>\n",
       "      <td>NaN</td>\n",
       "      <td>進行又は転移性固形がん患者を対象としたAZD8205の単独療法及び他の抗がん剤との併用療法に...</td>\n",
       "      <td>乳癌、胆道癌、卵巣癌、子宮内膜癌</td>\n",
       "      <td>乳癌、胆道癌、卵巣癌、子宮内膜癌</td>\n",
       "      <td>Breast Cancer, Biliary Tract Carcinoma, Ovaria...</td>\n",
       "      <td>進行又は転移性固形がんに対する治療法候補としての新規化合物AZD8205の研究</td>\n",
       "      <td>1-2</td>\n",
       "      <td>NaN</td>\n",
       "      <td>...</td>\n",
       "      <td>- Age18 years or more - Relapsed/metastatic so...</td>\n",
       "      <td>- Treatment with any of the following: 1. Nitr...</td>\n",
       "      <td>18age old over</td>\n",
       "      <td>No limit</td>\n",
       "      <td>Both</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>Sub-study1 AZD8205 Monotherapy AZD8205 is an a...</td>\n",
       "      <td>[Breast Cancer, Biliary Tract Carcinoma, Ovari...</td>\n",
       "      <td>[[tensor(-0.0986), tensor(-0.6295), tensor(-0....</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>593</th>\n",
       "      <td>jRCT2031210560</td>\n",
       "      <td>NCT05061823</td>\n",
       "      <td>NaN</td>\n",
       "      <td>複数のBintrafusp alfa（M7824）臨床試験の被験者を対象に長期データを収集す...</td>\n",
       "      <td>固形癌</td>\n",
       "      <td>固形癌</td>\n",
       "      <td>Cancer</td>\n",
       "      <td>本治験は、実施中のbintrafusp alfaの親試験( (NCT02517398, NC...</td>\n",
       "      <td>3</td>\n",
       "      <td>NaN</td>\n",
       "      <td>...</td>\n",
       "      <td>- Participants who are currently enrolled in a...</td>\n",
       "      <td>- Participants who are pregnant or currently i...</td>\n",
       "      <td>18age old over</td>\n",
       "      <td>No limit</td>\n",
       "      <td>Both</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>Participants who are continuing treatment with...</td>\n",
       "      <td>[Cancer]</td>\n",
       "      <td>[[tensor(-0.2145), tensor(-0.7045), tensor(-0....</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>600</th>\n",
       "      <td>jRCT2041210123</td>\n",
       "      <td>NCT05114746</td>\n",
       "      <td>NaN</td>\n",
       "      <td>進行性PSMA陽性転移性去勢抵抗性前立腺がん（mCRPC）を有する患者を対象に177Lu-P...</td>\n",
       "      <td>前立腺癌</td>\n",
       "      <td>前立腺癌</td>\n",
       "      <td>Prostate Cancer</td>\n",
       "      <td>本治験の目的は、進行性PSMA陽性mCRPCの日本人患者に177Lu-PSMA-617を投与...</td>\n",
       "      <td>2</td>\n",
       "      <td>NaN</td>\n",
       "      <td>...</td>\n",
       "      <td>- ECOG performance status: 1. Post-taxane popu...</td>\n",
       "      <td>- Previous treatment with any of the following...</td>\n",
       "      <td>20age old over</td>\n",
       "      <td>No limit</td>\n",
       "      <td>Male</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>Radiation: 177Lu-PSMA-617 administered intrave...</td>\n",
       "      <td>[Prostate Cancer]</td>\n",
       "      <td>[[tensor(0.0202), tensor(-0.7650), tensor(-0.5...</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>629</th>\n",
       "      <td>jRCT2052210099</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>乳がんを有する成人女性及び健康成人女性を対象とした乳房用マイクロ波画像診断装置IGS-000...</td>\n",
       "      <td>乳がん</td>\n",
       "      <td>乳がん</td>\n",
       "      <td>Breast cancer</td>\n",
       "      <td>本治験の目的は,乳がん又はその疑いのある者,並びに乳がん又はその疑いがない者を対象に乳房画像...</td>\n",
       "      <td>2</td>\n",
       "      <td>NaN</td>\n",
       "      <td>...</td>\n",
       "      <td>Breast cancer (including suspected) (1) One of...</td>\n",
       "      <td>(1) There is trauma with bleeding on the breas...</td>\n",
       "      <td>20age old over</td>\n",
       "      <td>No limit</td>\n",
       "      <td>Female</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>The test using IGS-0001 will be performed twic...</td>\n",
       "      <td>[Breast Cancer]</td>\n",
       "      <td>[[tensor(-0.0986), tensor(-0.6295), tensor(-0....</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>691</th>\n",
       "      <td>jRCT2031200384</td>\n",
       "      <td>NCT04221542</td>\n",
       "      <td>NaN</td>\n",
       "      <td>転移性去勢抵抗性前立腺がん患者を対象としたAMG 509の安全性、忍容性、薬物動態及び有効性...</td>\n",
       "      <td>前立腺がん</td>\n",
       "      <td>前立腺がん</td>\n",
       "      <td>Prostate Cancer</td>\n",
       "      <td>成人被験者のAMG 509の安全性及び忍容性を評価し、最大耐量(MTD)又は第II 相推奨用...</td>\n",
       "      <td>1</td>\n",
       "      <td>NaN</td>\n",
       "      <td>...</td>\n",
       "      <td>- Parts 1, 2, and 5: Participants with histolo...</td>\n",
       "      <td>- Pathological finding consistent with pure sm...</td>\n",
       "      <td>18age old over</td>\n",
       "      <td>No limit</td>\n",
       "      <td>Male</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>- Experimental: Part 1: AMG 509 Intravenous (I...</td>\n",
       "      <td>[Prostate Cancer]</td>\n",
       "      <td>[[tensor(0.0202), tensor(-0.7650), tensor(-0.5...</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>698</th>\n",
       "      <td>jRCT2071200086</td>\n",
       "      <td>NCT04640623</td>\n",
       "      <td>NaN</td>\n",
       "      <td>膀胱全摘除術に不適格である，又は膀胱全摘除術を選択しなかった，カルメット・ゲラン桿菌（BCG...</td>\n",
       "      <td>膀胱癌</td>\n",
       "      <td>膀胱癌</td>\n",
       "      <td>Bladder Cancer</td>\n",
       "      <td>試験の目的は、高グレードTa又はT1乳頭状癌の併発の有無を問わない上皮内癌(CIS)患者にT...</td>\n",
       "      <td>2</td>\n",
       "      <td>NaN</td>\n",
       "      <td>...</td>\n",
       "      <td>- Histologically confirmed diagnosis of persis...</td>\n",
       "      <td>- Histologically confirmed, muscle-invasive, l...</td>\n",
       "      <td>18age old over</td>\n",
       "      <td>No limit</td>\n",
       "      <td>Both</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>TAR-200 Cohort 1: TAR-200 and Cetrelimab: TAR-...</td>\n",
       "      <td>[Bladder Cancer]</td>\n",
       "      <td>[[tensor(-0.2062), tensor(-0.5714), tensor(-0....</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>729</th>\n",
       "      <td>jRCT2031200057</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>治療抵抗性乳がんを対象としたTDM-812の腫瘍内投与法の安全性評価を目的とした第I相試験</td>\n",
       "      <td>乳がん</td>\n",
       "      <td>乳がん</td>\n",
       "      <td>Breast cancer</td>\n",
       "      <td>TDM-812の腫瘍内投与の安全性および忍容性を評価し、今後の臨床評価に用いる腫瘍内投与にお...</td>\n",
       "      <td>1</td>\n",
       "      <td>NaN</td>\n",
       "      <td>...</td>\n",
       "      <td>&lt;Disease Characteristics&gt; 1)Patients with inop...</td>\n",
       "      <td>1)Patients correspond to either of the followi...</td>\n",
       "      <td>20age old over</td>\n",
       "      <td>No limit</td>\n",
       "      <td>Female</td>\n",
       "      <td>NaN</td>\n",
       "      <td>Breast cancer</td>\n",
       "      <td>For the target tumor, the dose settled at each...</td>\n",
       "      <td>[Breast Cancer]</td>\n",
       "      <td>[[tensor(-0.0986), tensor(-0.6295), tensor(-0....</td>\n",
       "    </tr>\n",
       "  </tbody>\n",
       "</table>\n",
       "<p>27 rows × 41 columns</p>\n",
       "</div>"
      ],
      "text/plain": [
       "            JRCT ID       NCT No JapicCTI No  \\\n",
       "31   jRCT2061240081  NCT06393374         NaN   \n",
       "32   jRCT2031240476  NCT06312176         NaN   \n",
       "157  jRCT2052240059          NaN         NaN   \n",
       "170  jRCT2031240096  NCT06380751         NaN   \n",
       "209  jRCT2031230750  NCT06188559         NaN   \n",
       "215  jRCT2031230723  NCT06112379         NaN   \n",
       "224  jRCT2051230196  NCT06105632         NaN   \n",
       "234  jRCT2031230677  NCT06206837         NaN   \n",
       "240  jRCT2061230102  NCT06103864         NaN   \n",
       "307  jRCT2061230074  NCT05952557         NaN   \n",
       "342  jRCT2051230094  NCT05794906         NaN   \n",
       "351  jRCT2031230279  NCT05862285         NaN   \n",
       "385  jRCT2031230121  NCT04557449         NaN   \n",
       "388  jRCT2031230109  NCT05514054         NaN   \n",
       "391  jRCT2031230090          NaN         NaN   \n",
       "392  jRCT2031230096  NCT05774951         NaN   \n",
       "402  jRCT2061230009  NCT05485766         NaN   \n",
       "463  jRCT2061220087  NCT05629585         NaN   \n",
       "515  jRCT2031220276  NCT05307705         NaN   \n",
       "555  jRCT2031220034  NCT04862663         NaN   \n",
       "590  jRCT2031210585  NCT05123482         NaN   \n",
       "593  jRCT2031210560  NCT05061823         NaN   \n",
       "600  jRCT2041210123  NCT05114746         NaN   \n",
       "629  jRCT2052210099          NaN         NaN   \n",
       "691  jRCT2031200384  NCT04221542         NaN   \n",
       "698  jRCT2071200086  NCT04640623         NaN   \n",
       "729  jRCT2031200057          NaN         NaN   \n",
       "\n",
       "                                                 Title  \\\n",
       "31   術前薬物療法後の手術時に病理学的完全奏効を達成していないトリプルネガティブ乳癌患者を対象に、...   \n",
       "32   ホルモン受容体陽性・ヒト上皮成長因子受容体2陰性（HR+/HER2-）の切除不能な局所進行又...   \n",
       "157  遺伝子HSD17B4高メチル化を有するHER2陽性ER陰性乳癌における非手術療法の有用性を評...   \n",
       "170  本治験の主要目的は、BRCA1、BRCA2 又は PALB2 変異を有し、HR 陽性、HER...   \n",
       "209  治療歴のある HER2 陽性又は HER2 低発現の切除不能又は転移性乳癌の被験者を対象に，...   \n",
       "215  未治療のトリプルネガティブ又はホルモン受容体低発現／HER2陰性乳癌の成人患者を対象として、...   \n",
       "224  CDK4/6 阻害薬による前治療で進行したホルモン受容体陽性HER2 陰性の進行または転移乳...   \n",
       "234  18歳以上のER陽性／HER2陰性の進行または転移乳癌患者を対象として，経口タンパク質分解誘...   \n",
       "240  Programmed death-ligand（PD-L1）陽性の局所再発手術不能又は転移性...   \n",
       "307  根治的局所治療（化学療法の併用または非併用）を受けて疾患の兆候のない、再発リスクが中間～高リ...   \n",
       "342  前立腺癌の高リスク生化学的再発（BCR）患者を対象としたアンドロゲン遮断療法（ADT）を併用...   \n",
       "351  GENENTECH社及び／又はF.HOFFMANN-LA ROCHE LTDが依頼した試験に...   \n",
       "385  進行固形がん患者を対象として，PF-07220060 の単剤投与および併用投与の安全性，忍容...   \n",
       "388  EMBER-4：2～5年間の術後内分泌療法による前治療歴を有する再発高リスクのER+、HER...   \n",
       "391  MELK阻害剤OTS167POにおける転移性・進行性乳がん患者を対象とした安全性、忍容性およ...   \n",
       "392  根治的局所領域療法（化学療法の併用または非併用）および標準補助内分泌療法（ET）を少なくとも...   \n",
       "402  gBRCA1/2遺伝子変異を有するトリプルネガティブ原発乳がんに対するプラチナ製剤、PARP...   \n",
       "463  術前薬物療法後の外科的切除時に乳房及び／又は腋窩リンパ節に浸潤性残存病変を有するステージI～...   \n",
       "515  PIK3CA H1047R変異を有する進行乳がん患者及びその他の固形がん患者を対象としたLO...   \n",
       "555  局所進行（切除不能）又は転移性のホルモン受容体陽性／ヒト上皮細胞成長因子受容体2 陰性乳癌を...   \n",
       "590  進行又は転移性固形がん患者を対象としたAZD8205の単独療法及び他の抗がん剤との併用療法に...   \n",
       "593  複数のBintrafusp alfa（M7824）臨床試験の被験者を対象に長期データを収集す...   \n",
       "600  進行性PSMA陽性転移性去勢抵抗性前立腺がん（mCRPC）を有する患者を対象に177Lu-P...   \n",
       "629  乳がんを有する成人女性及び健康成人女性を対象とした乳房用マイクロ波画像診断装置IGS-000...   \n",
       "691  転移性去勢抵抗性前立腺がん患者を対象としたAMG 509の安全性、忍容性、薬物動態及び有効性...   \n",
       "698  膀胱全摘除術に不適格である，又は膀胱全摘除術を選択しなかった，カルメット・ゲラン桿菌（BCG...   \n",
       "729      治療抵抗性乳がんを対象としたTDM-812の腫瘍内投与法の安全性評価を目的とした第I相試験   \n",
       "\n",
       "                                         TargetJ  \\\n",
       "31                                   トリプルネガティブ乳癌   \n",
       "32                  HR+/HER2-の切除不能局所進行又は転移性乳癌の患者   \n",
       "157                                          乳がん   \n",
       "170                                         進行乳癌   \n",
       "209               HER2 陽性又は HER2 低発現の切除不能又は転移性乳癌   \n",
       "215                                           乳癌   \n",
       "224  ホルモン受容体（HR）陽性ヒト上皮増殖因子受容体2（HER2）陰性の進行または転移乳癌   \n",
       "234       エストロゲン受容体陽性／ヒト上皮増殖因子受容体 2 陰性の進行または転移乳癌   \n",
       "240                                           乳癌   \n",
       "307                                    乳がん、早期乳がん   \n",
       "342                                         前立腺癌   \n",
       "351                                            癌   \n",
       "385                  進行または転移性の乳癌，前立腺癌，およびその他の固形癌   \n",
       "388                                           乳癌   \n",
       "391            再発・難治性の局所進行性・転移性乳がん及びトリプルネガティブ乳がん   \n",
       "392                                    乳がん、早期乳がん   \n",
       "402                                 トリプルネガティブ乳がん   \n",
       "463                                           乳癌   \n",
       "515                                          乳がん   \n",
       "555                            局所進行（切除不能）又は転移性乳癌   \n",
       "590                             乳癌、胆道癌、卵巣癌、子宮内膜癌   \n",
       "593                                          固形癌   \n",
       "600                                         前立腺癌   \n",
       "629                                          乳がん   \n",
       "691                                        前立腺がん   \n",
       "698                                          膀胱癌   \n",
       "729                                          乳がん   \n",
       "\n",
       "                                          Target  \\\n",
       "31                                   トリプルネガティブ乳癌   \n",
       "32                  HR+/HER2-の切除不能局所進行又は転移性乳癌の患者   \n",
       "157                                          乳がん   \n",
       "170                                         進行乳癌   \n",
       "209               HER2 陽性又は HER2 低発現の切除不能又は転移性乳癌   \n",
       "215                                           乳癌   \n",
       "224  ホルモン受容体(HR)陽性ヒト上皮増殖因子受容体2(HER2)陰性の進行または転移乳癌   \n",
       "234       エストロゲン受容体陽性/ヒト上皮増殖因子受容体 2 陰性の進行または転移乳癌   \n",
       "240                                           乳癌   \n",
       "307                                    乳がん、早期乳がん   \n",
       "342                                         前立腺癌   \n",
       "351                                            癌   \n",
       "385                  進行または転移性の乳癌,前立腺癌,およびその他の固形癌   \n",
       "388                                           乳癌   \n",
       "391            再発・難治性の局所進行性・転移性乳がん及びトリプルネガティブ乳がん   \n",
       "392                                    乳がん、早期乳がん   \n",
       "402                                 トリプルネガティブ乳がん   \n",
       "463                                           乳癌   \n",
       "515                                          乳がん   \n",
       "555                            局所進行(切除不能)又は転移性乳癌   \n",
       "590                             乳癌、胆道癌、卵巣癌、子宮内膜癌   \n",
       "593                                          固形癌   \n",
       "600                                         前立腺癌   \n",
       "629                                          乳がん   \n",
       "691                                        前立腺がん   \n",
       "698                                          膀胱癌   \n",
       "729                                          乳がん   \n",
       "\n",
       "                                         TargetEnglish  \\\n",
       "31                       Triple negative breast cancer   \n",
       "32   HR+/HER2 unresectable locally advanced or meta...   \n",
       "157                                      Breast cancer   \n",
       "170                             Advanced Breast Cancer   \n",
       "209  HER2-positive or HER2-low unresectable or meta...   \n",
       "215                                      Breast Cancer   \n",
       "224  HR-positive, HER2-negative Advanced or Metasta...   \n",
       "234     ER+/HER2- advanced or metastatic breast cancer   \n",
       "240                                      Breast Cancer   \n",
       "307                 Breast Cancer, Early Breast Cancer   \n",
       "342                                    prostate cancer   \n",
       "351                                             Cancer   \n",
       "385  advanced or metastatic BC, prostate cancer or ...   \n",
       "388                                   Breast Neoplasms   \n",
       "391  Relapsed/Refractory Locally Advanced or Metast...   \n",
       "392                 Breast Cancer, Early Breast Cancer   \n",
       "402                   Triple Negative Breast Neoplasms   \n",
       "463                                      Breast Cancer   \n",
       "515                                      Breast Cancer   \n",
       "555  Locally Advanced (Inoperable) or Metastatic Br...   \n",
       "590  Breast Cancer, Biliary Tract Carcinoma, Ovaria...   \n",
       "593                                             Cancer   \n",
       "600                                    Prostate Cancer   \n",
       "629                                      Breast cancer   \n",
       "691                                    Prostate Cancer   \n",
       "698                                     Bladder Cancer   \n",
       "729                                      Breast cancer   \n",
       "\n",
       "                                              研究・治験の目的 試験等のフェーズ 試験の種類  ...  \\\n",
       "31   術前薬物療法後の手術時に病理学的完全奏効を達成していないトリプルネガティブ乳癌患者を対象に、...        3   NaN  ...   \n",
       "32   HR+/HER2-の切除不能な局所進行又は転移性乳癌患者を対象にMK-2870の単剤及びペム...        3   NaN  ...   \n",
       "157  HSD17B4高メチル化(HSD17B4 hypermethylation:HH)を有するH...        2   NaN  ...   \n",
       "170                                          Treatment        3   NaN  ...   \n",
       "209                                                 治療        2   NaN  ...   \n",
       "215  未治療のトリプルネガティブ又はホルモン受容体低発現/HER2陰性乳癌の成人患者を対象として、...        3   NaN  ...   \n",
       "224  CDK4/6阻害薬による前治療で進行したHR陽性HER2陰性の進行または転移乳癌患者を対象に...        3   NaN  ...   \n",
       "234  進行乳癌患者を対象として,vepdegestrantとPF-07220060を併用投与したと...      1-2   NaN  ...   \n",
       "240  PD-L1陽性の局所再発手術不能または転移性TNBC患者を対象に、デュルバルマブ併用または非...        3   NaN  ...   \n",
       "307                                          Treatment        3   NaN  ...   \n",
       "342  BAY 1841788(darolutamide)とADTの24ヵ月間併用投与によって、プラ...        3   NaN  ...   \n",
       "351  本継続投与試験の目的は,親治験から移行する時点でまだ治験治療を受けており,その地域でその治療...        3   NaN  ...   \n",
       "385  パート1(用量漸増,食事の影響,DDIの評価パート) 1A: PF-07220060のMTD...        1   NaN  ...   \n",
       "388            早期乳癌患者を対象としたimlunestrantと標準的な内分泌療法の比較試験        3   NaN  ...   \n",
       "391  再発・難治性の局所進行性または転移性乳がん患者に対して、OTS167を経口カプセルで投与する...        1   NaN  ...   \n",
       "392                                          Treatment        3   NaN  ...   \n",
       "402  術前療法としてgBRCA変異陽性手術可能または局所進行TNBCに対してペムブロリズマブ+パク...        2   NaN  ...   \n",
       "463  術前薬物療法後の外科的切除時に乳房および/または腋窩リンパ節に浸潤性残存病変を有するI~II...        3   NaN  ...   \n",
       "515  LOXO-783 の単独投与及び他の抗がん剤との併用投与における 第 2 相試験の推奨用量、...        1   NaN  ...   \n",
       "555                                          Treatment        3   NaN  ...   \n",
       "590            進行又は転移性固形がんに対する治療法候補としての新規化合物AZD8205の研究      1-2   NaN  ...   \n",
       "593  本治験は、実施中のbintrafusp alfaの親試験( (NCT02517398, NC...        3   NaN  ...   \n",
       "600  本治験の目的は、進行性PSMA陽性mCRPCの日本人患者に177Lu-PSMA-617を投与...        2   NaN  ...   \n",
       "629  本治験の目的は,乳がん又はその疑いのある者,並びに乳がん又はその疑いがない者を対象に乳房画像...        2   NaN  ...   \n",
       "691  成人被験者のAMG 509の安全性及び忍容性を評価し、最大耐量(MTD)又は第II 相推奨用...        1   NaN  ...   \n",
       "698  試験の目的は、高グレードTa又はT1乳頭状癌の併発の有無を問わない上皮内癌(CIS)患者にT...        2   NaN  ...   \n",
       "729  TDM-812の腫瘍内投与の安全性および忍容性を評価し、今後の臨床評価に用いる腫瘍内投与にお...        1   NaN  ...   \n",
       "\n",
       "                                    Inclusion Criteria  \\\n",
       "31   -Has centrally confirmed TNBC, as defined by t...   \n",
       "32   -Has unresectable locally advanced or metastat...   \n",
       "157  1. Histologically confirmed invasive breast ca...   \n",
       "170  - Adult females, pre/peri-menopausal and/or po...   \n",
       "209  (1) Male or female, aged >=18 years at the tim...   \n",
       "215  1. Participant must be >= 18 years at the time...   \n",
       "224  * Histological confirmation of breast cancer w...   \n",
       "234  Inclusion Criteria: * Histological or cytologi...   \n",
       "240  Histologically or cytologically documented loc...   \n",
       "307  - Women and Men; 18 years or more at the time ...   \n",
       "342  - Histologically or cytologically confirmed ad...   \n",
       "351  - Eligible for continuing Roche IMP-based ther...   \n",
       "385  Disease * Part 1A/1D/1E: HR+HER2-BC, HR+HER2+B...   \n",
       "388  -Have a diagnosis of ER+, HER2- early-stage, r...   \n",
       "391  Dose Escalation and Dose Expansion Cohorts 1. ...   \n",
       "392  - Women and Men, greater than or equal to 18 y...   \n",
       "402  1)Male/female subjects who are at least 18 yea...   \n",
       "463  Participant must be >= 18 years at the time of...   \n",
       "515  -Have advanced breast cancer or another solid ...   \n",
       "555  Key inclusion criteria for both phases: 1. Adu...   \n",
       "590  - Age18 years or more - Relapsed/metastatic so...   \n",
       "593  - Participants who are currently enrolled in a...   \n",
       "600  - ECOG performance status: 1. Post-taxane popu...   \n",
       "629  Breast cancer (including suspected) (1) One of...   \n",
       "691  - Parts 1, 2, and 5: Participants with histolo...   \n",
       "698  - Histologically confirmed diagnosis of persis...   \n",
       "729  <Disease Characteristics> 1)Patients with inop...   \n",
       "\n",
       "                                    Exclusion Criteria       Age Minimum  \\\n",
       "31   -Has a known germline breast cancer gene (BRCA...    18age old over   \n",
       "32   -Has breast cancer amenable to treatment with ...    18age old over   \n",
       "157  1. History of other malignancy within the last...    20age old over   \n",
       "170  - Participants with history of MDS/AML or with...    18age old over   \n",
       "209  (1) Presence of brain or subdural metastases, ...    18age old over   \n",
       "215  1. As judged by the investigator, any evidence...    18age old over   \n",
       "224  * Any medical or psychiatric condition that ma...    18age old over   \n",
       "234  Exclusion Criteria: * visceral crisis at risk ...    18age old over   \n",
       "240  As judged by investigator, severe or uncontrol...    18age old over   \n",
       "307  - Inoperable locally advanced or metastatic br...    18age old over   \n",
       "342  - Pathological finding consistent with small c...    18age old over   \n",
       "351  - Meet any of the study treatment discontinuat...    18age old over   \n",
       "385  * Part 1D: participants who have had a gastrec...    18age old over   \n",
       "388  -Have any evidence of metastatic disease (incl...    18age old over   \n",
       "391  Dose Escalation and Expansion Cohorts 1. Women...    18age old over   \n",
       "392  - Inoperable locally advanced or metastatic br...    18age old over   \n",
       "402  1) Subjects who has a positive urine pregnancy...  18age old exceed   \n",
       "463  Stage IV (metastatic) TNBC. History of prior i...    18age old over   \n",
       "515  -Medical Conditions -Colorectal cancer -Endome...    18age old over   \n",
       "555  Key exclusion criteria for both phases: 1. His...    18age old over   \n",
       "590  - Treatment with any of the following: 1. Nitr...    18age old over   \n",
       "593  - Participants who are pregnant or currently i...    18age old over   \n",
       "600  - Previous treatment with any of the following...    20age old over   \n",
       "629  (1) There is trauma with bleeding on the breas...    20age old over   \n",
       "691  - Pathological finding consistent with pure sm...    18age old over   \n",
       "698  - Histologically confirmed, muscle-invasive, l...    18age old over   \n",
       "729  1)Patients correspond to either of the followi...    20age old over   \n",
       "\n",
       "          Age Maximum  Gender Discontinuation Criteria  \\\n",
       "31           No limit    Both                      NaN   \n",
       "32           No limit    Both                      NaN   \n",
       "157          No limit  Female                      NaN   \n",
       "170          No limit    Both                      NaN   \n",
       "209          No limit    Both                      NaN   \n",
       "215          No limit    Both                      NaN   \n",
       "224          No limit    Both                      NaN   \n",
       "234          No limit    Both                      NaN   \n",
       "240          No limit    Both                      NaN   \n",
       "307          No limit     NaN                      NaN   \n",
       "342          No limit    Male                      NaN   \n",
       "351          No limit    Both                      NaN   \n",
       "385          No limit    Both                      NaN   \n",
       "388          No limit    Both                      NaN   \n",
       "391          No limit  Female                      NaN   \n",
       "392          No limit    Both                      NaN   \n",
       "402          No limit    Both                      NaN   \n",
       "463  130age old under    Both                      NaN   \n",
       "515          No limit    Both                      NaN   \n",
       "555   99age old under    Both                      NaN   \n",
       "590          No limit    Both                      NaN   \n",
       "593          No limit    Both                      NaN   \n",
       "600          No limit    Male                      NaN   \n",
       "629          No limit  Female                      NaN   \n",
       "691          No limit    Male                      NaN   \n",
       "698          No limit    Both                      NaN   \n",
       "729          No limit  Female                      NaN   \n",
       "\n",
       "                                               Keyword  \\\n",
       "31                                                 NaN   \n",
       "32                                                 NaN   \n",
       "157                                                NaN   \n",
       "170                                                NaN   \n",
       "209                                                NaN   \n",
       "215                                                NaN   \n",
       "224                                                NaN   \n",
       "234                                                NaN   \n",
       "240                                                NaN   \n",
       "307                                                NaN   \n",
       "342                                                NaN   \n",
       "351                                                NaN   \n",
       "385                                                NaN   \n",
       "388                                                NaN   \n",
       "391                                                NaN   \n",
       "392                                                NaN   \n",
       "402  Triple Negative Breast Cancer, Breast Neoplasm...   \n",
       "463                                                NaN   \n",
       "515                                                NaN   \n",
       "555                                                NaN   \n",
       "590                                                NaN   \n",
       "593                                                NaN   \n",
       "600                                                NaN   \n",
       "629                                                NaN   \n",
       "691                                                NaN   \n",
       "698                                                NaN   \n",
       "729                                      Breast cancer   \n",
       "\n",
       "                                       Intervention(s)  \\\n",
       "31   -Arm 1: MK-2870 4mg/kg intravenous (IV) every ...   \n",
       "32   Arm A:MK-2870 Participants receive 4 mg/kg of ...   \n",
       "157  Omitting breast surgery after preoperative che...   \n",
       "170  Experimental: Arm 1: saruparib (AZD5305) plus ...   \n",
       "209  Generic Name:NA Study Treatment in dose optimi...   \n",
       "215  - Experimental arm: Dato-DXd plus durvalumab n...   \n",
       "224  Arm A (Experimental Arm) PF-07220060 plus Fulv...   \n",
       "234  Drug: vepdegestrant Daily oral dosages of vepd...   \n",
       "240  Arm 1: Dato-DXd + durvalumab Arm 2: Investigat...   \n",
       "307  arm A: continue with SoC ET as directed by inv...   \n",
       "342  Drug: Darolutamide (BAY1841788, Nubeqa) Coated...   \n",
       "351  Ipatasertib: Ipatasertib will be administered ...   \n",
       "385  * Part 1A/1D/1E: PF-07220060 (CDK4 inhibitor) ...   \n",
       "388  -Drug: Imlunestrant Administered orally. Other...   \n",
       "391  This is a Phase I dose escalation/expansion mu...   \n",
       "392  arm A: continue with SoC ET as directed by inv...   \n",
       "402  Drug: Pembrolizumab 200 mg fixed dose, IV, eve...   \n",
       "463  Arm 1: Dato-DXd 6 mg/kg IV Q3W x 8 cycles + Du...   \n",
       "515  -Drug: LOXO-783 Oral Other Name: LY3849524 -Dr...   \n",
       "555  Capivasertib: PO BD 4 days on /3 days off per ...   \n",
       "590  Sub-study1 AZD8205 Monotherapy AZD8205 is an a...   \n",
       "593  Participants who are continuing treatment with...   \n",
       "600  Radiation: 177Lu-PSMA-617 administered intrave...   \n",
       "629  The test using IGS-0001 will be performed twic...   \n",
       "691  - Experimental: Part 1: AMG 509 Intravenous (I...   \n",
       "698  TAR-200 Cohort 1: TAR-200 and Cetrelimab: TAR-...   \n",
       "729  For the target tumor, the dose settled at each...   \n",
       "\n",
       "                                            TargetWord  \\\n",
       "31                    [Triple Negative, Breast Cancer]   \n",
       "32   [Hr+/Her2 Unresectable Locally Advanced, Metas...   \n",
       "157                                    [Breast Cancer]   \n",
       "170                           [Advanced Breast Cancer]   \n",
       "209  [Her2-Positive, Her2-Low Unresectable, Metasta...   \n",
       "215                                    [Breast Cancer]   \n",
       "224  [Hr-Positive, Her2-Negative Advanced, Metastat...   \n",
       "234     [Er+/Her2- Advanced, Metastatic Breast Cancer]   \n",
       "240                                    [Breast Cancer]   \n",
       "307               [Breast Cancer, Early Breast Cancer]   \n",
       "342                                  [Prostate Cancer]   \n",
       "351                                           [Cancer]   \n",
       "385  [Advanced, Metastatic Bc, Prostate Cancer, Oth...   \n",
       "388                                 [Breast Neoplasms]   \n",
       "391  [Relapsed/Refract, Y Locally Advanced, Metasta...   \n",
       "392               [Breast Cancer, Early Breast Cancer]   \n",
       "402                [Triple Negative, Breast Neoplasms]   \n",
       "463                                    [Breast Cancer]   \n",
       "515                                    [Breast Cancer]   \n",
       "555  [Locally Advanced (Inoperable), Metastatic Bre...   \n",
       "590  [Breast Cancer, Biliary Tract Carcinoma, Ovari...   \n",
       "593                                           [Cancer]   \n",
       "600                                  [Prostate Cancer]   \n",
       "629                                    [Breast Cancer]   \n",
       "691                                  [Prostate Cancer]   \n",
       "698                                   [Bladder Cancer]   \n",
       "729                                    [Breast Cancer]   \n",
       "\n",
       "                                             TargetVec  \n",
       "31   [[tensor(-0.2188), tensor(-0.4577), tensor(-0....  \n",
       "32   [[tensor(0.0489), tensor(-0.7584), tensor(-0.4...  \n",
       "157  [[tensor(-0.0986), tensor(-0.6295), tensor(-0....  \n",
       "170  [[tensor(-0.2116), tensor(-0.5020), tensor(-0....  \n",
       "209  [[tensor(-0.0889), tensor(-0.8979), tensor(-0....  \n",
       "215  [[tensor(-0.0986), tensor(-0.6295), tensor(-0....  \n",
       "224  [[tensor(-0.0832), tensor(-0.7572), tensor(-0....  \n",
       "234  [[tensor(-0.1549), tensor(-0.3944), tensor(-0....  \n",
       "240  [[tensor(-0.0986), tensor(-0.6295), tensor(-0....  \n",
       "307  [[tensor(-0.0986), tensor(-0.6295), tensor(-0....  \n",
       "342  [[tensor(0.0202), tensor(-0.7650), tensor(-0.5...  \n",
       "351  [[tensor(-0.2145), tensor(-0.7045), tensor(-0....  \n",
       "385  [[tensor(-0.4654), tensor(-0.4115), tensor(-0....  \n",
       "388  [[tensor(-0.3342), tensor(-0.6973), tensor(-0....  \n",
       "391  [[tensor(-0.0153), tensor(-0.4994), tensor(-0....  \n",
       "392  [[tensor(-0.0986), tensor(-0.6295), tensor(-0....  \n",
       "402  [[tensor(-0.2188), tensor(-0.4577), tensor(-0....  \n",
       "463  [[tensor(-0.0986), tensor(-0.6295), tensor(-0....  \n",
       "515  [[tensor(-0.0986), tensor(-0.6295), tensor(-0....  \n",
       "555  [[tensor(-0.1555), tensor(-0.8327), tensor(-0....  \n",
       "590  [[tensor(-0.0986), tensor(-0.6295), tensor(-0....  \n",
       "593  [[tensor(-0.2145), tensor(-0.7045), tensor(-0....  \n",
       "600  [[tensor(0.0202), tensor(-0.7650), tensor(-0.5...  \n",
       "629  [[tensor(-0.0986), tensor(-0.6295), tensor(-0....  \n",
       "691  [[tensor(0.0202), tensor(-0.7650), tensor(-0.5...  \n",
       "698  [[tensor(-0.2062), tensor(-0.5714), tensor(-0....  \n",
       "729  [[tensor(-0.0986), tensor(-0.6295), tensor(-0....  \n",
       "\n",
       "[27 rows x 41 columns]"
      ]
     },
     "execution_count": 24,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "get_matched_df(basedf, \"breast cancer\", model, threshold=0.925)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": []
  },
  {
   "cell_type": "code",
   "execution_count": 16,
   "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>JRCT ID</th>\n",
       "      <th>NCT No</th>\n",
       "      <th>JapicCTI No</th>\n",
       "      <th>Title</th>\n",
       "      <th>TargetJ</th>\n",
       "      <th>Target</th>\n",
       "      <th>TargetEnglish</th>\n",
       "      <th>研究・治験の目的</th>\n",
       "      <th>試験等のフェーズ</th>\n",
       "      <th>試験の種類</th>\n",
       "      <th>...</th>\n",
       "      <th>purpose</th>\n",
       "      <th>Inclusion Criteria</th>\n",
       "      <th>Exclusion Criteria</th>\n",
       "      <th>Age Minimum</th>\n",
       "      <th>Age Maximum</th>\n",
       "      <th>Gender</th>\n",
       "      <th>Discontinuation Criteria</th>\n",
       "      <th>Keyword</th>\n",
       "      <th>Intervention(s)</th>\n",
       "      <th>TargetWord</th>\n",
       "    </tr>\n",
       "  </thead>\n",
       "  <tbody>\n",
       "    <tr>\n",
       "      <th>31</th>\n",
       "      <td>jRCT2061240081</td>\n",
       "      <td>NCT06393374</td>\n",
       "      <td>NaN</td>\n",
       "      <td>術前薬物療法後の手術時に病理学的完全奏効を達成していないトリプルネガティブ乳癌患者を対象に、...</td>\n",
       "      <td>トリプルネガティブ乳癌</td>\n",
       "      <td>トリプルネガティブ乳癌</td>\n",
       "      <td>Triple negative breast cancer</td>\n",
       "      <td>術前薬物療法後の手術時に病理学的完全奏効を達成していないトリプルネガティブ乳癌患者を対象に、...</td>\n",
       "      <td>3</td>\n",
       "      <td>NaN</td>\n",
       "      <td>...</td>\n",
       "      <td>treatment purpose</td>\n",
       "      <td>-Has centrally confirmed TNBC, as defined by t...</td>\n",
       "      <td>-Has a known germline breast cancer gene (BRCA...</td>\n",
       "      <td>18age old over</td>\n",
       "      <td>No limit</td>\n",
       "      <td>Both</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>-Arm 1: MK-2870 4mg/kg intravenous (IV) every ...</td>\n",
       "      <td>[Triple Negative, Breast Cancer]</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>32</th>\n",
       "      <td>jRCT2031240476</td>\n",
       "      <td>NCT06312176</td>\n",
       "      <td>NaN</td>\n",
       "      <td>ホルモン受容体陽性・ヒト上皮成長因子受容体2陰性（HR+/HER2-）の切除不能な局所進行又...</td>\n",
       "      <td>HR+/HER2-の切除不能局所進行又は転移性乳癌の患者</td>\n",
       "      <td>HR+/HER2-の切除不能局所進行又は転移性乳癌の患者</td>\n",
       "      <td>HR+/HER2 unresectable locally advanced or meta...</td>\n",
       "      <td>HR+/HER2-の切除不能な局所進行又は転移性乳癌患者を対象にMK-2870の単剤及びペム...</td>\n",
       "      <td>3</td>\n",
       "      <td>NaN</td>\n",
       "      <td>...</td>\n",
       "      <td>treatment purpose</td>\n",
       "      <td>-Has unresectable locally advanced or metastat...</td>\n",
       "      <td>-Has breast cancer amenable to treatment with ...</td>\n",
       "      <td>18age old over</td>\n",
       "      <td>No limit</td>\n",
       "      <td>Both</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>Arm A:MK-2870 Participants receive 4 mg/kg of ...</td>\n",
       "      <td>[Hr+/Her2 Unresectable Locally Advanced, Metas...</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>157</th>\n",
       "      <td>jRCT2052240059</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>遺伝子HSD17B4高メチル化を有するHER2陽性ER陰性乳癌における非手術療法の有用性を評...</td>\n",
       "      <td>乳がん</td>\n",
       "      <td>乳がん</td>\n",
       "      <td>Breast cancer</td>\n",
       "      <td>HSD17B4高メチル化(HSD17B4 hypermethylation:HH)を有するH...</td>\n",
       "      <td>2</td>\n",
       "      <td>NaN</td>\n",
       "      <td>...</td>\n",
       "      <td>diagnostic purpose</td>\n",
       "      <td>1. Histologically confirmed invasive breast ca...</td>\n",
       "      <td>1. History of other malignancy within the last...</td>\n",
       "      <td>20age old over</td>\n",
       "      <td>No limit</td>\n",
       "      <td>Female</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>Omitting breast surgery after preoperative che...</td>\n",
       "      <td>[Breast Cancer]</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>170</th>\n",
       "      <td>jRCT2031240096</td>\n",
       "      <td>NCT06380751</td>\n",
       "      <td>NaN</td>\n",
       "      <td>本治験の主要目的は、BRCA1、BRCA2 又は PALB2 変異を有し、HR 陽性、HER...</td>\n",
       "      <td>進行乳癌</td>\n",
       "      <td>進行乳癌</td>\n",
       "      <td>Advanced Breast Cancer</td>\n",
       "      <td>Treatment</td>\n",
       "      <td>3</td>\n",
       "      <td>NaN</td>\n",
       "      <td>...</td>\n",
       "      <td>treatment purpose</td>\n",
       "      <td>- Adult females, pre/peri-menopausal and/or po...</td>\n",
       "      <td>- Participants with history of MDS/AML or with...</td>\n",
       "      <td>18age old over</td>\n",
       "      <td>No limit</td>\n",
       "      <td>Both</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>Experimental: Arm 1: saruparib (AZD5305) plus ...</td>\n",
       "      <td>[Advanced Breast Cancer]</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>209</th>\n",
       "      <td>jRCT2031230750</td>\n",
       "      <td>NCT06188559</td>\n",
       "      <td>NaN</td>\n",
       "      <td>治療歴のある HER2 陽性又は HER2 低発現の切除不能又は転移性乳癌の被験者を対象に，...</td>\n",
       "      <td>HER2 陽性又は HER2 低発現の切除不能又は転移性乳癌</td>\n",
       "      <td>HER2 陽性又は HER2 低発現の切除不能又は転移性乳癌</td>\n",
       "      <td>HER2-positive or HER2-low unresectable or meta...</td>\n",
       "      <td>治療</td>\n",
       "      <td>2</td>\n",
       "      <td>NaN</td>\n",
       "      <td>...</td>\n",
       "      <td>treatment purpose</td>\n",
       "      <td>(1) Male or female, aged &gt;=18 years at the tim...</td>\n",
       "      <td>(1) Presence of brain or subdural metastases, ...</td>\n",
       "      <td>18age old over</td>\n",
       "      <td>No limit</td>\n",
       "      <td>Both</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>Generic Name:NA Study Treatment in dose optimi...</td>\n",
       "      <td>[Her2-Positive, Her2-Low Unresectable, Metasta...</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>215</th>\n",
       "      <td>jRCT2031230723</td>\n",
       "      <td>NCT06112379</td>\n",
       "      <td>NaN</td>\n",
       "      <td>未治療のトリプルネガティブ又はホルモン受容体低発現／HER2陰性乳癌の成人患者を対象として、...</td>\n",
       "      <td>乳癌</td>\n",
       "      <td>乳癌</td>\n",
       "      <td>Breast Cancer</td>\n",
       "      <td>未治療のトリプルネガティブ又はホルモン受容体低発現/HER2陰性乳癌の成人患者を対象として、...</td>\n",
       "      <td>3</td>\n",
       "      <td>NaN</td>\n",
       "      <td>...</td>\n",
       "      <td>treatment purpose</td>\n",
       "      <td>1. Participant must be &gt;= 18 years at the time...</td>\n",
       "      <td>1. As judged by the investigator, any evidence...</td>\n",
       "      <td>18age old over</td>\n",
       "      <td>No limit</td>\n",
       "      <td>Both</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>- Experimental arm: Dato-DXd plus durvalumab n...</td>\n",
       "      <td>[Breast Cancer]</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>224</th>\n",
       "      <td>jRCT2051230196</td>\n",
       "      <td>NCT06105632</td>\n",
       "      <td>NaN</td>\n",
       "      <td>CDK4/6 阻害薬による前治療で進行したホルモン受容体陽性HER2 陰性の進行または転移乳...</td>\n",
       "      <td>ホルモン受容体（HR）陽性ヒト上皮増殖因子受容体2（HER2）陰性の進行または転移乳癌</td>\n",
       "      <td>ホルモン受容体(HR)陽性ヒト上皮増殖因子受容体2(HER2)陰性の進行または転移乳癌</td>\n",
       "      <td>HR-positive, HER2-negative Advanced or Metasta...</td>\n",
       "      <td>CDK4/6阻害薬による前治療で進行したHR陽性HER2陰性の進行または転移乳癌患者を対象に...</td>\n",
       "      <td>3</td>\n",
       "      <td>NaN</td>\n",
       "      <td>...</td>\n",
       "      <td>treatment purpose</td>\n",
       "      <td>* Histological confirmation of breast cancer w...</td>\n",
       "      <td>* Any medical or psychiatric condition that ma...</td>\n",
       "      <td>18age old over</td>\n",
       "      <td>No limit</td>\n",
       "      <td>Both</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>Arm A (Experimental Arm) PF-07220060 plus Fulv...</td>\n",
       "      <td>[Hr-Positive, Her2-Negative Advanced, Metastat...</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>234</th>\n",
       "      <td>jRCT2031230677</td>\n",
       "      <td>NCT06206837</td>\n",
       "      <td>NaN</td>\n",
       "      <td>18歳以上のER陽性／HER2陰性の進行または転移乳癌患者を対象として，経口タンパク質分解誘...</td>\n",
       "      <td>エストロゲン受容体陽性／ヒト上皮増殖因子受容体 2 陰性の進行または転移乳癌</td>\n",
       "      <td>エストロゲン受容体陽性/ヒト上皮増殖因子受容体 2 陰性の進行または転移乳癌</td>\n",
       "      <td>ER+/HER2- advanced or metastatic breast cancer</td>\n",
       "      <td>進行乳癌患者を対象として,vepdegestrantとPF-07220060を併用投与したと...</td>\n",
       "      <td>1-2</td>\n",
       "      <td>NaN</td>\n",
       "      <td>...</td>\n",
       "      <td>treatment purpose</td>\n",
       "      <td>Inclusion Criteria: * Histological or cytologi...</td>\n",
       "      <td>Exclusion Criteria: * visceral crisis at risk ...</td>\n",
       "      <td>18age old over</td>\n",
       "      <td>No limit</td>\n",
       "      <td>Both</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>Drug: vepdegestrant Daily oral dosages of vepd...</td>\n",
       "      <td>[Er+/Her2- Advanced, Metastatic Breast Cancer]</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>240</th>\n",
       "      <td>jRCT2061230102</td>\n",
       "      <td>NCT06103864</td>\n",
       "      <td>NaN</td>\n",
       "      <td>Programmed death-ligand（PD-L1）陽性の局所再発手術不能又は転移性...</td>\n",
       "      <td>乳癌</td>\n",
       "      <td>乳癌</td>\n",
       "      <td>Breast Cancer</td>\n",
       "      <td>PD-L1陽性の局所再発手術不能または転移性TNBC患者を対象に、デュルバルマブ併用または非...</td>\n",
       "      <td>3</td>\n",
       "      <td>NaN</td>\n",
       "      <td>...</td>\n",
       "      <td>treatment purpose</td>\n",
       "      <td>Histologically or cytologically documented loc...</td>\n",
       "      <td>As judged by investigator, severe or uncontrol...</td>\n",
       "      <td>18age old over</td>\n",
       "      <td>No limit</td>\n",
       "      <td>Both</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>Arm 1: Dato-DXd + durvalumab Arm 2: Investigat...</td>\n",
       "      <td>[Breast Cancer]</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>307</th>\n",
       "      <td>jRCT2061230074</td>\n",
       "      <td>NCT05952557</td>\n",
       "      <td>NaN</td>\n",
       "      <td>根治的局所治療（化学療法の併用または非併用）を受けて疾患の兆候のない、再発リスクが中間～高リ...</td>\n",
       "      <td>乳がん、早期乳がん</td>\n",
       "      <td>乳がん、早期乳がん</td>\n",
       "      <td>Breast Cancer, Early Breast Cancer</td>\n",
       "      <td>Treatment</td>\n",
       "      <td>3</td>\n",
       "      <td>NaN</td>\n",
       "      <td>...</td>\n",
       "      <td>treatment purpose</td>\n",
       "      <td>- Women and Men; 18 years or more at the time ...</td>\n",
       "      <td>- Inoperable locally advanced or metastatic br...</td>\n",
       "      <td>18age old over</td>\n",
       "      <td>No limit</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>arm A: continue with SoC ET as directed by inv...</td>\n",
       "      <td>[Breast Cancer, Early Breast Cancer]</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>342</th>\n",
       "      <td>jRCT2051230094</td>\n",
       "      <td>NCT05794906</td>\n",
       "      <td>NaN</td>\n",
       "      <td>前立腺癌の高リスク生化学的再発（BCR）患者を対象としたアンドロゲン遮断療法（ADT）を併用...</td>\n",
       "      <td>前立腺癌</td>\n",
       "      <td>前立腺癌</td>\n",
       "      <td>prostate cancer</td>\n",
       "      <td>BAY 1841788(darolutamide)とADTの24ヵ月間併用投与によって、プラ...</td>\n",
       "      <td>3</td>\n",
       "      <td>NaN</td>\n",
       "      <td>...</td>\n",
       "      <td>treatment purpose</td>\n",
       "      <td>- Histologically or cytologically confirmed ad...</td>\n",
       "      <td>- Pathological finding consistent with small c...</td>\n",
       "      <td>18age old over</td>\n",
       "      <td>No limit</td>\n",
       "      <td>Male</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>Drug: Darolutamide (BAY1841788, Nubeqa) Coated...</td>\n",
       "      <td>[Prostate Cancer]</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>351</th>\n",
       "      <td>jRCT2031230279</td>\n",
       "      <td>NCT05862285</td>\n",
       "      <td>NaN</td>\n",
       "      <td>GENENTECH社及び／又はF.HOFFMANN-LA ROCHE LTDが依頼した試験に...</td>\n",
       "      <td>癌</td>\n",
       "      <td>癌</td>\n",
       "      <td>Cancer</td>\n",
       "      <td>本継続投与試験の目的は,親治験から移行する時点でまだ治験治療を受けており,その地域でその治療...</td>\n",
       "      <td>3</td>\n",
       "      <td>NaN</td>\n",
       "      <td>...</td>\n",
       "      <td>treatment purpose</td>\n",
       "      <td>- Eligible for continuing Roche IMP-based ther...</td>\n",
       "      <td>- Meet any of the study treatment discontinuat...</td>\n",
       "      <td>18age old over</td>\n",
       "      <td>No limit</td>\n",
       "      <td>Both</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>Ipatasertib: Ipatasertib will be administered ...</td>\n",
       "      <td>[Cancer]</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>385</th>\n",
       "      <td>jRCT2031230121</td>\n",
       "      <td>NCT04557449</td>\n",
       "      <td>NaN</td>\n",
       "      <td>進行固形がん患者を対象として，PF-07220060 の単剤投与および併用投与の安全性，忍容...</td>\n",
       "      <td>進行または転移性の乳癌，前立腺癌，およびその他の固形癌</td>\n",
       "      <td>進行または転移性の乳癌,前立腺癌,およびその他の固形癌</td>\n",
       "      <td>advanced or metastatic BC, prostate cancer or ...</td>\n",
       "      <td>パート1(用量漸増,食事の影響,DDIの評価パート) 1A: PF-07220060のMTD...</td>\n",
       "      <td>1</td>\n",
       "      <td>NaN</td>\n",
       "      <td>...</td>\n",
       "      <td>treatment purpose</td>\n",
       "      <td>Disease * Part 1A/1D/1E: HR+HER2-BC, HR+HER2+B...</td>\n",
       "      <td>* Part 1D: participants who have had a gastrec...</td>\n",
       "      <td>18age old over</td>\n",
       "      <td>No limit</td>\n",
       "      <td>Both</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>* Part 1A/1D/1E: PF-07220060 (CDK4 inhibitor) ...</td>\n",
       "      <td>[Advanced, Metastatic Bc, Prostate Cancer, Oth...</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>388</th>\n",
       "      <td>jRCT2031230109</td>\n",
       "      <td>NCT05514054</td>\n",
       "      <td>NaN</td>\n",
       "      <td>EMBER-4：2～5年間の術後内分泌療法による前治療歴を有する再発高リスクのER+、HER...</td>\n",
       "      <td>乳癌</td>\n",
       "      <td>乳癌</td>\n",
       "      <td>Breast Neoplasms</td>\n",
       "      <td>早期乳癌患者を対象としたimlunestrantと標準的な内分泌療法の比較試験</td>\n",
       "      <td>3</td>\n",
       "      <td>NaN</td>\n",
       "      <td>...</td>\n",
       "      <td>treatment purpose</td>\n",
       "      <td>-Have a diagnosis of ER+, HER2- early-stage, r...</td>\n",
       "      <td>-Have any evidence of metastatic disease (incl...</td>\n",
       "      <td>18age old over</td>\n",
       "      <td>No limit</td>\n",
       "      <td>Both</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>-Drug: Imlunestrant Administered orally. Other...</td>\n",
       "      <td>[Breast Neoplasms]</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>391</th>\n",
       "      <td>jRCT2031230090</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>MELK阻害剤OTS167POにおける転移性・進行性乳がん患者を対象とした安全性、忍容性およ...</td>\n",
       "      <td>再発・難治性の局所進行性・転移性乳がん及びトリプルネガティブ乳がん</td>\n",
       "      <td>再発・難治性の局所進行性・転移性乳がん及びトリプルネガティブ乳がん</td>\n",
       "      <td>Relapsed/Refractory Locally Advanced or Metast...</td>\n",
       "      <td>再発・難治性の局所進行性または転移性乳がん患者に対して、OTS167を経口カプセルで投与する...</td>\n",
       "      <td>1</td>\n",
       "      <td>NaN</td>\n",
       "      <td>...</td>\n",
       "      <td>treatment purpose</td>\n",
       "      <td>Dose Escalation and Dose Expansion Cohorts 1. ...</td>\n",
       "      <td>Dose Escalation and Expansion Cohorts 1. Women...</td>\n",
       "      <td>18age old over</td>\n",
       "      <td>No limit</td>\n",
       "      <td>Female</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>This is a Phase I dose escalation/expansion mu...</td>\n",
       "      <td>[Relapsed/Refract, Y Locally Advanced, Metasta...</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>392</th>\n",
       "      <td>jRCT2031230096</td>\n",
       "      <td>NCT05774951</td>\n",
       "      <td>NaN</td>\n",
       "      <td>根治的局所領域療法（化学療法の併用または非併用）および標準補助内分泌療法（ET）を少なくとも...</td>\n",
       "      <td>乳がん、早期乳がん</td>\n",
       "      <td>乳がん、早期乳がん</td>\n",
       "      <td>Breast Cancer, Early Breast Cancer</td>\n",
       "      <td>Treatment</td>\n",
       "      <td>3</td>\n",
       "      <td>NaN</td>\n",
       "      <td>...</td>\n",
       "      <td>treatment purpose</td>\n",
       "      <td>- Women and Men, greater than or equal to 18 y...</td>\n",
       "      <td>- Inoperable locally advanced or metastatic br...</td>\n",
       "      <td>18age old over</td>\n",
       "      <td>No limit</td>\n",
       "      <td>Both</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>arm A: continue with SoC ET as directed by inv...</td>\n",
       "      <td>[Breast Cancer, Early Breast Cancer]</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>402</th>\n",
       "      <td>jRCT2061230009</td>\n",
       "      <td>NCT05485766</td>\n",
       "      <td>NaN</td>\n",
       "      <td>gBRCA1/2遺伝子変異を有するトリプルネガティブ原発乳がんに対するプラチナ製剤、PARP...</td>\n",
       "      <td>トリプルネガティブ乳がん</td>\n",
       "      <td>トリプルネガティブ乳がん</td>\n",
       "      <td>Triple Negative Breast Neoplasms</td>\n",
       "      <td>術前療法としてgBRCA変異陽性手術可能または局所進行TNBCに対してペムブロリズマブ+パク...</td>\n",
       "      <td>2</td>\n",
       "      <td>NaN</td>\n",
       "      <td>...</td>\n",
       "      <td>treatment purpose</td>\n",
       "      <td>1)Male/female subjects who are at least 18 yea...</td>\n",
       "      <td>1) Subjects who has a positive urine pregnancy...</td>\n",
       "      <td>18age old exceed</td>\n",
       "      <td>No limit</td>\n",
       "      <td>Both</td>\n",
       "      <td>NaN</td>\n",
       "      <td>Triple Negative Breast Cancer, Breast Neoplasm...</td>\n",
       "      <td>Drug: Pembrolizumab 200 mg fixed dose, IV, eve...</td>\n",
       "      <td>[Triple Negative, Breast Neoplasms]</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>463</th>\n",
       "      <td>jRCT2061220087</td>\n",
       "      <td>NCT05629585</td>\n",
       "      <td>NaN</td>\n",
       "      <td>術前薬物療法後の外科的切除時に乳房及び／又は腋窩リンパ節に浸潤性残存病変を有するステージI～...</td>\n",
       "      <td>乳癌</td>\n",
       "      <td>乳癌</td>\n",
       "      <td>Breast Cancer</td>\n",
       "      <td>術前薬物療法後の外科的切除時に乳房および/または腋窩リンパ節に浸潤性残存病変を有するI~II...</td>\n",
       "      <td>3</td>\n",
       "      <td>NaN</td>\n",
       "      <td>...</td>\n",
       "      <td>treatment purpose</td>\n",
       "      <td>Participant must be &gt;= 18 years at the time of...</td>\n",
       "      <td>Stage IV (metastatic) TNBC. History of prior i...</td>\n",
       "      <td>18age old over</td>\n",
       "      <td>130age old under</td>\n",
       "      <td>Both</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>Arm 1: Dato-DXd 6 mg/kg IV Q3W x 8 cycles + Du...</td>\n",
       "      <td>[Breast Cancer]</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>515</th>\n",
       "      <td>jRCT2031220276</td>\n",
       "      <td>NCT05307705</td>\n",
       "      <td>NaN</td>\n",
       "      <td>PIK3CA H1047R変異を有する進行乳がん患者及びその他の固形がん患者を対象としたLO...</td>\n",
       "      <td>乳がん</td>\n",
       "      <td>乳がん</td>\n",
       "      <td>Breast Cancer</td>\n",
       "      <td>LOXO-783 の単独投与及び他の抗がん剤との併用投与における 第 2 相試験の推奨用量、...</td>\n",
       "      <td>1</td>\n",
       "      <td>NaN</td>\n",
       "      <td>...</td>\n",
       "      <td>treatment purpose</td>\n",
       "      <td>-Have advanced breast cancer or another solid ...</td>\n",
       "      <td>-Medical Conditions -Colorectal cancer -Endome...</td>\n",
       "      <td>18age old over</td>\n",
       "      <td>No limit</td>\n",
       "      <td>Both</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>-Drug: LOXO-783 Oral Other Name: LY3849524 -Dr...</td>\n",
       "      <td>[Breast Cancer]</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>555</th>\n",
       "      <td>jRCT2031220034</td>\n",
       "      <td>NCT04862663</td>\n",
       "      <td>NaN</td>\n",
       "      <td>局所進行（切除不能）又は転移性のホルモン受容体陽性／ヒト上皮細胞成長因子受容体2 陰性乳癌を...</td>\n",
       "      <td>局所進行（切除不能）又は転移性乳癌</td>\n",
       "      <td>局所進行(切除不能)又は転移性乳癌</td>\n",
       "      <td>Locally Advanced (Inoperable) or Metastatic Br...</td>\n",
       "      <td>Treatment</td>\n",
       "      <td>3</td>\n",
       "      <td>NaN</td>\n",
       "      <td>...</td>\n",
       "      <td>treatment purpose</td>\n",
       "      <td>Key inclusion criteria for both phases: 1. Adu...</td>\n",
       "      <td>Key exclusion criteria for both phases: 1. His...</td>\n",
       "      <td>18age old over</td>\n",
       "      <td>99age old under</td>\n",
       "      <td>Both</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>Capivasertib: PO BD 4 days on /3 days off per ...</td>\n",
       "      <td>[Locally Advanced (Inoperable), Metastatic Bre...</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>590</th>\n",
       "      <td>jRCT2031210585</td>\n",
       "      <td>NCT05123482</td>\n",
       "      <td>NaN</td>\n",
       "      <td>進行又は転移性固形がん患者を対象としたAZD8205の単独療法及び他の抗がん剤との併用療法に...</td>\n",
       "      <td>乳癌、胆道癌、卵巣癌、子宮内膜癌</td>\n",
       "      <td>乳癌、胆道癌、卵巣癌、子宮内膜癌</td>\n",
       "      <td>Breast Cancer, Biliary Tract Carcinoma, Ovaria...</td>\n",
       "      <td>進行又は転移性固形がんに対する治療法候補としての新規化合物AZD8205の研究</td>\n",
       "      <td>1-2</td>\n",
       "      <td>NaN</td>\n",
       "      <td>...</td>\n",
       "      <td>treatment purpose</td>\n",
       "      <td>- Age18 years or more - Relapsed/metastatic so...</td>\n",
       "      <td>- Treatment with any of the following: 1. Nitr...</td>\n",
       "      <td>18age old over</td>\n",
       "      <td>No limit</td>\n",
       "      <td>Both</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>Sub-study1 AZD8205 Monotherapy AZD8205 is an a...</td>\n",
       "      <td>[Breast Cancer, Biliary Tract Carcinoma, Ovari...</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>593</th>\n",
       "      <td>jRCT2031210560</td>\n",
       "      <td>NCT05061823</td>\n",
       "      <td>NaN</td>\n",
       "      <td>複数のBintrafusp alfa（M7824）臨床試験の被験者を対象に長期データを収集す...</td>\n",
       "      <td>固形癌</td>\n",
       "      <td>固形癌</td>\n",
       "      <td>Cancer</td>\n",
       "      <td>本治験は、実施中のbintrafusp alfaの親試験( (NCT02517398, NC...</td>\n",
       "      <td>3</td>\n",
       "      <td>NaN</td>\n",
       "      <td>...</td>\n",
       "      <td>treatment purpose</td>\n",
       "      <td>- Participants who are currently enrolled in a...</td>\n",
       "      <td>- Participants who are pregnant or currently i...</td>\n",
       "      <td>18age old over</td>\n",
       "      <td>No limit</td>\n",
       "      <td>Both</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>Participants who are continuing treatment with...</td>\n",
       "      <td>[Cancer]</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>600</th>\n",
       "      <td>jRCT2041210123</td>\n",
       "      <td>NCT05114746</td>\n",
       "      <td>NaN</td>\n",
       "      <td>進行性PSMA陽性転移性去勢抵抗性前立腺がん（mCRPC）を有する患者を対象に177Lu-P...</td>\n",
       "      <td>前立腺癌</td>\n",
       "      <td>前立腺癌</td>\n",
       "      <td>Prostate Cancer</td>\n",
       "      <td>本治験の目的は、進行性PSMA陽性mCRPCの日本人患者に177Lu-PSMA-617を投与...</td>\n",
       "      <td>2</td>\n",
       "      <td>NaN</td>\n",
       "      <td>...</td>\n",
       "      <td>treatment purpose</td>\n",
       "      <td>- ECOG performance status: 1. Post-taxane popu...</td>\n",
       "      <td>- Previous treatment with any of the following...</td>\n",
       "      <td>20age old over</td>\n",
       "      <td>No limit</td>\n",
       "      <td>Male</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>Radiation: 177Lu-PSMA-617 administered intrave...</td>\n",
       "      <td>[Prostate Cancer]</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>629</th>\n",
       "      <td>jRCT2052210099</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>乳がんを有する成人女性及び健康成人女性を対象とした乳房用マイクロ波画像診断装置IGS-000...</td>\n",
       "      <td>乳がん</td>\n",
       "      <td>乳がん</td>\n",
       "      <td>Breast cancer</td>\n",
       "      <td>本治験の目的は,乳がん又はその疑いのある者,並びに乳がん又はその疑いがない者を対象に乳房画像...</td>\n",
       "      <td>2</td>\n",
       "      <td>NaN</td>\n",
       "      <td>...</td>\n",
       "      <td>screening</td>\n",
       "      <td>Breast cancer (including suspected) (1) One of...</td>\n",
       "      <td>(1) There is trauma with bleeding on the breas...</td>\n",
       "      <td>20age old over</td>\n",
       "      <td>No limit</td>\n",
       "      <td>Female</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>The test using IGS-0001 will be performed twic...</td>\n",
       "      <td>[Breast Cancer]</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>691</th>\n",
       "      <td>jRCT2031200384</td>\n",
       "      <td>NCT04221542</td>\n",
       "      <td>NaN</td>\n",
       "      <td>転移性去勢抵抗性前立腺がん患者を対象としたAMG 509の安全性、忍容性、薬物動態及び有効性...</td>\n",
       "      <td>前立腺がん</td>\n",
       "      <td>前立腺がん</td>\n",
       "      <td>Prostate Cancer</td>\n",
       "      <td>成人被験者のAMG 509の安全性及び忍容性を評価し、最大耐量(MTD)又は第II 相推奨用...</td>\n",
       "      <td>1</td>\n",
       "      <td>NaN</td>\n",
       "      <td>...</td>\n",
       "      <td>treatment purpose</td>\n",
       "      <td>- Parts 1, 2, and 5: Participants with histolo...</td>\n",
       "      <td>- Pathological finding consistent with pure sm...</td>\n",
       "      <td>18age old over</td>\n",
       "      <td>No limit</td>\n",
       "      <td>Male</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>- Experimental: Part 1: AMG 509 Intravenous (I...</td>\n",
       "      <td>[Prostate Cancer]</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>698</th>\n",
       "      <td>jRCT2071200086</td>\n",
       "      <td>NCT04640623</td>\n",
       "      <td>NaN</td>\n",
       "      <td>膀胱全摘除術に不適格である，又は膀胱全摘除術を選択しなかった，カルメット・ゲラン桿菌（BCG...</td>\n",
       "      <td>膀胱癌</td>\n",
       "      <td>膀胱癌</td>\n",
       "      <td>Bladder Cancer</td>\n",
       "      <td>試験の目的は、高グレードTa又はT1乳頭状癌の併発の有無を問わない上皮内癌(CIS)患者にT...</td>\n",
       "      <td>2</td>\n",
       "      <td>NaN</td>\n",
       "      <td>...</td>\n",
       "      <td>treatment purpose</td>\n",
       "      <td>- Histologically confirmed diagnosis of persis...</td>\n",
       "      <td>- Histologically confirmed, muscle-invasive, l...</td>\n",
       "      <td>18age old over</td>\n",
       "      <td>No limit</td>\n",
       "      <td>Both</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>TAR-200 Cohort 1: TAR-200 and Cetrelimab: TAR-...</td>\n",
       "      <td>[Bladder Cancer]</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>729</th>\n",
       "      <td>jRCT2031200057</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>治療抵抗性乳がんを対象としたTDM-812の腫瘍内投与法の安全性評価を目的とした第I相試験</td>\n",
       "      <td>乳がん</td>\n",
       "      <td>乳がん</td>\n",
       "      <td>Breast cancer</td>\n",
       "      <td>TDM-812の腫瘍内投与の安全性および忍容性を評価し、今後の臨床評価に用いる腫瘍内投与にお...</td>\n",
       "      <td>1</td>\n",
       "      <td>NaN</td>\n",
       "      <td>...</td>\n",
       "      <td>treatment purpose</td>\n",
       "      <td>&lt;Disease Characteristics&gt; 1)Patients with inop...</td>\n",
       "      <td>1)Patients correspond to either of the followi...</td>\n",
       "      <td>20age old over</td>\n",
       "      <td>No limit</td>\n",
       "      <td>Female</td>\n",
       "      <td>NaN</td>\n",
       "      <td>Breast cancer</td>\n",
       "      <td>For the target tumor, the dose settled at each...</td>\n",
       "      <td>[Breast Cancer]</td>\n",
       "    </tr>\n",
       "  </tbody>\n",
       "</table>\n",
       "<p>27 rows × 40 columns</p>\n",
       "</div>"
      ],
      "text/plain": [
       "            JRCT ID       NCT No JapicCTI No  \\\n",
       "31   jRCT2061240081  NCT06393374         NaN   \n",
       "32   jRCT2031240476  NCT06312176         NaN   \n",
       "157  jRCT2052240059          NaN         NaN   \n",
       "170  jRCT2031240096  NCT06380751         NaN   \n",
       "209  jRCT2031230750  NCT06188559         NaN   \n",
       "215  jRCT2031230723  NCT06112379         NaN   \n",
       "224  jRCT2051230196  NCT06105632         NaN   \n",
       "234  jRCT2031230677  NCT06206837         NaN   \n",
       "240  jRCT2061230102  NCT06103864         NaN   \n",
       "307  jRCT2061230074  NCT05952557         NaN   \n",
       "342  jRCT2051230094  NCT05794906         NaN   \n",
       "351  jRCT2031230279  NCT05862285         NaN   \n",
       "385  jRCT2031230121  NCT04557449         NaN   \n",
       "388  jRCT2031230109  NCT05514054         NaN   \n",
       "391  jRCT2031230090          NaN         NaN   \n",
       "392  jRCT2031230096  NCT05774951         NaN   \n",
       "402  jRCT2061230009  NCT05485766         NaN   \n",
       "463  jRCT2061220087  NCT05629585         NaN   \n",
       "515  jRCT2031220276  NCT05307705         NaN   \n",
       "555  jRCT2031220034  NCT04862663         NaN   \n",
       "590  jRCT2031210585  NCT05123482         NaN   \n",
       "593  jRCT2031210560  NCT05061823         NaN   \n",
       "600  jRCT2041210123  NCT05114746         NaN   \n",
       "629  jRCT2052210099          NaN         NaN   \n",
       "691  jRCT2031200384  NCT04221542         NaN   \n",
       "698  jRCT2071200086  NCT04640623         NaN   \n",
       "729  jRCT2031200057          NaN         NaN   \n",
       "\n",
       "                                                 Title  \\\n",
       "31   術前薬物療法後の手術時に病理学的完全奏効を達成していないトリプルネガティブ乳癌患者を対象に、...   \n",
       "32   ホルモン受容体陽性・ヒト上皮成長因子受容体2陰性（HR+/HER2-）の切除不能な局所進行又...   \n",
       "157  遺伝子HSD17B4高メチル化を有するHER2陽性ER陰性乳癌における非手術療法の有用性を評...   \n",
       "170  本治験の主要目的は、BRCA1、BRCA2 又は PALB2 変異を有し、HR 陽性、HER...   \n",
       "209  治療歴のある HER2 陽性又は HER2 低発現の切除不能又は転移性乳癌の被験者を対象に，...   \n",
       "215  未治療のトリプルネガティブ又はホルモン受容体低発現／HER2陰性乳癌の成人患者を対象として、...   \n",
       "224  CDK4/6 阻害薬による前治療で進行したホルモン受容体陽性HER2 陰性の進行または転移乳...   \n",
       "234  18歳以上のER陽性／HER2陰性の進行または転移乳癌患者を対象として，経口タンパク質分解誘...   \n",
       "240  Programmed death-ligand（PD-L1）陽性の局所再発手術不能又は転移性...   \n",
       "307  根治的局所治療（化学療法の併用または非併用）を受けて疾患の兆候のない、再発リスクが中間～高リ...   \n",
       "342  前立腺癌の高リスク生化学的再発（BCR）患者を対象としたアンドロゲン遮断療法（ADT）を併用...   \n",
       "351  GENENTECH社及び／又はF.HOFFMANN-LA ROCHE LTDが依頼した試験に...   \n",
       "385  進行固形がん患者を対象として，PF-07220060 の単剤投与および併用投与の安全性，忍容...   \n",
       "388  EMBER-4：2～5年間の術後内分泌療法による前治療歴を有する再発高リスクのER+、HER...   \n",
       "391  MELK阻害剤OTS167POにおける転移性・進行性乳がん患者を対象とした安全性、忍容性およ...   \n",
       "392  根治的局所領域療法（化学療法の併用または非併用）および標準補助内分泌療法（ET）を少なくとも...   \n",
       "402  gBRCA1/2遺伝子変異を有するトリプルネガティブ原発乳がんに対するプラチナ製剤、PARP...   \n",
       "463  術前薬物療法後の外科的切除時に乳房及び／又は腋窩リンパ節に浸潤性残存病変を有するステージI～...   \n",
       "515  PIK3CA H1047R変異を有する進行乳がん患者及びその他の固形がん患者を対象としたLO...   \n",
       "555  局所進行（切除不能）又は転移性のホルモン受容体陽性／ヒト上皮細胞成長因子受容体2 陰性乳癌を...   \n",
       "590  進行又は転移性固形がん患者を対象としたAZD8205の単独療法及び他の抗がん剤との併用療法に...   \n",
       "593  複数のBintrafusp alfa（M7824）臨床試験の被験者を対象に長期データを収集す...   \n",
       "600  進行性PSMA陽性転移性去勢抵抗性前立腺がん（mCRPC）を有する患者を対象に177Lu-P...   \n",
       "629  乳がんを有する成人女性及び健康成人女性を対象とした乳房用マイクロ波画像診断装置IGS-000...   \n",
       "691  転移性去勢抵抗性前立腺がん患者を対象としたAMG 509の安全性、忍容性、薬物動態及び有効性...   \n",
       "698  膀胱全摘除術に不適格である，又は膀胱全摘除術を選択しなかった，カルメット・ゲラン桿菌（BCG...   \n",
       "729      治療抵抗性乳がんを対象としたTDM-812の腫瘍内投与法の安全性評価を目的とした第I相試験   \n",
       "\n",
       "                                         TargetJ  \\\n",
       "31                                   トリプルネガティブ乳癌   \n",
       "32                  HR+/HER2-の切除不能局所進行又は転移性乳癌の患者   \n",
       "157                                          乳がん   \n",
       "170                                         進行乳癌   \n",
       "209               HER2 陽性又は HER2 低発現の切除不能又は転移性乳癌   \n",
       "215                                           乳癌   \n",
       "224  ホルモン受容体（HR）陽性ヒト上皮増殖因子受容体2（HER2）陰性の進行または転移乳癌   \n",
       "234       エストロゲン受容体陽性／ヒト上皮増殖因子受容体 2 陰性の進行または転移乳癌   \n",
       "240                                           乳癌   \n",
       "307                                    乳がん、早期乳がん   \n",
       "342                                         前立腺癌   \n",
       "351                                            癌   \n",
       "385                  進行または転移性の乳癌，前立腺癌，およびその他の固形癌   \n",
       "388                                           乳癌   \n",
       "391            再発・難治性の局所進行性・転移性乳がん及びトリプルネガティブ乳がん   \n",
       "392                                    乳がん、早期乳がん   \n",
       "402                                 トリプルネガティブ乳がん   \n",
       "463                                           乳癌   \n",
       "515                                          乳がん   \n",
       "555                            局所進行（切除不能）又は転移性乳癌   \n",
       "590                             乳癌、胆道癌、卵巣癌、子宮内膜癌   \n",
       "593                                          固形癌   \n",
       "600                                         前立腺癌   \n",
       "629                                          乳がん   \n",
       "691                                        前立腺がん   \n",
       "698                                          膀胱癌   \n",
       "729                                          乳がん   \n",
       "\n",
       "                                          Target  \\\n",
       "31                                   トリプルネガティブ乳癌   \n",
       "32                  HR+/HER2-の切除不能局所進行又は転移性乳癌の患者   \n",
       "157                                          乳がん   \n",
       "170                                         進行乳癌   \n",
       "209               HER2 陽性又は HER2 低発現の切除不能又は転移性乳癌   \n",
       "215                                           乳癌   \n",
       "224  ホルモン受容体(HR)陽性ヒト上皮増殖因子受容体2(HER2)陰性の進行または転移乳癌   \n",
       "234       エストロゲン受容体陽性/ヒト上皮増殖因子受容体 2 陰性の進行または転移乳癌   \n",
       "240                                           乳癌   \n",
       "307                                    乳がん、早期乳がん   \n",
       "342                                         前立腺癌   \n",
       "351                                            癌   \n",
       "385                  進行または転移性の乳癌,前立腺癌,およびその他の固形癌   \n",
       "388                                           乳癌   \n",
       "391            再発・難治性の局所進行性・転移性乳がん及びトリプルネガティブ乳がん   \n",
       "392                                    乳がん、早期乳がん   \n",
       "402                                 トリプルネガティブ乳がん   \n",
       "463                                           乳癌   \n",
       "515                                          乳がん   \n",
       "555                            局所進行(切除不能)又は転移性乳癌   \n",
       "590                             乳癌、胆道癌、卵巣癌、子宮内膜癌   \n",
       "593                                          固形癌   \n",
       "600                                         前立腺癌   \n",
       "629                                          乳がん   \n",
       "691                                        前立腺がん   \n",
       "698                                          膀胱癌   \n",
       "729                                          乳がん   \n",
       "\n",
       "                                         TargetEnglish  \\\n",
       "31                       Triple negative breast cancer   \n",
       "32   HR+/HER2 unresectable locally advanced or meta...   \n",
       "157                                      Breast cancer   \n",
       "170                             Advanced Breast Cancer   \n",
       "209  HER2-positive or HER2-low unresectable or meta...   \n",
       "215                                      Breast Cancer   \n",
       "224  HR-positive, HER2-negative Advanced or Metasta...   \n",
       "234     ER+/HER2- advanced or metastatic breast cancer   \n",
       "240                                      Breast Cancer   \n",
       "307                 Breast Cancer, Early Breast Cancer   \n",
       "342                                    prostate cancer   \n",
       "351                                             Cancer   \n",
       "385  advanced or metastatic BC, prostate cancer or ...   \n",
       "388                                   Breast Neoplasms   \n",
       "391  Relapsed/Refractory Locally Advanced or Metast...   \n",
       "392                 Breast Cancer, Early Breast Cancer   \n",
       "402                   Triple Negative Breast Neoplasms   \n",
       "463                                      Breast Cancer   \n",
       "515                                      Breast Cancer   \n",
       "555  Locally Advanced (Inoperable) or Metastatic Br...   \n",
       "590  Breast Cancer, Biliary Tract Carcinoma, Ovaria...   \n",
       "593                                             Cancer   \n",
       "600                                    Prostate Cancer   \n",
       "629                                      Breast cancer   \n",
       "691                                    Prostate Cancer   \n",
       "698                                     Bladder Cancer   \n",
       "729                                      Breast cancer   \n",
       "\n",
       "                                              研究・治験の目的 試験等のフェーズ 試験の種類  ...  \\\n",
       "31   術前薬物療法後の手術時に病理学的完全奏効を達成していないトリプルネガティブ乳癌患者を対象に、...        3   NaN  ...   \n",
       "32   HR+/HER2-の切除不能な局所進行又は転移性乳癌患者を対象にMK-2870の単剤及びペム...        3   NaN  ...   \n",
       "157  HSD17B4高メチル化(HSD17B4 hypermethylation:HH)を有するH...        2   NaN  ...   \n",
       "170                                          Treatment        3   NaN  ...   \n",
       "209                                                 治療        2   NaN  ...   \n",
       "215  未治療のトリプルネガティブ又はホルモン受容体低発現/HER2陰性乳癌の成人患者を対象として、...        3   NaN  ...   \n",
       "224  CDK4/6阻害薬による前治療で進行したHR陽性HER2陰性の進行または転移乳癌患者を対象に...        3   NaN  ...   \n",
       "234  進行乳癌患者を対象として,vepdegestrantとPF-07220060を併用投与したと...      1-2   NaN  ...   \n",
       "240  PD-L1陽性の局所再発手術不能または転移性TNBC患者を対象に、デュルバルマブ併用または非...        3   NaN  ...   \n",
       "307                                          Treatment        3   NaN  ...   \n",
       "342  BAY 1841788(darolutamide)とADTの24ヵ月間併用投与によって、プラ...        3   NaN  ...   \n",
       "351  本継続投与試験の目的は,親治験から移行する時点でまだ治験治療を受けており,その地域でその治療...        3   NaN  ...   \n",
       "385  パート1(用量漸増,食事の影響,DDIの評価パート) 1A: PF-07220060のMTD...        1   NaN  ...   \n",
       "388            早期乳癌患者を対象としたimlunestrantと標準的な内分泌療法の比較試験        3   NaN  ...   \n",
       "391  再発・難治性の局所進行性または転移性乳がん患者に対して、OTS167を経口カプセルで投与する...        1   NaN  ...   \n",
       "392                                          Treatment        3   NaN  ...   \n",
       "402  術前療法としてgBRCA変異陽性手術可能または局所進行TNBCに対してペムブロリズマブ+パク...        2   NaN  ...   \n",
       "463  術前薬物療法後の外科的切除時に乳房および/または腋窩リンパ節に浸潤性残存病変を有するI~II...        3   NaN  ...   \n",
       "515  LOXO-783 の単独投与及び他の抗がん剤との併用投与における 第 2 相試験の推奨用量、...        1   NaN  ...   \n",
       "555                                          Treatment        3   NaN  ...   \n",
       "590            進行又は転移性固形がんに対する治療法候補としての新規化合物AZD8205の研究      1-2   NaN  ...   \n",
       "593  本治験は、実施中のbintrafusp alfaの親試験( (NCT02517398, NC...        3   NaN  ...   \n",
       "600  本治験の目的は、進行性PSMA陽性mCRPCの日本人患者に177Lu-PSMA-617を投与...        2   NaN  ...   \n",
       "629  本治験の目的は,乳がん又はその疑いのある者,並びに乳がん又はその疑いがない者を対象に乳房画像...        2   NaN  ...   \n",
       "691  成人被験者のAMG 509の安全性及び忍容性を評価し、最大耐量(MTD)又は第II 相推奨用...        1   NaN  ...   \n",
       "698  試験の目的は、高グレードTa又はT1乳頭状癌の併発の有無を問わない上皮内癌(CIS)患者にT...        2   NaN  ...   \n",
       "729  TDM-812の腫瘍内投与の安全性および忍容性を評価し、今後の臨床評価に用いる腫瘍内投与にお...        1   NaN  ...   \n",
       "\n",
       "                purpose                                 Inclusion Criteria  \\\n",
       "31    treatment purpose  -Has centrally confirmed TNBC, as defined by t...   \n",
       "32    treatment purpose  -Has unresectable locally advanced or metastat...   \n",
       "157  diagnostic purpose  1. Histologically confirmed invasive breast ca...   \n",
       "170   treatment purpose  - Adult females, pre/peri-menopausal and/or po...   \n",
       "209   treatment purpose  (1) Male or female, aged >=18 years at the tim...   \n",
       "215   treatment purpose  1. Participant must be >= 18 years at the time...   \n",
       "224   treatment purpose  * Histological confirmation of breast cancer w...   \n",
       "234   treatment purpose  Inclusion Criteria: * Histological or cytologi...   \n",
       "240   treatment purpose  Histologically or cytologically documented loc...   \n",
       "307   treatment purpose  - Women and Men; 18 years or more at the time ...   \n",
       "342   treatment purpose  - Histologically or cytologically confirmed ad...   \n",
       "351   treatment purpose  - Eligible for continuing Roche IMP-based ther...   \n",
       "385   treatment purpose  Disease * Part 1A/1D/1E: HR+HER2-BC, HR+HER2+B...   \n",
       "388   treatment purpose  -Have a diagnosis of ER+, HER2- early-stage, r...   \n",
       "391   treatment purpose  Dose Escalation and Dose Expansion Cohorts 1. ...   \n",
       "392   treatment purpose  - Women and Men, greater than or equal to 18 y...   \n",
       "402   treatment purpose  1)Male/female subjects who are at least 18 yea...   \n",
       "463   treatment purpose  Participant must be >= 18 years at the time of...   \n",
       "515   treatment purpose  -Have advanced breast cancer or another solid ...   \n",
       "555   treatment purpose  Key inclusion criteria for both phases: 1. Adu...   \n",
       "590   treatment purpose  - Age18 years or more - Relapsed/metastatic so...   \n",
       "593   treatment purpose  - Participants who are currently enrolled in a...   \n",
       "600   treatment purpose  - ECOG performance status: 1. Post-taxane popu...   \n",
       "629           screening  Breast cancer (including suspected) (1) One of...   \n",
       "691   treatment purpose  - Parts 1, 2, and 5: Participants with histolo...   \n",
       "698   treatment purpose  - Histologically confirmed diagnosis of persis...   \n",
       "729   treatment purpose  <Disease Characteristics> 1)Patients with inop...   \n",
       "\n",
       "                                    Exclusion Criteria       Age Minimum  \\\n",
       "31   -Has a known germline breast cancer gene (BRCA...    18age old over   \n",
       "32   -Has breast cancer amenable to treatment with ...    18age old over   \n",
       "157  1. History of other malignancy within the last...    20age old over   \n",
       "170  - Participants with history of MDS/AML or with...    18age old over   \n",
       "209  (1) Presence of brain or subdural metastases, ...    18age old over   \n",
       "215  1. As judged by the investigator, any evidence...    18age old over   \n",
       "224  * Any medical or psychiatric condition that ma...    18age old over   \n",
       "234  Exclusion Criteria: * visceral crisis at risk ...    18age old over   \n",
       "240  As judged by investigator, severe or uncontrol...    18age old over   \n",
       "307  - Inoperable locally advanced or metastatic br...    18age old over   \n",
       "342  - Pathological finding consistent with small c...    18age old over   \n",
       "351  - Meet any of the study treatment discontinuat...    18age old over   \n",
       "385  * Part 1D: participants who have had a gastrec...    18age old over   \n",
       "388  -Have any evidence of metastatic disease (incl...    18age old over   \n",
       "391  Dose Escalation and Expansion Cohorts 1. Women...    18age old over   \n",
       "392  - Inoperable locally advanced or metastatic br...    18age old over   \n",
       "402  1) Subjects who has a positive urine pregnancy...  18age old exceed   \n",
       "463  Stage IV (metastatic) TNBC. History of prior i...    18age old over   \n",
       "515  -Medical Conditions -Colorectal cancer -Endome...    18age old over   \n",
       "555  Key exclusion criteria for both phases: 1. His...    18age old over   \n",
       "590  - Treatment with any of the following: 1. Nitr...    18age old over   \n",
       "593  - Participants who are pregnant or currently i...    18age old over   \n",
       "600  - Previous treatment with any of the following...    20age old over   \n",
       "629  (1) There is trauma with bleeding on the breas...    20age old over   \n",
       "691  - Pathological finding consistent with pure sm...    18age old over   \n",
       "698  - Histologically confirmed, muscle-invasive, l...    18age old over   \n",
       "729  1)Patients correspond to either of the followi...    20age old over   \n",
       "\n",
       "          Age Maximum  Gender Discontinuation Criteria  \\\n",
       "31           No limit    Both                      NaN   \n",
       "32           No limit    Both                      NaN   \n",
       "157          No limit  Female                      NaN   \n",
       "170          No limit    Both                      NaN   \n",
       "209          No limit    Both                      NaN   \n",
       "215          No limit    Both                      NaN   \n",
       "224          No limit    Both                      NaN   \n",
       "234          No limit    Both                      NaN   \n",
       "240          No limit    Both                      NaN   \n",
       "307          No limit     NaN                      NaN   \n",
       "342          No limit    Male                      NaN   \n",
       "351          No limit    Both                      NaN   \n",
       "385          No limit    Both                      NaN   \n",
       "388          No limit    Both                      NaN   \n",
       "391          No limit  Female                      NaN   \n",
       "392          No limit    Both                      NaN   \n",
       "402          No limit    Both                      NaN   \n",
       "463  130age old under    Both                      NaN   \n",
       "515          No limit    Both                      NaN   \n",
       "555   99age old under    Both                      NaN   \n",
       "590          No limit    Both                      NaN   \n",
       "593          No limit    Both                      NaN   \n",
       "600          No limit    Male                      NaN   \n",
       "629          No limit  Female                      NaN   \n",
       "691          No limit    Male                      NaN   \n",
       "698          No limit    Both                      NaN   \n",
       "729          No limit  Female                      NaN   \n",
       "\n",
       "                                               Keyword  \\\n",
       "31                                                 NaN   \n",
       "32                                                 NaN   \n",
       "157                                                NaN   \n",
       "170                                                NaN   \n",
       "209                                                NaN   \n",
       "215                                                NaN   \n",
       "224                                                NaN   \n",
       "234                                                NaN   \n",
       "240                                                NaN   \n",
       "307                                                NaN   \n",
       "342                                                NaN   \n",
       "351                                                NaN   \n",
       "385                                                NaN   \n",
       "388                                                NaN   \n",
       "391                                                NaN   \n",
       "392                                                NaN   \n",
       "402  Triple Negative Breast Cancer, Breast Neoplasm...   \n",
       "463                                                NaN   \n",
       "515                                                NaN   \n",
       "555                                                NaN   \n",
       "590                                                NaN   \n",
       "593                                                NaN   \n",
       "600                                                NaN   \n",
       "629                                                NaN   \n",
       "691                                                NaN   \n",
       "698                                                NaN   \n",
       "729                                      Breast cancer   \n",
       "\n",
       "                                       Intervention(s)  \\\n",
       "31   -Arm 1: MK-2870 4mg/kg intravenous (IV) every ...   \n",
       "32   Arm A:MK-2870 Participants receive 4 mg/kg of ...   \n",
       "157  Omitting breast surgery after preoperative che...   \n",
       "170  Experimental: Arm 1: saruparib (AZD5305) plus ...   \n",
       "209  Generic Name:NA Study Treatment in dose optimi...   \n",
       "215  - Experimental arm: Dato-DXd plus durvalumab n...   \n",
       "224  Arm A (Experimental Arm) PF-07220060 plus Fulv...   \n",
       "234  Drug: vepdegestrant Daily oral dosages of vepd...   \n",
       "240  Arm 1: Dato-DXd + durvalumab Arm 2: Investigat...   \n",
       "307  arm A: continue with SoC ET as directed by inv...   \n",
       "342  Drug: Darolutamide (BAY1841788, Nubeqa) Coated...   \n",
       "351  Ipatasertib: Ipatasertib will be administered ...   \n",
       "385  * Part 1A/1D/1E: PF-07220060 (CDK4 inhibitor) ...   \n",
       "388  -Drug: Imlunestrant Administered orally. Other...   \n",
       "391  This is a Phase I dose escalation/expansion mu...   \n",
       "392  arm A: continue with SoC ET as directed by inv...   \n",
       "402  Drug: Pembrolizumab 200 mg fixed dose, IV, eve...   \n",
       "463  Arm 1: Dato-DXd 6 mg/kg IV Q3W x 8 cycles + Du...   \n",
       "515  -Drug: LOXO-783 Oral Other Name: LY3849524 -Dr...   \n",
       "555  Capivasertib: PO BD 4 days on /3 days off per ...   \n",
       "590  Sub-study1 AZD8205 Monotherapy AZD8205 is an a...   \n",
       "593  Participants who are continuing treatment with...   \n",
       "600  Radiation: 177Lu-PSMA-617 administered intrave...   \n",
       "629  The test using IGS-0001 will be performed twic...   \n",
       "691  - Experimental: Part 1: AMG 509 Intravenous (I...   \n",
       "698  TAR-200 Cohort 1: TAR-200 and Cetrelimab: TAR-...   \n",
       "729  For the target tumor, the dose settled at each...   \n",
       "\n",
       "                                            TargetWord  \n",
       "31                    [Triple Negative, Breast Cancer]  \n",
       "32   [Hr+/Her2 Unresectable Locally Advanced, Metas...  \n",
       "157                                    [Breast Cancer]  \n",
       "170                           [Advanced Breast Cancer]  \n",
       "209  [Her2-Positive, Her2-Low Unresectable, Metasta...  \n",
       "215                                    [Breast Cancer]  \n",
       "224  [Hr-Positive, Her2-Negative Advanced, Metastat...  \n",
       "234     [Er+/Her2- Advanced, Metastatic Breast Cancer]  \n",
       "240                                    [Breast Cancer]  \n",
       "307               [Breast Cancer, Early Breast Cancer]  \n",
       "342                                  [Prostate Cancer]  \n",
       "351                                           [Cancer]  \n",
       "385  [Advanced, Metastatic Bc, Prostate Cancer, Oth...  \n",
       "388                                 [Breast Neoplasms]  \n",
       "391  [Relapsed/Refract, Y Locally Advanced, Metasta...  \n",
       "392               [Breast Cancer, Early Breast Cancer]  \n",
       "402                [Triple Negative, Breast Neoplasms]  \n",
       "463                                    [Breast Cancer]  \n",
       "515                                    [Breast Cancer]  \n",
       "555  [Locally Advanced (Inoperable), Metastatic Bre...  \n",
       "590  [Breast Cancer, Biliary Tract Carcinoma, Ovari...  \n",
       "593                                           [Cancer]  \n",
       "600                                  [Prostate Cancer]  \n",
       "629                                    [Breast Cancer]  \n",
       "691                                  [Prostate Cancer]  \n",
       "698                                   [Bladder Cancer]  \n",
       "729                                    [Breast Cancer]  \n",
       "\n",
       "[27 rows x 40 columns]"
      ]
     },
     "execution_count": 16,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "# ターゲットリスト全体を処理\n",
    "matched_indices = []\n",
    "target_vecs_list = []\n",
    "cosine_scores_list = []\n",
    "for idx, target_words in enumerate(basedf['TargetWord']):\n",
    "    # ターゲット内の各単語をベクトル化\n",
    "    target_vecs = model.encode(target_words, convert_to_tensor=True)\n",
    "    # コサイン類似度を計算\n",
    "    cosine_scores = util.cos_sim(query_vec, target_vecs).squeeze()\n",
    "    target_vecs_list.append(target_vecs)\n",
    "    cosine_scores_list.append(cosine_scores)\n",
    "    # 閾値を超えるか確認\n",
    "    if (cosine_scores >= threshold).any():  # いずれかが閾値を超えている場合\n",
    "        matched_indices.append(idx)\n",
    "\n",
    "# 抽出\n",
    "matched_df = basedf.iloc[matched_indices]\n",
    "matched_df"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 92,
   "metadata": {},
   "outputs": [
    {
     "ename": "TypeError",
     "evalue": "list indices must be integers or slices, not list",
     "output_type": "error",
     "traceback": [
      "\u001b[0;31m---------------------------------------------------------------------------\u001b[0m",
      "\u001b[0;31mTypeError\u001b[0m                                 Traceback (most recent call last)",
      "Cell \u001b[0;32mIn[92], line 1\u001b[0m\n\u001b[0;32m----> 1\u001b[0m \u001b[43mcosine_scores_list\u001b[49m\u001b[43m \u001b[49m\u001b[43m[\u001b[49m\u001b[43mmatched_indices\u001b[49m\u001b[43m]\u001b[49m\n",
      "\u001b[0;31mTypeError\u001b[0m: list indices must be integers or slices, not list"
     ]
    }
   ],
   "source": [
    "cosine_scores_list [matched_indices]"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 17,
   "metadata": {},
   "outputs": [
    {
     "data": {
      "text/plain": [
       "[tensor([[-0.1260, -0.7142, -0.4861,  ...,  0.2338,  0.6205,  0.4827],\n",
       "         [-0.0648, -0.5602, -0.7288,  ...,  0.0539,  0.6869,  0.4918]],\n",
       "        device='mps:0'),\n",
       " tensor([[-0.4843, -0.6084, -0.5889,  ...,  0.1928,  0.4419,  0.4955],\n",
       "         [-0.2643, -0.8360, -0.8564,  ...,  0.1610,  0.5643,  0.6620]],\n",
       "        device='mps:0'),\n",
       " tensor([[-6.1750e-01, -4.2829e-01, -2.3934e-01, -9.4116e-01, -3.4548e-01,\n",
       "          -6.5496e-02, -6.3874e-01,  9.7828e-01, -9.5892e-02,  4.0096e-01,\n",
       "           1.4919e-01, -5.1603e-03,  2.0980e-01,  1.3680e-01, -6.9506e-01,\n",
       "           1.6740e-01,  4.6094e-01,  2.1159e-01,  3.5951e-01, -4.4329e-02,\n",
       "           7.9832e-01,  3.4245e-01,  1.8019e-01, -2.5205e-01, -4.5876e-01,\n",
       "           5.8994e-01, -2.4628e-01, -9.2014e-02,  2.0509e-01, -8.1949e-01,\n",
       "          -1.3586e-01, -4.9757e-02, -3.6281e-01,  8.7602e-01,  1.1948e-01,\n",
       "          -2.7310e-01, -1.5276e-01, -2.7996e-01,  2.3494e-01, -6.3637e-02,\n",
       "           3.7814e-01,  2.6592e-02, -2.9347e-01,  3.1386e-01, -1.6421e-01,\n",
       "          -1.4057e-01,  1.4047e-02, -7.8429e-01,  1.3666e-02,  1.8955e-01,\n",
       "           5.8851e-02, -4.7205e-02,  1.9231e-01,  3.7030e-01, -1.0617e+00,\n",
       "          -2.6348e-01,  2.7603e-01,  2.9849e-01, -1.7932e-01, -5.7784e-02,\n",
       "          -1.9989e-01,  9.0201e-01,  1.7949e-01, -2.7822e-01,  1.7824e-01,\n",
       "          -4.4899e-01, -5.4601e-01, -6.0781e-01, -1.0682e-01, -1.1967e-01,\n",
       "          -1.0002e-01,  9.7528e-02,  1.4009e-01, -2.9385e-01,  2.3170e-01,\n",
       "           2.1813e-01,  4.7024e-02, -2.7652e-02, -4.3158e-01, -2.6546e-01,\n",
       "           3.6947e-01,  3.4425e-02, -1.4730e-02,  2.0326e-01, -5.8744e-01,\n",
       "          -2.0226e-01,  3.7959e-01,  1.5703e-01,  2.5914e-01,  2.2662e-03,\n",
       "           5.9967e-01,  1.2024e-02,  8.0339e-02, -4.5369e-01,  5.8427e-01,\n",
       "          -5.4955e-01,  1.0329e-01, -1.6007e-01,  4.9532e-02, -5.2088e-01,\n",
       "          -9.8735e-02,  2.4494e-01, -1.8085e-01,  1.6053e-01, -4.6592e-01,\n",
       "          -3.2983e-02,  2.4514e-01,  3.0468e-01, -4.9183e-01,  4.2685e-02,\n",
       "          -6.5134e-03, -2.9791e-01,  7.3694e-02, -6.6407e-01, -8.5795e-01,\n",
       "           5.8354e-01,  1.8428e-02, -4.5240e-01, -2.3930e-01,  7.4443e-01,\n",
       "           2.9738e-01, -2.3201e-01, -5.3072e-02, -3.5548e-01,  2.0259e-01,\n",
       "          -1.8518e-01,  2.5689e-01, -2.7018e-01,  3.8439e-01,  5.5026e-01,\n",
       "           3.1581e-01,  2.8489e-01, -1.8327e-01,  2.7637e-01,  1.4465e-01,\n",
       "           3.6947e-02, -6.6614e-02, -1.5929e-01,  1.5180e-01,  1.2041e-01,\n",
       "          -2.9580e-01, -3.2999e-01,  5.4746e-03,  1.1182e-01, -3.3279e-01,\n",
       "          -1.0202e-01,  1.8680e-01,  3.4808e-01, -1.4092e-01, -2.5403e-01,\n",
       "          -7.8984e-01, -4.2328e-01,  1.0217e-01, -1.8560e-01, -1.7135e-01,\n",
       "          -4.1911e-01,  3.8121e-01, -6.6660e-02,  4.7638e-01,  3.4189e-01,\n",
       "           6.9978e-01, -1.5311e-01,  5.9776e-02, -5.6412e-01,  3.5197e-01,\n",
       "           3.8270e-02, -6.1110e-01, -1.7247e-01,  1.5680e-01, -4.1813e-04,\n",
       "           6.8195e-01,  7.0979e-02,  4.0124e-01,  1.2070e-01, -3.2553e-01,\n",
       "           8.2089e-03,  1.9890e-01, -1.2864e-01, -1.4297e-01,  7.3766e-01,\n",
       "          -5.5188e-01, -2.3029e+00,  1.0721e-01, -3.5844e-01, -3.6529e-01,\n",
       "          -1.8135e-01,  3.7255e-01,  3.3772e-02,  4.9966e-01,  1.5486e-01,\n",
       "           3.7598e-02,  1.6729e-01, -2.8025e-02,  3.6851e-01, -8.4399e-02,\n",
       "           2.6751e-01,  3.3341e-01,  1.2988e-01, -2.3298e-01, -1.0956e-01,\n",
       "          -3.9227e-01,  2.4031e-01, -9.8916e-02,  8.6010e-01,  2.3260e-01,\n",
       "           2.0943e-01, -4.3657e-03, -4.7262e-02,  4.8964e-01,  4.3099e-01,\n",
       "          -6.6541e-02, -1.6585e-01, -4.9522e-01,  4.6269e-03, -4.1279e-02,\n",
       "          -1.2963e-02, -4.5611e-01,  1.3753e-01,  4.2841e-01, -1.2416e-01,\n",
       "          -1.8354e-01,  1.5525e-01, -4.7715e-02, -3.1517e-01, -2.6492e-02,\n",
       "           6.8453e-01,  1.9080e-01, -2.2994e-01,  1.2377e-01, -1.1896e-01,\n",
       "           3.7006e-02, -4.6975e-01,  7.0044e-01,  6.1147e-01,  1.5585e-01,\n",
       "           1.6214e-02, -6.9943e-01,  2.3699e-01, -3.1574e-01, -4.7647e-03,\n",
       "          -1.5027e-01, -1.5731e-01, -4.2986e-01, -5.7285e-02, -5.5854e-02,\n",
       "           4.0115e-01,  7.9943e-02,  4.6653e-01,  1.1837e+00,  2.7973e-01,\n",
       "           1.9184e-02,  5.6097e-01,  2.0801e-01,  4.2492e-01, -4.0180e-01,\n",
       "           3.3082e-01,  8.8825e-02, -6.3896e-01,  2.7188e-01, -4.1593e-01,\n",
       "           4.4963e-02, -4.1432e-02, -3.7926e-01,  1.4208e-01, -2.1276e-02,\n",
       "           3.4237e-01,  3.3517e-01,  1.5186e-01,  1.6838e-01,  2.9411e-01,\n",
       "          -3.1322e-01,  5.9715e-01, -6.0809e-02, -1.1127e-01, -6.3492e-01,\n",
       "          -7.6732e-01,  9.5206e-02,  4.4743e-01, -1.2880e-01, -6.1279e-01,\n",
       "           1.0462e-01, -2.3866e-01, -1.2257e-01, -3.5064e-02,  2.1191e-01,\n",
       "           2.7455e-01,  5.0687e-02, -9.5436e-02,  6.4791e-02, -4.6950e-01,\n",
       "          -4.9277e-01,  4.9140e-02, -1.5758e-01, -1.3063e-01,  5.3918e-01,\n",
       "          -7.9415e-02,  1.1575e-01,  5.7221e-01,  1.9799e-02, -4.5976e-01,\n",
       "          -2.5640e-01,  1.4802e-01,  2.8667e-01, -7.2082e-02,  2.1981e-01,\n",
       "           3.4119e-01,  2.6467e-02,  5.1649e-01,  1.4823e-01, -2.1046e-01,\n",
       "          -2.7239e-01,  5.1173e-01, -1.5755e-01,  6.1403e-02, -1.2527e-01,\n",
       "          -8.7207e-01, -5.6150e-02, -7.8413e-02, -1.6833e-01, -5.4440e-01,\n",
       "           2.6566e-01,  1.4351e-01,  1.0561e-01,  2.2350e-01, -1.5984e-01,\n",
       "           7.0123e-01,  6.5285e-01, -3.3100e-01, -1.6943e-01, -4.0104e-01,\n",
       "           9.3922e-02, -6.8374e-02, -2.9329e-01, -1.4009e-02,  9.5368e-03,\n",
       "          -2.2765e-01, -1.2516e-01, -1.0805e-01,  2.8921e-01, -1.8876e-01,\n",
       "           5.2862e-01,  1.2189e-01,  8.3649e-02,  4.3873e-01,  2.0612e-02,\n",
       "           1.3071e-01,  3.6515e-01, -4.2117e-01,  1.2870e-01,  4.0424e-01,\n",
       "           4.0578e-01, -1.0551e+00, -8.3291e-01, -1.4936e-01,  3.6111e-01,\n",
       "           2.2860e-01,  2.5189e-01,  1.6999e-02, -7.2215e-03,  8.9455e-02,\n",
       "          -2.3402e-01,  3.5135e-01, -6.6415e-02, -5.7412e-01,  3.0561e-01,\n",
       "           2.3567e-01,  5.3263e-01, -2.7377e-01, -5.0579e-01, -6.5233e-02,\n",
       "           2.4616e-01, -3.6381e-01,  1.1392e-01,  2.3480e-01,  4.5466e-01,\n",
       "          -3.6015e-01, -5.6164e-01, -7.0282e-02, -7.3986e-01, -1.3578e-01,\n",
       "           3.2483e-01, -2.1437e-01,  2.7178e-02, -2.4758e-01,  1.5231e-01,\n",
       "          -1.7925e-02, -1.8879e-01,  6.0757e-01,  2.0522e-01, -2.5855e-01,\n",
       "           1.1981e-01, -2.3892e-01, -1.9475e-01,  5.0091e-01, -5.0226e-01,\n",
       "          -3.0348e-01, -5.0960e-02, -1.5349e-01, -3.2418e-01,  3.2181e-01,\n",
       "          -4.1389e-01, -1.4861e-01,  1.6748e-01, -6.4736e-02, -1.3068e-01,\n",
       "          -3.5233e-01,  1.4268e-01, -1.2707e-01, -4.9883e-02, -1.7654e-02,\n",
       "          -1.3353e-01,  6.0900e-02,  1.3308e-01,  6.4538e-02,  1.4388e-01,\n",
       "           5.3696e-01, -1.1464e-01,  2.7582e-03, -2.5900e-01, -2.2469e-01,\n",
       "           5.2498e-01, -9.4973e-02, -3.3117e-01,  5.4928e-01, -1.1156e+01,\n",
       "           1.0386e-01,  7.2631e-01, -5.5808e-01,  5.5128e-02, -2.8539e-01,\n",
       "           2.1085e-01, -3.0009e-01,  7.9101e-02,  2.4967e-01, -2.8460e-01,\n",
       "          -1.2277e-01, -2.7364e-01, -2.3193e-01, -4.0872e-01, -3.9742e-01,\n",
       "          -4.7229e-01, -5.6789e-01,  1.1772e-01, -1.7663e-01, -1.0996e-01,\n",
       "          -5.7036e-01, -1.9379e-01,  2.9696e-01, -4.9559e-03, -5.1069e-01,\n",
       "           5.9541e-02, -2.2948e-01,  2.6399e-01,  4.6126e-02,  2.2045e-01,\n",
       "          -1.9949e-02,  1.4108e-01,  2.0669e-01, -4.4946e-02, -4.5100e-01,\n",
       "          -4.4853e-01,  2.2129e-01, -1.5955e-01, -2.2267e-01, -5.4017e-02,\n",
       "          -3.3838e-01, -2.7689e-01,  1.3447e-01,  3.7594e-01,  6.2596e-01,\n",
       "           1.1595e+00,  5.6208e-01,  2.9245e-01, -1.0625e-01, -8.5438e-02,\n",
       "          -1.0762e+00,  5.5900e-01, -3.7318e-01,  8.1787e-02, -2.5119e-01,\n",
       "           6.6201e-01, -4.9906e-01,  5.4685e-01, -4.0754e-01, -2.5500e-01,\n",
       "           3.7179e-01,  3.0194e-01,  7.1104e-03,  2.8239e-02, -2.7432e-01,\n",
       "           4.0257e-01, -2.6094e-01, -3.6915e-02, -3.7819e-01,  4.0081e-01,\n",
       "          -3.9472e-01,  5.0770e-01, -5.6534e-02,  1.2945e-01,  1.8567e-01,\n",
       "          -3.5001e-01,  1.6610e-01,  1.0337e+00,  1.7414e-01,  4.1071e-01,\n",
       "           1.7420e-01,  5.1160e-01,  5.3064e-02, -8.8958e-01,  1.3691e-01,\n",
       "           2.0502e-01,  2.0789e-01, -6.7096e-02,  5.7131e-01, -2.5517e-02,\n",
       "          -6.5181e-01, -2.3094e-01, -3.2169e-01,  4.3108e-01,  6.5352e-01,\n",
       "          -2.2597e-01, -4.4343e-01,  2.3894e-01, -7.9776e-03, -5.3395e-01,\n",
       "           9.0074e-02, -2.2320e-01, -5.2792e-01,  8.7275e-02, -5.8914e-01,\n",
       "           1.4985e-01, -1.2428e-01,  5.4238e-02,  6.6887e-01, -4.0159e-01,\n",
       "          -1.6113e-01,  3.6602e-02, -4.4532e-02, -4.1270e-01, -4.9520e-01,\n",
       "          -7.7083e-02,  4.1073e-01, -1.9776e-01, -2.7976e-01,  7.0636e-02,\n",
       "          -3.0673e-01,  2.2584e-01, -1.7377e-01, -6.4935e-01,  3.2879e-01,\n",
       "          -6.0415e-01,  2.0578e-01,  2.5360e-01, -2.2507e-01,  4.9017e-01,\n",
       "           1.1337e-01, -8.1817e-02,  1.9397e-01, -9.9369e-02, -7.8554e-01,\n",
       "          -3.2573e-01, -2.7299e-02,  3.2662e-01, -4.2844e-01,  1.5776e-01,\n",
       "           1.5022e-01, -4.5752e-02, -2.2574e-01,  2.7904e-01,  1.0174e-01,\n",
       "           3.7762e-01,  3.9846e-01, -5.5006e-01,  1.1727e-01, -2.9867e-01,\n",
       "          -2.9262e-02, -6.1174e-02,  1.7952e-01,  2.3654e-01, -7.0764e-01,\n",
       "          -1.5224e-01, -2.6434e-01, -3.0965e-01, -2.1650e-01, -2.5120e-01,\n",
       "           2.0680e-01,  1.4038e-01, -1.5897e-02, -1.1308e-01,  5.8077e-01,\n",
       "           4.0814e-02, -1.1167e-01, -4.5997e-01,  3.1443e-01,  4.3178e-01,\n",
       "           2.1794e-01,  7.5875e-02, -1.2830e-01, -5.5532e-01, -1.1922e-02,\n",
       "          -4.3644e-01,  9.1081e-01, -7.3906e-01, -7.1970e-01,  6.6086e-02,\n",
       "           1.4951e-02, -8.8259e-04, -2.3450e-01, -5.0265e-01,  8.1298e-02,\n",
       "          -2.1017e-01, -1.4891e-01,  6.1550e-03, -1.0607e-02,  1.6299e-01,\n",
       "           4.0742e-01,  3.6838e-01,  1.5951e-01, -3.3938e-01, -4.4081e-01,\n",
       "          -6.0244e-01,  1.4567e-02, -1.6795e-01,  2.9330e-01, -2.6752e-01,\n",
       "          -3.2661e-01,  1.6043e-02, -9.1892e-03, -2.2272e-01,  5.2876e-01,\n",
       "           1.2427e-01,  1.6396e+00, -3.7287e-01, -4.4551e-01,  2.9018e-01,\n",
       "           7.2306e-02, -2.7073e-02, -2.8130e-01,  2.5074e-01,  9.5447e-02,\n",
       "           4.0194e-02, -3.6304e-01,  1.5023e-02, -2.0558e-02,  3.2773e-01,\n",
       "           3.2056e-01, -1.0703e+00, -2.0238e-01, -1.7993e-01,  5.5664e-01,\n",
       "           3.2388e-02,  6.3185e-01, -2.2305e-01,  3.0577e-01,  4.6106e-01,\n",
       "          -8.6667e-01, -2.3000e-01, -1.6129e-01,  2.7295e-01, -1.0583e-01,\n",
       "          -9.8735e-02, -1.6178e-01, -3.7165e-02, -1.6936e-01,  3.9538e-01,\n",
       "           7.4751e-01, -7.8426e-01,  7.4081e-01,  7.0467e-01, -9.8234e-02,\n",
       "           5.6815e-01,  9.4187e-02,  3.2641e-01, -1.4659e-01,  2.2669e-02,\n",
       "           2.6200e-01,  1.8542e-01, -6.3165e-01, -2.3962e-01,  2.4473e-01,\n",
       "           4.0158e-01, -1.1943e-01,  1.2108e-01,  7.9292e-02,  3.0960e-02,\n",
       "          -3.7691e-01, -1.0145e-01, -2.4552e-01, -3.9904e-01,  1.1356e-01,\n",
       "           1.4603e-02, -1.5001e-01, -8.6997e-02,  4.7709e-01, -1.7580e-01,\n",
       "           7.5067e-01,  4.2910e-01,  3.7145e-01, -3.5103e-01,  8.9950e-04,\n",
       "          -5.7565e-02,  6.0671e-01, -3.1670e-01,  7.7385e-01,  6.3337e-02,\n",
       "          -2.3140e-01,  2.7251e-01,  6.0547e-03,  1.7829e-01,  2.9892e-01,\n",
       "          -2.3519e-01,  1.2702e-01, -3.0718e-01, -2.3711e-01,  2.8330e-01,\n",
       "          -1.8640e-01,  1.1542e-01, -7.2270e-01, -6.3602e-03,  3.9857e-01,\n",
       "          -3.2051e-01,  4.9873e-01, -2.3074e-01,  6.1026e-01, -4.3509e-01,\n",
       "           3.2669e-02, -4.7470e-01, -4.3275e-01, -2.8508e-01,  5.1854e-02,\n",
       "          -8.6027e-01, -1.9453e-01,  1.7684e-01,  3.7405e-02,  2.1707e-01,\n",
       "           7.5155e-01, -3.3794e-01, -2.4771e-01, -2.0386e-01,  4.0190e-02,\n",
       "           1.8299e-01, -8.2071e-02, -3.5586e-01,  2.1161e-01, -3.5607e-01,\n",
       "          -1.0520e+00, -2.3117e-01,  8.9990e-02,  3.8419e-01, -3.5853e-01,\n",
       "           1.1570e+00,  2.5132e-02, -2.0226e-01,  6.7253e-01,  2.0808e-01,\n",
       "           6.4157e-01,  6.0323e-01,  1.3628e-03, -4.1259e-01, -5.4109e-02,\n",
       "          -1.5877e-01, -3.2881e-01,  1.5757e-01, -1.8346e-01,  5.0556e-01,\n",
       "           5.4802e-01, -2.0806e-02,  5.4756e-01]], device='mps:0'),\n",
       " tensor([[-0.2330, -0.4411, -0.4565,  ...,  0.0257,  0.5397,  0.4576],\n",
       "         [-0.2034, -0.5982, -0.6031,  ...,  0.0101,  0.6477,  0.4153]],\n",
       "        device='mps:0'),\n",
       " tensor([[-0.2839, -0.7334, -0.5583,  ...,  0.0945,  0.5009,  0.5019],\n",
       "         [-0.2512, -0.7003, -0.5509,  ...,  0.0127,  0.2749,  0.4704]],\n",
       "        device='mps:0'),\n",
       " tensor([[ 2.0972e-01, -7.0529e-01, -3.3110e-01, -5.2878e-01, -1.6621e-01,\n",
       "          -3.0437e-01, -3.9961e-01, -8.0757e-02,  2.1818e-03,  4.2855e-01,\n",
       "           1.4417e-01,  6.6552e-02, -3.1631e-02, -4.9837e-01, -8.3322e-01,\n",
       "          -2.6594e-01,  8.7027e-02,  4.7867e-01, -1.1181e-02,  7.8806e-01,\n",
       "           4.0823e-01,  1.8596e-01,  1.3284e-01,  4.7004e-02, -9.0578e-01,\n",
       "           1.7801e-01,  1.4466e-01,  2.4854e-01,  3.6524e-01, -7.8644e-02,\n",
       "          -6.9674e-02,  4.6506e-01, -5.5918e-01,  8.5882e-01, -1.2238e-01,\n",
       "          -1.7314e-01,  7.8814e-02, -3.8327e-01,  6.1792e-01,  4.5555e-03,\n",
       "           6.4068e-01,  5.9587e-01,  2.5859e-02, -8.0803e-02, -7.1639e-02,\n",
       "          -2.5329e-01, -4.0721e-01, -3.0355e-01, -3.1912e-01, -1.5915e-01,\n",
       "          -3.7934e-01, -2.5916e-01,  5.5773e-01,  1.4994e-01, -7.2842e-01,\n",
       "          -5.1068e-01, -1.1576e-01,  2.5061e-01, -1.2123e-01, -1.4712e-01,\n",
       "          -6.8262e-03,  3.4523e-01, -2.2683e-01,  2.7532e-01,  1.6315e-01,\n",
       "           3.2079e-01, -6.5868e-01, -8.1255e-01, -4.6365e-01,  1.7713e-01,\n",
       "           2.6524e-01,  2.2580e-01,  2.4552e-01,  3.0190e-02,  1.5116e-01,\n",
       "          -5.0478e-01,  2.1706e-01, -9.5926e-02, -5.9165e-01,  3.7624e-01,\n",
       "           6.6430e-02,  3.5996e-01, -1.7392e-01, -4.1417e-01,  3.9086e-01,\n",
       "          -1.8943e-01,  7.0977e-01, -9.0007e-02,  1.3985e-01, -1.1159e-01,\n",
       "           5.1320e-02, -1.1486e-01, -5.7764e-01, -4.6208e-01,  3.9386e-01,\n",
       "          -2.4277e-01,  1.8547e-02, -5.1043e-01, -1.5805e-01,  4.4308e-02,\n",
       "          -1.3199e-02,  1.5336e-01,  2.0327e-01,  4.3547e-01, -7.1462e-01,\n",
       "          -8.7322e-01,  9.9392e-02, -2.8265e-01, -7.2167e-01, -1.6001e-01,\n",
       "           6.2922e-02, -9.0481e-02,  4.7913e-02, -4.2210e-01, -6.6721e-02,\n",
       "           7.6288e-02,  2.5745e-01,  6.4695e-03, -2.8692e-02,  5.1667e-01,\n",
       "           2.7214e-01, -3.9246e-01,  2.3820e-01,  2.7097e-01, -8.8920e-02,\n",
       "          -7.6922e-02,  2.9875e-02, -2.4900e-01,  4.7170e-01,  4.2458e-01,\n",
       "          -7.9423e-02,  3.2631e-01, -1.5202e-01,  3.5193e-02, -3.0143e-02,\n",
       "           1.6530e-01,  5.4502e-01, -6.1917e-01,  3.3977e-01,  3.2973e-01,\n",
       "          -5.2898e-02, -9.3560e-03, -5.7116e-02, -1.5098e-01, -2.4331e-01,\n",
       "          -3.3420e-01,  1.7835e-01, -2.0305e-02,  1.0977e-01, -2.9185e-02,\n",
       "          -1.0098e+00, -7.4574e-01,  5.2415e-01, -1.6591e-01,  7.0199e-02,\n",
       "          -2.0178e-01,  6.9219e-01, -1.8301e-02, -5.1685e-01,  2.6321e-01,\n",
       "           6.2996e-01,  2.7620e-01,  3.5459e-01, -2.2095e-01,  4.2289e-02,\n",
       "           1.9981e-01,  3.7540e-01, -1.4578e-01,  8.7146e-03,  1.9156e-01,\n",
       "          -1.3175e-01, -2.5092e-01,  3.6083e-01, -1.5721e-01,  4.2581e-01,\n",
       "          -3.9400e-01,  2.2267e-01,  6.4155e-02, -3.0225e-01,  1.1343e-01,\n",
       "          -3.1985e-01, -1.0331e+00,  8.8725e-02, -5.7983e-01, -1.9259e-01,\n",
       "          -5.1127e-01,  3.5784e-01,  2.1798e-02,  2.6415e-01,  8.3662e-02,\n",
       "          -3.2261e-02,  4.4646e-01,  2.4876e-01,  2.3577e-01,  2.1502e-01,\n",
       "           1.1136e-01,  2.9488e-01,  3.2147e-01, -3.2246e-01,  9.4117e-02,\n",
       "          -7.1115e-01,  1.7365e-01,  2.1753e-01,  9.1414e-02,  1.9105e-02,\n",
       "           2.9200e-01, -6.3047e-02,  3.1459e-01,  1.2206e-01, -3.7295e-01,\n",
       "          -1.8798e-01, -2.2211e-01, -5.3811e-01, -2.7432e-01, -7.0177e-02,\n",
       "           4.7717e-01, -2.3528e-01,  1.9622e-02,  6.3332e-01,  5.0235e-02,\n",
       "          -1.0954e-01,  9.2971e-03,  9.0613e-02, -2.2507e-01, -1.1211e-02,\n",
       "           1.1896e-01, -1.6094e-01,  1.9216e-01, -3.7093e-01, -2.1987e-01,\n",
       "           6.4922e-01, -1.1753e-01,  7.4672e-01,  3.9565e-01, -4.9886e-02,\n",
       "          -2.0584e-01, -5.0605e-01,  2.3996e-01, -1.6363e-01, -7.7218e-02,\n",
       "          -3.4819e-01, -5.1441e-01, -2.8390e-01,  6.2091e-03,  1.3994e-01,\n",
       "           2.7951e-01, -9.5383e-02,  6.6054e-01,  4.8782e-01,  2.8877e-01,\n",
       "           1.8974e-01, -3.8515e-02,  3.0820e-01, -9.3956e-02, -4.4895e-01,\n",
       "          -2.7563e-01,  8.7065e-02, -6.1929e-01,  3.0097e-01, -3.2161e-01,\n",
       "          -2.2575e-01,  1.0745e-01, -9.5288e-02,  1.7359e-01, -2.8505e-01,\n",
       "           6.7464e-02,  4.3732e-01, -2.5137e-01, -3.3279e-01,  6.2822e-01,\n",
       "          -1.8165e-01,  4.5750e-01,  9.3487e-02, -6.0304e-01, -4.2069e-01,\n",
       "          -9.0815e-01,  3.4757e-01, -1.0545e-01,  1.2053e-02, -9.7710e-01,\n",
       "           7.8483e-02, -1.8584e-02,  1.2894e-01,  6.6227e-02, -1.3237e-01,\n",
       "           9.5549e-02,  2.2247e-01, -1.7537e-01, -3.1228e-01, -2.2662e-01,\n",
       "          -1.8137e-01,  1.2421e-01, -5.2104e-01,  2.5660e-01,  1.4212e-01,\n",
       "          -2.8165e-01,  2.1041e-01,  7.9427e-02, -2.8818e-01, -4.3592e-01,\n",
       "          -3.3109e-02, -1.0383e-01, -2.4781e-02,  8.4363e-02,  2.4129e-01,\n",
       "          -1.7132e-01,  3.3873e-01,  4.0942e-01, -4.7567e-01, -6.2251e-01,\n",
       "          -1.3890e-02,  6.8539e-01, -4.8763e-01, -6.9160e-01, -7.7904e-02,\n",
       "          -6.3890e-01,  4.8867e-02, -1.3534e-01, -9.9755e-02, -2.3977e-01,\n",
       "           1.7491e-01,  2.8601e-01,  9.2964e-02, -2.6267e-01, -8.9631e-02,\n",
       "           4.7405e-01,  7.3027e-01,  5.7866e-02, -1.0961e-02, -3.5887e-01,\n",
       "          -3.8540e-01, -2.9773e-01, -2.7849e-01, -7.6125e-01,  1.8886e-01,\n",
       "           9.3885e-02,  3.8784e-01,  1.6988e-01, -1.4772e-01, -3.2467e-01,\n",
       "           2.8299e-01, -4.4083e-01,  3.3340e-01,  3.3183e-01, -1.5024e-01,\n",
       "          -3.4911e-02,  2.5905e-01, -2.0019e-01, -2.9142e-01,  4.6257e-01,\n",
       "           6.0348e-01,  6.5144e-01, -3.8473e-01, -1.2540e-01,  3.5216e-01,\n",
       "           7.5292e-01, -1.0902e-01, -4.2839e-01,  5.3126e-01,  3.4355e-01,\n",
       "          -1.4022e-02,  2.3008e-01, -2.7344e-01, -9.9128e-02, -1.2134e-01,\n",
       "           1.8070e-01,  3.6711e-01, -3.5057e-01,  2.2257e-01, -9.6432e-02,\n",
       "          -1.5594e-01, -3.3102e-01, -1.8986e-01,  3.4150e-02,  7.2909e-01,\n",
       "          -6.8457e-02, -3.3301e-01,  6.3493e-01, -4.9171e-01, -7.2394e-02,\n",
       "          -5.6280e-03,  6.0677e-02, -3.5679e-02, -2.4571e-01, -4.5428e-01,\n",
       "          -4.0136e-01,  3.3425e-01,  8.5540e-01,  7.5585e-02, -6.5626e-02,\n",
       "           7.4385e-02, -2.8545e-02, -3.3322e-01,  5.2281e-01, -2.8884e-01,\n",
       "          -1.5549e-01,  2.8854e-01, -1.4161e-01,  1.0251e-01,  3.4239e-01,\n",
       "          -1.9153e-01,  1.7474e-02,  7.6514e-02, -2.5458e-02, -5.0728e-02,\n",
       "          -1.4719e-01, -5.5459e-02,  2.4034e-01, -5.1834e-02,  1.2937e-01,\n",
       "           6.2167e-01, -6.9003e-02, -1.9656e-01,  2.7261e-02, -6.6939e-02,\n",
       "           1.3318e-01, -3.5410e-01,  3.5835e-01, -1.2065e-01, -7.1993e-01,\n",
       "           7.9583e-01,  3.5395e-01,  3.6960e-01,  3.7927e-01, -1.1226e+01,\n",
       "           3.2127e-01,  4.4304e-01, -4.5927e-01, -1.2128e-01,  1.3327e-01,\n",
       "           2.4023e-02, -2.1125e-01,  1.8002e-01, -4.5764e-01,  3.9144e-02,\n",
       "          -4.3645e-01,  4.2428e-02,  1.0673e-01, -6.7005e-01, -1.3491e-01,\n",
       "           3.7732e-02, -8.2283e-01, -2.8878e-02,  4.4365e-02,  2.3819e-01,\n",
       "           1.1901e-01, -1.7147e-01,  6.8519e-02,  9.2784e-02, -2.3301e-01,\n",
       "           1.0584e-01, -8.4770e-02,  4.1060e-01,  1.9352e-01, -4.2459e-02,\n",
       "           5.9244e-02,  2.3006e-03, -7.7106e-02, -1.4286e-01, -5.4043e-01,\n",
       "           2.6263e-01,  3.2198e-01,  3.6035e-01,  1.5501e-01, -4.3664e-01,\n",
       "          -3.2591e-01, -1.5101e-01,  6.7418e-02,  4.6245e-01,  7.4016e-01,\n",
       "           9.0839e-01,  1.2240e+00,  4.0461e-01,  1.2912e-01, -5.2427e-01,\n",
       "          -9.4191e-01,  1.6540e-01, -5.2428e-01, -3.9597e-02, -1.5577e-01,\n",
       "           3.7424e-01, -6.3065e-01,  2.5987e-01, -4.0388e-01, -3.0071e-01,\n",
       "           4.3752e-01,  4.1406e-01,  3.8406e-02, -7.2177e-01, -2.4935e-01,\n",
       "           1.0612e-01,  1.3210e-01,  3.2934e-01, -2.3812e-01,  3.4319e-01,\n",
       "          -3.9774e-02,  3.7486e-01,  3.5411e-01, -4.7508e-01,  7.4228e-02,\n",
       "          -1.7236e-01, -1.1381e-01,  2.3639e+00, -1.6652e-02,  6.7754e-02,\n",
       "           2.1895e-01,  2.8834e-01, -7.6355e-01, -5.6113e-01, -1.6440e-01,\n",
       "           5.5703e-01, -5.6724e-02,  6.4883e-01,  3.5115e-01, -1.4751e-01,\n",
       "          -7.9965e-01,  2.1881e-01,  1.2099e-01,  3.3622e-01,  8.8443e-01,\n",
       "          -6.8296e-01, -3.7779e-01,  2.6321e-01,  4.2990e-01, -5.3446e-01,\n",
       "           3.9974e-03, -1.6391e-01, -2.5054e-01,  5.0749e-01, -1.0957e-01,\n",
       "           6.4377e-03, -1.0581e-01, -1.0388e-01,  2.9109e-01, -2.0991e-01,\n",
       "           3.2127e-01,  8.9717e-02,  5.9031e-02, -3.2271e-01, -3.0944e-01,\n",
       "           3.7983e-01,  1.0723e-01, -1.8961e-01, -5.5256e-01, -5.2849e-01,\n",
       "           4.8736e-01,  8.9276e-02,  2.7268e-01, -3.8981e-01,  5.5949e-01,\n",
       "          -4.6660e-01,  8.9849e-02,  4.6095e-02, -1.9025e-01,  1.3537e-01,\n",
       "           1.5886e-01,  3.1032e-02,  2.4252e-01, -5.5735e-01, -5.3326e-01,\n",
       "          -6.2157e-01,  1.3673e-01,  5.9913e-02, -7.2839e-01, -2.4078e-01,\n",
       "           8.6503e-03, -1.7342e-01, -2.9535e-01,  1.1929e-01, -7.0389e-02,\n",
       "           4.0934e-01,  1.3370e-01, -2.9600e-01,  4.2876e-01, -1.2138e-01,\n",
       "          -2.2217e-01,  1.1965e-01, -2.4771e-01,  2.7823e-01, -2.7950e-01,\n",
       "          -7.6121e-01, -3.7444e-01, -2.0078e-01,  2.4572e-01, -4.4340e-02,\n",
       "          -1.1244e-01,  1.3798e-02, -3.7875e-01,  6.3814e-01,  2.0263e-01,\n",
       "           2.5656e-01, -3.8560e-01,  1.3939e-01, -3.3343e-01, -9.0675e-02,\n",
       "           2.0255e-01,  3.5437e-01,  3.7570e-02, -2.1193e-01, -3.6009e-02,\n",
       "          -2.0876e-01,  1.2269e+00, -6.9662e-02, -4.4425e-01,  3.6663e-01,\n",
       "           3.8993e-01, -4.5317e-01,  3.0175e-01, -1.6764e-01,  4.2986e-02,\n",
       "          -3.3485e-02, -8.3087e-01,  2.9591e-01,  3.6022e-02,  2.7230e-01,\n",
       "           2.3092e-01,  4.7917e-02,  1.7008e-01,  2.6093e-01,  6.8821e-02,\n",
       "          -2.8306e-01, -1.1844e-01, -1.9740e-01, -1.0163e-01,  4.4790e-01,\n",
       "          -3.9521e-01,  1.7099e-01,  2.4826e-01, -1.0075e-01, -1.2352e-01,\n",
       "          -1.6472e-02,  9.7656e-01, -5.4410e-01, -2.1135e-01,  2.1996e-01,\n",
       "           8.9735e-02,  2.1825e-01,  3.8245e-01,  6.2827e-02,  7.7025e-02,\n",
       "           2.6363e-01,  1.4415e-02, -5.6465e-01,  5.0703e-02,  1.0461e-01,\n",
       "           3.3129e-01, -5.6308e-01, -7.9202e-01, -7.5339e-02,  1.1135e-01,\n",
       "          -5.2298e-01,  7.5735e-02, -4.0391e-01,  5.9075e-01,  2.0149e-01,\n",
       "          -8.2802e-01,  1.6371e-01, -7.1217e-01,  3.0165e-01, -1.8019e-02,\n",
       "          -3.3532e-01,  6.4956e-02, -4.1768e-01,  4.5480e-02,  2.6043e-01,\n",
       "           1.6325e-01, -1.6200e+00,  1.0498e-01,  5.3174e-01, -3.8898e-01,\n",
       "           2.7227e-01, -8.6638e-02,  3.0943e-01,  2.3013e-01, -1.4659e-01,\n",
       "          -3.0452e-02,  8.2237e-02, -6.2003e-01, -9.1796e-01,  7.1247e-02,\n",
       "           4.3417e-01,  1.9536e-02, -3.0194e-01, -5.8090e-02, -2.6994e-01,\n",
       "           1.5639e-01, -1.9183e-01,  1.0028e-01,  1.0232e-02, -4.3359e-02,\n",
       "           9.6693e-02, -6.2048e-02,  1.5903e-02,  2.6562e-01, -3.9515e-01,\n",
       "           4.4779e-01,  7.1628e-01,  8.3268e-01, -1.4977e-01, -6.0242e-02,\n",
       "           3.3866e-02,  1.4313e-01, -2.9810e-02,  5.5056e-01, -9.6904e-02,\n",
       "          -4.4854e-01,  2.7790e-01,  2.4515e-01, -2.0167e-01,  3.2183e-02,\n",
       "          -3.9025e-01,  4.2493e-01, -1.1276e-01, -2.5398e-01,  4.1216e-01,\n",
       "           1.1008e-01, -3.9448e-02, -7.1693e-01,  4.9330e-01,  4.6957e-01,\n",
       "          -9.5580e-02,  5.4183e-01,  1.7072e-01,  7.4580e-01, -3.2102e-01,\n",
       "           2.2552e-02, -9.1301e-01, -2.6662e-01, -6.5699e-02,  4.5665e-02,\n",
       "          -3.8437e-01, -1.6668e-01,  5.0396e-01,  4.7545e-02,  2.8247e-01,\n",
       "           5.2489e-01, -3.1823e-01, -7.7021e-01, -2.6016e-01, -4.0553e-02,\n",
       "           1.4974e-01, -1.3272e-01,  3.7139e-01, -6.3783e-01,  2.0332e-01,\n",
       "          -3.3532e-01, -5.7927e-01,  6.7625e-02,  6.1898e-01, -3.2455e-02,\n",
       "           5.9162e-01, -4.8033e-02,  1.9231e-01,  4.3555e-01,  4.1558e-01,\n",
       "           8.0359e-02,  6.0849e-01, -1.3917e-01, -9.1508e-01,  1.1777e-01,\n",
       "           1.8890e-01, -1.3081e-01,  3.7369e-01, -4.4057e-01, -2.0569e-02,\n",
       "           3.0017e-01,  6.1711e-01,  4.5551e-01]], device='mps:0'),\n",
       " tensor([[-1.1418e-01, -4.2743e-01, -7.2260e-01, -6.3933e-01, -7.8022e-02,\n",
       "          -5.6085e-01, -9.7444e-01,  7.0255e-01,  4.8843e-01,  7.1184e-01,\n",
       "           2.5635e-01,  2.2949e-01,  1.9804e-01, -2.2429e-01, -1.2982e-01,\n",
       "          -3.8039e-01, -1.8212e-02,  3.6377e-01,  7.4515e-02, -4.0144e-01,\n",
       "           8.3199e-01,  4.0793e-01, -1.1195e-01,  1.8140e-01, -6.1533e-01,\n",
       "           9.5521e-02,  5.5987e-02,  1.5592e-01,  3.2368e-01, -7.7032e-01,\n",
       "          -1.9274e-01,  4.8107e-01,  5.2586e-02,  6.9033e-01, -9.0843e-03,\n",
       "          -1.9389e-01,  2.4976e-01, -1.2960e-01,  1.8592e-01,  6.6149e-02,\n",
       "           3.1164e-01,  1.5772e-01,  3.5226e-01,  6.8565e-02,  4.4235e-01,\n",
       "          -2.0117e-01,  2.6766e-01, -6.4730e-01,  1.7283e-01, -2.2068e-01,\n",
       "          -4.5338e-01, -2.6728e-01,  3.5508e-01,  1.2777e-01, -7.7063e-01,\n",
       "          -6.1651e-01, -7.2258e-02,  1.5742e-01, -4.7190e-01,  3.6978e-01,\n",
       "           1.1932e-01,  4.1916e-01,  3.0417e-02,  1.1317e-01,  3.9937e-01,\n",
       "           4.5465e-01, -5.8629e-01, -6.8149e-01, -4.9264e-01, -8.9501e-02,\n",
       "           1.0030e-01,  3.3130e-01, -2.4906e-01, -5.8570e-01,  2.5348e-01,\n",
       "          -1.2351e-01,  4.7854e-01, -5.3089e-03, -6.0739e-01, -8.3552e-03,\n",
       "           4.2371e-01,  1.4891e-01, -5.4971e-02,  1.9372e-02, -9.7926e-02,\n",
       "          -5.6371e-02,  3.1711e-01,  1.8591e-01, -1.1715e-01,  3.5059e-01,\n",
       "           8.1878e-01,  4.2754e-01,  3.1761e-02, -4.4933e-01,  1.8662e-01,\n",
       "          -1.3193e-01,  3.9146e-01, -1.3390e-01,  7.1522e-02, -1.1976e-01,\n",
       "           4.0129e-02,  3.1847e-01,  2.8331e-01, -3.0023e-01, -2.6693e-01,\n",
       "          -7.9652e-01, -6.3445e-02, -5.8767e-02, -5.2738e-01, -5.1041e-03,\n",
       "          -1.3288e-01, -3.9180e-01,  1.3404e-01, -5.8206e-01, -1.1228e-01,\n",
       "           3.7008e-01, -9.4421e-02, -4.9702e-01, -3.5256e-01,  5.7060e-01,\n",
       "           6.1035e-02, -2.0046e-01, -2.8722e-01, -5.2835e-02,  5.5335e-01,\n",
       "          -4.3384e-01,  5.7172e-01, -5.4222e-01,  3.7980e-01,  8.6242e-01,\n",
       "           1.5968e-01,  6.8082e-02, -3.4286e-01, -1.3683e-01, -2.2591e-01,\n",
       "           1.7988e-01,  1.1767e-01, -5.1410e-01,  6.6868e-02,  7.0048e-02,\n",
       "          -1.0259e-01, -1.3545e-01, -5.1185e-01, -3.2670e-01, -3.6276e-02,\n",
       "          -3.1234e-01, -2.3606e-01,  4.4991e-01,  1.3007e-03, -3.5474e-01,\n",
       "          -5.9774e-01, -4.7919e-01,  4.3388e-01, -4.4451e-02, -1.0801e-03,\n",
       "          -3.0292e-01, -1.0861e-01,  7.7864e-02, -1.4194e-01,  8.6647e-02,\n",
       "           8.0961e-01, -1.1594e-01,  3.2994e-01, -2.8057e-01, -1.3921e-01,\n",
       "           3.3787e-01,  2.4864e-01, -1.3246e-01, -1.8179e-02,  6.0183e-02,\n",
       "           3.1254e-01, -1.8515e-01,  6.9839e-02, -4.7296e-01,  5.1302e-03,\n",
       "           3.9153e-02,  5.4970e-01,  4.1528e-01, -1.2645e-01, -2.3431e-01,\n",
       "           8.8300e-02, -1.9227e+00,  7.8008e-01, -8.2620e-01, -5.4370e-02,\n",
       "          -4.8668e-01,  4.3606e-01,  2.8575e-01,  1.8621e-01,  1.1944e-01,\n",
       "          -1.7396e-01, -1.1376e-01,  2.6503e-02,  3.8158e-01,  9.4875e-03,\n",
       "           6.8358e-01,  5.9196e-01, -1.3851e-01,  1.0958e-01,  1.0254e-01,\n",
       "          -3.0793e-01,  4.5035e-01, -5.5995e-02, -4.8428e-01,  1.3838e-01,\n",
       "           2.4356e-01, -3.0038e-01,  1.2927e-01,  1.7494e-01,  8.9433e-03,\n",
       "          -5.0228e-02, -2.0711e-01, -3.3613e-01,  2.7038e-01, -2.9817e-01,\n",
       "           5.2912e-01, -2.3759e-01,  9.6168e-03,  3.7388e-01,  1.2453e-01,\n",
       "          -2.6596e-01,  6.3262e-02, -7.5493e-02, -2.3458e-01, -8.1702e-02,\n",
       "           5.0799e-01, -5.6874e-01,  1.8403e-01, -2.8042e-01, -2.0181e-01,\n",
       "          -1.0137e-01, -1.0480e+00,  4.7027e-01,  3.3153e-01,  7.0918e-02,\n",
       "          -5.6316e-02, -8.4896e-02, -1.3306e-01, -3.4439e-01, -2.4876e-01,\n",
       "          -5.1092e-01, -3.8222e-01, -4.1384e-01, -6.6754e-04,  2.2540e-01,\n",
       "           2.8167e-02, -3.9951e-02,  1.1024e+00,  4.6301e-01,  7.7434e-01,\n",
       "           1.5994e-01,  2.0158e-01,  5.3376e-01,  1.1136e-01, -2.2741e-01,\n",
       "          -8.8031e-02,  7.5215e-03, -2.0065e-01,  4.1026e-01, -3.0941e-01,\n",
       "          -2.1875e-01,  2.3748e-01,  4.0660e-01,  7.5155e-01,  4.0889e-02,\n",
       "          -9.1460e-02,  3.8390e-03,  1.1481e-01, -1.5783e-01,  6.3135e-01,\n",
       "          -2.7045e-01,  5.5682e-02, -1.9083e-01, -3.1005e-01, -7.2249e-01,\n",
       "          -4.5068e-01,  1.9485e-01, -2.6671e-01,  2.5773e-01, -1.1227e+00,\n",
       "          -3.0850e-01,  3.8312e-01, -1.0940e-02,  4.2779e-01,  2.0810e-01,\n",
       "           3.1011e-01,  2.0690e-01,  2.0710e-01, -2.0779e-02, -1.8910e-01,\n",
       "          -3.5013e-02,  2.8509e-01, -3.9585e-01,  4.3052e-01,  3.0796e-01,\n",
       "          -3.9720e-01,  2.0534e-01,  2.2160e-01,  1.2022e-01, -2.9145e-01,\n",
       "          -4.5417e-01, -3.2472e-01, -1.6361e-01, -1.5181e-01, -1.0703e-01,\n",
       "          -1.9928e-01,  4.3994e-01,  5.6727e-01,  2.7638e-02, -2.7430e-01,\n",
       "          -5.7596e-02,  5.6100e-01, -5.7774e-02,  2.6694e-01, -1.9232e-01,\n",
       "          -1.0586e+00, -4.4672e-01, -2.4890e-01, -1.1880e-01, -3.5309e-01,\n",
       "           2.5164e-01, -2.7543e-02, -1.4043e-01, -1.5893e-01, -3.3290e-01,\n",
       "           2.0233e-01,  6.0997e-02,  2.8627e-01, -1.6957e-01,  1.8614e-01,\n",
       "          -2.1409e-01, -2.6816e-01,  2.2424e-02, -2.3625e-01,  9.3003e-02,\n",
       "           2.4144e-01,  6.6493e-02, -1.5103e-01,  4.1886e-01, -2.3275e-01,\n",
       "           7.8724e-01, -5.5086e-01,  1.1249e-02,  4.7414e-01, -3.5539e-01,\n",
       "          -6.1377e-01,  2.1641e-01, -4.2110e-01,  1.1372e-01,  5.7735e-02,\n",
       "           2.0022e-01, -5.1700e-01, -3.7121e-01, -4.8988e-01,  8.0017e-01,\n",
       "           2.8766e-01,  5.5485e-01, -6.7970e-01,  3.5377e-01,  2.8934e-01,\n",
       "          -4.6455e-01,  1.9734e-01, -3.0868e-01, -1.8893e-01, -3.1992e-01,\n",
       "           3.7759e-01,  3.1319e-01, -5.2707e-01, -1.1038e-01, -1.7950e-01,\n",
       "           2.6453e-01, -2.2520e-01,  1.3450e-01,  3.4013e-01,  7.2350e-01,\n",
       "          -2.4565e-01, -1.8269e-01,  3.1211e-01, -5.1684e-01, -4.4199e-01,\n",
       "           1.2182e-01,  2.6149e-02, -1.6446e-01, -1.2462e-02, -3.3089e-01,\n",
       "          -3.1200e-01,  2.1527e-01,  3.2137e-01,  3.2873e-01,  5.4179e-03,\n",
       "           1.2921e-01, -2.1754e-01, -2.7522e-01,  3.8505e-01, -1.9795e-01,\n",
       "          -1.3386e-01,  4.5056e-01, -1.2003e-01, -4.4023e-02,  2.9940e-01,\n",
       "          -2.5893e-01,  2.7013e-01,  3.8679e-01,  5.9588e-01, -3.3450e-01,\n",
       "          -3.6672e-01,  2.2656e-01,  4.1868e-01, -1.7451e-02,  1.8606e-01,\n",
       "          -4.7954e-01,  8.4423e-01, -5.5263e-01, -3.6410e-01,  3.6787e-01,\n",
       "          -2.3068e-01, -1.5708e-01,  2.1586e-01, -1.9405e-01, -3.4027e-01,\n",
       "           2.5154e-01,  4.8401e-02,  1.7694e-01,  2.1754e-01, -1.1128e+01,\n",
       "           2.1036e-01,  6.3077e-01, -3.8542e-01,  4.4238e-02,  3.5218e-02,\n",
       "          -1.3279e-01, -2.8615e-01,  1.1089e-01, -4.5168e-01, -3.2237e-01,\n",
       "           5.9403e-03,  8.6440e-02,  8.7702e-03, -3.3251e-01, -2.6889e-01,\n",
       "          -1.2792e-01, -8.4641e-01,  2.0372e-01, -2.0529e-01,  2.6673e-01,\n",
       "          -3.6992e-01,  3.0577e-01,  4.0037e-01, -3.7710e-01, -7.9932e-02,\n",
       "          -2.8592e-01,  1.0338e-01,  3.7273e-02,  8.3935e-02, -2.9265e-01,\n",
       "          -2.2893e-01, -8.4321e-02,  1.8480e-01, -2.0552e-01, -6.8101e-01,\n",
       "          -9.2089e-02,  1.3163e-01,  4.3866e-01,  1.9581e-01, -6.1002e-01,\n",
       "          -2.7602e-03, -4.4350e-01,  3.8802e-01,  4.6439e-01,  6.2744e-01,\n",
       "           1.2380e+00,  4.9424e-01, -4.9526e-03, -4.5367e-02,  7.9707e-02,\n",
       "          -7.2045e-01,  3.4287e-01, -6.4450e-01, -7.9565e-02, -2.0868e-01,\n",
       "           5.8680e-01, -5.4928e-01, -3.4396e-02, -2.3011e-01, -1.7896e-01,\n",
       "           1.4167e-02,  2.6237e-02,  3.8453e-01, -1.8165e-01, -4.2770e-01,\n",
       "           8.4162e-01, -7.7598e-02,  1.9936e-01, -2.1547e-01, -2.0975e-01,\n",
       "          -4.6069e-01,  2.6226e-01,  2.6892e-01, -3.8940e-01, -6.5375e-03,\n",
       "          -9.8964e-02, -6.4932e-02,  9.1792e-01, -6.8857e-02,  2.0661e-01,\n",
       "           2.0010e-01,  3.5173e-03, -5.3219e-01, -4.3487e-01, -3.0295e-01,\n",
       "           7.8755e-02,  1.4023e-01, -7.8579e-03,  3.5031e-01, -2.8077e-01,\n",
       "          -6.7702e-01,  3.8307e-02,  7.3619e-02,  1.2486e-01,  5.8775e-01,\n",
       "          -4.8320e-01, -2.7477e-01,  1.2170e-01,  2.0798e-01, -6.3947e-01,\n",
       "          -1.9498e-01, -2.6607e-01, -3.4058e-01,  3.5853e-01, -9.3284e-02,\n",
       "           3.2092e-01,  6.5622e-02, -5.1130e-02,  5.2738e-01, -2.1840e-01,\n",
       "           3.7076e-01, -2.7750e-01,  4.2136e-01, -4.9311e-01,  7.2964e-02,\n",
       "           3.7248e-01,  8.0890e-01, -3.3219e-01, -6.0434e-01, -1.7028e-01,\n",
       "           1.8848e-01,  2.6653e-01,  1.5358e-01, -7.7681e-01,  2.7171e-02,\n",
       "          -4.5704e-01,  9.5779e-03, -2.7549e-02,  3.9952e-01,  6.7410e-01,\n",
       "           1.4607e-01, -3.1008e-02, -2.2989e-01, -3.4152e-01, -1.0948e-01,\n",
       "          -3.3062e-01, -1.8914e-02,  4.7147e-01, -5.0398e-01, -1.6211e-01,\n",
       "          -1.5793e-01,  1.3629e-01, -3.4561e-01,  3.3825e-01, -8.6527e-02,\n",
       "           5.0523e-01,  5.7888e-01,  3.0291e-02,  5.7838e-01, -2.0869e-01,\n",
       "          -3.3156e-01, -4.3820e-01, -8.5286e-02, -5.8063e-02, -4.7024e-01,\n",
       "           4.1879e-02,  1.2707e-01, -4.6827e-01, -1.0929e-01, -2.9483e-02,\n",
       "          -5.3709e-01, -3.2703e-01, -3.7821e-01,  5.4790e-02,  3.7660e-01,\n",
       "           2.6391e-02, -5.5826e-01, -4.7311e-02,  6.9194e-02,  2.0596e-01,\n",
       "           2.6573e-01, -2.6262e-01, -3.5280e-01, -7.9648e-01,  8.9418e-02,\n",
       "           1.4000e-02,  9.4462e-01, -3.6996e-01, -5.4305e-01,  2.8718e-01,\n",
       "           2.6548e-01, -7.1391e-02, -1.5647e-01, -9.2560e-02,  1.5106e-01,\n",
       "          -1.8011e-01,  7.5540e-02,  2.7130e-01, -1.2569e-01,  1.2237e-01,\n",
       "           7.8234e-02,  4.9238e-01, -1.4760e-01,  2.8264e-01,  1.2660e-01,\n",
       "          -5.8340e-01, -9.1271e-02, -1.1782e+00, -5.9068e-02, -5.5899e-02,\n",
       "          -2.2886e-01,  1.5079e-01, -1.6813e-01, -2.7420e-02,  3.1934e-02,\n",
       "          -1.0851e-03,  1.0644e+00,  1.3662e-01, -9.3327e-03,  5.3231e-01,\n",
       "           2.3226e-01, -1.7349e-01,  1.8365e-01, -2.3434e-01,  1.2572e-01,\n",
       "           2.7873e-01,  1.7760e-01, -5.2205e-01,  1.5771e-01,  3.5325e-01,\n",
       "           5.0884e-01, -1.1552e+00, -7.6919e-01, -2.5133e-02,  4.3937e-01,\n",
       "          -1.4434e-01,  9.6027e-03, -1.9234e-01,  2.6255e-01,  4.9446e-01,\n",
       "          -1.2182e+00, -3.1310e-01,  5.8438e-01,  5.0454e-01, -3.6039e-02,\n",
       "          -6.0419e-01, -8.2513e-02,  9.2000e-03, -4.3843e-01,  5.8603e-01,\n",
       "           4.8111e-01, -5.5660e-01,  3.0458e-01,  2.3874e-01, -3.7205e-01,\n",
       "          -7.9385e-02, -1.3196e-01,  2.7204e-01,  2.9828e-01,  1.8684e-01,\n",
       "          -6.0796e-01,  3.0275e-01, -6.7057e-01, -3.9097e-01,  2.3010e-01,\n",
       "           5.0302e-01, -3.9938e-02,  2.8966e-01, -1.1752e-01, -1.9297e-03,\n",
       "          -2.1381e-01, -2.1942e-01, -3.8699e-01, -1.1027e-01, -6.6579e-02,\n",
       "           3.9786e-01, -1.0672e-01,  2.4226e-01,  3.1450e-01, -4.0286e-01,\n",
       "           1.9907e-01,  5.6838e-01,  2.5180e-01, -7.1677e-02, -1.2031e-01,\n",
       "          -1.0406e-01,  7.5263e-02, -2.7059e-01,  6.2964e-01, -2.4138e-01,\n",
       "          -3.3699e-01,  4.3508e-01,  2.5083e-02, -4.4687e-01, -1.9116e-01,\n",
       "          -4.1389e-01,  1.4291e-01,  1.2167e-01, -5.4617e-01,  2.6297e-01,\n",
       "           2.9411e-01,  1.9805e-01, -5.1015e-01,  2.3947e-01,  2.9114e-01,\n",
       "          -1.2864e-01,  1.0273e+00,  5.8633e-02,  4.9283e-01,  7.2642e-03,\n",
       "           1.4251e-01, -6.0643e-01, -1.6881e-01,  3.0661e-01,  3.9877e-01,\n",
       "          -2.5154e-01, -3.0881e-02,  3.0149e-01,  6.5951e-02,  9.4433e-01,\n",
       "           5.6249e-01, -2.6987e-02, -6.3389e-01, -6.4409e-02,  1.3366e-01,\n",
       "          -5.3730e-02, -3.5430e-01,  2.2848e-01, -7.4755e-02, -1.3501e-02,\n",
       "          -1.0133e+00, -4.8762e-01,  4.7826e-02,  1.6941e-01, -1.8491e-01,\n",
       "           7.6992e-01, -1.0579e-02,  1.4110e-01,  5.9030e-01,  1.1791e-01,\n",
       "           1.9761e-01,  5.8025e-01,  2.9968e-01, -4.3029e-01,  2.6831e-01,\n",
       "           2.4331e-01, -2.3397e-01,  7.7202e-02, -6.1282e-01,  5.6888e-02,\n",
       "           1.4553e-02,  1.1538e+00,  4.0106e-01]], device='mps:0'),\n",
       " tensor([[-0.0743, -0.4193, -0.5377,  ...,  0.2489,  0.3612,  0.6175],\n",
       "         [-0.2961, -0.6861, -0.6414,  ...,  0.3717,  0.2839,  0.5318]],\n",
       "        device='mps:0'),\n",
       " tensor([[-2.5187e-01, -4.0054e-01, -5.8007e-01, -8.3055e-01, -1.2868e-01,\n",
       "          -2.8804e-02, -7.1917e-01,  6.6701e-01,  1.2709e-01,  5.0855e-01,\n",
       "           1.2570e-01,  5.1916e-01,  5.1676e-02, -1.1521e-01, -1.6481e-01,\n",
       "          -4.9820e-01,  5.6222e-02,  3.7338e-01,  6.7187e-01, -2.4111e-02,\n",
       "           2.2844e-01,  5.7979e-01, -2.6323e-01, -1.2386e-02, -8.3013e-01,\n",
       "           2.9613e-01, -1.1576e-01,  2.2409e-01,  1.7718e-01, -6.3092e-01,\n",
       "           4.0999e-02,  4.7417e-01,  2.3068e-02,  1.3497e+00, -5.6080e-01,\n",
       "          -1.3230e-01,  8.2213e-02, -3.9453e-01, -2.3010e-01,  3.7283e-02,\n",
       "           6.1164e-01, -4.1315e-01,  1.3635e-01,  4.0254e-01, -6.1085e-02,\n",
       "           8.9076e-02,  1.3899e-01, -3.7744e-01,  2.5892e-01, -8.3961e-01,\n",
       "          -4.8318e-02, -2.0669e-01,  4.9144e-01,  6.2615e-01, -6.3176e-01,\n",
       "          -2.2975e-01,  2.7419e-01, -3.7716e-02, -3.3020e-02,  1.5349e-01,\n",
       "           9.7980e-02,  3.8582e-01, -2.9995e-01,  2.3257e-01,  3.3441e-01,\n",
       "           2.4458e-01, -3.0392e-01, -4.8889e-01, -5.4078e-01, -5.0111e-02,\n",
       "           9.1859e-03,  2.2490e-02,  1.4846e-01, -2.3272e-01,  4.1973e-01,\n",
       "          -7.1214e-01, -6.4354e-02, -8.9125e-02, -2.5973e-01,  2.9246e-01,\n",
       "          -4.5549e-02,  1.9719e-01,  3.1010e-01,  1.0268e-01,  1.2457e-01,\n",
       "           4.3419e-02,  2.7215e-01,  5.2932e-04,  4.7373e-02,  3.7533e-01,\n",
       "          -1.3375e-01, -8.1722e-02, -2.8313e-01, -3.5814e-01,  3.1229e-01,\n",
       "           4.0598e-02,  2.9907e-01, -1.9557e-01,  1.8579e-01, -3.2474e-01,\n",
       "           1.5588e-01,  1.7768e-01, -1.8083e-01,  5.7286e-02, -3.5172e-01,\n",
       "          -4.5475e-01,  2.3589e-02, -2.8569e-01, -4.8285e-01,  2.2697e-01,\n",
       "          -1.4211e-01, -3.0205e-01, -4.0922e-02, -3.8935e-01, -3.1524e-02,\n",
       "          -7.5164e-02,  2.6013e-01, -1.3230e-01, -3.4806e-01, -8.3033e-02,\n",
       "           2.0734e-01, -2.2808e-01,  1.6077e-01,  2.6882e-01,  1.2232e-01,\n",
       "          -9.6705e-02,  6.5251e-01,  1.8950e-02,  3.1674e-01,  2.5763e-01,\n",
       "           1.1699e-01, -6.5768e-02, -1.9745e-01, -2.3154e-01,  7.5988e-02,\n",
       "          -7.6941e-02,  1.0687e-01, -1.8182e-01, -2.2790e-01, -1.8508e-01,\n",
       "           3.1771e-01, -3.3097e-01, -6.0893e-01, -4.9260e-01, -1.6098e-01,\n",
       "          -1.6326e-01, -2.4804e-02, -8.2473e-03, -3.3693e-01, -3.1956e-01,\n",
       "          -6.7443e-01, -3.6717e-01,  8.1041e-01, -1.6831e-02, -3.3356e-01,\n",
       "          -1.6499e-01,  2.3208e-01,  3.1666e-01, -4.5959e-02,  3.7161e-01,\n",
       "           7.7250e-01, -4.7670e-02,  1.6786e-01, -5.1827e-01, -3.8195e-01,\n",
       "           4.8335e-02,  2.6658e-01, -2.7054e-01,  6.1879e-01,  4.8352e-02,\n",
       "           7.5627e-02, -6.6170e-02,  2.7430e-01, -3.2704e-01, -1.2324e-02,\n",
       "          -5.7573e-01,  3.7750e-01,  2.9539e-01, -3.9892e-01, -6.6567e-02,\n",
       "          -2.0834e-02, -1.2060e+00,  9.6964e-02, -8.3040e-01,  3.1236e-01,\n",
       "          -2.3110e-01,  1.7963e-01,  2.3389e-01,  2.9545e-01, -6.5320e-02,\n",
       "           2.5404e-01,  2.8725e-01,  2.5687e-01,  3.3609e-01, -3.4924e-02,\n",
       "           2.8019e-01,  2.3924e-01,  5.0790e-01,  4.1073e-01,  1.4769e-01,\n",
       "          -3.5440e-01,  2.9683e-01,  3.5781e-01,  9.0598e-02,  1.6954e-01,\n",
       "           9.9825e-02, -2.1636e-01,  2.1091e-01,  1.7501e-01, -1.6618e-01,\n",
       "          -2.2406e-01, -3.1100e-02, -4.6818e-01, -5.6655e-02, -4.8467e-01,\n",
       "           3.2557e-01, -5.1983e-01, -1.1912e-01,  3.4831e-01,  2.4340e-01,\n",
       "          -1.4896e-02,  1.8319e-01,  1.7196e-01, -2.4338e-01,  2.1735e-01,\n",
       "           3.7449e-01, -3.4618e-01, -3.9804e-02, -3.2197e-01, -2.9367e-01,\n",
       "           2.4357e-01, -5.6244e-01,  6.6635e-01,  4.0574e-01,  6.0165e-02,\n",
       "           1.5119e-01, -2.7920e-01, -5.1638e-02, -5.7408e-01,  3.5471e-01,\n",
       "          -3.2130e-01, -3.4955e-01, -5.4801e-01,  6.4487e-02,  4.2773e-01,\n",
       "           2.7565e-01, -6.4915e-02,  7.7868e-01,  4.2408e-01,  4.2813e-01,\n",
       "          -4.3003e-01, -1.5896e-02,  4.0377e-01, -4.1405e-02, -4.6059e-01,\n",
       "           6.7116e-02, -1.4962e-01, -5.1643e-01,  2.8000e-01, -2.6598e-01,\n",
       "          -1.2047e-01, -1.4913e-01,  3.3682e-01,  1.5365e-01, -2.1964e-01,\n",
       "          -3.8461e-01,  3.9587e-01, -1.4349e-01, -4.1723e-01,  4.7123e-01,\n",
       "          -3.5025e-01,  1.5209e-01,  1.0694e-01, -3.8441e-01, -6.2072e-01,\n",
       "          -3.2235e-01,  2.3090e-01,  1.5585e-01,  3.4799e-02, -6.6306e-01,\n",
       "          -4.0571e-01,  3.1197e-01,  2.4601e-01,  2.9976e-01,  2.9840e-01,\n",
       "           5.7427e-01, -3.0944e-01, -1.6702e-01,  1.4982e-01, -4.1377e-01,\n",
       "          -1.5704e-01, -3.3228e-02, -1.9995e-01,  2.9569e-01,  2.4768e-01,\n",
       "           1.1600e-01,  2.0367e-01,  4.3482e-01, -2.5272e-01, -3.8673e-01,\n",
       "          -7.1543e-02,  2.0161e-01, -9.3629e-02, -5.8306e-02,  2.3951e-01,\n",
       "          -6.3095e-02,  5.9650e-01,  7.2143e-01,  7.7439e-02, -6.4006e-01,\n",
       "           1.7696e-01,  5.2502e-01, -2.7253e-01, -4.0343e-02,  9.6365e-02,\n",
       "          -1.0344e+00, -2.1331e-01, -1.2883e-01, -2.0181e-01, -6.5352e-01,\n",
       "           8.8115e-02,  3.4863e-01,  4.6833e-03, -1.7864e-01, -3.0947e-01,\n",
       "           1.0448e-01,  4.8724e-02, -4.8980e-02,  4.3534e-01, -2.8901e-01,\n",
       "          -3.4797e-01, -2.3438e-01, -2.1532e-01, -5.4201e-01, -3.9968e-01,\n",
       "           3.5914e-01,  2.8129e-01,  1.1986e-01, -2.6486e-01, -5.1832e-01,\n",
       "           4.5577e-01, -6.8169e-01, -1.4632e-01,  2.4840e-01, -4.7702e-01,\n",
       "           1.4650e-02,  4.8990e-01,  1.4999e-01,  1.6336e-01,  6.5258e-02,\n",
       "           7.8557e-02, -3.3903e-01, -5.0030e-01, -1.5000e-01,  6.2323e-01,\n",
       "           3.2283e-01,  3.3506e-01, -1.7236e-01,  3.9096e-01,  4.5167e-01,\n",
       "          -2.9582e-01,  4.1670e-01,  1.0930e-02,  7.7439e-02, -6.0414e-02,\n",
       "           1.9048e-01,  3.4370e-01, -3.6842e-01,  2.0145e-01,  1.7929e-01,\n",
       "          -3.1634e-01, -3.2149e-01,  1.5691e-02,  2.0642e-01, -1.8580e-01,\n",
       "          -3.0875e-01, -2.0361e-01,  9.9698e-02, -3.4856e-01, -4.3104e-01,\n",
       "           2.9161e-01, -3.6543e-02,  7.0495e-02, -4.0357e-01, -4.3462e-01,\n",
       "          -2.0181e-01, -2.6055e-01,  6.8681e-01,  2.1064e-01, -1.9621e-01,\n",
       "           1.8720e-01, -1.0907e-01, -3.6330e-01,  4.7847e-01,  2.0162e-02,\n",
       "          -4.8754e-01,  4.5482e-01, -3.4822e-01,  1.0627e-01,  5.5756e-01,\n",
       "          -1.6769e-01, -1.0840e-01,  1.1008e-01,  1.0820e-01,  2.9026e-01,\n",
       "          -3.9884e-01,  1.5516e-01, -6.3476e-02, -3.1885e-01,  3.1820e-01,\n",
       "           2.6411e-01,  2.2781e-01, -1.7137e-01, -3.3426e-02,  3.5677e-02,\n",
       "           2.8782e-01, -6.0853e-02, -3.9888e-01, -4.7347e-01, -4.6132e-01,\n",
       "           1.2832e-01,  1.8927e-01,  5.6207e-01,  3.4144e-01, -1.1394e+01,\n",
       "           1.2938e-01,  7.8031e-01, -4.0713e-01, -8.2819e-02, -5.0320e-02,\n",
       "           1.0976e-02, -1.3708e-01,  4.2955e-01, -2.7734e-01, -4.4105e-01,\n",
       "          -3.5727e-01, -1.3308e-01,  1.5454e-01, -3.2189e-01, -2.6945e-01,\n",
       "          -1.4686e-02, -7.0935e-01,  4.0963e-01, -3.0286e-01,  5.0298e-01,\n",
       "          -2.3511e-01, -8.3798e-02,  1.5816e-01, -3.5851e-01, -1.3439e-02,\n",
       "           5.9766e-02, -1.6033e-01,  1.4651e-01,  4.2486e-01, -5.2072e-02,\n",
       "           3.3244e-02, -4.1258e-03,  4.8461e-02, -2.5723e-01, -5.9999e-01,\n",
       "           1.9296e-02, -1.4141e-01,  3.3422e-01, -4.4133e-02, -7.1265e-01,\n",
       "          -2.2533e-01, -2.7418e-01,  3.1407e-01,  2.7021e-01,  9.0598e-01,\n",
       "           2.2955e-01,  1.0686e+00,  1.5022e-01,  1.7748e-01,  2.6230e-02,\n",
       "          -7.8764e-01, -3.3269e-02, -4.1907e-01,  2.3280e-02,  1.0114e-01,\n",
       "           3.6576e-01, -7.9367e-01,  2.5251e-01, -3.5521e-01,  3.6487e-02,\n",
       "           4.9094e-01,  3.1768e-02,  2.7700e-01, -3.2298e-02, -7.8123e-02,\n",
       "           1.2089e-01, -2.8544e-02,  1.1098e-01, -2.2583e-01, -2.4084e-01,\n",
       "          -1.0177e-01,  5.9888e-01,  3.9592e-01, -7.2211e-01,  3.6112e-01,\n",
       "           4.9258e-02, -1.8622e-01,  1.4556e+00, -4.8573e-02,  1.8361e-02,\n",
       "           1.9800e-03,  2.9604e-01, -5.0236e-01, -7.3516e-01,  3.6511e-01,\n",
       "          -1.0211e-01,  5.6427e-02,  1.0127e-01,  4.8148e-01, -9.7481e-02,\n",
       "          -4.7940e-01, -1.8593e-03,  1.0556e-01, -1.3380e-01,  5.4710e-01,\n",
       "          -7.2824e-01, -1.6963e-01, -1.1223e-01,  3.1723e-01, -8.4054e-01,\n",
       "          -6.1126e-02, -7.0210e-01, -3.9328e-01,  5.0586e-01, -2.5896e-01,\n",
       "           1.5852e-01, -2.7443e-02, -4.3439e-02,  5.5574e-01, -2.1489e-01,\n",
       "           7.1555e-02,  2.2813e-01,  8.4995e-02, -1.6733e-01,  3.9349e-02,\n",
       "           4.1630e-01,  4.9571e-01, -4.2740e-01, -6.3911e-01, -7.6901e-03,\n",
       "           1.7180e-01,  1.4933e-01,  3.4019e-01, -5.3463e-01,  3.6431e-01,\n",
       "          -7.9162e-01, -1.8769e-01,  7.2949e-02, -4.3273e-02,  3.6175e-01,\n",
       "           2.3925e-01,  7.2038e-02,  2.2256e-01, -7.7756e-01, -6.4498e-01,\n",
       "          -6.5932e-01, -2.1445e-01,  2.6143e-01, -5.4236e-01, -8.3654e-02,\n",
       "          -1.1033e-01, -4.8837e-03, -5.0292e-02, -6.6102e-02,  1.1939e-02,\n",
       "           1.3632e-01,  2.3587e-01, -1.3023e-01, -1.3178e-01, -9.2464e-02,\n",
       "          -5.5566e-01, -1.1753e-01, -9.9591e-02, -1.2505e-01, -5.6783e-01,\n",
       "          -1.3797e-01,  3.3075e-02, -2.5373e-01, -6.8124e-02,  3.5608e-01,\n",
       "          -2.9832e-02, -2.7856e-02, -4.8850e-01,  7.2846e-01,  4.0279e-01,\n",
       "           5.2088e-02, -2.8298e-01, -5.5189e-01, -1.0567e-01,  2.9592e-01,\n",
       "           1.9555e-01, -6.7007e-02, -1.7709e-01, -2.4145e-01, -4.2522e-01,\n",
       "           1.4302e-01,  7.1147e-01, -6.6296e-02, -6.8115e-01,  2.6621e-01,\n",
       "           5.8223e-01, -1.2882e-01, -2.4432e-01,  2.0983e-02, -9.3284e-02,\n",
       "          -3.1851e-01, -6.0317e-01,  7.8821e-01, -1.1246e-01, -5.8638e-02,\n",
       "           2.2427e-01,  2.0359e-01, -9.2588e-03,  4.0209e-01, -1.5770e-01,\n",
       "          -3.5434e-01, -1.6477e-03, -7.3600e-01,  4.3257e-01,  6.1666e-02,\n",
       "          -3.1489e-01,  7.6082e-01,  2.7260e-01, -6.0593e-02, -4.5779e-02,\n",
       "           2.2863e-01,  7.9941e-01, -2.4636e-01,  6.4653e-02,  5.3779e-01,\n",
       "           8.8110e-02, -1.1229e-01,  3.9580e-02,  4.1665e-02,  4.5410e-02,\n",
       "           3.7012e-01, -3.2145e-02, -4.1564e-01, -6.1194e-02,  3.8924e-01,\n",
       "           3.9295e-01, -5.4610e-01, -7.3487e-01,  2.2333e-02,  6.8551e-01,\n",
       "          -2.9054e-01,  2.3654e-01, -3.2708e-02,  3.0880e-01,  1.2001e-01,\n",
       "          -9.2160e-01, -1.5016e-01, -1.0056e-01,  2.8229e-01, -2.5482e-02,\n",
       "          -1.6139e-01, -1.4727e-01, -2.3686e-01, -1.6208e-01,  6.8308e-01,\n",
       "           2.1991e-01, -6.9160e-01,  3.4058e-02,  5.7281e-01, -6.0551e-01,\n",
       "           4.6052e-01, -2.4577e-01, -3.4302e-02,  3.6435e-01,  1.3117e-01,\n",
       "           4.9067e-02,  1.7856e-01, -4.2138e-01, -4.0598e-01,  1.7824e-01,\n",
       "           9.1672e-01, -7.6613e-02,  1.1166e-01, -3.9384e-01, -1.7182e-01,\n",
       "          -7.1855e-02, -3.1913e-01, -3.1286e-01,  3.7185e-01,  2.0529e-01,\n",
       "          -3.2552e-02, -1.3271e-01, -2.2992e-01,  3.0816e-01,  1.1821e-01,\n",
       "           3.2536e-01,  5.8187e-01,  6.4368e-01, -1.0077e-02, -2.4936e-01,\n",
       "          -6.1235e-02,  1.4568e-01, -1.9524e-01,  5.2704e-01, -2.0627e-01,\n",
       "          -3.3453e-01, -3.8069e-03,  3.0134e-01, -6.1681e-01, -5.8725e-02,\n",
       "          -4.5426e-01,  4.2361e-01, -1.6955e-01,  3.7868e-02,  2.8833e-01,\n",
       "           4.0522e-02,  5.7925e-02, -2.2651e-01,  5.5807e-01,  4.4511e-01,\n",
       "           1.6609e-01,  4.5662e-01,  1.6104e-01,  4.5272e-02, -5.1597e-01,\n",
       "           2.1978e-01, -1.0213e+00, -9.4260e-02, -4.6640e-02, -1.5401e-01,\n",
       "          -4.4224e-01, -2.8126e-02,  1.6814e-01,  1.8538e-01,  6.9597e-01,\n",
       "           7.2967e-01, -3.0871e-01, -4.5290e-01, -1.8510e-01,  1.5304e-01,\n",
       "           2.7705e-01, -1.9517e-01,  3.5971e-01,  9.6103e-02,  2.4432e-01,\n",
       "          -5.0967e-01, -4.3165e-01, -4.2252e-02,  3.1411e-01, -2.0171e-01,\n",
       "           5.7118e-01, -2.0164e-01, -1.3863e-01,  9.0832e-02,  1.3770e-01,\n",
       "           1.6639e-01,  7.9664e-01,  4.3416e-01, -6.3425e-01,  8.7469e-02,\n",
       "          -1.3980e-02, -2.9988e-01,  2.5983e-02, -4.3577e-01,  2.8858e-01,\n",
       "           1.2984e-01,  4.9280e-01,  6.7504e-01]], device='mps:0'),\n",
       " tensor([[-0.2334, -0.6497, -0.4326,  ...,  0.3671,  0.0747,  0.6548],\n",
       "         [-0.1858, -0.6238, -0.5019,  ...,  0.3588, -0.2460,  0.6156]],\n",
       "        device='mps:0'),\n",
       " tensor([[-5.6159e-02, -7.4351e-01, -7.7055e-01, -5.1459e-01, -1.5011e-01,\n",
       "           9.5182e-02, -7.2901e-01,  3.7005e-01, -1.6330e-02,  3.6349e-01,\n",
       "           4.2817e-01,  7.0454e-02, -3.6471e-02, -1.3400e-01, -2.0862e-01,\n",
       "          -4.3869e-01, -1.9008e-01,  3.4799e-01,  6.2164e-01, -9.8459e-02,\n",
       "           8.3423e-01,  5.9048e-01, -3.3121e-01,  1.0551e-01, -4.0690e-01,\n",
       "           3.2365e-01, -1.6958e-01,  3.6947e-01,  3.9406e-01, -5.6992e-01,\n",
       "          -4.7269e-02,  4.4332e-01, -7.3384e-02,  1.1898e+00, -3.4662e-01,\n",
       "          -2.5311e-01,  3.3469e-01, -3.2480e-01,  1.9230e-01,  2.1643e-03,\n",
       "           7.5962e-01, -4.0600e-01, -1.0885e-01,  1.5042e-01,  1.3638e-01,\n",
       "           1.2538e-01,  3.4136e-01, -7.6988e-01,  2.7150e-01, -4.1205e-01,\n",
       "           2.6370e-03, -4.9210e-02,  3.8852e-01,  5.9392e-01, -1.9883e-01,\n",
       "          -2.3533e-01,  2.5692e-01,  6.5980e-03,  1.0126e-02,  2.9610e-01,\n",
       "          -7.3802e-02,  2.7981e-01, -2.9778e-01, -4.8221e-02, -8.6955e-03,\n",
       "           3.1026e-01, -4.0167e-01, -5.6211e-01, -2.7823e-01, -2.9055e-01,\n",
       "          -1.1360e-01,  1.7577e-01,  8.9641e-02, -5.0221e-03,  2.1681e-01,\n",
       "          -6.5290e-01, -7.0354e-02, -1.5169e-01, -4.4987e-01,  9.8436e-02,\n",
       "           1.7210e-01,  1.5427e-01,  1.5735e-01, -2.9861e-01, -1.0442e-01,\n",
       "           1.0065e-01,  2.0422e-01,  2.3507e-01,  6.0317e-02,  2.2189e-01,\n",
       "           7.2204e-02,  1.0577e-01, -3.0039e-01, -3.9517e-01,  2.7825e-01,\n",
       "           2.1193e-02,  4.2376e-01, -1.8206e-01, -2.9471e-02, -6.9033e-02,\n",
       "           2.6110e-01,  1.7334e-01,  2.0773e-01,  9.4309e-02, -1.0059e-01,\n",
       "          -3.5621e-01,  1.0553e-01, -2.5510e-01, -8.0903e-01, -1.7499e-02,\n",
       "          -4.0820e-01, -1.2544e-01,  8.8041e-02, -3.2272e-01, -3.0519e-01,\n",
       "          -1.2654e-01,  7.0498e-02, -1.2355e-02, -2.6439e-01,  8.8780e-02,\n",
       "           2.5118e-01, -2.9852e-01, -4.7264e-03,  1.4121e-01,  3.3932e-01,\n",
       "          -2.3967e-01,  6.0729e-01, -6.4247e-02,  4.4398e-01,  3.3240e-01,\n",
       "          -4.0654e-02, -1.0863e-01, -1.9193e-01, -4.6489e-01, -1.7091e-01,\n",
       "           2.4839e-02,  3.0924e-01, -2.8946e-01, -7.3990e-02,  8.0248e-02,\n",
       "           8.7193e-02, -2.6962e-01, -4.3079e-01, -7.7054e-02, -8.8653e-02,\n",
       "          -1.7805e-01,  6.9545e-02, -2.1758e-01, -8.3199e-01, -1.5756e-01,\n",
       "          -7.3660e-01, -7.8536e-01,  2.4839e-01, -5.1206e-02, -2.8460e-01,\n",
       "           7.5877e-02,  3.4014e-01,  6.1601e-02,  1.2929e-01,  1.6348e-01,\n",
       "           9.7163e-01, -3.8764e-01,  2.6410e-01, -7.3760e-01, -9.3853e-02,\n",
       "           6.4467e-02, -2.0643e-02, -4.7381e-03,  3.0378e-01, -9.8469e-02,\n",
       "           2.6185e-01, -2.7798e-01,  6.7913e-02, -5.6080e-01,  1.8383e-01,\n",
       "          -5.0073e-01,  4.4353e-01,  2.6466e-01, -1.0758e-01, -7.5751e-02,\n",
       "          -1.6589e-01, -1.8454e+00,  1.0864e-01, -6.4884e-01,  3.3750e-01,\n",
       "          -5.1450e-01,  1.8966e-01,  3.0533e-01,  2.2923e-01, -9.9950e-03,\n",
       "          -4.1960e-01,  4.6632e-02,  1.3557e-01,  4.4305e-01, -6.4869e-02,\n",
       "           4.9505e-01,  4.7846e-01,  3.8219e-01, -7.1430e-02, -1.1439e-02,\n",
       "          -2.1757e-01,  2.7094e-01,  1.5644e-01,  2.2700e-01, -6.2473e-02,\n",
       "           7.3156e-02, -1.8454e-02,  1.4273e-02,  6.4570e-01, -1.1737e-01,\n",
       "          -2.5117e-01, -1.1730e-01, -6.7967e-01, -1.6522e-01, -3.5929e-01,\n",
       "           2.5386e-01, -3.3645e-01,  3.5554e-02,  2.4848e-01,  3.6929e-01,\n",
       "           1.6485e-02, -1.3339e-02,  1.6464e-01, -4.4579e-01, -7.7400e-02,\n",
       "           4.0176e-01, -1.7038e-01, -3.7012e-01, -2.3195e-01, -3.4085e-01,\n",
       "           2.3194e-01, -5.1263e-01,  5.7738e-01,  1.0870e-01, -2.1791e-01,\n",
       "          -5.3271e-02, -4.3455e-01,  1.0590e-01, -4.0813e-01,  3.9324e-01,\n",
       "          -2.3281e-01, -5.6456e-01, -7.4241e-01,  2.5686e-01,  2.2820e-01,\n",
       "           1.4165e-01, -1.9226e-01,  7.2816e-01,  3.1414e-01,  5.7962e-01,\n",
       "          -2.2387e-01, -1.6135e-01,  3.2026e-01, -3.7061e-02, -3.9336e-01,\n",
       "          -3.6285e-02, -1.8368e-01, -2.6123e-01,  2.3718e-01, -1.9361e-01,\n",
       "          -1.4516e-01, -1.5468e-01,  3.8994e-01,  3.7796e-01, -3.9453e-01,\n",
       "          -1.8533e-01,  6.3714e-01, -4.2050e-01, -3.7792e-01,  4.0357e-01,\n",
       "          -4.4185e-01,  1.9187e-01,  1.4171e-01, -3.2502e-01, -6.2283e-01,\n",
       "          -6.9085e-01,  1.4602e-01, -8.6266e-02,  1.5525e-01, -6.8304e-01,\n",
       "           6.7388e-02,  6.1778e-01,  6.2236e-02,  3.7998e-01,  1.0552e-01,\n",
       "           4.1339e-01,  6.5618e-02,  6.2072e-02, -1.3801e-02, -5.1079e-01,\n",
       "           2.4395e-02,  2.6482e-01, -3.0951e-01,  3.9167e-01,  2.8439e-01,\n",
       "           9.0374e-02,  4.4214e-02,  1.8450e-01, -6.4599e-02, -1.5511e-01,\n",
       "           7.8145e-02,  3.1548e-01, -2.9666e-01, -8.9995e-02,  1.1054e-01,\n",
       "           4.1322e-02,  2.7090e-01,  5.6562e-01, -2.7847e-01, -5.3656e-01,\n",
       "           1.8116e-01,  9.2991e-01, -1.6970e-01,  6.5798e-02, -1.3364e-01,\n",
       "          -9.6649e-01, -3.1290e-01, -6.9977e-02, -2.6638e-01, -4.3554e-01,\n",
       "           3.7775e-01,  1.1252e-02, -1.4700e-02,  7.4612e-02, -4.2946e-01,\n",
       "           1.8518e-01,  1.6357e-01,  1.5992e-01,  2.0969e-01, -7.0329e-02,\n",
       "          -3.2355e-01,  6.0476e-03, -2.2460e-01, -2.9169e-01, -1.2995e-01,\n",
       "           3.5974e-01,  3.3233e-01,  6.4574e-02, -5.0331e-02, -3.9120e-01,\n",
       "           7.7419e-01, -2.6417e-01, -2.2123e-01,  1.6562e-01, -3.4219e-01,\n",
       "          -4.3943e-01,  4.5454e-01,  7.0638e-02,  1.6243e-02, -1.3814e-01,\n",
       "           1.9186e-01, -7.2527e-01, -3.5453e-01,  1.2730e-01,  5.4103e-01,\n",
       "           2.8882e-01,  3.9909e-01, -2.8542e-01,  4.6307e-01,  7.4782e-01,\n",
       "          -4.9132e-01,  1.6731e-01, -1.1661e-01, -1.2424e-02, -7.5087e-02,\n",
       "           2.2402e-01,  3.9763e-01, -3.4733e-01,  4.9282e-01,  3.2317e-01,\n",
       "           1.4432e-01, -2.4270e-01, -3.1217e-01,  2.0787e-01,  1.3050e-01,\n",
       "          -2.0530e-02, -3.5504e-01,  1.5250e-01, -2.4855e-01, -4.3503e-01,\n",
       "           6.2152e-02,  1.0324e-01,  2.9098e-02, -4.7503e-01, -4.9068e-01,\n",
       "          -9.4770e-02, -4.9648e-02,  7.5107e-01,  2.5692e-02, -1.9927e-01,\n",
       "           1.3370e-01, -6.7683e-02, -4.7579e-01,  3.0414e-01, -3.7149e-02,\n",
       "          -3.0627e-01,  3.1311e-01, -6.4816e-01,  2.3139e-01,  3.8684e-01,\n",
       "          -1.6321e-01,  1.1935e-01,  6.4666e-02,  1.0916e-01,  7.9050e-02,\n",
       "          -2.9369e-01,  3.5165e-01,  1.6051e-02, -1.4870e-01,  1.0526e-01,\n",
       "           2.7475e-01,  1.5304e-01, -1.8087e-01, -1.1984e-01,  2.9899e-02,\n",
       "           5.0685e-01, -1.8950e-01, -1.4922e-01, -3.8122e-01, -5.7133e-01,\n",
       "           5.4754e-01,  2.2708e-01,  1.7308e-01,  1.3466e-01, -1.1493e+01,\n",
       "           2.2249e-01,  8.3642e-01, -4.3672e-01,  3.1150e-02,  2.0615e-01,\n",
       "           2.7600e-01, -2.5874e-01,  3.9136e-01, -4.7329e-01, -3.8457e-01,\n",
       "          -2.2588e-01, -4.8776e-02,  2.3161e-01, -4.3731e-01, -1.2419e-01,\n",
       "           2.6972e-02, -8.3028e-01,  2.5371e-01, -3.2031e-01,  3.9712e-01,\n",
       "          -5.0866e-01,  1.8938e-01, -2.5241e-02, -1.9351e-01, -4.0729e-01,\n",
       "          -3.4177e-01, -2.0075e-01, -1.6324e-02,  4.3752e-01, -6.3947e-02,\n",
       "          -5.7404e-02, -1.7993e-01,  8.2828e-02, -3.9104e-01, -5.4100e-01,\n",
       "          -1.1565e-01,  4.0165e-03,  4.1684e-01, -8.0974e-02, -3.1982e-01,\n",
       "           2.3412e-01, -2.8751e-01,  3.9954e-01, -5.6369e-02,  7.3551e-01,\n",
       "           6.5880e-01,  6.9980e-01, -8.5097e-03,  8.7980e-03,  1.4689e-01,\n",
       "          -7.6033e-01,  2.8972e-01, -5.9246e-01,  3.8414e-01,  2.2412e-01,\n",
       "           8.0944e-01, -6.9841e-01, -4.4841e-02, -1.0660e-01, -9.3357e-02,\n",
       "           4.7294e-01, -5.5637e-02,  4.8759e-01, -3.4475e-01,  3.0868e-02,\n",
       "           7.2032e-02,  7.9746e-02,  2.2373e-01, -2.7790e-01, -1.5704e-01,\n",
       "          -1.5446e-01,  9.1127e-01,  3.4400e-01, -1.0391e+00,  9.4545e-02,\n",
       "           3.6472e-01, -5.0740e-02,  5.6374e-01,  2.8168e-01,  4.0491e-02,\n",
       "           2.1923e-01,  6.1624e-01, -7.8321e-01, -5.3852e-01, -1.9404e-01,\n",
       "          -3.4772e-02,  3.4376e-01,  1.4675e-01,  6.2819e-01, -3.3004e-01,\n",
       "          -1.9104e-01, -2.3222e-01,  3.3707e-01,  1.4037e-01,  7.6669e-01,\n",
       "          -7.6945e-01, -1.9820e-01,  1.0504e-02, -1.4504e-02, -4.7963e-01,\n",
       "          -2.4422e-02, -4.7716e-01, -5.6844e-01,  5.3022e-01, -3.1606e-01,\n",
       "           5.0685e-02, -1.4149e-01,  4.5036e-02,  3.5321e-01, -3.9756e-01,\n",
       "           2.2672e-01,  9.6424e-02,  1.6372e-01, -9.8921e-02,  8.1474e-02,\n",
       "           1.4077e-01,  2.4373e-01, -6.2190e-01, -4.1214e-01, -1.2577e-01,\n",
       "          -1.4496e-01,  3.6630e-01,  1.4898e-01, -4.8501e-01, -8.8424e-02,\n",
       "          -5.4292e-01,  3.6838e-01,  1.7384e-01,  8.1007e-03,  3.3378e-01,\n",
       "           2.4016e-01,  6.0958e-01,  2.4129e-01, -3.6967e-01, -7.1860e-01,\n",
       "          -4.8793e-01, -8.1743e-02,  2.6068e-01, -3.8153e-01, -1.0318e-01,\n",
       "          -5.2585e-02, -1.0677e-01, -1.1081e-01, -2.4153e-03, -1.2347e-01,\n",
       "           3.4951e-01,  1.2959e-01, -4.2962e-01, -1.5834e-01, -1.3520e-01,\n",
       "          -4.3249e-01, -1.2131e-01, -2.2921e-01, -7.0514e-02, -3.9360e-01,\n",
       "          -1.8314e-01,  2.6073e-01,  1.0812e-01, -3.1528e-01,  6.4496e-02,\n",
       "           1.9663e-02,  6.7912e-02, -4.0204e-01,  4.4789e-01,  1.1130e-01,\n",
       "           1.5459e-01, -3.1662e-01, -4.7862e-01,  9.4838e-02,  6.7072e-02,\n",
       "           7.8780e-02, -6.2361e-02, -1.3876e-01, -5.5376e-01, -1.9798e-01,\n",
       "           1.9208e-02,  9.8863e-01, -1.8085e-01, -4.3229e-01,  3.8511e-01,\n",
       "           1.5964e-01, -2.9967e-01,  4.2781e-02,  1.6043e-01,  2.2051e-01,\n",
       "          -3.3644e-01, -5.0811e-01,  4.0136e-01, -1.3525e-01,  1.9756e-01,\n",
       "           4.5579e-01,  7.8091e-02, -2.2815e-01,  6.1733e-01, -2.3571e-01,\n",
       "          -1.7423e-01,  1.1553e-02, -4.1933e-01,  2.9725e-02,  8.8170e-02,\n",
       "           2.4123e-02,  4.3459e-01, -1.2417e-01,  3.4082e-03,  2.2043e-01,\n",
       "           2.3360e-01,  9.4740e-01, -1.1127e-02,  1.7081e-01,  1.7051e-01,\n",
       "           1.0722e-01, -1.7336e-01, -6.4538e-03, -1.2495e-01,  2.6057e-01,\n",
       "           1.8539e-01, -1.4245e-01, -2.1543e-01, -8.5940e-02,  1.7562e-01,\n",
       "           4.5967e-01, -6.9803e-01, -6.4368e-01, -1.4504e-01,  8.4782e-01,\n",
       "           2.8236e-02,  3.5781e-01, -1.0816e-02,  4.1620e-01,  5.8393e-01,\n",
       "          -9.1321e-01, -1.1648e-02, -3.2229e-01,  1.1161e-01, -6.4522e-02,\n",
       "          -2.4181e-01, -1.7231e-01, -2.8317e-01, -3.1975e-01,  3.6586e-01,\n",
       "           3.5205e-01, -6.7886e-01,  1.1388e-01,  2.5582e-01, -5.7093e-01,\n",
       "           1.0890e-01, -2.7624e-01,  7.3417e-02,  2.7470e-01, -3.5543e-02,\n",
       "          -8.9339e-02,  2.6486e-01, -5.7170e-01, -4.3055e-01,  1.0443e-01,\n",
       "           8.0751e-01,  1.5443e-02,  1.9650e-01, -6.7370e-02, -4.2684e-01,\n",
       "          -8.2392e-03, -3.3569e-01, -2.7009e-01,  3.1159e-01, -1.2203e-01,\n",
       "          -8.0634e-02, -4.0915e-01,  3.0815e-02,  3.3067e-01,  2.7344e-01,\n",
       "           2.9140e-01,  6.2994e-01,  5.6328e-01,  2.0175e-02, -2.2754e-01,\n",
       "          -6.0781e-02,  2.5357e-02,  2.6738e-02,  4.6445e-01, -2.5649e-01,\n",
       "          -2.6895e-01,  1.5698e-01,  1.7507e-01, -6.0609e-01,  1.7570e-01,\n",
       "          -3.9487e-01,  3.1059e-01, -1.9035e-02,  2.0927e-02,  5.3007e-01,\n",
       "           2.3251e-01,  2.2164e-01,  1.3739e-01,  3.1040e-01,  1.8886e-01,\n",
       "          -1.7927e-01,  8.0972e-01,  3.0657e-01,  3.0160e-01, -3.1236e-01,\n",
       "           1.4227e-01, -5.5068e-01, -3.0320e-01,  6.9929e-02,  2.9764e-02,\n",
       "          -1.0115e-01, -2.3232e-01,  2.1528e-01, -3.1784e-03,  7.8758e-01,\n",
       "           6.5278e-01, -3.1137e-01, -4.9148e-01, -6.1546e-02,  2.4809e-01,\n",
       "           8.6847e-02, -2.5838e-01,  1.8832e-01,  1.1013e-01,  5.8205e-02,\n",
       "          -7.0137e-01, -5.1429e-01, -2.2007e-01,  1.0481e-01, -4.6790e-01,\n",
       "           6.2939e-01, -6.2612e-02, -1.6385e-01,  2.8843e-02,  3.2251e-01,\n",
       "          -1.1435e-01,  7.6348e-01,  2.9217e-01, -5.0775e-01,  1.2245e-01,\n",
       "           1.7504e-01, -3.2113e-01,  6.6884e-02, -2.8081e-01,  1.9413e-01,\n",
       "           2.7819e-01,  7.2585e-01,  8.5032e-01]], device='mps:0'),\n",
       " tensor([[-0.3858, -0.7009, -0.8680,  ...,  0.1991,  0.3679,  0.4232],\n",
       "         [-0.2804, -0.8977, -0.7813,  ...,  0.0441,  0.7521,  0.7393]],\n",
       "        device='mps:0'),\n",
       " tensor([[-2.0871e-01, -4.9876e-01, -5.9414e-01, -8.6958e-01, -2.0575e-01,\n",
       "          -2.6170e-01, -7.8189e-01,  5.2303e-01,  1.2626e-01,  4.1574e-01,\n",
       "           7.6426e-02,  3.6450e-01,  1.1199e-01, -1.2855e-01, -2.0884e-01,\n",
       "          -4.2214e-01,  1.2372e-01,  4.1274e-01,  6.8074e-01, -1.6973e-02,\n",
       "           3.1205e-01,  5.5668e-01, -2.3586e-01,  4.1445e-03, -6.5374e-01,\n",
       "           2.1499e-01, -1.5975e-02,  3.6234e-01,  8.0418e-02, -5.8193e-01,\n",
       "           1.8039e-01,  4.5218e-01, -8.9967e-02,  1.2530e+00, -4.0645e-01,\n",
       "          -2.5601e-01,  1.2561e-01, -4.0902e-01, -2.9123e-01,  1.7117e-01,\n",
       "           7.1885e-01, -3.5627e-01, -2.7945e-02,  4.0516e-01, -5.8456e-02,\n",
       "           3.7839e-02,  1.9860e-01, -3.3710e-01,  4.0057e-01, -7.1950e-01,\n",
       "          -8.9293e-02, -1.4620e-01,  4.1353e-01,  4.8482e-01, -6.2407e-01,\n",
       "          -3.8267e-01,  3.1417e-01,  7.4246e-02, -1.2824e-01,  1.0762e-01,\n",
       "          -2.2779e-03,  2.5272e-01, -2.9454e-01,  2.7098e-01,  1.8962e-01,\n",
       "           2.1571e-01, -4.0370e-01, -5.4935e-01, -6.1328e-01, -1.4206e-01,\n",
       "           5.7840e-02,  1.0825e-01,  2.1299e-01, -1.4520e-01,  4.0326e-01,\n",
       "          -7.7670e-01,  1.1136e-01,  2.5204e-02, -1.8099e-01,  2.9084e-01,\n",
       "          -1.8048e-01,  1.5599e-01,  2.4780e-01,  6.6927e-02,  1.7369e-01,\n",
       "           1.4270e-01,  3.7688e-01, -3.4489e-02,  6.6987e-02,  3.6293e-01,\n",
       "           7.6463e-03, -5.8744e-02, -1.4522e-01, -2.4543e-01,  2.4448e-01,\n",
       "           6.9287e-02,  3.9222e-01, -3.3125e-01,  3.8264e-03, -3.1764e-01,\n",
       "           1.8923e-01,  1.1586e-01, -1.4359e-01,  1.8456e-01, -4.8470e-01,\n",
       "          -4.5091e-01, -4.7240e-03, -2.3053e-01, -4.8299e-01,  1.7109e-01,\n",
       "          -2.9637e-01, -1.2506e-01,  8.1158e-03, -4.5159e-01,  8.7590e-02,\n",
       "          -2.1949e-01,  5.4356e-02, -9.1063e-02, -4.1634e-01,  1.5827e-02,\n",
       "           1.2608e-01, -2.5516e-01,  2.3546e-01,  1.2292e-01,  5.8368e-02,\n",
       "          -3.7871e-02,  6.2178e-01,  4.0911e-02,  2.3701e-01,  3.0235e-01,\n",
       "           3.7875e-02, -3.0389e-02, -2.1747e-01, -2.1432e-01, -1.1928e-01,\n",
       "           7.7718e-02,  2.5177e-01, -7.3117e-02, -4.0501e-01, -2.4091e-01,\n",
       "           3.7901e-01, -2.1554e-01, -4.9826e-01, -3.7829e-01, -1.7658e-01,\n",
       "          -2.4278e-01, -2.2922e-01, -7.5678e-02, -3.2968e-01, -4.9567e-01,\n",
       "          -5.3347e-01, -4.0128e-01,  8.1063e-01, -9.0763e-02, -2.5443e-01,\n",
       "          -1.9396e-01,  8.9328e-03,  1.4070e-01,  1.0753e-01,  4.9118e-01,\n",
       "           6.6787e-01, -2.0246e-01,  1.1035e-01, -4.3649e-01, -3.2070e-01,\n",
       "           1.1602e-01,  2.0033e-01, -1.1290e-01,  3.4975e-01,  7.3701e-02,\n",
       "           1.3978e-01, -9.5759e-02,  4.4649e-01, -3.0171e-01,  2.1104e-02,\n",
       "          -7.3016e-01,  4.4867e-01,  2.9467e-01, -3.4026e-01, -5.6754e-04,\n",
       "           1.1658e-02, -1.1293e+00,  1.2940e-01, -9.6823e-01,  5.0640e-01,\n",
       "          -1.1057e-01,  1.0070e-01,  2.0418e-01,  3.4089e-01, -1.2391e-01,\n",
       "           3.2333e-01,  2.9416e-01,  3.0931e-01,  3.8118e-01,  4.9760e-02,\n",
       "           2.1625e-01,  3.3902e-01,  4.2265e-01,  2.2721e-01,  2.4747e-01,\n",
       "          -5.0337e-01,  3.6799e-01,  3.6112e-01, -1.9144e-01,  3.0299e-01,\n",
       "           6.7005e-02, -3.1862e-01,  2.2171e-01,  6.7960e-03,  6.9859e-03,\n",
       "          -2.5506e-01, -2.2087e-01, -5.5809e-01, -1.3544e-01, -3.7872e-01,\n",
       "           2.1537e-01, -5.0278e-01, -1.8713e-01,  3.6958e-01,  1.6668e-01,\n",
       "           2.0573e-02,  2.1924e-01,  1.9831e-01, -2.8480e-01,  2.4394e-01,\n",
       "           4.0098e-01, -4.7788e-01,  1.3331e-02, -2.5628e-01, -2.7256e-01,\n",
       "           2.1740e-01, -6.1896e-01,  5.8841e-01,  2.2207e-01,  1.9294e-01,\n",
       "           1.4724e-01, -2.1071e-01,  3.4637e-02, -4.7396e-01,  1.8130e-01,\n",
       "          -4.1949e-01, -3.6765e-01, -6.1252e-01, -5.4634e-02,  3.1208e-01,\n",
       "           1.8688e-01, -7.9167e-02,  7.0842e-01,  5.2501e-01,  4.6890e-01,\n",
       "          -3.9962e-01, -8.4200e-02,  4.4696e-01, -8.6341e-02, -5.3561e-01,\n",
       "           3.0263e-01, -9.4906e-02, -4.7518e-01,  2.7505e-01, -2.5621e-01,\n",
       "          -1.2143e-01, -1.2812e-01,  3.9635e-01,  2.5964e-01, -1.8858e-01,\n",
       "          -4.2993e-01,  1.3426e-01, -5.3275e-02, -4.0008e-01,  5.9552e-01,\n",
       "          -2.6969e-01,  8.1095e-02, -7.9348e-02, -3.3630e-01, -6.5401e-01,\n",
       "          -3.1556e-01,  2.7823e-01,  2.1916e-01,  1.4341e-01, -7.1258e-01,\n",
       "          -4.4175e-01,  3.4650e-01,  2.1553e-01,  2.2412e-01,  2.2438e-01,\n",
       "           5.6507e-01, -4.4041e-01, -3.0624e-02,  8.2854e-02, -3.9815e-01,\n",
       "          -1.2046e-01,  2.5992e-02, -4.2635e-01,  2.6059e-01,  3.6052e-01,\n",
       "          -6.1154e-02,  3.2284e-01,  4.8810e-01, -2.6220e-01, -4.8795e-01,\n",
       "          -4.7808e-02,  2.2443e-01, -2.6101e-01,  5.9009e-02,  1.7934e-01,\n",
       "           2.0202e-03,  5.1871e-01,  6.2178e-01, -8.1714e-02, -6.3018e-01,\n",
       "           3.4833e-01,  5.9592e-01, -1.6984e-01, -2.0466e-01,  1.8202e-02,\n",
       "          -8.6129e-01, -1.7917e-01, -1.5752e-01, -3.7799e-01, -7.8793e-01,\n",
       "          -2.7920e-02,  1.7062e-01, -1.7550e-02, -1.0860e-01, -4.3495e-01,\n",
       "           2.3539e-01,  2.9332e-01,  2.4454e-02,  3.7016e-01, -1.4201e-01,\n",
       "          -4.4208e-01, -2.9621e-01, -2.9567e-01, -5.4917e-01, -5.9259e-01,\n",
       "           2.3549e-01,  2.5834e-01,  1.1563e-01, -2.8733e-01, -2.8735e-01,\n",
       "           3.6816e-01, -5.8478e-01, -2.1360e-01,  3.2014e-01, -4.9393e-01,\n",
       "           1.5572e-03,  3.4664e-01,  3.0914e-01,  6.3538e-02, -1.5430e-02,\n",
       "           2.9558e-02, -4.1877e-01, -5.2197e-01, -2.4938e-01,  5.1848e-01,\n",
       "           4.3515e-01,  3.5848e-01, -3.5052e-01,  3.2910e-01,  4.6797e-01,\n",
       "          -3.3508e-01,  3.8546e-01,  1.7046e-01,  1.9616e-02, -1.8682e-01,\n",
       "           2.3862e-01,  4.9492e-01, -3.9239e-01,  4.6878e-02,  1.8320e-01,\n",
       "          -2.2409e-01, -2.0040e-01, -8.6438e-02,  2.7361e-01, -2.3964e-01,\n",
       "          -2.6304e-01, -1.9998e-01,  1.0600e-01, -1.6629e-01, -2.9217e-01,\n",
       "           3.4140e-01,  8.4418e-02, -2.1524e-02, -3.3230e-01, -4.0470e-01,\n",
       "          -1.7235e-01, -2.9933e-01,  7.4696e-01,  1.6856e-01, -1.9638e-01,\n",
       "           2.9832e-01, -1.4078e-01, -3.9336e-01,  5.7599e-01,  2.8038e-02,\n",
       "          -5.8993e-01,  5.3573e-01, -1.8986e-01,  1.3499e-01,  5.0660e-01,\n",
       "          -1.5537e-01,  4.4607e-02,  1.2620e-01,  3.4026e-01,  3.6860e-01,\n",
       "          -3.2623e-01,  5.5726e-02, -5.2839e-02, -3.9130e-01,  3.4303e-01,\n",
       "           3.2119e-01, -6.2571e-03, -2.8304e-01,  7.7711e-02, -6.4027e-02,\n",
       "           1.9304e-01,  9.2611e-02, -4.6455e-01, -4.2820e-01, -3.9517e-01,\n",
       "           3.1106e-01,  1.2182e-01,  5.1230e-01,  3.1006e-01, -1.1311e+01,\n",
       "           9.2453e-02,  8.3868e-01, -4.8985e-01, -2.5004e-01, -1.2443e-01,\n",
       "          -2.1783e-02, -8.0032e-02,  3.9286e-01, -1.6461e-01, -4.8514e-01,\n",
       "          -4.3399e-01, -2.1754e-02,  1.7438e-01, -3.7778e-01, -2.5964e-01,\n",
       "           5.2741e-02, -5.8080e-01,  4.2005e-01, -1.6131e-01,  5.0881e-01,\n",
       "          -3.7730e-01,  2.0370e-03,  1.3650e-01, -3.9081e-01,  1.8804e-02,\n",
       "          -4.9863e-02,  1.0768e-02,  2.5991e-01,  5.4637e-01, -6.5177e-02,\n",
       "           1.2398e-01,  3.6075e-03,  6.8569e-03, -3.7757e-01, -4.9889e-01,\n",
       "           2.0720e-01, -1.4206e-01,  6.7757e-01,  1.0958e-02, -7.7031e-01,\n",
       "          -1.8135e-01, -3.1349e-01,  2.4677e-01,  3.6372e-01,  8.8464e-01,\n",
       "           2.8866e-01,  1.0830e+00,  1.5554e-01,  1.4368e-01, -1.0674e-01,\n",
       "          -8.3551e-01,  8.1663e-02, -3.9913e-01, -8.6957e-02,  2.2264e-01,\n",
       "           2.9286e-01, -7.7238e-01,  2.2482e-01, -4.9196e-01,  2.6579e-02,\n",
       "           5.4107e-01,  4.8679e-02,  3.4684e-01, -1.1709e-01, -1.5736e-01,\n",
       "           1.7237e-01, -9.9095e-02,  2.5332e-01, -3.1391e-01, -3.5410e-01,\n",
       "          -1.6731e-01,  5.6287e-01,  3.8139e-01, -3.2635e-01,  2.1897e-01,\n",
       "           7.9485e-02, -2.2604e-01,  1.7638e+00, -5.5713e-02, -2.0248e-02,\n",
       "           1.5795e-01,  2.6175e-01, -5.0635e-01, -8.0817e-01,  1.9013e-01,\n",
       "          -1.4018e-01, -7.8020e-02,  2.6989e-01,  4.6437e-01, -1.2132e-01,\n",
       "          -6.0735e-01, -9.8240e-02,  1.4428e-01,  4.2380e-02,  5.5259e-01,\n",
       "          -6.2712e-01, -1.5767e-01, -1.7066e-01,  2.6589e-01, -1.0672e+00,\n",
       "           6.5837e-03, -6.8871e-01, -4.4072e-01,  5.6763e-01, -1.4612e-01,\n",
       "           1.0127e-01, -1.5710e-01,  4.7437e-02,  6.6199e-01, -1.8641e-01,\n",
       "           1.3039e-01, -1.4096e-01,  1.7725e-01, -2.3238e-01,  1.5225e-02,\n",
       "           4.4049e-01,  4.8209e-01, -3.9154e-01, -6.5160e-01, -7.2929e-02,\n",
       "           2.1625e-01, -4.7424e-02,  5.6579e-01, -5.5888e-01,  2.4073e-01,\n",
       "          -8.1339e-01, -4.9427e-02,  1.1117e-01, -1.6542e-02,  4.0595e-01,\n",
       "           1.1279e-01, -1.0126e-01,  2.1397e-01, -7.7083e-01, -4.9387e-01,\n",
       "          -7.0551e-01, -3.2717e-01,  2.2691e-01, -6.3818e-01, -1.9376e-01,\n",
       "          -1.5667e-01, -3.4681e-02, -7.7231e-02, -3.9630e-02,  6.9016e-03,\n",
       "           2.3901e-01,  1.6592e-01, -1.2631e-01, -1.1272e-01, -8.5773e-02,\n",
       "          -6.2325e-01, -5.3097e-02, -2.0006e-02, -2.5865e-02, -4.3998e-01,\n",
       "          -2.3641e-01,  1.1166e-02, -3.7871e-01, -1.6376e-01,  2.4717e-01,\n",
       "          -2.2092e-02, -1.0909e-02, -5.8804e-01,  7.0666e-01,  4.2763e-01,\n",
       "           2.1919e-02, -1.1676e-01, -3.4897e-01, -1.9956e-01,  2.9844e-01,\n",
       "           2.8413e-01, -6.3579e-02, -5.6798e-02, -2.8518e-01, -6.1177e-01,\n",
       "           1.4855e-01,  7.6289e-01, -4.3263e-02, -7.1570e-01,  2.6172e-01,\n",
       "           4.4940e-01, -2.0404e-01, -1.5525e-01,  1.2551e-01,  1.6531e-02,\n",
       "          -2.6583e-01, -5.9601e-01,  6.3958e-01, -1.4412e-01, -1.5950e-02,\n",
       "           2.5946e-01,  2.0613e-02,  3.2402e-02,  5.1515e-01,  1.3069e-01,\n",
       "          -3.7251e-01, -1.4628e-02, -6.7722e-01,  3.5876e-01, -7.1502e-02,\n",
       "          -3.4729e-01,  8.9149e-01,  2.3754e-01, -3.2885e-02, -1.7075e-02,\n",
       "           2.9793e-01,  6.3697e-01, -1.2535e-01,  1.6127e-01,  4.7927e-01,\n",
       "           1.4719e-01,  2.7537e-01,  1.5673e-01,  2.5670e-01,  1.1651e-01,\n",
       "           2.8032e-01, -1.2606e-01, -3.0652e-01,  1.8786e-03,  3.5073e-01,\n",
       "           3.2673e-01, -4.8570e-01, -6.6871e-01,  7.8409e-02,  5.4406e-01,\n",
       "          -1.5096e-01,  2.6833e-01, -1.8636e-01,  4.1238e-01,  4.2298e-01,\n",
       "          -8.4562e-01, -1.7709e-01, -8.6022e-02,  1.1428e-01, -1.1581e-01,\n",
       "          -3.3104e-01, -2.2862e-02, -2.9539e-01, -8.9189e-02,  6.6447e-01,\n",
       "           3.5855e-01, -7.6365e-01,  1.4278e-02,  7.7191e-01, -5.1617e-01,\n",
       "           3.4129e-01, -4.5815e-01,  7.1097e-02,  3.4258e-01,  2.2138e-01,\n",
       "           1.3320e-01,  1.4954e-01, -5.2462e-01, -4.4942e-01,  1.5006e-01,\n",
       "           9.4127e-01, -6.4293e-02,  1.3915e-01, -4.1095e-01, -2.0782e-01,\n",
       "          -1.0980e-01, -2.9655e-01, -3.1087e-01,  2.4989e-01,  3.9635e-02,\n",
       "           1.3150e-02, -3.1746e-02, -1.9109e-01,  2.3038e-01,  1.3108e-01,\n",
       "           1.9111e-01,  7.2513e-01,  5.6131e-01,  5.1935e-02, -2.5852e-01,\n",
       "          -1.9926e-01,  1.0344e-01, -1.9144e-01,  4.5723e-01, -3.2750e-01,\n",
       "          -3.0354e-01, -7.0665e-02,  2.9589e-01, -6.1322e-01,  1.2346e-01,\n",
       "          -5.0940e-01,  3.2378e-01, -1.7703e-01,  1.5883e-02,  2.1255e-01,\n",
       "           8.8231e-02,  1.0681e-01, -3.5827e-01,  5.1949e-01,  5.0278e-01,\n",
       "           1.2184e-01,  3.6589e-01,  1.1083e-01,  9.6130e-02, -2.9205e-01,\n",
       "           1.3590e-01, -1.0323e+00, -1.4879e-01, -1.2165e-01, -9.5606e-02,\n",
       "          -3.4406e-01, -6.6448e-02,  3.7965e-01,  3.2564e-01,  7.6110e-01,\n",
       "           6.4569e-01, -1.2733e-01, -5.5783e-01, -1.3392e-01,  2.0765e-01,\n",
       "           3.0355e-01, -1.8993e-01,  4.6326e-01, -3.6799e-02,  1.4314e-01,\n",
       "          -4.8614e-01, -4.6370e-01,  1.5581e-02,  3.8327e-01, -4.2768e-02,\n",
       "           6.8209e-01, -1.6766e-01, -1.9290e-01,  1.2383e-01,  1.9252e-01,\n",
       "           8.7829e-02,  9.3827e-01,  2.9546e-01, -5.7379e-01,  1.1189e-01,\n",
       "          -1.3077e-01, -2.4141e-01,  2.3854e-01, -5.0509e-01,  2.2848e-01,\n",
       "           1.1474e-01,  7.3053e-01,  6.3622e-01]], device='mps:0'),\n",
       " tensor([[-5.3230e-01, -3.7434e-01, -2.7748e-01, -8.3750e-01,  4.6736e-01,\n",
       "          -7.4365e-01, -5.5296e-01,  4.8026e-01, -2.7583e-01,  2.6447e-01,\n",
       "           3.4042e-01,  3.6712e-01,  3.6379e-01,  1.5507e-01, -2.8911e-01,\n",
       "          -3.2637e-03, -8.4825e-03,  2.1439e-01, -1.1276e-01,  4.9243e-01,\n",
       "           4.5846e-01,  5.6978e-01, -4.5187e-01,  2.0561e-01, -4.8338e-01,\n",
       "          -1.8313e-01,  1.4641e-01,  1.3601e-01,  1.2147e-01, -4.6028e-01,\n",
       "          -8.6405e-02,  2.7917e-01, -2.2965e-01,  1.2273e+00, -1.6564e-01,\n",
       "          -4.7997e-01,  1.4113e-01, -4.1191e-01,  4.8541e-01, -1.2163e-01,\n",
       "           6.3939e-01, -1.4727e-01,  3.8029e-01, -1.8161e-01,  4.5200e-01,\n",
       "           2.0389e-02,  3.3293e-01, -5.4026e-01, -3.5189e-02, -6.8944e-01,\n",
       "          -4.2472e-01, -9.4401e-02,  1.7926e-02,  1.2926e-01, -1.0414e+00,\n",
       "          -7.2053e-01,  2.3273e-01,  3.6721e-01,  2.1758e-01, -1.6788e-01,\n",
       "          -4.2119e-01,  6.1769e-02,  1.7082e-01,  5.8996e-02, -2.6845e-01,\n",
       "           7.8872e-03, -3.5073e-01, -5.0660e-01, -4.7743e-01, -4.0506e-01,\n",
       "          -1.6617e-01,  1.4891e-01,  9.2513e-02,  1.3410e-01,  5.4631e-02,\n",
       "          -1.1196e-01,  4.8648e-01,  9.7832e-02, -1.7808e-01,  5.9061e-03,\n",
       "           1.7129e-01,  3.1498e-01,  8.1381e-02, -5.2303e-01,  8.5022e-02,\n",
       "           2.1870e-01,  6.2716e-01,  1.9437e-01, -5.9017e-02, -4.4161e-01,\n",
       "           4.2555e-01, -1.3424e-01, -2.5779e-01, -5.3450e-02,  3.8629e-01,\n",
       "          -2.8927e-02,  5.9372e-01, -6.8174e-01,  4.8779e-01, -1.7815e-01,\n",
       "          -2.8955e-01,  1.8944e-02,  3.8203e-02,  2.1037e-02, -5.4069e-01,\n",
       "          -6.6443e-01,  6.9967e-02,  2.0752e-01, -7.0623e-01,  3.0378e-01,\n",
       "           1.6624e-01, -4.3843e-01,  1.9408e-02, -2.2351e-01, -1.6801e-01,\n",
       "           2.1498e-01,  2.7790e-01,  7.5247e-02, -3.2597e-01,  3.0509e-01,\n",
       "           6.0853e-01, -4.1746e-01,  4.4037e-01,  2.9959e-03,  2.0515e-02,\n",
       "          -4.1476e-01,  2.3433e-01, -3.5780e-01,  1.9038e-01,  4.3458e-01,\n",
       "          -7.7449e-02, -1.1319e-02, -4.0862e-01,  9.2312e-02,  5.5144e-01,\n",
       "           1.7477e-01,  2.0581e-02, -2.2520e-01,  1.7076e-02,  1.1837e-01,\n",
       "           3.0854e-01, -1.6565e-01,  5.3021e-02, -3.5114e-02, -1.1192e-01,\n",
       "          -1.1187e-01, -1.4971e-01,  7.3960e-03, -1.9968e-01, -6.7628e-01,\n",
       "          -4.0996e-01, -4.6535e-01,  5.6852e-01,  2.2836e-01, -2.4100e-01,\n",
       "          -1.8745e-01,  7.1285e-01, -6.0576e-02,  1.5726e-01, -1.6336e-01,\n",
       "           3.6543e-01, -3.4895e-01, -1.1418e-01, -7.4171e-01,  1.0692e-02,\n",
       "           1.4290e-01, -2.0915e-01, -3.5380e-01,  1.8388e-01, -4.1876e-02,\n",
       "           1.3378e-01, -2.2297e-01,  6.0775e-01, -2.5886e-02,  1.1552e-01,\n",
       "          -6.4562e-01,  2.8683e-01, -1.0485e-01, -1.1777e-01,  1.7262e-01,\n",
       "           6.3191e-01, -1.7292e+00,  5.7387e-01, -3.1557e-01, -8.1795e-02,\n",
       "          -2.6888e-01, -3.5055e-02,  1.3827e-01,  7.9208e-01, -2.5043e-02,\n",
       "          -3.2718e-01, -3.6192e-02,  4.9462e-02,  4.0358e-01, -2.4289e-01,\n",
       "           2.5715e-01,  6.4252e-01,  1.1524e-01, -8.3911e-03,  2.3249e-01,\n",
       "          -5.9467e-01,  2.8160e-01,  4.4210e-02, -6.3666e-02, -6.1773e-02,\n",
       "           1.7378e-01, -3.8367e-01,  2.1466e-01,  3.2612e-01,  1.7329e-01,\n",
       "          -2.8217e-01, -4.7429e-01, -3.7362e-01,  1.5759e-01,  2.7506e-01,\n",
       "           5.5391e-01, -4.8149e-01,  3.2643e-01,  7.4560e-01,  3.4716e-01,\n",
       "           4.6769e-02,  1.0718e-01,  6.5965e-02,  5.2699e-02,  1.3819e-01,\n",
       "           4.9464e-01, -2.1187e-01,  2.7621e-02, -2.9196e-01, -5.5734e-02,\n",
       "          -2.5460e-01, -5.1236e-01,  6.1117e-01, -3.7956e-02, -2.6380e-02,\n",
       "           1.6835e-01, -1.8896e-01,  3.3723e-01,  4.5381e-01, -8.6729e-02,\n",
       "          -1.7718e-01, -4.9539e-01, -4.6380e-01,  1.8952e-01,  1.0787e-01,\n",
       "           4.2249e-01, -7.5633e-02,  1.6650e-01,  8.7875e-01,  4.5260e-01,\n",
       "           2.0047e-01,  7.6071e-02,  2.4738e-01,  2.4385e-01, -4.7367e-01,\n",
       "          -2.5442e-02,  2.5930e-01, -5.9698e-01,  5.4240e-01, -4.3615e-01,\n",
       "          -1.1024e-01,  4.8938e-03,  4.1022e-01,  6.0264e-01, -4.9241e-01,\n",
       "          -1.5806e-01,  8.6648e-01, -4.2046e-01, -1.4270e-01,  2.9633e-01,\n",
       "          -1.5103e-01,  3.4869e-01, -4.2407e-01, -2.7265e-01, -7.8390e-01,\n",
       "          -5.0319e-01,  4.8404e-03, -1.8550e-03,  5.1285e-02, -4.9531e-01,\n",
       "           3.1280e-01, -4.7841e-01, -2.3042e-01, -3.3602e-01, -5.4181e-01,\n",
       "           1.7470e-01, -2.0786e-01,  1.9024e-01, -6.0885e-01, -6.5904e-01,\n",
       "          -6.3671e-01,  9.9059e-02, -3.7861e-01, -2.9118e-01,  2.8663e-01,\n",
       "           1.7978e-01,  2.4327e-01,  2.4089e-01,  1.0699e-01, -6.7883e-01,\n",
       "          -9.3429e-02, -2.0576e-01, -4.1798e-02,  2.2823e-02,  3.6082e-01,\n",
       "           4.9263e-02,  3.4851e-01,  8.7198e-01, -1.4006e+00, -5.0425e-01,\n",
       "           1.8903e-01,  5.0604e-01, -1.1216e-01, -9.8676e-02, -1.2527e-01,\n",
       "          -1.1621e+00, -6.5192e-01, -7.3105e-01, -1.1744e-01, -3.2442e-01,\n",
       "          -2.9386e-03,  4.1493e-02, -2.8303e-01, -2.2407e-01,  2.3424e-01,\n",
       "           3.2200e-01,  1.0604e+00, -2.6451e-01,  4.4569e-01, -5.5983e-01,\n",
       "          -1.5369e-01, -2.0930e-01, -6.9985e-01, -4.2132e-01,  2.9607e-01,\n",
       "           7.8883e-01,  3.7814e-01, -1.9492e-01, -4.7823e-01, -8.7002e-02,\n",
       "           5.3735e-01, -3.3373e-01,  1.1109e-02,  5.5848e-01, -1.8769e-01,\n",
       "           4.2399e-01, -3.9516e-02,  7.4487e-02,  1.0242e-01,  1.0806e-01,\n",
       "          -1.0720e-02, -6.3601e-02,  3.7202e-01,  1.4944e-02,  6.2831e-01,\n",
       "           2.8829e-01,  2.5522e-02, -1.5944e-01,  2.9333e-01,  2.4616e-01,\n",
       "          -5.4602e-01,  2.6765e-02, -5.9415e-01, -3.0250e-01, -1.4120e-02,\n",
       "          -2.0698e-01,  3.4492e-01, -6.1123e-01,  2.4262e-01, -1.9477e-01,\n",
       "          -5.0365e-01, -5.0317e-01, -1.6089e-01, -1.3078e-02,  6.0296e-01,\n",
       "           2.2658e-02, -5.6395e-01, -2.0574e-01, -8.8919e-01, -2.1780e-01,\n",
       "           3.8817e-02, -1.9377e-01,  3.4397e-01, -6.6417e-01, -5.3322e-01,\n",
       "          -4.9852e-01, -2.2785e-01,  3.1839e-01,  4.4651e-01,  2.9522e-01,\n",
       "           4.5855e-02, -1.4396e-01, -8.2439e-03,  8.9682e-01,  8.5149e-02,\n",
       "          -1.1797e-01,  1.4183e-01, -7.7400e-01,  1.5284e-01, -5.3441e-02,\n",
       "          -5.6195e-01, -2.3426e-01,  2.9966e-01, -2.0841e-01,  4.0527e-01,\n",
       "          -3.1282e-01, -1.1345e-01,  5.1331e-01, -1.9962e-02, -6.2769e-02,\n",
       "          -7.3285e-02,  6.9835e-01, -2.5322e-01,  1.8115e-02,  2.4350e-01,\n",
       "           1.2071e-01, -3.0779e-01, -2.2747e-01,  4.0259e-01, -3.0085e-01,\n",
       "           5.1682e-01, -1.6437e-01, -4.7011e-02,  1.8385e-01, -1.0762e+01,\n",
       "           3.6586e-01,  6.5534e-01, -4.4795e-01, -1.0310e-01, -6.7439e-02,\n",
       "           2.5748e-01,  7.2740e-02,  5.6400e-02,  5.0059e-02, -3.9118e-01,\n",
       "          -5.4658e-01,  2.9601e-01,  4.5796e-01, -4.8568e-01, -4.1870e-01,\n",
       "          -2.6520e-01, -1.0797e+00,  3.9352e-02, -4.6453e-02,  3.2696e-01,\n",
       "          -3.1066e-01, -2.0245e-02,  3.1021e-01, -2.9549e-01, -1.0476e-01,\n",
       "          -2.2842e-01, -3.0065e-01,  4.8820e-01,  4.6100e-01,  2.4646e-01,\n",
       "           2.5389e-01,  1.2526e-01,  1.5482e-01, -3.8053e-01, -6.3707e-01,\n",
       "           1.5653e-01,  1.3925e-01,  6.2456e-01,  4.3380e-01, -2.9348e-02,\n",
       "          -5.1111e-01, -1.2092e-01,  4.6362e-01,  7.4935e-01,  7.9519e-01,\n",
       "           1.2107e+00,  1.1891e+00,  4.9694e-01,  8.6219e-02, -5.9371e-01,\n",
       "          -7.1803e-01,  2.0157e-01, -4.3192e-01,  1.4641e-01,  5.8745e-02,\n",
       "           2.3521e-01,  2.0899e-02,  1.4241e-01, -1.2085e-01,  7.8796e-02,\n",
       "           2.3736e-01,  1.9741e-01,  2.3760e-01, -4.9930e-01,  2.0935e-02,\n",
       "           1.3104e-01,  9.0919e-02,  7.0643e-02, -3.6307e-01,  4.3179e-03,\n",
       "           6.5936e-03, -1.2740e-02,  9.5049e-02,  2.3304e-01,  3.6611e-01,\n",
       "          -2.6338e-01, -4.7392e-01,  1.0544e+00, -3.0837e-01,  3.2186e-01,\n",
       "          -3.8621e-02, -2.6583e-01, -2.4214e-01, -7.3968e-01, -2.4816e-01,\n",
       "          -1.9444e-01, -3.4437e-01,  6.2381e-01,  2.4582e-01, -6.0803e-01,\n",
       "          -4.3154e-01, -1.1287e-02,  3.5091e-01,  3.6997e-01,  9.9784e-01,\n",
       "          -4.5563e-01, -1.3615e-01,  1.4798e-01,  3.8381e-01, -2.5122e-01,\n",
       "          -2.3582e-01, -6.4553e-02, -3.5431e-01, -1.8344e-02, -4.5881e-01,\n",
       "           6.5218e-02,  3.0300e-01,  1.3348e-02,  2.2037e-01, -6.6509e-02,\n",
       "           1.6781e-01, -2.0484e-01,  5.3065e-01, -4.0241e-01, -6.5079e-01,\n",
       "           3.1882e-01,  7.8334e-01, -3.0670e-01, -6.0818e-01, -9.8141e-02,\n",
       "           3.9265e-01,  4.0078e-01,  2.0486e-01, -3.7263e-01,  4.1634e-01,\n",
       "          -6.2793e-01, -1.9618e-01, -6.1121e-02,  3.6308e-01,  4.9872e-01,\n",
       "          -5.2018e-02, -6.5910e-01,  7.1759e-02, -4.2549e-01, -3.6892e-02,\n",
       "          -7.0741e-01, -2.7361e-01,  3.4803e-01, -7.5931e-01,  1.4836e-01,\n",
       "           6.6230e-02, -3.3253e-01, -4.7032e-01,  4.8963e-01,  1.5585e-01,\n",
       "          -2.3160e-01,  3.3784e-01,  2.1051e-01,  2.8534e-01, -3.0460e-02,\n",
       "           6.6039e-02, -3.3441e-03, -2.4155e-01,  1.6999e-01,  3.2059e-02,\n",
       "          -6.1326e-01, -8.3539e-02, -2.1572e-01, -7.5685e-01,  4.1046e-01,\n",
       "           2.9290e-01, -1.8804e-01, -3.4822e-01,  7.4446e-01,  1.4568e-01,\n",
       "          -8.0214e-03, -4.5680e-01, -4.3768e-01, -2.5367e-01,  2.0053e-01,\n",
       "           1.5430e-01,  1.0774e-01, -5.7480e-01, -4.1209e-01,  1.0424e-01,\n",
       "          -7.2817e-02,  9.6974e-01, -3.8407e-01, -3.7306e-01,  2.3132e-01,\n",
       "          -8.9216e-02, -2.2055e-01,  3.1798e-01, -1.8854e-01,  2.2558e-01,\n",
       "          -1.3249e-01, -4.2717e-01,  2.1329e-01, -1.5381e-02,  3.8435e-01,\n",
       "          -9.2393e-02,  5.7524e-01,  3.3509e-01,  3.6297e-01, -4.4852e-01,\n",
       "          -4.9839e-01, -2.7539e-01, -6.1184e-01, -2.0528e-01,  2.2091e-01,\n",
       "           3.9591e-01, -3.9450e-01,  1.6429e-01,  3.1604e-01, -9.4758e-02,\n",
       "           6.2051e-01,  1.1286e+00, -1.5887e-01,  4.7650e-02,  6.4530e-01,\n",
       "           1.7233e-01, -1.0008e-01,  3.0189e-02, -6.8580e-02,  8.0284e-02,\n",
       "           4.4204e-01,  1.4167e-02, -1.7450e-01,  7.7722e-02,  2.7198e-02,\n",
       "           2.0596e-01, -2.2528e-01, -6.7438e-01, -1.3766e-01,  5.8357e-01,\n",
       "          -2.6077e-01,  3.9860e-01, -4.8429e-01,  4.6246e-01,  2.8723e-01,\n",
       "          -3.5205e-01,  1.6034e-01, -2.3014e-01, -3.8359e-01, -4.6960e-01,\n",
       "          -1.5234e-01,  3.5962e-02, -4.9875e-01, -1.5892e-01,  1.5349e-01,\n",
       "           3.9535e-01, -7.9673e-01,  1.6656e-01,  4.4208e-01, -3.8840e-01,\n",
       "           3.4154e-01, -2.9838e-02,  5.9380e-02,  7.2580e-01, -1.0732e-01,\n",
       "          -1.3108e-01,  3.2219e-01, -9.0424e-01, -5.4071e-01,  1.4301e-01,\n",
       "           8.1563e-01,  1.0219e-01, -2.4513e-01, -2.0630e-01, -9.0575e-02,\n",
       "          -1.7266e-01, -1.1246e-02,  1.1110e-01,  7.1624e-02,  1.0531e-01,\n",
       "          -1.3956e-01,  1.4590e-01,  1.1162e-01,  1.6787e-01, -7.2645e-01,\n",
       "          -3.9216e-03,  3.8171e-01,  1.2996e+00,  4.7738e-02, -6.1268e-01,\n",
       "           3.3729e-02,  2.1100e-01, -2.6508e-01,  7.9902e-01,  2.1337e-02,\n",
       "          -1.8471e-01,  3.8837e-02,  8.5767e-02, -2.0767e-01,  6.8850e-02,\n",
       "          -4.4106e-01,  4.3241e-02,  7.9932e-02, -2.9122e-01,  8.3575e-01,\n",
       "          -6.0383e-02,  4.0893e-01, -8.2550e-01, -2.0033e-01,  3.5489e-01,\n",
       "          -2.0466e-01,  8.1066e-02,  3.5284e-01,  3.2908e-01,  2.6528e-01,\n",
       "          -3.0514e-01, -4.9820e-01,  1.3842e-01, -2.7749e-01, -2.3379e-01,\n",
       "          -2.4558e-01, -4.7470e-01,  2.5197e-01,  9.1438e-02,  7.5358e-01,\n",
       "           7.9870e-01,  8.6452e-02, -8.2657e-02, -4.7943e-01, -1.0593e-01,\n",
       "           4.1662e-01, -3.0806e-01,  1.9287e-01, -3.4246e-01, -5.0206e-02,\n",
       "          -1.5981e-01, -2.8099e-01,  2.1671e-01,  2.4583e-01, -7.0620e-02,\n",
       "           8.1360e-01,  1.2432e-01,  2.7185e-01,  7.5472e-01,  4.1060e-01,\n",
       "          -2.1709e-01,  8.0328e-01, -6.0699e-02, -8.1921e-01, -7.5192e-02,\n",
       "           5.5815e-01, -2.5974e-01,  4.8287e-01, -8.5323e-01,  1.7709e-02,\n",
       "           1.0471e-01,  7.5396e-01,  2.4214e-01]], device='mps:0'),\n",
       " tensor([[-0.4077, -0.7226, -0.3781,  ...,  0.1806, -0.0905,  0.5976],\n",
       "         [-0.3236, -0.8805, -0.2364,  ...,  0.5745, -0.0029,  0.3892]],\n",
       "        device='mps:0'),\n",
       " tensor([[-3.1341e-01, -5.7095e-01, -4.0753e-01, -7.5548e-01, -2.6188e-01,\n",
       "           9.8577e-03, -9.4136e-01,  4.8246e-01,  1.4272e-01,  3.0358e-01,\n",
       "           2.1854e-01,  4.2295e-01,  7.3523e-02, -4.6327e-02, -2.3695e-01,\n",
       "          -1.3734e-01, -8.3121e-03,  4.7490e-01,  3.4465e-01,  4.8662e-01,\n",
       "           2.7444e-01,  4.0487e-01,  2.4706e-02, -6.0735e-02, -6.1204e-01,\n",
       "           4.3389e-01, -6.2464e-02, -5.4892e-03,  1.2547e-01, -6.7002e-01,\n",
       "           8.8170e-02,  5.0359e-01,  6.5773e-02,  1.0928e+00, -2.1165e-01,\n",
       "          -1.8833e-01, -1.6424e-02, -4.6113e-01,  1.9583e-01, -4.4738e-01,\n",
       "           4.5307e-01, -3.4665e-01, -2.5262e-03,  2.1954e-01, -2.3533e-01,\n",
       "           1.8935e-01,  3.7966e-01, -8.3258e-01,  3.1579e-01, -6.1189e-01,\n",
       "           1.1260e-01,  2.1042e-01,  5.0233e-01,  4.5420e-01, -4.6159e-01,\n",
       "          -2.8339e-01,  3.3737e-01,  6.6392e-02, -2.6335e-01,  2.5477e-01,\n",
       "          -9.5593e-02,  8.2668e-01,  9.9869e-02,  9.7756e-03,  3.6351e-01,\n",
       "           2.5066e-01, -3.3492e-01, -6.1001e-01, -5.4615e-01, -4.1348e-02,\n",
       "           1.7667e-01,  2.7966e-01,  1.4960e-02, -2.0603e-01,  2.0130e-01,\n",
       "          -5.6013e-01,  4.7607e-02, -1.9005e-01, -4.6904e-01, -9.6338e-02,\n",
       "           6.9779e-02,  1.5947e-01,  2.8026e-01, -1.8601e-01,  1.4694e-02,\n",
       "          -4.0735e-02,  1.8870e-01,  1.6958e-01,  1.2658e-01,  2.6546e-01,\n",
       "           5.2096e-01, -8.3658e-02, -2.9224e-01, -4.3071e-01,  4.2480e-01,\n",
       "           7.4965e-03,  3.1755e-01, -2.8713e-01,  3.3587e-02, -1.6724e-01,\n",
       "           5.5991e-01,  2.5666e-01, -8.5053e-02,  2.5083e-01, -4.6546e-01,\n",
       "          -6.6041e-02, -1.9302e-02, -1.6191e-01, -4.7724e-01,  3.5736e-02,\n",
       "          -1.3128e-01, -4.4065e-01,  7.9840e-02, -5.4848e-01, -2.6817e-01,\n",
       "           6.3977e-02,  7.2839e-03, -1.7838e-01, -1.8428e-01,  3.8729e-01,\n",
       "           2.2280e-01, -4.9525e-01, -5.2452e-02,  8.4705e-02,  1.3760e-01,\n",
       "          -2.0320e-01,  6.5043e-01,  3.3509e-01,  3.7986e-01,  1.7712e-01,\n",
       "           2.3862e-01, -2.7346e-02, -3.2809e-01, -7.1634e-02, -4.3334e-01,\n",
       "          -4.9746e-02,  1.1753e-01, -3.2914e-01,  1.0242e-01, -1.7714e-01,\n",
       "           1.0124e-01, -1.4951e-01, -3.1045e-01, -3.3118e-01, -8.1615e-02,\n",
       "          -3.4844e-01,  2.5460e-01, -2.6378e-01, -3.5184e-02, -2.5451e-01,\n",
       "          -5.6548e-01, -6.2583e-01,  3.6178e-01, -1.0381e-01, -1.8738e-01,\n",
       "          -1.0420e-01,  4.5892e-01,  3.3769e-01, -9.8365e-02,  2.8162e-01,\n",
       "           1.0042e+00, -1.3108e-01,  1.3012e-01, -6.2318e-01, -2.9023e-02,\n",
       "          -1.8672e-01,  4.5440e-02, -6.5459e-02,  3.5226e-01, -1.2345e-01,\n",
       "           1.7781e-01,  5.1436e-02,  9.3528e-02, -1.2729e-02,  6.7686e-03,\n",
       "          -7.1223e-01,  2.2040e-01, -7.7533e-02, -1.4331e-01,  1.2482e-01,\n",
       "          -3.5166e-01, -1.1101e+00,  2.4160e-01, -6.5350e-02, -6.9801e-02,\n",
       "          -2.1810e-01,  7.1948e-02,  2.1497e-01,  3.3463e-01,  9.2393e-02,\n",
       "          -4.6142e-02,  1.1646e-01,  1.6264e-01,  3.3201e-01,  2.1348e-01,\n",
       "           3.1139e-01,  3.9493e-01, -1.3339e-01,  2.4941e-01,  7.5572e-02,\n",
       "          -9.2905e-01,  2.4085e-01,  3.3137e-01,  1.4103e-02,  6.4565e-02,\n",
       "           1.6532e-01, -2.6756e-01,  1.9304e-01,  3.2880e-01, -8.5909e-02,\n",
       "          -2.2081e-02, -5.4574e-02, -4.1131e-01,  3.0800e-02, -1.0591e-01,\n",
       "           3.4454e-01, -4.7465e-01, -1.6673e-01,  4.0538e-01,  3.6121e-01,\n",
       "           1.2434e-01,  1.0588e-01,  1.4007e-01, -2.7085e-01, -7.1562e-02,\n",
       "           3.9040e-01, -1.9457e-01, -1.1644e-01, -3.3084e-01, -4.3002e-01,\n",
       "          -4.9459e-02, -3.4986e-01,  5.8448e-01,  1.5919e-01, -2.0105e-01,\n",
       "           9.8654e-02, -4.6320e-01, -1.1824e-02, -4.7723e-01, -6.9138e-04,\n",
       "          -3.2795e-01, -3.2493e-01, -6.1019e-01,  1.3242e-01,  2.4047e-01,\n",
       "           3.8375e-01, -5.7811e-02,  8.6985e-01,  3.0024e-01,  2.8871e-01,\n",
       "          -3.2892e-01,  1.3970e-02,  3.5238e-01,  4.8349e-01, -3.6528e-01,\n",
       "           7.5964e-02, -4.7367e-02, -5.4691e-01,  2.3242e-01, -4.5432e-01,\n",
       "           5.2836e-02,  1.4607e-01,  1.0275e-01,  1.1803e-01, -2.0916e-01,\n",
       "          -1.2443e-01,  2.8925e-01, -1.0495e-01, -5.3029e-01,  5.7800e-01,\n",
       "          -4.1488e-01,  2.0768e-01,  7.7810e-02, -4.3718e-01, -5.3427e-01,\n",
       "          -5.6545e-01,  1.4112e-01,  4.0549e-01,  2.6621e-01, -7.5665e-01,\n",
       "          -2.3255e-01,  2.0705e-01,  8.8220e-02,  7.8199e-02,  7.1394e-02,\n",
       "           4.4066e-01, -2.5942e-01,  1.3981e-01,  2.9926e-01, -4.7121e-01,\n",
       "           4.6010e-02, -5.2244e-02, -1.6941e-01,  1.5912e-02,  3.2312e-01,\n",
       "          -3.9849e-01,  1.5039e-01,  2.4047e-01, -2.0000e-01, -3.0815e-01,\n",
       "           1.2394e-01, -6.0731e-02,  1.1596e-01, -5.7990e-02, -3.3582e-03,\n",
       "           2.5182e-01,  5.0693e-01,  2.2612e-01, -1.5501e-01, -5.9781e-01,\n",
       "          -1.1146e-01,  5.0861e-01, -1.1162e-01,  2.4937e-01,  9.6753e-02,\n",
       "          -1.0496e+00,  6.5761e-02, -2.1311e-01, -3.1728e-01, -7.3192e-01,\n",
       "           1.4806e-01,  9.6083e-02, -2.0472e-01,  7.6619e-02, -6.1380e-01,\n",
       "           2.9803e-01,  4.2681e-02,  6.2848e-02, -1.5193e-02, -4.8420e-01,\n",
       "           1.5610e-01, -7.1635e-01, -5.1463e-01, -3.3654e-01,  1.7288e-01,\n",
       "           8.0921e-02,  2.2109e-01,  2.1822e-01, -2.0376e-01, -3.7667e-01,\n",
       "           7.9054e-01, -4.4562e-01,  2.8469e-01,  2.0878e-01, -4.3663e-01,\n",
       "          -4.6714e-02,  3.5996e-01, -4.1555e-01, -5.7016e-02,  3.5138e-01,\n",
       "           3.4592e-01, -2.2654e-01, -3.8532e-01, -3.8368e-02,  7.5555e-01,\n",
       "           5.1247e-01,  3.7169e-01, -1.0941e-01,  3.0974e-01,  3.4570e-01,\n",
       "          -4.5993e-01,  5.6363e-01,  1.3277e-01, -4.5380e-02,  7.7913e-02,\n",
       "           1.1091e-01,  2.3592e-01, -4.8310e-01,  1.6505e-01, -4.0184e-01,\n",
       "           3.2343e-02, -3.7550e-01, -1.6358e-01,  3.1340e-01,  1.5119e-01,\n",
       "          -2.5759e-01, -7.1917e-01,  2.9901e-02, -1.7346e-01, -3.5152e-01,\n",
       "           3.3410e-01,  4.8810e-02, -3.7420e-02, -3.1801e-01, -6.1229e-02,\n",
       "          -6.3560e-02, -1.4276e-01,  6.1491e-01,  2.9910e-01, -3.0677e-01,\n",
       "           2.3434e-01, -2.9881e-01, -2.6499e-01,  6.6745e-01, -1.7061e-01,\n",
       "          -4.6629e-01,  5.3143e-01, -4.1484e-01, -9.9874e-02,  9.6444e-01,\n",
       "          -2.5741e-01,  1.6482e-01,  2.3685e-01, -1.0921e-02,  1.4634e-02,\n",
       "          -3.2154e-01,  3.6309e-01,  5.5193e-02, -2.7384e-01,  4.2327e-01,\n",
       "           8.3210e-02,  2.1747e-01, -1.8861e-01, -2.1404e-01, -1.0413e-01,\n",
       "           3.2530e-01, -2.6872e-01, -2.6064e-01, -3.2979e-01, -7.6837e-01,\n",
       "           5.4798e-01,  1.6131e-01,  4.8875e-01,  4.4584e-01, -1.1400e+01,\n",
       "           2.8342e-01,  8.1003e-01, -6.5410e-01, -1.6323e-01,  1.3335e-01,\n",
       "           7.5099e-02, -5.0939e-01,  4.2801e-01, -1.8365e-01, -2.0401e-01,\n",
       "          -4.7192e-01, -4.7072e-02,  1.2400e-01, -2.3607e-01, -6.0517e-01,\n",
       "          -6.5091e-02, -8.5080e-01,  2.5193e-01, -2.6283e-01,  4.6670e-01,\n",
       "          -2.6378e-01,  2.8205e-03,  2.0961e-01, -2.3548e-01, -2.3070e-01,\n",
       "          -3.2914e-01,  2.4508e-01,  3.8506e-01,  2.0288e-01,  6.3704e-02,\n",
       "           1.0473e-01, -2.6243e-02,  3.5746e-01, -1.0475e-01, -3.9611e-01,\n",
       "          -1.8350e-01, -1.1008e-01,  1.0008e-01, -3.4772e-01, -4.5893e-01,\n",
       "          -8.4113e-02, -2.1002e-01,  3.5841e-01,  4.1556e-01,  8.5826e-01,\n",
       "           6.2736e-01,  9.9876e-01,  2.3202e-01, -6.2909e-02,  6.4073e-02,\n",
       "          -8.8353e-01,  3.9291e-01, -7.0052e-01,  1.3040e-01,  1.1506e-01,\n",
       "           5.2461e-01, -7.5267e-01,  2.3667e-01, -2.7701e-01, -5.2831e-02,\n",
       "           2.5409e-01,  2.0178e-01,  2.4785e-02, -3.5551e-01,  1.9275e-01,\n",
       "           3.8966e-01, -1.2825e-01, -1.2054e-01, -4.0165e-02,  3.0375e-02,\n",
       "          -2.0463e-01,  6.7562e-01,  3.8821e-01, -3.7788e-01,  1.9271e-01,\n",
       "          -1.7668e-01, -7.2565e-02,  1.3805e+00, -6.2203e-02,  2.0208e-01,\n",
       "           2.5923e-01,  3.5580e-01, -2.0421e-01, -5.8102e-01,  4.1444e-01,\n",
       "          -7.0078e-02,  2.6108e-01,  4.3656e-01,  6.6531e-01,  2.0911e-01,\n",
       "          -7.0457e-01,  1.1270e-02,  9.7785e-02,  1.4612e-01,  5.3300e-01,\n",
       "          -5.3500e-01, -1.3245e-01,  7.4544e-02,  1.6588e-02, -6.3760e-01,\n",
       "          -1.2052e-01, -5.6344e-01, -4.3139e-01,  1.1516e-01, -4.2808e-01,\n",
       "           1.7599e-01, -5.0486e-02, -7.5371e-02,  4.8919e-01,  1.7079e-02,\n",
       "          -1.4432e-01, -1.4419e-01,  6.6394e-03, -4.7681e-02, -1.1487e-01,\n",
       "           2.0094e-01,  8.5936e-01, -6.3309e-01, -5.8914e-01, -3.5178e-01,\n",
       "          -3.5786e-01,  3.8473e-01,  1.3381e-01, -2.5522e-01,  7.1022e-01,\n",
       "          -4.1201e-01,  2.5257e-01,  1.2223e-01, -2.3180e-01,  3.0817e-01,\n",
       "           1.9926e-01, -5.3706e-02,  2.6143e-01, -3.9642e-01, -4.7648e-01,\n",
       "          -6.6862e-01, -1.6199e-01,  4.8756e-01, -2.4331e-01, -8.8962e-03,\n",
       "          -4.2210e-01,  2.3546e-01, -6.6005e-02,  1.4649e-01,  1.4088e-01,\n",
       "          -6.1891e-02,  1.7960e-01, -7.3315e-02,  2.3961e-01, -1.0460e-01,\n",
       "          -7.2381e-01,  1.8874e-02,  2.5986e-03, -8.8239e-02, -6.7532e-01,\n",
       "          -2.4571e-01, -1.4245e-02, -1.6801e-01, -1.1654e-01,  2.7921e-01,\n",
       "          -1.4666e-01, -6.8386e-02, -3.2985e-01,  6.7817e-01,  4.9017e-01,\n",
       "           1.5438e-01, -4.5613e-01, -1.7605e-01,  3.3145e-01,  1.8773e-01,\n",
       "           1.4792e-01,  1.6632e-01, -3.3628e-01, -3.8058e-01, -5.3098e-01,\n",
       "          -7.0421e-02,  6.8357e-01, -3.8562e-01, -5.8930e-01,  2.4393e-01,\n",
       "           3.6514e-01,  1.6591e-02, -2.3971e-01, -6.6128e-02, -1.6800e-01,\n",
       "          -4.9923e-01, -3.5262e-01,  5.4339e-01, -2.3365e-02,  5.5950e-02,\n",
       "           5.6255e-01,  1.1248e-01, -1.9319e-01,  7.5874e-01, -2.0498e-01,\n",
       "          -3.5044e-01, -3.4642e-02, -5.5229e-01,  1.9261e-01,  1.5678e-01,\n",
       "          -5.3358e-01,  5.2742e-01, -4.9824e-02, -1.6713e-01,  2.5695e-01,\n",
       "           4.4338e-01,  1.0443e+00, -5.3030e-01, -2.1690e-01,  5.9020e-01,\n",
       "           3.8117e-02,  2.4558e-02,  1.1598e-01, -2.2553e-01,  3.4843e-01,\n",
       "           3.0657e-01, -4.5122e-01, -2.7735e-01, -2.9579e-02,  1.0294e-01,\n",
       "           4.7629e-01, -8.7946e-01, -2.9600e-01, -1.4609e-02,  7.9245e-01,\n",
       "          -2.8161e-01,  3.6256e-01,  7.4960e-02,  2.7479e-01,  3.7889e-01,\n",
       "          -1.2627e+00, -1.8217e-01, -4.8292e-01,  5.8311e-02, -2.4320e-02,\n",
       "          -3.1454e-01, -3.0194e-01, -6.6876e-02, -1.0495e-01,  4.7766e-01,\n",
       "           1.4597e-01, -7.5507e-01,  1.6905e-01,  5.9199e-01, -7.5051e-01,\n",
       "           4.2879e-01, -5.5707e-02,  1.1767e-01,  3.0892e-01, -7.5338e-02,\n",
       "           1.4539e-01, -3.9236e-02, -5.7126e-01, -4.3480e-01,  3.8497e-01,\n",
       "           8.5192e-01,  1.4653e-02,  2.6704e-02,  1.3127e-01, -2.9082e-01,\n",
       "          -2.3343e-01, -1.0252e-01, -1.5074e-01, -2.4859e-01,  5.1916e-02,\n",
       "          -8.1242e-02, -3.8516e-01, -2.0871e-01,  2.5418e-01,  6.9897e-02,\n",
       "           2.2001e-01,  6.7344e-01,  4.2265e-01, -9.6083e-02, -1.1523e-01,\n",
       "          -5.0568e-02,  4.9249e-02, -8.5899e-02,  3.9826e-01, -4.6943e-01,\n",
       "          -3.7784e-01,  2.8655e-02,  2.6015e-02, -5.4447e-01,  1.7322e-01,\n",
       "          -3.7485e-01,  4.4039e-01, -4.3018e-01, -2.2480e-01,  5.3154e-01,\n",
       "           1.8020e-01, -3.1669e-05, -4.9497e-01,  1.3084e-01,  4.6058e-01,\n",
       "          -1.0373e-02,  2.9664e-01,  4.0699e-01,  2.2199e-01, -2.8259e-01,\n",
       "           1.9721e-01, -1.0816e+00, -1.0269e-01, -2.8844e-01,  1.2631e-01,\n",
       "          -8.5308e-02, -1.8965e-01,  2.7264e-01,  1.9703e-01,  5.6104e-01,\n",
       "           5.5033e-01, -2.9487e-01, -6.0137e-01, -1.0005e-01,  7.8713e-02,\n",
       "          -9.0349e-02, -3.2882e-03,  3.5229e-01,  2.2693e-02,  2.4294e-01,\n",
       "          -5.1701e-01, -3.8123e-01, -2.4116e-01,  3.9837e-01, -1.0992e-01,\n",
       "           9.6358e-01, -2.9994e-01, -5.5056e-02,  6.9097e-01,  6.5120e-02,\n",
       "           6.9286e-03,  8.8120e-01,  6.5911e-02, -5.7396e-01, -3.7155e-02,\n",
       "           3.0318e-02,  6.7666e-02, -1.7774e-01, -5.3636e-01,  3.2825e-01,\n",
       "           4.0580e-02,  3.4102e-01,  4.4494e-01]], device='mps:0'),\n",
       " tensor([[-0.1377, -0.6164, -0.4381,  ...,  0.2714,  0.0332,  0.5972],\n",
       "         [-0.3292, -0.2590, -0.5016,  ..., -0.0148,  0.3509,  0.1607]],\n",
       "        device='mps:0'),\n",
       " tensor([[-0.2330, -0.4411, -0.4565,  ...,  0.0257,  0.5397,  0.4576],\n",
       "         [-0.0118, -0.4835, -0.6737,  ...,  0.0717,  0.7745,  0.4633]],\n",
       "        device='mps:0'),\n",
       " tensor([[-3.6120e-01, -7.8905e-01, -5.0390e-01, -7.1751e-01, -5.0848e-01,\n",
       "          -2.5903e-01, -4.6083e-01,  8.4162e-01, -1.8370e-01,  3.5557e-01,\n",
       "           5.2342e-02, -4.4721e-02,  1.0798e-01, -3.5564e-01, -9.1665e-01,\n",
       "          -1.6942e-01,  3.7399e-02,  2.4044e-01,  6.0382e-01,  4.0347e-01,\n",
       "           7.6676e-01,  2.5545e-02,  1.3427e-01, -1.1481e-01, -4.1879e-01,\n",
       "           3.5593e-01, -1.7092e-01,  5.9363e-02,  5.1618e-01, -4.8417e-01,\n",
       "          -3.4153e-02,  4.2593e-01, -3.5590e-01,  9.3423e-01, -1.0669e-01,\n",
       "          -2.4920e-01,  1.3935e-01, -2.7026e-01,  2.7001e-01, -1.4552e-01,\n",
       "           3.6396e-01,  2.4605e-02,  3.2007e-01,  7.8465e-02,  1.7156e-01,\n",
       "           6.1884e-02, -6.2117e-02, -5.7154e-02, -1.5552e-01, -7.9268e-02,\n",
       "           2.0746e-01, -2.8734e-02,  4.0402e-01,  4.0476e-01, -5.9948e-01,\n",
       "          -3.0309e-01, -1.6778e-02,  9.9904e-02, -7.3945e-01,  2.2545e-01,\n",
       "           3.7677e-03,  7.4284e-01,  2.6843e-01,  1.7773e-01,  6.2669e-02,\n",
       "          -1.5334e-01, -6.8920e-01, -6.2121e-01, -5.2099e-01,  5.1783e-03,\n",
       "           1.3491e-01,  3.3806e-01,  4.3066e-02, -4.0168e-02,  1.4309e-01,\n",
       "           1.0541e-01,  1.2187e-01, -3.5372e-01, -7.7031e-02, -5.9915e-01,\n",
       "           6.9437e-02, -2.3010e-01, -6.6999e-02, -1.1261e-01, -2.8000e-01,\n",
       "          -2.5507e-01,  4.1908e-01,  5.8319e-03,  3.2745e-01, -1.8378e-02,\n",
       "           4.6842e-01,  6.0292e-02, -3.0946e-01, -2.0062e-01,  5.1059e-01,\n",
       "          -3.7545e-01,  4.8624e-01, -3.0415e-01, -3.1175e-01, -3.0402e-01,\n",
       "           9.5717e-02,  3.6020e-01, -4.1639e-02,  1.9650e-01, -6.0948e-01,\n",
       "          -3.6407e-01,  1.8219e-01,  1.0345e-01, -6.8707e-01, -6.0942e-02,\n",
       "          -1.2020e-01, -3.4137e-01, -6.7478e-03, -5.5366e-01, -3.4280e-01,\n",
       "           3.3223e-02, -3.0145e-01, -3.7612e-01, -2.2604e-01,  8.4775e-01,\n",
       "           5.2491e-01, -3.0803e-01,  1.1399e-01, -1.5878e-01,  9.5237e-02,\n",
       "          -2.2838e-01,  4.0001e-02, -4.6209e-02,  2.4139e-01,  3.7528e-01,\n",
       "           2.0085e-02,  6.8270e-02, -2.6732e-01,  5.9443e-02, -9.6104e-03,\n",
       "          -2.7478e-01,  5.1452e-02, -5.3484e-01,  6.1006e-02,  2.5778e-01,\n",
       "          -3.3596e-01,  3.0767e-02, -3.0067e-01,  3.9098e-02,  1.2264e-01,\n",
       "          -8.4187e-02,  9.1865e-02, -5.3766e-02, -3.2444e-01, -1.0787e-02,\n",
       "          -6.4091e-01, -4.1750e-01,  3.8454e-01, -1.8327e-01,  6.1051e-02,\n",
       "          -2.4053e-01,  7.0274e-01, -1.3700e-02, -1.6221e-01,  1.8193e-01,\n",
       "           6.6760e-01, -1.5030e-01,  2.1051e-01, -2.6279e-01,  3.6893e-02,\n",
       "           3.1640e-01, -1.7744e-01, -2.9518e-01, -6.1691e-02,  1.7223e-02,\n",
       "           7.4439e-02, -1.7474e-01, -8.7299e-02, -8.1241e-02, -5.8571e-02,\n",
       "          -2.8273e-01,  1.1313e-01,  2.0861e-01, -2.2429e-01,  1.9823e-01,\n",
       "           1.9230e-01, -1.7416e+00,  2.5365e-02,  1.5863e-02, -1.0367e-01,\n",
       "          -2.3848e-01,  3.1032e-01,  1.5422e-01,  2.9009e-01,  2.9835e-02,\n",
       "           1.0126e-01, -1.1426e-01,  3.3732e-01,  4.4724e-01,  2.3651e-01,\n",
       "           2.7471e-01,  5.7314e-01,  3.8651e-01,  1.5104e-01, -1.4669e-01,\n",
       "          -6.0205e-01,  1.4928e-01, -1.0620e-01,  9.4374e-01,  4.0608e-02,\n",
       "           1.5254e-01, -1.9116e-01, -1.8022e-01,  2.5309e-01, -2.8234e-02,\n",
       "          -9.2637e-02, -1.0012e-01, -3.4675e-01, -3.0097e-01, -9.9249e-02,\n",
       "           4.2358e-01, -9.2871e-02, -2.4689e-01,  4.2208e-01,  2.5082e-01,\n",
       "          -3.8324e-01,  1.2687e-01, -1.3844e-01, -2.5892e-01, -2.9616e-01,\n",
       "           3.3238e-01, -2.7032e-02,  3.1468e-02, -1.9035e-01, -2.1735e-01,\n",
       "          -3.9601e-03, -5.8736e-01,  5.5944e-01,  1.9640e-01, -2.4654e-01,\n",
       "          -4.1387e-01, -7.5450e-01,  1.9970e-01, -7.5315e-02,  9.1461e-02,\n",
       "          -2.5202e-01, -3.1641e-01, -5.2487e-01,  3.5947e-02,  2.0697e-01,\n",
       "           4.4926e-01,  1.9772e-01,  8.9947e-01,  7.2866e-01,  2.4648e-01,\n",
       "          -2.5856e-01, -1.4761e-01,  4.1776e-01,  1.1088e-01, -3.4336e-01,\n",
       "           1.0788e-01, -2.2348e-02, -3.6824e-01,  2.0439e-01, -5.9988e-01,\n",
       "          -2.5088e-01, -2.1491e-03,  1.0569e-01,  4.9361e-01, -2.5127e-01,\n",
       "           2.6580e-01,  1.4168e-01, -4.9975e-01, -1.1628e-01,  1.3331e-01,\n",
       "          -3.3109e-01,  8.9008e-01,  3.5144e-01, -4.4793e-01, -6.8973e-01,\n",
       "          -5.0734e-01, -3.8761e-01, -1.0096e-01,  1.1021e-01, -5.3170e-01,\n",
       "           3.0667e-01, -3.4302e-01,  2.8320e-02, -1.6571e-01, -2.2510e-01,\n",
       "           1.2014e-01,  2.8105e-01, -9.3450e-02, -2.1394e-01, -4.2610e-01,\n",
       "          -1.0801e-01,  6.4642e-02, -3.2187e-01,  2.1907e-01,  1.3559e-01,\n",
       "           1.0015e-01,  4.5731e-02,  1.2892e-01, -1.1062e-02,  1.9028e-01,\n",
       "          -2.2997e-01, -4.4981e-01,  2.1816e-02, -3.8993e-01, -2.9428e-01,\n",
       "           5.8222e-01, -5.8712e-03,  8.0696e-01, -1.7285e-01, -5.9574e-01,\n",
       "           1.5104e-02,  6.0376e-01, -1.1727e-01, -1.8771e-01, -1.4679e-01,\n",
       "          -1.2361e+00, -2.1945e-01, -9.8679e-02, -8.2909e-02, -2.3480e-01,\n",
       "           5.3374e-02, -1.5989e-01,  8.0825e-03, -1.1905e-01, -3.4139e-01,\n",
       "           1.0438e+00,  7.8848e-01,  2.3246e-01, -3.0861e-01, -2.0016e-01,\n",
       "           6.3138e-03, -3.8141e-02, -2.0345e-01, -1.0253e-01,  1.7503e-01,\n",
       "           7.7049e-02,  6.2584e-01,  1.1042e-02, -2.0044e-02, -3.4368e-01,\n",
       "           4.2383e-01, -3.3043e-02, -1.7615e-01,  5.8453e-01, -1.7298e-01,\n",
       "          -2.0774e-01,  4.2771e-01, -2.0503e-01,  1.3829e-02,  1.4883e-01,\n",
       "           2.6899e-01, -4.7596e-02, -3.4460e-01,  8.2540e-02,  3.5717e-01,\n",
       "           6.2109e-01,  3.3953e-01, -2.7760e-01,  6.5552e-01,  3.2699e-01,\n",
       "          -2.0191e-01,  4.1142e-01,  1.3820e-01, -1.7924e-01, -2.0539e-01,\n",
       "           2.0168e-01,  6.9807e-02, -7.4488e-01, -8.4617e-02, -4.7329e-01,\n",
       "           5.7393e-01, -6.8078e-01,  7.9306e-02,  2.8730e-01,  3.8121e-01,\n",
       "          -4.0480e-01, -5.3148e-01,  3.8241e-01, -4.7768e-01, -3.0693e-01,\n",
       "          -3.4384e-02, -8.6439e-02,  1.4523e-01, -1.2218e-01, -7.9246e-02,\n",
       "          -1.9341e-01,  3.4088e-03,  7.2867e-01,  5.6793e-01, -3.1465e-01,\n",
       "          -1.4702e-01, -1.3017e-01, -2.8323e-01,  6.2295e-01, -1.7019e-01,\n",
       "          -5.7448e-02,  8.0345e-02, -3.2670e-01, -8.4425e-02,  3.2321e-01,\n",
       "          -5.2073e-01, -3.5352e-01,  3.2990e-02, -2.9764e-01, -2.8259e-01,\n",
       "          -3.3876e-01,  2.4112e-01,  6.9097e-01, -5.9782e-02,  4.3552e-02,\n",
       "           1.4855e-01,  1.8282e-01, -2.7335e-01,  2.5850e-01,  1.2036e-01,\n",
       "           3.5476e-01, -4.7735e-01,  2.8174e-01, -8.3400e-02, -5.8719e-01,\n",
       "           7.5692e-01,  2.2249e-01,  2.4640e-01,  4.7898e-01, -1.1493e+01,\n",
       "           7.3623e-02,  5.1149e-01, -5.4270e-01, -2.3046e-01,  3.6556e-02,\n",
       "           2.7208e-01, -5.5615e-01,  1.8624e-01, -4.5838e-01, -2.3841e-01,\n",
       "          -6.6034e-02, -4.8877e-03,  6.7520e-03, -4.1436e-01, -8.8732e-02,\n",
       "          -3.5048e-01, -7.8613e-01,  1.1187e-01, -1.7526e-01,  1.9979e-01,\n",
       "           7.1418e-02,  1.0158e-01,  1.9436e-01,  1.1062e-02, -2.7425e-01,\n",
       "           2.0853e-02,  1.8423e-01,  4.6734e-01,  1.7640e-01,  3.9518e-02,\n",
       "           2.1239e-01, -1.6040e-01,  1.3806e-01, -2.8911e-01, -5.4310e-01,\n",
       "          -6.4348e-02,  3.7330e-04, -6.7531e-02,  1.7136e-01, -2.9623e-01,\n",
       "          -4.1043e-01, -1.2209e-01,  4.0116e-02,  2.5722e-01,  7.0231e-01,\n",
       "           1.1589e+00,  3.7990e-01,  3.1284e-01,  1.4258e-01, -4.0650e-01,\n",
       "          -7.5050e-01,  2.7479e-01, -6.8614e-01,  7.7542e-02, -1.7055e-01,\n",
       "           7.3101e-01, -6.8846e-01,  3.0977e-02, -5.3695e-01, -2.8627e-01,\n",
       "           3.5933e-01,  2.4379e-01, -9.9686e-02, -4.6647e-01, -1.2278e-01,\n",
       "           6.7903e-01,  1.5290e-01,  1.1141e-01, -2.2945e-01,  3.5839e-01,\n",
       "          -6.7356e-01,  7.7488e-01,  1.8900e-01, -2.2858e-01, -1.3857e-01,\n",
       "           1.7395e-01,  1.4820e-01,  1.3816e+00,  1.1599e-01,  6.3929e-02,\n",
       "           1.8410e-01,  2.6066e-01, -3.8152e-01, -8.9436e-01,  2.8663e-02,\n",
       "           2.1578e-01,  2.1592e-01, -2.3035e-01,  4.3544e-01, -1.5236e-01,\n",
       "          -2.8447e-01, -1.2155e-01, -2.1023e-01,  2.4203e-01,  5.7358e-01,\n",
       "          -5.5475e-01,  2.1427e-01,  2.2306e-01,  1.5120e-01, -3.9240e-01,\n",
       "           5.6700e-02,  4.9140e-02, -4.5574e-01,  2.1555e-01, -1.4491e-01,\n",
       "           2.5648e-01, -2.5836e-01, -4.7821e-03,  2.3491e-01, -5.0771e-01,\n",
       "           2.3405e-01,  6.3044e-02, -1.6749e-02, -2.1802e-01, -1.1216e-01,\n",
       "           1.6698e-01,  4.1186e-01, -3.4030e-01, -3.4897e-01,  8.3229e-02,\n",
       "          -6.3409e-02, -6.6233e-02, -5.9560e-04, -4.2535e-01,  3.8539e-01,\n",
       "          -5.9813e-01, -6.2702e-02,  2.1950e-01,  1.5265e-01,  2.0760e-01,\n",
       "           2.0895e-01,  7.6257e-02,  3.9178e-01, -3.0957e-01, -7.2999e-01,\n",
       "          -8.2714e-01, -1.2870e-02, -1.5439e-02, -2.9529e-01,  2.0102e-03,\n",
       "           6.3339e-02,  3.9385e-02,  4.4908e-02,  2.6216e-01,  8.9147e-02,\n",
       "           6.4812e-01,  3.4904e-01, -7.6467e-02, -1.0895e-01, -3.8254e-01,\n",
       "          -3.1574e-01, -8.3585e-02,  5.5418e-02,  1.0959e-01, -2.7562e-01,\n",
       "          -3.6509e-01, -2.1296e-01,  1.1393e-02,  1.7875e-01, -1.7594e-01,\n",
       "          -2.6665e-01, -4.1213e-01,  2.4575e-01,  1.6257e-01,  4.0556e-01,\n",
       "           1.3812e-01, -3.0917e-01, -2.1799e-01,  1.8940e-01,  1.6605e-01,\n",
       "           1.3211e-01,  1.7243e-01, -1.3321e-01, -6.5244e-01, -1.4800e-01,\n",
       "          -1.6010e-01,  9.7176e-01, -2.6523e-01, -6.9528e-01,  2.2610e-01,\n",
       "           3.8961e-01, -1.8501e-01, -8.7357e-02, -1.7382e-01,  2.4024e-01,\n",
       "           1.8289e-02, -3.1846e-01,  2.3319e-01,  3.5450e-01,  3.7156e-01,\n",
       "           8.5839e-01,  6.5356e-02, -2.8349e-01,  3.7613e-01, -6.9142e-02,\n",
       "          -4.8594e-01,  2.9152e-02, -2.5754e-01,  1.4371e-01,  9.3908e-02,\n",
       "          -2.3990e-01, -3.1825e-01, -8.5579e-02,  1.6091e-01, -1.0908e-01,\n",
       "          -1.3584e-01,  1.5362e+00, -1.7029e-01, -1.8210e-01,  5.1552e-01,\n",
       "           2.9895e-01,  1.2060e-01,  4.1176e-01,  2.6028e-01,  2.5675e-01,\n",
       "           3.1944e-01, -2.3775e-01, -1.0824e-01,  1.7429e-01,  2.5753e-01,\n",
       "           6.7871e-01, -9.5930e-01, -2.2058e-01,  6.5388e-03,  2.3299e-01,\n",
       "           2.0779e-01,  4.4212e-01, -4.6028e-01,  2.8910e-01,  6.0566e-01,\n",
       "          -8.5687e-01,  2.0489e-01, -3.1282e-01,  3.5310e-01, -6.3758e-03,\n",
       "          -1.9207e-02, -1.4746e-01, -2.4634e-01, -9.2791e-02,  2.9471e-01,\n",
       "           4.3620e-01, -8.8294e-01,  2.7629e-01,  4.8844e-01, -2.0529e-01,\n",
       "           1.7227e-01, -1.3486e-01,  4.6878e-01,  1.3991e-01,  8.1479e-02,\n",
       "           4.5127e-02,  1.8888e-01, -2.5279e-01, -4.7563e-01,  2.8581e-01,\n",
       "           3.6522e-01,  2.5596e-03, -1.5496e-01,  2.0276e-01, -1.7676e-01,\n",
       "          -2.0991e-01, -3.3415e-01, -2.7939e-01, -1.4067e-01, -3.4600e-01,\n",
       "          -7.4126e-02, -1.3004e-01,  2.9175e-02,  4.7739e-01, -4.8251e-01,\n",
       "           5.7450e-01,  5.1800e-01,  6.0023e-01,  1.3326e-01,  2.9019e-02,\n",
       "           1.2872e-02,  2.2699e-01, -2.6209e-01,  5.3419e-01,  9.2327e-03,\n",
       "          -2.8983e-01,  2.7334e-01,  2.3794e-01, -3.0262e-01,  3.3767e-01,\n",
       "          -3.9539e-01, -8.1388e-02, -8.6262e-03, -2.3816e-01,  1.5794e-01,\n",
       "           1.4496e-02,  1.5213e-01, -5.0711e-01,  1.5700e-01,  1.8925e-01,\n",
       "          -3.6244e-01,  6.6234e-01, -1.7975e-01,  7.6817e-01, -2.9022e-01,\n",
       "           3.5644e-01, -7.6970e-01, -3.8740e-01,  8.0646e-02, -6.3140e-02,\n",
       "          -7.2002e-01, -4.1574e-01, -1.7238e-01,  5.8208e-01,  5.2535e-01,\n",
       "           1.5061e-01, -3.1460e-02, -7.5419e-01, -4.3918e-01,  6.7152e-02,\n",
       "           1.2295e-02, -1.5178e-01,  6.9128e-02, -1.5086e-01,  2.7358e-01,\n",
       "          -4.3002e-01, -4.0269e-01,  7.5083e-02,  1.7688e-01, -3.3978e-01,\n",
       "           8.2535e-01, -5.2140e-02,  5.6301e-02,  4.0559e-01,  1.7516e-01,\n",
       "           5.8121e-01,  5.5259e-01, -6.4897e-02, -6.8323e-01, -1.6898e-01,\n",
       "           3.8740e-02, -1.5738e-01,  1.8994e-03, -6.9056e-01,  7.2676e-01,\n",
       "           2.1885e-01,  3.3740e-01,  6.0192e-01]], device='mps:0'),\n",
       " tensor([[-2.2870e-01, -8.8700e-01, -6.8135e-01, -5.1080e-01, -4.1040e-01,\n",
       "          -5.5338e-01, -7.8598e-01,  3.7378e-01,  2.5134e-02,  4.8663e-01,\n",
       "           2.6921e-01, -1.1940e-01,  9.6979e-02, -4.0224e-01, -4.0943e-01,\n",
       "          -2.8067e-01, -2.1810e-02,  4.2545e-01,  6.1063e-01, -4.8945e-02,\n",
       "           6.7133e-01,  3.6942e-01, -1.0432e-01, -3.4328e-01, -5.5362e-01,\n",
       "           1.8481e-01, -1.3357e-01,  4.8337e-02,  4.8337e-01, -9.8779e-01,\n",
       "          -2.5068e-01,  8.3833e-01,  5.5728e-02,  5.8454e-01,  6.8700e-02,\n",
       "          -6.8008e-02,  8.6611e-02,  1.6401e-01, -2.7600e-02,  1.0803e-01,\n",
       "           1.9714e-01,  2.0738e-01,  3.5773e-01,  3.0351e-01,  8.3137e-01,\n",
       "           8.6267e-02,  2.7429e-01, -7.9806e-01,  2.1738e-01, -4.5040e-01,\n",
       "          -1.8865e-01, -4.8025e-01,  1.1329e-01,  2.9939e-01, -3.1065e-01,\n",
       "          -6.9581e-01,  1.7679e-01,  4.5688e-02, -3.3919e-01,  2.2163e-01,\n",
       "           1.4830e-01,  9.0352e-01, -2.0059e-01,  5.1915e-03,  2.3493e-01,\n",
       "           1.2875e-01, -6.4311e-01, -2.6861e-01, -4.0767e-01, -3.4631e-01,\n",
       "          -2.5357e-01,  4.7475e-01, -4.7245e-01, -3.9913e-01,  4.8095e-01,\n",
       "          -3.2435e-01,  2.9375e-01, -2.6899e-01, -4.8477e-01, -7.5755e-02,\n",
       "           4.2490e-01, -6.2055e-02,  3.9069e-02, -8.9886e-03, -3.8113e-01,\n",
       "          -3.3598e-01,  4.3189e-01,  1.7966e-01,  1.9423e-01,  2.3311e-01,\n",
       "           2.9699e-01,  1.0820e-01,  3.5883e-01, -3.0064e-01,  2.5480e-01,\n",
       "          -2.7369e-01,  6.0998e-01, -1.4611e-01,  1.1977e-01, -2.0550e-01,\n",
       "           2.3192e-01,  2.3844e-01,  1.1043e-01, -2.2939e-02, -4.3267e-01,\n",
       "          -4.2490e-01,  3.4806e-01,  1.7001e-01, -7.1048e-01,  1.0899e-01,\n",
       "          -1.6945e-01,  7.3260e-02,  3.5270e-01, -7.3340e-01, -5.1518e-01,\n",
       "           1.2889e-01,  7.9233e-02, -4.6810e-01, -2.6803e-01,  6.2721e-01,\n",
       "           3.4699e-01, -2.0806e-01,  3.3957e-01,  2.0204e-01,  4.3609e-01,\n",
       "          -2.4110e-01,  3.1290e-01, -8.4763e-02,  2.5244e-01,  3.4877e-01,\n",
       "          -2.1669e-01, -4.4014e-02, -3.5329e-01,  4.3750e-02,  2.4931e-01,\n",
       "           1.3383e-01,  1.3523e-02, -4.6624e-01,  2.0303e-01,  2.1580e-02,\n",
       "          -3.9259e-01,  1.6003e-01, -6.0204e-01, -1.1459e-01,  3.5156e-02,\n",
       "           8.1904e-02, -1.7506e-01,  3.9603e-02, -2.6438e-01, -1.6971e-01,\n",
       "          -5.6958e-01, -6.5652e-01,  8.3907e-01, -5.0883e-02,  7.2179e-02,\n",
       "          -4.9150e-02,  2.1174e-02, -9.4097e-02,  3.0636e-01,  2.9277e-01,\n",
       "           4.9046e-01, -3.5381e-01,  1.2420e-01, -3.2924e-01, -2.0875e-01,\n",
       "           4.5272e-01,  1.7476e-01, -4.5295e-01, -2.6092e-02,  7.6062e-02,\n",
       "           4.1214e-01, -2.3759e-01, -2.3001e-01, -7.5819e-01, -1.2371e-01,\n",
       "          -4.1405e-01,  1.4379e-01,  4.0819e-01, -1.3564e-01, -6.8051e-02,\n",
       "           2.1813e-01, -1.8734e+00,  3.7262e-01, -6.5120e-01,  2.4834e-01,\n",
       "          -4.6237e-01,  2.3539e-01,  1.7476e-01,  2.8680e-01, -1.1470e-01,\n",
       "           1.9708e-01, -7.3023e-02,  4.0751e-01,  5.5726e-01, -9.6310e-02,\n",
       "           1.1501e-02,  8.0742e-01,  1.2549e-01,  3.1699e-01, -7.4929e-02,\n",
       "          -3.4589e-01,  4.9148e-01, -3.2728e-01, -1.0780e-01,  2.9765e-02,\n",
       "           1.4736e-01, -5.5367e-01,  2.4935e-01,  1.0860e-01, -8.4753e-02,\n",
       "          -1.2487e-01, -1.8458e-01, -3.7864e-01, -3.2858e-01, -3.5586e-01,\n",
       "           6.9168e-02, -1.2408e-01, -1.4366e-02,  1.9077e-01,  3.5159e-01,\n",
       "          -1.5650e-01,  1.0434e-03, -4.0249e-02, -1.3225e-01, -3.0542e-01,\n",
       "           4.1200e-01, -9.8697e-02, -2.5005e-02, -9.5056e-02, -3.4129e-01,\n",
       "          -1.7922e-02, -6.7035e-01,  7.1945e-01,  5.5658e-01, -1.2536e-01,\n",
       "          -6.2325e-01,  2.7624e-02, -1.4257e-01, -3.2465e-02, -1.0131e-01,\n",
       "          -3.5984e-01, -5.1292e-01, -9.8263e-01,  1.6108e-02,  8.0363e-02,\n",
       "          -4.4372e-02, -1.2976e-01,  1.0835e+00,  4.7376e-01,  3.1055e-01,\n",
       "          -1.1502e-01,  9.8451e-03,  3.4919e-01,  3.6776e-01, -1.3280e-01,\n",
       "           1.3500e-01,  9.5632e-02, -2.6003e-01,  4.1018e-01,  1.0877e-01,\n",
       "          -3.3265e-01, -5.1670e-02, -1.5925e-02,  4.8073e-01, -2.8239e-01,\n",
       "           2.5055e-01,  1.1687e-02, -2.1440e-01, -4.1279e-02,  1.1783e-01,\n",
       "          -3.5343e-01,  1.6912e-01, -7.0205e-01, -3.5846e-01, -7.5197e-01,\n",
       "          -5.2475e-01,  2.0746e-01,  1.1569e-01,  4.8625e-01, -9.1457e-01,\n",
       "          -2.2642e-01, -1.6266e-01,  7.9728e-03,  1.6252e-01, -2.2655e-01,\n",
       "          -2.0754e-02,  6.7752e-03, -8.8470e-03,  4.0114e-02, -5.6009e-01,\n",
       "          -1.6505e-01,  5.5866e-01, -2.0033e-01,  4.0891e-01,  2.3090e-01,\n",
       "          -1.4765e-01,  1.6253e-01,  9.6199e-02, -1.9533e-01,  1.9001e-01,\n",
       "          -5.4567e-01, -4.8515e-01, -2.1186e-01, -4.1888e-01, -3.4564e-01,\n",
       "           4.1775e-01,  1.4993e-01,  3.8311e-01, -4.2379e-01, -2.4834e-01,\n",
       "          -1.8432e-02,  7.5069e-01, -5.5868e-03, -2.9016e-01, -4.8554e-01,\n",
       "          -1.5035e+00, -7.2283e-02, -2.6747e-01, -2.2801e-01, -2.7903e-01,\n",
       "           2.3181e-01,  1.0901e-01, -2.1356e-01, -8.0815e-02, -1.9570e-01,\n",
       "           4.6539e-02,  4.4082e-01, -1.3702e-01, -2.9372e-01,  3.0851e-01,\n",
       "          -3.4794e-01, -7.1680e-02, -1.7839e-01, -7.8423e-02,  4.4446e-01,\n",
       "           2.1991e-01,  4.4595e-01, -9.3823e-02, -1.0186e-01, -4.5129e-01,\n",
       "           9.1779e-01, -3.9616e-01,  2.9585e-01,  4.9386e-01, -5.7145e-01,\n",
       "           1.5549e-01,  3.0122e-01, -1.7437e-01,  1.0383e-01, -3.8355e-01,\n",
       "           3.8566e-02, -4.4544e-01, -3.5709e-01, -1.4900e-01,  6.7017e-01,\n",
       "           3.9898e-01,  4.3181e-01, -5.1608e-01,  1.2213e-01,  2.9053e-01,\n",
       "          -3.5009e-01,  2.5124e-01,  3.1827e-02,  1.4622e-02, -3.1347e-01,\n",
       "           3.0557e-01,  3.7943e-01, -6.7065e-01,  7.5271e-02,  1.1177e-01,\n",
       "           2.1661e-01, -6.6084e-01, -4.6453e-02,  1.4868e-01,  4.0579e-01,\n",
       "          -1.6617e-01, -2.5089e-01,  3.9574e-01, -5.1983e-01, -1.8592e-01,\n",
       "           1.5268e-01,  2.8492e-01,  1.5412e-01,  8.6882e-02, -7.1585e-01,\n",
       "          -2.4998e-01,  2.4460e-01,  5.2979e-01,  8.0768e-01,  3.2857e-01,\n",
       "           2.6091e-01, -1.4823e-01, -8.6614e-02,  2.0798e-01,  1.3959e-01,\n",
       "           2.7236e-02,  5.7876e-01, -2.1398e-01,  1.9698e-01,  2.0142e-01,\n",
       "          -2.9425e-01, -1.3855e-01,  3.9904e-02,  1.9139e-01, -2.6975e-01,\n",
       "          -5.3084e-01,  4.5452e-01,  3.6166e-01, -6.5311e-02,  1.6501e-02,\n",
       "           3.1112e-02,  5.3745e-01, -2.2483e-01,  4.6276e-01,  4.5712e-01,\n",
       "           2.1299e-02,  6.2443e-02, -8.1643e-02,  1.5101e-01, -4.5383e-01,\n",
       "           7.2576e-01,  8.6553e-02,  2.0610e-01,  3.6012e-01, -1.1119e+01,\n",
       "          -7.8650e-02,  9.8512e-01, -7.1784e-01,  1.3836e-01,  1.2976e-01,\n",
       "          -1.4800e-01, -3.0315e-01,  3.0477e-01, -5.9469e-01, -1.6600e-01,\n",
       "           6.5890e-02,  2.8277e-01,  1.6499e-01, -5.1816e-02, -2.2043e-01,\n",
       "          -2.7286e-01, -8.1320e-01,  7.0796e-02,  5.0479e-02,  4.7736e-01,\n",
       "           1.3338e-01,  3.2252e-01,  1.6235e-02,  6.3680e-02, -3.3066e-01,\n",
       "          -3.7762e-01, -3.3155e-01, -2.1581e-01,  1.2106e-01, -1.0570e-01,\n",
       "           1.9730e-02,  2.5462e-01, -5.8040e-03, -1.9459e-01, -6.7289e-01,\n",
       "          -3.1245e-02, -2.9032e-01,  1.6792e-01,  1.2950e-01, -2.7323e-01,\n",
       "           1.4651e-01, -1.7177e-01,  4.0901e-01,  2.4496e-01,  5.7064e-01,\n",
       "           9.4648e-01,  6.1512e-01,  2.6684e-01,  3.5616e-02, -1.3224e-01,\n",
       "          -7.0815e-01,  2.3694e-01, -8.8688e-01, -2.3235e-02, -2.7380e-01,\n",
       "           5.4967e-01, -3.9036e-01,  1.3185e-01,  1.3166e-01, -2.5649e-01,\n",
       "           4.2904e-01, -9.4908e-04,  3.6444e-01, -1.6780e-01, -2.5881e-01,\n",
       "           6.5304e-01,  2.0482e-01,  2.2949e-01,  9.2168e-02,  5.4488e-02,\n",
       "          -3.8491e-01,  1.0538e+00,  3.9007e-01,  1.4053e-01,  2.3492e-01,\n",
       "           7.7697e-02,  5.4342e-02,  9.7469e-01, -2.9494e-01, -2.2920e-01,\n",
       "          -1.2597e-01,  3.1078e-01, -8.1914e-01, -8.3974e-01, -4.8350e-02,\n",
       "           1.5754e-01,  2.1465e-01,  2.4761e-01,  1.3753e-01, -1.2731e-01,\n",
       "          -1.5899e-01, -1.4790e-01,  1.7663e-02,  2.7450e-01,  4.2146e-01,\n",
       "          -3.2015e-01, -1.1937e-01,  9.6745e-02, -1.4334e-01, -9.9337e-01,\n",
       "          -6.4899e-02, -1.5450e-01, -1.8219e-02, -1.7600e-02, -1.9829e-01,\n",
       "           2.2113e-02,  3.1336e-01,  2.0718e-01,  4.3399e-01, -5.4310e-01,\n",
       "           2.0945e-01, -2.3456e-01,  2.2782e-01, -1.0508e-01, -2.8217e-01,\n",
       "           1.0562e-01,  7.1099e-01, -2.5874e-01, -6.7787e-01, -5.5078e-01,\n",
       "          -1.3140e-01,  3.1113e-01, -3.9926e-03, -4.9705e-01,  4.7478e-02,\n",
       "          -6.2704e-01, -4.6129e-02,  1.6997e-02,  1.1453e-01,  2.8324e-01,\n",
       "           7.8601e-02,  3.8694e-01,  2.8706e-01, -2.5741e-01, -5.0207e-01,\n",
       "          -3.2953e-01,  3.6087e-01,  4.3282e-01, -2.5988e-01, -3.0314e-01,\n",
       "          -2.9128e-01,  1.3189e-01, -2.2452e-01,  3.2494e-01,  1.9085e-02,\n",
       "           4.2063e-01,  3.7017e-01,  1.5178e-01, -9.9054e-02, -1.7656e-01,\n",
       "          -5.0238e-01, -2.0770e-01,  1.7114e-01, -3.2072e-01, -2.7242e-01,\n",
       "          -1.8504e-01,  4.0900e-01, -2.0912e-01, -1.2875e-01,  1.6579e-01,\n",
       "          -1.4625e-01, -4.1546e-01, -4.2819e-01,  1.2809e-01,  3.5295e-01,\n",
       "          -7.9260e-02, -3.6993e-01, -1.3415e-01, -8.1793e-03,  2.7689e-01,\n",
       "           2.4495e-01, -3.7170e-02, -6.9409e-01, -3.0790e-01,  4.9375e-02,\n",
       "           2.7446e-02,  9.7193e-01, -2.5416e-01, -2.0671e-01,  4.3594e-01,\n",
       "          -1.0716e-01, -9.7609e-02,  2.9042e-01, -1.7053e-01,  1.5842e-01,\n",
       "          -2.8897e-01, -1.0981e-01,  4.0742e-01, -8.3312e-02,  4.3273e-01,\n",
       "           2.6199e-01,  7.3869e-02, -3.7512e-01,  1.7171e-01, -3.3571e-02,\n",
       "          -5.0566e-01,  3.2303e-02, -3.2812e-01, -9.8650e-02, -8.1062e-03,\n",
       "           1.6711e-01, -2.0885e-01, -1.9130e-01,  1.6912e-01,  4.7614e-01,\n",
       "           6.3085e-02,  1.3273e+00,  2.2209e-01, -3.2508e-01,  1.8304e-01,\n",
       "           1.5626e-01, -7.1832e-01,  1.1164e-01,  1.9761e-02,  4.2644e-01,\n",
       "           2.8502e-01,  3.4148e-01, -1.6274e-01,  3.3630e-01,  6.6245e-02,\n",
       "           4.8457e-01, -1.1360e+00, -4.9980e-01,  4.1099e-02,  4.8159e-01,\n",
       "          -1.5322e-02,  4.1813e-01, -2.5115e-01,  5.1466e-01,  5.6221e-01,\n",
       "          -1.1060e+00, -1.6374e-01, -1.3842e-01,  6.4103e-01, -8.5121e-02,\n",
       "          -7.6155e-01, -3.9883e-01, -3.5841e-01, -4.4867e-01,  7.2701e-02,\n",
       "           5.4930e-01, -7.0501e-01,  3.3940e-01,  1.0614e-01, -3.1229e-01,\n",
       "          -9.8814e-02, -3.8682e-01,  2.7848e-01,  2.1951e-01,  3.2815e-02,\n",
       "           1.9455e-01,  3.4930e-01, -4.8403e-01, -5.5503e-01,  3.0239e-01,\n",
       "           3.5093e-01, -1.6249e-01, -1.3997e-03, -1.7992e-02,  3.8519e-01,\n",
       "          -1.8636e-01, -4.4194e-02, -2.0312e-01,  1.9192e-01,  1.4584e-01,\n",
       "           3.0377e-01, -2.1070e-01, -4.3036e-03,  3.6583e-01, -3.0202e-01,\n",
       "           2.0309e-01,  3.0854e-01,  6.6888e-01, -1.2178e-01, -2.3346e-01,\n",
       "           2.5899e-01,  2.7084e-01, -2.3244e-01,  5.1641e-01, -3.0897e-01,\n",
       "          -4.0342e-01,  6.5178e-01, -1.0920e-01, -4.6288e-01,  6.4695e-02,\n",
       "          -2.2374e-01,  2.1219e-01,  1.4344e-01, -2.2669e-01,  1.9221e-01,\n",
       "          -3.7374e-02,  8.5593e-02, -8.6913e-01,  8.5155e-02,  2.0573e-02,\n",
       "          -4.6300e-01,  9.2102e-01,  5.9993e-02,  4.3186e-01, -2.8205e-01,\n",
       "           3.2247e-01, -2.5365e-01, -5.8313e-01,  3.8689e-02, -4.0746e-01,\n",
       "          -3.5299e-01, -3.3450e-01,  1.6947e-01,  1.9136e-01,  1.0613e+00,\n",
       "           1.0243e-01, -1.1753e-02, -4.3755e-01, -5.1485e-01,  2.8997e-02,\n",
       "           9.5061e-02, -3.1968e-01,  5.6185e-01,  1.0401e-01,  4.6401e-01,\n",
       "          -7.9376e-01, -5.2997e-01, -1.3187e-01,  2.3249e-01, -2.2896e-01,\n",
       "           4.8413e-01, -5.2940e-01,  7.9023e-01,  4.7559e-01,  1.6946e-01,\n",
       "           4.2180e-01,  4.8242e-01,  2.2670e-01, -5.1347e-01, -8.7485e-03,\n",
       "           3.0991e-01, -7.6557e-01,  2.6752e-01, -5.7792e-01,  3.2053e-01,\n",
       "           9.9477e-02,  7.1084e-01,  7.7773e-01]], device='mps:0'),\n",
       " tensor([[-0.2188, -0.4577, -0.5994,  ...,  0.3145,  0.5079,  0.5582],\n",
       "         [-0.0986, -0.6295, -0.5474,  ...,  0.1507,  0.8416,  0.6531]],\n",
       "        device='mps:0'),\n",
       " tensor([[ 0.0489, -0.7584, -0.4917,  ...,  0.2771,  0.1922,  0.6268],\n",
       "         [-0.1689, -0.4450, -0.6101,  ...,  0.4169,  0.1256,  0.6819]],\n",
       "        device='mps:0'),\n",
       " tensor([[-2.1557e-01, -3.4302e-01, -5.3454e-01, -8.8363e-01, -6.0228e-02,\n",
       "          -3.9555e-01, -8.3520e-01,  4.2932e-01,  4.9146e-02,  3.3053e-01,\n",
       "           1.1544e-01,  4.9200e-01,  1.7665e-01,  4.1852e-02, -3.5439e-01,\n",
       "          -3.9834e-02,  2.2093e-01,  3.5368e-01,  3.0812e-01,  2.4063e-01,\n",
       "           4.6833e-01,  4.8268e-01, -1.1595e-01,  1.7440e-02, -8.2687e-01,\n",
       "           1.7049e-01,  9.8705e-03,  3.3937e-01,  2.8929e-01, -3.9326e-01,\n",
       "          -1.2633e-01,  2.4473e-01, -1.4527e-01,  1.5918e+00, -9.4493e-02,\n",
       "          -7.4941e-02,  1.3339e-01, -5.9292e-01,  1.0284e-01, -9.0184e-03,\n",
       "           5.2707e-01, -2.3729e-01,  1.2671e-01,  3.9361e-01,  2.2218e-01,\n",
       "           1.2874e-01,  1.3737e-01, -2.3741e-01,  2.3237e-01, -6.2971e-01,\n",
       "          -1.1962e-01, -2.5275e-01,  5.5227e-01,  5.1938e-01, -1.1180e+00,\n",
       "          -5.9156e-01,  3.6817e-01,  2.0554e-01, -3.6088e-01,  2.2772e-02,\n",
       "          -5.7591e-02,  2.7615e-01,  2.1165e-02,  3.2266e-01,  1.8788e-02,\n",
       "           3.7221e-01, -2.1358e-01, -6.3098e-01, -5.3598e-01, -4.7268e-01,\n",
       "           1.4808e-02,  1.0864e-01,  1.4762e-01, -2.1165e-01,  3.3026e-01,\n",
       "          -3.7047e-01,  8.7954e-02, -1.7560e-01, -1.6127e-01,  5.7955e-03,\n",
       "          -7.2312e-02,  2.2787e-01,  1.2575e-01, -1.1595e-01,  6.9053e-02,\n",
       "           1.8631e-01,  1.5945e-01,  1.3073e-01,  1.1838e-01, -3.0293e-02,\n",
       "          -3.7750e-02,  2.0769e-03, -3.5665e-01, -2.6140e-01,  3.0009e-01,\n",
       "          -8.8599e-02,  2.5662e-01, -3.8743e-01,  7.2947e-02, -6.4576e-02,\n",
       "           4.3362e-02,  5.5173e-02, -8.0903e-02,  2.4139e-01, -2.8904e-01,\n",
       "          -6.1432e-01,  1.7252e-01, -1.0068e-01, -8.1571e-01,  1.5752e-01,\n",
       "          -1.2342e-01, -4.1377e-01,  2.0792e-02, -3.1927e-01, -2.3805e-01,\n",
       "          -8.8598e-02, -1.9253e-01,  1.1267e-01, -2.4556e-01,  3.1407e-01,\n",
       "           3.2215e-01, -2.2466e-01,  2.5970e-01, -9.4584e-02,  1.8047e-01,\n",
       "          -2.7668e-01,  2.8358e-01, -1.3223e-02,  2.9831e-01,  2.9925e-01,\n",
       "           1.0013e-01,  8.0034e-02, -2.3931e-01, -3.9883e-02,  1.3174e-01,\n",
       "           2.0585e-01,  2.1442e-01, -2.3368e-01,  1.7978e-02,  6.2208e-02,\n",
       "           2.0043e-01, -1.3184e-02, -2.2168e-01, -1.6869e-01, -2.2059e-01,\n",
       "          -1.7857e-01, -1.4671e-01,  5.8244e-02, -3.5367e-01, -4.0686e-01,\n",
       "          -6.0569e-01, -5.0516e-01,  6.7528e-01, -7.9958e-02, -1.1554e-01,\n",
       "          -1.6982e-01,  4.6091e-01, -4.3617e-02,  3.4217e-03,  1.7970e-01,\n",
       "           7.1405e-01,  2.7194e-02,  1.6640e-01, -4.7625e-01, -2.0394e-01,\n",
       "           2.7045e-01, -3.0122e-01, -2.0669e-01, -5.5958e-02,  1.4700e-01,\n",
       "           1.8914e-01, -1.0384e-01,  2.8586e-01,  1.8930e-02,  2.3565e-01,\n",
       "          -7.3573e-01,  5.1941e-01,  2.9407e-01, -2.3449e-01,  6.8919e-02,\n",
       "          -1.5530e-01, -1.2998e+00,  2.1974e-01, -7.6685e-01,  1.8615e-01,\n",
       "          -1.3486e-01,  1.8778e-01,  3.4968e-01,  3.4987e-01, -2.3343e-02,\n",
       "           1.8538e-01,  1.8583e-01,  2.9852e-01,  3.8594e-01, -1.0652e-01,\n",
       "           3.3285e-01,  3.9289e-01,  2.6937e-01,  2.5171e-01,  3.0164e-01,\n",
       "          -6.5569e-01,  3.8671e-01,  3.1694e-01,  4.6357e-01,  1.0670e-01,\n",
       "           1.4295e-01,  1.5333e-01,  1.2618e-01,  2.3161e-01,  1.3248e-02,\n",
       "          -3.2979e-01, -1.5194e-01, -5.2744e-01, -1.0320e-02, -1.3782e-01,\n",
       "           3.7245e-01, -1.9296e-01,  1.4482e-01,  7.2240e-01,  3.0177e-01,\n",
       "          -5.9742e-02,  3.0168e-01,  1.2129e-02, -2.5151e-01,  7.9077e-02,\n",
       "           4.2383e-01, -1.5031e-01,  1.5279e-01, -4.6546e-01, -1.8859e-01,\n",
       "           3.7013e-01, -3.5821e-01,  6.3186e-01,  4.6392e-01,  2.1484e-02,\n",
       "           1.9729e-01, -3.8772e-01, -2.4794e-02, -1.3292e-01,  4.2606e-01,\n",
       "          -3.2235e-01, -4.4201e-01, -4.6025e-01,  1.2488e-01,  5.1963e-01,\n",
       "           2.7507e-01,  1.8342e-02,  4.0816e-01,  5.8068e-01,  4.3071e-01,\n",
       "          -2.1999e-01, -1.9365e-01,  3.5483e-01,  9.1989e-02, -4.9810e-01,\n",
       "           3.0726e-01, -1.4577e-02, -5.0317e-01,  3.5902e-01, -5.3025e-01,\n",
       "          -5.5532e-02, -5.9393e-02,  4.5818e-01,  4.0688e-01, -5.1826e-01,\n",
       "          -3.4261e-02,  4.0496e-01, -1.3117e-01, -3.0446e-01,  3.2324e-01,\n",
       "          -3.4511e-01,  3.1175e-01,  1.6773e-01, -5.5057e-01, -5.7871e-01,\n",
       "          -4.2803e-01, -2.4415e-01, -8.8823e-02,  1.4967e-02, -7.8610e-01,\n",
       "          -2.8046e-01,  2.0665e-01, -5.1283e-03, -8.6086e-02,  5.7148e-02,\n",
       "           2.8681e-01, -2.0106e-01, -9.2440e-02, -1.5680e-01, -6.0368e-01,\n",
       "          -1.5160e-01, -1.0235e-01, -3.0615e-01,  3.1186e-01,  8.6952e-02,\n",
       "           7.5664e-02,  1.9435e-01,  2.7866e-01, -6.4456e-02, -4.0782e-01,\n",
       "          -1.2747e-01, -2.3504e-02, -3.1469e-01, -2.1963e-01,  1.4944e-01,\n",
       "          -3.3305e-02,  2.7266e-01,  8.1616e-01, -3.8768e-01, -5.5249e-01,\n",
       "           1.4606e-01,  8.3344e-01, -2.9283e-01, -9.5223e-02, -1.1903e-01,\n",
       "          -9.2530e-01, -3.7072e-01, -1.9451e-01, -2.9672e-01, -4.8813e-01,\n",
       "           1.0980e-01, -1.7958e-01, -1.1784e-01, -1.8766e-01, -3.3024e-01,\n",
       "           3.9487e-01,  6.7966e-01, -1.1788e-01,  3.2904e-01, -3.4598e-01,\n",
       "          -3.1677e-01, -9.7884e-02, -4.3956e-01, -5.7858e-01, -2.7805e-01,\n",
       "           2.5600e-01,  3.4740e-01,  2.1232e-02, -6.8528e-02, -3.8176e-01,\n",
       "           5.3752e-01, -4.1048e-01,  3.0514e-02,  3.4183e-01, -4.6879e-01,\n",
       "           1.0197e-01,  4.2102e-01, -1.3648e-01,  1.4493e-01, -5.2486e-02,\n",
       "           1.1283e-01, -3.4785e-01, -4.6263e-01,  1.3005e-02,  6.9578e-01,\n",
       "           3.8538e-01,  1.9462e-01, -2.2188e-01,  4.6105e-01,  4.0659e-01,\n",
       "          -5.2163e-01,  5.0398e-01,  1.0944e-01, -2.3511e-01, -4.7327e-01,\n",
       "           2.1476e-01,  3.1016e-01, -3.7805e-01,  1.6156e-01,  1.3281e-01,\n",
       "          -1.4074e-01, -2.5158e-01, -1.6374e-01,  4.7408e-01,  4.5011e-01,\n",
       "          -1.2427e-01, -4.3088e-01,  1.1413e-01, -5.0812e-01, -2.0407e-01,\n",
       "           2.0856e-01, -1.2761e-01,  2.1117e-01, -4.1330e-01, -3.9195e-01,\n",
       "          -3.3138e-01, -1.6117e-01,  7.0991e-01,  3.2262e-01, -3.7908e-01,\n",
       "           1.0666e-01, -2.6697e-01, -3.6592e-01,  4.0021e-01,  9.3521e-02,\n",
       "          -2.9539e-01,  3.4456e-01, -4.6624e-01, -3.8041e-02, -1.1845e-03,\n",
       "          -1.4180e-01, -5.5793e-02, -5.7221e-02, -5.7840e-02,  3.9790e-01,\n",
       "          -2.7578e-01, -4.4203e-03, -2.5783e-02, -9.3828e-02,  2.5156e-01,\n",
       "           1.9253e-01,  1.4145e-01, -4.2573e-01,  3.0091e-02,  2.0523e-01,\n",
       "           1.9929e-01, -1.9835e-01,  5.9970e-02, -3.5347e-01, -3.4772e-01,\n",
       "           4.7067e-01,  4.5500e-01,  2.8334e-01,  2.9687e-01, -1.1389e+01,\n",
       "          -1.2259e-01,  5.9564e-01, -4.0830e-01, -2.9001e-01,  8.8802e-02,\n",
       "          -4.6091e-02, -2.1754e-01,  1.8197e-01, -1.2609e-01, -6.0791e-01,\n",
       "          -5.4574e-01, -2.0938e-01,  2.8982e-01, -3.4635e-01, -4.4516e-01,\n",
       "           1.3640e-01, -8.0581e-01,  1.5269e-01, -2.0424e-01,  5.1419e-01,\n",
       "          -3.2018e-01,  1.7545e-03,  2.7812e-01, -2.5248e-01, -2.5937e-02,\n",
       "           1.0715e-01, -3.0715e-03,  5.4778e-01,  3.6972e-01,  9.6707e-02,\n",
       "           2.7368e-01, -3.1309e-01, -1.4697e-01, -3.1938e-01, -5.8170e-01,\n",
       "          -5.9212e-02, -3.0166e-01,  3.7046e-01,  2.0678e-01, -5.3534e-01,\n",
       "          -3.2953e-01, -3.5016e-01,  2.2099e-01,  2.9859e-01,  9.2455e-01,\n",
       "           6.2430e-01,  1.0096e+00,  3.2782e-01,  6.1630e-02, -1.0016e-01,\n",
       "          -7.7906e-01,  2.3107e-01, -2.1678e-01, -3.8133e-01, -1.3936e-02,\n",
       "           2.9604e-01, -4.9503e-01,  1.0404e-01, -4.3391e-01, -9.8243e-02,\n",
       "           6.7254e-01,  2.0503e-01,  2.2486e-01, -6.3817e-02, -1.8371e-01,\n",
       "           1.5121e-01, -5.9580e-02,  1.9754e-01, -1.2314e-01, -2.3528e-01,\n",
       "           3.7950e-02,  3.5330e-01,  3.8059e-02, -6.4610e-01,  2.7982e-01,\n",
       "          -5.7172e-02, -2.8368e-01,  1.8774e+00,  2.4910e-01,  9.0204e-02,\n",
       "           1.3192e-01,  1.3170e-01, -5.0931e-01, -8.2021e-01,  4.1625e-02,\n",
       "          -1.1771e-01,  4.8621e-02,  3.3109e-01,  3.1045e-01, -3.9686e-01,\n",
       "          -8.3503e-01,  8.8458e-03,  2.6393e-01,  3.8405e-01,  6.6810e-01,\n",
       "          -7.3522e-01, -2.7382e-02,  1.6784e-01, -5.0340e-03, -5.4275e-01,\n",
       "           7.3825e-02, -3.0804e-01, -1.8592e-01,  4.3392e-01, -4.9519e-01,\n",
       "           9.3395e-02, -9.9280e-02, -3.3757e-01,  3.8242e-01, -3.0130e-01,\n",
       "           1.6792e-01,  1.3893e-01,  2.3818e-02, -3.0528e-01,  3.1779e-03,\n",
       "           2.5971e-01,  6.3491e-01, -4.7380e-01, -3.5711e-01, -1.2010e-01,\n",
       "           2.6295e-01,  1.3416e-01,  3.5675e-01, -4.7070e-01,  5.1508e-01,\n",
       "          -7.6393e-01, -5.4993e-02, -1.5303e-02,  2.6561e-02,  3.8799e-01,\n",
       "          -1.0307e-01,  3.6686e-02,  4.1945e-01, -6.9520e-01, -6.0791e-01,\n",
       "          -5.1056e-01, -6.3155e-02,  6.9854e-02, -2.9517e-01,  4.0342e-02,\n",
       "          -1.3685e-01, -4.2188e-02, -1.4064e-01,  2.0590e-01,  5.1737e-03,\n",
       "           3.2727e-02,  3.1345e-01,  4.1379e-03,  1.0171e-01, -8.6417e-02,\n",
       "          -5.1170e-01,  1.9350e-01, -5.6769e-02,  1.1530e-01, -4.9557e-01,\n",
       "          -4.6112e-01, -9.4338e-02, -6.6168e-02, -5.0070e-01,  1.3697e-01,\n",
       "          -1.0336e-01, -3.4211e-02, -2.9169e-01,  4.1859e-01,  4.4790e-01,\n",
       "           1.7379e-01, -3.4664e-01, -1.1183e-01, -2.1416e-01,  2.6304e-01,\n",
       "           1.7069e-01,  4.0728e-01, -2.9340e-01, -7.2700e-01, -1.3937e-01,\n",
       "          -4.5551e-02,  8.0850e-01, -6.0790e-02, -4.7980e-01,  1.0876e-01,\n",
       "           2.9569e-01, -2.2702e-01, -2.5532e-01,  2.9731e-02,  6.5002e-02,\n",
       "          -1.6618e-01, -7.1052e-01,  5.5661e-01,  8.3718e-02,  1.9604e-01,\n",
       "           4.0477e-01,  1.5657e-01,  1.3425e-01,  7.1581e-01, -2.3745e-01,\n",
       "          -6.2046e-01, -2.6741e-01, -7.9407e-01,  1.7069e-01,  1.4890e-01,\n",
       "          -6.3217e-02,  4.8177e-01,  1.0144e-01, -8.9269e-02, -6.0832e-02,\n",
       "           4.0118e-01,  1.1128e+00, -1.2449e-01,  2.6457e-01,  8.0649e-01,\n",
       "           3.0339e-01, -4.5361e-02,  2.0610e-01,  1.6160e-01,  2.6019e-01,\n",
       "           3.8251e-01, -2.0437e-01, -3.2468e-01, -9.1600e-02,  2.0393e-01,\n",
       "           5.3296e-01, -4.6040e-01, -6.9971e-01, -3.2672e-01,  6.8627e-01,\n",
       "          -3.8964e-01,  2.6576e-01, -2.7233e-01,  3.1561e-01,  4.6440e-01,\n",
       "          -8.5009e-01,  7.0507e-03, -1.8646e-01, -1.7795e-02, -1.9321e-03,\n",
       "          -3.6371e-01,  1.5946e-01, -4.0737e-01, -2.0887e-01,  5.2176e-01,\n",
       "           2.4735e-01, -8.0778e-01,  2.0780e-01,  8.5386e-01, -5.5973e-01,\n",
       "           3.2365e-01, -3.9568e-01, -1.8308e-01,  4.0536e-01,  1.4305e-01,\n",
       "          -1.1124e-01,  2.2893e-01, -4.9468e-01, -5.9884e-01, -4.8032e-02,\n",
       "           9.5249e-01, -4.7501e-02,  3.9376e-02, -2.3031e-01, -7.3316e-02,\n",
       "          -2.5871e-01, -3.7012e-01, -3.0622e-01,  1.4150e-01, -9.1555e-02,\n",
       "           7.6066e-02, -1.6942e-02, -1.9552e-01,  3.3901e-01, -2.0967e-01,\n",
       "           2.8537e-01,  3.9990e-01,  9.0599e-01, -3.4863e-02, -3.4870e-01,\n",
       "          -3.8202e-02,  5.6629e-02, -1.1396e-01,  3.8273e-01, -3.0769e-01,\n",
       "          -2.9748e-01,  1.1371e-03,  7.0561e-02, -4.7597e-01,  1.8526e-02,\n",
       "          -3.9717e-01,  1.2734e-01,  6.2316e-02, -1.6889e-01,  4.9766e-01,\n",
       "           3.5772e-02, -5.1320e-03, -2.3181e-01,  3.6451e-01,  3.9618e-01,\n",
       "          -6.6660e-02,  5.4402e-01,  2.1621e-01,  1.7946e-01, -1.1713e-01,\n",
       "           7.7112e-02, -1.0672e+00, -1.3640e-01, -6.8354e-02, -3.5718e-01,\n",
       "          -3.6618e-01, -3.4641e-01,  3.2972e-01,  4.2438e-01,  5.2024e-01,\n",
       "           4.7299e-01,  2.4308e-02, -6.1962e-01, -2.7716e-01,  9.3200e-02,\n",
       "           2.0286e-01, -2.2097e-01,  3.1465e-01,  6.9508e-02,  3.2527e-01,\n",
       "          -4.7197e-01, -4.3926e-01,  1.9456e-02,  3.8353e-01,  7.0162e-02,\n",
       "           9.9717e-01, -6.1842e-02, -1.4776e-01,  1.2019e-01,  7.2328e-02,\n",
       "           1.6920e-01,  7.5268e-01, -3.0741e-02, -6.7453e-01, -2.4769e-02,\n",
       "           2.5609e-04, -3.2638e-01,  3.4564e-01, -8.4256e-01,  1.0561e-01,\n",
       "           2.7495e-01,  4.3042e-01,  4.8383e-01]], device='mps:0'),\n",
       " tensor([[-1.4630e-01, -7.9251e-01, -6.9310e-01, -7.3117e-01, -3.2860e-01,\n",
       "          -3.7026e-01, -8.9840e-01,  6.6354e-01,  6.3545e-02,  3.0733e-01,\n",
       "           2.2657e-02,  1.8670e-01,  4.6891e-01, -2.9963e-01, -3.9510e-01,\n",
       "          -4.9224e-01,  1.1991e-02,  6.1488e-01,  5.9763e-01,  4.8349e-01,\n",
       "           3.3779e-01,  3.8441e-01, -3.3708e-02,  8.3856e-02, -5.4307e-01,\n",
       "           4.0013e-01,  3.1069e-02, -6.9905e-02,  1.4332e-01, -2.0293e-01,\n",
       "          -2.3009e-01,  3.8228e-01, -1.7115e-01,  1.1910e+00, -1.9500e-01,\n",
       "          -2.3382e-01,  3.6068e-01, -6.2367e-01,  2.2717e-01,  1.0226e-01,\n",
       "           2.7379e-01, -4.9091e-01,  1.3973e-01,  2.6480e-01,  4.0718e-01,\n",
       "           4.4308e-01, -5.3546e-02, -2.5398e-01,  3.5203e-01, -6.3471e-01,\n",
       "           1.7551e-01,  3.4193e-03,  6.7050e-01,  6.4895e-01, -5.3316e-01,\n",
       "          -3.9018e-01,  3.0881e-01, -8.8802e-02, -3.4488e-01,  3.9579e-01,\n",
       "          -1.8767e-01,  6.1251e-01, -9.6296e-04,  2.8345e-01,  2.7874e-01,\n",
       "           3.3636e-01, -4.7783e-01, -6.2564e-01, -5.4264e-01, -8.4037e-02,\n",
       "          -2.9905e-02,  2.0660e-01,  9.9408e-02, -1.5713e-02,  2.0715e-02,\n",
       "          -3.1796e-01,  1.8027e-02, -3.0938e-02, -2.8592e-01, -2.4189e-01,\n",
       "          -4.1209e-02, -2.9189e-02,  9.6763e-02,  2.4732e-01, -5.9468e-02,\n",
       "          -1.4308e-01,  3.3899e-01,  1.1621e-01,  2.5397e-01,  1.5085e-01,\n",
       "           3.0354e-01, -2.0206e-01, -3.5366e-02, -4.9758e-01,  3.4507e-01,\n",
       "           1.1604e-01,  6.1926e-01, -2.0807e-01, -7.5440e-02, -1.5419e-01,\n",
       "           1.4013e-01,  2.1290e-01, -1.9514e-01,  2.7389e-01, -4.1395e-01,\n",
       "          -7.2166e-01,  3.1085e-02, -2.0840e-01, -4.1386e-01, -1.9010e-03,\n",
       "          -4.0163e-01, -3.9329e-01,  7.5585e-02, -4.2258e-01, -2.5417e-01,\n",
       "          -1.6044e-01, -5.0910e-01, -1.9146e-01, -2.7703e-01, -1.0837e-01,\n",
       "           1.0499e-01, -2.2850e-01,  3.4841e-01,  3.6293e-01, -6.3678e-03,\n",
       "          -4.8861e-02,  2.1589e-01,  2.3887e-01,  6.3388e-01,  2.1304e-01,\n",
       "          -3.6472e-02, -4.1830e-02, -2.3633e-01, -6.6500e-02,  4.2866e-02,\n",
       "          -2.1207e-01,  1.6930e-01, -3.5317e-01,  8.9090e-02, -4.3066e-02,\n",
       "          -4.8462e-02,  1.1935e-01, -1.5119e-01, -2.7856e-01, -1.6776e-01,\n",
       "          -3.2720e-02,  2.9903e-01,  3.4140e-02, -2.1048e-01, -2.2048e-01,\n",
       "          -3.1430e-01, -6.5559e-01,  6.2799e-01, -5.9629e-02, -2.8641e-01,\n",
       "           4.4859e-02,  6.0757e-01,  4.2018e-03, -1.6625e-01,  2.9867e-01,\n",
       "           1.0068e+00,  4.8861e-02,  3.0247e-01, -4.1704e-01, -3.7135e-01,\n",
       "           1.2190e-01, -2.6572e-02, -1.2934e-01,  2.1203e-01, -1.6714e-01,\n",
       "           3.0004e-01, -2.2733e-01,  3.7549e-01,  1.6823e-02,  5.6855e-02,\n",
       "          -6.3391e-01,  3.5107e-01,  4.1541e-01, -1.7639e-01,  4.4847e-02,\n",
       "           7.0998e-02, -1.1756e+00,  7.9291e-02, -3.5005e-01, -1.0385e-01,\n",
       "           7.8621e-03,  1.0582e-01,  3.4599e-01,  1.6086e-01,  7.1312e-02,\n",
       "           2.8843e-01,  3.9821e-01,  3.3006e-01,  3.5603e-01,  5.2826e-02,\n",
       "           2.8486e-01,  4.8478e-01,  4.6160e-01,  3.1432e-01, -4.7989e-02,\n",
       "          -5.7373e-01,  1.6029e-01,  2.0317e-01,  4.0426e-01, -7.0360e-03,\n",
       "           3.7839e-01, -3.2642e-02,  4.6185e-02, -6.1657e-03, -1.8631e-01,\n",
       "          -2.2355e-01, -1.5501e-01, -6.4298e-01, -1.0289e-01, -3.6857e-01,\n",
       "           1.6990e-01, -4.2436e-01,  2.2354e-02,  4.2487e-01,  3.8828e-01,\n",
       "          -1.4099e-01,  6.0539e-02,  1.6553e-01, -2.2271e-01, -1.5832e-01,\n",
       "           3.3000e-01, -3.0234e-01, -2.6573e-02, -3.8758e-01, -2.8190e-01,\n",
       "           4.1923e-01, -2.9581e-01,  8.0504e-01,  5.9524e-01, -1.1642e-01,\n",
       "          -9.1544e-02, -6.5181e-01,  2.0564e-01, -5.7395e-01,  4.0385e-01,\n",
       "          -4.7506e-01, -3.6862e-01, -7.1266e-01, -1.5201e-01,  3.7210e-01,\n",
       "           2.5994e-01,  7.1605e-02,  1.0307e+00,  3.9066e-01,  3.1873e-01,\n",
       "          -2.9203e-01, -3.7214e-01,  6.4106e-01, -1.0742e-01, -4.5257e-01,\n",
       "           1.6376e-02,  5.1204e-03, -7.3345e-01,  4.5041e-01, -1.4714e-01,\n",
       "          -1.6372e-01, -2.6356e-01,  1.9030e-01,  3.5571e-01, -1.7469e-01,\n",
       "           1.5327e-01,  5.2084e-01, -1.2266e-01, -3.6037e-01,  3.3316e-01,\n",
       "          -5.1441e-01,  4.4012e-01,  1.6585e-01, -4.0384e-01, -4.2312e-01,\n",
       "          -5.6768e-01, -3.1273e-01, -1.6540e-01,  5.5195e-02, -6.3327e-01,\n",
       "           1.3910e-02, -1.3078e-01,  1.6768e-02,  1.8336e-01, -7.0515e-02,\n",
       "           3.9942e-01,  4.5085e-02, -2.2706e-01, -2.9942e-02, -3.7346e-01,\n",
       "          -6.4376e-03, -1.6163e-01, -2.1477e-01,  1.6253e-01, -1.9010e-01,\n",
       "          -9.3596e-03, -1.6828e-01,  3.5796e-01, -2.4359e-01, -5.6952e-01,\n",
       "           5.5954e-02,  2.4998e-02, -2.1326e-01, -1.2744e-01,  7.3623e-02,\n",
       "           2.2688e-01,  4.2906e-01,  6.1472e-01, -3.7363e-02, -5.6059e-01,\n",
       "          -3.2649e-02,  7.4165e-01, -1.6573e-01, -5.7834e-01, -1.7233e-01,\n",
       "          -1.3045e+00, -8.3945e-02,  1.5209e-02, -1.2450e-01, -5.8125e-01,\n",
       "           3.7668e-01,  2.5227e-01, -1.7158e-01, -4.0422e-02, -1.4156e-01,\n",
       "           1.4886e-01,  3.2365e-01,  3.1846e-01,  7.7305e-02, -5.1830e-01,\n",
       "          -1.3914e-01, -6.3613e-01, -3.0754e-01, -4.2334e-01, -2.8684e-01,\n",
       "           1.1542e-01,  2.0513e-01,  2.2956e-03, -2.3481e-01, -4.1576e-01,\n",
       "           6.0223e-01, -4.8758e-01, -5.2105e-03,  3.0859e-01, -9.7527e-02,\n",
       "           1.9868e-03,  4.4093e-01, -2.5132e-01, -1.8444e-01,  3.6213e-01,\n",
       "           3.0272e-01, -2.4578e-01, -4.5284e-01,  1.1036e-01,  6.8156e-01,\n",
       "           1.9740e-01,  1.8420e-01, -2.1100e-01,  5.6744e-01,  2.3765e-01,\n",
       "          -1.6602e-01,  4.2053e-01, -3.6183e-01, -3.6309e-02, -2.3875e-01,\n",
       "           2.4922e-01,  3.0795e-01, -4.4151e-01,  1.4042e-02, -1.3937e-02,\n",
       "          -1.5103e-01, -6.5904e-01, -2.6720e-01,  2.3578e-01,  3.6689e-01,\n",
       "          -2.7697e-01, -2.0292e-01,  4.5104e-01, -5.9952e-01, -5.4924e-01,\n",
       "          -2.8320e-02, -9.6803e-02, -4.1828e-03, -4.6916e-01, -4.9386e-01,\n",
       "           3.9016e-02, -3.9614e-01,  8.4743e-01,  3.6801e-01, -1.7668e-01,\n",
       "           3.7860e-01, -1.9428e-01, -3.8622e-01,  3.9809e-01,  8.4527e-03,\n",
       "          -4.1863e-01,  4.2506e-01, -4.9132e-01, -7.6521e-02,  8.8567e-01,\n",
       "          -2.5614e-01,  1.8194e-01,  3.4979e-01, -8.6159e-02,  1.5291e-01,\n",
       "          -2.1399e-01, -2.9577e-02, -1.1771e-01, -3.4967e-01,  2.0677e-01,\n",
       "           6.1273e-02,  5.8935e-02, -4.4512e-01,  7.8430e-02,  3.1248e-01,\n",
       "           3.2087e-01, -1.6028e-01, -2.0960e-01, -5.1734e-02, -6.8023e-01,\n",
       "           9.2279e-01,  4.6335e-01,  7.0312e-01,  2.8211e-01, -1.1211e+01,\n",
       "           2.0385e-01,  7.4361e-01, -4.0061e-01, -1.4166e-01, -2.7592e-02,\n",
       "           9.8451e-02, -4.3630e-01,  2.7281e-01, -4.7154e-01, -2.2382e-01,\n",
       "          -3.1581e-01, -4.0021e-02,  1.3873e-01, -4.1939e-02, -1.3283e-01,\n",
       "          -1.7140e-01, -6.0269e-01,  3.2287e-01, -2.3182e-01,  5.5424e-01,\n",
       "          -1.8120e-01, -1.7540e-01,  2.0420e-01, -2.6560e-01, -1.6085e-01,\n",
       "           1.6732e-01,  1.2975e-02,  5.1715e-01,  2.1178e-01, -3.3382e-01,\n",
       "          -8.6462e-02, -2.3926e-01,  1.4191e-01, -1.1086e-01, -9.4255e-01,\n",
       "           5.8082e-02, -1.0912e-01,  2.6004e-01, -1.4603e-01, -5.8077e-01,\n",
       "          -1.3595e-01, -5.4518e-01,  1.4391e-01,  4.6493e-01,  8.8153e-01,\n",
       "           5.0384e-01,  8.2347e-01,  4.0584e-01,  5.0860e-02, -3.8778e-01,\n",
       "          -6.4219e-01,  1.4699e-01, -5.1705e-01, -5.2795e-02,  2.6455e-02,\n",
       "           5.2227e-01, -6.2205e-01,  9.5526e-02, -4.2470e-01, -6.3488e-02,\n",
       "           5.3065e-01,  2.6747e-01, -3.3949e-01, -2.5863e-01, -1.5555e-01,\n",
       "           1.1383e-01, -1.5481e-01,  1.1925e-01, -7.6042e-02,  3.3589e-02,\n",
       "          -2.0709e-01,  7.0947e-01,  2.4386e-01, -7.8921e-01,  2.0214e-01,\n",
       "           1.2805e-01, -2.0939e-01,  1.8761e+00, -2.4680e-01,  1.9883e-01,\n",
       "          -1.4179e-02,  3.2276e-01, -5.4327e-01, -6.3310e-01,  5.8102e-02,\n",
       "           4.6173e-01, -4.6532e-02,  1.3061e-01,  6.0219e-01, -1.8206e-01,\n",
       "          -4.7409e-01,  3.1363e-01,  7.2379e-02,  6.9707e-02,  5.7185e-01,\n",
       "          -7.3037e-01, -3.5687e-01,  2.9579e-01,  4.1378e-01, -4.6856e-01,\n",
       "           3.6807e-02, -3.7544e-01, -5.1599e-01,  6.0722e-01,  1.0862e-01,\n",
       "           2.8904e-01, -1.0212e-01, -2.5866e-01,  5.8037e-01,  2.4017e-02,\n",
       "           3.0239e-01,  2.1101e-03, -7.9067e-02, -1.4887e-01,  5.6925e-02,\n",
       "           1.8465e-01,  5.6707e-01, -2.7518e-01, -5.4210e-01, -4.3478e-01,\n",
       "          -2.8499e-01,  2.1369e-01,  6.0937e-01, -2.0701e-01,  4.2202e-01,\n",
       "          -5.0721e-01,  6.8652e-02,  1.3642e-01, -4.6690e-01,  3.6702e-01,\n",
       "           4.5155e-01,  3.1140e-01,  3.2128e-01, -5.9587e-01, -6.6967e-01,\n",
       "          -4.3990e-01,  1.7339e-02,  2.1829e-01, -2.5044e-01, -6.2980e-02,\n",
       "          -3.2282e-01, -1.2042e-01,  9.4750e-03, -1.5805e-01,  2.3326e-01,\n",
       "           1.1761e-01,  2.9066e-01, -2.0440e-01, -1.1834e-02, -1.7658e-01,\n",
       "          -4.7385e-01, -2.0605e-01, -1.2630e-01,  2.5257e-01, -6.1352e-01,\n",
       "          -4.0958e-01, -7.9283e-02, -1.7123e-01,  5.4608e-03,  2.6499e-01,\n",
       "           1.0020e-01,  6.6518e-02, -1.7164e-01,  4.8964e-01,  1.9304e-01,\n",
       "           1.1898e-01, -4.7008e-01, -2.0581e-01, -1.8162e-01,  3.4117e-01,\n",
       "           8.1158e-02,  2.4952e-01, -4.0347e-01,  3.3343e-03, -4.2802e-01,\n",
       "          -2.7253e-02,  7.2108e-01, -2.2954e-01, -5.1967e-01,  2.8050e-01,\n",
       "           3.7886e-01, -1.7378e-01, -2.4494e-01,  1.2434e-01, -3.3678e-01,\n",
       "          -4.8744e-01, -6.1469e-01,  6.0120e-01,  1.2612e-01,  1.3376e-01,\n",
       "           7.9128e-01, -3.3319e-02, -3.1081e-01,  2.1440e-01, -9.9935e-02,\n",
       "          -2.8965e-01, -2.1110e-01, -4.5055e-01,  3.0963e-01,  3.9221e-02,\n",
       "          -1.3523e-01,  1.1681e-01,  4.0168e-02, -2.9986e-01,  2.5512e-02,\n",
       "           1.0839e-01,  9.7708e-01, -2.7883e-01, -1.6153e-01,  7.7632e-01,\n",
       "           1.5228e-01, -6.8561e-02,  2.1012e-02,  1.8178e-02,  1.8777e-01,\n",
       "           4.0595e-01, -4.3674e-01, -3.2473e-01, -8.4031e-02,  3.6678e-01,\n",
       "           5.8085e-01, -6.4715e-01, -3.4738e-01, -1.7644e-01,  3.8821e-01,\n",
       "          -2.7499e-02,  3.0043e-01, -2.4895e-01,  1.8185e-01,  3.8042e-01,\n",
       "          -9.8289e-01, -2.3022e-01, -3.6784e-01,  4.5363e-01,  2.4602e-02,\n",
       "           1.7636e-01, -1.8567e-01, -7.1477e-01, -5.7281e-02,  6.1942e-01,\n",
       "           5.1205e-02, -7.8463e-01,  3.0576e-01,  5.9371e-01, -8.8049e-01,\n",
       "           6.8456e-02, -4.1607e-01,  3.2485e-01, -3.2874e-04, -4.0835e-03,\n",
       "           5.2362e-01,  1.9315e-01, -3.2539e-01, -4.6767e-01,  2.3766e-01,\n",
       "           9.8888e-01,  2.7662e-02,  1.7452e-01, -1.8709e-01,  1.1638e-01,\n",
       "          -4.2507e-01, -2.0436e-01, -2.1225e-01, -6.0749e-03, -1.0501e-01,\n",
       "           8.9331e-02, -8.7254e-02, -2.3491e-01,  4.0844e-01, -9.9004e-02,\n",
       "           5.3533e-01,  7.6529e-01,  3.5295e-01,  1.8604e-01, -5.0560e-01,\n",
       "           1.1334e-01,  9.2954e-02, -1.4297e-01,  3.1813e-01,  1.3630e-03,\n",
       "          -1.7414e-01,  1.4545e-01,  1.1285e-01, -4.5274e-01, -1.4542e-01,\n",
       "          -2.8380e-01,  3.9477e-01, -2.1710e-01,  3.9840e-02,  3.4370e-01,\n",
       "           6.7820e-02,  1.8945e-01, -1.9217e-01,  2.8941e-01,  3.2305e-01,\n",
       "           5.6995e-03,  6.1399e-01,  8.9551e-02,  5.4981e-01, -8.5539e-01,\n",
       "           2.0032e-01, -1.0665e+00, -3.0704e-01, -8.0143e-02, -2.8560e-01,\n",
       "          -4.2380e-01, -3.4568e-01,  4.7146e-01,  2.2463e-01,  3.7087e-01,\n",
       "           3.7657e-01,  1.4386e-01, -8.5684e-01, -3.0876e-01,  1.1623e-01,\n",
       "           6.6759e-02, -4.4421e-02,  3.6648e-01,  3.0752e-01,  4.0087e-01,\n",
       "          -4.9949e-01, -6.0897e-01, -7.6167e-02, -6.5376e-03, -2.0788e-01,\n",
       "           8.7933e-01, -1.5636e-01, -1.2506e-01, -4.7335e-02,  2.7363e-01,\n",
       "           7.0157e-02,  8.9630e-01, -4.1512e-02, -2.8172e-01, -3.1699e-01,\n",
       "           5.1205e-02, -1.0251e-02, -5.8274e-02, -4.4208e-01,  2.3358e-01,\n",
       "           3.0278e-01,  2.8278e-01,  7.3155e-01]], device='mps:0'),\n",
       " tensor([[ 7.8665e-02, -3.3319e-01, -5.5660e-01, -5.8965e-01, -4.6930e-01,\n",
       "          -1.1946e-01, -4.5759e-01,  6.5134e-01,  1.4655e-01,  5.9324e-01,\n",
       "           3.8617e-01,  5.1988e-01, -8.6744e-02, -4.0391e-02, -8.0315e-01,\n",
       "          -3.4874e-01, -1.0696e-01,  2.4962e-01,  4.9079e-01,  4.1632e-01,\n",
       "           7.9087e-01,  2.9980e-01, -8.4391e-02, -1.0321e-01, -7.6051e-01,\n",
       "           3.8608e-01,  2.4686e-01,  8.7815e-02,  7.6643e-02, -8.2045e-01,\n",
       "          -3.6573e-01,  4.6250e-01,  1.1736e-01,  8.3404e-01, -8.5751e-02,\n",
       "          -2.6677e-01,  1.8969e-01, -2.5406e-01,  3.9238e-01, -1.6812e-01,\n",
       "           6.7438e-01, -1.0123e-01,  5.0173e-01, -6.9290e-02,  4.7742e-01,\n",
       "          -4.2209e-02,  1.8865e-01, -7.1643e-01,  7.9959e-02, -4.2033e-01,\n",
       "          -1.9924e-01, -2.5431e-01,  2.0197e-01,  2.5903e-01, -2.9010e-01,\n",
       "          -4.9244e-01,  9.3847e-02,  1.9423e-01, -1.0342e-01, -2.1171e-01,\n",
       "          -4.2479e-01,  8.4228e-01, -3.5377e-01,  4.3697e-02, -1.7282e-01,\n",
       "           6.7371e-02,  2.0693e-01, -6.5276e-01, -6.2630e-01, -1.0272e-01,\n",
       "          -1.3679e-01,  1.2663e-01, -1.3620e-02, -2.5428e-01,  6.3548e-01,\n",
       "          -3.5617e-01,  1.6481e-01,  2.7260e-03, -4.1286e-01, -2.4351e-02,\n",
       "           1.7587e-01,  2.5331e-01,  1.2718e-01, -1.0982e-01, -5.1255e-01,\n",
       "           1.5244e-01,  4.7846e-01,  6.2648e-01,  1.8116e-01, -2.6247e-01,\n",
       "           2.8822e-01,  1.7173e-01, -3.4797e-01, -1.5619e-01,  5.4757e-01,\n",
       "          -1.7455e-01,  1.9786e-01, -3.1488e-01,  2.4265e-01, -4.8853e-01,\n",
       "           2.0382e-01,  3.1247e-01, -8.9069e-02,  3.8070e-01, -3.5779e-01,\n",
       "          -9.2581e-01,  4.3997e-01, -1.9861e-01, -5.5595e-01, -1.9809e-01,\n",
       "          -1.5441e-01, -1.6270e-01,  2.9113e-01, -2.9993e-01, -1.5597e-01,\n",
       "           8.5736e-02,  1.8774e-01, -1.0887e-01, -4.8866e-01,  3.0041e-01,\n",
       "           3.7222e-01, -3.4038e-01, -7.7277e-02, -1.3002e-03,  1.0354e-01,\n",
       "           1.3987e-01,  2.0961e-01, -1.3358e-01,  5.4239e-01,  4.0550e-01,\n",
       "           1.3351e-03,  3.3908e-01, -4.4286e-01, -1.5048e-02,  1.3303e-01,\n",
       "           2.1415e-01, -2.0294e-01, -1.6943e-01,  6.1144e-01,  1.7822e-01,\n",
       "          -1.9778e-01, -4.4103e-02, -7.0901e-02, -1.9709e-01,  2.2189e-01,\n",
       "          -2.5595e-02,  1.2318e-02, -1.4898e-01, -2.1898e-01, -8.5230e-02,\n",
       "          -5.4516e-01, -8.4766e-01,  3.2739e-01, -2.5227e-01,  2.0967e-01,\n",
       "          -2.8692e-01,  3.8498e-01,  8.0672e-02,  2.1787e-01,  3.7474e-01,\n",
       "           3.3423e-01, -9.3499e-02, -8.7517e-03, -2.8127e-01, -4.4604e-02,\n",
       "           3.4312e-01, -4.3486e-02, -4.0885e-01, -2.4033e-01,  2.6412e-03,\n",
       "           1.4530e-02, -4.5320e-01,  1.1431e-01, -3.2538e-01,  2.0005e-02,\n",
       "          -2.9642e-01,  6.3012e-03,  3.2302e-02, -4.4854e-01,  1.9007e-01,\n",
       "          -5.6012e-01, -1.6311e+00,  4.7305e-01,  5.1473e-02, -3.7522e-01,\n",
       "          -4.8794e-01,  3.7521e-01,  3.5346e-01,  3.8928e-01, -2.2185e-02,\n",
       "          -2.5600e-01,  1.9699e-02,  1.9123e-01,  5.9596e-01, -3.3575e-01,\n",
       "           4.6698e-01,  4.4459e-01, -1.4062e-01,  9.7715e-03,  1.6836e-01,\n",
       "          -5.3519e-01,  5.9995e-02,  1.0718e-01, -6.0225e-01,  1.8609e-01,\n",
       "           3.5584e-02,  1.1801e-01, -2.0560e-02,  1.1724e-01,  1.1455e-01,\n",
       "          -4.7473e-02, -2.2635e-01,  2.9356e-02, -2.1258e-01, -3.2194e-01,\n",
       "           6.5296e-01, -3.0155e-02,  1.5382e-01,  5.1350e-01, -2.2872e-02,\n",
       "          -2.2212e-01,  7.3842e-02, -1.0983e-01, -7.0626e-02, -3.2899e-01,\n",
       "           9.9057e-02,  6.7015e-02, -8.4346e-02, -3.1247e-01, -3.1246e-01,\n",
       "          -2.6931e-01, -3.6587e-01,  8.3386e-01,  2.5966e-01, -3.9696e-01,\n",
       "          -2.1199e-01, -2.7683e-01, -2.0610e-01,  1.9071e-01,  2.4327e-01,\n",
       "          -1.4497e-01, -3.9168e-01, -6.3954e-01,  9.7886e-02,  1.0352e-01,\n",
       "           9.2401e-02, -9.5361e-02,  4.3674e-01,  5.9194e-01,  4.4844e-01,\n",
       "          -3.2206e-02, -2.2028e-01,  3.6476e-01,  3.3383e-01, -8.0178e-02,\n",
       "           1.8134e-01,  9.4323e-02, -1.4522e-01,  4.4022e-01, -3.3692e-01,\n",
       "           2.3695e-01,  8.0931e-02,  2.8936e-02,  4.5324e-01, -3.3168e-01,\n",
       "          -1.3309e-01,  3.4396e-01, -3.6911e-01,  6.4961e-02,  1.9793e-01,\n",
       "          -2.9062e-01,  3.7940e-01,  2.3827e-01, -3.3976e-01, -7.8926e-01,\n",
       "          -2.1933e-01,  4.9726e-01, -3.1672e-03, -1.2541e-01, -9.5543e-01,\n",
       "          -1.8762e-01, -3.3156e-03,  9.2503e-02, -6.4099e-02, -4.0994e-01,\n",
       "           1.2834e-01,  3.9177e-01, -5.0183e-01, -8.7568e-02, -6.1278e-01,\n",
       "           6.9972e-02,  2.5074e-01, -2.8204e-01,  3.7159e-01,  3.0243e-01,\n",
       "          -2.4428e-01,  2.8653e-01,  7.5605e-02,  1.1205e-01, -3.2914e-01,\n",
       "          -8.2701e-02, -1.7332e-01, -7.7504e-02,  1.8209e-01,  4.2936e-02,\n",
       "           2.9511e-01,  3.2389e-01,  3.8615e-01, -7.4800e-01, -6.1691e-01,\n",
       "          -2.8275e-01,  6.6626e-01, -9.6199e-02, -5.5451e-01, -2.2432e-01,\n",
       "          -9.9932e-01, -7.7640e-02, -3.0153e-01, -1.8693e-01, -5.6794e-01,\n",
       "           3.2395e-01,  1.5117e-01, -3.3903e-01,  5.8060e-02, -1.8543e-01,\n",
       "           3.9446e-01,  5.7926e-01, -4.6635e-01, -4.7312e-02, -1.8110e-01,\n",
       "          -7.0000e-01, -1.6876e-01,  9.6926e-03, -1.4039e-01, -2.6525e-01,\n",
       "           1.4828e-01,  8.2458e-03, -3.1826e-02, -3.1465e-02, -4.7696e-01,\n",
       "           4.8306e-01, -7.8942e-02, -8.0065e-02, -6.8700e-02, -4.6410e-01,\n",
       "           7.7896e-01,  5.3654e-01, -2.3148e-01, -2.5147e-01, -1.1381e-01,\n",
       "          -2.4422e-01, -1.8979e-02, -2.6958e-01, -7.2210e-02,  6.7415e-01,\n",
       "           5.6705e-01,  1.3674e-01, -3.2257e-01,  5.3214e-01,  4.5663e-01,\n",
       "          -4.6946e-01,  4.6575e-01, -1.0605e-01, -2.4862e-01,  5.9298e-02,\n",
       "           1.4316e-01,  1.8748e-01, -4.5911e-01,  1.2969e-01,  1.3473e-01,\n",
       "           2.4729e-01, -5.4253e-01,  1.5897e-01, -1.3442e-02,  6.4844e-01,\n",
       "          -4.0979e-01, -4.1103e-01,  2.6310e-01, -5.7506e-01, -1.4183e-01,\n",
       "           1.1065e-01,  2.7836e-01,  1.3613e-01, -1.2344e-03, -2.2089e-01,\n",
       "          -4.2060e-01,  4.6985e-01,  1.0382e+00,  4.9191e-01, -1.1880e-01,\n",
       "           2.7936e-01, -1.7533e-01, -2.2582e-01,  5.2580e-01, -3.0311e-01,\n",
       "          -2.6708e-01,  7.1871e-01, -6.8122e-01,  1.5384e-01,  4.6540e-01,\n",
       "          -3.0831e-01, -2.4656e-01, -3.2853e-01, -2.3824e-01, -2.7128e-01,\n",
       "          -4.1425e-01,  2.6325e-01, -3.0651e-01, -5.8545e-03, -5.0543e-02,\n",
       "           1.9667e-01,  7.0085e-02, -1.3518e-01,  5.5212e-02, -3.5810e-01,\n",
       "           4.2658e-02, -5.1103e-02,  1.3118e-01,  1.1394e-01, -2.7586e-01,\n",
       "           5.3951e-01,  1.0687e-01,  5.8638e-02,  3.7220e-01, -1.1108e+01,\n",
       "           8.8746e-02,  3.3313e-01, -2.9655e-01, -1.2104e-01, -9.0374e-02,\n",
       "           1.4931e-01, -1.8101e-01,  1.0486e-01, -3.0765e-01, -2.0723e-01,\n",
       "          -1.2570e-01,  2.4640e-01,  2.5130e-01,  2.2168e-01, -3.4182e-01,\n",
       "          -1.6470e-01, -9.1021e-01,  1.6431e-01,  2.0580e-01, -3.6130e-02,\n",
       "          -2.9418e-01,  1.5450e-01, -7.8763e-03,  1.3719e-02,  2.8503e-01,\n",
       "          -3.2059e-01, -8.4449e-02,  3.6967e-01,  3.7890e-01,  2.7910e-01,\n",
       "           1.1647e-01,  2.1658e-01,  5.1481e-02, -4.6570e-01, -5.1569e-01,\n",
       "           1.0081e-01, -6.3294e-02, -1.6165e-01,  2.9209e-01, -3.6465e-01,\n",
       "          -2.3637e-01, -3.2496e-01,  3.9977e-01,  4.1944e-01,  8.8848e-01,\n",
       "           9.2062e-01,  4.7680e-01,  5.0283e-01,  1.4722e-01, -5.4062e-02,\n",
       "          -7.3410e-01,  3.3623e-02, -7.9774e-01, -5.3108e-02, -1.3550e-01,\n",
       "           6.7615e-01, -3.5227e-01, -9.0720e-02, -4.1440e-01, -2.6125e-01,\n",
       "           2.4691e-01,  4.5322e-01, -1.4443e-01, -8.2877e-01,  5.2842e-02,\n",
       "           2.2913e-01, -3.6616e-02,  1.3933e-01, -2.1148e-02,  1.1532e-01,\n",
       "           1.8874e-02,  4.7326e-01,  2.0297e-01, -1.6483e-01,  4.1378e-01,\n",
       "           2.5998e-01, -6.2633e-03,  1.2631e+00, -7.9458e-02, -1.4828e-01,\n",
       "          -1.2580e-01,  2.2602e-02, -5.4676e-01, -7.3130e-01, -1.8601e-01,\n",
       "           3.1138e-01,  3.4070e-02,  1.2881e-01,  1.2845e-01, -2.6901e-01,\n",
       "          -1.0709e+00, -2.1814e-01,  3.4137e-01,  5.9279e-01,  9.6023e-01,\n",
       "          -8.9445e-01, -1.3622e-01,  2.3819e-01,  6.5576e-02, -6.3389e-01,\n",
       "           3.7269e-02, -3.6797e-01, -1.8228e-01,  1.0887e-01,  3.8440e-02,\n",
       "          -6.1551e-02,  9.8307e-02, -2.9388e-02,  2.0752e-01, -4.1445e-01,\n",
       "           2.3701e-01,  4.4028e-01,  1.2296e-01, -1.2347e-01,  1.6682e-01,\n",
       "           7.1984e-02,  7.3137e-01, -8.2626e-02, -6.5632e-01, -3.7716e-02,\n",
       "          -1.0649e-02,  1.5745e-01, -8.1279e-02,  4.8702e-02,  5.2501e-01,\n",
       "          -3.5379e-01, -7.3176e-02,  2.9441e-01,  3.2810e-01,  4.1030e-01,\n",
       "           6.5463e-02, -6.3273e-01, -1.7626e-02, -8.8577e-01, -9.4784e-01,\n",
       "          -7.1769e-01,  1.8862e-01,  1.2131e-03, -1.6380e-01, -3.8259e-01,\n",
       "          -1.8723e-02,  2.6946e-01, -3.9179e-01,  6.9487e-02,  2.3106e-01,\n",
       "           7.5315e-01,  3.0494e-01, -3.1046e-01,  3.7063e-01, -2.1554e-01,\n",
       "          -5.9244e-01, -3.7050e-01, -1.7002e-01,  7.9902e-02, -3.8246e-01,\n",
       "          -5.5448e-01, -1.4587e-01, -4.0493e-01, -3.0235e-01,  3.4117e-01,\n",
       "          -3.8965e-01,  6.2961e-02, -1.3742e-01,  3.6226e-01,  3.7745e-01,\n",
       "           2.5163e-01, -6.4800e-02,  9.0067e-02, -1.9536e-02,  5.0779e-01,\n",
       "          -8.2760e-02,  2.2725e-01, -4.8690e-01, -6.5028e-02,  7.8279e-02,\n",
       "           4.7451e-01,  8.5143e-01,  1.1698e-02, -1.3860e-01,  1.3725e-01,\n",
       "           4.1962e-01, -2.4000e-01,  5.8429e-02,  1.3686e-01, -4.2022e-01,\n",
       "           1.6675e-02, -6.0083e-01,  2.4538e-01, -3.3244e-02,  3.8033e-01,\n",
       "          -1.9480e-01, -1.0302e-01, -9.5638e-02,  6.1240e-01, -7.3715e-02,\n",
       "          -3.1113e-01, -2.5702e-01, -3.2976e-01,  4.1188e-03,  2.8295e-01,\n",
       "          -1.1001e-01, -2.8407e-02, -2.9404e-01, -2.0039e-01,  1.7325e-01,\n",
       "          -4.8451e-02,  1.0851e+00,  2.0519e-01, -6.9019e-01,  6.2926e-01,\n",
       "           3.4115e-01, -4.9966e-01,  1.0309e-01, -3.3142e-01,  3.3400e-01,\n",
       "           4.1797e-01, -1.2241e-02,  1.6279e-01,  2.2284e-01,  6.1621e-02,\n",
       "           5.4012e-01, -6.8948e-01, -5.9747e-01,  8.5144e-02,  2.1068e-01,\n",
       "          -2.6200e-01,  5.0605e-01,  3.2392e-02,  5.1016e-01,  7.9295e-01,\n",
       "          -1.2491e+00, -2.1483e-01, -6.5459e-01,  4.1457e-01,  1.2664e-01,\n",
       "          -2.8600e-01, -9.1929e-02, -4.4405e-01, -4.8891e-01,  3.4095e-01,\n",
       "           9.6014e-01, -1.3052e+00,  1.0671e-01,  4.2998e-01,  2.0346e-01,\n",
       "           2.0621e-01, -3.8512e-01,  3.3660e-01, -3.0197e-01,  1.4346e-01,\n",
       "          -2.9912e-01,  3.6593e-01, -6.6939e-01, -7.8222e-01, -1.3041e-03,\n",
       "           4.4873e-01,  1.9597e-01,  1.3924e-01,  5.5806e-03,  1.9202e-01,\n",
       "          -1.3225e-01, -9.4832e-02, -9.5299e-02,  4.6108e-01, -2.0293e-01,\n",
       "           1.8000e-01, -4.9486e-01,  8.6916e-02,  2.5310e-01,  6.8945e-02,\n",
       "           2.4457e-03,  4.5603e-01,  9.4221e-01, -2.2006e-02, -1.0648e-01,\n",
       "           4.3890e-01,  5.1716e-01, -1.8882e-01,  6.7642e-01, -3.1450e-01,\n",
       "          -2.7973e-01,  5.7423e-01, -3.3632e-02, -3.4042e-01,  3.1184e-01,\n",
       "          -2.3210e-01,  2.5750e-01,  1.5642e-01, -1.8786e-01,  4.9289e-02,\n",
       "           1.3476e-01, -4.2449e-02, -5.0464e-01,  8.7000e-02,  2.9638e-01,\n",
       "          -3.9737e-02,  9.0737e-01, -2.9828e-01,  5.2502e-01, -5.4668e-01,\n",
       "           1.1240e-01, -5.7810e-01, -1.9621e-01, -3.5303e-01, -3.1059e-01,\n",
       "          -5.0994e-01, -1.6831e-01,  1.5122e-01,  3.3280e-01,  9.1844e-01,\n",
       "           4.4060e-01, -4.7953e-01, -4.6967e-01, -1.0615e-01,  3.8615e-01,\n",
       "          -9.7167e-02, -1.0981e-01,  3.1683e-01,  1.5071e-01,  4.4850e-01,\n",
       "          -5.9487e-01, -4.0341e-01, -4.0880e-02,  3.4785e-01, -2.0486e-01,\n",
       "           6.1918e-01, -5.8090e-02,  2.8287e-01,  2.0884e-01,  1.3691e-01,\n",
       "          -1.4738e-01,  2.0868e-01, -2.8917e-01, -5.5648e-01, -9.5854e-02,\n",
       "           4.5684e-02, -5.6348e-01, -2.9590e-02, -6.8213e-01,  1.5402e-01,\n",
       "           4.2150e-01,  8.7482e-01,  5.2121e-01]], device='mps:0'),\n",
       " tensor([[-3.3529e-01, -5.5695e-01, -8.4054e-01, -8.6970e-01, -1.0327e-01,\n",
       "          -3.5986e-01, -9.1283e-01,  4.4275e-02,  3.9933e-01,  1.5726e-01,\n",
       "           3.5663e-01, -4.5460e-02, -9.4596e-02, -4.2587e-01, -8.7757e-01,\n",
       "          -1.6967e-01, -1.7055e-01,  1.2418e-01,  3.8021e-01,  4.4836e-01,\n",
       "           2.6924e-01,  5.1149e-01, -1.1942e-01,  7.3242e-02, -3.2963e-01,\n",
       "           2.4435e-01, -4.9213e-01,  9.7354e-02,  1.8819e-01, -1.0696e+00,\n",
       "           1.4278e-01,  5.2179e-01, -2.9633e-01,  1.1085e+00, -1.7575e-01,\n",
       "          -2.1160e-01, -1.2116e-01, -6.1950e-01,  7.6242e-01, -2.0050e-01,\n",
       "           5.4080e-01,  1.1953e-01,  3.9375e-01,  2.0621e-01,  1.5852e-01,\n",
       "           7.1616e-03, -1.0983e-02, -1.8919e-01,  4.6501e-03, -6.9389e-01,\n",
       "          -2.3482e-01, -2.1155e-01,  5.4513e-01, -1.2077e-01, -8.2809e-01,\n",
       "          -1.2844e-01,  3.0575e-02,  2.0131e-01, -6.9866e-02, -4.4673e-02,\n",
       "          -1.6635e-01,  7.3618e-01,  3.0765e-01,  2.1647e-01, -2.9052e-01,\n",
       "           4.4825e-01, -1.7892e-01, -5.6214e-01, -5.5512e-01,  6.3202e-02,\n",
       "           1.2786e-01,  3.7092e-01,  4.3577e-01, -3.6696e-01,  5.7286e-01,\n",
       "          -2.9309e-01,  4.7598e-01, -2.0932e-01, -5.4969e-01,  2.4779e-01,\n",
       "           7.4719e-02,  4.1131e-01,  1.1097e-01, -4.7479e-02,  1.3622e-01,\n",
       "           1.7638e-01,  4.5388e-02,  3.8438e-01,  1.3830e-01, -1.9977e-01,\n",
       "           2.6303e-01,  4.5818e-01, -6.6144e-01, -3.6104e-01,  1.5321e-01,\n",
       "          -2.9794e-01, -8.0938e-02, -3.9256e-01,  5.7533e-02, -3.4790e-01,\n",
       "          -9.8799e-03,  2.3344e-01,  2.1109e-01,  3.4638e-01, -4.9737e-01,\n",
       "          -3.9576e-01,  3.2313e-01, -1.2020e-01, -7.0432e-01, -2.0962e-01,\n",
       "          -1.3006e-02, -3.8313e-01,  5.3954e-01, -3.4508e-02, -1.1136e-01,\n",
       "           2.1438e-01,  1.0900e-01,  2.5418e-01,  2.3662e-01,  1.0424e+00,\n",
       "           3.6374e-01, -5.0564e-01, -2.7898e-01, -1.7338e-01,  3.2408e-01,\n",
       "          -2.8136e-01,  1.5686e-01, -6.8353e-02,  1.6191e-01,  2.7083e-02,\n",
       "           1.6245e-04,  5.1865e-01, -5.9792e-01, -7.9283e-02,  3.0276e-01,\n",
       "           5.3850e-01,  7.9713e-02, -7.3409e-01,  1.1896e-01, -3.7446e-01,\n",
       "           3.3275e-01,  2.1202e-01, -2.8856e-02,  1.6605e-01, -2.5051e-02,\n",
       "           6.9883e-02, -1.2737e-01, -1.1701e-01, -2.6197e-01, -1.4611e-01,\n",
       "          -4.3669e-01, -1.9172e-01,  3.4160e-01, -2.6245e-02, -6.5248e-02,\n",
       "          -3.6720e-01,  2.3432e-01, -2.5630e-01, -1.0019e-01,  1.7937e-01,\n",
       "           6.5403e-01, -1.0494e-01,  8.3473e-02, -2.6796e-01, -3.1901e-01,\n",
       "          -2.1523e-02,  1.0725e-01,  1.4855e-01, -4.4633e-01,  1.9099e-01,\n",
       "           1.3518e-01,  1.3184e-01,  1.8968e-01, -2.2125e-01,  1.0548e-01,\n",
       "          -3.6703e-01,  5.3186e-01,  2.9482e-01, -1.5419e-01,  2.7374e-01,\n",
       "           4.7464e-01, -2.0031e+00,  3.6990e-01, -7.5532e-01,  1.8056e-01,\n",
       "          -2.7083e-01,  9.1213e-02,  2.7334e-01,  3.8874e-01,  1.3406e-01,\n",
       "           6.5247e-02,  1.5217e-01,  1.8347e-01,  6.0201e-01,  7.3613e-02,\n",
       "           3.3592e-01,  6.2523e-01, -1.2319e-01,  1.0060e-01,  8.7082e-02,\n",
       "          -8.1716e-01, -8.7146e-02,  2.8494e-02,  9.7412e-02, -1.3172e-01,\n",
       "           2.2619e-01, -1.3090e-01,  8.8061e-02,  1.6564e-01,  6.2145e-02,\n",
       "          -8.9520e-02, -5.0376e-01, -4.0012e-01, -1.7090e-01, -1.1732e-01,\n",
       "           3.5617e-01, -4.1647e-01, -1.6510e-01,  6.5971e-01,  7.3331e-02,\n",
       "          -1.8613e-01, -1.8285e-01,  4.4938e-02,  1.1407e-02, -3.3100e-01,\n",
       "           5.1969e-01, -1.5921e-01,  1.7989e-02, -5.0805e-01, -2.5790e-01,\n",
       "           1.2061e-01, -6.2274e-01,  7.0156e-01,  2.4446e-01, -8.6059e-02,\n",
       "          -1.7086e-01, -1.2065e-01,  5.3225e-01,  2.0112e-01,  1.9566e-02,\n",
       "          -4.7084e-01, -1.2730e-01, -6.6461e-01,  8.0638e-02,  1.2436e-01,\n",
       "           1.4310e-01, -4.5433e-02,  9.9732e-01,  6.4227e-01,  6.2124e-01,\n",
       "           5.1036e-03, -2.7630e-01,  2.7658e-01,  1.9885e-01, -3.9093e-01,\n",
       "          -8.8990e-02, -1.0632e-02, -9.5598e-02,  7.2259e-02, -4.1124e-01,\n",
       "          -3.3184e-01,  2.8270e-01,  3.0472e-01,  7.6216e-01, -6.0472e-01,\n",
       "          -1.7223e-01,  6.4059e-02, -1.2385e-01, -1.4897e-01,  3.3320e-01,\n",
       "          -4.0914e-01,  4.4383e-01, -1.6787e-01, -8.8467e-02, -5.4149e-01,\n",
       "          -5.5487e-01,  5.3320e-02, -2.3943e-01,  2.0777e-01, -1.0308e+00,\n",
       "          -7.6955e-01,  8.2450e-02,  6.2357e-02,  7.3931e-02, -3.6115e-01,\n",
       "           1.1828e-01, -8.8758e-02,  2.1464e-02, -1.5277e-01, -2.9661e-01,\n",
       "          -2.6314e-01, -1.3524e-01, -5.9292e-02,  7.1922e-02,  1.1989e-01,\n",
       "          -3.3245e-01,  2.8847e-01, -2.7146e-02, -2.2728e-02,  5.2040e-02,\n",
       "          -3.7792e-01,  2.1989e-01,  9.5255e-02, -2.6523e-01,  1.5729e-01,\n",
       "           1.6152e-01,  2.9194e-01,  6.9849e-01, -6.1525e-01, -6.2157e-01,\n",
       "           7.5035e-02,  7.0693e-01, -1.9720e-01, -4.7100e-01,  1.4466e-01,\n",
       "          -8.4779e-01,  3.4785e-01, -2.0519e-02, -2.4880e-01, -2.4321e-01,\n",
       "           1.7346e-01, -1.0691e-01, -4.4648e-01, -3.1177e-01,  9.2852e-02,\n",
       "           2.4880e-01,  7.0440e-01, -6.8641e-02,  1.6716e-01, -1.7223e-01,\n",
       "          -2.2402e-01,  3.4534e-01, -3.9622e-01, -8.3563e-01,  1.7354e-01,\n",
       "           4.9434e-01,  3.5907e-01, -1.5715e-01, -1.2043e-01, -3.0419e-01,\n",
       "           6.5450e-01, -5.9340e-01, -5.0877e-01,  3.1210e-01, -2.1272e-01,\n",
       "          -1.3408e-01,  2.5574e-01, -2.5393e-01,  3.2492e-01, -2.3003e-01,\n",
       "          -4.3464e-02,  8.3218e-02, -2.9416e-01, -3.7247e-02,  4.2719e-01,\n",
       "           2.3335e-01,  1.3971e-01, -3.8892e-01,  4.4994e-01,  4.5657e-01,\n",
       "          -4.0787e-01,  3.1882e-01, -2.3519e-01, -4.6167e-01, -1.7471e-01,\n",
       "           2.1261e-01,  8.1308e-02, -3.8943e-01,  6.1397e-02,  8.1698e-02,\n",
       "          -7.4577e-03, -5.5001e-01,  1.6889e-01,  8.3712e-02,  7.0067e-01,\n",
       "          -1.7219e-01, -3.9944e-01,  5.6773e-01, -4.6941e-01, -1.7702e-01,\n",
       "          -3.3626e-01,  1.1710e-01,  5.2610e-02, -1.2257e-01, -4.2065e-01,\n",
       "          -1.0150e-01,  2.9701e-01,  3.3799e-01,  2.9341e-01,  6.3504e-02,\n",
       "           7.7026e-02,  8.4817e-03, -4.6622e-01,  3.9888e-01,  5.2766e-02,\n",
       "          -2.6363e-01,  1.7619e-01, -2.1826e-01,  3.0687e-01, -1.9057e-01,\n",
       "          -3.1337e-01,  3.9906e-01, -1.0522e-01, -1.7919e-01,  5.2159e-02,\n",
       "          -5.1421e-01,  4.1538e-02,  3.6812e-01, -5.9000e-01,  2.7319e-01,\n",
       "           7.5081e-02,  5.1476e-01,  1.0246e-01,  2.6803e-01, -3.1989e-01,\n",
       "          -3.4255e-01, -1.2507e-01, -3.3801e-02, -4.3155e-02, -3.6830e-01,\n",
       "           3.9592e-01,  1.3746e-01,  1.6197e-01,  4.0490e-01, -1.1023e+01,\n",
       "           1.1612e-01,  5.9746e-01, -2.9122e-01, -7.7146e-03, -2.7595e-02,\n",
       "           5.0322e-01, -3.5427e-01,  8.4343e-02, -5.1099e-01, -4.1381e-01,\n",
       "          -2.6194e-01,  4.1312e-01,  1.1991e-01, -9.0194e-01,  7.8903e-02,\n",
       "          -1.9512e-01, -8.1112e-01, -5.6010e-02,  2.0953e-01,  6.2940e-01,\n",
       "          -3.0623e-01,  2.5414e-01,  9.2121e-02, -1.5419e-01, -3.4674e-01,\n",
       "          -1.8781e-01, -2.1818e-01,  5.5155e-01,  3.6189e-01, -1.4923e-01,\n",
       "          -4.1026e-01, -8.3995e-02, -9.0928e-02, -4.4083e-01, -3.7081e-01,\n",
       "           1.0075e-01, -2.9621e-02,  4.4231e-01,  1.3875e-01, -2.7354e-01,\n",
       "          -4.3459e-01,  9.9936e-02,  2.7248e-01,  5.1273e-01,  6.7197e-01,\n",
       "           1.4942e+00,  8.5894e-01,  3.3874e-01,  1.6958e-01, -1.2377e-01,\n",
       "          -2.5145e-01,  8.1828e-02, -2.0351e-01,  3.8534e-01, -2.8548e-01,\n",
       "           3.5737e-01, -8.4920e-01, -1.4052e-01, -6.5247e-01, -3.0196e-02,\n",
       "           4.3634e-01,  6.0688e-01, -3.4377e-01, -2.5001e-01, -3.1517e-01,\n",
       "           4.0359e-01, -2.4541e-01,  2.0456e-01, -5.6815e-01,  6.3176e-02,\n",
       "          -1.8364e-01,  2.0698e-01, -1.2214e-01,  4.2981e-01,  4.1995e-01,\n",
       "          -1.1593e-01, -3.8172e-02,  1.6613e+00, -1.1600e-01,  8.6016e-02,\n",
       "           7.9507e-02, -8.9963e-02, -3.1830e-01, -6.2713e-01, -2.0212e-01,\n",
       "          -3.7415e-02,  4.8784e-02,  9.0497e-02,  3.4870e-01, -4.0055e-01,\n",
       "          -1.1280e+00, -1.4547e-01,  1.3901e-01,  7.1138e-01,  7.2689e-01,\n",
       "          -6.8133e-01, -1.7299e-01, -1.4980e-01,  1.5908e-01, -6.2619e-01,\n",
       "          -1.1279e-01, -3.8793e-01, -5.0256e-01,  3.1044e-01, -1.2263e-01,\n",
       "          -2.8357e-01,  1.6195e-01,  3.1403e-01,  1.1930e-01, -5.9241e-01,\n",
       "           5.1828e-02,  3.1778e-01,  2.4612e-01, -2.0883e-01, -1.9944e-01,\n",
       "           3.9510e-01,  5.3273e-01, -1.7999e-01, -5.3599e-01, -1.3628e-01,\n",
       "           3.9232e-01,  3.3415e-01,  3.2852e-01, -2.8132e-01,  5.6967e-01,\n",
       "          -8.8329e-01, -1.1318e-02,  1.4594e-01,  2.5863e-02,  4.4457e-01,\n",
       "           5.9376e-02, -2.7644e-01,  3.9410e-01, -6.1087e-01, -4.1939e-01,\n",
       "          -7.4041e-01,  6.6022e-02,  4.6935e-01, -7.9662e-01, -2.8870e-01,\n",
       "          -2.5932e-01,  2.1474e-02,  7.2215e-02,  2.0485e-02,  2.0597e-01,\n",
       "           2.7391e-01,  5.0489e-01, -2.3855e-02,  1.1569e-01, -6.3854e-02,\n",
       "          -2.5995e-01, -2.7688e-01, -2.6475e-01, -1.2558e-01, -4.0941e-01,\n",
       "          -4.8413e-01, -3.7644e-02, -2.6387e-01,  5.0808e-02,  1.3354e-02,\n",
       "           2.0187e-01,  1.9350e-01, -3.1742e-01,  5.6049e-01,  3.8809e-01,\n",
       "           2.6070e-01, -5.7720e-01,  2.0828e-01, -7.9890e-02,  1.4128e-01,\n",
       "          -2.7630e-01,  3.2847e-01, -3.0280e-01, -4.0458e-01, -3.0516e-01,\n",
       "           2.4541e-01,  9.7052e-01, -2.3993e-02, -5.3738e-01,  1.2382e-01,\n",
       "           1.2199e-01, -1.2581e-01,  2.7061e-01, -3.4579e-03,  2.8632e-01,\n",
       "          -1.5006e-01, -5.9731e-01,  7.5602e-01, -5.7531e-02,  4.0097e-01,\n",
       "          -2.9658e-02,  8.5531e-02, -3.4589e-02,  6.1693e-01,  2.3741e-02,\n",
       "          -4.4298e-01, -5.5860e-02, -1.7387e-01,  1.5157e-02,  9.4419e-02,\n",
       "           9.4317e-02,  1.9909e-01,  5.6437e-02,  2.2971e-01,  4.4989e-02,\n",
       "           1.0705e-01,  1.2622e+00, -3.1746e-01, -1.2460e-01,  7.0705e-01,\n",
       "           2.3401e-01,  3.8188e-01,  3.4845e-01, -1.4136e-01,  3.5877e-01,\n",
       "           2.7152e-01,  8.0584e-02, -4.5650e-01,  5.0036e-01,  3.0545e-01,\n",
       "           1.1938e-01, -8.6055e-01, -5.2725e-01, -1.5572e-01,  7.6573e-01,\n",
       "          -3.4261e-01,  3.3450e-01, -4.1011e-01,  5.1298e-01,  5.4985e-01,\n",
       "          -7.3228e-01, -1.0309e-01, -6.7335e-01,  4.8638e-01, -1.1037e-01,\n",
       "          -6.4741e-01,  1.5509e-01, -2.4022e-01, -3.7983e-02,  1.1236e-01,\n",
       "           4.5340e-01, -1.1022e+00, -1.5132e-01,  4.1022e-01, -3.4871e-01,\n",
       "           1.9629e-01, -3.0116e-02,  9.1455e-02,  2.1342e-02,  2.2915e-01,\n",
       "          -3.3271e-01,  2.1307e-03, -4.2180e-01, -8.1493e-01, -4.9809e-02,\n",
       "           6.7291e-01,  1.4094e-01,  8.9490e-02,  7.0496e-02, -5.7160e-01,\n",
       "          -6.2607e-02, -4.2725e-01, -1.5950e-01, -2.8809e-02, -2.4354e-02,\n",
       "          -1.9924e-02, -8.1866e-02,  1.8339e-01,  1.5830e-01, -2.9097e-01,\n",
       "           1.8530e-01,  4.4434e-01,  9.5524e-01, -3.3169e-01, -3.0913e-01,\n",
       "          -4.7081e-01,  4.3880e-01, -5.9913e-02,  4.6908e-01, -2.9761e-01,\n",
       "          -7.0704e-02,  2.1634e-01,  9.5406e-02, -3.9899e-01,  5.1429e-02,\n",
       "          -2.4013e-01,  5.5306e-01, -7.0142e-02, -4.0348e-01,  4.8335e-01,\n",
       "           3.9087e-01,  2.2048e-01, -8.6751e-01, -9.5981e-03,  3.1538e-01,\n",
       "           6.6823e-02,  2.5007e-01,  1.7633e-01,  3.4156e-01, -2.4660e-01,\n",
       "           4.0166e-03, -8.8546e-01,  1.4054e-01,  5.2862e-02, -2.6542e-02,\n",
       "          -1.4982e-01, -5.8322e-01,  4.4505e-01,  6.5995e-01,  6.3707e-01,\n",
       "           3.8145e-01, -9.8880e-02, -5.5021e-01, -5.2458e-01,  3.8749e-01,\n",
       "           1.3073e-01, -2.5049e-01,  5.4600e-01, -1.2101e-01,  2.9498e-01,\n",
       "          -4.9739e-01, -4.9227e-02, -1.2361e-01, -6.4576e-02, -4.7342e-02,\n",
       "           7.1756e-01, -4.7029e-01,  4.5654e-01,  1.7864e-01,  2.9922e-01,\n",
       "           5.6852e-01,  3.4726e-01,  3.1100e-01, -9.1697e-01, -4.1464e-02,\n",
       "           4.8286e-01, -1.1174e-01, -2.6011e-01, -9.0510e-01,  1.8993e-01,\n",
       "           3.2996e-01,  5.6809e-01,  6.7878e-01]], device='mps:0'),\n",
       " tensor([[ 2.5094e-03, -7.7603e-01, -6.4354e-01, -7.8739e-01, -1.7765e-01,\n",
       "          -4.9358e-01, -4.8214e-01,  2.9645e-01,  2.2467e-01,  3.0159e-01,\n",
       "           6.6638e-02, -9.6306e-02,  2.3930e-01,  2.2297e-01, -3.1967e-01,\n",
       "          -2.9578e-01,  2.6966e-02,  3.6815e-01, -6.4729e-02,  8.0701e-01,\n",
       "           3.5496e-01,  3.8282e-01, -2.9154e-03, -4.7794e-02, -3.3786e-01,\n",
       "           4.1325e-01,  8.9081e-02,  1.0875e-01,  5.7795e-01,  1.0654e-01,\n",
       "          -1.7245e-01,  5.0537e-01, -2.6512e-01,  1.0421e+00, -7.3412e-02,\n",
       "          -4.0096e-01,  4.2324e-01, -5.3976e-01,  7.2103e-01,  3.4765e-01,\n",
       "           3.9276e-01,  1.9420e-01, -1.4392e-01,  2.7615e-01,  3.0925e-01,\n",
       "           1.8988e-01, -8.0851e-02, -5.3630e-01,  3.2571e-01, -3.0943e-01,\n",
       "          -1.7616e-02, -2.8855e-01,  3.7303e-01,  2.4016e-01, -4.4431e-01,\n",
       "          -2.4423e-01,  1.7877e-01, -6.9982e-02, -4.8404e-01,  3.2689e-01,\n",
       "          -8.9055e-02,  6.0931e-01, -1.2791e-01,  9.7445e-02, -1.5709e-01,\n",
       "           2.2240e-01, -4.2557e-01, -6.1972e-01, -3.6467e-01, -2.9191e-01,\n",
       "          -4.6546e-02,  2.2044e-01,  3.0243e-02,  1.8793e-01, -1.0593e-02,\n",
       "          -3.4950e-01,  1.9529e-01, -2.4812e-01, -2.7753e-01, -1.1452e-01,\n",
       "           9.8439e-02,  1.9154e-01, -1.7799e-02, -3.2264e-01, -1.0873e-01,\n",
       "           8.9044e-02,  1.3174e-01,  3.0381e-01, -4.9622e-03, -1.6806e-01,\n",
       "           9.2048e-02, -1.6027e-01, -4.4083e-01, -2.9422e-01,  2.1513e-01,\n",
       "           9.7358e-02,  4.6479e-01, -1.5215e-01,  1.0510e-01, -3.2864e-02,\n",
       "           4.4830e-02,  8.5912e-02,  4.9136e-02,  3.6633e-01, -8.0000e-02,\n",
       "          -6.0276e-01,  4.2129e-01, -3.1155e-01, -6.2110e-01, -2.8760e-01,\n",
       "          -2.3092e-01, -3.2808e-01,  1.0239e-01, -3.3374e-01,  1.6613e-01,\n",
       "           6.5269e-02, -2.7505e-01, -4.9941e-02, -1.3055e-01, -1.1246e-02,\n",
       "           1.1995e-01, -2.4769e-01,  1.8099e-01,  1.1720e-01,  1.6843e-01,\n",
       "           7.0295e-02,  1.5321e-01,  2.2263e-01,  6.3613e-01,  2.9430e-01,\n",
       "           1.0461e-01,  2.3573e-01, -1.9072e-01, -2.6784e-01,  1.2418e-01,\n",
       "           4.6954e-02,  3.6396e-01, -2.7249e-01,  1.3857e-01,  9.8303e-02,\n",
       "           4.6398e-02,  3.7335e-02, -1.8128e-01, -2.9439e-01,  4.9317e-02,\n",
       "          -3.4762e-01, -1.6075e-01, -1.3688e-02, -3.3187e-01, -3.3683e-02,\n",
       "          -4.2825e-01, -1.0789e+00,  5.3182e-01, -2.1047e-01, -7.4814e-02,\n",
       "          -1.2806e-01,  7.7973e-01, -4.0525e-01, -1.3506e-02,  2.5355e-01,\n",
       "           9.2344e-01, -4.9283e-02,  5.5602e-01,  3.8889e-02, -3.9356e-02,\n",
       "           1.3161e-01, -5.0963e-02, -9.1892e-02, -1.3065e-02,  7.0292e-02,\n",
       "           1.8449e-02, -2.1710e-01,  3.7095e-01,  1.3632e-01,  3.9298e-01,\n",
       "          -5.7324e-01,  2.7231e-01,  1.2358e-01, -7.4571e-02,  2.8339e-01,\n",
       "           2.8194e-01, -9.4886e-01, -3.2595e-01, -5.9470e-01, -1.5677e-01,\n",
       "          -2.4295e-01,  4.0924e-01,  2.7556e-01,  2.5921e-01,  1.4709e-01,\n",
       "           2.7380e-01,  2.2903e-01,  3.4499e-01,  3.3422e-01, -9.9684e-03,\n",
       "           3.9742e-02,  4.7342e-01,  2.3564e-01, -1.3079e-01,  2.4556e-01,\n",
       "          -7.9959e-01,  4.3222e-01,  9.2285e-02,  1.4512e-01, -4.2333e-04,\n",
       "           2.8226e-01,  6.4513e-03,  1.7959e-01,  9.3057e-02, -2.7955e-01,\n",
       "          -1.8548e-01, -1.8081e-01, -4.3689e-01, -1.4264e-01, -2.9027e-01,\n",
       "           1.6625e-01, -1.4734e-01, -1.0121e-01,  5.2489e-01,  2.4208e-01,\n",
       "          -1.3778e-01,  1.4587e-01,  1.5196e-01, -3.6433e-01, -2.2526e-02,\n",
       "           3.1293e-01, -2.7385e-01,  2.1622e-02, -2.7672e-01, -2.0485e-01,\n",
       "           3.7328e-01, -2.4940e-02,  5.2086e-01,  3.5396e-01, -1.7102e-02,\n",
       "          -1.6500e-01, -6.9645e-01,  6.5025e-02, -3.4316e-01,  3.2847e-02,\n",
       "          -4.9146e-01, -4.5275e-01, -5.8290e-01, -1.6520e-01,  6.1031e-01,\n",
       "           3.6340e-01, -2.6389e-02,  7.4453e-01,  3.6381e-01,  3.4445e-01,\n",
       "          -3.4769e-01, -2.5924e-01,  7.1685e-01,  1.4084e-01, -6.0077e-01,\n",
       "          -9.7348e-02, -4.1927e-02, -5.2857e-01,  5.0694e-01, -8.1073e-03,\n",
       "           1.5075e-01, -2.6151e-01,  2.9972e-01,  3.4847e-01, -4.4663e-01,\n",
       "           2.4421e-01,  8.4559e-01,  7.5328e-03, -2.5513e-01,  5.2429e-01,\n",
       "          -4.3819e-01,  8.2773e-01,  2.5817e-01, -6.1569e-01, -4.0724e-01,\n",
       "          -4.5738e-01, -4.2084e-01, -3.8514e-01,  2.5854e-03, -8.6744e-01,\n",
       "          -1.1858e-01, -9.3166e-02,  1.3690e-01,  1.5934e-01, -1.8017e-01,\n",
       "           2.7613e-01,  1.9645e-01, -3.4796e-01, -2.0851e-01, -6.6412e-02,\n",
       "          -3.2466e-02,  5.3390e-02, -5.4696e-01,  2.8592e-01,  7.9877e-02,\n",
       "          -1.5638e-01,  1.3765e-01,  2.3398e-01, -3.6833e-01, -2.2967e-01,\n",
       "           2.6672e-01, -2.2140e-01, -6.0803e-01, -1.5790e-01,  4.0555e-01,\n",
       "           1.8342e-01,  2.1137e-01,  2.7013e-01, -3.8388e-01, -7.0383e-01,\n",
       "           6.2139e-02,  8.1809e-01, -2.5049e-01, -4.2503e-01, -1.7861e-01,\n",
       "          -1.1579e+00, -7.0929e-02, -6.4058e-02, -2.3525e-01, -5.5601e-01,\n",
       "           2.4539e-01,  9.9789e-03, -2.4518e-01, -2.0366e-01, -2.0472e-01,\n",
       "           2.9062e-01,  4.1707e-01,  3.1636e-01,  1.5095e-01, -5.0305e-01,\n",
       "          -3.4081e-01, -5.9977e-01, -9.2189e-01, -3.0730e-01,  1.6302e-01,\n",
       "           1.5218e-01,  2.9625e-01, -1.7935e-01, -7.9909e-02, -3.2240e-01,\n",
       "           3.6202e-01, -5.0540e-01,  9.1929e-03,  1.4106e-01, -3.6083e-01,\n",
       "          -1.3676e-01,  4.0543e-01, -1.0324e-01, -2.5268e-01,  2.0782e-01,\n",
       "           2.8613e-01,  5.7365e-02, -3.8435e-01,  3.2368e-02,  4.4046e-01,\n",
       "           2.4860e-01,  2.5931e-02, -4.4375e-01,  4.4556e-02,  3.4612e-01,\n",
       "          -2.3934e-01,  2.8167e-01, -5.1758e-01, -1.5571e-01, -4.4858e-01,\n",
       "           2.9094e-01,  3.5566e-01, -3.4882e-01,  6.2367e-02, -2.0425e-01,\n",
       "           9.7989e-02, -2.1348e-01, -3.8678e-01,  2.3560e-01,  5.9552e-01,\n",
       "          -7.3198e-02, -4.3558e-01,  3.3462e-01, -2.4124e-01, -1.1284e-01,\n",
       "           3.6448e-02, -6.6252e-02,  1.8014e-01, -3.8357e-01, -8.8642e-02,\n",
       "          -9.5383e-02, -8.7215e-02,  8.2821e-01,  2.1047e-02, -4.1697e-01,\n",
       "           8.7230e-03, -1.7249e-01, -3.7600e-01,  9.3535e-01, -6.1636e-02,\n",
       "          -1.1247e-01,  6.5870e-01, -2.7035e-01,  1.1696e-01,  6.7684e-01,\n",
       "          -1.0856e-01,  1.3190e-01, -1.1005e-04,  2.2437e-01,  1.3304e-01,\n",
       "          -1.3636e-01, -8.5588e-02,  4.0402e-01, -6.7084e-02, -6.0046e-02,\n",
       "           1.9464e-01, -1.9850e-01, -3.3869e-01,  2.4727e-01,  1.3828e-01,\n",
       "           4.0977e-01, -3.3825e-01,  1.1032e-01, -1.2469e-01, -5.0683e-01,\n",
       "           9.9436e-01,  3.9197e-01,  6.0586e-01,  2.2938e-01, -1.1408e+01,\n",
       "           1.6005e-01,  6.9706e-01, -5.4557e-01, -1.8859e-01,  7.0108e-02,\n",
       "           4.4122e-02, -5.4401e-01,  3.0451e-01, -8.5404e-02, -2.5098e-01,\n",
       "          -5.6050e-01, -1.7060e-01,  3.0824e-01, -4.1696e-01, -3.8478e-03,\n",
       "           6.2855e-02, -6.6196e-01,  3.2733e-01, -2.6619e-01,  3.9636e-01,\n",
       "          -3.7908e-01, -1.2275e-01,  1.7463e-02, -9.4297e-02, -2.3741e-01,\n",
       "           2.4045e-02,  2.7681e-01,  6.5524e-01,  3.5509e-01, -1.7527e-01,\n",
       "           1.3547e-01, -4.5203e-01,  6.6564e-02, -1.5049e-01, -7.4502e-01,\n",
       "           1.7099e-01, -1.9092e-01,  3.6802e-01,  3.8176e-02, -4.1595e-01,\n",
       "           2.8943e-01, -5.8077e-01,  1.1266e-01,  4.5644e-01,  8.6001e-01,\n",
       "           4.9959e-01,  9.5201e-01,  8.8797e-02,  2.3604e-01, -2.9558e-01,\n",
       "          -7.7910e-01,  2.9195e-01, -2.7124e-01, -3.9676e-01, -2.6435e-01,\n",
       "           5.7985e-01, -5.9084e-01, -6.6055e-02, -4.6488e-01,  9.3852e-04,\n",
       "           5.3239e-01,  3.3204e-01, -2.9056e-01, -6.7767e-01, -1.7784e-01,\n",
       "           3.4346e-02,  1.2536e-01,  1.9901e-01, -2.7111e-01,  3.9503e-01,\n",
       "          -1.9847e-01,  5.5901e-01,  2.8058e-01, -4.7682e-01,  1.2545e-01,\n",
       "          -1.3087e-01, -1.3123e-01,  1.9487e+00, -1.7052e-01,  1.1300e-01,\n",
       "           1.2546e-01,  3.6745e-01, -5.3082e-01, -6.2872e-01,  1.3637e-01,\n",
       "           6.5491e-01,  1.0107e-02,  4.7437e-01,  3.4576e-01, -7.9695e-02,\n",
       "          -1.0278e+00,  1.2009e-01,  2.0648e-01,  5.2507e-02,  8.2554e-01,\n",
       "          -9.2932e-01,  4.4135e-02,  5.5268e-01,  2.0314e-01, -2.2036e-01,\n",
       "           5.9647e-02, -1.9972e-01, -7.5509e-01,  5.6636e-01, -3.7235e-01,\n",
       "           1.3626e-01, -4.1534e-01,  6.2337e-02,  4.3691e-01, -1.0082e-01,\n",
       "           2.8439e-01,  2.1725e-01,  8.4393e-02, -1.8147e-01, -2.6968e-01,\n",
       "          -9.8012e-02,  1.8613e-01, -4.1148e-01, -7.6009e-01, -4.0703e-01,\n",
       "          -2.0349e-01,  2.4760e-01,  4.9482e-01, -1.9291e-01,  1.5096e-01,\n",
       "          -3.6524e-01, -8.5209e-03,  2.1909e-01, -3.9130e-01,  1.9638e-01,\n",
       "           2.7801e-01,  2.4265e-01,  3.0093e-01, -4.9366e-01, -2.5754e-01,\n",
       "          -5.2037e-01, -7.2051e-02,  9.5461e-02, -2.6055e-01, -2.3577e-01,\n",
       "          -1.9921e-02, -3.0184e-01, -1.6654e-01,  1.9820e-01,  5.1052e-02,\n",
       "           5.6111e-01,  4.5226e-01, -1.0799e-01,  3.3802e-02, -4.4723e-02,\n",
       "          -3.4893e-01,  1.7807e-01, -3.3366e-01,  5.5716e-01, -6.7077e-01,\n",
       "          -5.6626e-01, -1.4746e-01, -9.5920e-02, -5.4778e-01, -1.4881e-03,\n",
       "           9.0524e-02, -1.3194e-02, -3.9910e-02,  3.6070e-01,  7.2567e-02,\n",
       "           1.5352e-01, -6.1252e-01, -1.6015e-01, -2.1059e-01,  1.2832e-01,\n",
       "           3.0693e-01,  1.4800e-01, -2.0756e-01, -2.7357e-01, -2.0801e-01,\n",
       "          -2.0887e-02,  5.3623e-01, -3.4345e-01, -4.6612e-01,  1.7962e-01,\n",
       "           3.2527e-01, -3.1491e-01, -9.8617e-02,  6.7383e-02,  4.9989e-02,\n",
       "          -2.0493e-01, -7.3032e-01,  2.6624e-01, -5.2014e-02,  4.9027e-01,\n",
       "           3.9670e-01,  9.5413e-02, -5.9441e-03,  2.0561e-01,  6.5076e-02,\n",
       "          -2.9551e-01, -4.7805e-02, -4.4282e-01,  9.9429e-02,  5.6958e-01,\n",
       "          -1.5964e-01,  5.3416e-01,  4.6404e-02, -5.5900e-01,  6.4402e-02,\n",
       "           2.1737e-01,  1.0254e+00, -1.5598e-01, -2.2978e-01,  5.5609e-01,\n",
       "           9.3542e-02, -1.2028e-01,  1.6378e-01, -2.9565e-01,  9.1217e-02,\n",
       "           4.0566e-01, -1.1364e-01, -3.7475e-01, -1.5086e-01,  7.4672e-02,\n",
       "           5.2457e-01, -5.3479e-01, -5.8809e-01, -4.4621e-01,  1.4686e-01,\n",
       "           4.6963e-02,  5.0071e-01, -2.2754e-01,  3.0338e-01,  2.3438e-01,\n",
       "          -8.8726e-01,  1.1958e-01, -5.2252e-01,  2.8393e-01,  3.1834e-02,\n",
       "          -1.9481e-01, -3.0872e-02, -5.6573e-01,  5.2379e-02,  3.5380e-01,\n",
       "          -1.1614e-01, -9.4304e-01,  1.5220e-01,  5.1933e-01, -6.3573e-01,\n",
       "           8.9088e-02, -4.1933e-01,  2.0593e-01,  2.6074e-02, -1.3882e-01,\n",
       "           2.9851e-01,  2.4470e-01, -4.7253e-01, -6.5147e-01, -1.2383e-01,\n",
       "           5.5470e-01, -1.0652e-01,  1.4964e-01,  1.9418e-02, -1.3342e-01,\n",
       "          -3.3293e-01, -1.9107e-01, -4.7163e-02, -7.0295e-02,  1.2775e-01,\n",
       "          -2.7494e-02, -1.9072e-01, -1.8151e-02,  3.8853e-01,  8.2250e-02,\n",
       "           4.0289e-01,  8.7941e-01,  6.2624e-01,  1.7382e-01, -6.5339e-01,\n",
       "          -2.8310e-01, -1.1725e-01, -2.8089e-02,  3.7968e-01, -1.9638e-01,\n",
       "          -8.9095e-03,  1.8355e-01,  1.6837e-01, -3.7815e-01,  2.2121e-01,\n",
       "          -2.3200e-01,  1.9327e-01, -2.6335e-01,  1.2638e-01,  2.6221e-01,\n",
       "           2.8087e-01,  8.3241e-02, -2.0840e-01,  2.1564e-01,  1.7866e-01,\n",
       "          -4.7031e-02,  7.0805e-01,  2.2444e-01,  5.3166e-01, -4.1613e-01,\n",
       "           1.9231e-03, -8.0530e-01, -2.4678e-01,  3.4441e-02, -2.2563e-01,\n",
       "          -1.5505e-01, -4.4638e-01,  6.0944e-01,  4.3643e-02,  4.1664e-01,\n",
       "           1.7721e-01,  2.3549e-01, -1.0460e+00, -3.8034e-01,  1.5483e-01,\n",
       "          -7.7975e-04,  7.3941e-02,  1.1436e-01,  5.0715e-01,  4.4412e-01,\n",
       "          -6.8081e-01, -4.4889e-01, -1.1779e-01,  8.6630e-02, -1.3658e-01,\n",
       "           9.2892e-01, -6.5533e-02,  4.7638e-03, -1.5389e-01,  2.0314e-01,\n",
       "          -7.2813e-02,  7.6990e-01, -5.5826e-02, -6.6628e-01, -6.6394e-02,\n",
       "          -4.9148e-02, -1.9445e-01,  2.2660e-01, -4.4205e-01,  4.4218e-02,\n",
       "           4.1154e-01,  5.8815e-01,  4.0953e-01]], device='mps:0'),\n",
       " tensor([[-0.3664, -0.3331, -0.5190,  ...,  0.2090,  0.5275,  0.2347],\n",
       "         [-0.0626, -0.6265, -0.7028,  ...,  0.1217,  0.4033,  0.6177]],\n",
       "        device='mps:0'),\n",
       " tensor([[-0.0278, -0.7575, -0.5307,  ...,  0.0801,  0.9419,  0.7043],\n",
       "         [-0.0114, -0.7980, -0.7141,  ...,  0.1732,  0.4728,  0.5635],\n",
       "         [-0.0956, -0.4754, -0.5534,  ...,  0.1531,  0.6314,  0.4641]],\n",
       "        device='mps:0'),\n",
       " tensor([[-1.6720e-01, -5.2486e-01, -4.5395e-01, -6.3577e-01, -4.3561e-01,\n",
       "          -1.5533e-01, -5.8268e-01,  2.5578e-01, -1.8350e-01,  5.0250e-01,\n",
       "           1.6126e-01,  1.3783e-01, -2.1897e-01, -4.6204e-01, -6.9128e-01,\n",
       "          -2.0657e-01, -2.5177e-01,  1.9475e-01,  4.1034e-01,  5.6174e-01,\n",
       "           2.8700e-01,  1.6722e-01, -3.7843e-01,  2.1455e-01, -5.1055e-01,\n",
       "           1.5812e-01, -2.8156e-01, -6.3669e-02,  4.5597e-01,  6.8516e-02,\n",
       "          -2.1018e-02,  4.7623e-01, -2.8390e-02,  1.0709e+00, -7.7172e-02,\n",
       "          -3.0626e-01,  1.0760e-01, -4.1569e-01,  8.9012e-01, -5.6131e-01,\n",
       "           6.8910e-01,  6.2264e-01, -1.5577e-01,  1.5282e-02,  2.3088e-01,\n",
       "           1.5974e-01,  5.1535e-02, -5.3995e-01, -5.6558e-01, -1.5427e-01,\n",
       "           1.6123e-01,  1.9663e-01,  5.4635e-01,  6.3307e-02, -7.5459e-01,\n",
       "          -2.6525e-01, -1.3650e-01,  3.0473e-01,  1.7296e-02,  3.5135e-01,\n",
       "          -7.4312e-02,  5.0134e-01,  5.1710e-02, -6.9956e-02,  1.0589e-01,\n",
       "           2.1145e-01, -6.0512e-01, -2.5610e-01, -5.7211e-01,  2.0313e-01,\n",
       "           2.0375e-01,  5.2893e-01,  1.1646e-01, -4.0534e-01,  1.4546e-01,\n",
       "           7.6205e-02,  9.5666e-02, -2.8535e-02, -5.2510e-01, -1.6272e-01,\n",
       "           2.0181e-01,  3.7399e-01,  2.1510e-01, -2.0973e-01,  1.4960e-01,\n",
       "           2.5140e-01,  5.5226e-01, -5.2582e-02, -1.1641e-01, -4.8629e-02,\n",
       "           2.9244e-01, -2.7514e-01, -7.8764e-01, -3.1798e-01, -1.0430e-01,\n",
       "          -3.4914e-01,  7.9176e-02, -4.2834e-01, -2.7512e-01, -3.3882e-01,\n",
       "           1.5050e-01,  1.0200e-01,  2.5842e-01,  7.5482e-02, -7.5174e-01,\n",
       "          -4.1705e-01,  4.6029e-01,  8.0138e-02, -4.2192e-01, -3.1245e-01,\n",
       "          -1.3741e-01, -2.1173e-01,  9.7620e-02, -4.0036e-01, -8.6561e-02,\n",
       "          -1.3006e-01,  2.5022e-01, -1.4378e-01,  5.4056e-02,  4.9777e-01,\n",
       "           2.5708e-01, -5.8312e-01,  1.0353e-01,  2.2601e-01,  4.0089e-02,\n",
       "           3.0807e-02,  6.0830e-01, -5.3486e-02,  4.9134e-01,  1.2285e-01,\n",
       "           2.6257e-01,  1.5138e-01, -6.2596e-01, -5.3560e-01, -2.0734e-01,\n",
       "           1.3350e-01, -3.3082e-01, -8.4107e-01,  3.7865e-01,  3.1799e-02,\n",
       "           5.1144e-02, -9.6587e-02, -6.7862e-01, -3.0496e-01, -3.9669e-01,\n",
       "          -2.7230e-01,  1.4708e-01, -3.5011e-01, -1.3439e-01, -2.7277e-01,\n",
       "          -6.7571e-01, -8.5650e-01, -1.1642e-03, -1.3369e-01,  4.2598e-01,\n",
       "          -5.3225e-01,  4.3086e-01, -2.3121e-01, -5.0388e-01,  4.1057e-01,\n",
       "           5.9947e-01, -1.7783e-01,  2.8322e-01, -5.0157e-01,  3.3619e-02,\n",
       "           3.4604e-01, -3.8918e-02, -5.9573e-02, -3.8183e-02,  1.3359e-01,\n",
       "           3.4058e-01,  2.0341e-01,  1.0286e-01, -2.3079e-01,  3.2100e-01,\n",
       "          -3.7378e-01, -5.5154e-03, -2.6355e-01,  1.6162e-01,  3.2661e-02,\n",
       "           2.9978e-01, -1.6971e+00, -2.5911e-01,  2.0065e-02, -1.8571e-01,\n",
       "          -1.9740e-01,  3.3240e-01,  1.4453e-01,  2.5148e-01, -2.5533e-01,\n",
       "           2.1988e-01,  9.8764e-02, -1.4769e-01, -3.0128e-01,  5.8231e-02,\n",
       "           3.8842e-01,  6.9484e-01,  3.7541e-01,  7.7099e-02, -3.0723e-01,\n",
       "          -7.5503e-01,  1.8407e-01,  3.3852e-01, -1.1523e-01,  5.9383e-01,\n",
       "           3.0489e-02, -8.6229e-01,  1.9019e-01,  2.9946e-02,  2.6756e-01,\n",
       "           5.7319e-02, -2.9941e-01, -4.1668e-01, -3.5307e-01, -3.9315e-01,\n",
       "           2.3395e-01, -1.8355e-01, -3.1845e-01,  4.8844e-01,  8.1310e-02,\n",
       "          -3.0776e-02,  4.2831e-01,  6.0508e-02, -4.4367e-01,  1.9303e-01,\n",
       "           4.4887e-01,  2.6861e-02,  1.1130e-01, -2.7214e-01, -4.0374e-02,\n",
       "          -6.2142e-02, -3.2118e-01,  3.6896e-01,  7.7114e-01, -7.4506e-02,\n",
       "           3.7060e-01, -3.0406e-01,  5.1180e-01, -1.9654e-01,  5.8690e-02,\n",
       "          -1.1981e-01,  3.4078e-02, -2.6240e-01,  2.4533e-01, -1.1898e-01,\n",
       "           9.4908e-02,  8.5883e-02,  7.2222e-01,  3.7433e-01,  7.9341e-02,\n",
       "           3.1509e-01, -4.4737e-03,  1.1637e-02,  5.8059e-01, -3.3724e-01,\n",
       "           3.9433e-01, -1.0171e-01, -3.7982e-01,  3.7113e-01, -4.7675e-01,\n",
       "           3.1991e-01, -1.8581e-01,  1.0010e-01,  2.5551e-01,  3.0602e-02,\n",
       "          -2.9218e-01,  6.1325e-01,  2.2058e-01, -4.4818e-01,  5.2200e-01,\n",
       "          -2.5795e-01,  7.1632e-01, -2.0896e-01, -1.6155e-01, -7.6790e-02,\n",
       "          -9.9737e-01, -4.3677e-01,  1.9994e-01,  1.8271e-01, -6.8430e-01,\n",
       "          -4.9875e-01,  2.8702e-01,  1.1765e-01,  1.0125e-01,  4.2891e-01,\n",
       "           3.8920e-01,  9.2143e-02,  1.0493e-01, -4.8046e-01, -3.6847e-01,\n",
       "          -2.1420e-01,  9.0059e-02, -8.5629e-04, -7.5374e-02,  2.0220e-01,\n",
       "           5.9414e-01,  9.6554e-02,  1.8048e-01,  3.6746e-02, -2.8779e-01,\n",
       "          -1.0518e-01,  1.6372e-01,  4.2762e-01, -5.4888e-01, -2.0137e-01,\n",
       "           6.8173e-02,  4.3129e-02,  4.5140e-01, -3.3691e-01, -3.2667e-01,\n",
       "          -4.2742e-02,  4.8308e-01, -1.2353e-01, -7.1202e-02,  3.3410e-02,\n",
       "          -7.8725e-01,  2.7660e-01, -4.5062e-02, -2.6552e-02, -3.0430e-01,\n",
       "           2.3290e-01,  3.4297e-02, -2.8492e-01, -1.1149e-01, -2.7156e-01,\n",
       "           5.8752e-01,  5.5235e-01,  3.3698e-01, -5.6671e-01, -3.5461e-01,\n",
       "           4.4066e-02, -7.0730e-01,  9.8556e-02, -4.9156e-01,  2.2488e-01,\n",
       "           1.4708e-01,  3.3198e-01, -1.1607e-01,  1.3552e-01, -1.6474e-01,\n",
       "           4.9723e-01, -6.3908e-01,  1.2761e-01,  5.5720e-01, -5.3885e-01,\n",
       "          -2.3301e-01,  4.1839e-01, -5.5797e-01,  1.2634e-01,  4.1453e-01,\n",
       "           3.9631e-01,  4.6106e-02, -4.1602e-01, -7.3707e-02,  9.1249e-01,\n",
       "           5.2639e-01,  1.6638e-01, -6.9733e-01,  9.7526e-02,  3.4043e-01,\n",
       "          -2.0714e-01,  9.5206e-02, -3.2600e-01, -4.7544e-01, -2.1026e-01,\n",
       "          -8.4302e-02,  3.9153e-01, -2.6238e-01,  3.5886e-02,  2.8572e-01,\n",
       "           3.6819e-01, -4.7808e-01, -2.3588e-01,  3.2227e-01,  6.0455e-01,\n",
       "          -1.8222e-01, -3.6901e-01,  1.6512e-01, -5.0214e-01,  2.6335e-01,\n",
       "          -1.4632e-01, -3.4035e-01,  2.7259e-01, -1.2606e-01, -3.7457e-01,\n",
       "          -1.8250e-01, -4.7967e-01,  2.5449e-01,  9.2148e-02, -1.2652e-01,\n",
       "           2.8609e-01, -1.1966e-01, -3.8868e-01,  8.2449e-01, -5.2187e-01,\n",
       "           4.2853e-03,  5.3276e-01, -2.7521e-01, -1.8249e-01,  2.4158e-01,\n",
       "          -2.6392e-01, -1.9683e-02,  1.8527e-02,  9.4766e-01,  1.2345e-01,\n",
       "          -2.7053e-03, -1.1552e-01,  5.1146e-01, -3.8026e-01,  1.5931e-01,\n",
       "          -3.3320e-01,  3.1171e-01, -1.6158e-01,  1.8223e-02, -1.0254e+00,\n",
       "           4.4946e-01, -3.7111e-01, -2.1751e-01, -5.0362e-01, -3.0765e-01,\n",
       "           9.4840e-02, -6.3165e-02,  2.3746e-01,  6.3366e-01, -1.1037e+01,\n",
       "           8.9129e-02,  6.0403e-01, -7.6489e-02,  1.1652e-01,  4.1141e-01,\n",
       "           1.7540e-01, -4.2006e-01,  5.2776e-01, -3.4203e-01,  1.1525e-01,\n",
       "          -4.7982e-01, -1.3280e-01, -8.5294e-02, -3.7948e-01,  3.6349e-01,\n",
       "          -9.6031e-02, -6.4688e-01,  2.4074e-01, -1.3490e-01, -1.7654e-02,\n",
       "          -2.2095e-01,  1.1407e-01,  4.3684e-01, -4.6076e-01, -4.4456e-01,\n",
       "           7.1777e-02, -2.7274e-01,  5.1195e-01,  2.8186e-01, -1.1255e-01,\n",
       "           1.1615e-01,  1.5217e-01,  3.0053e-01, -2.7216e-01, -3.1454e-01,\n",
       "           1.2732e-01, -3.4997e-01, -1.8961e-01, -1.0421e+00, -1.5631e-01,\n",
       "          -2.0656e-01, -4.6750e-01, -1.7710e-02,  6.5233e-01,  7.8743e-01,\n",
       "           1.6358e+00,  9.2174e-01,  3.2668e-01,  1.9802e-01, -2.2250e-01,\n",
       "          -4.6619e-01,  1.1894e-01, -6.7995e-02, -7.0546e-01, -2.4133e-01,\n",
       "           2.8985e-01, -7.5931e-01,  5.3131e-01, -4.7860e-01, -1.3681e-01,\n",
       "           4.2374e-01,  3.9276e-02, -4.2065e-02, -5.9925e-01, -5.3467e-01,\n",
       "          -4.9082e-02,  1.2925e-01,  4.8245e-02, -5.7979e-01,  1.4884e-01,\n",
       "           5.0590e-02,  7.6896e-01, -3.4175e-01, -5.9383e-01,  5.7234e-02,\n",
       "          -1.2763e-01,  3.6795e-02,  1.6636e+00, -1.7915e-01, -4.1931e-02,\n",
       "           1.2354e-01,  4.9556e-02, -6.9680e-01, -4.7087e-01, -4.3059e-01,\n",
       "          -2.7038e-01,  2.5252e-01,  6.6455e-01,  2.0274e-01,  9.0694e-02,\n",
       "          -4.6503e-01, -1.2589e-01,  9.3002e-02,  3.9245e-01,  1.0908e+00,\n",
       "          -5.4949e-01,  2.6147e-01, -7.5939e-02,  2.6322e-01,  1.2335e-01,\n",
       "          -3.6152e-01, -6.2216e-02,  3.9533e-02,  1.6284e-02,  4.5349e-01,\n",
       "           8.6971e-02,  4.2124e-01,  6.5534e-02,  1.2702e-01, -5.0361e-01,\n",
       "          -3.5058e-01, -2.8066e-01, -1.1609e-01, -4.6047e-01, -4.4892e-02,\n",
       "          -2.8268e-02,  1.6753e-01, -1.7086e-01, -2.9519e-01, -2.9979e-01,\n",
       "           2.9427e-01,  2.0283e-01, -6.9070e-02,  9.5939e-03,  6.0460e-01,\n",
       "          -6.3640e-01, -1.8307e-01, -9.4406e-02,  8.6813e-02,  6.9132e-02,\n",
       "           2.1208e-01, -3.8688e-01, -1.1241e-01, -7.3959e-02, -8.4460e-01,\n",
       "          -9.6512e-01, -9.7748e-02,  5.1010e-01, -2.4332e-01,  1.0866e-01,\n",
       "           6.7349e-02,  1.2474e-02, -3.2385e-01,  2.6710e-01, -1.9218e-01,\n",
       "           1.1645e-01,  3.9439e-01, -5.4670e-01,  1.5963e-01, -3.1155e-01,\n",
       "           1.8497e-02, -2.0876e-02, -6.5711e-02, -3.8026e-01, -5.2860e-01,\n",
       "          -7.6356e-01, -6.8179e-02,  5.1232e-02,  5.6135e-01, -3.2886e-02,\n",
       "           1.0488e-01, -1.0398e-01,  7.7290e-02,  7.2752e-01,  8.5206e-01,\n",
       "           2.6106e-01, -3.5137e-01, -4.9530e-01,  5.0728e-02, -2.6647e-02,\n",
       "           5.7946e-02,  2.1021e-01,  7.9023e-02, -2.2641e-01, -1.4403e-01,\n",
       "          -2.6117e-01,  1.1740e+00, -2.3018e-02, -2.5236e-01,  3.1798e-04,\n",
       "           1.2502e+00, -5.8815e-01, -1.4557e-01,  1.7334e-01,  2.6380e-01,\n",
       "          -5.2810e-01, -3.4023e-01,  5.6863e-01, -2.1703e-01,  1.5201e-01,\n",
       "           3.9180e-01,  1.8677e-01,  1.0221e-01, -4.4054e-02,  8.4521e-02,\n",
       "          -1.5903e-01,  3.3655e-01,  3.0166e-02, -1.4937e-01,  7.3811e-01,\n",
       "           1.1911e-01,  8.0888e-02, -1.4277e-01, -2.4375e-01, -2.6556e-01,\n",
       "           1.9428e-01,  1.2395e+00, -4.2394e-01, -5.8109e-01,  8.7338e-01,\n",
       "          -1.4019e-01,  3.5768e-01,  6.0006e-01, -3.9499e-02,  1.3149e-02,\n",
       "           5.7371e-01,  1.1784e-01, -6.0531e-01,  2.3557e-01,  3.0019e-01,\n",
       "           1.5461e-01, -8.7253e-01, -2.0869e-01, -8.1431e-02, -1.2343e-01,\n",
       "          -8.4444e-02,  5.6766e-01,  6.7880e-02,  3.7625e-01,  3.3336e-01,\n",
       "          -8.6361e-01,  2.9748e-01,  3.9963e-02,  8.7689e-01,  3.6637e-02,\n",
       "          -3.2763e-01, -2.5701e-01, -2.0231e-01, -2.1073e-01,  3.1221e-01,\n",
       "           2.4523e-01, -1.3614e+00, -8.3672e-02,  4.5844e-01, -8.1731e-02,\n",
       "           4.3829e-01, -7.9391e-02,  2.9827e-01,  3.0846e-01,  4.5552e-02,\n",
       "          -5.2861e-02,  7.4971e-02, -2.2854e-01, -2.7639e-01,  4.5622e-01,\n",
       "           5.5347e-01, -5.8603e-02, -7.4470e-02,  2.2723e-01, -2.7374e-01,\n",
       "          -3.2137e-01, -2.2843e-01, -1.8679e-01, -2.2410e-01, -1.2894e-01,\n",
       "          -3.6518e-01, -6.9317e-01, -3.1199e-02, -5.0747e-02, -4.5982e-02,\n",
       "           3.4356e-01,  2.6472e-01,  6.5181e-01, -1.2203e-01,  1.5073e-01,\n",
       "          -1.0406e-01,  4.4593e-01, -2.4264e-01,  7.3864e-01,  3.8785e-01,\n",
       "          -1.7432e-01,  2.7267e-01,  1.0448e-01, -2.3479e-01,  1.3287e-01,\n",
       "          -2.8065e-01,  2.5046e-01, -4.0412e-01, -4.0210e-01, -5.9590e-02,\n",
       "          -1.5547e-01, -1.1097e-01, -2.4685e-01,  2.4733e-01,  3.3605e-01,\n",
       "          -8.1116e-02,  9.9147e-01, -2.5425e-01,  6.5276e-01, -3.0278e-01,\n",
       "          -9.8080e-02, -6.8560e-01,  3.0100e-01,  1.5904e-01, -8.1888e-02,\n",
       "          -3.9633e-01, -2.5967e-01, -3.2959e-01, -4.0276e-01,  6.6570e-01,\n",
       "           7.1379e-01, -5.7936e-01, -7.3617e-01, -2.6704e-01, -1.7923e-01,\n",
       "          -2.0926e-02,  5.7887e-02,  3.7235e-02, -2.4464e-01,  1.1875e-01,\n",
       "          -8.1963e-01,  1.9554e-01,  2.5091e-02,  3.7079e-01, -2.2153e-01,\n",
       "           7.4679e-01, -2.9177e-02, -4.3907e-02,  9.2664e-01,  2.8644e-01,\n",
       "           4.0754e-02,  8.6899e-01, -1.2868e-01, -3.3744e-01,  1.2946e-01,\n",
       "           8.9648e-02, -3.6052e-01, -2.4705e-02, -5.5916e-01,  3.7844e-01,\n",
       "           3.7713e-01,  2.0163e-01,  5.3400e-01]], device='mps:0'),\n",
       " tensor([[-1.6720e-01, -5.2486e-01, -4.5395e-01, -6.3577e-01, -4.3561e-01,\n",
       "          -1.5533e-01, -5.8268e-01,  2.5578e-01, -1.8350e-01,  5.0250e-01,\n",
       "           1.6126e-01,  1.3783e-01, -2.1897e-01, -4.6204e-01, -6.9128e-01,\n",
       "          -2.0657e-01, -2.5177e-01,  1.9475e-01,  4.1034e-01,  5.6174e-01,\n",
       "           2.8700e-01,  1.6722e-01, -3.7843e-01,  2.1455e-01, -5.1055e-01,\n",
       "           1.5812e-01, -2.8156e-01, -6.3669e-02,  4.5597e-01,  6.8516e-02,\n",
       "          -2.1018e-02,  4.7623e-01, -2.8390e-02,  1.0709e+00, -7.7172e-02,\n",
       "          -3.0626e-01,  1.0760e-01, -4.1569e-01,  8.9012e-01, -5.6131e-01,\n",
       "           6.8910e-01,  6.2264e-01, -1.5577e-01,  1.5282e-02,  2.3088e-01,\n",
       "           1.5974e-01,  5.1535e-02, -5.3995e-01, -5.6558e-01, -1.5427e-01,\n",
       "           1.6123e-01,  1.9663e-01,  5.4635e-01,  6.3307e-02, -7.5459e-01,\n",
       "          -2.6525e-01, -1.3650e-01,  3.0473e-01,  1.7296e-02,  3.5135e-01,\n",
       "          -7.4312e-02,  5.0134e-01,  5.1710e-02, -6.9956e-02,  1.0589e-01,\n",
       "           2.1145e-01, -6.0512e-01, -2.5610e-01, -5.7211e-01,  2.0313e-01,\n",
       "           2.0375e-01,  5.2893e-01,  1.1646e-01, -4.0534e-01,  1.4546e-01,\n",
       "           7.6205e-02,  9.5666e-02, -2.8535e-02, -5.2510e-01, -1.6272e-01,\n",
       "           2.0181e-01,  3.7399e-01,  2.1510e-01, -2.0973e-01,  1.4960e-01,\n",
       "           2.5140e-01,  5.5226e-01, -5.2582e-02, -1.1641e-01, -4.8629e-02,\n",
       "           2.9244e-01, -2.7514e-01, -7.8764e-01, -3.1798e-01, -1.0430e-01,\n",
       "          -3.4914e-01,  7.9176e-02, -4.2834e-01, -2.7512e-01, -3.3882e-01,\n",
       "           1.5050e-01,  1.0200e-01,  2.5842e-01,  7.5482e-02, -7.5174e-01,\n",
       "          -4.1705e-01,  4.6029e-01,  8.0138e-02, -4.2192e-01, -3.1245e-01,\n",
       "          -1.3741e-01, -2.1173e-01,  9.7620e-02, -4.0036e-01, -8.6561e-02,\n",
       "          -1.3006e-01,  2.5022e-01, -1.4378e-01,  5.4056e-02,  4.9777e-01,\n",
       "           2.5708e-01, -5.8312e-01,  1.0353e-01,  2.2601e-01,  4.0089e-02,\n",
       "           3.0807e-02,  6.0830e-01, -5.3486e-02,  4.9134e-01,  1.2285e-01,\n",
       "           2.6257e-01,  1.5138e-01, -6.2596e-01, -5.3560e-01, -2.0734e-01,\n",
       "           1.3350e-01, -3.3082e-01, -8.4107e-01,  3.7865e-01,  3.1799e-02,\n",
       "           5.1144e-02, -9.6587e-02, -6.7862e-01, -3.0496e-01, -3.9669e-01,\n",
       "          -2.7230e-01,  1.4708e-01, -3.5011e-01, -1.3439e-01, -2.7277e-01,\n",
       "          -6.7571e-01, -8.5650e-01, -1.1642e-03, -1.3369e-01,  4.2598e-01,\n",
       "          -5.3225e-01,  4.3086e-01, -2.3121e-01, -5.0388e-01,  4.1057e-01,\n",
       "           5.9947e-01, -1.7783e-01,  2.8322e-01, -5.0157e-01,  3.3619e-02,\n",
       "           3.4604e-01, -3.8918e-02, -5.9573e-02, -3.8183e-02,  1.3359e-01,\n",
       "           3.4058e-01,  2.0341e-01,  1.0286e-01, -2.3079e-01,  3.2100e-01,\n",
       "          -3.7378e-01, -5.5154e-03, -2.6355e-01,  1.6162e-01,  3.2661e-02,\n",
       "           2.9978e-01, -1.6971e+00, -2.5911e-01,  2.0065e-02, -1.8571e-01,\n",
       "          -1.9740e-01,  3.3240e-01,  1.4453e-01,  2.5148e-01, -2.5533e-01,\n",
       "           2.1988e-01,  9.8764e-02, -1.4769e-01, -3.0128e-01,  5.8231e-02,\n",
       "           3.8842e-01,  6.9484e-01,  3.7541e-01,  7.7099e-02, -3.0723e-01,\n",
       "          -7.5503e-01,  1.8407e-01,  3.3852e-01, -1.1523e-01,  5.9383e-01,\n",
       "           3.0489e-02, -8.6229e-01,  1.9019e-01,  2.9946e-02,  2.6756e-01,\n",
       "           5.7319e-02, -2.9941e-01, -4.1668e-01, -3.5307e-01, -3.9315e-01,\n",
       "           2.3395e-01, -1.8355e-01, -3.1845e-01,  4.8844e-01,  8.1310e-02,\n",
       "          -3.0776e-02,  4.2831e-01,  6.0508e-02, -4.4367e-01,  1.9303e-01,\n",
       "           4.4887e-01,  2.6861e-02,  1.1130e-01, -2.7214e-01, -4.0374e-02,\n",
       "          -6.2142e-02, -3.2118e-01,  3.6896e-01,  7.7114e-01, -7.4506e-02,\n",
       "           3.7060e-01, -3.0406e-01,  5.1180e-01, -1.9654e-01,  5.8690e-02,\n",
       "          -1.1981e-01,  3.4078e-02, -2.6240e-01,  2.4533e-01, -1.1898e-01,\n",
       "           9.4908e-02,  8.5883e-02,  7.2222e-01,  3.7433e-01,  7.9341e-02,\n",
       "           3.1509e-01, -4.4737e-03,  1.1637e-02,  5.8059e-01, -3.3724e-01,\n",
       "           3.9433e-01, -1.0171e-01, -3.7982e-01,  3.7113e-01, -4.7675e-01,\n",
       "           3.1991e-01, -1.8581e-01,  1.0010e-01,  2.5551e-01,  3.0602e-02,\n",
       "          -2.9218e-01,  6.1325e-01,  2.2058e-01, -4.4818e-01,  5.2200e-01,\n",
       "          -2.5795e-01,  7.1632e-01, -2.0896e-01, -1.6155e-01, -7.6790e-02,\n",
       "          -9.9737e-01, -4.3677e-01,  1.9994e-01,  1.8271e-01, -6.8430e-01,\n",
       "          -4.9875e-01,  2.8702e-01,  1.1765e-01,  1.0125e-01,  4.2891e-01,\n",
       "           3.8920e-01,  9.2143e-02,  1.0493e-01, -4.8046e-01, -3.6847e-01,\n",
       "          -2.1420e-01,  9.0059e-02, -8.5629e-04, -7.5374e-02,  2.0220e-01,\n",
       "           5.9414e-01,  9.6554e-02,  1.8048e-01,  3.6746e-02, -2.8779e-01,\n",
       "          -1.0518e-01,  1.6372e-01,  4.2762e-01, -5.4888e-01, -2.0137e-01,\n",
       "           6.8173e-02,  4.3129e-02,  4.5140e-01, -3.3691e-01, -3.2667e-01,\n",
       "          -4.2742e-02,  4.8308e-01, -1.2353e-01, -7.1202e-02,  3.3410e-02,\n",
       "          -7.8725e-01,  2.7660e-01, -4.5062e-02, -2.6552e-02, -3.0430e-01,\n",
       "           2.3290e-01,  3.4297e-02, -2.8492e-01, -1.1149e-01, -2.7156e-01,\n",
       "           5.8752e-01,  5.5235e-01,  3.3698e-01, -5.6671e-01, -3.5461e-01,\n",
       "           4.4066e-02, -7.0730e-01,  9.8556e-02, -4.9156e-01,  2.2488e-01,\n",
       "           1.4708e-01,  3.3198e-01, -1.1607e-01,  1.3552e-01, -1.6474e-01,\n",
       "           4.9723e-01, -6.3908e-01,  1.2761e-01,  5.5720e-01, -5.3885e-01,\n",
       "          -2.3301e-01,  4.1839e-01, -5.5797e-01,  1.2634e-01,  4.1453e-01,\n",
       "           3.9631e-01,  4.6106e-02, -4.1602e-01, -7.3707e-02,  9.1249e-01,\n",
       "           5.2639e-01,  1.6638e-01, -6.9733e-01,  9.7526e-02,  3.4043e-01,\n",
       "          -2.0714e-01,  9.5206e-02, -3.2600e-01, -4.7544e-01, -2.1026e-01,\n",
       "          -8.4302e-02,  3.9153e-01, -2.6238e-01,  3.5886e-02,  2.8572e-01,\n",
       "           3.6819e-01, -4.7808e-01, -2.3588e-01,  3.2227e-01,  6.0455e-01,\n",
       "          -1.8222e-01, -3.6901e-01,  1.6512e-01, -5.0214e-01,  2.6335e-01,\n",
       "          -1.4632e-01, -3.4035e-01,  2.7259e-01, -1.2606e-01, -3.7457e-01,\n",
       "          -1.8250e-01, -4.7967e-01,  2.5449e-01,  9.2148e-02, -1.2652e-01,\n",
       "           2.8609e-01, -1.1966e-01, -3.8868e-01,  8.2449e-01, -5.2187e-01,\n",
       "           4.2853e-03,  5.3276e-01, -2.7521e-01, -1.8249e-01,  2.4158e-01,\n",
       "          -2.6392e-01, -1.9683e-02,  1.8527e-02,  9.4766e-01,  1.2345e-01,\n",
       "          -2.7053e-03, -1.1552e-01,  5.1146e-01, -3.8026e-01,  1.5931e-01,\n",
       "          -3.3320e-01,  3.1171e-01, -1.6158e-01,  1.8223e-02, -1.0254e+00,\n",
       "           4.4946e-01, -3.7111e-01, -2.1751e-01, -5.0362e-01, -3.0765e-01,\n",
       "           9.4840e-02, -6.3165e-02,  2.3746e-01,  6.3366e-01, -1.1037e+01,\n",
       "           8.9129e-02,  6.0403e-01, -7.6489e-02,  1.1652e-01,  4.1141e-01,\n",
       "           1.7540e-01, -4.2006e-01,  5.2776e-01, -3.4203e-01,  1.1525e-01,\n",
       "          -4.7982e-01, -1.3280e-01, -8.5294e-02, -3.7948e-01,  3.6349e-01,\n",
       "          -9.6031e-02, -6.4688e-01,  2.4074e-01, -1.3490e-01, -1.7654e-02,\n",
       "          -2.2095e-01,  1.1407e-01,  4.3684e-01, -4.6076e-01, -4.4456e-01,\n",
       "           7.1777e-02, -2.7274e-01,  5.1195e-01,  2.8186e-01, -1.1255e-01,\n",
       "           1.1615e-01,  1.5217e-01,  3.0053e-01, -2.7216e-01, -3.1454e-01,\n",
       "           1.2732e-01, -3.4997e-01, -1.8961e-01, -1.0421e+00, -1.5631e-01,\n",
       "          -2.0656e-01, -4.6750e-01, -1.7710e-02,  6.5233e-01,  7.8743e-01,\n",
       "           1.6358e+00,  9.2174e-01,  3.2668e-01,  1.9802e-01, -2.2250e-01,\n",
       "          -4.6619e-01,  1.1894e-01, -6.7995e-02, -7.0546e-01, -2.4133e-01,\n",
       "           2.8985e-01, -7.5931e-01,  5.3131e-01, -4.7860e-01, -1.3681e-01,\n",
       "           4.2374e-01,  3.9276e-02, -4.2065e-02, -5.9925e-01, -5.3467e-01,\n",
       "          -4.9082e-02,  1.2925e-01,  4.8245e-02, -5.7979e-01,  1.4884e-01,\n",
       "           5.0590e-02,  7.6896e-01, -3.4175e-01, -5.9383e-01,  5.7234e-02,\n",
       "          -1.2763e-01,  3.6795e-02,  1.6636e+00, -1.7915e-01, -4.1931e-02,\n",
       "           1.2354e-01,  4.9556e-02, -6.9680e-01, -4.7087e-01, -4.3059e-01,\n",
       "          -2.7038e-01,  2.5252e-01,  6.6455e-01,  2.0274e-01,  9.0694e-02,\n",
       "          -4.6503e-01, -1.2589e-01,  9.3002e-02,  3.9245e-01,  1.0908e+00,\n",
       "          -5.4949e-01,  2.6147e-01, -7.5939e-02,  2.6322e-01,  1.2335e-01,\n",
       "          -3.6152e-01, -6.2216e-02,  3.9533e-02,  1.6284e-02,  4.5349e-01,\n",
       "           8.6971e-02,  4.2124e-01,  6.5534e-02,  1.2702e-01, -5.0361e-01,\n",
       "          -3.5058e-01, -2.8066e-01, -1.1609e-01, -4.6047e-01, -4.4892e-02,\n",
       "          -2.8268e-02,  1.6753e-01, -1.7086e-01, -2.9519e-01, -2.9979e-01,\n",
       "           2.9427e-01,  2.0283e-01, -6.9070e-02,  9.5939e-03,  6.0460e-01,\n",
       "          -6.3640e-01, -1.8307e-01, -9.4406e-02,  8.6813e-02,  6.9132e-02,\n",
       "           2.1208e-01, -3.8688e-01, -1.1241e-01, -7.3959e-02, -8.4460e-01,\n",
       "          -9.6512e-01, -9.7748e-02,  5.1010e-01, -2.4332e-01,  1.0866e-01,\n",
       "           6.7349e-02,  1.2474e-02, -3.2385e-01,  2.6710e-01, -1.9218e-01,\n",
       "           1.1645e-01,  3.9439e-01, -5.4670e-01,  1.5963e-01, -3.1155e-01,\n",
       "           1.8497e-02, -2.0876e-02, -6.5711e-02, -3.8026e-01, -5.2860e-01,\n",
       "          -7.6356e-01, -6.8179e-02,  5.1232e-02,  5.6135e-01, -3.2886e-02,\n",
       "           1.0488e-01, -1.0398e-01,  7.7290e-02,  7.2752e-01,  8.5206e-01,\n",
       "           2.6106e-01, -3.5137e-01, -4.9530e-01,  5.0728e-02, -2.6647e-02,\n",
       "           5.7946e-02,  2.1021e-01,  7.9023e-02, -2.2641e-01, -1.4403e-01,\n",
       "          -2.6117e-01,  1.1740e+00, -2.3018e-02, -2.5236e-01,  3.1798e-04,\n",
       "           1.2502e+00, -5.8815e-01, -1.4557e-01,  1.7334e-01,  2.6380e-01,\n",
       "          -5.2810e-01, -3.4023e-01,  5.6863e-01, -2.1703e-01,  1.5201e-01,\n",
       "           3.9180e-01,  1.8677e-01,  1.0221e-01, -4.4054e-02,  8.4521e-02,\n",
       "          -1.5903e-01,  3.3655e-01,  3.0166e-02, -1.4937e-01,  7.3811e-01,\n",
       "           1.1911e-01,  8.0888e-02, -1.4277e-01, -2.4375e-01, -2.6556e-01,\n",
       "           1.9428e-01,  1.2395e+00, -4.2394e-01, -5.8109e-01,  8.7338e-01,\n",
       "          -1.4019e-01,  3.5768e-01,  6.0006e-01, -3.9499e-02,  1.3149e-02,\n",
       "           5.7371e-01,  1.1784e-01, -6.0531e-01,  2.3557e-01,  3.0019e-01,\n",
       "           1.5461e-01, -8.7253e-01, -2.0869e-01, -8.1431e-02, -1.2343e-01,\n",
       "          -8.4444e-02,  5.6766e-01,  6.7880e-02,  3.7625e-01,  3.3336e-01,\n",
       "          -8.6361e-01,  2.9748e-01,  3.9963e-02,  8.7689e-01,  3.6637e-02,\n",
       "          -3.2763e-01, -2.5701e-01, -2.0231e-01, -2.1073e-01,  3.1221e-01,\n",
       "           2.4523e-01, -1.3614e+00, -8.3672e-02,  4.5844e-01, -8.1731e-02,\n",
       "           4.3829e-01, -7.9391e-02,  2.9827e-01,  3.0846e-01,  4.5552e-02,\n",
       "          -5.2861e-02,  7.4971e-02, -2.2854e-01, -2.7639e-01,  4.5622e-01,\n",
       "           5.5347e-01, -5.8603e-02, -7.4470e-02,  2.2723e-01, -2.7374e-01,\n",
       "          -3.2137e-01, -2.2843e-01, -1.8679e-01, -2.2410e-01, -1.2894e-01,\n",
       "          -3.6518e-01, -6.9317e-01, -3.1199e-02, -5.0747e-02, -4.5982e-02,\n",
       "           3.4356e-01,  2.6472e-01,  6.5181e-01, -1.2203e-01,  1.5073e-01,\n",
       "          -1.0406e-01,  4.4593e-01, -2.4264e-01,  7.3864e-01,  3.8785e-01,\n",
       "          -1.7432e-01,  2.7267e-01,  1.0448e-01, -2.3479e-01,  1.3287e-01,\n",
       "          -2.8065e-01,  2.5046e-01, -4.0412e-01, -4.0210e-01, -5.9590e-02,\n",
       "          -1.5547e-01, -1.1097e-01, -2.4685e-01,  2.4733e-01,  3.3605e-01,\n",
       "          -8.1116e-02,  9.9147e-01, -2.5425e-01,  6.5276e-01, -3.0278e-01,\n",
       "          -9.8080e-02, -6.8560e-01,  3.0100e-01,  1.5904e-01, -8.1888e-02,\n",
       "          -3.9633e-01, -2.5967e-01, -3.2959e-01, -4.0276e-01,  6.6570e-01,\n",
       "           7.1379e-01, -5.7936e-01, -7.3617e-01, -2.6704e-01, -1.7923e-01,\n",
       "          -2.0926e-02,  5.7887e-02,  3.7235e-02, -2.4464e-01,  1.1875e-01,\n",
       "          -8.1963e-01,  1.9554e-01,  2.5091e-02,  3.7079e-01, -2.2153e-01,\n",
       "           7.4679e-01, -2.9177e-02, -4.3907e-02,  9.2664e-01,  2.8644e-01,\n",
       "           4.0754e-02,  8.6899e-01, -1.2868e-01, -3.3744e-01,  1.2946e-01,\n",
       "           8.9648e-02, -3.6052e-01, -2.4705e-02, -5.5916e-01,  3.7844e-01,\n",
       "           3.7713e-01,  2.0163e-01,  5.3400e-01]], device='mps:0'),\n",
       " tensor([[-1.8026e-01, -4.4232e-01, -6.3530e-01, -8.3882e-01, -1.2546e-01,\n",
       "          -2.6156e-01, -7.2987e-01,  8.8834e-01,  1.3742e-01,  6.1187e-01,\n",
       "           1.2638e-01,  2.9957e-01, -5.2388e-02,  5.2491e-02, -3.7168e-03,\n",
       "          -3.5088e-01, -4.0738e-01,  3.9836e-01,  1.0616e+00,  2.7886e-01,\n",
       "           4.9073e-01,  5.2902e-01, -2.0238e-01, -2.9319e-01, -8.1767e-01,\n",
       "           4.1088e-01, -1.5851e-02,  1.7507e-01,  2.3728e-01, -8.4636e-01,\n",
       "           2.9442e-01,  3.1880e-01,  1.0864e-01,  9.2454e-01, -2.1080e-01,\n",
       "          -4.3420e-01,  3.1791e-01, -6.7135e-01, -3.8613e-02, -3.1581e-02,\n",
       "           5.4255e-01,  1.1325e-01,  1.4228e-01,  2.7688e-01,  1.6871e-01,\n",
       "           1.3075e-01,  7.4409e-02, -2.3563e-01,  1.4124e-01, -9.0595e-01,\n",
       "          -4.8242e-02, -4.2305e-03,  5.2371e-01,  5.5715e-01, -7.8440e-01,\n",
       "           1.7397e-02,  4.8661e-01,  1.6303e-01, -1.1662e-02,  1.9559e-01,\n",
       "          -4.5218e-02,  4.7434e-01, -2.7927e-01, -2.6301e-02,  2.4026e-01,\n",
       "          -2.5824e-02, -4.8488e-01, -4.7341e-01, -5.2653e-01,  1.2508e-01,\n",
       "          -2.9974e-01,  4.7228e-01,  3.8499e-01, -3.3062e-01,  4.4722e-01,\n",
       "          -4.2223e-01,  1.7017e-02, -2.9650e-01, -5.1168e-01,  3.3759e-01,\n",
       "           2.8954e-01,  4.7824e-01,  1.7831e-01, -1.4169e-01, -1.1066e-01,\n",
       "          -8.0921e-02,  2.5016e-01,  1.4041e-01,  1.6673e-02,  1.0552e-01,\n",
       "           9.9268e-02, -8.3575e-02, -2.5016e-01, -2.6953e-02,  3.8451e-01,\n",
       "          -5.5873e-01,  1.8280e-01, -1.8129e-01,  1.0947e-01, -3.0075e-01,\n",
       "          -4.7085e-02,  2.6777e-01, -3.1902e-01,  2.1156e-01, -1.5542e-01,\n",
       "          -4.1404e-01,  3.2307e-01, -2.3314e-01, -4.9899e-01,  2.4707e-01,\n",
       "           1.2749e-01, -2.6736e-01,  2.6279e-02, -2.8323e-01, -1.3229e-01,\n",
       "           1.0844e-01,  2.0034e-01, -1.2294e-01, -5.3174e-02,  1.7989e-01,\n",
       "           1.7100e-01, -2.0534e-01, -2.6044e-01,  4.5634e-02, -5.9688e-02,\n",
       "          -3.9615e-01,  7.6362e-01,  1.0248e-01,  3.4818e-01,  2.7312e-01,\n",
       "           2.4619e-01,  8.9235e-02, -3.2009e-01, -2.4333e-01, -6.9398e-02,\n",
       "           2.5774e-01,  1.5973e-01, -4.1872e-01, -1.7944e-01,  1.8768e-01,\n",
       "           6.7977e-02, -1.8645e-01, -4.4626e-01, -1.2813e-01, -1.2079e-01,\n",
       "          -8.0812e-02,  9.7628e-02, -2.4063e-01, -1.1392e-01, -1.2523e-01,\n",
       "          -5.2096e-01, -6.6918e-01,  3.2367e-01, -1.0139e-02, -1.5958e-01,\n",
       "          -5.2467e-01,  1.8129e-01,  1.0995e-01,  1.7554e-01,  8.1101e-02,\n",
       "           1.1493e+00,  9.9297e-02,  2.7102e-01, -3.2497e-01, -4.1237e-01,\n",
       "          -1.7473e-02, -2.9013e-03, -3.1320e-01,  4.0578e-01,  1.2205e-02,\n",
       "           1.0240e-01, -5.4999e-02,  4.4965e-01,  1.3480e-02,  3.0615e-01,\n",
       "          -4.0436e-01,  5.5507e-01,  4.0592e-01, -1.1821e-01,  2.6252e-01,\n",
       "           1.7698e-01, -1.4803e+00,  5.3950e-02, -5.4983e-01,  2.0314e-01,\n",
       "          -2.4999e-02,  4.5161e-01,  2.5972e-01,  8.7789e-02,  2.9821e-03,\n",
       "           2.3626e-01,  3.4403e-01, -3.5241e-02,  2.3832e-01, -1.3102e-01,\n",
       "           3.3415e-01,  5.1664e-01,  4.8128e-01,  4.6320e-01, -8.8351e-02,\n",
       "          -5.2857e-01,  2.1771e-01,  2.4819e-01, -4.0674e-02,  2.8556e-01,\n",
       "          -3.2432e-02, -4.3560e-01, -1.0964e-01,  4.9456e-02, -6.8739e-02,\n",
       "          -2.9088e-01,  1.7615e-02, -6.2833e-01,  1.2241e-01, -5.0066e-01,\n",
       "           3.1367e-01, -2.5182e-01, -9.0362e-02,  5.3746e-01,  2.4618e-02,\n",
       "          -1.6689e-01,  2.8526e-01,  2.1940e-01, -2.9726e-01,  8.5165e-02,\n",
       "           4.2408e-01, -3.6217e-02, -1.3894e-01, -1.9763e-01, -1.8102e-01,\n",
       "           1.6181e-01, -6.6410e-01,  9.8319e-01,  2.0733e-01,  8.7583e-02,\n",
       "           1.7656e-01, -2.0487e-01,  1.5232e-02, -4.4265e-01,  3.1788e-01,\n",
       "          -3.1135e-01, -5.4811e-01, -7.2913e-01,  1.7607e-01,  2.1675e-01,\n",
       "           1.7503e-01, -2.7322e-01,  6.3306e-01,  5.0026e-01,  4.4801e-01,\n",
       "          -4.3550e-01,  1.6988e-02,  2.6386e-01,  1.0550e-01, -2.3404e-01,\n",
       "           1.8069e-01, -2.6100e-01, -6.4202e-01,  3.6663e-01, -1.2811e-01,\n",
       "           1.2761e-02, -7.2446e-04,  2.7533e-01,  2.6907e-01, -3.9351e-01,\n",
       "          -5.6766e-01,  4.6157e-01,  3.1542e-02, -3.7814e-01,  6.4209e-01,\n",
       "          -3.4561e-01,  3.3086e-01, -5.6323e-02, -2.5670e-01, -4.2761e-01,\n",
       "          -6.7598e-01, -4.4768e-02,  6.8458e-03, -3.1332e-01, -9.0600e-01,\n",
       "          -4.7059e-01,  3.9810e-01,  1.4332e-01,  1.6931e-01,  3.2656e-01,\n",
       "           5.1251e-01, -8.7150e-02,  6.0687e-02,  6.3879e-02, -1.9244e-01,\n",
       "          -1.3550e-01, -3.6948e-02, -2.7511e-01,  3.5439e-01,  4.2138e-01,\n",
       "           3.4471e-01,  2.8218e-01,  3.9223e-01, -2.1797e-01, -1.2740e-01,\n",
       "          -1.7961e-01, -3.5648e-03, -5.9737e-01, -2.8762e-01,  5.2643e-01,\n",
       "          -7.2141e-02,  5.9034e-01,  6.0938e-01,  2.3492e-01, -4.2682e-01,\n",
       "           1.6498e-01,  5.0961e-01, -2.2448e-01, -2.5854e-01,  4.2514e-01,\n",
       "          -9.0746e-01, -1.6219e-01,  8.7168e-02,  8.4752e-02, -3.0387e-01,\n",
       "           1.5755e-01,  4.6468e-01,  1.7898e-02, -2.0119e-01, -3.1256e-01,\n",
       "           3.4475e-01,  8.6661e-02,  8.4651e-02,  2.1642e-01, -4.3123e-01,\n",
       "          -1.7722e-01, -2.1063e-01, -2.3184e-01, -3.0326e-01, -2.1592e-01,\n",
       "           1.7322e-01,  4.4754e-02, -2.6345e-01, -1.8248e-01, -9.8883e-02,\n",
       "           3.6461e-01, -5.0345e-01, -5.3008e-01,  2.9953e-01, -1.9090e-01,\n",
       "          -1.5948e-01,  2.2633e-01, -2.7313e-01,  1.2923e-01,  2.1944e-01,\n",
       "           1.3498e-01, -3.9947e-01, -2.2111e-01, -6.0296e-02,  7.1042e-01,\n",
       "           3.5863e-01,  4.0341e-01, -3.5135e-01,  5.5041e-01,  2.5020e-01,\n",
       "          -2.6894e-01,  3.2724e-01, -1.9149e-02, -3.0023e-01, -4.4553e-01,\n",
       "           9.8974e-02,  3.1975e-01, -5.2677e-01,  2.1332e-01, -1.0215e-01,\n",
       "          -5.5534e-03, -3.1889e-02, -2.8833e-01,  8.0205e-02, -3.9064e-02,\n",
       "          -3.9047e-01, -5.7563e-01,  1.8567e-01, -3.1113e-01, -3.4037e-01,\n",
       "           2.1586e-01, -4.5828e-03,  1.1810e-01, -3.3710e-01, -3.1393e-01,\n",
       "          -1.2998e-01, -9.0811e-02,  7.3466e-01,  1.5843e-01, -3.0909e-01,\n",
       "           2.2874e-01, -2.6187e-01, -2.6227e-01,  1.4137e-01, -3.6443e-01,\n",
       "          -4.9663e-01,  3.3695e-01, -3.1186e-01, -4.3420e-02,  4.9534e-01,\n",
       "          -2.5107e-01,  1.6927e-01, -2.3024e-01,  3.3764e-01,  1.7367e-01,\n",
       "          -4.4418e-01,  6.5778e-02,  2.3232e-01, -4.1725e-01,  4.0709e-01,\n",
       "           8.3966e-02,  4.4366e-01, -2.7170e-01, -1.0515e-02, -8.4301e-02,\n",
       "           3.4906e-01, -1.8763e-01, -3.9492e-01, -6.3277e-01, -3.0224e-01,\n",
       "           2.0110e-01,  4.3343e-01,  2.6295e-01,  2.7897e-01, -1.1152e+01,\n",
       "           3.7999e-01,  5.8346e-01, -4.1866e-01, -4.6594e-02, -1.1951e-01,\n",
       "           1.0405e-01,  7.2082e-02,  5.2849e-01, -4.9644e-01, -3.7150e-01,\n",
       "          -5.4389e-01, -2.1424e-01, -1.0351e-01, -3.5362e-01, -2.0435e-01,\n",
       "           1.0909e-01, -9.0689e-01,  5.3989e-01, -2.4808e-01,  2.1677e-01,\n",
       "          -2.7082e-02, -1.4863e-02,  1.2968e-01, -5.0828e-01, -1.9091e-02,\n",
       "          -2.5551e-02, -1.7479e-01,  6.0314e-02,  3.8721e-01,  2.1110e-01,\n",
       "           2.5503e-02,  5.3285e-02, -2.4445e-01, -3.5459e-01, -2.8498e-01,\n",
       "          -3.1118e-01, -2.4057e-01, -1.9513e-01, -2.8490e-01, -6.0315e-01,\n",
       "          -8.6188e-01, -2.5326e-01,  4.0296e-01,  1.8047e-01,  7.8639e-01,\n",
       "           6.6804e-01,  9.0338e-01,  2.1034e-01, -3.3941e-03, -6.1751e-02,\n",
       "          -9.7980e-01,  6.2658e-02, -2.5847e-01, -2.9654e-01, -1.8345e-01,\n",
       "           5.1266e-01, -9.3716e-01,  3.5514e-01, -4.8838e-01,  5.7181e-02,\n",
       "           3.8232e-01,  1.6466e-01,  1.8833e-01, -1.6122e-01, -2.6733e-01,\n",
       "          -3.3428e-01, -1.9846e-01, -1.2450e-01, -2.4957e-01, -1.4324e-01,\n",
       "          -2.7619e-01,  5.0767e-01,  3.9902e-01, -3.4528e-01,  2.8240e-02,\n",
       "          -8.6671e-02, -2.9111e-01,  1.6935e+00,  1.3792e-01,  1.7165e-01,\n",
       "          -7.0476e-02,  5.3283e-01, -7.5773e-01, -7.6288e-01,  3.3184e-01,\n",
       "          -2.5133e-01,  4.2806e-01,  2.2950e-01,  2.4008e-01, -2.3805e-01,\n",
       "          -7.9595e-01, -2.5749e-01,  1.7075e-01,  6.0917e-02,  7.1170e-01,\n",
       "          -6.8082e-01, -8.9317e-02, -5.4308e-02,  2.3923e-01, -4.0630e-01,\n",
       "          -1.4717e-01, -5.9971e-01, -5.7581e-01,  2.7959e-01, -4.5525e-01,\n",
       "           1.3217e-01,  3.5520e-01, -2.0370e-01,  3.9210e-01, -4.7474e-01,\n",
       "           3.1491e-01,  3.8010e-01, -2.2435e-02,  2.6213e-02, -2.4900e-02,\n",
       "           6.6466e-01,  4.0200e-01, -4.8305e-01, -4.9664e-01,  5.3772e-02,\n",
       "           1.8671e-01,  2.3887e-01,  5.4003e-02, -3.4584e-01,  2.5884e-01,\n",
       "          -5.5322e-01, -1.1062e-01,  3.8136e-01,  1.6960e-01, -6.4809e-02,\n",
       "           1.3202e-01, -4.6467e-02,  5.0027e-01, -6.6744e-01, -6.3988e-01,\n",
       "          -9.3126e-01, -1.5069e-01,  4.7730e-01, -1.2454e-01,  8.7141e-02,\n",
       "          -4.7498e-01, -3.6818e-02, -4.8013e-02, -9.5999e-03,  5.3335e-02,\n",
       "           1.2719e-02,  1.9869e-01, -2.9641e-01, -4.9786e-02, -1.0996e-01,\n",
       "          -5.5739e-01,  3.4814e-02, -3.4192e-01, -3.1428e-01, -6.6642e-01,\n",
       "          -4.2152e-01, -4.3145e-02, -1.6507e-01, -1.0444e-03,  9.7781e-02,\n",
       "           1.7126e-01, -3.2799e-02, -4.6856e-01,  6.1721e-01,  7.3483e-01,\n",
       "           1.6169e-01, -5.3097e-01, -5.4671e-01, -1.6490e-01,  2.1195e-01,\n",
       "           1.9858e-01,  7.4742e-02,  1.0437e-01, -4.7710e-01, -5.4726e-01,\n",
       "          -1.2380e-01,  7.8853e-01,  1.6040e-02, -6.8430e-01,  3.9007e-01,\n",
       "           7.3871e-01, -3.5237e-03, -2.1670e-02, -1.0618e-01, -6.8319e-02,\n",
       "          -6.6187e-01, -3.7287e-01,  5.2372e-01, -1.8563e-01,  1.2308e-01,\n",
       "           1.9600e-01, -9.2570e-02, -2.0449e-02,  5.9991e-01, -1.1621e-01,\n",
       "          -2.3472e-01, -8.0068e-02, -4.2323e-01,  5.0298e-01,  3.3807e-01,\n",
       "          -1.2082e-01,  6.0316e-01,  2.1889e-01, -1.7355e-01, -1.0556e-01,\n",
       "           3.1984e-01,  1.1170e+00, -3.9562e-01, -7.6102e-02,  8.4811e-01,\n",
       "           7.3972e-02, -2.5433e-01,  8.4302e-02, -3.0754e-02, -1.2848e-01,\n",
       "           1.3537e-01,  1.6327e-01, -2.5349e-01, -1.7150e-02,  3.7064e-01,\n",
       "           3.8045e-01, -7.0193e-01, -7.4534e-01, -1.9640e-01,  1.0131e+00,\n",
       "           1.1395e-01,  6.6263e-01, -3.2265e-02,  5.2020e-01,  6.0150e-01,\n",
       "          -1.0529e+00,  6.5484e-02, -4.2606e-01,  4.5084e-01,  3.2758e-01,\n",
       "          -3.6939e-02, -6.6010e-02, -2.0651e-01, -4.4844e-01,  6.3488e-01,\n",
       "           4.4696e-01, -8.3246e-01, -1.4416e-01,  6.5571e-01, -5.4269e-01,\n",
       "           5.1106e-01, -4.9280e-02,  1.5034e-01,  1.3029e-01,  3.0136e-01,\n",
       "          -5.5170e-02,  2.6973e-01, -6.6353e-01, -4.7970e-01, -1.1991e-01,\n",
       "           7.1953e-01,  2.7125e-01,  1.0736e-01, -1.4688e-01, -9.1519e-02,\n",
       "          -1.1150e-01, -5.4870e-01, -1.7689e-01,  2.8381e-01,  4.4404e-02,\n",
       "           3.7778e-02, -2.1687e-01, -1.6643e-01,  3.6815e-01, -1.0943e-01,\n",
       "           2.8784e-01,  2.9604e-01,  7.1616e-01, -2.0103e-01,  8.5123e-03,\n",
       "          -3.0136e-01,  2.5629e-01, -2.1876e-01,  3.5586e-01, -9.7485e-03,\n",
       "          -1.6554e-01,  1.7756e-01,  7.1069e-02, -5.7272e-01,  1.7963e-01,\n",
       "          -4.6242e-01,  6.8298e-01, -8.8670e-02, -2.4391e-01,  2.9843e-01,\n",
       "           3.1923e-02,  4.5043e-02, -1.2651e-01,  3.4254e-01,  7.6554e-02,\n",
       "           1.5754e-01,  7.8944e-01,  5.3788e-02,  1.0196e-01, -7.9301e-01,\n",
       "           7.2680e-02, -6.4816e-01, -6.4566e-02,  2.0911e-02, -2.3719e-01,\n",
       "          -3.1386e-01, -1.5047e-01,  1.3386e-01,  7.0262e-02,  7.0962e-01,\n",
       "           8.4021e-01, -3.5686e-01, -6.6060e-01, -3.3593e-01,  1.2414e-01,\n",
       "           7.1324e-02, -1.2284e-01,  5.9506e-02,  8.8759e-02,  5.5121e-01,\n",
       "          -6.2762e-01, -1.4263e-01, -3.0292e-01,  1.6980e-01, -2.9811e-01,\n",
       "           4.5968e-01, -2.3501e-01,  1.2388e-01,  2.2884e-01,  2.8990e-01,\n",
       "           3.2426e-03,  6.2036e-01,  2.1862e-01, -5.2221e-01,  6.1076e-02,\n",
       "          -1.5430e-01, -4.4659e-01,  1.1350e-02, -3.2200e-01,  3.9374e-01,\n",
       "           1.2570e-01,  5.3074e-01,  8.7903e-01]], device='mps:0'),\n",
       " tensor([[-1.2655e-01, -7.4520e-01, -7.3208e-01, -2.8851e-01, -4.0142e-01,\n",
       "          -1.1156e-01, -1.7252e-01,  5.3901e-01,  7.7116e-02,  7.3915e-01,\n",
       "           1.7537e-01,  1.7667e-01, -2.6012e-02, -1.4756e-01, -3.8676e-01,\n",
       "          -3.7443e-01, -3.3357e-02,  2.8249e-01, -1.9368e-01, -2.2135e-01,\n",
       "           6.0315e-01,  4.3964e-01, -2.4971e-01,  8.9217e-02, -6.7105e-01,\n",
       "          -4.0270e-02,  1.1560e-01,  9.3162e-02,  8.8618e-02, -7.6946e-01,\n",
       "          -1.3872e-01,  6.3594e-01,  1.0944e-01,  1.1640e+00, -2.1243e-01,\n",
       "          -2.0144e-01,  6.2456e-02, -3.6353e-01,  3.5678e-01, -5.0662e-01,\n",
       "           6.4547e-01,  9.5906e-02,  1.0728e-01,  4.9590e-01,  3.0941e-01,\n",
       "          -9.9768e-02,  3.8522e-01, -9.3185e-01,  5.9163e-01, -5.7836e-01,\n",
       "          -4.5920e-01,  2.5531e-01,  3.5370e-01,  2.4685e-01, -5.3818e-01,\n",
       "          -3.4115e-01,  4.4585e-01,  4.0757e-01, -3.7288e-01,  2.8399e-01,\n",
       "          -2.7777e-01,  4.8607e-01, -9.5103e-02,  1.2683e-01,  3.3310e-01,\n",
       "           2.5259e-01, -3.6833e-01, -4.1241e-01, -3.6330e-01, -1.6269e-01,\n",
       "          -1.7272e-02,  3.9606e-01, -9.4364e-02, -4.2791e-01,  2.8922e-01,\n",
       "          -2.7780e-01, -2.6952e-02, -3.1996e-01, -5.3741e-01, -1.5066e-01,\n",
       "           2.3701e-01,  4.1333e-01,  1.2565e-01, -2.6415e-01,  1.6644e-01,\n",
       "           5.9443e-02,  3.9176e-01,  4.9960e-01,  2.5199e-01, -1.6932e-01,\n",
       "           7.0013e-01,  1.1491e-01, -4.5725e-01, -3.0624e-01,  4.2559e-01,\n",
       "          -1.8026e-01,  4.8127e-01, -2.2585e-01, -1.7340e-01, -2.0582e-01,\n",
       "           2.9714e-02,  1.4234e-01,  1.3911e-01, -1.5366e-01, -1.9418e-01,\n",
       "          -4.5523e-01,  9.2212e-02, -1.0325e-01, -7.3849e-01, -3.2521e-01,\n",
       "          -2.1782e-01,  1.0146e-01,  2.7080e-01, -6.0420e-01, -4.1818e-02,\n",
       "           5.6733e-03,  1.6661e-01, -2.6615e-02, -2.2333e-01,  5.1674e-01,\n",
       "           2.3316e-01, -3.5313e-01, -1.5852e-01,  2.8876e-01,  4.5897e-01,\n",
       "          -5.7108e-01,  7.3468e-01, -3.4321e-01,  3.1998e-01,  4.8956e-01,\n",
       "           9.4281e-02,  8.1907e-02, -4.1177e-01, -1.0418e-01,  2.1088e-01,\n",
       "          -6.6784e-02,  4.1300e-01, -4.0179e-01,  1.9421e-02, -3.8683e-02,\n",
       "           1.6348e-01, -1.7412e-01, -4.3610e-01, -1.1717e-02, -1.1581e-01,\n",
       "          -1.6510e-01, -1.8304e-01, -7.2298e-02, -2.9771e-01, -5.0788e-01,\n",
       "          -5.4121e-01, -5.5751e-01,  1.0337e-01, -2.3788e-01,  1.0912e-01,\n",
       "          -3.0113e-01,  2.7625e-01,  2.0260e-01, -8.1330e-03,  1.9222e-01,\n",
       "           7.9019e-01, -2.1016e-01,  2.6126e-01, -8.5382e-01, -7.1905e-02,\n",
       "          -4.8229e-02, -4.6440e-02, -1.7337e-01,  8.4039e-03,  4.1042e-02,\n",
       "           3.5682e-01, -2.8487e-01, -9.8864e-02, -4.7375e-01,  1.5659e-01,\n",
       "          -5.6279e-01,  4.2615e-01,  1.3419e-01, -4.2396e-02, -1.4768e-01,\n",
       "           9.7335e-02, -1.8931e+00, -1.5465e-01, -2.6779e-02, -1.4878e-01,\n",
       "          -4.4806e-01,  3.9898e-01,  3.0941e-01,  3.3377e-01, -1.3627e-01,\n",
       "          -6.6707e-01,  9.2625e-02,  2.3287e-02,  1.6853e-01,  1.2024e-01,\n",
       "           4.5454e-01,  4.6832e-01,  9.4002e-03,  2.9393e-02,  3.2991e-01,\n",
       "          -6.1158e-01,  2.3558e-01,  4.2741e-02, -2.0119e-01,  7.1801e-02,\n",
       "           1.7236e-01, -2.4868e-01,  1.9310e-01,  1.3628e-01,  1.6636e-01,\n",
       "          -2.4036e-01, -2.2848e-01, -1.5175e-01, -2.2029e-02, -4.4728e-01,\n",
       "           1.2618e+00,  4.6907e-02, -1.5209e-01,  4.7050e-01,  2.5533e-01,\n",
       "          -1.1167e-01, -4.1617e-02,  8.5119e-02, -2.4530e-01, -9.3295e-02,\n",
       "           6.0232e-01, -5.8611e-01, -9.9963e-02, -1.4179e-01, -2.6297e-01,\n",
       "          -1.9659e-02, -6.8554e-01,  6.7727e-01, -2.9161e-01,  2.3121e-02,\n",
       "          -2.3610e-01, -5.6084e-02, -3.2733e-03, -2.4264e-01,  1.1758e-01,\n",
       "          -5.1666e-01, -5.9031e-01, -8.7811e-01,  2.3734e-01,  2.1112e-01,\n",
       "           3.2798e-02, -2.9125e-01,  8.7574e-01,  4.0826e-01,  5.6717e-01,\n",
       "          -2.9647e-02, -2.5191e-01,  3.9790e-01, -2.7886e-01, -1.5353e-01,\n",
       "          -3.6532e-02, -1.0843e-01, -2.4601e-01,  4.9387e-01, -1.4118e-01,\n",
       "          -1.1985e-01, -1.7838e-01,  2.8209e-01,  7.1989e-01, -1.3719e-01,\n",
       "           1.5138e-01,  1.4714e-01, -4.7592e-01, -2.4229e-01,  4.8534e-01,\n",
       "          -9.8380e-02,  1.7773e-01, -2.5375e-01, -5.9610e-01, -5.0816e-01,\n",
       "          -4.2394e-01,  3.6723e-02, -4.1870e-01,  4.6948e-01, -7.8169e-01,\n",
       "          -4.6660e-01,  2.7831e-01, -1.4913e-01,  9.1576e-02, -2.5437e-01,\n",
       "           4.2342e-01,  1.6156e-01,  3.1737e-01, -2.6485e-01, -5.5156e-01,\n",
       "          -5.8013e-02,  2.6101e-01, -6.4054e-02, -2.2062e-02,  1.4868e-01,\n",
       "          -1.6116e-02,  3.8746e-01, -2.7389e-02, -5.7911e-02, -2.5913e-01,\n",
       "           2.2320e-02, -1.8898e-01, -1.6678e-01, -1.5143e-01,  1.2601e-01,\n",
       "           1.4057e-02,  1.0538e-01,  1.8138e-01, -5.9508e-01, -7.2537e-01,\n",
       "           1.5597e-02,  5.9317e-01, -1.4668e-01, -1.0549e-01, -2.0807e-01,\n",
       "          -8.5347e-01, -4.2241e-01, -2.0503e-01, -2.0471e-01, -4.2933e-02,\n",
       "           1.1471e-01,  1.5247e-01, -8.3550e-02, -2.9233e-02, -4.9752e-01,\n",
       "           2.7221e-01,  3.8839e-01,  6.6993e-01,  1.1004e-01, -1.4645e-01,\n",
       "          -2.8724e-01, -2.9969e-01, -2.9549e-01, -6.2650e-03, -2.0145e-01,\n",
       "           2.8245e-01,  2.1621e-01, -5.0797e-01, -2.7820e-01, -8.5465e-02,\n",
       "           5.3849e-01, -5.6729e-01, -5.8864e-01,  1.3343e-01, -3.1772e-01,\n",
       "          -6.6511e-01,  1.3537e-01, -6.8609e-01, -7.7676e-02,  1.6837e-01,\n",
       "           3.0898e-01, -3.5734e-01,  1.5134e-01, -3.4125e-01,  7.0115e-01,\n",
       "           5.0160e-01,  4.2641e-01, -4.4835e-01,  3.9933e-01,  4.5206e-01,\n",
       "          -5.1036e-01,  1.6760e-02, -2.3424e-01, -1.6616e-01, -1.3416e-01,\n",
       "           1.4652e-01,  3.2764e-01, -6.9834e-01,  1.5715e-01,  3.1416e-01,\n",
       "           5.0575e-02, -1.9972e-01, -3.9472e-01,  2.8528e-01, -3.0174e-01,\n",
       "          -1.8285e-01, -3.6691e-01,  1.1957e-01, -1.7406e-01, -3.2604e-01,\n",
       "           2.0167e-01,  7.2607e-02, -6.4363e-02, -2.7870e-01, -1.4528e-01,\n",
       "          -2.5803e-01,  3.0054e-01,  7.2737e-01,  7.0952e-02,  2.7148e-01,\n",
       "           1.0595e-01, -5.6026e-02, -1.7841e-01,  4.6604e-01,  5.7957e-02,\n",
       "          -2.9957e-01,  2.5715e-01, -4.8969e-02,  2.3108e-01,  7.1128e-01,\n",
       "          -3.2841e-01, -4.9112e-01,  9.4450e-02, -1.2226e-01, -2.4214e-01,\n",
       "          -3.2170e-01,  2.9251e-01,  3.9917e-01,  1.2972e-01,  7.2342e-02,\n",
       "          -2.2402e-01,  6.4680e-01, -3.7128e-01, -2.2534e-01, -2.1277e-01,\n",
       "           1.5045e-01, -2.3928e-02, -3.7103e-02, -5.8279e-02, -3.2639e-01,\n",
       "           3.5380e-01,  2.3950e-01,  4.7152e-01,  1.3765e-01, -1.1137e+01,\n",
       "           2.3304e-01,  8.5987e-01, -5.3979e-01,  2.0917e-01,  1.1097e-01,\n",
       "           3.8024e-01, -2.2548e-01,  1.9945e-01, -2.2175e-01, -2.0747e-01,\n",
       "           1.0544e-01, -1.0698e-01,  2.6923e-01, -5.2697e-01,  2.6734e-01,\n",
       "          -9.4496e-02, -6.8301e-01,  4.8033e-01, -3.7921e-01,  5.5370e-01,\n",
       "          -4.9093e-01,  3.7755e-01,  2.4613e-01, -2.4723e-01, -5.5835e-01,\n",
       "          -1.1016e-01, -5.9247e-01,  2.0945e-01,  6.5333e-01, -1.2674e-01,\n",
       "           7.5986e-02,  1.2314e-01,  3.9375e-01, -2.7636e-01, -7.2495e-01,\n",
       "           3.3798e-01, -2.5869e-01,  2.7356e-01,  2.8488e-01, -3.2794e-01,\n",
       "           1.7858e-01, -3.0247e-01,  6.9807e-01,  2.2586e-01,  7.0655e-01,\n",
       "           1.0020e+00,  7.1939e-01, -3.5037e-02, -6.7434e-01,  1.0608e-01,\n",
       "          -7.8312e-01,  4.9115e-02, -7.6924e-01,  1.9184e-01, -1.7609e-01,\n",
       "           4.8911e-01, -5.8598e-01,  8.3167e-02,  9.0617e-02, -5.0307e-03,\n",
       "           1.6894e-01, -1.6678e-01,  2.9473e-01, -2.3873e-01, -1.1305e-01,\n",
       "           3.5025e-01, -1.2861e-01,  2.4635e-01,  1.6825e-01, -4.8248e-01,\n",
       "          -3.1677e-01,  4.1630e-01,  9.9644e-02, -6.3425e-01,  2.9325e-01,\n",
       "           3.9251e-02, -2.3113e-02,  5.3637e-01, -4.7048e-02, -1.3664e-01,\n",
       "           3.8503e-01,  4.1584e-01, -1.0700e-01, -5.4898e-01, -3.8213e-01,\n",
       "          -1.1865e-01,  7.3741e-02,  2.9014e-01,  5.7460e-01, -3.2655e-01,\n",
       "          -3.7755e-01, -3.0112e-01,  1.5459e-01, -1.8642e-03,  1.0263e+00,\n",
       "          -7.1321e-01,  5.3698e-01,  3.1682e-01, -2.2188e-01, -4.7782e-01,\n",
       "          -4.2439e-02, -1.6548e-01, -3.3178e-01,  1.4035e-01, -3.9319e-01,\n",
       "           5.2460e-02,  3.1456e-01,  1.7819e-01,  4.4850e-01, -1.3422e-01,\n",
       "           3.7151e-01,  2.0540e-01,  1.0344e-01, -2.9791e-01,  1.8818e-01,\n",
       "           3.0356e-01,  7.1717e-01, -4.2188e-01, -1.9580e-01, -4.0693e-01,\n",
       "           7.0743e-02,  5.6798e-01,  4.0898e-02, -3.8230e-01,  9.8394e-02,\n",
       "          -7.2955e-01, -9.2036e-02, -2.4510e-01, -3.1327e-02,  4.7172e-01,\n",
       "           1.5715e-01, -2.1307e-02,  2.0088e-02, -2.1966e-01, -3.9113e-01,\n",
       "          -9.2195e-01, -1.6279e-01,  1.2524e-01, -4.8603e-02,  2.1508e-01,\n",
       "          -8.5762e-02, -1.4127e-01, -3.4117e-02,  6.5544e-01, -1.5354e-02,\n",
       "           7.1977e-01,  2.2196e-01, -1.3854e-01,  1.5714e-01, -6.1457e-02,\n",
       "          -4.3722e-01, -2.1751e-01, -2.9514e-01, -1.8853e-01, -2.8292e-01,\n",
       "           6.0776e-02,  1.7421e-01,  1.5628e-01, -3.5800e-01,  4.9556e-01,\n",
       "           5.3650e-03, -1.7080e-01, -1.1970e-01,  6.6134e-02,  1.4497e-01,\n",
       "           1.0977e-01, -3.4114e-01, -2.4418e-01, -9.5924e-02,  7.2654e-02,\n",
       "           2.3113e-01, -4.7076e-01,  3.5924e-02, -5.8183e-01,  1.4300e-01,\n",
       "           1.8358e-01,  1.3775e+00, -3.3167e-02, -5.4050e-01,  1.5789e-01,\n",
       "           6.0053e-01, -3.3132e-01,  1.5013e-01, -1.6961e-02,  2.0315e-01,\n",
       "          -4.3932e-01, -1.6432e-01,  3.2769e-01, -9.7204e-02,  2.7980e-01,\n",
       "           6.6650e-03,  2.8495e-01, -4.1759e-01,  3.4674e-01,  1.1297e-01,\n",
       "          -3.7993e-01, -1.3553e-01, -5.6543e-01, -3.4098e-02,  6.2812e-01,\n",
       "           3.0632e-02, -1.5477e-01, -2.2456e-01,  4.4538e-03, -1.7448e-01,\n",
       "           4.5311e-01,  1.2193e+00,  2.2650e-02,  4.3392e-02,  4.1220e-01,\n",
       "           1.3233e-02, -3.5122e-01,  2.5082e-01,  7.7446e-02,  6.5709e-02,\n",
       "           4.1433e-01,  2.8429e-01, -3.9413e-01,  3.2738e-01,  1.8144e-01,\n",
       "          -1.2196e-02, -8.6635e-01, -7.4945e-01, -1.8081e-01,  2.9443e-01,\n",
       "           2.3014e-01,  3.1770e-01, -6.9946e-02,  5.6758e-01,  8.6579e-01,\n",
       "          -1.0240e+00, -3.6591e-01,  2.1984e-01,  1.1269e-01, -1.3523e-01,\n",
       "          -2.8316e-01,  3.6406e-02, -6.6481e-02, -4.8127e-01,  2.5986e-01,\n",
       "           7.4845e-01, -8.2243e-01,  1.0108e-01,  3.3324e-01, -4.8112e-01,\n",
       "          -4.6681e-02,  1.0242e-02,  2.7858e-02,  3.4290e-01,  4.9584e-02,\n",
       "          -5.8812e-01,  5.7447e-01, -4.3146e-01, -5.1602e-01,  3.3867e-01,\n",
       "           6.1310e-01, -1.7420e-02, -1.5538e-02,  2.5800e-01, -1.9679e-01,\n",
       "           1.0622e-01, -3.8051e-01, -3.0642e-01, -4.1422e-02, -3.3001e-01,\n",
       "           4.2411e-02, -1.9060e-01,  3.5958e-01,  1.5858e-01, -6.1822e-01,\n",
       "           2.4092e-01,  7.2764e-01,  8.2494e-01, -4.0542e-02, -4.1354e-01,\n",
       "           4.3800e-01, -4.4073e-02, -3.2619e-01,  4.6795e-01, -3.7749e-02,\n",
       "          -1.7259e-01,  3.5766e-01,  1.9563e-01, -2.8715e-01, -1.5703e-01,\n",
       "          -2.8124e-01,  2.8023e-01, -6.0972e-02, -3.8436e-01,  2.7954e-01,\n",
       "          -1.8757e-01,  3.1244e-01, -6.0115e-01,  1.0007e-01,  3.1193e-01,\n",
       "          -6.1689e-02,  7.0562e-01,  1.3887e-01,  4.2675e-01, -1.3264e-01,\n",
       "           1.2326e-01, -4.0292e-01, -4.7626e-01,  1.4668e-02,  1.1085e-01,\n",
       "          -3.0529e-01,  2.2641e-01,  1.2448e-01,  3.0816e-02,  7.6298e-01,\n",
       "           7.3448e-01, -4.3314e-02, -6.5195e-01,  4.6754e-02,  2.6127e-02,\n",
       "           1.4634e-01, -1.0814e-01,  3.7987e-01, -2.9931e-01, -7.7688e-02,\n",
       "          -8.8698e-01, -3.9418e-01,  1.6635e-01,  2.5230e-01, -2.3809e-01,\n",
       "           5.2611e-01, -1.3072e-01,  2.8637e-01,  3.2407e-01,  7.9897e-02,\n",
       "           3.1870e-01,  9.1416e-01,  2.1795e-01, -5.4236e-01,  2.1027e-01,\n",
       "           5.3772e-01, -1.5431e-01, -3.0445e-01, -8.9030e-01,  2.3249e-01,\n",
       "           1.3295e-02,  7.6987e-01,  1.5070e-01]], device='mps:0'),\n",
       " tensor([[-1.7181e-01, -6.4549e-01, -5.7448e-01, -7.1045e-01, -9.6808e-02,\n",
       "          -2.5044e-01, -3.9752e-01,  3.9469e-01, -1.4308e-01,  7.3040e-01,\n",
       "           7.2152e-02,  5.7757e-02,  1.2357e-01,  3.6355e-01, -5.0211e-01,\n",
       "          -1.9126e-01, -1.9438e-01,  1.0118e-01,  6.6771e-01,  2.7755e-01,\n",
       "           6.0022e-01,  3.8175e-01,  5.9648e-02,  3.6110e-02, -4.0495e-01,\n",
       "           1.0908e-01,  5.6896e-02, -3.3160e-01,  1.6069e-01, -4.0980e-01,\n",
       "          -8.6072e-02,  3.4855e-01, -4.2609e-03,  1.0719e+00,  1.3050e-02,\n",
       "          -3.3559e-01,  6.4208e-01, -2.3921e-01,  1.3912e-01, -1.8475e-01,\n",
       "           3.9069e-01, -5.0211e-01,  5.8790e-01,  2.7712e-02,  2.6614e-01,\n",
       "           2.3363e-02, -8.8759e-02, -3.5566e-01,  3.6817e-01, -6.2354e-01,\n",
       "          -4.7039e-01, -6.1892e-02,  5.0552e-01,  1.4938e-01, -6.9017e-01,\n",
       "          -6.4498e-01,  2.9070e-01, -9.6768e-02, -4.2830e-01, -9.0308e-02,\n",
       "          -1.1524e-01,  2.9932e-01, -9.8185e-02,  1.1844e-01,  2.5667e-01,\n",
       "          -2.1253e-01, -2.8266e-01, -6.6095e-01, -3.2323e-01, -2.1927e-01,\n",
       "           1.6468e-02, -8.9914e-02,  2.7893e-01, -7.1324e-02, -3.4760e-02,\n",
       "          -3.7109e-02,  2.0370e-01, -1.0604e-01, -2.8808e-01,  5.1780e-01,\n",
       "           1.2277e-01,  1.3383e-01, -2.7375e-02, -1.8511e-01, -4.0588e-01,\n",
       "          -3.2465e-01,  5.2697e-01,  9.7599e-02,  2.8124e-02, -1.7601e-01,\n",
       "           2.8564e-02, -2.7181e-01, -4.2351e-01, -4.2479e-01,  3.8401e-01,\n",
       "           2.3568e-02,  4.5413e-01, -1.3306e-01,  4.0519e-01, -4.1151e-01,\n",
       "          -6.0371e-02,  2.1479e-01, -1.5709e-01,  1.1077e-01, -4.1740e-01,\n",
       "          -3.2331e-01,  1.4327e-01, -4.2038e-01, -5.9865e-01,  2.0443e-01,\n",
       "          -1.0664e-01, -3.8976e-01,  2.0470e-01, -5.0003e-01,  2.1729e-02,\n",
       "           5.9807e-02,  4.6482e-01, -1.0017e-01, -4.5139e-01, -5.9348e-02,\n",
       "           2.6206e-01, -6.8922e-02, -3.3320e-01,  3.6134e-01,  2.3817e-01,\n",
       "          -3.8331e-01,  6.7504e-01,  1.8415e-03,  3.4180e-01,  4.2603e-01,\n",
       "          -1.0778e-01,  1.3478e-01, -5.4941e-01,  2.4809e-01,  1.3738e-01,\n",
       "           9.6144e-02, -3.3339e-02, -4.9153e-01, -3.3870e-01, -1.2708e-01,\n",
       "           1.2475e-01, -1.1541e-01, -2.4505e-01, -1.8222e-01,  3.4313e-01,\n",
       "          -3.6158e-01, -1.5829e-02, -3.8306e-01,  1.8278e-01, -3.7292e-01,\n",
       "          -6.1296e-01, -6.4233e-01,  4.3687e-01,  8.4491e-02,  5.8013e-03,\n",
       "          -3.7342e-01,  7.6568e-01,  2.7921e-01,  9.6937e-02,  5.3783e-02,\n",
       "           7.6019e-01,  9.8583e-02,  1.9684e-01, -5.0296e-01,  1.0274e-01,\n",
       "          -3.2161e-02,  1.5414e-01, -4.1540e-01,  2.5067e-01,  3.1259e-01,\n",
       "           1.2741e-01, -1.9924e-01,  1.1486e-01, -2.0733e-01,  4.9529e-01,\n",
       "          -1.7659e-01,  4.7378e-02,  1.1059e-01, -3.4261e-01, -1.7713e-01,\n",
       "           1.6862e-01, -1.1129e+00,  2.5922e-01, -1.1392e-01, -4.8828e-01,\n",
       "          -4.7882e-01,  1.7053e-01,  3.5474e-02,  1.0687e-01, -4.1283e-02,\n",
       "          -4.7283e-02,  2.5526e-01,  6.2058e-02,  3.9659e-01, -1.3854e-01,\n",
       "           3.8102e-01,  6.0958e-01,  4.3639e-01,  3.3495e-02,  1.0572e-01,\n",
       "          -1.0720e-01, -1.0676e-01,  6.0501e-02,  1.5202e-01,  4.2106e-01,\n",
       "           4.3287e-01, -4.0701e-01, -1.0864e-01,  1.9089e-01, -1.1266e-01,\n",
       "           9.3747e-02, -3.0103e-01, -3.1379e-01,  2.1820e-01, -2.1367e-01,\n",
       "           5.8365e-01, -5.7616e-01, -1.4920e-01,  5.2008e-01,  2.8533e-01,\n",
       "           7.7920e-02,  1.8443e-02,  3.1502e-01, -2.8398e-01,  3.7277e-02,\n",
       "           4.2105e-01, -7.9739e-02, -8.3890e-02, -4.6432e-01,  7.2891e-04,\n",
       "           3.4598e-02, -8.5204e-01,  3.9526e-01,  5.5524e-01, -2.1651e-01,\n",
       "           7.0030e-02, -1.2946e-01,  2.8644e-01, -8.9730e-03,  1.2486e-01,\n",
       "          -4.2913e-01, -5.6476e-01, -8.6001e-01, -8.5230e-02, -1.4568e-01,\n",
       "           4.8711e-01, -3.2702e-01,  4.9277e-01,  6.1041e-01,  5.0842e-01,\n",
       "           3.5494e-02,  2.7759e-01,  1.9492e-01,  2.8990e-01, -7.8568e-01,\n",
       "          -3.3317e-01,  2.9332e-01, -4.3850e-01,  3.8266e-01, -1.5574e-01,\n",
       "          -2.3426e-01,  6.9124e-02,  1.7158e-01,  2.2771e-01, -1.8435e-01,\n",
       "          -6.7281e-02,  3.6134e-01, -1.6210e-01, -3.9969e-01,  5.5423e-01,\n",
       "          -2.7505e-01,  2.3456e-01, -2.6616e-01, -3.5822e-01, -6.6034e-01,\n",
       "          -4.9783e-01, -2.3744e-02,  3.1836e-01,  1.6063e-01, -8.4115e-01,\n",
       "           3.8415e-02, -6.1992e-02, -1.1358e-01, -9.0689e-02, -4.3225e-01,\n",
       "           4.0372e-01,  8.2375e-02, -1.1005e-01, -4.0145e-02, -2.3812e-01,\n",
       "          -4.9792e-01,  5.3299e-01, -2.8395e-01, -1.2137e-01,  2.3736e-01,\n",
       "          -8.1076e-02,  2.0710e-01,  1.7370e-01, -2.0886e-01, -3.3641e-01,\n",
       "          -7.5963e-02, -1.6021e-01,  2.7031e-01, -8.5882e-02,  6.3641e-01,\n",
       "           4.1257e-01,  4.0651e-01,  3.0501e-01, -1.5275e-01, -5.5305e-01,\n",
       "          -1.8490e-01,  3.7508e-01, -2.1432e-01, -6.6863e-01,  3.0617e-02,\n",
       "          -1.1647e+00, -2.8565e-01, -1.4212e-01, -2.1825e-01, -6.2887e-01,\n",
       "          -7.6079e-02,  5.7582e-01, -5.7489e-02,  9.2841e-02,  6.4088e-02,\n",
       "           6.5148e-01,  2.6157e-01, -5.5557e-02,  5.3558e-01, -7.8311e-01,\n",
       "          -7.0398e-02, -6.7994e-01, -3.5415e-01, -4.5159e-01, -9.8750e-02,\n",
       "           4.7609e-01,  7.1101e-02,  1.7522e-01, -4.4968e-01, -5.8390e-01,\n",
       "           4.3794e-01, -5.6580e-01, -5.6455e-01,  2.3257e-01, -3.5567e-01,\n",
       "           7.3726e-01,  2.7841e-02, -3.0041e-01, -3.2738e-01,  6.3258e-01,\n",
       "           5.9889e-02, -2.5262e-01, -2.1002e-01, -2.8881e-01,  6.9236e-01,\n",
       "           2.1988e-01,  1.0852e-01, -4.1732e-01,  3.6433e-01,  5.4488e-01,\n",
       "          -5.1712e-01,  1.5520e-01, -6.0163e-01, -4.7408e-01,  2.6998e-01,\n",
       "           2.2893e-01,  4.7710e-01, -3.1875e-01,  2.5945e-01, -2.1413e-01,\n",
       "           1.0299e-01, -2.8592e-01,  1.0023e-01, -1.6998e-01,  1.1399e-01,\n",
       "           1.8562e-01, -3.6355e-01,  2.3670e-01, -6.4295e-01, -2.3344e-01,\n",
       "          -4.4202e-01, -7.5024e-02,  8.8807e-02, -4.9439e-01, -1.0065e-02,\n",
       "          -2.2013e-01, -1.3759e-01,  1.0585e+00,  4.3212e-02,  1.2852e-01,\n",
       "           1.4746e-01,  1.6325e-01,  1.4503e-01,  8.1781e-01,  1.7731e-01,\n",
       "          -1.3330e-01,  4.5455e-01, -5.1306e-01,  2.0038e-01,  7.2396e-01,\n",
       "          -4.4904e-01, -1.8923e-02,  6.1173e-02,  3.4376e-01,  4.4394e-01,\n",
       "          -2.2634e-01, -1.0942e-01,  2.8729e-01, -1.3221e-01, -7.8852e-03,\n",
       "          -1.6275e-01,  6.1887e-01, -4.4128e-01,  1.5656e-01, -7.9889e-02,\n",
       "          -1.2789e-01, -1.1183e-01, -1.1787e-01, -1.7309e-01, -4.4213e-01,\n",
       "           6.5310e-01,  1.6819e-01, -3.9524e-02,  3.6444e-01, -1.1199e+01,\n",
       "           5.2703e-01,  7.6985e-01, -4.4771e-01, -7.6510e-02, -1.5621e-01,\n",
       "           9.5667e-02, -1.5871e-01,  3.3999e-01, -3.9560e-01,  3.6265e-02,\n",
       "          -3.1791e-01, -8.0143e-02,  3.5560e-01, -3.6357e-01, -2.6766e-01,\n",
       "          -1.2937e-01, -6.1193e-01,  1.7725e-01, -5.8939e-02,  2.9077e-01,\n",
       "          -1.8182e-01, -1.8564e-01, -9.6228e-02, -8.4335e-02,  1.5674e-01,\n",
       "          -8.6843e-02, -3.4911e-01,  4.6851e-01,  2.9896e-01, -1.4806e-01,\n",
       "          -4.4073e-02,  3.2559e-01,  6.2351e-01, -6.2128e-02, -8.7160e-01,\n",
       "           2.4800e-01,  1.1561e-01, -1.3724e-01, -1.8255e-02, -3.6080e-01,\n",
       "           1.3373e-01, -2.2341e-01,  2.0027e-01,  3.8150e-01,  6.1344e-01,\n",
       "           1.0415e-01,  5.2508e-01,  3.2747e-01, -5.5177e-02, -2.0798e-01,\n",
       "          -8.5479e-01,  1.3399e-01, -8.4326e-01,  6.5109e-02, -7.0365e-02,\n",
       "           6.6030e-01, -4.3562e-01,  1.7603e-01, -3.2717e-01, -1.9379e-01,\n",
       "          -5.7700e-02,  1.6666e-01, -3.0321e-01, -5.6551e-01, -3.6769e-01,\n",
       "          -4.6205e-02, -2.0928e-01, -3.1683e-02, -5.5590e-01,  4.2717e-03,\n",
       "          -2.3000e-01,  3.5375e-01,  3.1628e-01,  1.7550e-01,  2.8804e-01,\n",
       "          -4.0143e-01, -2.3534e-01,  2.1452e+00, -4.3451e-01,  3.2584e-01,\n",
       "          -1.1520e-01,  1.0291e-01, -5.6665e-01, -5.6359e-01, -1.3205e-01,\n",
       "           3.1509e-02, -5.0538e-02,  4.3502e-01,  5.3258e-01, -1.1445e-01,\n",
       "          -6.8055e-01,  4.2053e-02,  4.4455e-01,  8.3219e-02,  1.0224e+00,\n",
       "          -4.0814e-01, -4.4237e-01, -6.7758e-02,  1.5016e-01, -4.0676e-01,\n",
       "          -2.1966e-01, -3.0815e-01, -4.0644e-01,  1.8817e-01, -3.9074e-01,\n",
       "           3.8911e-02, -3.9507e-02,  3.4119e-01,  4.4342e-01,  9.1317e-03,\n",
       "           2.3687e-01,  2.8360e-01, -1.4593e-01, -3.3315e-01, -5.6356e-01,\n",
       "           3.4643e-01,  4.9624e-01, -5.7022e-02, -6.7289e-01, -1.6730e-01,\n",
       "          -2.4478e-01,  2.1247e-01, -4.2319e-02, -5.1464e-01,  2.6012e-01,\n",
       "          -4.5878e-01,  9.4800e-02,  3.9247e-01, -2.8573e-01,  6.3406e-01,\n",
       "           4.2629e-01, -3.6110e-01, -2.0545e-01, -2.2754e-01, -3.2389e-01,\n",
       "          -7.2641e-01, -3.1915e-01,  3.6843e-01, -1.3623e-01, -5.0094e-02,\n",
       "           3.0149e-01, -6.9470e-02, -2.1966e-01, -1.3374e-01,  3.5540e-01,\n",
       "           3.2266e-01,  5.0087e-01,  8.7539e-02,  2.1457e-01, -1.9546e-01,\n",
       "          -3.1801e-01, -4.3533e-02, -3.6899e-01, -4.3137e-02, -4.4314e-01,\n",
       "          -2.4621e-02, -2.5187e-01, -2.6282e-01, -4.7259e-01,  2.0462e-01,\n",
       "           6.6081e-01, -1.5446e-01, -3.9355e-01,  5.5619e-01,  7.1274e-02,\n",
       "           1.1669e-01, -2.6400e-01, -2.2193e-01, -4.8790e-02,  3.7464e-01,\n",
       "           3.2862e-01, -2.1371e-01, -6.1154e-01, -1.0732e-01, -3.8806e-01,\n",
       "          -1.4537e-01,  8.7285e-01, -3.0314e-01, -6.3566e-01,  3.3248e-01,\n",
       "           4.3234e-01, -1.6519e-01,  9.2976e-03,  2.1159e-01, -1.9991e-01,\n",
       "          -2.4712e-01,  1.7348e-01,  4.6567e-01, -2.5039e-01,  2.0177e-01,\n",
       "           2.4127e-01, -2.1334e-01,  8.7048e-02,  4.3581e-01, -2.3658e-01,\n",
       "           2.2431e-02, -4.3233e-03, -2.0896e-01, -2.1368e-01,  3.2164e-01,\n",
       "          -3.2487e-01,  2.5705e-02,  3.4141e-01, -3.6298e-01, -9.5012e-03,\n",
       "           6.2922e-01,  6.6706e-01, -1.6613e-01, -7.9492e-02,  4.8050e-01,\n",
       "           3.7800e-01,  3.5405e-01,  1.6053e-01,  1.3443e-01, -1.2153e-01,\n",
       "           3.2495e-01,  2.2780e-02, -3.5713e-01, -2.2073e-02,  3.2642e-01,\n",
       "           3.8563e-01, -5.0308e-01, -5.7262e-01,  1.6706e-01,  3.3994e-01,\n",
       "           4.5630e-02,  3.9967e-01, -5.3573e-02,  3.1322e-01,  4.6684e-01,\n",
       "          -1.0005e+00, -1.2907e-01,  7.4049e-02, -4.3060e-01, -9.6275e-02,\n",
       "          -7.0479e-01, -2.6385e-01, -2.5949e-01, -2.1462e-01,  4.4147e-01,\n",
       "           4.9308e-01, -9.1658e-01,  2.8788e-01,  6.1745e-01, -2.6333e-01,\n",
       "           5.6974e-01,  2.1035e-01,  1.7002e-01,  3.9422e-01, -1.8134e-01,\n",
       "          -1.8103e-01,  4.2924e-01, -3.1173e-01, -5.4250e-01,  7.0610e-02,\n",
       "           8.0354e-01,  1.5099e-01, -1.0007e-01,  1.2480e-01, -3.9904e-01,\n",
       "          -2.4378e-02, -3.1114e-01, -5.0025e-02,  4.0179e-02,  3.1669e-01,\n",
       "           1.3859e-01, -4.9635e-01, -2.0442e-01,  2.0001e-01, -9.1190e-02,\n",
       "           3.8837e-01,  5.8407e-01,  4.2776e-01,  1.3402e-01, -5.3121e-01,\n",
       "          -3.6254e-03,  7.0843e-01, -4.9830e-01,  7.3488e-01, -3.6373e-01,\n",
       "          -4.2233e-01,  4.5096e-02, -7.1095e-04, -1.7474e-01, -8.0840e-02,\n",
       "          -4.0276e-01,  4.6111e-01, -3.1691e-01, -3.4208e-02,  4.5268e-01,\n",
       "          -2.0881e-01, -9.4136e-02, -6.9643e-01,  1.1976e-02,  1.8568e-01,\n",
       "          -4.7976e-02,  7.7040e-01,  1.1633e-01,  3.1298e-01, -5.5760e-01,\n",
       "          -2.4051e-01, -6.1813e-01, -6.6689e-02,  2.5255e-01, -3.4532e-01,\n",
       "          -1.4907e-01, -1.9114e-01, -1.5805e-01,  9.0005e-02,  6.5291e-01,\n",
       "           8.6521e-01,  1.4978e-01, -4.6998e-01, -2.1504e-01,  3.5965e-01,\n",
       "           2.4172e-01,  3.3037e-01,  5.3420e-01, -1.0693e-01, -1.4779e-02,\n",
       "          -5.1098e-01, -2.0526e-01,  5.3655e-03,  2.5108e-01, -2.8082e-01,\n",
       "           4.9497e-01,  2.4166e-01, -2.0945e-02,  7.4267e-02,  2.5052e-01,\n",
       "           4.0047e-01,  8.0278e-01,  2.0147e-01, -4.7077e-01, -7.0836e-02,\n",
       "           2.8668e-01, -1.8076e-01, -3.7951e-02, -5.5466e-01,  4.5767e-01,\n",
       "           2.1555e-01,  4.9546e-01,  2.9799e-01]], device='mps:0'),\n",
       " tensor([[-9.3597e-02, -7.5983e-01, -5.5698e-01, -6.1297e-01, -4.6662e-01,\n",
       "          -2.7231e-01, -3.4142e-01,  3.1485e-01, -3.5417e-02,  5.0401e-01,\n",
       "           1.3099e-01,  2.6525e-01,  1.8735e-01, -2.1442e-01, -6.1470e-02,\n",
       "          -4.6995e-01,  1.7552e-01,  7.5934e-01,  3.3963e-01,  1.3952e-01,\n",
       "           6.4924e-01,  1.6016e-01,  1.2924e-01,  8.3972e-02, -8.3433e-01,\n",
       "           6.9432e-02, -1.6656e-01, -1.7115e-01,  1.5443e-01, -6.3208e-01,\n",
       "          -4.4632e-01,  2.0927e-01, -5.7606e-02,  1.0884e+00,  2.4827e-01,\n",
       "           5.1564e-02,  2.0354e-01, -2.5232e-01,  5.9929e-01,  4.3575e-02,\n",
       "           3.6600e-01, -2.9288e-01,  1.7371e-01,  1.6505e-01,  6.9309e-01,\n",
       "           1.0386e-01,  1.7947e-02, -4.9602e-01,  5.0459e-01, -2.8994e-01,\n",
       "           1.5237e-01,  7.3949e-02,  7.3697e-01,  4.6847e-01, -8.4447e-01,\n",
       "          -5.2782e-01,  3.5190e-01,  1.3362e-01, -5.9164e-01,  3.0314e-01,\n",
       "          -5.7999e-02,  7.0011e-01,  1.0683e-01, -6.9916e-02, -8.7986e-02,\n",
       "           2.0294e-01, -7.4331e-01, -7.9600e-01, -3.6518e-01, -2.5943e-01,\n",
       "          -8.6892e-02,  6.1974e-01, -2.2544e-01, -5.1863e-03, -3.0266e-02,\n",
       "          -7.5362e-02,  1.4054e-01, -4.3458e-02, -3.0610e-01, -2.9767e-02,\n",
       "           4.8715e-02,  4.7370e-02, -1.9225e-01, -1.2266e-01, -3.0489e-01,\n",
       "          -1.0192e-01,  5.5554e-01,  1.9764e-01,  3.7427e-01, -1.9865e-01,\n",
       "           7.0895e-01, -1.4617e-01, -1.5420e-01, -7.8688e-01,  2.8885e-01,\n",
       "          -1.9214e-01,  4.6453e-01, -3.4607e-01,  2.6179e-02, -1.9432e-01,\n",
       "           2.3194e-01,  1.7143e-01, -1.7954e-01,  2.9637e-01, -2.7705e-01,\n",
       "          -7.2748e-01, -2.8403e-04, -2.6417e-01, -5.6623e-01, -1.1346e-02,\n",
       "          -3.4650e-01, -3.8877e-01,  1.8319e-01, -4.8122e-01, -2.3497e-02,\n",
       "          -2.1353e-01, -4.6003e-02, -1.4494e-01, -4.1947e-01,  3.2399e-01,\n",
       "          -3.5408e-02, -3.7632e-01,  3.2656e-01,  3.1135e-01,  2.1448e-01,\n",
       "          -3.7562e-01,  4.6586e-01,  8.4879e-02,  1.7949e-01,  4.5167e-01,\n",
       "           3.1249e-02,  1.7941e-01, -1.1019e-01, -5.7802e-02, -2.7086e-01,\n",
       "          -1.5395e-01,  2.3654e-01, -3.7179e-01,  1.7203e-01, -1.3186e-01,\n",
       "           3.1420e-02,  9.1810e-02, -2.8656e-01, -1.9174e-01, -2.1169e-02,\n",
       "          -7.9043e-03,  1.7603e-01, -2.0972e-01, -3.3410e-01, -4.1969e-01,\n",
       "          -5.7823e-01, -5.5476e-01,  5.8192e-01, -3.4491e-02, -1.0956e-01,\n",
       "          -1.6419e-01,  6.5303e-01,  1.8472e-01, -2.0602e-01,  3.3415e-01,\n",
       "           6.7706e-01, -8.5874e-02,  2.5695e-01, -1.4574e-01, -2.2965e-01,\n",
       "           3.6555e-01, -3.4229e-01, -1.6325e-01, -4.9086e-01, -5.0465e-02,\n",
       "           4.5928e-01, -2.7112e-01,  2.6551e-01,  1.6495e-01,  1.0085e-01,\n",
       "          -3.6704e-01,  3.8210e-01,  4.0732e-01, -3.2077e-01, -1.4745e-01,\n",
       "          -3.8122e-01, -1.4696e+00,  3.3253e-02,  5.2824e-01, -1.6110e-01,\n",
       "          -6.2049e-01,  4.0829e-01,  5.0781e-01,  7.9148e-02, -1.4124e-01,\n",
       "           1.4493e-01, -7.9055e-02,  1.2379e-01,  2.9555e-01, -2.0766e-01,\n",
       "           4.4080e-01,  5.6117e-01,  2.3712e-01,  2.5024e-01, -2.6885e-02,\n",
       "          -4.7575e-01,  2.7925e-02,  1.4449e-01,  2.3830e-01,  1.0214e-01,\n",
       "           1.3973e-01, -1.6265e-02,  1.1292e-01,  4.4491e-01, -1.1834e-01,\n",
       "          -7.0917e-02, -2.0032e-01, -6.2531e-01, -8.0498e-02,  1.2015e-01,\n",
       "           7.0054e-01, -1.9387e-01, -2.0850e-01,  4.7122e-01,  1.2109e-01,\n",
       "          -4.0161e-01,  1.4024e-01, -1.9411e-01, -3.0374e-01, -2.4821e-01,\n",
       "           7.8137e-01, -2.3559e-01, -2.0329e-02,  2.7035e-02, -2.6760e-01,\n",
       "           4.0398e-01, -1.6142e-01,  5.9135e-01,  4.2020e-01, -3.1670e-01,\n",
       "          -1.6174e-01, -6.0103e-01, -8.6454e-02, -1.0472e-01,  8.8193e-02,\n",
       "          -4.8491e-01, -4.4630e-01, -5.6140e-01, -4.0957e-01,  2.3041e-01,\n",
       "           4.0387e-01, -1.5316e-01,  7.7280e-01,  4.2006e-01,  3.1546e-01,\n",
       "          -5.1706e-02, -3.8068e-02,  3.9765e-01, -6.1892e-02, -5.1506e-01,\n",
       "           3.1321e-01,  2.5350e-01, -5.8001e-01,  5.2014e-01, -3.6669e-01,\n",
       "          -2.3412e-01, -1.1919e-01,  2.2131e-01,  3.8168e-01, -2.7421e-01,\n",
       "           2.0880e-01, -5.2366e-02, -6.0127e-01, -3.1833e-01,  2.0320e-01,\n",
       "          -2.6223e-01,  5.4548e-01,  4.4564e-01, -4.1260e-01, -5.7563e-01,\n",
       "          -6.8329e-01, -1.3737e-01, -4.6682e-01, -7.5511e-02, -7.7820e-01,\n",
       "          -3.6736e-01,  9.5497e-02, -1.8770e-01,  6.1109e-02, -3.4672e-01,\n",
       "           1.9687e-01,  2.5179e-01,  4.2597e-02,  8.8744e-02, -4.0441e-01,\n",
       "          -4.2355e-01, -3.8965e-02, -1.6272e-01,  5.2542e-03, -1.8680e-01,\n",
       "           3.1795e-01, -1.5576e-01, -1.0721e-01,  6.5366e-02, -4.3936e-01,\n",
       "           9.1155e-02, -3.7708e-02,  2.5777e-01, -2.6785e-01, -2.2282e-01,\n",
       "           3.3625e-02,  1.1606e-01,  4.0706e-01,  1.1018e-02, -6.1862e-01,\n",
       "          -1.5919e-01,  6.2148e-01,  9.6874e-02, -2.9395e-01, -3.8271e-02,\n",
       "          -1.0804e+00, -6.6971e-02,  1.3629e-02, -6.2282e-03, -3.1650e-01,\n",
       "           1.7415e-01,  1.9339e-01, -2.0113e-01,  3.3341e-01, -4.6807e-01,\n",
       "           5.7890e-01, -1.0380e-01,  1.2937e-01,  7.4676e-02, -5.1971e-01,\n",
       "           9.8827e-03, -8.0048e-01, -1.9369e-01, -1.8455e-01,  1.3700e-02,\n",
       "          -1.7151e-02,  8.1660e-02,  1.0934e-01,  4.0339e-01, -2.8279e-01,\n",
       "           4.1096e-01, -4.1676e-01, -6.9008e-03,  4.1939e-01, -3.5093e-01,\n",
       "           1.6325e-01,  4.1581e-01, -8.3627e-01, -1.2011e-01,  2.1667e-01,\n",
       "           1.7826e-01, -1.4035e-01, -1.3732e-01, -1.5056e-01,  6.1045e-01,\n",
       "           4.3765e-01,  2.5115e-01, -3.3101e-01,  4.4451e-01,  1.1785e-01,\n",
       "          -3.0285e-01,  3.2584e-01, -2.1722e-01, -3.0494e-01, -3.5750e-01,\n",
       "           2.0461e-01,  1.8817e-01, -3.9370e-01, -3.0719e-01, -3.1983e-01,\n",
       "           2.0295e-01, -4.0609e-01, -1.1631e-01,  1.8603e-01,  4.4755e-01,\n",
       "          -2.5024e-01, -3.6661e-01,  1.4363e-01, -3.0050e-01, -3.0193e-01,\n",
       "          -6.6494e-02, -9.5558e-02,  1.0112e-01, -2.1945e-01, -1.6687e-01,\n",
       "          -1.9864e-01, -2.5196e-01,  6.8554e-01,  1.7021e-01, -4.8878e-02,\n",
       "           8.3042e-02,  1.2628e-01, -4.7308e-01,  5.9135e-01,  1.7433e-01,\n",
       "          -1.4584e-01,  4.0896e-01, -2.0128e-01, -3.1161e-02,  3.2916e-01,\n",
       "          -1.8061e-01,  1.2818e-01,  3.7661e-02,  2.9169e-01,  1.0268e-01,\n",
       "          -3.8614e-01, -5.1338e-03,  2.2410e-01, -8.9810e-03, -4.2967e-02,\n",
       "          -1.0091e-02,  7.8181e-01, -4.1975e-01,  5.2181e-01,  1.8406e-01,\n",
       "           1.8304e-01, -8.5270e-03,  1.0952e-01, -5.8070e-02, -5.9997e-01,\n",
       "           6.8344e-01,  4.3650e-01,  5.9719e-01,  5.8294e-01, -1.1421e+01,\n",
       "          -2.0900e-01,  5.7592e-01, -5.9389e-01, -9.3166e-02,  9.0981e-03,\n",
       "          -8.3851e-02, -3.6349e-01,  1.2614e-01, -5.2635e-01, -1.8346e-01,\n",
       "           4.9221e-02, -4.8337e-02,  2.2034e-01,  5.4671e-02, -1.8041e-01,\n",
       "          -2.3008e-01, -5.2879e-01,  2.0529e-01, -1.9773e-01,  4.8123e-01,\n",
       "          -3.6504e-01,  9.4994e-02,  3.6662e-01, -2.9320e-02, -1.9452e-01,\n",
       "          -2.3291e-01, -3.4894e-02,  5.8308e-01,  2.3971e-01, -2.4359e-01,\n",
       "           7.2476e-02,  1.5783e-01,  4.6654e-01,  3.7865e-02, -5.1482e-01,\n",
       "           5.1384e-02, -4.0425e-01,  1.1864e-01, -4.5566e-01, -1.9738e-01,\n",
       "          -1.1095e-01, -1.2582e-01,  3.2041e-01,  3.8734e-01,  7.0563e-01,\n",
       "           4.4580e-01,  6.7177e-01,  3.0459e-01, -4.1876e-01,  1.0243e-01,\n",
       "          -8.4663e-01,  2.8578e-01, -6.8960e-01, -5.2603e-01,  2.2449e-02,\n",
       "           5.8338e-01, -4.8274e-01,  1.6627e-01, -2.3953e-01, -4.2443e-01,\n",
       "           4.2245e-01,  5.1365e-02, -2.7767e-01, -4.5201e-01, -3.0597e-01,\n",
       "           4.6072e-01,  7.1043e-02,  6.2598e-02, -3.9400e-01, -2.3640e-02,\n",
       "          -1.4686e-01,  4.5383e-01,  4.1260e-01, -6.9185e-02, -5.3301e-03,\n",
       "          -5.9392e-02,  2.5969e-02,  1.6861e+00, -1.5753e-01,  1.5971e-01,\n",
       "           2.0630e-01,  4.1698e-01, -8.0137e-02, -9.5487e-01,  8.4358e-02,\n",
       "           1.5397e-01,  5.1517e-01,  1.3777e-01,  4.1932e-01, -1.2751e-01,\n",
       "          -6.5198e-01,  1.7260e-01,  2.0560e-01,  1.0725e-01,  5.0898e-01,\n",
       "          -7.0628e-01,  1.2042e-01,  3.5785e-01, -3.7031e-02,  6.2978e-02,\n",
       "           6.4718e-02, -3.9540e-01, -3.2088e-01,  3.0584e-01,  1.9250e-01,\n",
       "           2.3504e-01, -9.2298e-04,  3.7540e-01,  3.4406e-01, -1.1479e-01,\n",
       "           1.7374e-01,  3.7105e-02,  2.6185e-02, -1.6075e-01, -1.2843e-01,\n",
       "          -3.6167e-03,  5.3672e-01, -1.9667e-01, -2.3686e-01, -2.7043e-01,\n",
       "          -1.9796e-01, -7.2273e-02,  3.9042e-01, -5.2087e-01,  6.8360e-01,\n",
       "          -4.7674e-01, -5.9429e-02, -3.9537e-02, -1.0713e-01,  3.3279e-01,\n",
       "           3.3193e-01,  4.2532e-01,  1.3776e-01, -5.0578e-01, -6.5116e-01,\n",
       "          -6.6603e-01,  2.4701e-01,  1.4962e-01,  8.3788e-02,  1.9820e-01,\n",
       "          -2.9411e-01,  1.3010e-02, -2.9285e-01,  1.9266e-01,  2.3436e-01,\n",
       "           3.6344e-01,  3.7243e-01, -5.3114e-02,  2.2692e-01, -2.4446e-01,\n",
       "          -4.9383e-01, -5.8904e-02, -2.4828e-01,  3.1423e-01, -5.8450e-01,\n",
       "          -1.5967e-01, -1.5713e-01, -2.6303e-01, -3.2462e-01, -7.0354e-02,\n",
       "          -1.3516e-01,  1.0222e-01,  1.1857e-01,  1.8078e-01,  1.8026e-01,\n",
       "           2.0332e-01, -2.2026e-01,  1.0099e-01, -2.3403e-01,  3.0323e-01,\n",
       "          -3.6483e-03, -4.2025e-02, -2.6213e-01, -2.8150e-01, -1.8752e-01,\n",
       "           1.8846e-01,  5.4278e-01, -1.6907e-01, -6.8827e-01,  3.6096e-01,\n",
       "           4.0697e-01, -1.3475e-01, -3.8549e-01, -9.7766e-02, -3.5457e-01,\n",
       "          -2.6875e-01, -4.8925e-01,  3.7058e-01,  1.3636e-01,  5.3200e-01,\n",
       "           2.9279e-01, -2.5359e-01, -4.6617e-01,  6.7052e-01,  1.0603e-01,\n",
       "          -4.0416e-01, -2.5572e-01, -5.0023e-01, -1.6621e-01,  2.6028e-01,\n",
       "           1.2584e-01, -2.7521e-01, -5.2184e-02, -1.9500e-01,  1.2183e-02,\n",
       "           4.0016e-01,  7.3369e-01,  4.0673e-03, -1.7632e-01,  8.4599e-01,\n",
       "           3.5400e-01,  1.9013e-02,  3.0118e-01,  3.2074e-01,  2.9208e-01,\n",
       "           4.6073e-01, -1.2851e-01, -4.5651e-01,  8.4055e-02,  7.0494e-02,\n",
       "           5.0691e-01, -7.5998e-01, -2.0263e-01, -1.2255e-01,  1.8136e-01,\n",
       "           8.8292e-02,  3.5883e-01, -4.0135e-01,  6.4197e-03,  2.8978e-01,\n",
       "          -1.1751e+00, -3.7752e-02, -3.8941e-01,  3.0874e-02,  7.6326e-02,\n",
       "          -2.2482e-01, -1.3984e-01, -7.8640e-01, -8.0862e-02,  4.4159e-01,\n",
       "           1.5406e-01, -6.3360e-01,  1.7597e-01,  7.3459e-01, -4.5057e-01,\n",
       "           4.4052e-02, -2.8862e-01,  1.1468e-01,  1.8486e-01,  2.2464e-01,\n",
       "           2.9377e-02,  2.9499e-01,  2.3095e-01, -5.0464e-01,  2.6942e-01,\n",
       "           6.8098e-01, -1.0646e-02,  3.2904e-02,  1.5718e-01,  2.5972e-01,\n",
       "          -2.8712e-01, -2.8700e-01, -2.0037e-01, -8.8218e-03, -2.8909e-01,\n",
       "           1.6438e-01, -1.7414e-01, -7.9182e-02,  1.9763e-01, -2.0708e-01,\n",
       "           3.9544e-01,  5.6423e-01,  8.8157e-01,  5.0242e-03, -5.1714e-01,\n",
       "           1.5460e-01,  1.4937e-01,  8.3920e-03,  3.0561e-01, -1.5197e-01,\n",
       "          -2.6716e-01,  1.4870e-01,  1.2910e-01, -1.7071e-01, -1.8418e-02,\n",
       "          -5.3501e-03,  3.3123e-02, -1.3857e-01, -3.8094e-01,  3.8825e-01,\n",
       "           7.7403e-04,  6.7490e-02, -5.1346e-01, -9.5019e-02,  4.7311e-01,\n",
       "          -2.8433e-01,  6.3161e-01, -2.0024e-01,  3.7446e-01, -3.0193e-01,\n",
       "           2.6294e-02, -1.0022e+00, -2.7971e-01,  2.5827e-01, -3.0287e-01,\n",
       "          -4.6068e-01,  1.0824e-02,  1.2644e-01,  8.8389e-02,  5.5480e-01,\n",
       "           3.0829e-01,  3.2608e-02, -8.6013e-01, -3.9834e-01,  1.9373e-01,\n",
       "          -6.8738e-02, -1.0422e-01,  3.5955e-01,  2.2445e-01,  2.9625e-01,\n",
       "          -8.2328e-01, -5.2759e-01, -1.3057e-01, -4.0218e-04, -3.0653e-01,\n",
       "           9.2733e-01,  2.4616e-02, -1.6542e-01,  2.4822e-01,  2.5689e-01,\n",
       "           3.2649e-01,  7.9899e-01, -1.2685e-01, -3.3525e-02, -1.9437e-01,\n",
       "           2.2384e-01, -6.4748e-02, -2.9126e-02, -8.1102e-01,  6.0876e-01,\n",
       "           1.2083e-01,  2.7683e-01,  4.5590e-01]], device='mps:0'),\n",
       " tensor([[-7.2281e-02, -6.9185e-01, -3.4258e-01, -6.6365e-01, -5.4581e-01,\n",
       "          -2.9957e-01, -3.5949e-01,  3.1618e-01, -1.3398e-01,  5.4146e-01,\n",
       "           1.4823e-01,  4.5465e-01,  1.7889e-02, -5.0086e-01, -4.6828e-01,\n",
       "          -4.8138e-01, -1.1317e-01,  5.0224e-01,  6.0980e-01,  3.1430e-01,\n",
       "           1.8159e-01,  2.1725e-01, -1.5609e-01,  1.5228e-01, -5.5009e-01,\n",
       "           3.4362e-02, -1.2777e-01, -7.0927e-03,  2.7773e-01, -1.3275e-01,\n",
       "          -3.7560e-01,  4.8500e-01, -7.5535e-02,  1.1362e+00,  5.6913e-02,\n",
       "          -1.0761e-01,  2.3571e-01, -2.0861e-01,  7.6994e-01, -3.3664e-01,\n",
       "           6.4467e-01, -4.4352e-02, -2.5230e-02,  7.8207e-02,  5.1560e-01,\n",
       "           1.6018e-01,  2.5531e-01, -6.0353e-01, -3.3082e-02, -6.3613e-01,\n",
       "          -7.3729e-02, -1.4125e-01,  5.8847e-01,  2.6750e-01, -5.9945e-01,\n",
       "          -2.2787e-01,  1.4229e-01,  2.7933e-01, -4.3886e-01,  1.1570e-01,\n",
       "          -7.2328e-02,  5.9530e-01, -6.0887e-02,  3.4634e-02,  1.5063e-01,\n",
       "           3.3015e-01, -4.5835e-01, -3.6721e-01, -5.1874e-01,  1.7533e-01,\n",
       "           8.9242e-02,  3.9951e-01, -7.1672e-02, -3.2469e-01, -1.8945e-02,\n",
       "           4.9643e-02,  1.9437e-01, -1.2438e-02, -4.7411e-01, -1.7206e-01,\n",
       "          -1.2158e-01,  3.8255e-01, -6.8788e-02, -9.2526e-04, -1.0220e-01,\n",
       "          -2.3351e-02,  4.5734e-01,  1.1756e-01, -4.3198e-02, -5.0371e-02,\n",
       "           2.4945e-01, -1.9103e-01, -6.3509e-01, -3.6354e-01,  6.7100e-02,\n",
       "          -1.2449e-01,  3.4944e-01, -3.9771e-01, -3.4730e-02, -2.1893e-01,\n",
       "           2.5272e-01,  7.2620e-02, -5.8370e-02,  2.8303e-01, -5.9521e-01,\n",
       "          -6.6510e-01,  1.4882e-01,  8.9286e-02, -2.4349e-01, -1.6036e-01,\n",
       "          -2.8482e-01, -1.9338e-01,  9.5176e-02, -4.9022e-01, -3.6298e-02,\n",
       "          -1.0709e-01,  1.0010e-01, -1.3913e-01, -1.8419e-01,  2.3759e-01,\n",
       "           9.7741e-02, -6.1953e-01,  2.2338e-01,  3.1028e-01,  2.4990e-03,\n",
       "          -1.2854e-01,  6.4286e-01, -1.2584e-02,  4.1519e-01,  2.6425e-01,\n",
       "           2.0656e-02,  2.0370e-01, -5.2542e-01, -1.0000e-01, -2.2381e-01,\n",
       "           1.9307e-01, -1.4715e-01, -3.1675e-01,  5.8968e-01, -1.4393e-01,\n",
       "           6.2090e-02,  5.4673e-02, -3.7938e-01, -5.6488e-01, -2.5118e-01,\n",
       "          -3.5816e-01,  3.0669e-01, -4.5867e-01, -4.1208e-01, -3.1451e-01,\n",
       "          -7.2210e-01, -7.3879e-01,  2.6551e-01, -1.5251e-01,  2.8810e-01,\n",
       "          -2.4155e-01,  4.4817e-01, -4.7467e-02, -3.3717e-01,  5.0049e-01,\n",
       "           5.6891e-01, -7.3731e-02,  3.0168e-01, -4.4340e-01, -1.1017e-01,\n",
       "           6.6977e-02, -2.7160e-01, -1.4082e-01, -1.5725e-01, -3.1042e-02,\n",
       "           4.2308e-01, -7.1408e-02,  2.0555e-01, -2.1342e-01,  2.1569e-01,\n",
       "          -4.7944e-01, -9.4788e-03,  4.2678e-02, -2.0745e-01, -3.1569e-01,\n",
       "           9.4412e-02, -1.1082e+00, -2.4643e-01,  4.3187e-01, -3.8425e-01,\n",
       "          -4.8048e-01,  2.3294e-01,  2.7108e-01,  1.9753e-01, -2.5306e-01,\n",
       "           4.6446e-02,  7.1553e-02,  1.7226e-01, -4.2147e-02,  1.0841e-01,\n",
       "           3.7498e-01,  5.9785e-01,  3.1734e-01,  3.2101e-01, -1.4534e-01,\n",
       "          -8.2788e-01,  2.9911e-02,  1.0337e-01,  1.5490e-01,  2.9849e-01,\n",
       "           7.1560e-02, -4.1603e-01,  2.2774e-01,  3.5943e-02, -1.9868e-01,\n",
       "           4.8173e-02, -2.1426e-01, -4.8691e-01, -2.0303e-01, -2.4576e-01,\n",
       "           4.8171e-01, -2.7265e-01, -1.2361e-01,  6.5467e-01,  6.1939e-03,\n",
       "           6.3541e-02,  4.0877e-01,  8.4793e-02, -3.6411e-01,  7.2143e-02,\n",
       "           4.7340e-01, -2.1637e-02,  1.3643e-01, -3.1372e-01, -2.9558e-01,\n",
       "          -8.5747e-02, -8.5290e-02,  3.6826e-01,  8.2005e-01, -3.3191e-01,\n",
       "           3.5512e-02, -2.6988e-01, -1.0295e-02, -1.9440e-01,  2.1288e-01,\n",
       "          -5.1439e-01, -1.6321e-01, -3.4201e-01, -8.0423e-02,  6.0386e-02,\n",
       "           3.2998e-01, -1.8332e-01,  8.3076e-01,  4.5491e-01,  1.9047e-01,\n",
       "           1.6944e-01, -1.5894e-03,  5.6516e-02,  2.4126e-01, -4.7432e-01,\n",
       "           3.2262e-01, -2.8505e-02, -5.2562e-01,  4.2455e-01, -3.2002e-01,\n",
       "           8.3011e-02,  1.2041e-02, -7.8104e-02,  1.4548e-01, -5.3577e-02,\n",
       "          -1.1621e-01,  4.2937e-01,  9.5680e-02, -4.5882e-01,  4.0013e-01,\n",
       "          -3.9800e-01,  6.5099e-01,  1.0680e-01, -3.0722e-01, -3.5025e-01,\n",
       "          -5.9310e-01, -1.3453e-01,  2.7389e-02,  1.9121e-01, -7.2501e-01,\n",
       "          -4.5692e-01,  1.8554e-01, -1.3024e-01,  1.1806e-01, -6.4532e-02,\n",
       "           3.6955e-01,  2.2539e-01,  4.0137e-03, -3.5288e-01, -4.7116e-01,\n",
       "          -1.4978e-01,  3.7069e-02, -1.7111e-01, -6.2556e-02, -7.0342e-03,\n",
       "           4.1910e-01, -4.3977e-02,  5.0995e-03,  1.8554e-01, -4.2366e-01,\n",
       "           2.5740e-01,  5.4041e-02,  2.0707e-01, -3.5506e-01, -7.5737e-02,\n",
       "           4.0751e-02,  2.7297e-01,  5.8966e-01, -2.8456e-01, -4.6985e-01,\n",
       "          -1.6502e-01,  4.8983e-01, -1.1739e-01, -1.4517e-01, -9.7460e-02,\n",
       "          -1.0150e+00,  1.5726e-01, -1.8178e-01, -1.0723e-01, -3.8570e-01,\n",
       "           1.9995e-01,  2.9396e-01, -1.3408e-01,  7.8658e-02, -2.3332e-01,\n",
       "           5.0993e-01,  2.7779e-01, -5.0237e-02, -3.0032e-01, -5.1631e-01,\n",
       "           1.4851e-01, -8.4764e-01, -4.9120e-03, -3.9366e-01, -2.1585e-01,\n",
       "           4.0157e-02,  2.6847e-01,  2.7191e-03,  2.0411e-02, -3.8873e-01,\n",
       "           4.4270e-01, -3.3690e-01,  2.2332e-01,  3.7323e-01, -3.7687e-01,\n",
       "          -9.5667e-03,  4.1144e-01, -5.7726e-01,  8.2607e-02,  3.4840e-01,\n",
       "           2.8551e-01, -5.9086e-02, -1.9446e-01, -1.5625e-01,  6.1673e-01,\n",
       "           7.0465e-01,  2.4878e-01, -5.6078e-01,  3.6748e-01,  1.1692e-01,\n",
       "          -2.8998e-01,  4.5885e-01, -3.4448e-01, -3.4872e-01, -1.0854e-01,\n",
       "          -6.7318e-02,  3.9443e-01, -2.2911e-01,  6.8505e-02,  2.0781e-01,\n",
       "           2.5453e-01, -3.9350e-01, -1.7844e-01,  4.2788e-01,  3.6663e-01,\n",
       "           1.9232e-02, -3.3072e-01,  2.1522e-01, -3.1081e-01,  3.6235e-02,\n",
       "          -1.9556e-01, -2.5768e-01,  4.5876e-01, -2.8025e-02, -2.7881e-01,\n",
       "          -1.3176e-01, -3.7814e-01,  4.7108e-01, -1.4123e-01, -7.9774e-02,\n",
       "           3.1665e-01, -7.2995e-03, -2.2364e-01,  9.6249e-01, -4.7976e-02,\n",
       "          -7.5719e-02,  5.8226e-01, -2.9790e-01, -1.7097e-01,  5.1247e-01,\n",
       "          -1.4655e-01, -1.7604e-02,  9.6143e-02,  6.6502e-01,  1.9150e-01,\n",
       "          -3.0932e-01,  5.1563e-02,  3.6619e-01, -1.3722e-01,  1.1420e-01,\n",
       "          -3.6637e-01,  5.5122e-01, -1.7589e-01,  5.7103e-02, -6.6622e-01,\n",
       "           3.4620e-01, -2.1315e-01,  1.1591e-03, -3.6949e-01, -3.9610e-01,\n",
       "           3.0544e-01,  3.2008e-01,  4.3200e-01,  6.8009e-01, -1.1549e+01,\n",
       "           3.3955e-01,  5.9483e-01, -3.9941e-01,  1.4958e-01,  3.9016e-01,\n",
       "           1.5672e-01, -3.4863e-01,  3.9525e-01, -3.7131e-01,  1.5078e-02,\n",
       "          -1.6361e-01, -2.8015e-02,  9.5829e-04, -1.4131e-01,  1.9519e-01,\n",
       "          -9.2042e-02, -4.7943e-01,  1.7870e-01, -1.7089e-01,  3.4778e-01,\n",
       "          -3.1615e-01,  1.0651e-01,  4.8648e-01, -1.6307e-01, -4.5827e-01,\n",
       "          -1.4875e-01, -1.1987e-01,  6.5764e-01,  2.4417e-01, -1.5909e-01,\n",
       "           3.6878e-02,  1.7233e-01,  3.3901e-01, -1.7929e-01, -5.0929e-01,\n",
       "           2.8215e-01, -4.9047e-01, -4.2486e-02, -8.6983e-01, -1.9595e-01,\n",
       "           1.0394e-02, -3.9118e-01,  2.7440e-01,  5.0539e-01,  7.7292e-01,\n",
       "           5.1752e-01,  9.1307e-01,  3.1279e-01, -1.2214e-01, -9.0046e-02,\n",
       "          -6.6627e-01,  1.9552e-01, -5.4846e-01, -7.6458e-01, -3.3742e-02,\n",
       "           5.6542e-01, -4.4924e-01,  4.3954e-01, -5.0893e-01, -2.7452e-01,\n",
       "           1.8719e-01, -1.5364e-02, -5.5913e-02, -6.4497e-01, -4.6050e-01,\n",
       "           1.7425e-01,  3.1120e-02,  3.5228e-02, -5.4707e-01,  1.0436e-01,\n",
       "           2.2434e-01,  5.8871e-01, -1.0984e-01, -5.4829e-01,  1.1932e-01,\n",
       "           7.8217e-02, -6.5347e-02,  2.2664e+00, -1.7103e-01,  9.2378e-02,\n",
       "           6.0007e-02,  3.1923e-01, -5.3674e-01, -5.9841e-01, -4.1218e-01,\n",
       "          -1.9955e-01,  4.9157e-01,  3.9181e-01,  3.3989e-01, -1.0601e-01,\n",
       "          -6.5948e-01,  4.5256e-02,  1.8310e-01,  4.5858e-01,  7.8016e-01,\n",
       "          -4.0351e-01,  3.2070e-03,  9.7878e-02, -9.8831e-02, -1.5589e-01,\n",
       "          -1.9266e-01, -2.6854e-01, -1.1365e-01,  7.1787e-02,  3.8020e-01,\n",
       "           2.3545e-01,  2.1094e-01,  1.0226e-01,  1.3365e-01, -3.0364e-01,\n",
       "          -1.9603e-01, -2.8747e-01, -1.1181e-01, -4.2839e-01, -3.2432e-01,\n",
       "           1.4049e-02,  2.3407e-01, -2.0449e-01, -4.6587e-01, -2.8722e-01,\n",
       "          -3.0625e-02,  7.2199e-02,  1.5879e-01, -1.6850e-01,  6.6371e-01,\n",
       "          -5.0832e-01, -1.9388e-01, -1.9577e-01, -1.7664e-01,  3.5256e-01,\n",
       "           5.6302e-01, -1.5916e-01, -2.1202e-01, -3.0961e-01, -4.9411e-01,\n",
       "          -5.5858e-01, -1.6044e-02,  3.3507e-01,  7.3736e-03,  6.2692e-02,\n",
       "          -1.5843e-01,  2.8507e-02, -4.8350e-01,  3.0446e-01,  3.3065e-04,\n",
       "           3.4665e-01,  3.3237e-01, -1.9577e-01,  3.6725e-01, -7.0677e-02,\n",
       "          -1.8469e-01,  1.8630e-01, -2.4029e-01, -1.5606e-02, -6.4073e-01,\n",
       "          -4.9464e-01, -2.0679e-01, -1.2671e-02,  1.3205e-01,  5.8994e-02,\n",
       "          -8.8187e-02,  1.1721e-01, -8.7948e-02,  8.2735e-01,  4.2399e-01,\n",
       "           2.8646e-01, -1.8365e-01, -4.0143e-01, -5.4097e-02,  2.2815e-01,\n",
       "           2.1257e-01,  5.7036e-02, -1.0395e-01, -2.1743e-01, -7.0277e-02,\n",
       "          -1.2339e-02,  1.0442e+00, -7.7569e-02, -3.5018e-01,  1.5343e-01,\n",
       "           9.1526e-01, -2.5940e-01, -1.7950e-01,  3.2151e-01, -2.5541e-01,\n",
       "          -5.1138e-01, -4.6133e-01,  5.2929e-01, -2.3651e-01,  3.7757e-01,\n",
       "           4.6583e-01, -2.8303e-01, -2.6763e-01,  4.1238e-01,  1.7368e-01,\n",
       "          -4.8959e-01,  1.1017e-01, -2.4596e-01, -1.3131e-01,  5.3706e-01,\n",
       "           3.5547e-03, -1.5342e-01, -1.0340e-01, -2.5336e-01, -1.1470e-01,\n",
       "           4.3776e-01,  7.0975e-01, -2.1507e-01, -4.3984e-01,  6.4477e-01,\n",
       "           5.7750e-04,  2.1932e-01,  4.4324e-01,  1.3474e-01,  2.1500e-01,\n",
       "           6.3178e-01,  1.3649e-01, -5.5685e-01, -1.1680e-01,  2.0233e-01,\n",
       "           1.4124e-01, -6.5828e-01, -1.7324e-01, -2.5129e-01,  3.5229e-02,\n",
       "           1.2001e-02,  4.6418e-01,  1.4807e-02,  2.5932e-01,  3.8887e-01,\n",
       "          -1.0296e+00,  7.8896e-02, -7.9192e-03,  3.4998e-01,  2.4655e-02,\n",
       "          -2.0271e-01, -2.3621e-01, -5.7472e-01, -2.0474e-01,  3.8118e-01,\n",
       "           2.5816e-01, -9.3037e-01, -9.4606e-02,  5.7856e-01, -9.7991e-02,\n",
       "           2.6405e-01, -2.5393e-01,  1.8824e-01,  4.6050e-01,  9.7241e-03,\n",
       "          -1.2412e-02,  1.4051e-01,  1.1043e-01, -4.3793e-01,  4.5555e-01,\n",
       "           5.7030e-01,  4.2216e-02,  1.2997e-03,  4.5222e-02, -1.6374e-01,\n",
       "          -2.6979e-01, -3.6939e-01, -8.7372e-02, -3.8896e-01, -2.0323e-01,\n",
       "          -3.1087e-02, -3.5730e-01, -2.1444e-03,  1.9043e-01, -1.8307e-01,\n",
       "           4.1174e-01,  5.0495e-01,  8.0158e-01, -8.7977e-03, -2.2749e-01,\n",
       "          -3.9679e-02,  4.0529e-01, -1.3290e-01,  5.6575e-01,  1.3134e-01,\n",
       "          -3.4543e-01,  1.9743e-01,  1.2002e-01, -3.4943e-01, -7.3915e-02,\n",
       "          -1.0198e-01,  2.2211e-01, -3.0122e-01, -3.0077e-01,  6.5550e-02,\n",
       "          -8.1374e-02, -1.0427e-01, -5.0241e-01,  1.2193e-01,  4.2772e-01,\n",
       "           3.8353e-02,  6.3113e-01, -9.8871e-02,  3.5137e-01, -1.1593e-01,\n",
       "          -7.2015e-03, -9.6583e-01,  1.0177e-02,  2.9836e-01, -2.1274e-01,\n",
       "          -5.9995e-01, -1.5584e-01, -1.3440e-01, -3.3661e-01,  6.1933e-01,\n",
       "           6.2129e-01, -3.5296e-01, -8.2083e-01, -2.3006e-01,  7.0003e-02,\n",
       "          -5.7159e-02, -3.1712e-02,  1.9876e-01, -3.0568e-03,  7.5187e-02,\n",
       "          -9.0143e-01, -3.2166e-02, -2.7428e-02,  4.1436e-01, -3.1593e-01,\n",
       "           6.7230e-01, -1.3408e-01,  1.4795e-02,  6.6765e-01,  3.5368e-01,\n",
       "          -3.2434e-03,  6.7825e-01, -1.1593e-01, -3.7210e-01, -9.9699e-02,\n",
       "           7.6772e-02, -7.5527e-02,  9.0247e-02, -6.4630e-01,  3.6341e-01,\n",
       "           2.4509e-01,  2.8732e-03,  5.7475e-01]], device='mps:0'),\n",
       " tensor([[-0.1272, -0.7714, -0.4614,  ...,  0.1841,  0.6774,  0.6739],\n",
       "         [-0.0583, -0.6757, -0.8037,  ...,  0.1694,  0.7852,  0.5016]],\n",
       "        device='mps:0'),\n",
       " tensor([[-2.5187e-01, -4.0054e-01, -5.8007e-01, -8.3055e-01, -1.2868e-01,\n",
       "          -2.8804e-02, -7.1917e-01,  6.6701e-01,  1.2709e-01,  5.0855e-01,\n",
       "           1.2570e-01,  5.1916e-01,  5.1676e-02, -1.1521e-01, -1.6481e-01,\n",
       "          -4.9820e-01,  5.6222e-02,  3.7338e-01,  6.7187e-01, -2.4111e-02,\n",
       "           2.2844e-01,  5.7979e-01, -2.6323e-01, -1.2386e-02, -8.3013e-01,\n",
       "           2.9613e-01, -1.1576e-01,  2.2409e-01,  1.7718e-01, -6.3092e-01,\n",
       "           4.0999e-02,  4.7417e-01,  2.3068e-02,  1.3497e+00, -5.6080e-01,\n",
       "          -1.3230e-01,  8.2213e-02, -3.9453e-01, -2.3010e-01,  3.7283e-02,\n",
       "           6.1164e-01, -4.1315e-01,  1.3635e-01,  4.0254e-01, -6.1085e-02,\n",
       "           8.9076e-02,  1.3899e-01, -3.7744e-01,  2.5892e-01, -8.3961e-01,\n",
       "          -4.8318e-02, -2.0669e-01,  4.9144e-01,  6.2615e-01, -6.3176e-01,\n",
       "          -2.2975e-01,  2.7419e-01, -3.7716e-02, -3.3020e-02,  1.5349e-01,\n",
       "           9.7980e-02,  3.8582e-01, -2.9995e-01,  2.3257e-01,  3.3441e-01,\n",
       "           2.4458e-01, -3.0392e-01, -4.8889e-01, -5.4078e-01, -5.0111e-02,\n",
       "           9.1859e-03,  2.2490e-02,  1.4846e-01, -2.3272e-01,  4.1973e-01,\n",
       "          -7.1214e-01, -6.4354e-02, -8.9125e-02, -2.5973e-01,  2.9246e-01,\n",
       "          -4.5549e-02,  1.9719e-01,  3.1010e-01,  1.0268e-01,  1.2457e-01,\n",
       "           4.3419e-02,  2.7215e-01,  5.2932e-04,  4.7373e-02,  3.7533e-01,\n",
       "          -1.3375e-01, -8.1722e-02, -2.8313e-01, -3.5814e-01,  3.1229e-01,\n",
       "           4.0598e-02,  2.9907e-01, -1.9557e-01,  1.8579e-01, -3.2474e-01,\n",
       "           1.5588e-01,  1.7768e-01, -1.8083e-01,  5.7286e-02, -3.5172e-01,\n",
       "          -4.5475e-01,  2.3589e-02, -2.8569e-01, -4.8285e-01,  2.2697e-01,\n",
       "          -1.4211e-01, -3.0205e-01, -4.0922e-02, -3.8935e-01, -3.1524e-02,\n",
       "          -7.5164e-02,  2.6013e-01, -1.3230e-01, -3.4806e-01, -8.3033e-02,\n",
       "           2.0734e-01, -2.2808e-01,  1.6077e-01,  2.6882e-01,  1.2232e-01,\n",
       "          -9.6705e-02,  6.5251e-01,  1.8950e-02,  3.1674e-01,  2.5763e-01,\n",
       "           1.1699e-01, -6.5768e-02, -1.9745e-01, -2.3154e-01,  7.5988e-02,\n",
       "          -7.6941e-02,  1.0687e-01, -1.8182e-01, -2.2790e-01, -1.8508e-01,\n",
       "           3.1771e-01, -3.3097e-01, -6.0893e-01, -4.9260e-01, -1.6098e-01,\n",
       "          -1.6326e-01, -2.4804e-02, -8.2473e-03, -3.3693e-01, -3.1956e-01,\n",
       "          -6.7443e-01, -3.6717e-01,  8.1041e-01, -1.6831e-02, -3.3356e-01,\n",
       "          -1.6499e-01,  2.3208e-01,  3.1666e-01, -4.5959e-02,  3.7161e-01,\n",
       "           7.7250e-01, -4.7670e-02,  1.6786e-01, -5.1827e-01, -3.8195e-01,\n",
       "           4.8335e-02,  2.6658e-01, -2.7054e-01,  6.1879e-01,  4.8352e-02,\n",
       "           7.5627e-02, -6.6170e-02,  2.7430e-01, -3.2704e-01, -1.2324e-02,\n",
       "          -5.7573e-01,  3.7750e-01,  2.9539e-01, -3.9892e-01, -6.6567e-02,\n",
       "          -2.0834e-02, -1.2060e+00,  9.6964e-02, -8.3040e-01,  3.1236e-01,\n",
       "          -2.3110e-01,  1.7963e-01,  2.3389e-01,  2.9545e-01, -6.5320e-02,\n",
       "           2.5404e-01,  2.8725e-01,  2.5687e-01,  3.3609e-01, -3.4924e-02,\n",
       "           2.8019e-01,  2.3924e-01,  5.0790e-01,  4.1073e-01,  1.4769e-01,\n",
       "          -3.5440e-01,  2.9683e-01,  3.5781e-01,  9.0598e-02,  1.6954e-01,\n",
       "           9.9825e-02, -2.1636e-01,  2.1091e-01,  1.7501e-01, -1.6618e-01,\n",
       "          -2.2406e-01, -3.1100e-02, -4.6818e-01, -5.6655e-02, -4.8467e-01,\n",
       "           3.2557e-01, -5.1983e-01, -1.1912e-01,  3.4831e-01,  2.4340e-01,\n",
       "          -1.4896e-02,  1.8319e-01,  1.7196e-01, -2.4338e-01,  2.1735e-01,\n",
       "           3.7449e-01, -3.4618e-01, -3.9804e-02, -3.2197e-01, -2.9367e-01,\n",
       "           2.4357e-01, -5.6244e-01,  6.6635e-01,  4.0574e-01,  6.0165e-02,\n",
       "           1.5119e-01, -2.7920e-01, -5.1638e-02, -5.7408e-01,  3.5471e-01,\n",
       "          -3.2130e-01, -3.4955e-01, -5.4801e-01,  6.4487e-02,  4.2773e-01,\n",
       "           2.7565e-01, -6.4915e-02,  7.7868e-01,  4.2408e-01,  4.2813e-01,\n",
       "          -4.3003e-01, -1.5896e-02,  4.0377e-01, -4.1405e-02, -4.6059e-01,\n",
       "           6.7116e-02, -1.4962e-01, -5.1643e-01,  2.8000e-01, -2.6598e-01,\n",
       "          -1.2047e-01, -1.4913e-01,  3.3682e-01,  1.5365e-01, -2.1964e-01,\n",
       "          -3.8461e-01,  3.9587e-01, -1.4349e-01, -4.1723e-01,  4.7123e-01,\n",
       "          -3.5025e-01,  1.5209e-01,  1.0694e-01, -3.8441e-01, -6.2072e-01,\n",
       "          -3.2235e-01,  2.3090e-01,  1.5585e-01,  3.4799e-02, -6.6306e-01,\n",
       "          -4.0571e-01,  3.1197e-01,  2.4601e-01,  2.9976e-01,  2.9840e-01,\n",
       "           5.7427e-01, -3.0944e-01, -1.6702e-01,  1.4982e-01, -4.1377e-01,\n",
       "          -1.5704e-01, -3.3228e-02, -1.9995e-01,  2.9569e-01,  2.4768e-01,\n",
       "           1.1600e-01,  2.0367e-01,  4.3482e-01, -2.5272e-01, -3.8673e-01,\n",
       "          -7.1543e-02,  2.0161e-01, -9.3629e-02, -5.8306e-02,  2.3951e-01,\n",
       "          -6.3095e-02,  5.9650e-01,  7.2143e-01,  7.7439e-02, -6.4006e-01,\n",
       "           1.7696e-01,  5.2502e-01, -2.7253e-01, -4.0343e-02,  9.6365e-02,\n",
       "          -1.0344e+00, -2.1331e-01, -1.2883e-01, -2.0181e-01, -6.5352e-01,\n",
       "           8.8115e-02,  3.4863e-01,  4.6833e-03, -1.7864e-01, -3.0947e-01,\n",
       "           1.0448e-01,  4.8724e-02, -4.8980e-02,  4.3534e-01, -2.8901e-01,\n",
       "          -3.4797e-01, -2.3438e-01, -2.1532e-01, -5.4201e-01, -3.9968e-01,\n",
       "           3.5914e-01,  2.8129e-01,  1.1986e-01, -2.6486e-01, -5.1832e-01,\n",
       "           4.5577e-01, -6.8169e-01, -1.4632e-01,  2.4840e-01, -4.7702e-01,\n",
       "           1.4650e-02,  4.8990e-01,  1.4999e-01,  1.6336e-01,  6.5258e-02,\n",
       "           7.8557e-02, -3.3903e-01, -5.0030e-01, -1.5000e-01,  6.2323e-01,\n",
       "           3.2283e-01,  3.3506e-01, -1.7236e-01,  3.9096e-01,  4.5167e-01,\n",
       "          -2.9582e-01,  4.1670e-01,  1.0930e-02,  7.7439e-02, -6.0414e-02,\n",
       "           1.9048e-01,  3.4370e-01, -3.6842e-01,  2.0145e-01,  1.7929e-01,\n",
       "          -3.1634e-01, -3.2149e-01,  1.5691e-02,  2.0642e-01, -1.8580e-01,\n",
       "          -3.0875e-01, -2.0361e-01,  9.9698e-02, -3.4856e-01, -4.3104e-01,\n",
       "           2.9161e-01, -3.6543e-02,  7.0495e-02, -4.0357e-01, -4.3462e-01,\n",
       "          -2.0181e-01, -2.6055e-01,  6.8681e-01,  2.1064e-01, -1.9621e-01,\n",
       "           1.8720e-01, -1.0907e-01, -3.6330e-01,  4.7847e-01,  2.0162e-02,\n",
       "          -4.8754e-01,  4.5482e-01, -3.4822e-01,  1.0627e-01,  5.5756e-01,\n",
       "          -1.6769e-01, -1.0840e-01,  1.1008e-01,  1.0820e-01,  2.9026e-01,\n",
       "          -3.9884e-01,  1.5516e-01, -6.3476e-02, -3.1885e-01,  3.1820e-01,\n",
       "           2.6411e-01,  2.2781e-01, -1.7137e-01, -3.3426e-02,  3.5677e-02,\n",
       "           2.8782e-01, -6.0853e-02, -3.9888e-01, -4.7347e-01, -4.6132e-01,\n",
       "           1.2832e-01,  1.8927e-01,  5.6207e-01,  3.4144e-01, -1.1394e+01,\n",
       "           1.2938e-01,  7.8031e-01, -4.0713e-01, -8.2819e-02, -5.0320e-02,\n",
       "           1.0976e-02, -1.3708e-01,  4.2955e-01, -2.7734e-01, -4.4105e-01,\n",
       "          -3.5727e-01, -1.3308e-01,  1.5454e-01, -3.2189e-01, -2.6945e-01,\n",
       "          -1.4686e-02, -7.0935e-01,  4.0963e-01, -3.0286e-01,  5.0298e-01,\n",
       "          -2.3511e-01, -8.3798e-02,  1.5816e-01, -3.5851e-01, -1.3439e-02,\n",
       "           5.9766e-02, -1.6033e-01,  1.4651e-01,  4.2486e-01, -5.2072e-02,\n",
       "           3.3244e-02, -4.1258e-03,  4.8461e-02, -2.5723e-01, -5.9999e-01,\n",
       "           1.9296e-02, -1.4141e-01,  3.3422e-01, -4.4133e-02, -7.1265e-01,\n",
       "          -2.2533e-01, -2.7418e-01,  3.1407e-01,  2.7021e-01,  9.0598e-01,\n",
       "           2.2955e-01,  1.0686e+00,  1.5022e-01,  1.7748e-01,  2.6230e-02,\n",
       "          -7.8764e-01, -3.3269e-02, -4.1907e-01,  2.3280e-02,  1.0114e-01,\n",
       "           3.6576e-01, -7.9367e-01,  2.5251e-01, -3.5521e-01,  3.6487e-02,\n",
       "           4.9094e-01,  3.1768e-02,  2.7700e-01, -3.2298e-02, -7.8123e-02,\n",
       "           1.2089e-01, -2.8544e-02,  1.1098e-01, -2.2583e-01, -2.4084e-01,\n",
       "          -1.0177e-01,  5.9888e-01,  3.9592e-01, -7.2211e-01,  3.6112e-01,\n",
       "           4.9258e-02, -1.8622e-01,  1.4556e+00, -4.8573e-02,  1.8361e-02,\n",
       "           1.9800e-03,  2.9604e-01, -5.0236e-01, -7.3516e-01,  3.6511e-01,\n",
       "          -1.0211e-01,  5.6427e-02,  1.0127e-01,  4.8148e-01, -9.7481e-02,\n",
       "          -4.7940e-01, -1.8593e-03,  1.0556e-01, -1.3380e-01,  5.4710e-01,\n",
       "          -7.2824e-01, -1.6963e-01, -1.1223e-01,  3.1723e-01, -8.4054e-01,\n",
       "          -6.1126e-02, -7.0210e-01, -3.9328e-01,  5.0586e-01, -2.5896e-01,\n",
       "           1.5852e-01, -2.7443e-02, -4.3439e-02,  5.5574e-01, -2.1489e-01,\n",
       "           7.1555e-02,  2.2813e-01,  8.4995e-02, -1.6733e-01,  3.9349e-02,\n",
       "           4.1630e-01,  4.9571e-01, -4.2740e-01, -6.3911e-01, -7.6901e-03,\n",
       "           1.7180e-01,  1.4933e-01,  3.4019e-01, -5.3463e-01,  3.6431e-01,\n",
       "          -7.9162e-01, -1.8769e-01,  7.2949e-02, -4.3273e-02,  3.6175e-01,\n",
       "           2.3925e-01,  7.2038e-02,  2.2256e-01, -7.7756e-01, -6.4498e-01,\n",
       "          -6.5932e-01, -2.1445e-01,  2.6143e-01, -5.4236e-01, -8.3654e-02,\n",
       "          -1.1033e-01, -4.8837e-03, -5.0292e-02, -6.6102e-02,  1.1939e-02,\n",
       "           1.3632e-01,  2.3587e-01, -1.3023e-01, -1.3178e-01, -9.2464e-02,\n",
       "          -5.5566e-01, -1.1753e-01, -9.9591e-02, -1.2505e-01, -5.6783e-01,\n",
       "          -1.3797e-01,  3.3075e-02, -2.5373e-01, -6.8124e-02,  3.5608e-01,\n",
       "          -2.9832e-02, -2.7856e-02, -4.8850e-01,  7.2846e-01,  4.0279e-01,\n",
       "           5.2088e-02, -2.8298e-01, -5.5189e-01, -1.0567e-01,  2.9592e-01,\n",
       "           1.9555e-01, -6.7007e-02, -1.7709e-01, -2.4145e-01, -4.2522e-01,\n",
       "           1.4302e-01,  7.1147e-01, -6.6296e-02, -6.8115e-01,  2.6621e-01,\n",
       "           5.8223e-01, -1.2882e-01, -2.4432e-01,  2.0983e-02, -9.3284e-02,\n",
       "          -3.1851e-01, -6.0317e-01,  7.8821e-01, -1.1246e-01, -5.8638e-02,\n",
       "           2.2427e-01,  2.0359e-01, -9.2588e-03,  4.0209e-01, -1.5770e-01,\n",
       "          -3.5434e-01, -1.6477e-03, -7.3600e-01,  4.3257e-01,  6.1666e-02,\n",
       "          -3.1489e-01,  7.6082e-01,  2.7260e-01, -6.0593e-02, -4.5779e-02,\n",
       "           2.2863e-01,  7.9941e-01, -2.4636e-01,  6.4653e-02,  5.3779e-01,\n",
       "           8.8110e-02, -1.1229e-01,  3.9580e-02,  4.1665e-02,  4.5410e-02,\n",
       "           3.7012e-01, -3.2145e-02, -4.1564e-01, -6.1194e-02,  3.8924e-01,\n",
       "           3.9295e-01, -5.4610e-01, -7.3487e-01,  2.2333e-02,  6.8551e-01,\n",
       "          -2.9054e-01,  2.3654e-01, -3.2708e-02,  3.0880e-01,  1.2001e-01,\n",
       "          -9.2160e-01, -1.5016e-01, -1.0056e-01,  2.8229e-01, -2.5482e-02,\n",
       "          -1.6139e-01, -1.4727e-01, -2.3686e-01, -1.6208e-01,  6.8308e-01,\n",
       "           2.1991e-01, -6.9160e-01,  3.4058e-02,  5.7281e-01, -6.0551e-01,\n",
       "           4.6052e-01, -2.4577e-01, -3.4302e-02,  3.6435e-01,  1.3117e-01,\n",
       "           4.9067e-02,  1.7856e-01, -4.2138e-01, -4.0598e-01,  1.7824e-01,\n",
       "           9.1672e-01, -7.6613e-02,  1.1166e-01, -3.9384e-01, -1.7182e-01,\n",
       "          -7.1855e-02, -3.1913e-01, -3.1286e-01,  3.7185e-01,  2.0529e-01,\n",
       "          -3.2552e-02, -1.3271e-01, -2.2992e-01,  3.0816e-01,  1.1821e-01,\n",
       "           3.2536e-01,  5.8187e-01,  6.4368e-01, -1.0077e-02, -2.4936e-01,\n",
       "          -6.1235e-02,  1.4568e-01, -1.9524e-01,  5.2704e-01, -2.0627e-01,\n",
       "          -3.3453e-01, -3.8069e-03,  3.0134e-01, -6.1681e-01, -5.8725e-02,\n",
       "          -4.5426e-01,  4.2361e-01, -1.6955e-01,  3.7868e-02,  2.8833e-01,\n",
       "           4.0522e-02,  5.7925e-02, -2.2651e-01,  5.5807e-01,  4.4511e-01,\n",
       "           1.6609e-01,  4.5662e-01,  1.6104e-01,  4.5272e-02, -5.1597e-01,\n",
       "           2.1978e-01, -1.0213e+00, -9.4260e-02, -4.6640e-02, -1.5401e-01,\n",
       "          -4.4224e-01, -2.8126e-02,  1.6814e-01,  1.8538e-01,  6.9597e-01,\n",
       "           7.2967e-01, -3.0871e-01, -4.5290e-01, -1.8510e-01,  1.5304e-01,\n",
       "           2.7705e-01, -1.9517e-01,  3.5971e-01,  9.6103e-02,  2.4432e-01,\n",
       "          -5.0967e-01, -4.3165e-01, -4.2252e-02,  3.1411e-01, -2.0171e-01,\n",
       "           5.7118e-01, -2.0164e-01, -1.3863e-01,  9.0832e-02,  1.3770e-01,\n",
       "           1.6639e-01,  7.9664e-01,  4.3416e-01, -6.3425e-01,  8.7469e-02,\n",
       "          -1.3980e-02, -2.9988e-01,  2.5983e-02, -4.3577e-01,  2.8858e-01,\n",
       "           1.2984e-01,  4.9280e-01,  6.7504e-01]], device='mps:0'),\n",
       " tensor([[-0.1672, -0.5249, -0.4540,  ...,  0.3771,  0.2016,  0.5340],\n",
       "         [-0.3292, -0.2590, -0.5016,  ..., -0.0148,  0.3509,  0.1607]],\n",
       "        device='mps:0'),\n",
       " tensor([[-1.5136e-01, -7.5586e-01, -8.1384e-01, -9.1980e-01, -5.2126e-01,\n",
       "          -3.2844e-01, -6.0972e-01,  1.2110e-01,  1.6680e-01,  2.5252e-01,\n",
       "           3.7932e-01,  1.2372e-01, -2.7941e-01, -3.0830e-01, -5.7015e-01,\n",
       "          -1.7003e-01, -3.1723e-01,  1.7830e-01,  5.8713e-01,  4.0714e-01,\n",
       "           7.5345e-01,  3.7571e-01, -1.0960e-01,  1.5255e-01, -2.4459e-01,\n",
       "          -1.5226e-01, -3.6247e-01, -3.9301e-01,  4.3934e-01, -4.4308e-01,\n",
       "          -2.2458e-01,  7.2268e-01, -1.2187e-01,  1.0884e+00, -2.6833e-01,\n",
       "          -1.9729e-01, -1.1235e-01, -1.9969e-02,  4.9466e-01, -3.6754e-01,\n",
       "           4.2648e-01,  4.8437e-01, -1.5944e-01, -3.8274e-02,  4.1515e-01,\n",
       "           3.4124e-01,  3.4597e-02, -1.0402e+00, -2.6556e-01, -4.1580e-01,\n",
       "          -1.5103e-01, -2.2391e-01,  3.1307e-02,  3.2039e-02, -5.6966e-01,\n",
       "          -6.4446e-01,  3.6746e-01,  4.3025e-01, -3.3519e-01,  4.2449e-01,\n",
       "           1.6695e-01,  5.7898e-01, -3.2483e-01,  7.1463e-02,  2.9963e-01,\n",
       "           1.2657e-01, -1.8916e-01,  2.3430e-02, -5.9626e-01,  2.3319e-01,\n",
       "          -2.4495e-01, -3.2527e-01, -3.8707e-01, -3.2053e-01, -3.8760e-01,\n",
       "          -4.0921e-01,  4.9236e-01,  1.0977e-01, -7.8428e-01, -1.0021e-01,\n",
       "           1.4690e-01,  2.0643e-01,  3.6060e-01, -3.9667e-01, -6.7072e-02,\n",
       "          -1.3855e-01,  5.4452e-01,  3.6637e-01, -1.9874e-01, -2.8721e-01,\n",
       "           2.1328e-03,  9.2310e-03, -8.3384e-02, -2.6789e-02, -5.2821e-02,\n",
       "          -3.1838e-01,  7.5880e-02,  3.7851e-02,  1.4477e-02, -2.5927e-01,\n",
       "           1.4979e-01, -4.7950e-02,  1.2924e-02,  3.2758e-01, -7.2160e-01,\n",
       "          -1.6211e-01,  1.9544e-01,  3.8454e-01, -1.0742e+00, -2.2150e-01,\n",
       "          -1.0660e-01, -6.0180e-01,  3.5667e-01, -7.7395e-01,  9.9449e-02,\n",
       "           3.1683e-01,  2.1767e-01, -2.3871e-01, -4.8071e-02,  2.3616e-02,\n",
       "           9.0168e-02, -5.3738e-01,  2.0883e-01,  3.7058e-01,  4.4826e-02,\n",
       "           1.0232e-01,  7.5834e-01, -2.0404e-02,  2.9376e-01, -7.0217e-02,\n",
       "          -1.6667e-01, -5.8728e-02, -2.7217e-01, -1.1385e-01,  6.7200e-02,\n",
       "          -1.5064e-01, -4.4954e-01, -4.1478e-01,  2.3839e-01, -1.7561e-01,\n",
       "          -1.2304e-01,  1.3956e-01, -1.6829e-01, -6.7839e-02,  5.4122e-02,\n",
       "          -3.5982e-01,  1.2483e-01, -5.5054e-01, -7.7597e-02, -5.6098e-01,\n",
       "          -8.4665e-01, -7.7005e-01,  4.6723e-01, -7.7317e-02,  9.5109e-02,\n",
       "          -3.0340e-01,  5.1211e-01, -8.0929e-02, -4.5807e-01,  2.3248e-01,\n",
       "           5.1955e-01, -1.3894e-01,  3.6038e-01, -2.8268e-01,  9.5521e-02,\n",
       "           3.1819e-01, -4.8392e-01, -3.0411e-01,  4.8015e-01,  6.4552e-02,\n",
       "           5.4493e-01, -1.8575e-01,  4.8536e-01,  3.3529e-02,  7.0859e-02,\n",
       "          -5.3047e-01, -1.4043e-01, -2.4773e-01,  8.1425e-02, -2.0647e-01,\n",
       "           7.6947e-01, -1.8773e+00, -6.8594e-02, -4.0633e-01, -5.0227e-01,\n",
       "          -7.4195e-01,  5.9844e-02,  1.4247e-01,  2.9475e-01, -3.5434e-01,\n",
       "           3.8264e-01, -1.4103e-01, -5.3223e-02, -6.1844e-02, -3.6088e-01,\n",
       "           2.0156e-01,  6.9581e-01, -4.7764e-02,  2.8092e-01, -1.4545e-02,\n",
       "          -5.4543e-01,  4.7419e-01,  1.7242e-02, -1.7857e-01,  1.2477e-01,\n",
       "           4.5569e-02, -3.9765e-01,  3.8486e-01, -1.9301e-02, -3.3858e-02,\n",
       "           3.3231e-01, -1.5043e-01, -2.6527e-01, -3.6944e-01, -3.6290e-01,\n",
       "           4.9682e-01, -8.2745e-02, -2.7547e-01,  4.4828e-01,  3.1405e-01,\n",
       "           1.9693e-01, -2.5512e-01,  3.5234e-02, -9.5527e-02,  9.4319e-02,\n",
       "           4.2944e-01,  5.2496e-02,  8.0113e-02, -1.8984e-01, -5.3948e-02,\n",
       "          -1.8223e-01, -4.3966e-01,  2.3930e-01,  8.6482e-01,  1.5895e-01,\n",
       "          -6.5677e-02,  4.8381e-01,  1.8769e-01, -6.6619e-01, -1.2295e-01,\n",
       "          -6.1654e-02, -4.4951e-01, -2.3619e-01,  1.3015e-01,  3.2170e-01,\n",
       "           2.4061e-01, -2.9828e-01,  6.2679e-01,  6.5494e-01,  3.5102e-01,\n",
       "           1.2191e-01,  6.8163e-02,  3.2044e-01,  5.8914e-01, -2.5390e-01,\n",
       "          -3.5122e-01, -2.6185e-01, -8.0897e-02,  3.0693e-01, -1.3989e-01,\n",
       "          -3.9008e-03,  9.6840e-03,  1.4231e-01,  2.3466e-01, -3.8002e-01,\n",
       "           2.3948e-01,  4.3379e-01,  5.2116e-02, -4.0920e-01,  1.9851e-01,\n",
       "          -4.7578e-01,  5.4326e-01, -5.4287e-01, -1.6239e-01, -5.6266e-01,\n",
       "          -1.1720e+00, -3.6596e-01,  3.0435e-01,  3.8556e-01, -6.6375e-01,\n",
       "          -4.4063e-01, -1.5712e-01,  2.3455e-01, -1.2639e-01,  2.2004e-02,\n",
       "           6.0797e-01,  1.8632e-02, -2.7533e-01, -3.7525e-01, -1.5134e-01,\n",
       "          -2.1408e-01,  3.1023e-01, -3.7581e-01, -7.1781e-03,  7.3526e-02,\n",
       "           1.5049e-01,  3.8825e-01,  2.0030e-01,  1.2658e-01, -3.4927e-01,\n",
       "           2.2997e-01,  3.0363e-01,  1.0879e-01, -5.8432e-01, -1.3193e-01,\n",
       "           3.3130e-01,  2.4730e-01,  2.4390e-01, -2.1613e-01, -1.5271e-01,\n",
       "          -2.1170e-01,  5.7792e-01, -8.6089e-02,  3.6916e-01,  1.1893e-01,\n",
       "          -1.2018e+00,  1.5474e-01, -3.0351e-01, -1.7213e-01, -2.2296e-01,\n",
       "           4.9387e-01,  1.7435e-01,  3.6068e-03, -6.3411e-02,  1.6763e-01,\n",
       "           6.4545e-02,  6.5791e-01,  2.1606e-01, -1.4580e-01,  9.9922e-03,\n",
       "          -1.7031e-03, -3.6968e-01, -1.5085e-01, -1.9520e-01, -2.8417e-02,\n",
       "           3.3851e-01,  3.3543e-01,  7.1366e-02, -3.4919e-01,  4.0577e-02,\n",
       "           6.8291e-01, -2.8802e-01, -1.6444e-01,  3.1672e-01, -4.8397e-01,\n",
       "           6.1017e-02,  4.9453e-02, -3.1211e-01,  7.8324e-02,  5.8786e-02,\n",
       "           3.4805e-01, -1.2552e-01, -3.3886e-01,  8.9838e-02,  6.0651e-01,\n",
       "           5.2233e-01,  3.6151e-01, -2.3630e-01, -6.7023e-02,  4.0056e-01,\n",
       "          -3.6999e-01,  4.6654e-01, -1.8530e-02, -2.3368e-01,  2.7612e-01,\n",
       "           3.1782e-01,  3.3012e-01, -1.9079e-01,  9.0653e-02,  1.7389e-01,\n",
       "           7.9576e-02, -2.8089e-01, -1.7136e-01,  5.1910e-01,  4.2178e-01,\n",
       "          -2.3109e-01, -6.4818e-01,  3.7474e-01, -2.2280e-01, -2.4887e-01,\n",
       "           6.2658e-02,  2.6196e-02, -1.3083e-02, -6.7907e-02, -3.9152e-01,\n",
       "          -3.1219e-01, -4.1153e-02,  2.1500e-01,  4.5415e-01, -3.5176e-01,\n",
       "           6.4365e-02, -4.8427e-02, -8.7736e-02,  7.2485e-01, -8.2825e-02,\n",
       "           1.2977e-01,  6.2105e-01,  2.5941e-01,  8.7971e-02,  5.0688e-01,\n",
       "          -2.1968e-01, -3.0563e-01,  1.2088e-01,  2.4942e-02, -5.4484e-02,\n",
       "          -1.0436e-01,  2.7450e-02,  5.0762e-01, -3.3658e-01, -1.1933e-01,\n",
       "          -9.9179e-01,  2.5511e-01,  3.8789e-01,  3.4282e-01, -3.3047e-01,\n",
       "           3.8866e-01, -1.0132e-01,  1.3878e-01, -4.8119e-01,  1.1502e-02,\n",
       "           4.0276e-01,  1.6503e-01,  2.5691e-01,  9.3528e-01, -1.1160e+01,\n",
       "           6.8179e-01,  8.0300e-01,  8.0442e-02, -2.2256e-01,  1.4825e-01,\n",
       "           1.3920e-01, -1.6389e-01,  3.4991e-01, -3.9364e-01, -6.4644e-02,\n",
       "          -2.7783e-01, -2.5025e-01, -1.2225e-01, -3.6500e-01,  1.0035e-01,\n",
       "          -2.3303e-02, -7.0200e-01,  4.0127e-01,  5.9541e-02,  4.0224e-01,\n",
       "          -1.1860e-01,  2.4353e-01,  3.7284e-01, -4.9035e-01, -4.1969e-02,\n",
       "           3.2856e-03, -3.6592e-01,  1.3950e-01,  3.0475e-01, -2.1594e-01,\n",
       "           4.7233e-02,  2.8216e-02,  4.5392e-01, -4.7154e-01, -2.6509e-01,\n",
       "           8.6872e-02, -5.2238e-01,  3.3344e-01, -4.0733e-01, -4.8060e-01,\n",
       "           5.8748e-01, -4.7829e-01,  1.1407e-01,  4.5538e-01,  7.5794e-01,\n",
       "           1.4339e+00,  8.3883e-01, -1.2703e-01, -1.7992e-02, -4.9264e-03,\n",
       "          -8.2394e-01, -7.4832e-02, -7.3050e-01,  1.9121e-01, -4.1946e-02,\n",
       "           4.9045e-01, -7.1533e-01,  2.1530e-01, -3.9709e-01,  1.9434e-01,\n",
       "          -1.5212e-01,  4.4016e-02,  5.0940e-01, -5.5294e-01, -5.2070e-01,\n",
       "           3.1121e-01,  2.3738e-01,  2.0496e-01, -8.1316e-01,  1.3610e-01,\n",
       "           2.4623e-01,  3.2467e-01,  1.1656e-01, -3.0166e-01,  3.6217e-01,\n",
       "           3.6646e-01, -2.2560e-01,  1.2670e+00, -4.7642e-01, -4.2682e-02,\n",
       "          -2.2889e-01,  5.4941e-02, -8.5887e-02, -6.3624e-01, -3.5596e-01,\n",
       "           2.6584e-01,  1.8777e-01,  5.5298e-01,  4.4232e-01, -1.3906e-01,\n",
       "          -3.4432e-01,  1.0121e-01,  9.5618e-02,  2.6158e-01,  6.8637e-01,\n",
       "          -3.4487e-01, -3.6082e-02,  8.9440e-03,  1.8037e-01, -7.3768e-02,\n",
       "          -3.4593e-01, -2.8760e-01, -2.6471e-01,  2.8607e-01,  2.7817e-01,\n",
       "           1.1018e-01,  2.4931e-01,  5.9552e-01,  2.1092e-01, -2.9968e-01,\n",
       "          -2.4228e-01, -2.1867e-01,  3.1249e-02, -5.7933e-01, -6.6221e-01,\n",
       "           1.1008e-01,  1.8517e-01, -3.0263e-01, -8.0752e-01, -5.9058e-01,\n",
       "          -2.2687e-01,  3.1117e-01,  2.5546e-01, -3.5573e-01,  6.1044e-01,\n",
       "          -2.0076e-01,  7.4884e-03,  2.2000e-01,  2.3360e-01,  3.8192e-01,\n",
       "           4.6632e-01, -6.3975e-02,  3.5949e-01, -1.7871e-01, -3.2916e-01,\n",
       "          -6.1769e-01,  5.2273e-02,  4.3087e-01, -5.6046e-02,  4.1120e-01,\n",
       "           2.1191e-01, -2.7603e-01, -1.3528e-01,  2.1660e-01,  2.3636e-01,\n",
       "           1.6861e-01,  4.9719e-01, -2.7396e-01, -2.0366e-01, -1.9574e-01,\n",
       "          -2.3413e-01, -2.8023e-02, -3.0199e-01, -3.4568e-01, -5.0850e-01,\n",
       "          -8.0719e-02, -1.9801e-01, -2.9696e-02,  4.9834e-01, -1.0420e-02,\n",
       "          -1.4309e-01, -1.5234e-01, -2.4274e-01,  8.5199e-01,  3.2406e-01,\n",
       "          -7.0302e-02, -3.4655e-01, -4.8790e-01, -6.0330e-02,  2.0181e-01,\n",
       "           3.0854e-01, -1.9956e-01, -3.8062e-01, -1.0060e-01,  2.1768e-01,\n",
       "           1.9098e-01,  1.1828e+00, -6.5476e-01, -2.7473e-01, -6.0527e-03,\n",
       "           3.3533e-01, -2.2773e-01, -3.0782e-01,  5.2094e-01, -2.5529e-01,\n",
       "          -1.5442e-01, -1.6367e-01,  2.5357e-01, -1.5772e-01,  8.0009e-02,\n",
       "           6.2643e-01,  4.5658e-02,  1.8714e-01,  1.7549e-01, -5.2621e-01,\n",
       "          -1.8228e-01,  3.1855e-02,  6.6168e-03, -2.9340e-01,  2.7452e-01,\n",
       "          -4.0433e-01,  2.9787e-01, -6.2657e-01, -5.5859e-01,  4.8579e-01,\n",
       "           3.5936e-01,  1.2429e+00, -9.9552e-02, -6.7378e-01,  4.6484e-01,\n",
       "           3.1666e-01, -4.3585e-01,  4.3006e-01, -3.9853e-02,  7.6519e-02,\n",
       "           5.4779e-01,  2.4986e-01, -8.2126e-01, -1.5716e-01,  3.1502e-01,\n",
       "           2.7757e-01, -3.8188e-01, -5.0202e-01, -1.9937e-01, -2.0487e-01,\n",
       "          -1.5409e-01,  4.0612e-01,  1.3366e-01,  7.0740e-02,  3.2261e-02,\n",
       "          -9.2541e-01, -1.3949e-01,  5.3396e-01,  4.2065e-01, -1.0356e-01,\n",
       "          -5.7435e-01, -3.3747e-01, -2.2956e-01, -4.4699e-01, -4.4595e-02,\n",
       "           6.4759e-01, -8.1308e-01,  2.4580e-01,  1.7554e-01, -3.4362e-01,\n",
       "           4.2627e-01,  2.2459e-01,  2.9799e-01,  2.2486e-01, -5.6201e-02,\n",
       "          -5.6559e-02,  7.8461e-02, -3.5867e-02, -2.7324e-01,  4.9894e-01,\n",
       "           4.2878e-01, -5.4827e-02,  1.0165e-01,  1.6486e-01, -3.4116e-01,\n",
       "          -1.9494e-01, -1.1452e-01, -1.4168e-01, -2.7533e-01, -4.0694e-02,\n",
       "           8.9495e-02, -4.5870e-01,  5.1001e-02, -1.5601e-01, -2.2101e-01,\n",
       "           1.0593e-01,  3.9124e-01,  5.8633e-01, -4.2341e-01, -3.4222e-01,\n",
       "          -1.6960e-02,  5.5202e-01, -2.7298e-01,  7.8831e-01,  1.8761e-01,\n",
       "          -1.7201e-01,  3.6569e-01,  3.8595e-01, -3.9224e-01,  1.7014e-01,\n",
       "          -3.9326e-01,  1.0144e-01, -6.2075e-01, -7.3016e-02,  3.2609e-02,\n",
       "           1.3212e-01, -1.7849e-01, -3.1477e-01, -2.0189e-01,  1.8516e-01,\n",
       "          -1.1255e-01,  7.2665e-01,  4.2921e-02,  7.1932e-01, -1.9012e-01,\n",
       "          -9.8441e-02, -5.0702e-01, -2.5493e-02,  3.3866e-01, -2.2337e-01,\n",
       "          -2.8937e-01, -4.3076e-01, -8.7455e-02, -1.1562e-01,  8.6517e-01,\n",
       "           5.9388e-01, -5.4413e-02, -3.1162e-01, -3.2387e-01,  7.3285e-03,\n",
       "           1.7597e-01,  3.2424e-01,  1.4821e-02,  2.5427e-01, -9.2756e-02,\n",
       "          -7.8851e-01,  2.6514e-01, -2.5584e-01,  2.6861e-01, -2.3964e-01,\n",
       "           8.4253e-01, -2.9942e-01, -3.4974e-02,  1.1885e+00,  8.9965e-02,\n",
       "          -2.2190e-01,  6.6023e-01, -7.7138e-02, -5.2935e-01, -1.0319e-02,\n",
       "          -3.6080e-02, -3.6365e-01,  2.3211e-01, -6.0540e-01,  8.6115e-02,\n",
       "          -2.9785e-01,  2.8131e-01,  2.9262e-01]], device='mps:0'),\n",
       " tensor([[-2.5187e-01, -4.0054e-01, -5.8007e-01, -8.3055e-01, -1.2868e-01,\n",
       "          -2.8804e-02, -7.1917e-01,  6.6701e-01,  1.2709e-01,  5.0855e-01,\n",
       "           1.2570e-01,  5.1916e-01,  5.1676e-02, -1.1521e-01, -1.6481e-01,\n",
       "          -4.9820e-01,  5.6222e-02,  3.7338e-01,  6.7187e-01, -2.4111e-02,\n",
       "           2.2844e-01,  5.7979e-01, -2.6323e-01, -1.2386e-02, -8.3013e-01,\n",
       "           2.9613e-01, -1.1576e-01,  2.2409e-01,  1.7718e-01, -6.3092e-01,\n",
       "           4.0999e-02,  4.7417e-01,  2.3068e-02,  1.3497e+00, -5.6080e-01,\n",
       "          -1.3230e-01,  8.2213e-02, -3.9453e-01, -2.3010e-01,  3.7283e-02,\n",
       "           6.1164e-01, -4.1315e-01,  1.3635e-01,  4.0254e-01, -6.1085e-02,\n",
       "           8.9076e-02,  1.3899e-01, -3.7744e-01,  2.5892e-01, -8.3961e-01,\n",
       "          -4.8318e-02, -2.0669e-01,  4.9144e-01,  6.2615e-01, -6.3176e-01,\n",
       "          -2.2975e-01,  2.7419e-01, -3.7716e-02, -3.3020e-02,  1.5349e-01,\n",
       "           9.7980e-02,  3.8582e-01, -2.9995e-01,  2.3257e-01,  3.3441e-01,\n",
       "           2.4458e-01, -3.0392e-01, -4.8889e-01, -5.4078e-01, -5.0111e-02,\n",
       "           9.1859e-03,  2.2490e-02,  1.4846e-01, -2.3272e-01,  4.1973e-01,\n",
       "          -7.1214e-01, -6.4354e-02, -8.9125e-02, -2.5973e-01,  2.9246e-01,\n",
       "          -4.5549e-02,  1.9719e-01,  3.1010e-01,  1.0268e-01,  1.2457e-01,\n",
       "           4.3419e-02,  2.7215e-01,  5.2932e-04,  4.7373e-02,  3.7533e-01,\n",
       "          -1.3375e-01, -8.1722e-02, -2.8313e-01, -3.5814e-01,  3.1229e-01,\n",
       "           4.0598e-02,  2.9907e-01, -1.9557e-01,  1.8579e-01, -3.2474e-01,\n",
       "           1.5588e-01,  1.7768e-01, -1.8083e-01,  5.7286e-02, -3.5172e-01,\n",
       "          -4.5475e-01,  2.3589e-02, -2.8569e-01, -4.8285e-01,  2.2697e-01,\n",
       "          -1.4211e-01, -3.0205e-01, -4.0922e-02, -3.8935e-01, -3.1524e-02,\n",
       "          -7.5164e-02,  2.6013e-01, -1.3230e-01, -3.4806e-01, -8.3033e-02,\n",
       "           2.0734e-01, -2.2808e-01,  1.6077e-01,  2.6882e-01,  1.2232e-01,\n",
       "          -9.6705e-02,  6.5251e-01,  1.8950e-02,  3.1674e-01,  2.5763e-01,\n",
       "           1.1699e-01, -6.5768e-02, -1.9745e-01, -2.3154e-01,  7.5988e-02,\n",
       "          -7.6941e-02,  1.0687e-01, -1.8182e-01, -2.2790e-01, -1.8508e-01,\n",
       "           3.1771e-01, -3.3097e-01, -6.0893e-01, -4.9260e-01, -1.6098e-01,\n",
       "          -1.6326e-01, -2.4804e-02, -8.2473e-03, -3.3693e-01, -3.1956e-01,\n",
       "          -6.7443e-01, -3.6717e-01,  8.1041e-01, -1.6831e-02, -3.3356e-01,\n",
       "          -1.6499e-01,  2.3208e-01,  3.1666e-01, -4.5959e-02,  3.7161e-01,\n",
       "           7.7250e-01, -4.7670e-02,  1.6786e-01, -5.1827e-01, -3.8195e-01,\n",
       "           4.8335e-02,  2.6658e-01, -2.7054e-01,  6.1879e-01,  4.8352e-02,\n",
       "           7.5627e-02, -6.6170e-02,  2.7430e-01, -3.2704e-01, -1.2324e-02,\n",
       "          -5.7573e-01,  3.7750e-01,  2.9539e-01, -3.9892e-01, -6.6567e-02,\n",
       "          -2.0834e-02, -1.2060e+00,  9.6964e-02, -8.3040e-01,  3.1236e-01,\n",
       "          -2.3110e-01,  1.7963e-01,  2.3389e-01,  2.9545e-01, -6.5320e-02,\n",
       "           2.5404e-01,  2.8725e-01,  2.5687e-01,  3.3609e-01, -3.4924e-02,\n",
       "           2.8019e-01,  2.3924e-01,  5.0790e-01,  4.1073e-01,  1.4769e-01,\n",
       "          -3.5440e-01,  2.9683e-01,  3.5781e-01,  9.0598e-02,  1.6954e-01,\n",
       "           9.9825e-02, -2.1636e-01,  2.1091e-01,  1.7501e-01, -1.6618e-01,\n",
       "          -2.2406e-01, -3.1100e-02, -4.6818e-01, -5.6655e-02, -4.8467e-01,\n",
       "           3.2557e-01, -5.1983e-01, -1.1912e-01,  3.4831e-01,  2.4340e-01,\n",
       "          -1.4896e-02,  1.8319e-01,  1.7196e-01, -2.4338e-01,  2.1735e-01,\n",
       "           3.7449e-01, -3.4618e-01, -3.9804e-02, -3.2197e-01, -2.9367e-01,\n",
       "           2.4357e-01, -5.6244e-01,  6.6635e-01,  4.0574e-01,  6.0165e-02,\n",
       "           1.5119e-01, -2.7920e-01, -5.1638e-02, -5.7408e-01,  3.5471e-01,\n",
       "          -3.2130e-01, -3.4955e-01, -5.4801e-01,  6.4487e-02,  4.2773e-01,\n",
       "           2.7565e-01, -6.4915e-02,  7.7868e-01,  4.2408e-01,  4.2813e-01,\n",
       "          -4.3003e-01, -1.5896e-02,  4.0377e-01, -4.1405e-02, -4.6059e-01,\n",
       "           6.7116e-02, -1.4962e-01, -5.1643e-01,  2.8000e-01, -2.6598e-01,\n",
       "          -1.2047e-01, -1.4913e-01,  3.3682e-01,  1.5365e-01, -2.1964e-01,\n",
       "          -3.8461e-01,  3.9587e-01, -1.4349e-01, -4.1723e-01,  4.7123e-01,\n",
       "          -3.5025e-01,  1.5209e-01,  1.0694e-01, -3.8441e-01, -6.2072e-01,\n",
       "          -3.2235e-01,  2.3090e-01,  1.5585e-01,  3.4799e-02, -6.6306e-01,\n",
       "          -4.0571e-01,  3.1197e-01,  2.4601e-01,  2.9976e-01,  2.9840e-01,\n",
       "           5.7427e-01, -3.0944e-01, -1.6702e-01,  1.4982e-01, -4.1377e-01,\n",
       "          -1.5704e-01, -3.3228e-02, -1.9995e-01,  2.9569e-01,  2.4768e-01,\n",
       "           1.1600e-01,  2.0367e-01,  4.3482e-01, -2.5272e-01, -3.8673e-01,\n",
       "          -7.1543e-02,  2.0161e-01, -9.3629e-02, -5.8306e-02,  2.3951e-01,\n",
       "          -6.3095e-02,  5.9650e-01,  7.2143e-01,  7.7439e-02, -6.4006e-01,\n",
       "           1.7696e-01,  5.2502e-01, -2.7253e-01, -4.0343e-02,  9.6365e-02,\n",
       "          -1.0344e+00, -2.1331e-01, -1.2883e-01, -2.0181e-01, -6.5352e-01,\n",
       "           8.8115e-02,  3.4863e-01,  4.6833e-03, -1.7864e-01, -3.0947e-01,\n",
       "           1.0448e-01,  4.8724e-02, -4.8980e-02,  4.3534e-01, -2.8901e-01,\n",
       "          -3.4797e-01, -2.3438e-01, -2.1532e-01, -5.4201e-01, -3.9968e-01,\n",
       "           3.5914e-01,  2.8129e-01,  1.1986e-01, -2.6486e-01, -5.1832e-01,\n",
       "           4.5577e-01, -6.8169e-01, -1.4632e-01,  2.4840e-01, -4.7702e-01,\n",
       "           1.4650e-02,  4.8990e-01,  1.4999e-01,  1.6336e-01,  6.5258e-02,\n",
       "           7.8557e-02, -3.3903e-01, -5.0030e-01, -1.5000e-01,  6.2323e-01,\n",
       "           3.2283e-01,  3.3506e-01, -1.7236e-01,  3.9096e-01,  4.5167e-01,\n",
       "          -2.9582e-01,  4.1670e-01,  1.0930e-02,  7.7439e-02, -6.0414e-02,\n",
       "           1.9048e-01,  3.4370e-01, -3.6842e-01,  2.0145e-01,  1.7929e-01,\n",
       "          -3.1634e-01, -3.2149e-01,  1.5691e-02,  2.0642e-01, -1.8580e-01,\n",
       "          -3.0875e-01, -2.0361e-01,  9.9698e-02, -3.4856e-01, -4.3104e-01,\n",
       "           2.9161e-01, -3.6543e-02,  7.0495e-02, -4.0357e-01, -4.3462e-01,\n",
       "          -2.0181e-01, -2.6055e-01,  6.8681e-01,  2.1064e-01, -1.9621e-01,\n",
       "           1.8720e-01, -1.0907e-01, -3.6330e-01,  4.7847e-01,  2.0162e-02,\n",
       "          -4.8754e-01,  4.5482e-01, -3.4822e-01,  1.0627e-01,  5.5756e-01,\n",
       "          -1.6769e-01, -1.0840e-01,  1.1008e-01,  1.0820e-01,  2.9026e-01,\n",
       "          -3.9884e-01,  1.5516e-01, -6.3476e-02, -3.1885e-01,  3.1820e-01,\n",
       "           2.6411e-01,  2.2781e-01, -1.7137e-01, -3.3426e-02,  3.5677e-02,\n",
       "           2.8782e-01, -6.0853e-02, -3.9888e-01, -4.7347e-01, -4.6132e-01,\n",
       "           1.2832e-01,  1.8927e-01,  5.6207e-01,  3.4144e-01, -1.1394e+01,\n",
       "           1.2938e-01,  7.8031e-01, -4.0713e-01, -8.2819e-02, -5.0320e-02,\n",
       "           1.0976e-02, -1.3708e-01,  4.2955e-01, -2.7734e-01, -4.4105e-01,\n",
       "          -3.5727e-01, -1.3308e-01,  1.5454e-01, -3.2189e-01, -2.6945e-01,\n",
       "          -1.4686e-02, -7.0935e-01,  4.0963e-01, -3.0286e-01,  5.0298e-01,\n",
       "          -2.3511e-01, -8.3798e-02,  1.5816e-01, -3.5851e-01, -1.3439e-02,\n",
       "           5.9766e-02, -1.6033e-01,  1.4651e-01,  4.2486e-01, -5.2072e-02,\n",
       "           3.3244e-02, -4.1258e-03,  4.8461e-02, -2.5723e-01, -5.9999e-01,\n",
       "           1.9296e-02, -1.4141e-01,  3.3422e-01, -4.4133e-02, -7.1265e-01,\n",
       "          -2.2533e-01, -2.7418e-01,  3.1407e-01,  2.7021e-01,  9.0598e-01,\n",
       "           2.2955e-01,  1.0686e+00,  1.5022e-01,  1.7748e-01,  2.6230e-02,\n",
       "          -7.8764e-01, -3.3269e-02, -4.1907e-01,  2.3280e-02,  1.0114e-01,\n",
       "           3.6576e-01, -7.9367e-01,  2.5251e-01, -3.5521e-01,  3.6487e-02,\n",
       "           4.9094e-01,  3.1768e-02,  2.7700e-01, -3.2298e-02, -7.8123e-02,\n",
       "           1.2089e-01, -2.8544e-02,  1.1098e-01, -2.2583e-01, -2.4084e-01,\n",
       "          -1.0177e-01,  5.9888e-01,  3.9592e-01, -7.2211e-01,  3.6112e-01,\n",
       "           4.9258e-02, -1.8622e-01,  1.4556e+00, -4.8573e-02,  1.8361e-02,\n",
       "           1.9800e-03,  2.9604e-01, -5.0236e-01, -7.3516e-01,  3.6511e-01,\n",
       "          -1.0211e-01,  5.6427e-02,  1.0127e-01,  4.8148e-01, -9.7481e-02,\n",
       "          -4.7940e-01, -1.8593e-03,  1.0556e-01, -1.3380e-01,  5.4710e-01,\n",
       "          -7.2824e-01, -1.6963e-01, -1.1223e-01,  3.1723e-01, -8.4054e-01,\n",
       "          -6.1126e-02, -7.0210e-01, -3.9328e-01,  5.0586e-01, -2.5896e-01,\n",
       "           1.5852e-01, -2.7443e-02, -4.3439e-02,  5.5574e-01, -2.1489e-01,\n",
       "           7.1555e-02,  2.2813e-01,  8.4995e-02, -1.6733e-01,  3.9349e-02,\n",
       "           4.1630e-01,  4.9571e-01, -4.2740e-01, -6.3911e-01, -7.6901e-03,\n",
       "           1.7180e-01,  1.4933e-01,  3.4019e-01, -5.3463e-01,  3.6431e-01,\n",
       "          -7.9162e-01, -1.8769e-01,  7.2949e-02, -4.3273e-02,  3.6175e-01,\n",
       "           2.3925e-01,  7.2038e-02,  2.2256e-01, -7.7756e-01, -6.4498e-01,\n",
       "          -6.5932e-01, -2.1445e-01,  2.6143e-01, -5.4236e-01, -8.3654e-02,\n",
       "          -1.1033e-01, -4.8837e-03, -5.0292e-02, -6.6102e-02,  1.1939e-02,\n",
       "           1.3632e-01,  2.3587e-01, -1.3023e-01, -1.3178e-01, -9.2464e-02,\n",
       "          -5.5566e-01, -1.1753e-01, -9.9591e-02, -1.2505e-01, -5.6783e-01,\n",
       "          -1.3797e-01,  3.3075e-02, -2.5373e-01, -6.8124e-02,  3.5608e-01,\n",
       "          -2.9832e-02, -2.7856e-02, -4.8850e-01,  7.2846e-01,  4.0279e-01,\n",
       "           5.2088e-02, -2.8298e-01, -5.5189e-01, -1.0567e-01,  2.9592e-01,\n",
       "           1.9555e-01, -6.7007e-02, -1.7709e-01, -2.4145e-01, -4.2522e-01,\n",
       "           1.4302e-01,  7.1147e-01, -6.6296e-02, -6.8115e-01,  2.6621e-01,\n",
       "           5.8223e-01, -1.2882e-01, -2.4432e-01,  2.0983e-02, -9.3284e-02,\n",
       "          -3.1851e-01, -6.0317e-01,  7.8821e-01, -1.1246e-01, -5.8638e-02,\n",
       "           2.2427e-01,  2.0359e-01, -9.2588e-03,  4.0209e-01, -1.5770e-01,\n",
       "          -3.5434e-01, -1.6477e-03, -7.3600e-01,  4.3257e-01,  6.1666e-02,\n",
       "          -3.1489e-01,  7.6082e-01,  2.7260e-01, -6.0593e-02, -4.5779e-02,\n",
       "           2.2863e-01,  7.9941e-01, -2.4636e-01,  6.4653e-02,  5.3779e-01,\n",
       "           8.8110e-02, -1.1229e-01,  3.9580e-02,  4.1665e-02,  4.5410e-02,\n",
       "           3.7012e-01, -3.2145e-02, -4.1564e-01, -6.1194e-02,  3.8924e-01,\n",
       "           3.9295e-01, -5.4610e-01, -7.3487e-01,  2.2333e-02,  6.8551e-01,\n",
       "          -2.9054e-01,  2.3654e-01, -3.2708e-02,  3.0880e-01,  1.2001e-01,\n",
       "          -9.2160e-01, -1.5016e-01, -1.0056e-01,  2.8229e-01, -2.5482e-02,\n",
       "          -1.6139e-01, -1.4727e-01, -2.3686e-01, -1.6208e-01,  6.8308e-01,\n",
       "           2.1991e-01, -6.9160e-01,  3.4058e-02,  5.7281e-01, -6.0551e-01,\n",
       "           4.6052e-01, -2.4577e-01, -3.4302e-02,  3.6435e-01,  1.3117e-01,\n",
       "           4.9067e-02,  1.7856e-01, -4.2138e-01, -4.0598e-01,  1.7824e-01,\n",
       "           9.1672e-01, -7.6613e-02,  1.1166e-01, -3.9384e-01, -1.7182e-01,\n",
       "          -7.1855e-02, -3.1913e-01, -3.1286e-01,  3.7185e-01,  2.0529e-01,\n",
       "          -3.2552e-02, -1.3271e-01, -2.2992e-01,  3.0816e-01,  1.1821e-01,\n",
       "           3.2536e-01,  5.8187e-01,  6.4368e-01, -1.0077e-02, -2.4936e-01,\n",
       "          -6.1235e-02,  1.4568e-01, -1.9524e-01,  5.2704e-01, -2.0627e-01,\n",
       "          -3.3453e-01, -3.8069e-03,  3.0134e-01, -6.1681e-01, -5.8725e-02,\n",
       "          -4.5426e-01,  4.2361e-01, -1.6955e-01,  3.7868e-02,  2.8833e-01,\n",
       "           4.0522e-02,  5.7925e-02, -2.2651e-01,  5.5807e-01,  4.4511e-01,\n",
       "           1.6609e-01,  4.5662e-01,  1.6104e-01,  4.5272e-02, -5.1597e-01,\n",
       "           2.1978e-01, -1.0213e+00, -9.4260e-02, -4.6640e-02, -1.5401e-01,\n",
       "          -4.4224e-01, -2.8126e-02,  1.6814e-01,  1.8538e-01,  6.9597e-01,\n",
       "           7.2967e-01, -3.0871e-01, -4.5290e-01, -1.8510e-01,  1.5304e-01,\n",
       "           2.7705e-01, -1.9517e-01,  3.5971e-01,  9.6103e-02,  2.4432e-01,\n",
       "          -5.0967e-01, -4.3165e-01, -4.2252e-02,  3.1411e-01, -2.0171e-01,\n",
       "           5.7118e-01, -2.0164e-01, -1.3863e-01,  9.0832e-02,  1.3770e-01,\n",
       "           1.6639e-01,  7.9664e-01,  4.3416e-01, -6.3425e-01,  8.7469e-02,\n",
       "          -1.3980e-02, -2.9988e-01,  2.5983e-02, -4.3577e-01,  2.8858e-01,\n",
       "           1.2984e-01,  4.9280e-01,  6.7504e-01]], device='mps:0'),\n",
       " tensor([[-1.2282e-01, -1.0009e+00, -8.0313e-01, -3.3379e-01, -6.7105e-01,\n",
       "           3.0037e-01, -3.2686e-01,  7.1374e-01,  1.3524e-01,  6.7131e-01,\n",
       "           4.0025e-01, -2.8247e-01,  7.3509e-02, -3.6023e-01, -5.3436e-01,\n",
       "          -4.6410e-01,  2.7697e-01,  3.8337e-01,  4.6139e-01,  4.4027e-01,\n",
       "           5.6217e-01,  2.6119e-01, -1.1473e-01, -4.6630e-02, -7.1912e-01,\n",
       "           2.3983e-01,  1.3843e-02,  5.2805e-02,  4.4453e-01, -5.6120e-01,\n",
       "          -5.8281e-02,  4.2175e-01, -6.9438e-03,  9.3467e-01, -2.3796e-01,\n",
       "          -1.8945e-01,  1.3743e-01,  6.5932e-02,  4.1065e-01, -2.9866e-01,\n",
       "           7.2111e-01,  5.4093e-02,  3.1360e-01,  2.1983e-01,  3.8153e-01,\n",
       "          -1.1718e-01,  1.4883e-01, -5.6135e-01, -2.6654e-01, -2.7610e-01,\n",
       "           3.9481e-03, -3.0586e-01,  6.2366e-01,  4.0539e-01, -1.3481e-01,\n",
       "          -2.1925e-01,  1.1953e-01,  5.0661e-01, -4.9889e-01,  1.4443e-01,\n",
       "           1.0496e-02,  7.6254e-01,  4.7347e-02,  1.2913e-01,  2.7537e-02,\n",
       "           2.1706e-01, -7.0500e-01, -4.7582e-01, -2.7160e-01,  2.2755e-01,\n",
       "           6.1030e-04,  2.0272e-01, -1.8811e-01, -1.4186e-01,  4.0872e-01,\n",
       "          -5.5364e-01,  2.2408e-01, -1.4214e-01, -7.3817e-01, -2.6348e-01,\n",
       "           3.3931e-01,  1.4292e-01, -2.2960e-01, -2.6672e-01, -2.5289e-01,\n",
       "          -2.3525e-02,  4.1953e-01,  1.5162e-01,  1.4770e-01, -3.0398e-03,\n",
       "           3.4429e-01,  1.5448e-01, -3.5274e-01, -3.7306e-01,  4.4490e-01,\n",
       "          -3.6430e-01,  4.3425e-01, -8.4080e-02, -1.8731e-01, -1.9265e-01,\n",
       "          -6.1049e-02,  4.0043e-01,  8.6827e-02,  8.8159e-02, -3.7495e-01,\n",
       "          -4.3747e-01,  1.9772e-01, -1.4565e-01, -5.7050e-01,  3.6462e-02,\n",
       "          -1.7092e-01, -7.0719e-02,  2.9210e-01, -7.4185e-01, -3.1999e-01,\n",
       "          -4.1333e-02,  1.8365e-01, -2.5661e-01, -3.6474e-01,  6.1925e-01,\n",
       "           3.0277e-01, -6.0844e-01,  1.8690e-01,  2.4292e-01,  2.1558e-01,\n",
       "          -3.0072e-01,  2.6922e-01, -1.1955e-01,  1.8721e-01,  4.2216e-01,\n",
       "          -2.8266e-01,  1.9653e-02, -1.2952e-01, -1.2699e-01, -1.0186e-01,\n",
       "          -7.9259e-02,  3.1118e-01, -7.3181e-01,  2.9215e-01,  1.3941e-01,\n",
       "          -1.3217e-01,  2.9565e-02, -4.7224e-01, -4.4789e-02,  2.0047e-01,\n",
       "          -9.3192e-02,  2.0226e-01, -5.4880e-03, -4.0916e-01, -1.6269e-01,\n",
       "          -6.9393e-01, -5.1840e-01,  4.2203e-01,  7.7163e-02,  5.3019e-02,\n",
       "           1.7631e-01,  1.9299e-01,  5.1817e-02, -1.7905e-01,  1.1806e-01,\n",
       "           6.4804e-01,  3.9183e-03,  2.9397e-01, -3.6751e-01,  1.2895e-02,\n",
       "           3.2534e-02,  2.3527e-01, -2.2775e-01,  2.5041e-01,  1.6326e-01,\n",
       "           7.2998e-02, -4.6181e-01,  3.1106e-02, -4.0375e-01,  1.0158e-01,\n",
       "          -4.1834e-01,  7.7058e-02,  3.3918e-01, -2.2947e-01, -5.9240e-02,\n",
       "          -2.1314e-02, -1.8483e+00,  4.5683e-02, -2.5595e-01,  2.6390e-01,\n",
       "          -3.0407e-01,  4.5791e-01, -7.6800e-03,  3.3370e-01, -2.6174e-01,\n",
       "          -2.1348e-01, -2.4916e-01,  5.2349e-03,  2.5424e-01,  1.7514e-01,\n",
       "           3.5905e-01,  6.5585e-01,  3.0718e-01, -3.1080e-01, -2.1680e-01,\n",
       "          -4.6333e-01,  9.3005e-02, -1.3645e-01,  5.7550e-01,  1.4159e-01,\n",
       "           2.9845e-01, -4.4450e-01, -7.4372e-02,  3.4584e-01, -1.7871e-01,\n",
       "           2.8515e-02, -3.6597e-01, -4.3026e-01, -2.0238e-01, -2.0622e-01,\n",
       "           5.3534e-01, -4.3609e-02, -1.4388e-01,  2.8290e-01,  3.5539e-01,\n",
       "          -2.0161e-01, -1.5003e-01, -4.8581e-02, -4.7491e-01, -2.4064e-01,\n",
       "           1.5279e-01, -2.0908e-01, -6.4798e-02, -4.1998e-01, -9.7233e-02,\n",
       "           1.5801e-01, -5.1636e-01,  7.5361e-01,  3.4870e-01, -2.5193e-01,\n",
       "          -3.7018e-01, -2.3858e-01,  4.1901e-02, -3.9680e-01,  7.9643e-02,\n",
       "          -4.0522e-01, -7.3651e-01, -7.1697e-01,  5.1355e-02,  1.9663e-01,\n",
       "           6.7605e-02, -3.0550e-01,  1.0826e+00,  2.4535e-01,  1.7938e-01,\n",
       "          -1.9605e-01, -7.9267e-02,  1.7770e-01, -2.0493e-02, -1.6561e-02,\n",
       "          -6.8134e-02,  6.6405e-02, -1.8555e-01,  5.1919e-01, -4.2570e-01,\n",
       "          -3.7851e-02, -3.3567e-02, -1.2244e-01,  4.4230e-01, -2.7502e-01,\n",
       "           6.1031e-01,  2.1376e-01, -6.9119e-01, -7.8894e-02,  2.3231e-01,\n",
       "          -5.8821e-01,  4.3038e-01,  3.8286e-01, -4.0441e-01, -5.1756e-01,\n",
       "          -8.1046e-01,  9.1969e-02, -3.9989e-01, -1.7811e-02, -9.7618e-01,\n",
       "           1.6797e-02, -1.2557e-01,  2.1986e-01,  1.7411e-01, -3.2327e-02,\n",
       "           7.8134e-02,  3.8281e-01, -5.2944e-02, -8.2577e-02, -5.4203e-01,\n",
       "           6.5732e-03,  3.5112e-01, -3.0607e-01,  3.9488e-01,  2.4681e-01,\n",
       "          -1.4167e-01,  6.0896e-03,  1.2175e-01,  8.2526e-02, -1.9617e-01,\n",
       "          -3.2019e-01, -1.9687e-01, -9.8245e-02, -1.7265e-01, -1.3268e-01,\n",
       "           1.3543e-01,  1.6066e-01,  9.2753e-01, -2.0777e-01, -4.9319e-01,\n",
       "           1.0971e-01,  4.7147e-01, -1.2898e-01,  1.1297e-01, -3.5562e-01,\n",
       "          -1.4190e+00, -1.8720e-01, -1.2377e-01, -3.1198e-02, -2.7850e-01,\n",
       "           4.8128e-01,  1.3939e-01,  2.1192e-01, -5.5510e-02, -3.2444e-01,\n",
       "           4.1317e-01,  2.5505e-01,  2.0660e-02, -1.7565e-01, -7.0306e-02,\n",
       "          -1.5080e-01, -2.3740e-02, -6.3960e-02, -2.9343e-02,  3.4454e-01,\n",
       "           2.3612e-01,  4.1912e-01,  2.4774e-01,  2.7942e-02, -3.5550e-01,\n",
       "           8.2605e-01, -1.7123e-01, -2.3247e-01,  5.3222e-01, -4.7600e-01,\n",
       "          -4.6041e-01,  3.9855e-01,  2.6403e-01, -1.4477e-01,  1.3391e-01,\n",
       "           4.1646e-01,  1.9749e-01, -2.3777e-01,  1.1411e-01,  2.9440e-01,\n",
       "           4.8666e-01,  4.8244e-01, -2.1270e-01,  5.5613e-01,  7.9894e-01,\n",
       "          -1.1233e-01,  1.3821e-01, -1.2602e-02, -1.3257e-01, -1.1910e-01,\n",
       "           2.9295e-01,  3.1526e-01, -5.4161e-01,  6.8883e-02, -1.0962e-01,\n",
       "           1.6683e-01, -5.3993e-01, -1.5984e-01,  1.2706e-01,  2.3646e-01,\n",
       "          -3.2021e-01, -4.2415e-01,  8.1832e-01, -4.3098e-01, -5.3917e-01,\n",
       "           2.6176e-02,  1.6421e-01,  4.9367e-02, -2.2133e-01, -7.7805e-01,\n",
       "          -3.2374e-01,  8.8946e-02,  8.2102e-01,  2.0789e-01,  2.0069e-01,\n",
       "          -1.7580e-01, -8.6511e-02, -2.8754e-01,  3.0112e-01, -4.6145e-02,\n",
       "          -1.2611e-01,  2.2645e-01, -2.4535e-01,  1.6982e-01,  2.6074e-01,\n",
       "          -1.7538e-01, -1.2990e-01, -2.0046e-02,  1.2391e-01, -2.1355e-01,\n",
       "          -4.2072e-01,  1.2400e-01,  4.3528e-01, -1.6031e-01,  5.0517e-02,\n",
       "           7.5126e-02,  2.7463e-01, -1.4845e-01,  1.1507e-01,  1.1176e-01,\n",
       "           3.4408e-01, -2.6469e-01, -9.9935e-02, -1.1457e-01, -7.7510e-01,\n",
       "           5.8753e-01,  3.5017e-01,  5.7762e-01,  1.9253e-01, -1.1207e+01,\n",
       "           4.3065e-01,  6.2007e-01, -6.0640e-01,  2.2182e-03,  1.2837e-01,\n",
       "           1.2923e-01, -5.2519e-01, -1.0567e-01, -4.4544e-01, -3.0250e-01,\n",
       "           9.8305e-02,  6.7697e-02,  1.4094e-01, -2.6816e-01,  8.9631e-02,\n",
       "          -2.4555e-01, -9.4602e-01,  2.6917e-01,  2.3038e-01,  2.2556e-01,\n",
       "           1.0170e-01,  1.7094e-01, -1.3086e-01,  6.8527e-02, -5.9959e-01,\n",
       "           1.8633e-01, -4.2963e-01,  4.0315e-03,  1.6478e-01, -1.3790e-01,\n",
       "           1.8061e-01,  3.4425e-01,  2.2897e-01, -3.3746e-01, -5.4996e-01,\n",
       "           1.1886e-01,  2.2965e-01,  1.8025e-01,  1.1416e-01, -1.2993e-01,\n",
       "          -6.1662e-02, -3.0087e-01,  2.8323e-01, -1.6089e-01,  7.3798e-01,\n",
       "           8.4623e-01,  8.6409e-01, -2.0904e-02,  3.2439e-02, -1.1059e-01,\n",
       "          -8.4712e-01,  3.6877e-01, -1.0788e+00,  3.0544e-01,  2.0325e-01,\n",
       "           6.2066e-01, -7.0502e-01,  6.6395e-02, -2.0172e-01, -4.6060e-01,\n",
       "           3.0533e-01, -2.6946e-04,  5.0181e-01, -2.1084e-01, -3.0979e-01,\n",
       "           3.1959e-01,  1.6816e-01,  3.4269e-01, -1.5653e-01,  3.9434e-01,\n",
       "          -2.9682e-01,  4.1430e-01,  4.0736e-01, -7.7463e-01, -9.1483e-02,\n",
       "           8.3009e-02,  1.5028e-01,  6.3094e-01,  5.5129e-02, -6.6208e-02,\n",
       "           1.1210e-01,  3.5883e-01, -5.9032e-01, -6.2614e-01, -2.9249e-02,\n",
       "           1.9467e-01,  1.4047e-01,  2.0442e-01,  7.0878e-01, -2.2115e-01,\n",
       "          -6.1920e-01, -1.0030e-01, -1.9728e-01,  1.7532e-01,  6.3285e-01,\n",
       "          -4.3159e-01, -4.6845e-02,  1.4070e-01,  4.3348e-02, -7.4768e-01,\n",
       "          -2.0005e-01, -3.3668e-02, -2.8464e-01,  4.3781e-01,  1.0388e-01,\n",
       "           5.6217e-02,  1.2794e-01,  1.0908e-01,  1.6960e-01, -6.0826e-01,\n",
       "           5.5654e-01, -6.1068e-02,  6.5867e-02, -1.8301e-01, -5.3224e-01,\n",
       "          -1.9937e-02,  3.5888e-01, -3.6460e-01, -4.7315e-01, -3.7036e-01,\n",
       "           7.6572e-02,  1.8403e-01, -1.2566e-01, -4.2222e-01,  2.5333e-01,\n",
       "          -5.6398e-01, -1.7998e-02, -2.7319e-01,  1.0585e-01,  2.8376e-01,\n",
       "           1.1761e-01,  7.6720e-01,  3.2489e-01, -1.5523e-01, -8.3045e-01,\n",
       "          -7.3913e-01,  2.4104e-02,  1.0286e-01, -2.8027e-01, -1.0998e-01,\n",
       "           2.1459e-02, -1.1331e-01, -1.2950e-01,  1.0336e-01, -2.5993e-01,\n",
       "           6.0969e-01,  4.3006e-01,  5.0938e-03, -1.5371e-01, -2.2367e-01,\n",
       "          -1.7876e-01, -4.3676e-01, -1.9685e-01, -4.9070e-02, -4.4307e-01,\n",
       "          -2.7480e-01,  2.5340e-02, -3.9396e-01,  4.3587e-01, -1.2331e-01,\n",
       "           6.2844e-02, -2.1139e-01, -1.0869e-01,  1.3021e-01,  2.4948e-01,\n",
       "           1.4432e-01, -2.0088e-01, -8.7266e-01,  1.2105e-03,  1.0548e-01,\n",
       "           8.6826e-02, -2.6941e-01, -1.6187e-01, -5.3528e-01,  1.4577e-01,\n",
       "          -1.0671e-01,  1.0990e+00, -1.9868e-01, -4.3621e-01,  3.2971e-01,\n",
       "           7.4673e-01, -1.5844e-01,  1.6846e-01, -1.6015e-01,  3.3279e-01,\n",
       "          -1.2037e-01, -3.9456e-01,  3.5135e-01,  1.1294e-01,  3.3777e-01,\n",
       "           4.6435e-01,  1.7662e-01, -1.6769e-01,  1.3047e-01, -3.6915e-01,\n",
       "          -3.2789e-01, -1.2379e-01, -1.2752e-01, -9.3829e-03,  2.7743e-01,\n",
       "           8.5338e-02, -3.6565e-02, -6.1144e-02,  1.5004e-01,  2.4275e-01,\n",
       "           1.8513e-01,  1.1725e+00, -1.1618e-01, -6.6647e-03,  2.1380e-01,\n",
       "           1.4384e-01, -4.1668e-01,  1.4292e-01,  1.3996e-01,  2.2563e-01,\n",
       "           3.9302e-01,  1.5897e-01, -3.9491e-01,  1.0915e-01,  1.3083e-01,\n",
       "           1.6617e-01, -6.4212e-01, -4.0982e-01, -2.3292e-01,  5.4243e-01,\n",
       "           2.6012e-01,  1.5623e-01, -3.0076e-01,  5.1075e-01,  5.9605e-01,\n",
       "          -9.5970e-01,  1.4219e-02, -2.1583e-01,  5.5636e-01,  2.5232e-02,\n",
       "          -6.3485e-01, -3.0340e-01, -2.3314e-01, -1.8778e-01,  3.7244e-01,\n",
       "           3.7025e-01, -8.5616e-01,  1.7786e-01,  6.3513e-02, -4.7794e-01,\n",
       "           1.4626e-01, -2.2941e-01,  3.6366e-01,  2.4091e-01, -2.1493e-01,\n",
       "          -2.9202e-01,  3.5941e-01, -1.8032e-01, -5.8955e-01,  5.7702e-02,\n",
       "           5.2778e-01, -1.3529e-01, -1.0543e-01, -4.5490e-02, -5.0637e-01,\n",
       "           4.3923e-02, -2.3218e-01, -1.9070e-01, -1.1636e-01, -1.7815e-01,\n",
       "          -8.8682e-02, -4.2483e-01,  1.3395e-01,  4.7818e-01, -4.5795e-01,\n",
       "           6.8551e-01,  7.2864e-01,  4.2991e-01, -2.3582e-01,  2.5261e-01,\n",
       "          -1.0374e-01,  1.9663e-01, -1.8388e-01,  6.6632e-01, -8.3145e-02,\n",
       "          -5.6816e-01,  5.3113e-01,  9.5585e-02, -1.9265e-01, -1.9473e-01,\n",
       "          -3.9884e-01,  3.5734e-01,  2.0654e-01, -2.5408e-01,  3.6258e-01,\n",
       "           3.0309e-01, -1.4590e-03, -4.8905e-01,  2.4644e-01,  7.0648e-02,\n",
       "          -3.6446e-01,  9.9253e-01, -2.2580e-02,  6.2195e-01, -6.9831e-02,\n",
       "           3.3236e-01, -5.8784e-01, -6.0821e-01,  1.0415e-01, -2.9728e-02,\n",
       "          -4.2693e-01,  1.6494e-04,  4.0958e-01,  7.1183e-02,  6.9952e-01,\n",
       "           6.1421e-01, -3.3958e-01, -7.3405e-01, -1.3101e-01,  8.0231e-02,\n",
       "           2.6322e-01, -2.6491e-01,  3.5553e-01, -3.7240e-01,  2.3845e-01,\n",
       "          -9.3591e-01, -5.5461e-01,  1.3333e-01,  1.1252e-01, -5.6967e-01,\n",
       "           4.0954e-01, -2.2807e-01,  6.0493e-02,  6.3642e-01,  2.4896e-01,\n",
       "           5.1331e-01,  4.4721e-01,  1.4418e-01, -6.3073e-01,  5.7258e-02,\n",
       "           1.3861e-01, -1.8973e-01,  1.0174e-01, -3.9414e-01,  4.1617e-01,\n",
       "           1.2926e-01,  5.3336e-01,  7.5406e-01]], device='mps:0'),\n",
       " tensor([[-5.9986e-01, -5.8071e-01, -4.4628e-01, -9.4837e-01, -2.2188e-01,\n",
       "          -2.4522e-01, -4.5167e-01,  4.3818e-01, -1.0890e-01,  6.1997e-01,\n",
       "           1.7055e-01,  3.6053e-01,  2.2687e-01,  1.3967e-01, -3.4491e-01,\n",
       "          -1.7004e-01,  2.8656e-01,  4.2862e-01,  6.6314e-01,  1.9541e-01,\n",
       "           3.8983e-01,  5.7991e-01, -2.7285e-02,  1.2275e-01, -6.3986e-01,\n",
       "          -3.8570e-02, -6.4205e-02, -2.0677e-01,  8.6848e-03, -3.3889e-01,\n",
       "           3.3805e-02,  5.6181e-01, -1.2109e-01,  1.3670e+00, -3.6849e-01,\n",
       "          -2.5548e-01,  8.0624e-02, -6.8563e-01, -2.5481e-01, -1.6507e-01,\n",
       "           5.7922e-01, -4.2658e-01,  6.3166e-01,  2.3412e-01, -9.0888e-02,\n",
       "           1.2489e-01,  2.3260e-01, -7.8522e-01, -1.4019e-01, -8.9620e-01,\n",
       "          -2.0266e-01, -3.6797e-01,  1.8880e-01,  4.3463e-01, -7.4965e-01,\n",
       "          -6.6265e-01,  3.6612e-01,  2.3338e-01, -7.7975e-02, -6.5850e-02,\n",
       "           2.8344e-03,  2.6418e-01, -2.2152e-01,  9.0237e-02,  2.3685e-01,\n",
       "           1.5629e-01, -4.2910e-01, -7.1475e-01, -2.9186e-01, -2.3040e-01,\n",
       "          -1.3608e-01, -1.4039e-01,  6.7188e-02, -1.9737e-01,  1.7806e-01,\n",
       "          -1.0404e-01,  5.1380e-01, -6.8830e-02, -2.5658e-01,  1.0894e-01,\n",
       "           4.8938e-02,  2.0825e-01,  1.6417e-01, -8.3588e-02, -2.4607e-02,\n",
       "           1.1586e-01,  7.1500e-01,  8.4506e-02, -1.2176e-01,  2.2346e-01,\n",
       "          -9.5694e-02, -1.4458e-01, -2.4290e-01, -3.0207e-01,  5.0057e-01,\n",
       "           1.7735e-01,  4.8836e-01, -4.4711e-01,  4.0626e-01, -1.1815e-01,\n",
       "           4.8650e-02, -9.3685e-02, -1.8465e-01, -6.6936e-02, -5.3100e-01,\n",
       "          -4.1456e-01,  7.1318e-03, -6.0075e-02, -2.8061e-01,  2.7221e-01,\n",
       "           3.9479e-02, -2.3094e-01, -2.0551e-02, -4.6682e-01, -2.1256e-01,\n",
       "           4.1566e-02,  4.1221e-01, -1.2608e-02, -5.6256e-01,  2.0088e-01,\n",
       "           2.6442e-01, -3.6168e-02,  2.2522e-01,  1.3851e-01,  2.6032e-01,\n",
       "          -2.9102e-01,  5.8988e-01, -2.0270e-01,  2.3524e-02,  2.2378e-01,\n",
       "          -1.2902e-01,  1.5941e-01, -4.6271e-01,  8.4240e-02,  2.7372e-01,\n",
       "          -4.5096e-02, -1.1628e-01,  4.3517e-02, -4.1653e-01, -2.7984e-01,\n",
       "           3.5924e-01, -1.6150e-02, -4.9139e-01, -5.2152e-01, -9.1633e-02,\n",
       "          -4.2473e-01,  1.0916e-01, -1.4203e-01, -1.9560e-01, -7.8867e-01,\n",
       "          -7.0640e-01, -2.0061e-01,  1.1558e+00, -1.0171e-01, -1.2236e-01,\n",
       "          -2.5873e-01,  4.8501e-01,  8.1900e-02,  1.0112e-01,  3.9156e-02,\n",
       "           6.9647e-01, -1.8913e-02, -1.4151e-01, -7.4296e-01, -1.4178e-01,\n",
       "          -4.3916e-02,  4.7051e-02, -6.6918e-01,  4.0712e-01,  2.2043e-01,\n",
       "           9.4892e-02,  7.5169e-02,  2.3205e-01, -2.0188e-01, -3.8960e-02,\n",
       "          -1.7136e-01,  4.3342e-01,  8.7772e-02, -2.7035e-01,  3.5288e-02,\n",
       "           1.4378e-01, -1.2897e+00,  4.6665e-01, -7.4589e-01, -4.8585e-01,\n",
       "          -2.4211e-01, -1.4207e-01,  4.1869e-02,  4.8758e-01, -1.4803e-01,\n",
       "           1.9242e-01,  5.8966e-01,  2.4058e-01,  2.6846e-01, -1.3921e-01,\n",
       "          -4.4660e-02,  5.5277e-01,  4.8228e-01, -1.8062e-03,  2.3137e-01,\n",
       "          -2.9311e-01,  2.0736e-01, -1.2034e-01,  2.1864e-01,  1.1130e-01,\n",
       "           3.6631e-01, -1.4582e-01,  1.4176e-01,  1.7793e-01, -1.6939e-01,\n",
       "           1.1129e-01, -3.0893e-01, -2.5650e-01,  6.1634e-03, -9.0794e-02,\n",
       "           3.2502e-01, -4.7708e-01, -1.4587e-01,  5.8896e-01,  5.7457e-02,\n",
       "           1.7198e-01,  5.1557e-02,  3.4031e-01,  6.4284e-03,  2.2144e-01,\n",
       "           4.4961e-01, -3.0284e-01,  9.0476e-03, -9.8446e-02, -3.5095e-02,\n",
       "           8.6274e-04, -6.3861e-01,  5.9653e-01,  6.9014e-01, -8.3351e-02,\n",
       "           2.1762e-01, -2.1398e-01,  2.9664e-01, -2.7373e-01,  2.0512e-01,\n",
       "          -1.0999e-01, -5.2604e-01, -4.4272e-01,  6.4359e-02,  2.8096e-01,\n",
       "           4.8166e-01, -1.8238e-01,  7.7372e-01,  7.7686e-01,  2.4199e-01,\n",
       "          -2.8317e-02,  1.8981e-01,  2.5127e-01,  1.9115e-01, -2.0013e-01,\n",
       "          -3.8745e-03,  1.5356e-02, -8.0386e-01,  4.2256e-01, -3.2633e-01,\n",
       "          -3.7168e-01,  6.6421e-03, -2.2219e-01,  3.2776e-01, -1.7996e-01,\n",
       "           1.3581e-01,  3.3111e-01, -1.7782e-01, -2.9296e-01,  3.2896e-01,\n",
       "          -3.1901e-01, -4.1062e-02, -1.1323e-01, -2.2944e-01, -6.6221e-01,\n",
       "          -5.9880e-01,  2.3148e-01,  4.3017e-01,  1.1508e-01, -1.7638e-01,\n",
       "          -2.1126e-01, -4.4052e-02, -1.6650e-01, -3.3492e-01, -3.8023e-01,\n",
       "           4.3403e-01, -4.5984e-01, -3.3340e-01, -2.7603e-01, -5.7840e-01,\n",
       "          -5.8306e-01,  6.3763e-02, -2.9589e-01, -9.6459e-02,  3.4789e-01,\n",
       "          -4.9175e-02,  3.9024e-01,  4.0372e-01,  6.2507e-02, -6.1863e-01,\n",
       "           1.4525e-01,  1.8597e-01,  2.7145e-01, -5.9086e-02,  2.4495e-01,\n",
       "           2.2551e-01,  2.3777e-01,  4.5625e-01, -4.3213e-01, -5.0559e-01,\n",
       "          -2.9505e-02,  4.1987e-01, -1.0874e-01, -9.6742e-02, -8.5808e-02,\n",
       "          -1.1752e+00, -4.3661e-01, -4.5948e-01, -1.5567e-01, -8.4134e-01,\n",
       "           4.8342e-02,  5.2300e-01,  4.6483e-02,  2.6314e-01,  2.5546e-02,\n",
       "           6.2046e-02,  8.9853e-01, -2.0643e-01,  5.1348e-01, -3.3940e-01,\n",
       "          -2.0692e-01, -1.7875e-01, -5.1900e-01, -3.9995e-01, -2.9075e-01,\n",
       "           4.8032e-01,  4.7372e-01,  1.3095e-01, -5.3437e-01, -1.6496e-01,\n",
       "           5.2851e-01, -3.9292e-01,  1.4314e-01,  4.8007e-01, -2.5479e-01,\n",
       "           3.4619e-01,  3.3901e-01,  9.0033e-02,  9.2645e-02,  3.3060e-01,\n",
       "           7.8486e-02, -1.1409e-01, -4.7858e-01, -1.1296e-01,  5.4602e-01,\n",
       "           4.6376e-01,  2.8676e-01,  2.2057e-01,  1.0901e-01,  3.3245e-01,\n",
       "          -2.4023e-01,  4.4442e-01, -4.0759e-01, -2.5812e-01,  3.5659e-01,\n",
       "          -3.4919e-02,  2.7185e-01, -4.6329e-01,  1.7144e-01,  9.3924e-02,\n",
       "          -2.1936e-01, -3.6436e-01,  2.1228e-01, -1.7398e-02,  2.2033e-01,\n",
       "          -4.0648e-02, -4.8108e-01,  3.9207e-02, -5.6739e-01, -9.4617e-02,\n",
       "           1.5975e-01, -1.2313e-01,  3.9231e-01, -5.3728e-01, -6.6231e-01,\n",
       "          -2.3772e-01, -2.4685e-01,  7.6016e-01,  2.4157e-01,  1.4446e-01,\n",
       "           3.4416e-01,  1.5863e-02, -5.7323e-02,  8.8062e-01,  3.3542e-01,\n",
       "          -1.4566e-01,  2.0275e-01, -2.6462e-01,  1.8027e-01,  6.2857e-01,\n",
       "          -2.8791e-01, -1.3841e-01,  1.8741e-01, -2.2345e-01,  2.3410e-01,\n",
       "          -1.8064e-01, -1.3175e-01,  1.7376e-01, -3.4965e-01, -1.1742e-01,\n",
       "          -1.2156e-01,  2.7231e-01, -2.1691e-01,  1.4711e-01, -4.5798e-02,\n",
       "           9.9564e-02, -1.1695e-01, -1.8197e-01,  2.1748e-01, -6.1760e-01,\n",
       "           1.6528e-01, -3.9170e-02,  1.8686e-01,  3.7895e-01, -1.1254e+01,\n",
       "           2.6038e-01,  9.8944e-01, -2.4246e-01, -3.8202e-02, -2.2010e-01,\n",
       "           2.1267e-01, -2.5252e-01,  1.6194e-01, -9.0646e-02, -4.1135e-01,\n",
       "          -5.1867e-01, -2.3361e-01,  1.7279e-01, -3.4155e-01, -2.7158e-01,\n",
       "          -3.8114e-02, -7.5712e-01, -1.2757e-01, -5.1241e-03,  3.9250e-01,\n",
       "          -1.2016e-01, -3.5186e-01,  2.3144e-01, -2.0463e-01, -2.9645e-01,\n",
       "          -1.6739e-01, -2.9232e-01,  1.2473e-01,  5.0514e-01, -6.4091e-02,\n",
       "           8.9342e-02,  1.7904e-01,  3.9205e-01, -9.6837e-02, -7.3186e-01,\n",
       "           1.0360e-01, -1.2176e-02,  2.3037e-01,  1.4088e-01, -2.9825e-01,\n",
       "          -2.2175e-01, -2.7830e-01,  1.8859e-01,  2.9146e-01,  8.1229e-01,\n",
       "           7.0421e-01,  1.1076e+00,  3.9757e-01,  4.4792e-02, -8.3668e-02,\n",
       "          -7.2496e-01, -6.7451e-02, -8.1609e-01,  1.8780e-01,  2.0281e-01,\n",
       "           5.4801e-01, -4.7107e-01,  1.4166e-01, -4.5254e-01,  1.7928e-02,\n",
       "           2.9979e-01, -7.0214e-02,  3.5143e-01, -8.1181e-02, -3.1965e-01,\n",
       "           4.2025e-01,  8.7954e-02,  1.0001e-01, -3.5756e-01, -2.3236e-01,\n",
       "           8.2218e-02,  2.4489e-01,  3.2222e-01,  3.8734e-01,  6.7507e-01,\n",
       "          -1.4487e-02, -3.9065e-01,  1.3785e+00, -7.0608e-01,  4.7467e-02,\n",
       "          -2.5235e-01,  2.7406e-01, -3.9264e-01, -8.5014e-01,  3.1788e-01,\n",
       "          -2.5193e-01, -3.2085e-01,  7.0430e-01,  3.4121e-01, -2.2194e-01,\n",
       "          -3.9095e-01,  5.1890e-02,  5.8582e-02,  6.0583e-02,  4.0850e-01,\n",
       "          -2.1762e-01, -3.0236e-01,  5.3737e-03,  3.2022e-01, -1.0583e+00,\n",
       "          -6.2216e-02, -2.5567e-01, -1.8609e-01,  2.7690e-01, -2.7576e-01,\n",
       "          -1.2412e-01,  3.2938e-01,  6.6555e-02,  6.5445e-01, -1.6146e-01,\n",
       "           3.9251e-03,  6.1427e-02, -1.1930e-02, -3.7871e-01, -8.0631e-01,\n",
       "           2.7528e-01,  5.6966e-01, -3.7859e-01, -8.2134e-01, -1.4595e-01,\n",
       "          -1.1708e-01,  8.5684e-02,  2.3751e-01, -4.8229e-01,  3.5016e-01,\n",
       "          -7.4996e-01, -1.5208e-01,  4.7453e-01, -6.7663e-02,  6.3427e-01,\n",
       "           1.4394e-01, -3.4639e-01,  1.4120e-01, -4.3305e-01, -1.9071e-01,\n",
       "          -7.7708e-01, -3.9132e-01,  2.0163e-01, -3.6706e-01,  2.5267e-01,\n",
       "           1.2395e-01, -1.3826e-02, -1.4353e-01,  4.9503e-02,  1.1154e-01,\n",
       "          -2.6673e-01,  1.8547e-01,  2.3176e-02,  9.9754e-02,  9.2615e-02,\n",
       "          -1.5581e-03, -2.3249e-01, -1.9478e-01, -1.1495e-01, -4.8389e-01,\n",
       "          -1.3852e-01, -8.8434e-03, -2.5839e-01, -1.9299e-01,  2.9182e-01,\n",
       "           1.0305e-01,  8.9879e-02, -3.9552e-01,  8.1877e-01,  1.5140e-01,\n",
       "          -1.0440e-01, -2.0240e-01, -1.9365e-01, -2.2718e-01,  2.4061e-01,\n",
       "           2.4864e-01, -3.1960e-01, -3.8043e-01, -3.2884e-01, -5.2058e-01,\n",
       "          -1.6539e-02,  8.8023e-01, -3.9396e-01, -3.7615e-01,  4.1426e-01,\n",
       "           6.0415e-01, -2.2164e-02,  6.2259e-02,  4.4959e-03, -1.4484e-01,\n",
       "          -1.5212e-01, -3.8485e-01,  5.8360e-01, -2.1811e-02,  1.8711e-01,\n",
       "           5.3818e-01,  4.5634e-01,  2.3414e-01,  1.7039e-01, -3.3867e-01,\n",
       "          -5.7250e-01, -7.4190e-02, -2.9400e-01,  7.2116e-02,  3.4898e-02,\n",
       "          -2.3751e-01,  2.8734e-02,  1.6718e-01, -3.5497e-01,  7.4745e-02,\n",
       "           7.4725e-01,  7.3856e-01, -1.7929e-01,  2.5100e-02,  2.6173e-01,\n",
       "           3.6917e-01, -5.7477e-02, -1.6215e-01,  3.0763e-01, -1.1986e-01,\n",
       "           3.2669e-01, -1.6222e-01, -1.3975e-01, -9.7786e-02,  3.5065e-01,\n",
       "           3.2258e-01, -5.4643e-01, -3.7769e-01, -2.5287e-03,  4.3014e-01,\n",
       "          -4.5815e-01,  3.5067e-01, -3.6820e-02,  2.7189e-01,  1.2290e-01,\n",
       "          -9.0975e-01, -2.1728e-01,  2.5947e-01, -5.0382e-01, -3.3665e-01,\n",
       "          -1.4679e-01, -2.9161e-01, -4.9621e-01, -1.9801e-01,  5.6206e-01,\n",
       "           5.8575e-01, -8.0312e-01,  2.0735e-01,  6.1602e-01, -3.9550e-01,\n",
       "           3.1894e-01, -2.3266e-01,  2.5177e-01,  5.5562e-01, -2.2694e-02,\n",
       "          -6.4449e-02,  7.7840e-02, -2.6984e-01, -4.0713e-01,  3.1932e-02,\n",
       "           8.4565e-01,  5.4950e-02, -2.1370e-01, -4.6435e-01, -1.5990e-01,\n",
       "          -1.0541e-01, -2.0914e-01, -3.0816e-01, -5.7932e-03,  4.8358e-01,\n",
       "          -1.5778e-02, -1.4820e-03, -2.6261e-01,  1.2562e-02, -6.1486e-02,\n",
       "           3.3883e-01,  5.7289e-01,  7.7821e-01,  1.9031e-01, -6.6526e-01,\n",
       "          -2.2225e-02,  5.7024e-01, -4.0762e-01,  8.4612e-01,  2.0897e-02,\n",
       "          -5.9238e-01, -5.4270e-02, -7.1482e-03, -3.1483e-01, -1.1298e-01,\n",
       "          -3.8393e-01,  3.1011e-01, -2.1960e-01, -4.5321e-02,  4.2697e-01,\n",
       "          -4.2276e-02, -4.5525e-02, -8.0352e-01,  8.4910e-02,  5.8983e-01,\n",
       "           4.6830e-02,  6.9970e-02,  2.3678e-01,  1.5260e-01, -2.5731e-01,\n",
       "           2.8113e-02, -4.8932e-01, -4.4836e-02, -6.6732e-02, -1.2363e-01,\n",
       "          -5.4123e-01, -1.9316e-01,  1.2913e-01, -1.0969e-02,  6.8609e-01,\n",
       "           5.8661e-01, -1.3685e-01, -2.6877e-01, -2.0460e-01,  7.8201e-02,\n",
       "           4.5422e-01, -1.5288e-01,  3.4858e-01, -4.3469e-01, -6.3057e-02,\n",
       "          -4.0346e-01, -3.1861e-01,  7.8064e-02,  3.3152e-01, -3.1102e-02,\n",
       "           7.1171e-01, -3.0483e-01,  3.8613e-01,  3.9274e-01,  2.6121e-01,\n",
       "           4.0228e-03,  8.3439e-01,  3.7727e-01, -4.4734e-01, -1.1801e-01,\n",
       "           1.6814e-01, -1.2430e-01,  3.3140e-01, -5.1814e-01,  2.9914e-01,\n",
       "           1.7445e-01,  6.1147e-01,  2.5416e-01]], device='mps:0'),\n",
       " tensor([[-2.3519e-01, -6.3371e-01, -3.9896e-01, -1.0186e+00, -9.4941e-02,\n",
       "          -3.6410e-01, -5.2584e-01,  1.9083e-01, -2.2695e-01,  5.6346e-01,\n",
       "           2.3786e-01,  7.7069e-01,  3.1128e-01, -1.7864e-01, -2.2765e-01,\n",
       "          -3.1705e-02, -4.1422e-01,  6.8270e-01,  2.0461e-02,  2.5459e-01,\n",
       "           1.9514e-01,  3.9301e-01,  3.5860e-02,  1.3255e-01, -3.7996e-01,\n",
       "           3.1960e-01, -9.8958e-02, -4.4341e-02,  1.1014e-01, -6.0124e-01,\n",
       "          -2.9173e-02,  3.4534e-01,  6.3153e-04,  1.1014e+00, -6.8774e-02,\n",
       "          -1.0427e-01,  9.9469e-02, -5.7804e-01,  1.2380e+00, -7.7512e-02,\n",
       "           6.3180e-01,  2.3657e-01,  3.9966e-01,  3.2439e-01,  2.3408e-01,\n",
       "           4.1047e-01,  5.0762e-02, -5.9244e-01,  8.9461e-01, -5.7127e-01,\n",
       "          -2.3595e-01,  2.2244e-01,  4.5986e-01,  4.9957e-01, -7.0989e-01,\n",
       "          -4.6618e-01,  9.7421e-02,  2.3866e-01, -2.3782e-01,  1.0917e-01,\n",
       "           1.0503e-02,  1.9244e-01,  1.6234e-01, -8.7667e-02,  3.7997e-01,\n",
       "          -1.4994e-01, -4.8794e-01, -6.0677e-01, -1.2908e-01, -3.1650e-01,\n",
       "           2.1112e-01,  6.6671e-01,  4.6587e-01, -2.1917e-01, -1.6820e-01,\n",
       "          -3.0276e-01, -2.6932e-02,  2.7908e-01, -4.3165e-01, -1.8088e-01,\n",
       "           1.9407e-01, -8.5480e-02,  3.7844e-01,  9.5261e-02,  1.7946e-01,\n",
       "          -8.0224e-02,  5.2334e-01,  4.6948e-02,  3.4248e-01, -1.0172e-01,\n",
       "           4.6834e-01, -8.7651e-02, -4.0746e-01, -4.8606e-01,  2.4618e-01,\n",
       "           5.2486e-02,  4.7303e-01, -2.1050e-01, -7.5806e-02, -1.5849e-01,\n",
       "          -1.6630e-01,  1.9285e-01, -1.9686e-01,  5.9283e-01, -7.5639e-01,\n",
       "          -6.1828e-01, -6.0258e-03, -1.1869e-01, -1.2987e-01,  1.4105e-01,\n",
       "          -6.9070e-02, -1.2430e-01, -1.6752e-01, -4.2096e-01, -2.3481e-01,\n",
       "           2.7533e-02,  7.0664e-02,  2.5699e-01, -1.9355e-01,  2.5887e-01,\n",
       "           2.8549e-01, -2.9140e-01,  2.3055e-01, -1.5196e-01,  8.2926e-02,\n",
       "          -3.2676e-01,  3.6760e-01, -1.1617e-01,  2.5373e-01,  3.0852e-01,\n",
       "           5.2067e-02,  1.9400e-01, -1.7803e-01, -4.4125e-02, -5.7992e-02,\n",
       "          -6.2594e-02, -3.5072e-03, -5.3249e-01,  3.9039e-01,  8.3587e-02,\n",
       "           2.3886e-01, -2.6332e-01, -2.1664e-01, -9.4093e-02, -3.6729e-01,\n",
       "           7.5407e-02, -1.7772e-01, -4.2243e-01, -1.1474e-01, -4.2612e-01,\n",
       "          -1.1239e+00, -6.7834e-01,  4.3189e-01, -3.5821e-01,  3.1799e-01,\n",
       "          -3.2297e-01,  3.9620e-01,  2.7187e-01, -1.1693e-01,  2.6186e-01,\n",
       "           8.3698e-01, -9.0484e-02,  2.4875e-01, -8.1845e-01, -8.3810e-02,\n",
       "           1.9045e-01, -7.6055e-02, -1.6250e-01,  8.3675e-02, -5.8983e-02,\n",
       "           3.9856e-01, -3.0380e-01,  2.0018e-01, -3.1733e-02,  9.5918e-02,\n",
       "          -6.5778e-01,  3.8481e-01,  1.4684e-01, -1.7515e-01, -3.3264e-01,\n",
       "          -7.0890e-02, -1.3925e+00,  1.5630e-01, -2.1264e-02, -1.3510e-01,\n",
       "          -1.8223e-02,  7.9344e-02,  1.4757e-02,  2.5762e-01, -2.3566e-01,\n",
       "          -2.8826e-01, -5.5773e-02,  4.0725e-02,  4.3680e-01,  2.4141e-01,\n",
       "           6.9008e-01,  2.9695e-01,  2.0432e-01, -1.6986e-02, -4.8535e-02,\n",
       "          -7.0739e-01,  1.5223e-02,  1.2974e-01, -6.1104e-02,  2.2293e-02,\n",
       "           3.2859e-01,  1.8100e-01, -3.6520e-02,  1.8201e-01,  3.5168e-01,\n",
       "          -3.0208e-01, -1.1276e-01, -2.2296e-01, -7.6326e-02, -5.9510e-01,\n",
       "           7.7247e-02, -3.9291e-01,  8.5230e-02,  5.6418e-01,  1.7057e-01,\n",
       "           9.5080e-03,  9.8574e-02,  1.1306e-01, -2.4409e-01,  8.0452e-02,\n",
       "           3.9646e-01, -2.8620e-01, -2.6840e-01,  3.3672e-01, -3.8027e-01,\n",
       "          -6.8344e-02, -3.5444e-01,  5.2986e-01,  4.0494e-01, -1.4672e-01,\n",
       "          -5.9116e-02, -2.6650e-01,  2.8812e-01,  4.7766e-02, -1.1440e-01,\n",
       "          -1.9670e-01, -3.6151e-01, -8.6920e-01,  7.0158e-02, -2.6073e-02,\n",
       "           6.0727e-01,  1.4739e-01,  8.9074e-01,  3.9516e-01,  1.9055e-01,\n",
       "          -3.9342e-01, -2.2987e-02,  3.8868e-01,  5.4269e-03, -4.2464e-01,\n",
       "           4.8969e-01, -1.3321e-01, -5.2717e-01,  6.0358e-01, -1.2439e-01,\n",
       "          -5.9346e-01, -2.6736e-01,  8.9781e-02,  2.7475e-01,  2.4901e-01,\n",
       "          -3.4798e-02,  3.4537e-01,  3.3240e-02, -2.8898e-01,  6.3712e-01,\n",
       "          -4.9811e-01,  2.0182e-01, -1.0242e-01, -3.3093e-01, -6.0300e-01,\n",
       "          -1.1063e+00,  8.0643e-02,  1.5260e-01,  1.6624e-01, -6.8187e-01,\n",
       "           2.9465e-01,  2.9830e-03,  6.5832e-02, -2.1558e-02,  1.7557e-01,\n",
       "           3.2120e-01,  6.1378e-02,  3.4677e-01, -7.3442e-01, -3.4666e-01,\n",
       "           7.9347e-03,  6.0914e-03, -4.5698e-01, -2.1402e-01,  2.8246e-01,\n",
       "           2.3872e-01, -2.4038e-01,  2.9901e-01,  1.0012e-01, -3.8955e-01,\n",
       "          -5.1804e-02,  4.2279e-01,  1.8728e-01, -5.8764e-01,  4.3584e-01,\n",
       "          -3.7265e-02,  3.4849e-01,  5.4026e-01, -9.4126e-01, -3.4182e-01,\n",
       "           9.1396e-02,  6.4538e-01, -7.9455e-02, -3.5486e-01, -2.3750e-01,\n",
       "          -1.1628e+00, -3.6779e-01, -3.9861e-01, -7.8205e-03, -7.7351e-01,\n",
       "           9.5785e-02,  6.3411e-01, -1.3320e-01, -9.2686e-02, -4.5884e-01,\n",
       "           4.0870e-01,  5.1783e-01,  1.9097e-01,  1.3172e-01, -2.2311e-01,\n",
       "           3.3804e-01, -8.0207e-01, -1.5963e-01, -1.5756e-01,  4.9796e-01,\n",
       "           2.3887e-01, -9.5163e-02,  3.0496e-01, -1.3906e-01, -3.2453e-01,\n",
       "           3.1943e-01, -3.1365e-01,  2.1230e-01,  1.7999e-01, -5.2442e-01,\n",
       "           3.2090e-02,  3.7148e-01, -9.8575e-02, -4.4444e-01,  1.5591e-01,\n",
       "           1.8764e-01, -2.1052e-01, -1.7040e-01,  1.0312e-01,  3.5871e-01,\n",
       "           4.4465e-01, -3.9133e-02, -7.2567e-02,  2.1684e-01,  3.1887e-01,\n",
       "          -2.6293e-01,  2.0067e-01, -4.1917e-01, -3.6049e-01,  1.8195e-01,\n",
       "           9.3196e-02,  1.3420e-01, -4.6248e-01, -9.2089e-02,  2.4409e-01,\n",
       "           3.4584e-01, -1.1816e-01,  9.4128e-02,  1.3821e-01,  2.6819e-01,\n",
       "          -4.5113e-01, -2.9048e-01,  3.7013e-01, -3.3869e-01, -1.8053e-01,\n",
       "           1.4875e-01,  1.0612e-01, -8.1610e-02, -3.5766e-01, -3.9509e-01,\n",
       "           1.1346e-01, -4.0654e-01,  5.1584e-01,  1.1625e-01, -6.7147e-02,\n",
       "           5.4858e-01,  9.0857e-02, -5.7833e-01,  7.2298e-01, -8.8219e-02,\n",
       "          -5.3808e-01, -7.8019e-02, -6.7699e-01, -1.9453e-03,  1.8447e-01,\n",
       "          -1.7365e-01, -3.1420e-02,  6.3506e-01,  2.4134e-01,  4.3929e-02,\n",
       "          -6.0979e-01,  1.1633e-01,  1.3870e-01,  3.3333e-02,  6.4044e-02,\n",
       "           2.8829e-01,  2.1046e-01, -2.1454e-01,  7.2217e-02, -4.1837e-01,\n",
       "           2.4538e-01,  1.4237e-01, -1.3511e-02, -1.4005e-02, -3.1026e-01,\n",
       "           4.2512e-01,  3.9391e-01,  2.8842e-01,  4.0465e-01, -1.1093e+01,\n",
       "           1.2324e-01,  8.5030e-01, -4.6468e-01, -1.7704e-01,  1.5362e-01,\n",
       "           2.1652e-01, -4.8169e-01,  2.5172e-01, -4.1415e-01,  2.7277e-02,\n",
       "          -6.0716e-01,  2.7891e-02,  1.3998e-01, -3.4088e-01, -7.2235e-02,\n",
       "           1.6796e-02, -6.4857e-01,  3.1892e-01, -2.6817e-01,  2.9598e-01,\n",
       "          -2.2346e-01,  2.6760e-01,  4.7341e-01, -6.0102e-01, -1.5885e-01,\n",
       "          -5.0118e-01, -4.1849e-01,  3.9901e-01,  4.8130e-01,  3.1760e-02,\n",
       "           1.6640e-01, -2.1786e-01,  2.3033e-01, -9.9465e-02, -5.2744e-01,\n",
       "          -2.0885e-01, -3.2477e-01,  2.2176e-01, -3.3734e-01, -5.4205e-01,\n",
       "          -3.3749e-01, -7.7168e-01,  1.1410e-01,  2.8914e-01,  6.5261e-01,\n",
       "           1.2087e+00,  3.1873e-01, -8.6040e-02, -3.8570e-01, -2.9106e-01,\n",
       "          -1.0073e+00,  2.8618e-01, -5.7262e-01, -5.4699e-02, -2.7942e-01,\n",
       "           3.5488e-01, -7.0110e-01,  1.7907e-01, -7.0943e-01, -4.8413e-01,\n",
       "           2.9894e-01,  2.4673e-01, -2.1047e-01, -3.7920e-01, -3.4121e-01,\n",
       "           3.2177e-01,  1.1175e-03,  7.5296e-03, -1.9650e-01, -1.1444e-01,\n",
       "          -1.3737e-01,  3.9299e-02, -1.0066e-01,  2.4705e-02,  2.7256e-01,\n",
       "          -7.1737e-02, -2.2164e-02,  1.9646e+00,  2.9096e-02,  5.5644e-02,\n",
       "          -1.2761e-02,  7.8561e-02, -4.0744e-01, -3.5399e-01, -5.3691e-02,\n",
       "          -1.5385e-01,  3.5542e-01,  6.7684e-02,  8.5775e-01, -1.5889e-01,\n",
       "          -3.2763e-01,  2.0298e-01,  1.0178e-01,  2.9140e-01,  7.9947e-01,\n",
       "          -8.8464e-01,  1.9690e-01,  1.9176e-01,  1.0403e-01, -7.3814e-01,\n",
       "          -9.7782e-02, -9.6754e-02, -1.7766e-01, -9.9169e-03,  1.5399e-02,\n",
       "           2.7463e-02,  1.0818e-01, -2.9960e-01,  4.8171e-01, -6.7675e-01,\n",
       "           3.5388e-01,  1.6850e-01,  1.8789e-01, -5.1128e-01,  2.1864e-01,\n",
       "          -8.2798e-02,  5.2555e-01, -1.4564e-01, -5.1377e-01, -1.8494e-01,\n",
       "          -2.0897e-01, -1.3103e-01,  1.4477e-01, -2.8716e-02,  3.8990e-01,\n",
       "          -4.6774e-01,  1.3049e-01,  9.0041e-02, -2.8606e-01,  2.7283e-01,\n",
       "           6.6209e-01, -3.7374e-01,  6.5701e-01, -8.3634e-01, -8.5767e-01,\n",
       "          -2.6895e-01, -8.0204e-01,  3.6235e-01,  1.0043e-01,  1.7741e-01,\n",
       "           5.6806e-02,  1.6685e-01, -1.6483e-01,  3.8109e-01, -1.3905e-02,\n",
       "           1.7838e-01,  3.0797e-01, -2.8051e-01,  2.4082e-01, -8.2916e-02,\n",
       "           5.6197e-02,  5.6677e-02, -1.1972e-01, -1.1306e-01, -3.5441e-01,\n",
       "          -6.5374e-01, -1.1587e-01, -2.9301e-01, -1.2123e-01,  2.9440e-01,\n",
       "          -1.5864e-01,  1.5346e-01, -3.7590e-01,  2.6037e-01,  5.0231e-01,\n",
       "          -1.6963e-01, -6.3625e-01, -8.1928e-02, -2.1210e-01, -2.9557e-01,\n",
       "           5.7016e-01,  5.9257e-02, -5.6882e-01, -5.6617e-01, -4.5835e-01,\n",
       "          -1.1816e-01,  1.1832e+00, -3.7872e-01, -5.3999e-01,  1.8885e-01,\n",
       "           2.4026e-01,  1.3029e-01,  1.3243e-01, -3.5065e-01,  3.2850e-01,\n",
       "          -5.3982e-02, -7.5493e-01,  5.3914e-01, -4.5606e-01,  9.5240e-02,\n",
       "           6.0410e-01,  1.2052e-01,  1.1295e-01,  2.9613e-01, -4.9317e-02,\n",
       "          -3.9007e-01,  4.5396e-02, -1.5194e-01,  1.4599e-01,  1.9684e-01,\n",
       "          -3.3648e-01,  2.6243e-01, -1.9767e-02, -1.9612e-01,  2.2976e-01,\n",
       "           9.2581e-03,  1.5046e+00, -1.5516e-01,  2.4871e-02,  7.9564e-01,\n",
       "           1.3917e-02,  3.0821e-01,  6.2921e-01,  1.8131e-01,  1.0842e-01,\n",
       "           5.0243e-01, -3.0356e-01, -1.8693e-01,  2.3025e-01,  1.3750e-02,\n",
       "           3.0597e-01, -4.2537e-01, -1.5688e-01, -7.7421e-02,  7.5666e-01,\n",
       "          -1.7739e-01,  4.7975e-01, -2.4294e-01,  3.2017e-01,  4.2784e-01,\n",
       "          -1.4040e+00,  6.2424e-02, -3.0111e-01,  2.6861e-01, -1.0852e-01,\n",
       "          -4.4028e-01, -1.9086e-01,  2.0205e-01, -1.5534e-01,  4.9482e-01,\n",
       "           1.2247e-01, -7.6575e-01,  1.7275e-01,  8.6826e-01, -5.0838e-01,\n",
       "           3.6167e-01, -7.4507e-02,  4.1481e-01,  4.1777e-01, -9.2003e-02,\n",
       "           2.9110e-01,  9.6200e-02, -4.1398e-01, -1.9687e-01,  2.9120e-01,\n",
       "           4.9325e-01,  7.3612e-02, -3.4189e-02,  4.3856e-02, -1.4902e-01,\n",
       "          -1.0980e-01, -8.8664e-02, -1.1207e-01,  2.0956e-01,  4.1601e-01,\n",
       "          -3.8450e-02, -2.9285e-01, -1.8421e-01,  3.8168e-01, -5.6923e-02,\n",
       "           4.1079e-01,  3.5718e-01,  4.2790e-01, -7.2897e-02, -7.0941e-02,\n",
       "           2.9322e-01,  2.6045e-01,  1.0653e-01,  4.4932e-01, -6.6769e-01,\n",
       "          -7.6971e-01, -7.1048e-01,  5.7672e-02, -4.6665e-01, -1.1900e-01,\n",
       "          -2.8608e-01,  6.4861e-01, -5.5923e-02, -1.9439e-01,  8.5390e-01,\n",
       "          -1.2660e-01,  3.0225e-01, -3.3585e-01,  6.7554e-01,  3.3760e-01,\n",
       "          -3.0843e-01,  5.0248e-01,  1.2138e-01,  4.1639e-01, -3.1148e-01,\n",
       "           1.5618e-01, -8.5468e-01, -1.3177e-01, -5.7319e-03,  8.5468e-02,\n",
       "          -3.0946e-01, -2.5879e-01, -2.9772e-01,  8.2040e-02,  4.1993e-01,\n",
       "           1.0467e+00, -2.2017e-01, -7.3593e-01, -4.4707e-01,  1.7262e-01,\n",
       "           2.5308e-01, -2.0193e-01,  3.0355e-01, -1.6407e-01,  1.4232e-01,\n",
       "          -4.1417e-01, -3.4498e-01, -5.9067e-02,  9.5870e-02, -2.3439e-01,\n",
       "           8.4371e-01, -1.4760e-01,  3.0592e-02,  1.9323e-01,  8.1209e-02,\n",
       "           5.7252e-02,  7.4917e-01,  1.4256e-01, -4.5192e-01, -2.8155e-01,\n",
       "           1.1717e-01, -6.8659e-02,  4.1136e-02, -6.4484e-01,  4.5948e-01,\n",
       "           2.6039e-02,  1.2186e-01,  4.1427e-01]], device='mps:0'),\n",
       " tensor([[-2.9288e-01, -8.7314e-01, -8.3036e-01, -9.0729e-01, -8.9148e-02,\n",
       "           2.0944e-01, -4.8184e-01,  5.6757e-01,  7.9446e-02,  4.4134e-01,\n",
       "           3.3554e-01,  2.9026e-02,  4.0956e-02, -4.8051e-01, -7.7979e-01,\n",
       "          -2.2033e-01,  4.6334e-02,  5.2241e-01,  5.2651e-01,  1.5342e-01,\n",
       "           8.3315e-01,  2.6826e-01, -1.6943e-01,  1.9120e-01, -3.4117e-01,\n",
       "           1.8159e-01,  8.5857e-02, -1.5625e-01,  1.4711e-01, -1.1017e+00,\n",
       "           2.0672e-01,  6.7655e-01, -2.5047e-01,  9.6070e-01, -3.0931e-01,\n",
       "          -5.0099e-02,  7.1713e-02, -2.5040e-01,  3.4745e-01, -2.5904e-01,\n",
       "           8.6024e-01, -1.0390e-01,  3.1400e-01, -3.1817e-02,  3.6707e-01,\n",
       "          -1.6134e-01,  1.7569e-01, -7.6486e-01,  1.0781e-01, -4.0035e-01,\n",
       "          -1.8631e-01, -8.2935e-02,  4.6524e-01,  4.4788e-01, -6.8111e-01,\n",
       "          -3.8431e-01,  2.5490e-01,  4.1637e-01, -6.2341e-01,  1.9573e-01,\n",
       "          -3.2629e-01,  2.6769e-01, -4.5053e-02, -7.2807e-02,  6.7779e-02,\n",
       "           2.7192e-02, -7.3119e-02, -3.8734e-01, -1.8034e-01, -1.5174e-02,\n",
       "           2.9954e-01,  3.1360e-01, -1.9374e-01, -2.7117e-01,  5.7572e-01,\n",
       "          -2.2220e-01,  7.0932e-02,  2.5184e-01, -6.3231e-01,  1.9246e-01,\n",
       "          -1.9254e-01,  2.6830e-01, -2.3702e-01, -2.6101e-01,  1.5371e-01,\n",
       "          -1.1992e-01,  2.8678e-01,  2.4379e-01,  2.0629e-01, -3.7043e-01,\n",
       "           1.9684e-01, -4.3262e-02, -4.6578e-02,  4.4053e-02,  2.9039e-01,\n",
       "           5.2806e-02,  3.3489e-01, -3.5769e-01,  3.6432e-02, -4.3117e-01,\n",
       "          -3.4411e-02,  3.3684e-01, -2.3292e-02, -1.0907e-01, -7.2862e-01,\n",
       "          -8.3141e-01,  3.5424e-02, -2.4503e-01, -9.6142e-01, -2.5235e-01,\n",
       "          -1.8151e-01,  8.8420e-02,  3.0328e-01, -6.7734e-01, -2.1008e-01,\n",
       "           3.7976e-02, -4.7202e-02, -9.3493e-02, -3.1586e-01,  3.4311e-01,\n",
       "          -2.8059e-02, -1.8120e-01,  1.5784e-01, -3.5670e-02,  4.0841e-02,\n",
       "          -3.0222e-01,  8.1187e-01, -2.6322e-01,  4.9310e-01,  3.4010e-01,\n",
       "          -1.9004e-01,  1.0029e-01, -4.0861e-01, -1.4823e-01,  7.2706e-02,\n",
       "          -1.1439e-01,  2.1916e-01, -5.7457e-01, -1.2218e-01,  2.3276e-01,\n",
       "          -1.9743e-01, -3.3937e-01,  1.3113e-01, -1.7131e-02,  1.6094e-01,\n",
       "          -7.4423e-02,  3.0840e-01, -1.7384e-01, -7.0308e-02, -4.3296e-01,\n",
       "          -7.3100e-01, -9.2010e-01,  3.7805e-01, -2.4666e-01,  9.3405e-02,\n",
       "          -1.0113e-01,  2.6415e-01,  3.0387e-01, -5.5042e-03,  2.0101e-01,\n",
       "           8.3920e-01,  3.9848e-02,  2.6922e-01, -9.1562e-01, -8.0707e-02,\n",
       "          -2.3819e-01,  1.6616e-01, -1.2619e-01,  9.5852e-03, -3.4357e-02,\n",
       "           3.0667e-01, -1.1769e-01,  2.0016e-01, -5.7027e-01,  4.0579e-01,\n",
       "          -7.3660e-01,  5.0358e-01,  1.8270e-01, -2.3639e-01, -1.3372e-01,\n",
       "           1.2247e-01, -1.2062e+00,  4.9831e-01, -3.6601e-01, -1.0942e-01,\n",
       "          -5.6333e-01,  1.0090e+00,  1.6895e-01,  3.0506e-01,  1.4368e-01,\n",
       "           1.3587e-01, -3.1782e-01,  4.4816e-03,  5.4274e-01,  1.5228e-01,\n",
       "           5.5595e-01,  4.7985e-01,  7.4151e-02, -1.2919e-01,  4.7412e-02,\n",
       "          -7.5892e-01,  4.8252e-02, -3.2468e-02,  8.5681e-02,  3.1971e-01,\n",
       "           5.5460e-01, -1.2940e-01,  1.9787e-01,  1.5882e-02,  2.3008e-01,\n",
       "          -1.8434e-01, -2.6429e-01, -6.4579e-01, -1.9565e-01, -4.7367e-01,\n",
       "           4.0957e-01, -1.9305e-01, -3.1216e-01,  3.0269e-01,  2.1693e-01,\n",
       "          -2.7159e-01, -3.7334e-01,  3.5312e-02, -4.7512e-01,  9.6402e-02,\n",
       "           4.0799e-01, -4.7326e-01,  5.4479e-02, -8.0748e-02, -3.0695e-01,\n",
       "           1.7636e-01, -8.7167e-01,  6.4255e-01,  2.6578e-01, -2.5909e-01,\n",
       "          -4.6777e-01, -2.2385e-01,  1.7326e-01, -1.2252e-01, -3.0211e-01,\n",
       "          -5.4488e-01, -6.5918e-01, -5.3320e-01, -5.8488e-02, -6.5193e-02,\n",
       "           2.5191e-01, -2.2753e-01,  8.6861e-01,  4.1936e-01,  3.0692e-01,\n",
       "           6.2163e-02, -4.7275e-01,  9.1706e-02, -2.6539e-01, -3.7433e-01,\n",
       "          -4.7550e-01, -2.3982e-01, -2.4279e-01,  4.3422e-01, -3.0742e-01,\n",
       "          -5.3396e-01, -3.3148e-02,  1.7141e-01,  2.6970e-01, -4.3827e-02,\n",
       "          -2.9972e-02, -4.1887e-02, -3.0860e-01, -2.1206e-01,  5.0152e-01,\n",
       "          -6.9803e-02,  3.2745e-02, -2.5440e-01, -6.8108e-01, -5.4692e-01,\n",
       "          -3.9722e-01,  2.4398e-01, -4.3672e-01,  6.3305e-01, -3.7595e-01,\n",
       "          -2.0573e-01,  8.0469e-02, -5.5669e-02,  1.8577e-02, -1.1917e-01,\n",
       "           2.1133e-01,  1.9120e-02,  1.1580e-02, -4.1703e-01, -5.6093e-01,\n",
       "          -1.1776e-01,  2.3510e-01, -3.0787e-01,  1.4083e-01,  1.4746e-01,\n",
       "           2.2955e-01,  1.5456e-01,  3.3469e-01,  2.5366e-01, -5.0728e-01,\n",
       "          -2.0619e-01, -7.4513e-02,  1.8979e-02,  2.3842e-02,  2.7430e-01,\n",
       "          -3.3728e-01,  4.5499e-01,  6.1394e-01, -2.4935e-01, -7.2121e-01,\n",
       "           1.5674e-01,  5.6095e-01, -1.6541e-01, -1.0497e-01, -8.8263e-02,\n",
       "          -9.7190e-01, -2.3514e-01, -2.9094e-01,  2.6353e-02, -3.4220e-01,\n",
       "           3.3554e-01, -3.0276e-02,  2.2698e-01,  1.1166e-01, -3.6041e-01,\n",
       "           1.1786e-01, -2.5092e-01,  3.1932e-01,  2.6953e-01, -1.4665e-01,\n",
       "          -1.0777e-01, -4.1049e-01, -3.0109e-01, -8.6413e-02, -1.1207e-01,\n",
       "           2.6766e-01,  3.8201e-01, -1.0363e-01, -1.1280e-01, -6.9667e-02,\n",
       "           5.0100e-01, -2.2810e-01, -8.0220e-01,  3.4421e-01, -3.1339e-01,\n",
       "          -5.6208e-01,  1.9574e-01, -1.6346e-01, -2.3518e-01,  3.5715e-01,\n",
       "           3.2643e-01, -3.0058e-01, -2.8244e-01, -5.2943e-02,  6.0010e-01,\n",
       "           4.7667e-01,  5.0677e-01, -3.5451e-01,  3.1741e-01,  4.0610e-01,\n",
       "          -3.3666e-01,  3.6258e-02, -4.7560e-01,  5.5196e-03,  5.1393e-02,\n",
       "           8.9190e-02,  4.1719e-01, -3.9122e-01,  1.9928e-01,  3.4317e-01,\n",
       "          -2.0007e-01,  1.0609e-01, -1.4943e-01,  1.5650e-01, -8.4282e-02,\n",
       "          -1.8275e-01, -3.7689e-01,  4.3468e-01, -1.7396e-01, -2.0849e-01,\n",
       "           3.4059e-01,  4.0850e-01, -2.0967e-01, -4.6286e-01, -2.4343e-01,\n",
       "          -2.6897e-01, -9.2810e-02,  1.0455e+00,  7.4038e-02,  5.7296e-02,\n",
       "           1.1045e-01,  1.9602e-01, -1.8049e-01,  1.8980e-01,  3.4227e-01,\n",
       "          -2.5638e-01, -4.6622e-02, -3.5637e-01,  3.4872e-01,  4.8857e-01,\n",
       "          -3.2944e-01, -3.1593e-01,  2.4620e-01, -2.4486e-01, -5.2003e-02,\n",
       "          -5.1237e-01,  7.6474e-02,  4.6747e-01, -2.3285e-01,  2.0770e-02,\n",
       "           6.0287e-02,  9.4645e-01, -3.8485e-01,  1.4038e-01,  5.3313e-02,\n",
       "           3.1083e-01,  8.8567e-02, -2.2433e-01, -2.3987e-01, -5.3709e-01,\n",
       "           4.6842e-01,  5.9782e-01,  1.0582e-02,  2.3366e-01, -1.1143e+01,\n",
       "          -2.2806e-01,  5.6687e-01, -4.1450e-01,  3.9290e-01, -1.3661e-01,\n",
       "           4.0053e-01, -2.6131e-01,  9.5148e-02, -3.0096e-01,  1.1145e-01,\n",
       "          -4.1732e-01,  7.0476e-02, -2.2120e-01, -4.9903e-01, -2.4542e-01,\n",
       "          -6.3652e-02, -6.2133e-01,  5.0232e-01, -1.0347e-01,  6.3477e-01,\n",
       "          -2.5830e-01,  3.0063e-01,  9.9521e-02, -1.5226e-01, -2.5000e-01,\n",
       "          -8.9859e-02, -5.6227e-01,  3.3807e-02,  4.1182e-01,  1.0179e-01,\n",
       "           4.3537e-02,  3.0651e-01,  2.3129e-01, -2.1486e-01, -6.6703e-01,\n",
       "          -9.2876e-02, -2.0670e-01,  1.4230e-01, -1.4062e-01, -2.1868e-01,\n",
       "          -2.5932e-01,  2.4003e-02,  5.0342e-01,  1.3067e-01,  7.3937e-01,\n",
       "           8.0422e-01,  3.6202e-01,  9.9462e-02, -1.9998e-01,  7.6297e-02,\n",
       "          -7.7912e-01,  3.4062e-01, -1.1797e+00,  9.0450e-02,  8.0357e-02,\n",
       "           4.7560e-01, -6.1620e-01,  1.2372e-01, -1.8336e-01, -9.9995e-02,\n",
       "           3.8890e-01, -9.7863e-02,  6.5083e-02, -3.4913e-01, -2.9623e-01,\n",
       "           4.1237e-01, -5.4118e-01,  2.1745e-01, -2.7497e-01, -1.1712e-01,\n",
       "          -3.8724e-01,  2.9677e-01,  3.1264e-01, -4.6093e-01, -1.6928e-01,\n",
       "           4.5249e-02, -4.8456e-01,  1.1604e+00, -2.4378e-02,  9.5393e-02,\n",
       "           2.3049e-01,  2.9555e-01, -4.3821e-01, -5.9859e-01, -2.3646e-01,\n",
       "           3.6533e-02, -1.8133e-01,  3.8760e-02,  8.5763e-01, -1.0779e-01,\n",
       "          -5.8605e-01,  8.3385e-02,  4.9138e-01,  2.0744e-01,  6.9150e-01,\n",
       "          -6.4472e-01,  2.8242e-01,  1.0758e-01, -6.2577e-02, -5.0886e-01,\n",
       "           2.0881e-01, -2.3286e-01, -6.1711e-01,  5.1433e-01,  4.5532e-01,\n",
       "           1.4984e-01,  6.9505e-02,  3.8049e-02,  4.2275e-01, -4.8852e-01,\n",
       "           4.0405e-01,  9.1673e-02,  2.7653e-01, -6.1840e-01,  1.8680e-03,\n",
       "           4.1207e-01,  7.2896e-01, -4.9047e-02, -4.1595e-01, -3.6885e-01,\n",
       "           2.0188e-01,  2.6490e-01, -5.1289e-02, -4.7215e-01, -1.3233e-01,\n",
       "          -5.3236e-01, -1.5674e-02, -9.7173e-02,  1.5745e-01,  6.0949e-01,\n",
       "           1.3918e-01,  2.5085e-01, -3.8581e-02, -3.5964e-01, -6.9492e-01,\n",
       "          -4.8527e-01, -1.8366e-02,  2.0670e-01, -4.3197e-01,  1.4607e-01,\n",
       "           2.1760e-01, -1.0143e-01, -2.5897e-01,  2.8592e-01,  9.6152e-02,\n",
       "           2.1557e-01,  1.9749e-01, -1.1818e-01,  8.7119e-02, -3.5730e-01,\n",
       "          -2.4573e-01, -6.0373e-02, -4.2607e-02, -1.1479e-01, -3.9210e-01,\n",
       "          -1.0046e-01,  1.8307e-01, -8.6005e-02, -1.3861e-01, -1.3424e-01,\n",
       "           3.1648e-01, -6.8344e-03,  2.6353e-02,  4.0484e-01, -2.6006e-03,\n",
       "           8.3145e-02, -4.2339e-01, -3.0836e-01, -1.8390e-01,  1.5819e-02,\n",
       "           2.7405e-01, -3.1654e-01, -4.1169e-01, -3.4185e-01,  8.0299e-02,\n",
       "           1.9597e-01,  1.3697e+00, -3.0183e-01, -7.0392e-01,  3.6088e-01,\n",
       "           1.9914e-01,  1.0551e-01,  2.0523e-01, -3.4951e-01,  5.1806e-01,\n",
       "          -6.3885e-01, -3.6709e-02,  2.3357e-01, -1.4136e-01,  3.5182e-01,\n",
       "           2.7864e-01, -1.2618e-02, -2.1570e-01,  1.2628e+00, -7.6990e-02,\n",
       "          -1.6995e-01, -3.3124e-01, -4.5241e-01, -9.8361e-02,  3.0101e-01,\n",
       "          -1.2247e-01,  2.8961e-01,  8.3488e-02,  2.7243e-01,  1.4445e-01,\n",
       "           4.1018e-01,  9.7578e-01, -2.1737e-01,  3.8046e-01,  3.9125e-01,\n",
       "           2.3063e-01, -6.6879e-02,  2.5823e-01,  1.8687e-01,  2.0360e-01,\n",
       "           3.3333e-01, -1.2319e-01, -2.7763e-01, -1.4537e-01,  1.9444e-01,\n",
       "           1.4965e-01, -6.6735e-01, -4.0481e-01, -1.0311e-01,  4.8277e-01,\n",
       "           7.4379e-03,  5.2215e-01, -1.5726e-01,  5.7742e-01,  8.6073e-01,\n",
       "          -1.0186e+00, -2.2159e-01, -1.5484e-01,  2.8118e-01, -4.1177e-02,\n",
       "          -2.3571e-01,  1.6776e-01,  3.4610e-02, -1.6184e-01,  3.5677e-01,\n",
       "           3.3243e-01, -6.7393e-01,  3.4110e-01,  3.3694e-01, -4.8014e-01,\n",
       "          -3.6084e-02,  1.1975e-01,  4.2998e-01,  5.9728e-02,  1.0868e-01,\n",
       "          -1.3316e-01,  7.0184e-02,  7.0692e-02, -7.8885e-01,  2.0709e-01,\n",
       "           5.1896e-01,  3.4803e-02,  1.5692e-01,  2.0201e-01, -1.3900e-01,\n",
       "          -1.2885e-01, -2.6163e-01, -2.9764e-01,  1.0740e-01, -1.2160e-01,\n",
       "           1.5702e-02, -5.3228e-01,  1.6256e-01,  2.7426e-01, -2.8053e-01,\n",
       "           4.1407e-01,  7.4598e-01,  5.9338e-01, -9.0275e-03, -3.5227e-02,\n",
       "           2.0519e-01,  1.9254e-01, -3.2242e-02,  4.0414e-01, -5.2218e-02,\n",
       "          -4.3299e-01,  1.2565e-01,  2.8156e-01, -1.7107e-01,  1.9911e-01,\n",
       "          -4.6248e-01,  3.3742e-01, -2.5453e-02, -1.1417e-01,  5.4168e-01,\n",
       "           1.3928e-01,  5.8937e-01, -5.6335e-01,  2.9540e-01,  1.8699e-01,\n",
       "           5.4253e-02,  8.1286e-01,  3.1516e-01,  2.3072e-01, -7.2708e-02,\n",
       "           5.6683e-02, -3.8544e-01, -5.3989e-01,  2.5237e-01, -8.0678e-02,\n",
       "          -3.8260e-01, -4.7103e-02,  1.5997e-01,  3.5062e-01,  7.3176e-01,\n",
       "           8.9605e-01, -5.6440e-02, -6.9474e-01, -4.2234e-01,  9.3208e-03,\n",
       "          -5.2694e-02, -5.6176e-01,  5.2667e-02, -7.4479e-02,  6.4665e-02,\n",
       "          -8.3254e-01, -8.8518e-01, -6.7289e-02, -1.6431e-01, -4.6478e-01,\n",
       "           6.2077e-01, -2.1732e-01,  4.6311e-01,  5.4961e-02,  4.0741e-01,\n",
       "           8.3780e-02,  8.0015e-01,  2.1262e-01, -5.1639e-01,  6.7346e-02,\n",
       "           3.7895e-01, -1.3906e-01,  1.2845e-01, -3.7148e-01,  2.5928e-01,\n",
       "           1.2472e-01,  6.2208e-01,  5.9939e-01]], device='mps:0'),\n",
       " tensor([[-0.2258, -0.7099, -0.3778,  ...,  0.3292,  0.7059,  0.7494],\n",
       "         [-0.2479, -0.6708, -0.6417,  ...,  0.4683,  0.3422,  0.5940],\n",
       "         [-0.0292, -0.8824, -0.7142,  ...,  0.1563,  0.7993,  0.5188],\n",
       "         [-0.1833, -0.7244, -0.8752,  ...,  0.1263,  0.8326,  0.4855]],\n",
       "        device='mps:0'),\n",
       " tensor([[-5.6903e-02, -5.8641e-01, -3.8721e-01, -6.3056e-01, -2.4671e-01,\n",
       "          -3.0499e-01, -5.8315e-01,  2.4842e-01, -6.4447e-02,  3.6267e-01,\n",
       "           4.8309e-01,  7.9601e-02, -2.2305e-01, -3.2887e-01, -7.5675e-01,\n",
       "           5.7158e-02, -5.0681e-01,  1.4361e-01,  3.0380e-01,  1.3730e-01,\n",
       "           3.3055e-01,  1.9436e-01, -5.4206e-01,  5.4293e-02, -6.0092e-01,\n",
       "           1.6735e-01, -8.2495e-02,  2.7757e-01,  3.7720e-01, -2.7628e-01,\n",
       "          -4.8833e-02,  5.1316e-01, -2.7488e-01,  1.0115e+00,  5.3101e-02,\n",
       "          -3.9116e-01,  2.1174e-01, -1.4576e-01,  4.4837e-01, -5.5390e-01,\n",
       "           4.2510e-01, -8.7731e-02,  9.3562e-02,  1.8301e-01,  4.9244e-02,\n",
       "          -1.5626e-01,  5.1249e-02, -5.8329e-01, -1.3319e-01, -5.9850e-01,\n",
       "          -1.7161e-01, -3.7777e-01,  4.1485e-01,  1.6862e-01, -2.4272e-01,\n",
       "          -1.9685e-01, -1.1184e-01,  3.5731e-02, -6.3605e-02,  4.6973e-01,\n",
       "          -2.5097e-01,  8.0534e-01, -7.4380e-02, -6.7786e-02,  2.1091e-02,\n",
       "           3.0267e-01, -5.3761e-01, -5.8244e-01, -4.6578e-01,  4.5849e-02,\n",
       "           2.0566e-01,  4.4067e-01,  4.5226e-01, -4.5072e-01,  3.9990e-01,\n",
       "           1.5220e-01,  2.1060e-01, -4.5266e-02, -3.1224e-01,  1.4000e-01,\n",
       "           4.1026e-01,  4.5923e-01,  1.5326e-01,  4.0364e-01,  6.1462e-03,\n",
       "          -1.7192e-02,  2.4875e-01, -3.0160e-02,  2.6356e-03,  3.0803e-01,\n",
       "          -4.2584e-02,  3.1753e-02, -2.3757e-01, -4.0439e-01,  1.9259e-02,\n",
       "          -2.6839e-01,  1.4189e-01, -4.1171e-01, -1.9750e-01, -2.3773e-01,\n",
       "           2.6473e-01,  1.7897e-01,  3.4631e-01,  4.6945e-02, -6.5065e-01,\n",
       "          -3.0759e-01,  2.1990e-01, -1.2570e-01, -2.8246e-01, -1.3274e-01,\n",
       "          -1.5414e-01,  8.7244e-02,  4.3948e-02, -2.0630e-01,  3.3771e-02,\n",
       "           2.7440e-01,  9.7223e-02, -2.2189e-01,  1.3560e-01,  5.0062e-01,\n",
       "           1.6194e-01, -2.3034e-01,  1.9645e-01,  2.4779e-01,  3.2566e-01,\n",
       "           2.3118e-01,  6.1527e-01,  3.1987e-02,  3.7861e-01,  1.9559e-01,\n",
       "           4.9017e-02,  2.4681e-01, -2.3478e-01, -2.5762e-01, -2.8451e-03,\n",
       "           2.0728e-01, -1.5865e-01, -5.5846e-01,  5.0950e-01, -3.7913e-01,\n",
       "           1.0748e-01,  1.0434e-01, -2.4657e-01, -2.4882e-01, -3.1677e-01,\n",
       "          -2.0537e-01,  3.0558e-01,  1.6177e-01, -6.3913e-01, -1.9608e-01,\n",
       "          -7.0240e-01, -8.0006e-01,  2.2135e-01,  2.1072e-02,  9.4958e-02,\n",
       "          -3.2916e-01,  1.0398e-01, -1.0511e-01, -2.3261e-01,  1.1680e-01,\n",
       "           6.7396e-01, -4.1217e-01,  8.0606e-02, -5.0417e-01, -2.0337e-03,\n",
       "          -1.5892e-01,  1.0615e-01,  1.7212e-02,  4.9882e-01,  2.2270e-01,\n",
       "           2.8616e-01,  6.6406e-03,  6.5187e-02, -7.3384e-01,  1.6905e-01,\n",
       "          -3.6936e-02,  6.7118e-03,  3.4385e-01, -2.1454e-01, -1.3293e-01,\n",
       "           5.3220e-02, -1.7484e+00,  1.2324e-01, -3.9274e-01,  1.7318e-01,\n",
       "           2.1113e-01,  2.6336e-01, -1.0125e-01,  3.6618e-01, -1.2408e-01,\n",
       "          -1.8917e-01,  1.1115e-01,  2.0870e-01,  3.2696e-01,  2.4943e-02,\n",
       "           2.6605e-01,  7.0772e-01,  3.0965e-01,  1.4486e-01, -1.8212e-01,\n",
       "          -4.9094e-01,  2.1998e-01,  2.9899e-01,  3.8237e-01,  7.8184e-02,\n",
       "           2.1056e-01, -7.5875e-01,  7.5754e-02,  3.8253e-01, -2.2340e-02,\n",
       "          -1.5876e-01, -2.9896e-01, -5.1655e-01, -1.3805e-01, -6.5851e-01,\n",
       "           1.1350e-01, -2.3885e-01, -4.4590e-02,  5.9355e-01,  1.3376e-01,\n",
       "          -2.0169e-01,  3.6954e-01,  2.0442e-01, -3.5316e-01,  1.4404e-01,\n",
       "           6.5105e-01,  9.5968e-02, -2.4719e-02, -1.7733e-01, -4.9374e-01,\n",
       "          -7.1718e-02, -5.1033e-01,  7.2807e-01,  4.8571e-01, -2.0841e-01,\n",
       "           1.8011e-01, -5.2912e-01,  8.5262e-02, -1.9391e-01, -4.5951e-02,\n",
       "          -1.8129e-01, -8.2823e-02, -5.3264e-01,  1.7527e-01,  3.4959e-02,\n",
       "           1.4781e-01,  4.8752e-02,  8.1369e-01,  5.9641e-01,  3.1503e-01,\n",
       "           1.2854e-01,  2.9832e-01,  2.6065e-01,  6.9199e-01, -2.6353e-01,\n",
       "           4.3880e-01,  2.9800e-02, -3.8826e-01,  3.3455e-01, -3.1503e-02,\n",
       "           6.9781e-02, -3.3639e-01, -1.7904e-03,  3.3258e-01, -1.6937e-01,\n",
       "           1.0432e-01,  3.8080e-01,  2.6967e-02, -3.3577e-01,  5.2354e-01,\n",
       "          -5.0699e-01,  5.9370e-01,  2.0575e-01, -2.3646e-01, -5.0810e-01,\n",
       "          -7.9763e-01, -1.0557e-01, -4.8342e-02,  2.2293e-01, -9.2350e-01,\n",
       "          -3.5545e-01,  5.2613e-01,  1.5440e-01,  3.3226e-01,  1.7132e-01,\n",
       "           2.2515e-01, -1.0299e-02,  1.6039e-01, -2.4707e-01, -6.0023e-01,\n",
       "          -1.5093e-01,  1.0024e-01, -2.2311e-01,  8.2777e-02,  2.5673e-01,\n",
       "           1.6824e-01,  2.0340e-01,  1.3925e-01, -1.9015e-01, -1.5548e-01,\n",
       "          -2.2247e-01,  2.6442e-01, -3.3694e-02, -4.6560e-01, -3.6676e-02,\n",
       "           1.2635e-01,  1.2769e-01,  5.5031e-01, -8.6039e-02, -3.3811e-01,\n",
       "          -1.1500e-01,  4.5787e-01, -1.2383e-01, -2.6480e-01, -4.9292e-02,\n",
       "          -7.9521e-01,  1.6463e-01,  1.3696e-01, -1.9046e-01, -5.6897e-01,\n",
       "           3.2259e-01, -1.2629e-01, -3.7848e-01,  1.2466e-02, -4.8989e-02,\n",
       "           4.8388e-01,  3.5461e-01,  2.2299e-01, -2.4737e-01, -2.1371e-01,\n",
       "          -4.5278e-02, -2.6071e-01, -3.7906e-02, -1.9623e-01,  2.2065e-01,\n",
       "           2.0307e-01,  3.6967e-01, -1.0660e-01,  6.3183e-02, -1.9617e-01,\n",
       "           8.4112e-01, -5.1709e-01, -1.3619e-01,  4.4108e-01, -2.1099e-01,\n",
       "          -1.0738e-01,  7.3803e-01,  3.5581e-01,  3.2068e-02,  7.2176e-02,\n",
       "           1.7139e-01, -2.4860e-01, -3.2834e-01,  7.5825e-03,  5.4314e-01,\n",
       "           3.6471e-01,  6.1288e-02, -4.3456e-01,  1.7110e-01,  3.8287e-01,\n",
       "          -3.7150e-01, -9.5523e-02, -3.5868e-01, -3.3038e-01, -2.3326e-02,\n",
       "           2.2574e-01,  5.3576e-01, -2.7101e-01,  1.1662e-01,  4.5883e-01,\n",
       "           2.3244e-01, -4.8899e-01, -1.3544e-01,  4.7106e-01,  1.6318e-01,\n",
       "           1.3058e-01, -4.1783e-01,  1.4401e-01, -3.3637e-01, -2.7604e-03,\n",
       "          -1.7857e-01,  9.8866e-02,  1.7141e-01, -1.5689e-01, -4.9977e-01,\n",
       "          -9.2468e-02, -1.8406e-01,  3.9149e-01,  1.7543e-01, -5.2595e-02,\n",
       "           3.0261e-01,  8.5742e-03, -4.0160e-01,  7.0656e-01, -2.6254e-01,\n",
       "          -2.6860e-01,  3.7187e-01, -6.4792e-01, -3.1495e-03,  1.1215e-01,\n",
       "          -1.3224e-01, -1.4070e-01,  9.4788e-02,  9.0739e-01,  1.2407e-01,\n",
       "          -2.2069e-01, -1.5799e-02,  9.8980e-02, -2.9262e-01,  2.8874e-01,\n",
       "          -9.9073e-02,  6.8600e-01, -3.0231e-01,  2.7561e-01, -5.3189e-01,\n",
       "           5.7169e-02, -3.2847e-01, -4.4404e-01, -1.8546e-01, -2.0863e-01,\n",
       "          -1.7868e-01,  1.3611e-01, -1.8452e-01,  9.3011e-02, -1.1454e+01,\n",
       "           5.3142e-02,  7.5984e-01, -4.0551e-01,  8.5388e-02,  2.4382e-01,\n",
       "           2.4605e-01, -3.8249e-01,  4.9774e-01, -4.7641e-01, -2.6436e-01,\n",
       "          -2.9402e-01,  2.1042e-01,  6.7028e-03, -6.6546e-01,  2.1164e-01,\n",
       "          -1.9778e-01, -6.3890e-01,  1.2717e-01,  4.2410e-02,  4.3289e-01,\n",
       "          -1.8935e-01,  2.0722e-01,  1.5612e-01, -3.1001e-02, -5.3600e-01,\n",
       "          -3.8579e-01, -4.3957e-01,  2.4735e-01,  3.7999e-01, -1.9054e-01,\n",
       "           1.3813e-01,  1.9379e-02,  2.1396e-01, -3.5660e-01, -5.3706e-01,\n",
       "           5.9854e-01, -3.3344e-01,  1.9525e-01, -7.6677e-01, -2.1315e-01,\n",
       "          -2.0185e-01,  1.3241e-02,  2.7459e-01,  2.2009e-01,  5.9105e-01,\n",
       "           1.1653e+00,  8.7336e-01,  8.8747e-02,  3.9697e-01,  3.6810e-02,\n",
       "          -7.2668e-01,  2.7609e-01, -6.5259e-01, -6.9626e-02, -1.6356e-01,\n",
       "           1.9012e-01, -5.5752e-01,  1.8687e-01, -2.2854e-01, -1.7084e-01,\n",
       "           5.1440e-01,  2.1067e-01,  6.1831e-03, -3.6248e-01, -6.1038e-01,\n",
       "           3.8786e-01, -1.4220e-01,  1.4322e-01, -5.8122e-01,  2.2277e-01,\n",
       "           7.7920e-02,  4.9184e-01,  6.7620e-02, -2.3288e-01,  1.5460e-01,\n",
       "          -2.3866e-01,  2.0751e-01,  8.3186e-01, -2.8920e-01,  2.2385e-03,\n",
       "           8.5739e-02,  7.8352e-03, -2.2555e-01, -2.6210e-01, -2.0830e-01,\n",
       "          -2.3700e-02,  4.6804e-01,  3.5913e-02,  5.7259e-01, -5.8937e-02,\n",
       "          -5.6660e-01, -1.5944e-01, -1.2947e-01,  3.8242e-01,  8.9312e-01,\n",
       "          -5.0260e-01,  4.6019e-03, -3.1203e-01, -1.1694e-01, -4.2042e-01,\n",
       "          -2.2088e-01,  1.4430e-01, -3.2399e-01, -5.0702e-02,  1.1103e-01,\n",
       "          -1.9105e-01,  1.8969e-01,  2.3967e-02,  2.7560e-01, -6.5583e-01,\n",
       "          -1.7311e-01, -1.3828e-02,  4.7960e-02, -5.6071e-01, -3.4282e-01,\n",
       "           1.4288e-01,  6.3603e-01, -2.7256e-01, -6.0526e-01, -2.7077e-02,\n",
       "           3.3434e-01,  5.9012e-02, -2.3505e-01, -2.8072e-01,  2.6762e-01,\n",
       "          -1.0249e+00, -2.2160e-01, -2.3878e-01, -5.3404e-03, -9.7232e-03,\n",
       "           3.7631e-02, -2.7658e-01, -2.4121e-01, -1.3280e-01, -5.3771e-01,\n",
       "          -6.0428e-01,  1.1543e-03,  6.3281e-01, -4.2160e-01, -2.6007e-01,\n",
       "          -3.5964e-02,  1.4040e-01, -3.7589e-01,  3.0430e-01, -2.7456e-01,\n",
       "           2.1244e-01,  2.5662e-01, -2.0654e-01,  4.2366e-02,  8.9708e-02,\n",
       "          -1.4249e-01, -2.4488e-01,  1.4569e-01, -4.6223e-02, -8.1391e-01,\n",
       "          -2.6014e-01,  2.9221e-01, -5.2530e-01,  1.0002e-01, -1.2237e-01,\n",
       "           1.9736e-01,  3.4906e-02, -3.9669e-01,  7.2917e-01,  3.4346e-01,\n",
       "           3.5724e-02, -2.0192e-01, -4.5568e-01,  1.2616e-01,  1.3150e-01,\n",
       "          -1.6022e-01,  1.5410e-01, -1.2206e-01, -4.1575e-01, -1.9556e-01,\n",
       "           2.0641e-01,  1.3183e+00,  6.5166e-02, -2.9417e-01,  4.1270e-01,\n",
       "           8.0899e-01, -1.6499e-01,  8.3801e-02, -3.0989e-01,  3.2795e-01,\n",
       "          -6.1007e-01, -3.2796e-01,  6.8495e-01, -2.7846e-01,  3.5306e-01,\n",
       "           3.1553e-01,  3.0101e-02, -1.9090e-01,  3.2920e-01,  1.3395e-01,\n",
       "          -4.0242e-01,  1.1046e-01, -5.6028e-01, -1.2113e-02,  3.6860e-01,\n",
       "          -2.1671e-02,  2.1833e-01, -1.6844e-01, -1.7775e-01,  7.0732e-03,\n",
       "           5.6642e-02,  1.1880e+00, -2.7733e-01, -9.7671e-02,  2.7998e-01,\n",
       "          -3.2664e-01,  6.2594e-01,  1.6714e-01, -2.0379e-01,  1.5029e-01,\n",
       "           3.3695e-01, -6.4550e-02, -4.9348e-01,  4.7882e-01,  1.4254e-01,\n",
       "          -1.4128e-01, -7.4071e-01, -5.5050e-01, -4.9202e-02,  6.0453e-01,\n",
       "          -9.0969e-02,  2.9555e-01, -5.6990e-02,  6.2127e-01,  3.2031e-01,\n",
       "          -1.0168e+00,  2.0116e-01, -4.9862e-01,  5.8338e-01,  8.4300e-02,\n",
       "          -6.6380e-01, -2.5403e-01,  4.2309e-02, -2.5634e-01,  2.4247e-01,\n",
       "           4.8668e-01, -1.1312e+00, -2.3301e-01,  4.3346e-01,  1.7139e-01,\n",
       "           3.7335e-01, -2.5902e-01,  1.9894e-01,  2.9403e-01, -1.0647e-01,\n",
       "          -4.2044e-01,  2.7717e-01, -1.1361e-01, -4.5778e-01,  1.2818e-01,\n",
       "           4.2428e-01, -3.3571e-02,  3.2037e-02, -2.0371e-01, -2.5786e-02,\n",
       "          -2.1611e-01, -9.7089e-02, -3.8251e-01, -1.2936e-01,  3.4589e-01,\n",
       "          -1.5230e-01, -4.7016e-01,  1.0652e-02,  1.3186e-01,  1.0095e-01,\n",
       "           5.0995e-01,  3.5783e-01,  5.0124e-01, -2.7550e-01,  1.0115e-01,\n",
       "          -5.0503e-01,  4.0939e-01, -1.2563e-01,  7.2823e-01,  1.8234e-02,\n",
       "          -3.0793e-01,  1.9007e-01,  1.3380e-01, -1.2915e-01,  2.4596e-02,\n",
       "          -2.0123e-01,  2.4790e-01, -2.0185e-01, -3.4348e-01,  3.1102e-01,\n",
       "          -5.6585e-02,  4.8833e-02, -3.6904e-01,  1.8462e-01, -3.1064e-02,\n",
       "          -2.2500e-01,  1.0184e+00, -1.6743e-01,  3.3291e-01,  3.1289e-03,\n",
       "           3.5709e-02, -7.5324e-01,  2.9754e-03,  1.7759e-01,  5.0277e-03,\n",
       "          -5.4714e-01, -8.0906e-03, -1.7833e-02, -9.2659e-02,  5.8473e-01,\n",
       "           4.9283e-01, -6.6887e-01, -3.5047e-01, -6.8579e-02, -7.3142e-03,\n",
       "           1.7722e-01, -3.5007e-01,  1.9330e-01, -2.7346e-01, -1.4945e-01,\n",
       "          -8.4373e-01, -4.7345e-02, -1.2118e-01,  5.5447e-02, -2.0985e-01,\n",
       "           7.1069e-01, -6.4661e-02, -9.2506e-02,  7.0238e-01,  7.3527e-02,\n",
       "           2.0257e-01,  5.7521e-01,  1.6096e-01, -5.5514e-01, -2.8329e-03,\n",
       "           1.1694e-01, -1.7472e-01,  4.7835e-02, -3.9509e-01,  6.3804e-01,\n",
       "           1.2624e-01,  1.9815e-01,  7.6799e-01]], device='mps:0'),\n",
       " tensor([[-4.7943e-01, -1.9705e-01, -5.4760e-01, -7.8593e-01, -1.3950e-01,\n",
       "          -3.0171e-01, -6.0835e-01,  5.0343e-01,  3.6838e-01,  5.0234e-01,\n",
       "          -3.2067e-01, -1.3565e-02,  1.0303e-01, -5.0750e-02, -3.1975e-01,\n",
       "           1.6656e-01,  3.3741e-02,  3.1802e-01,  5.3083e-01,  3.8236e-01,\n",
       "           4.0426e-01,  1.3260e-02,  7.6264e-02, -1.2599e-01, -5.0356e-01,\n",
       "          -9.1020e-02,  3.4169e-02, -1.0382e-01,  6.5516e-02, -3.6354e-01,\n",
       "          -5.8098e-02,  3.4610e-01, -2.5540e-01,  7.4857e-01, -2.8612e-01,\n",
       "          -3.4460e-01,  1.4211e-01, -4.1346e-01,  3.1871e-01,  4.9761e-01,\n",
       "           7.2806e-01,  3.1244e-01,  5.3113e-01, -9.1034e-02,  4.7372e-01,\n",
       "           3.4793e-01,  3.5225e-01, -6.1514e-01,  1.7372e-01, -5.3511e-01,\n",
       "          -4.5483e-01, -4.1875e-02,  4.7265e-01,  2.7920e-01, -7.2626e-01,\n",
       "          -1.7113e-01, -1.5823e-01,  4.8104e-01, -2.9617e-02,  1.1153e-01,\n",
       "          -4.6001e-02,  6.2479e-01,  2.9680e-01, -7.0388e-02,  2.1994e-01,\n",
       "          -8.8303e-02, -5.6360e-01, -6.1257e-01, -5.8798e-01, -2.0691e-01,\n",
       "           1.3773e-01,  2.3561e-01, -2.0983e-02, -2.8879e-01,  3.7623e-01,\n",
       "          -4.5048e-01,  5.0740e-01, -4.0084e-01, -4.0793e-01,  2.2501e-01,\n",
       "           4.1490e-01,  2.1588e-01, -2.5389e-01, -5.2098e-02, -1.9971e-01,\n",
       "          -2.3700e-01,  7.4074e-01,  1.8664e-01,  7.4432e-02,  2.5919e-01,\n",
       "           8.8462e-01, -1.9194e-01, -7.7801e-01, -3.4788e-01,  3.7935e-01,\n",
       "          -5.6256e-01,  1.7237e-01, -3.6364e-03,  4.6391e-01, -3.6419e-01,\n",
       "           3.6107e-02,  3.3617e-01,  2.7522e-01,  1.9697e-01, -6.6149e-01,\n",
       "          -4.7943e-01,  9.9648e-02, -2.7465e-02, -3.6349e-01, -3.0831e-03,\n",
       "           1.6936e-01, -2.1800e-01,  4.3730e-01, -4.1454e-01, -5.5688e-01,\n",
       "           1.2851e-01,  4.4460e-01, -3.1378e-01, -5.5519e-03,  4.1981e-01,\n",
       "           1.8983e-02, -4.1536e-02, -5.2749e-02, -1.7621e-01,  2.1082e-01,\n",
       "          -7.2790e-01,  5.1964e-01, -1.3656e-01,  4.0531e-01,  2.8229e-02,\n",
       "           1.2166e-01,  3.4042e-01, -3.5849e-01, -2.7387e-01, -3.5963e-01,\n",
       "           2.0716e-01,  4.7604e-01, -4.9633e-01, -9.7041e-02,  5.1572e-01,\n",
       "          -7.9949e-02,  5.7793e-02, -1.3518e-01, -3.8281e-01,  2.7487e-02,\n",
       "          -2.6958e-01, -1.4323e-01, -4.3254e-01, -2.5653e-01, -1.6475e-01,\n",
       "          -2.5959e-01, -7.3536e-01,  6.1303e-01,  1.3975e-01,  1.2850e-01,\n",
       "          -3.9435e-01,  1.6789e-01, -2.3736e-02,  2.3193e-02, -6.9538e-02,\n",
       "           4.5712e-01,  1.2744e-01,  1.4623e-01, -4.5866e-01, -3.2706e-01,\n",
       "           5.4829e-02, -1.4122e-01, -2.7836e-01, -1.4841e-01,  2.6909e-01,\n",
       "           2.6851e-01, -2.3492e-02,  5.3631e-01, -4.1410e-01,  5.0866e-01,\n",
       "          -7.1012e-01,  4.7134e-01,  1.9337e-01,  9.6823e-02,  1.6391e-01,\n",
       "          -3.2657e-02, -1.2090e+00,  3.7071e-01, -5.0272e-01, -1.4703e-01,\n",
       "          -7.7724e-02,  3.5066e-01,  9.7891e-02, -2.2797e-02, -3.0873e-01,\n",
       "          -3.2902e-01,  2.4984e-01,  3.0091e-02,  4.6498e-01,  7.4432e-02,\n",
       "           1.9505e-01,  8.3460e-01,  3.4008e-01,  3.7986e-02,  2.7769e-01,\n",
       "          -6.5466e-01,  1.4616e-01,  6.3511e-02,  9.2721e-02, -4.3168e-02,\n",
       "           1.1949e-01, -3.2321e-01,  4.9601e-02, -6.0283e-02, -8.6669e-02,\n",
       "           1.6797e-01, -4.2450e-01, -5.0590e-01,  1.7384e-01,  1.4933e-01,\n",
       "           4.6791e-01, -1.3370e-01, -2.6736e-03,  7.3415e-01,  2.8876e-01,\n",
       "           2.6414e-01,  1.2397e-01, -3.3699e-01,  1.2464e-02, -6.6147e-02,\n",
       "           1.4986e-02, -3.1438e-01,  4.0716e-01, -1.9547e-01, -8.1311e-02,\n",
       "           1.7181e-01, -4.4105e-01,  1.0493e+00,  3.8437e-01,  4.3501e-03,\n",
       "           5.0645e-02, -1.7898e-01,  2.3299e-01, -4.5522e-01, -3.1716e-01,\n",
       "          -5.4502e-01, -6.9350e-01, -6.5505e-01, -9.9141e-02, -3.9081e-02,\n",
       "           3.7729e-01, -5.9069e-01,  9.0784e-01,  2.1024e-01,  3.4085e-01,\n",
       "          -4.3085e-02, -3.2439e-01,  6.3413e-02,  3.5943e-01, -2.4527e-01,\n",
       "           1.2542e-01, -4.2240e-03, -6.1343e-01,  4.3789e-01, -2.5911e-01,\n",
       "          -4.8381e-01,  1.4034e-01,  2.7221e-01,  5.4676e-01,  2.5608e-02,\n",
       "           1.3398e-01,  2.6428e-01,  2.5647e-01, -1.5805e-01,  2.8059e-01,\n",
       "          -4.1073e-01, -5.3212e-02, -2.6070e-01, -4.5493e-01, -6.7124e-01,\n",
       "          -8.2097e-01, -1.3943e-01, -1.2024e-01,  7.2665e-02, -1.0215e+00,\n",
       "          -1.1167e-01,  7.5830e-02, -2.7773e-02, -2.0016e-01,  3.9324e-03,\n",
       "           1.6496e-01, -3.1459e-02,  7.2284e-02,  2.9418e-01, -5.0102e-01,\n",
       "          -6.8648e-02,  2.4896e-01, -1.8367e-01, -8.9186e-02,  3.0417e-01,\n",
       "          -4.2414e-01, -8.3475e-02,  2.8315e-01,  9.2343e-02,  1.4871e-02,\n",
       "          -2.5901e-01, -2.1078e-01, -1.8769e-01, -1.6972e-01, -4.0960e-02,\n",
       "          -1.0523e-01,  7.2559e-01,  6.8936e-02, -3.6577e-01, -5.5049e-01,\n",
       "          -1.5934e-01,  4.1323e-01, -1.7708e-01, -8.8511e-02,  1.5563e-02,\n",
       "          -9.7471e-01, -3.8352e-01, -2.7679e-01, -3.7832e-01, -4.2656e-01,\n",
       "           9.8468e-02,  2.2019e-01,  6.9746e-02,  1.6256e-01, -2.7425e-01,\n",
       "           5.3859e-01,  2.9426e-01, -3.4102e-02, -1.2627e-01, -4.7904e-01,\n",
       "           2.7069e-01, -6.6177e-01, -3.7560e-01, -3.4729e-01,  7.8782e-02,\n",
       "           6.6355e-03, -1.0030e-01, -2.9933e-01, -1.9086e-01, -1.2774e-01,\n",
       "           8.4845e-01, -4.6335e-01,  3.8324e-01,  3.2362e-01, -1.5688e-01,\n",
       "           6.1425e-02,  3.5903e-02, -9.2271e-02, -2.7449e-01,  2.2175e-01,\n",
       "           2.8540e-01, -3.5791e-01,  2.0224e-01, -1.4085e-01,  7.1359e-01,\n",
       "           7.1478e-01,  5.4179e-01, -2.3104e-01,  1.3415e-01,  2.5223e-01,\n",
       "          -3.1822e-01,  3.0073e-01, -1.9893e-01, -2.7328e-01, -2.5193e-01,\n",
       "           1.0184e-01,  1.9944e-01, -3.4389e-01, -1.9380e-01, -8.6784e-02,\n",
       "           2.3624e-01, -5.5886e-02,  1.4168e-01,  1.0846e-01,  2.2833e-01,\n",
       "          -1.4380e-01, -3.7876e-01,  3.0981e-01, -6.9943e-02,  3.3132e-01,\n",
       "          -3.4841e-02,  1.2818e-01, -1.1597e-02, -1.2363e-01, -1.5007e-01,\n",
       "          -2.2221e-01, -7.4494e-02,  7.2501e-01, -1.2101e-01,  2.4141e-01,\n",
       "           6.0182e-01, -2.1779e-01,  8.3859e-02,  3.4790e-01, -3.8177e-01,\n",
       "          -3.7318e-01,  2.5595e-01, -2.9714e-01,  1.1078e-01,  3.6392e-01,\n",
       "          -2.1017e-01, -9.6035e-02,  3.7645e-01,  4.8171e-01,  2.0747e-01,\n",
       "          -4.8316e-01,  4.7687e-02,  4.0243e-01,  2.6259e-01,  2.2231e-01,\n",
       "           3.4923e-01,  3.2138e-01, -5.5031e-02,  2.7603e-01, -3.3429e-01,\n",
       "           1.9949e-01, -3.5768e-01,  9.0598e-03,  6.0915e-02, -4.1723e-01,\n",
       "           4.2616e-01,  4.0338e-01,  1.3780e-01,  1.6319e-01, -1.1113e+01,\n",
       "           4.9643e-01,  7.6167e-01, -7.4459e-01,  2.0755e-02,  2.0664e-01,\n",
       "           2.4591e-01, -3.4959e-01,  2.1494e-01, -1.9413e-01, -3.5031e-01,\n",
       "          -3.2401e-01, -1.7908e-01, -1.7792e-01, -4.9521e-01, -4.5081e-01,\n",
       "          -2.3440e-01, -8.6870e-01,  5.9256e-02, -3.8355e-01,  1.4032e-01,\n",
       "          -2.9152e-01,  1.0481e-01,  6.2631e-02, -2.9472e-01, -1.6983e-01,\n",
       "          -1.9608e-01,  3.9777e-01,  1.3194e-01,  8.3314e-02,  1.5971e-01,\n",
       "           3.2665e-01, -8.3579e-02,  2.0492e-01, -1.0971e-01, -7.3515e-01,\n",
       "          -1.2687e-01, -6.4678e-03, -1.2942e-01, -3.6354e-01, -3.0642e-01,\n",
       "          -1.8299e-01, -6.5248e-01,  2.8287e-01,  1.9401e-01,  7.0420e-01,\n",
       "           5.5715e-01,  4.7601e-01,  3.0062e-01, -2.2889e-02, -6.8408e-02,\n",
       "          -4.3802e-01,  1.7584e-01, -1.9310e-01, -8.1311e-01, -1.7806e-01,\n",
       "           7.1860e-01, -8.4850e-01,  1.6995e-01, -3.0902e-01, -1.3177e-01,\n",
       "          -1.3822e-01, -5.6797e-02,  4.4247e-02, -3.3055e-01, -3.5446e-01,\n",
       "           3.0186e-01,  2.2667e-01,  4.4369e-01, -2.3864e-01,  5.5604e-02,\n",
       "          -1.0274e-01,  4.6945e-01,  3.1695e-01,  1.4181e-01, -3.7237e-01,\n",
       "          -3.1944e-01, -1.4762e-01,  2.3896e+00, -1.8011e-01,  2.5493e-01,\n",
       "           6.8610e-01,  2.9107e-01, -6.1802e-01, -4.1720e-01,  6.1975e-02,\n",
       "          -5.3475e-02, -1.7325e-01,  3.2883e-01,  3.8709e-01, -9.6989e-02,\n",
       "          -7.0527e-01, -3.6515e-01,  2.3969e-01,  6.5959e-01,  6.0581e-01,\n",
       "           5.2579e-02,  3.5638e-01, -7.4214e-02, -1.1966e-01, -6.8616e-01,\n",
       "          -7.3796e-02, -3.3403e-01, -3.6157e-01,  2.5560e-01,  4.1829e-01,\n",
       "          -6.0994e-02, -3.4286e-02, -1.7481e-02,  2.5273e-01, -3.4032e-01,\n",
       "           6.0386e-01, -1.4149e-01,  2.1206e-02, -1.7514e-01, -5.3499e-01,\n",
       "           3.7974e-01,  6.2354e-01, -4.3369e-01, -5.7695e-01, -9.0980e-02,\n",
       "          -1.8109e-02,  3.0438e-01,  1.6802e-01, -5.7573e-01,  2.0533e-01,\n",
       "          -3.7580e-01,  8.6789e-02,  2.8433e-01,  1.2774e-01,  4.6036e-01,\n",
       "          -2.1457e-01, -4.0898e-01, -2.3380e-01, -1.2138e-02, -2.7964e-01,\n",
       "          -5.7688e-01, -1.7625e-01,  4.3789e-01, -2.4453e-01,  1.4403e-01,\n",
       "          -3.6241e-01,  1.5150e-01, -3.6422e-01,  2.3070e-01,  3.6386e-01,\n",
       "          -1.9077e-01,  3.1956e-01,  6.2048e-02,  4.7842e-01, -2.9983e-01,\n",
       "          -2.5537e-01,  1.9828e-01, -1.0835e-01,  3.7140e-01, -3.3127e-01,\n",
       "          -2.6564e-01, -1.8376e-01, -4.0560e-01, -2.7477e-01,  4.5608e-02,\n",
       "           1.5732e-01, -2.6897e-02, -7.2871e-01,  1.5133e-01,  5.6762e-01,\n",
       "           2.1604e-01, -6.3013e-01, -4.1365e-01, -3.2858e-01, -1.3765e-01,\n",
       "           4.8992e-01,  1.9091e-01, -1.0518e-01, -3.0626e-01, -1.5564e-01,\n",
       "          -3.3159e-01,  7.4845e-01, -2.5967e-01, -2.7541e-01,  3.7041e-01,\n",
       "           9.3324e-01,  9.8147e-02,  3.0462e-02, -6.8602e-01,  5.5830e-02,\n",
       "          -3.7831e-01, -2.0496e-01,  5.5431e-01, -1.4922e-01,  8.0585e-01,\n",
       "          -6.9625e-02,  1.1009e-02,  2.0215e-01,  7.2499e-01,  3.3675e-01,\n",
       "          -2.1344e-01, -2.2668e-01, -4.1563e-01, -3.2143e-01,  2.1629e-01,\n",
       "          -2.0207e-02,  3.3537e-01, -8.3927e-02, -1.2455e-01, -2.8564e-01,\n",
       "           2.9138e-01,  8.1068e-01, -5.8126e-01, -1.6206e-01,  7.3237e-01,\n",
       "           5.3573e-01,  1.9851e-01,  2.0442e-01,  3.0043e-01, -1.1168e-01,\n",
       "           2.8522e-01,  9.0870e-02, -2.3934e-01, -3.7911e-02,  3.4673e-01,\n",
       "           7.6102e-01, -9.9298e-01, -3.8995e-01,  2.4754e-01,  4.1007e-01,\n",
       "          -8.2150e-01,  2.5516e-01, -2.5163e-01,  4.8939e-01,  6.7302e-01,\n",
       "          -1.0234e+00, -7.9343e-02, -2.0605e-01,  2.9998e-01, -1.2701e-01,\n",
       "          -5.5667e-01,  2.7000e-02, -3.9768e-01, -3.2761e-01,  3.8632e-01,\n",
       "           3.4577e-01, -9.4042e-01,  1.2071e-01,  7.2172e-01, -4.5951e-01,\n",
       "           8.8156e-02, -3.6302e-03,  7.9584e-02, -1.0191e-01,  1.8995e-01,\n",
       "          -2.6877e-01,  1.7081e-01, -7.6699e-01, -1.0823e+00,  2.0237e-01,\n",
       "           5.8696e-01,  6.5665e-02, -1.2358e-01,  1.4185e-01,  2.9387e-02,\n",
       "           3.1174e-02, -5.8682e-01, -1.6205e-01, -2.2526e-01,  1.7667e-01,\n",
       "           5.8478e-02, -3.2344e-01,  4.9439e-03,  4.2668e-01, -6.0041e-01,\n",
       "           3.5830e-01,  2.6293e-01,  5.8916e-01,  1.4088e-01, -2.5362e-01,\n",
       "          -1.2133e-01,  4.2209e-01, -8.3828e-02,  3.9020e-01, -2.9269e-02,\n",
       "          -4.0789e-01,  2.9854e-02,  1.9986e-01, -4.1768e-01, -2.0956e-01,\n",
       "          -2.3791e-01,  4.7287e-01, -3.8440e-02, -2.3953e-01,  4.2919e-02,\n",
       "          -1.9629e-01,  3.8706e-02, -7.6449e-01,  4.1498e-01,  3.4545e-01,\n",
       "          -2.0748e-01,  6.4744e-01,  4.3222e-01,  6.2502e-01,  1.3381e-01,\n",
       "          -1.3405e-01, -7.9160e-01, -1.8083e-01, -5.2220e-02,  9.3393e-02,\n",
       "           6.4782e-02, -3.4813e-01,  1.2166e-01,  1.1973e-01,  6.5218e-01,\n",
       "           5.6592e-01,  5.1663e-02, -9.5732e-01, -4.8932e-01,  9.0817e-02,\n",
       "          -2.3929e-01,  3.5125e-02,  3.6935e-02,  5.6364e-03,  3.0536e-01,\n",
       "          -5.2311e-01, -4.4795e-01, -2.1609e-01,  3.7351e-01, -2.8401e-01,\n",
       "           5.2386e-01, -2.6663e-01,  3.7622e-01,  4.0234e-01,  9.6415e-02,\n",
       "           3.4200e-01,  4.6911e-01,  5.0758e-02, -2.4893e-01, -1.8420e-01,\n",
       "           2.7977e-01, -1.8086e-01,  3.0597e-01, -7.8489e-01,  2.6396e-01,\n",
       "           2.0244e-01,  6.8786e-01,  3.1523e-01]], device='mps:0'),\n",
       " tensor([[-0.0932, -0.6293, -0.4879,  ...,  0.3041, -0.1958,  0.5576],\n",
       "         [-0.1672, -0.5249, -0.4540,  ...,  0.3771,  0.2016,  0.5340],\n",
       "         [-0.2082, -0.4731, -0.5731,  ...,  0.2261,  0.3546,  0.5161]],\n",
       "        device='mps:0'),\n",
       " tensor([[-1.7365e-01, -6.0521e-01, -4.8893e-01, -7.6425e-01, -3.5536e-01,\n",
       "          -5.2173e-02, -5.4721e-01,  5.1896e-01,  9.2863e-02,  5.7445e-01,\n",
       "           1.8773e-02,  5.1101e-01, -2.7094e-04, -1.4121e-01, -1.2958e-01,\n",
       "          -4.0469e-01, -5.3262e-01,  4.0700e-01,  5.7400e-01, -1.2836e-01,\n",
       "           2.7941e-01,  5.3287e-01, -4.6403e-01,  2.0407e-02, -9.9954e-01,\n",
       "           9.0094e-02,  5.4413e-02,  2.8502e-01,  2.8373e-01, -8.4723e-01,\n",
       "          -3.7371e-02,  4.5598e-01,  1.4021e-01,  1.5409e+00, -4.0482e-01,\n",
       "          -1.2192e-01,  8.7089e-02, -6.0884e-01,  1.3140e-01, -2.3987e-01,\n",
       "           5.9388e-01, -1.1633e-01,  1.5245e-01,  4.2147e-01, -1.9290e-02,\n",
       "          -8.5204e-03,  3.8905e-01, -4.5460e-01,  4.3782e-01, -8.1275e-01,\n",
       "          -1.8224e-01, -2.4678e-02,  4.9019e-01,  4.0885e-01, -5.8255e-01,\n",
       "          -4.8741e-01,  4.7792e-01,  2.2046e-01, -1.9713e-01,  2.2699e-01,\n",
       "          -2.1041e-01,  4.1121e-01, -1.2305e-01,  3.1800e-01,  1.4866e-01,\n",
       "           1.0803e-01, -4.3384e-01, -2.6755e-01, -5.2116e-01, -2.4428e-01,\n",
       "          -1.0353e-02,  2.0226e-01,  7.0166e-02, -1.5633e-02,  6.1524e-01,\n",
       "          -5.9877e-01, -2.6641e-01, -2.2043e-01, -2.9464e-01,  5.0098e-02,\n",
       "          -2.1909e-01,  3.8045e-01,  4.7421e-01,  1.6099e-02,  2.0554e-01,\n",
       "          -9.3745e-02,  3.6554e-01,  1.9841e-01,  5.4626e-02,  2.7241e-01,\n",
       "           4.1719e-03,  1.0701e-01, -1.8449e-01, -1.9886e-01,  2.4174e-01,\n",
       "          -2.6877e-01,  5.6201e-01, -9.3755e-02, -6.4750e-03, -3.3758e-01,\n",
       "           5.8448e-02,  1.5405e-01, -5.9612e-03, -1.3377e-02, -1.8391e-01,\n",
       "          -5.9330e-01, -7.1101e-02, -3.1543e-01, -8.4009e-01,  6.8321e-02,\n",
       "          -8.2465e-02, -1.1920e-01, -2.8461e-02, -7.5575e-01, -1.0069e-01,\n",
       "          -1.1372e-01,  4.1687e-01, -4.5738e-02, -3.1860e-01, -4.6225e-02,\n",
       "           2.3628e-01, -3.4806e-01,  1.9969e-01,  2.2902e-01,  3.0422e-01,\n",
       "          -3.0139e-01,  7.8027e-01, -3.1544e-02,  3.4120e-01,  2.5369e-01,\n",
       "           7.2768e-02, -2.6166e-02, -1.4090e-01, -5.7074e-02,  1.4612e-01,\n",
       "          -5.5540e-02,  3.4574e-01,  5.5989e-02,  6.0720e-02, -2.4942e-01,\n",
       "           2.6388e-01, -3.5834e-01, -4.7120e-01, -1.1100e-01, -1.5177e-02,\n",
       "          -2.0426e-01,  9.0032e-03, -3.4135e-01, -2.2890e-01, -3.0804e-01,\n",
       "          -5.8986e-01, -4.1812e-01,  3.2717e-01, -2.2889e-01, -8.6319e-02,\n",
       "          -1.3509e-01,  2.9342e-02,  2.7247e-01, -1.1554e-01,  3.5010e-01,\n",
       "           1.3823e+00, -1.2746e-01,  2.3111e-01, -6.9600e-01, -2.5666e-01,\n",
       "          -2.5462e-01,  1.3897e-01, -2.4862e-01,  3.3040e-01, -1.8839e-01,\n",
       "           1.7549e-01, -7.8527e-02,  4.5808e-02, -3.6278e-01,  9.1009e-02,\n",
       "          -8.0103e-01,  5.2738e-01,  1.3883e-01, -1.8624e-02, -3.3765e-01,\n",
       "           1.7784e-01, -1.1684e+00, -2.3791e-01, -2.4528e-01,  1.5445e-01,\n",
       "          -3.2721e-01,  1.4372e-01,  2.7757e-01,  2.1295e-01,  8.5266e-03,\n",
       "          -1.4577e-01,  4.4095e-01,  3.0554e-01,  2.9397e-01,  5.7244e-02,\n",
       "           2.8239e-01,  2.8472e-01,  6.0840e-01,  4.5792e-01,  1.3345e-01,\n",
       "          -2.9677e-01,  2.3646e-01,  4.4149e-03, -9.3967e-02, -4.3160e-02,\n",
       "           6.6917e-02, -8.8529e-02,  2.5131e-01,  1.3798e-01,  1.3470e-03,\n",
       "          -3.1839e-01, -7.7328e-02, -4.3570e-01, -8.7815e-02, -5.7056e-01,\n",
       "           6.1187e-01, -2.3497e-01, -9.0042e-02,  3.1988e-01,  1.7902e-01,\n",
       "           3.2903e-02,  5.9280e-02,  2.9360e-01, -2.8735e-01,  1.1270e-01,\n",
       "           5.6981e-01, -4.5904e-01,  2.1052e-01, -2.4293e-01, -2.4495e-01,\n",
       "           3.4652e-01, -6.7653e-01,  6.0298e-01, -6.1563e-02,  2.4440e-01,\n",
       "           5.1617e-02, -2.8467e-01, -8.6212e-02, -3.2153e-01,  2.5548e-01,\n",
       "          -6.0622e-01, -4.5084e-01, -8.0597e-01,  1.6491e-01,  3.0480e-01,\n",
       "           3.4768e-01, -1.2553e-01,  7.9463e-01,  3.1520e-01,  4.6370e-01,\n",
       "          -3.6454e-01, -4.1172e-01,  2.2748e-01, -1.9795e-01, -2.5553e-01,\n",
       "          -1.8345e-01, -2.1838e-01, -4.0190e-01,  3.7516e-01, -1.1286e-01,\n",
       "          -1.4414e-01, -2.5778e-01,  4.2261e-01,  3.4187e-01, -5.9433e-02,\n",
       "          -2.1422e-01,  7.9390e-02, -3.2845e-01, -1.8757e-01,  4.2486e-01,\n",
       "          -3.6646e-01,  8.7340e-02,  1.5011e-01, -4.4196e-01, -5.0876e-01,\n",
       "          -1.6671e-01,  3.4433e-01,  3.3656e-02,  3.7497e-01, -8.6811e-01,\n",
       "          -6.6793e-01,  3.3937e-01,  4.9178e-02,  1.4245e-01,  2.7107e-01,\n",
       "           6.1380e-01, -2.0973e-01, -2.9792e-02, -8.7256e-02, -4.5333e-01,\n",
       "          -4.8164e-02, -2.8547e-02, -1.2605e-01,  5.3627e-02,  1.6174e-01,\n",
       "           2.7959e-01,  1.4621e-01,  2.0760e-01,  4.1144e-02, -3.6209e-01,\n",
       "           8.2500e-05,  2.0645e-02, -1.4630e-01, -7.8761e-02,  2.0669e-01,\n",
       "          -9.4222e-02,  4.8781e-01,  4.6651e-01, -1.5176e-01, -8.4299e-01,\n",
       "           1.6623e-01,  3.7859e-01, -2.0847e-01, -8.4822e-03,  1.1683e-02,\n",
       "          -8.5689e-01, -2.6224e-01, -3.9478e-02, -1.8666e-01, -1.6264e-01,\n",
       "          -3.9523e-02,  3.6815e-01,  7.5647e-02, -5.8827e-02, -5.1407e-01,\n",
       "           2.2193e-01,  3.1436e-01,  2.2781e-01,  2.9344e-01, -1.7430e-01,\n",
       "          -1.8739e-01, -3.7203e-01, -3.2899e-01, -4.6635e-01, -5.2188e-01,\n",
       "           3.7793e-01,  3.2523e-01, -1.4899e-01, -2.5656e-01, -2.4943e-01,\n",
       "           4.2126e-01, -7.2891e-01, -4.7008e-01,  1.6416e-01, -4.0861e-01,\n",
       "          -5.1945e-01,  4.9533e-01, -4.2333e-01,  6.4556e-02,  1.2214e-01,\n",
       "           2.0372e-01, -4.1686e-01, -1.2643e-01, -1.6071e-01,  5.1965e-01,\n",
       "           5.5214e-01,  4.4166e-01, -1.7568e-01,  3.9089e-01,  3.7936e-01,\n",
       "          -2.3190e-01,  3.3155e-01, -2.5298e-01,  2.1932e-01, -1.4975e-01,\n",
       "           4.7046e-02,  3.5250e-01, -6.9045e-01,  1.3206e-01,  2.5244e-01,\n",
       "          -2.8573e-01, -3.3056e-01, -1.7352e-01,  2.8385e-01, -2.0805e-01,\n",
       "          -3.3376e-01, -2.9907e-01,  5.2522e-02,  2.2873e-02, -1.2435e-01,\n",
       "           4.6982e-01,  6.4071e-02,  1.9047e-01, -3.7265e-01, -2.9794e-01,\n",
       "          -2.0368e-01,  1.3210e-02,  7.3120e-01, -3.6242e-02, -3.6823e-02,\n",
       "           1.7504e-02, -1.1176e-01, -1.6023e-01,  7.5185e-01,  9.7705e-02,\n",
       "          -5.5185e-01,  2.7978e-01,  1.6985e-02,  7.0694e-02,  5.7346e-01,\n",
       "          -2.1871e-01, -1.4957e-01, -2.1050e-02, -3.5421e-02, -2.7988e-02,\n",
       "          -2.8130e-01,  3.0359e-01,  2.2880e-01, -9.3723e-02,  2.5413e-01,\n",
       "           3.7027e-01,  2.5631e-01, -4.9875e-01, -1.1718e-01,  1.1090e-02,\n",
       "           1.4668e-01, -4.1129e-02, -1.0215e-01, -6.1850e-01, -4.5658e-01,\n",
       "           3.2395e-01,  4.0024e-01,  6.7697e-01,  3.1178e-01, -1.1266e+01,\n",
       "           3.7360e-01,  8.5277e-01, -4.6481e-01,  1.2759e-02,  2.0652e-01,\n",
       "           3.2077e-01, -1.8772e-01,  5.1744e-01, -9.7468e-02, -3.4391e-01,\n",
       "          -7.6599e-02, -2.3395e-01,  1.1837e-01, -3.7392e-01, -2.5743e-02,\n",
       "           2.6901e-02, -6.7948e-01,  6.8444e-01, -2.4779e-01,  4.9145e-01,\n",
       "          -3.5104e-01,  1.6295e-01,  2.1764e-01, -5.7189e-01, -2.1191e-01,\n",
       "           7.9445e-02, -2.6870e-01,  3.4715e-01,  6.8729e-01, -5.5151e-02,\n",
       "           1.3270e-01,  2.4529e-02,  8.0028e-02, -4.1907e-01, -6.0072e-01,\n",
       "           1.2458e-01, -3.0097e-01,  4.9566e-01,  5.5341e-02, -3.6186e-01,\n",
       "          -8.4715e-02, -9.8033e-02,  4.9700e-01,  1.2856e-01,  6.8896e-01,\n",
       "           6.0841e-01,  7.7473e-01,  1.9114e-02, -4.1309e-01,  2.4269e-01,\n",
       "          -1.0750e+00, -6.2496e-02, -5.4137e-01,  8.6889e-04, -3.4974e-02,\n",
       "           4.2891e-01, -8.1942e-01,  3.1201e-01, -1.1282e-01, -2.4048e-02,\n",
       "           5.7438e-01, -1.2207e-01,  9.4243e-02, -2.9175e-01,  1.4004e-02,\n",
       "           2.8490e-01, -1.1348e-01,  3.1717e-01, -7.3247e-02, -3.1326e-01,\n",
       "          -3.5685e-01,  6.8181e-01,  4.8985e-01, -5.9240e-01,  1.4252e-01,\n",
       "           1.9440e-01, -2.4868e-01,  1.7931e+00,  1.6264e-01, -1.4107e-01,\n",
       "           2.9478e-02,  2.2498e-01, -3.9107e-01, -4.5466e-01,  1.9977e-01,\n",
       "          -1.1395e-01,  1.7371e-01,  2.4123e-01,  3.7107e-01, -1.7930e-01,\n",
       "          -4.2296e-01, -1.0254e-01,  1.5780e-01, -1.4469e-01,  7.2007e-01,\n",
       "          -1.0424e+00,  6.7988e-02,  2.1149e-01, -1.6420e-02, -6.2597e-01,\n",
       "           4.1726e-02, -4.2537e-01, -4.2227e-01,  4.0702e-01, -1.3935e-01,\n",
       "          -1.0427e-01, -3.6507e-02,  8.1093e-02,  3.9013e-01, -1.0261e-02,\n",
       "           2.0793e-01,  2.9305e-01,  7.4550e-02, -1.4051e-01,  2.5627e-01,\n",
       "           5.5070e-01,  5.8657e-01, -4.5700e-01, -3.7258e-01, -1.5872e-01,\n",
       "           1.7076e-01,  2.7615e-01,  3.1854e-01, -7.3882e-02,  4.5360e-01,\n",
       "          -6.6939e-01, -2.7130e-01,  3.9645e-02, -1.3322e-01,  1.5713e-01,\n",
       "           1.6773e-01,  1.0803e-01,  1.3557e-01, -4.2337e-01, -2.2174e-01,\n",
       "          -8.4059e-01, -2.6599e-01,  1.1606e-01, -1.7242e-01, -4.3838e-02,\n",
       "          -8.7855e-02, -1.1833e-01,  8.9517e-02,  2.7409e-01,  1.4283e-01,\n",
       "           4.0897e-01,  1.4803e-01, -4.7874e-01,  6.1070e-02, -1.4947e-01,\n",
       "          -6.1474e-01,  1.6703e-01, -3.1721e-01,  1.3127e-03, -2.9079e-01,\n",
       "          -2.5875e-01,  4.9758e-02, -6.7728e-02, -5.2831e-01,  3.5582e-01,\n",
       "          -2.8725e-02,  7.1028e-02, -4.3343e-01,  4.1554e-01,  2.0487e-01,\n",
       "           2.6759e-02, -3.8398e-01, -3.9704e-01, -1.2265e-01,  1.7679e-01,\n",
       "           3.0650e-01, -3.2042e-01,  4.3845e-02, -4.2510e-01, -2.2416e-01,\n",
       "          -5.8356e-02,  1.0031e+00,  2.2956e-01, -7.8291e-01,  2.6870e-01,\n",
       "           8.3847e-01, -3.0303e-01,  4.8139e-02,  5.2892e-02, -7.8655e-02,\n",
       "          -6.0784e-01, -2.4458e-01,  6.5344e-01, -1.5486e-01,  1.0818e-01,\n",
       "           3.9831e-02,  2.4545e-01, -3.5897e-01,  2.9855e-01, -1.6963e-01,\n",
       "          -4.0057e-01,  4.9056e-02, -6.8206e-01,  3.3116e-01,  1.9574e-01,\n",
       "          -4.4569e-01,  2.4331e-01,  6.3450e-02, -1.4575e-02, -2.6418e-01,\n",
       "           2.1485e-01,  6.5055e-01, -1.7290e-01,  2.8123e-02,  5.3741e-01,\n",
       "          -2.5026e-02, -2.4267e-01,  2.8803e-01,  9.9112e-02, -5.1954e-02,\n",
       "           2.6703e-01, -2.9070e-02, -2.4585e-01, -3.7770e-02,  2.9516e-02,\n",
       "           1.4813e-01, -7.9955e-01, -9.5084e-01, -2.2358e-02,  3.8439e-01,\n",
       "          -1.0012e-01,  2.7254e-01, -3.3573e-02,  3.8565e-01,  4.4163e-01,\n",
       "          -1.0450e+00, -3.3045e-01, -7.5275e-02,  2.1010e-01, -2.9979e-02,\n",
       "          -4.3269e-02, -1.0389e-01, -4.6724e-01, -3.8535e-01,  4.7262e-01,\n",
       "           5.2273e-01, -6.5987e-01, -4.1867e-02,  5.2306e-01, -4.2618e-01,\n",
       "           4.1542e-01, -1.0737e-01, -3.2170e-02,  4.6298e-01,  1.0735e-01,\n",
       "          -1.2095e-01,  2.7621e-01, -4.1062e-01, -5.1064e-01,  2.1178e-01,\n",
       "           9.1987e-01, -1.7667e-01,  1.6079e-01, -2.2317e-01, -2.7954e-01,\n",
       "          -8.6583e-02, -4.5509e-01, -3.6335e-01,  1.4705e-01,  9.4002e-03,\n",
       "           1.9365e-02, -6.9529e-03,  6.7343e-02,  2.7688e-01, -2.0386e-01,\n",
       "           1.9497e-01,  7.7627e-01,  1.0166e+00,  1.4677e-01, -2.4814e-01,\n",
       "           1.3630e-01,  4.0572e-02, -2.8357e-01,  3.4766e-01, -1.8538e-01,\n",
       "          -3.5074e-01,  6.3041e-02,  1.9903e-01, -6.7512e-01, -2.7503e-01,\n",
       "          -2.8906e-01,  3.0544e-01, -1.3470e-01, -1.9002e-01,  2.7396e-01,\n",
       "          -1.0885e-01,  2.8519e-01, -2.3913e-01,  4.2042e-01,  4.1865e-01,\n",
       "           1.6928e-01,  5.6661e-01,  1.8559e-01,  1.8026e-01, -4.5929e-01,\n",
       "           3.0212e-01, -9.3606e-01, -4.3186e-01, -3.7995e-02, -2.2053e-01,\n",
       "          -3.5666e-01,  2.1509e-01,  2.1856e-02,  3.0167e-01,  8.3403e-01,\n",
       "           7.3587e-01, -1.1088e-01, -8.3563e-01, -2.2532e-01, -4.7178e-02,\n",
       "           1.7178e-01, -2.3846e-01,  1.2287e-01, -2.5986e-01,  9.2987e-02,\n",
       "          -5.0824e-01, -3.9447e-01, -2.5987e-02,  3.1790e-01, -2.4947e-01,\n",
       "           6.1633e-01, -1.2458e-01,  8.6852e-02,  3.8443e-01, -2.7125e-02,\n",
       "           1.5641e-01,  9.5795e-01,  3.0130e-01, -7.5649e-01,  1.8470e-01,\n",
       "           1.5053e-01, -7.4613e-02, -4.5115e-02, -5.7978e-01,  3.0133e-01,\n",
       "          -2.7175e-02,  6.7042e-01,  4.6435e-01]], device='mps:0'),\n",
       " tensor([[-0.1139, -0.4936, -0.5114,  ...,  0.1629,  0.1837,  0.4606],\n",
       "         [-0.4643, -0.3750, -0.2078,  ...,  0.2835,  0.5093,  0.4110]],\n",
       "        device='mps:0'),\n",
       " tensor([[-5.1126e-02, -2.8501e-01, -3.8529e-01, -7.2268e-01, -5.4203e-01,\n",
       "          -3.0172e-02, -5.3191e-01,  6.3552e-01,  1.0589e-01,  4.4171e-01,\n",
       "           3.0159e-01,  3.8554e-01,  1.2016e-02,  1.7563e-01, -7.0440e-01,\n",
       "          -3.1046e-01,  2.1502e-03,  2.0064e-01,  6.3957e-01,  5.4072e-01,\n",
       "           6.9292e-01,  2.0383e-02,  1.7950e-01, -8.3376e-03, -7.4422e-01,\n",
       "           3.1356e-01, -1.1968e-01,  2.7317e-02,  9.5952e-02, -4.1735e-01,\n",
       "          -3.9705e-01,  4.2076e-01, -1.7393e-01,  8.7339e-01, -1.6545e-01,\n",
       "          -2.6821e-01,  1.5444e-01, -2.3351e-01,  1.7837e-01, -1.6480e-01,\n",
       "           4.2368e-01, -1.2277e-01,  3.5561e-01,  2.3946e-02,  2.1155e-01,\n",
       "          -2.3307e-02,  7.0184e-02, -5.7189e-01, -4.7950e-02, -2.7651e-01,\n",
       "          -1.6402e-01, -1.5482e-01,  1.6208e-01,  1.6600e-01, -8.5511e-01,\n",
       "          -6.2832e-01,  1.5788e-01,  2.2054e-01, -5.6742e-01,  4.9031e-03,\n",
       "          -3.2868e-01,  9.4137e-01, -1.7578e-01,  4.7376e-02, -1.6037e-01,\n",
       "           4.4001e-02,  2.6169e-01, -9.9389e-01, -5.3365e-01, -1.4769e-01,\n",
       "          -1.2559e-01,  2.4374e-01, -1.8493e-01, -2.3438e-01,  5.2561e-01,\n",
       "          -1.9246e-01,  2.1888e-01,  1.9117e-02, -3.3819e-01, -1.1843e-01,\n",
       "           3.2432e-01,  2.8103e-01,  3.4206e-02, -1.5008e-01, -5.8829e-01,\n",
       "          -1.3160e-02,  1.6431e-01,  6.4090e-01,  3.4175e-01, -2.9114e-01,\n",
       "           3.2411e-01,  5.1269e-02, -2.5643e-01, -2.8929e-01,  4.6619e-01,\n",
       "          -1.4166e-01,  1.0833e-01, -1.5827e-01,  1.5258e-01, -5.1464e-01,\n",
       "           2.0347e-01,  3.3114e-01, -1.3157e-01,  3.9068e-01, -4.4874e-01,\n",
       "          -6.6845e-01,  4.9142e-01,  2.7571e-02, -8.3493e-01, -1.6651e-01,\n",
       "          -1.5476e-01, -4.9717e-01,  2.2506e-01, -4.0295e-01, -4.4835e-01,\n",
       "           4.3978e-02, -1.7085e-01,  1.6180e-02, -4.2683e-01,  3.9505e-01,\n",
       "           3.3727e-01, -1.5996e-01,  6.1175e-02,  9.7501e-02, -4.3175e-02,\n",
       "          -1.8437e-01,  2.4336e-01,  2.3235e-02,  3.4376e-01,  3.3634e-01,\n",
       "           3.6402e-02,  1.7338e-01, -4.8474e-01,  1.2624e-01,  1.2892e-01,\n",
       "           1.9847e-01, -2.7336e-01, -2.3810e-01,  4.4674e-01,  4.8440e-02,\n",
       "          -1.9511e-01,  5.7870e-02, -3.6001e-02, -1.6339e-02,  1.3800e-01,\n",
       "           1.1280e-02,  1.8276e-01, -1.9202e-01, -2.1533e-01, -1.4171e-01,\n",
       "          -3.1961e-01, -6.3153e-01,  2.8212e-01, -2.8258e-01, -5.1169e-03,\n",
       "          -2.9066e-01,  7.1131e-01, -5.0904e-02, -2.3606e-01,  3.4000e-01,\n",
       "           5.1940e-01, -1.7617e-01,  1.8724e-01, -2.3978e-01, -8.9669e-03,\n",
       "           3.8528e-01, -5.4324e-01, -3.1497e-01, -2.7645e-01,  1.9782e-01,\n",
       "           1.5774e-01, -3.0053e-01,  1.8198e-01, -2.3273e-01, -3.7135e-01,\n",
       "          -3.6919e-01, -1.4353e-02, -1.1401e-01, -3.8424e-01,  3.4923e-01,\n",
       "          -4.6807e-01, -1.7083e+00,  3.6938e-01,  4.6546e-01, -4.4169e-01,\n",
       "          -5.6420e-01,  3.3808e-01,  4.6760e-01,  3.2077e-01, -1.8681e-01,\n",
       "           1.5719e-01, -2.0055e-01,  2.0034e-01,  6.5039e-01, -3.3121e-01,\n",
       "           4.1034e-01,  3.9286e-01, -1.6473e-01,  4.4456e-02,  8.8823e-02,\n",
       "          -6.3506e-01,  1.3750e-01,  2.3443e-01,  6.4788e-02,  9.1671e-02,\n",
       "           5.5027e-02,  2.5430e-01, -1.4151e-01,  9.3404e-02,  1.7616e-01,\n",
       "           1.3217e-01, -2.1121e-01, -4.9334e-02, -6.6079e-02,  1.6504e-01,\n",
       "           6.1088e-01, -9.1898e-02,  9.7443e-02,  6.8057e-01, -4.0278e-02,\n",
       "          -3.7259e-01,  2.6361e-01, -1.1001e-01, -5.9590e-02, -4.4861e-01,\n",
       "           1.9721e-01,  2.8491e-02,  1.6895e-02, -5.5469e-01, -4.3106e-01,\n",
       "          -2.1533e-01, -3.8761e-01,  5.3494e-01,  3.9313e-01, -3.3004e-01,\n",
       "          -1.3308e-01, -1.8949e-01, -8.4665e-02,  7.2744e-02,  1.8211e-01,\n",
       "          -1.5552e-01, -3.0108e-01, -5.9139e-01, -6.6608e-02,  1.4961e-01,\n",
       "           1.9409e-01, -5.9524e-02,  4.0885e-01,  6.1225e-01,  2.6222e-01,\n",
       "           1.3163e-01, -1.2710e-01,  3.8273e-01,  4.2423e-01, -2.3683e-01,\n",
       "           2.5812e-01,  1.0466e-01, -1.4294e-01,  3.8938e-01, -4.4998e-01,\n",
       "           1.8749e-01,  2.2638e-01,  7.3271e-02,  4.2201e-01, -4.1587e-01,\n",
       "           1.9679e-01,  3.3357e-01, -2.8398e-01, -1.6133e-01, -2.5279e-02,\n",
       "          -3.1659e-01,  7.0654e-01,  2.0220e-02, -4.1284e-01, -6.7732e-01,\n",
       "          -2.8434e-01,  1.1274e-01, -5.0061e-02,  8.3906e-02, -8.1079e-01,\n",
       "          -1.6014e-01,  8.0252e-02, -1.7744e-02, -1.3805e-01, -1.3427e-01,\n",
       "           1.7793e-01,  3.8390e-01, -3.8900e-01, -4.1669e-02, -4.3060e-01,\n",
       "          -5.3813e-03, -5.4836e-02, -2.8503e-01,  4.7195e-01,  2.4164e-01,\n",
       "          -3.1330e-01,  4.1537e-02,  2.8968e-03,  2.2939e-01, -2.9854e-01,\n",
       "          -2.1427e-01, -2.5644e-01,  1.6988e-01, -5.5428e-02,  4.1859e-02,\n",
       "           3.3249e-01,  2.7264e-01,  6.3262e-01, -6.2714e-01, -6.5367e-01,\n",
       "          -2.9808e-01,  7.7644e-01, -1.9541e-01, -2.7986e-01, -1.7401e-01,\n",
       "          -1.1282e+00,  3.1393e-02, -1.3006e-01, -4.1241e-01, -3.7452e-01,\n",
       "           2.6864e-01, -2.5032e-01, -2.9976e-01,  4.9473e-02, -3.5195e-01,\n",
       "           6.8677e-01,  6.2315e-01, -4.4296e-01, -2.3640e-02, -2.8714e-01,\n",
       "          -5.3809e-01, -1.0001e-01, -1.9415e-01, -1.3235e-01,  4.5764e-02,\n",
       "           2.1269e-02,  3.1943e-02,  1.2693e-01, -8.2345e-02, -5.2606e-01,\n",
       "           3.2320e-01, -3.8177e-02,  2.7711e-01,  9.9190e-02, -3.4102e-01,\n",
       "           4.7458e-01,  5.3597e-01, -7.7505e-01, -1.3130e-01,  1.2168e-02,\n",
       "          -1.0826e-02, -3.0743e-01, -3.4958e-01,  6.8695e-03,  5.5977e-01,\n",
       "           3.7047e-01,  2.2973e-01, -3.7028e-01,  5.9347e-01,  3.3665e-01,\n",
       "          -4.4719e-01,  5.7400e-01, -1.3493e-02, -4.9672e-01,  2.7300e-01,\n",
       "           2.6518e-01,  7.8713e-02, -3.8528e-01, -1.2151e-01, -2.1130e-01,\n",
       "           3.2781e-01, -5.9963e-01,  1.4531e-01,  1.9579e-01,  8.7510e-01,\n",
       "          -3.4190e-01, -5.5311e-01,  4.2372e-01, -4.9152e-01, -3.4192e-01,\n",
       "           3.7694e-02,  9.3534e-02,  1.3797e-02, -6.5537e-02, -1.5126e-01,\n",
       "          -3.4497e-01,  1.6372e-01,  9.5668e-01,  3.9847e-01, -3.6969e-01,\n",
       "           1.2659e-01, -1.5552e-01, -2.3435e-01,  5.3556e-01, -1.1919e-01,\n",
       "          -1.3737e-01,  5.2284e-01, -2.7189e-01,  1.5629e-01,  4.0450e-01,\n",
       "          -1.9844e-01, -3.0718e-01, -2.2003e-01, -1.7618e-01, -1.9988e-02,\n",
       "          -2.9517e-01,  7.5906e-02, -3.1435e-01,  3.6730e-02,  5.3758e-03,\n",
       "          -1.4090e-01,  3.4359e-01,  8.5591e-02,  1.6030e-01, -2.5526e-01,\n",
       "           6.3391e-02, -1.9289e-01,  2.1254e-01,  4.6103e-02, -4.5114e-01,\n",
       "           7.3039e-01,  1.9319e-01,  3.3147e-01,  7.2208e-01, -1.1280e+01,\n",
       "           4.5015e-02,  2.6078e-01, -3.0236e-01, -5.4087e-02, -3.6394e-02,\n",
       "           5.8352e-02, -1.5857e-01,  1.3550e-01, -2.4423e-01, -1.7170e-01,\n",
       "          -3.8143e-02,  3.5513e-02,  1.7697e-01,  1.1254e-01, -3.1194e-01,\n",
       "          -1.6458e-02, -8.3978e-01,  7.5213e-02, -6.3580e-02,  2.5615e-01,\n",
       "          -3.3918e-01,  2.0090e-01,  1.7773e-01,  2.5706e-02,  2.4989e-01,\n",
       "          -4.0555e-02, -1.3146e-01,  6.6927e-01,  2.1160e-01,  2.4003e-01,\n",
       "           2.9471e-01,  2.2402e-01, -1.5102e-01, -2.7826e-01, -3.4201e-01,\n",
       "          -2.4018e-01, -2.3187e-01,  1.2014e-02,  2.6670e-02, -3.1118e-01,\n",
       "          -1.1623e-01, -3.2685e-01,  2.5905e-01,  5.4841e-01,  7.6415e-01,\n",
       "           1.1161e+00,  3.9738e-01,  4.5882e-01,  1.5860e-01,  4.5109e-02,\n",
       "          -8.8534e-01,  1.5949e-01, -7.7589e-01,  1.7843e-01, -2.6090e-01,\n",
       "           9.2598e-01, -5.6313e-01,  3.2246e-02, -5.3656e-01, -3.4600e-01,\n",
       "           3.6217e-01,  4.2020e-01, -3.2840e-01, -4.4346e-01,  7.9233e-02,\n",
       "           2.3218e-01, -2.6861e-01,  1.1006e-01, -1.6650e-01,  3.1470e-01,\n",
       "          -2.1785e-01,  1.5494e-01,  1.4058e-01, -4.0322e-01,  1.8147e-01,\n",
       "           1.7401e-01,  1.7939e-03,  1.4425e+00, -1.0687e-01, -7.9383e-02,\n",
       "           5.8210e-02,  1.3425e-01, -2.2002e-01, -9.4803e-01, -9.5233e-02,\n",
       "           3.1095e-01,  2.1332e-02,  2.6035e-01,  2.4617e-01, -3.2873e-01,\n",
       "          -7.4567e-01, -1.1429e-01,  1.3796e-01,  5.5189e-01,  4.9469e-01,\n",
       "          -8.2520e-01, -1.3089e-01,  1.2396e-01,  1.7384e-01, -6.3882e-01,\n",
       "          -1.6101e-02, -3.9591e-01, -2.2820e-01,  4.5508e-02, -7.9702e-02,\n",
       "           6.9409e-02, -6.1165e-02,  5.5654e-02,  5.0870e-02, -3.3012e-01,\n",
       "           1.3823e-01,  1.5222e-01,  2.0560e-02, -1.5455e-01,  9.0791e-02,\n",
       "          -1.4331e-03,  5.2172e-01, -1.2646e-01, -5.2627e-01,  8.5098e-02,\n",
       "          -1.3950e-01,  2.1497e-01,  3.5242e-02, -9.9167e-02,  6.0032e-01,\n",
       "          -4.6024e-01, -2.5221e-02,  2.8253e-01,  3.8809e-01,  3.5782e-01,\n",
       "           2.2473e-02, -2.0231e-01,  4.0447e-01, -9.2568e-01, -9.9207e-01,\n",
       "          -6.0392e-01,  1.6451e-01,  3.0170e-02, -2.4210e-01, -1.6654e-01,\n",
       "           2.0493e-01,  1.8613e-01, -2.0664e-01,  2.6577e-01,  4.3194e-01,\n",
       "           6.5809e-01,  6.3892e-01, -3.4559e-01,  1.7555e-01, -2.8982e-01,\n",
       "          -6.6889e-01, -2.5098e-01, -2.7379e-02, -1.3643e-01, -4.9237e-01,\n",
       "          -2.7474e-01, -1.3966e-01, -5.7761e-01, -1.1064e-01,  3.0852e-01,\n",
       "          -2.5571e-01,  7.6489e-02,  7.4931e-02,  1.6453e-01,  3.4748e-01,\n",
       "           2.7378e-01, -4.2897e-01,  7.3819e-02,  9.0574e-02,  5.2725e-01,\n",
       "           5.4063e-02,  3.0601e-01, -5.5019e-01, -8.3042e-02, -4.0384e-02,\n",
       "           1.8055e-01,  8.2633e-01, -8.6766e-02, -2.2549e-01, -6.6482e-03,\n",
       "           4.1994e-01, -1.9313e-02, -2.5039e-01,  2.3708e-01, -2.6592e-01,\n",
       "          -1.4271e-02, -6.5444e-01,  1.4642e-01,  1.5499e-01,  4.8188e-01,\n",
       "           1.2150e-01, -1.1962e-01, -1.1160e-01,  3.4576e-01, -8.1212e-03,\n",
       "          -3.8413e-01, -1.4821e-01, -3.7468e-01,  8.2507e-02, -2.1171e-03,\n",
       "          -2.5816e-01, -8.7727e-02, -7.4530e-02,  8.7949e-02,  4.9076e-01,\n",
       "           1.3156e-01,  1.2498e+00,  1.3982e-01, -4.3459e-01,  9.6834e-01,\n",
       "           3.2720e-01, -1.2725e-01,  3.2857e-01, -1.7345e-01,  3.0437e-01,\n",
       "           3.2208e-01,  2.6840e-02,  1.0006e-01,  1.7471e-01,  1.8776e-01,\n",
       "           5.1972e-01, -8.5473e-01, -2.6929e-01, -1.5768e-01,  3.0280e-01,\n",
       "          -2.1880e-01,  4.6427e-01, -4.0641e-03,  3.6329e-01,  6.3658e-01,\n",
       "          -1.1663e+00, -1.0814e-01, -2.8106e-01,  4.8889e-01,  2.1263e-01,\n",
       "          -2.9070e-01, -1.2494e-01, -2.0081e-01, -4.1487e-01,  3.2257e-01,\n",
       "           6.8477e-01, -1.0990e+00,  1.6568e-01,  5.2005e-01, -3.7122e-02,\n",
       "           2.1747e-01, -2.9042e-01,  2.4947e-01, -3.5699e-01,  4.0594e-01,\n",
       "          -9.0899e-02,  1.4992e-01, -3.1097e-01, -6.8830e-01,  1.9605e-01,\n",
       "           5.2137e-01,  2.5203e-01,  8.2648e-02,  7.6049e-03,  8.4035e-02,\n",
       "          -2.0300e-01, -2.1429e-01, -2.1871e-01,  2.6814e-01, -2.4822e-01,\n",
       "           5.0949e-02, -3.7260e-01,  2.4027e-01,  1.8445e-01,  5.6072e-02,\n",
       "           1.7156e-01,  4.0126e-01,  6.4522e-01, -1.5220e-01, -1.0017e-01,\n",
       "           3.1798e-01,  3.4738e-01, -1.7020e-01,  6.2769e-01, -1.6404e-01,\n",
       "          -4.8061e-01,  4.1298e-01,  1.8282e-01, -2.0559e-01,  4.9050e-01,\n",
       "          -3.3075e-01,  7.5206e-02,  6.5926e-03, -4.3996e-01,  6.2009e-02,\n",
       "          -3.2148e-02, -5.9382e-02, -4.4113e-01,  1.0753e-01,  4.8802e-01,\n",
       "          -2.3655e-01,  1.0645e+00, -3.1276e-01,  5.7433e-01, -7.6750e-01,\n",
       "           2.9171e-01, -1.0004e+00, -2.0646e-01,  8.2582e-04, -5.1013e-01,\n",
       "          -5.2405e-01, -2.0525e-01,  1.1584e-01,  2.9036e-01,  6.4602e-01,\n",
       "           3.6055e-01, -2.8261e-01, -7.9490e-01, -3.5681e-01,  3.5302e-01,\n",
       "          -2.4094e-02,  2.1132e-03,  3.8885e-01,  2.4644e-01,  3.3682e-01,\n",
       "          -7.2307e-01, -4.1258e-01, -7.6408e-02,  4.0116e-01, -3.0530e-01,\n",
       "           6.8111e-01,  1.2574e-01, -8.3136e-02,  1.6910e-01,  3.4142e-02,\n",
       "           4.3701e-01,  3.4014e-01, -3.8497e-01, -7.2370e-01, -1.6265e-01,\n",
       "          -9.8423e-02, -5.1496e-01,  9.6989e-02, -8.8662e-01,  3.7690e-01,\n",
       "           4.0243e-01,  5.4996e-01,  6.3497e-01]], device='mps:0'),\n",
       " tensor([[-0.2485, -0.5584, -0.4741,  ...,  0.2258,  0.4312,  0.7559],\n",
       "         [-0.2427, -0.6267, -0.1893,  ...,  0.2335,  0.0749,  0.3544],\n",
       "         [-0.3292, -0.2590, -0.5016,  ..., -0.0148,  0.3509,  0.1607]],\n",
       "        device='mps:0'),\n",
       " tensor([[-1.9400e-01, -4.9422e-01, -6.2531e-01, -5.6268e-01, -4.6672e-01,\n",
       "          -2.2676e-01, -6.5807e-01,  2.7116e-01, -3.0350e-01,  5.1783e-01,\n",
       "           1.4256e-01, -6.4564e-02,  9.4674e-02,  1.1344e-01, -5.0581e-01,\n",
       "          -3.4350e-01,  1.5847e-01,  5.5723e-01,  7.0455e-01,  4.0245e-01,\n",
       "           7.3582e-01,  2.5639e-01, -1.3035e-01,  1.4386e-01, -8.5328e-01,\n",
       "           4.1517e-01, -1.5595e-01,  8.9291e-02,  3.2285e-01, -8.6096e-01,\n",
       "          -9.0626e-02,  1.2928e-01, -4.7080e-01,  1.0615e+00, -2.4960e-01,\n",
       "          -2.9692e-01,  1.5351e-01, -7.8985e-01,  2.6603e-01, -1.4188e-01,\n",
       "           3.7648e-01, -6.1428e-02,  1.6802e-01,  2.0368e-01,  3.1514e-01,\n",
       "           1.0768e-02,  1.2834e-03, -5.1247e-01, -2.2490e-02, -1.9487e-01,\n",
       "           2.1118e-01,  1.0015e-01,  5.5624e-01,  5.2776e-01, -1.0144e+00,\n",
       "          -7.6294e-01,  1.3003e-01,  2.7043e-01, -3.2939e-01,  3.4255e-01,\n",
       "          -2.2605e-01,  5.4455e-01,  2.5090e-01,  2.3896e-01, -3.1222e-01,\n",
       "           7.2382e-02, -4.4337e-01, -6.2591e-01, -3.7993e-01, -2.9316e-01,\n",
       "           8.4630e-02,  1.2320e-01,  3.3439e-01, -2.0319e-01,  1.3937e-01,\n",
       "          -4.9484e-01,  1.3212e-01, -1.7603e-01, -4.1495e-01, -4.9826e-02,\n",
       "           4.5542e-01,  1.0303e-02, -2.7600e-02, -9.3869e-02, -1.2975e-01,\n",
       "           1.0553e-01,  5.8671e-01,  2.9060e-01,  4.4662e-01,  3.0419e-02,\n",
       "           1.6668e-01,  1.3469e-01, -4.1910e-01, -4.5835e-01,  3.4583e-01,\n",
       "          -4.0973e-01,  1.7416e-01, -2.5138e-01,  1.1001e-01, -2.2705e-01,\n",
       "           1.6164e-01,  3.0715e-01, -5.5869e-02,  3.6037e-01, -6.3292e-01,\n",
       "          -4.2189e-01, -9.6632e-02, -2.7297e-01, -8.1894e-01, -1.1278e-01,\n",
       "          -9.6049e-02, -2.6028e-01,  1.2394e-01, -4.6499e-01, -6.3680e-01,\n",
       "          -1.6504e-01, -2.3578e-01,  2.8164e-01, -2.0064e-01,  7.1296e-01,\n",
       "           2.9546e-01,  2.8437e-02,  2.5572e-01,  2.0410e-01,  3.8045e-01,\n",
       "          -5.0013e-01,  3.4844e-01, -2.9484e-02,  3.9427e-01,  4.4584e-01,\n",
       "          -9.8696e-02,  1.0963e-01, -1.1878e-01, -2.8733e-01,  1.0365e-01,\n",
       "          -2.1877e-01,  1.4233e-01, -5.0690e-01, -2.4474e-01,  2.0514e-01,\n",
       "          -2.8813e-01, -3.0925e-02,  2.6758e-02,  1.3106e-01, -1.1313e-01,\n",
       "           1.0749e-01,  7.9991e-02, -1.2717e-01, -2.3391e-01, -2.6376e-01,\n",
       "          -5.0670e-01, -3.2999e-01,  2.6123e-01, -4.1238e-02, -2.3992e-01,\n",
       "          -2.1807e-01,  5.0003e-01, -6.9080e-02, -2.0149e-01, -7.2599e-02,\n",
       "           8.4191e-01,  8.1708e-02,  2.9127e-01, -3.8673e-01, -7.1779e-02,\n",
       "           3.3893e-01, -1.7026e-01,  6.0155e-03, -7.8037e-03,  3.2157e-02,\n",
       "           2.0963e-01, -1.1523e-01,  2.8010e-01, -4.1352e-02,  9.0722e-02,\n",
       "          -6.3774e-01,  4.9111e-01,  1.4262e-01,  1.7635e-01, -2.5070e-01,\n",
       "          -7.5605e-01, -1.8745e+00,  3.5670e-01, -3.2605e-01,  3.5185e-01,\n",
       "          -2.7486e-01,  4.0581e-01,  3.1805e-01,  2.1925e-02, -1.1416e-01,\n",
       "           1.8002e-02,  4.3369e-02,  2.9385e-01,  5.9928e-01, -2.5712e-01,\n",
       "           4.4481e-01,  5.4815e-01,  3.6279e-01,  2.0757e-01, -1.6196e-01,\n",
       "          -6.4526e-01,  2.8719e-01,  8.0839e-02,  3.5264e-01, -3.1187e-02,\n",
       "           3.6331e-01,  1.4893e-01, -6.6528e-02,  4.5502e-01, -4.0972e-02,\n",
       "          -7.8619e-02, -5.2894e-01, -5.3711e-01, -2.5801e-01, -1.6033e-01,\n",
       "           4.5533e-01, -2.1475e-01, -2.4017e-01,  3.5092e-01,  2.5787e-01,\n",
       "          -2.4175e-01, -3.0517e-01, -2.9779e-01, -3.2930e-01, -4.5461e-01,\n",
       "           2.9046e-01, -2.9904e-01, -1.5997e-02, -1.6093e-01, -2.5913e-01,\n",
       "           6.4656e-01, -6.1734e-01,  7.3106e-01,  1.2839e-01,  1.0675e-01,\n",
       "          -4.8161e-02, -4.2873e-01,  7.7549e-02, -2.2963e-01,  1.8372e-01,\n",
       "           5.5366e-02, -6.2915e-01, -6.3638e-01,  1.7201e-01,  2.3098e-01,\n",
       "           4.2752e-01, -1.3956e-01,  6.3251e-01,  5.4821e-01,  4.1150e-01,\n",
       "           3.6794e-02, -1.6338e-01,  2.7844e-01,  2.8758e-01, -2.9783e-01,\n",
       "           3.8240e-01, -2.2844e-02, -3.8414e-01,  7.8917e-02, -6.7327e-01,\n",
       "          -2.7974e-01, -1.6353e-01,  1.7876e-01,  2.2380e-01, -2.7251e-01,\n",
       "           1.0734e-01,  3.7475e-01, -3.1694e-01, -8.2077e-02, -1.6679e-02,\n",
       "          -3.7002e-01,  5.4541e-01,  2.8331e-02, -2.9312e-01, -6.0521e-01,\n",
       "          -6.0153e-01,  1.3094e-01, -6.0761e-02,  7.7238e-02, -8.8505e-01,\n",
       "           2.1323e-01,  5.9244e-01, -1.5904e-01, -2.9963e-02, -2.8886e-01,\n",
       "           1.9218e-01,  5.7932e-02, -1.2457e-01,  8.1214e-02, -8.2646e-01,\n",
       "          -3.1652e-01,  3.1422e-02, -1.8173e-01,  3.5255e-01,  2.4611e-01,\n",
       "           1.6074e-02, -4.1914e-01,  1.0984e-01,  2.1972e-02, -3.1093e-01,\n",
       "           1.6964e-02, -6.4328e-02, -5.3006e-02, -1.2690e-01, -2.6615e-01,\n",
       "          -1.2943e-01,  2.2449e-01,  4.0817e-01, -2.2190e-01, -6.3576e-01,\n",
       "           1.0995e-02,  7.9174e-01, -7.6358e-02, -1.6495e-02, -4.9436e-02,\n",
       "          -1.2160e+00, -2.8229e-01,  1.4753e-01, -3.2710e-01, -2.8888e-01,\n",
       "           4.5095e-01, -6.2748e-01,  9.5548e-02,  8.3817e-02, -4.7348e-01,\n",
       "           4.0565e-01,  6.7499e-01, -4.5424e-02,  8.4168e-02, -3.7722e-01,\n",
       "           5.2154e-02,  1.0145e-01, -9.1103e-03, -2.5627e-01, -3.6931e-02,\n",
       "           1.5962e-01,  3.1507e-01, -2.9503e-02, -1.5136e-01, -1.9913e-01,\n",
       "           4.6195e-01, -1.1944e-01,  1.8693e-01,  1.6665e-01, -7.1570e-02,\n",
       "          -2.9111e-01,  5.5803e-01, -2.4376e-01, -1.2211e-01, -4.0299e-01,\n",
       "           9.5140e-02,  4.3383e-02, -3.5889e-02,  3.2394e-01,  7.3829e-01,\n",
       "           3.6792e-01,  1.8927e-01, -8.2351e-02,  7.0964e-01,  4.8409e-01,\n",
       "          -5.5335e-01,  1.3180e-01,  5.1269e-01, -1.4870e-01, -1.6526e-01,\n",
       "           8.6647e-02,  3.1457e-01, -3.0487e-01,  9.5268e-03,  1.9314e-01,\n",
       "           4.1095e-01, -5.4423e-01,  1.5988e-01,  2.4126e-01,  2.9669e-01,\n",
       "          -1.2125e-01, -3.2533e-01,  3.7381e-01, -4.8872e-01, -6.3806e-01,\n",
       "           2.0963e-01, -1.1946e-01,  5.8886e-02, -3.6657e-01, -2.0282e-01,\n",
       "          -3.2470e-01, -7.3259e-02,  5.6744e-01,  1.8299e-01, -3.0588e-01,\n",
       "           7.5567e-02, -9.5894e-02, -3.5132e-01,  9.8344e-02, -2.6695e-01,\n",
       "          -2.8672e-01,  4.9516e-02, -1.5842e-01,  6.3465e-02,  3.3160e-02,\n",
       "          -3.2098e-01, -1.0028e-01, -1.6388e-02, -2.0612e-01,  4.0933e-02,\n",
       "          -3.6975e-01,  6.6398e-02, -1.1624e-02, -6.9892e-02, -9.4167e-03,\n",
       "           4.9075e-01,  6.3873e-01, -4.4224e-01,  5.2307e-01, -4.2838e-02,\n",
       "           8.2747e-02, -3.7947e-01,  2.0474e-01, -8.3240e-02, -7.1869e-01,\n",
       "           7.5591e-01,  5.7971e-01,  2.1585e-01,  2.4918e-01, -1.1289e+01,\n",
       "          -2.0507e-01,  7.2147e-01, -4.5404e-01, -7.6179e-02,  4.1232e-02,\n",
       "           2.7294e-01, -2.3829e-01, -2.1903e-01, -1.1365e-01, -4.2311e-01,\n",
       "          -1.0917e-01,  6.9672e-02, -6.4551e-02, -4.0637e-01, -2.5870e-01,\n",
       "          -3.4716e-01, -6.1491e-01,  2.7619e-01, -6.6507e-02,  4.3677e-01,\n",
       "          -2.2608e-01,  3.5006e-01,  9.8410e-02, -2.4560e-01, -2.0605e-01,\n",
       "          -2.0218e-01, -1.0883e-01,  3.1177e-01,  2.8157e-01, -2.7214e-01,\n",
       "           2.4058e-01, -1.9657e-01,  3.2429e-01, -2.9483e-01, -5.0567e-01,\n",
       "          -8.8671e-02, -2.5229e-01,  4.7805e-01, -2.6967e-01, -3.0327e-01,\n",
       "          -2.9805e-01, -5.0126e-01,  1.3977e-01,  8.1988e-02,  7.6412e-01,\n",
       "           1.1732e+00,  4.8170e-01,  2.9421e-01,  8.9610e-02,  2.3257e-01,\n",
       "          -6.7898e-01,  3.1162e-01, -3.9110e-01,  1.0328e-01, -7.6706e-02,\n",
       "           7.7747e-01, -6.3993e-01, -2.7178e-01,  2.3355e-01, -1.7761e-01,\n",
       "           4.4284e-01,  2.5765e-01,  1.6110e-01, -1.7149e-01, -3.0838e-01,\n",
       "           7.7001e-01,  4.6394e-02,  3.8015e-01, -3.1799e-01,  4.6805e-02,\n",
       "          -1.9610e-02,  3.7031e-01,  3.4700e-01, -1.0117e-01, -1.4431e-01,\n",
       "          -8.5810e-02,  1.5959e-01,  1.0883e+00,  2.0140e-01,  1.3138e-02,\n",
       "           7.4184e-01,  2.0786e-01, -2.6059e-01, -8.9454e-01, -2.1374e-01,\n",
       "           2.7278e-01,  8.4782e-02,  8.9263e-02,  8.8882e-01, -3.0656e-01,\n",
       "          -5.5283e-01, -3.5697e-02, -7.9839e-04,  4.5909e-01,  3.6647e-01,\n",
       "          -3.5070e-01, -1.2368e-01, -1.1991e-01,  1.4865e-01, -4.5196e-01,\n",
       "           9.5934e-02, -2.8260e-01, -3.5705e-01,  2.5315e-01, -3.4640e-02,\n",
       "          -4.1625e-02, -1.6119e-02,  1.3105e-01,  2.2807e-01, -6.2291e-01,\n",
       "           3.8671e-01, -2.2148e-01, -2.7668e-01, -2.9436e-01, -3.0792e-01,\n",
       "           1.7060e-01,  5.8156e-01, -2.3175e-01, -2.8216e-01,  9.4710e-02,\n",
       "           1.4969e-01, -6.3152e-02,  2.3605e-01, -2.5962e-01,  6.3450e-01,\n",
       "          -6.2117e-01,  3.2064e-01,  2.5945e-01,  2.0480e-01,  5.3191e-01,\n",
       "          -4.8356e-02,  2.3656e-01,  6.5329e-01, -3.3035e-01, -9.1497e-01,\n",
       "          -6.0952e-01,  3.9055e-02, -1.8467e-02, -5.9156e-01,  7.2190e-02,\n",
       "           1.0407e-01,  2.0986e-01, -9.9768e-03, -1.7587e-01,  2.9035e-01,\n",
       "           1.6211e-01,  4.0786e-01, -2.6625e-01, -3.1131e-02, -2.1695e-01,\n",
       "          -5.1629e-01,  6.2543e-02, -3.3420e-02,  2.3582e-01, -2.1396e-01,\n",
       "           1.0633e-01,  6.6051e-02, -2.9417e-01, -1.7749e-01,  1.0545e-01,\n",
       "          -1.3524e-01,  2.2664e-02, -9.9179e-02, -1.3461e-01,  3.0105e-01,\n",
       "           3.1261e-01, -5.6458e-01,  9.1092e-03, -1.0901e-01,  2.8804e-01,\n",
       "           2.0548e-02,  6.9845e-02, -1.9505e-01, -4.3997e-01, -1.8082e-01,\n",
       "           1.0130e-01,  1.1442e+00, -3.2982e-01, -3.4694e-01,  1.9047e-01,\n",
       "           3.5305e-01, -6.9584e-02, -3.3983e-01, -2.4206e-01,  2.2857e-01,\n",
       "          -1.2823e-01, -6.1195e-01,  5.6061e-01, -8.0105e-02,  4.1579e-01,\n",
       "           3.7348e-01, -1.8169e-02, -2.2192e-01,  4.5069e-01, -5.4892e-01,\n",
       "          -2.8723e-01, -1.6285e-01, -3.5184e-01, -2.9182e-01,  5.1067e-02,\n",
       "          -3.3332e-01,  2.9471e-01, -9.8883e-02,  6.2791e-02,  2.4088e-01,\n",
       "           1.6187e-01,  1.2151e+00, -2.7946e-01, -7.9952e-02,  5.5759e-01,\n",
       "           5.2717e-01,  1.1489e-01, -1.8795e-02,  1.8102e-01,  6.8021e-02,\n",
       "           3.5146e-01, -9.9870e-02, -1.0578e-01,  1.1036e-01,  2.3472e-01,\n",
       "           7.7802e-01, -7.0832e-01, -3.3332e-01,  8.3063e-02,  4.4044e-01,\n",
       "          -2.3589e-01,  6.9998e-04, -2.7963e-01,  2.3579e-01,  4.2248e-01,\n",
       "          -1.2745e+00, -4.4542e-02, -2.7405e-01,  3.6496e-01, -7.6979e-02,\n",
       "           1.4013e-02,  3.4424e-02, -1.4951e-01, -1.5381e-01,  3.0384e-01,\n",
       "           2.6126e-01, -8.8536e-01,  3.5167e-01,  4.5480e-01, -4.4099e-01,\n",
       "           2.8381e-01, -2.8135e-01,  2.4144e-01,  1.4778e-01,  4.0163e-01,\n",
       "          -1.4972e-01,  2.2471e-01, -4.9484e-01, -7.8929e-01,  2.4902e-01,\n",
       "           7.9506e-01,  4.6182e-02, -1.7489e-01, -1.1340e-02,  3.7384e-01,\n",
       "          -1.8991e-01, -5.8737e-01, -3.2367e-01,  5.0769e-02, -1.0663e-01,\n",
       "          -2.3647e-02, -2.7170e-01,  7.7789e-02,  1.8997e-01, -1.9009e-01,\n",
       "           5.7409e-01,  5.1269e-01,  7.5134e-01, -3.0422e-01, -5.7460e-02,\n",
       "           6.3156e-02,  3.4594e-01, -7.8665e-02,  3.1580e-01, -2.1759e-01,\n",
       "          -3.6017e-01,  2.5146e-01,  6.1842e-02, -7.2129e-02,  3.3746e-01,\n",
       "          -2.7120e-01,  2.8629e-01,  1.5002e-02, -2.3587e-01,  5.5121e-01,\n",
       "          -5.9226e-02,  6.4832e-02, -1.4502e-01,  3.4001e-01,  1.7527e-01,\n",
       "          -4.6809e-01,  9.4722e-01, -1.7249e-02,  5.6974e-01, -1.6431e-01,\n",
       "           3.9491e-02, -9.7174e-01, -4.5802e-01, -5.3830e-02, -1.0273e-01,\n",
       "          -1.0214e-01, -4.2787e-02,  1.7357e-01,  4.2013e-02,  4.8634e-01,\n",
       "           4.1660e-01, -2.8879e-01, -6.7143e-01, -3.9708e-01,  4.4053e-01,\n",
       "           1.0391e-01, -1.4212e-02, -1.5177e-02,  4.8121e-02,  3.7661e-02,\n",
       "          -5.9154e-01, -7.6089e-01, -3.0103e-01,  1.2848e-01, -2.6071e-01,\n",
       "           8.6395e-01, -6.6638e-02, -2.8080e-01,  7.5815e-02,  3.8157e-02,\n",
       "           2.2677e-01,  5.7373e-01,  2.9766e-03, -3.7409e-01, -1.5480e-01,\n",
       "          -1.0157e-01, -3.3284e-01,  2.3746e-02, -6.6055e-01,  6.3345e-01,\n",
       "           3.6691e-01,  9.7053e-02,  7.4482e-01]], device='mps:0'),\n",
       " tensor([[-1.2245e-01, -5.7781e-01, -3.8958e-01, -7.6396e-01, -2.7582e-02,\n",
       "          -1.6325e-01, -4.0784e-01,  5.5577e-02, -5.2986e-01,  5.5490e-01,\n",
       "           6.6294e-02,  3.2568e-01,  6.8944e-02, -1.0107e-01, -5.8537e-01,\n",
       "           1.0273e-01,  4.8949e-01,  7.9202e-01,  2.2680e-01,  3.4107e-01,\n",
       "           3.4389e-01,  3.7029e-01, -2.7859e-02,  1.4883e-01, -1.3906e-01,\n",
       "          -2.9529e-01, -2.2164e-01, -2.8248e-01, -1.5838e-01, -2.1350e-01,\n",
       "           1.5787e-02,  3.8044e-01,  3.3309e-01,  8.7431e-01, -1.5470e-01,\n",
       "          -9.2261e-02,  3.1864e-01, -4.1380e-01,  8.3176e-01, -1.2767e-01,\n",
       "           6.7482e-01, -2.9526e-01,  2.8905e-01,  2.8700e-01,  1.6726e-01,\n",
       "          -3.2651e-01,  2.1048e-02, -6.7547e-01,  3.6165e-01, -2.8836e-01,\n",
       "          -1.3908e-01,  3.3092e-02,  5.8982e-01,  2.1875e-01, -4.9037e-01,\n",
       "          -3.4248e-01,  8.8899e-02,  3.8226e-01, -3.0909e-01,  6.7864e-02,\n",
       "          -4.8682e-02,  2.8856e-01, -1.5308e-01, -3.3922e-01,  4.4150e-01,\n",
       "          -4.3517e-01, -3.9718e-01, -4.9870e-01, -3.0109e-01,  1.8430e-02,\n",
       "           1.0152e-01,  2.1445e-01,  7.7467e-02, -2.8755e-01,  3.3672e-01,\n",
       "          -4.0882e-01,  7.2813e-02,  9.9625e-02, -3.0996e-01,  2.4040e-01,\n",
       "           1.4034e-01,  9.2429e-02,  1.2657e-01, -2.1012e-02, -1.5462e-01,\n",
       "          -4.9138e-01,  4.8570e-01,  1.7899e-01,  2.7682e-01,  5.3955e-02,\n",
       "           8.6116e-02, -1.2011e-01, -3.8080e-01, -2.9571e-01,  3.3823e-02,\n",
       "           1.1369e-01,  4.5901e-01, -1.3141e-01,  9.8330e-02, -5.4636e-01,\n",
       "           9.4681e-02,  1.0914e-01,  1.6927e-01,  5.4382e-01, -7.3915e-01,\n",
       "          -3.5657e-01, -1.1137e-02,  9.5517e-02, -7.5739e-01, -2.2284e-01,\n",
       "           1.4985e-01, -1.3659e-01,  1.7036e-01, -3.8239e-01, -3.0824e-01,\n",
       "          -9.8159e-02,  1.5064e-01,  1.4366e-01, -3.6514e-01,  2.7753e-01,\n",
       "           3.0526e-01, -2.9768e-01, -1.2574e-01, -2.9089e-01,  4.2636e-02,\n",
       "          -1.2425e-01,  6.6274e-01, -3.9488e-01,  1.7964e-01,  1.8463e-01,\n",
       "          -2.5211e-02,  1.7945e-01, -3.3309e-01, -2.7254e-01,  7.8234e-02,\n",
       "           6.9266e-03,  4.2576e-01, -5.4737e-01,  6.1962e-02,  1.5746e-01,\n",
       "           5.7722e-02, -2.7750e-01, -1.4573e-01, -2.6733e-01, -1.0586e-01,\n",
       "          -2.4358e-01,  8.5503e-03, -4.9788e-01, -1.2120e-01, -3.4548e-01,\n",
       "          -1.0180e+00, -8.1517e-01,  4.5468e-01, -7.5471e-02,  3.5421e-01,\n",
       "          -2.2898e-01,  4.2709e-01,  3.7557e-01,  1.3645e-01,  9.8688e-02,\n",
       "           3.1447e-01,  2.8680e-02,  2.9820e-01, -3.7642e-01, -4.0584e-03,\n",
       "          -1.9951e-01,  2.1875e-01, -3.8459e-01,  1.7380e-01,  2.1561e-01,\n",
       "           4.0179e-01, -1.6316e-01,  8.3338e-02, -2.3812e-01,  5.9257e-01,\n",
       "          -3.6920e-01,  1.7679e-01,  6.3019e-02, -4.4760e-01, -2.3796e-01,\n",
       "           2.3727e-01, -1.5272e+00,  3.2048e-01, -6.3608e-02, -5.5412e-01,\n",
       "          -4.9239e-01,  2.3388e-01,  5.0960e-02, -6.9563e-02,  2.4825e-02,\n",
       "          -2.3335e-01, -1.7445e-01,  2.0905e-01,  3.3678e-02,  7.9070e-02,\n",
       "           1.1382e-01,  8.6032e-01,  2.4227e-01, -2.6465e-02,  1.6451e-01,\n",
       "          -4.3021e-01,  1.7101e-01,  2.6315e-01, -1.9545e-01,  1.7528e-01,\n",
       "           4.7480e-01, -2.3228e-01, -1.3536e-01,  4.0256e-01,  6.3072e-02,\n",
       "          -4.8707e-02, -2.1529e-01, -3.3282e-01,  1.5149e-01, -2.0686e-01,\n",
       "           4.0066e-01, -4.4691e-01, -1.9808e-01,  7.9317e-01,  1.6929e-01,\n",
       "           8.7514e-02, -1.4191e-01,  1.0905e-01, -2.0628e-01,  1.6963e-01,\n",
       "           3.3177e-01,  1.0414e-01,  7.5869e-02,  8.9200e-03, -1.2354e-01,\n",
       "          -1.4794e-01, -4.4333e-01,  3.1511e-01,  6.6038e-01, -2.0833e-01,\n",
       "          -3.9876e-01, -1.8492e-01,  4.1386e-01, -3.0567e-01, -1.3112e-01,\n",
       "          -3.6188e-01, -6.2118e-01, -6.7222e-01, -1.1476e-01, -1.3031e-01,\n",
       "           2.6493e-01, -3.9383e-01,  7.1166e-01,  4.7271e-01,  2.8841e-01,\n",
       "           1.4825e-01,  3.1720e-02,  2.2944e-01,  4.6645e-01, -3.6666e-01,\n",
       "           9.5270e-02,  1.9249e-01, -3.2722e-01,  2.3899e-01, -3.6536e-01,\n",
       "          -4.7326e-01,  8.1880e-02, -1.3397e-01,  3.0030e-01,  2.5877e-01,\n",
       "           5.9657e-01,  3.9240e-01, -4.1703e-02, -1.7073e-01,  4.6831e-01,\n",
       "          -4.6186e-01,  3.6412e-03, -1.3622e-01, -6.6397e-01, -6.6737e-01,\n",
       "          -9.9108e-01, -3.9231e-01,  3.5252e-01,  3.1310e-01, -9.7573e-01,\n",
       "          -2.7705e-01, -2.0298e-03, -1.0710e-01, -2.8120e-01, -4.7766e-02,\n",
       "           6.6853e-02,  1.0973e-01,  1.0132e-01, -3.1542e-01, -3.5033e-01,\n",
       "           4.7391e-02,  1.5852e-01, -2.2432e-01,  6.6087e-02,  2.1104e-01,\n",
       "          -5.2562e-01,  2.7128e-01,  4.2518e-01,  8.4050e-03, -2.4438e-01,\n",
       "          -3.3975e-01,  4.2294e-01,  2.6264e-02, -1.4985e-01,  2.5075e-01,\n",
       "           1.5131e-01,  3.7266e-01,  5.6206e-01, -3.4032e-01, -3.8364e-01,\n",
       "          -5.1922e-02,  6.1384e-01, -9.3325e-02, -4.0134e-01, -1.9554e-01,\n",
       "          -8.4299e-01, -1.2032e-01, -3.9545e-01, -1.0136e-01, -7.9280e-01,\n",
       "           3.1383e-02,  5.7563e-01, -2.7175e-01,  2.6416e-01, -2.5523e-01,\n",
       "           3.7680e-01,  4.3516e-01,  5.4362e-02,  2.6257e-02, -3.5394e-01,\n",
       "           8.1611e-02, -6.5917e-01, -1.9982e-01, -3.0926e-01, -8.0198e-02,\n",
       "           4.9156e-01,  3.1095e-01, -9.2148e-02, -3.8973e-02, -3.1183e-01,\n",
       "           6.7835e-01, -3.5728e-01, -6.2542e-02,  3.1373e-01, -4.4525e-01,\n",
       "           2.8770e-01,  3.7958e-01, -4.3665e-01, -5.4877e-01,  6.2342e-01,\n",
       "           3.9367e-01, -3.1148e-01, -5.6768e-01,  1.4511e-02,  6.7250e-01,\n",
       "           8.1918e-01,  4.8919e-01, -1.6672e-01, -6.9026e-02,  5.8866e-01,\n",
       "          -4.7197e-01,  1.0334e-01, -1.6481e-01, -4.4863e-01, -1.0603e-02,\n",
       "           4.7552e-01,  3.2897e-01, -1.3447e-01,  2.4254e-01,  2.1837e-01,\n",
       "           7.2488e-03, -2.6205e-01, -1.9449e-01,  5.8611e-03, -1.0413e-01,\n",
       "          -2.3968e-01, -6.2146e-01,  4.0149e-01, -5.4296e-01, -2.1666e-01,\n",
       "          -2.1174e-02, -2.0481e-01, -1.0555e-01,  4.9926e-02, -1.9305e-01,\n",
       "          -1.4429e-01, -2.4448e-01,  5.1375e-01, -1.8378e-01, -1.4526e-01,\n",
       "           4.1174e-01, -3.2762e-02, -7.3499e-02,  6.3983e-01,  1.5631e-01,\n",
       "          -2.9186e-01,  2.1402e-01, -7.5152e-01,  4.4602e-02,  4.5640e-01,\n",
       "          -4.9454e-01, -6.2713e-02,  2.3795e-01,  4.6874e-01,  1.4375e-01,\n",
       "          -4.3789e-01,  1.8928e-02,  1.2892e-01,  3.4386e-01,  1.7597e-01,\n",
       "          -1.7538e-01,  1.0277e-01,  1.3256e-01,  2.3569e-01, -4.1563e-01,\n",
       "           3.3259e-01, -7.4330e-02, -2.0785e-01,  6.0598e-02, -2.7260e-01,\n",
       "           4.1346e-01,  2.9612e-01,  1.2060e-01,  4.7756e-01, -1.1282e+01,\n",
       "           3.4846e-01,  7.1861e-01, -4.6049e-01, -5.2191e-02, -6.8471e-02,\n",
       "           1.0716e-01, -3.8360e-01,  3.4603e-01, -1.8431e-01,  6.5323e-02,\n",
       "          -3.9245e-01, -1.8136e-02,  3.0199e-01, -2.8758e-01, -4.7102e-01,\n",
       "          -1.0458e-01, -7.3859e-01,  2.2092e-01, -1.8834e-01,  3.8300e-01,\n",
       "          -2.6688e-01,  2.0120e-02,  5.4724e-02, -1.3875e-01,  3.9925e-02,\n",
       "          -2.6712e-01, -1.2768e-01,  3.6339e-01,  1.9476e-01, -1.0833e-01,\n",
       "           2.3304e-01,  1.2832e-01,  5.2911e-01,  2.4326e-04, -8.8647e-01,\n",
       "           1.5746e-01, -2.4732e-01, -1.2812e-01, -3.0047e-01, -2.8109e-01,\n",
       "          -8.4073e-02, -6.5727e-01,  5.2620e-01,  1.9589e-01,  6.0847e-01,\n",
       "           3.8309e-01,  5.4310e-01,  1.2076e-01, -4.9185e-01,  9.8005e-02,\n",
       "          -4.8713e-01,  3.0827e-01, -7.2678e-01,  1.2453e-01, -6.3194e-02,\n",
       "           1.7286e-01, -6.4417e-01,  8.5422e-02, -4.0248e-01, -2.7547e-01,\n",
       "           1.7887e-02, -2.2071e-02, -1.2420e-01, -3.9905e-01, -3.2315e-01,\n",
       "           4.4037e-01,  1.1355e-01, -9.8643e-04, -6.8071e-02,  9.7733e-02,\n",
       "           1.6351e-01,  3.0069e-01,  1.4166e-01,  3.2929e-02,  5.0860e-01,\n",
       "          -1.2825e-01, -2.6614e-01,  1.4092e+00, -1.6027e-01,  2.0432e-01,\n",
       "           1.0025e-01, -1.5621e-02, -4.5483e-01, -3.7854e-01, -2.5644e-01,\n",
       "          -1.1403e-01,  7.3303e-02,  1.3781e-01,  9.4019e-01, -3.2990e-01,\n",
       "          -6.0432e-01, -1.1892e-01, -2.2482e-02,  4.5873e-01,  5.2051e-01,\n",
       "          -1.6010e-01,  3.2659e-01,  3.5514e-01, -4.7585e-02, -8.1981e-01,\n",
       "          -4.0362e-02,  1.4318e-01, -3.2729e-01,  1.2148e-01, -8.5972e-02,\n",
       "           2.8706e-01, -9.9108e-02,  9.0974e-02,  5.7230e-01, -2.1913e-01,\n",
       "           2.7006e-01,  1.1437e-01, -8.5017e-02, -2.9742e-01, -5.8119e-01,\n",
       "           5.5245e-01,  6.1337e-01, -4.0501e-01, -4.8951e-01, -3.0976e-01,\n",
       "          -3.1603e-01,  1.3729e-01, -1.6313e-01, -7.3593e-01,  1.2753e-01,\n",
       "          -2.1902e-01,  3.4415e-02,  2.8760e-02, -5.0055e-01,  2.4628e-01,\n",
       "           3.8246e-01, -2.6106e-01,  1.3447e-01, -5.5106e-01, -4.8841e-01,\n",
       "          -4.3955e-01, -2.1827e-01,  6.3352e-01,  2.3813e-01,  8.9218e-02,\n",
       "           2.3891e-01, -2.7735e-02, -3.0639e-01,  2.2481e-02,  1.4725e-01,\n",
       "           6.5081e-02,  6.0419e-01, -2.5781e-02,  1.2054e-01, -2.0253e-01,\n",
       "          -1.9812e-01, -2.6124e-01, -1.6061e-01, -1.6594e-01, -4.3133e-01,\n",
       "          -2.6963e-01, -1.2346e-01, -2.1350e-01, -1.4637e-01,  1.7440e-01,\n",
       "           3.7959e-01,  6.0042e-02, -2.0123e-01,  5.1146e-01,  3.1948e-01,\n",
       "          -6.4279e-02, -4.9575e-01, -1.4014e-01, -4.6734e-02,  5.0299e-02,\n",
       "           6.3072e-01, -3.3846e-01, -4.1659e-01, -3.5880e-01, -2.7362e-01,\n",
       "          -3.0836e-01,  1.1714e+00, -3.2289e-01, -4.0536e-01,  1.3974e-01,\n",
       "           3.0288e-01,  1.6032e-01, -2.9487e-02, -1.4742e-01,  2.6654e-01,\n",
       "          -2.6470e-01, -2.3047e-01,  5.5499e-01, -1.2951e-01,  3.6912e-01,\n",
       "           4.9956e-01,  2.9191e-02, -2.9182e-02,  7.7666e-01,  3.3349e-02,\n",
       "          -2.6236e-02, -4.6004e-02, -2.2133e-01, -9.1059e-02,  6.3970e-01,\n",
       "          -3.9021e-01,  3.5781e-01, -1.6293e-02, -4.9791e-01, -2.4402e-01,\n",
       "           4.0487e-01,  9.6132e-01, -3.1109e-02, -1.0535e-01,  6.1484e-01,\n",
       "           4.2048e-02,  5.6598e-01,  2.0697e-01,  2.5105e-01,  1.5937e-01,\n",
       "           5.1065e-01, -1.7016e-01, -3.7724e-01,  2.5914e-01,  3.8037e-01,\n",
       "           5.6390e-01, -8.2133e-01, -4.1921e-01,  1.7426e-01,  1.6803e-01,\n",
       "          -3.6207e-01,  4.4320e-01, -7.7486e-02,  5.4751e-01,  3.5323e-01,\n",
       "          -1.2105e+00, -6.0771e-02, -1.2738e-02, -2.1907e-01, -1.9580e-01,\n",
       "          -4.7930e-01, -4.2360e-01, -4.5427e-02, -2.8831e-01,  2.2131e-01,\n",
       "           5.5160e-01, -9.6258e-01,  1.7292e-01,  5.8355e-01, -9.7257e-02,\n",
       "           1.6400e-01,  1.9748e-01,  4.0745e-01, -4.0205e-02, -9.8905e-02,\n",
       "           2.8405e-01,  2.6230e-01, -4.4277e-01, -5.3872e-01,  7.9842e-02,\n",
       "           7.2294e-01,  1.9525e-01, -2.4948e-01,  2.2715e-01, -2.7463e-01,\n",
       "           1.6132e-01, -7.2458e-02, -1.2597e-01, -1.5573e-01,  1.9978e-01,\n",
       "           2.3655e-01, -6.9787e-01, -2.4726e-01,  4.3994e-01,  2.0281e-02,\n",
       "           4.3018e-01,  5.1091e-01,  3.8675e-01, -2.8266e-02,  1.7235e-02,\n",
       "          -1.7242e-01,  4.4095e-01, -4.0217e-02,  5.9193e-01, -3.4883e-01,\n",
       "          -4.0975e-01, -1.9659e-01,  1.5489e-01, -5.1623e-01, -9.3763e-02,\n",
       "          -2.3352e-01,  3.6941e-01, -2.9031e-01,  7.1783e-03,  3.8026e-01,\n",
       "          -1.2332e-01,  1.9070e-01, -9.3956e-01,  3.8546e-01, -4.2914e-02,\n",
       "           2.0208e-01,  3.2139e-01,  3.7902e-01,  2.8658e-01, -3.8607e-01,\n",
       "          -4.4020e-02, -8.5683e-01, -3.7021e-01,  1.2781e-01, -2.2206e-02,\n",
       "          -6.9262e-02, -6.3737e-02, -2.6712e-01, -1.5855e-01,  7.4498e-01,\n",
       "           8.0704e-01, -5.6676e-02, -6.7529e-01, -3.6390e-01,  3.4909e-02,\n",
       "           2.8099e-01,  7.3834e-02,  8.0311e-01,  3.0284e-01, -2.0016e-01,\n",
       "          -1.0172e+00, -8.5778e-02, -4.6928e-02,  3.0515e-01, -1.2532e-01,\n",
       "           7.0273e-01, -2.8806e-01, -3.8807e-03,  9.2491e-03,  1.0487e-01,\n",
       "           3.9578e-01,  7.4112e-01,  9.3885e-02, -3.0965e-01,  1.5731e-02,\n",
       "           2.0802e-01, -2.7304e-01, -1.7051e-01, -7.2546e-01,  2.7109e-01,\n",
       "           2.4429e-02,  3.1394e-01,  1.9063e-01]], device='mps:0'),\n",
       " tensor([[-2.8108e-01, -3.7484e-01, -6.1831e-01, -9.3117e-01, -3.1959e-01,\n",
       "          -2.3352e-01, -7.1554e-01,  5.7815e-01, -1.3307e-01,  5.1129e-01,\n",
       "           1.1047e-01,  3.2588e-01,  1.2867e-01, -8.3931e-02, -1.8435e-01,\n",
       "          -4.2552e-01,  2.1278e-01,  5.3735e-01,  8.1682e-01,  1.6719e-01,\n",
       "           4.4930e-01,  4.0088e-01, -1.0182e-01, -7.2512e-02, -1.0008e+00,\n",
       "           4.2358e-01, -2.2976e-01,  1.6857e-01,  2.2502e-01, -7.3631e-01,\n",
       "          -1.5737e-01,  1.4557e-01, -2.2995e-02,  1.3010e+00, -4.6968e-01,\n",
       "          -5.9723e-02,  3.0219e-02, -6.3651e-01, -2.0261e-01, -5.9931e-02,\n",
       "           4.0038e-01, -2.9294e-01, -1.7059e-03,  3.8977e-01,  3.5409e-02,\n",
       "           7.8028e-02,  5.2568e-03, -3.7887e-01, -2.6745e-02, -5.1159e-01,\n",
       "           1.8451e-01, -5.9270e-02,  5.6395e-01,  6.6091e-01, -1.1404e+00,\n",
       "          -5.7429e-01,  4.0468e-01,  2.3865e-02, -1.5114e-01,  3.1327e-01,\n",
       "          -8.6331e-02,  4.2793e-01,  1.5154e-02,  3.5662e-01,  1.1785e-01,\n",
       "           1.6499e-01, -2.8226e-01, -6.8019e-01, -5.3820e-01, -2.0708e-01,\n",
       "          -1.3663e-02,  1.7471e-01,  1.7975e-01, -1.9909e-01,  1.2095e-01,\n",
       "          -5.6156e-01,  6.0610e-02, -1.2110e-01, -2.8883e-01,  6.2282e-02,\n",
       "           7.2962e-02,  3.5870e-02,  3.1402e-01, -8.7292e-03, -4.1873e-02,\n",
       "           1.1678e-01,  2.2429e-01,  1.6244e-01,  2.6359e-01,  2.4968e-01,\n",
       "          -2.4917e-02, -2.6614e-02, -2.0652e-01, -3.6364e-01,  4.1822e-01,\n",
       "          -2.1498e-01,  3.1010e-01, -1.3065e-01,  1.0402e-02, -3.4767e-01,\n",
       "           6.7269e-02,  2.4863e-01, -2.6340e-01,  1.4501e-01, -3.6625e-01,\n",
       "          -5.6101e-01,  9.0186e-03, -2.7620e-01, -7.3542e-01,  9.6348e-02,\n",
       "          -4.4488e-02, -4.5805e-01, -1.5478e-01, -3.9537e-01, -3.6075e-01,\n",
       "          -1.1748e-01, -5.6123e-02, -5.7623e-02, -3.0033e-01,  2.6778e-01,\n",
       "           2.6859e-01, -1.0462e-01,  2.4816e-01,  1.7601e-01,  4.5036e-02,\n",
       "          -2.6697e-01,  4.3527e-01,  1.6238e-01,  3.1707e-01,  2.7942e-01,\n",
       "           1.2728e-01,  1.0740e-01, -1.3354e-01, -2.4394e-01,  1.7467e-01,\n",
       "          -1.4773e-01,  1.3850e-01, -3.0067e-01, -2.9541e-01, -1.8862e-01,\n",
       "           1.6769e-01, -1.5491e-01, -3.9755e-01, -2.0876e-01, -1.4398e-01,\n",
       "          -1.6545e-01, -6.4801e-02, -2.2403e-03, -2.9620e-01, -3.3344e-01,\n",
       "          -4.8860e-01, -2.4960e-01,  6.9749e-01, -1.2152e-01, -4.3519e-01,\n",
       "          -2.1778e-01,  4.9052e-01,  1.7072e-01, -2.4193e-01,  2.6751e-01,\n",
       "           9.0571e-01,  5.9500e-02,  1.6854e-01, -3.4540e-01, -3.2128e-01,\n",
       "           2.8603e-01, -7.5859e-02, -2.5540e-01,  4.0637e-01,  5.7010e-04,\n",
       "           1.9466e-01, -3.2531e-02,  2.7885e-01,  1.0085e-01, -2.7389e-02,\n",
       "          -6.2531e-01,  7.0003e-01,  3.2776e-01, -3.0784e-01,  5.9444e-02,\n",
       "          -3.2736e-01, -1.2749e+00,  1.8470e-01, -6.5597e-01,  2.0378e-01,\n",
       "          -2.3159e-01,  3.3212e-01,  4.3813e-01,  7.7201e-02, -7.4739e-02,\n",
       "           3.7823e-01,  2.9356e-01,  3.4726e-01,  4.8691e-01, -2.4386e-01,\n",
       "           3.7561e-01,  2.8681e-01,  4.3175e-01,  5.4078e-01,  2.8823e-02,\n",
       "          -5.1905e-01,  2.0450e-01,  2.7353e-01,  3.7485e-01,  1.4859e-01,\n",
       "           1.0860e-01, -1.5709e-02,  1.2544e-01,  2.7541e-01,  1.1719e-02,\n",
       "          -1.4565e-01, -1.4870e-01, -5.7577e-01, -1.3231e-01, -3.6377e-01,\n",
       "           4.3795e-01, -3.9672e-01, -1.8917e-01,  3.6049e-01,  2.8940e-01,\n",
       "          -1.3827e-01,  1.5530e-01, -1.1313e-01, -2.2979e-01, -7.7242e-02,\n",
       "           4.9397e-01, -1.6361e-01,  9.0261e-02, -1.7600e-01, -2.7260e-01,\n",
       "           4.1994e-01, -6.4171e-01,  6.9182e-01,  3.2239e-01,  1.6845e-01,\n",
       "           1.4385e-01, -3.3087e-01, -2.9645e-02, -5.0241e-01,  3.9076e-01,\n",
       "          -2.1210e-01, -4.5514e-01, -4.7640e-01,  2.8700e-02,  4.6243e-01,\n",
       "           3.4534e-01, -9.3707e-02,  6.5947e-01,  6.6354e-01,  3.4031e-01,\n",
       "          -3.0604e-01, -1.9752e-01,  4.1994e-01,  6.8240e-02, -4.4426e-01,\n",
       "           3.5850e-01, -1.3570e-01, -5.4099e-01,  1.4622e-01, -4.6548e-01,\n",
       "          -2.0782e-01, -9.8246e-02,  2.7137e-01,  1.4882e-01, -3.5781e-01,\n",
       "          -1.7280e-01,  3.3482e-01, -1.4327e-01, -3.3125e-01,  9.8567e-02,\n",
       "          -3.9540e-01,  3.0713e-01,  3.4015e-01, -5.9188e-01, -6.0065e-01,\n",
       "          -4.3830e-01,  9.5429e-02,  1.6812e-01, -3.2600e-03, -6.1841e-01,\n",
       "          -2.9303e-01,  4.2058e-01,  7.2919e-02,  9.1323e-02,  2.3675e-01,\n",
       "           4.8938e-01, -2.1500e-01, -2.6744e-01,  3.7505e-01, -4.7863e-01,\n",
       "          -1.9335e-01, -1.3922e-01, -1.9257e-01,  3.5934e-01,  1.1179e-01,\n",
       "           1.0549e-01,  7.2627e-02,  1.5979e-01, -4.1320e-02, -3.0642e-01,\n",
       "           2.5070e-02,  9.9457e-02, -1.5275e-01, -2.4673e-01, -8.3315e-02,\n",
       "          -5.2572e-02,  4.6199e-01,  5.1846e-01,  7.2075e-02, -6.6386e-01,\n",
       "           1.2292e-01,  8.0272e-01, -1.2040e-01, -4.5757e-02,  5.7011e-02,\n",
       "          -9.8548e-01, -1.0134e-01,  5.8204e-02, -1.5064e-01, -6.1723e-01,\n",
       "           2.5079e-01, -1.1671e-01,  1.6290e-02, -1.7362e-01, -4.7266e-01,\n",
       "           3.7491e-01,  2.3265e-01, -2.8585e-02,  3.5426e-01, -2.7959e-01,\n",
       "          -2.6724e-01, -2.0814e-01, -1.3485e-01, -4.1039e-01, -3.2286e-01,\n",
       "           1.5096e-01,  3.4455e-01,  6.4357e-02, -1.9744e-01, -4.1764e-01,\n",
       "           4.2601e-01, -4.8655e-01,  4.2497e-02,  1.5369e-01, -3.5111e-01,\n",
       "           5.2216e-02,  6.3828e-01, -2.0453e-01,  8.4669e-02, -1.6801e-01,\n",
       "           7.7905e-02, -1.9418e-01, -4.8186e-01,  1.0964e-01,  6.9387e-01,\n",
       "           3.2260e-01,  2.0983e-01, -1.4092e-01,  5.7346e-01,  3.5759e-01,\n",
       "          -4.1690e-01,  5.4171e-01,  2.0572e-01, -4.8109e-02, -2.2554e-01,\n",
       "           8.4920e-02,  2.9016e-01, -4.1639e-01, -8.1974e-02,  3.9692e-02,\n",
       "          -3.8818e-02, -5.1860e-01,  7.8393e-02,  2.8840e-01,  6.5195e-02,\n",
       "          -3.6269e-01, -2.5624e-01,  1.4209e-01, -4.0619e-01, -4.5918e-01,\n",
       "           3.1795e-01, -8.6771e-02,  3.8608e-02, -3.5063e-01, -1.4096e-01,\n",
       "          -2.2800e-01, -1.1640e-01,  6.2209e-01,  5.1960e-01, -3.8957e-01,\n",
       "           6.4594e-02, -8.5264e-02, -4.3011e-01,  2.6618e-01, -6.0306e-02,\n",
       "          -3.7877e-01,  3.1368e-01, -1.8524e-01,  1.4171e-02,  3.1551e-01,\n",
       "          -2.6033e-01, -8.1554e-02,  5.8563e-02, -5.1083e-03,  1.9449e-01,\n",
       "          -3.1312e-01,  1.6631e-01,  1.4865e-02, -2.4466e-01,  1.9123e-01,\n",
       "           2.5123e-01,  2.9101e-01, -2.5301e-01,  2.9332e-01,  8.8253e-02,\n",
       "           2.3496e-01, -1.3800e-01, -1.8703e-01, -5.7631e-01, -5.5742e-01,\n",
       "           4.2981e-01,  4.8780e-01,  4.9861e-01,  4.5671e-01, -1.1267e+01,\n",
       "          -2.0310e-01,  7.4742e-01, -5.2037e-01, -3.0688e-01, -8.4139e-02,\n",
       "           2.7017e-03, -2.0190e-01,  3.2917e-01, -2.5079e-01, -3.9686e-01,\n",
       "          -3.5932e-01, -2.2528e-01,  8.1906e-02, -2.2862e-01, -4.5197e-01,\n",
       "          -2.0565e-01, -7.0668e-01,  3.3178e-01, -2.2724e-01,  6.4384e-01,\n",
       "          -3.3765e-01, -8.9058e-03,  3.0109e-01, -3.5991e-01, -5.6521e-02,\n",
       "           1.8853e-01, -7.0181e-02,  2.3019e-01,  4.5990e-01,  4.3363e-02,\n",
       "           2.0083e-01, -1.2387e-01,  3.8655e-02, -1.2718e-01, -4.6201e-01,\n",
       "          -1.4443e-01, -3.1684e-01,  2.9109e-01, -3.4202e-02, -7.7406e-01,\n",
       "          -4.9367e-01, -2.9740e-01,  3.1915e-01,  2.2301e-01,  1.0165e+00,\n",
       "           5.7538e-01,  7.6913e-01,  2.7018e-01,  4.9769e-02,  1.5411e-01,\n",
       "          -7.5628e-01,  1.3697e-01, -3.7861e-01, -1.3582e-01,  5.1104e-02,\n",
       "           5.2194e-01, -8.9648e-01,  2.0023e-01, -2.4535e-01, -1.4727e-01,\n",
       "           5.4986e-01,  1.1639e-01,  1.6371e-01, -4.3491e-02, -2.5623e-01,\n",
       "           3.3150e-01,  4.5641e-03,  1.5994e-01, -1.8476e-01, -3.1513e-01,\n",
       "          -1.9579e-01,  6.5769e-01,  3.4536e-01, -4.7494e-01,  1.9761e-01,\n",
       "           8.1165e-02, -1.9386e-01,  1.6662e+00,  3.1453e-02, -4.9347e-03,\n",
       "           1.9063e-01,  4.1998e-01, -4.5761e-01, -9.4665e-01,  2.8397e-01,\n",
       "           9.6266e-02,  2.1432e-01,  9.8688e-02,  4.8840e-01, -9.2841e-02,\n",
       "          -6.6376e-01,  2.4334e-02,  1.9303e-02,  7.8559e-02,  2.4837e-01,\n",
       "          -7.9352e-01, -9.4598e-02,  8.9948e-02,  2.5429e-01, -6.7536e-01,\n",
       "           2.8760e-02, -5.7775e-01, -2.2888e-01,  5.3991e-01, -2.2899e-01,\n",
       "           4.3570e-02, -4.5893e-02, -7.8857e-02,  4.4974e-01, -2.6140e-01,\n",
       "           1.2188e-01,  1.3980e-01, -8.8829e-02, -1.0057e-01,  5.8633e-02,\n",
       "           3.9425e-01,  5.0096e-01, -2.4497e-01, -5.5933e-01, -1.5415e-03,\n",
       "           2.6228e-02, -6.4130e-03,  3.7466e-01, -4.2420e-01,  6.1031e-01,\n",
       "          -7.0925e-01,  1.6798e-03,  2.7996e-01, -4.1459e-02,  2.3841e-01,\n",
       "           5.0644e-02,  3.7118e-01,  5.1204e-01, -8.1640e-01, -8.2573e-01,\n",
       "          -6.2267e-01,  3.5791e-02,  7.7117e-02, -4.3876e-01,  8.3958e-02,\n",
       "           1.6146e-02,  1.9810e-01,  3.8795e-02, -1.0065e-01,  2.1069e-01,\n",
       "           2.1221e-02,  4.3617e-01, -2.1362e-01, -7.9418e-02, -2.0523e-01,\n",
       "          -6.5736e-01,  1.7868e-02, -4.2143e-02,  4.8392e-02, -5.4735e-01,\n",
       "          -1.2454e-01,  7.0928e-02, -2.9275e-01, -1.1627e-01,  2.6611e-01,\n",
       "          -1.4969e-01, -4.9682e-02, -2.7746e-01,  3.9071e-01,  5.1983e-01,\n",
       "           2.7812e-01, -5.1713e-01, -2.9790e-01, -2.3527e-01,  3.3415e-01,\n",
       "           1.7495e-01,  1.3537e-01, -1.3876e-01, -2.7087e-01, -3.5144e-01,\n",
       "           1.4493e-01,  7.3790e-01, -7.6043e-02, -6.3279e-01,  3.0019e-01,\n",
       "           5.3820e-01, -5.6642e-02, -4.0626e-01, -6.4452e-02, -1.7511e-01,\n",
       "          -3.4487e-01, -7.2674e-01,  8.3132e-01,  8.9558e-02,  1.5114e-01,\n",
       "           4.4484e-01, -1.8393e-02, -2.5200e-01,  4.9365e-01, -5.1058e-01,\n",
       "          -4.9474e-01, -1.2428e-01, -5.4508e-01,  3.6016e-01,  4.3502e-03,\n",
       "          -4.2744e-01,  5.5704e-01,  1.7197e-01, -1.7541e-02,  1.6173e-01,\n",
       "           1.0641e-01,  1.0622e+00, -2.5700e-01, -9.4455e-02,  7.4805e-01,\n",
       "           3.4912e-01, -1.4396e-02,  2.2145e-02,  1.1840e-01,  8.3677e-02,\n",
       "           3.7081e-01, -1.7142e-01, -2.5759e-01,  5.4964e-02,  2.9541e-01,\n",
       "           6.7422e-01, -6.4760e-01, -5.5052e-01, -7.8796e-03,  6.9252e-01,\n",
       "          -2.1513e-01,  1.7361e-01, -2.0184e-01,  1.7118e-01,  2.6863e-01,\n",
       "          -1.1650e+00, -1.8193e-01, -2.8844e-01,  4.8998e-01,  9.1345e-02,\n",
       "          -6.0027e-02,  4.3161e-02, -3.8979e-01, -1.3649e-01,  5.4023e-01,\n",
       "           1.3175e-01, -7.6431e-01,  1.4301e-01,  8.0370e-01, -5.7560e-01,\n",
       "           4.1548e-01, -2.2725e-01,  2.8262e-02,  1.2628e-01,  4.4508e-01,\n",
       "           1.7719e-01,  1.9609e-01, -4.0154e-01, -5.1532e-01,  2.0903e-01,\n",
       "           9.6834e-01, -1.7445e-02, -2.8939e-02, -2.4019e-01,  2.4422e-01,\n",
       "          -1.0414e-01, -4.1415e-01, -4.2177e-01,  2.1629e-01,  4.3331e-02,\n",
       "          -4.2721e-02, -1.4487e-01, -7.9199e-02,  2.0656e-01,  6.1359e-02,\n",
       "           3.2456e-01,  6.1022e-01,  7.6481e-01, -1.5083e-01, -3.4408e-01,\n",
       "           3.7209e-02,  1.2895e-01, -2.6819e-01,  4.0818e-01, -1.7941e-01,\n",
       "          -3.1292e-01,  2.2776e-01,  2.6177e-01, -4.6524e-01,  1.0064e-01,\n",
       "          -3.0071e-01,  2.9056e-01, -1.0643e-01, -9.6355e-02,  2.9301e-01,\n",
       "          -1.4668e-01,  2.2089e-02, -7.8386e-02,  3.9884e-01,  4.2614e-01,\n",
       "          -1.6194e-01,  6.8209e-01, -1.1344e-02,  1.9053e-01, -5.9657e-01,\n",
       "           3.0322e-01, -1.1714e+00, -2.8642e-01,  3.7194e-02, -1.9029e-01,\n",
       "          -4.0353e-01, -3.7620e-02, -1.9468e-02,  3.6856e-01,  5.0498e-01,\n",
       "           4.6566e-01, -1.5479e-01, -6.8562e-01, -3.9315e-01,  2.8324e-01,\n",
       "           1.6281e-01, -1.7404e-01,  3.0982e-01,  2.2466e-01,  2.2239e-01,\n",
       "          -5.3061e-01, -5.2652e-01, -2.8442e-01,  3.2200e-01, -1.0477e-01,\n",
       "           8.5644e-01, -1.0205e-01, -2.8816e-01,  3.1324e-02, -7.0642e-03,\n",
       "           3.3603e-01,  8.1370e-01,  1.7417e-01, -6.3298e-01, -6.1838e-02,\n",
       "          -1.3623e-01, -2.3077e-01, -9.0610e-02, -5.8411e-01,  5.6895e-01,\n",
       "           3.0931e-01,  9.1718e-02,  7.4464e-01]], device='mps:0'),\n",
       " tensor([[-1.1097e-02, -5.0567e-01, -4.6699e-01, -6.4640e-01, -3.9556e-01,\n",
       "          -2.5266e-01, -5.5990e-01,  3.0801e-01, -1.2536e-01,  1.4189e-01,\n",
       "           2.9133e-01,  5.3879e-01, -3.6706e-01, -2.8859e-01, -4.8290e-01,\n",
       "          -5.3699e-02, -6.1741e-01, -1.6918e-02,  3.4690e-01,  3.0491e-01,\n",
       "           5.3718e-01,  1.8073e-01, -4.1129e-01,  2.9149e-01, -3.7959e-01,\n",
       "           6.6561e-02, -3.4985e-01, -3.1546e-01,  2.6210e-01,  2.8012e-01,\n",
       "           5.3618e-02,  7.7692e-01, -2.4307e-02,  1.0962e+00, -1.8413e-01,\n",
       "          -2.1778e-01,  1.5359e-01,  2.3320e-01,  8.6439e-01, -6.2204e-01,\n",
       "           5.6395e-01,  3.6489e-01,  2.7586e-02, -4.0063e-02,  2.7845e-01,\n",
       "           3.5451e-01,  1.4567e-01, -5.7913e-01, -2.7161e-01, -3.4037e-01,\n",
       "          -1.2979e-01, -1.5164e-01,  5.0058e-01, -2.3292e-01, -7.9546e-01,\n",
       "          -1.9253e-01,  1.1288e-02,  2.6982e-01,  1.1117e-01,  2.3591e-01,\n",
       "           2.8900e-02,  7.9425e-01, -1.9339e-01, -3.2966e-03,  2.0559e-01,\n",
       "           2.5502e-01, -7.6243e-01, -1.2873e-01, -7.2822e-01,  1.1548e-01,\n",
       "           2.1413e-01,  3.5162e-01,  1.0500e-01, -2.6816e-01, -1.9644e-01,\n",
       "           6.0243e-02,  1.6956e-01,  1.8610e-01, -2.0035e-01,  3.5967e-02,\n",
       "           2.1140e-01,  4.4140e-01,  4.5980e-01,  5.9922e-02,  1.6117e-01,\n",
       "           5.6960e-02,  2.0080e-01, -3.9869e-02, -3.1738e-02,  1.7672e-01,\n",
       "           1.1984e-01, -9.9577e-02, -3.7483e-01, -4.3956e-01, -9.5205e-02,\n",
       "          -3.0408e-01,  3.2250e-02, -2.9469e-01, -2.3954e-01, -3.3261e-01,\n",
       "           2.5982e-01,  9.2762e-02,  3.2750e-01, -9.3082e-02, -6.6514e-01,\n",
       "          -3.5248e-01,  4.0832e-01,  3.4586e-03, -2.8129e-01, -3.8448e-01,\n",
       "          -6.6713e-02, -4.9346e-01,  1.2765e-01, -4.4650e-01,  1.9893e-01,\n",
       "           1.0835e-01,  1.8686e-01, -3.2670e-01,  6.5172e-02,  8.8101e-02,\n",
       "           6.1068e-02, -5.0938e-01,  2.7421e-01,  1.8326e-01,  9.3046e-02,\n",
       "           2.2489e-01,  6.8015e-01, -1.6210e-01,  4.3454e-01,  3.5722e-01,\n",
       "           3.1169e-01,  1.6221e-01, -4.1686e-01, -5.0259e-01, -7.5302e-02,\n",
       "           1.8488e-02, -5.9908e-01, -5.7442e-01,  6.1723e-02, -2.2868e-01,\n",
       "           1.6627e-01, -7.3083e-02, -5.1768e-01, -5.4307e-01, -3.1494e-01,\n",
       "          -3.0772e-01, -2.0517e-02, -3.2393e-01, -4.8159e-01, -2.5398e-01,\n",
       "          -6.1867e-01, -7.2891e-01,  1.9443e-01,  1.5164e-01,  2.7714e-01,\n",
       "          -4.3163e-01,  5.4580e-01, -3.0612e-01, -4.8554e-01,  3.7642e-01,\n",
       "           4.3432e-01, -5.0330e-01,  2.9526e-01, -6.3499e-01,  5.1171e-02,\n",
       "           2.0480e-01,  2.5291e-02, -3.4147e-02,  4.2807e-01,  1.1727e-01,\n",
       "           3.9831e-01,  2.7152e-02,  1.2476e-01, -2.9187e-01,  1.5015e-01,\n",
       "          -4.5427e-02, -3.2147e-01, -1.1176e-01,  1.7279e-02, -5.2967e-02,\n",
       "           5.4936e-01, -1.7306e+00, -1.7852e-01,  7.0060e-02, -2.7460e-01,\n",
       "          -1.3173e-01,  2.1473e-01,  5.0407e-03,  3.2283e-01, -3.5674e-01,\n",
       "           8.3698e-02,  8.7128e-02,  8.4654e-02, -1.0725e-01, -5.0983e-02,\n",
       "           2.9101e-01,  7.5929e-01,  4.0793e-01,  9.7463e-02, -2.6657e-01,\n",
       "          -6.0645e-01,  4.1554e-01,  1.9081e-01,  2.0730e-01,  3.9000e-01,\n",
       "           6.7931e-02, -6.9343e-01,  2.9674e-01,  1.9859e-01,  3.7002e-01,\n",
       "           2.5546e-02, -5.8496e-02, -4.8962e-01, -2.4682e-01, -4.3705e-01,\n",
       "           2.7759e-01, -1.3436e-01, -2.5922e-01,  6.3173e-01,  1.9273e-01,\n",
       "          -5.3531e-02,  3.9653e-01,  1.8733e-01, -4.1041e-01,  4.1633e-01,\n",
       "           5.2814e-01,  7.9114e-02,  1.4573e-01, -2.8055e-01, -1.6673e-01,\n",
       "          -3.1532e-01, -4.4420e-01,  4.4703e-01,  8.3941e-01, -7.2084e-03,\n",
       "           5.5343e-01, -3.5757e-01,  5.1174e-01, -3.6424e-01, -6.7779e-03,\n",
       "          -4.6486e-02,  6.8700e-02, -3.3988e-01,  1.6284e-01,  1.5551e-01,\n",
       "           1.0340e-01,  1.9709e-01,  8.5425e-01,  4.6494e-01,  1.4684e-01,\n",
       "           4.1031e-01,  2.5417e-02,  3.2080e-01,  7.8684e-01, -3.0254e-01,\n",
       "           3.4990e-01, -7.1578e-03, -4.1842e-01,  5.6461e-01, -3.5139e-01,\n",
       "           1.4248e-01, -2.3020e-01,  2.2987e-01,  1.7866e-01, -2.7854e-01,\n",
       "           2.1426e-01,  6.5273e-01,  9.5860e-02, -3.9283e-01,  7.3417e-01,\n",
       "          -4.0214e-01,  7.1262e-01, -3.0138e-02, -2.6992e-02, -1.0120e-01,\n",
       "          -7.4577e-01, -6.8569e-01,  1.4294e-01,  3.0808e-01, -7.6332e-01,\n",
       "          -3.4295e-01,  1.2380e-01,  1.3255e-01,  3.5068e-01,  2.4156e-01,\n",
       "           4.6144e-01, -1.7016e-01, -4.7185e-02, -4.4024e-01, -4.6015e-01,\n",
       "           1.1917e-01,  7.3311e-02,  9.5311e-02, -9.8199e-02,  9.2587e-02,\n",
       "           6.1067e-01,  2.4297e-01,  2.6799e-01,  1.4946e-01, -3.4610e-01,\n",
       "          -2.7723e-02,  1.7135e-01,  3.7874e-01, -6.1900e-01, -2.1807e-01,\n",
       "           5.3543e-02,  8.3164e-02,  6.6826e-01, -2.1832e-01, -1.1546e-01,\n",
       "          -1.2450e-01,  5.9739e-01, -2.2108e-01,  6.8002e-02,  1.0269e-01,\n",
       "          -6.8586e-01,  2.7619e-01, -5.9939e-02, -8.5297e-02, -5.2861e-01,\n",
       "           2.2997e-01, -5.7787e-02, -1.6068e-01, -7.3269e-03, -7.7601e-02,\n",
       "           2.2454e-01,  1.3406e-01,  2.8044e-01, -5.6270e-01, -4.2092e-01,\n",
       "           1.1781e-01, -3.2365e-01, -6.6016e-02, -4.5861e-01,  3.6766e-01,\n",
       "           1.3890e-01,  4.3829e-01, -1.0837e-01, -2.8576e-02, -1.3190e-01,\n",
       "           5.2866e-01, -6.2780e-01,  1.5573e-02,  4.8525e-01, -3.5595e-01,\n",
       "          -1.8139e-01,  4.7644e-01, -2.8011e-01,  4.9664e-02,  2.2982e-01,\n",
       "           3.7738e-01, -1.1530e-01, -4.7105e-01, -5.5247e-02,  9.3761e-01,\n",
       "           4.8980e-01,  2.0874e-02, -6.3820e-01,  1.2560e-01,  4.2066e-01,\n",
       "          -3.4219e-01,  1.0777e-01, -3.7280e-01, -2.3519e-01,  8.6456e-02,\n",
       "           3.3260e-02,  5.0344e-01, -5.8998e-02,  1.9741e-01,  4.6778e-01,\n",
       "           1.7011e-01, -6.9331e-01, -3.0557e-01,  6.3124e-01,  5.4670e-01,\n",
       "          -1.2247e-01, -5.4832e-01,  1.1480e-01, -6.4563e-01,  3.1198e-02,\n",
       "          -3.8623e-03, -3.5755e-01,  2.9653e-01, -1.4962e-01, -2.5683e-01,\n",
       "          -2.0003e-01, -5.3748e-01,  4.4973e-03,  1.0291e-01, -6.9595e-02,\n",
       "           4.1738e-01, -7.9610e-03, -3.1154e-01,  1.0058e+00, -2.7292e-01,\n",
       "          -3.1356e-02,  7.4139e-01, -5.4234e-01,  1.0457e-02,  2.4642e-01,\n",
       "          -1.3433e-01, -1.2416e-02,  2.4323e-02,  1.0660e+00,  1.1515e-05,\n",
       "           3.6402e-02, -2.0346e-01,  4.2203e-01, -4.7659e-01,  1.9394e-01,\n",
       "          -4.9958e-01,  4.8310e-01, -6.9399e-02,  2.4253e-01, -8.2329e-01,\n",
       "           4.3257e-01, -3.0604e-01, -3.2040e-01, -6.3922e-01,  5.0359e-02,\n",
       "          -5.0711e-02,  6.6524e-02, -5.8707e-02,  6.0787e-01, -1.0953e+01,\n",
       "           1.0946e-01,  8.4108e-01, -3.2618e-02,  1.3653e-01,  4.3336e-01,\n",
       "           1.2158e-01, -4.6684e-01,  7.2509e-01, -2.3242e-01, -6.6092e-03,\n",
       "          -3.5861e-01, -1.2444e-01, -1.3819e-01, -6.3519e-01,  4.0758e-01,\n",
       "          -1.3255e-01, -5.9114e-01,  2.8540e-01, -1.6945e-01,  2.4978e-01,\n",
       "          -2.3651e-01,  1.5671e-01,  5.9352e-01, -6.1010e-01, -3.3943e-01,\n",
       "          -1.1958e-01, -2.9991e-01,  5.1723e-01,  4.0217e-01, -2.3672e-01,\n",
       "           2.9096e-01,  3.3529e-02,  2.9455e-01, -5.5694e-01, -4.8133e-01,\n",
       "           4.3369e-01, -3.9950e-01,  2.1957e-01, -9.9943e-01, -3.3265e-01,\n",
       "          -1.8082e-01, -5.0951e-01, -2.7870e-02,  5.3812e-01,  7.5524e-01,\n",
       "           1.2821e+00,  1.1534e+00,  9.8198e-02,  4.6362e-01, -1.0523e-01,\n",
       "          -6.6935e-01,  5.1619e-02, -5.6525e-01, -4.1224e-01,  8.7196e-02,\n",
       "           2.2385e-01, -5.4352e-01,  4.7854e-01, -5.9897e-01,  1.8912e-01,\n",
       "           1.4329e-01,  7.9852e-02,  1.7368e-01, -5.1897e-01, -5.5039e-01,\n",
       "          -1.3282e-01,  8.1201e-02,  8.6324e-02, -6.7881e-01,  1.1881e-01,\n",
       "           1.7006e-01,  6.8197e-01, -2.5397e-01, -4.4268e-01,  4.1137e-01,\n",
       "          -2.2746e-03,  9.3809e-02,  1.3158e+00, -2.6036e-01, -1.7413e-01,\n",
       "          -4.8428e-02, -2.3194e-01, -3.3484e-01, -3.9656e-01, -4.5613e-01,\n",
       "          -2.5020e-02,  3.6573e-01,  5.1948e-01,  2.6062e-01, -2.3329e-02,\n",
       "          -3.0054e-01, -4.1445e-02,  5.1977e-02,  3.3862e-01,  9.6772e-01,\n",
       "          -6.6887e-01,  2.6280e-01, -1.9148e-01,  9.2178e-02,  2.2223e-01,\n",
       "          -5.7106e-01, -1.6871e-02,  2.9434e-03,  1.3975e-02,  5.7281e-01,\n",
       "           1.9705e-01,  2.6453e-01,  3.1190e-02,  2.2859e-01, -4.1682e-01,\n",
       "          -4.4044e-01,  2.9125e-04, -1.3190e-01, -5.8294e-01, -4.1034e-01,\n",
       "          -1.3514e-01,  1.3712e-01, -2.8585e-01, -5.3585e-01, -2.5012e-01,\n",
       "           1.4192e-01,  2.5900e-03,  4.9952e-02, -3.3857e-01,  4.7379e-01,\n",
       "          -7.6918e-01, -2.1429e-01, -2.7578e-01, -3.8495e-03,  1.4633e-01,\n",
       "           2.5508e-01, -5.1518e-01, -2.7838e-01, -1.0123e-01, -9.2314e-01,\n",
       "          -8.4489e-01,  2.2927e-02,  7.4620e-01, -1.1960e-01,  1.1861e-01,\n",
       "           8.5428e-02, -3.1671e-03, -3.7708e-01,  2.5194e-01, -2.4644e-01,\n",
       "           2.8149e-01,  4.3265e-01, -4.9583e-01,  2.9773e-02, -1.2792e-01,\n",
       "          -1.8992e-02, -1.0887e-01, -8.8647e-02, -5.4131e-01, -6.1371e-01,\n",
       "          -4.1197e-01,  1.3736e-01, -2.2747e-01,  4.1516e-01,  6.8072e-02,\n",
       "           1.2582e-01, -2.0826e-01,  9.8224e-02,  8.2540e-01,  6.0197e-01,\n",
       "           2.3014e-01, -3.2920e-01, -7.7666e-01,  3.4140e-02,  4.1642e-02,\n",
       "          -1.6450e-01,  2.1317e-01, -1.6384e-01, -3.0657e-01, -1.0112e-01,\n",
       "           1.2620e-02,  1.0002e+00, -2.0790e-01, -1.9155e-01,  1.0507e-01,\n",
       "           7.4626e-01, -6.0834e-01, -5.7221e-01,  3.7612e-01,  5.0418e-02,\n",
       "          -5.6831e-01, -1.9450e-01,  4.0315e-01, -1.7111e-01,  8.9921e-02,\n",
       "           3.7572e-01,  1.7821e-01, -1.8612e-01,  2.2711e-01, -6.7291e-02,\n",
       "          -1.7001e-01,  3.2731e-01, -1.6533e-01, -7.7254e-02,  4.0384e-01,\n",
       "           9.9517e-02,  1.5584e-01, -2.5262e-01, -4.8444e-01, -2.7025e-01,\n",
       "           1.5439e-01,  1.1110e+00, -1.7989e-01, -3.2527e-01,  6.6881e-01,\n",
       "          -3.0956e-01,  3.8928e-01,  4.8478e-01, -2.2611e-01, -5.0807e-02,\n",
       "           5.8049e-01,  3.0288e-01, -7.3236e-01,  2.2417e-01,  3.1670e-01,\n",
       "          -1.7335e-01, -5.3095e-01, -3.2838e-01, -2.1330e-01, -3.5189e-02,\n",
       "          -3.6356e-01,  5.2132e-01,  1.2255e-01,  3.9253e-01, -1.0843e-01,\n",
       "          -8.7207e-01,  4.7185e-01,  2.6128e-01,  8.0250e-01, -6.2568e-03,\n",
       "          -4.4880e-01, -3.5541e-01, -3.5790e-01, -6.6453e-02,  3.6543e-01,\n",
       "           2.0116e-01, -1.0333e+00,  4.6071e-02,  4.0507e-01, -3.9812e-02,\n",
       "           4.7083e-01, -1.4254e-01,  1.9634e-01,  4.7891e-01, -4.3918e-02,\n",
       "          -2.1856e-01,  1.3161e-01, -3.3027e-02, -2.7190e-01,  4.2861e-01,\n",
       "           4.6755e-01,  1.6062e-02,  2.5454e-02,  6.4881e-02, -4.2427e-01,\n",
       "          -3.4980e-01,  7.7064e-02, -1.7966e-01, -3.5675e-01, -1.4295e-01,\n",
       "          -4.0726e-01, -5.1772e-01, -6.6164e-02, -1.1357e-01,  1.3233e-01,\n",
       "           3.1032e-01,  1.3036e-01,  3.8431e-01, -2.3708e-01, -1.2541e-01,\n",
       "          -2.7977e-01,  4.5653e-01, -2.7595e-01,  8.0808e-01,  2.0566e-01,\n",
       "          -1.4423e-01,  2.0720e-01,  2.1514e-01, -2.9046e-01,  1.3028e-01,\n",
       "          -2.7828e-01,  2.8810e-01, -5.7798e-01, -3.4110e-01, -5.9710e-02,\n",
       "          -7.7948e-02, -1.5049e-01, -2.8505e-01, -1.0179e-01,  1.0608e-01,\n",
       "          -9.9043e-02,  9.7868e-01, -4.1759e-01,  6.3833e-01, -1.0342e-01,\n",
       "          -1.1350e-01, -7.5089e-01,  3.9397e-01,  3.6376e-01, -1.4798e-01,\n",
       "          -5.6434e-01, -2.1653e-01, -8.9158e-02, -4.2284e-01,  8.8621e-01,\n",
       "           5.8157e-01, -7.4673e-01, -3.7246e-01, -1.5919e-01, -2.4113e-01,\n",
       "          -4.7389e-02,  6.9820e-02, -3.2710e-01, -1.3295e-02, -1.6580e-01,\n",
       "          -9.8715e-01,  2.7641e-01, -3.6473e-02,  4.9205e-01, -2.5466e-01,\n",
       "           7.6979e-01, -8.6017e-02, -1.2749e-01,  1.1690e+00,  1.1169e-01,\n",
       "          -6.1844e-02,  5.9531e-01, -1.6837e-01, -6.2973e-01, -7.0295e-02,\n",
       "           4.4239e-02, -3.1098e-01,  8.7670e-02, -4.7880e-01,  2.9353e-01,\n",
       "           2.3648e-01, -7.1226e-02,  4.3877e-01]], device='mps:0'),\n",
       " tensor([[-0.2565, -0.5583, -0.5201,  ...,  0.2883, -0.0725,  0.5769],\n",
       "         [-0.2806, -0.7437, -0.2562,  ...,  0.1512,  0.1475,  0.3835]],\n",
       "        device='mps:0'),\n",
       " tensor([[-1.7487e-01, -6.9704e-01, -5.7816e-01, -5.4792e-01, -7.8680e-01,\n",
       "          -3.8896e-01, -4.1577e-01,  6.4769e-01, -8.4102e-02,  6.5170e-01,\n",
       "           2.6998e-02, -1.4299e-01,  1.2171e-01, -5.5801e-01, -4.3555e-01,\n",
       "          -3.5320e-01,  1.5178e-01,  5.8032e-01,  8.5628e-01,  2.7401e-01,\n",
       "           6.0110e-01,  1.6607e-01,  5.7805e-02, -9.7357e-02, -1.0252e+00,\n",
       "           2.6563e-01, -2.7809e-01,  5.0935e-02,  3.9403e-01, -9.6591e-01,\n",
       "          -3.5555e-02,  3.3393e-01, -7.3095e-02,  7.5360e-01, -8.8335e-03,\n",
       "          -9.0089e-02,  9.3240e-02, -4.3505e-01,  4.7106e-02, -2.2909e-01,\n",
       "           5.9218e-01,  8.4240e-02,  5.0037e-01,  7.7741e-02,  5.0087e-01,\n",
       "          -1.0385e-02,  2.9208e-02, -4.6014e-01, -1.0377e-01, -2.2889e-01,\n",
       "           4.0216e-01, -4.3623e-02,  5.2191e-01,  4.1412e-01, -9.8408e-01,\n",
       "          -5.2273e-01, -1.1914e-01,  5.8914e-02, -5.4251e-01,  3.7908e-01,\n",
       "          -3.7878e-02,  5.2135e-01,  3.3377e-01,  3.1719e-01, -4.1832e-02,\n",
       "           6.0309e-02, -6.1103e-01, -7.2427e-01, -3.9978e-01, -7.7754e-02,\n",
       "          -2.0233e-02,  5.1361e-01, -8.9101e-02, -1.3041e-01,  1.3617e-01,\n",
       "          -2.8355e-01,  1.4675e-01, -2.1771e-01, -3.1324e-01, -5.1016e-01,\n",
       "           1.3073e-01, -1.1495e-01, -2.4514e-01, -2.6599e-01, -3.2105e-01,\n",
       "          -4.3338e-02,  4.0341e-01,  1.5786e-01,  3.0000e-01,  1.1056e-02,\n",
       "           3.3503e-01,  1.7395e-01, -3.5189e-01, -3.2193e-01,  4.5337e-01,\n",
       "          -5.8255e-01,  3.2454e-01, -2.6040e-01, -1.3350e-01, -3.3616e-01,\n",
       "           1.4620e-01,  3.9234e-01, -1.1758e-01,  2.5510e-01, -8.6180e-01,\n",
       "          -5.8757e-01,  1.7690e-01, -2.9393e-02, -7.3296e-01,  4.7878e-02,\n",
       "          -7.8943e-02, -3.5927e-01, -2.2699e-02, -5.3469e-01, -4.3103e-01,\n",
       "          -1.5761e-01, -1.7578e-01, -3.1504e-01, -2.3421e-01,  7.8093e-01,\n",
       "           4.0324e-01, -3.9294e-01,  3.0568e-01, -4.0809e-04,  2.3331e-01,\n",
       "          -1.2559e-01,  6.6333e-02, -6.3980e-02,  3.5359e-01,  4.7728e-01,\n",
       "          -8.7652e-02,  1.4458e-01, -1.7384e-01,  3.4721e-02, -5.7257e-02,\n",
       "          -1.3096e-01,  1.6342e-01, -6.4833e-01,  8.1126e-02,  1.4958e-01,\n",
       "          -1.1885e-01,  8.7146e-02, -2.8717e-01,  2.4933e-01,  2.0620e-02,\n",
       "          -2.3268e-02,  2.4376e-01,  6.8033e-02, -3.6266e-01, -1.2138e-01,\n",
       "          -6.9132e-01, -1.6077e-01,  5.6006e-01, -2.4038e-01, -2.7189e-02,\n",
       "          -2.1384e-01,  4.7440e-01, -7.4919e-02, -3.0735e-01,  9.6006e-02,\n",
       "           7.2034e-01, -7.0909e-02,  2.2096e-01, -3.3948e-01, -1.0040e-01,\n",
       "           4.0406e-01, -1.0799e-01, -1.3565e-01, -3.3751e-02, -9.6651e-02,\n",
       "           9.7769e-02, -3.0323e-01,  1.2762e-01,  1.1152e-01,  9.1916e-02,\n",
       "          -4.6493e-01,  3.9442e-01,  4.4154e-01, -1.8759e-01, -3.0969e-02,\n",
       "          -1.3808e-01, -1.6728e+00,  2.5699e-03,  8.6223e-02,  1.5797e-01,\n",
       "          -3.5474e-01,  5.2948e-01,  2.4474e-01,  1.2140e-01, -2.8890e-01,\n",
       "           4.4515e-01, -1.5694e-01,  2.5350e-01,  5.0596e-01,  7.1384e-03,\n",
       "           3.9109e-01,  4.1970e-01,  4.0489e-01,  4.0523e-01, -8.1335e-02,\n",
       "          -5.8243e-01,  1.1016e-01, -1.2054e-01,  8.0941e-01,  1.4626e-01,\n",
       "           2.0901e-01, -3.0917e-01, -2.2309e-01,  1.8957e-01,  1.2465e-01,\n",
       "           2.2946e-02, -1.9455e-01, -5.3978e-01, -2.6211e-01, -3.2100e-02,\n",
       "           5.2135e-01, -2.8532e-02, -4.8199e-01,  3.5992e-01,  2.7053e-01,\n",
       "          -2.4625e-01, -2.7593e-02, -1.7606e-01, -3.3474e-01, -2.7109e-01,\n",
       "           2.9029e-01, -8.5422e-02, -2.9669e-02,  1.8353e-02, -1.6153e-01,\n",
       "           1.8974e-01, -6.2877e-01,  7.1589e-01,  4.5639e-01, -1.4926e-01,\n",
       "          -4.6907e-01, -6.8285e-01, -7.4131e-02, -6.3212e-02,  1.4680e-01,\n",
       "          -2.6285e-01, -4.5176e-01, -5.0414e-01,  1.6577e-02,  1.8325e-01,\n",
       "           3.6659e-01, -9.8025e-03,  8.5184e-01,  4.4624e-01,  4.2394e-02,\n",
       "          -1.8212e-01, -1.8370e-01,  3.8151e-01, -4.9672e-02, -2.4277e-01,\n",
       "           2.4891e-01,  4.8388e-02, -4.0590e-01,  2.0033e-01, -4.9239e-01,\n",
       "          -3.5087e-01,  4.5282e-02,  6.1958e-02,  3.1507e-01, -1.8047e-01,\n",
       "           2.0918e-01,  1.9550e-01, -5.0246e-01, -1.0615e-01, -5.1722e-02,\n",
       "          -2.9383e-01,  8.0257e-01,  4.9303e-01, -3.6923e-01, -5.0436e-01,\n",
       "          -8.2990e-01, -1.6095e-01, -3.7148e-01,  1.5781e-01, -7.4194e-01,\n",
       "           1.4855e-01, -3.3535e-02,  1.5345e-01, -1.8508e-01, -1.0484e-01,\n",
       "           4.3430e-02,  1.5246e-01,  5.8665e-02,  7.4600e-02, -5.1988e-01,\n",
       "          -1.0571e-01,  1.1810e-01, -3.3137e-01,  3.1147e-01,  6.6065e-02,\n",
       "           1.4536e-01, -2.6249e-01,  2.3162e-02,  1.3533e-01,  7.6275e-02,\n",
       "          -2.1078e-01, -1.9608e-01, -2.1165e-01, -3.7800e-01, -4.7228e-01,\n",
       "           2.5956e-01,  1.1281e-01,  5.6944e-01, -1.9618e-01, -6.0321e-01,\n",
       "           3.1888e-02,  7.7975e-01,  1.2558e-02, -1.1474e-02, -2.0536e-01,\n",
       "          -1.3278e+00, -8.9873e-02, -1.2785e-02,  1.4524e-02, -2.0426e-01,\n",
       "           2.4222e-01, -1.6542e-01,  9.5205e-02,  2.2250e-03, -4.4374e-01,\n",
       "           7.2836e-01,  2.7344e-01,  7.1463e-02, -4.4850e-01, -1.9688e-01,\n",
       "          -2.8655e-02, -1.7610e-01,  1.8572e-02, -1.4647e-01,  1.4897e-01,\n",
       "           1.5541e-01,  5.5521e-01,  1.4206e-01,  1.6747e-01, -3.2408e-01,\n",
       "           4.0051e-01, -5.4100e-02, -1.0645e-01,  5.4058e-01, -3.2256e-01,\n",
       "          -5.0044e-01,  5.4764e-01, -2.2490e-01, -6.2869e-02, -1.4289e-01,\n",
       "           2.7681e-01,  9.9871e-02, -4.7728e-01,  2.1545e-01,  4.1552e-01,\n",
       "           5.3482e-01,  3.9514e-01, -3.0473e-01,  7.9740e-01,  2.5778e-01,\n",
       "          -1.4541e-01,  3.9751e-01,  1.7578e-01, -1.7144e-01, -4.1768e-01,\n",
       "           1.8533e-01,  1.6502e-01, -6.7161e-01, -3.0245e-01, -2.3023e-01,\n",
       "           3.8032e-01, -4.3857e-01,  6.4291e-02,  3.0836e-01,  6.2598e-01,\n",
       "          -6.3280e-01, -3.6928e-01,  5.0905e-01, -4.1955e-01, -2.8117e-01,\n",
       "          -3.3483e-02,  6.8662e-02,  1.3040e-01, -2.0236e-01,  1.8790e-02,\n",
       "          -3.3788e-01,  8.1686e-03,  6.7058e-01,  3.3045e-01, -2.8401e-01,\n",
       "          -1.5254e-01, -6.7215e-02, -3.8495e-01,  5.8046e-01,  5.0972e-02,\n",
       "          -1.6410e-02,  1.9046e-01, -1.5517e-01, -3.3603e-02,  1.4793e-02,\n",
       "          -3.8112e-01, -9.6584e-03,  8.1798e-03, -2.9520e-01, -8.9407e-02,\n",
       "          -4.6536e-01,  2.8712e-02,  6.2865e-01, -9.2617e-02, -1.9962e-03,\n",
       "           2.5004e-01,  1.8693e-01, -4.2499e-01,  5.2363e-01,  2.3818e-01,\n",
       "           1.6574e-01, -2.3984e-01,  1.9745e-01, -3.2244e-01, -6.2366e-01,\n",
       "           8.8577e-01,  4.7145e-01,  5.0358e-01,  5.8973e-01, -1.1286e+01,\n",
       "           1.0123e-02,  4.7429e-01, -5.4042e-01, -2.6382e-01,  1.6563e-01,\n",
       "           1.8828e-02, -3.8497e-01,  3.6875e-02, -3.9392e-01, -3.5082e-01,\n",
       "           8.2596e-03, -2.7845e-02, -1.7876e-02, -5.3743e-02, -1.4759e-01,\n",
       "          -4.0597e-01, -7.0174e-01,  2.0152e-01, -6.8782e-02,  4.0022e-01,\n",
       "           6.1135e-02,  2.0245e-01,  1.3882e-01, -5.2201e-02, -5.2477e-01,\n",
       "           7.2585e-02,  2.1164e-03,  2.4050e-01,  3.1452e-01, -2.9791e-02,\n",
       "           1.7824e-01, -1.3006e-01,  7.8083e-02, -1.6836e-01, -3.2528e-01,\n",
       "          -9.3686e-02, -1.9412e-01,  2.1021e-01,  5.5705e-02, -4.6791e-01,\n",
       "          -4.0887e-01, -1.5246e-01,  1.5269e-01,  1.8018e-01,  8.1653e-01,\n",
       "           1.1163e+00,  3.8507e-01,  4.3569e-01,  4.9180e-03, -1.2600e-02,\n",
       "          -8.4492e-01,  2.4195e-01, -7.5448e-01, -6.4268e-02, -5.4643e-03,\n",
       "           6.6932e-01, -8.1059e-01,  5.7351e-02, -3.5014e-01, -3.3550e-01,\n",
       "           5.2953e-01,  2.4699e-01,  1.2031e-01, -3.3482e-01, -2.8508e-01,\n",
       "           5.4002e-01,  2.0452e-01,  2.0374e-01, -2.3721e-01,  1.1125e-01,\n",
       "          -5.6418e-01,  6.1734e-01,  2.4000e-01, -2.7011e-01, -1.8844e-02,\n",
       "           5.7638e-02,  2.1412e-04,  1.1028e+00,  1.0428e-01, -8.2062e-02,\n",
       "           2.7263e-01,  4.7224e-01, -5.1629e-01, -9.0890e-01,  2.8869e-04,\n",
       "           1.9498e-01,  5.0678e-01, -6.6236e-02,  4.4454e-01, -5.3385e-02,\n",
       "          -4.1239e-01, -9.2015e-02, -2.5619e-02,  2.1073e-01,  1.6605e-01,\n",
       "          -4.7017e-01,  4.9633e-02,  3.0226e-01,  1.8064e-01, -3.4695e-01,\n",
       "          -1.3893e-01, -1.4963e-01, -1.2571e-01,  3.6069e-01, -9.4967e-02,\n",
       "           1.6524e-01, -2.2040e-01, -3.3572e-03,  2.6630e-01, -5.5434e-01,\n",
       "           3.6098e-01, -8.9638e-02, -6.3091e-02, -2.2058e-01, -1.2092e-01,\n",
       "           3.5314e-01,  3.4706e-01, -3.8404e-01, -3.4073e-01, -2.5000e-01,\n",
       "           2.7946e-01, -1.2378e-01, -1.5761e-02, -5.7207e-01,  7.1549e-01,\n",
       "          -5.3027e-01,  2.3663e-01,  2.0593e-01,  2.4032e-01,  1.5276e-01,\n",
       "           7.9512e-02,  6.2344e-01,  6.5359e-01, -3.5815e-01, -7.8679e-01,\n",
       "          -7.9041e-01, -1.7375e-02,  3.4195e-02, -2.3421e-01,  1.5630e-01,\n",
       "           8.4958e-02,  2.1768e-01, -3.0117e-02,  2.1162e-01,  1.7308e-01,\n",
       "           6.1032e-01,  3.6579e-01, -5.1869e-02, -1.9627e-01, -4.9920e-01,\n",
       "          -4.0617e-01, -1.7625e-01,  6.2134e-02,  2.0632e-01, -3.4951e-01,\n",
       "          -2.3005e-01, -2.4755e-01, -1.8158e-01,  2.7680e-01, -4.5833e-02,\n",
       "          -4.2563e-01, -3.4909e-01,  1.3696e-01,  2.2049e-01,  3.2748e-01,\n",
       "           1.7614e-01, -3.2842e-01, -3.1858e-01, -6.9246e-02,  7.4694e-02,\n",
       "           8.6884e-02,  2.7079e-01, -8.0245e-02, -4.3233e-01, -2.4885e-02,\n",
       "          -5.1160e-03,  8.4681e-01, -1.7817e-01, -6.5517e-01,  2.3112e-01,\n",
       "           4.8178e-01, -1.8096e-01, -3.0238e-01, -2.3428e-01, -2.8093e-02,\n",
       "          -2.0610e-01, -4.5873e-01,  5.8179e-01,  4.5081e-01,  4.1855e-01,\n",
       "           7.7069e-01, -2.8882e-03, -4.3614e-01,  5.4695e-01, -2.5362e-01,\n",
       "          -5.7714e-01, -2.1787e-01, -3.2134e-01, -1.0514e-01,  2.9347e-02,\n",
       "          -6.3368e-02, -1.2028e-01,  3.6711e-02, -1.1952e-01,  2.3536e-01,\n",
       "          -1.7535e-01,  1.2186e+00, -8.0169e-02, -2.2117e-01,  5.2108e-01,\n",
       "           4.9229e-01, -7.4179e-02,  1.0815e-01,  4.8907e-01,  2.9439e-01,\n",
       "           3.7798e-01, -1.9474e-01, -1.5123e-01,  2.1955e-01,  1.4308e-01,\n",
       "           7.8665e-01, -8.3584e-01, -2.7811e-01,  5.7133e-02,  4.5449e-01,\n",
       "           2.5229e-01,  2.3693e-01, -3.6992e-01,  1.1288e-01,  6.4871e-01,\n",
       "          -1.0897e+00,  7.7743e-03, -2.5594e-01,  6.4265e-01,  1.6222e-02,\n",
       "          -1.7110e-01, -1.4210e-01, -3.9511e-01, -2.2048e-01,  1.7086e-01,\n",
       "           3.6244e-01, -9.9991e-01,  1.4855e-01,  6.2576e-01, -5.3339e-01,\n",
       "           2.1243e-02, -3.7816e-01,  6.0137e-01,  1.3975e-01,  2.8528e-01,\n",
       "           2.5756e-01,  1.0201e-01, -6.8275e-02, -5.0270e-01,  2.4387e-01,\n",
       "           4.9225e-01, -7.4623e-02, -9.7867e-02,  9.1640e-02,  2.0072e-01,\n",
       "          -2.4799e-01, -5.5229e-01, -1.5302e-01, -6.4715e-02, -2.1767e-01,\n",
       "          -9.6733e-03, -1.8603e-01,  9.9003e-02,  3.0483e-01, -4.3539e-01,\n",
       "           4.6263e-01,  5.5518e-01,  8.0166e-01, -2.3406e-01, -7.3756e-04,\n",
       "           1.1018e-01,  1.6826e-01, -4.1178e-02,  4.4991e-01, -2.3209e-01,\n",
       "          -3.5686e-01,  4.3760e-01,  9.3853e-02, -1.5347e-01,  2.4318e-01,\n",
       "          -3.0115e-01,  4.7205e-03,  2.0706e-01, -3.9119e-01,  1.8953e-01,\n",
       "          -3.0704e-03, -2.3337e-02, -4.2014e-01,  8.5252e-02,  1.8308e-01,\n",
       "          -5.0971e-01,  8.0649e-01, -1.6245e-01,  7.8271e-01, -4.2675e-02,\n",
       "           2.3313e-01, -8.4019e-01, -4.5496e-01,  2.3778e-02, -1.3770e-01,\n",
       "          -5.1340e-01, -1.5414e-01, -1.1330e-01,  6.7107e-01,  4.6943e-01,\n",
       "           7.7583e-02, -1.0141e-01, -7.8928e-01, -5.8133e-01,  1.3443e-01,\n",
       "           8.8643e-02, -1.9750e-01,  7.7909e-02, -1.2201e-01,  2.8997e-01,\n",
       "          -4.7258e-01, -6.1747e-01, -4.5629e-02,  1.6377e-01, -2.6216e-01,\n",
       "           9.2567e-01, -2.2844e-01, -1.1146e-01,  3.6226e-01,  2.2668e-01,\n",
       "           2.4421e-01,  5.4398e-01,  4.6265e-02, -5.3723e-01, -1.4465e-01,\n",
       "           1.3253e-01, -2.7262e-01, -1.8598e-02, -3.7761e-01,  7.3732e-01,\n",
       "           1.8963e-01,  3.1977e-01,  6.4131e-01]], device='mps:0'),\n",
       " tensor([[-2.5187e-01, -4.0054e-01, -5.8007e-01, -8.3055e-01, -1.2868e-01,\n",
       "          -2.8804e-02, -7.1917e-01,  6.6701e-01,  1.2709e-01,  5.0855e-01,\n",
       "           1.2570e-01,  5.1916e-01,  5.1676e-02, -1.1521e-01, -1.6481e-01,\n",
       "          -4.9820e-01,  5.6222e-02,  3.7338e-01,  6.7187e-01, -2.4111e-02,\n",
       "           2.2844e-01,  5.7979e-01, -2.6323e-01, -1.2386e-02, -8.3013e-01,\n",
       "           2.9613e-01, -1.1576e-01,  2.2409e-01,  1.7718e-01, -6.3092e-01,\n",
       "           4.0999e-02,  4.7417e-01,  2.3068e-02,  1.3497e+00, -5.6080e-01,\n",
       "          -1.3230e-01,  8.2213e-02, -3.9453e-01, -2.3010e-01,  3.7283e-02,\n",
       "           6.1164e-01, -4.1315e-01,  1.3635e-01,  4.0254e-01, -6.1085e-02,\n",
       "           8.9076e-02,  1.3899e-01, -3.7744e-01,  2.5892e-01, -8.3961e-01,\n",
       "          -4.8318e-02, -2.0669e-01,  4.9144e-01,  6.2615e-01, -6.3176e-01,\n",
       "          -2.2975e-01,  2.7419e-01, -3.7716e-02, -3.3020e-02,  1.5349e-01,\n",
       "           9.7980e-02,  3.8582e-01, -2.9995e-01,  2.3257e-01,  3.3441e-01,\n",
       "           2.4458e-01, -3.0392e-01, -4.8889e-01, -5.4078e-01, -5.0111e-02,\n",
       "           9.1859e-03,  2.2490e-02,  1.4846e-01, -2.3272e-01,  4.1973e-01,\n",
       "          -7.1214e-01, -6.4354e-02, -8.9125e-02, -2.5973e-01,  2.9246e-01,\n",
       "          -4.5549e-02,  1.9719e-01,  3.1010e-01,  1.0268e-01,  1.2457e-01,\n",
       "           4.3419e-02,  2.7215e-01,  5.2932e-04,  4.7373e-02,  3.7533e-01,\n",
       "          -1.3375e-01, -8.1722e-02, -2.8313e-01, -3.5814e-01,  3.1229e-01,\n",
       "           4.0598e-02,  2.9907e-01, -1.9557e-01,  1.8579e-01, -3.2474e-01,\n",
       "           1.5588e-01,  1.7768e-01, -1.8083e-01,  5.7286e-02, -3.5172e-01,\n",
       "          -4.5475e-01,  2.3589e-02, -2.8569e-01, -4.8285e-01,  2.2697e-01,\n",
       "          -1.4211e-01, -3.0205e-01, -4.0922e-02, -3.8935e-01, -3.1524e-02,\n",
       "          -7.5164e-02,  2.6013e-01, -1.3230e-01, -3.4806e-01, -8.3033e-02,\n",
       "           2.0734e-01, -2.2808e-01,  1.6077e-01,  2.6882e-01,  1.2232e-01,\n",
       "          -9.6705e-02,  6.5251e-01,  1.8950e-02,  3.1674e-01,  2.5763e-01,\n",
       "           1.1699e-01, -6.5768e-02, -1.9745e-01, -2.3154e-01,  7.5988e-02,\n",
       "          -7.6941e-02,  1.0687e-01, -1.8182e-01, -2.2790e-01, -1.8508e-01,\n",
       "           3.1771e-01, -3.3097e-01, -6.0893e-01, -4.9260e-01, -1.6098e-01,\n",
       "          -1.6326e-01, -2.4804e-02, -8.2473e-03, -3.3693e-01, -3.1956e-01,\n",
       "          -6.7443e-01, -3.6717e-01,  8.1041e-01, -1.6831e-02, -3.3356e-01,\n",
       "          -1.6499e-01,  2.3208e-01,  3.1666e-01, -4.5959e-02,  3.7161e-01,\n",
       "           7.7250e-01, -4.7670e-02,  1.6786e-01, -5.1827e-01, -3.8195e-01,\n",
       "           4.8335e-02,  2.6658e-01, -2.7054e-01,  6.1879e-01,  4.8352e-02,\n",
       "           7.5627e-02, -6.6170e-02,  2.7430e-01, -3.2704e-01, -1.2324e-02,\n",
       "          -5.7573e-01,  3.7750e-01,  2.9539e-01, -3.9892e-01, -6.6567e-02,\n",
       "          -2.0834e-02, -1.2060e+00,  9.6964e-02, -8.3040e-01,  3.1236e-01,\n",
       "          -2.3110e-01,  1.7963e-01,  2.3389e-01,  2.9545e-01, -6.5320e-02,\n",
       "           2.5404e-01,  2.8725e-01,  2.5687e-01,  3.3609e-01, -3.4924e-02,\n",
       "           2.8019e-01,  2.3924e-01,  5.0790e-01,  4.1073e-01,  1.4769e-01,\n",
       "          -3.5440e-01,  2.9683e-01,  3.5781e-01,  9.0598e-02,  1.6954e-01,\n",
       "           9.9825e-02, -2.1636e-01,  2.1091e-01,  1.7501e-01, -1.6618e-01,\n",
       "          -2.2406e-01, -3.1100e-02, -4.6818e-01, -5.6655e-02, -4.8467e-01,\n",
       "           3.2557e-01, -5.1983e-01, -1.1912e-01,  3.4831e-01,  2.4340e-01,\n",
       "          -1.4896e-02,  1.8319e-01,  1.7196e-01, -2.4338e-01,  2.1735e-01,\n",
       "           3.7449e-01, -3.4618e-01, -3.9804e-02, -3.2197e-01, -2.9367e-01,\n",
       "           2.4357e-01, -5.6244e-01,  6.6635e-01,  4.0574e-01,  6.0165e-02,\n",
       "           1.5119e-01, -2.7920e-01, -5.1638e-02, -5.7408e-01,  3.5471e-01,\n",
       "          -3.2130e-01, -3.4955e-01, -5.4801e-01,  6.4487e-02,  4.2773e-01,\n",
       "           2.7565e-01, -6.4915e-02,  7.7868e-01,  4.2408e-01,  4.2813e-01,\n",
       "          -4.3003e-01, -1.5896e-02,  4.0377e-01, -4.1405e-02, -4.6059e-01,\n",
       "           6.7116e-02, -1.4962e-01, -5.1643e-01,  2.8000e-01, -2.6598e-01,\n",
       "          -1.2047e-01, -1.4913e-01,  3.3682e-01,  1.5365e-01, -2.1964e-01,\n",
       "          -3.8461e-01,  3.9587e-01, -1.4349e-01, -4.1723e-01,  4.7123e-01,\n",
       "          -3.5025e-01,  1.5209e-01,  1.0694e-01, -3.8441e-01, -6.2072e-01,\n",
       "          -3.2235e-01,  2.3090e-01,  1.5585e-01,  3.4799e-02, -6.6306e-01,\n",
       "          -4.0571e-01,  3.1197e-01,  2.4601e-01,  2.9976e-01,  2.9840e-01,\n",
       "           5.7427e-01, -3.0944e-01, -1.6702e-01,  1.4982e-01, -4.1377e-01,\n",
       "          -1.5704e-01, -3.3228e-02, -1.9995e-01,  2.9569e-01,  2.4768e-01,\n",
       "           1.1600e-01,  2.0367e-01,  4.3482e-01, -2.5272e-01, -3.8673e-01,\n",
       "          -7.1543e-02,  2.0161e-01, -9.3629e-02, -5.8306e-02,  2.3951e-01,\n",
       "          -6.3095e-02,  5.9650e-01,  7.2143e-01,  7.7439e-02, -6.4006e-01,\n",
       "           1.7696e-01,  5.2502e-01, -2.7253e-01, -4.0343e-02,  9.6365e-02,\n",
       "          -1.0344e+00, -2.1331e-01, -1.2883e-01, -2.0181e-01, -6.5352e-01,\n",
       "           8.8115e-02,  3.4863e-01,  4.6833e-03, -1.7864e-01, -3.0947e-01,\n",
       "           1.0448e-01,  4.8724e-02, -4.8980e-02,  4.3534e-01, -2.8901e-01,\n",
       "          -3.4797e-01, -2.3438e-01, -2.1532e-01, -5.4201e-01, -3.9968e-01,\n",
       "           3.5914e-01,  2.8129e-01,  1.1986e-01, -2.6486e-01, -5.1832e-01,\n",
       "           4.5577e-01, -6.8169e-01, -1.4632e-01,  2.4840e-01, -4.7702e-01,\n",
       "           1.4650e-02,  4.8990e-01,  1.4999e-01,  1.6336e-01,  6.5258e-02,\n",
       "           7.8557e-02, -3.3903e-01, -5.0030e-01, -1.5000e-01,  6.2323e-01,\n",
       "           3.2283e-01,  3.3506e-01, -1.7236e-01,  3.9096e-01,  4.5167e-01,\n",
       "          -2.9582e-01,  4.1670e-01,  1.0930e-02,  7.7439e-02, -6.0414e-02,\n",
       "           1.9048e-01,  3.4370e-01, -3.6842e-01,  2.0145e-01,  1.7929e-01,\n",
       "          -3.1634e-01, -3.2149e-01,  1.5691e-02,  2.0642e-01, -1.8580e-01,\n",
       "          -3.0875e-01, -2.0361e-01,  9.9698e-02, -3.4856e-01, -4.3104e-01,\n",
       "           2.9161e-01, -3.6543e-02,  7.0495e-02, -4.0357e-01, -4.3462e-01,\n",
       "          -2.0181e-01, -2.6055e-01,  6.8681e-01,  2.1064e-01, -1.9621e-01,\n",
       "           1.8720e-01, -1.0907e-01, -3.6330e-01,  4.7847e-01,  2.0162e-02,\n",
       "          -4.8754e-01,  4.5482e-01, -3.4822e-01,  1.0627e-01,  5.5756e-01,\n",
       "          -1.6769e-01, -1.0840e-01,  1.1008e-01,  1.0820e-01,  2.9026e-01,\n",
       "          -3.9884e-01,  1.5516e-01, -6.3476e-02, -3.1885e-01,  3.1820e-01,\n",
       "           2.6411e-01,  2.2781e-01, -1.7137e-01, -3.3426e-02,  3.5677e-02,\n",
       "           2.8782e-01, -6.0853e-02, -3.9888e-01, -4.7347e-01, -4.6132e-01,\n",
       "           1.2832e-01,  1.8927e-01,  5.6207e-01,  3.4144e-01, -1.1394e+01,\n",
       "           1.2938e-01,  7.8031e-01, -4.0713e-01, -8.2819e-02, -5.0320e-02,\n",
       "           1.0976e-02, -1.3708e-01,  4.2955e-01, -2.7734e-01, -4.4105e-01,\n",
       "          -3.5727e-01, -1.3308e-01,  1.5454e-01, -3.2189e-01, -2.6945e-01,\n",
       "          -1.4686e-02, -7.0935e-01,  4.0963e-01, -3.0286e-01,  5.0298e-01,\n",
       "          -2.3511e-01, -8.3798e-02,  1.5816e-01, -3.5851e-01, -1.3439e-02,\n",
       "           5.9766e-02, -1.6033e-01,  1.4651e-01,  4.2486e-01, -5.2072e-02,\n",
       "           3.3244e-02, -4.1258e-03,  4.8461e-02, -2.5723e-01, -5.9999e-01,\n",
       "           1.9296e-02, -1.4141e-01,  3.3422e-01, -4.4133e-02, -7.1265e-01,\n",
       "          -2.2533e-01, -2.7418e-01,  3.1407e-01,  2.7021e-01,  9.0598e-01,\n",
       "           2.2955e-01,  1.0686e+00,  1.5022e-01,  1.7748e-01,  2.6230e-02,\n",
       "          -7.8764e-01, -3.3269e-02, -4.1907e-01,  2.3280e-02,  1.0114e-01,\n",
       "           3.6576e-01, -7.9367e-01,  2.5251e-01, -3.5521e-01,  3.6487e-02,\n",
       "           4.9094e-01,  3.1768e-02,  2.7700e-01, -3.2298e-02, -7.8123e-02,\n",
       "           1.2089e-01, -2.8544e-02,  1.1098e-01, -2.2583e-01, -2.4084e-01,\n",
       "          -1.0177e-01,  5.9888e-01,  3.9592e-01, -7.2211e-01,  3.6112e-01,\n",
       "           4.9258e-02, -1.8622e-01,  1.4556e+00, -4.8573e-02,  1.8361e-02,\n",
       "           1.9800e-03,  2.9604e-01, -5.0236e-01, -7.3516e-01,  3.6511e-01,\n",
       "          -1.0211e-01,  5.6427e-02,  1.0127e-01,  4.8148e-01, -9.7481e-02,\n",
       "          -4.7940e-01, -1.8593e-03,  1.0556e-01, -1.3380e-01,  5.4710e-01,\n",
       "          -7.2824e-01, -1.6963e-01, -1.1223e-01,  3.1723e-01, -8.4054e-01,\n",
       "          -6.1126e-02, -7.0210e-01, -3.9328e-01,  5.0586e-01, -2.5896e-01,\n",
       "           1.5852e-01, -2.7443e-02, -4.3439e-02,  5.5574e-01, -2.1489e-01,\n",
       "           7.1555e-02,  2.2813e-01,  8.4995e-02, -1.6733e-01,  3.9349e-02,\n",
       "           4.1630e-01,  4.9571e-01, -4.2740e-01, -6.3911e-01, -7.6901e-03,\n",
       "           1.7180e-01,  1.4933e-01,  3.4019e-01, -5.3463e-01,  3.6431e-01,\n",
       "          -7.9162e-01, -1.8769e-01,  7.2949e-02, -4.3273e-02,  3.6175e-01,\n",
       "           2.3925e-01,  7.2038e-02,  2.2256e-01, -7.7756e-01, -6.4498e-01,\n",
       "          -6.5932e-01, -2.1445e-01,  2.6143e-01, -5.4236e-01, -8.3654e-02,\n",
       "          -1.1033e-01, -4.8837e-03, -5.0292e-02, -6.6102e-02,  1.1939e-02,\n",
       "           1.3632e-01,  2.3587e-01, -1.3023e-01, -1.3178e-01, -9.2464e-02,\n",
       "          -5.5566e-01, -1.1753e-01, -9.9591e-02, -1.2505e-01, -5.6783e-01,\n",
       "          -1.3797e-01,  3.3075e-02, -2.5373e-01, -6.8124e-02,  3.5608e-01,\n",
       "          -2.9832e-02, -2.7856e-02, -4.8850e-01,  7.2846e-01,  4.0279e-01,\n",
       "           5.2088e-02, -2.8298e-01, -5.5189e-01, -1.0567e-01,  2.9592e-01,\n",
       "           1.9555e-01, -6.7007e-02, -1.7709e-01, -2.4145e-01, -4.2522e-01,\n",
       "           1.4302e-01,  7.1147e-01, -6.6296e-02, -6.8115e-01,  2.6621e-01,\n",
       "           5.8223e-01, -1.2882e-01, -2.4432e-01,  2.0983e-02, -9.3284e-02,\n",
       "          -3.1851e-01, -6.0317e-01,  7.8821e-01, -1.1246e-01, -5.8638e-02,\n",
       "           2.2427e-01,  2.0359e-01, -9.2588e-03,  4.0209e-01, -1.5770e-01,\n",
       "          -3.5434e-01, -1.6477e-03, -7.3600e-01,  4.3257e-01,  6.1666e-02,\n",
       "          -3.1489e-01,  7.6082e-01,  2.7260e-01, -6.0593e-02, -4.5779e-02,\n",
       "           2.2863e-01,  7.9941e-01, -2.4636e-01,  6.4653e-02,  5.3779e-01,\n",
       "           8.8110e-02, -1.1229e-01,  3.9580e-02,  4.1665e-02,  4.5410e-02,\n",
       "           3.7012e-01, -3.2145e-02, -4.1564e-01, -6.1194e-02,  3.8924e-01,\n",
       "           3.9295e-01, -5.4610e-01, -7.3487e-01,  2.2333e-02,  6.8551e-01,\n",
       "          -2.9054e-01,  2.3654e-01, -3.2708e-02,  3.0880e-01,  1.2001e-01,\n",
       "          -9.2160e-01, -1.5016e-01, -1.0056e-01,  2.8229e-01, -2.5482e-02,\n",
       "          -1.6139e-01, -1.4727e-01, -2.3686e-01, -1.6208e-01,  6.8308e-01,\n",
       "           2.1991e-01, -6.9160e-01,  3.4058e-02,  5.7281e-01, -6.0551e-01,\n",
       "           4.6052e-01, -2.4577e-01, -3.4302e-02,  3.6435e-01,  1.3117e-01,\n",
       "           4.9067e-02,  1.7856e-01, -4.2138e-01, -4.0598e-01,  1.7824e-01,\n",
       "           9.1672e-01, -7.6613e-02,  1.1166e-01, -3.9384e-01, -1.7182e-01,\n",
       "          -7.1855e-02, -3.1913e-01, -3.1286e-01,  3.7185e-01,  2.0529e-01,\n",
       "          -3.2552e-02, -1.3271e-01, -2.2992e-01,  3.0816e-01,  1.1821e-01,\n",
       "           3.2536e-01,  5.8187e-01,  6.4368e-01, -1.0077e-02, -2.4936e-01,\n",
       "          -6.1235e-02,  1.4568e-01, -1.9524e-01,  5.2704e-01, -2.0627e-01,\n",
       "          -3.3453e-01, -3.8069e-03,  3.0134e-01, -6.1681e-01, -5.8725e-02,\n",
       "          -4.5426e-01,  4.2361e-01, -1.6955e-01,  3.7868e-02,  2.8833e-01,\n",
       "           4.0522e-02,  5.7925e-02, -2.2651e-01,  5.5807e-01,  4.4511e-01,\n",
       "           1.6609e-01,  4.5662e-01,  1.6104e-01,  4.5272e-02, -5.1597e-01,\n",
       "           2.1978e-01, -1.0213e+00, -9.4260e-02, -4.6640e-02, -1.5401e-01,\n",
       "          -4.4224e-01, -2.8126e-02,  1.6814e-01,  1.8538e-01,  6.9597e-01,\n",
       "           7.2967e-01, -3.0871e-01, -4.5290e-01, -1.8510e-01,  1.5304e-01,\n",
       "           2.7705e-01, -1.9517e-01,  3.5971e-01,  9.6103e-02,  2.4432e-01,\n",
       "          -5.0967e-01, -4.3165e-01, -4.2252e-02,  3.1411e-01, -2.0171e-01,\n",
       "           5.7118e-01, -2.0164e-01, -1.3863e-01,  9.0832e-02,  1.3770e-01,\n",
       "           1.6639e-01,  7.9664e-01,  4.3416e-01, -6.3425e-01,  8.7469e-02,\n",
       "          -1.3980e-02, -2.9988e-01,  2.5983e-02, -4.3577e-01,  2.8858e-01,\n",
       "           1.2984e-01,  4.9280e-01,  6.7504e-01]], device='mps:0'),\n",
       " tensor([[-0.0932, -0.6293, -0.4879,  ...,  0.3041, -0.1958,  0.5576],\n",
       "         [-0.1242, -0.5179, -0.4093,  ...,  0.3691, -0.0529,  0.5134],\n",
       "         [-0.1672, -0.5249, -0.4540,  ...,  0.3771,  0.2016,  0.5340]],\n",
       "        device='mps:0'),\n",
       " tensor([[-9.3362e-03, -7.6677e-01, -2.1541e-01, -7.6450e-01, -2.4034e-01,\n",
       "          -3.4842e-01, -7.6284e-01,  5.6302e-02,  1.5109e-01,  4.0040e-01,\n",
       "           1.1321e-01,  5.7407e-01,  1.6392e-02, -1.0792e-01, -5.6478e-01,\n",
       "          -8.0959e-03, -5.1203e-01,  2.3219e-01,  5.8300e-01,  4.0974e-01,\n",
       "           1.1817e-01,  3.7767e-01, -9.5571e-02,  1.8691e-01, -3.9888e-01,\n",
       "          -1.0256e-01,  1.2107e-01, -7.5519e-02, -4.5555e-02, -4.6745e-01,\n",
       "           2.0564e-01,  5.9403e-01, -1.5869e-01,  9.4833e-01,  1.9096e-01,\n",
       "          -3.1064e-01,  1.9828e-01, -5.3826e-01,  2.5114e-01, -8.7310e-02,\n",
       "           3.7418e-01, -8.8314e-02,  1.6757e-01,  2.1145e-01,  6.5556e-02,\n",
       "           2.1201e-01,  7.2850e-02, -2.7350e-01,  2.2207e-01,  6.6317e-02,\n",
       "          -2.3835e-01, -2.7685e-01,  5.3458e-01,  1.5658e-01, -2.7097e-01,\n",
       "          -2.9253e-01,  1.5293e-02,  3.6885e-01, -3.2769e-01, -2.7033e-01,\n",
       "          -3.1601e-01,  2.4631e-01, -2.4872e-01, -8.3147e-02,  3.6152e-02,\n",
       "          -2.6122e-01, -6.6692e-01, -4.1990e-01, -8.7783e-01, -1.1621e-01,\n",
       "           2.2628e-01,  2.5113e-01,  1.8853e-01,  1.9692e-01,  4.3793e-01,\n",
       "          -1.9610e-01,  1.6006e-01, -2.4895e-01, -6.1875e-01,  2.1313e-03,\n",
       "          -6.5234e-02,  2.1058e-01,  8.2293e-02, -2.5059e-01, -9.1222e-02,\n",
       "          -3.8137e-01,  6.0491e-01, -2.2968e-02, -4.5486e-02, -1.5314e-01,\n",
       "           1.4532e-01, -1.0920e-01, -3.5060e-01,  8.0505e-02,  1.4198e-01,\n",
       "          -2.7496e-02,  3.7310e-01, -4.2341e-01, -3.9907e-01, -5.6702e-01,\n",
       "           2.4140e-01, -8.5457e-02,  1.9135e-01,  5.0658e-01, -3.2913e-01,\n",
       "          -5.9091e-01,  2.0181e-01, -2.4193e-01, -5.7645e-01, -1.9080e-01,\n",
       "          -3.2410e-02, -7.9350e-02,  4.1442e-01, -3.7657e-01, -2.6307e-01,\n",
       "           1.2284e-01,  1.1624e-01, -4.0876e-02, -3.3125e-01,  6.1128e-02,\n",
       "           1.8405e-01, -3.2039e-02, -6.1523e-02,  5.7802e-02, -8.9551e-02,\n",
       "           6.8527e-02,  5.1348e-01, -8.9487e-02,  3.7025e-01,  4.2176e-01,\n",
       "          -2.3509e-01,  2.9254e-01, -2.9448e-01,  5.6386e-02, -3.2099e-03,\n",
       "           2.6384e-02, -1.1502e-01,  1.8555e-02,  1.8462e-01,  8.9972e-03,\n",
       "          -2.4547e-01, -1.0367e-01, -2.0584e-02,  1.9049e-01,  3.1628e-01,\n",
       "          -1.6252e-01, -3.1133e-02, -3.3074e-01, -1.3156e-01, -3.3504e-01,\n",
       "          -7.1071e-01, -6.4999e-01,  1.9419e-01, -1.6550e-01,  1.0181e-01,\n",
       "          -1.4422e-01,  1.8076e-01,  3.1225e-01, -2.0700e-02,  4.2905e-02,\n",
       "           1.0355e+00,  2.8224e-02,  1.1028e-01, -4.2242e-01,  6.6875e-02,\n",
       "          -1.0803e-01,  2.5337e-01, -1.1155e-01, -1.5244e-01, -1.8900e-02,\n",
       "          -8.6447e-02, -2.1378e-01,  4.4614e-01, -1.0831e-01,  2.4949e-01,\n",
       "          -3.5918e-01,  4.9993e-01, -4.2215e-02, -2.0303e-01, -1.7452e-01,\n",
       "           1.7889e-01, -1.4422e+00,  1.3163e-01, -1.1238e-01, -3.1916e-01,\n",
       "          -7.1467e-01,  2.0827e-01, -2.6912e-01,  2.8816e-01,  1.5817e-01,\n",
       "          -2.0754e-01,  1.2273e-02,  3.8470e-01,  1.2810e-01, -3.5067e-02,\n",
       "           2.6208e-01,  9.1924e-01,  5.3252e-01,  9.1512e-02, -5.6009e-03,\n",
       "          -4.2044e-01,  1.1729e-01,  3.3410e-01,  1.7715e-01,  3.2705e-01,\n",
       "           4.3609e-01, -1.3549e-01,  2.2114e-01,  1.9472e-01, -9.9983e-02,\n",
       "          -2.4581e-01, -1.8751e-01, -5.6995e-01, -1.1028e-01, -6.7493e-01,\n",
       "           5.4235e-01, -2.9585e-01, -7.0956e-02,  3.9030e-01, -1.4294e-01,\n",
       "          -2.9792e-01, -2.8527e-01,  1.2801e-01, -3.8556e-01,  1.2344e-01,\n",
       "           1.4992e-02,  5.2858e-02,  2.5381e-01, -7.0780e-02, -1.2696e-01,\n",
       "           1.6956e-01, -6.3319e-01,  4.8779e-01,  2.6226e-01, -2.5770e-01,\n",
       "          -5.0705e-02, -8.9804e-03,  1.4307e-01, -4.3268e-02, -1.5705e-01,\n",
       "          -4.0808e-01, -3.1880e-01, -7.1499e-01, -1.0101e-01, -1.6671e-01,\n",
       "           1.6526e-01, -1.8692e-01,  6.8668e-01,  5.0023e-01,  5.0139e-01,\n",
       "           1.0721e-01, -1.0809e-01,  2.3825e-01,  2.3724e-01, -4.5239e-01,\n",
       "          -6.8748e-03,  1.5108e-01, -1.5344e-01,  1.9884e-01, -2.2085e-01,\n",
       "          -1.6174e-01,  2.9263e-02,  1.0911e-01,  5.4624e-01,  4.7460e-02,\n",
       "          -1.2276e-01,  1.7854e-01, -2.3804e-01, -1.5214e-01,  2.6616e-01,\n",
       "          -2.3826e-01,  3.0464e-01,  2.3556e-01, -7.0255e-01, -5.5893e-01,\n",
       "          -6.7597e-01, -3.5512e-01,  1.4726e-01,  3.2621e-01, -8.5043e-01,\n",
       "          -6.6184e-02, -5.3878e-02, -2.7010e-01,  5.9152e-02, -3.9423e-01,\n",
       "           1.2526e-01,  1.9927e-02, -1.5439e-01, -3.7634e-01, -4.9778e-01,\n",
       "          -1.6433e-01,  1.1344e-01, -6.4549e-01,  5.5198e-02,  2.6885e-01,\n",
       "          -2.4222e-01,  3.9591e-01,  1.8553e-01,  5.0692e-02, -2.2584e-01,\n",
       "          -2.4917e-01,  2.6325e-01, -1.3841e-04,  3.5895e-02, -4.4445e-02,\n",
       "           1.7311e-01,  3.3362e-01,  6.4096e-01, -3.0470e-01, -6.8165e-01,\n",
       "          -3.2324e-01,  8.2645e-01, -4.8813e-02, -4.8314e-01, -5.6699e-02,\n",
       "          -1.0637e+00, -3.7420e-01, -5.5169e-02, -2.6771e-01, -5.0605e-01,\n",
       "           2.1276e-01,  1.6916e-01, -4.7763e-02,  1.0619e-01,  2.3367e-02,\n",
       "           6.0488e-01,  5.1779e-01,  2.5483e-01,  4.2313e-01, -2.1606e-01,\n",
       "          -2.0005e-01, -6.2941e-01, -2.0123e-01, -4.2634e-01, -2.6481e-01,\n",
       "           1.3306e-01,  5.4079e-01, -3.4426e-01, -6.4218e-02, -2.5055e-01,\n",
       "           6.1514e-01, -2.8470e-01, -4.9141e-01,  6.4503e-01, -2.2475e-01,\n",
       "           4.5148e-01,  2.1801e-01, -1.0176e-01, -1.0093e-01,  2.4131e-01,\n",
       "          -1.2493e-01,  6.1917e-02,  8.8688e-02, -1.5489e-01,  4.0513e-01,\n",
       "           4.1299e-01, -1.8455e-02, -4.0942e-01,  3.0867e-01,  2.7989e-01,\n",
       "          -3.2938e-01,  3.1920e-02, -2.7842e-01,  2.7697e-02,  1.8487e-01,\n",
       "           2.8759e-01,  4.3853e-01, -5.1904e-01,  2.3803e-01,  5.1241e-02,\n",
       "           2.2027e-01, -9.9217e-02, -1.7484e-01,  9.4205e-02, -4.2853e-02,\n",
       "          -2.1999e-01, -5.7029e-01,  1.4148e-01, -4.3884e-01,  5.7149e-02,\n",
       "           1.9019e-01,  1.8446e-01, -2.2901e-01, -2.2170e-01, -2.3721e-01,\n",
       "          -1.9180e-01, -7.0943e-02,  9.2870e-01,  5.3026e-01, -2.6835e-01,\n",
       "           2.4453e-01, -6.4669e-02, -1.1915e-01,  4.8114e-01, -1.9753e-01,\n",
       "          -4.1824e-01,  2.9277e-01, -5.6200e-01,  3.7823e-01,  4.4028e-01,\n",
       "          -5.2738e-01, -2.2726e-01,  1.7392e-01,  6.9855e-01,  6.9265e-02,\n",
       "          -1.9761e-01, -1.0830e-02,  2.5953e-01, -1.0218e-01,  2.1655e-01,\n",
       "           3.1190e-01,  9.0676e-01, -3.3991e-01,  3.2129e-01, -3.6893e-01,\n",
       "           2.0846e-01, -1.9687e-01,  1.6508e-01, -3.0328e-01, -1.5812e-01,\n",
       "           6.7094e-01,  2.7710e-01,  2.5110e-01,  2.3429e-01, -1.1389e+01,\n",
       "           6.5502e-01,  6.9774e-01, -6.3505e-01, -1.8791e-01, -1.0213e-01,\n",
       "           2.6884e-01,  9.7986e-04,  2.4583e-01, -3.0139e-01, -1.7789e-01,\n",
       "          -5.2046e-01,  1.5956e-01,  1.6775e-01, -4.2504e-01, -1.6963e-01,\n",
       "          -1.0109e-01, -5.8410e-01,  3.6942e-01, -6.6914e-02,  4.6283e-01,\n",
       "          -1.0696e-01, -1.2464e-02,  1.0441e-01, -2.7326e-01,  7.8947e-02,\n",
       "           1.4064e-01,  2.6105e-01,  4.2993e-01,  2.2448e-01, -3.6510e-02,\n",
       "           1.2017e-01,  4.0608e-02,  3.1029e-01, -5.2329e-01, -6.1520e-01,\n",
       "           2.3862e-01, -3.0433e-01,  2.5140e-01, -2.2966e-01, -4.9148e-02,\n",
       "          -1.3000e-01, -1.2184e-01,  4.4620e-01,  1.1073e-01,  6.0459e-01,\n",
       "           1.1515e+00,  3.7076e-01,  1.7997e-01,  1.2114e-01,  1.7640e-01,\n",
       "          -8.5286e-01,  3.4642e-01, -6.1470e-01,  3.7445e-02, -1.3153e-01,\n",
       "           3.7917e-01, -7.2483e-01, -5.7047e-02, -2.8894e-01, -3.8283e-01,\n",
       "          -9.2371e-02,  2.4421e-01, -2.6092e-02, -7.1778e-01, -9.1306e-02,\n",
       "           2.2197e-01, -2.0549e-01,  2.0653e-01, -3.8558e-01, -4.5315e-02,\n",
       "          -3.1685e-02,  1.2349e-01,  3.1747e-01,  2.7599e-01,  2.0832e-01,\n",
       "          -1.7998e-01,  5.9894e-02,  1.5225e+00, -3.9188e-01,  7.3388e-02,\n",
       "          -2.0444e-01,  1.7054e-02, -1.8317e-01, -1.3891e-01, -3.9470e-01,\n",
       "          -8.3557e-02, -7.9324e-03,  1.9629e-01,  5.2355e-01, -3.6208e-01,\n",
       "          -4.5763e-01,  1.5164e-01,  1.0735e-01,  3.5039e-01,  9.5346e-01,\n",
       "          -9.4353e-01,  2.1876e-02,  6.5925e-02, -1.3981e-02, -6.1750e-01,\n",
       "           2.2193e-01, -8.7891e-02, -7.2904e-01,  2.2076e-01,  2.9418e-01,\n",
       "           1.2917e-01, -2.7932e-01,  1.1589e-01,  3.1183e-01, -1.2318e-01,\n",
       "           3.6136e-01, -4.3605e-02,  1.1499e-01, -2.6866e-01, -6.4522e-01,\n",
       "           3.0025e-01,  6.9398e-01, -1.4134e-01, -5.0138e-01,  1.2401e-01,\n",
       "           1.2598e-01, -8.5035e-02,  2.2707e-01, -3.5131e-01,  3.4162e-02,\n",
       "          -4.5417e-01, -6.0492e-02, -3.6292e-02, -1.3435e-01,  3.8186e-01,\n",
       "           2.4095e-01, -1.0717e+00, -2.5678e-01, -4.6240e-01, -4.8925e-01,\n",
       "          -8.1995e-01, -2.7147e-01,  3.2315e-01, -2.7225e-01, -3.1544e-01,\n",
       "          -3.0173e-02, -1.0815e-01, -2.6250e-01,  2.5386e-02,  2.2402e-01,\n",
       "           1.1678e-01,  2.2783e-01, -5.0519e-01,  1.5716e-01,  6.7321e-02,\n",
       "          -1.2715e-01, -1.7574e-01, -2.9794e-01,  2.3782e-01, -1.4374e-01,\n",
       "          -3.0633e-01,  1.4332e-01, -3.3587e-01, -4.8872e-01, -4.2970e-01,\n",
       "           6.7526e-01,  9.1294e-04, -6.3423e-01,  8.7640e-01,  1.5532e-01,\n",
       "           1.0565e-01, -5.3176e-01,  9.8671e-02, -2.1196e-01,  1.1312e-01,\n",
       "           1.6191e-01, -1.9886e-01, -3.4610e-01, -5.8781e-01, -9.0459e-03,\n",
       "           1.1273e-01,  1.0472e+00, -2.0181e-01, -4.8700e-01,  1.0948e-01,\n",
       "           5.0907e-01,  5.4538e-02,  5.7157e-02,  2.5832e-02,  2.5558e-01,\n",
       "          -1.9220e-01, -8.7473e-02,  3.4582e-01, -2.0488e-01,  3.3239e-01,\n",
       "           1.3991e-01, -4.3841e-02, -4.0026e-01,  9.7974e-02, -4.6212e-02,\n",
       "          -3.2926e-01, -1.7781e-01, -3.2328e-01, -2.2887e-01,  2.3503e-01,\n",
       "          -5.5779e-01,  1.8577e-01, -1.1891e-01,  2.1361e-01, -3.3514e-01,\n",
       "           3.2549e-01,  1.1205e+00,  5.5355e-02, -3.2342e-01,  6.5373e-01,\n",
       "           2.9508e-01,  6.7862e-01,  3.4052e-01,  3.7321e-01,  1.1422e-01,\n",
       "           6.8802e-02, -1.0208e-01, -3.3644e-01,  2.2852e-01,  1.6952e-01,\n",
       "           6.6221e-01, -6.3757e-01, -2.0130e-01,  3.1177e-01,  2.4171e-01,\n",
       "          -6.0267e-02,  3.9591e-01, -3.3592e-01,  5.6899e-01,  7.7964e-01,\n",
       "          -1.2127e+00, -9.3061e-02, -2.5760e-02, -2.2144e-03, -2.0823e-03,\n",
       "          -7.3736e-01, -2.8508e-01, -8.7894e-02, -3.2744e-01,  2.4900e-01,\n",
       "           6.6695e-01, -9.7905e-01,  4.6716e-02,  7.5940e-01, -2.5606e-02,\n",
       "           2.8667e-01, -2.4440e-01,  2.3361e-01,  1.6516e-03,  1.8606e-01,\n",
       "          -2.0503e-02,  2.3058e-01, -7.4546e-01, -4.8741e-01,  3.6689e-01,\n",
       "           5.4729e-01, -6.2968e-02,  3.9003e-02, -5.1263e-02,  1.8634e-01,\n",
       "          -9.7204e-03,  9.9246e-02, -1.2613e-01, -1.4622e-01, -3.8010e-02,\n",
       "           1.5440e-01, -3.8744e-01,  1.0084e-01,  6.0546e-01, -3.1769e-01,\n",
       "           2.6450e-01,  4.5097e-01,  5.9714e-01, -1.1272e-01, -8.5926e-02,\n",
       "           1.6573e-01,  3.8779e-01, -2.8603e-01,  3.8524e-01,  6.2453e-02,\n",
       "          -4.0516e-01, -1.0955e-01,  5.0687e-02, -4.7914e-01,  3.5469e-01,\n",
       "          -3.2568e-01,  1.9325e-02, -2.1964e-01, -8.7912e-02,  3.0504e-01,\n",
       "           6.5893e-02,  3.0175e-01, -4.7765e-01,  2.8358e-02,  2.1910e-01,\n",
       "           1.8291e-01,  1.1073e+00, -3.5656e-02,  3.7359e-01, -1.9200e-01,\n",
       "           4.2825e-02, -7.8978e-01, -9.3282e-02,  2.9709e-02, -2.9800e-01,\n",
       "          -2.5966e-01, -3.4971e-01, -6.1916e-02,  1.6474e-01,  9.1079e-01,\n",
       "           7.0980e-01, -2.3285e-02, -5.0526e-01, -4.5522e-01, -1.5850e-02,\n",
       "           2.8135e-01,  1.4917e-01,  5.7719e-01,  2.6700e-01,  1.8490e-01,\n",
       "          -7.5903e-01, -3.2534e-01, -1.0213e-01,  4.1504e-01, -1.4955e-01,\n",
       "           8.6783e-01,  1.4243e-01, -3.6938e-02,  5.3273e-01,  1.2353e-01,\n",
       "           2.5421e-01,  5.3421e-01, -2.9279e-01, -7.6181e-01, -3.6082e-02,\n",
       "           3.5354e-01, -2.0502e-01,  1.1581e-01, -8.5590e-01,  2.6546e-01,\n",
       "           3.5680e-01,  6.7867e-01,  7.1248e-01]], device='mps:0'),\n",
       " tensor([[-4.5993e-01, -5.5362e-01, -4.3222e-01, -8.5138e-01,  1.6672e-02,\n",
       "          -5.6179e-01, -3.2278e-01,  6.3303e-01, -3.4063e-01,  7.0084e-01,\n",
       "           1.7761e-01,  9.4700e-02,  2.5333e-01,  2.8317e-01, -5.4294e-01,\n",
       "          -3.4952e-01, -6.0846e-02,  3.8834e-01,  4.8692e-01,  2.5059e-01,\n",
       "           4.0799e-01,  4.8037e-01, -1.4129e-01, -8.8928e-02, -1.7045e-01,\n",
       "           1.9112e-01,  1.1345e-02, -6.8891e-02,  1.8440e-01, -2.9933e-01,\n",
       "           9.6148e-02,  4.0075e-01, -1.8974e-01,  1.1887e+00, -2.2727e-01,\n",
       "          -4.0091e-01,  2.8867e-01, -4.6887e-01,  8.2426e-02, -1.4291e-01,\n",
       "           7.1590e-01, -2.6750e-01,  2.4529e-01,  4.8493e-03, -1.8979e-01,\n",
       "           2.5745e-02,  1.9280e-01, -3.3482e-01,  3.5466e-01, -8.3483e-01,\n",
       "          -5.3604e-01, -1.8416e-01,  2.0978e-01,  3.8217e-01, -7.0795e-01,\n",
       "          -5.8362e-01,  3.0541e-01,  4.4934e-02, -3.6326e-01,  1.2290e-01,\n",
       "          -1.2605e-01,  3.5731e-01, -3.8981e-02,  1.8661e-01,  8.3088e-02,\n",
       "          -2.6203e-03, -4.8224e-01, -2.9802e-01, -3.1033e-01, -2.9062e-02,\n",
       "           9.8921e-03,  3.1789e-01,  1.1525e-01,  9.7677e-02,  7.2359e-02,\n",
       "          -1.5005e-01,  4.0906e-01,  5.4104e-02,  1.5327e-01,  1.6452e-01,\n",
       "          -3.9623e-02,  2.8683e-01,  1.2109e-01,  7.4832e-03, -7.2980e-02,\n",
       "          -1.0339e-01,  6.7938e-01, -3.2434e-02,  1.3032e-01,  1.1854e-01,\n",
       "           2.8072e-01, -1.3293e-02, -2.6796e-01, -1.5814e-01,  4.6240e-01,\n",
       "           3.3992e-02,  6.0497e-01, -1.6469e-01,  2.8352e-01, -2.8332e-01,\n",
       "          -7.7636e-02, -7.2880e-02, -2.0909e-01,  1.1373e-01, -2.6614e-01,\n",
       "          -2.8088e-01,  7.2642e-02, -1.4830e-01, -2.8166e-01,  2.5782e-01,\n",
       "          -1.0221e-01, -2.7325e-01,  4.6092e-02, -4.2246e-01, -2.6597e-01,\n",
       "          -6.3725e-02,  1.8987e-01,  6.2450e-02, -4.9651e-01,  1.1041e-01,\n",
       "           3.1975e-01, -1.5889e-01,  2.0613e-01, -2.0693e-02,  4.2340e-02,\n",
       "          -2.1489e-01,  6.7447e-01, -1.6136e-01,  8.3545e-02,  4.6881e-01,\n",
       "          -1.5642e-02,  3.6852e-02, -3.8681e-01,  2.0587e-01,  1.8319e-01,\n",
       "           5.8282e-02, -3.3157e-01, -1.2672e-01, -1.8939e-01,  4.9540e-02,\n",
       "           2.0019e-01, -3.7137e-01, -3.4241e-01, -4.4561e-01, -3.5659e-02,\n",
       "          -4.7467e-01,  1.3322e-01, -2.7377e-01, -7.2389e-02, -5.6886e-01,\n",
       "          -5.5641e-01, -5.8016e-02,  9.3289e-01,  2.7031e-02, -1.5282e-02,\n",
       "          -1.6398e-02,  2.6797e-01, -1.7225e-01,  2.7468e-01,  8.3999e-02,\n",
       "           7.6757e-01, -1.9212e-01, -1.9223e-01, -5.1872e-01,  6.0627e-03,\n",
       "          -9.3505e-04,  1.4530e-01, -5.0927e-01,  1.8013e-01,  1.3595e-02,\n",
       "           1.3701e-01, -6.5411e-02,  2.4723e-01, -1.2483e-01, -1.7866e-01,\n",
       "          -6.7839e-01,  9.6356e-02,  3.2594e-02, -8.9517e-02, -1.2465e-01,\n",
       "           3.7897e-01, -1.0536e+00,  2.7615e-01, -3.3306e-01, -2.5856e-01,\n",
       "          -2.7364e-01, -3.2196e-02,  1.5536e-01,  2.7892e-01,  1.1584e-01,\n",
       "          -1.3100e-01, -1.4517e-01, -1.0146e-01,  2.5224e-01, -2.9863e-01,\n",
       "           1.2539e-01,  6.0411e-01,  4.5658e-01,  1.2033e-01, -8.3314e-05,\n",
       "          -3.8162e-01,  1.0519e-01, -1.0172e-01, -6.0927e-03,  1.2036e-01,\n",
       "           1.7749e-01, -4.6927e-01,  6.6624e-02,  1.6120e-01,  4.6699e-02,\n",
       "          -1.5706e-01, -2.0425e-01, -2.9699e-01, -1.5202e-01, -1.3974e-01,\n",
       "           3.3998e-02, -3.6657e-01, -9.5448e-02,  4.8061e-01,  1.1887e-01,\n",
       "           4.4784e-02,  8.8612e-02,  1.4218e-01, -1.4530e-01,  1.4258e-01,\n",
       "           3.9321e-01, -3.6568e-01, -1.0475e-01, -6.3098e-02,  1.0535e-02,\n",
       "           3.1746e-02, -9.9502e-01,  4.6614e-01,  2.4682e-01,  1.2204e-01,\n",
       "          -7.8257e-02, -1.9979e-01,  3.2038e-01,  1.3557e-01, -9.8935e-02,\n",
       "          -1.4186e-01, -5.6998e-01, -5.7191e-01,  9.2645e-02, -2.0530e-02,\n",
       "           5.5570e-01, -6.9240e-02,  5.5460e-01,  8.5452e-01,  5.3477e-01,\n",
       "           9.9161e-02, -8.4261e-03,  3.7271e-01,  4.1103e-01, -4.8643e-01,\n",
       "           8.4926e-02,  1.3323e-01, -4.8680e-01,  1.3204e-01, -3.3167e-01,\n",
       "          -2.6373e-01,  8.8352e-02,  9.5563e-02,  1.6208e-01, -3.3610e-02,\n",
       "          -5.8742e-02,  3.3399e-01, -1.0795e-01, -2.7766e-01,  2.1658e-01,\n",
       "          -3.3470e-01,  2.2870e-01, -3.8470e-01, -1.5378e-01, -7.2561e-01,\n",
       "          -4.2176e-01,  2.2724e-02,  2.9976e-01,  9.3408e-02, -4.7802e-01,\n",
       "           7.3651e-02, -4.5770e-01, -2.3165e-01, -3.2072e-01, -2.2605e-01,\n",
       "           4.1884e-01, -1.8974e-01, -1.5361e-01, -2.1398e-01, -3.2636e-01,\n",
       "          -5.3041e-01,  3.0537e-01, -4.9065e-01, -2.4090e-01,  3.2913e-01,\n",
       "          -3.2194e-01,  2.1737e-01,  2.3534e-01, -7.6615e-02, -5.3070e-01,\n",
       "           3.0883e-01,  1.0704e-01, -9.1067e-02,  8.8578e-02,  3.3531e-01,\n",
       "           2.7937e-01,  2.1696e-01,  5.3015e-01, -4.6834e-01, -3.8131e-01,\n",
       "           7.1578e-02,  4.6001e-01, -1.2921e-01, -3.5379e-01, -7.3017e-03,\n",
       "          -1.1683e+00, -3.2771e-01, -5.2199e-01, -1.9734e-01, -4.4616e-01,\n",
       "           1.0628e-02,  4.3719e-01,  3.3842e-02, -1.0367e-01, -1.9964e-01,\n",
       "           4.8676e-01,  6.0754e-01, -3.8314e-01,  4.5103e-01, -4.3573e-01,\n",
       "           8.4715e-02, -2.3954e-01, -6.6433e-01, -2.6841e-01, -3.3141e-02,\n",
       "           3.9675e-01,  3.7643e-01,  2.6132e-01, -3.4787e-01, -3.5854e-01,\n",
       "           3.3782e-01, -3.3030e-01, -2.4625e-01,  1.3215e-01, -1.3091e-02,\n",
       "           1.4185e-01,  2.1752e-01,  1.9407e-02,  2.7133e-01,  5.6749e-01,\n",
       "           2.8202e-02,  1.8957e-02, -1.7300e-01, -3.3849e-01,  4.6411e-01,\n",
       "           4.8827e-01,  3.7523e-01, -2.5776e-01,  4.4371e-01,  3.8938e-01,\n",
       "          -3.8292e-01,  3.4950e-01, -4.6086e-01, -2.6901e-01,  2.1047e-01,\n",
       "           3.5162e-02,  4.1637e-01, -4.5098e-01,  6.9078e-03,  6.8094e-02,\n",
       "          -2.7275e-01, -3.4366e-01,  2.5949e-01,  1.1783e-01, -2.0829e-01,\n",
       "           1.2642e-01, -3.7644e-01,  1.7955e-01, -3.2186e-01, -1.4946e-01,\n",
       "          -7.1921e-02, -2.0017e-01,  4.6318e-02, -2.1325e-01, -1.6507e-01,\n",
       "          -1.5653e-01, -2.3056e-02,  6.1870e-01, -2.1583e-02,  1.5104e-01,\n",
       "           1.2189e-01, -2.0771e-01, -1.6833e-01,  8.4097e-01,  1.2230e-01,\n",
       "          -2.9291e-01,  3.0549e-01, -2.2157e-01,  3.2390e-02,  3.2594e-01,\n",
       "          -4.3915e-01,  3.7179e-02,  1.4310e-01,  2.3976e-01,  2.7307e-01,\n",
       "          -2.5837e-01, -2.3253e-02,  5.2575e-01, -4.4018e-01,  5.3508e-02,\n",
       "          -4.8297e-02,  6.3187e-01, -3.0088e-01, -8.5843e-02, -1.6794e-01,\n",
       "           2.7918e-01, -1.6132e-01, -9.3240e-02,  1.8760e-01, -6.8891e-01,\n",
       "           5.8867e-01, -2.2608e-01, -1.0430e-01,  4.3828e-01, -1.1609e+01,\n",
       "           3.3185e-01,  8.4323e-01, -5.3138e-01,  1.3896e-01, -2.7244e-01,\n",
       "           2.2795e-01, -2.2564e-01,  3.0217e-01, -1.0194e-01, -7.2425e-02,\n",
       "          -4.5716e-01, -1.1619e-01,  1.8957e-01, -1.6579e-01, -7.7557e-02,\n",
       "          -1.4945e-02, -5.9391e-01,  2.1829e-01, -2.1056e-02,  2.4668e-01,\n",
       "          -2.4188e-01, -1.9728e-01,  2.3085e-01, -3.9367e-01, -1.2757e-01,\n",
       "          -2.9842e-02, -8.7676e-02,  3.2412e-01,  4.9388e-01, -3.6276e-02,\n",
       "           2.3771e-01,  1.9911e-01,  3.3781e-01, -2.7726e-01, -5.8092e-01,\n",
       "           1.1093e-01, -8.7556e-02,  1.2332e-01,  9.6892e-02, -4.6864e-01,\n",
       "          -2.3892e-01, -2.3466e-01,  2.8627e-01,  4.9407e-01,  5.7456e-01,\n",
       "           3.2551e-01,  9.3170e-01,  4.4898e-02,  1.4999e-01, -3.0855e-01,\n",
       "          -9.1416e-01,  2.0184e-01, -6.3387e-01,  4.4706e-01,  1.7213e-02,\n",
       "           6.7288e-01, -2.8100e-01,  5.3293e-01, -4.4858e-01, -2.4356e-02,\n",
       "           4.9962e-02,  6.0832e-02,  5.1910e-02, -5.8280e-01, -1.2768e-01,\n",
       "           3.5761e-01,  2.8770e-02,  1.1484e-01, -3.9105e-01,  7.6919e-02,\n",
       "          -1.8234e-01,  3.3540e-01,  1.6100e-01,  3.8859e-01,  2.5134e-01,\n",
       "          -4.2611e-02, -1.5988e-01,  2.1140e+00, -5.3707e-01,  2.1415e-01,\n",
       "          -1.0793e-01,  2.3151e-01, -3.6280e-01, -9.0239e-01,  7.1985e-02,\n",
       "          -2.2859e-01, -2.2749e-01,  4.2394e-01,  3.4885e-01, -2.7646e-01,\n",
       "          -1.3131e-01, -1.2671e-01, -2.2864e-02, -9.7145e-02,  8.2701e-01,\n",
       "          -1.8999e-01, -1.8237e-01, -1.7885e-01,  2.3422e-01, -8.5222e-01,\n",
       "          -2.3404e-01, -2.8530e-02, -2.4553e-01,  1.0123e-01, -1.5895e-01,\n",
       "           7.3166e-02,  7.3475e-02, -1.6989e-02,  3.2925e-01, -3.0363e-01,\n",
       "           3.5753e-02, -5.7775e-02,  8.0447e-02, -1.2671e-01, -5.8670e-01,\n",
       "           3.0562e-01,  4.6267e-01, -2.7374e-01, -5.8811e-01, -2.1015e-01,\n",
       "          -3.9689e-02,  1.3719e-01,  1.3683e-01, -4.4215e-01,  4.3428e-01,\n",
       "          -8.0134e-01, -5.4971e-02,  3.6436e-01,  1.3190e-01,  4.2940e-01,\n",
       "           1.2198e-01, -5.0292e-01, -7.2441e-02, -2.6340e-01, -3.8471e-02,\n",
       "          -6.7604e-01, -4.4493e-01,  3.3042e-01, -5.1753e-01,  1.3454e-01,\n",
       "           2.5266e-01, -1.1031e-02, -3.6403e-01,  3.1486e-01,  1.0523e-01,\n",
       "           1.2728e-01,  4.0824e-01, -6.0782e-02,  1.1628e-01,  2.2110e-02,\n",
       "          -2.4945e-02, -2.5258e-01, -2.1018e-01,  9.6722e-02, -4.9583e-01,\n",
       "          -3.1264e-01,  3.5294e-02, -1.1356e-01, -3.3498e-01,  3.5926e-01,\n",
       "           5.2842e-01, -9.9262e-02, -3.6157e-01,  6.7794e-01,  1.8527e-01,\n",
       "          -7.8573e-02, -2.1666e-01, -4.4501e-01, -1.4649e-01,  2.4374e-01,\n",
       "           3.2116e-01, -3.8388e-03, -4.5512e-01, -1.9092e-01, -3.8693e-01,\n",
       "          -4.3243e-01,  7.7709e-01, -4.1281e-01, -4.1133e-01,  2.4360e-01,\n",
       "           4.6751e-01, -7.9611e-02,  1.4519e-01,  1.5992e-01,  2.7603e-02,\n",
       "          -2.1619e-01, -1.5785e-01,  4.0646e-01, -9.9734e-02,  2.0570e-01,\n",
       "           3.2170e-01,  1.8707e-01, -1.6333e-01,  2.3132e-02, -1.3422e-01,\n",
       "          -4.1734e-01,  1.5040e-01, -1.3064e-01, -1.6464e-01,  2.7439e-02,\n",
       "          -3.7394e-02, -9.8458e-02,  2.0955e-01,  1.1083e-02, -1.6035e-01,\n",
       "           4.5640e-01,  7.6676e-01, -5.1684e-02, -4.6713e-02,  5.3290e-01,\n",
       "           3.3603e-01, -7.4324e-02,  2.6426e-01,  3.2087e-01,  5.5218e-02,\n",
       "           5.4859e-01, -1.5440e-01, -1.3511e-01, -6.3772e-02,  4.1494e-01,\n",
       "           2.6000e-01, -4.2989e-01, -3.8409e-01, -1.7394e-01,  4.5164e-01,\n",
       "          -2.1264e-01,  4.4061e-01, -1.2257e-01,  2.9824e-01,  2.6655e-01,\n",
       "          -7.1614e-01, -1.0508e-01,  2.6590e-02, -6.8347e-01, -2.7572e-01,\n",
       "           6.6241e-03, -4.3288e-01, -1.5869e-01, -8.1605e-02,  4.2886e-01,\n",
       "           3.8285e-01, -9.0564e-01,  2.6764e-01,  6.7166e-01, -3.7137e-01,\n",
       "           3.9517e-01, -1.2374e-01,  2.9050e-01,  7.8764e-01, -7.1568e-02,\n",
       "          -5.4920e-02,  5.4519e-02, -3.1557e-01, -5.7119e-01,  1.0121e-01,\n",
       "           6.4058e-01,  1.6697e-02, -2.9841e-02, -3.9605e-01, -3.3039e-01,\n",
       "          -3.3471e-01, -3.7206e-01, -1.7797e-01,  4.0217e-02,  4.4390e-01,\n",
       "          -3.0920e-01,  1.5030e-01, -8.7314e-02,  7.7903e-02, -1.7131e-01,\n",
       "           3.6589e-01,  4.7877e-01,  4.1254e-01,  1.9297e-01, -4.1855e-01,\n",
       "          -2.8506e-02,  2.6738e-01, -3.6805e-01,  8.4016e-01, -1.9787e-01,\n",
       "          -5.5973e-01, -5.7310e-02,  8.3423e-02, -3.2776e-01,  2.0985e-01,\n",
       "          -5.2006e-01,  4.5335e-02, -1.9439e-01, -9.3013e-02,  5.5208e-01,\n",
       "           6.7503e-02,  3.1376e-01, -8.2895e-01,  4.8301e-02,  5.0351e-01,\n",
       "          -4.9560e-02,  2.4715e-01,  1.0509e-01,  1.3685e-01, -8.6074e-02,\n",
       "          -1.3845e-01, -4.5627e-01, -1.6032e-01,  3.4061e-01, -9.2519e-02,\n",
       "          -4.3411e-01, -2.8077e-01, -6.5692e-02, -1.4789e-02,  7.0380e-01,\n",
       "           7.6870e-01, -1.6223e-01, -5.8238e-01, -2.2804e-01,  7.3916e-02,\n",
       "           2.2059e-01,  4.5209e-02, -5.5407e-02, -3.5007e-01, -2.1852e-02,\n",
       "          -3.8795e-01, -3.3181e-01,  9.6164e-02,  4.3261e-01, -6.7885e-02,\n",
       "           4.7760e-01, -4.6733e-02, -8.1343e-02,  4.7360e-01,  3.7053e-01,\n",
       "           7.3408e-03,  8.0001e-01,  5.7293e-02, -3.8322e-01, -2.2769e-01,\n",
       "           1.8730e-01, -2.8557e-02,  3.3610e-01, -7.1255e-01,  5.3459e-01,\n",
       "           1.2295e-01,  5.0004e-01,  4.8792e-01]], device='mps:0'),\n",
       " tensor([[-2.5187e-01, -4.0054e-01, -5.8007e-01, -8.3055e-01, -1.2868e-01,\n",
       "          -2.8804e-02, -7.1917e-01,  6.6701e-01,  1.2709e-01,  5.0855e-01,\n",
       "           1.2570e-01,  5.1916e-01,  5.1676e-02, -1.1521e-01, -1.6481e-01,\n",
       "          -4.9820e-01,  5.6222e-02,  3.7338e-01,  6.7187e-01, -2.4111e-02,\n",
       "           2.2844e-01,  5.7979e-01, -2.6323e-01, -1.2386e-02, -8.3013e-01,\n",
       "           2.9613e-01, -1.1576e-01,  2.2409e-01,  1.7718e-01, -6.3092e-01,\n",
       "           4.0999e-02,  4.7417e-01,  2.3068e-02,  1.3497e+00, -5.6080e-01,\n",
       "          -1.3230e-01,  8.2213e-02, -3.9453e-01, -2.3010e-01,  3.7283e-02,\n",
       "           6.1164e-01, -4.1315e-01,  1.3635e-01,  4.0254e-01, -6.1085e-02,\n",
       "           8.9076e-02,  1.3899e-01, -3.7744e-01,  2.5892e-01, -8.3961e-01,\n",
       "          -4.8318e-02, -2.0669e-01,  4.9144e-01,  6.2615e-01, -6.3176e-01,\n",
       "          -2.2975e-01,  2.7419e-01, -3.7716e-02, -3.3020e-02,  1.5349e-01,\n",
       "           9.7980e-02,  3.8582e-01, -2.9995e-01,  2.3257e-01,  3.3441e-01,\n",
       "           2.4458e-01, -3.0392e-01, -4.8889e-01, -5.4078e-01, -5.0111e-02,\n",
       "           9.1859e-03,  2.2490e-02,  1.4846e-01, -2.3272e-01,  4.1973e-01,\n",
       "          -7.1214e-01, -6.4354e-02, -8.9125e-02, -2.5973e-01,  2.9246e-01,\n",
       "          -4.5549e-02,  1.9719e-01,  3.1010e-01,  1.0268e-01,  1.2457e-01,\n",
       "           4.3419e-02,  2.7215e-01,  5.2932e-04,  4.7373e-02,  3.7533e-01,\n",
       "          -1.3375e-01, -8.1722e-02, -2.8313e-01, -3.5814e-01,  3.1229e-01,\n",
       "           4.0598e-02,  2.9907e-01, -1.9557e-01,  1.8579e-01, -3.2474e-01,\n",
       "           1.5588e-01,  1.7768e-01, -1.8083e-01,  5.7286e-02, -3.5172e-01,\n",
       "          -4.5475e-01,  2.3589e-02, -2.8569e-01, -4.8285e-01,  2.2697e-01,\n",
       "          -1.4211e-01, -3.0205e-01, -4.0922e-02, -3.8935e-01, -3.1524e-02,\n",
       "          -7.5164e-02,  2.6013e-01, -1.3230e-01, -3.4806e-01, -8.3033e-02,\n",
       "           2.0734e-01, -2.2808e-01,  1.6077e-01,  2.6882e-01,  1.2232e-01,\n",
       "          -9.6705e-02,  6.5251e-01,  1.8950e-02,  3.1674e-01,  2.5763e-01,\n",
       "           1.1699e-01, -6.5768e-02, -1.9745e-01, -2.3154e-01,  7.5988e-02,\n",
       "          -7.6941e-02,  1.0687e-01, -1.8182e-01, -2.2790e-01, -1.8508e-01,\n",
       "           3.1771e-01, -3.3097e-01, -6.0893e-01, -4.9260e-01, -1.6098e-01,\n",
       "          -1.6326e-01, -2.4804e-02, -8.2473e-03, -3.3693e-01, -3.1956e-01,\n",
       "          -6.7443e-01, -3.6717e-01,  8.1041e-01, -1.6831e-02, -3.3356e-01,\n",
       "          -1.6499e-01,  2.3208e-01,  3.1666e-01, -4.5959e-02,  3.7161e-01,\n",
       "           7.7250e-01, -4.7670e-02,  1.6786e-01, -5.1827e-01, -3.8195e-01,\n",
       "           4.8335e-02,  2.6658e-01, -2.7054e-01,  6.1879e-01,  4.8352e-02,\n",
       "           7.5627e-02, -6.6170e-02,  2.7430e-01, -3.2704e-01, -1.2324e-02,\n",
       "          -5.7573e-01,  3.7750e-01,  2.9539e-01, -3.9892e-01, -6.6567e-02,\n",
       "          -2.0834e-02, -1.2060e+00,  9.6964e-02, -8.3040e-01,  3.1236e-01,\n",
       "          -2.3110e-01,  1.7963e-01,  2.3389e-01,  2.9545e-01, -6.5320e-02,\n",
       "           2.5404e-01,  2.8725e-01,  2.5687e-01,  3.3609e-01, -3.4924e-02,\n",
       "           2.8019e-01,  2.3924e-01,  5.0790e-01,  4.1073e-01,  1.4769e-01,\n",
       "          -3.5440e-01,  2.9683e-01,  3.5781e-01,  9.0598e-02,  1.6954e-01,\n",
       "           9.9825e-02, -2.1636e-01,  2.1091e-01,  1.7501e-01, -1.6618e-01,\n",
       "          -2.2406e-01, -3.1100e-02, -4.6818e-01, -5.6655e-02, -4.8467e-01,\n",
       "           3.2557e-01, -5.1983e-01, -1.1912e-01,  3.4831e-01,  2.4340e-01,\n",
       "          -1.4896e-02,  1.8319e-01,  1.7196e-01, -2.4338e-01,  2.1735e-01,\n",
       "           3.7449e-01, -3.4618e-01, -3.9804e-02, -3.2197e-01, -2.9367e-01,\n",
       "           2.4357e-01, -5.6244e-01,  6.6635e-01,  4.0574e-01,  6.0165e-02,\n",
       "           1.5119e-01, -2.7920e-01, -5.1638e-02, -5.7408e-01,  3.5471e-01,\n",
       "          -3.2130e-01, -3.4955e-01, -5.4801e-01,  6.4487e-02,  4.2773e-01,\n",
       "           2.7565e-01, -6.4915e-02,  7.7868e-01,  4.2408e-01,  4.2813e-01,\n",
       "          -4.3003e-01, -1.5896e-02,  4.0377e-01, -4.1405e-02, -4.6059e-01,\n",
       "           6.7116e-02, -1.4962e-01, -5.1643e-01,  2.8000e-01, -2.6598e-01,\n",
       "          -1.2047e-01, -1.4913e-01,  3.3682e-01,  1.5365e-01, -2.1964e-01,\n",
       "          -3.8461e-01,  3.9587e-01, -1.4349e-01, -4.1723e-01,  4.7123e-01,\n",
       "          -3.5025e-01,  1.5209e-01,  1.0694e-01, -3.8441e-01, -6.2072e-01,\n",
       "          -3.2235e-01,  2.3090e-01,  1.5585e-01,  3.4799e-02, -6.6306e-01,\n",
       "          -4.0571e-01,  3.1197e-01,  2.4601e-01,  2.9976e-01,  2.9840e-01,\n",
       "           5.7427e-01, -3.0944e-01, -1.6702e-01,  1.4982e-01, -4.1377e-01,\n",
       "          -1.5704e-01, -3.3228e-02, -1.9995e-01,  2.9569e-01,  2.4768e-01,\n",
       "           1.1600e-01,  2.0367e-01,  4.3482e-01, -2.5272e-01, -3.8673e-01,\n",
       "          -7.1543e-02,  2.0161e-01, -9.3629e-02, -5.8306e-02,  2.3951e-01,\n",
       "          -6.3095e-02,  5.9650e-01,  7.2143e-01,  7.7439e-02, -6.4006e-01,\n",
       "           1.7696e-01,  5.2502e-01, -2.7253e-01, -4.0343e-02,  9.6365e-02,\n",
       "          -1.0344e+00, -2.1331e-01, -1.2883e-01, -2.0181e-01, -6.5352e-01,\n",
       "           8.8115e-02,  3.4863e-01,  4.6833e-03, -1.7864e-01, -3.0947e-01,\n",
       "           1.0448e-01,  4.8724e-02, -4.8980e-02,  4.3534e-01, -2.8901e-01,\n",
       "          -3.4797e-01, -2.3438e-01, -2.1532e-01, -5.4201e-01, -3.9968e-01,\n",
       "           3.5914e-01,  2.8129e-01,  1.1986e-01, -2.6486e-01, -5.1832e-01,\n",
       "           4.5577e-01, -6.8169e-01, -1.4632e-01,  2.4840e-01, -4.7702e-01,\n",
       "           1.4650e-02,  4.8990e-01,  1.4999e-01,  1.6336e-01,  6.5258e-02,\n",
       "           7.8557e-02, -3.3903e-01, -5.0030e-01, -1.5000e-01,  6.2323e-01,\n",
       "           3.2283e-01,  3.3506e-01, -1.7236e-01,  3.9096e-01,  4.5167e-01,\n",
       "          -2.9582e-01,  4.1670e-01,  1.0930e-02,  7.7439e-02, -6.0414e-02,\n",
       "           1.9048e-01,  3.4370e-01, -3.6842e-01,  2.0145e-01,  1.7929e-01,\n",
       "          -3.1634e-01, -3.2149e-01,  1.5691e-02,  2.0642e-01, -1.8580e-01,\n",
       "          -3.0875e-01, -2.0361e-01,  9.9698e-02, -3.4856e-01, -4.3104e-01,\n",
       "           2.9161e-01, -3.6543e-02,  7.0495e-02, -4.0357e-01, -4.3462e-01,\n",
       "          -2.0181e-01, -2.6055e-01,  6.8681e-01,  2.1064e-01, -1.9621e-01,\n",
       "           1.8720e-01, -1.0907e-01, -3.6330e-01,  4.7847e-01,  2.0162e-02,\n",
       "          -4.8754e-01,  4.5482e-01, -3.4822e-01,  1.0627e-01,  5.5756e-01,\n",
       "          -1.6769e-01, -1.0840e-01,  1.1008e-01,  1.0820e-01,  2.9026e-01,\n",
       "          -3.9884e-01,  1.5516e-01, -6.3476e-02, -3.1885e-01,  3.1820e-01,\n",
       "           2.6411e-01,  2.2781e-01, -1.7137e-01, -3.3426e-02,  3.5677e-02,\n",
       "           2.8782e-01, -6.0853e-02, -3.9888e-01, -4.7347e-01, -4.6132e-01,\n",
       "           1.2832e-01,  1.8927e-01,  5.6207e-01,  3.4144e-01, -1.1394e+01,\n",
       "           1.2938e-01,  7.8031e-01, -4.0713e-01, -8.2819e-02, -5.0320e-02,\n",
       "           1.0976e-02, -1.3708e-01,  4.2955e-01, -2.7734e-01, -4.4105e-01,\n",
       "          -3.5727e-01, -1.3308e-01,  1.5454e-01, -3.2189e-01, -2.6945e-01,\n",
       "          -1.4686e-02, -7.0935e-01,  4.0963e-01, -3.0286e-01,  5.0298e-01,\n",
       "          -2.3511e-01, -8.3798e-02,  1.5816e-01, -3.5851e-01, -1.3439e-02,\n",
       "           5.9766e-02, -1.6033e-01,  1.4651e-01,  4.2486e-01, -5.2072e-02,\n",
       "           3.3244e-02, -4.1258e-03,  4.8461e-02, -2.5723e-01, -5.9999e-01,\n",
       "           1.9296e-02, -1.4141e-01,  3.3422e-01, -4.4133e-02, -7.1265e-01,\n",
       "          -2.2533e-01, -2.7418e-01,  3.1407e-01,  2.7021e-01,  9.0598e-01,\n",
       "           2.2955e-01,  1.0686e+00,  1.5022e-01,  1.7748e-01,  2.6230e-02,\n",
       "          -7.8764e-01, -3.3269e-02, -4.1907e-01,  2.3280e-02,  1.0114e-01,\n",
       "           3.6576e-01, -7.9367e-01,  2.5251e-01, -3.5521e-01,  3.6487e-02,\n",
       "           4.9094e-01,  3.1768e-02,  2.7700e-01, -3.2298e-02, -7.8123e-02,\n",
       "           1.2089e-01, -2.8544e-02,  1.1098e-01, -2.2583e-01, -2.4084e-01,\n",
       "          -1.0177e-01,  5.9888e-01,  3.9592e-01, -7.2211e-01,  3.6112e-01,\n",
       "           4.9258e-02, -1.8622e-01,  1.4556e+00, -4.8573e-02,  1.8361e-02,\n",
       "           1.9800e-03,  2.9604e-01, -5.0236e-01, -7.3516e-01,  3.6511e-01,\n",
       "          -1.0211e-01,  5.6427e-02,  1.0127e-01,  4.8148e-01, -9.7481e-02,\n",
       "          -4.7940e-01, -1.8593e-03,  1.0556e-01, -1.3380e-01,  5.4710e-01,\n",
       "          -7.2824e-01, -1.6963e-01, -1.1223e-01,  3.1723e-01, -8.4054e-01,\n",
       "          -6.1126e-02, -7.0210e-01, -3.9328e-01,  5.0586e-01, -2.5896e-01,\n",
       "           1.5852e-01, -2.7443e-02, -4.3439e-02,  5.5574e-01, -2.1489e-01,\n",
       "           7.1555e-02,  2.2813e-01,  8.4995e-02, -1.6733e-01,  3.9349e-02,\n",
       "           4.1630e-01,  4.9571e-01, -4.2740e-01, -6.3911e-01, -7.6901e-03,\n",
       "           1.7180e-01,  1.4933e-01,  3.4019e-01, -5.3463e-01,  3.6431e-01,\n",
       "          -7.9162e-01, -1.8769e-01,  7.2949e-02, -4.3273e-02,  3.6175e-01,\n",
       "           2.3925e-01,  7.2038e-02,  2.2256e-01, -7.7756e-01, -6.4498e-01,\n",
       "          -6.5932e-01, -2.1445e-01,  2.6143e-01, -5.4236e-01, -8.3654e-02,\n",
       "          -1.1033e-01, -4.8837e-03, -5.0292e-02, -6.6102e-02,  1.1939e-02,\n",
       "           1.3632e-01,  2.3587e-01, -1.3023e-01, -1.3178e-01, -9.2464e-02,\n",
       "          -5.5566e-01, -1.1753e-01, -9.9591e-02, -1.2505e-01, -5.6783e-01,\n",
       "          -1.3797e-01,  3.3075e-02, -2.5373e-01, -6.8124e-02,  3.5608e-01,\n",
       "          -2.9832e-02, -2.7856e-02, -4.8850e-01,  7.2846e-01,  4.0279e-01,\n",
       "           5.2088e-02, -2.8298e-01, -5.5189e-01, -1.0567e-01,  2.9592e-01,\n",
       "           1.9555e-01, -6.7007e-02, -1.7709e-01, -2.4145e-01, -4.2522e-01,\n",
       "           1.4302e-01,  7.1147e-01, -6.6296e-02, -6.8115e-01,  2.6621e-01,\n",
       "           5.8223e-01, -1.2882e-01, -2.4432e-01,  2.0983e-02, -9.3284e-02,\n",
       "          -3.1851e-01, -6.0317e-01,  7.8821e-01, -1.1246e-01, -5.8638e-02,\n",
       "           2.2427e-01,  2.0359e-01, -9.2588e-03,  4.0209e-01, -1.5770e-01,\n",
       "          -3.5434e-01, -1.6477e-03, -7.3600e-01,  4.3257e-01,  6.1666e-02,\n",
       "          -3.1489e-01,  7.6082e-01,  2.7260e-01, -6.0593e-02, -4.5779e-02,\n",
       "           2.2863e-01,  7.9941e-01, -2.4636e-01,  6.4653e-02,  5.3779e-01,\n",
       "           8.8110e-02, -1.1229e-01,  3.9580e-02,  4.1665e-02,  4.5410e-02,\n",
       "           3.7012e-01, -3.2145e-02, -4.1564e-01, -6.1194e-02,  3.8924e-01,\n",
       "           3.9295e-01, -5.4610e-01, -7.3487e-01,  2.2333e-02,  6.8551e-01,\n",
       "          -2.9054e-01,  2.3654e-01, -3.2708e-02,  3.0880e-01,  1.2001e-01,\n",
       "          -9.2160e-01, -1.5016e-01, -1.0056e-01,  2.8229e-01, -2.5482e-02,\n",
       "          -1.6139e-01, -1.4727e-01, -2.3686e-01, -1.6208e-01,  6.8308e-01,\n",
       "           2.1991e-01, -6.9160e-01,  3.4058e-02,  5.7281e-01, -6.0551e-01,\n",
       "           4.6052e-01, -2.4577e-01, -3.4302e-02,  3.6435e-01,  1.3117e-01,\n",
       "           4.9067e-02,  1.7856e-01, -4.2138e-01, -4.0598e-01,  1.7824e-01,\n",
       "           9.1672e-01, -7.6613e-02,  1.1166e-01, -3.9384e-01, -1.7182e-01,\n",
       "          -7.1855e-02, -3.1913e-01, -3.1286e-01,  3.7185e-01,  2.0529e-01,\n",
       "          -3.2552e-02, -1.3271e-01, -2.2992e-01,  3.0816e-01,  1.1821e-01,\n",
       "           3.2536e-01,  5.8187e-01,  6.4368e-01, -1.0077e-02, -2.4936e-01,\n",
       "          -6.1235e-02,  1.4568e-01, -1.9524e-01,  5.2704e-01, -2.0627e-01,\n",
       "          -3.3453e-01, -3.8069e-03,  3.0134e-01, -6.1681e-01, -5.8725e-02,\n",
       "          -4.5426e-01,  4.2361e-01, -1.6955e-01,  3.7868e-02,  2.8833e-01,\n",
       "           4.0522e-02,  5.7925e-02, -2.2651e-01,  5.5807e-01,  4.4511e-01,\n",
       "           1.6609e-01,  4.5662e-01,  1.6104e-01,  4.5272e-02, -5.1597e-01,\n",
       "           2.1978e-01, -1.0213e+00, -9.4260e-02, -4.6640e-02, -1.5401e-01,\n",
       "          -4.4224e-01, -2.8126e-02,  1.6814e-01,  1.8538e-01,  6.9597e-01,\n",
       "           7.2967e-01, -3.0871e-01, -4.5290e-01, -1.8510e-01,  1.5304e-01,\n",
       "           2.7705e-01, -1.9517e-01,  3.5971e-01,  9.6103e-02,  2.4432e-01,\n",
       "          -5.0967e-01, -4.3165e-01, -4.2252e-02,  3.1411e-01, -2.0171e-01,\n",
       "           5.7118e-01, -2.0164e-01, -1.3863e-01,  9.0832e-02,  1.3770e-01,\n",
       "           1.6639e-01,  7.9664e-01,  4.3416e-01, -6.3425e-01,  8.7469e-02,\n",
       "          -1.3980e-02, -2.9988e-01,  2.5983e-02, -4.3577e-01,  2.8858e-01,\n",
       "           1.2984e-01,  4.9280e-01,  6.7504e-01]], device='mps:0'),\n",
       " tensor([[-0.1932, -0.4948, -0.6750,  ...,  0.2888,  0.2386,  0.7024],\n",
       "         [-0.0583, -0.6757, -0.8037,  ...,  0.1694,  0.7852,  0.5016]],\n",
       "        device='mps:0'),\n",
       " tensor([[-1.6720e-01, -5.2486e-01, -4.5395e-01, -6.3577e-01, -4.3561e-01,\n",
       "          -1.5533e-01, -5.8268e-01,  2.5578e-01, -1.8350e-01,  5.0250e-01,\n",
       "           1.6126e-01,  1.3783e-01, -2.1897e-01, -4.6204e-01, -6.9128e-01,\n",
       "          -2.0657e-01, -2.5177e-01,  1.9475e-01,  4.1034e-01,  5.6174e-01,\n",
       "           2.8700e-01,  1.6722e-01, -3.7843e-01,  2.1455e-01, -5.1055e-01,\n",
       "           1.5812e-01, -2.8156e-01, -6.3669e-02,  4.5597e-01,  6.8516e-02,\n",
       "          -2.1018e-02,  4.7623e-01, -2.8390e-02,  1.0709e+00, -7.7172e-02,\n",
       "          -3.0626e-01,  1.0760e-01, -4.1569e-01,  8.9012e-01, -5.6131e-01,\n",
       "           6.8910e-01,  6.2264e-01, -1.5577e-01,  1.5282e-02,  2.3088e-01,\n",
       "           1.5974e-01,  5.1535e-02, -5.3995e-01, -5.6558e-01, -1.5427e-01,\n",
       "           1.6123e-01,  1.9663e-01,  5.4635e-01,  6.3307e-02, -7.5459e-01,\n",
       "          -2.6525e-01, -1.3650e-01,  3.0473e-01,  1.7296e-02,  3.5135e-01,\n",
       "          -7.4312e-02,  5.0134e-01,  5.1710e-02, -6.9956e-02,  1.0589e-01,\n",
       "           2.1145e-01, -6.0512e-01, -2.5610e-01, -5.7211e-01,  2.0313e-01,\n",
       "           2.0375e-01,  5.2893e-01,  1.1646e-01, -4.0534e-01,  1.4546e-01,\n",
       "           7.6205e-02,  9.5666e-02, -2.8535e-02, -5.2510e-01, -1.6272e-01,\n",
       "           2.0181e-01,  3.7399e-01,  2.1510e-01, -2.0973e-01,  1.4960e-01,\n",
       "           2.5140e-01,  5.5226e-01, -5.2582e-02, -1.1641e-01, -4.8629e-02,\n",
       "           2.9244e-01, -2.7514e-01, -7.8764e-01, -3.1798e-01, -1.0430e-01,\n",
       "          -3.4914e-01,  7.9176e-02, -4.2834e-01, -2.7512e-01, -3.3882e-01,\n",
       "           1.5050e-01,  1.0200e-01,  2.5842e-01,  7.5482e-02, -7.5174e-01,\n",
       "          -4.1705e-01,  4.6029e-01,  8.0138e-02, -4.2192e-01, -3.1245e-01,\n",
       "          -1.3741e-01, -2.1173e-01,  9.7620e-02, -4.0036e-01, -8.6561e-02,\n",
       "          -1.3006e-01,  2.5022e-01, -1.4378e-01,  5.4056e-02,  4.9777e-01,\n",
       "           2.5708e-01, -5.8312e-01,  1.0353e-01,  2.2601e-01,  4.0089e-02,\n",
       "           3.0807e-02,  6.0830e-01, -5.3486e-02,  4.9134e-01,  1.2285e-01,\n",
       "           2.6257e-01,  1.5138e-01, -6.2596e-01, -5.3560e-01, -2.0734e-01,\n",
       "           1.3350e-01, -3.3082e-01, -8.4107e-01,  3.7865e-01,  3.1799e-02,\n",
       "           5.1144e-02, -9.6587e-02, -6.7862e-01, -3.0496e-01, -3.9669e-01,\n",
       "          -2.7230e-01,  1.4708e-01, -3.5011e-01, -1.3439e-01, -2.7277e-01,\n",
       "          -6.7571e-01, -8.5650e-01, -1.1642e-03, -1.3369e-01,  4.2598e-01,\n",
       "          -5.3225e-01,  4.3086e-01, -2.3121e-01, -5.0388e-01,  4.1057e-01,\n",
       "           5.9947e-01, -1.7783e-01,  2.8322e-01, -5.0157e-01,  3.3619e-02,\n",
       "           3.4604e-01, -3.8918e-02, -5.9573e-02, -3.8183e-02,  1.3359e-01,\n",
       "           3.4058e-01,  2.0341e-01,  1.0286e-01, -2.3079e-01,  3.2100e-01,\n",
       "          -3.7378e-01, -5.5154e-03, -2.6355e-01,  1.6162e-01,  3.2661e-02,\n",
       "           2.9978e-01, -1.6971e+00, -2.5911e-01,  2.0065e-02, -1.8571e-01,\n",
       "          -1.9740e-01,  3.3240e-01,  1.4453e-01,  2.5148e-01, -2.5533e-01,\n",
       "           2.1988e-01,  9.8764e-02, -1.4769e-01, -3.0128e-01,  5.8231e-02,\n",
       "           3.8842e-01,  6.9484e-01,  3.7541e-01,  7.7099e-02, -3.0723e-01,\n",
       "          -7.5503e-01,  1.8407e-01,  3.3852e-01, -1.1523e-01,  5.9383e-01,\n",
       "           3.0489e-02, -8.6229e-01,  1.9019e-01,  2.9946e-02,  2.6756e-01,\n",
       "           5.7319e-02, -2.9941e-01, -4.1668e-01, -3.5307e-01, -3.9315e-01,\n",
       "           2.3395e-01, -1.8355e-01, -3.1845e-01,  4.8844e-01,  8.1310e-02,\n",
       "          -3.0776e-02,  4.2831e-01,  6.0508e-02, -4.4367e-01,  1.9303e-01,\n",
       "           4.4887e-01,  2.6861e-02,  1.1130e-01, -2.7214e-01, -4.0374e-02,\n",
       "          -6.2142e-02, -3.2118e-01,  3.6896e-01,  7.7114e-01, -7.4506e-02,\n",
       "           3.7060e-01, -3.0406e-01,  5.1180e-01, -1.9654e-01,  5.8690e-02,\n",
       "          -1.1981e-01,  3.4078e-02, -2.6240e-01,  2.4533e-01, -1.1898e-01,\n",
       "           9.4908e-02,  8.5883e-02,  7.2222e-01,  3.7433e-01,  7.9341e-02,\n",
       "           3.1509e-01, -4.4737e-03,  1.1637e-02,  5.8059e-01, -3.3724e-01,\n",
       "           3.9433e-01, -1.0171e-01, -3.7982e-01,  3.7113e-01, -4.7675e-01,\n",
       "           3.1991e-01, -1.8581e-01,  1.0010e-01,  2.5551e-01,  3.0602e-02,\n",
       "          -2.9218e-01,  6.1325e-01,  2.2058e-01, -4.4818e-01,  5.2200e-01,\n",
       "          -2.5795e-01,  7.1632e-01, -2.0896e-01, -1.6155e-01, -7.6790e-02,\n",
       "          -9.9737e-01, -4.3677e-01,  1.9994e-01,  1.8271e-01, -6.8430e-01,\n",
       "          -4.9875e-01,  2.8702e-01,  1.1765e-01,  1.0125e-01,  4.2891e-01,\n",
       "           3.8920e-01,  9.2143e-02,  1.0493e-01, -4.8046e-01, -3.6847e-01,\n",
       "          -2.1420e-01,  9.0059e-02, -8.5629e-04, -7.5374e-02,  2.0220e-01,\n",
       "           5.9414e-01,  9.6554e-02,  1.8048e-01,  3.6746e-02, -2.8779e-01,\n",
       "          -1.0518e-01,  1.6372e-01,  4.2762e-01, -5.4888e-01, -2.0137e-01,\n",
       "           6.8173e-02,  4.3129e-02,  4.5140e-01, -3.3691e-01, -3.2667e-01,\n",
       "          -4.2742e-02,  4.8308e-01, -1.2353e-01, -7.1202e-02,  3.3410e-02,\n",
       "          -7.8725e-01,  2.7660e-01, -4.5062e-02, -2.6552e-02, -3.0430e-01,\n",
       "           2.3290e-01,  3.4297e-02, -2.8492e-01, -1.1149e-01, -2.7156e-01,\n",
       "           5.8752e-01,  5.5235e-01,  3.3698e-01, -5.6671e-01, -3.5461e-01,\n",
       "           4.4066e-02, -7.0730e-01,  9.8556e-02, -4.9156e-01,  2.2488e-01,\n",
       "           1.4708e-01,  3.3198e-01, -1.1607e-01,  1.3552e-01, -1.6474e-01,\n",
       "           4.9723e-01, -6.3908e-01,  1.2761e-01,  5.5720e-01, -5.3885e-01,\n",
       "          -2.3301e-01,  4.1839e-01, -5.5797e-01,  1.2634e-01,  4.1453e-01,\n",
       "           3.9631e-01,  4.6106e-02, -4.1602e-01, -7.3707e-02,  9.1249e-01,\n",
       "           5.2639e-01,  1.6638e-01, -6.9733e-01,  9.7526e-02,  3.4043e-01,\n",
       "          -2.0714e-01,  9.5206e-02, -3.2600e-01, -4.7544e-01, -2.1026e-01,\n",
       "          -8.4302e-02,  3.9153e-01, -2.6238e-01,  3.5886e-02,  2.8572e-01,\n",
       "           3.6819e-01, -4.7808e-01, -2.3588e-01,  3.2227e-01,  6.0455e-01,\n",
       "          -1.8222e-01, -3.6901e-01,  1.6512e-01, -5.0214e-01,  2.6335e-01,\n",
       "          -1.4632e-01, -3.4035e-01,  2.7259e-01, -1.2606e-01, -3.7457e-01,\n",
       "          -1.8250e-01, -4.7967e-01,  2.5449e-01,  9.2148e-02, -1.2652e-01,\n",
       "           2.8609e-01, -1.1966e-01, -3.8868e-01,  8.2449e-01, -5.2187e-01,\n",
       "           4.2853e-03,  5.3276e-01, -2.7521e-01, -1.8249e-01,  2.4158e-01,\n",
       "          -2.6392e-01, -1.9683e-02,  1.8527e-02,  9.4766e-01,  1.2345e-01,\n",
       "          -2.7053e-03, -1.1552e-01,  5.1146e-01, -3.8026e-01,  1.5931e-01,\n",
       "          -3.3320e-01,  3.1171e-01, -1.6158e-01,  1.8223e-02, -1.0254e+00,\n",
       "           4.4946e-01, -3.7111e-01, -2.1751e-01, -5.0362e-01, -3.0765e-01,\n",
       "           9.4840e-02, -6.3165e-02,  2.3746e-01,  6.3366e-01, -1.1037e+01,\n",
       "           8.9129e-02,  6.0403e-01, -7.6489e-02,  1.1652e-01,  4.1141e-01,\n",
       "           1.7540e-01, -4.2006e-01,  5.2776e-01, -3.4203e-01,  1.1525e-01,\n",
       "          -4.7982e-01, -1.3280e-01, -8.5294e-02, -3.7948e-01,  3.6349e-01,\n",
       "          -9.6031e-02, -6.4688e-01,  2.4074e-01, -1.3490e-01, -1.7654e-02,\n",
       "          -2.2095e-01,  1.1407e-01,  4.3684e-01, -4.6076e-01, -4.4456e-01,\n",
       "           7.1777e-02, -2.7274e-01,  5.1195e-01,  2.8186e-01, -1.1255e-01,\n",
       "           1.1615e-01,  1.5217e-01,  3.0053e-01, -2.7216e-01, -3.1454e-01,\n",
       "           1.2732e-01, -3.4997e-01, -1.8961e-01, -1.0421e+00, -1.5631e-01,\n",
       "          -2.0656e-01, -4.6750e-01, -1.7710e-02,  6.5233e-01,  7.8743e-01,\n",
       "           1.6358e+00,  9.2174e-01,  3.2668e-01,  1.9802e-01, -2.2250e-01,\n",
       "          -4.6619e-01,  1.1894e-01, -6.7995e-02, -7.0546e-01, -2.4133e-01,\n",
       "           2.8985e-01, -7.5931e-01,  5.3131e-01, -4.7860e-01, -1.3681e-01,\n",
       "           4.2374e-01,  3.9276e-02, -4.2065e-02, -5.9925e-01, -5.3467e-01,\n",
       "          -4.9082e-02,  1.2925e-01,  4.8245e-02, -5.7979e-01,  1.4884e-01,\n",
       "           5.0590e-02,  7.6896e-01, -3.4175e-01, -5.9383e-01,  5.7234e-02,\n",
       "          -1.2763e-01,  3.6795e-02,  1.6636e+00, -1.7915e-01, -4.1931e-02,\n",
       "           1.2354e-01,  4.9556e-02, -6.9680e-01, -4.7087e-01, -4.3059e-01,\n",
       "          -2.7038e-01,  2.5252e-01,  6.6455e-01,  2.0274e-01,  9.0694e-02,\n",
       "          -4.6503e-01, -1.2589e-01,  9.3002e-02,  3.9245e-01,  1.0908e+00,\n",
       "          -5.4949e-01,  2.6147e-01, -7.5939e-02,  2.6322e-01,  1.2335e-01,\n",
       "          -3.6152e-01, -6.2216e-02,  3.9533e-02,  1.6284e-02,  4.5349e-01,\n",
       "           8.6971e-02,  4.2124e-01,  6.5534e-02,  1.2702e-01, -5.0361e-01,\n",
       "          -3.5058e-01, -2.8066e-01, -1.1609e-01, -4.6047e-01, -4.4892e-02,\n",
       "          -2.8268e-02,  1.6753e-01, -1.7086e-01, -2.9519e-01, -2.9979e-01,\n",
       "           2.9427e-01,  2.0283e-01, -6.9070e-02,  9.5939e-03,  6.0460e-01,\n",
       "          -6.3640e-01, -1.8307e-01, -9.4406e-02,  8.6813e-02,  6.9132e-02,\n",
       "           2.1208e-01, -3.8688e-01, -1.1241e-01, -7.3959e-02, -8.4460e-01,\n",
       "          -9.6512e-01, -9.7748e-02,  5.1010e-01, -2.4332e-01,  1.0866e-01,\n",
       "           6.7349e-02,  1.2474e-02, -3.2385e-01,  2.6710e-01, -1.9218e-01,\n",
       "           1.1645e-01,  3.9439e-01, -5.4670e-01,  1.5963e-01, -3.1155e-01,\n",
       "           1.8497e-02, -2.0876e-02, -6.5711e-02, -3.8026e-01, -5.2860e-01,\n",
       "          -7.6356e-01, -6.8179e-02,  5.1232e-02,  5.6135e-01, -3.2886e-02,\n",
       "           1.0488e-01, -1.0398e-01,  7.7290e-02,  7.2752e-01,  8.5206e-01,\n",
       "           2.6106e-01, -3.5137e-01, -4.9530e-01,  5.0728e-02, -2.6647e-02,\n",
       "           5.7946e-02,  2.1021e-01,  7.9023e-02, -2.2641e-01, -1.4403e-01,\n",
       "          -2.6117e-01,  1.1740e+00, -2.3018e-02, -2.5236e-01,  3.1798e-04,\n",
       "           1.2502e+00, -5.8815e-01, -1.4557e-01,  1.7334e-01,  2.6380e-01,\n",
       "          -5.2810e-01, -3.4023e-01,  5.6863e-01, -2.1703e-01,  1.5201e-01,\n",
       "           3.9180e-01,  1.8677e-01,  1.0221e-01, -4.4054e-02,  8.4521e-02,\n",
       "          -1.5903e-01,  3.3655e-01,  3.0166e-02, -1.4937e-01,  7.3811e-01,\n",
       "           1.1911e-01,  8.0888e-02, -1.4277e-01, -2.4375e-01, -2.6556e-01,\n",
       "           1.9428e-01,  1.2395e+00, -4.2394e-01, -5.8109e-01,  8.7338e-01,\n",
       "          -1.4019e-01,  3.5768e-01,  6.0006e-01, -3.9499e-02,  1.3149e-02,\n",
       "           5.7371e-01,  1.1784e-01, -6.0531e-01,  2.3557e-01,  3.0019e-01,\n",
       "           1.5461e-01, -8.7253e-01, -2.0869e-01, -8.1431e-02, -1.2343e-01,\n",
       "          -8.4444e-02,  5.6766e-01,  6.7880e-02,  3.7625e-01,  3.3336e-01,\n",
       "          -8.6361e-01,  2.9748e-01,  3.9963e-02,  8.7689e-01,  3.6637e-02,\n",
       "          -3.2763e-01, -2.5701e-01, -2.0231e-01, -2.1073e-01,  3.1221e-01,\n",
       "           2.4523e-01, -1.3614e+00, -8.3672e-02,  4.5844e-01, -8.1731e-02,\n",
       "           4.3829e-01, -7.9391e-02,  2.9827e-01,  3.0846e-01,  4.5552e-02,\n",
       "          -5.2861e-02,  7.4971e-02, -2.2854e-01, -2.7639e-01,  4.5622e-01,\n",
       "           5.5347e-01, -5.8603e-02, -7.4470e-02,  2.2723e-01, -2.7374e-01,\n",
       "          -3.2137e-01, -2.2843e-01, -1.8679e-01, -2.2410e-01, -1.2894e-01,\n",
       "          -3.6518e-01, -6.9317e-01, -3.1199e-02, -5.0747e-02, -4.5982e-02,\n",
       "           3.4356e-01,  2.6472e-01,  6.5181e-01, -1.2203e-01,  1.5073e-01,\n",
       "          -1.0406e-01,  4.4593e-01, -2.4264e-01,  7.3864e-01,  3.8785e-01,\n",
       "          -1.7432e-01,  2.7267e-01,  1.0448e-01, -2.3479e-01,  1.3287e-01,\n",
       "          -2.8065e-01,  2.5046e-01, -4.0412e-01, -4.0210e-01, -5.9590e-02,\n",
       "          -1.5547e-01, -1.1097e-01, -2.4685e-01,  2.4733e-01,  3.3605e-01,\n",
       "          -8.1116e-02,  9.9147e-01, -2.5425e-01,  6.5276e-01, -3.0278e-01,\n",
       "          -9.8080e-02, -6.8560e-01,  3.0100e-01,  1.5904e-01, -8.1888e-02,\n",
       "          -3.9633e-01, -2.5967e-01, -3.2959e-01, -4.0276e-01,  6.6570e-01,\n",
       "           7.1379e-01, -5.7936e-01, -7.3617e-01, -2.6704e-01, -1.7923e-01,\n",
       "          -2.0926e-02,  5.7887e-02,  3.7235e-02, -2.4464e-01,  1.1875e-01,\n",
       "          -8.1963e-01,  1.9554e-01,  2.5091e-02,  3.7079e-01, -2.2153e-01,\n",
       "           7.4679e-01, -2.9177e-02, -4.3907e-02,  9.2664e-01,  2.8644e-01,\n",
       "           4.0754e-02,  8.6899e-01, -1.2868e-01, -3.3744e-01,  1.2946e-01,\n",
       "           8.9648e-02, -3.6052e-01, -2.4705e-02, -5.5916e-01,  3.7844e-01,\n",
       "           3.7713e-01,  2.0163e-01,  5.3400e-01]], device='mps:0'),\n",
       " tensor([[-1.7731e-01, -3.8604e-01, -5.6476e-01, -2.2650e-01, -1.3208e-01,\n",
       "          -3.1625e-01, -5.9292e-01,  4.5825e-01,  2.2151e-01,  5.2940e-01,\n",
       "           3.8502e-01,  1.4343e-01,  6.5615e-02, -2.5563e-01, -4.4222e-01,\n",
       "          -1.6916e-01,  1.6884e-01,  5.0107e-01,  3.6126e-01, -7.6046e-03,\n",
       "           5.9085e-01,  3.7327e-01, -2.0317e-01,  2.5562e-01, -7.5220e-01,\n",
       "           2.6747e-02, -1.6815e-01, -2.5680e-02,  9.6832e-02, -3.0143e-01,\n",
       "          -1.1712e-01,  2.7595e-01, -3.7936e-01,  7.2144e-01, -1.4269e-01,\n",
       "          -2.3945e-01,  1.6259e-01, -1.0450e-01,  5.3649e-01, -1.3024e-01,\n",
       "           5.7769e-01, -2.9111e-01,  4.9321e-01,  3.4162e-02,  2.2711e-01,\n",
       "           1.3206e-01,  2.9819e-01, -4.0790e-01,  1.3721e-01, -5.3671e-01,\n",
       "          -1.6362e-02, -7.3465e-02,  4.1546e-01,  1.0724e-01, -6.3685e-01,\n",
       "          -2.7220e-01, -7.3533e-02,  4.2370e-01, -3.4661e-01,  1.4839e-01,\n",
       "           1.4452e-01,  4.6875e-01,  1.2890e-01,  1.1326e-01, -3.3348e-01,\n",
       "           3.0093e-01, -7.0323e-01, -8.0455e-01, -5.7621e-01, -1.8468e-01,\n",
       "           2.6210e-01,  2.4685e-01,  3.2711e-01, -6.3178e-01,  2.8882e-01,\n",
       "          -5.5500e-01,  4.7778e-01, -3.7641e-02, -3.8107e-01,  2.4976e-02,\n",
       "           4.6662e-01,  1.6673e-01, -2.3893e-01,  2.6411e-01,  2.9965e-01,\n",
       "          -6.2089e-02,  4.4818e-01, -2.9706e-02,  5.6407e-02,  3.8863e-01,\n",
       "           5.0392e-01,  2.6142e-01, -5.4248e-01, -2.1306e-01,  2.3546e-01,\n",
       "          -3.7940e-01,  2.9621e-01, -1.0776e-01,  2.0922e-02, -8.6342e-02,\n",
       "          -3.6881e-02,  3.4334e-01,  3.1658e-01, -6.2528e-02, -1.0724e+00,\n",
       "          -4.3983e-01, -2.4311e-02, -2.7814e-01, -3.4665e-01, -1.0267e-02,\n",
       "           7.0862e-03, -2.1454e-01,  4.5842e-01, -4.4114e-01, -3.9926e-01,\n",
       "           1.3807e-01, -3.8646e-02, -1.5082e-02, -1.4568e-01,  6.0048e-01,\n",
       "           1.6661e-01, -5.3057e-02,  1.5341e-01,  8.8933e-02,  4.9861e-01,\n",
       "          -3.9381e-01,  3.7161e-01, -4.2227e-01,  3.0903e-01,  6.8681e-02,\n",
       "          -2.0183e-02, -7.2510e-02, -2.5207e-01, -2.0895e-01, -2.4617e-02,\n",
       "           2.1165e-02,  2.8627e-01, -4.8458e-01,  5.5727e-02,  5.6803e-02,\n",
       "           2.5004e-01, -1.2498e-01, -4.2667e-01, -1.9318e-01, -6.4563e-02,\n",
       "          -4.0302e-02,  1.9246e-01, -1.1367e-01, -2.7591e-01, -2.6032e-01,\n",
       "          -7.7253e-01, -4.4007e-01,  3.9722e-01,  1.0952e-01,  9.2693e-02,\n",
       "           1.1479e-01,  2.1754e-01, -1.9775e-01, -1.8512e-01, -1.5673e-01,\n",
       "           5.8319e-01, -9.5195e-02,  6.6991e-02, -3.6372e-01, -3.0891e-01,\n",
       "           7.3167e-02,  1.7092e-01, -1.7339e-01,  4.3863e-02,  1.4590e-02,\n",
       "           2.2225e-01, -8.8855e-02, -1.0134e-03, -7.5909e-01, -3.0093e-02,\n",
       "          -2.8356e-01,  3.3814e-01,  2.7840e-01,  1.8768e-01, -2.4537e-01,\n",
       "           5.0899e-02, -1.7294e+00,  7.8671e-01, -5.9166e-01,  2.4877e-01,\n",
       "          -1.8150e-01,  3.1718e-01,  9.1189e-02,  1.8509e-01, -2.9267e-01,\n",
       "          -1.2026e-01, -3.9309e-02,  4.1623e-02,  2.9227e-01, -8.4757e-02,\n",
       "           4.8559e-01,  5.5627e-01,  3.3170e-01,  2.0436e-01, -9.9033e-02,\n",
       "          -6.0977e-01,  3.3235e-01,  1.6388e-01, -2.7455e-01,  2.9857e-02,\n",
       "           2.4361e-01, -3.2791e-01, -5.7703e-02,  3.4800e-01,  3.3556e-02,\n",
       "          -8.5587e-02, -3.0852e-01, -4.0912e-01, -1.8006e-01, -1.6990e-01,\n",
       "           3.0904e-01, -2.3500e-01, -2.0646e-01,  7.4699e-01,  9.5515e-02,\n",
       "          -2.2466e-02, -7.9463e-02, -2.8277e-01, -4.4039e-01, -2.0940e-01,\n",
       "           2.9576e-01, -5.3435e-01, -1.7453e-01, -2.8854e-01, -4.6020e-01,\n",
       "           4.6218e-01, -9.4786e-01,  6.3073e-01,  2.5225e-01,  1.6028e-01,\n",
       "           6.5808e-02, -8.2977e-02, -1.3264e-01, -3.6344e-01, -1.9702e-02,\n",
       "          -2.1377e-02, -5.7642e-01, -9.0486e-01,  1.5472e-01, -7.9254e-02,\n",
       "           1.2900e-01, -4.1901e-02,  6.2316e-01,  4.2972e-01,  5.9118e-01,\n",
       "           2.3140e-01,  5.5355e-03,  4.6999e-01,  4.3441e-01, -4.1017e-01,\n",
       "           4.6498e-01,  1.8871e-01, -4.7838e-02,  6.9646e-02, -3.2186e-01,\n",
       "          -9.1404e-01,  2.0856e-02,  3.2947e-01,  4.9930e-01, -1.7114e-01,\n",
       "           4.2813e-01,  2.6662e-01, -1.3998e-01, -1.7031e-01,  3.1347e-01,\n",
       "          -4.8944e-01, -1.5127e-01,  9.5415e-02, -1.4029e-01, -6.5634e-01,\n",
       "          -4.8841e-01,  2.7357e-02, -2.9199e-01,  2.0484e-01, -9.9702e-01,\n",
       "          -2.6273e-01,  5.6883e-01, -2.8623e-02,  9.3732e-02, -2.3345e-01,\n",
       "           7.0240e-02, -1.0750e-01,  1.0928e-01, -1.5161e-01, -8.9822e-01,\n",
       "          -1.4258e-01,  9.6331e-02, -2.2318e-01,  3.2530e-01,  3.7809e-01,\n",
       "          -2.0083e-01, -2.9617e-01,  2.8364e-01, -9.0550e-02, -2.5039e-01,\n",
       "          -1.3545e-01, -3.7063e-01, -2.4607e-02, -1.3576e-01, -1.1638e-01,\n",
       "           6.9800e-02,  4.9085e-01,  4.8805e-01, -7.7597e-01, -5.5824e-01,\n",
       "          -2.2992e-02,  5.0756e-01, -1.6302e-01,  3.7266e-02, -2.2967e-01,\n",
       "          -1.0825e+00, -2.0374e-01, -1.1835e-01, -3.9640e-01, -3.8528e-01,\n",
       "           5.0161e-01, -1.9830e-01, -2.0671e-01,  9.6594e-02, -3.4539e-01,\n",
       "           3.4160e-01,  8.4200e-01, -4.9032e-02, -4.7416e-02, -2.9725e-01,\n",
       "           1.7477e-02, -1.8692e-01,  4.8213e-02, -1.9462e-01, -2.5485e-02,\n",
       "           3.1570e-01,  1.2741e-01, -1.8351e-01, -8.2635e-02, -1.7372e-01,\n",
       "           7.9347e-01, -5.6947e-01,  1.3088e-01,  3.8738e-01, -2.4876e-01,\n",
       "          -1.3259e-01,  4.8066e-01,  1.4439e-01,  1.9491e-01, -1.0024e-01,\n",
       "          -2.6493e-02, -2.8608e-01, -1.8423e-01, -3.1248e-01,  9.6616e-01,\n",
       "           3.5980e-01,  2.2368e-01, -3.0496e-01,  5.7904e-01,  5.6810e-01,\n",
       "          -3.8525e-01,  7.6086e-02,  8.8516e-02,  3.4151e-03, -3.1432e-01,\n",
       "           3.2810e-01,  2.4729e-01, -4.2766e-01, -2.5123e-02,  4.5374e-03,\n",
       "           1.4485e-01, -5.5797e-01,  1.6191e-01,  1.0782e-01,  8.5195e-02,\n",
       "          -1.4724e-03, -2.6111e-01,  5.2888e-01, -2.4531e-01, -2.8912e-01,\n",
       "          -1.0092e-01, -1.2182e-01,  4.1012e-02, -2.4548e-01, -3.6887e-01,\n",
       "          -2.7428e-01,  2.4193e-02,  3.5074e-01,  1.4285e-02,  7.4821e-02,\n",
       "           2.5544e-01, -2.6796e-01, -3.4638e-01,  4.5500e-01, -1.7660e-02,\n",
       "          -1.6810e-01,  4.0768e-01, -2.4890e-01, -1.1639e-01,  1.3881e-01,\n",
       "           6.2926e-02,  1.0677e-01, -5.8094e-02,  3.7998e-01,  1.7175e-01,\n",
       "          -2.8992e-01,  3.6344e-02,  2.1408e-01, -8.0245e-02,  3.1665e-01,\n",
       "           5.1861e-02,  8.2278e-01, -3.7109e-01, -2.7164e-02, -3.8441e-01,\n",
       "          -1.2821e-01, -1.6499e-01,  7.5490e-03,  1.5024e-01, -7.5773e-01,\n",
       "           3.0131e-01,  2.8842e-01,  3.4034e-01,  1.1331e-01, -1.1458e+01,\n",
       "           1.6305e-01,  6.4218e-01, -5.1374e-01,  1.8438e-01,  1.7594e-01,\n",
       "          -1.4040e-01, -3.9794e-01,  4.0875e-02, -3.4785e-01, -2.6394e-01,\n",
       "          -6.5590e-02,  1.9075e-01, -1.2756e-02, -5.5478e-01,  3.3680e-02,\n",
       "          -2.5398e-01, -6.3192e-01,  2.6160e-01, -9.0323e-02,  4.1335e-01,\n",
       "          -3.0876e-01,  1.8731e-01,  2.3832e-01, -3.6034e-01, -2.7614e-02,\n",
       "          -5.1547e-01, -5.5807e-02,  3.7042e-01,  3.2770e-01, -3.5254e-01,\n",
       "           2.2243e-01, -1.8729e-01,  3.8908e-01, -2.6449e-01, -5.2363e-01,\n",
       "           1.3092e-01, -9.6993e-02,  3.7952e-01,  1.1884e-02, -2.4878e-01,\n",
       "          -2.5432e-02, -5.2584e-01,  5.9952e-02,  2.0270e-01,  3.4279e-01,\n",
       "           1.2541e+00,  9.4924e-01,  1.3760e-01, -1.3198e-01,  1.6614e-01,\n",
       "          -7.3513e-01,  2.0677e-01, -6.2811e-01, -8.1585e-02, -6.4376e-02,\n",
       "           8.3128e-01, -4.3261e-01, -3.3743e-01,  7.3012e-02, -8.0493e-02,\n",
       "           3.1300e-01, -7.9471e-02,  2.6202e-01, -3.1186e-01,  2.9341e-03,\n",
       "           6.1375e-01, -1.8722e-02,  3.8063e-01, -4.3721e-01, -6.2672e-02,\n",
       "          -4.2457e-02,  2.9525e-01,  3.7279e-01,  6.2687e-02, -2.5474e-01,\n",
       "          -4.1450e-01,  4.2406e-02,  5.2243e-01,  1.3990e-01,  1.4541e-01,\n",
       "           5.8211e-01, -2.3165e-01, -4.1432e-01, -6.0104e-01, -3.9034e-01,\n",
       "          -7.6601e-03,  1.3684e-01, -3.9867e-01,  6.4750e-01, -4.2253e-01,\n",
       "          -2.7567e-01, -1.6427e-01, -8.8628e-02,  3.8882e-01,  5.0639e-01,\n",
       "          -3.3627e-01,  8.0837e-02, -1.8585e-01, -4.4707e-03, -6.7852e-01,\n",
       "          -3.5458e-01, -8.4538e-02, -2.9987e-01,  2.3231e-01,  1.7305e-01,\n",
       "          -1.0020e-01,  9.0697e-03,  1.2051e-01,  7.3101e-02, -5.4356e-01,\n",
       "           3.0479e-01, -1.0749e-01,  6.8380e-02, -3.9586e-01, -5.1756e-01,\n",
       "           4.0833e-01,  7.0232e-01, -4.2415e-01, -3.3873e-01,  9.8320e-02,\n",
       "           1.3073e-01,  1.8252e-01,  1.2615e-01, -6.4414e-01,  1.1900e-01,\n",
       "          -4.1931e-01, -1.3812e-01,  1.3512e-01,  7.2335e-01,  6.6950e-01,\n",
       "          -5.3703e-02, -3.2049e-01,  2.4632e-01, -5.0640e-02, -2.5765e-01,\n",
       "          -7.0535e-01, -2.9464e-02,  2.1372e-01, -5.3812e-01,  9.7345e-02,\n",
       "          -3.3445e-01,  1.9755e-01, -1.7248e-01,  1.0825e-01, -7.4644e-02,\n",
       "           4.1662e-01,  4.1987e-01,  7.6810e-02,  4.3515e-01, -1.7908e-01,\n",
       "          -2.1990e-01,  1.9709e-01, -1.7846e-01, -1.1703e-01, -1.7569e-01,\n",
       "          -1.7366e-01,  9.7459e-02, -4.0722e-01, -3.6435e-02,  1.7903e-01,\n",
       "          -1.2723e-01,  6.7948e-02, -3.4229e-01,  2.2739e-01,  3.3538e-01,\n",
       "          -1.0467e-01, -3.7966e-01, -2.3884e-01,  4.7521e-02,  7.1481e-02,\n",
       "           3.8731e-01, -2.6195e-01, -1.0102e-01, -7.3811e-01, -2.0636e-01,\n",
       "           3.6787e-02,  1.0410e+00, -3.7965e-02, -3.0778e-01,  3.5156e-01,\n",
       "           3.9051e-01, -2.4726e-01, -3.4583e-01, -1.8699e-01,  7.4840e-01,\n",
       "          -2.3598e-01, -1.5465e-01,  2.2099e-01,  2.5510e-02,  6.2240e-01,\n",
       "           1.3154e-01,  2.2300e-01, -1.8512e-01,  4.7019e-01,  2.1848e-01,\n",
       "          -4.4102e-01, -1.5073e-01, -3.9303e-01, -3.2472e-01,  4.3754e-01,\n",
       "          -6.9776e-02,  2.7454e-01, -1.1790e-01,  8.1553e-02,  1.4253e-02,\n",
       "           5.4206e-02,  1.1645e+00, -2.0185e-01, -2.3960e-01,  4.7684e-01,\n",
       "           2.7079e-01,  3.4262e-01,  3.7917e-02, -1.3477e-01,  9.5921e-02,\n",
       "           4.9705e-01, -1.6706e-01, -4.9624e-01,  1.5664e-01,  3.6272e-01,\n",
       "           5.9108e-01, -8.7613e-01, -3.6622e-01,  2.0509e-01,  2.2296e-01,\n",
       "          -3.2121e-01,  1.9818e-01, -7.6228e-02,  3.2510e-01,  3.6266e-01,\n",
       "          -1.2771e+00,  1.7642e-01,  7.1262e-02,  1.5943e-01, -1.5269e-01,\n",
       "          -6.7715e-01, -2.5385e-01,  1.0595e-01, -6.2916e-02,  4.2076e-01,\n",
       "           4.4789e-01, -7.1509e-01,  1.2452e-01,  3.1511e-01, -3.7691e-01,\n",
       "           1.2320e-01, -2.7571e-01,  1.6643e-01,  5.2170e-01,  1.6129e-01,\n",
       "          -4.2505e-01,  2.5049e-01, -5.0826e-01, -6.1173e-01,  1.4561e-01,\n",
       "           6.9575e-01,  2.1324e-02, -1.1322e-01, -1.1804e-01, -2.9289e-01,\n",
       "          -1.9107e-01, -3.1790e-01, -3.4967e-01, -2.7542e-02,  1.4802e-01,\n",
       "           2.4431e-02, -3.6606e-02,  2.4884e-01,  1.1143e-01, -2.2300e-01,\n",
       "           5.8287e-01,  4.8747e-01,  7.4334e-01, -1.5062e-02,  1.1963e-02,\n",
       "          -2.2207e-01,  2.9970e-01, -4.7196e-02,  4.7552e-01, -3.4433e-01,\n",
       "          -2.0911e-01,  2.2006e-01,  3.6212e-02, -6.2294e-01, -1.0381e-01,\n",
       "          -3.5492e-01,  3.0044e-01,  1.1075e-02, -2.7609e-01,  4.8880e-01,\n",
       "           2.0553e-02,  1.8280e-01, -5.5884e-01,  4.6756e-01,  1.6596e-01,\n",
       "           2.2073e-02,  9.8054e-01,  1.1685e-01,  5.3145e-01,  5.8221e-01,\n",
       "           1.1828e-01, -8.1138e-01, -1.8763e-01, -7.2256e-02,  1.5035e-01,\n",
       "           1.1544e-02,  1.5872e-03,  1.5741e-01,  2.8459e-02,  9.3799e-01,\n",
       "           4.5776e-01, -2.2136e-01, -3.9686e-01, -3.7782e-01,  1.8491e-01,\n",
       "          -8.4548e-02, -3.3508e-01,  2.0442e-01, -2.2187e-01,  1.7388e-02,\n",
       "          -5.8935e-01, -5.5885e-01, -1.2302e-01,  2.7758e-01, -2.6880e-01,\n",
       "           5.4517e-01,  4.6664e-02,  5.2447e-03,  3.4141e-01, -2.9763e-02,\n",
       "           2.1521e-01,  6.7382e-01,  2.2993e-01, -2.9428e-01,  1.8031e-01,\n",
       "           2.1968e-01, -1.3524e-01,  2.0995e-01, -8.6762e-01,  3.6296e-01,\n",
       "          -5.1872e-02,  5.4038e-01,  4.7469e-01]], device='mps:0'),\n",
       " tensor([[-6.8440e-02, -6.1567e-01, -6.8551e-01, -9.0571e-01, -3.6638e-01,\n",
       "          -4.1167e-01, -9.6670e-01,  1.1460e-01,  4.6579e-01,  1.4290e-01,\n",
       "           9.8639e-02,  7.6311e-01,  1.9776e-01, -6.4306e-01, -2.6599e-01,\n",
       "          -2.0143e-01, -3.5707e-01,  4.3854e-01,  3.0526e-01,  3.8213e-01,\n",
       "           6.4217e-01, -8.2762e-02,  2.7324e-01,  1.1692e-02, -1.8514e-01,\n",
       "           3.6500e-01,  6.6316e-02, -5.9700e-02, -2.2576e-02, -3.3517e-01,\n",
       "           4.5611e-02,  4.5763e-01, -1.6025e-01,  1.1059e+00,  1.3205e-01,\n",
       "          -3.0036e-01,  3.2734e-01, -3.5608e-01,  7.5722e-01,  3.8600e-01,\n",
       "           4.8290e-01, -4.4890e-02,  2.9644e-01, -2.1108e-02,  3.3282e-01,\n",
       "           3.0103e-01,  4.3903e-01, -4.0593e-01,  3.2835e-01, -3.0832e-01,\n",
       "          -2.3030e-01, -9.5423e-03,  6.3261e-01,  3.5270e-01, -4.3699e-01,\n",
       "          -3.9819e-01,  6.4942e-02,  3.8610e-02, -4.4479e-01,  2.2105e-01,\n",
       "          -2.3588e-01,  5.3986e-01,  2.1186e-02, -3.6906e-02,  5.6112e-01,\n",
       "           5.0361e-01, -7.0454e-01,  8.2692e-03, -1.5009e-01, -1.7516e-01,\n",
       "          -4.3843e-02,  2.2956e-01,  1.9022e-01, -1.9426e-01,  3.1354e-01,\n",
       "          -4.5359e-01,  1.2511e-01, -5.5034e-02, -2.6176e-01,  1.6612e-01,\n",
       "           1.7452e-02, -1.1973e-01,  1.5225e-03,  1.6402e-01, -7.8709e-02,\n",
       "          -3.4534e-01,  5.0996e-01,  6.8605e-02,  3.0883e-02,  2.6672e-01,\n",
       "           4.2743e-01,  7.1080e-02, -2.0594e-01, -6.8634e-01,  1.4389e-01,\n",
       "           1.7825e-01,  2.0870e-01, -2.6405e-01,  2.7510e-01, -1.4522e-01,\n",
       "           2.0643e-01, -1.7576e-01, -2.3019e-02,  2.8568e-01, -7.5668e-01,\n",
       "          -6.3125e-01,  8.6547e-02, -4.1034e-01, -2.8925e-01,  4.6023e-02,\n",
       "          -5.3326e-01, -3.4498e-01,  1.6879e-01, -3.0012e-01, -2.1372e-01,\n",
       "          -9.2202e-02, -1.2911e-01, -1.8546e-01, -1.4728e-01,  4.1628e-01,\n",
       "          -2.3979e-01, -2.4835e-01,  4.3662e-01,  1.8691e-01, -1.7726e-01,\n",
       "          -1.5903e-01,  3.5384e-01, -3.6026e-02,  5.4647e-01,  7.2925e-01,\n",
       "          -1.8875e-01, -3.4967e-03, -7.9559e-02,  2.0667e-01, -3.4355e-01,\n",
       "          -2.2621e-03,  1.9575e-01, -4.4328e-01,  3.4295e-01,  3.0954e-01,\n",
       "          -4.0244e-02,  1.3387e-01, -2.3284e-02, -1.1288e-01, -1.3513e-01,\n",
       "           2.2343e-02, -1.3022e-01, -3.6468e-01, -4.2739e-01, -1.0881e-01,\n",
       "          -8.1378e-01, -1.1007e+00,  3.5853e-01, -1.5357e-01,  3.3957e-04,\n",
       "           2.6908e-01,  5.8922e-01, -3.7131e-02, -1.8346e-01,  1.4983e-01,\n",
       "           8.3747e-01, -1.9017e-01,  1.3692e-01, -5.4608e-01, -3.9811e-01,\n",
       "           4.0121e-01, -1.4787e-01, -5.2749e-02, -1.4801e-01, -7.4842e-02,\n",
       "           9.5451e-02, -4.7606e-01,  6.2954e-01, -5.8887e-02,  3.4851e-01,\n",
       "          -7.1842e-01, -1.3438e-01,  2.0717e-01, -6.6760e-02, -1.5866e-01,\n",
       "           2.4434e-01, -1.3012e+00,  2.9778e-01, -5.2696e-03,  2.2341e-01,\n",
       "          -2.8454e-01,  1.1818e-01,  1.4453e-01,  6.6674e-02,  1.0480e-01,\n",
       "           1.4952e-01,  1.9171e-01,  2.3239e-01,  5.2520e-01,  1.5651e-01,\n",
       "           2.1228e-01,  9.0922e-01,  4.6329e-02, -1.0329e-01,  6.6856e-02,\n",
       "          -2.4414e-01,  2.1584e-01,  3.3212e-02,  2.2433e-01,  5.7168e-03,\n",
       "           2.6629e-01,  1.1888e-02,  1.1274e-01, -1.8366e-02, -1.6825e-01,\n",
       "          -1.4094e-02,  1.1814e-01, -7.8362e-01,  2.7416e-01, -1.5072e-01,\n",
       "           1.4248e-01, -4.0110e-01, -2.6179e-02,  4.6046e-01,  8.5497e-02,\n",
       "           5.7754e-03,  2.0294e-01,  2.3967e-01, -5.4230e-02, -1.2138e-01,\n",
       "           2.7332e-01, -2.4406e-01,  1.0481e-01, -8.2899e-02, -3.2789e-01,\n",
       "           5.4369e-01, -3.0406e-01,  6.8193e-01,  6.4307e-01, -2.1470e-01,\n",
       "           2.5228e-03, -1.5402e-01, -1.0553e-01, -4.4043e-01,  1.4932e-01,\n",
       "          -4.5821e-01, -4.9816e-01, -4.3410e-01,  3.4489e-02,  2.2044e-01,\n",
       "           1.1179e-01,  1.3677e-01,  1.1448e+00,  3.9932e-01,  4.2509e-01,\n",
       "           2.0026e-01, -8.8041e-02,  2.8139e-01, -4.9453e-02, -3.9008e-01,\n",
       "           3.2494e-01,  1.6439e-01, -5.5436e-01,  4.5891e-01,  1.1481e-01,\n",
       "          -2.2092e-01,  7.6933e-03,  9.5860e-02,  4.1336e-01, -9.0223e-02,\n",
       "           8.9799e-02,  7.5296e-01, -1.6053e-01, -4.3947e-01,  4.5455e-01,\n",
       "          -4.6984e-01,  4.2360e-01, -2.4254e-01,  1.5036e-01, -7.1209e-01,\n",
       "          -8.5463e-01, -1.4327e-01, -2.3435e-01,  2.0269e-01, -9.8882e-01,\n",
       "           8.2125e-02, -1.4310e-01, -1.6680e-01,  1.3198e-01, -1.2436e-02,\n",
       "           4.2453e-01, -4.5945e-02,  1.2416e-01, -4.7197e-01, -2.7331e-01,\n",
       "           4.5780e-02, -2.6536e-05, -6.4607e-01,  1.6084e-01,  1.9963e-01,\n",
       "          -5.7620e-01, -9.0077e-02, -1.7872e-02, -1.7255e-01, -6.7646e-01,\n",
       "          -1.1339e-01, -2.9593e-02,  3.7503e-02, -2.9412e-01,  9.1477e-02,\n",
       "           2.2788e-01,  5.6036e-01,  2.0159e-01, -3.1651e-01, -5.2153e-01,\n",
       "          -1.3919e-01,  4.6196e-01, -1.4350e-01, -3.5782e-01, -1.9500e-01,\n",
       "          -8.6859e-01,  2.0336e-02, -8.5532e-02, -2.9870e-01, -6.4456e-01,\n",
       "           3.4463e-02,  4.8347e-01, -7.7219e-02, -6.9886e-02, -2.6082e-01,\n",
       "           1.9429e-01,  1.3409e-01,  1.1823e-02,  3.6727e-02, -4.5774e-01,\n",
       "          -5.8844e-02, -5.6964e-01, -2.5480e-01, -4.2066e-01,  1.3881e-01,\n",
       "           1.0163e-01, -1.5564e-01,  4.6162e-01, -2.1236e-01, -4.6308e-01,\n",
       "           7.7805e-01, -5.5288e-01,  4.9542e-01,  5.1126e-01, -2.5418e-01,\n",
       "          -1.2231e-01,  2.8878e-01,  2.9830e-02, -2.2866e-01,  3.4908e-01,\n",
       "           5.0759e-01, -2.0301e-01, -5.0787e-01, -1.4423e-01,  5.4634e-01,\n",
       "           6.2664e-01,  2.6607e-02, -5.2209e-01,  1.0827e-01,  2.0350e-01,\n",
       "          -3.7798e-01,  2.7532e-01, -4.2631e-01, -1.0673e-01, -3.3490e-01,\n",
       "           3.4900e-01,  5.7334e-01, -3.5473e-01,  1.8844e-01,  3.8423e-02,\n",
       "           2.0787e-01, -3.2998e-03, -1.4752e-01,  2.4446e-01,  6.3058e-01,\n",
       "           5.2356e-03, -3.0765e-01,  2.9022e-01, -1.2732e-01, -4.8072e-01,\n",
       "           1.2679e-01,  2.0867e-01, -2.4724e-01, -2.5492e-01, -3.0377e-01,\n",
       "           1.7978e-02, -2.0453e-01,  6.2768e-01, -2.4218e-02, -3.7363e-01,\n",
       "           4.7951e-01, -3.0343e-01, -3.2949e-01,  8.6097e-01, -4.0169e-02,\n",
       "          -6.7772e-01,  5.6794e-01, -3.3788e-01, -1.4797e-01,  1.6652e-01,\n",
       "          -2.2268e-01,  3.7729e-01,  2.9616e-01,  4.2201e-01,  1.1322e-01,\n",
       "          -5.7225e-01, -1.0068e-01, -1.8026e-01, -2.1607e-01,  4.8807e-01,\n",
       "          -8.3315e-02,  6.0486e-01, -1.7046e-01, -1.6590e-02,  2.1921e-01,\n",
       "           3.6653e-01, -2.4128e-01,  1.4542e-02,  2.8682e-01, -4.9042e-01,\n",
       "           1.1415e+00,  3.2186e-01,  3.1149e-01,  4.5361e-01, -1.1071e+01,\n",
       "           5.2935e-01,  8.6400e-01, -2.8949e-01, -1.1855e-01,  4.0762e-01,\n",
       "           3.0067e-02, -5.6444e-01,  3.0209e-01, -4.4027e-01, -1.3890e-01,\n",
       "          -9.3478e-02, -1.6436e-01,  3.6105e-01, -3.3074e-01, -3.9431e-01,\n",
       "           1.2346e-01, -6.9892e-01, -2.4912e-02, -1.4803e-01,  4.8552e-01,\n",
       "          -1.9101e-02, -6.0469e-02,  3.7632e-01, -2.3898e-01,  5.4751e-02,\n",
       "          -2.0981e-01,  3.5063e-01,  4.3703e-01,  2.4707e-01,  9.8359e-02,\n",
       "          -2.3600e-01, -3.2755e-02, -1.6841e-01, -3.2207e-01, -8.7129e-01,\n",
       "          -1.7835e-01, -5.8865e-01,  2.4012e-01, -8.2689e-01, -4.3580e-01,\n",
       "           2.4104e-02, -6.5527e-01,  1.5929e-01,  4.2353e-01,  8.7764e-01,\n",
       "           9.9047e-01,  9.4393e-01, -1.3787e-01, -2.2072e-01, -2.3588e-01,\n",
       "          -6.5016e-01,  7.4608e-02, -4.3903e-01, -8.4857e-03, -5.6144e-02,\n",
       "           9.1847e-01, -7.4441e-01,  2.2063e-01, -7.3157e-01, -1.4435e-01,\n",
       "           5.4313e-01,  9.5912e-02, -5.9593e-02, -3.9118e-01, -8.7961e-02,\n",
       "           4.4757e-02, -1.4224e-01,  2.1528e-01, -1.4801e-01,  5.4066e-02,\n",
       "          -1.5805e-01,  3.6870e-01,  2.0149e-01, -4.5900e-01,  3.6061e-01,\n",
       "          -6.0429e-02, -1.5538e-01,  1.4964e+00,  2.5752e-02,  1.0927e-01,\n",
       "          -1.2265e-01,  2.3113e-01, -1.2835e-01, -4.9498e-01, -2.5396e-02,\n",
       "           2.3290e-01,  1.0978e-01,  2.3910e-01,  2.7299e-01, -1.1906e-01,\n",
       "          -3.0919e-01,  2.9157e-01,  2.6256e-01,  2.4434e-01,  4.5100e-01,\n",
       "          -9.0948e-01,  1.6498e-02,  2.5183e-01,  1.8562e-01, -5.5324e-01,\n",
       "           3.1595e-02, -4.0827e-01, -5.7318e-01,  5.4527e-01, -2.6323e-01,\n",
       "           1.7727e-01, -1.5405e-01, -3.9527e-01,  4.5459e-01, -5.4141e-02,\n",
       "           1.0417e-01,  3.4943e-01, -9.5500e-02, -3.7475e-01,  5.2868e-03,\n",
       "           1.1371e-01,  2.3882e-01, -3.0584e-01, -2.1562e-01, -2.5764e-01,\n",
       "          -2.8334e-02,  4.2491e-01,  5.2654e-01, -7.2695e-01,  4.6824e-01,\n",
       "          -4.7134e-01,  3.4289e-01, -1.8236e-02, -2.4621e-02,  3.2297e-01,\n",
       "           2.8516e-01, -4.2229e-01, -1.5763e-01, -3.2971e-01, -5.6693e-01,\n",
       "          -5.2879e-01, -1.2382e-01,  5.8274e-01,  2.4729e-01,  4.6219e-02,\n",
       "          -2.4490e-01, -1.5102e-01, -4.8222e-02,  2.9591e-01,  1.6306e-01,\n",
       "          -2.7028e-03,  3.1591e-01, -6.8332e-01,  1.7536e-02,  4.9712e-02,\n",
       "          -3.8844e-01, -3.4469e-01, -1.6842e-01, -1.7756e-02, -5.3132e-01,\n",
       "          -1.2535e-01, -2.3976e-01, -3.0059e-01, -4.5478e-01,  2.2271e-02,\n",
       "          -1.0925e-01, -1.4158e-02, -2.2029e-01,  7.3141e-01,  4.5972e-01,\n",
       "           2.1203e-01, -4.9215e-01, -6.3569e-01, -1.8737e-03,  2.7079e-02,\n",
       "           1.0130e-01,  4.2634e-01, -6.1714e-01, -7.1684e-02, -6.1537e-01,\n",
       "          -4.7320e-01,  4.1885e-01, -3.9122e-01, -5.3615e-01,  1.3807e-01,\n",
       "           2.1119e-01,  3.7869e-01,  5.8091e-02,  1.8201e-01, -3.6851e-01,\n",
       "          -5.4496e-01, -4.0979e-01,  2.6359e-01, -2.9032e-01,  3.1234e-01,\n",
       "           9.1288e-01, -2.4419e-01,  5.4744e-02,  3.9268e-01, -1.0293e-01,\n",
       "          -2.1390e-01, -1.1739e-01, -3.8167e-01,  8.7756e-03,  1.4015e-01,\n",
       "          -1.3432e-01,  4.2543e-01,  1.4473e-01, -3.3980e-01,  1.9419e-01,\n",
       "           1.9926e-01,  1.0079e+00, -1.9778e-01,  1.4163e-01,  7.1735e-01,\n",
       "          -1.9607e-01, -5.1626e-02,  4.5361e-01,  1.0530e-01,  8.3014e-02,\n",
       "           2.5955e-01, -3.8604e-01,  2.3659e-02, -1.0134e-01,  3.6612e-01,\n",
       "           6.7566e-01, -4.2284e-01, -1.9604e-01, -2.7321e-01,  5.4320e-01,\n",
       "           2.6088e-02,  4.9048e-01, -2.6245e-01,  5.7649e-01,  1.1524e-01,\n",
       "          -1.0882e+00, -2.3719e-01, -1.3276e-02,  1.8184e-01, -1.0124e-01,\n",
       "          -3.7704e-01, -2.0758e-01, -3.3661e-01, -6.2287e-02,  4.7111e-01,\n",
       "           5.5141e-02, -5.0755e-01,  2.1111e-01,  7.4249e-01, -5.2565e-01,\n",
       "           3.2329e-02, -5.1750e-01,  2.0179e-01,  2.1586e-01, -8.0050e-02,\n",
       "           3.9363e-01,  4.5234e-02, -6.8901e-01, -5.1770e-01,  2.2277e-01,\n",
       "           8.1475e-01, -1.1874e-01,  7.9284e-02, -1.9400e-01, -3.5176e-01,\n",
       "          -4.5660e-01,  2.0728e-01,  6.0365e-02, -2.0284e-01, -3.6111e-02,\n",
       "           1.7920e-01, -2.1904e-01, -2.2794e-01,  5.8693e-01, -1.0044e-01,\n",
       "           1.8279e-01,  4.1854e-01, -8.1510e-02,  1.1730e-01, -3.1282e-01,\n",
       "           2.8236e-01,  2.7894e-01,  9.1241e-02,  4.5320e-01, -4.8311e-01,\n",
       "          -4.4237e-01, -1.3241e-01, -9.7428e-02, -6.6949e-01, -2.6617e-01,\n",
       "          -6.5273e-01,  5.2143e-01, -5.2971e-01,  9.4244e-02,  4.9967e-01,\n",
       "           3.4637e-01,  1.5813e-01, -6.6386e-01,  4.0425e-01,  1.6590e-01,\n",
       "           1.0383e-01,  4.1510e-01,  1.8080e-01,  5.1758e-01, -3.3488e-01,\n",
       "           9.9620e-02, -8.2595e-01, -2.0650e-01,  1.1093e-02, -2.6528e-01,\n",
       "          -5.9410e-02, -5.2484e-01,  5.2501e-01,  2.7803e-02,  5.2912e-01,\n",
       "           4.3911e-01, -3.2744e-01, -7.9372e-01, -3.5859e-01,  1.5475e-01,\n",
       "           1.2856e-01,  1.3181e-01,  3.8671e-01,  2.4999e-01,  4.8117e-01,\n",
       "          -5.2437e-01, -7.8164e-01, -3.2135e-01,  1.2448e-01, -7.6998e-02,\n",
       "           4.5971e-01, -3.8042e-01, -5.9902e-01,  2.3923e-01,  1.0586e-01,\n",
       "           9.7645e-02,  6.7776e-01, -1.3657e-01, -4.5368e-01, -2.3307e-01,\n",
       "           2.1161e-01,  2.3228e-02,  6.3233e-01, -5.3866e-01,  1.1376e-01,\n",
       "          -3.3485e-03,  4.0446e-01,  6.5569e-01]], device='mps:0'),\n",
       " tensor([[-6.4825e-02, -5.2106e-01, -4.7129e-01, -7.1352e-01, -4.1874e-01,\n",
       "          -3.3978e-01, -5.6321e-01,  2.5766e-01, -1.8484e-01,  2.8946e-01,\n",
       "           1.6020e-01,  5.0809e-01, -2.9761e-01, -3.1342e-01, -3.6341e-01,\n",
       "          -1.8069e-01, -5.8500e-01, -2.6408e-03,  4.5701e-01,  3.3428e-01,\n",
       "           4.4373e-01,  2.1492e-01, -3.3609e-01,  2.7432e-01, -2.7496e-01,\n",
       "           7.8362e-02, -3.6782e-01, -2.2060e-01,  2.5464e-01,  4.0353e-01,\n",
       "          -2.6057e-03,  7.0182e-01, -1.2743e-01,  1.1250e+00, -1.5166e-01,\n",
       "          -2.5849e-01,  1.8462e-01,  1.6912e-01,  6.6740e-01, -4.7031e-01,\n",
       "           5.2026e-01,  3.7073e-01, -4.2569e-02, -2.6648e-02,  2.1226e-01,\n",
       "           2.9048e-01,  2.0070e-01, -4.8075e-01, -2.3366e-01, -4.5804e-01,\n",
       "          -1.2683e-01, -1.4647e-01,  5.7650e-01, -1.2586e-01, -8.0363e-01,\n",
       "          -2.9385e-01,  3.5176e-02,  3.2682e-01, -9.1530e-02,  2.7774e-01,\n",
       "          -4.5191e-02,  6.2563e-01, -1.7904e-01, -4.3002e-02,  2.4446e-01,\n",
       "           2.1190e-01, -7.4972e-01,  4.7833e-02, -7.4761e-01,  1.0685e-01,\n",
       "           2.6541e-01,  2.7565e-01,  4.1902e-02, -2.3926e-01, -1.9666e-01,\n",
       "           1.2829e-01,  2.5094e-01,  2.8252e-01, -2.9091e-01,  4.0238e-02,\n",
       "           1.3003e-01,  5.3658e-01,  4.0210e-01,  1.1421e-01,  5.4004e-02,\n",
       "           1.1214e-01,  4.0195e-01, -5.9318e-02, -5.2039e-03,  1.1265e-01,\n",
       "           2.8745e-01, -1.0246e-01, -4.5026e-01, -4.4526e-01, -1.1998e-02,\n",
       "          -2.5569e-01,  1.1069e-01, -3.2805e-01, -1.3091e-01, -3.6731e-01,\n",
       "           2.3212e-01,  2.0121e-01,  3.0390e-01, -8.9589e-02, -5.9814e-01,\n",
       "          -2.9427e-01,  3.1589e-01,  2.8366e-02, -2.7184e-01, -3.8099e-01,\n",
       "          -1.3858e-01, -5.0067e-01,  2.9704e-02, -5.9905e-01,  1.6921e-01,\n",
       "           6.6048e-02,  1.9034e-01, -2.6980e-01, -6.1016e-02,  3.0875e-02,\n",
       "           1.9304e-02, -5.7870e-01,  1.9193e-01,  1.9284e-01,  6.5058e-02,\n",
       "           1.1261e-01,  7.7020e-01, -7.6319e-02,  3.4631e-01,  3.9914e-01,\n",
       "           1.5024e-01,  2.0123e-01, -3.8401e-01, -3.5889e-01, -2.5630e-01,\n",
       "           3.6826e-02, -6.7761e-01, -4.7202e-01,  1.8507e-01, -1.7973e-01,\n",
       "           4.8701e-02, -5.8634e-02, -4.2142e-01, -6.6156e-01, -2.0305e-01,\n",
       "          -3.0240e-01, -3.5202e-02, -2.5152e-01, -5.2988e-01, -2.1979e-01,\n",
       "          -5.3207e-01, -6.8560e-01,  1.1391e-01,  1.2911e-01,  3.0541e-01,\n",
       "          -3.9329e-01,  4.4104e-01, -2.0136e-01, -4.1893e-01,  4.3976e-01,\n",
       "           5.0969e-01, -3.8342e-01,  2.3417e-01, -5.8400e-01,  1.4660e-01,\n",
       "           1.8595e-01, -8.8508e-02, -7.4275e-02,  4.4892e-01,  1.0856e-01,\n",
       "           3.5695e-01,  2.8331e-02,  1.2614e-01, -3.0004e-01,  1.1666e-01,\n",
       "          -1.7105e-01, -4.2594e-01, -8.2680e-02, -1.1702e-01, -1.5351e-01,\n",
       "           5.9227e-01, -1.4698e+00, -2.5325e-01, -5.4114e-02, -3.1088e-01,\n",
       "          -2.0718e-01,  5.1802e-02, -3.7429e-02,  3.1005e-01, -3.5773e-01,\n",
       "          -3.0028e-02,  7.6997e-02,  1.0992e-01, -5.2481e-02, -7.9266e-02,\n",
       "           1.8482e-01,  8.4826e-01,  4.5106e-01,  1.3314e-01, -2.8838e-01,\n",
       "          -5.4012e-01,  4.5734e-01,  1.5561e-01,  1.8193e-01,  4.3073e-01,\n",
       "           4.4635e-02, -6.0490e-01,  3.0504e-01,  1.1000e-01,  2.5177e-01,\n",
       "          -4.2604e-02, -1.6907e-01, -3.9960e-01, -2.6950e-01, -4.1066e-01,\n",
       "           2.5943e-01, -1.7532e-01, -1.7243e-01,  6.3180e-01,  5.4959e-02,\n",
       "           6.5588e-03,  3.6443e-01,  7.3443e-02, -3.3393e-01,  3.6316e-01,\n",
       "           4.9857e-01,  1.3469e-01,  2.0749e-01, -3.2029e-01, -9.1574e-02,\n",
       "          -3.0071e-01, -3.6468e-01,  4.0803e-01,  8.6293e-01, -1.5206e-01,\n",
       "           5.5927e-01, -3.5270e-01,  5.3560e-01, -3.9017e-01,  8.1285e-02,\n",
       "          -9.3146e-02,  3.9584e-02, -3.7348e-01,  1.5369e-01,  1.6798e-01,\n",
       "           9.5925e-02,  1.1447e-01,  7.8298e-01,  4.9105e-01,  1.9894e-01,\n",
       "           3.8166e-01,  2.6511e-02,  2.7216e-01,  7.3381e-01, -2.2718e-01,\n",
       "           5.0500e-01, -5.7285e-02, -4.4514e-01,  5.6800e-01, -3.3556e-01,\n",
       "           7.3583e-02, -2.5305e-01,  2.1730e-01,  2.8859e-01, -2.1042e-01,\n",
       "           1.7457e-01,  7.3826e-01,  2.1812e-01, -4.7769e-01,  6.5969e-01,\n",
       "          -4.3635e-01,  6.9529e-01,  4.4339e-02,  7.3822e-03, -2.3651e-01,\n",
       "          -6.8127e-01, -6.0514e-01,  1.9676e-01,  3.0464e-01, -6.5879e-01,\n",
       "          -4.5763e-01,  6.3686e-02,  1.2136e-01,  3.0831e-01,  2.7192e-01,\n",
       "           4.0691e-01, -1.4668e-01, -9.3748e-02, -4.7597e-01, -3.9844e-01,\n",
       "           3.4784e-02,  8.6306e-02,  1.7161e-02, -1.7234e-01,  1.1311e-01,\n",
       "           6.3554e-01,  2.4728e-01,  2.7233e-01,  2.0309e-01, -3.0888e-01,\n",
       "           1.0501e-01,  1.3520e-01,  4.8356e-01, -5.5674e-01, -1.8321e-01,\n",
       "           1.0654e-01,  4.1640e-02,  6.0277e-01, -1.9599e-01, -1.9078e-01,\n",
       "          -4.3756e-02,  6.1045e-01, -1.8914e-01,  1.0181e-01,  5.5625e-02,\n",
       "          -6.9975e-01,  3.2056e-01, -1.1154e-01, -1.7081e-01, -5.1389e-01,\n",
       "           2.2244e-01, -2.0140e-01, -1.6287e-01, -9.5493e-02, -1.7268e-01,\n",
       "           2.8308e-01,  1.3893e-01,  1.7448e-01, -5.6188e-01, -4.2752e-01,\n",
       "           3.2014e-01, -2.7909e-01,  4.3503e-02, -4.8550e-01,  4.2742e-01,\n",
       "           1.1431e-01,  3.8558e-01, -7.8663e-02,  6.2418e-02, -2.0310e-01,\n",
       "           6.0218e-01, -5.2397e-01,  1.9654e-01,  2.8141e-01, -3.8314e-01,\n",
       "          -2.1339e-01,  4.1385e-01, -2.3207e-01,  1.2589e-01,  2.8636e-01,\n",
       "           4.1314e-01, -1.4282e-01, -3.8034e-01, -5.0178e-02,  7.5572e-01,\n",
       "           6.7215e-01,  1.3754e-01, -6.2772e-01,  5.3940e-02,  4.3708e-01,\n",
       "          -3.9106e-01,  2.0241e-01, -3.6752e-01, -2.1631e-01,  2.7133e-02,\n",
       "          -1.1264e-01,  4.6458e-01, -1.5988e-01,  1.7771e-01,  4.9879e-01,\n",
       "           1.8651e-01, -5.4533e-01, -1.7627e-01,  6.7831e-01,  4.8863e-01,\n",
       "          -5.9119e-02, -6.0099e-01, -1.2013e-02, -6.6729e-01, -4.5641e-02,\n",
       "           2.2328e-02, -4.1547e-01,  2.8862e-01, -3.3416e-02, -3.0317e-01,\n",
       "          -1.1642e-01, -5.4113e-01, -2.8084e-02,  1.1841e-02, -1.5296e-02,\n",
       "           3.6090e-01,  4.2264e-02, -2.1034e-01,  9.8636e-01, -1.7617e-01,\n",
       "          -1.7216e-01,  6.6492e-01, -3.5360e-01, -2.4789e-02,  3.5074e-01,\n",
       "          -1.2685e-01, -1.7760e-01,  1.0776e-01,  1.0341e+00,  1.8563e-01,\n",
       "          -1.0997e-01, -2.2289e-01,  4.5971e-01, -4.7821e-01,  1.4294e-01,\n",
       "          -3.9929e-01,  3.9268e-01, -9.7826e-02,  1.7502e-01, -7.3251e-01,\n",
       "           3.9438e-01, -2.7307e-01, -2.7071e-01, -5.5340e-01,  4.7237e-02,\n",
       "          -1.3060e-01,  8.5511e-02, -8.7546e-02,  7.6618e-01, -1.1043e+01,\n",
       "           6.0728e-02,  7.9958e-01, -9.5852e-02,  1.5617e-01,  4.5071e-01,\n",
       "           1.9176e-01, -4.9120e-01,  7.1129e-01, -2.7299e-01,  1.0486e-02,\n",
       "          -2.9627e-01, -1.1472e-01, -7.9883e-02, -5.9069e-01,  3.8218e-01,\n",
       "          -1.0732e-01, -5.6742e-01,  2.8002e-01, -1.2831e-01,  3.6383e-01,\n",
       "          -3.2844e-01,  1.6256e-01,  6.2258e-01, -5.6404e-01, -3.9973e-01,\n",
       "          -1.8337e-01, -1.5875e-01,  6.0639e-01,  3.4210e-01, -1.3924e-01,\n",
       "           2.8654e-01,  1.1465e-01,  2.7886e-01, -5.8166e-01, -6.1478e-01,\n",
       "           4.9869e-01, -4.0690e-01,  2.4524e-01, -1.0549e+00, -3.5975e-01,\n",
       "          -1.0347e-01, -5.2712e-01,  1.1584e-01,  6.1502e-01,  8.3241e-01,\n",
       "           9.4745e-01,  1.1350e+00,  3.6312e-02,  3.9092e-01, -2.3656e-01,\n",
       "          -6.7364e-01,  3.7568e-02, -5.6739e-01, -3.6089e-01,  1.0510e-01,\n",
       "           2.5856e-01, -4.8878e-01,  5.4276e-01, -5.0351e-01,  1.8606e-01,\n",
       "           1.1487e-01,  5.9217e-02,  2.6631e-01, -4.8675e-01, -5.9340e-01,\n",
       "          -4.0694e-02,  5.0454e-02,  9.6270e-02, -5.5815e-01,  2.1392e-01,\n",
       "           2.6225e-01,  6.0538e-01, -1.8202e-01, -3.0828e-01,  3.9334e-01,\n",
       "           1.5842e-01,  1.0993e-01,  1.5678e+00, -3.1461e-01, -1.5477e-01,\n",
       "          -2.5324e-02, -2.6904e-01, -2.8078e-01, -4.5965e-01, -3.6256e-01,\n",
       "          -5.0609e-02,  2.8533e-01,  6.1110e-01,  3.6001e-01, -9.7555e-03,\n",
       "          -2.6114e-01, -4.1609e-02,  5.3602e-02,  4.3446e-01,  8.7102e-01,\n",
       "          -4.7274e-01,  2.7725e-01, -3.3923e-01,  1.1110e-01,  1.2157e-01,\n",
       "          -5.2017e-01, -1.2595e-01,  3.2368e-02, -3.7792e-02,  5.6994e-01,\n",
       "           2.1239e-01,  4.4888e-01,  6.0452e-02,  1.2190e-01, -4.2370e-01,\n",
       "          -4.8563e-01, -6.1457e-02, -1.5502e-01, -5.1597e-01, -2.6759e-01,\n",
       "          -6.4046e-02,  5.5104e-02, -2.5323e-01, -5.4375e-01, -3.2244e-01,\n",
       "           1.4182e-01, -3.1462e-02,  1.2303e-01, -2.0925e-01,  5.6036e-01,\n",
       "          -9.1892e-01, -1.3865e-01, -3.2977e-01, -5.8584e-02,  1.5673e-01,\n",
       "           2.6501e-01, -4.1691e-01, -3.1252e-01, -1.2731e-01, -8.9877e-01,\n",
       "          -8.0599e-01,  4.6310e-02,  7.4018e-01, -1.0141e-01,  9.8598e-02,\n",
       "           1.3335e-01,  9.5394e-02, -3.9534e-01,  3.2461e-01, -1.9912e-01,\n",
       "           2.5577e-01,  4.6972e-01, -4.6662e-01,  4.5361e-02, -4.3112e-02,\n",
       "          -7.4609e-02, -1.7800e-01, -7.5994e-02, -3.7163e-01, -6.2177e-01,\n",
       "          -2.7218e-01,  2.6677e-02, -1.4632e-01,  3.3523e-01,  2.6146e-02,\n",
       "           1.3773e-01, -2.4567e-01,  5.7964e-02,  8.4792e-01,  6.0189e-01,\n",
       "           2.9340e-01, -2.4105e-01, -7.3028e-01,  7.9905e-03,  8.0459e-02,\n",
       "          -5.8068e-02,  3.1108e-01, -5.8308e-02, -1.8405e-01, -1.3325e-01,\n",
       "          -1.1517e-01,  9.9034e-01, -2.4143e-01, -1.4203e-01, -2.7258e-02,\n",
       "           8.2295e-01, -5.7484e-01, -5.9305e-01,  4.5987e-01, -7.3865e-02,\n",
       "          -5.7696e-01, -2.6757e-01,  4.5631e-01, -3.1296e-01,  1.5606e-01,\n",
       "           4.7921e-01,  1.1156e-01, -1.9008e-01,  2.0635e-01,  7.1951e-03,\n",
       "          -1.8519e-01,  2.7335e-01, -1.5825e-02, -6.4675e-02,  3.2909e-01,\n",
       "          -4.2979e-02,  4.1402e-02, -2.0500e-01, -5.2587e-01, -1.5853e-01,\n",
       "           9.3770e-02,  9.6391e-01, -2.8928e-01, -3.7168e-01,  6.8675e-01,\n",
       "          -2.1419e-01,  3.2345e-01,  5.2334e-01, -1.1961e-01,  5.2370e-02,\n",
       "           6.3989e-01,  2.2575e-01, -7.7426e-01,  1.3355e-01,  3.8001e-01,\n",
       "          -1.7476e-01, -5.1716e-01, -3.5556e-01, -2.6843e-01, -9.0906e-02,\n",
       "          -3.1223e-01,  4.5868e-01,  1.5797e-01,  4.1484e-01, -1.8044e-01,\n",
       "          -7.5125e-01,  2.9581e-01,  2.7307e-01,  6.3136e-01, -5.3841e-02,\n",
       "          -3.1481e-01, -3.8860e-01, -3.3587e-01, -1.4710e-01,  3.6450e-01,\n",
       "           1.9902e-01, -9.8030e-01,  2.5868e-02,  5.2103e-01, -1.3043e-01,\n",
       "           4.2693e-01, -2.3898e-01,  2.0990e-01,  4.7655e-01, -3.3392e-02,\n",
       "          -2.5199e-01,  1.4633e-01, -4.5813e-02, -3.9709e-01,  3.9194e-01,\n",
       "           5.2896e-01,  3.2633e-02, -9.6709e-03, -3.7031e-02, -3.7811e-01,\n",
       "          -2.4963e-01,  1.4607e-02, -7.6037e-02, -4.3888e-01, -4.3815e-02,\n",
       "          -3.7991e-01, -5.2297e-01, -9.5728e-02, -2.1129e-02,  7.9186e-02,\n",
       "           3.5542e-01,  2.6620e-01,  3.1958e-01, -1.4079e-01, -6.3270e-02,\n",
       "          -2.9629e-01,  4.8844e-01, -2.4669e-01,  8.5712e-01,  1.3263e-01,\n",
       "          -2.0121e-01,  2.0207e-01,  1.6567e-01, -3.4834e-01,  1.1668e-01,\n",
       "          -2.1062e-01,  2.7849e-01, -5.5939e-01, -2.2947e-01, -1.4995e-01,\n",
       "          -1.1785e-01, -2.5136e-01, -4.5111e-01,  7.0217e-03,  9.6189e-02,\n",
       "           2.1270e-02,  7.3320e-01, -3.4671e-01,  5.9496e-01, -1.2969e-01,\n",
       "          -1.0662e-01, -9.3499e-01,  3.5159e-01,  3.5436e-01, -1.9950e-01,\n",
       "          -7.2710e-01, -1.9952e-01, -1.1804e-01, -4.6280e-01,  7.6148e-01,\n",
       "           6.1392e-01, -7.6495e-01, -5.6779e-01, -9.0494e-02, -9.1732e-02,\n",
       "          -6.1294e-02,  8.4652e-02, -3.4115e-01, -6.5393e-02, -1.8825e-01,\n",
       "          -9.8893e-01,  2.4605e-01, -1.1326e-01,  5.6495e-01, -2.4930e-01,\n",
       "           6.7239e-01, -1.4412e-01, -1.2005e-01,  1.0570e+00,  2.1607e-01,\n",
       "          -1.0211e-01,  6.0083e-01, -2.0121e-01, -7.8225e-01, -2.2944e-01,\n",
       "          -3.5991e-02, -2.8073e-01,  1.3715e-01, -5.8237e-01,  2.9603e-01,\n",
       "           3.5317e-01,  6.1116e-02,  4.5349e-01]], device='mps:0'),\n",
       " tensor([[-0.1672, -0.5249, -0.4540,  ...,  0.3771,  0.2016,  0.5340],\n",
       "         [-0.3292, -0.2590, -0.5016,  ..., -0.0148,  0.3509,  0.1607]],\n",
       "        device='mps:0'),\n",
       " tensor([[-2.1428e-01, -4.6304e-01, -6.0226e-01, -8.6245e-01, -1.2912e-01,\n",
       "          -5.7483e-02, -7.2137e-01,  6.0820e-01,  1.5343e-01,  5.5940e-01,\n",
       "           5.8560e-02,  4.7535e-01,  8.4764e-02, -1.3874e-01, -2.3861e-01,\n",
       "          -3.2028e-01,  1.4151e-01,  3.5338e-01,  6.8410e-01, -8.5543e-02,\n",
       "           1.2135e-01,  5.2795e-01, -1.3223e-01, -1.1121e-01, -7.1389e-01,\n",
       "           3.1484e-01, -8.6552e-02,  1.6783e-01,  2.1801e-01, -5.2886e-01,\n",
       "           1.0182e-01,  3.0299e-01, -6.9980e-02,  1.2869e+00, -5.2201e-01,\n",
       "           6.2602e-03, -2.8222e-02, -5.4752e-01, -2.5678e-01,  8.2743e-02,\n",
       "           5.9293e-01, -1.6014e-01,  3.0883e-01,  3.2941e-01, -1.8708e-01,\n",
       "           1.7853e-01,  1.3944e-01, -2.5515e-01,  1.9385e-01, -6.8852e-01,\n",
       "          -1.4266e-01, -2.0075e-01,  5.2545e-01,  6.8220e-01, -8.2848e-01,\n",
       "          -1.6025e-01,  2.4301e-01, -4.2233e-02, -1.8426e-01,  1.0217e-01,\n",
       "           6.2543e-02,  3.8665e-01, -3.4417e-01,  2.8762e-01,  3.5981e-01,\n",
       "           1.8542e-02, -2.5932e-01, -3.9333e-01, -6.1644e-01,  1.5318e-02,\n",
       "          -7.8782e-02, -1.7713e-02,  3.7853e-01, -1.3523e-01,  4.4894e-01,\n",
       "          -8.2229e-01,  8.7743e-03, -2.0194e-01, -1.3997e-01,  1.5729e-01,\n",
       "          -6.5853e-02,  2.1279e-01,  2.7503e-01,  2.4115e-01,  1.2334e-01,\n",
       "          -1.0445e-01,  3.4167e-01, -8.0665e-02,  2.4802e-02,  4.8506e-01,\n",
       "          -2.2436e-01, -9.7363e-02, -2.5658e-01, -3.9750e-01,  3.2020e-01,\n",
       "           1.8333e-02,  2.9975e-01, -1.6075e-01,  2.9202e-01, -4.6087e-01,\n",
       "           3.7702e-02,  1.3259e-01, -2.2855e-01,  5.9943e-02, -4.7864e-01,\n",
       "          -3.6210e-01, -3.6125e-02, -2.6244e-01, -3.7540e-01,  2.5262e-01,\n",
       "          -1.3188e-01, -3.7669e-01, -9.1665e-02, -4.8708e-01,  1.2301e-01,\n",
       "          -5.7559e-02,  3.4175e-01, -1.4204e-01, -3.5765e-01, -1.2174e-01,\n",
       "           3.0066e-01, -2.6226e-01,  1.7729e-01,  1.9696e-01,  8.3961e-03,\n",
       "          -5.9714e-02,  6.5854e-01,  1.6283e-02,  5.8174e-02,  2.4064e-01,\n",
       "           1.6383e-01, -1.5761e-01, -3.5365e-01, -2.1838e-01,  7.2090e-02,\n",
       "          -2.0142e-02,  1.1417e-01, -2.2350e-01, -1.8440e-01, -1.9552e-01,\n",
       "           3.2831e-01, -3.8148e-01, -7.3583e-01, -5.2855e-01, -1.6312e-01,\n",
       "          -1.5302e-01, -8.8458e-02, -1.5035e-01, -1.8632e-01, -3.5559e-01,\n",
       "          -5.0672e-01, -3.4680e-01,  9.0192e-01,  9.6753e-03, -3.6184e-01,\n",
       "          -2.5239e-01,  3.9113e-02,  3.5390e-01, -1.1306e-01,  3.9913e-01,\n",
       "           7.0255e-01, -2.3614e-01,  1.4010e-01, -4.8793e-01, -4.0324e-01,\n",
       "           2.5649e-02,  1.4548e-01, -4.0725e-01,  5.8930e-01, -6.6257e-02,\n",
       "          -3.4239e-02, -1.6170e-01,  3.6137e-01, -3.6273e-01, -1.1464e-01,\n",
       "          -5.3731e-01,  2.8337e-01,  3.3102e-01, -3.7572e-01,  6.9872e-04,\n",
       "           2.1670e-01, -1.1249e+00,  8.2914e-02, -7.7449e-01,  3.4764e-01,\n",
       "          -6.5806e-02, -6.1831e-02,  1.8922e-01,  3.5135e-01, -9.1391e-03,\n",
       "           4.4689e-01,  3.1136e-01,  2.3280e-01,  3.2164e-01, -1.2703e-01,\n",
       "           2.1965e-01,  3.2291e-01,  5.3628e-01,  4.7929e-01,  2.3282e-01,\n",
       "          -3.8256e-01,  2.5821e-01,  4.2981e-01,  2.0540e-01,  1.7850e-02,\n",
       "           5.0731e-02, -2.8605e-01,  1.7270e-01,  2.4311e-02, -1.4891e-01,\n",
       "          -1.0208e-01,  6.4905e-02, -4.9560e-01,  1.0395e-01, -5.0843e-01,\n",
       "           8.2774e-02, -5.1396e-01, -1.9048e-01,  3.1099e-01,  1.6981e-01,\n",
       "           7.2414e-02,  1.8958e-01,  7.6842e-02, -2.5865e-01,  1.6483e-01,\n",
       "           2.4488e-01, -3.4713e-01,  4.3987e-03, -3.6783e-01, -3.0241e-01,\n",
       "           3.1041e-01, -4.5224e-01,  6.5660e-01,  3.3129e-01,  1.3460e-01,\n",
       "           3.0637e-01, -2.8168e-01,  6.9004e-02, -5.4571e-01,  2.8282e-01,\n",
       "          -2.8113e-01, -2.5194e-01, -4.5118e-01,  3.0442e-02,  4.0773e-01,\n",
       "           2.7572e-01,  3.2304e-02,  7.5658e-01,  4.8197e-01,  5.0615e-01,\n",
       "          -2.9445e-01,  7.2967e-02,  4.8133e-01,  3.2189e-02, -5.1304e-01,\n",
       "           1.6526e-01, -2.4754e-02, -5.0285e-01,  2.0693e-01, -3.2379e-01,\n",
       "          -2.7975e-01, -3.5164e-01,  4.5878e-01,  1.9159e-01, -2.3987e-01,\n",
       "          -4.8394e-01,  4.3540e-01, -5.4785e-02, -4.0376e-01,  4.0913e-01,\n",
       "          -3.1471e-01,  1.0174e-01,  1.2366e-01, -2.1547e-01, -6.9059e-01,\n",
       "          -3.8851e-01,  1.5111e-01,  3.1103e-01, -6.3791e-02, -6.9181e-01,\n",
       "          -3.0950e-01,  1.1535e-01,  2.6901e-01,  1.2915e-01,  3.3375e-01,\n",
       "           5.0176e-01, -4.1352e-01, -2.2396e-01,  1.6950e-01, -3.0506e-01,\n",
       "          -2.0975e-01, -1.0372e-01, -1.5325e-01,  2.4864e-01,  2.6606e-01,\n",
       "           1.8939e-02,  2.0752e-01,  4.4019e-01, -1.8740e-01, -3.8957e-01,\n",
       "          -1.5907e-01,  2.4763e-01, -2.0147e-02, -3.3015e-02,  2.9910e-01,\n",
       "           1.8917e-02,  5.3926e-01,  7.2600e-01,  2.8633e-02, -6.2766e-01,\n",
       "           1.2612e-01,  3.5168e-01, -3.6703e-01, -1.1328e-01,  6.6088e-02,\n",
       "          -7.6485e-01, -8.7589e-02, -1.4349e-01, -2.4346e-01, -6.4515e-01,\n",
       "           8.3794e-02,  4.5887e-01, -5.3197e-02, -1.3335e-01, -3.7209e-01,\n",
       "           1.0595e-01,  7.3932e-02, -1.0391e-01,  4.7521e-01, -6.8557e-01,\n",
       "          -2.1753e-01, -3.0296e-01, -2.4104e-01, -6.6678e-01, -5.0683e-01,\n",
       "           4.6987e-01,  1.8278e-01,  1.8438e-01, -1.7562e-01, -4.8127e-01,\n",
       "           4.3590e-01, -5.6693e-01, -1.2143e-01,  2.5726e-01, -3.3640e-01,\n",
       "           1.7891e-01,  4.8011e-01,  3.3950e-01,  1.8893e-01,  2.3249e-01,\n",
       "           1.5596e-01, -1.9976e-01, -4.7932e-01, -2.6261e-01,  5.4734e-01,\n",
       "           3.5794e-01,  3.6781e-01, -8.0499e-02,  3.8106e-01,  3.5050e-01,\n",
       "          -2.7070e-01,  5.4636e-01,  2.8723e-03,  1.2758e-01, -7.9144e-02,\n",
       "           2.1310e-01,  3.2451e-01, -3.5524e-01,  1.4103e-01,  1.1267e-01,\n",
       "          -3.8518e-01, -3.7926e-01,  1.1588e-01,  1.5864e-01, -1.7886e-01,\n",
       "          -3.8864e-01, -2.3529e-01,  1.4944e-01, -3.7024e-01, -4.5368e-01,\n",
       "           3.2418e-01, -1.3036e-01,  3.8865e-02, -5.2801e-01, -3.6243e-01,\n",
       "          -9.4692e-02, -2.3962e-01,  6.5581e-01,  2.0142e-01, -2.0718e-01,\n",
       "           2.0176e-01, -2.5030e-01, -3.8638e-01,  5.7391e-01,  6.9535e-02,\n",
       "          -5.8044e-01,  5.0094e-01, -2.7802e-01,  1.6906e-02,  6.4205e-01,\n",
       "          -2.9221e-01, -1.6703e-01,  1.2033e-01,  6.6737e-02,  2.6250e-01,\n",
       "          -3.7788e-01,  1.1051e-01, -1.2816e-01, -3.5933e-01,  3.7737e-01,\n",
       "           2.9999e-01,  9.0089e-02,  1.6393e-02, -1.4371e-01,  1.5188e-01,\n",
       "           3.2675e-01, -8.6354e-02, -2.8780e-01, -4.2978e-01, -4.9046e-01,\n",
       "           2.6547e-01,  8.3203e-03,  5.6817e-01,  4.3292e-01, -1.1338e+01,\n",
       "           1.7135e-01,  7.8132e-01, -2.2592e-01, -2.1613e-02, -6.7641e-02,\n",
       "           9.5591e-04, -8.7411e-02,  5.4393e-01, -2.9185e-01, -3.7458e-01,\n",
       "          -3.6948e-01, -2.1535e-01,  1.0600e-01, -5.4939e-01, -2.4387e-01,\n",
       "           1.4022e-01, -6.3944e-01,  3.7958e-01, -3.5501e-01,  5.3722e-01,\n",
       "          -1.3738e-01, -3.6763e-01,  2.4554e-01, -3.9900e-01,  2.0619e-01,\n",
       "           4.1831e-02, -3.2975e-02,  3.4585e-02,  3.6884e-01, -3.6430e-02,\n",
       "           2.5146e-05, -1.0966e-01, -5.2364e-02, -2.6173e-01, -5.6603e-01,\n",
       "          -8.1707e-02, -9.8585e-02,  1.6140e-01,  8.7934e-02, -6.3779e-01,\n",
       "          -2.1041e-01, -1.9558e-01,  3.0270e-01,  4.7164e-01,  8.0925e-01,\n",
       "           1.5508e-01,  8.9130e-01,  2.6148e-01,  1.3292e-01, -2.2298e-02,\n",
       "          -6.7888e-01, -9.1548e-03, -2.2088e-01,  5.4206e-02,  8.5933e-02,\n",
       "           2.1174e-01, -8.2170e-01,  4.0946e-01, -4.2569e-01,  1.2735e-01,\n",
       "           5.1140e-01,  3.3094e-02,  2.1213e-01, -9.7305e-02,  1.2035e-01,\n",
       "           2.1069e-01, -4.0514e-02, -1.9188e-02, -3.1068e-01, -3.1422e-02,\n",
       "          -1.3687e-01,  5.3603e-01,  4.0176e-01, -4.9771e-01,  3.9253e-01,\n",
       "           4.5616e-02, -1.7747e-01,  1.7428e+00, -1.5386e-02,  1.1326e-01,\n",
       "          -2.2217e-02,  2.5099e-01, -4.3465e-01, -6.0920e-01,  4.3427e-01,\n",
       "          -2.0288e-01, -2.1842e-02,  9.0202e-02,  4.1744e-01, -8.6282e-02,\n",
       "          -3.8784e-01, -5.4755e-02,  9.3908e-02, -1.2271e-01,  4.6021e-01,\n",
       "          -7.8547e-01, -7.9199e-02, -2.5631e-01,  3.6829e-01, -6.4138e-01,\n",
       "          -2.6593e-02, -6.2716e-01, -2.4110e-01,  4.0700e-01, -5.7576e-02,\n",
       "           2.3343e-01,  8.0581e-02, -9.0688e-02,  4.2352e-01, -1.0327e-01,\n",
       "          -6.2373e-02,  3.4294e-01,  1.1820e-01, -1.4864e-01, -1.7376e-01,\n",
       "           5.8752e-01,  3.0245e-01, -4.6349e-01, -5.7227e-01, -3.5248e-02,\n",
       "           2.3541e-01,  2.1669e-01,  4.5424e-01, -6.0876e-01,  5.6510e-01,\n",
       "          -9.5120e-01, -1.9399e-01,  2.6580e-01, -3.0090e-02,  4.1343e-01,\n",
       "           4.0114e-01,  7.2082e-02,  1.1682e-01, -7.5717e-01, -7.1291e-01,\n",
       "          -7.6576e-01, -1.3369e-01,  5.0864e-01, -5.4733e-01, -4.5594e-02,\n",
       "          -1.2579e-01, -4.7622e-02, -1.9596e-02, -1.3762e-01,  4.7523e-02,\n",
       "           1.8685e-01,  1.8771e-01, -2.3898e-01, -3.0205e-02, -7.4963e-02,\n",
       "          -4.6358e-01, -1.8491e-01, -5.6194e-02, -2.4988e-01, -5.7585e-01,\n",
       "          -1.0491e-01, -1.8886e-01, -2.0614e-01,  6.5077e-02,  3.5719e-01,\n",
       "           4.4115e-02,  1.9234e-02, -4.1817e-01,  8.4708e-01,  5.6776e-01,\n",
       "          -2.0947e-02, -2.4130e-01, -5.9824e-01, -1.2950e-01,  2.1301e-01,\n",
       "           2.6551e-01,  7.8602e-02, -1.1793e-01, -2.7361e-02, -5.1194e-01,\n",
       "          -5.1687e-03,  6.7920e-01, -9.8279e-02, -7.7002e-01,  1.8635e-01,\n",
       "           5.9155e-01, -1.6688e-01, -2.6021e-01,  7.2578e-02, -2.2754e-01,\n",
       "          -2.0489e-01, -4.7234e-01,  7.9726e-01, -1.1101e-01,  1.4213e-02,\n",
       "           2.7818e-01,  9.8228e-02, -7.9742e-02,  4.9572e-01, -1.1436e-01,\n",
       "          -3.9446e-01,  7.3313e-02, -6.9074e-01,  5.0796e-01,  1.6727e-03,\n",
       "          -4.3511e-01,  8.9881e-01,  2.2767e-01,  2.1836e-02, -2.3823e-01,\n",
       "           2.6335e-01,  6.6281e-01, -4.0820e-01,  1.1652e-01,  7.1162e-01,\n",
       "           4.0125e-02, -5.1415e-03,  2.7865e-02, -9.9275e-02,  8.6166e-03,\n",
       "           3.6497e-01, -2.2016e-01, -4.0130e-01, -1.4801e-01,  6.0868e-01,\n",
       "           5.3130e-01, -4.5236e-01, -7.6377e-01, -1.4272e-01,  6.6346e-01,\n",
       "          -3.5432e-01,  2.8854e-01,  4.4811e-02,  2.7632e-01, -5.0098e-02,\n",
       "          -7.2778e-01, -8.2344e-02, -7.4880e-02,  2.1927e-01, -1.0379e-01,\n",
       "          -2.8176e-02, -2.4004e-01, -3.0857e-01,  2.2252e-02,  5.7282e-01,\n",
       "           1.6107e-01, -6.8974e-01, -7.9529e-02,  6.2764e-01, -7.9667e-01,\n",
       "           5.9815e-01, -2.4122e-01, -1.4434e-01,  2.4667e-01,  1.4368e-01,\n",
       "           1.9957e-01,  5.1885e-02, -4.3656e-01, -3.9333e-01,  2.2229e-01,\n",
       "           9.6937e-01, -1.0492e-01,  5.8533e-02, -5.4956e-01, -2.1251e-01,\n",
       "          -3.4972e-02, -3.1622e-01, -3.5043e-01,  3.8746e-01,  1.5761e-01,\n",
       "          -1.1143e-01, -4.3634e-02, -1.3781e-01,  3.3610e-01,  1.5341e-01,\n",
       "           2.5290e-01,  4.0495e-01,  6.2447e-01,  1.2754e-01, -2.1692e-01,\n",
       "          -9.5806e-02,  3.4124e-01, -2.5732e-01,  5.6458e-01, -1.1250e-01,\n",
       "          -3.5087e-01, -9.5994e-02,  4.0499e-01, -5.7023e-01, -4.9129e-02,\n",
       "          -4.1155e-01,  2.8773e-01, -2.9653e-01,  1.0916e-01,  2.1131e-01,\n",
       "          -9.8838e-02,  4.1761e-02, -2.6606e-01,  5.4474e-01,  4.1066e-01,\n",
       "           4.1814e-01,  4.7646e-01,  2.5447e-02,  1.3216e-02, -4.8603e-01,\n",
       "           2.0912e-01, -1.1070e+00,  1.2480e-01, -9.7393e-03, -1.5381e-01,\n",
       "          -5.0934e-01, -1.2143e-01,  8.9973e-02,  3.9806e-01,  5.2069e-01,\n",
       "           7.0266e-01, -2.7664e-01, -5.1060e-01, -1.7154e-01,  1.0853e-01,\n",
       "           2.0677e-01, -1.5980e-01,  4.7214e-01,  1.1590e-01,  2.1906e-01,\n",
       "          -3.8513e-01, -4.1463e-01, -1.7415e-01,  3.1501e-01, -2.0280e-01,\n",
       "           5.5551e-01, -1.7613e-01, -1.2810e-01,  2.6468e-01,  2.1503e-01,\n",
       "           1.0100e-01,  6.2326e-01,  2.2550e-01, -7.0986e-01,  7.6456e-02,\n",
       "           1.6193e-02, -2.9241e-01,  4.0279e-03, -4.1249e-01,  1.4813e-01,\n",
       "           1.2812e-01,  4.5238e-01,  6.1597e-01]], device='mps:0'),\n",
       " tensor([[-0.1672, -0.5249, -0.4540,  ...,  0.3771,  0.2016,  0.5340],\n",
       "         [-0.3292, -0.2590, -0.5016,  ..., -0.0148,  0.3509,  0.1607]],\n",
       "        device='mps:0'),\n",
       " tensor([[-0.2974, -0.7044, -0.3158,  ...,  0.0674,  0.7277,  0.4297],\n",
       "         [-0.4058, -0.5658, -0.4077,  ...,  0.3815,  0.8412,  0.1457],\n",
       "         [-0.0308, -0.6415, -0.5830,  ...,  0.3218,  0.5289,  0.9014],\n",
       "         [-0.1656, -0.6272, -0.3633,  ...,  0.3083,  0.7874,  0.1520],\n",
       "         [-0.3486, -0.5854, -0.6592,  ...,  0.0641,  0.3814,  0.3555]],\n",
       "        device='mps:0'),\n",
       " tensor([[-2.6669e-01, -4.3706e-01, -4.7596e-01, -6.7304e-01, -1.3138e-01,\n",
       "           6.4374e-04, -6.0597e-01,  5.9090e-01,  6.7669e-03,  6.4348e-01,\n",
       "           1.3153e-01,  2.7624e-01,  1.3447e-02, -2.7649e-01, -5.7975e-01,\n",
       "          -5.0599e-01,  2.4398e-01,  5.8716e-01,  7.5801e-01,  3.8037e-01,\n",
       "           2.3905e-01,  6.1888e-01, -2.5811e-01, -1.7227e-01, -9.2400e-01,\n",
       "           3.5704e-01, -1.1709e-01,  3.4438e-01,  2.4135e-01, -5.9088e-01,\n",
       "           4.9509e-04,  3.3321e-01, -2.1608e-02,  1.2997e+00, -5.0299e-01,\n",
       "          -2.6962e-01,  1.0103e-01, -7.9520e-01, -1.3835e-01, -7.2011e-02,\n",
       "           7.8072e-01, -2.2671e-01, -5.0898e-02,  2.4541e-01,  9.0136e-02,\n",
       "          -6.5820e-02,  1.4531e-01, -3.5947e-01, -1.6407e-01, -6.3451e-01,\n",
       "           8.1026e-02,  2.8061e-02,  5.0322e-01,  6.3953e-01, -6.0556e-01,\n",
       "          -3.1544e-01,  4.4217e-02,  5.1450e-03, -1.3031e-01,  1.6744e-01,\n",
       "          -1.5517e-01,  2.2217e-01, -3.8968e-02,  1.1978e-01,  2.4615e-01,\n",
       "           2.1713e-01, -2.0339e-01, -4.9474e-01, -4.5195e-01, -3.0046e-03,\n",
       "           2.9934e-02,  2.4908e-01,  1.0440e-01, -3.2297e-01,  4.5519e-01,\n",
       "          -5.1012e-01, -7.7386e-02, -8.9061e-02, -4.8942e-01,  6.7425e-02,\n",
       "          -4.9286e-02,  3.0154e-01,  1.3302e-01, -1.6004e-01,  2.2767e-01,\n",
       "           1.7675e-01,  4.7197e-01,  3.8563e-02,  1.1828e-02,  5.7524e-02,\n",
       "           9.6185e-02, -2.4340e-01, -5.5609e-01, -1.9307e-01,  2.3140e-01,\n",
       "          -1.4339e-01,  1.6008e-01, -3.7226e-01, -1.8733e-02, -2.6769e-01,\n",
       "           1.1829e-02,  1.4753e-01, -1.9106e-01,  9.5430e-02, -4.1773e-01,\n",
       "          -5.4564e-01,  1.4099e-01, -1.5372e-01, -7.1785e-01,  2.0021e-01,\n",
       "          -1.0204e-01, -1.5055e-02,  1.0202e-01, -2.1675e-01, -3.1958e-01,\n",
       "          -2.0219e-01,  2.5920e-01, -1.1339e-02, -2.2266e-01,  3.2632e-01,\n",
       "           3.5906e-01, -3.0213e-01,  1.1375e-01,  9.2287e-02,  4.5701e-02,\n",
       "          -2.1335e-01,  6.0362e-01,  2.4028e-02,  4.3650e-01,  1.7077e-01,\n",
       "           1.0153e-01,  7.7920e-02, -4.5629e-01, -2.6933e-01,  1.2104e-01,\n",
       "           1.3717e-01,  2.6540e-01, -4.9232e-01,  2.4692e-02,  1.4262e-01,\n",
       "           1.2387e-01, -2.1885e-01, -5.6537e-01, -2.1133e-01, -1.1971e-01,\n",
       "          -2.5224e-01,  1.1765e-02, -2.2651e-01, -1.1036e-01, -2.6001e-01,\n",
       "          -6.5849e-01, -6.2506e-01,  5.7964e-01, -1.9432e-01, -1.3871e-01,\n",
       "          -2.2300e-01,  3.5561e-01,  4.9680e-02, -7.4630e-02,  3.6573e-01,\n",
       "           7.6619e-01,  5.6569e-02,  4.3075e-02, -4.2795e-01, -3.5467e-01,\n",
       "           2.2589e-01,  2.4553e-01, -3.5334e-01,  1.3043e-01,  2.0456e-01,\n",
       "           1.5307e-01, -1.0832e-02,  2.7463e-01, -2.2198e-01,  2.4185e-01,\n",
       "          -6.6922e-01,  6.6483e-01,  2.0678e-01, -2.7953e-01,  5.1875e-02,\n",
       "          -1.0417e-01, -1.2402e+00,  1.6376e-01, -5.8857e-01,  1.7824e-01,\n",
       "          -2.0540e-01,  3.5234e-01,  3.8504e-01,  2.1492e-01, -3.8152e-02,\n",
       "           2.7009e-01,  2.5973e-01,  1.5486e-01,  2.7364e-01, -5.6590e-02,\n",
       "           5.4587e-01,  4.4938e-01,  4.5008e-01,  2.7313e-01,  1.2965e-01,\n",
       "          -5.3941e-01,  3.9300e-02,  4.9174e-01, -6.9616e-02,  3.6034e-01,\n",
       "           1.5707e-01, -3.9769e-01,  6.1221e-02,  3.8432e-02, -4.7404e-02,\n",
       "          -2.0827e-01, -1.9611e-01, -5.9737e-01, -2.0864e-01, -3.5310e-01,\n",
       "           2.7306e-01, -3.6915e-01, -2.3400e-01,  4.6606e-01,  1.4711e-01,\n",
       "          -2.4735e-02,  2.2147e-01,  3.1945e-02, -3.3354e-01,  1.0792e-01,\n",
       "           2.8013e-01, -2.4420e-01, -1.5576e-02, -4.3904e-01, -1.5632e-01,\n",
       "           3.3399e-01, -4.3383e-01,  6.1239e-01,  2.8382e-01, -3.3190e-02,\n",
       "           2.0303e-02, -1.2578e-01,  2.8065e-02, -3.3142e-01,  3.7946e-01,\n",
       "          -3.2269e-01, -4.5179e-01, -4.3334e-01,  2.1128e-01,  1.2049e-01,\n",
       "           2.3977e-01, -2.4026e-01,  6.3917e-01,  5.0884e-01,  3.2966e-01,\n",
       "          -2.8186e-01, -6.5457e-02,  1.7808e-01,  9.9536e-03, -4.1537e-01,\n",
       "           8.7899e-02, -1.8099e-01, -6.1234e-01,  1.6878e-01, -4.1831e-01,\n",
       "           1.2814e-01, -5.9025e-02,  1.2498e-01,  1.6226e-01, -8.4770e-02,\n",
       "          -5.2078e-01,  4.5037e-01,  4.3352e-02, -3.8129e-01,  3.8771e-01,\n",
       "          -3.4068e-01,  1.8967e-01, -3.2728e-02, -4.5297e-01, -4.3795e-01,\n",
       "          -6.7658e-01,  1.5078e-01,  1.5339e-01, -1.0142e-03, -6.8293e-01,\n",
       "          -2.7571e-01,  4.1142e-01,  1.0819e-01,  8.2559e-02,  4.5421e-01,\n",
       "           5.0181e-01, -5.2925e-02, -5.9691e-02,  9.9037e-02, -4.2089e-01,\n",
       "          -3.2972e-01,  3.9187e-02, -2.7262e-01,  1.9374e-01,  3.2319e-01,\n",
       "           2.4645e-01,  1.0160e-01,  2.5225e-01, -1.3414e-01, -4.3414e-01,\n",
       "          -1.8845e-01,  1.9938e-01, -1.6081e-01, -8.2010e-02,  1.9281e-01,\n",
       "          -5.1205e-02,  6.0268e-01,  5.4264e-01, -1.6438e-01, -5.8651e-01,\n",
       "           2.0490e-01,  5.6551e-01, -1.8465e-01, -1.8872e-01,  1.0269e-01,\n",
       "          -9.6697e-01, -1.8496e-01,  6.4934e-02, -8.9887e-02, -5.7390e-01,\n",
       "           1.0109e-01,  3.4863e-01, -1.5122e-01, -1.4832e-01, -4.1673e-01,\n",
       "           3.4936e-01,  5.3502e-01, -2.7166e-02,  2.5459e-01, -3.1492e-01,\n",
       "          -3.9519e-01, -5.3522e-01, -8.9294e-02, -5.2208e-01, -5.0580e-01,\n",
       "           3.5540e-01,  2.7834e-01, -8.6759e-02, -1.2290e-01, -4.0563e-01,\n",
       "           3.5520e-01, -6.0802e-01, -8.6681e-02,  3.4036e-01, -4.8614e-01,\n",
       "          -8.3471e-02,  4.8270e-01, -2.6538e-01,  4.0032e-02,  2.1176e-01,\n",
       "           1.5432e-02, -2.6222e-01, -4.6079e-01, -8.0798e-02,  7.4098e-01,\n",
       "           3.6566e-01,  2.7895e-01, -2.5122e-01,  3.9871e-01,  4.0345e-01,\n",
       "          -3.0865e-01,  4.2468e-01, -8.6272e-02, -1.9507e-01, -4.0550e-01,\n",
       "           1.0274e-01,  2.9723e-01, -4.4318e-01,  1.3295e-01,  1.4606e-01,\n",
       "          -1.2874e-01, -1.7816e-01, -4.2518e-02,  1.0418e-01, -2.1360e-02,\n",
       "          -3.2653e-01, -1.2787e-01,  2.7417e-01, -2.3525e-01, -6.4287e-02,\n",
       "           7.0331e-02, -7.3733e-02,  2.2464e-01, -3.2135e-01, -4.2933e-01,\n",
       "          -3.0237e-01, -2.0227e-01,  8.8525e-01,  2.9675e-01, -2.5456e-01,\n",
       "           2.9469e-01, -2.3396e-01, -3.9108e-01,  2.4452e-01, -3.5424e-01,\n",
       "          -3.0656e-01,  4.2482e-01, -2.9462e-01, -1.0329e-01,  5.6171e-01,\n",
       "          -1.7964e-01, -1.4489e-01, -4.5861e-04,  1.5629e-01,  3.1616e-01,\n",
       "          -2.5958e-01,  1.3797e-01,  7.5066e-02, -2.9145e-01,  2.5187e-01,\n",
       "           3.7264e-01,  4.4885e-02, -2.0926e-01, -1.5341e-01, -3.0785e-01,\n",
       "           2.7671e-01, -1.4718e-01, -3.6931e-01, -5.0086e-01, -6.5042e-01,\n",
       "           3.4967e-01,  2.6955e-01,  3.8810e-01,  3.5468e-01, -1.1377e+01,\n",
       "           2.0013e-02,  5.0456e-01, -4.2194e-01,  2.0937e-03, -1.5111e-01,\n",
       "           1.2082e-01, -1.6535e-01,  4.3084e-01, -2.7223e-01, -3.2614e-01,\n",
       "          -6.9926e-01, -1.1167e-01,  1.2102e-01, -3.1484e-01, -1.9683e-01,\n",
       "          -1.1599e-01, -7.5337e-01,  3.1500e-01, -1.7982e-01,  1.7717e-01,\n",
       "          -2.7249e-01, -1.7803e-01,  1.5075e-01, -2.9849e-01, -2.8032e-02,\n",
       "           3.1434e-01, -1.1683e-01,  2.6480e-01,  3.9000e-01,  1.2891e-01,\n",
       "           6.1290e-02,  3.2118e-02,  5.6437e-02, -8.4044e-02, -5.0284e-01,\n",
       "          -2.5283e-01, -1.7709e-01, -2.0110e-02, -2.5312e-01, -5.8619e-01,\n",
       "          -3.7296e-01, -2.3237e-01,  3.9207e-01,  2.7948e-01,  9.5278e-01,\n",
       "           3.9367e-01,  8.1366e-01,  4.0400e-01,  1.3884e-01,  5.1258e-02,\n",
       "          -6.0169e-01,  2.3417e-02, -5.7934e-02, -4.7860e-01, -1.5314e-01,\n",
       "           3.9025e-01, -8.4697e-01,  4.0645e-01, -4.9992e-01, -1.6174e-01,\n",
       "           5.0095e-01,  1.5167e-01,  2.1177e-02, -1.5478e-01, -1.1835e-01,\n",
       "           2.9361e-02,  2.8784e-02,  9.4043e-02, -1.6661e-01, -2.7722e-01,\n",
       "          -1.2523e-01,  8.5651e-01, -1.5642e-03, -6.9428e-01, -8.4744e-02,\n",
       "           3.6425e-02, -2.6455e-01,  1.8203e+00,  4.2966e-02,  5.3375e-02,\n",
       "           1.0681e-01,  4.9972e-01, -8.7441e-01, -7.4027e-01,  1.6256e-01,\n",
       "          -2.2990e-01, -2.1354e-02,  2.7314e-01,  4.1352e-01, -1.5877e-01,\n",
       "          -7.9939e-01, -1.7606e-01,  1.3340e-01,  1.2208e-01,  8.1355e-01,\n",
       "          -6.7446e-01, -1.7792e-01,  1.4477e-01,  2.9987e-01, -7.1651e-01,\n",
       "           4.9153e-02, -4.7479e-01, -1.7856e-01,  4.4306e-01, -3.2465e-01,\n",
       "           4.9508e-02,  2.0475e-02, -2.0761e-01,  4.4174e-01, -3.3644e-01,\n",
       "           9.1400e-02, -6.2004e-02,  1.0473e-01, -1.6457e-01,  1.0357e-01,\n",
       "           5.4552e-01,  4.2707e-01, -3.5946e-01, -3.3354e-01, -1.5038e-01,\n",
       "           3.0716e-01,  1.8745e-01,  1.7077e-01, -2.5328e-01,  5.3390e-01,\n",
       "          -5.5036e-01, -1.4622e-01,  3.2583e-01,  4.8665e-02,  2.0818e-01,\n",
       "           1.5231e-01, -9.6376e-02,  1.7477e-01, -6.7863e-01, -7.5909e-01,\n",
       "          -8.0018e-01, -2.4358e-01,  1.5439e-01, -4.4855e-01, -1.4862e-01,\n",
       "          -2.3419e-01,  7.3974e-02, -5.1005e-02,  1.4870e-02, -2.5686e-02,\n",
       "           1.6097e-02,  1.6317e-01, -2.4016e-01,  1.0677e-01, -2.4245e-01,\n",
       "          -3.3148e-01,  1.2349e-02, -2.0928e-01, -9.6694e-02, -5.2250e-01,\n",
       "          -6.1963e-01, -3.7462e-02, -1.0307e-01,  1.8196e-01,  1.0656e-01,\n",
       "           1.0292e-01, -1.7712e-02, -4.3886e-01,  6.1135e-01,  5.1758e-01,\n",
       "           1.6445e-01, -3.3695e-01, -3.2353e-01, -1.0853e-01,  1.7682e-01,\n",
       "           2.0129e-01,  2.2317e-01, -1.2276e-02, -1.6320e-01, -5.0478e-01,\n",
       "          -9.8563e-02,  9.8739e-01,  1.3774e-01, -4.7945e-01,  2.5363e-01,\n",
       "           9.6138e-01, -1.3354e-01,  1.6805e-01, -2.9215e-02,  1.1194e-01,\n",
       "          -4.4008e-01, -7.0675e-01,  8.0018e-01, -3.6636e-02,  1.2877e-01,\n",
       "           2.9275e-01,  1.2451e-01, -7.7632e-03,  5.2487e-01,  1.2346e-02,\n",
       "          -3.8720e-01,  3.6415e-02, -4.3676e-01,  2.8684e-01,  4.6460e-01,\n",
       "          -2.2969e-01,  5.9022e-01,  1.9322e-01,  3.9133e-02, -1.9496e-01,\n",
       "           2.8252e-01,  9.0577e-01, -3.9320e-01, -2.3260e-01,  9.4466e-01,\n",
       "           1.8525e-01,  1.9796e-02,  1.4980e-01,  1.0566e-01,  9.9099e-02,\n",
       "           4.4159e-01, -1.1097e-01, -1.8993e-01, -4.2436e-02,  2.9372e-01,\n",
       "           5.2662e-01, -7.4004e-01, -7.0469e-01,  5.2254e-02,  6.6033e-01,\n",
       "          -2.6628e-01,  3.4612e-01, -1.2615e-02,  4.7841e-01,  7.4639e-01,\n",
       "          -9.3128e-01, -1.1187e-01, -3.5959e-01,  4.2181e-01,  9.3330e-02,\n",
       "          -3.0429e-02,  2.0069e-02, -2.4715e-01, -2.1373e-01,  5.5320e-01,\n",
       "           2.8602e-01, -1.0652e+00, -1.0496e-01,  6.4288e-01, -5.5859e-01,\n",
       "           3.2355e-01, -1.0464e-01,  9.4316e-02,  9.5133e-02,  1.8910e-01,\n",
       "           1.5151e-01,  9.9232e-02, -5.0697e-01, -3.8816e-01,  1.8828e-01,\n",
       "           9.9329e-01, -7.8510e-02,  3.1882e-02, -2.5602e-01, -1.4773e-01,\n",
       "          -2.4202e-02, -4.7863e-01, -2.7923e-01,  3.7689e-01,  2.1389e-02,\n",
       "          -7.8835e-02, -3.5621e-01, -2.7949e-01,  1.0702e-01,  6.3132e-02,\n",
       "           2.4612e-01,  5.1465e-01,  7.9325e-01,  2.3308e-02, -7.1828e-02,\n",
       "           1.0114e-01,  9.7149e-02, -1.3890e-01,  4.7611e-01,  5.3982e-02,\n",
       "          -4.0961e-01,  1.5278e-01,  1.0081e-01, -4.1020e-01, -1.5330e-03,\n",
       "          -4.2404e-01,  4.4031e-01,  2.5315e-03, -1.7649e-01,  2.8251e-01,\n",
       "          -1.4685e-01,  5.0563e-02, -1.5809e-01,  6.3118e-01,  4.4867e-01,\n",
       "           7.4203e-02,  6.3175e-01,  1.7669e-01,  2.0984e-01, -4.9467e-01,\n",
       "           1.4856e-01, -7.5069e-01, -2.0604e-01, -1.3657e-01, -1.3802e-01,\n",
       "          -3.1078e-01, -4.1866e-02,  1.9206e-02,  2.2115e-01,  5.3296e-01,\n",
       "           8.6237e-01, -2.9525e-01, -6.2137e-01, -3.1109e-01,  9.9329e-02,\n",
       "           1.2273e-01, -8.8691e-02,  4.1858e-01, -8.8168e-02,  4.0425e-01,\n",
       "          -4.9122e-01, -2.8049e-01, -3.0361e-02,  4.3579e-01, -2.2783e-01,\n",
       "           7.0317e-01, -8.0943e-04, -3.5070e-02,  8.6417e-02,  3.1666e-01,\n",
       "           6.4023e-02,  8.4298e-01,  2.8706e-01, -4.1373e-01,  1.4737e-01,\n",
       "           1.7830e-02, -3.3744e-01,  1.9099e-01, -5.7530e-01,  2.2969e-01,\n",
       "           1.4163e-01,  5.5768e-01,  7.2432e-01]], device='mps:0'),\n",
       " tensor([[-3.6355e-01, -5.7639e-01, -4.1457e-01, -5.0283e-01, -6.0284e-02,\n",
       "          -5.9142e-01, -2.9968e-01,  7.0318e-01, -2.3932e-01,  1.0118e+00,\n",
       "           1.6173e-01,  7.3109e-02,  1.0005e-01,  4.7311e-01, -4.8394e-01,\n",
       "          -1.6809e-01, -3.6919e-01,  2.7615e-01,  6.8393e-01,  1.7351e-01,\n",
       "           7.0548e-01,  5.9816e-01, -1.0212e-02, -4.0329e-02, -4.2510e-01,\n",
       "           4.0677e-01,  3.2944e-02, -8.7254e-02,  1.3014e-01, -4.7322e-01,\n",
       "           3.5176e-01,  2.7871e-01,  1.0769e-01,  1.0955e+00,  5.2976e-02,\n",
       "          -4.6911e-01,  5.1179e-01, -4.0847e-01, -1.2514e-01, -1.6568e-01,\n",
       "           5.6654e-01, -2.4748e-01,  3.2461e-01, -4.2481e-02,  5.6546e-02,\n",
       "           1.3156e-01, -1.7127e-01, -9.7535e-02,  4.1906e-01, -9.0828e-01,\n",
       "          -2.6304e-01, -5.1435e-02,  3.9086e-01,  6.1185e-01, -6.9639e-01,\n",
       "          -3.8863e-01,  3.8993e-01, -1.4479e-01, -1.4661e-01,  5.2356e-02,\n",
       "           1.0030e-01,  1.6524e-01, -1.9967e-01,  1.6365e-01,  1.6468e-01,\n",
       "          -1.1189e-01, -5.9249e-01, -4.2536e-01, -1.9803e-01, -1.6094e-01,\n",
       "          -1.3677e-01,  3.1963e-02,  4.1627e-01, -2.9232e-01,  8.0891e-02,\n",
       "           2.4143e-02,  1.5719e-01, -9.4937e-02,  9.4487e-02,  8.4795e-02,\n",
       "           9.8399e-02, -7.1774e-02,  1.7304e-01, -8.8617e-02, -8.5615e-03,\n",
       "          -5.2789e-03,  7.0553e-01, -2.3303e-01, -3.6160e-02,  2.2880e-01,\n",
       "           2.9545e-01,  1.1454e-02, -4.4803e-01, -2.8487e-01,  5.2200e-01,\n",
       "          -4.1697e-01,  5.1858e-01, -2.2271e-01,  1.0827e-01, -1.7546e-01,\n",
       "          -7.2692e-02,  1.9464e-01, -2.9051e-01,  6.4835e-02, -3.1452e-01,\n",
       "          -1.9899e-01, -2.5383e-02, -3.6231e-01, -4.6059e-01,  3.9690e-01,\n",
       "           3.8400e-02,  1.9193e-02, -1.1212e-01, -3.8796e-01, -1.0809e-01,\n",
       "          -1.0892e-02,  2.5472e-01, -9.5143e-02, -4.2694e-01,  3.3498e-01,\n",
       "           5.0107e-01, -5.0206e-02,  3.4072e-02,  1.9349e-02,  2.5892e-01,\n",
       "          -7.8038e-02,  7.1748e-01, -1.3444e-01,  2.4802e-01,  5.8337e-01,\n",
       "           9.7632e-02,  1.7167e-01, -4.0961e-01,  1.7680e-01,  1.5526e-01,\n",
       "          -1.0146e-01, -3.1777e-01, -2.1626e-01, -3.2992e-01, -1.6378e-01,\n",
       "           3.4069e-02, -2.7105e-01, -5.6292e-01, -4.9561e-01, -5.6035e-02,\n",
       "          -1.8110e-01,  1.4349e-01, -2.3207e-01,  3.2760e-01, -3.9283e-01,\n",
       "          -5.3651e-01, -2.9925e-01,  7.3020e-01,  6.7211e-04,  9.5871e-02,\n",
       "          -2.7969e-01,  5.7788e-01,  8.9136e-02,  3.2768e-01, -7.2801e-02,\n",
       "           9.8230e-01,  5.7133e-02, -4.7675e-02, -7.4918e-01, -4.7643e-02,\n",
       "          -5.2528e-02,  1.3923e-01, -4.9983e-01,  2.5507e-01,  7.6985e-02,\n",
       "           1.5098e-01, -2.0293e-01, -9.0335e-03, -1.3282e-01,  2.0459e-01,\n",
       "          -1.7443e-01,  3.1851e-01,  2.4223e-01, -1.4078e-01, -2.0150e-01,\n",
       "           2.4757e-01, -1.2566e+00,  2.6479e-01, -3.6115e-01, -2.3872e-01,\n",
       "          -1.3641e-01,  1.4451e-01, -5.5870e-02,  2.2890e-02,  3.8009e-02,\n",
       "          -2.6097e-01,  3.6324e-02, -4.5817e-01,  2.0586e-01, -2.0680e-01,\n",
       "           1.0416e-01,  4.2564e-01,  6.7318e-01,  3.2049e-01,  1.6586e-01,\n",
       "          -4.8888e-03,  1.4397e-01, -2.4480e-01,  1.3897e-01,  1.0399e-01,\n",
       "           3.8237e-01, -6.5473e-01, -9.8087e-02,  2.6900e-01,  2.2620e-01,\n",
       "          -3.7550e-01, -3.0692e-02, -4.4326e-01, -1.9498e-01, -2.7130e-01,\n",
       "           5.9519e-02, -5.0578e-01, -1.4336e-01,  5.2758e-01,  1.5470e-01,\n",
       "          -2.8267e-02,  7.2433e-02,  2.4700e-01, -2.4854e-01,  1.8642e-01,\n",
       "           5.7958e-01, -3.2799e-01, -2.6224e-01, -1.2151e-01,  4.6325e-02,\n",
       "          -1.5202e-01, -8.6695e-01,  5.4390e-01,  4.5353e-01,  2.9829e-02,\n",
       "          -8.7949e-02, -6.0956e-01,  4.1739e-01,  1.4364e-01,  2.0726e-01,\n",
       "           2.3091e-02, -5.1714e-01, -5.1283e-01,  2.3668e-01, -1.0143e-01,\n",
       "           4.5889e-01, -3.7359e-02,  3.8348e-01,  7.7194e-01,  5.6539e-01,\n",
       "          -2.9230e-01,  1.5241e-01,  5.2744e-01,  3.0603e-01, -5.8258e-01,\n",
       "          -2.6138e-01,  1.5836e-01, -6.8704e-01,  1.0400e-01, -3.0415e-01,\n",
       "          -2.7689e-01, -1.2015e-01,  1.8692e-01,  1.9290e-01,  1.1260e-02,\n",
       "           1.9354e-01,  1.1238e-01, -8.9099e-02, -3.4396e-01,  1.0924e-01,\n",
       "          -4.0331e-01,  3.9422e-01, -2.3491e-01, -8.9328e-02, -7.1923e-01,\n",
       "          -4.6336e-01, -3.4141e-02,  2.8713e-01, -4.3271e-02, -7.4487e-01,\n",
       "           8.1011e-02,  2.4181e-02, -2.1292e-01, -1.2444e-01, -3.8278e-01,\n",
       "           5.7042e-01, -1.6574e-01, -5.3670e-02, -3.1788e-01, -1.8751e-01,\n",
       "          -5.9294e-01,  3.7134e-01, -1.7055e-01, -4.5362e-02,  3.1057e-01,\n",
       "           1.4260e-01,  1.5375e-01,  5.7561e-01, -1.3875e-01, -3.7599e-01,\n",
       "           1.9233e-01,  3.6105e-01, -9.7490e-02,  3.4068e-02,  4.3382e-01,\n",
       "           5.8086e-01,  2.4874e-01,  4.6015e-01, -3.3827e-01, -2.1499e-01,\n",
       "          -4.5326e-02,  5.1555e-01, -1.1089e-01, -5.9192e-01,  1.3044e-01,\n",
       "          -1.0595e+00, -2.7029e-01, -6.3880e-02,  1.0124e-02, -3.8105e-01,\n",
       "          -3.7074e-02,  6.1840e-01, -1.1614e-01,  6.9434e-02, -5.5580e-02,\n",
       "           4.0490e-01,  8.3382e-01, -1.2020e-01,  1.6949e-01, -6.8952e-01,\n",
       "          -5.5152e-02, -1.7167e-01, -5.0060e-01, -3.1424e-01, -1.7019e-02,\n",
       "           2.9608e-01,  2.5314e-01,  4.4125e-01, -2.5558e-01, -4.2526e-01,\n",
       "           4.5961e-01, -3.6450e-01, -5.7582e-01,  3.0228e-01,  2.8991e-02,\n",
       "           9.7237e-02,  3.9841e-01, -6.8601e-02,  3.7154e-02,  6.2877e-01,\n",
       "          -7.4783e-02, -2.1692e-02, -3.9929e-01, -4.4706e-01,  6.4240e-01,\n",
       "           4.6178e-01,  1.5984e-01, -2.2428e-01,  6.7277e-01,  3.8775e-01,\n",
       "          -3.8551e-01,  2.6219e-01, -5.8901e-01, -4.4882e-01,  2.3775e-01,\n",
       "           2.5052e-01,  2.5837e-01, -2.4308e-01, -1.0043e-01, -5.8987e-03,\n",
       "          -7.1583e-02, -4.5932e-01,  1.0813e-01, -6.9623e-02,  1.5392e-01,\n",
       "          -1.4782e-01, -2.0250e-01,  1.8869e-01, -7.4750e-01, -2.7405e-01,\n",
       "          -2.9653e-01, -9.8162e-02,  6.5668e-02, -3.3928e-01, -1.6678e-01,\n",
       "          -3.9604e-02, -5.9445e-03,  7.9989e-01, -9.8573e-03,  5.3335e-02,\n",
       "           3.5260e-01, -7.7638e-02, -2.3873e-01,  8.0848e-01,  6.6808e-02,\n",
       "          -2.3597e-01,  4.8079e-01, -3.1659e-01, -1.9737e-02,  3.4057e-01,\n",
       "          -4.5335e-01,  2.9726e-01, -5.5556e-02,  1.8911e-01,  2.2718e-01,\n",
       "          -3.1041e-01, -1.4550e-01,  4.5487e-01, -6.6704e-01, -3.5466e-02,\n",
       "           2.9962e-01,  6.8267e-01, -6.0315e-01, -1.2318e-03,  1.7526e-02,\n",
       "           1.8750e-01, -8.5368e-02, -1.6232e-01,  4.0410e-02, -8.7599e-01,\n",
       "           3.8119e-01, -2.1660e-01, -8.7728e-02,  3.6628e-01, -1.1149e+01,\n",
       "           2.3726e-01,  8.8183e-01, -3.2766e-01, -3.9771e-02, -3.4749e-01,\n",
       "           1.9291e-01, -7.7958e-02,  2.6107e-01, -2.8036e-01, -9.5322e-02,\n",
       "          -2.5367e-01,  2.2686e-02,  8.0204e-02, -2.5156e-01,  4.3455e-02,\n",
       "          -1.7673e-01, -7.2054e-01,  2.6437e-01,  5.9248e-02, -1.2207e-01,\n",
       "          -1.2074e-01, -2.3925e-01,  1.4175e-01, -6.3142e-01,  5.2380e-02,\n",
       "          -2.2003e-01, -2.4293e-01,  8.3487e-04,  6.9521e-01, -2.1532e-01,\n",
       "          -1.2654e-02,  9.7817e-02,  2.8130e-01, -3.9372e-01, -5.1365e-01,\n",
       "           1.2387e-02, -3.3278e-02, -1.5469e-01,  1.7484e-01, -5.7002e-01,\n",
       "          -5.8714e-01, -2.1127e-02,  7.5472e-02,  1.8655e-01,  5.5475e-01,\n",
       "           2.5084e-01,  9.3274e-01,  4.4707e-01, -1.4224e-01, -3.4981e-01,\n",
       "          -1.0015e+00,  6.3469e-02, -9.5381e-01,  3.5078e-01, -5.4649e-02,\n",
       "           7.9073e-01, -4.0021e-01,  2.5469e-01, -3.8538e-01,  1.6499e-02,\n",
       "           1.8450e-01,  2.7618e-01, -6.1267e-02, -4.4336e-01, -6.7098e-02,\n",
       "           4.2739e-01, -2.0773e-01,  3.6439e-02, -4.7796e-01, -3.1042e-01,\n",
       "          -2.9442e-01,  4.4250e-01,  1.9411e-02,  3.9886e-01,  3.5074e-01,\n",
       "          -2.4336e-01, -2.0888e-01,  1.7993e+00, -4.0366e-01,  1.7553e-01,\n",
       "          -1.8192e-01,  1.8362e-01, -4.9901e-01, -8.3152e-01,  1.8161e-01,\n",
       "          -2.0930e-01,  1.5689e-01,  4.5942e-02,  6.2053e-01, -3.4065e-01,\n",
       "          -2.6528e-01,  1.0245e-02, -6.2038e-02, -2.3191e-01,  1.0345e+00,\n",
       "          -1.2127e-01, -3.6922e-01, -1.4760e-01,  3.0626e-01, -7.1314e-01,\n",
       "          -2.8442e-01, -1.3455e-01, -2.7820e-01,  2.0215e-01, -3.3942e-01,\n",
       "           3.5390e-02, -6.0351e-02, -3.4419e-02,  5.7237e-01, -5.5415e-01,\n",
       "           1.4902e-01,  4.3460e-01,  6.0812e-03, -2.2517e-01, -5.3994e-01,\n",
       "           4.8159e-01,  4.4345e-01, -3.1941e-01, -6.1522e-01, -5.2188e-02,\n",
       "          -1.5662e-01, -1.9574e-02, -1.0654e-01, -5.6184e-01,  3.3130e-01,\n",
       "          -7.0560e-01, -7.4080e-02,  5.3353e-01, -3.8275e-02,  4.1007e-01,\n",
       "           2.8641e-01, -4.8787e-01,  2.8231e-01, -3.1828e-01, -1.1510e-01,\n",
       "          -8.0165e-01, -6.6641e-01,  3.2451e-01, -4.8866e-01,  2.8968e-01,\n",
       "           1.8796e-02, -7.0031e-02, -2.7368e-01,  9.3806e-02, -1.5728e-01,\n",
       "           1.2878e-01,  3.6399e-01, -4.5294e-02, -1.2584e-01, -1.1594e-01,\n",
       "           1.6907e-02, -4.1819e-01, -3.8143e-01,  1.1122e-01, -5.4353e-01,\n",
       "          -1.6795e-01, -1.0167e-01, -1.3214e-01, -1.4681e-01,  5.2087e-01,\n",
       "           6.4475e-01, -9.6996e-02, -3.0489e-01,  3.5089e-01,  3.5642e-01,\n",
       "          -2.2080e-01,  1.1692e-01, -5.9096e-01, -1.0907e-01,  2.1660e-01,\n",
       "           1.0487e-01, -1.5369e-01, -3.0138e-01, -2.5180e-01, -7.0079e-01,\n",
       "          -2.2514e-01,  1.0950e+00, -5.1845e-01, -6.8783e-01,  4.6600e-01,\n",
       "           3.0278e-01, -1.9977e-01,  6.3924e-02, -1.2250e-01,  2.6434e-02,\n",
       "          -3.5658e-01,  1.3026e-01,  7.1617e-01,  1.5163e-01,  1.6427e-01,\n",
       "           3.9115e-01,  3.5391e-01, -3.4353e-01,  1.0958e-01, -1.4495e-01,\n",
       "          -4.7247e-01,  1.4219e-01, -4.0098e-02, -1.6355e-01,  9.1140e-02,\n",
       "          -5.1558e-02, -2.6708e-01,  2.7695e-01, -2.7505e-01,  3.2110e-03,\n",
       "           2.7308e-01,  8.5689e-01, -1.1081e-01, -6.9538e-02,  3.1955e-01,\n",
       "           3.9728e-01,  1.3435e-01,  1.4905e-01,  3.7763e-01,  8.3303e-02,\n",
       "           4.2342e-01,  2.3370e-01, -1.5310e-01, -1.7052e-01,  1.5115e-01,\n",
       "           4.1113e-01, -4.1120e-01, -5.8758e-01,  1.1081e-01,  5.7065e-01,\n",
       "          -1.2180e-01,  6.0302e-01, -1.6362e-01,  1.7591e-01,  4.8836e-01,\n",
       "          -9.6338e-01,  6.1833e-02,  2.0248e-01, -6.0485e-01, -2.1245e-01,\n",
       "          -2.2919e-01, -3.4192e-01, -1.6229e-01, -4.7573e-01,  4.6485e-01,\n",
       "           5.6997e-01, -1.2268e+00,  2.3353e-01,  7.0952e-01, -4.3914e-01,\n",
       "           6.0637e-01, -4.7956e-02,  4.7979e-01,  8.8963e-01, -2.2181e-01,\n",
       "          -9.5748e-02,  1.5582e-01, -3.1432e-01, -4.2579e-01,  5.9131e-02,\n",
       "           6.2730e-01,  4.2211e-02,  1.8738e-02, -2.1620e-01, -5.1581e-01,\n",
       "          -4.3150e-01, -3.2225e-01, -2.5392e-01,  1.6589e-01,  6.2313e-01,\n",
       "          -2.2271e-01,  3.8019e-02, -2.0308e-01,  2.4913e-01, -6.9309e-02,\n",
       "           5.3801e-01,  2.8776e-01,  4.7944e-01,  1.9747e-01, -5.4127e-01,\n",
       "          -8.7731e-02,  5.7235e-01, -3.8376e-01,  8.6869e-01, -1.8586e-01,\n",
       "          -3.9331e-01, -8.4133e-02,  1.1606e-01, -1.8034e-01,  2.3142e-01,\n",
       "          -5.2697e-01,  3.8899e-01, -1.8176e-01,  6.6721e-03,  5.8575e-01,\n",
       "          -6.6039e-02,  2.7508e-01, -7.1980e-01, -8.3032e-03,  1.9168e-01,\n",
       "           8.7176e-02,  5.6715e-01, -1.5991e-01,  8.0517e-02, -4.7620e-01,\n",
       "          -2.1484e-01, -1.7699e-01, -9.6950e-02,  4.5661e-01, -1.2172e-01,\n",
       "          -4.9477e-01, -3.3108e-01, -2.3593e-01,  2.1215e-01,  6.7934e-01,\n",
       "           7.3356e-01, -2.0065e-01, -5.0491e-01, -2.2002e-01,  1.5968e-01,\n",
       "           1.9439e-01,  7.4096e-02,  1.6166e-01, -2.6325e-01, -1.9598e-01,\n",
       "          -4.4662e-01, -4.0758e-01,  7.3051e-02,  4.8675e-02, -2.5397e-01,\n",
       "           5.3172e-01, -8.5962e-02,  8.8738e-02,  5.3409e-01,  3.9084e-01,\n",
       "           4.9408e-02,  6.9127e-01,  1.7864e-01, -4.7942e-01, -1.7557e-01,\n",
       "           1.3188e-01, -9.3891e-02,  8.5702e-02, -3.3488e-01,  6.4587e-01,\n",
       "           1.9185e-01,  2.7023e-01,  4.4925e-01]], device='mps:0'),\n",
       " tensor([[-0.1672, -0.5249, -0.4540,  ...,  0.3771,  0.2016,  0.5340],\n",
       "         [-0.3073, -0.7515, -0.5212,  ...,  0.3863,  0.5295,  0.7535]],\n",
       "        device='mps:0'),\n",
       " tensor([[-2.5187e-01, -4.0054e-01, -5.8007e-01, -8.3055e-01, -1.2868e-01,\n",
       "          -2.8804e-02, -7.1917e-01,  6.6701e-01,  1.2709e-01,  5.0855e-01,\n",
       "           1.2570e-01,  5.1916e-01,  5.1676e-02, -1.1521e-01, -1.6481e-01,\n",
       "          -4.9820e-01,  5.6222e-02,  3.7338e-01,  6.7187e-01, -2.4111e-02,\n",
       "           2.2844e-01,  5.7979e-01, -2.6323e-01, -1.2386e-02, -8.3013e-01,\n",
       "           2.9613e-01, -1.1576e-01,  2.2409e-01,  1.7718e-01, -6.3092e-01,\n",
       "           4.0999e-02,  4.7417e-01,  2.3068e-02,  1.3497e+00, -5.6080e-01,\n",
       "          -1.3230e-01,  8.2213e-02, -3.9453e-01, -2.3010e-01,  3.7283e-02,\n",
       "           6.1164e-01, -4.1315e-01,  1.3635e-01,  4.0254e-01, -6.1085e-02,\n",
       "           8.9076e-02,  1.3899e-01, -3.7744e-01,  2.5892e-01, -8.3961e-01,\n",
       "          -4.8318e-02, -2.0669e-01,  4.9144e-01,  6.2615e-01, -6.3176e-01,\n",
       "          -2.2975e-01,  2.7419e-01, -3.7716e-02, -3.3020e-02,  1.5349e-01,\n",
       "           9.7980e-02,  3.8582e-01, -2.9995e-01,  2.3257e-01,  3.3441e-01,\n",
       "           2.4458e-01, -3.0392e-01, -4.8889e-01, -5.4078e-01, -5.0111e-02,\n",
       "           9.1859e-03,  2.2490e-02,  1.4846e-01, -2.3272e-01,  4.1973e-01,\n",
       "          -7.1214e-01, -6.4354e-02, -8.9125e-02, -2.5973e-01,  2.9246e-01,\n",
       "          -4.5549e-02,  1.9719e-01,  3.1010e-01,  1.0268e-01,  1.2457e-01,\n",
       "           4.3419e-02,  2.7215e-01,  5.2932e-04,  4.7373e-02,  3.7533e-01,\n",
       "          -1.3375e-01, -8.1722e-02, -2.8313e-01, -3.5814e-01,  3.1229e-01,\n",
       "           4.0598e-02,  2.9907e-01, -1.9557e-01,  1.8579e-01, -3.2474e-01,\n",
       "           1.5588e-01,  1.7768e-01, -1.8083e-01,  5.7286e-02, -3.5172e-01,\n",
       "          -4.5475e-01,  2.3589e-02, -2.8569e-01, -4.8285e-01,  2.2697e-01,\n",
       "          -1.4211e-01, -3.0205e-01, -4.0922e-02, -3.8935e-01, -3.1524e-02,\n",
       "          -7.5164e-02,  2.6013e-01, -1.3230e-01, -3.4806e-01, -8.3033e-02,\n",
       "           2.0734e-01, -2.2808e-01,  1.6077e-01,  2.6882e-01,  1.2232e-01,\n",
       "          -9.6705e-02,  6.5251e-01,  1.8950e-02,  3.1674e-01,  2.5763e-01,\n",
       "           1.1699e-01, -6.5768e-02, -1.9745e-01, -2.3154e-01,  7.5988e-02,\n",
       "          -7.6941e-02,  1.0687e-01, -1.8182e-01, -2.2790e-01, -1.8508e-01,\n",
       "           3.1771e-01, -3.3097e-01, -6.0893e-01, -4.9260e-01, -1.6098e-01,\n",
       "          -1.6326e-01, -2.4804e-02, -8.2473e-03, -3.3693e-01, -3.1956e-01,\n",
       "          -6.7443e-01, -3.6717e-01,  8.1041e-01, -1.6831e-02, -3.3356e-01,\n",
       "          -1.6499e-01,  2.3208e-01,  3.1666e-01, -4.5959e-02,  3.7161e-01,\n",
       "           7.7250e-01, -4.7670e-02,  1.6786e-01, -5.1827e-01, -3.8195e-01,\n",
       "           4.8335e-02,  2.6658e-01, -2.7054e-01,  6.1879e-01,  4.8352e-02,\n",
       "           7.5627e-02, -6.6170e-02,  2.7430e-01, -3.2704e-01, -1.2324e-02,\n",
       "          -5.7573e-01,  3.7750e-01,  2.9539e-01, -3.9892e-01, -6.6567e-02,\n",
       "          -2.0834e-02, -1.2060e+00,  9.6964e-02, -8.3040e-01,  3.1236e-01,\n",
       "          -2.3110e-01,  1.7963e-01,  2.3389e-01,  2.9545e-01, -6.5320e-02,\n",
       "           2.5404e-01,  2.8725e-01,  2.5687e-01,  3.3609e-01, -3.4924e-02,\n",
       "           2.8019e-01,  2.3924e-01,  5.0790e-01,  4.1073e-01,  1.4769e-01,\n",
       "          -3.5440e-01,  2.9683e-01,  3.5781e-01,  9.0598e-02,  1.6954e-01,\n",
       "           9.9825e-02, -2.1636e-01,  2.1091e-01,  1.7501e-01, -1.6618e-01,\n",
       "          -2.2406e-01, -3.1100e-02, -4.6818e-01, -5.6655e-02, -4.8467e-01,\n",
       "           3.2557e-01, -5.1983e-01, -1.1912e-01,  3.4831e-01,  2.4340e-01,\n",
       "          -1.4896e-02,  1.8319e-01,  1.7196e-01, -2.4338e-01,  2.1735e-01,\n",
       "           3.7449e-01, -3.4618e-01, -3.9804e-02, -3.2197e-01, -2.9367e-01,\n",
       "           2.4357e-01, -5.6244e-01,  6.6635e-01,  4.0574e-01,  6.0165e-02,\n",
       "           1.5119e-01, -2.7920e-01, -5.1638e-02, -5.7408e-01,  3.5471e-01,\n",
       "          -3.2130e-01, -3.4955e-01, -5.4801e-01,  6.4487e-02,  4.2773e-01,\n",
       "           2.7565e-01, -6.4915e-02,  7.7868e-01,  4.2408e-01,  4.2813e-01,\n",
       "          -4.3003e-01, -1.5896e-02,  4.0377e-01, -4.1405e-02, -4.6059e-01,\n",
       "           6.7116e-02, -1.4962e-01, -5.1643e-01,  2.8000e-01, -2.6598e-01,\n",
       "          -1.2047e-01, -1.4913e-01,  3.3682e-01,  1.5365e-01, -2.1964e-01,\n",
       "          -3.8461e-01,  3.9587e-01, -1.4349e-01, -4.1723e-01,  4.7123e-01,\n",
       "          -3.5025e-01,  1.5209e-01,  1.0694e-01, -3.8441e-01, -6.2072e-01,\n",
       "          -3.2235e-01,  2.3090e-01,  1.5585e-01,  3.4799e-02, -6.6306e-01,\n",
       "          -4.0571e-01,  3.1197e-01,  2.4601e-01,  2.9976e-01,  2.9840e-01,\n",
       "           5.7427e-01, -3.0944e-01, -1.6702e-01,  1.4982e-01, -4.1377e-01,\n",
       "          -1.5704e-01, -3.3228e-02, -1.9995e-01,  2.9569e-01,  2.4768e-01,\n",
       "           1.1600e-01,  2.0367e-01,  4.3482e-01, -2.5272e-01, -3.8673e-01,\n",
       "          -7.1543e-02,  2.0161e-01, -9.3629e-02, -5.8306e-02,  2.3951e-01,\n",
       "          -6.3095e-02,  5.9650e-01,  7.2143e-01,  7.7439e-02, -6.4006e-01,\n",
       "           1.7696e-01,  5.2502e-01, -2.7253e-01, -4.0343e-02,  9.6365e-02,\n",
       "          -1.0344e+00, -2.1331e-01, -1.2883e-01, -2.0181e-01, -6.5352e-01,\n",
       "           8.8115e-02,  3.4863e-01,  4.6833e-03, -1.7864e-01, -3.0947e-01,\n",
       "           1.0448e-01,  4.8724e-02, -4.8980e-02,  4.3534e-01, -2.8901e-01,\n",
       "          -3.4797e-01, -2.3438e-01, -2.1532e-01, -5.4201e-01, -3.9968e-01,\n",
       "           3.5914e-01,  2.8129e-01,  1.1986e-01, -2.6486e-01, -5.1832e-01,\n",
       "           4.5577e-01, -6.8169e-01, -1.4632e-01,  2.4840e-01, -4.7702e-01,\n",
       "           1.4650e-02,  4.8990e-01,  1.4999e-01,  1.6336e-01,  6.5258e-02,\n",
       "           7.8557e-02, -3.3903e-01, -5.0030e-01, -1.5000e-01,  6.2323e-01,\n",
       "           3.2283e-01,  3.3506e-01, -1.7236e-01,  3.9096e-01,  4.5167e-01,\n",
       "          -2.9582e-01,  4.1670e-01,  1.0930e-02,  7.7439e-02, -6.0414e-02,\n",
       "           1.9048e-01,  3.4370e-01, -3.6842e-01,  2.0145e-01,  1.7929e-01,\n",
       "          -3.1634e-01, -3.2149e-01,  1.5691e-02,  2.0642e-01, -1.8580e-01,\n",
       "          -3.0875e-01, -2.0361e-01,  9.9698e-02, -3.4856e-01, -4.3104e-01,\n",
       "           2.9161e-01, -3.6543e-02,  7.0495e-02, -4.0357e-01, -4.3462e-01,\n",
       "          -2.0181e-01, -2.6055e-01,  6.8681e-01,  2.1064e-01, -1.9621e-01,\n",
       "           1.8720e-01, -1.0907e-01, -3.6330e-01,  4.7847e-01,  2.0162e-02,\n",
       "          -4.8754e-01,  4.5482e-01, -3.4822e-01,  1.0627e-01,  5.5756e-01,\n",
       "          -1.6769e-01, -1.0840e-01,  1.1008e-01,  1.0820e-01,  2.9026e-01,\n",
       "          -3.9884e-01,  1.5516e-01, -6.3476e-02, -3.1885e-01,  3.1820e-01,\n",
       "           2.6411e-01,  2.2781e-01, -1.7137e-01, -3.3426e-02,  3.5677e-02,\n",
       "           2.8782e-01, -6.0853e-02, -3.9888e-01, -4.7347e-01, -4.6132e-01,\n",
       "           1.2832e-01,  1.8927e-01,  5.6207e-01,  3.4144e-01, -1.1394e+01,\n",
       "           1.2938e-01,  7.8031e-01, -4.0713e-01, -8.2819e-02, -5.0320e-02,\n",
       "           1.0976e-02, -1.3708e-01,  4.2955e-01, -2.7734e-01, -4.4105e-01,\n",
       "          -3.5727e-01, -1.3308e-01,  1.5454e-01, -3.2189e-01, -2.6945e-01,\n",
       "          -1.4686e-02, -7.0935e-01,  4.0963e-01, -3.0286e-01,  5.0298e-01,\n",
       "          -2.3511e-01, -8.3798e-02,  1.5816e-01, -3.5851e-01, -1.3439e-02,\n",
       "           5.9766e-02, -1.6033e-01,  1.4651e-01,  4.2486e-01, -5.2072e-02,\n",
       "           3.3244e-02, -4.1258e-03,  4.8461e-02, -2.5723e-01, -5.9999e-01,\n",
       "           1.9296e-02, -1.4141e-01,  3.3422e-01, -4.4133e-02, -7.1265e-01,\n",
       "          -2.2533e-01, -2.7418e-01,  3.1407e-01,  2.7021e-01,  9.0598e-01,\n",
       "           2.2955e-01,  1.0686e+00,  1.5022e-01,  1.7748e-01,  2.6230e-02,\n",
       "          -7.8764e-01, -3.3269e-02, -4.1907e-01,  2.3280e-02,  1.0114e-01,\n",
       "           3.6576e-01, -7.9367e-01,  2.5251e-01, -3.5521e-01,  3.6487e-02,\n",
       "           4.9094e-01,  3.1768e-02,  2.7700e-01, -3.2298e-02, -7.8123e-02,\n",
       "           1.2089e-01, -2.8544e-02,  1.1098e-01, -2.2583e-01, -2.4084e-01,\n",
       "          -1.0177e-01,  5.9888e-01,  3.9592e-01, -7.2211e-01,  3.6112e-01,\n",
       "           4.9258e-02, -1.8622e-01,  1.4556e+00, -4.8573e-02,  1.8361e-02,\n",
       "           1.9800e-03,  2.9604e-01, -5.0236e-01, -7.3516e-01,  3.6511e-01,\n",
       "          -1.0211e-01,  5.6427e-02,  1.0127e-01,  4.8148e-01, -9.7481e-02,\n",
       "          -4.7940e-01, -1.8593e-03,  1.0556e-01, -1.3380e-01,  5.4710e-01,\n",
       "          -7.2824e-01, -1.6963e-01, -1.1223e-01,  3.1723e-01, -8.4054e-01,\n",
       "          -6.1126e-02, -7.0210e-01, -3.9328e-01,  5.0586e-01, -2.5896e-01,\n",
       "           1.5852e-01, -2.7443e-02, -4.3439e-02,  5.5574e-01, -2.1489e-01,\n",
       "           7.1555e-02,  2.2813e-01,  8.4995e-02, -1.6733e-01,  3.9349e-02,\n",
       "           4.1630e-01,  4.9571e-01, -4.2740e-01, -6.3911e-01, -7.6901e-03,\n",
       "           1.7180e-01,  1.4933e-01,  3.4019e-01, -5.3463e-01,  3.6431e-01,\n",
       "          -7.9162e-01, -1.8769e-01,  7.2949e-02, -4.3273e-02,  3.6175e-01,\n",
       "           2.3925e-01,  7.2038e-02,  2.2256e-01, -7.7756e-01, -6.4498e-01,\n",
       "          -6.5932e-01, -2.1445e-01,  2.6143e-01, -5.4236e-01, -8.3654e-02,\n",
       "          -1.1033e-01, -4.8837e-03, -5.0292e-02, -6.6102e-02,  1.1939e-02,\n",
       "           1.3632e-01,  2.3587e-01, -1.3023e-01, -1.3178e-01, -9.2464e-02,\n",
       "          -5.5566e-01, -1.1753e-01, -9.9591e-02, -1.2505e-01, -5.6783e-01,\n",
       "          -1.3797e-01,  3.3075e-02, -2.5373e-01, -6.8124e-02,  3.5608e-01,\n",
       "          -2.9832e-02, -2.7856e-02, -4.8850e-01,  7.2846e-01,  4.0279e-01,\n",
       "           5.2088e-02, -2.8298e-01, -5.5189e-01, -1.0567e-01,  2.9592e-01,\n",
       "           1.9555e-01, -6.7007e-02, -1.7709e-01, -2.4145e-01, -4.2522e-01,\n",
       "           1.4302e-01,  7.1147e-01, -6.6296e-02, -6.8115e-01,  2.6621e-01,\n",
       "           5.8223e-01, -1.2882e-01, -2.4432e-01,  2.0983e-02, -9.3284e-02,\n",
       "          -3.1851e-01, -6.0317e-01,  7.8821e-01, -1.1246e-01, -5.8638e-02,\n",
       "           2.2427e-01,  2.0359e-01, -9.2588e-03,  4.0209e-01, -1.5770e-01,\n",
       "          -3.5434e-01, -1.6477e-03, -7.3600e-01,  4.3257e-01,  6.1666e-02,\n",
       "          -3.1489e-01,  7.6082e-01,  2.7260e-01, -6.0593e-02, -4.5779e-02,\n",
       "           2.2863e-01,  7.9941e-01, -2.4636e-01,  6.4653e-02,  5.3779e-01,\n",
       "           8.8110e-02, -1.1229e-01,  3.9580e-02,  4.1665e-02,  4.5410e-02,\n",
       "           3.7012e-01, -3.2145e-02, -4.1564e-01, -6.1194e-02,  3.8924e-01,\n",
       "           3.9295e-01, -5.4610e-01, -7.3487e-01,  2.2333e-02,  6.8551e-01,\n",
       "          -2.9054e-01,  2.3654e-01, -3.2708e-02,  3.0880e-01,  1.2001e-01,\n",
       "          -9.2160e-01, -1.5016e-01, -1.0056e-01,  2.8229e-01, -2.5482e-02,\n",
       "          -1.6139e-01, -1.4727e-01, -2.3686e-01, -1.6208e-01,  6.8308e-01,\n",
       "           2.1991e-01, -6.9160e-01,  3.4058e-02,  5.7281e-01, -6.0551e-01,\n",
       "           4.6052e-01, -2.4577e-01, -3.4302e-02,  3.6435e-01,  1.3117e-01,\n",
       "           4.9067e-02,  1.7856e-01, -4.2138e-01, -4.0598e-01,  1.7824e-01,\n",
       "           9.1672e-01, -7.6613e-02,  1.1166e-01, -3.9384e-01, -1.7182e-01,\n",
       "          -7.1855e-02, -3.1913e-01, -3.1286e-01,  3.7185e-01,  2.0529e-01,\n",
       "          -3.2552e-02, -1.3271e-01, -2.2992e-01,  3.0816e-01,  1.1821e-01,\n",
       "           3.2536e-01,  5.8187e-01,  6.4368e-01, -1.0077e-02, -2.4936e-01,\n",
       "          -6.1235e-02,  1.4568e-01, -1.9524e-01,  5.2704e-01, -2.0627e-01,\n",
       "          -3.3453e-01, -3.8069e-03,  3.0134e-01, -6.1681e-01, -5.8725e-02,\n",
       "          -4.5426e-01,  4.2361e-01, -1.6955e-01,  3.7868e-02,  2.8833e-01,\n",
       "           4.0522e-02,  5.7925e-02, -2.2651e-01,  5.5807e-01,  4.4511e-01,\n",
       "           1.6609e-01,  4.5662e-01,  1.6104e-01,  4.5272e-02, -5.1597e-01,\n",
       "           2.1978e-01, -1.0213e+00, -9.4260e-02, -4.6640e-02, -1.5401e-01,\n",
       "          -4.4224e-01, -2.8126e-02,  1.6814e-01,  1.8538e-01,  6.9597e-01,\n",
       "           7.2967e-01, -3.0871e-01, -4.5290e-01, -1.8510e-01,  1.5304e-01,\n",
       "           2.7705e-01, -1.9517e-01,  3.5971e-01,  9.6103e-02,  2.4432e-01,\n",
       "          -5.0967e-01, -4.3165e-01, -4.2252e-02,  3.1411e-01, -2.0171e-01,\n",
       "           5.7118e-01, -2.0164e-01, -1.3863e-01,  9.0832e-02,  1.3770e-01,\n",
       "           1.6639e-01,  7.9664e-01,  4.3416e-01, -6.3425e-01,  8.7469e-02,\n",
       "          -1.3980e-02, -2.9988e-01,  2.5983e-02, -4.3577e-01,  2.8858e-01,\n",
       "           1.2984e-01,  4.9280e-01,  6.7504e-01]], device='mps:0'),\n",
       " tensor([[-0.0111, -0.5057, -0.4670,  ...,  0.2365, -0.0712,  0.4388],\n",
       "         [-0.2138, -0.7115, -0.0545,  ..., -0.0400,  1.0202,  0.1528]],\n",
       "        device='mps:0'),\n",
       " tensor([[-0.0111, -0.5057, -0.4670,  ...,  0.2365, -0.0712,  0.4388],\n",
       "         [-0.3292, -0.2590, -0.5016,  ..., -0.0148,  0.3509,  0.1607]],\n",
       "        device='mps:0'),\n",
       " tensor([[-3.0149e-01, -6.6211e-01, -6.9301e-01, -1.0462e+00, -5.1587e-01,\n",
       "          -2.4685e-01, -6.5106e-01,  8.8509e-01,  8.6229e-02,  4.2184e-01,\n",
       "           2.2681e-01,  5.5254e-01,  3.6319e-01, -4.5176e-01, -7.4929e-01,\n",
       "          -1.8310e-01, -1.5045e-01,  4.0498e-01,  6.0556e-01,  4.3528e-01,\n",
       "           6.8072e-01,  7.7191e-02, -1.3795e-01, -5.8510e-02, -2.0373e-01,\n",
       "           4.0880e-01, -1.9485e-02, -5.9545e-02,  2.7157e-01, -7.3432e-01,\n",
       "           1.7094e-02,  5.9785e-01, -1.2438e-01,  1.2329e+00, -1.0087e-01,\n",
       "          -1.1610e-01,  2.4528e-01, -1.8679e-01,  8.5582e-01, -4.7573e-01,\n",
       "           8.0489e-01,  1.7412e-01,  2.7970e-01,  2.4407e-01,  4.2429e-01,\n",
       "           1.3282e-01,  3.6367e-01, -5.9261e-01,  6.9939e-02, -5.6682e-01,\n",
       "          -1.9591e-01, -2.1175e-02,  4.0060e-01,  4.0411e-01, -5.6681e-01,\n",
       "          -4.2669e-01,  4.8966e-02,  3.9787e-01, -9.8613e-01,  3.2912e-01,\n",
       "          -1.1695e-01,  7.0120e-01,  5.1015e-01,  2.8693e-02,  1.0263e-01,\n",
       "           1.1395e-02, -4.3822e-01, -4.1292e-01, -4.4885e-01,  6.1657e-02,\n",
       "           2.3342e-02,  4.2783e-01,  2.9783e-02, -2.1898e-01,  7.6891e-01,\n",
       "          -2.5723e-01,  5.8233e-02, -2.4487e-01, -4.8788e-01, -2.2056e-01,\n",
       "          -1.9548e-01,  1.8204e-01, -3.8798e-01, -1.2284e-01, -1.4915e-01,\n",
       "          -1.6267e-01,  6.9790e-02,  9.7007e-02,  1.4820e-01, -5.4348e-03,\n",
       "           2.0449e-01, -6.2253e-02, -3.5966e-01, -3.2979e-01,  1.8241e-01,\n",
       "           1.0753e-01,  3.1368e-01,  1.4063e-01, -1.6909e-01, -2.0057e-01,\n",
       "           2.6503e-01,  2.5026e-01,  9.0673e-02,  1.4981e-01, -7.0591e-01,\n",
       "          -7.8928e-01,  4.6700e-02, -7.7030e-02, -7.1297e-01, -2.2066e-01,\n",
       "          -3.8239e-01, -4.2176e-01,  3.1230e-01, -4.7721e-01,  8.7683e-02,\n",
       "          -1.4820e-01, -1.8889e-01, -2.2456e-01, -1.8180e-01,  6.8231e-01,\n",
       "          -1.1088e-01, -3.5657e-01, -5.7816e-02,  6.9311e-02,  1.5681e-01,\n",
       "          -1.7802e-01,  4.1718e-01, -1.0340e-02,  2.0044e-01,  1.3454e-01,\n",
       "           1.1747e-01, -1.6800e-02, -4.2919e-01,  5.5664e-02, -2.6592e-01,\n",
       "           4.3293e-01,  1.2309e-02, -3.4961e-01,  1.7686e-01, -6.8040e-02,\n",
       "          -9.5958e-02,  2.5224e-02, -1.5626e-01,  1.9536e-01, -9.0955e-02,\n",
       "          -1.9166e-01, -7.6367e-02, -2.0750e-01, -3.4427e-01, -2.6126e-01,\n",
       "          -3.8710e-01, -4.6725e-01,  1.4831e-01, -3.3261e-01,  7.7033e-02,\n",
       "           5.3501e-03, -1.2201e-02, -8.4118e-02, -6.3044e-02,  3.9849e-01,\n",
       "           8.8994e-01, -1.2221e-01,  2.7348e-01, -5.4010e-01,  9.3024e-02,\n",
       "          -5.0941e-02, -1.7295e-01, -6.7410e-02, -1.4414e-01,  8.0231e-02,\n",
       "           1.2269e-01, -2.8960e-01, -8.2604e-02, -4.6220e-01,  7.8118e-02,\n",
       "          -1.0382e+00,  2.3921e-01,  2.1753e-01, -1.6458e-01, -3.1370e-01,\n",
       "           2.9818e-01, -1.1032e+00,  1.1456e-01, -5.7877e-01, -7.3260e-02,\n",
       "          -4.4016e-01,  5.6995e-01,  3.9981e-01,  2.0091e-01,  1.5740e-01,\n",
       "           6.5769e-02, -3.7287e-01,  1.5966e-01,  4.4396e-01,  3.3513e-01,\n",
       "           3.2179e-01,  2.6112e-01, -9.1485e-02,  7.3118e-03, -8.3452e-02,\n",
       "          -6.9589e-01,  4.8267e-02, -1.0536e-01,  2.8194e-01,  3.8240e-01,\n",
       "           3.9192e-01, -2.4711e-01,  1.0728e-01,  2.5877e-02,  4.5443e-02,\n",
       "          -1.1905e-01,  1.4878e-01, -4.0438e-01, -2.4969e-01, -9.1650e-02,\n",
       "           5.8958e-01, -3.4102e-01, -2.2780e-01,  4.4939e-01,  2.1616e-01,\n",
       "          -1.6737e-01,  2.1891e-01,  1.3805e-01, -1.5000e-01, -9.3797e-02,\n",
       "           2.9097e-01, -1.9896e-01,  3.3684e-01, -4.1328e-01, -3.4776e-01,\n",
       "          -2.0598e-02, -5.6601e-01,  2.9598e-01,  5.0112e-01, -1.5784e-01,\n",
       "          -7.3784e-02, -2.9341e-01,  3.3785e-01, -4.7242e-01,  2.2371e-01,\n",
       "          -5.1516e-01, -2.6530e-01, -4.9741e-01, -3.3121e-01, -1.4924e-03,\n",
       "           2.9940e-01,  1.6183e-01,  1.1368e+00,  1.1539e-01,  1.9940e-01,\n",
       "          -2.5972e-01, -4.7254e-01,  4.8447e-01,  3.0748e-02, -3.7296e-01,\n",
       "          -2.5839e-01, -1.0365e-01, -1.6010e-01,  3.2487e-01, -2.1141e-01,\n",
       "           3.1868e-02, -3.4954e-02,  1.5336e-01,  4.7139e-01, -3.8786e-01,\n",
       "           7.5400e-02,  3.2093e-01, -1.7837e-01, -3.4280e-01,  4.1751e-01,\n",
       "          -5.1837e-01,  4.1929e-01,  2.1803e-01, -3.9737e-01, -5.7979e-01,\n",
       "          -4.1193e-01,  1.8021e-01, -3.7550e-01,  1.0709e-01, -7.2701e-01,\n",
       "          -4.2120e-01,  2.4189e-02, -8.9674e-02,  5.0182e-02,  1.4171e-01,\n",
       "           1.9428e-01,  1.5182e-01, -1.4555e-01, -5.1129e-01, -4.1474e-01,\n",
       "          -7.1766e-03, -1.3559e-02, -5.6397e-01,  4.0346e-01,  2.7608e-01,\n",
       "          -4.4769e-01,  6.5804e-02,  2.5160e-01,  2.8332e-01, -4.5983e-01,\n",
       "          -2.0237e-01, -3.4603e-01, -1.5293e-01, -1.7055e-01,  2.0178e-01,\n",
       "           1.4764e-01,  3.2234e-01,  7.1860e-01, -3.1833e-01, -8.4506e-01,\n",
       "           6.4835e-02,  7.2226e-01, -1.8505e-01, -2.7852e-02, -1.0830e-01,\n",
       "          -1.3005e+00, -1.8073e-01, -5.8205e-01, -3.1176e-01, -3.9207e-01,\n",
       "           6.9240e-02,  4.4920e-02,  6.3851e-02, -1.2146e-01, -5.1301e-01,\n",
       "           1.1379e-01,  3.2215e-01,  3.5238e-03,  5.2452e-02, -4.0787e-01,\n",
       "          -4.0028e-02, -4.2300e-01, -3.0547e-01, -4.9134e-02,  2.2767e-01,\n",
       "          -4.9487e-02,  1.1026e-01,  9.0243e-02,  2.0215e-01, -3.9532e-01,\n",
       "           5.9136e-01, -2.2641e-01, -1.8379e-01,  4.0002e-01, -3.6463e-01,\n",
       "          -2.3052e-01,  1.8526e-01, -5.8671e-01,  1.1582e-01,  3.1139e-01,\n",
       "           2.5448e-01, -5.1068e-01, -7.3737e-01,  1.1132e-02,  5.3357e-01,\n",
       "           4.6448e-01,  6.1394e-01, -6.1725e-01,  6.4766e-02,  1.8701e-01,\n",
       "          -1.3023e-01,  4.5004e-01, -3.0363e-01, -1.1110e-01, -1.2154e-02,\n",
       "           2.8932e-01,  3.3598e-01, -2.4193e-01,  1.0934e-01, -2.1118e-01,\n",
       "           2.1851e-01, -9.3692e-02,  7.6811e-02,  3.2007e-01,  1.9439e-01,\n",
       "          -1.6163e-01, -4.5144e-01,  7.9689e-01, -2.8161e-02, -4.3164e-01,\n",
       "           3.2932e-02,  1.2716e-01, -1.6311e-01,  5.6596e-02, -2.5591e-01,\n",
       "          -1.7767e-01,  7.7688e-02,  9.1522e-01,  1.3013e-01, -4.6080e-02,\n",
       "           2.6572e-01, -2.1941e-01, -1.7357e-01,  4.9046e-01, -1.8026e-02,\n",
       "          -4.0357e-01,  3.8359e-01, -1.3195e-01,  3.2654e-02,  1.3832e-01,\n",
       "          -2.1124e-01,  3.8524e-02,  2.3328e-01,  2.6290e-02,  9.9947e-02,\n",
       "          -3.7240e-01,  4.8670e-02,  1.6473e-01, -1.6647e-01,  2.7749e-01,\n",
       "          -2.4931e-01,  7.1132e-01, -2.4312e-02,  9.2098e-02,  5.6558e-02,\n",
       "           2.1331e-01, -2.2674e-01,  2.1369e-01, -1.3341e-01, -5.1877e-01,\n",
       "           6.3759e-01,  3.8673e-01,  4.8767e-01,  5.6823e-01, -1.1214e+01,\n",
       "           3.5605e-01,  5.4424e-01, -3.9178e-01,  2.6273e-02,  2.4363e-01,\n",
       "           2.6908e-01, -3.6216e-01,  2.2127e-01, -1.2938e-01, -1.6477e-01,\n",
       "          -2.2090e-01,  2.8619e-01, -5.9194e-02, -5.9279e-02, -2.8318e-01,\n",
       "           1.6415e-01, -7.0844e-01,  1.0695e-01, -1.1860e-01,  7.5527e-01,\n",
       "          -5.8532e-02,  1.4597e-01,  3.6262e-01, -4.9921e-01, -4.3758e-01,\n",
       "           2.0899e-01, -1.3209e-01,  1.6965e-01,  9.8185e-02, -1.2722e-01,\n",
       "           1.7887e-01, -1.9330e-01, -9.9810e-02, -2.5810e-01, -3.9631e-01,\n",
       "          -1.1723e-01, -1.5591e-01,  7.5878e-01, -3.1705e-01, -5.5775e-02,\n",
       "          -2.8780e-01, -1.9313e-01,  3.8049e-01,  4.6353e-01,  7.3074e-01,\n",
       "           1.2051e+00,  5.5980e-01,  1.7294e-01, -1.7826e-01,  6.9183e-02,\n",
       "          -7.9845e-01,  4.6391e-01, -8.1605e-01,  2.2795e-01, -9.7347e-02,\n",
       "           5.0062e-01, -9.0607e-01, -1.1043e-01, -5.3145e-01, -2.4753e-01,\n",
       "           1.8785e-01, -1.4819e-01, -3.3125e-02, -1.6844e-01, -4.4202e-01,\n",
       "           4.6035e-01, -2.5114e-01,  1.5195e-01, -2.8833e-01,  3.0307e-01,\n",
       "          -3.7993e-01,  8.4476e-02,  1.8084e-01, -8.6361e-01,  1.3124e-01,\n",
       "           9.1054e-02, -1.2822e-01,  1.6344e+00, -1.9378e-01, -2.3397e-02,\n",
       "           2.1542e-01,  6.5722e-02, -2.7130e-01, -8.1910e-01, -2.0294e-02,\n",
       "           1.4245e-01,  5.9657e-01,  5.6578e-01,  4.9260e-01, -1.3493e-01,\n",
       "          -3.3287e-01,  8.7146e-02,  1.2506e-01,  3.1924e-01,  7.0830e-01,\n",
       "          -7.3072e-01,  9.9208e-03,  1.2706e-01,  2.6859e-02, -6.0301e-01,\n",
       "           6.7688e-02, -2.1844e-01, -5.0919e-01,  1.0351e-01,  4.4144e-01,\n",
       "           2.2914e-02, -3.7978e-01, -8.4219e-02,  3.0913e-01, -4.6576e-01,\n",
       "           4.8572e-01, -3.0103e-02, -5.1929e-02, -4.2352e-01,  1.5845e-01,\n",
       "           9.3424e-02,  5.9586e-01, -1.8097e-01, -3.9262e-01, -2.2164e-01,\n",
       "           1.3566e-01,  3.9388e-01, -2.2994e-01, -1.4540e-01,  5.6474e-01,\n",
       "          -5.7056e-01, -2.7646e-01, -1.9670e-01, -6.3229e-02,  4.8794e-01,\n",
       "           2.5863e-01,  5.0222e-02,  9.0066e-02, -2.5316e-01, -5.8084e-01,\n",
       "          -6.6420e-01, -5.7965e-02,  5.1636e-01, -1.6749e-01, -1.6315e-01,\n",
       "           8.9398e-02,  1.7235e-01, -3.1593e-01,  6.7563e-01,  2.6293e-01,\n",
       "           4.6157e-01,  3.8274e-01, -2.7932e-01,  2.3472e-01, -1.1386e-01,\n",
       "          -2.5763e-01,  2.8480e-01, -3.1194e-02, -1.0856e-01, -6.7501e-01,\n",
       "          -1.6200e-01, -2.5868e-01, -4.7356e-01,  1.3400e-01,  1.3255e-01,\n",
       "          -4.2844e-02,  1.5866e-01,  2.3246e-01,  4.5309e-01,  4.9598e-01,\n",
       "           8.5208e-02, -1.5076e-01, -4.3563e-01,  1.3657e-01, -2.4837e-02,\n",
       "          -2.0115e-01,  6.7437e-02, -5.2360e-01, -4.0240e-01, -1.2355e-01,\n",
       "          -1.9977e-01,  1.0274e+00, -2.9984e-01, -5.2231e-01,  3.9339e-01,\n",
       "           5.7084e-01,  1.3371e-01, -1.1469e-01,  5.9781e-02, -5.1145e-02,\n",
       "          -4.3716e-01, -2.6367e-01,  4.4876e-01, -6.2744e-02,  1.6814e-01,\n",
       "           3.2170e-01,  9.4986e-03,  6.0426e-02,  2.9850e-01,  1.7695e-01,\n",
       "          -6.1890e-01,  1.7748e-01, -5.2546e-01,  1.2992e-01,  2.0528e-01,\n",
       "          -2.0412e-01,  4.6561e-01,  1.5737e-01, -3.0518e-03,  2.3958e-01,\n",
       "          -5.3423e-02,  8.3199e-01,  1.9241e-02,  2.7669e-01,  7.5782e-01,\n",
       "           1.5191e-01,  2.5222e-01,  4.7285e-01,  2.4662e-01,  3.2063e-01,\n",
       "           3.8712e-01, -3.4046e-01, -4.8040e-01, -3.1192e-01,  1.6407e-01,\n",
       "           4.4875e-01, -4.5698e-01, -3.1041e-01, -3.2118e-01,  5.0280e-01,\n",
       "          -3.6554e-02,  5.9122e-01, -2.2456e-01,  2.5967e-01,  6.0552e-01,\n",
       "          -8.5501e-01, -1.8939e-01,  2.7121e-01,  4.7677e-01,  2.5451e-02,\n",
       "          -4.8132e-01, -9.3847e-02, -9.9959e-02, -3.7225e-01,  3.3732e-01,\n",
       "           1.6031e-01, -9.1128e-01, -1.6897e-02,  6.2482e-01, -6.2240e-01,\n",
       "           1.3513e-01, -1.8874e-01,  1.8584e-01,  3.5508e-01,  1.9337e-01,\n",
       "          -3.9813e-01,  4.3248e-02, -2.2636e-01, -6.6278e-01,  3.4547e-01,\n",
       "           4.2400e-01,  1.7936e-02,  1.6725e-01,  1.1480e-01, -3.5369e-01,\n",
       "          -2.4815e-01, -2.6298e-01, -1.9551e-01, -2.0242e-01, -2.3159e-01,\n",
       "          -6.9567e-02, -1.9504e-01,  2.1318e-01,  4.5516e-01, -4.0071e-01,\n",
       "           4.6827e-01,  5.4213e-01,  3.9522e-01,  1.5309e-02, -1.0326e-01,\n",
       "          -3.0216e-01,  1.0549e-02, -1.0255e-01,  5.7495e-01, -1.1509e-01,\n",
       "          -5.0896e-01,  1.8755e-02,  1.4274e-01, -3.5408e-01,  3.5184e-01,\n",
       "          -6.2009e-01,  3.1706e-01, -1.6246e-01, -3.2547e-01,  4.0369e-01,\n",
       "           1.1709e-02,  2.3623e-01, -4.8849e-01,  1.9521e-01,  1.1755e-01,\n",
       "           5.2542e-02,  7.2613e-01, -6.5458e-02,  3.8255e-01, -1.3803e-01,\n",
       "           1.2770e-01, -9.5706e-01, -1.7840e-01,  1.6083e-01, -7.5761e-02,\n",
       "          -1.3471e-01, -2.3569e-01,  4.8303e-01,  1.4003e-01,  9.7210e-01,\n",
       "           3.8847e-01,  5.6024e-04, -1.1442e+00, -3.2922e-01, -5.9417e-02,\n",
       "           1.6317e-01, -2.9767e-01,  2.4602e-01, -2.8561e-01,  4.7147e-01,\n",
       "          -7.9551e-01, -3.6861e-01, -5.6060e-03, -2.4818e-02, -2.0548e-01,\n",
       "           5.5167e-01, -8.9934e-02,  9.4105e-02,  2.3195e-01,  3.1128e-02,\n",
       "           4.5577e-01,  7.2594e-01, -1.0334e-01, -5.4650e-01, -2.8453e-01,\n",
       "           2.6135e-01, -5.5831e-02,  1.1167e-01, -7.1042e-01,  1.3746e-01,\n",
       "           1.6669e-01,  5.7022e-01,  5.8746e-01]], device='mps:0'),\n",
       " tensor([[ 1.1277e-01, -5.1182e-01, -5.1456e-01, -9.9081e-02, -3.0116e-01,\n",
       "           3.3007e-03, -4.0947e-01,  4.0427e-01, -5.8056e-02,  7.3590e-01,\n",
       "           2.2527e-01,  8.2791e-02, -3.8724e-02, -7.5126e-02, -1.5840e-01,\n",
       "          -7.3986e-01,  3.8029e-02,  3.1506e-01,  2.4049e-01,  1.5258e-01,\n",
       "           8.4897e-01,  3.7601e-01,  3.7846e-02,  1.5338e-01, -8.9169e-01,\n",
       "           2.4549e-02,  1.7534e-01, -3.8121e-01,  3.0474e-01, -9.4057e-01,\n",
       "          -3.1590e-02,  4.1782e-01, -3.6549e-01,  9.4881e-01, -2.1749e-01,\n",
       "          -2.3738e-01,  2.7768e-01,  1.1423e-01,  1.5727e-03,  8.7533e-02,\n",
       "           6.3608e-01, -3.9554e-01,  3.5800e-01,  1.7469e-01,  5.4999e-01,\n",
       "          -9.7317e-02,  1.1774e-02, -5.5940e-01,  4.2788e-01, -2.4478e-01,\n",
       "          -8.0416e-02, -1.6983e-01,  3.4295e-01,  4.9191e-01, -6.8602e-01,\n",
       "          -5.2333e-01, -6.2734e-02,  1.8164e-01, -2.1619e-01,  2.9696e-01,\n",
       "           1.3140e-01,  5.4967e-01,  1.4880e-01,  8.9218e-02, -9.8997e-02,\n",
       "           2.7962e-01, -6.9163e-01, -1.1292e-01, -4.6817e-01, -1.7222e-01,\n",
       "           7.9175e-02,  3.5231e-01, -2.4967e-01, -7.3913e-02,  7.4744e-01,\n",
       "          -3.1754e-01,  3.1779e-01, -1.0437e-01, -5.2653e-01,  2.5923e-01,\n",
       "           5.3624e-01,  2.2665e-01, -3.8162e-01, -4.3491e-01, -2.9896e-01,\n",
       "           3.1739e-01,  3.6286e-01,  2.1409e-01,  4.1664e-01,  2.3004e-01,\n",
       "           6.9510e-01,  1.1862e-01, -9.0323e-01, -1.9789e-01,  3.0084e-01,\n",
       "          -2.8430e-01,  3.6802e-01, -1.0774e-01,  1.9963e-01,  2.4580e-01,\n",
       "           9.4402e-02,  1.0678e-01, -1.2301e-01,  3.3043e-01, -7.3431e-01,\n",
       "          -3.8387e-01,  2.7538e-01,  7.5278e-02, -7.1095e-01, -5.3743e-02,\n",
       "          -1.3470e-01,  2.0865e-01,  1.6720e-01, -4.9918e-01, -2.6117e-01,\n",
       "           8.9412e-02, -1.5191e-01,  3.4209e-02, -4.2780e-01,  3.6346e-01,\n",
       "           2.9865e-01, -4.8785e-01,  1.3818e-01,  1.4371e-01,  4.4715e-01,\n",
       "          -4.1340e-01,  3.8903e-01, -2.5446e-01,  2.2889e-01,  8.2835e-01,\n",
       "          -1.2485e-01,  2.6252e-02, -4.0830e-02, -1.7487e-01, -2.8440e-01,\n",
       "          -3.4081e-01,  2.2259e-01, -7.0968e-01, -2.4259e-01,  1.8416e-01,\n",
       "           2.6748e-01, -3.7059e-02, -4.8123e-01, -1.1090e-01,  2.2158e-01,\n",
       "          -2.0902e-01, -3.0049e-01,  7.2718e-02, -2.7801e-01, -3.0796e-01,\n",
       "          -1.0773e+00, -6.4338e-01,  3.3934e-01,  1.0304e-01, -8.3425e-02,\n",
       "          -3.2985e-02,  2.6094e-01,  1.2604e-01, -1.1825e-01,  2.1861e-02,\n",
       "           1.4631e-01,  1.5870e-01,  4.0601e-01, -7.2045e-01, -2.6279e-01,\n",
       "           5.1222e-01, -6.9927e-02, -1.1101e-01, -3.8262e-01,  1.1619e-01,\n",
       "           9.3115e-02, -4.6417e-01,  1.4190e-02, -5.3166e-01,  2.5739e-01,\n",
       "          -6.9917e-01,  2.6343e-01,  7.8697e-02,  4.6975e-02, -5.2005e-01,\n",
       "          -5.1269e-01, -1.6613e+00,  4.4969e-02, -1.1536e-01,  4.2832e-01,\n",
       "          -6.4780e-01,  6.8155e-01,  4.9667e-01,  1.3957e-01, -1.0725e-01,\n",
       "          -1.1613e-01, -2.0495e-01, -1.3738e-01,  4.8684e-01, -1.1487e-01,\n",
       "           4.7349e-01,  6.1281e-01,  3.6342e-01,  1.0458e-01,  3.8659e-02,\n",
       "          -5.1090e-01,  3.2178e-01, -6.3780e-02,  6.5283e-02, -4.0803e-03,\n",
       "           5.2045e-01, -1.5508e-01, -5.5974e-02,  4.7542e-01, -9.0455e-02,\n",
       "           2.1110e-02, -5.0237e-01, -2.0562e-01, -1.7373e-01,  3.3842e-02,\n",
       "           6.1372e-01,  9.7513e-02, -1.1752e-01,  6.5636e-01,  6.4063e-01,\n",
       "          -4.3595e-02, -1.1716e-01, -2.8002e-01, -4.5596e-01, -3.9645e-01,\n",
       "           4.4838e-01, -4.1673e-01, -3.8331e-02, -1.3722e-01, -2.2336e-03,\n",
       "           2.9366e-01, -7.7720e-01,  7.8321e-01,  2.7313e-01,  1.2754e-01,\n",
       "          -2.9649e-01, -3.8133e-01, -8.9398e-02,  1.1919e-01,  8.8294e-02,\n",
       "           8.6567e-02, -6.9762e-01, -4.4501e-01,  2.6233e-01, -1.7112e-02,\n",
       "           2.9408e-02, -4.8063e-01,  9.5151e-01,  3.2366e-02,  2.1866e-01,\n",
       "          -8.1848e-02, -2.3095e-01,  2.8605e-01,  5.2168e-02, -4.2972e-01,\n",
       "          -1.1705e-01, -5.8341e-02, -3.5230e-01,  3.8333e-01, -5.6102e-01,\n",
       "          -2.8672e-01,  4.4379e-02,  2.3252e-01,  3.8893e-01, -1.8287e-02,\n",
       "           3.8934e-01, -9.1873e-02, -5.2896e-01, -2.6387e-01,  3.8154e-01,\n",
       "          -2.8519e-01,  3.6897e-01, -4.2110e-01, -2.0175e-01, -6.3157e-01,\n",
       "          -7.1356e-01,  3.0949e-01, -4.4762e-01,  2.3137e-01, -9.6326e-01,\n",
       "          -4.6904e-01,  1.5038e-01, -3.9511e-02,  7.9127e-02, -2.8288e-01,\n",
       "           3.9424e-01,  2.3657e-01,  1.0638e-01, -1.3625e-02, -6.8624e-01,\n",
       "          -1.7361e-01,  5.1713e-01, -3.0739e-01,  3.0249e-01,  3.2178e-01,\n",
       "          -1.5219e-01, -2.7008e-01,  2.0505e-01, -6.8380e-02, -1.3222e-01,\n",
       "          -2.4847e-01, -2.3051e-01,  7.3440e-02, -2.9512e-01, -4.0589e-01,\n",
       "          -1.9683e-01,  3.5653e-01,  3.1357e-01, -1.9608e-01, -4.0936e-01,\n",
       "          -3.1297e-02,  6.9414e-01, -8.3245e-02,  2.4383e-01, -1.2362e-01,\n",
       "          -1.4305e+00, -4.0833e-01, -1.8540e-02, -1.3066e-01, -2.5434e-01,\n",
       "           2.5155e-01,  6.0078e-02, -4.4774e-02, -2.3862e-02, -3.9032e-01,\n",
       "           4.2100e-01,  9.6530e-02, -5.5141e-01,  1.4950e-01, -2.0711e-01,\n",
       "          -1.2827e-01, -1.0196e-01,  5.8219e-02,  1.1231e-02, -1.1398e-01,\n",
       "           1.6417e-01, -2.6502e-02,  2.1900e-01,  8.7820e-02, -3.6880e-01,\n",
       "           5.6536e-01, -4.0455e-01, -9.2397e-02,  2.2993e-01, -6.7463e-01,\n",
       "          -5.8048e-01,  3.5097e-01,  2.4469e-01, -1.4927e-01,  8.5738e-02,\n",
       "           2.5571e-01, -1.1733e-01, -1.8922e-01,  1.1565e-01,  5.0862e-01,\n",
       "           1.6608e-01,  1.5372e-01, -3.3662e-01,  6.7240e-01,  6.1452e-01,\n",
       "          -6.7974e-01,  3.0024e-01, -8.6431e-02, -7.1191e-02, -7.8438e-02,\n",
       "           2.5682e-01,  2.9384e-01, -2.6061e-01,  3.1474e-01,  1.1061e-01,\n",
       "           8.3275e-02, -1.8821e-02, -4.8232e-02,  1.8014e-01,  6.5893e-01,\n",
       "          -6.6205e-02, -4.0549e-01,  3.5343e-01, -4.1343e-01, -2.6268e-01,\n",
       "          -7.5984e-02, -3.7208e-02,  6.5495e-02, -3.1240e-01, -3.4070e-01,\n",
       "          -4.4867e-01, -4.5409e-02,  5.9015e-01, -1.3339e-01, -1.2243e-01,\n",
       "           6.4085e-03,  1.1937e-01, -3.9825e-01,  4.0693e-01, -1.6292e-01,\n",
       "          -6.7273e-03,  1.6511e-01, -4.9061e-01,  5.9405e-01,  2.2158e-01,\n",
       "          -1.9118e-01,  7.7159e-02,  3.7849e-01,  2.9315e-01,  7.3291e-02,\n",
       "          -6.2785e-01, -1.9814e-01,  1.2483e-01,  8.7468e-02,  1.3790e-01,\n",
       "           5.2048e-01,  8.7935e-01, -4.8997e-01,  1.1694e-01, -4.3027e-02,\n",
       "           1.9913e-01, -2.3995e-01,  3.2426e-01,  2.5199e-02, -6.5157e-01,\n",
       "           5.7959e-01, -5.5682e-02,  1.2983e-01, -8.3751e-04, -1.1148e+01,\n",
       "          -2.4697e-02,  6.7893e-01, -5.5838e-01,  2.0324e-01,  1.4167e-01,\n",
       "           2.7728e-01, -3.5224e-01,  2.1719e-04, -1.7645e-01, -4.1684e-01,\n",
       "          -2.9137e-01,  1.3225e-01,  2.1139e-01, -2.9905e-01,  8.3671e-02,\n",
       "          -6.6111e-02, -7.4170e-01,  4.0852e-01,  6.7087e-02,  3.1935e-01,\n",
       "          -8.0690e-02,  3.8597e-01, -1.1090e-01, -1.3520e-02, -3.8995e-01,\n",
       "           1.1861e-01,  2.4902e-02,  5.2355e-02,  3.9441e-02, -2.2822e-01,\n",
       "           2.3118e-01,  4.8091e-02,  4.5015e-01, -2.8330e-01, -7.0757e-01,\n",
       "          -1.9601e-01, -1.5070e-02,  1.7927e-02, -1.7013e-02, -2.8357e-01,\n",
       "           6.6553e-02, -5.8297e-01,  2.2333e-01,  1.1373e-01,  8.5199e-01,\n",
       "           9.8358e-01,  6.7080e-01,  7.6940e-02, -5.4386e-02, -1.6916e-02,\n",
       "          -5.6321e-01,  1.6130e-01, -8.1082e-01, -1.3062e-01, -2.1928e-01,\n",
       "           7.0352e-01, -3.6822e-01,  1.1476e-01,  2.6222e-01, -3.4481e-01,\n",
       "           2.0382e-01,  3.7136e-02,  3.1790e-01, -4.3110e-01, -5.0590e-01,\n",
       "           5.2232e-01, -7.6970e-03,  1.8870e-01, -5.3201e-01,  2.1314e-01,\n",
       "          -3.1748e-01,  2.6052e-01,  1.8880e-01, -5.2874e-01, -2.0760e-01,\n",
       "          -4.9469e-01,  1.3613e-01,  8.4277e-01,  1.4568e-01,  7.7079e-02,\n",
       "           3.6946e-01,  2.0022e-01, -4.6711e-01, -7.1561e-01, -4.9342e-01,\n",
       "           2.4731e-01, -2.3498e-01,  3.4844e-01,  5.7382e-01, -7.8887e-02,\n",
       "          -4.7290e-01,  1.1404e-01,  1.6852e-01,  2.4731e-01,  6.4383e-01,\n",
       "          -3.0459e-01, -4.1848e-02, -2.9485e-02,  9.0120e-02, -3.1542e-01,\n",
       "          -3.7731e-01, -3.2564e-01, -1.7041e-01,  4.5536e-01, -4.5159e-02,\n",
       "           3.8940e-01,  2.2186e-01,  5.9770e-02,  1.9004e-01, -8.8353e-01,\n",
       "           3.4893e-01, -2.2614e-01,  3.7178e-01, -1.8391e-01, -4.1715e-01,\n",
       "           2.0961e-01,  5.8875e-01, -2.8693e-01, -4.0336e-01, -2.5849e-01,\n",
       "           2.3852e-01,  1.3124e-01, -1.3747e-01, -8.3652e-01,  1.2572e-01,\n",
       "          -6.3280e-01, -8.3095e-02, -2.7914e-02,  5.4686e-01,  8.0290e-01,\n",
       "           2.7624e-01, -6.7492e-02,  8.6376e-02, -3.8386e-01, -6.1207e-01,\n",
       "          -6.3800e-01,  1.3163e-02, -1.3512e-01, -7.8098e-01,  2.1135e-02,\n",
       "           2.8876e-02, -2.0300e-02,  5.8427e-02, -2.3546e-01,  1.7729e-01,\n",
       "           4.8705e-01,  3.6141e-01, -1.2408e-01,  5.9419e-02, -3.7922e-01,\n",
       "          -6.2190e-01, -8.6745e-02,  7.7854e-02,  1.6172e-01, -5.1278e-01,\n",
       "          -2.9601e-01,  1.1747e-01, -3.1322e-02,  2.7528e-02, -1.9899e-01,\n",
       "           1.3669e-01, -2.2503e-01, -1.2000e-02,  2.7866e-01,  4.6635e-01,\n",
       "           1.1528e-02, -2.7282e-01, -4.3857e-01, -1.8906e-02,  1.5881e-01,\n",
       "           2.4698e-01, -2.2711e-01, -3.0300e-01, -3.8354e-01,  2.0940e-01,\n",
       "           2.9705e-01,  1.0639e+00, -4.8833e-01, -4.0899e-01,  1.9583e-01,\n",
       "           3.5230e-01, -1.0565e-01, -2.8806e-01, -3.5628e-01,  2.0647e-01,\n",
       "           1.1352e-01, -8.0433e-02, -3.4162e-02, -7.7830e-02,  5.1550e-01,\n",
       "           1.6482e-01,  4.7041e-01,  3.0757e-01,  4.8310e-01,  1.9579e-01,\n",
       "          -1.9577e-01, -8.1087e-02, -6.1276e-01, -5.7265e-01,  1.7477e-01,\n",
       "          -3.4696e-02,  1.7945e-01,  2.9475e-01, -5.4680e-01,  3.5253e-01,\n",
       "           3.9298e-01,  1.0477e+00,  3.8225e-02, -1.9548e-01,  4.0182e-01,\n",
       "           3.8969e-01, -4.5758e-01, -5.5331e-02,  3.1775e-01,  9.0078e-02,\n",
       "           4.2217e-01,  2.6969e-01, -2.5045e-01, -2.6832e-01,  1.3747e-01,\n",
       "           3.6461e-01, -5.7917e-01, -3.7681e-01, -1.5167e-02,  5.5110e-01,\n",
       "          -2.4375e-01,  2.1840e-01,  4.5848e-02,  3.7804e-01,  3.7463e-01,\n",
       "          -1.0699e+00,  1.6807e-01, -3.2041e-01,  2.7435e-01, -2.4271e-02,\n",
       "          -7.8250e-01, -3.4543e-02, -1.9427e-01, -1.1596e-01,  3.3735e-01,\n",
       "           4.1202e-01, -7.9805e-01,  3.3972e-01,  2.3964e-01, -3.8364e-01,\n",
       "           4.7577e-02, -1.7531e-01,  2.4049e-01,  3.3544e-01,  3.2362e-02,\n",
       "          -1.4892e-01,  1.8644e-01, -1.9447e-01, -8.0775e-01,  5.7712e-01,\n",
       "           6.0096e-01, -2.1005e-01,  1.4164e-01,  1.1858e-01, -8.4184e-02,\n",
       "          -3.2385e-01, -3.8343e-01, -1.2416e-01,  1.6063e-01,  9.6145e-02,\n",
       "          -9.8046e-02, -4.7037e-01, -4.0440e-02,  3.8794e-01, -3.9096e-01,\n",
       "           4.3372e-01,  3.4804e-01,  5.3842e-01, -2.8332e-01, -4.2575e-02,\n",
       "           1.7443e-01,  5.7542e-01, -1.1207e-01,  3.8094e-01, -3.8443e-01,\n",
       "          -8.6311e-03,  4.5158e-01, -1.4772e-01,  7.1035e-03,  2.3732e-01,\n",
       "          -5.0991e-01,  2.9103e-01,  1.0851e-01, -5.6297e-01,  5.5767e-01,\n",
       "           5.0509e-01, -7.6896e-02, -4.0047e-01,  1.5744e-01,  2.9919e-01,\n",
       "           3.9453e-02,  7.8684e-01,  2.8627e-01,  3.8282e-01,  2.7073e-01,\n",
       "          -6.7450e-02, -5.4737e-01, -3.2683e-01, -5.6482e-03,  2.6335e-01,\n",
       "           1.1920e-01, -3.6386e-01,  3.2594e-01,  1.8073e-01,  5.2975e-01,\n",
       "           6.1907e-01, -1.8066e-01, -9.0544e-01, -5.6162e-01,  9.6928e-02,\n",
       "          -1.8944e-01, -1.1782e-01, -1.4313e-01,  5.4460e-02,  1.6554e-01,\n",
       "          -7.7022e-01, -5.7397e-01, -1.9516e-01,  2.1820e-01, -3.5735e-01,\n",
       "           6.6382e-01,  4.6434e-02, -3.8504e-02,  3.3171e-01,  3.8933e-01,\n",
       "           2.1859e-01,  3.6118e-01, -5.0593e-02, -5.7660e-01,  1.3136e-02,\n",
       "           9.6468e-02, -4.2191e-01, -1.8386e-01, -4.0805e-01,  5.0626e-01,\n",
       "           3.3973e-01,  9.6001e-01,  5.3169e-01]], device='mps:0'),\n",
       " tensor([[-4.7052e-03, -2.8964e-01, -4.8786e-01, -7.0258e-01, -5.2877e-01,\n",
       "          -3.8291e-02, -5.6498e-01,  6.0469e-01,  3.2463e-02,  4.2240e-01,\n",
       "           3.3467e-01,  4.3544e-01,  1.6649e-02,  2.1653e-01, -8.9128e-01,\n",
       "          -2.2569e-01,  2.5929e-02,  3.1582e-01,  6.6690e-01,  5.4626e-01,\n",
       "           6.7861e-01,  1.5369e-01,  1.5123e-01, -5.4915e-02, -8.2267e-01,\n",
       "           2.9110e-01, -1.2212e-01,  1.9921e-02,  9.3266e-02, -4.1448e-01,\n",
       "          -3.3536e-01,  4.0438e-01, -1.5177e-01,  9.1833e-01, -2.1048e-01,\n",
       "          -2.1881e-01,  2.0108e-01, -3.0485e-01,  3.3645e-01, -1.9360e-01,\n",
       "           4.5021e-01, -6.4852e-02,  2.7733e-01,  5.2829e-02,  3.7394e-01,\n",
       "           8.6825e-02,  8.8577e-02, -4.6033e-01, -8.2400e-02, -3.0934e-01,\n",
       "           5.5321e-03, -1.8231e-01,  2.6481e-01,  1.0155e-01, -6.6594e-01,\n",
       "          -6.4626e-01,  7.3161e-02,  3.0356e-01, -7.6759e-01, -3.3102e-02,\n",
       "          -3.5284e-01,  8.8272e-01, -1.4331e-01,  1.2698e-01, -1.2803e-01,\n",
       "           3.1562e-02,  1.6231e-01, -8.0131e-01, -6.2177e-01, -2.0136e-01,\n",
       "          -2.9173e-03,  1.7380e-01, -5.9514e-02, -2.9058e-01,  6.6827e-01,\n",
       "          -2.6645e-01,  2.5016e-01, -1.4395e-01, -3.9422e-01, -2.0033e-01,\n",
       "           1.6207e-01,  2.7108e-01, -5.8034e-03, -1.1140e-01, -6.2601e-01,\n",
       "          -2.7907e-02,  2.8242e-01,  5.8652e-01,  3.0536e-01, -2.9037e-01,\n",
       "           2.5447e-01,  8.4271e-02, -1.9337e-01, -1.2893e-01,  4.7694e-01,\n",
       "          -1.8865e-01,  5.7017e-02, -5.7121e-02,  2.4393e-01, -5.5938e-01,\n",
       "           1.7060e-01,  3.4899e-01, -1.4797e-01,  3.4014e-01, -5.0014e-01,\n",
       "          -7.4363e-01,  3.7880e-01,  2.5241e-02, -9.1301e-01, -1.9715e-01,\n",
       "          -9.9766e-02, -4.2456e-01,  2.7552e-01, -5.0931e-01, -3.5171e-01,\n",
       "          -3.9503e-02, -4.4978e-02,  9.7122e-02, -4.4658e-01,  3.3929e-01,\n",
       "           3.4658e-01, -2.5348e-01,  7.5713e-02,  1.1001e-01,  1.1829e-02,\n",
       "          -3.4932e-02,  1.6259e-01, -3.4304e-02,  4.4033e-01,  3.2669e-01,\n",
       "           6.7625e-02,  1.4640e-01, -5.5088e-01,  4.3575e-02, -9.6247e-04,\n",
       "           1.2883e-01, -2.8999e-01, -2.1327e-01,  4.5755e-01,  8.0405e-02,\n",
       "          -2.1551e-01,  6.7557e-02, -3.2944e-02,  8.2292e-02,  1.4665e-01,\n",
       "          -5.9250e-02,  2.9072e-01, -2.1572e-01, -2.7339e-01, -1.5397e-01,\n",
       "          -3.0666e-01, -6.5934e-01,  1.8231e-01, -1.7701e-01,  1.4187e-02,\n",
       "          -2.6869e-01,  7.2470e-01, -4.8729e-02, -2.7939e-02,  2.8769e-01,\n",
       "           5.3997e-01, -1.8589e-02,  1.9243e-01, -1.4651e-01,  5.6034e-02,\n",
       "           3.3883e-01, -3.0829e-01, -3.8587e-01, -3.8174e-01,  8.6224e-02,\n",
       "           9.3780e-02, -3.0712e-01,  1.6981e-01, -1.4381e-01, -2.2844e-01,\n",
       "          -4.8882e-01,  1.1794e-01, -7.5567e-02, -3.2981e-01,  3.7590e-01,\n",
       "          -5.9765e-01, -1.6226e+00,  2.7806e-01,  2.9293e-01, -4.5808e-01,\n",
       "          -5.6601e-01,  3.7231e-01,  4.1788e-01,  3.5291e-01, -1.2671e-01,\n",
       "           2.9683e-01, -3.3162e-01,  2.7602e-01,  6.1184e-01, -3.1995e-01,\n",
       "           4.1990e-01,  3.7486e-01, -1.5027e-02,  5.7791e-02,  3.4821e-02,\n",
       "          -7.5119e-01,  4.9343e-02,  2.6673e-01, -1.2592e-01,  6.4485e-02,\n",
       "           1.2320e-01,  3.8651e-01, -1.6197e-01,  5.6027e-02,  2.1140e-01,\n",
       "           9.4408e-02, -2.2668e-01, -8.9645e-02, -8.8328e-02,  5.6669e-02,\n",
       "           4.3653e-01,  1.6818e-02,  5.6391e-02,  6.3548e-01,  3.6890e-03,\n",
       "          -3.0647e-01,  1.4464e-01, -1.9891e-01, -1.1201e-01, -4.0697e-01,\n",
       "           2.1326e-01,  3.1009e-02,  3.0046e-02, -5.1332e-01, -4.5532e-01,\n",
       "          -1.0647e-01, -4.3802e-01,  3.6539e-01,  4.6897e-01, -3.1519e-01,\n",
       "          -2.4277e-01, -2.8747e-01, -7.1588e-02,  1.1594e-01,  2.5484e-01,\n",
       "          -1.9323e-01, -2.8014e-01, -6.9694e-01, -3.9154e-02,  1.6952e-01,\n",
       "           2.7978e-01, -8.7181e-02,  5.5669e-01,  5.1616e-01,  2.2391e-01,\n",
       "           7.2825e-02, -1.7346e-01,  4.4753e-01,  3.3826e-01, -2.7248e-01,\n",
       "           2.8831e-01,  1.3558e-01, -5.8160e-02,  4.0515e-01, -4.4470e-01,\n",
       "           5.1072e-02,  1.1007e-01,  5.6167e-02,  5.2287e-01, -5.5110e-01,\n",
       "           7.1630e-02,  3.6739e-01, -3.1045e-01, -2.1033e-02,  1.3506e-01,\n",
       "          -3.9937e-01,  6.4969e-01,  2.6148e-01, -5.0733e-01, -6.9277e-01,\n",
       "          -3.3234e-01,  1.6590e-01, -1.3700e-01, -4.8068e-02, -8.9607e-01,\n",
       "          -2.3665e-01,  1.6386e-02,  7.7052e-02, -1.2783e-01, -2.8466e-01,\n",
       "           1.6166e-01,  3.8236e-01, -5.0754e-01, -6.4609e-02, -6.4521e-01,\n",
       "           6.0019e-02, -6.9526e-02, -2.9084e-01,  5.5703e-01,  2.5027e-01,\n",
       "          -2.5521e-01,  1.5577e-01,  5.8269e-02,  1.9295e-01, -3.3633e-01,\n",
       "          -1.0639e-01, -2.7958e-01,  6.8882e-02,  1.3967e-02,  1.5194e-01,\n",
       "           4.0629e-01,  2.8010e-01,  8.6252e-01, -6.3028e-01, -7.1276e-01,\n",
       "          -2.6839e-01,  7.7791e-01, -1.4459e-01, -2.5539e-01, -2.0474e-01,\n",
       "          -1.1396e+00, -6.5938e-02, -1.5500e-01, -3.7426e-01, -4.6809e-01,\n",
       "           2.8563e-01, -1.6780e-01, -2.2452e-01,  3.7282e-02, -3.2439e-01,\n",
       "           5.9310e-01,  7.5024e-01, -5.5199e-01, -9.2968e-02, -2.0907e-01,\n",
       "          -4.8898e-01, -5.8562e-02, -1.7874e-01, -2.5544e-01, -1.7150e-01,\n",
       "           5.7639e-02,  1.1225e-01,  1.6898e-02, -1.2802e-01, -4.6394e-01,\n",
       "           3.7808e-01,  9.5179e-03,  2.2656e-01,  3.3500e-02, -3.1722e-01,\n",
       "           4.5618e-01,  5.1201e-01, -8.3795e-01, -1.5467e-01, -1.1835e-01,\n",
       "          -7.3888e-02, -2.4488e-01, -3.6613e-01,  2.6136e-02,  6.0463e-01,\n",
       "           4.5444e-01,  2.6448e-01, -3.5098e-01,  6.2728e-01,  4.0659e-01,\n",
       "          -4.2125e-01,  5.9106e-01,  3.9261e-04, -3.6594e-01,  2.3924e-01,\n",
       "           1.7602e-01,  1.3159e-01, -3.7787e-01,  7.5391e-02, -2.0637e-01,\n",
       "           3.2059e-01, -6.8267e-01,  1.1234e-01,  1.7178e-01,  7.1229e-01,\n",
       "          -3.1058e-01, -4.7909e-01,  5.7494e-01, -4.7029e-01, -3.1743e-01,\n",
       "           4.0040e-02,  9.7216e-02,  3.2540e-02,  4.1689e-02, -1.8212e-01,\n",
       "          -3.2061e-01,  2.2126e-01,  9.4326e-01,  3.9329e-01, -3.3742e-01,\n",
       "           1.3991e-01, -1.9865e-01, -2.4052e-01,  4.4569e-01, -2.5823e-01,\n",
       "          -9.0030e-02,  6.8934e-01, -2.0009e-01,  9.8263e-02,  2.8468e-01,\n",
       "          -2.3035e-01, -2.9077e-01, -2.4253e-01, -3.9984e-01, -3.1827e-02,\n",
       "          -2.7401e-01,  1.7750e-01, -3.2325e-01,  3.5016e-02, -6.2991e-02,\n",
       "          -7.3809e-02,  2.6584e-01,  8.8470e-02,  8.3958e-02, -2.4245e-01,\n",
       "           8.6168e-02, -2.5228e-01,  2.3715e-01, -2.2619e-02, -4.3186e-01,\n",
       "           6.6163e-01,  2.9422e-01,  4.1671e-01,  6.0674e-01, -1.1291e+01,\n",
       "          -9.1759e-02,  3.4720e-01, -2.9895e-01, -9.0161e-02,  1.4151e-02,\n",
       "           5.2342e-02, -2.9192e-01,  4.4950e-02, -1.5200e-01, -2.4135e-01,\n",
       "          -7.4682e-02,  5.9989e-02,  1.0660e-01,  6.3530e-02, -2.6779e-01,\n",
       "          -6.8554e-02, -7.9546e-01,  1.0541e-01,  1.1291e-02,  1.8130e-01,\n",
       "          -2.2885e-01,  1.6540e-01,  1.8337e-01, -7.3581e-02,  2.4724e-01,\n",
       "          -1.3380e-01, -1.1550e-01,  5.3803e-01,  3.5058e-01,  1.9950e-01,\n",
       "           2.8758e-01,  2.0595e-01, -1.6196e-01, -3.6112e-01, -3.6233e-01,\n",
       "          -2.4526e-01, -2.4733e-01, -8.8502e-02,  7.0408e-02, -1.8735e-01,\n",
       "          -1.0226e-01, -3.1100e-01,  2.5691e-01,  4.3285e-01,  7.9375e-01,\n",
       "           1.3030e+00,  5.2712e-01,  4.4888e-01,  1.6357e-01,  7.4202e-02,\n",
       "          -8.3520e-01,  2.7388e-01, -6.1586e-01, -5.3542e-03, -1.8850e-01,\n",
       "           8.4532e-01, -5.4538e-01, -1.5915e-02, -4.5384e-01, -2.5253e-01,\n",
       "           2.9089e-01,  3.1911e-01, -2.7775e-01, -4.4961e-01,  1.3650e-01,\n",
       "           2.1080e-01, -2.4010e-01,  1.2801e-01, -9.1529e-02,  2.5831e-01,\n",
       "          -1.6036e-01,  2.1684e-01,  5.5463e-02, -5.2496e-01,  1.4923e-01,\n",
       "           2.9207e-01, -9.2721e-02,  1.2220e+00, -1.4293e-03, -1.6301e-01,\n",
       "           3.1573e-02, -3.8681e-02, -2.5054e-01, -9.1565e-01, -2.3575e-01,\n",
       "           2.9389e-01,  6.1115e-03,  3.9116e-01,  2.0628e-01, -2.3681e-01,\n",
       "          -8.6150e-01, -1.6264e-01,  1.6574e-01,  5.4260e-01,  3.5161e-01,\n",
       "          -9.3751e-01, -1.3459e-01,  1.0793e-01,  1.8788e-02, -5.5041e-01,\n",
       "           8.2199e-02, -3.8965e-01, -1.7687e-01,  1.1790e-01,  3.7443e-02,\n",
       "           7.4850e-02, -2.2264e-01, -8.1625e-02,  1.2454e-01, -3.6823e-01,\n",
       "           1.6344e-01,  2.1583e-01,  8.4510e-03, -8.7869e-02,  1.1922e-01,\n",
       "           3.7711e-02,  5.9793e-01, -1.8163e-01, -5.5343e-01,  5.8368e-02,\n",
       "          -1.3352e-01,  2.5648e-01,  7.7454e-02,  3.8482e-02,  6.6556e-01,\n",
       "          -3.5589e-01, -2.7879e-02,  3.1497e-01,  4.5051e-01,  4.7129e-01,\n",
       "           1.5801e-02, -1.9220e-01,  3.4147e-01, -7.9931e-01, -1.0203e+00,\n",
       "          -6.3159e-01,  2.0522e-01, -8.4502e-02, -3.0027e-01, -2.5152e-01,\n",
       "           1.1410e-01,  2.1459e-01, -2.9448e-01,  2.0046e-01,  2.9798e-01,\n",
       "           7.7084e-01,  5.4432e-01, -3.8212e-01,  1.5064e-01, -3.3429e-01,\n",
       "          -5.5963e-01, -1.9811e-01, -1.7518e-01, -2.6842e-02, -4.8014e-01,\n",
       "          -3.2286e-01, -1.7132e-01, -3.6921e-01, -1.7506e-01,  2.8414e-01,\n",
       "          -2.9731e-01,  7.1301e-02,  5.9975e-02,  1.7024e-01,  3.7358e-01,\n",
       "           3.0326e-01, -2.3328e-01,  1.8896e-01,  7.9992e-02,  5.2761e-01,\n",
       "           9.4885e-02,  3.3434e-01, -5.4254e-01, -5.0649e-02, -1.5205e-02,\n",
       "           2.4292e-01,  8.0821e-01, -1.0107e-01, -1.0328e-01, -4.7611e-03,\n",
       "           3.6858e-01, -8.3820e-02, -1.4482e-01,  1.3701e-01, -2.4956e-01,\n",
       "           9.2389e-03, -4.5992e-01,  3.5870e-01,  1.3574e-01,  4.3240e-01,\n",
       "           1.7296e-01, -2.2619e-01, -1.0825e-01,  3.7505e-01, -8.8768e-02,\n",
       "          -3.9186e-01, -1.5501e-01, -2.5055e-01,  5.2027e-02, -5.6252e-02,\n",
       "          -3.4667e-01, -7.6041e-02, -2.9249e-01,  4.3328e-02,  2.5671e-01,\n",
       "           1.9691e-02,  1.3760e+00,  1.0725e-01, -5.2535e-01,  1.0009e+00,\n",
       "           3.2173e-01, -2.6737e-02,  3.4089e-01, -1.5695e-01,  3.8457e-01,\n",
       "           4.4414e-01, -1.8778e-01,  1.2676e-01,  1.2305e-01,  1.4650e-01,\n",
       "           5.4956e-01, -8.2059e-01, -3.4561e-01, -1.3916e-01,  2.2998e-01,\n",
       "          -1.8918e-01,  4.2392e-01,  8.7230e-02,  3.7625e-01,  6.9564e-01,\n",
       "          -1.0938e+00, -1.2957e-01, -4.3566e-01,  4.9788e-01,  1.8736e-01,\n",
       "          -3.2428e-01, -1.2393e-01, -1.9897e-01, -5.0226e-01,  3.5444e-01,\n",
       "           7.3221e-01, -1.0105e+00,  1.3526e-01,  4.9224e-01, -1.4000e-02,\n",
       "           2.3589e-01, -2.4623e-01,  2.3014e-01, -2.5124e-01,  3.4803e-01,\n",
       "          -2.9266e-02,  1.7715e-01, -3.7720e-01, -8.1968e-01,  2.4956e-01,\n",
       "           5.3095e-01,  2.5859e-01,  9.7470e-02,  2.5063e-02,  7.1963e-02,\n",
       "          -1.8989e-01, -2.7334e-01, -2.3794e-01,  3.0958e-01, -2.2524e-01,\n",
       "           9.2092e-02, -2.8657e-01,  1.8384e-01,  1.5047e-01,  2.6005e-02,\n",
       "           2.4709e-01,  3.9804e-01,  7.6380e-01, -1.4634e-01, -2.0663e-01,\n",
       "           3.2347e-01,  4.5175e-01, -1.2248e-01,  5.9067e-01, -2.1228e-01,\n",
       "          -4.4475e-01,  4.1615e-01,  1.8901e-01, -2.4644e-01,  4.1677e-01,\n",
       "          -3.4223e-01,  7.6530e-02,  4.0049e-02, -2.9351e-01,  1.4830e-01,\n",
       "           5.4184e-03, -1.8467e-02, -4.3710e-01,  7.1453e-02,  3.8704e-01,\n",
       "          -1.8139e-01,  9.8383e-01, -2.7297e-01,  5.4875e-01, -7.1134e-01,\n",
       "           3.9573e-01, -9.0719e-01, -1.5457e-01, -7.6269e-02, -4.7067e-01,\n",
       "          -5.0326e-01, -2.3791e-01,  1.5941e-01,  3.3386e-01,  6.7009e-01,\n",
       "           2.3475e-01, -2.2873e-01, -6.4823e-01, -3.3609e-01,  3.7911e-01,\n",
       "          -1.7678e-01,  6.2914e-02,  2.9701e-01,  1.4306e-01,  3.8930e-01,\n",
       "          -6.9229e-01, -3.7959e-01, -4.0079e-02,  3.6859e-01, -2.8177e-01,\n",
       "           7.1107e-01,  4.8831e-02,  6.2038e-03,  2.0659e-01,  3.1494e-02,\n",
       "           3.6890e-01,  4.3326e-01, -3.5486e-01, -7.3276e-01, -1.4664e-01,\n",
       "          -8.3731e-02, -5.6326e-01, -3.5421e-02, -9.6357e-01,  2.6725e-01,\n",
       "           4.0771e-01,  5.6247e-01,  6.5069e-01]], device='mps:0'),\n",
       " tensor([[-0.4654, -0.4115, -0.4370,  ...,  0.1625,  0.4338,  0.5908],\n",
       "         [ 0.0130, -0.7446, -0.5741,  ...,  0.2998,  0.4681,  0.4815]],\n",
       "        device='mps:0'),\n",
       " tensor([[-3.0527e-01, -5.0767e-01, -5.4826e-01, -8.1678e-01, -2.0262e-01,\n",
       "          -3.4384e-01, -5.1098e-01,  4.4753e-01, -2.2169e-01,  2.5475e-01,\n",
       "           1.1888e-01, -1.6182e-01,  7.3755e-02,  2.0782e-01, -6.0317e-01,\n",
       "           5.7815e-02,  1.3021e-01,  1.4951e-01,  5.0639e-01,  6.0814e-02,\n",
       "           9.6302e-01,  4.7046e-01,  3.4311e-02, -6.2618e-02, -5.8057e-01,\n",
       "          -6.0525e-03, -2.9453e-01,  2.6144e-01,  2.2644e-01, -4.5028e-01,\n",
       "          -9.1428e-02,  2.4006e-01, -2.9972e-01,  1.1072e+00, -6.8620e-02,\n",
       "          -8.8843e-02,  5.5203e-02, -7.0518e-01,  2.1604e-01, -2.9081e-01,\n",
       "          -5.1905e-02,  6.3968e-02,  4.0090e-01,  5.6925e-01,  3.6641e-01,\n",
       "           1.5456e-01, -9.3559e-02, -3.0134e-01, -4.9171e-02, -3.1805e-01,\n",
       "          -2.7202e-01,  4.4876e-02,  2.8892e-01,  3.4796e-01, -6.8269e-01,\n",
       "          -4.3743e-01,  7.0707e-01,  2.7768e-01, -4.1565e-01,  2.1039e-01,\n",
       "          -3.4419e-01,  6.7574e-01,  4.7932e-02, -6.1339e-03, -5.2823e-02,\n",
       "           2.1578e-02, -3.3343e-01, -7.9349e-01, -1.1822e-01, -2.1921e-01,\n",
       "           2.4821e-02, -8.8630e-02,  1.1309e-01, -2.3542e-01, -1.1408e-01,\n",
       "           4.7945e-02,  1.9183e-01,  5.5437e-02, -5.8177e-01, -2.2593e-03,\n",
       "          -4.8062e-02, -1.4935e-01,  4.2555e-01, -1.8557e-01, -2.6337e-01,\n",
       "           2.5821e-02,  3.3047e-01,  2.0148e-01,  4.2564e-01,  6.1848e-02,\n",
       "           6.3744e-01, -1.8943e-01, -3.0159e-02, -7.0648e-01,  5.0329e-01,\n",
       "          -1.5119e-01,  2.8734e-01, -9.8546e-02,  5.6219e-02, -4.2336e-01,\n",
       "          -3.5849e-02,  1.0008e-01, -1.1925e-01, -1.3023e-01, -2.0746e-01,\n",
       "          -4.6692e-01,  3.3509e-01,  2.1711e-01, -1.1894e+00, -2.4856e-02,\n",
       "           7.9642e-02, -2.0049e-01, -1.5365e-01, -6.3680e-01, -2.7360e-01,\n",
       "          -1.1119e-01,  2.5672e-02, -5.0558e-02, -2.8266e-01,  5.5576e-01,\n",
       "           4.1733e-01, -1.5723e-01, -1.4753e-01,  1.0433e-01,  2.0569e-01,\n",
       "          -5.0770e-01,  2.0889e-01,  1.0338e-01,  3.6859e-02,  3.4040e-01,\n",
       "           7.8629e-03, -1.3139e-01, -3.0936e-01, -1.1731e-01,  1.7506e-01,\n",
       "          -1.7897e-01,  3.3063e-01, -1.8507e-01, -9.5570e-02, -1.0367e-01,\n",
       "          -3.6557e-01,  7.1585e-02,  7.8899e-02, -1.3873e-01,  1.5008e-02,\n",
       "          -4.3042e-01,  2.6277e-01,  2.5443e-03, -7.3255e-02, -1.6284e-01,\n",
       "          -4.8917e-01, -3.7474e-01,  4.1611e-01, -7.9513e-02, -2.4438e-02,\n",
       "          -5.6420e-01,  8.7666e-01, -3.9440e-02,  1.5899e-01,  1.7491e-02,\n",
       "           8.9089e-01,  1.6844e-01,  2.6206e-01, -4.5416e-01, -1.8454e-01,\n",
       "           2.7490e-01, -1.9658e-01, -2.0556e-01, -1.0123e-01,  2.1526e-01,\n",
       "           2.6036e-01,  2.2029e-03,  3.9397e-01, -7.5508e-02, -1.8033e-01,\n",
       "          -3.2681e-01,  2.3943e-01,  2.9262e-01, -1.4561e-01,  3.3078e-01,\n",
       "           1.2514e-01, -1.5132e+00,  3.3903e-01, -3.9769e-01, -6.0739e-01,\n",
       "          -8.0181e-02,  2.8904e-01,  7.6061e-02,  3.2670e-01,  9.1556e-02,\n",
       "          -1.7607e-01,  9.4792e-04,  1.2908e-01,  3.5186e-01, -1.3683e-01,\n",
       "           1.9703e-01,  5.1852e-01,  7.6627e-02, -5.6828e-02, -2.6248e-02,\n",
       "          -4.0614e-01,  4.5635e-01, -7.8520e-02,  8.2835e-01,  3.0775e-01,\n",
       "           2.0302e-01,  8.9879e-02,  1.4894e-01,  3.8968e-01,  1.0700e-01,\n",
       "           5.1331e-02, -1.0860e-01, -6.6148e-01, -1.3455e-01, -4.5611e-01,\n",
       "           4.9172e-01, -3.3636e-01, -4.9978e-01,  1.2000e-01,  2.8709e-01,\n",
       "          -2.0632e-01, -8.1132e-02, -6.9061e-03, -1.7189e-01, -2.6253e-01,\n",
       "           1.0805e+00, -1.7711e-01,  3.2960e-02,  1.4755e-01,  9.2513e-02,\n",
       "          -9.4842e-02, -7.9861e-01,  5.8281e-01,  3.2999e-01,  3.1799e-02,\n",
       "          -1.7208e-01, -3.6683e-01,  3.9288e-01, -8.8705e-02, -1.1671e-01,\n",
       "          -2.4913e-01, -4.6160e-01, -5.3674e-01, -3.1948e-01,  1.9893e-01,\n",
       "           2.2246e-01, -1.6421e-01,  6.1919e-01,  5.8383e-01,  2.9446e-01,\n",
       "          -2.8869e-01, -2.1954e-01, -8.7718e-02,  7.6787e-02, -2.8610e-01,\n",
       "           1.5097e-01,  7.7329e-02, -5.3462e-01,  3.6507e-01, -3.6626e-01,\n",
       "          -4.2319e-01, -9.3988e-02,  2.7016e-01,  4.3481e-01, -4.2295e-01,\n",
       "           4.3312e-01,  2.1454e-01, -1.3488e-01,  3.5386e-01,  2.0873e-01,\n",
       "          -1.9545e-01,  5.2453e-01,  4.6953e-01, -9.8075e-01, -4.2734e-01,\n",
       "          -4.0591e-01, -3.9144e-01,  2.6483e-01, -2.0910e-01, -3.9558e-01,\n",
       "          -9.3475e-03, -2.7615e-02, -1.5159e-01, -2.1858e-01, -1.1814e-01,\n",
       "          -1.0013e-01,  2.2221e-01, -6.0817e-02, -1.8369e-01, -3.0054e-01,\n",
       "          -5.5287e-01,  1.2606e-01,  4.8161e-02,  5.2273e-02,  2.0814e-01,\n",
       "           5.4045e-03,  1.3845e-01,  2.3043e-01,  3.6255e-01, -1.4606e-01,\n",
       "          -2.6486e-01, -3.5285e-01,  5.1076e-01, -3.9188e-01,  9.8662e-02,\n",
       "           1.7789e-01,  6.1696e-02,  6.0978e-01, -1.2674e-01, -5.4442e-01,\n",
       "          -8.8265e-02,  7.8787e-01, -4.9785e-03, -2.8758e-01,  1.8791e-02,\n",
       "          -1.0020e+00, -2.1507e-01,  7.2948e-02,  4.0501e-03, -4.3006e-01,\n",
       "           1.3914e-01, -2.7258e-01, -2.8297e-01,  3.8414e-01, -9.7462e-02,\n",
       "           5.7410e-01,  8.6555e-01, -2.4044e-01, -4.6988e-02, -4.1481e-02,\n",
       "          -2.0422e-01, -9.0367e-02, -8.9520e-02, -2.7225e-01, -4.0525e-01,\n",
       "          -3.4472e-03,  4.0062e-01,  2.5179e-01,  1.9823e-01, -2.4842e-01,\n",
       "           6.3762e-01,  2.3478e-02,  4.6470e-02,  3.4225e-01, -4.6188e-01,\n",
       "           7.1827e-02,  3.5409e-01, -1.0316e+00, -5.5195e-03, -7.0413e-02,\n",
       "           2.1759e-01, -6.3729e-01, -1.2032e-01,  5.0954e-02,  4.4042e-01,\n",
       "           3.2636e-01,  2.2479e-01,  3.8254e-01,  5.9016e-01,  5.7894e-01,\n",
       "          -2.5902e-01,  5.5651e-01,  1.7059e-01, -4.3517e-01, -3.9112e-01,\n",
       "           4.2695e-01,  1.9891e-01, -7.6350e-01, -3.6127e-01, -3.3999e-01,\n",
       "           1.8391e-01, -7.6569e-01,  2.1020e-01,  7.9145e-02,  1.1698e-01,\n",
       "          -2.9001e-01, -7.5465e-01,  5.5427e-02, -8.7542e-01, -4.8592e-01,\n",
       "           7.0627e-02, -6.4067e-02,  1.9579e-01, -4.3997e-01, -2.4823e-01,\n",
       "          -3.4863e-01,  2.5193e-02,  3.1886e-01,  4.9269e-01, -1.7228e-01,\n",
       "           2.5469e-01,  1.8219e-01, -1.6011e-01, -2.7885e-02, -1.8336e-01,\n",
       "          -1.0676e-01,  1.8495e-01, -1.5903e-01, -6.1748e-02,  3.1832e-01,\n",
       "          -3.6281e-01,  2.3495e-01,  1.7155e-01, -1.2138e-03,  1.0303e-01,\n",
       "          -4.5579e-01,  1.1581e-01, -1.1956e-01, -4.4599e-01,  2.9208e-01,\n",
       "          -3.3458e-01,  1.0199e-01, -4.3284e-01,  4.9563e-01,  2.3482e-01,\n",
       "           1.8214e-01, -2.2978e-01, -6.6505e-02, -2.0288e-01, -3.4091e-01,\n",
       "           5.5028e-01,  2.0809e-01,  1.8878e-01,  3.2488e-01, -1.1158e+01,\n",
       "          -1.7915e-01,  9.0973e-01, -3.0310e-01, -4.7787e-01, -1.0118e-01,\n",
       "           1.0664e-01, -5.6257e-01,  3.1451e-01, -1.8771e-01, -4.2395e-01,\n",
       "          -2.9024e-01, -5.5197e-02,  1.5427e-01, -4.4419e-01, -6.6630e-01,\n",
       "          -1.6191e-01, -7.1028e-01,  1.2267e-01, -2.3634e-01,  2.6155e-01,\n",
       "          -7.0415e-02,  3.6331e-01,  4.9140e-01, -3.0460e-01, -4.5503e-01,\n",
       "          -2.4066e-01, -1.2548e-01,  2.3188e-01,  4.4537e-01, -4.1870e-02,\n",
       "          -2.0463e-03,  4.0884e-01, -9.1099e-02, -1.1565e-01, -5.6082e-01,\n",
       "          -2.5906e-01, -1.7325e-01,  6.6560e-01, -1.3452e-01, -5.8319e-01,\n",
       "          -4.7523e-01, -1.2258e-01,  2.3167e-01,  3.5006e-01,  6.8819e-01,\n",
       "           1.2556e+00,  4.8780e-01,  5.4290e-01, -5.3707e-02, -3.8721e-01,\n",
       "          -9.5886e-01,  1.9823e-01, -3.1037e-01, -1.0214e-01, -1.7801e-01,\n",
       "           6.3427e-01, -6.7212e-01, -8.8442e-02, -3.0560e-01, -3.3437e-01,\n",
       "           4.9204e-01,  1.2309e-01,  2.4810e-02, -1.5629e-01, -2.9678e-01,\n",
       "           5.1157e-01,  9.3190e-02,  1.1001e-01, -2.9384e-01, -8.3219e-02,\n",
       "          -1.7970e-01,  7.6828e-01,  4.6436e-01,  5.2733e-01,  5.2942e-01,\n",
       "          -2.8577e-01, -2.4431e-01,  2.0757e+00, -1.1043e-01,  1.2289e-01,\n",
       "           3.9211e-02,  2.2259e-01, -3.2279e-01, -8.0706e-01,  1.4595e-01,\n",
       "           2.8421e-01,  1.8264e-01,  4.4424e-01,  5.8978e-01, -2.5530e-02,\n",
       "          -7.5532e-01, -1.6410e-01,  2.3382e-01,  5.1213e-01,  3.8887e-01,\n",
       "          -2.6337e-01,  1.9466e-01,  1.4636e-01,  3.7525e-01, -5.3591e-01,\n",
       "          -1.3160e-01, -4.8239e-01, -2.1180e-01,  1.2655e-01, -2.1834e-01,\n",
       "          -6.7611e-03, -6.5546e-02,  2.4392e-01,  4.2268e-01, -4.7342e-01,\n",
       "           1.4063e-01, -1.0337e-01, -8.7956e-02, -2.3678e-01, -2.2007e-01,\n",
       "           2.2546e-01,  6.5683e-01, -1.5269e-01, -4.6425e-01, -1.5952e-01,\n",
       "           6.1512e-02,  2.1052e-02,  1.6707e-01, -3.9651e-01, -3.4138e-03,\n",
       "          -7.4184e-01,  6.2061e-02,  1.7511e-01, -1.3371e-01,  5.4687e-01,\n",
       "          -8.5739e-02,  2.5884e-01,  6.4255e-01, -4.3813e-01, -6.1902e-01,\n",
       "          -5.5055e-01,  1.6335e-01,  1.0877e-01, -3.6114e-01,  1.7931e-01,\n",
       "           3.3609e-01,  9.2736e-02, -7.0020e-02,  4.9498e-01,  1.8608e-01,\n",
       "           1.2569e-01,  3.7132e-01, -1.1446e-01,  1.9235e-01, -3.1295e-01,\n",
       "          -1.8431e-01, -1.7607e-01, -1.8472e-03,  1.1790e-01, -5.0641e-01,\n",
       "           1.6587e-01, -4.8732e-02,  1.8925e-02, -1.7115e-01,  1.6459e-01,\n",
       "          -2.7767e-02,  4.2223e-03,  2.4984e-01,  8.6035e-02,  5.9357e-01,\n",
       "           1.9666e-01, -5.2434e-01, -2.6514e-02, -2.5965e-01,  3.8822e-01,\n",
       "           1.8582e-01,  1.6113e-02, -1.1383e-01, -5.8547e-01, -4.0181e-01,\n",
       "          -3.4921e-01,  5.9282e-01, -3.3673e-01, -6.0076e-01,  4.0522e-01,\n",
       "           8.9741e-02,  1.3607e-01, -1.8550e-01,  8.2654e-02, -1.1011e-01,\n",
       "          -4.2433e-01, -2.5947e-01,  4.7620e-01,  2.7828e-01,  2.8481e-01,\n",
       "           5.1538e-01,  1.7491e-01, -2.3449e-01,  5.1456e-01, -5.5801e-01,\n",
       "          -2.9023e-01,  1.1111e-01, -2.8837e-01,  2.3234e-01,  1.0321e-01,\n",
       "          -6.2105e-01, -2.1517e-01, -7.8464e-02, -1.6388e-01,  1.9043e-01,\n",
       "           3.1181e-01,  1.4949e+00, -6.1620e-01, -1.7219e-01,  7.7003e-01,\n",
       "           2.6837e-01,  9.4473e-02, -2.2260e-01,  1.2587e-01,  1.7176e-01,\n",
       "           1.9671e-01, -1.2275e-01, -2.1932e-01,  5.0274e-02,  3.6854e-01,\n",
       "           4.2866e-01, -1.0827e+00, -3.5564e-01, -1.8527e-02,  2.7633e-01,\n",
       "           2.3386e-01,  3.5494e-01, -6.1587e-01,  1.0233e-01,  5.0949e-01,\n",
       "          -1.0722e+00, -1.0060e-01, -4.8356e-01,  3.1442e-01,  1.8018e-01,\n",
       "          -1.4962e-01,  5.2234e-02, -2.8888e-01, -4.2402e-01,  3.6563e-01,\n",
       "           6.3428e-01, -1.0135e+00,  5.6003e-02,  5.6683e-01, -2.4677e-01,\n",
       "           3.5608e-01,  1.2685e-01,  2.2172e-01,  4.8363e-02,  2.5739e-01,\n",
       "          -3.0885e-01,  6.7267e-01, -5.5282e-01, -5.0918e-01, -1.2764e-01,\n",
       "           5.4209e-01,  1.2805e-01, -2.6452e-01, -1.3689e-01,  4.3621e-01,\n",
       "           2.4490e-02, -3.7324e-01, -2.9477e-01, -3.9558e-02, -9.2844e-02,\n",
       "           2.1229e-01,  1.6194e-01,  1.1577e-01,  1.8920e-01, -3.9392e-01,\n",
       "           4.1678e-01,  6.8339e-01,  8.8206e-01, -1.6203e-01, -4.7772e-01,\n",
       "          -9.3459e-03,  2.3458e-01, -3.7819e-01,  4.8734e-01,  2.7296e-01,\n",
       "          -3.7126e-01,  4.7690e-01,  1.7119e-01, -1.0447e-01,  3.0072e-01,\n",
       "          -1.5839e-02,  1.7504e-01, -1.8615e-01, -9.1598e-02,  2.6163e-01,\n",
       "          -3.0184e-01,  1.8436e-01, -6.4333e-01,  9.4018e-03,  1.2201e-01,\n",
       "          -3.5793e-01,  3.1134e-01,  8.7767e-03,  3.0447e-01, -3.2334e-01,\n",
       "           1.1106e-01, -5.4512e-01, -3.2700e-01,  3.2340e-01, -4.4251e-01,\n",
       "          -6.2610e-01, -2.0274e-01, -2.2315e-02,  4.2191e-01,  5.6178e-01,\n",
       "           1.8109e-01,  2.0605e-01, -4.2351e-01, -3.8336e-01,  2.3075e-01,\n",
       "           1.3959e-01, -2.8108e-01,  1.9494e-01, -5.0174e-01, -6.3197e-02,\n",
       "          -6.3457e-01, -3.4867e-01, -2.7937e-01, -7.5508e-02, -1.0797e-02,\n",
       "           9.8403e-01, -3.5592e-01, -1.1933e-01,  4.5155e-01,  1.0826e-01,\n",
       "           4.7504e-01,  7.8958e-01,  2.6721e-01, -4.0573e-01, -3.3979e-02,\n",
       "          -2.2849e-02, -1.8233e-01, -1.4337e-01, -4.7415e-01,  5.4684e-01,\n",
       "           2.4229e-01,  4.3592e-01,  5.1030e-01]], device='mps:0'),\n",
       " tensor([[-6.1750e-01, -4.2829e-01, -2.3934e-01, -9.4116e-01, -3.4548e-01,\n",
       "          -6.5496e-02, -6.3874e-01,  9.7828e-01, -9.5892e-02,  4.0096e-01,\n",
       "           1.4919e-01, -5.1603e-03,  2.0980e-01,  1.3680e-01, -6.9506e-01,\n",
       "           1.6740e-01,  4.6094e-01,  2.1159e-01,  3.5951e-01, -4.4329e-02,\n",
       "           7.9832e-01,  3.4245e-01,  1.8019e-01, -2.5205e-01, -4.5876e-01,\n",
       "           5.8994e-01, -2.4628e-01, -9.2014e-02,  2.0509e-01, -8.1949e-01,\n",
       "          -1.3586e-01, -4.9757e-02, -3.6281e-01,  8.7602e-01,  1.1948e-01,\n",
       "          -2.7310e-01, -1.5276e-01, -2.7996e-01,  2.3494e-01, -6.3637e-02,\n",
       "           3.7814e-01,  2.6592e-02, -2.9347e-01,  3.1386e-01, -1.6421e-01,\n",
       "          -1.4057e-01,  1.4047e-02, -7.8429e-01,  1.3666e-02,  1.8955e-01,\n",
       "           5.8851e-02, -4.7205e-02,  1.9231e-01,  3.7030e-01, -1.0617e+00,\n",
       "          -2.6348e-01,  2.7603e-01,  2.9849e-01, -1.7932e-01, -5.7784e-02,\n",
       "          -1.9989e-01,  9.0201e-01,  1.7949e-01, -2.7822e-01,  1.7824e-01,\n",
       "          -4.4899e-01, -5.4601e-01, -6.0781e-01, -1.0682e-01, -1.1967e-01,\n",
       "          -1.0002e-01,  9.7528e-02,  1.4009e-01, -2.9385e-01,  2.3170e-01,\n",
       "           2.1813e-01,  4.7024e-02, -2.7652e-02, -4.3158e-01, -2.6546e-01,\n",
       "           3.6947e-01,  3.4425e-02, -1.4730e-02,  2.0326e-01, -5.8744e-01,\n",
       "          -2.0226e-01,  3.7959e-01,  1.5703e-01,  2.5914e-01,  2.2662e-03,\n",
       "           5.9967e-01,  1.2024e-02,  8.0339e-02, -4.5369e-01,  5.8427e-01,\n",
       "          -5.4955e-01,  1.0329e-01, -1.6007e-01,  4.9532e-02, -5.2088e-01,\n",
       "          -9.8735e-02,  2.4494e-01, -1.8085e-01,  1.6053e-01, -4.6592e-01,\n",
       "          -3.2983e-02,  2.4514e-01,  3.0468e-01, -4.9183e-01,  4.2685e-02,\n",
       "          -6.5134e-03, -2.9791e-01,  7.3694e-02, -6.6407e-01, -8.5795e-01,\n",
       "           5.8354e-01,  1.8428e-02, -4.5240e-01, -2.3930e-01,  7.4443e-01,\n",
       "           2.9738e-01, -2.3201e-01, -5.3072e-02, -3.5548e-01,  2.0259e-01,\n",
       "          -1.8518e-01,  2.5689e-01, -2.7018e-01,  3.8439e-01,  5.5026e-01,\n",
       "           3.1581e-01,  2.8489e-01, -1.8327e-01,  2.7637e-01,  1.4465e-01,\n",
       "           3.6947e-02, -6.6614e-02, -1.5929e-01,  1.5180e-01,  1.2041e-01,\n",
       "          -2.9580e-01, -3.2999e-01,  5.4746e-03,  1.1182e-01, -3.3279e-01,\n",
       "          -1.0202e-01,  1.8680e-01,  3.4808e-01, -1.4092e-01, -2.5403e-01,\n",
       "          -7.8984e-01, -4.2328e-01,  1.0217e-01, -1.8560e-01, -1.7135e-01,\n",
       "          -4.1911e-01,  3.8121e-01, -6.6660e-02,  4.7638e-01,  3.4189e-01,\n",
       "           6.9978e-01, -1.5311e-01,  5.9776e-02, -5.6412e-01,  3.5197e-01,\n",
       "           3.8270e-02, -6.1110e-01, -1.7247e-01,  1.5680e-01, -4.1813e-04,\n",
       "           6.8195e-01,  7.0979e-02,  4.0124e-01,  1.2070e-01, -3.2553e-01,\n",
       "           8.2089e-03,  1.9890e-01, -1.2864e-01, -1.4297e-01,  7.3766e-01,\n",
       "          -5.5188e-01, -2.3029e+00,  1.0721e-01, -3.5844e-01, -3.6529e-01,\n",
       "          -1.8135e-01,  3.7255e-01,  3.3772e-02,  4.9966e-01,  1.5486e-01,\n",
       "           3.7598e-02,  1.6729e-01, -2.8025e-02,  3.6851e-01, -8.4399e-02,\n",
       "           2.6751e-01,  3.3341e-01,  1.2988e-01, -2.3298e-01, -1.0956e-01,\n",
       "          -3.9227e-01,  2.4031e-01, -9.8916e-02,  8.6010e-01,  2.3260e-01,\n",
       "           2.0943e-01, -4.3657e-03, -4.7262e-02,  4.8964e-01,  4.3099e-01,\n",
       "          -6.6541e-02, -1.6585e-01, -4.9522e-01,  4.6269e-03, -4.1279e-02,\n",
       "          -1.2963e-02, -4.5611e-01,  1.3753e-01,  4.2841e-01, -1.2416e-01,\n",
       "          -1.8354e-01,  1.5525e-01, -4.7715e-02, -3.1517e-01, -2.6492e-02,\n",
       "           6.8453e-01,  1.9080e-01, -2.2994e-01,  1.2377e-01, -1.1896e-01,\n",
       "           3.7006e-02, -4.6975e-01,  7.0044e-01,  6.1147e-01,  1.5585e-01,\n",
       "           1.6214e-02, -6.9943e-01,  2.3699e-01, -3.1574e-01, -4.7647e-03,\n",
       "          -1.5027e-01, -1.5731e-01, -4.2986e-01, -5.7285e-02, -5.5854e-02,\n",
       "           4.0115e-01,  7.9943e-02,  4.6653e-01,  1.1837e+00,  2.7973e-01,\n",
       "           1.9184e-02,  5.6097e-01,  2.0801e-01,  4.2492e-01, -4.0180e-01,\n",
       "           3.3082e-01,  8.8825e-02, -6.3896e-01,  2.7188e-01, -4.1593e-01,\n",
       "           4.4963e-02, -4.1432e-02, -3.7926e-01,  1.4208e-01, -2.1276e-02,\n",
       "           3.4237e-01,  3.3517e-01,  1.5186e-01,  1.6838e-01,  2.9411e-01,\n",
       "          -3.1322e-01,  5.9715e-01, -6.0809e-02, -1.1127e-01, -6.3492e-01,\n",
       "          -7.6732e-01,  9.5206e-02,  4.4743e-01, -1.2880e-01, -6.1279e-01,\n",
       "           1.0462e-01, -2.3866e-01, -1.2257e-01, -3.5064e-02,  2.1191e-01,\n",
       "           2.7455e-01,  5.0687e-02, -9.5436e-02,  6.4791e-02, -4.6950e-01,\n",
       "          -4.9277e-01,  4.9140e-02, -1.5758e-01, -1.3063e-01,  5.3918e-01,\n",
       "          -7.9415e-02,  1.1575e-01,  5.7221e-01,  1.9799e-02, -4.5976e-01,\n",
       "          -2.5640e-01,  1.4802e-01,  2.8667e-01, -7.2082e-02,  2.1981e-01,\n",
       "           3.4119e-01,  2.6467e-02,  5.1649e-01,  1.4823e-01, -2.1046e-01,\n",
       "          -2.7239e-01,  5.1173e-01, -1.5755e-01,  6.1403e-02, -1.2527e-01,\n",
       "          -8.7207e-01, -5.6150e-02, -7.8413e-02, -1.6833e-01, -5.4440e-01,\n",
       "           2.6566e-01,  1.4351e-01,  1.0561e-01,  2.2350e-01, -1.5984e-01,\n",
       "           7.0123e-01,  6.5285e-01, -3.3100e-01, -1.6943e-01, -4.0104e-01,\n",
       "           9.3922e-02, -6.8374e-02, -2.9329e-01, -1.4009e-02,  9.5368e-03,\n",
       "          -2.2765e-01, -1.2516e-01, -1.0805e-01,  2.8921e-01, -1.8876e-01,\n",
       "           5.2862e-01,  1.2189e-01,  8.3649e-02,  4.3873e-01,  2.0612e-02,\n",
       "           1.3071e-01,  3.6515e-01, -4.2117e-01,  1.2870e-01,  4.0424e-01,\n",
       "           4.0578e-01, -1.0551e+00, -8.3291e-01, -1.4936e-01,  3.6111e-01,\n",
       "           2.2860e-01,  2.5189e-01,  1.6999e-02, -7.2215e-03,  8.9455e-02,\n",
       "          -2.3402e-01,  3.5135e-01, -6.6415e-02, -5.7412e-01,  3.0561e-01,\n",
       "           2.3567e-01,  5.3263e-01, -2.7377e-01, -5.0579e-01, -6.5233e-02,\n",
       "           2.4616e-01, -3.6381e-01,  1.1392e-01,  2.3480e-01,  4.5466e-01,\n",
       "          -3.6015e-01, -5.6164e-01, -7.0282e-02, -7.3986e-01, -1.3578e-01,\n",
       "           3.2483e-01, -2.1437e-01,  2.7178e-02, -2.4758e-01,  1.5231e-01,\n",
       "          -1.7925e-02, -1.8879e-01,  6.0757e-01,  2.0522e-01, -2.5855e-01,\n",
       "           1.1981e-01, -2.3892e-01, -1.9475e-01,  5.0091e-01, -5.0226e-01,\n",
       "          -3.0348e-01, -5.0960e-02, -1.5349e-01, -3.2418e-01,  3.2181e-01,\n",
       "          -4.1389e-01, -1.4861e-01,  1.6748e-01, -6.4736e-02, -1.3068e-01,\n",
       "          -3.5233e-01,  1.4268e-01, -1.2707e-01, -4.9883e-02, -1.7654e-02,\n",
       "          -1.3353e-01,  6.0900e-02,  1.3308e-01,  6.4538e-02,  1.4388e-01,\n",
       "           5.3696e-01, -1.1464e-01,  2.7582e-03, -2.5900e-01, -2.2469e-01,\n",
       "           5.2498e-01, -9.4973e-02, -3.3117e-01,  5.4928e-01, -1.1156e+01,\n",
       "           1.0386e-01,  7.2631e-01, -5.5808e-01,  5.5128e-02, -2.8539e-01,\n",
       "           2.1085e-01, -3.0009e-01,  7.9101e-02,  2.4967e-01, -2.8460e-01,\n",
       "          -1.2277e-01, -2.7364e-01, -2.3193e-01, -4.0872e-01, -3.9742e-01,\n",
       "          -4.7229e-01, -5.6789e-01,  1.1772e-01, -1.7663e-01, -1.0996e-01,\n",
       "          -5.7036e-01, -1.9379e-01,  2.9696e-01, -4.9559e-03, -5.1069e-01,\n",
       "           5.9541e-02, -2.2948e-01,  2.6399e-01,  4.6126e-02,  2.2045e-01,\n",
       "          -1.9949e-02,  1.4108e-01,  2.0669e-01, -4.4946e-02, -4.5100e-01,\n",
       "          -4.4853e-01,  2.2129e-01, -1.5955e-01, -2.2267e-01, -5.4017e-02,\n",
       "          -3.3838e-01, -2.7689e-01,  1.3447e-01,  3.7594e-01,  6.2596e-01,\n",
       "           1.1595e+00,  5.6208e-01,  2.9245e-01, -1.0625e-01, -8.5438e-02,\n",
       "          -1.0762e+00,  5.5900e-01, -3.7318e-01,  8.1787e-02, -2.5119e-01,\n",
       "           6.6201e-01, -4.9906e-01,  5.4685e-01, -4.0754e-01, -2.5500e-01,\n",
       "           3.7179e-01,  3.0194e-01,  7.1104e-03,  2.8239e-02, -2.7432e-01,\n",
       "           4.0257e-01, -2.6094e-01, -3.6915e-02, -3.7819e-01,  4.0081e-01,\n",
       "          -3.9472e-01,  5.0770e-01, -5.6534e-02,  1.2945e-01,  1.8567e-01,\n",
       "          -3.5001e-01,  1.6610e-01,  1.0337e+00,  1.7414e-01,  4.1071e-01,\n",
       "           1.7420e-01,  5.1160e-01,  5.3064e-02, -8.8958e-01,  1.3691e-01,\n",
       "           2.0502e-01,  2.0789e-01, -6.7096e-02,  5.7131e-01, -2.5517e-02,\n",
       "          -6.5181e-01, -2.3094e-01, -3.2169e-01,  4.3108e-01,  6.5352e-01,\n",
       "          -2.2597e-01, -4.4343e-01,  2.3894e-01, -7.9776e-03, -5.3395e-01,\n",
       "           9.0074e-02, -2.2320e-01, -5.2792e-01,  8.7275e-02, -5.8914e-01,\n",
       "           1.4985e-01, -1.2428e-01,  5.4238e-02,  6.6887e-01, -4.0159e-01,\n",
       "          -1.6113e-01,  3.6602e-02, -4.4532e-02, -4.1270e-01, -4.9520e-01,\n",
       "          -7.7083e-02,  4.1073e-01, -1.9776e-01, -2.7976e-01,  7.0636e-02,\n",
       "          -3.0673e-01,  2.2584e-01, -1.7377e-01, -6.4935e-01,  3.2879e-01,\n",
       "          -6.0415e-01,  2.0578e-01,  2.5360e-01, -2.2507e-01,  4.9017e-01,\n",
       "           1.1337e-01, -8.1817e-02,  1.9397e-01, -9.9369e-02, -7.8554e-01,\n",
       "          -3.2573e-01, -2.7299e-02,  3.2662e-01, -4.2844e-01,  1.5776e-01,\n",
       "           1.5022e-01, -4.5752e-02, -2.2574e-01,  2.7904e-01,  1.0174e-01,\n",
       "           3.7762e-01,  3.9846e-01, -5.5006e-01,  1.1727e-01, -2.9867e-01,\n",
       "          -2.9262e-02, -6.1174e-02,  1.7952e-01,  2.3654e-01, -7.0764e-01,\n",
       "          -1.5224e-01, -2.6434e-01, -3.0965e-01, -2.1650e-01, -2.5120e-01,\n",
       "           2.0680e-01,  1.4038e-01, -1.5897e-02, -1.1308e-01,  5.8077e-01,\n",
       "           4.0814e-02, -1.1167e-01, -4.5997e-01,  3.1443e-01,  4.3178e-01,\n",
       "           2.1794e-01,  7.5875e-02, -1.2830e-01, -5.5532e-01, -1.1922e-02,\n",
       "          -4.3644e-01,  9.1081e-01, -7.3906e-01, -7.1970e-01,  6.6086e-02,\n",
       "           1.4951e-02, -8.8259e-04, -2.3450e-01, -5.0265e-01,  8.1298e-02,\n",
       "          -2.1017e-01, -1.4891e-01,  6.1550e-03, -1.0607e-02,  1.6299e-01,\n",
       "           4.0742e-01,  3.6838e-01,  1.5951e-01, -3.3938e-01, -4.4081e-01,\n",
       "          -6.0244e-01,  1.4567e-02, -1.6795e-01,  2.9330e-01, -2.6752e-01,\n",
       "          -3.2661e-01,  1.6043e-02, -9.1892e-03, -2.2272e-01,  5.2876e-01,\n",
       "           1.2427e-01,  1.6396e+00, -3.7287e-01, -4.4551e-01,  2.9018e-01,\n",
       "           7.2306e-02, -2.7073e-02, -2.8130e-01,  2.5074e-01,  9.5447e-02,\n",
       "           4.0194e-02, -3.6304e-01,  1.5023e-02, -2.0558e-02,  3.2773e-01,\n",
       "           3.2056e-01, -1.0703e+00, -2.0238e-01, -1.7993e-01,  5.5664e-01,\n",
       "           3.2388e-02,  6.3185e-01, -2.2305e-01,  3.0577e-01,  4.6106e-01,\n",
       "          -8.6667e-01, -2.3000e-01, -1.6129e-01,  2.7295e-01, -1.0583e-01,\n",
       "          -9.8735e-02, -1.6178e-01, -3.7165e-02, -1.6936e-01,  3.9538e-01,\n",
       "           7.4751e-01, -7.8426e-01,  7.4081e-01,  7.0467e-01, -9.8234e-02,\n",
       "           5.6815e-01,  9.4187e-02,  3.2641e-01, -1.4659e-01,  2.2669e-02,\n",
       "           2.6200e-01,  1.8542e-01, -6.3165e-01, -2.3962e-01,  2.4473e-01,\n",
       "           4.0158e-01, -1.1943e-01,  1.2108e-01,  7.9292e-02,  3.0960e-02,\n",
       "          -3.7691e-01, -1.0145e-01, -2.4552e-01, -3.9904e-01,  1.1356e-01,\n",
       "           1.4603e-02, -1.5001e-01, -8.6997e-02,  4.7709e-01, -1.7580e-01,\n",
       "           7.5067e-01,  4.2910e-01,  3.7145e-01, -3.5103e-01,  8.9950e-04,\n",
       "          -5.7565e-02,  6.0671e-01, -3.1670e-01,  7.7385e-01,  6.3337e-02,\n",
       "          -2.3140e-01,  2.7251e-01,  6.0547e-03,  1.7829e-01,  2.9892e-01,\n",
       "          -2.3519e-01,  1.2702e-01, -3.0718e-01, -2.3711e-01,  2.8330e-01,\n",
       "          -1.8640e-01,  1.1542e-01, -7.2270e-01, -6.3602e-03,  3.9857e-01,\n",
       "          -3.2051e-01,  4.9873e-01, -2.3074e-01,  6.1026e-01, -4.3509e-01,\n",
       "           3.2669e-02, -4.7470e-01, -4.3275e-01, -2.8508e-01,  5.1854e-02,\n",
       "          -8.6027e-01, -1.9453e-01,  1.7684e-01,  3.7405e-02,  2.1707e-01,\n",
       "           7.5155e-01, -3.3794e-01, -2.4771e-01, -2.0386e-01,  4.0190e-02,\n",
       "           1.8299e-01, -8.2071e-02, -3.5586e-01,  2.1161e-01, -3.5607e-01,\n",
       "          -1.0520e+00, -2.3117e-01,  8.9990e-02,  3.8419e-01, -3.5853e-01,\n",
       "           1.1570e+00,  2.5132e-02, -2.0226e-01,  6.7253e-01,  2.0808e-01,\n",
       "           6.4157e-01,  6.0323e-01,  1.3628e-03, -4.1259e-01, -5.4109e-02,\n",
       "          -1.5877e-01, -3.2881e-01,  1.5757e-01, -1.8346e-01,  5.0556e-01,\n",
       "           5.4802e-01, -2.0806e-02,  5.4756e-01]], device='mps:0'),\n",
       " tensor([[ 3.4071e-02, -2.9717e-01, -5.2041e-01, -6.2457e-01, -6.1773e-01,\n",
       "          -2.0190e-01, -5.3291e-01,  5.5376e-01,  1.4886e-02,  6.1137e-01,\n",
       "           3.8470e-01,  3.6079e-01, -2.0804e-02,  3.9639e-02, -7.7629e-01,\n",
       "          -3.6499e-01,  8.0625e-02,  3.1471e-01,  6.8561e-01,  5.1176e-01,\n",
       "           7.7385e-01,  2.6021e-01, -2.2049e-02, -1.5625e-01, -9.3796e-01,\n",
       "           3.5399e-01,  6.0321e-02,  1.1601e-01,  1.3489e-01, -8.1185e-01,\n",
       "          -4.2179e-01,  3.0112e-01,  2.2480e-02,  9.2568e-01, -5.0239e-02,\n",
       "          -2.2757e-01,  1.9493e-01, -4.5913e-01,  1.7824e-01, -2.1328e-01,\n",
       "           5.3217e-01, -7.8240e-02,  3.4865e-01,  4.0189e-02,  4.6586e-01,\n",
       "          -9.0774e-02,  6.7549e-02, -6.0353e-01, -1.1865e-01, -2.5073e-01,\n",
       "           4.4826e-02, -1.5488e-01,  2.4971e-01,  2.9482e-01, -6.1421e-01,\n",
       "          -7.4125e-01,  1.9515e-01,  1.8580e-01, -2.7521e-01, -8.0030e-03,\n",
       "          -5.0449e-01,  8.1756e-01, -1.1247e-01,  1.8228e-01, -1.9152e-01,\n",
       "           4.7189e-02,  1.9949e-01, -8.9550e-01, -6.1592e-01, -1.6989e-01,\n",
       "          -1.2305e-01,  1.9182e-01, -1.5044e-02, -2.4941e-01,  4.3821e-01,\n",
       "          -3.4872e-01,  1.4540e-01, -5.4957e-02, -4.2302e-01, -1.3897e-01,\n",
       "           1.2322e-01,  1.3994e-01,  1.0151e-01, -9.1235e-02, -5.8038e-01,\n",
       "           1.7111e-01,  3.5946e-01,  6.0728e-01,  3.3556e-01, -2.7603e-01,\n",
       "           3.1813e-01,  1.7052e-01, -2.8351e-01, -2.0442e-01,  5.3840e-01,\n",
       "          -2.7790e-01,  1.7323e-01, -2.8934e-01,  1.6458e-01, -5.7143e-01,\n",
       "           2.2644e-01,  4.1025e-01, -1.2435e-01,  3.9394e-01, -3.8573e-01,\n",
       "          -9.3342e-01,  3.3152e-01, -2.4020e-01, -7.8356e-01, -1.9075e-01,\n",
       "          -6.4547e-02, -3.1519e-01,  1.8819e-01, -4.2984e-01, -2.0328e-01,\n",
       "          -5.5353e-02,  4.9543e-02, -1.0054e-02, -4.6012e-01,  3.8859e-01,\n",
       "           3.3196e-01, -3.1168e-01,  3.3389e-02,  8.9257e-03,  5.1470e-02,\n",
       "          -5.7345e-03,  1.9232e-01, -5.8692e-02,  5.0628e-01,  3.8166e-01,\n",
       "           4.9400e-02,  4.2834e-01, -4.3317e-01, -1.1126e-01,  1.8834e-01,\n",
       "           2.1228e-01, -1.9140e-01, -2.2247e-01,  4.0818e-01,  1.1462e-01,\n",
       "          -2.1289e-01,  1.9026e-02, -7.3806e-02,  1.6737e-02,  1.3096e-01,\n",
       "          -6.5309e-02,  7.1146e-02, -1.9181e-01, -2.3137e-01, -1.3201e-01,\n",
       "          -3.4504e-01, -6.7160e-01,  3.8480e-01, -2.1937e-01,  3.2838e-02,\n",
       "          -2.9042e-01,  6.1155e-01, -8.9098e-02,  1.4369e-01,  2.9825e-01,\n",
       "           5.3258e-01, -1.7637e-02,  7.8580e-02, -1.9677e-01, -6.3400e-02,\n",
       "           4.6641e-01, -2.2236e-01, -3.1901e-01, -2.5433e-01,  1.4914e-02,\n",
       "           1.3072e-01, -3.0401e-01,  1.7263e-01, -1.6900e-02, -5.2437e-02,\n",
       "          -4.2313e-01,  2.9258e-01,  6.7456e-02, -3.6111e-01,  2.3742e-01,\n",
       "          -6.8381e-01, -1.5976e+00,  3.2836e-01,  1.1163e-01, -3.3053e-01,\n",
       "          -4.5970e-01,  3.8186e-01,  4.7125e-01,  2.8728e-01, -1.2735e-01,\n",
       "          -1.3115e-02, -7.1566e-02,  2.7043e-01,  6.7322e-01, -4.0633e-01,\n",
       "           5.2789e-01,  4.5672e-01, -7.8586e-02,  1.1915e-01,  6.5401e-02,\n",
       "          -5.8183e-01,  8.5160e-02,  1.3994e-01, -2.6073e-01,  1.3540e-01,\n",
       "           3.2890e-02,  2.3874e-01, -3.7870e-02,  1.6278e-01,  2.4395e-01,\n",
       "          -2.8766e-02, -2.4849e-01, -1.7426e-01, -2.3914e-01, -2.3835e-01,\n",
       "           6.6633e-01,  1.5901e-02,  1.6619e-02,  4.8022e-01,  9.9257e-02,\n",
       "          -3.2100e-01,  4.5557e-02, -2.2564e-01, -8.6619e-02, -4.8562e-01,\n",
       "           2.4222e-01,  1.3933e-01, -4.9343e-02, -2.5172e-01, -2.6136e-01,\n",
       "          -7.5472e-02, -4.4399e-01,  7.7235e-01,  2.7280e-01, -2.0181e-01,\n",
       "          -2.3415e-01, -3.2583e-01, -1.3526e-01,  1.5481e-01,  3.3654e-01,\n",
       "          -1.4982e-01, -4.4813e-01, -6.0247e-01,  7.4521e-03,  2.0382e-01,\n",
       "           1.6158e-01, -1.3563e-01,  4.2506e-01,  6.4306e-01,  2.9921e-01,\n",
       "          -2.2230e-02, -2.8336e-01,  3.9504e-01,  2.7751e-01, -8.9055e-02,\n",
       "           2.9258e-01,  1.2287e-01, -1.1063e-01,  3.3827e-01, -4.4641e-01,\n",
       "           1.9426e-01,  9.9069e-02,  1.3837e-01,  4.1584e-01, -3.8734e-01,\n",
       "          -8.1174e-02,  2.7581e-01, -3.3925e-01,  4.9830e-02,  5.2198e-02,\n",
       "          -3.2506e-01,  5.1286e-01,  3.9266e-01, -5.1345e-01, -7.1922e-01,\n",
       "          -2.9595e-01,  3.6623e-01,  5.9341e-02, -1.1917e-01, -8.3279e-01,\n",
       "          -2.5885e-01,  6.0394e-02,  7.4094e-02, -8.6705e-02, -3.5151e-01,\n",
       "           1.8034e-01,  4.1036e-01, -5.2998e-01,  1.2523e-01, -5.8288e-01,\n",
       "           2.9689e-02,  8.3561e-02, -2.1037e-01,  4.6055e-01,  2.4163e-01,\n",
       "          -1.9267e-01,  1.4231e-01, -1.1987e-02,  2.2184e-01, -2.5850e-01,\n",
       "          -8.6415e-02, -2.4300e-01, -2.6806e-02,  7.0254e-02, -7.5894e-02,\n",
       "           2.3873e-01,  2.0719e-01,  3.9665e-01, -6.1996e-01, -7.5848e-01,\n",
       "          -2.8338e-01,  8.3559e-01, -2.3204e-02, -4.4689e-01, -1.7349e-01,\n",
       "          -1.0530e+00, -1.9959e-03, -1.0907e-01, -8.7653e-02, -5.0403e-01,\n",
       "           3.5114e-01, -8.7987e-02, -3.2037e-01,  2.8363e-02, -3.4979e-01,\n",
       "           4.3701e-01,  6.4103e-01, -3.9464e-01, -1.1180e-01, -1.4781e-01,\n",
       "          -6.7924e-01, -1.2432e-01,  2.7084e-02, -1.3874e-01, -1.6327e-01,\n",
       "           7.8122e-02,  1.1048e-01, -2.3561e-03, -3.5861e-02, -4.1968e-01,\n",
       "           4.1994e-01, -6.6101e-02, -1.8726e-02, -4.9698e-02, -3.9231e-01,\n",
       "           6.1119e-01,  6.0076e-01, -4.4191e-01, -2.4376e-01, -2.4516e-01,\n",
       "          -2.5308e-01, -1.3153e-01, -2.8943e-01,  1.3050e-01,  6.8407e-01,\n",
       "           5.3290e-01,  1.4919e-01, -2.5693e-01,  5.9957e-01,  4.0764e-01,\n",
       "          -4.8672e-01,  5.0133e-01,  1.4056e-01, -2.7022e-01, -1.9986e-02,\n",
       "           1.1927e-01,  1.1699e-01, -4.3865e-01, -8.5986e-02,  3.7310e-02,\n",
       "           3.4347e-01, -6.3874e-01,  2.1628e-01,  2.0757e-02,  7.1496e-01,\n",
       "          -4.8469e-01, -4.1512e-01,  3.8438e-01, -5.0230e-01, -2.3507e-01,\n",
       "           1.1113e-01,  1.8432e-01,  1.5330e-01,  4.7329e-02, -8.6986e-02,\n",
       "          -3.8965e-01,  3.4337e-01,  9.0853e-01,  6.0038e-01, -2.5430e-01,\n",
       "           1.6047e-01, -1.2856e-01, -3.4747e-01,  3.9901e-01, -2.1123e-01,\n",
       "          -1.7307e-01,  6.5494e-01, -4.4190e-01,  8.8023e-02,  3.4354e-01,\n",
       "          -3.3712e-01, -2.0045e-01, -2.8739e-01, -3.2300e-01, -2.6419e-01,\n",
       "          -3.4803e-01,  2.2835e-01, -2.8171e-01, -4.2671e-03, -8.7564e-02,\n",
       "           2.0761e-01,  1.2157e-01, -1.8962e-01,  2.5995e-01, -2.8690e-01,\n",
       "          -4.3789e-02, -1.8549e-01,  1.1950e-01, -2.7063e-02, -3.7213e-01,\n",
       "           6.7906e-01,  3.2331e-01,  1.8718e-01,  4.6417e-01, -1.1074e+01,\n",
       "          -2.1597e-01,  2.8665e-01, -2.8784e-01, -2.8138e-01, -1.5864e-01,\n",
       "           8.2164e-02, -1.3391e-01,  6.9360e-02, -1.8699e-01, -2.9262e-01,\n",
       "          -8.8357e-02,  1.4367e-01,  1.5637e-01,  2.5714e-01, -4.0578e-01,\n",
       "          -2.2528e-01, -8.0358e-01,  1.2928e-01,  1.1862e-01,  1.3066e-01,\n",
       "          -3.6952e-01,  1.6723e-01,  4.5664e-04, -1.9071e-02,  2.8871e-01,\n",
       "          -2.1046e-01, -1.0137e-01,  3.9008e-01,  4.8009e-01,  1.9010e-01,\n",
       "           2.4369e-01,  1.9152e-01, -3.3188e-02, -3.3922e-01, -4.0925e-01,\n",
       "          -1.4547e-02, -1.4779e-01, -7.9594e-02,  2.2375e-01, -4.5438e-01,\n",
       "          -3.9426e-01, -3.2699e-01,  3.4436e-01,  3.6354e-01,  9.7861e-01,\n",
       "           1.1778e+00,  4.2192e-01,  6.4805e-01,  1.9158e-01,  7.5951e-02,\n",
       "          -7.8776e-01,  1.7477e-01, -7.1873e-01, -1.5000e-01, -1.2848e-01,\n",
       "           7.2423e-01, -5.0907e-01, -1.1991e-01, -3.0939e-01, -2.9208e-01,\n",
       "           4.6592e-01,  4.7968e-01, -1.7955e-01, -6.4371e-01, -4.2703e-02,\n",
       "           3.5308e-01, -3.7811e-03,  1.8482e-01, -2.9810e-02,  5.5819e-02,\n",
       "          -9.3985e-02,  4.8551e-01,  1.7053e-01, -1.1403e-01,  3.6616e-01,\n",
       "           2.4289e-01, -8.3583e-02,  1.4266e+00, -8.9431e-02, -1.7276e-01,\n",
       "           2.2648e-02,  1.1213e-01, -4.9845e-01, -9.1351e-01, -5.7625e-02,\n",
       "           4.1514e-01,  9.2678e-02,  1.1419e-01,  1.9845e-01, -1.7386e-01,\n",
       "          -1.1254e+00, -1.8576e-01,  3.1368e-01,  6.2105e-01,  7.0829e-01,\n",
       "          -8.1636e-01, -1.4641e-01,  2.6708e-01,  1.3351e-02, -5.4293e-01,\n",
       "           7.9445e-02, -4.1453e-01, -1.1206e-01,  1.8613e-01, -1.3869e-02,\n",
       "          -6.9609e-02, -1.0051e-02, -4.8137e-03,  1.9084e-01, -4.3096e-01,\n",
       "           1.9583e-01,  3.4802e-01, -3.7286e-02, -3.5483e-02,  1.4860e-01,\n",
       "           1.0632e-01,  7.6008e-01, -8.1066e-02, -5.6025e-01, -3.6568e-02,\n",
       "          -1.9936e-02,  3.8990e-02, -3.4845e-03,  4.1490e-02,  6.7276e-01,\n",
       "          -3.6501e-01,  6.9752e-02,  4.4149e-01,  3.4104e-01,  2.9183e-01,\n",
       "           3.1824e-02, -1.2069e-01,  2.5289e-01, -9.0341e-01, -1.0087e+00,\n",
       "          -6.0779e-01,  2.5024e-01, -8.6653e-02, -1.4295e-01, -2.1301e-01,\n",
       "           7.9791e-02,  3.7651e-01, -2.7189e-01, -1.3323e-02,  2.9782e-01,\n",
       "           6.8293e-01,  4.3474e-01, -2.5443e-01,  2.6671e-01, -2.7666e-01,\n",
       "          -6.8733e-01, -3.1430e-01, -9.0457e-02,  7.7285e-02, -4.4675e-01,\n",
       "          -4.2408e-01, -1.2925e-01, -3.8201e-01, -3.0012e-01,  2.9838e-01,\n",
       "          -3.9862e-01,  4.6507e-02, -1.2730e-02,  1.5868e-01,  4.7722e-01,\n",
       "           3.6471e-01, -2.3131e-01,  1.5733e-01, -1.6013e-01,  5.1678e-01,\n",
       "          -3.0885e-02,  2.5692e-01, -5.1283e-01, -1.4919e-01, -2.2796e-02,\n",
       "           4.9508e-01,  8.5019e-01, -4.0133e-02, -1.9382e-01,  1.7164e-01,\n",
       "           4.5099e-01, -2.0145e-01, -1.1957e-01,  9.4911e-02, -3.7469e-01,\n",
       "          -1.0728e-01, -6.8026e-01,  4.4311e-01,  5.1975e-02,  4.8214e-01,\n",
       "           6.8142e-02, -2.1494e-01, -2.9588e-01,  6.1760e-01, -2.6862e-01,\n",
       "          -4.0375e-01, -2.2065e-01, -2.6866e-01, -1.6066e-02,  1.3546e-01,\n",
       "          -2.2933e-01, -9.3142e-02, -2.5046e-01, -1.0680e-01,  2.6696e-01,\n",
       "          -1.2456e-01,  1.1305e+00,  1.3017e-01, -6.7789e-01,  8.4928e-01,\n",
       "           4.7960e-01, -3.8383e-01,  2.5386e-02, -2.0896e-01,  3.3782e-01,\n",
       "           3.4867e-01, -1.3754e-01,  2.2247e-01,  2.6582e-01,  5.6541e-02,\n",
       "           6.9955e-01, -8.3170e-01, -4.8994e-01,  2.3153e-02,  2.0593e-01,\n",
       "          -1.9732e-01,  4.1382e-01, -2.8635e-02,  3.5398e-01,  8.1252e-01,\n",
       "          -1.3768e+00, -2.0799e-01, -6.4739e-01,  5.2412e-01,  2.0404e-01,\n",
       "          -2.1514e-01, -4.1946e-02, -5.6845e-01, -5.0671e-01,  2.8356e-01,\n",
       "           8.4850e-01, -1.3043e+00,  1.3794e-01,  6.0523e-01,  1.1913e-01,\n",
       "           2.2594e-01, -4.0642e-01,  3.0005e-01, -3.6022e-01,  3.1950e-01,\n",
       "          -1.8065e-01,  3.2386e-01, -5.8660e-01, -8.2759e-01,  4.6760e-02,\n",
       "           5.4682e-01,  2.1500e-01,  4.6258e-02, -5.3910e-02,  4.1127e-01,\n",
       "          -1.9036e-01, -3.4804e-01, -1.9176e-01,  3.0959e-01, -3.1851e-01,\n",
       "           1.5883e-01, -4.6210e-01,  1.5840e-01,  1.7938e-01,  1.0998e-01,\n",
       "           6.1544e-02,  4.7191e-01,  9.9154e-01, -1.8248e-01, -1.9861e-01,\n",
       "           3.2506e-01,  4.5359e-01, -1.5732e-01,  5.4380e-01, -3.4592e-01,\n",
       "          -3.2962e-01,  6.2331e-01,  2.3865e-02, -2.3366e-01,  3.8240e-01,\n",
       "          -1.8577e-01,  1.6025e-01,  1.7408e-01, -2.3838e-01,  9.7084e-02,\n",
       "          -3.1464e-02,  1.2104e-02, -3.6001e-01,  1.0211e-01,  2.6003e-01,\n",
       "          -2.0636e-01,  1.0343e+00, -3.1966e-01,  5.4610e-01, -6.3046e-01,\n",
       "           2.1512e-01, -7.7928e-01, -2.7500e-01, -2.6732e-01, -3.4566e-01,\n",
       "          -4.7396e-01, -1.1280e-01,  1.4559e-01,  5.1882e-01,  7.5153e-01,\n",
       "           2.5154e-01, -2.9438e-01, -6.1250e-01, -2.7328e-01,  4.4171e-01,\n",
       "          -1.3850e-01, -1.0870e-01,  3.0023e-01,  2.1620e-01,  4.2932e-01,\n",
       "          -5.9806e-01, -4.6431e-01, -2.1360e-01,  3.0910e-01, -1.8723e-01,\n",
       "           7.6997e-01,  5.6949e-03,  1.4243e-01,  1.7298e-01,  5.0684e-02,\n",
       "           3.3295e-02,  3.3228e-01, -3.3804e-01, -5.8652e-01, -1.3629e-01,\n",
       "          -1.5068e-01, -5.9636e-01, -8.1112e-03, -7.5316e-01,  3.8938e-01,\n",
       "           4.1198e-01,  5.6101e-01,  5.5965e-01]], device='mps:0'),\n",
       " tensor([[-2.5187e-01, -4.0054e-01, -5.8007e-01, -8.3055e-01, -1.2868e-01,\n",
       "          -2.8804e-02, -7.1917e-01,  6.6701e-01,  1.2709e-01,  5.0855e-01,\n",
       "           1.2570e-01,  5.1916e-01,  5.1676e-02, -1.1521e-01, -1.6481e-01,\n",
       "          -4.9820e-01,  5.6222e-02,  3.7338e-01,  6.7187e-01, -2.4111e-02,\n",
       "           2.2844e-01,  5.7979e-01, -2.6323e-01, -1.2386e-02, -8.3013e-01,\n",
       "           2.9613e-01, -1.1576e-01,  2.2409e-01,  1.7718e-01, -6.3092e-01,\n",
       "           4.0999e-02,  4.7417e-01,  2.3068e-02,  1.3497e+00, -5.6080e-01,\n",
       "          -1.3230e-01,  8.2213e-02, -3.9453e-01, -2.3010e-01,  3.7283e-02,\n",
       "           6.1164e-01, -4.1315e-01,  1.3635e-01,  4.0254e-01, -6.1085e-02,\n",
       "           8.9076e-02,  1.3899e-01, -3.7744e-01,  2.5892e-01, -8.3961e-01,\n",
       "          -4.8318e-02, -2.0669e-01,  4.9144e-01,  6.2615e-01, -6.3176e-01,\n",
       "          -2.2975e-01,  2.7419e-01, -3.7716e-02, -3.3020e-02,  1.5349e-01,\n",
       "           9.7980e-02,  3.8582e-01, -2.9995e-01,  2.3257e-01,  3.3441e-01,\n",
       "           2.4458e-01, -3.0392e-01, -4.8889e-01, -5.4078e-01, -5.0111e-02,\n",
       "           9.1859e-03,  2.2490e-02,  1.4846e-01, -2.3272e-01,  4.1973e-01,\n",
       "          -7.1214e-01, -6.4354e-02, -8.9125e-02, -2.5973e-01,  2.9246e-01,\n",
       "          -4.5549e-02,  1.9719e-01,  3.1010e-01,  1.0268e-01,  1.2457e-01,\n",
       "           4.3419e-02,  2.7215e-01,  5.2932e-04,  4.7373e-02,  3.7533e-01,\n",
       "          -1.3375e-01, -8.1722e-02, -2.8313e-01, -3.5814e-01,  3.1229e-01,\n",
       "           4.0598e-02,  2.9907e-01, -1.9557e-01,  1.8579e-01, -3.2474e-01,\n",
       "           1.5588e-01,  1.7768e-01, -1.8083e-01,  5.7286e-02, -3.5172e-01,\n",
       "          -4.5475e-01,  2.3589e-02, -2.8569e-01, -4.8285e-01,  2.2697e-01,\n",
       "          -1.4211e-01, -3.0205e-01, -4.0922e-02, -3.8935e-01, -3.1524e-02,\n",
       "          -7.5164e-02,  2.6013e-01, -1.3230e-01, -3.4806e-01, -8.3033e-02,\n",
       "           2.0734e-01, -2.2808e-01,  1.6077e-01,  2.6882e-01,  1.2232e-01,\n",
       "          -9.6705e-02,  6.5251e-01,  1.8950e-02,  3.1674e-01,  2.5763e-01,\n",
       "           1.1699e-01, -6.5768e-02, -1.9745e-01, -2.3154e-01,  7.5988e-02,\n",
       "          -7.6941e-02,  1.0687e-01, -1.8182e-01, -2.2790e-01, -1.8508e-01,\n",
       "           3.1771e-01, -3.3097e-01, -6.0893e-01, -4.9260e-01, -1.6098e-01,\n",
       "          -1.6326e-01, -2.4804e-02, -8.2473e-03, -3.3693e-01, -3.1956e-01,\n",
       "          -6.7443e-01, -3.6717e-01,  8.1041e-01, -1.6831e-02, -3.3356e-01,\n",
       "          -1.6499e-01,  2.3208e-01,  3.1666e-01, -4.5959e-02,  3.7161e-01,\n",
       "           7.7250e-01, -4.7670e-02,  1.6786e-01, -5.1827e-01, -3.8195e-01,\n",
       "           4.8335e-02,  2.6658e-01, -2.7054e-01,  6.1879e-01,  4.8352e-02,\n",
       "           7.5627e-02, -6.6170e-02,  2.7430e-01, -3.2704e-01, -1.2324e-02,\n",
       "          -5.7573e-01,  3.7750e-01,  2.9539e-01, -3.9892e-01, -6.6567e-02,\n",
       "          -2.0834e-02, -1.2060e+00,  9.6964e-02, -8.3040e-01,  3.1236e-01,\n",
       "          -2.3110e-01,  1.7963e-01,  2.3389e-01,  2.9545e-01, -6.5320e-02,\n",
       "           2.5404e-01,  2.8725e-01,  2.5687e-01,  3.3609e-01, -3.4924e-02,\n",
       "           2.8019e-01,  2.3924e-01,  5.0790e-01,  4.1073e-01,  1.4769e-01,\n",
       "          -3.5440e-01,  2.9683e-01,  3.5781e-01,  9.0598e-02,  1.6954e-01,\n",
       "           9.9825e-02, -2.1636e-01,  2.1091e-01,  1.7501e-01, -1.6618e-01,\n",
       "          -2.2406e-01, -3.1100e-02, -4.6818e-01, -5.6655e-02, -4.8467e-01,\n",
       "           3.2557e-01, -5.1983e-01, -1.1912e-01,  3.4831e-01,  2.4340e-01,\n",
       "          -1.4896e-02,  1.8319e-01,  1.7196e-01, -2.4338e-01,  2.1735e-01,\n",
       "           3.7449e-01, -3.4618e-01, -3.9804e-02, -3.2197e-01, -2.9367e-01,\n",
       "           2.4357e-01, -5.6244e-01,  6.6635e-01,  4.0574e-01,  6.0165e-02,\n",
       "           1.5119e-01, -2.7920e-01, -5.1638e-02, -5.7408e-01,  3.5471e-01,\n",
       "          -3.2130e-01, -3.4955e-01, -5.4801e-01,  6.4487e-02,  4.2773e-01,\n",
       "           2.7565e-01, -6.4915e-02,  7.7868e-01,  4.2408e-01,  4.2813e-01,\n",
       "          -4.3003e-01, -1.5896e-02,  4.0377e-01, -4.1405e-02, -4.6059e-01,\n",
       "           6.7116e-02, -1.4962e-01, -5.1643e-01,  2.8000e-01, -2.6598e-01,\n",
       "          -1.2047e-01, -1.4913e-01,  3.3682e-01,  1.5365e-01, -2.1964e-01,\n",
       "          -3.8461e-01,  3.9587e-01, -1.4349e-01, -4.1723e-01,  4.7123e-01,\n",
       "          -3.5025e-01,  1.5209e-01,  1.0694e-01, -3.8441e-01, -6.2072e-01,\n",
       "          -3.2235e-01,  2.3090e-01,  1.5585e-01,  3.4799e-02, -6.6306e-01,\n",
       "          -4.0571e-01,  3.1197e-01,  2.4601e-01,  2.9976e-01,  2.9840e-01,\n",
       "           5.7427e-01, -3.0944e-01, -1.6702e-01,  1.4982e-01, -4.1377e-01,\n",
       "          -1.5704e-01, -3.3228e-02, -1.9995e-01,  2.9569e-01,  2.4768e-01,\n",
       "           1.1600e-01,  2.0367e-01,  4.3482e-01, -2.5272e-01, -3.8673e-01,\n",
       "          -7.1543e-02,  2.0161e-01, -9.3629e-02, -5.8306e-02,  2.3951e-01,\n",
       "          -6.3095e-02,  5.9650e-01,  7.2143e-01,  7.7439e-02, -6.4006e-01,\n",
       "           1.7696e-01,  5.2502e-01, -2.7253e-01, -4.0343e-02,  9.6365e-02,\n",
       "          -1.0344e+00, -2.1331e-01, -1.2883e-01, -2.0181e-01, -6.5352e-01,\n",
       "           8.8115e-02,  3.4863e-01,  4.6833e-03, -1.7864e-01, -3.0947e-01,\n",
       "           1.0448e-01,  4.8724e-02, -4.8980e-02,  4.3534e-01, -2.8901e-01,\n",
       "          -3.4797e-01, -2.3438e-01, -2.1532e-01, -5.4201e-01, -3.9968e-01,\n",
       "           3.5914e-01,  2.8129e-01,  1.1986e-01, -2.6486e-01, -5.1832e-01,\n",
       "           4.5577e-01, -6.8169e-01, -1.4632e-01,  2.4840e-01, -4.7702e-01,\n",
       "           1.4650e-02,  4.8990e-01,  1.4999e-01,  1.6336e-01,  6.5258e-02,\n",
       "           7.8557e-02, -3.3903e-01, -5.0030e-01, -1.5000e-01,  6.2323e-01,\n",
       "           3.2283e-01,  3.3506e-01, -1.7236e-01,  3.9096e-01,  4.5167e-01,\n",
       "          -2.9582e-01,  4.1670e-01,  1.0930e-02,  7.7439e-02, -6.0414e-02,\n",
       "           1.9048e-01,  3.4370e-01, -3.6842e-01,  2.0145e-01,  1.7929e-01,\n",
       "          -3.1634e-01, -3.2149e-01,  1.5691e-02,  2.0642e-01, -1.8580e-01,\n",
       "          -3.0875e-01, -2.0361e-01,  9.9698e-02, -3.4856e-01, -4.3104e-01,\n",
       "           2.9161e-01, -3.6543e-02,  7.0495e-02, -4.0357e-01, -4.3462e-01,\n",
       "          -2.0181e-01, -2.6055e-01,  6.8681e-01,  2.1064e-01, -1.9621e-01,\n",
       "           1.8720e-01, -1.0907e-01, -3.6330e-01,  4.7847e-01,  2.0162e-02,\n",
       "          -4.8754e-01,  4.5482e-01, -3.4822e-01,  1.0627e-01,  5.5756e-01,\n",
       "          -1.6769e-01, -1.0840e-01,  1.1008e-01,  1.0820e-01,  2.9026e-01,\n",
       "          -3.9884e-01,  1.5516e-01, -6.3476e-02, -3.1885e-01,  3.1820e-01,\n",
       "           2.6411e-01,  2.2781e-01, -1.7137e-01, -3.3426e-02,  3.5677e-02,\n",
       "           2.8782e-01, -6.0853e-02, -3.9888e-01, -4.7347e-01, -4.6132e-01,\n",
       "           1.2832e-01,  1.8927e-01,  5.6207e-01,  3.4144e-01, -1.1394e+01,\n",
       "           1.2938e-01,  7.8031e-01, -4.0713e-01, -8.2819e-02, -5.0320e-02,\n",
       "           1.0976e-02, -1.3708e-01,  4.2955e-01, -2.7734e-01, -4.4105e-01,\n",
       "          -3.5727e-01, -1.3308e-01,  1.5454e-01, -3.2189e-01, -2.6945e-01,\n",
       "          -1.4686e-02, -7.0935e-01,  4.0963e-01, -3.0286e-01,  5.0298e-01,\n",
       "          -2.3511e-01, -8.3798e-02,  1.5816e-01, -3.5851e-01, -1.3439e-02,\n",
       "           5.9766e-02, -1.6033e-01,  1.4651e-01,  4.2486e-01, -5.2072e-02,\n",
       "           3.3244e-02, -4.1258e-03,  4.8461e-02, -2.5723e-01, -5.9999e-01,\n",
       "           1.9296e-02, -1.4141e-01,  3.3422e-01, -4.4133e-02, -7.1265e-01,\n",
       "          -2.2533e-01, -2.7418e-01,  3.1407e-01,  2.7021e-01,  9.0598e-01,\n",
       "           2.2955e-01,  1.0686e+00,  1.5022e-01,  1.7748e-01,  2.6230e-02,\n",
       "          -7.8764e-01, -3.3269e-02, -4.1907e-01,  2.3280e-02,  1.0114e-01,\n",
       "           3.6576e-01, -7.9367e-01,  2.5251e-01, -3.5521e-01,  3.6487e-02,\n",
       "           4.9094e-01,  3.1768e-02,  2.7700e-01, -3.2298e-02, -7.8123e-02,\n",
       "           1.2089e-01, -2.8544e-02,  1.1098e-01, -2.2583e-01, -2.4084e-01,\n",
       "          -1.0177e-01,  5.9888e-01,  3.9592e-01, -7.2211e-01,  3.6112e-01,\n",
       "           4.9258e-02, -1.8622e-01,  1.4556e+00, -4.8573e-02,  1.8361e-02,\n",
       "           1.9800e-03,  2.9604e-01, -5.0236e-01, -7.3516e-01,  3.6511e-01,\n",
       "          -1.0211e-01,  5.6427e-02,  1.0127e-01,  4.8148e-01, -9.7481e-02,\n",
       "          -4.7940e-01, -1.8593e-03,  1.0556e-01, -1.3380e-01,  5.4710e-01,\n",
       "          -7.2824e-01, -1.6963e-01, -1.1223e-01,  3.1723e-01, -8.4054e-01,\n",
       "          -6.1126e-02, -7.0210e-01, -3.9328e-01,  5.0586e-01, -2.5896e-01,\n",
       "           1.5852e-01, -2.7443e-02, -4.3439e-02,  5.5574e-01, -2.1489e-01,\n",
       "           7.1555e-02,  2.2813e-01,  8.4995e-02, -1.6733e-01,  3.9349e-02,\n",
       "           4.1630e-01,  4.9571e-01, -4.2740e-01, -6.3911e-01, -7.6901e-03,\n",
       "           1.7180e-01,  1.4933e-01,  3.4019e-01, -5.3463e-01,  3.6431e-01,\n",
       "          -7.9162e-01, -1.8769e-01,  7.2949e-02, -4.3273e-02,  3.6175e-01,\n",
       "           2.3925e-01,  7.2038e-02,  2.2256e-01, -7.7756e-01, -6.4498e-01,\n",
       "          -6.5932e-01, -2.1445e-01,  2.6143e-01, -5.4236e-01, -8.3654e-02,\n",
       "          -1.1033e-01, -4.8837e-03, -5.0292e-02, -6.6102e-02,  1.1939e-02,\n",
       "           1.3632e-01,  2.3587e-01, -1.3023e-01, -1.3178e-01, -9.2464e-02,\n",
       "          -5.5566e-01, -1.1753e-01, -9.9591e-02, -1.2505e-01, -5.6783e-01,\n",
       "          -1.3797e-01,  3.3075e-02, -2.5373e-01, -6.8124e-02,  3.5608e-01,\n",
       "          -2.9832e-02, -2.7856e-02, -4.8850e-01,  7.2846e-01,  4.0279e-01,\n",
       "           5.2088e-02, -2.8298e-01, -5.5189e-01, -1.0567e-01,  2.9592e-01,\n",
       "           1.9555e-01, -6.7007e-02, -1.7709e-01, -2.4145e-01, -4.2522e-01,\n",
       "           1.4302e-01,  7.1147e-01, -6.6296e-02, -6.8115e-01,  2.6621e-01,\n",
       "           5.8223e-01, -1.2882e-01, -2.4432e-01,  2.0983e-02, -9.3284e-02,\n",
       "          -3.1851e-01, -6.0317e-01,  7.8821e-01, -1.1246e-01, -5.8638e-02,\n",
       "           2.2427e-01,  2.0359e-01, -9.2588e-03,  4.0209e-01, -1.5770e-01,\n",
       "          -3.5434e-01, -1.6477e-03, -7.3600e-01,  4.3257e-01,  6.1666e-02,\n",
       "          -3.1489e-01,  7.6082e-01,  2.7260e-01, -6.0593e-02, -4.5779e-02,\n",
       "           2.2863e-01,  7.9941e-01, -2.4636e-01,  6.4653e-02,  5.3779e-01,\n",
       "           8.8110e-02, -1.1229e-01,  3.9580e-02,  4.1665e-02,  4.5410e-02,\n",
       "           3.7012e-01, -3.2145e-02, -4.1564e-01, -6.1194e-02,  3.8924e-01,\n",
       "           3.9295e-01, -5.4610e-01, -7.3487e-01,  2.2333e-02,  6.8551e-01,\n",
       "          -2.9054e-01,  2.3654e-01, -3.2708e-02,  3.0880e-01,  1.2001e-01,\n",
       "          -9.2160e-01, -1.5016e-01, -1.0056e-01,  2.8229e-01, -2.5482e-02,\n",
       "          -1.6139e-01, -1.4727e-01, -2.3686e-01, -1.6208e-01,  6.8308e-01,\n",
       "           2.1991e-01, -6.9160e-01,  3.4058e-02,  5.7281e-01, -6.0551e-01,\n",
       "           4.6052e-01, -2.4577e-01, -3.4302e-02,  3.6435e-01,  1.3117e-01,\n",
       "           4.9067e-02,  1.7856e-01, -4.2138e-01, -4.0598e-01,  1.7824e-01,\n",
       "           9.1672e-01, -7.6613e-02,  1.1166e-01, -3.9384e-01, -1.7182e-01,\n",
       "          -7.1855e-02, -3.1913e-01, -3.1286e-01,  3.7185e-01,  2.0529e-01,\n",
       "          -3.2552e-02, -1.3271e-01, -2.2992e-01,  3.0816e-01,  1.1821e-01,\n",
       "           3.2536e-01,  5.8187e-01,  6.4368e-01, -1.0077e-02, -2.4936e-01,\n",
       "          -6.1235e-02,  1.4568e-01, -1.9524e-01,  5.2704e-01, -2.0627e-01,\n",
       "          -3.3453e-01, -3.8069e-03,  3.0134e-01, -6.1681e-01, -5.8725e-02,\n",
       "          -4.5426e-01,  4.2361e-01, -1.6955e-01,  3.7868e-02,  2.8833e-01,\n",
       "           4.0522e-02,  5.7925e-02, -2.2651e-01,  5.5807e-01,  4.4511e-01,\n",
       "           1.6609e-01,  4.5662e-01,  1.6104e-01,  4.5272e-02, -5.1597e-01,\n",
       "           2.1978e-01, -1.0213e+00, -9.4260e-02, -4.6640e-02, -1.5401e-01,\n",
       "          -4.4224e-01, -2.8126e-02,  1.6814e-01,  1.8538e-01,  6.9597e-01,\n",
       "           7.2967e-01, -3.0871e-01, -4.5290e-01, -1.8510e-01,  1.5304e-01,\n",
       "           2.7705e-01, -1.9517e-01,  3.5971e-01,  9.6103e-02,  2.4432e-01,\n",
       "          -5.0967e-01, -4.3165e-01, -4.2252e-02,  3.1411e-01, -2.0171e-01,\n",
       "           5.7118e-01, -2.0164e-01, -1.3863e-01,  9.0832e-02,  1.3770e-01,\n",
       "           1.6639e-01,  7.9664e-01,  4.3416e-01, -6.3425e-01,  8.7469e-02,\n",
       "          -1.3980e-02, -2.9988e-01,  2.5983e-02, -4.3577e-01,  2.8858e-01,\n",
       "           1.2984e-01,  4.9280e-01,  6.7504e-01]], device='mps:0'),\n",
       " tensor([[-3.6270e-01, -6.5215e-01, -5.0988e-01, -8.0487e-01,  1.7802e-01,\n",
       "          -6.8133e-01, -3.7467e-01,  2.8280e-01, -7.2522e-01,  4.4208e-01,\n",
       "           4.5373e-01, -2.9458e-02,  1.9788e-01,  1.0903e-01, -2.7139e-01,\n",
       "          -3.4415e-01, -2.4150e-02,  2.4970e-01,  3.6300e-01,  5.2085e-01,\n",
       "           6.8739e-01,  3.8730e-01, -3.5588e-01,  1.3672e-01, -2.0690e-01,\n",
       "           1.0850e-02, -1.3338e-01, -2.6168e-01,  1.6891e-01, -3.6702e-01,\n",
       "           5.2982e-02,  1.6025e-01, -3.5756e-01,  7.6654e-01, -1.6591e-01,\n",
       "          -6.2621e-01,  3.8187e-01, -2.9036e-01, -1.6128e-02, -1.4638e-01,\n",
       "           7.2402e-01, -1.0238e-01, -4.5463e-02, -3.4140e-01,  6.4434e-02,\n",
       "           7.2586e-02,  2.6605e-01, -6.0425e-01, -2.1140e-01, -5.3489e-01,\n",
       "          -3.3882e-01, -3.8213e-02, -1.0454e-01,  1.3405e-01, -5.4525e-01,\n",
       "          -5.9167e-01,  2.8984e-01,  1.7095e-01,  1.2780e-01,  3.7134e-01,\n",
       "          -4.9405e-01,  2.2868e-01,  5.2417e-02, -7.6663e-02, -4.7597e-01,\n",
       "          -1.2381e-01, -6.5609e-01, -5.8729e-01, -3.0152e-01, -2.9800e-01,\n",
       "           3.3472e-02,  3.4462e-01,  2.8241e-02,  1.7713e-01, -8.0296e-02,\n",
       "          -1.2805e-01,  7.0756e-01,  7.9893e-02, -6.4112e-02,  1.7404e-01,\n",
       "           3.0660e-01,  3.1728e-01,  7.4921e-02, -2.6456e-01, -7.8233e-02,\n",
       "           2.5605e-01,  7.5829e-01,  2.6791e-01,  7.0584e-02,  6.8077e-02,\n",
       "           7.2070e-01, -2.3763e-02,  2.3272e-02,  5.5358e-02,  4.4119e-01,\n",
       "          -4.7452e-02,  6.1673e-01, -3.7039e-01,  3.4372e-01, -3.1355e-01,\n",
       "          -1.0121e-01, -4.7478e-02, -1.6188e-02, -1.1358e-01, -4.5126e-01,\n",
       "          -4.9278e-01, -8.1535e-02,  2.0063e-02, -3.1114e-01,  4.2545e-02,\n",
       "          -1.2792e-01, -3.4478e-01, -3.5124e-03, -2.5416e-01, -1.3504e-01,\n",
       "           7.5419e-02,  1.7931e-01, -7.9693e-02, -2.4554e-01,  6.6228e-01,\n",
       "           4.7316e-01, -2.6959e-01,  2.6640e-01,  2.5193e-01, -6.3158e-02,\n",
       "          -3.0547e-01,  4.2531e-01, -1.4265e-01,  5.1440e-02,  2.6812e-01,\n",
       "           4.0500e-02, -3.0608e-01, -3.8535e-01,  1.7838e-01,  3.6891e-01,\n",
       "           5.6730e-02, -1.4521e-01, -1.8076e-01, -3.0777e-01, -6.7973e-02,\n",
       "           2.6486e-01, -2.3957e-01, -1.7611e-01, -2.2361e-01,  1.1938e-01,\n",
       "          -2.2428e-01, -4.1886e-02, -1.0626e-01, -1.6847e-01, -6.4419e-01,\n",
       "          -3.7255e-01, -2.4861e-01,  5.0068e-01,  1.5010e-01, -1.8235e-01,\n",
       "          -3.8709e-02,  4.4646e-01, -1.2241e-01,  1.2574e-01,  9.4958e-02,\n",
       "           8.8366e-01, -5.4372e-01, -1.6146e-01, -5.5897e-01,  4.3608e-02,\n",
       "           1.9553e-01,  1.5869e-01, -4.2238e-01,  3.5905e-01, -1.5884e-01,\n",
       "           2.6286e-01, -1.1542e-01,  4.6535e-01, -1.5467e-01, -2.8009e-01,\n",
       "          -6.6449e-01,  1.0754e-01, -1.8776e-01,  6.5808e-02, -1.5899e-01,\n",
       "           6.1530e-01, -1.7201e+00,  4.3007e-01, -1.2610e-02, -2.3068e-01,\n",
       "          -2.9953e-01,  6.3734e-02,  8.8103e-02,  5.0743e-01, -8.5834e-02,\n",
       "          -5.9973e-02, -5.2050e-02,  9.4586e-02,  4.5724e-01, -3.4299e-01,\n",
       "           2.3568e-01,  6.9477e-01,  2.5899e-01, -9.7720e-02, -1.3246e-01,\n",
       "          -4.3979e-01,  2.0488e-01, -1.0982e-01, -4.1112e-01,  1.2262e-01,\n",
       "          -1.1700e-01, -7.2489e-01,  1.5881e-01,  1.5312e-01,  3.1458e-02,\n",
       "          -2.9101e-02, -5.0807e-01, -3.5976e-01, -1.1326e-01,  9.0930e-02,\n",
       "           4.6663e-01, -4.8601e-01, -5.6244e-02,  4.5361e-01,  1.9120e-01,\n",
       "           2.9845e-01, -2.3202e-01,  1.8345e-01,  4.4429e-03, -1.9501e-01,\n",
       "           5.0441e-01, -5.6907e-01, -2.4975e-01, -4.9835e-02, -8.1999e-02,\n",
       "          -2.7322e-01, -1.0355e+00,  6.0225e-01, -2.5870e-01, -1.5653e-02,\n",
       "          -1.5236e-01, -9.2166e-02,  3.7864e-01,  1.9006e-01, -4.0710e-01,\n",
       "          -1.2410e-01, -6.0976e-01, -6.7407e-01,  7.0621e-02, -2.2560e-01,\n",
       "           6.4178e-01, -1.0950e-01,  5.1073e-01,  9.4721e-01,  4.0389e-01,\n",
       "           2.6916e-01,  1.5936e-01,  5.2482e-01,  5.0305e-01, -4.9883e-01,\n",
       "           1.5769e-01,  1.1583e-01, -4.3552e-01,  1.8467e-01, -2.9198e-01,\n",
       "          -2.8877e-01, -3.4814e-02,  1.5723e-01,  5.0557e-01, -2.6652e-01,\n",
       "          -2.1721e-01,  7.5306e-01, -4.4490e-01, -2.6333e-01,  1.5671e-01,\n",
       "          -3.0001e-01,  2.9512e-01, -3.4950e-01, -9.7230e-02, -6.6615e-01,\n",
       "          -3.1250e-01,  3.8925e-01,  1.0772e-01, -2.2841e-03, -3.1736e-01,\n",
       "           2.5772e-01, -1.3272e-01, -2.6435e-01, -1.2742e-01, -2.7513e-01,\n",
       "           4.2279e-01, -2.4325e-01,  1.2035e-01, -3.6201e-01, -5.0028e-01,\n",
       "          -5.5430e-01,  2.6018e-01, -4.8079e-01, -2.5354e-01,  4.7773e-01,\n",
       "          -2.5403e-01,  2.3220e-01,  1.7084e-01, -2.0042e-01, -4.9656e-01,\n",
       "           3.1345e-01, -6.5787e-02, -1.6607e-01, -1.2914e-02,  1.1665e-01,\n",
       "           2.3871e-01,  4.2832e-01,  6.0972e-01, -9.5562e-01, -5.5234e-01,\n",
       "           9.2377e-02,  4.8833e-01,  5.2371e-03, -1.4947e-01,  1.1713e-03,\n",
       "          -1.3175e+00, -3.4648e-01, -3.8330e-01, -1.9188e-01, -2.8640e-01,\n",
       "           2.2791e-01,  2.0245e-01, -2.5386e-01, -4.1770e-03, -1.1878e-01,\n",
       "           3.5018e-01,  8.4992e-01, -2.3189e-01,  2.1018e-01, -3.1639e-01,\n",
       "          -7.1295e-02, -3.6755e-01, -6.4900e-01, -7.5903e-02,  4.1987e-02,\n",
       "           5.8455e-01,  4.5367e-01, -1.2757e-01, -6.6992e-01,  3.9684e-02,\n",
       "           5.0619e-01, -3.1295e-01, -1.4824e-02,  3.9301e-01, -1.9960e-02,\n",
       "          -1.3317e-01,  1.4705e-01, -1.5024e-01,  1.8284e-01,  2.8640e-01,\n",
       "           8.3827e-02,  2.8530e-02,  2.4696e-01, -9.5439e-02,  5.0265e-01,\n",
       "           2.1515e-01,  3.1393e-01, -3.7892e-01,  1.8050e-01,  4.1358e-01,\n",
       "          -4.6528e-01, -7.2354e-03, -7.5776e-01, -1.1383e-01,  1.2942e-01,\n",
       "          -5.8990e-02,  3.5951e-01, -4.6891e-01,  1.0069e-01,  5.8809e-02,\n",
       "          -4.0930e-01, -5.8632e-01, -6.9460e-02, -1.3555e-01,  9.7353e-02,\n",
       "           1.3683e-01, -5.9959e-01,  6.8004e-02, -5.1309e-01, -1.5065e-01,\n",
       "          -1.6959e-02,  3.1878e-02,  3.3508e-02, -3.1471e-01, -1.8807e-01,\n",
       "          -2.3449e-01, -2.1379e-01,  5.3599e-01,  4.3381e-01,  3.7322e-01,\n",
       "           2.7614e-01,  1.9204e-02,  3.4715e-02,  8.5715e-01,  1.3239e-01,\n",
       "          -1.1231e-01,  4.4765e-01, -3.5604e-01,  2.8132e-01,  2.7218e-01,\n",
       "          -5.8259e-01,  7.1325e-02,  5.3821e-01,  5.2840e-01,  2.5765e-01,\n",
       "          -2.9380e-01,  5.9266e-02,  7.4316e-01, -3.7811e-01, -1.3408e-01,\n",
       "          -2.2333e-01,  9.1640e-01, -2.4506e-01, -4.8249e-02, -2.4009e-01,\n",
       "           4.6469e-02, -2.9129e-01, -4.2219e-01,  5.4288e-01, -5.1785e-01,\n",
       "           8.2026e-01, -4.4863e-01,  2.5458e-01,  2.3676e-01, -1.1104e+01,\n",
       "           3.9364e-01,  8.8448e-01, -5.2927e-01,  1.5315e-01, -2.3072e-01,\n",
       "           1.7764e-01, -8.8837e-02,  1.3392e-01, -1.4442e-02,  5.3465e-02,\n",
       "          -1.4674e-01,  2.9250e-01,  1.5271e-01, -1.6173e-01, -1.3114e-01,\n",
       "          -3.2141e-01, -1.0367e+00,  1.5804e-01, -4.3071e-02,  2.9477e-01,\n",
       "          -3.9575e-01, -7.5358e-02,  3.0770e-01, -4.8385e-01, -3.9994e-01,\n",
       "          -2.7091e-01, -3.2145e-01,  1.5822e-01,  4.8909e-01,  1.4553e-01,\n",
       "           2.6673e-01,  2.8539e-01,  4.2807e-01, -2.5985e-01, -5.7315e-01,\n",
       "           1.9557e-01,  3.5491e-01,  6.0529e-01,  3.0779e-01, -1.1977e-01,\n",
       "          -3.7191e-01, -7.3323e-02,  4.4910e-01,  8.2825e-01,  6.4901e-01,\n",
       "           1.3692e+00,  1.0846e+00,  3.9550e-01, -2.0117e-02, -3.8815e-01,\n",
       "          -7.3249e-01,  9.0892e-02, -8.5125e-01,  5.0925e-01, -2.3303e-02,\n",
       "           4.8555e-01, -1.2607e-01,  4.2978e-01,  1.1250e-01,  6.3109e-02,\n",
       "           5.8798e-03, -1.8851e-02,  2.5583e-01, -8.3729e-01,  7.8096e-02,\n",
       "           4.7570e-01,  2.9132e-01,  1.6944e-01, -7.4783e-01,  2.7922e-02,\n",
       "          -4.0768e-01,  4.3394e-01,  3.1936e-01,  4.0201e-01,  3.5720e-02,\n",
       "          -2.2940e-01, -1.7205e-01,  5.5768e-01, -7.1074e-01,  8.8897e-02,\n",
       "          -5.1904e-02,  3.0008e-01, -2.1599e-01, -6.4271e-01, -2.6945e-01,\n",
       "          -1.9801e-01, -2.7184e-01,  5.9395e-01,  2.1972e-01, -3.8144e-01,\n",
       "           3.1547e-02, -1.4324e-01, -7.3949e-02, -4.7354e-02,  7.9006e-01,\n",
       "          -3.6566e-01, -1.4887e-01,  5.9493e-03,  3.6977e-01, -5.0459e-01,\n",
       "          -3.6637e-01,  6.3689e-02, -4.6381e-01, -3.5531e-02, -3.7982e-02,\n",
       "          -1.6031e-01,  5.9399e-03,  1.6690e-01,  2.2444e-01, -2.2804e-01,\n",
       "          -5.9420e-02, -2.6232e-01,  4.3701e-01, -2.9491e-01, -9.6894e-01,\n",
       "           2.0977e-01,  5.7049e-01, -1.6526e-01, -5.9852e-01, -4.3393e-02,\n",
       "           9.6894e-02,  3.3377e-01,  1.6330e-01, -1.2391e-01,  3.6261e-01,\n",
       "          -5.1523e-01,  2.9856e-01,  3.6329e-01,  2.9748e-01,  5.3401e-01,\n",
       "           1.1691e-01, -3.8662e-01, -3.2452e-02, -1.1149e-01, -8.8181e-03,\n",
       "          -7.3665e-01, -2.6495e-01,  3.8657e-01, -7.3934e-01,  1.2326e-01,\n",
       "           1.8665e-01, -2.0697e-01, -4.2096e-01,  5.0210e-01,  3.8878e-01,\n",
       "           9.8851e-02,  2.0805e-01, -1.5532e-01, -5.8096e-02, -1.1509e-01,\n",
       "           1.3412e-03, -4.0164e-01, -3.9889e-01,  1.8618e-01, -1.2665e-01,\n",
       "          -3.4566e-01,  1.9871e-01, -2.0355e-01, -6.0255e-01,  3.6210e-01,\n",
       "           4.4467e-01, -1.7413e-01, -3.6313e-01,  8.6638e-01,  1.8786e-02,\n",
       "          -1.0293e-01, -5.0308e-01, -5.2215e-01, -2.4729e-01,  1.5831e-01,\n",
       "           4.1857e-01, -3.6632e-01, -5.7050e-01, -5.4976e-02, -2.0804e-01,\n",
       "          -1.3428e-01,  1.0619e+00, -4.3190e-01, -4.1496e-01,  5.4167e-01,\n",
       "           1.2294e-01, -1.2418e-01,  2.5114e-01, -4.5661e-02,  3.1203e-01,\n",
       "          -2.5742e-01, -2.8379e-02,  1.4288e-01, -1.1158e-01,  6.2508e-01,\n",
       "           7.9240e-02,  2.2458e-01, -2.5234e-01, -4.4185e-02, -1.7237e-01,\n",
       "          -3.8215e-01, -5.9907e-02, -1.8766e-02, -1.3430e-01,  3.0048e-02,\n",
       "           2.3452e-01, -2.9096e-01,  1.5172e-01,  3.3685e-01,  2.0404e-01,\n",
       "           4.6090e-01,  1.2033e+00, -1.6995e-01, -4.8221e-01,  9.5354e-02,\n",
       "          -4.3690e-02,  1.7427e-01, -4.3596e-02, -3.2821e-02, -8.8184e-02,\n",
       "           3.0450e-01,  8.2787e-03, -1.9764e-01,  2.6004e-01,  1.9134e-02,\n",
       "           1.8363e-01, -5.4772e-01, -3.8338e-01, -1.0452e-01,  2.2003e-01,\n",
       "          -7.2212e-02,  4.9270e-01, -2.1244e-01,  4.4858e-01,  1.2847e-01,\n",
       "          -5.3646e-01,  4.3220e-02, -1.8903e-01, -4.0923e-01, -4.2418e-01,\n",
       "          -1.1562e-01, -3.2983e-01, -2.9603e-01, -3.8993e-03,  2.4771e-01,\n",
       "           4.2492e-01, -9.7099e-01,  2.0999e-01,  2.5179e-01, -3.1100e-01,\n",
       "           2.6868e-01, -2.7205e-02,  4.9881e-01,  7.6139e-01,  8.5859e-02,\n",
       "           1.3097e-01,  3.0086e-01, -7.2706e-01, -4.8737e-01,  1.8239e-01,\n",
       "           8.1475e-01,  3.4519e-02, -2.7122e-01, -2.1324e-01, -6.4977e-02,\n",
       "          -2.4349e-01,  1.9247e-01,  8.0165e-02, -1.0089e-01,  2.6856e-01,\n",
       "          -3.8208e-01, -1.6545e-02,  3.0341e-01,  5.8069e-02, -2.9012e-01,\n",
       "          -5.8341e-02,  4.9297e-01,  6.7625e-01,  1.4584e-01, -5.2831e-01,\n",
       "          -2.4353e-01,  3.5675e-01, -3.4948e-01,  9.5828e-01, -1.3007e-02,\n",
       "          -3.2563e-01,  3.4790e-01,  1.4617e-01, -3.3546e-01,  4.9105e-01,\n",
       "          -5.2320e-01, -4.0881e-02, -1.4062e-01, -2.5149e-01,  6.5662e-01,\n",
       "           1.3665e-01,  5.3128e-01, -7.0242e-01, -2.9404e-01,  3.0185e-01,\n",
       "          -2.4620e-01,  3.5851e-01,  2.1633e-01,  4.3270e-01, -2.2250e-02,\n",
       "          -3.2621e-01, -1.3892e-01, -1.4768e-01, -4.6209e-02, -8.5268e-02,\n",
       "          -1.5176e-01, -3.8354e-01, -1.1540e-01, -3.4726e-02,  9.4009e-01,\n",
       "           6.9876e-01, -2.4898e-01, -1.3096e-01, -4.3854e-01,  2.7761e-02,\n",
       "           2.3517e-01, -2.1449e-01,  9.3143e-02, -3.2967e-01, -2.6774e-01,\n",
       "          -3.4250e-01, -4.9609e-01,  6.5761e-02,  2.4082e-01, -1.1732e-01,\n",
       "           6.9279e-01,  1.4597e-02,  1.0235e-01,  5.8834e-01,  4.6978e-01,\n",
       "          -1.0664e-01,  9.3548e-01, -9.3373e-03, -7.6442e-01, -2.1719e-01,\n",
       "           3.9957e-01, -1.1130e-01,  2.4418e-01, -5.3271e-01,  4.1766e-01,\n",
       "          -1.5317e-02,  7.0326e-01,  4.9588e-01]], device='mps:0'),\n",
       " tensor([[-2.5187e-01, -4.0054e-01, -5.8007e-01, -8.3055e-01, -1.2868e-01,\n",
       "          -2.8804e-02, -7.1917e-01,  6.6701e-01,  1.2709e-01,  5.0855e-01,\n",
       "           1.2570e-01,  5.1916e-01,  5.1676e-02, -1.1521e-01, -1.6481e-01,\n",
       "          -4.9820e-01,  5.6222e-02,  3.7338e-01,  6.7187e-01, -2.4111e-02,\n",
       "           2.2844e-01,  5.7979e-01, -2.6323e-01, -1.2386e-02, -8.3013e-01,\n",
       "           2.9613e-01, -1.1576e-01,  2.2409e-01,  1.7718e-01, -6.3092e-01,\n",
       "           4.0999e-02,  4.7417e-01,  2.3068e-02,  1.3497e+00, -5.6080e-01,\n",
       "          -1.3230e-01,  8.2213e-02, -3.9453e-01, -2.3010e-01,  3.7283e-02,\n",
       "           6.1164e-01, -4.1315e-01,  1.3635e-01,  4.0254e-01, -6.1085e-02,\n",
       "           8.9076e-02,  1.3899e-01, -3.7744e-01,  2.5892e-01, -8.3961e-01,\n",
       "          -4.8318e-02, -2.0669e-01,  4.9144e-01,  6.2615e-01, -6.3176e-01,\n",
       "          -2.2975e-01,  2.7419e-01, -3.7716e-02, -3.3020e-02,  1.5349e-01,\n",
       "           9.7980e-02,  3.8582e-01, -2.9995e-01,  2.3257e-01,  3.3441e-01,\n",
       "           2.4458e-01, -3.0392e-01, -4.8889e-01, -5.4078e-01, -5.0111e-02,\n",
       "           9.1859e-03,  2.2490e-02,  1.4846e-01, -2.3272e-01,  4.1973e-01,\n",
       "          -7.1214e-01, -6.4354e-02, -8.9125e-02, -2.5973e-01,  2.9246e-01,\n",
       "          -4.5549e-02,  1.9719e-01,  3.1010e-01,  1.0268e-01,  1.2457e-01,\n",
       "           4.3419e-02,  2.7215e-01,  5.2932e-04,  4.7373e-02,  3.7533e-01,\n",
       "          -1.3375e-01, -8.1722e-02, -2.8313e-01, -3.5814e-01,  3.1229e-01,\n",
       "           4.0598e-02,  2.9907e-01, -1.9557e-01,  1.8579e-01, -3.2474e-01,\n",
       "           1.5588e-01,  1.7768e-01, -1.8083e-01,  5.7286e-02, -3.5172e-01,\n",
       "          -4.5475e-01,  2.3589e-02, -2.8569e-01, -4.8285e-01,  2.2697e-01,\n",
       "          -1.4211e-01, -3.0205e-01, -4.0922e-02, -3.8935e-01, -3.1524e-02,\n",
       "          -7.5164e-02,  2.6013e-01, -1.3230e-01, -3.4806e-01, -8.3033e-02,\n",
       "           2.0734e-01, -2.2808e-01,  1.6077e-01,  2.6882e-01,  1.2232e-01,\n",
       "          -9.6705e-02,  6.5251e-01,  1.8950e-02,  3.1674e-01,  2.5763e-01,\n",
       "           1.1699e-01, -6.5768e-02, -1.9745e-01, -2.3154e-01,  7.5988e-02,\n",
       "          -7.6941e-02,  1.0687e-01, -1.8182e-01, -2.2790e-01, -1.8508e-01,\n",
       "           3.1771e-01, -3.3097e-01, -6.0893e-01, -4.9260e-01, -1.6098e-01,\n",
       "          -1.6326e-01, -2.4804e-02, -8.2473e-03, -3.3693e-01, -3.1956e-01,\n",
       "          -6.7443e-01, -3.6717e-01,  8.1041e-01, -1.6831e-02, -3.3356e-01,\n",
       "          -1.6499e-01,  2.3208e-01,  3.1666e-01, -4.5959e-02,  3.7161e-01,\n",
       "           7.7250e-01, -4.7670e-02,  1.6786e-01, -5.1827e-01, -3.8195e-01,\n",
       "           4.8335e-02,  2.6658e-01, -2.7054e-01,  6.1879e-01,  4.8352e-02,\n",
       "           7.5627e-02, -6.6170e-02,  2.7430e-01, -3.2704e-01, -1.2324e-02,\n",
       "          -5.7573e-01,  3.7750e-01,  2.9539e-01, -3.9892e-01, -6.6567e-02,\n",
       "          -2.0834e-02, -1.2060e+00,  9.6964e-02, -8.3040e-01,  3.1236e-01,\n",
       "          -2.3110e-01,  1.7963e-01,  2.3389e-01,  2.9545e-01, -6.5320e-02,\n",
       "           2.5404e-01,  2.8725e-01,  2.5687e-01,  3.3609e-01, -3.4924e-02,\n",
       "           2.8019e-01,  2.3924e-01,  5.0790e-01,  4.1073e-01,  1.4769e-01,\n",
       "          -3.5440e-01,  2.9683e-01,  3.5781e-01,  9.0598e-02,  1.6954e-01,\n",
       "           9.9825e-02, -2.1636e-01,  2.1091e-01,  1.7501e-01, -1.6618e-01,\n",
       "          -2.2406e-01, -3.1100e-02, -4.6818e-01, -5.6655e-02, -4.8467e-01,\n",
       "           3.2557e-01, -5.1983e-01, -1.1912e-01,  3.4831e-01,  2.4340e-01,\n",
       "          -1.4896e-02,  1.8319e-01,  1.7196e-01, -2.4338e-01,  2.1735e-01,\n",
       "           3.7449e-01, -3.4618e-01, -3.9804e-02, -3.2197e-01, -2.9367e-01,\n",
       "           2.4357e-01, -5.6244e-01,  6.6635e-01,  4.0574e-01,  6.0165e-02,\n",
       "           1.5119e-01, -2.7920e-01, -5.1638e-02, -5.7408e-01,  3.5471e-01,\n",
       "          -3.2130e-01, -3.4955e-01, -5.4801e-01,  6.4487e-02,  4.2773e-01,\n",
       "           2.7565e-01, -6.4915e-02,  7.7868e-01,  4.2408e-01,  4.2813e-01,\n",
       "          -4.3003e-01, -1.5896e-02,  4.0377e-01, -4.1405e-02, -4.6059e-01,\n",
       "           6.7116e-02, -1.4962e-01, -5.1643e-01,  2.8000e-01, -2.6598e-01,\n",
       "          -1.2047e-01, -1.4913e-01,  3.3682e-01,  1.5365e-01, -2.1964e-01,\n",
       "          -3.8461e-01,  3.9587e-01, -1.4349e-01, -4.1723e-01,  4.7123e-01,\n",
       "          -3.5025e-01,  1.5209e-01,  1.0694e-01, -3.8441e-01, -6.2072e-01,\n",
       "          -3.2235e-01,  2.3090e-01,  1.5585e-01,  3.4799e-02, -6.6306e-01,\n",
       "          -4.0571e-01,  3.1197e-01,  2.4601e-01,  2.9976e-01,  2.9840e-01,\n",
       "           5.7427e-01, -3.0944e-01, -1.6702e-01,  1.4982e-01, -4.1377e-01,\n",
       "          -1.5704e-01, -3.3228e-02, -1.9995e-01,  2.9569e-01,  2.4768e-01,\n",
       "           1.1600e-01,  2.0367e-01,  4.3482e-01, -2.5272e-01, -3.8673e-01,\n",
       "          -7.1543e-02,  2.0161e-01, -9.3629e-02, -5.8306e-02,  2.3951e-01,\n",
       "          -6.3095e-02,  5.9650e-01,  7.2143e-01,  7.7439e-02, -6.4006e-01,\n",
       "           1.7696e-01,  5.2502e-01, -2.7253e-01, -4.0343e-02,  9.6365e-02,\n",
       "          -1.0344e+00, -2.1331e-01, -1.2883e-01, -2.0181e-01, -6.5352e-01,\n",
       "           8.8115e-02,  3.4863e-01,  4.6833e-03, -1.7864e-01, -3.0947e-01,\n",
       "           1.0448e-01,  4.8724e-02, -4.8980e-02,  4.3534e-01, -2.8901e-01,\n",
       "          -3.4797e-01, -2.3438e-01, -2.1532e-01, -5.4201e-01, -3.9968e-01,\n",
       "           3.5914e-01,  2.8129e-01,  1.1986e-01, -2.6486e-01, -5.1832e-01,\n",
       "           4.5577e-01, -6.8169e-01, -1.4632e-01,  2.4840e-01, -4.7702e-01,\n",
       "           1.4650e-02,  4.8990e-01,  1.4999e-01,  1.6336e-01,  6.5258e-02,\n",
       "           7.8557e-02, -3.3903e-01, -5.0030e-01, -1.5000e-01,  6.2323e-01,\n",
       "           3.2283e-01,  3.3506e-01, -1.7236e-01,  3.9096e-01,  4.5167e-01,\n",
       "          -2.9582e-01,  4.1670e-01,  1.0930e-02,  7.7439e-02, -6.0414e-02,\n",
       "           1.9048e-01,  3.4370e-01, -3.6842e-01,  2.0145e-01,  1.7929e-01,\n",
       "          -3.1634e-01, -3.2149e-01,  1.5691e-02,  2.0642e-01, -1.8580e-01,\n",
       "          -3.0875e-01, -2.0361e-01,  9.9698e-02, -3.4856e-01, -4.3104e-01,\n",
       "           2.9161e-01, -3.6543e-02,  7.0495e-02, -4.0357e-01, -4.3462e-01,\n",
       "          -2.0181e-01, -2.6055e-01,  6.8681e-01,  2.1064e-01, -1.9621e-01,\n",
       "           1.8720e-01, -1.0907e-01, -3.6330e-01,  4.7847e-01,  2.0162e-02,\n",
       "          -4.8754e-01,  4.5482e-01, -3.4822e-01,  1.0627e-01,  5.5756e-01,\n",
       "          -1.6769e-01, -1.0840e-01,  1.1008e-01,  1.0820e-01,  2.9026e-01,\n",
       "          -3.9884e-01,  1.5516e-01, -6.3476e-02, -3.1885e-01,  3.1820e-01,\n",
       "           2.6411e-01,  2.2781e-01, -1.7137e-01, -3.3426e-02,  3.5677e-02,\n",
       "           2.8782e-01, -6.0853e-02, -3.9888e-01, -4.7347e-01, -4.6132e-01,\n",
       "           1.2832e-01,  1.8927e-01,  5.6207e-01,  3.4144e-01, -1.1394e+01,\n",
       "           1.2938e-01,  7.8031e-01, -4.0713e-01, -8.2819e-02, -5.0320e-02,\n",
       "           1.0976e-02, -1.3708e-01,  4.2955e-01, -2.7734e-01, -4.4105e-01,\n",
       "          -3.5727e-01, -1.3308e-01,  1.5454e-01, -3.2189e-01, -2.6945e-01,\n",
       "          -1.4686e-02, -7.0935e-01,  4.0963e-01, -3.0286e-01,  5.0298e-01,\n",
       "          -2.3511e-01, -8.3798e-02,  1.5816e-01, -3.5851e-01, -1.3439e-02,\n",
       "           5.9766e-02, -1.6033e-01,  1.4651e-01,  4.2486e-01, -5.2072e-02,\n",
       "           3.3244e-02, -4.1258e-03,  4.8461e-02, -2.5723e-01, -5.9999e-01,\n",
       "           1.9296e-02, -1.4141e-01,  3.3422e-01, -4.4133e-02, -7.1265e-01,\n",
       "          -2.2533e-01, -2.7418e-01,  3.1407e-01,  2.7021e-01,  9.0598e-01,\n",
       "           2.2955e-01,  1.0686e+00,  1.5022e-01,  1.7748e-01,  2.6230e-02,\n",
       "          -7.8764e-01, -3.3269e-02, -4.1907e-01,  2.3280e-02,  1.0114e-01,\n",
       "           3.6576e-01, -7.9367e-01,  2.5251e-01, -3.5521e-01,  3.6487e-02,\n",
       "           4.9094e-01,  3.1768e-02,  2.7700e-01, -3.2298e-02, -7.8123e-02,\n",
       "           1.2089e-01, -2.8544e-02,  1.1098e-01, -2.2583e-01, -2.4084e-01,\n",
       "          -1.0177e-01,  5.9888e-01,  3.9592e-01, -7.2211e-01,  3.6112e-01,\n",
       "           4.9258e-02, -1.8622e-01,  1.4556e+00, -4.8573e-02,  1.8361e-02,\n",
       "           1.9800e-03,  2.9604e-01, -5.0236e-01, -7.3516e-01,  3.6511e-01,\n",
       "          -1.0211e-01,  5.6427e-02,  1.0127e-01,  4.8148e-01, -9.7481e-02,\n",
       "          -4.7940e-01, -1.8593e-03,  1.0556e-01, -1.3380e-01,  5.4710e-01,\n",
       "          -7.2824e-01, -1.6963e-01, -1.1223e-01,  3.1723e-01, -8.4054e-01,\n",
       "          -6.1126e-02, -7.0210e-01, -3.9328e-01,  5.0586e-01, -2.5896e-01,\n",
       "           1.5852e-01, -2.7443e-02, -4.3439e-02,  5.5574e-01, -2.1489e-01,\n",
       "           7.1555e-02,  2.2813e-01,  8.4995e-02, -1.6733e-01,  3.9349e-02,\n",
       "           4.1630e-01,  4.9571e-01, -4.2740e-01, -6.3911e-01, -7.6901e-03,\n",
       "           1.7180e-01,  1.4933e-01,  3.4019e-01, -5.3463e-01,  3.6431e-01,\n",
       "          -7.9162e-01, -1.8769e-01,  7.2949e-02, -4.3273e-02,  3.6175e-01,\n",
       "           2.3925e-01,  7.2038e-02,  2.2256e-01, -7.7756e-01, -6.4498e-01,\n",
       "          -6.5932e-01, -2.1445e-01,  2.6143e-01, -5.4236e-01, -8.3654e-02,\n",
       "          -1.1033e-01, -4.8837e-03, -5.0292e-02, -6.6102e-02,  1.1939e-02,\n",
       "           1.3632e-01,  2.3587e-01, -1.3023e-01, -1.3178e-01, -9.2464e-02,\n",
       "          -5.5566e-01, -1.1753e-01, -9.9591e-02, -1.2505e-01, -5.6783e-01,\n",
       "          -1.3797e-01,  3.3075e-02, -2.5373e-01, -6.8124e-02,  3.5608e-01,\n",
       "          -2.9832e-02, -2.7856e-02, -4.8850e-01,  7.2846e-01,  4.0279e-01,\n",
       "           5.2088e-02, -2.8298e-01, -5.5189e-01, -1.0567e-01,  2.9592e-01,\n",
       "           1.9555e-01, -6.7007e-02, -1.7709e-01, -2.4145e-01, -4.2522e-01,\n",
       "           1.4302e-01,  7.1147e-01, -6.6296e-02, -6.8115e-01,  2.6621e-01,\n",
       "           5.8223e-01, -1.2882e-01, -2.4432e-01,  2.0983e-02, -9.3284e-02,\n",
       "          -3.1851e-01, -6.0317e-01,  7.8821e-01, -1.1246e-01, -5.8638e-02,\n",
       "           2.2427e-01,  2.0359e-01, -9.2588e-03,  4.0209e-01, -1.5770e-01,\n",
       "          -3.5434e-01, -1.6477e-03, -7.3600e-01,  4.3257e-01,  6.1666e-02,\n",
       "          -3.1489e-01,  7.6082e-01,  2.7260e-01, -6.0593e-02, -4.5779e-02,\n",
       "           2.2863e-01,  7.9941e-01, -2.4636e-01,  6.4653e-02,  5.3779e-01,\n",
       "           8.8110e-02, -1.1229e-01,  3.9580e-02,  4.1665e-02,  4.5410e-02,\n",
       "           3.7012e-01, -3.2145e-02, -4.1564e-01, -6.1194e-02,  3.8924e-01,\n",
       "           3.9295e-01, -5.4610e-01, -7.3487e-01,  2.2333e-02,  6.8551e-01,\n",
       "          -2.9054e-01,  2.3654e-01, -3.2708e-02,  3.0880e-01,  1.2001e-01,\n",
       "          -9.2160e-01, -1.5016e-01, -1.0056e-01,  2.8229e-01, -2.5482e-02,\n",
       "          -1.6139e-01, -1.4727e-01, -2.3686e-01, -1.6208e-01,  6.8308e-01,\n",
       "           2.1991e-01, -6.9160e-01,  3.4058e-02,  5.7281e-01, -6.0551e-01,\n",
       "           4.6052e-01, -2.4577e-01, -3.4302e-02,  3.6435e-01,  1.3117e-01,\n",
       "           4.9067e-02,  1.7856e-01, -4.2138e-01, -4.0598e-01,  1.7824e-01,\n",
       "           9.1672e-01, -7.6613e-02,  1.1166e-01, -3.9384e-01, -1.7182e-01,\n",
       "          -7.1855e-02, -3.1913e-01, -3.1286e-01,  3.7185e-01,  2.0529e-01,\n",
       "          -3.2552e-02, -1.3271e-01, -2.2992e-01,  3.0816e-01,  1.1821e-01,\n",
       "           3.2536e-01,  5.8187e-01,  6.4368e-01, -1.0077e-02, -2.4936e-01,\n",
       "          -6.1235e-02,  1.4568e-01, -1.9524e-01,  5.2704e-01, -2.0627e-01,\n",
       "          -3.3453e-01, -3.8069e-03,  3.0134e-01, -6.1681e-01, -5.8725e-02,\n",
       "          -4.5426e-01,  4.2361e-01, -1.6955e-01,  3.7868e-02,  2.8833e-01,\n",
       "           4.0522e-02,  5.7925e-02, -2.2651e-01,  5.5807e-01,  4.4511e-01,\n",
       "           1.6609e-01,  4.5662e-01,  1.6104e-01,  4.5272e-02, -5.1597e-01,\n",
       "           2.1978e-01, -1.0213e+00, -9.4260e-02, -4.6640e-02, -1.5401e-01,\n",
       "          -4.4224e-01, -2.8126e-02,  1.6814e-01,  1.8538e-01,  6.9597e-01,\n",
       "           7.2967e-01, -3.0871e-01, -4.5290e-01, -1.8510e-01,  1.5304e-01,\n",
       "           2.7705e-01, -1.9517e-01,  3.5971e-01,  9.6103e-02,  2.4432e-01,\n",
       "          -5.0967e-01, -4.3165e-01, -4.2252e-02,  3.1411e-01, -2.0171e-01,\n",
       "           5.7118e-01, -2.0164e-01, -1.3863e-01,  9.0832e-02,  1.3770e-01,\n",
       "           1.6639e-01,  7.9664e-01,  4.3416e-01, -6.3425e-01,  8.7469e-02,\n",
       "          -1.3980e-02, -2.9988e-01,  2.5983e-02, -4.3577e-01,  2.8858e-01,\n",
       "           1.2984e-01,  4.9280e-01,  6.7504e-01]], device='mps:0'),\n",
       " tensor([[-0.2258, -0.7099, -0.3778,  ...,  0.3292,  0.7059,  0.7494],\n",
       "         [-0.3141, -0.3773, -0.4768,  ..., -0.0974,  0.4994,  0.5352],\n",
       "         [-0.3619, -0.2640, -0.6956,  ...,  0.2612,  0.3177,  0.4225],\n",
       "         [-0.3258, -0.2937, -0.7479,  ...,  0.3141,  1.0802, -0.0097]],\n",
       "        device='mps:0'),\n",
       " tensor([[ 0.0328, -0.4838, -0.3429,  ...,  0.1665,  0.3770,  0.5988],\n",
       "         [-0.1242, -0.5179, -0.4093,  ...,  0.3691, -0.0529,  0.5134],\n",
       "         [-0.3292, -0.2590, -0.5016,  ..., -0.0148,  0.3509,  0.1607]],\n",
       "        device='mps:0'),\n",
       " tensor([[-1.6463e-01, -6.2987e-01, -5.9491e-01, -8.7478e-01, -2.1700e-01,\n",
       "          -2.0336e-01, -8.2936e-01,  4.8916e-01,  2.4250e-01,  3.8665e-01,\n",
       "           1.1729e-01,  3.1057e-01,  3.8316e-02, -1.2584e-01, -2.9434e-01,\n",
       "          -2.6972e-01,  1.2207e-01,  4.2729e-01,  6.4028e-01, -6.6002e-02,\n",
       "           4.4404e-01,  5.9261e-01, -2.6728e-01, -6.5947e-03, -6.7512e-01,\n",
       "           2.2431e-01, -1.7137e-02,  4.1736e-01,  1.3046e-01, -6.0291e-01,\n",
       "           2.0874e-01,  4.7479e-01, -8.7799e-02,  1.3142e+00, -3.8592e-01,\n",
       "          -2.0139e-01,  8.5795e-02, -3.2643e-01, -1.5136e-01,  1.7652e-01,\n",
       "           6.5120e-01, -2.4499e-01, -1.0717e-01,  4.6042e-01,  2.2148e-02,\n",
       "           3.1235e-02,  1.8368e-01, -3.5098e-01,  4.9288e-01, -6.4433e-01,\n",
       "          -1.3215e-01, -1.2080e-01,  3.4523e-01,  3.6739e-01, -6.3042e-01,\n",
       "          -3.5579e-01,  3.1771e-01,  1.2236e-01, -2.0105e-02,  5.2927e-02,\n",
       "           1.4721e-02,  1.9159e-01, -3.1728e-01,  3.5220e-01,  1.0779e-01,\n",
       "           1.6108e-01, -4.2660e-01, -6.0987e-01, -5.8009e-01, -1.1271e-01,\n",
       "           5.6932e-02,  1.3088e-01,  2.4465e-01, -1.1914e-01,  3.9873e-01,\n",
       "          -8.6018e-01,  1.5567e-01, -3.7925e-02, -2.1723e-01,  2.6124e-01,\n",
       "          -2.0660e-01,  1.4472e-01,  2.7131e-01,  1.0551e-01,  2.3004e-01,\n",
       "           1.2352e-01,  3.8836e-01, -2.2385e-02,  8.0606e-02,  3.7966e-01,\n",
       "           9.5459e-03, -1.6299e-02, -4.9791e-02, -2.7074e-01,  1.4597e-01,\n",
       "           1.0283e-02,  3.7803e-01, -3.7144e-01, -2.2207e-02, -3.3699e-01,\n",
       "           1.4116e-01,  1.5095e-01, -2.6062e-02,  2.5036e-01, -5.4525e-01,\n",
       "          -4.5018e-01, -4.9731e-02, -2.5837e-01, -5.4973e-01,  1.4000e-01,\n",
       "          -3.0289e-01, -1.6754e-01,  6.9134e-02, -5.3704e-01,  1.6621e-01,\n",
       "          -1.4252e-01,  6.7407e-02, -2.7345e-02, -3.6052e-01, -5.3335e-02,\n",
       "           1.6791e-01, -3.6935e-01,  2.2958e-01,  7.7249e-02,  7.6346e-02,\n",
       "          -1.5792e-02,  6.3841e-01,  6.7344e-04,  2.3178e-01,  4.5671e-01,\n",
       "           1.3759e-01, -1.4423e-01, -2.4303e-01, -2.5499e-01, -1.1037e-01,\n",
       "           9.4719e-02,  2.4416e-01, -1.1443e-01, -4.2458e-01, -1.7138e-01,\n",
       "           3.8709e-01, -2.2040e-01, -4.7081e-01, -2.9592e-01, -2.4868e-01,\n",
       "          -3.2852e-01, -2.7708e-01, -1.1309e-01, -3.4374e-01, -5.5468e-01,\n",
       "          -6.0400e-01, -4.2181e-01,  8.4459e-01, -1.9217e-01, -2.6822e-01,\n",
       "          -1.8096e-01,  5.9869e-02,  1.5524e-01,  9.7654e-02,  4.8522e-01,\n",
       "           7.0164e-01, -3.0016e-01,  1.6184e-01, -5.3280e-01, -2.9208e-01,\n",
       "           2.0326e-01,  2.1472e-01, -6.0359e-02,  3.2878e-01,  2.2210e-02,\n",
       "           1.8229e-01, -1.2904e-01,  4.6832e-01, -2.7730e-01,  2.5480e-02,\n",
       "          -7.4897e-01,  5.3443e-01,  3.0671e-01, -3.5066e-01,  4.1661e-02,\n",
       "           8.9642e-02, -1.2211e+00,  7.0465e-02, -9.3871e-01,  5.1877e-01,\n",
       "          -7.7421e-02,  1.7014e-01,  2.1269e-01,  2.9861e-01, -1.2347e-01,\n",
       "           2.7480e-01,  1.3661e-01,  3.1198e-01,  3.5359e-01, -2.7697e-03,\n",
       "           2.2140e-01,  3.0792e-01,  4.0054e-01,  2.3283e-01,  2.2982e-01,\n",
       "          -5.4856e-01,  3.4385e-01,  3.7915e-01, -2.9584e-01,  2.8055e-01,\n",
       "           5.2847e-02, -3.6500e-01,  1.8934e-01, -4.8190e-02,  8.7815e-02,\n",
       "          -2.7963e-01, -2.3524e-01, -5.7314e-01, -1.7597e-01, -4.7477e-01,\n",
       "           2.0949e-01, -5.3230e-01, -2.6017e-01,  3.1297e-01,  1.8246e-01,\n",
       "          -2.9728e-02,  2.1614e-01,  1.6065e-01, -3.0667e-01,  1.5972e-01,\n",
       "           4.9853e-01, -5.3870e-01,  1.5897e-02, -1.6202e-01, -3.1372e-01,\n",
       "           2.6124e-01, -6.7454e-01,  4.9924e-01,  2.2036e-01,  2.4722e-01,\n",
       "           1.4398e-01, -2.0770e-01,  5.2168e-04, -5.4831e-01,  1.6056e-01,\n",
       "          -3.7938e-01, -3.8020e-01, -6.3491e-01, -2.3992e-03,  2.6465e-01,\n",
       "           1.2630e-01, -5.3641e-02,  6.8822e-01,  5.0221e-01,  4.9185e-01,\n",
       "          -3.7572e-01, -1.0667e-01,  4.9007e-01, -1.8869e-02, -4.9092e-01,\n",
       "           3.0384e-01, -6.0874e-02, -3.9836e-01,  2.9423e-01, -3.4384e-01,\n",
       "          -2.1663e-01, -1.5437e-01,  4.1186e-01,  2.7257e-01, -2.6826e-01,\n",
       "          -3.4368e-01,  8.1981e-02, -7.8358e-03, -2.9391e-01,  6.3611e-01,\n",
       "          -2.3484e-01,  1.6300e-02, -1.2838e-01, -3.2362e-01, -6.5488e-01,\n",
       "          -3.5663e-01,  3.2779e-01,  2.0480e-01,  1.2334e-01, -7.1848e-01,\n",
       "          -4.0515e-01,  4.5434e-01,  3.4615e-01,  2.4363e-01,  1.9003e-01,\n",
       "           5.2387e-01, -4.9199e-01,  5.6070e-02,  1.0032e-01, -4.2104e-01,\n",
       "          -5.2305e-02,  8.2874e-03, -5.0788e-01,  3.6136e-01,  3.6076e-01,\n",
       "          -4.1847e-02,  3.3360e-01,  4.9221e-01, -1.6382e-01, -4.5178e-01,\n",
       "          -4.8624e-02,  3.4233e-01, -3.5644e-01,  3.5331e-02,  1.6714e-01,\n",
       "           1.2276e-02,  5.5694e-01,  6.1463e-01, -4.5377e-02, -5.5626e-01,\n",
       "           3.5769e-01,  6.4229e-01, -1.5019e-01, -2.2446e-01,  2.1123e-02,\n",
       "          -8.1552e-01, -6.7352e-02, -1.6750e-01, -3.8302e-01, -8.4950e-01,\n",
       "          -6.2288e-02,  1.6949e-01, -4.6078e-02, -1.6614e-01, -5.0993e-01,\n",
       "           2.4136e-01,  2.9618e-01,  1.0102e-01,  2.4892e-01, -1.5047e-01,\n",
       "          -4.7821e-01, -2.0784e-01, -3.1303e-01, -5.7959e-01, -6.2894e-01,\n",
       "           2.7068e-01,  3.2294e-01,  5.1956e-02, -2.1313e-01, -2.2139e-01,\n",
       "           3.9853e-01, -5.6407e-01, -2.7864e-01,  3.1518e-01, -3.8869e-01,\n",
       "          -7.4271e-02,  3.1308e-01,  4.2101e-01, -4.4025e-02, -4.6946e-02,\n",
       "          -7.8398e-03, -5.0189e-01, -5.5504e-01, -2.6592e-01,  5.7629e-01,\n",
       "           4.7239e-01,  2.6343e-01, -4.3606e-01,  3.2872e-01,  4.9554e-01,\n",
       "          -3.3007e-01,  2.9133e-01,  2.1073e-01,  4.6633e-02, -2.4488e-01,\n",
       "           2.9313e-01,  5.3606e-01, -3.6562e-01,  5.3967e-02,  1.9392e-01,\n",
       "          -2.2816e-01, -2.1657e-01, -1.8747e-01,  2.7192e-01, -2.8113e-01,\n",
       "          -3.1234e-01, -1.8871e-01,  1.7532e-02, -2.0115e-01, -3.4979e-01,\n",
       "           3.4833e-01,  8.5896e-02, -7.9513e-02, -3.5707e-01, -4.1229e-01,\n",
       "          -1.6507e-01, -3.2432e-01,  6.9834e-01,  1.5338e-01, -1.4228e-01,\n",
       "           3.3665e-01, -9.0682e-02, -4.0066e-01,  4.6967e-01, -7.0365e-03,\n",
       "          -5.2839e-01,  5.5518e-01, -1.2325e-01,  1.0326e-01,  3.9681e-01,\n",
       "          -1.4070e-01,  7.0161e-02,  1.2451e-01,  3.8194e-01,  3.6549e-01,\n",
       "          -2.7507e-01,  6.2256e-02,  8.2569e-02, -3.8922e-01,  3.9837e-01,\n",
       "           3.2644e-01, -6.7971e-02, -2.3194e-01,  1.2435e-01, -6.1024e-02,\n",
       "           2.0518e-01,  9.0217e-02, -4.8852e-01, -5.8374e-01, -3.3657e-01,\n",
       "           3.2339e-01,  1.4435e-01,  5.4899e-01,  2.5475e-01, -1.1235e+01,\n",
       "           7.4443e-02,  8.9649e-01, -4.5782e-01, -3.3704e-01, -1.4448e-01,\n",
       "          -8.0281e-02, -9.8865e-02,  4.0282e-01, -2.1118e-01, -4.8351e-01,\n",
       "          -4.0287e-01,  1.0217e-02,  1.0951e-01, -4.5019e-01, -2.9744e-01,\n",
       "           2.6881e-02, -5.6494e-01,  4.1648e-01, -1.6170e-01,  5.4023e-01,\n",
       "          -3.9491e-01,  5.9613e-02,  1.6446e-01, -3.5262e-01,  1.1604e-01,\n",
       "           1.5545e-02, -1.4863e-02,  2.6880e-01,  6.0715e-01, -1.1844e-01,\n",
       "           2.1563e-01, -2.9074e-02, -1.7177e-02, -4.3625e-01, -4.2844e-01,\n",
       "           2.0475e-01, -1.3481e-01,  7.2580e-01,  2.8171e-02, -8.1600e-01,\n",
       "          -1.8184e-01, -3.3844e-01,  2.4657e-01,  3.6242e-01,  8.3710e-01,\n",
       "           5.3237e-01,  1.1983e+00,  1.4850e-01,  1.8626e-01, -9.1346e-02,\n",
       "          -8.3781e-01,  1.3912e-01, -2.7175e-01,  1.1582e-03,  1.7560e-01,\n",
       "           2.4348e-01, -7.4361e-01,  2.2386e-01, -4.8452e-01,  2.7767e-02,\n",
       "           6.6164e-01,  5.2310e-02,  3.8033e-01, -2.4301e-01, -1.5607e-01,\n",
       "           2.0409e-01, -8.6334e-02,  2.8390e-01, -3.5716e-01, -3.8284e-01,\n",
       "          -1.9444e-01,  6.1009e-01,  3.7326e-01, -1.8706e-01,  2.6314e-01,\n",
       "           9.4880e-02, -2.5121e-01,  1.6192e+00,  7.9465e-02, -8.4777e-03,\n",
       "           1.8613e-01,  2.9204e-01, -4.3477e-01, -7.4140e-01,  1.5682e-01,\n",
       "          -1.4927e-01, -8.4614e-03,  1.8699e-01,  3.9803e-01, -1.2880e-01,\n",
       "          -6.7369e-01, -1.5170e-01,  2.0668e-01,  4.3763e-02,  5.5442e-01,\n",
       "          -7.2413e-01, -1.9327e-01, -2.1508e-01,  2.6601e-01, -1.0039e+00,\n",
       "           3.4616e-02, -6.7629e-01, -4.5550e-01,  5.7226e-01, -1.8857e-01,\n",
       "           1.3537e-01, -1.8299e-01,  9.0334e-02,  6.8122e-01, -1.5635e-01,\n",
       "           1.2207e-01, -4.6982e-02,  1.9227e-01, -2.8020e-01, -5.9959e-02,\n",
       "           3.9894e-01,  3.5282e-01, -4.0664e-01, -6.4802e-01, -5.4346e-02,\n",
       "           2.7406e-01, -5.4682e-02,  5.7432e-01, -6.0554e-01,  1.0472e-01,\n",
       "          -7.6609e-01,  1.7642e-03,  1.4169e-01, -5.0839e-02,  3.7317e-01,\n",
       "           7.9992e-02, -6.6711e-02,  2.9936e-01, -7.3280e-01, -4.3822e-01,\n",
       "          -6.8158e-01, -3.4103e-01,  2.5181e-01, -6.5217e-01, -1.9971e-01,\n",
       "          -1.9098e-01, -7.6230e-02, -3.4876e-02, -7.4858e-03, -4.6451e-02,\n",
       "           2.9607e-01,  1.2213e-01, -1.9058e-01, -1.0918e-01, -6.7776e-02,\n",
       "          -6.7437e-01, -7.0749e-02, -4.5643e-02, -2.1728e-02, -3.0115e-01,\n",
       "          -1.3757e-01, -3.1506e-03, -4.1711e-01, -2.3814e-01,  1.8353e-01,\n",
       "          -6.8982e-02, -6.8373e-02, -5.3997e-01,  8.1370e-01,  3.8834e-01,\n",
       "           2.7003e-02, -1.3485e-01, -3.8264e-01, -1.8808e-01,  2.4116e-01,\n",
       "           2.8589e-01, -9.1656e-02,  2.1301e-02, -3.9404e-01, -5.7616e-01,\n",
       "           8.4876e-02,  7.9514e-01, -1.3994e-01, -7.3410e-01,  1.6504e-01,\n",
       "           3.3562e-01, -2.2904e-01, -2.1350e-01,  1.3330e-01,  5.1994e-02,\n",
       "          -1.8567e-01, -5.3589e-01,  5.9451e-01, -1.2079e-01, -5.7221e-03,\n",
       "           1.9094e-01, -3.8488e-02, -1.5186e-02,  5.7142e-01,  2.3102e-01,\n",
       "          -3.6051e-01, -1.3530e-02, -6.3962e-01,  3.2304e-01, -1.0206e-01,\n",
       "          -3.2911e-01,  8.8233e-01,  1.4349e-01, -3.9878e-02, -5.8033e-02,\n",
       "           2.8933e-01,  7.6854e-01, -6.1313e-02,  1.5558e-01,  5.2002e-01,\n",
       "           1.4286e-01,  3.0896e-01,  2.1588e-01,  2.6079e-01,  9.8689e-02,\n",
       "           1.8055e-01, -2.0737e-01, -2.8836e-01, -1.5873e-02,  3.2246e-01,\n",
       "           3.3397e-01, -5.1201e-01, -6.6558e-01,  1.1189e-01,  5.2264e-01,\n",
       "          -2.3232e-02,  2.9175e-01, -2.6605e-01,  3.8845e-01,  4.5679e-01,\n",
       "          -8.3972e-01, -1.5682e-01, -1.2207e-02,  1.2224e-01, -1.0263e-01,\n",
       "          -3.7550e-01,  1.2262e-01, -2.1591e-01, -1.0958e-01,  6.7103e-01,\n",
       "           4.5117e-01, -7.5916e-01,  2.0372e-02,  7.2381e-01, -4.4436e-01,\n",
       "           3.1189e-01, -4.7181e-01,  1.3724e-01,  3.9517e-01,  2.6571e-01,\n",
       "           1.0533e-01,  1.8197e-01, -6.3467e-01, -4.1184e-01,  1.5987e-01,\n",
       "           9.4693e-01, -1.0792e-01,  1.2216e-01, -3.7231e-01, -2.3794e-01,\n",
       "          -1.1578e-01, -3.7518e-01, -3.3161e-01,  2.0421e-01, -7.0193e-02,\n",
       "          -2.3596e-04, -4.7337e-02, -1.7414e-01,  2.4538e-01,  7.4256e-02,\n",
       "           1.4345e-01,  6.9872e-01,  5.6507e-01, -2.2021e-02, -1.6569e-01,\n",
       "          -2.6327e-01,  9.2934e-02, -1.3871e-01,  3.6364e-01, -3.2259e-01,\n",
       "          -2.4633e-01, -6.5084e-02,  2.7859e-01, -6.3065e-01,  1.4644e-01,\n",
       "          -5.8094e-01,  3.1744e-01, -1.5674e-01, -1.6243e-02,  2.5572e-01,\n",
       "           1.1499e-01,  1.5172e-01, -3.1603e-01,  5.0661e-01,  4.5451e-01,\n",
       "           1.7379e-01,  3.8553e-01,  5.8656e-02,  9.9802e-02, -3.1322e-01,\n",
       "           1.0911e-01, -9.5036e-01, -1.3614e-01, -2.3175e-01,  7.2286e-02,\n",
       "          -2.7482e-01, -5.7624e-02,  4.9664e-01,  3.4990e-01,  8.4452e-01,\n",
       "           6.1083e-01, -1.3134e-01, -5.5092e-01, -1.1634e-01,  1.5570e-01,\n",
       "           3.0229e-01, -1.8740e-01,  4.1225e-01,  3.7789e-02,  1.7598e-01,\n",
       "          -4.7409e-01, -4.6931e-01, -1.6163e-02,  3.9329e-01,  4.6267e-03,\n",
       "           7.3195e-01, -1.6405e-01, -2.5740e-01,  3.0336e-02,  1.4914e-01,\n",
       "          -5.6990e-03,  9.9034e-01,  2.8344e-01, -5.6885e-01,  1.7748e-01,\n",
       "          -1.2519e-01, -2.1606e-01,  2.7562e-01, -4.1079e-01,  2.2364e-01,\n",
       "           8.3334e-02,  6.8286e-01,  7.2325e-01]], device='mps:0'),\n",
       " tensor([[-2.5187e-01, -4.0054e-01, -5.8007e-01, -8.3055e-01, -1.2868e-01,\n",
       "          -2.8804e-02, -7.1917e-01,  6.6701e-01,  1.2709e-01,  5.0855e-01,\n",
       "           1.2570e-01,  5.1916e-01,  5.1676e-02, -1.1521e-01, -1.6481e-01,\n",
       "          -4.9820e-01,  5.6222e-02,  3.7338e-01,  6.7187e-01, -2.4111e-02,\n",
       "           2.2844e-01,  5.7979e-01, -2.6323e-01, -1.2386e-02, -8.3013e-01,\n",
       "           2.9613e-01, -1.1576e-01,  2.2409e-01,  1.7718e-01, -6.3092e-01,\n",
       "           4.0999e-02,  4.7417e-01,  2.3068e-02,  1.3497e+00, -5.6080e-01,\n",
       "          -1.3230e-01,  8.2213e-02, -3.9453e-01, -2.3010e-01,  3.7283e-02,\n",
       "           6.1164e-01, -4.1315e-01,  1.3635e-01,  4.0254e-01, -6.1085e-02,\n",
       "           8.9076e-02,  1.3899e-01, -3.7744e-01,  2.5892e-01, -8.3961e-01,\n",
       "          -4.8318e-02, -2.0669e-01,  4.9144e-01,  6.2615e-01, -6.3176e-01,\n",
       "          -2.2975e-01,  2.7419e-01, -3.7716e-02, -3.3020e-02,  1.5349e-01,\n",
       "           9.7980e-02,  3.8582e-01, -2.9995e-01,  2.3257e-01,  3.3441e-01,\n",
       "           2.4458e-01, -3.0392e-01, -4.8889e-01, -5.4078e-01, -5.0111e-02,\n",
       "           9.1859e-03,  2.2490e-02,  1.4846e-01, -2.3272e-01,  4.1973e-01,\n",
       "          -7.1214e-01, -6.4354e-02, -8.9125e-02, -2.5973e-01,  2.9246e-01,\n",
       "          -4.5549e-02,  1.9719e-01,  3.1010e-01,  1.0268e-01,  1.2457e-01,\n",
       "           4.3419e-02,  2.7215e-01,  5.2932e-04,  4.7373e-02,  3.7533e-01,\n",
       "          -1.3375e-01, -8.1722e-02, -2.8313e-01, -3.5814e-01,  3.1229e-01,\n",
       "           4.0598e-02,  2.9907e-01, -1.9557e-01,  1.8579e-01, -3.2474e-01,\n",
       "           1.5588e-01,  1.7768e-01, -1.8083e-01,  5.7286e-02, -3.5172e-01,\n",
       "          -4.5475e-01,  2.3589e-02, -2.8569e-01, -4.8285e-01,  2.2697e-01,\n",
       "          -1.4211e-01, -3.0205e-01, -4.0922e-02, -3.8935e-01, -3.1524e-02,\n",
       "          -7.5164e-02,  2.6013e-01, -1.3230e-01, -3.4806e-01, -8.3033e-02,\n",
       "           2.0734e-01, -2.2808e-01,  1.6077e-01,  2.6882e-01,  1.2232e-01,\n",
       "          -9.6705e-02,  6.5251e-01,  1.8950e-02,  3.1674e-01,  2.5763e-01,\n",
       "           1.1699e-01, -6.5768e-02, -1.9745e-01, -2.3154e-01,  7.5988e-02,\n",
       "          -7.6941e-02,  1.0687e-01, -1.8182e-01, -2.2790e-01, -1.8508e-01,\n",
       "           3.1771e-01, -3.3097e-01, -6.0893e-01, -4.9260e-01, -1.6098e-01,\n",
       "          -1.6326e-01, -2.4804e-02, -8.2473e-03, -3.3693e-01, -3.1956e-01,\n",
       "          -6.7443e-01, -3.6717e-01,  8.1041e-01, -1.6831e-02, -3.3356e-01,\n",
       "          -1.6499e-01,  2.3208e-01,  3.1666e-01, -4.5959e-02,  3.7161e-01,\n",
       "           7.7250e-01, -4.7670e-02,  1.6786e-01, -5.1827e-01, -3.8195e-01,\n",
       "           4.8335e-02,  2.6658e-01, -2.7054e-01,  6.1879e-01,  4.8352e-02,\n",
       "           7.5627e-02, -6.6170e-02,  2.7430e-01, -3.2704e-01, -1.2324e-02,\n",
       "          -5.7573e-01,  3.7750e-01,  2.9539e-01, -3.9892e-01, -6.6567e-02,\n",
       "          -2.0834e-02, -1.2060e+00,  9.6964e-02, -8.3040e-01,  3.1236e-01,\n",
       "          -2.3110e-01,  1.7963e-01,  2.3389e-01,  2.9545e-01, -6.5320e-02,\n",
       "           2.5404e-01,  2.8725e-01,  2.5687e-01,  3.3609e-01, -3.4924e-02,\n",
       "           2.8019e-01,  2.3924e-01,  5.0790e-01,  4.1073e-01,  1.4769e-01,\n",
       "          -3.5440e-01,  2.9683e-01,  3.5781e-01,  9.0598e-02,  1.6954e-01,\n",
       "           9.9825e-02, -2.1636e-01,  2.1091e-01,  1.7501e-01, -1.6618e-01,\n",
       "          -2.2406e-01, -3.1100e-02, -4.6818e-01, -5.6655e-02, -4.8467e-01,\n",
       "           3.2557e-01, -5.1983e-01, -1.1912e-01,  3.4831e-01,  2.4340e-01,\n",
       "          -1.4896e-02,  1.8319e-01,  1.7196e-01, -2.4338e-01,  2.1735e-01,\n",
       "           3.7449e-01, -3.4618e-01, -3.9804e-02, -3.2197e-01, -2.9367e-01,\n",
       "           2.4357e-01, -5.6244e-01,  6.6635e-01,  4.0574e-01,  6.0165e-02,\n",
       "           1.5119e-01, -2.7920e-01, -5.1638e-02, -5.7408e-01,  3.5471e-01,\n",
       "          -3.2130e-01, -3.4955e-01, -5.4801e-01,  6.4487e-02,  4.2773e-01,\n",
       "           2.7565e-01, -6.4915e-02,  7.7868e-01,  4.2408e-01,  4.2813e-01,\n",
       "          -4.3003e-01, -1.5896e-02,  4.0377e-01, -4.1405e-02, -4.6059e-01,\n",
       "           6.7116e-02, -1.4962e-01, -5.1643e-01,  2.8000e-01, -2.6598e-01,\n",
       "          -1.2047e-01, -1.4913e-01,  3.3682e-01,  1.5365e-01, -2.1964e-01,\n",
       "          -3.8461e-01,  3.9587e-01, -1.4349e-01, -4.1723e-01,  4.7123e-01,\n",
       "          -3.5025e-01,  1.5209e-01,  1.0694e-01, -3.8441e-01, -6.2072e-01,\n",
       "          -3.2235e-01,  2.3090e-01,  1.5585e-01,  3.4799e-02, -6.6306e-01,\n",
       "          -4.0571e-01,  3.1197e-01,  2.4601e-01,  2.9976e-01,  2.9840e-01,\n",
       "           5.7427e-01, -3.0944e-01, -1.6702e-01,  1.4982e-01, -4.1377e-01,\n",
       "          -1.5704e-01, -3.3228e-02, -1.9995e-01,  2.9569e-01,  2.4768e-01,\n",
       "           1.1600e-01,  2.0367e-01,  4.3482e-01, -2.5272e-01, -3.8673e-01,\n",
       "          -7.1543e-02,  2.0161e-01, -9.3629e-02, -5.8306e-02,  2.3951e-01,\n",
       "          -6.3095e-02,  5.9650e-01,  7.2143e-01,  7.7439e-02, -6.4006e-01,\n",
       "           1.7696e-01,  5.2502e-01, -2.7253e-01, -4.0343e-02,  9.6365e-02,\n",
       "          -1.0344e+00, -2.1331e-01, -1.2883e-01, -2.0181e-01, -6.5352e-01,\n",
       "           8.8115e-02,  3.4863e-01,  4.6833e-03, -1.7864e-01, -3.0947e-01,\n",
       "           1.0448e-01,  4.8724e-02, -4.8980e-02,  4.3534e-01, -2.8901e-01,\n",
       "          -3.4797e-01, -2.3438e-01, -2.1532e-01, -5.4201e-01, -3.9968e-01,\n",
       "           3.5914e-01,  2.8129e-01,  1.1986e-01, -2.6486e-01, -5.1832e-01,\n",
       "           4.5577e-01, -6.8169e-01, -1.4632e-01,  2.4840e-01, -4.7702e-01,\n",
       "           1.4650e-02,  4.8990e-01,  1.4999e-01,  1.6336e-01,  6.5258e-02,\n",
       "           7.8557e-02, -3.3903e-01, -5.0030e-01, -1.5000e-01,  6.2323e-01,\n",
       "           3.2283e-01,  3.3506e-01, -1.7236e-01,  3.9096e-01,  4.5167e-01,\n",
       "          -2.9582e-01,  4.1670e-01,  1.0930e-02,  7.7439e-02, -6.0414e-02,\n",
       "           1.9048e-01,  3.4370e-01, -3.6842e-01,  2.0145e-01,  1.7929e-01,\n",
       "          -3.1634e-01, -3.2149e-01,  1.5691e-02,  2.0642e-01, -1.8580e-01,\n",
       "          -3.0875e-01, -2.0361e-01,  9.9698e-02, -3.4856e-01, -4.3104e-01,\n",
       "           2.9161e-01, -3.6543e-02,  7.0495e-02, -4.0357e-01, -4.3462e-01,\n",
       "          -2.0181e-01, -2.6055e-01,  6.8681e-01,  2.1064e-01, -1.9621e-01,\n",
       "           1.8720e-01, -1.0907e-01, -3.6330e-01,  4.7847e-01,  2.0162e-02,\n",
       "          -4.8754e-01,  4.5482e-01, -3.4822e-01,  1.0627e-01,  5.5756e-01,\n",
       "          -1.6769e-01, -1.0840e-01,  1.1008e-01,  1.0820e-01,  2.9026e-01,\n",
       "          -3.9884e-01,  1.5516e-01, -6.3476e-02, -3.1885e-01,  3.1820e-01,\n",
       "           2.6411e-01,  2.2781e-01, -1.7137e-01, -3.3426e-02,  3.5677e-02,\n",
       "           2.8782e-01, -6.0853e-02, -3.9888e-01, -4.7347e-01, -4.6132e-01,\n",
       "           1.2832e-01,  1.8927e-01,  5.6207e-01,  3.4144e-01, -1.1394e+01,\n",
       "           1.2938e-01,  7.8031e-01, -4.0713e-01, -8.2819e-02, -5.0320e-02,\n",
       "           1.0976e-02, -1.3708e-01,  4.2955e-01, -2.7734e-01, -4.4105e-01,\n",
       "          -3.5727e-01, -1.3308e-01,  1.5454e-01, -3.2189e-01, -2.6945e-01,\n",
       "          -1.4686e-02, -7.0935e-01,  4.0963e-01, -3.0286e-01,  5.0298e-01,\n",
       "          -2.3511e-01, -8.3798e-02,  1.5816e-01, -3.5851e-01, -1.3439e-02,\n",
       "           5.9766e-02, -1.6033e-01,  1.4651e-01,  4.2486e-01, -5.2072e-02,\n",
       "           3.3244e-02, -4.1258e-03,  4.8461e-02, -2.5723e-01, -5.9999e-01,\n",
       "           1.9296e-02, -1.4141e-01,  3.3422e-01, -4.4133e-02, -7.1265e-01,\n",
       "          -2.2533e-01, -2.7418e-01,  3.1407e-01,  2.7021e-01,  9.0598e-01,\n",
       "           2.2955e-01,  1.0686e+00,  1.5022e-01,  1.7748e-01,  2.6230e-02,\n",
       "          -7.8764e-01, -3.3269e-02, -4.1907e-01,  2.3280e-02,  1.0114e-01,\n",
       "           3.6576e-01, -7.9367e-01,  2.5251e-01, -3.5521e-01,  3.6487e-02,\n",
       "           4.9094e-01,  3.1768e-02,  2.7700e-01, -3.2298e-02, -7.8123e-02,\n",
       "           1.2089e-01, -2.8544e-02,  1.1098e-01, -2.2583e-01, -2.4084e-01,\n",
       "          -1.0177e-01,  5.9888e-01,  3.9592e-01, -7.2211e-01,  3.6112e-01,\n",
       "           4.9258e-02, -1.8622e-01,  1.4556e+00, -4.8573e-02,  1.8361e-02,\n",
       "           1.9800e-03,  2.9604e-01, -5.0236e-01, -7.3516e-01,  3.6511e-01,\n",
       "          -1.0211e-01,  5.6427e-02,  1.0127e-01,  4.8148e-01, -9.7481e-02,\n",
       "          -4.7940e-01, -1.8593e-03,  1.0556e-01, -1.3380e-01,  5.4710e-01,\n",
       "          -7.2824e-01, -1.6963e-01, -1.1223e-01,  3.1723e-01, -8.4054e-01,\n",
       "          -6.1126e-02, -7.0210e-01, -3.9328e-01,  5.0586e-01, -2.5896e-01,\n",
       "           1.5852e-01, -2.7443e-02, -4.3439e-02,  5.5574e-01, -2.1489e-01,\n",
       "           7.1555e-02,  2.2813e-01,  8.4995e-02, -1.6733e-01,  3.9349e-02,\n",
       "           4.1630e-01,  4.9571e-01, -4.2740e-01, -6.3911e-01, -7.6901e-03,\n",
       "           1.7180e-01,  1.4933e-01,  3.4019e-01, -5.3463e-01,  3.6431e-01,\n",
       "          -7.9162e-01, -1.8769e-01,  7.2949e-02, -4.3273e-02,  3.6175e-01,\n",
       "           2.3925e-01,  7.2038e-02,  2.2256e-01, -7.7756e-01, -6.4498e-01,\n",
       "          -6.5932e-01, -2.1445e-01,  2.6143e-01, -5.4236e-01, -8.3654e-02,\n",
       "          -1.1033e-01, -4.8837e-03, -5.0292e-02, -6.6102e-02,  1.1939e-02,\n",
       "           1.3632e-01,  2.3587e-01, -1.3023e-01, -1.3178e-01, -9.2464e-02,\n",
       "          -5.5566e-01, -1.1753e-01, -9.9591e-02, -1.2505e-01, -5.6783e-01,\n",
       "          -1.3797e-01,  3.3075e-02, -2.5373e-01, -6.8124e-02,  3.5608e-01,\n",
       "          -2.9832e-02, -2.7856e-02, -4.8850e-01,  7.2846e-01,  4.0279e-01,\n",
       "           5.2088e-02, -2.8298e-01, -5.5189e-01, -1.0567e-01,  2.9592e-01,\n",
       "           1.9555e-01, -6.7007e-02, -1.7709e-01, -2.4145e-01, -4.2522e-01,\n",
       "           1.4302e-01,  7.1147e-01, -6.6296e-02, -6.8115e-01,  2.6621e-01,\n",
       "           5.8223e-01, -1.2882e-01, -2.4432e-01,  2.0983e-02, -9.3284e-02,\n",
       "          -3.1851e-01, -6.0317e-01,  7.8821e-01, -1.1246e-01, -5.8638e-02,\n",
       "           2.2427e-01,  2.0359e-01, -9.2588e-03,  4.0209e-01, -1.5770e-01,\n",
       "          -3.5434e-01, -1.6477e-03, -7.3600e-01,  4.3257e-01,  6.1666e-02,\n",
       "          -3.1489e-01,  7.6082e-01,  2.7260e-01, -6.0593e-02, -4.5779e-02,\n",
       "           2.2863e-01,  7.9941e-01, -2.4636e-01,  6.4653e-02,  5.3779e-01,\n",
       "           8.8110e-02, -1.1229e-01,  3.9580e-02,  4.1665e-02,  4.5410e-02,\n",
       "           3.7012e-01, -3.2145e-02, -4.1564e-01, -6.1194e-02,  3.8924e-01,\n",
       "           3.9295e-01, -5.4610e-01, -7.3487e-01,  2.2333e-02,  6.8551e-01,\n",
       "          -2.9054e-01,  2.3654e-01, -3.2708e-02,  3.0880e-01,  1.2001e-01,\n",
       "          -9.2160e-01, -1.5016e-01, -1.0056e-01,  2.8229e-01, -2.5482e-02,\n",
       "          -1.6139e-01, -1.4727e-01, -2.3686e-01, -1.6208e-01,  6.8308e-01,\n",
       "           2.1991e-01, -6.9160e-01,  3.4058e-02,  5.7281e-01, -6.0551e-01,\n",
       "           4.6052e-01, -2.4577e-01, -3.4302e-02,  3.6435e-01,  1.3117e-01,\n",
       "           4.9067e-02,  1.7856e-01, -4.2138e-01, -4.0598e-01,  1.7824e-01,\n",
       "           9.1672e-01, -7.6613e-02,  1.1166e-01, -3.9384e-01, -1.7182e-01,\n",
       "          -7.1855e-02, -3.1913e-01, -3.1286e-01,  3.7185e-01,  2.0529e-01,\n",
       "          -3.2552e-02, -1.3271e-01, -2.2992e-01,  3.0816e-01,  1.1821e-01,\n",
       "           3.2536e-01,  5.8187e-01,  6.4368e-01, -1.0077e-02, -2.4936e-01,\n",
       "          -6.1235e-02,  1.4568e-01, -1.9524e-01,  5.2704e-01, -2.0627e-01,\n",
       "          -3.3453e-01, -3.8069e-03,  3.0134e-01, -6.1681e-01, -5.8725e-02,\n",
       "          -4.5426e-01,  4.2361e-01, -1.6955e-01,  3.7868e-02,  2.8833e-01,\n",
       "           4.0522e-02,  5.7925e-02, -2.2651e-01,  5.5807e-01,  4.4511e-01,\n",
       "           1.6609e-01,  4.5662e-01,  1.6104e-01,  4.5272e-02, -5.1597e-01,\n",
       "           2.1978e-01, -1.0213e+00, -9.4260e-02, -4.6640e-02, -1.5401e-01,\n",
       "          -4.4224e-01, -2.8126e-02,  1.6814e-01,  1.8538e-01,  6.9597e-01,\n",
       "           7.2967e-01, -3.0871e-01, -4.5290e-01, -1.8510e-01,  1.5304e-01,\n",
       "           2.7705e-01, -1.9517e-01,  3.5971e-01,  9.6103e-02,  2.4432e-01,\n",
       "          -5.0967e-01, -4.3165e-01, -4.2252e-02,  3.1411e-01, -2.0171e-01,\n",
       "           5.7118e-01, -2.0164e-01, -1.3863e-01,  9.0832e-02,  1.3770e-01,\n",
       "           1.6639e-01,  7.9664e-01,  4.3416e-01, -6.3425e-01,  8.7469e-02,\n",
       "          -1.3980e-02, -2.9988e-01,  2.5983e-02, -4.3577e-01,  2.8858e-01,\n",
       "           1.2984e-01,  4.9280e-01,  6.7504e-01]], device='mps:0'),\n",
       " tensor([[-1.8332e-01, -7.2438e-01, -8.7525e-01, -2.5644e-01, -2.5149e-01,\n",
       "          -1.2527e-01, -6.5986e-01,  3.6405e-01,  2.3049e-01,  4.2381e-01,\n",
       "           3.8290e-01,  7.8719e-02,  5.4802e-02, -2.8136e-01, -2.2979e-01,\n",
       "          -5.7622e-01, -1.7107e-01,  4.9884e-01,  3.2726e-01, -4.6910e-03,\n",
       "           8.8036e-01,  5.7748e-01, -2.3380e-01,  1.6707e-01, -3.8255e-01,\n",
       "           1.8482e-01, -6.7564e-02,  3.7116e-01,  2.8865e-01, -4.4743e-01,\n",
       "           1.7430e-03,  3.8959e-01, -9.4508e-02,  9.7863e-01, -7.5988e-02,\n",
       "          -5.7700e-02,  1.4284e-01, -4.0241e-01,  4.7999e-01,  5.5063e-03,\n",
       "           6.8700e-01, -4.7432e-01,  1.6722e-01,  2.9559e-01,  3.2660e-01,\n",
       "           3.4560e-01,  4.3420e-01, -8.8774e-01,  5.1386e-01, -7.1964e-01,\n",
       "          -1.4476e-02, -2.7074e-02,  1.1187e-01,  6.2514e-01, -7.2580e-01,\n",
       "          -3.8295e-01,  2.1666e-01,  1.9125e-01, -3.9105e-01, -2.1341e-02,\n",
       "          -2.6666e-01,  3.2058e-01,  1.8470e-01,  1.4999e-01, -7.9156e-02,\n",
       "           8.4236e-02, -6.9837e-01, -5.9362e-01, -5.1399e-01, -3.7028e-02,\n",
       "          -1.1436e-01,  3.3685e-01, -4.5435e-02, -2.0401e-01,  2.9615e-01,\n",
       "          -9.9932e-02,  2.2851e-01, -2.4196e-01, -6.7510e-01,  1.3662e-02,\n",
       "          -2.2797e-02,  2.4326e-01,  3.9243e-02, -8.5249e-02, -1.3411e-01,\n",
       "          -1.1328e-01,  4.7107e-01,  3.2481e-01,  7.9724e-02, -5.8508e-02,\n",
       "           3.5639e-01,  3.1304e-01, -1.4553e-01, -5.5810e-01,  3.5265e-01,\n",
       "          -1.6931e-02,  5.3550e-01, -2.4219e-01, -3.6003e-01, -3.5663e-01,\n",
       "           1.1511e-01,  1.2240e-01,  2.4277e-01,  2.4884e-01, -5.2827e-01,\n",
       "          -4.8009e-01,  2.7202e-01, -1.4471e-01, -6.1234e-01, -7.5459e-02,\n",
       "          -2.5642e-01, -2.4830e-01,  2.9544e-01, -3.6353e-01, -4.4267e-01,\n",
       "          -2.7090e-01,  2.0336e-01,  7.7228e-02, -3.4075e-01,  1.4327e-03,\n",
       "           5.7242e-02, -4.3556e-01,  1.9931e-01, -8.0626e-02,  3.3423e-01,\n",
       "          -2.3213e-01,  4.1390e-01, -1.0970e-01,  4.1768e-01,  1.5587e-01,\n",
       "          -7.4716e-02,  1.4742e-01, -2.8066e-01, -3.1899e-01,  1.2995e-01,\n",
       "          -2.9082e-02,  4.7610e-01, -4.5572e-01, -1.1686e-01, -1.0082e-01,\n",
       "           2.2881e-01, -2.2490e-01, -2.4221e-01,  1.3001e-01,  2.2031e-01,\n",
       "           7.3697e-02,  1.6684e-01, -1.7537e-01, -7.3393e-01, -3.4434e-01,\n",
       "          -8.8694e-01, -5.5817e-01,  1.5397e-01, -2.1409e-01, -8.0406e-03,\n",
       "           4.5495e-02,  2.0095e-01,  1.6682e-01, -2.3001e-01,  1.3984e-01,\n",
       "           6.7154e-01, -4.2881e-02,  3.7780e-01, -7.4070e-01, -3.3200e-01,\n",
       "           1.2667e-01,  7.1895e-03, -2.6433e-01, -2.3303e-01,  7.1865e-02,\n",
       "           3.5388e-01, -4.4592e-01, -1.4785e-01, -1.9408e-01,  1.7217e-01,\n",
       "          -4.0676e-01,  6.4152e-01,  2.1730e-01, -1.0083e-01,  1.2704e-01,\n",
       "           2.1838e-01, -2.0000e+00,  2.8294e-01, -2.8886e-01,  1.6784e-01,\n",
       "          -5.4526e-01,  3.8811e-01,  2.3489e-01,  4.4722e-01, -2.1272e-01,\n",
       "          -9.4920e-02, -3.1138e-01, -1.1883e-01,  2.5742e-01, -5.8573e-03,\n",
       "           3.7469e-01,  6.6313e-01,  4.1108e-01,  6.3316e-02,  1.3313e-01,\n",
       "          -5.8583e-01, -3.5396e-02,  2.3830e-01, -8.6961e-02,  2.4312e-01,\n",
       "           2.6874e-01, -4.2812e-01,  1.0149e-02,  2.9867e-01,  1.8795e-01,\n",
       "          -2.0775e-01, -1.3160e-02, -4.8778e-01, -2.4187e-02,  3.3788e-02,\n",
       "           6.4865e-01, -1.0118e-01,  4.5710e-02,  3.3207e-01,  7.5426e-02,\n",
       "          -3.6287e-01, -2.1074e-01,  4.0609e-02, -3.8136e-01, -3.8896e-02,\n",
       "           3.6470e-01, -2.2800e-01, -4.4522e-02, -1.0183e-01, -1.8044e-01,\n",
       "           2.4037e-01, -6.5717e-01,  4.4408e-01,  1.6536e-01, -1.6480e-01,\n",
       "          -5.1437e-01, -2.6095e-01,  1.2952e-01, -4.4660e-01,  3.9390e-02,\n",
       "          -6.4723e-01, -5.3995e-01, -7.9655e-01,  1.0909e-01, -1.6119e-01,\n",
       "           8.9640e-02, -1.2742e-01,  7.9211e-01,  6.0049e-01,  4.2956e-01,\n",
       "          -1.8547e-01, -2.6278e-03,  5.4752e-01, -3.0130e-01, -4.5609e-01,\n",
       "           1.8478e-01, -9.9347e-02,  1.9418e-02,  3.3722e-01, -2.1890e-01,\n",
       "          -3.5537e-01,  1.0883e-01,  2.9167e-01,  5.8885e-01,  1.8813e-01,\n",
       "           2.5401e-01, -2.0887e-01, -3.2936e-01, -1.5263e-01,  7.5104e-02,\n",
       "          -3.1025e-01,  1.5256e-01,  1.2107e-01, -4.3278e-01, -6.1611e-01,\n",
       "          -6.8927e-01,  3.3710e-01, -3.0987e-01,  2.5024e-01, -1.0265e+00,\n",
       "          -5.6543e-01,  4.2943e-01,  1.8202e-01, -5.4662e-02, -2.0747e-01,\n",
       "           2.1418e-01,  3.1981e-01,  6.7559e-02, -2.2816e-01, -6.3556e-01,\n",
       "          -3.0442e-03,  7.1619e-02, -5.3167e-01,  2.2049e-01,  2.5164e-01,\n",
       "           1.4453e-01,  3.9930e-01,  1.3431e-01,  1.6175e-01, -2.0358e-01,\n",
       "          -2.4859e-01,  1.6120e-01, -2.8840e-01, -6.2211e-02,  4.2187e-01,\n",
       "           1.8892e-01,  1.7930e-01,  6.8368e-01, -3.4660e-01, -4.6559e-01,\n",
       "          -1.1441e-01,  9.2723e-01, -1.3208e-02, -3.5383e-01, -2.4646e-01,\n",
       "          -1.3597e+00, -5.2525e-01, -4.6231e-01, -9.3234e-02, -2.8637e-01,\n",
       "           2.4446e-01,  1.4151e-01, -1.9918e-02, -1.1167e-01, -2.9387e-01,\n",
       "           4.5517e-01, -1.5686e-01,  4.9929e-01,  3.2594e-01,  2.1136e-01,\n",
       "          -4.1029e-01, -2.0823e-02, -1.2996e-01, -1.7482e-01, -1.2486e-01,\n",
       "           1.0927e-01,  2.2212e-01, -2.1518e-01,  1.5935e-01, -2.7294e-01,\n",
       "           5.6755e-01, -6.4543e-01, -5.7783e-01,  3.8596e-01, -2.6960e-01,\n",
       "          -1.1790e-01,  1.3982e-01, -3.0007e-01,  5.9478e-02,  4.5598e-02,\n",
       "          -1.2186e-02, -4.7714e-01, -6.6287e-02, -3.9943e-01,  3.6997e-01,\n",
       "           3.8057e-01,  4.5823e-01, -6.5774e-01,  3.0753e-01,  5.7713e-01,\n",
       "          -2.0986e-01, -2.9089e-05, -1.5052e-01, -1.8446e-01, -1.3540e-02,\n",
       "           1.9492e-01,  4.2102e-01, -2.9737e-01,  2.5434e-02, -3.9104e-02,\n",
       "           1.8675e-01, -4.2155e-01, -1.2577e-01,  1.4998e-01, -1.2105e-01,\n",
       "          -2.6093e-01, -1.2665e-01,  3.3444e-01,  6.9633e-02, -3.4834e-01,\n",
       "           1.9213e-01, -1.0111e-01, -5.3438e-02, -2.0106e-01, -5.8994e-01,\n",
       "          -1.6697e-01,  2.5821e-01,  7.8488e-01, -6.6566e-02,  1.7711e-01,\n",
       "           2.4406e-02,  8.6003e-02, -4.0306e-01,  3.7620e-01,  2.0656e-01,\n",
       "          -3.4339e-01,  2.9986e-01, -3.3380e-01,  5.8752e-01,  1.6874e-02,\n",
       "          -2.3149e-01, -3.8828e-01, -7.5848e-02,  5.5638e-01, -4.8195e-02,\n",
       "          -3.3489e-01,  3.6324e-01,  1.3722e-01,  5.9183e-02, -1.3275e-01,\n",
       "          -1.1957e-01,  5.7160e-01, -2.4397e-01,  2.0537e-01,  1.3177e-01,\n",
       "           3.2875e-01,  2.7354e-01,  2.5113e-01, -1.7291e-01, -4.3471e-01,\n",
       "           3.3734e-01,  5.9419e-01,  3.2455e-01,  1.8283e-01, -1.1262e+01,\n",
       "           3.3799e-01,  5.6984e-01, -5.0219e-01, -2.4939e-01, -1.8082e-01,\n",
       "           2.6280e-01, -1.4204e-01, -7.1985e-02, -2.2231e-01, -1.8224e-01,\n",
       "           1.4612e-01, -8.0753e-02,  2.0127e-01, -1.8275e-01,  3.7482e-01,\n",
       "          -3.6146e-01, -5.7213e-01,  2.9674e-01, -1.9440e-01,  5.8827e-01,\n",
       "          -2.9932e-01,  8.7200e-02,  1.6422e-01, -2.4964e-01, -5.1988e-01,\n",
       "          -3.2428e-01, -5.0724e-01,  1.0438e-01,  2.4762e-01, -1.4827e-01,\n",
       "           9.6583e-02,  1.1348e-01,  2.3969e-01, -5.1005e-01, -5.9942e-01,\n",
       "           5.2909e-02,  1.4968e-02,  4.4089e-01,  5.6792e-02, -5.0891e-01,\n",
       "           4.0910e-01, -2.9556e-01,  7.0119e-01,  6.8304e-02,  5.2597e-01,\n",
       "           5.5916e-01,  7.0934e-01, -1.9516e-01, -3.9743e-01,  1.0970e-01,\n",
       "          -9.4502e-01,  4.3058e-01, -8.2898e-01,  4.5020e-01,  4.2705e-02,\n",
       "           7.6496e-01, -3.7083e-01, -2.7284e-01, -2.5445e-01, -2.3356e-01,\n",
       "          -1.2144e-02, -2.7867e-01,  3.2251e-01, -5.0032e-01, -2.7192e-01,\n",
       "           3.0919e-01, -2.5615e-01,  3.8508e-01,  1.1110e-01, -2.1137e-01,\n",
       "          -6.1234e-02,  7.0210e-01, -1.5456e-01, -3.6549e-01,  2.6180e-01,\n",
       "           1.4712e-01, -1.8376e-02,  8.3926e-01,  7.1895e-02, -2.6972e-01,\n",
       "           2.5500e-01,  3.2243e-01, -1.5236e-01, -1.0408e+00, -5.1389e-01,\n",
       "          -1.7640e-01,  4.2322e-01, -2.2992e-01,  5.4269e-01, -3.2342e-01,\n",
       "          -2.1304e-01, -3.1732e-01,  1.7136e-02,  4.5663e-02,  8.0201e-01,\n",
       "          -6.8621e-01, -4.9653e-02,  4.4076e-02, -2.4919e-01, -6.2463e-01,\n",
       "          -8.3347e-02, -3.3138e-01, -5.0451e-01,  4.0932e-01,  1.5670e-01,\n",
       "          -1.0517e-01,  1.5200e-01,  5.0823e-01,  1.7908e-01, -4.0933e-01,\n",
       "           3.8597e-01, -1.3021e-01,  1.0531e-01, -1.6598e-01,  2.6723e-02,\n",
       "          -1.4589e-01,  6.7957e-01, -3.3056e-01, -2.6960e-01, -1.9234e-01,\n",
       "          -2.0489e-01,  1.3479e-01,  8.9132e-02, -6.3614e-01,  9.2379e-02,\n",
       "          -4.0752e-01, -8.4602e-02, -2.7118e-01,  2.2690e-02,  3.3843e-01,\n",
       "           4.6592e-02,  1.9533e-01, -1.4758e-02, -4.5796e-01, -6.7539e-01,\n",
       "          -5.4055e-01, -3.4621e-01,  1.3717e-01, -1.8460e-01, -5.4600e-02,\n",
       "          -1.8066e-01,  1.3747e-01, -1.4400e-01,  1.8668e-01, -5.7220e-02,\n",
       "           7.2531e-01,  3.7169e-01, -1.9779e-01,  2.7252e-02, -1.2525e-01,\n",
       "          -3.0619e-01,  1.5451e-02, -2.3260e-01, -1.9310e-01, -1.3549e-01,\n",
       "          -7.3330e-02,  2.1881e-02, -3.1155e-01,  2.9836e-02,  8.2558e-02,\n",
       "           1.8872e-01,  1.9881e-01, -1.5595e-01,  7.6407e-03,  1.2808e-01,\n",
       "           7.2090e-02, -4.0444e-01, -3.7342e-01, -1.0064e-01, -1.4194e-01,\n",
       "           2.3570e-01, -3.8599e-01, -2.9720e-02, -4.1641e-01,  1.7611e-02,\n",
       "           7.7914e-02,  1.1646e+00, -3.1025e-03, -4.3762e-01,  2.5491e-01,\n",
       "           6.7458e-01, -5.4860e-01,  1.0857e-02,  1.9788e-01,  1.5754e-01,\n",
       "           1.8101e-01, -2.1334e-01,  2.4919e-01, -1.7005e-01,  4.4857e-01,\n",
       "          -1.8341e-01,  2.3484e-02, -3.7814e-01,  3.2655e-02,  5.8057e-02,\n",
       "          -4.3370e-01, -2.5977e-01, -2.9185e-01, -5.6297e-02,  4.0633e-01,\n",
       "           1.6821e-01, -2.5156e-01, -1.7722e-01, -5.4817e-01,  3.0127e-01,\n",
       "           2.7813e-01,  1.0428e+00,  1.5348e-01,  2.1650e-01,  4.4582e-01,\n",
       "           5.1633e-01, -2.6262e-01,  5.9237e-01,  1.5909e-01,  2.6757e-01,\n",
       "           4.4979e-01,  3.6196e-01, -5.3503e-01,  2.6115e-01, -9.5359e-02,\n",
       "           4.3531e-01, -9.0298e-01, -5.4062e-01,  1.2545e-01,  4.9702e-01,\n",
       "          -1.5558e-01,  3.1162e-01, -3.9899e-01,  4.5694e-01,  7.8833e-01,\n",
       "          -9.7417e-01, -1.2414e-02, -6.3937e-02,  8.4419e-02, -1.9608e-01,\n",
       "          -3.8586e-01, -1.8465e-01, -1.8027e-02, -3.3592e-01,  3.4807e-01,\n",
       "           8.3323e-01, -3.4550e-01,  2.4038e-01,  2.6212e-01, -3.8956e-01,\n",
       "           3.5604e-02,  1.4271e-02, -1.1771e-01,  3.6776e-01, -4.6373e-02,\n",
       "          -1.0945e-01,  1.4753e-01, -2.5346e-01, -6.1653e-01,  2.7508e-01,\n",
       "           5.8519e-01, -1.2902e-01,  1.9299e-01,  1.1829e-01,  2.5737e-02,\n",
       "           8.2178e-02, -3.5963e-01, -1.1115e-01,  3.0495e-01,  1.4859e-01,\n",
       "           8.4016e-02, -1.4457e-01,  2.1217e-01,  3.2048e-01, -4.2547e-01,\n",
       "           5.1481e-01,  4.1607e-01,  4.8572e-01, -2.4968e-01, -2.4925e-01,\n",
       "           4.2485e-01,  2.4004e-01, -5.2260e-02,  6.1632e-01, -2.5387e-01,\n",
       "          -1.4369e-01,  1.1187e-01,  3.2568e-01, -4.2599e-01, -3.4498e-02,\n",
       "          -4.6300e-01,  2.3380e-01,  1.9683e-01, -3.2076e-01,  2.8881e-01,\n",
       "           2.9917e-01,  2.7347e-01, -3.2099e-01,  1.7177e-01,  5.1380e-01,\n",
       "          -1.1966e-01,  7.5201e-01,  6.5008e-03,  4.2565e-01,  7.5157e-02,\n",
       "          -1.8744e-03, -4.5695e-01, -3.9664e-01,  2.3967e-01,  3.0314e-01,\n",
       "          -3.7560e-01,  1.3550e-02,  2.1045e-01, -2.1683e-02,  7.7996e-01,\n",
       "           6.4882e-01,  3.3040e-02, -1.0021e+00, -3.5022e-01,  1.8220e-01,\n",
       "          -6.7047e-02, -1.2337e-01,  2.2226e-01,  2.9011e-01,  3.5022e-01,\n",
       "          -6.5618e-01, -2.8090e-01,  1.2028e-02, -9.9280e-02, -4.2958e-01,\n",
       "           6.3138e-01,  2.1441e-01,  5.5112e-01,  2.8615e-01,  1.3451e-01,\n",
       "           2.6572e-01,  7.8408e-01,  1.3794e-01, -5.8388e-01,  5.7469e-02,\n",
       "           5.4969e-01, -1.6234e-01, -1.4643e-01, -8.5972e-01,  2.3437e-01,\n",
       "           1.2633e-01,  8.3264e-01,  4.8552e-01]], device='mps:0'),\n",
       " tensor([[-1.7054e-01, -4.9387e-01, -5.1411e-01, -5.2485e-01, -1.1492e-01,\n",
       "          -3.0931e-01, -2.9657e-01,  6.0303e-01, -1.6415e-01,  8.7564e-01,\n",
       "           1.9912e-01,  2.3115e-01,  1.5256e-01,  5.0925e-01, -7.1364e-01,\n",
       "          -1.1486e-01, -2.3050e-01,  2.5049e-01,  6.6615e-01,  2.9257e-01,\n",
       "           7.1177e-01,  6.0786e-01,  3.0138e-02,  1.3639e-02, -3.8598e-01,\n",
       "           3.3203e-01,  2.2912e-01, -1.2879e-01,  2.1343e-01, -3.2683e-01,\n",
       "          -1.6979e-02,  2.6252e-01,  9.0765e-02,  1.1244e+00, -4.5286e-03,\n",
       "          -3.4734e-01,  5.4858e-01, -4.7631e-01, -1.9333e-01, -2.0515e-01,\n",
       "           4.1463e-01, -3.5166e-01,  4.2091e-01,  9.8138e-02,  2.9819e-01,\n",
       "           1.0746e-01, -2.2290e-01, -2.1062e-01,  4.2635e-01, -8.8741e-01,\n",
       "          -3.0651e-01, -1.3930e-01,  4.4423e-01,  4.4088e-01, -4.8751e-01,\n",
       "          -3.9498e-01,  4.6716e-01, -1.3957e-01, -7.4222e-02,  1.9950e-01,\n",
       "          -2.4606e-02,  2.3934e-01, -3.0492e-01,  1.8132e-01,  1.8292e-01,\n",
       "          -2.1812e-01, -5.2127e-01, -5.3344e-01, -3.4299e-01, -6.0217e-02,\n",
       "          -2.9849e-01, -8.3764e-02,  2.3195e-01, -2.6467e-01,  7.3269e-02,\n",
       "          -8.0860e-02,  1.2149e-01, -9.4898e-02, -1.0792e-01,  3.5444e-01,\n",
       "           5.1695e-02,  8.4225e-02,  4.4869e-02, -6.5909e-02, -7.7347e-02,\n",
       "           3.4296e-03,  4.8959e-01, -1.6582e-01,  3.9882e-02,  8.0303e-02,\n",
       "           1.6689e-02, -8.9686e-02, -2.9799e-01, -3.1378e-01,  3.8809e-01,\n",
       "          -3.0610e-01,  6.2506e-01,  9.6232e-02,  3.2242e-01, -2.6947e-01,\n",
       "           4.3462e-02,  1.8032e-01, -1.3492e-01,  9.2855e-02, -2.6593e-01,\n",
       "          -2.5137e-01,  7.2530e-02, -4.5559e-01, -6.3116e-01,  3.7394e-01,\n",
       "           7.7754e-02, -4.1964e-01, -1.6542e-02, -3.7512e-01, -1.5200e-02,\n",
       "           4.2850e-02,  2.9707e-01, -1.2385e-01, -3.8798e-01,  1.2938e-01,\n",
       "           1.9612e-01, -4.0967e-02,  6.6517e-03,  3.7980e-01,  3.8502e-01,\n",
       "          -2.4481e-01,  6.8376e-01, -1.8963e-01,  2.6652e-01,  5.1353e-01,\n",
       "           3.5211e-02, -1.3207e-01, -3.9070e-01,  2.9249e-01,  2.1947e-01,\n",
       "           6.0006e-02, -1.7387e-01, -3.4473e-01, -3.7988e-01, -2.0878e-01,\n",
       "           6.3815e-02, -2.7146e-01, -5.1579e-01, -2.9292e-01,  1.1763e-01,\n",
       "          -2.8118e-01,  2.5032e-01, -3.3725e-01,  3.6895e-01, -4.2952e-01,\n",
       "          -5.4249e-01, -3.7521e-01,  3.9407e-01,  9.7182e-02, -8.5246e-02,\n",
       "          -2.5514e-01,  5.7547e-01,  1.9097e-01,  9.8289e-02,  2.4899e-02,\n",
       "           9.3485e-01,  5.7181e-02,  2.1462e-02, -5.6206e-01,  4.3594e-02,\n",
       "           3.0797e-02,  2.3718e-01, -6.4775e-01,  3.1567e-01,  2.2884e-01,\n",
       "           9.5511e-02, -2.2434e-01, -1.2301e-01, -2.2429e-01,  2.0446e-01,\n",
       "          -2.8764e-01,  1.2707e-01,  1.0427e-01, -3.0508e-01, -5.0377e-01,\n",
       "           2.7555e-01, -1.1999e+00,  3.4006e-01, -2.5306e-01, -1.8299e-01,\n",
       "          -1.8584e-01,  4.0360e-01,  5.1267e-02, -6.0772e-02, -1.1567e-01,\n",
       "           1.1089e-01,  1.6649e-01, -2.0127e-01,  3.5430e-01, -4.2488e-01,\n",
       "           2.6610e-01,  5.2358e-01,  4.9975e-01,  3.3914e-01,  1.1583e-01,\n",
       "           2.4449e-01,  2.1050e-01, -1.3824e-01,  3.0396e-01,  1.6047e-01,\n",
       "           3.4639e-01, -5.5525e-01, -7.2642e-02,  2.7300e-01,  4.8506e-02,\n",
       "          -2.1988e-01,  3.5733e-02, -5.1577e-01,  9.4071e-03, -2.6012e-01,\n",
       "           1.8923e-01, -6.3669e-01, -4.4311e-02,  5.0913e-01,  5.5571e-02,\n",
       "          -8.5723e-03,  1.1601e-01,  2.6010e-01, -2.0315e-01,  2.2660e-01,\n",
       "           6.7093e-01, -2.1795e-01, -4.6390e-02, -3.8545e-01,  4.9164e-02,\n",
       "          -1.3578e-02, -8.2797e-01,  3.9172e-01,  4.2955e-01,  3.8406e-02,\n",
       "           3.6295e-02, -4.3737e-01,  2.2554e-01, -6.4286e-02,  2.3661e-01,\n",
       "           1.2875e-02, -5.7261e-01, -7.8607e-01,  5.1913e-02, -2.0507e-01,\n",
       "           2.5782e-01, -5.7877e-02,  3.6990e-01,  7.1524e-01,  5.7725e-01,\n",
       "          -4.1705e-02,  1.0699e-01,  3.6357e-01,  2.1261e-01, -6.5736e-01,\n",
       "          -4.6119e-01,  1.6183e-01, -7.5531e-01,  1.7054e-01, -1.6727e-01,\n",
       "          -1.3581e-01, -2.3793e-01,  3.0746e-01,  1.9123e-01, -2.0728e-01,\n",
       "           2.0443e-01,  3.2025e-01, -2.4744e-01, -4.5379e-01,  5.1146e-01,\n",
       "          -4.1741e-01,  1.6295e-01, -2.8200e-01, -4.7797e-02, -6.7946e-01,\n",
       "          -3.5914e-01,  2.4124e-02,  2.5774e-01, -1.4144e-01, -8.5936e-01,\n",
       "          -1.3546e-01, -7.4397e-02, -4.5771e-02, -3.0526e-02, -3.0894e-01,\n",
       "           4.7677e-01, -2.0260e-01,  4.6706e-02, -1.4003e-01, -2.0501e-01,\n",
       "          -5.4965e-01,  4.2522e-01, -2.1008e-01, -3.5740e-02,  1.8355e-01,\n",
       "           2.2508e-01,  2.2409e-01,  3.5566e-01, -1.1459e-01, -3.2763e-01,\n",
       "          -6.4275e-02,  3.9969e-01,  4.3399e-02, -1.2590e-01,  5.1310e-01,\n",
       "           5.8997e-01,  3.7250e-01,  4.2607e-01, -1.8770e-03, -3.7734e-01,\n",
       "          -1.6818e-01,  4.4803e-01, -1.1110e-01, -6.9094e-01,  1.3637e-01,\n",
       "          -9.4969e-01, -2.6031e-01, -6.2166e-02, -1.6009e-01, -2.8089e-01,\n",
       "          -5.7917e-02,  5.9507e-01, -2.0315e-01,  1.6388e-01,  1.3481e-01,\n",
       "           1.9363e-01,  5.4063e-01, -2.5430e-01,  2.8983e-01, -6.6649e-01,\n",
       "          -1.5239e-01, -1.1831e-01, -3.7930e-01, -4.3319e-01, -1.3292e-01,\n",
       "           5.2327e-01,  1.8709e-01,  4.3492e-01, -3.5690e-01, -5.6909e-01,\n",
       "           6.3063e-01, -5.9611e-01, -3.5736e-01,  3.0883e-01, -7.5716e-03,\n",
       "           5.2402e-01,  2.0851e-01, -2.1622e-01,  3.7749e-02,  6.8215e-01,\n",
       "           4.8982e-02, -1.6088e-01, -3.2711e-01, -4.5600e-01,  8.6702e-01,\n",
       "           2.6429e-01,  2.1121e-01, -2.8251e-01,  4.2864e-01,  3.6725e-01,\n",
       "          -4.5543e-01,  1.4229e-01, -5.1324e-01, -3.9014e-01,  1.5928e-01,\n",
       "           3.2841e-01,  3.8020e-01, -2.2367e-01,  6.7550e-02, -1.3124e-01,\n",
       "          -7.2757e-02, -4.3816e-01,  9.8568e-02, -3.5433e-02,  2.7402e-01,\n",
       "          -4.7287e-02, -2.2351e-01,  1.5070e-01, -7.5772e-01, -3.7646e-01,\n",
       "          -2.4968e-01, -5.4006e-02,  9.6048e-02, -2.7468e-01, -2.9298e-01,\n",
       "          -1.8641e-01, -1.8393e-02,  8.0092e-01,  1.3113e-01,  1.7250e-01,\n",
       "           3.6787e-01,  3.0653e-02, -8.4331e-02,  6.4409e-01,  5.2438e-02,\n",
       "          -1.0406e-01,  5.8718e-01, -2.8499e-01,  1.3248e-01,  5.6765e-01,\n",
       "          -3.6911e-01,  1.0153e-01, -8.0079e-02,  3.4592e-01,  3.5057e-01,\n",
       "          -1.9395e-01, -2.1964e-01,  3.0564e-01, -5.8916e-01, -1.2937e-01,\n",
       "          -7.5152e-02,  7.2319e-01, -6.5201e-01,  1.4879e-01,  2.3101e-01,\n",
       "           2.0772e-02, -1.7172e-01, -2.1383e-01, -4.9890e-02, -6.4440e-01,\n",
       "           5.0202e-01, -1.1176e-01, -1.9665e-01,  5.6100e-01, -1.1244e+01,\n",
       "           2.4094e-01,  1.0308e+00, -2.6176e-01, -5.0414e-03, -3.9913e-01,\n",
       "          -8.2861e-02, -1.3976e-01,  4.4184e-01, -4.1533e-01, -9.9397e-02,\n",
       "          -8.1215e-02,  7.4910e-02,  1.0815e-01, -3.4932e-01,  3.0263e-02,\n",
       "           5.5584e-03, -7.1436e-01,  3.5027e-01, -4.9089e-02,  5.1350e-02,\n",
       "          -4.9075e-02, -3.2869e-01,  8.1533e-02, -4.0534e-01,  2.2461e-01,\n",
       "          -2.8624e-01, -3.0957e-01, -4.8850e-02,  3.4395e-01, -3.7450e-01,\n",
       "          -6.5420e-02,  2.2747e-01,  3.3205e-01, -4.1436e-01, -6.6852e-01,\n",
       "           1.7592e-01, -1.3890e-02, -3.3633e-02, -1.3614e-02, -3.3589e-01,\n",
       "           4.4648e-02, -2.1253e-01,  2.2994e-01,  3.2788e-01,  6.2771e-01,\n",
       "           8.6975e-02,  9.2662e-01,  3.8077e-01, -1.2834e-01, -2.8222e-01,\n",
       "          -9.4287e-01,  5.9663e-02, -8.1384e-01,  5.2345e-01, -1.6063e-01,\n",
       "           7.1708e-01, -3.5336e-01,  1.6902e-01, -1.1646e-01, -1.6772e-02,\n",
       "           1.2937e-01,  1.3997e-01,  3.5616e-02, -4.0166e-01, -3.0913e-01,\n",
       "           2.9320e-01, -1.1489e-01,  2.4463e-02, -5.4903e-01, -9.7749e-02,\n",
       "          -9.2376e-02,  5.8203e-01,  2.4777e-01,  3.5646e-02,  3.9252e-01,\n",
       "          -3.6635e-01, -1.7855e-01,  1.6584e+00, -4.1690e-01,  1.6925e-01,\n",
       "          -1.3019e-01,  1.8309e-01, -4.6575e-01, -7.3579e-01, -2.8352e-02,\n",
       "          -8.4889e-02,  8.3755e-02,  2.6374e-01,  4.4616e-01, -2.1009e-01,\n",
       "          -2.7736e-01, -9.6165e-03,  1.0650e-01, -3.3188e-01,  9.5192e-01,\n",
       "          -1.5522e-01, -3.5791e-01, -1.3382e-01,  5.0800e-01, -5.2944e-01,\n",
       "          -3.7771e-01, -1.7082e-01, -3.9186e-01,  1.2860e-01, -3.2342e-01,\n",
       "          -3.9128e-02,  2.1437e-02,  1.1737e-01,  4.6058e-01, -2.7624e-01,\n",
       "           1.3337e-01,  3.2932e-01, -6.0613e-02, -2.5589e-01, -5.4609e-01,\n",
       "           3.8651e-01,  4.7169e-01, -2.5523e-01, -7.5225e-01, -2.1781e-01,\n",
       "          -5.8366e-02,  1.3076e-01, -3.0651e-02, -7.3154e-01,  2.2444e-01,\n",
       "          -7.0318e-01, -1.8821e-02,  3.6778e-01, -9.3804e-02,  4.6718e-01,\n",
       "           3.1302e-02, -5.6592e-01, -1.5239e-02, -1.5399e-01, -2.4461e-01,\n",
       "          -7.1411e-01, -2.9851e-01,  6.2052e-01, -3.6523e-01,  1.7140e-01,\n",
       "          -4.0230e-03, -1.2485e-01, -2.9250e-01,  1.3033e-01,  5.0603e-02,\n",
       "           1.7836e-01,  5.6655e-01, -1.0789e-01, -1.2127e-01, -9.4068e-02,\n",
       "          -2.6060e-01, -1.9935e-01, -2.9844e-01, -7.0782e-02, -5.9702e-01,\n",
       "           9.8918e-02, -1.9025e-01, -1.8541e-01, -3.1158e-01,  4.8898e-01,\n",
       "           5.9697e-01,  2.3817e-02, -4.3575e-01,  4.5708e-01,  2.9765e-01,\n",
       "          -2.3640e-01, -7.3912e-02, -4.9759e-01, -3.4130e-02,  2.7560e-01,\n",
       "          -6.5194e-03, -4.1088e-01, -3.3084e-01, -6.9681e-02, -5.5520e-01,\n",
       "          -2.4616e-01,  1.1699e+00, -5.3562e-01, -4.7702e-01,  4.5323e-01,\n",
       "           2.8407e-01, -2.7133e-01, -2.8665e-02,  1.9075e-01, -1.4462e-01,\n",
       "          -2.5558e-01,  2.1249e-01,  6.7685e-01, -1.4503e-01,  2.4307e-01,\n",
       "           3.3380e-01,  3.3995e-01,  6.3902e-03,  2.0006e-01, -2.0316e-01,\n",
       "          -2.7993e-01,  2.6705e-02, -4.6810e-02, -3.9281e-01,  1.0593e-01,\n",
       "          -5.0282e-03, -9.7650e-02,  3.8234e-01, -4.2132e-01,  1.3469e-02,\n",
       "           4.6372e-01,  7.2732e-01, -1.7017e-01, -2.2650e-01,  4.4526e-01,\n",
       "           3.4611e-01,  6.3445e-02,  1.0502e-01,  2.0967e-01,  3.7757e-02,\n",
       "           3.8135e-01,  2.6705e-01, -3.8417e-01, -7.0238e-02,  3.4253e-01,\n",
       "           2.8993e-01, -4.4476e-01, -8.2231e-01,  2.8567e-03,  4.0977e-01,\n",
       "          -1.2729e-01,  4.8023e-01, -9.0667e-02,  1.5972e-01,  2.8327e-02,\n",
       "          -1.1278e+00, -1.9778e-02,  3.6163e-01, -4.4955e-01, -2.2944e-01,\n",
       "          -5.4591e-01, -2.8148e-01, -7.1565e-02, -5.4813e-01,  3.3507e-01,\n",
       "           5.3387e-01, -9.7460e-01,  1.8274e-01,  4.7832e-01, -4.3798e-01,\n",
       "           6.5366e-01, -1.0436e-03,  3.4711e-01,  8.3076e-01, -5.9087e-02,\n",
       "          -2.4970e-01,  1.9153e-01, -1.3481e-01, -3.0593e-01,  8.3091e-02,\n",
       "           7.1746e-01,  1.1761e-01,  6.5277e-02, -1.3372e-01, -4.8267e-01,\n",
       "          -3.0525e-01, -1.8489e-01, -2.5455e-01,  1.1235e-01,  5.5861e-01,\n",
       "          -1.7064e-01, -3.6106e-01, -2.4716e-01,  2.5236e-01, -1.2118e-01,\n",
       "           3.4906e-01,  4.3570e-01,  2.4265e-01,  8.9844e-02, -5.0399e-01,\n",
       "          -1.3935e-01,  6.4379e-01, -4.2123e-01,  8.7078e-01, -1.0261e-01,\n",
       "          -3.5936e-01,  6.7971e-02,  1.9287e-01,  3.0803e-04, -8.2998e-02,\n",
       "          -6.4654e-01,  3.3301e-01, -3.3241e-01, -1.2421e-01,  5.3943e-01,\n",
       "          -1.6688e-01,  9.2077e-02, -5.0576e-01,  1.1921e-01, -6.3591e-02,\n",
       "          -2.3395e-02,  6.5120e-01, -3.0362e-02,  7.5460e-02, -5.2069e-01,\n",
       "          -9.7736e-02, -2.7798e-01, -1.4503e-01,  2.8300e-01, -3.3488e-01,\n",
       "          -3.4350e-01, -2.9192e-01, -8.4859e-02, -7.2215e-02,  7.0874e-01,\n",
       "           7.9103e-01, -1.9978e-01, -3.9188e-01, -1.1146e-01,  2.1243e-02,\n",
       "           1.5319e-01,  3.3228e-01,  8.1613e-02, -1.0225e-01, -1.7234e-01,\n",
       "          -5.3567e-01, -3.3244e-01, -2.0864e-02,  5.1238e-02, -2.3764e-01,\n",
       "           4.9697e-01, -4.1592e-02, -5.0235e-02,  3.8631e-01,  3.8037e-01,\n",
       "           2.6909e-01,  7.2729e-01,  1.8015e-01, -5.6700e-01, -2.5092e-01,\n",
       "           3.2247e-02, -1.9859e-01,  5.0858e-02, -3.8425e-01,  5.3339e-01,\n",
       "           1.8824e-01,  5.3826e-01,  4.4649e-01]], device='mps:0'),\n",
       " tensor([[-0.2974, -0.7044, -0.3158,  ...,  0.0674,  0.7277,  0.4297],\n",
       "         [-0.3013, -0.5480, -0.4200,  ...,  0.2718,  0.6160,  0.1351],\n",
       "         [-0.1401, -0.5549, -0.4022,  ...,  0.1990,  0.5515,  0.4581]],\n",
       "        device='mps:0'),\n",
       " tensor([[-0.1992, -0.7009,  0.0205,  ...,  0.5747,  1.0341,  0.2235],\n",
       "         [-0.1957, -0.6270, -0.4668,  ...,  0.5565,  0.9487,  0.4234]],\n",
       "        device='mps:0'),\n",
       " tensor([[-0.0037, -0.8362, -0.4347,  ...,  0.1675,  0.0198,  0.2123],\n",
       "         [-0.3292, -0.2590, -0.5016,  ..., -0.0148,  0.3509,  0.1607]],\n",
       "        device='mps:0'),\n",
       " tensor([[-1.0486e-01, -5.3601e-01, -3.5893e-01, -7.6706e-01, -2.1581e-01,\n",
       "          -8.0611e-02, -8.7259e-01,  2.7287e-01,  2.0348e-01,  3.6979e-01,\n",
       "           1.8290e-01,  6.1081e-01,  1.3175e-02, -1.1300e-02, -9.2947e-02,\n",
       "          -2.1458e-01, -1.7472e-01,  8.5034e-02,  8.1684e-01,  3.9081e-01,\n",
       "           1.8465e-01,  5.3788e-01, -1.3812e-01, -1.2944e-01, -4.9896e-01,\n",
       "           4.6718e-01,  4.1100e-02,  2.9071e-01,  1.9937e-01, -7.8236e-01,\n",
       "           2.2647e-01,  3.8815e-01, -1.9998e-01,  1.2759e+00, -2.7271e-01,\n",
       "          -4.0070e-01,  3.5463e-01, -5.2667e-01, -9.6079e-02,  9.3694e-02,\n",
       "           3.7319e-01,  2.2627e-01, -3.6428e-02,  1.7345e-01, -1.8849e-01,\n",
       "           3.8984e-01,  6.2129e-02, -3.8030e-02,  1.3808e-01, -6.8087e-01,\n",
       "          -2.7998e-01, -1.4845e-01,  4.5519e-01,  5.7747e-01, -2.3839e-01,\n",
       "          -3.9239e-01,  5.9324e-01,  7.4043e-02,  1.0622e-01,  2.6006e-01,\n",
       "          -1.9110e-01,  3.1945e-01, -2.3206e-01,  2.4882e-01,  1.8321e-01,\n",
       "           1.9101e-01, -5.2608e-01, -3.7308e-01, -6.9822e-01,  5.5332e-02,\n",
       "          -4.0892e-03,  3.7389e-01,  2.7160e-01,  5.7313e-02,  4.1172e-01,\n",
       "          -3.3860e-01,  2.6273e-01, -2.2130e-01, -4.2314e-01,  2.2084e-01,\n",
       "           2.6210e-02,  4.7560e-01,  3.8959e-01,  1.9689e-02, -5.4573e-02,\n",
       "          -1.3649e-01,  2.2125e-01,  1.8338e-02,  1.4460e-01,  1.4875e-01,\n",
       "          -6.9810e-02, -1.0500e-01, -3.4707e-01,  7.3270e-02,  1.5708e-01,\n",
       "          -1.9005e-01,  2.4547e-01, -1.9698e-01,  5.0065e-02, -3.2070e-01,\n",
       "           2.5155e-01, -6.0162e-02, -1.2735e-01,  5.3491e-01, -7.0988e-02,\n",
       "          -4.0439e-01,  2.5481e-01, -3.4024e-01, -3.7555e-01,  5.1713e-02,\n",
       "          -1.7387e-01, -2.3570e-01,  3.3367e-02, -4.0287e-01, -1.2646e-01,\n",
       "           1.6285e-01, -1.4059e-01,  1.8273e-02, -1.2550e-01,  3.2936e-01,\n",
       "           4.1520e-02, -1.1609e-02, -7.6940e-02,  3.7814e-01,  3.3483e-02,\n",
       "          -2.0045e-01,  5.9305e-01,  2.0417e-01,  5.3051e-01,  5.3354e-01,\n",
       "           1.1752e-01, -1.6815e-01, -4.1534e-01, -1.4020e-01,  1.5979e-01,\n",
       "           2.2762e-01,  8.2210e-03, -3.4301e-01, -2.5394e-03,  6.1370e-02,\n",
       "           1.2202e-01, -3.2980e-02, -1.1340e-01, -2.1859e-01, -1.2771e-01,\n",
       "          -1.4481e-01, -1.6129e-01, -4.0362e-01, -3.4868e-01, -2.3058e-01,\n",
       "          -7.1667e-01, -4.3733e-01,  3.2691e-01,  1.3085e-01, -2.3434e-01,\n",
       "          -4.3902e-01,  2.2984e-01, -5.5376e-02, -1.3619e-02,  4.7475e-02,\n",
       "           1.2331e+00,  1.0182e-01, -9.9619e-02, -7.3238e-02, -3.5539e-01,\n",
       "           3.4609e-01,  1.2031e-01, -1.9029e-01,  3.1905e-01, -1.3387e-01,\n",
       "          -1.3002e-01,  7.2752e-02,  7.5036e-01,  2.3113e-01,  6.1207e-02,\n",
       "          -9.3413e-01,  4.9095e-01, -3.8670e-02, -2.3170e-02,  2.9148e-01,\n",
       "           1.0049e-02, -1.2063e+00,  2.3395e-01, -6.2548e-01,  2.7620e-01,\n",
       "          -2.0665e-01,  8.1580e-02,  2.6410e-01,  8.9019e-02,  1.1608e-01,\n",
       "          -3.7473e-02,  3.3946e-01,  1.9610e-02,  3.9886e-01, -3.0981e-01,\n",
       "           2.4018e-01,  6.6626e-01,  5.0366e-01,  1.3529e-01, -9.5345e-02,\n",
       "          -3.3333e-01,  2.6221e-01,  5.4055e-01,  3.8259e-02,  1.5785e-01,\n",
       "           6.9331e-02, -3.9878e-02,  3.5450e-02,  3.8859e-02, -1.5360e-01,\n",
       "          -2.2989e-01, -1.8371e-01, -7.1280e-01, -8.1218e-02, -3.4430e-01,\n",
       "           2.7908e-01, -2.1509e-01, -1.4399e-01,  3.8890e-01,  9.2114e-02,\n",
       "          -3.4393e-01,  1.8920e-01, -2.6990e-02, -1.5679e-01,  9.8014e-03,\n",
       "           2.8955e-01, -1.0359e-01,  1.5303e-01, -2.1623e-01, -1.6988e-01,\n",
       "           2.4156e-01, -7.0874e-01,  7.6494e-01,  1.2414e-01, -5.3892e-02,\n",
       "           3.7866e-01, -2.9283e-01,  6.4252e-02, -4.7484e-01,  5.1700e-01,\n",
       "          -1.6717e-01, -3.7301e-01, -5.2210e-01,  1.8746e-01,  2.7419e-01,\n",
       "           2.9298e-01, -1.0229e-01,  7.9204e-01,  4.9391e-01,  6.1842e-01,\n",
       "          -1.0424e-01, -1.1653e-01,  3.5880e-01,  3.2708e-01, -2.0710e-01,\n",
       "           4.7858e-02, -2.8606e-01, -5.3529e-01,  6.8794e-02, -2.0805e-01,\n",
       "           9.9736e-02,  7.2293e-02,  4.8909e-01,  4.2747e-01, -4.1131e-01,\n",
       "          -6.3348e-01,  4.2085e-01,  1.9179e-01, -3.0217e-01,  5.8260e-01,\n",
       "          -3.2137e-01,  1.2840e-01,  1.5209e-01, -1.6372e-01, -5.6603e-01,\n",
       "          -4.0578e-01,  2.0824e-02,  1.0868e-02, -1.6669e-01, -8.3007e-01,\n",
       "          -4.3202e-01,  3.6153e-01, -4.1381e-02,  1.3210e-01,  6.9324e-02,\n",
       "           5.1596e-01, -5.0107e-03, -1.4229e-01, -2.0018e-01, -3.6341e-01,\n",
       "          -7.5663e-02, -3.4772e-02, -5.2673e-01,  1.5632e-01,  5.0937e-01,\n",
       "          -3.3486e-02,  3.0552e-01,  3.5797e-01, -4.3510e-01, -1.1857e-01,\n",
       "          -3.1095e-02,  1.9313e-01, -3.3707e-01, -7.1217e-02,  6.5632e-02,\n",
       "          -7.5525e-02,  5.5351e-01,  4.6624e-01, -1.0282e-01, -5.1982e-01,\n",
       "          -5.8142e-02,  7.8116e-01, -7.9228e-02, -5.9659e-01,  3.0996e-01,\n",
       "          -6.8841e-01, -1.3190e-01,  5.7654e-02, -1.2805e-01, -4.0229e-01,\n",
       "           1.4272e-01, -2.1224e-02,  1.2753e-01, -6.1101e-02, -1.9678e-01,\n",
       "           2.1080e-01,  2.1651e-01, -3.9756e-02,  3.9877e-01, -3.6275e-01,\n",
       "          -5.2730e-02, -5.6466e-02, -2.9264e-01, -5.7367e-01, -2.8541e-01,\n",
       "           6.7996e-02,  6.6373e-02, -2.7862e-01, -3.2026e-01, -7.3058e-03,\n",
       "           7.0271e-01, -4.8197e-01,  1.8868e-01,  3.1588e-01,  1.9705e-01,\n",
       "           3.5528e-02,  2.3983e-01, -7.3134e-02,  1.2889e-01,  1.6566e-01,\n",
       "          -1.4724e-01, -6.6411e-02, -5.0056e-02, -5.0920e-02,  6.7969e-01,\n",
       "           4.8682e-01,  8.5642e-02, -2.8156e-01,  5.7910e-01,  3.0537e-01,\n",
       "          -2.3738e-01,  2.3758e-01, -4.7328e-01, -3.9946e-02, -2.2136e-01,\n",
       "           6.5421e-02,  3.5829e-01, -3.5054e-01,  2.4173e-01,  2.9411e-01,\n",
       "           5.5328e-02, -3.6598e-02, -2.3353e-01,  2.3757e-01,  1.4383e-01,\n",
       "          -1.4323e-01, -5.7803e-01,  1.6946e-01, -8.9424e-02, -6.0076e-01,\n",
       "           2.4112e-01,  1.6550e-01, -7.3927e-02, -2.2862e-01, -4.8743e-01,\n",
       "          -6.5441e-02, -4.4828e-01,  7.9482e-01,  4.2033e-01, -3.7090e-01,\n",
       "           3.5447e-01, -1.7168e-01, -3.5721e-01,  3.5838e-01, -4.2097e-01,\n",
       "          -2.9924e-01,  2.0246e-01, -1.2491e-01,  1.6942e-01,  2.5449e-01,\n",
       "          -1.4595e-01, -1.3892e-01,  6.2940e-02,  3.6163e-01,  3.7230e-01,\n",
       "          -3.7111e-01,  9.2690e-02,  3.0468e-01, -5.5662e-01,  2.9395e-01,\n",
       "           2.1541e-01,  4.5352e-01, -3.4087e-01,  1.0466e-01, -1.4394e-01,\n",
       "           2.6700e-01, -3.3411e-01, -5.0746e-02, -4.9214e-01, -3.5550e-01,\n",
       "           3.7704e-01,  2.9707e-01,  1.6051e-01,  3.0916e-01, -1.1395e+01,\n",
       "           5.0216e-01,  7.7299e-01, -6.2035e-01, -1.0031e-01, -1.4953e-01,\n",
       "           1.8810e-01, -5.2872e-02,  2.4485e-01, -3.6069e-01, -1.8492e-01,\n",
       "          -4.7764e-01, -9.7103e-02,  7.0571e-02, -5.5446e-01, -1.9833e-01,\n",
       "           1.8462e-01, -1.0886e+00,  6.1645e-02, -1.4430e-01,  2.7955e-01,\n",
       "          -1.8315e-01, -1.2617e-01, -2.1344e-03, -2.0223e-01,  8.3598e-02,\n",
       "           1.2330e-01, -6.3327e-02,  3.1546e-01,  3.9036e-01,  7.2979e-02,\n",
       "           2.5666e-01,  2.3514e-02,  1.2064e-01, -4.4183e-01, -5.0231e-01,\n",
       "          -1.4799e-01, -1.2101e-01, -2.3866e-02, -2.9374e-01, -6.7107e-01,\n",
       "          -3.2756e-01, -3.5622e-01,  2.3970e-01,  3.8276e-01,  7.8277e-01,\n",
       "           7.9542e-01,  7.8141e-01,  2.6962e-01,  5.2778e-01, -2.8432e-02,\n",
       "          -9.8863e-01,  3.6455e-02, -2.0232e-01, -2.4846e-01, -2.5055e-02,\n",
       "           6.0834e-01, -9.4217e-01,  1.3856e-01, -5.6142e-01, -8.1499e-02,\n",
       "           4.1827e-01,  2.6638e-01,  1.6390e-01, -4.2560e-01, -1.2654e-01,\n",
       "          -8.8788e-02, -1.8427e-01,  1.8037e-01, -6.0691e-02, -3.3238e-01,\n",
       "          -6.8539e-02,  2.9958e-01, -6.7374e-02, -1.7890e-01, -7.9715e-02,\n",
       "           2.7396e-01, -2.1063e-01,  1.7747e+00,  7.8265e-02,  8.2473e-02,\n",
       "          -3.4137e-01,  3.4461e-01, -3.5723e-01, -5.0788e-01, -2.7328e-02,\n",
       "          -2.5367e-02,  1.0869e-02,  4.5878e-01,  4.0122e-01, -2.3137e-01,\n",
       "          -7.6030e-01, -1.1834e-02,  1.9015e-01,  2.3382e-01,  9.1034e-01,\n",
       "          -8.6425e-01, -5.4724e-01, -1.7222e-01,  3.6006e-01, -4.2146e-01,\n",
       "           6.4214e-02, -3.6597e-01, -6.7849e-01,  2.4626e-01, -1.4442e-01,\n",
       "           4.8236e-02,  2.3850e-01, -2.8502e-01,  3.5012e-01, -3.3872e-01,\n",
       "          -7.2552e-04,  3.8824e-01, -2.7851e-01, -3.5204e-01, -2.2237e-01,\n",
       "           1.1332e-01,  5.0203e-01, -2.3897e-01, -3.5643e-01, -2.1489e-01,\n",
       "           2.7116e-01,  1.3018e-01,  5.4696e-01, -2.5157e-01,  3.0745e-01,\n",
       "          -8.0733e-01,  4.6956e-03,  1.0945e-01, -3.2501e-01,  9.6737e-02,\n",
       "           2.3839e-01, -1.9459e-01,  2.6948e-01, -5.7368e-01, -5.6375e-01,\n",
       "          -8.3874e-01,  9.6012e-02,  3.1493e-01, -4.1790e-01, -1.3918e-01,\n",
       "          -2.6670e-01, -2.5988e-01,  1.5031e-01, -3.1356e-01,  2.2381e-01,\n",
       "          -5.6428e-02,  2.5481e-01, -5.8111e-01, -1.8531e-02,  2.4594e-02,\n",
       "          -5.8631e-01, -1.1254e-01, -3.8051e-01, -1.7367e-01, -6.9592e-01,\n",
       "          -3.3344e-01, -3.6257e-02, -1.3639e-01, -6.8849e-03,  4.6538e-02,\n",
       "           3.5882e-01,  5.8525e-02, -4.6289e-01,  9.0323e-01,  4.9007e-01,\n",
       "           2.1807e-01, -5.6047e-01, -1.4672e-01, -3.7731e-01,  3.2671e-01,\n",
       "           1.4712e-01,  1.6146e-01, -1.8580e-01, -3.6078e-01, -6.8422e-01,\n",
       "          -1.6671e-01,  7.0454e-01, -1.3091e-01, -2.7965e-01,  9.8876e-02,\n",
       "           8.0949e-01, -4.3372e-02, -1.0390e-01,  8.4266e-02, -1.9434e-01,\n",
       "          -3.6510e-01, -5.4925e-01,  5.5146e-01, -3.5906e-01,  1.1233e-01,\n",
       "           2.8254e-01,  1.7668e-01,  6.6171e-02,  2.1807e-02, -1.8632e-01,\n",
       "          -2.1569e-01, -2.5239e-02, -5.3016e-01,  1.7037e-01,  1.6437e-01,\n",
       "          -2.3914e-01,  6.8222e-01,  1.8471e-01, -2.7460e-01, -1.9350e-01,\n",
       "           1.5185e-01,  8.9474e-01, -3.3915e-01, -4.9715e-03,  8.1372e-01,\n",
       "           2.3365e-01, -3.0094e-02,  3.5710e-01,  1.4938e-01, -2.9522e-02,\n",
       "           8.6137e-02,  3.6403e-01, -3.0497e-01, -1.7526e-01,  3.7252e-01,\n",
       "           3.8911e-01, -6.7271e-01, -4.9631e-01, -7.5726e-02,  7.0105e-01,\n",
       "          -3.3820e-01,  4.0592e-01, -3.4346e-01,  5.0571e-01,  3.3863e-01,\n",
       "          -8.9915e-01, -2.0228e-02, -5.2276e-01,  3.4349e-01,  8.8250e-02,\n",
       "          -1.6746e-01,  7.4544e-02, -3.6558e-01, -2.7047e-01,  5.7370e-01,\n",
       "           2.1989e-01, -7.5707e-01, -1.2515e-01,  7.9826e-01, -5.1233e-01,\n",
       "           5.0982e-01, -2.4809e-01,  6.5301e-02, -1.5332e-01,  2.9279e-01,\n",
       "          -1.8047e-01,  8.2819e-02, -8.1055e-01, -5.7428e-01,  1.1292e-01,\n",
       "           6.1343e-01,  1.1614e-01,  9.9949e-02, -2.3068e-01,  1.6325e-01,\n",
       "          -2.3078e-01, -7.0953e-01, -5.9680e-02,  2.1554e-02, -1.1767e-01,\n",
       "           1.3395e-01, -3.2228e-02, -2.3234e-01,  3.8241e-01, -3.4936e-01,\n",
       "           1.1679e-01,  3.2336e-01,  5.6387e-01, -6.9460e-03, -1.1546e-01,\n",
       "          -2.7952e-01,  3.9812e-01, -2.0712e-01,  2.1981e-01,  3.6326e-02,\n",
       "          -3.0179e-01,  1.4696e-01,  1.9345e-01, -2.8181e-01,  1.9343e-01,\n",
       "          -6.4841e-01,  3.5364e-01, -1.7642e-01, -2.4022e-01,  1.5531e-01,\n",
       "           2.4154e-01,  3.1540e-01, -3.3483e-01,  2.2801e-01,  9.8886e-02,\n",
       "           1.1009e-01,  8.4901e-01,  1.8797e-01,  1.4130e-01, -6.5756e-01,\n",
       "           1.3368e-01, -7.6076e-01,  1.9439e-01, -2.6222e-01, -2.9079e-01,\n",
       "          -4.5464e-01, -3.1719e-01,  4.6791e-01,  2.5750e-02,  7.5538e-01,\n",
       "           7.4206e-01, -4.0278e-01, -9.5844e-01, -3.8468e-01,  2.0052e-01,\n",
       "           1.9187e-01,  6.1374e-02,  6.4367e-02,  1.7682e-01,  4.9661e-01,\n",
       "          -6.8685e-01, -1.1379e-01, -3.4984e-01,  3.5958e-01, -1.3679e-01,\n",
       "           3.6278e-01, -3.0666e-01, -4.6967e-02,  2.6889e-01,  1.0795e-01,\n",
       "           3.8738e-01,  4.8260e-01,  4.6116e-02, -4.1121e-01, -1.1503e-01,\n",
       "          -2.8475e-01, -2.7054e-01,  2.1038e-01, -7.0437e-01,  2.5747e-01,\n",
       "           4.4000e-01,  4.7605e-01,  8.9460e-01]], device='mps:0'),\n",
       " tensor([[-0.0548, -0.3751, -0.4766,  ...,  0.0464,  0.8423,  0.4111],\n",
       "         [-0.0583, -0.6757, -0.8037,  ...,  0.1694,  0.7852,  0.5016],\n",
       "         [-0.0566, -0.6197, -0.6624,  ...,  0.0110,  0.6389,  0.6625],\n",
       "         [ 0.1128, -0.5118, -0.5146,  ...,  0.3397,  0.9600,  0.5317]],\n",
       "        device='mps:0'),\n",
       " tensor([[-0.2590, -0.5176, -0.5255,  ...,  0.2942,  0.3270,  0.3736],\n",
       "         [ 0.0506, -0.6230, -0.6446,  ..., -0.0875,  0.4018,  0.3101],\n",
       "         [-0.5117, -0.5096, -0.2144,  ...,  0.4028,  1.0750,  0.2706]],\n",
       "        device='mps:0'),\n",
       " tensor([[ 4.1132e-02, -6.5094e-01, -2.8039e-01, -4.6021e-01, -1.2238e-01,\n",
       "          -1.3136e-01, -5.1280e-01,  4.4307e-01,  5.7225e-02,  3.7066e-01,\n",
       "           4.8848e-01,  1.4614e-01, -1.4815e-01, -1.4364e-01, -6.0122e-01,\n",
       "          -1.1221e-01, -3.7243e-01,  1.7685e-01,  3.2952e-01,  7.6030e-02,\n",
       "           4.6740e-01,  1.5353e-01, -5.2199e-01,  1.5487e-01, -7.7357e-01,\n",
       "           1.6558e-01, -1.5291e-02,  2.6168e-01,  3.7716e-01,  1.6756e-01,\n",
       "          -7.8669e-02,  7.2494e-01, -2.5487e-01,  9.1740e-01, -6.1728e-02,\n",
       "          -4.4202e-01,  3.7011e-01, -2.7153e-01,  4.8546e-01, -4.0695e-01,\n",
       "           4.4420e-01,  2.1482e-01,  8.3356e-02,  1.0495e-01, -2.7440e-02,\n",
       "           3.6323e-02,  4.3063e-02, -6.7142e-01, -2.9019e-01, -4.7403e-01,\n",
       "          -2.8729e-01, -3.6718e-01,  2.7165e-01,  1.1617e-01, -2.0159e-01,\n",
       "           2.2293e-02, -1.9416e-02,  2.3446e-02,  1.5310e-02,  3.0861e-01,\n",
       "          -2.2729e-01,  7.6836e-01, -3.0620e-02, -2.4100e-01, -9.4278e-02,\n",
       "           3.3939e-01, -6.6890e-01, -8.4290e-01, -5.4687e-01,  7.6755e-02,\n",
       "           1.5117e-01,  4.6605e-01,  4.7911e-01, -3.0583e-01,  2.7517e-01,\n",
       "           2.7804e-01,  2.1219e-01,  7.4238e-02, -3.6545e-01,  2.9948e-01,\n",
       "           4.0606e-01,  5.0235e-01,  1.0887e-01,  3.9552e-01,  2.4578e-02,\n",
       "           3.4714e-02,  3.1331e-01,  5.8279e-02,  1.1009e-01,  2.5478e-01,\n",
       "           5.0218e-02,  3.1285e-02, -3.5764e-01, -4.0359e-01,  3.1919e-02,\n",
       "          -1.6280e-01,  1.6231e-02, -4.8137e-01, -1.7811e-01, -1.2596e-01,\n",
       "           1.4128e-01,  2.4167e-01,  5.0545e-01, -1.5737e-01, -5.6046e-01,\n",
       "          -4.5775e-01,  1.7645e-01, -2.4639e-01, -3.3275e-02, -1.0876e-01,\n",
       "          -1.9173e-01,  2.5222e-01,  2.3957e-02, -3.4033e-01,  1.7702e-02,\n",
       "           3.2974e-01,  2.1459e-01, -1.5883e-01,  1.5417e-02,  4.2555e-01,\n",
       "           8.1813e-03, -2.5026e-01,  1.4198e-01,  2.9485e-01,  3.5092e-01,\n",
       "           1.2175e-01,  6.0253e-01, -1.1732e-01,  5.0078e-01,  3.2430e-01,\n",
       "           1.6795e-01,  2.3016e-01, -1.5698e-01, -3.6270e-01, -1.8049e-02,\n",
       "           1.4076e-01, -1.5064e-01, -3.7584e-01,  3.4022e-01, -2.0237e-01,\n",
       "           1.8063e-01, -9.9318e-02, -3.8738e-01, -2.9703e-01, -4.4299e-01,\n",
       "          -2.3431e-01,  1.6575e-01,  1.0372e-01, -6.9681e-01, -2.8872e-01,\n",
       "          -1.0631e+00, -6.7549e-01,  1.1503e-01, -5.5806e-02,  4.4024e-02,\n",
       "          -3.4441e-01,  1.9147e-01, -7.5336e-02, -2.9653e-01,  1.2778e-01,\n",
       "           8.1944e-01, -3.2766e-01,  4.7357e-02, -6.0783e-01,  8.6093e-02,\n",
       "          -5.2935e-02,  2.5008e-01,  3.3529e-02,  5.0613e-01,  1.7636e-01,\n",
       "           2.7926e-01, -1.9791e-01,  1.3606e-01, -7.0622e-01,  1.9965e-01,\n",
       "          -2.6802e-01,  9.9100e-02,  2.2822e-01, -3.8104e-01, -3.8077e-02,\n",
       "          -1.6908e-01, -1.8076e+00,  2.5336e-01, -3.1851e-01, -7.9081e-02,\n",
       "           3.0525e-02,  2.4922e-01, -1.0249e-01,  2.9005e-01, -2.9121e-01,\n",
       "          -3.2274e-01,  7.3114e-02,  2.4787e-01,  3.3753e-01,  9.0921e-02,\n",
       "           2.3441e-01,  5.0322e-01,  4.1451e-01, -2.1576e-01, -2.2682e-01,\n",
       "          -3.6321e-01,  2.3242e-01,  1.6370e-01,  3.7013e-01,  8.9903e-02,\n",
       "           1.3376e-01, -5.0049e-01,  6.0623e-02,  4.8659e-01, -6.9196e-03,\n",
       "          -1.5877e-01, -2.6798e-01, -4.7478e-01, -1.5133e-01, -7.3747e-01,\n",
       "           4.2395e-01, -3.9372e-01, -3.7828e-02,  4.4435e-01,  1.0416e-01,\n",
       "          -2.0827e-01,  3.6525e-01,  7.7069e-02, -4.2904e-01,  1.5995e-01,\n",
       "           6.2629e-01, -5.6713e-02, -1.0165e-01, -3.1610e-01, -3.8458e-01,\n",
       "           1.0811e-01, -5.7431e-01,  7.2183e-01,  4.2628e-01, -1.5128e-01,\n",
       "           4.0855e-01, -6.3010e-01,  1.7838e-01, -2.0987e-01,  2.4020e-01,\n",
       "          -2.7833e-01, -5.8647e-02, -5.2566e-01,  2.0458e-01,  1.4023e-03,\n",
       "           1.2732e-01,  3.8060e-02,  7.5138e-01,  5.3144e-01,  3.6914e-01,\n",
       "           5.6516e-02,  2.1748e-01,  3.5617e-01,  4.4463e-01, -1.4929e-01,\n",
       "           3.7078e-01, -8.0329e-02, -3.9264e-01,  5.2472e-01, -2.6272e-01,\n",
       "           4.5876e-02, -3.2677e-01, -6.0130e-02,  9.9844e-02, -1.2725e-01,\n",
       "          -1.2379e-01,  2.8413e-01,  7.3361e-02, -3.3675e-01,  5.8009e-01,\n",
       "          -5.0567e-01,  4.3898e-01,  4.7327e-01, -1.4953e-01, -4.9640e-01,\n",
       "          -5.3207e-01,  8.9080e-02, -3.4797e-02,  2.1530e-02, -7.0674e-01,\n",
       "          -4.7943e-01,  7.8465e-01,  2.6753e-01,  5.5243e-01,  1.6220e-01,\n",
       "           4.3081e-01,  2.1679e-02,  3.9715e-02, -3.3750e-01, -6.3403e-01,\n",
       "          -8.8967e-02,  2.9600e-02, -2.7991e-01,  1.0012e-01,  2.6089e-01,\n",
       "           4.5869e-01,  1.7305e-01,  1.8555e-01, -1.3221e-01, -2.8844e-01,\n",
       "          -6.6876e-02,  9.8902e-02,  1.3098e-01, -3.7536e-01,  3.8760e-02,\n",
       "          -3.9376e-02,  8.6162e-02,  5.6427e-01, -1.4292e-01, -3.5885e-01,\n",
       "          -4.2238e-02,  6.2941e-01, -1.6959e-01, -2.9701e-01,  3.8654e-02,\n",
       "          -7.1529e-01,  1.1136e-01,  1.4990e-01, -1.0957e-01, -5.4880e-01,\n",
       "           3.3329e-01,  7.6895e-02, -3.4422e-01,  8.0773e-02, -6.0784e-02,\n",
       "           4.0142e-01,  2.7023e-01,  4.0927e-01, -1.6985e-01, -1.5928e-01,\n",
       "           2.9303e-02, -3.3732e-01, -1.1418e-01, -3.2113e-01,  2.0889e-01,\n",
       "           1.6329e-02,  2.2674e-01, -1.3801e-01,  4.2453e-02, -8.2166e-02,\n",
       "           7.6514e-01, -4.7394e-01, -1.7635e-01,  3.3653e-01, -2.5326e-01,\n",
       "          -2.2561e-01,  6.2738e-01,  1.4655e-01,  6.7112e-02,  1.8565e-01,\n",
       "           1.1934e-01, -3.2687e-01, -3.1685e-01, -1.2307e-01,  6.4555e-01,\n",
       "           3.7142e-01,  2.2266e-02, -4.8760e-01,  2.0533e-01,  4.8009e-01,\n",
       "          -2.5562e-01, -1.2897e-01, -4.8516e-01, -3.8131e-01,  1.3690e-02,\n",
       "           9.9852e-02,  5.1155e-01, -2.5217e-01,  1.3352e-01,  2.6213e-01,\n",
       "           3.2591e-01, -6.8976e-01, -1.0543e-01,  4.1210e-01,  1.9576e-01,\n",
       "           1.0644e-01, -3.8138e-01,  4.9826e-02, -3.9606e-01, -1.1003e-01,\n",
       "          -1.2580e-01, -1.5131e-02,  1.3449e-01, -3.3877e-01, -4.6836e-01,\n",
       "          -2.0455e-02, -2.4626e-01,  4.8386e-01,  7.2755e-03, -9.0649e-03,\n",
       "           4.4505e-01,  1.4026e-01, -5.1379e-01,  5.5697e-01,  1.8597e-02,\n",
       "          -1.6599e-01,  4.1698e-01, -8.9188e-01,  1.6389e-01,  9.9280e-02,\n",
       "          -1.0108e-01, -1.4909e-01,  1.6488e-01,  8.3273e-01,  8.8939e-02,\n",
       "          -6.7929e-02,  3.4738e-02,  1.5425e-01, -1.4843e-01,  2.1800e-01,\n",
       "          -1.0692e-01,  5.8324e-01, -3.5548e-01,  1.6582e-01, -5.1506e-01,\n",
       "           1.5932e-01, -2.7850e-01, -5.2713e-01, -3.5828e-01, -2.3778e-01,\n",
       "          -1.9636e-01,  2.0226e-01,  4.5100e-02,  1.3833e-01, -1.1349e+01,\n",
       "          -1.4395e-01,  7.7830e-01, -4.1822e-01,  3.9926e-02,  6.8306e-02,\n",
       "           1.9056e-01, -3.5542e-01,  4.7115e-01, -4.5302e-01, -1.9384e-01,\n",
       "          -3.1288e-01,  7.1537e-02, -2.6220e-02, -5.8757e-01,  1.5506e-01,\n",
       "          -2.9810e-01, -4.9651e-01,  1.4024e-01, -7.3113e-02,  4.3695e-01,\n",
       "          -3.4481e-01,  1.5004e-01,  2.4557e-01, -1.5831e-01, -6.3952e-01,\n",
       "          -2.8741e-01, -5.0649e-01,  4.2068e-01,  4.5354e-01, -2.7369e-01,\n",
       "           1.4215e-01, -2.1815e-03,  1.4770e-01, -3.8729e-01, -5.7043e-01,\n",
       "           6.3144e-01, -1.3675e-01,  2.7155e-01, -6.0021e-01, -2.7520e-01,\n",
       "          -2.7646e-01, -2.1347e-01,  2.1472e-01,  1.7041e-01,  7.6957e-01,\n",
       "           1.1814e+00,  9.6204e-01,  1.6587e-01,  4.8739e-01,  1.1785e-01,\n",
       "          -8.6079e-01,  2.7474e-01, -6.4854e-01, -1.3556e-01, -1.1003e-01,\n",
       "           2.4886e-01, -5.1948e-01,  3.6370e-04, -2.4824e-01, -1.8949e-01,\n",
       "           5.4208e-01,  1.3288e-01,  6.1658e-02, -5.5035e-01, -5.6367e-01,\n",
       "           2.6491e-01, -7.3126e-02,  1.5304e-01, -5.5112e-01, -8.4650e-02,\n",
       "           1.5119e-01,  4.2429e-01,  7.9231e-02, -2.9660e-01,  1.1608e-01,\n",
       "          -1.5055e-01,  3.4075e-01,  1.0314e+00, -1.1081e-01,  3.5571e-02,\n",
       "           8.1396e-02,  2.7912e-01, -2.1881e-01, -1.6171e-01, -1.1837e-01,\n",
       "          -2.9392e-02,  4.1182e-01,  2.5885e-01,  5.7565e-01, -1.3764e-01,\n",
       "          -3.9801e-01, -1.4571e-01, -2.0992e-02,  2.8188e-01,  9.3231e-01,\n",
       "          -6.6734e-01,  2.2213e-02, -3.2601e-01, -1.1738e-01, -4.3866e-01,\n",
       "          -2.7857e-01,  5.6346e-02, -4.6713e-01,  1.0397e-01,  1.8817e-01,\n",
       "          -2.0765e-01,  2.0419e-01,  7.3303e-03,  2.7743e-01, -5.3783e-01,\n",
       "          -8.0897e-03,  2.3300e-01,  2.0491e-02, -6.4356e-01, -1.5450e-01,\n",
       "           3.8496e-02,  5.0112e-01, -1.9103e-01, -5.9185e-01, -1.3956e-01,\n",
       "           2.8907e-01,  5.9271e-02, -2.0275e-01, -1.9295e-01,  3.0953e-01,\n",
       "          -9.7582e-01, -3.3245e-01, -3.9477e-01,  7.4162e-02,  2.4147e-01,\n",
       "          -1.1080e-02, -2.2870e-01, -3.0761e-01, -2.9010e-01, -5.6766e-01,\n",
       "          -5.7391e-01, -7.5919e-02,  4.7712e-01, -4.4155e-01, -2.1580e-01,\n",
       "           1.0518e-02,  2.2349e-01, -2.7807e-01,  3.4129e-01, -3.2055e-01,\n",
       "           2.5534e-01,  2.4092e-01, -2.0539e-01,  1.0026e-01,  1.1992e-01,\n",
       "          -1.7309e-01, -2.1708e-01,  1.5221e-01, -8.1765e-02, -6.2045e-01,\n",
       "          -2.2079e-01,  2.6851e-01, -5.0321e-01,  2.3702e-01, -2.0676e-01,\n",
       "           2.7667e-01, -4.1757e-03, -2.9095e-01,  6.2407e-01,  3.9510e-01,\n",
       "           1.0110e-01, -3.4573e-01, -6.8441e-01,  8.5734e-02,  1.0479e-01,\n",
       "          -1.5183e-01, -2.3342e-02, -1.2350e-01, -4.9551e-01, -1.5726e-01,\n",
       "           1.4517e-01,  1.1799e+00,  3.2304e-02, -2.4388e-01,  4.7821e-01,\n",
       "           9.1742e-01, -3.5471e-01, -8.8470e-02, -1.7455e-01,  2.7227e-01,\n",
       "          -4.4190e-01, -4.6159e-01,  5.7691e-01, -3.6048e-01,  3.3605e-01,\n",
       "           1.0486e-01,  4.5831e-03, -7.5772e-03,  1.3232e-01,  9.8754e-02,\n",
       "          -4.6397e-01,  1.2711e-01, -5.0975e-01,  1.4201e-02,  4.0812e-01,\n",
       "          -1.0557e-03,  1.6510e-01, -3.5410e-02, -1.3642e-01,  2.4239e-02,\n",
       "           4.8952e-02,  1.1543e+00, -1.8132e-01, -3.9343e-02,  3.8609e-01,\n",
       "          -2.1701e-01,  6.2528e-01,  1.0961e-01, -2.1857e-01,  1.1182e-02,\n",
       "           3.6284e-01, -7.1999e-03, -4.6813e-01,  3.1468e-01,  1.1344e-01,\n",
       "          -1.8300e-01, -5.4540e-01, -4.2191e-01, -3.1568e-02,  5.4659e-01,\n",
       "          -2.8378e-01,  2.5502e-01, -3.2172e-02,  7.1086e-01,  2.7295e-01,\n",
       "          -1.0617e+00,  1.8740e-01, -4.2071e-01,  4.8835e-01,  1.0198e-01,\n",
       "          -5.7575e-01, -2.0667e-01,  5.2290e-02, -2.8079e-01,  4.6482e-01,\n",
       "           3.8359e-01, -1.1508e+00, -1.1696e-01,  4.3873e-01,  2.1275e-01,\n",
       "           3.0361e-01, -2.4762e-01,  1.3308e-01,  4.3007e-01, -1.1649e-01,\n",
       "          -3.5368e-01,  3.3771e-01, -1.2187e-01, -2.6344e-01, -4.8553e-02,\n",
       "           5.6909e-01, -1.0751e-04,  8.7843e-02, -3.7297e-01, -9.6441e-04,\n",
       "          -3.3292e-01, -1.1477e-01, -4.4278e-01,  5.2731e-03,  3.5934e-01,\n",
       "          -2.7647e-01, -4.4583e-01, -4.3651e-02,  2.0049e-01,  4.6175e-02,\n",
       "           5.9854e-01,  4.7369e-01,  5.0791e-01, -1.5297e-01,  5.0240e-02,\n",
       "          -2.8092e-01,  2.7318e-01, -1.6921e-01,  7.6149e-01, -9.8986e-03,\n",
       "          -2.2186e-01,  2.7967e-01,  1.5983e-01, -2.7847e-01, -2.1665e-02,\n",
       "          -2.2587e-01,  1.8053e-01, -1.1312e-01, -2.6131e-01,  3.4921e-01,\n",
       "           2.9430e-03, -2.1970e-03, -2.6998e-01,  2.7315e-01,  3.1188e-02,\n",
       "          -2.4773e-01,  1.2001e+00, -3.0669e-01,  2.4079e-01,  3.9930e-02,\n",
       "           8.6195e-02, -7.1725e-01, -3.3217e-02,  3.4360e-02,  8.8904e-03,\n",
       "          -5.7878e-01, -6.5391e-03,  1.3440e-01, -3.4574e-01,  6.6957e-01,\n",
       "           6.3616e-01, -7.2325e-01, -3.8989e-01,  4.6111e-02, -9.7906e-02,\n",
       "          -2.0112e-02, -5.0668e-01,  4.6546e-02, -2.1752e-01, -2.2184e-01,\n",
       "          -8.4370e-01, -5.6811e-02, -9.4653e-02,  1.0908e-01, -2.5984e-01,\n",
       "           8.2307e-01,  1.9247e-01, -1.2989e-01,  5.2229e-01,  5.5212e-02,\n",
       "           2.4035e-01,  7.2202e-01,  1.0181e-01, -5.7555e-01, -1.0718e-01,\n",
       "           1.0970e-01, -7.9635e-02,  1.3697e-01, -3.2271e-01,  5.5832e-01,\n",
       "           2.3963e-01,  2.5500e-01,  8.3173e-01]], device='mps:0'),\n",
       " tensor([[-0.3412, -0.4120, -0.3786,  ...,  0.1858,  0.4336, -0.0910],\n",
       "         [-0.2348, -0.4913, -0.4917,  ..., -0.0969,  0.7168,  0.5748]],\n",
       "        device='mps:0'),\n",
       " tensor([[-0.4654, -0.4115, -0.4370,  ...,  0.1625,  0.4338,  0.5908],\n",
       "         [-0.2137, -0.4179, -0.6477,  ...,  0.2712,  0.3571,  0.6939]],\n",
       "        device='mps:0'),\n",
       " tensor([[-1.9537e-01, -7.6714e-01, -4.1059e-01, -4.8067e-01, -2.7894e-01,\n",
       "          -1.7681e-01, -3.0621e-01,  5.4667e-01,  1.3083e-01,  8.4706e-01,\n",
       "           4.8280e-02, -2.0061e-02, -1.6855e-01, -4.7247e-01, -8.5564e-01,\n",
       "          -1.2239e-01,  2.4513e-01,  3.8850e-01,  4.5007e-01,  5.0489e-01,\n",
       "           3.9469e-01,  4.0784e-01,  1.5992e-01, -2.3120e-01, -1.0119e+00,\n",
       "           3.4897e-01, -2.1685e-01,  1.2550e-01,  2.8339e-01, -8.0494e-01,\n",
       "          -3.0360e-01,  7.3205e-01,  7.9466e-02,  1.0476e+00, -1.7694e-01,\n",
       "          -3.4208e-01,  1.3059e-01, -8.5001e-01,  3.4618e-01, -6.3179e-01,\n",
       "           5.9245e-01,  1.7569e-01,  3.2807e-01,  2.2345e-01,  6.2118e-01,\n",
       "          -5.8073e-02, -6.5462e-02, -3.6346e-01, -3.8076e-01, -1.2826e-01,\n",
       "          -1.0445e-01, -6.0510e-02,  4.9052e-01,  3.0901e-01, -5.7944e-01,\n",
       "          -3.8171e-01, -9.8867e-02,  1.6028e-01, -6.5294e-02, -1.1130e-01,\n",
       "          -4.5160e-01,  4.7487e-01,  6.1346e-02, -4.0955e-03, -4.3434e-02,\n",
       "          -6.8878e-02, -4.9020e-01, -6.1341e-01, -2.5455e-01, -5.5058e-02,\n",
       "           1.3208e-01,  6.0824e-01, -4.1064e-02, -4.0168e-01,  2.3355e-01,\n",
       "           1.9210e-01,  1.0912e-01, -4.5555e-01, -3.2905e-01, -1.2580e-01,\n",
       "           1.0480e-01,  2.6658e-01, -2.0855e-01, -6.0193e-01,  1.4558e-01,\n",
       "          -1.7034e-01,  5.9732e-01,  1.8657e-01,  1.7989e-01, -1.7151e-01,\n",
       "          -2.6500e-02,  6.3029e-02, -5.1233e-01, -3.0025e-01,  2.1281e-01,\n",
       "          -1.0926e-01, -3.6602e-02, -4.8659e-01, -4.7392e-02, -3.6469e-01,\n",
       "          -1.6833e-01,  3.2088e-02,  8.7177e-02,  9.3714e-02, -8.5195e-01,\n",
       "          -8.5715e-01,  4.5008e-01, -1.9254e-01, -5.9856e-01, -1.0673e-01,\n",
       "          -2.2602e-01,  1.5468e-01,  1.3722e-01, -3.2935e-01, -2.3870e-01,\n",
       "           1.4012e-01,  3.3867e-01,  1.4196e-01, -1.3241e-01,  8.6341e-01,\n",
       "           5.8075e-01, -2.5752e-01,  4.4042e-02,  1.1515e-01, -9.6395e-03,\n",
       "          -2.5079e-01,  4.7678e-01, -3.2103e-01,  5.9793e-01,  2.7615e-01,\n",
       "          -8.6729e-02,  1.2581e-01, -9.1913e-01, -4.2270e-02,  4.7953e-01,\n",
       "           5.4162e-02,  5.4788e-01, -4.5409e-01,  1.6243e-01,  3.0345e-01,\n",
       "          -1.8243e-01,  1.4328e-01, -1.5712e-01, -1.6577e-01, -3.8783e-01,\n",
       "          -4.5147e-01,  4.0010e-02, -2.9055e-01, -6.3229e-01, -1.8957e-01,\n",
       "          -1.0062e+00, -6.2997e-01,  3.5553e-01, -4.6137e-01,  1.6245e-01,\n",
       "          -5.1931e-01,  3.7583e-01, -2.3878e-02,  6.2885e-02,  1.9039e-01,\n",
       "           6.8568e-01, -9.9560e-02,  4.8855e-02, -8.5325e-01, -2.0990e-01,\n",
       "           3.5100e-01, -2.1200e-01, -2.5522e-01, -9.6026e-02, -3.1099e-02,\n",
       "           2.0845e-02, -4.0892e-01,  1.8966e-01, -1.5122e-01,  2.1871e-01,\n",
       "          -1.6423e-01,  1.4329e-01,  8.6569e-02, -2.2639e-01, -3.4781e-03,\n",
       "          -3.3602e-01, -1.6926e+00,  3.8608e-01, -2.0593e-01, -4.5013e-01,\n",
       "          -2.5551e-01,  5.0951e-01,  3.0924e-01,  2.4000e-01, -3.8578e-02,\n",
       "           2.9219e-01,  2.8489e-02,  1.1661e-01,  1.9160e-01,  6.2049e-02,\n",
       "           4.8478e-01,  5.5407e-01,  1.6881e-01,  1.3276e-01,  2.9077e-02,\n",
       "          -6.8876e-01,  1.7794e-02,  1.5747e-01,  4.8294e-01,  4.8952e-01,\n",
       "           9.2107e-02, -2.7803e-01,  1.0156e-01, -1.0773e-01,  6.1992e-02,\n",
       "          -1.5445e-01, -3.6377e-04, -4.5930e-01, -4.3462e-02, -4.2244e-01,\n",
       "           7.1031e-01, -9.9517e-02, -1.9589e-01,  3.0375e-01,  3.6502e-01,\n",
       "          -3.2689e-01,  3.3224e-01,  3.0513e-01, -2.9362e-01, -2.3010e-01,\n",
       "           2.8915e-01, -4.5202e-02,  1.0527e-01, -2.0429e-02, -3.1965e-01,\n",
       "           4.0509e-01, -4.5139e-01,  5.8180e-01,  2.6353e-01, -4.3359e-01,\n",
       "          -2.1276e-01, -1.5718e-01,  1.7456e-01, -8.6073e-02,  4.2702e-01,\n",
       "           3.4052e-03, -2.9327e-01, -3.3557e-01,  1.8396e-01, -4.6975e-02,\n",
       "           6.9673e-02, -2.0385e-01,  4.5260e-01,  9.3595e-01,  2.7460e-01,\n",
       "          -1.9949e-01, -1.8500e-01, -2.4242e-02, -1.5749e-01, -3.0378e-01,\n",
       "          -3.9930e-02,  1.1053e-01, -5.3137e-01,  1.1907e-01, -1.9843e-01,\n",
       "           1.7604e-01,  1.5906e-01, -1.5546e-01,  4.1803e-02, -1.7419e-01,\n",
       "          -3.3974e-01,  2.6306e-01, -2.0475e-01, -7.9363e-02,  3.3274e-02,\n",
       "          -7.4604e-02,  3.8399e-01,  1.2518e-01, -6.7589e-01, -2.4568e-01,\n",
       "          -1.0708e+00,  4.2555e-02, -5.1824e-02,  1.5406e-01, -7.5728e-01,\n",
       "           3.6909e-01,  1.5409e-01,  6.7395e-02, -2.2829e-01,  9.0973e-02,\n",
       "           7.5692e-02,  1.6026e-01,  6.8940e-02, -2.3737e-01, -5.5521e-01,\n",
       "          -1.2644e-01,  9.2649e-03, -3.5186e-01,  2.1910e-01,  3.3237e-01,\n",
       "           5.7214e-01,  2.3372e-01, -3.2698e-02,  2.8902e-02, -2.1253e-01,\n",
       "          -3.0721e-01,  4.4089e-02, -3.8157e-02, -9.4220e-02,  4.2251e-01,\n",
       "           1.9876e-01,  2.2269e-01,  8.6879e-02, -5.1053e-01, -5.8577e-01,\n",
       "           3.8258e-02,  8.3114e-01, -1.4490e-01, -5.4109e-01, -1.3452e-01,\n",
       "          -1.1654e+00, -1.9811e-01, -3.0409e-02,  7.4882e-02, -4.1199e-01,\n",
       "           2.0681e-01,  2.4486e-01,  8.2729e-02,  1.4480e-01, -2.2369e-01,\n",
       "           5.1506e-01,  6.8733e-01,  2.5394e-01, -3.5501e-01, -3.8037e-01,\n",
       "          -7.3257e-01, -5.0407e-01, -9.3412e-02, -3.1341e-01,  9.9567e-02,\n",
       "           1.0736e-01,  3.9702e-01, -3.0007e-01, -1.0238e-01, -5.0481e-01,\n",
       "           3.9953e-01, -2.6294e-01, -1.3230e-01,  6.4711e-01, -3.3720e-01,\n",
       "           7.5133e-03,  4.8357e-01, -3.7441e-01, -4.6051e-01,  1.1917e-01,\n",
       "           5.3933e-02, -2.1986e-01, -7.0389e-01, -1.1915e-01,  8.8815e-01,\n",
       "           5.4531e-01,  5.4494e-02, -6.3993e-01,  8.0498e-01,  3.0454e-01,\n",
       "          -1.0816e-01,  4.1903e-01, -1.8087e-01, -1.9390e-01, -2.9748e-01,\n",
       "           1.9756e-02,  5.2382e-01, -6.4113e-01,  2.5933e-02, -1.8499e-01,\n",
       "           5.0127e-01, -2.4819e-01, -2.0370e-01,  1.7671e-01,  6.3950e-01,\n",
       "          -7.1860e-02, -4.3216e-01,  3.1474e-01, -3.5591e-01, -1.8207e-01,\n",
       "          -1.1534e-01,  5.4537e-02,  4.1720e-01, -1.9655e-01, -3.9537e-01,\n",
       "          -3.3187e-01, -8.6248e-02,  7.2264e-01,  4.8571e-01, -3.2905e-01,\n",
       "           5.7301e-03,  1.6697e-01, -1.6034e-01,  2.6351e-01, -1.2808e-01,\n",
       "           2.7616e-02,  1.6283e-01, -3.0258e-01, -4.2947e-02,  2.5007e-01,\n",
       "          -2.1037e-01, -6.2846e-01, -2.4788e-01,  2.6898e-01, -1.0574e-01,\n",
       "          -3.8148e-01,  1.0107e-01,  3.1893e-01, -1.1224e-01, -2.1849e-02,\n",
       "          -1.2107e-01,  3.1918e-01, -1.8073e-01,  3.4043e-01, -3.6997e-01,\n",
       "           6.5849e-02, -4.5943e-02,  1.4730e-01, -1.1673e-01, -7.7141e-01,\n",
       "           5.1742e-01,  2.5305e-01,  1.3082e-01,  3.7735e-01, -1.1095e+01,\n",
       "           1.9016e-01,  4.5721e-01, -3.0617e-01,  1.7128e-01,  5.3342e-02,\n",
       "           1.4361e-01, -1.9525e-01,  2.8227e-01, -5.8959e-01, -2.0194e-01,\n",
       "          -5.3687e-01,  1.6488e-01,  3.9339e-02, -3.4982e-01, -2.7180e-01,\n",
       "          -7.8787e-02, -7.2682e-01,  9.2477e-02, -1.3003e-01,  3.1331e-01,\n",
       "          -1.8143e-01,  5.9639e-02,  1.4304e-01,  2.1936e-01, -5.7738e-01,\n",
       "           5.0720e-01, -4.2395e-01,  2.2492e-01,  4.6967e-01,  4.7615e-01,\n",
       "           1.3119e-01, -9.8655e-02,  1.9298e-01,  7.1838e-03, -3.5686e-01,\n",
       "           7.1532e-02, -2.6908e-01, -1.9580e-01,  1.9840e-03, -1.2954e-01,\n",
       "          -6.9429e-01, -1.5279e-01,  1.2213e-01,  1.7531e-01,  8.4611e-01,\n",
       "           9.6822e-01,  5.3302e-01,  3.9281e-01, -1.6773e-01, -1.4992e-01,\n",
       "          -7.6913e-01,  1.8752e-01, -3.5508e-01, -5.2967e-01, -3.0684e-01,\n",
       "           5.7806e-01, -3.5854e-01,  3.1392e-01, -6.0300e-01, -4.7590e-01,\n",
       "           8.8613e-01,  3.5909e-01, -1.7575e-01, -5.4210e-01, -4.9923e-01,\n",
       "          -8.8135e-02, -1.4138e-01,  2.8655e-02, -4.0381e-02, -4.1353e-02,\n",
       "           1.0747e-02,  6.1466e-01, -2.5217e-02, -3.1572e-01,  8.3480e-02,\n",
       "           4.6434e-02, -2.9425e-02,  1.6458e+00, -1.8524e-01,  1.0895e-01,\n",
       "          -2.0233e-01,  5.4196e-01, -8.1604e-01, -8.2807e-02, -6.0871e-01,\n",
       "          -4.2682e-02,  2.7058e-01,  1.9541e-01,  1.7041e-01, -1.4085e-01,\n",
       "          -8.5221e-01, -5.6336e-02,  1.2630e-01,  3.7564e-01,  8.2503e-01,\n",
       "          -8.2924e-01, -2.8343e-02,  3.9973e-01, -1.3083e-01, -4.5911e-01,\n",
       "           2.6798e-01, -8.6372e-02, -1.0158e-01,  8.6426e-02, -2.8385e-01,\n",
       "           1.5089e-01,  2.7594e-01, -2.2961e-01,  4.9597e-01, -3.1075e-01,\n",
       "           3.3576e-01,  2.0607e-01, -7.1535e-02, -2.4966e-01, -4.3049e-02,\n",
       "           6.0865e-01,  3.1363e-01, -1.3859e-01,  5.1616e-02, -5.4945e-02,\n",
       "           1.8833e-01,  1.0547e-01, -2.4046e-01, -4.2935e-02,  3.4240e-01,\n",
       "          -4.3607e-01, -5.0706e-02,  3.5839e-01,  2.2877e-01,  3.6365e-02,\n",
       "           1.6230e-01, -8.0707e-02, -2.6896e-02, -5.9514e-01, -4.8213e-01,\n",
       "          -8.9715e-01, -7.5202e-02,  2.1443e-02, -5.9406e-02, -6.4208e-02,\n",
       "           1.3018e-01,  1.3069e-01, -2.3841e-01,  3.2301e-01, -5.3169e-02,\n",
       "           5.7278e-01,  2.3114e-01,  1.3487e-02,  2.1960e-01, -1.5535e-01,\n",
       "          -1.0139e-01, -1.0605e-02, -2.1860e-01,  3.7201e-03, -6.8153e-01,\n",
       "          -5.8923e-01, -2.6921e-02, -1.3825e-02, -1.2907e-01, -3.8364e-01,\n",
       "          -4.7298e-01, -1.8725e-01, -6.8051e-02,  3.0955e-01,  2.9182e-01,\n",
       "           2.1892e-01, -4.0791e-01,  9.2710e-02, -1.2747e-01,  2.6471e-01,\n",
       "           1.7017e-01,  3.5056e-01,  9.7865e-02, -4.9885e-01, -2.1817e-03,\n",
       "          -2.1149e-01,  1.3831e+00, -3.0621e-02, -2.8556e-01,  3.2007e-01,\n",
       "           9.0687e-01, -3.8079e-02,  1.4560e-03, -1.2780e-01, -1.3599e-01,\n",
       "          -2.6386e-01, -6.9848e-01,  5.2983e-01,  1.1066e-01,  3.0149e-01,\n",
       "           3.7633e-01,  7.7778e-02, -2.3054e-01,  1.9589e-01,  3.8864e-01,\n",
       "          -4.8070e-01,  1.1797e-01, -3.9946e-01,  2.9509e-02,  6.7189e-01,\n",
       "          -1.3123e-01, -8.0630e-02, -5.6906e-02, -6.1161e-02, -7.9774e-02,\n",
       "           3.7219e-02,  1.4750e+00, -2.3294e-01, -2.3089e-01,  5.3119e-01,\n",
       "           3.4550e-01,  1.4422e-01,  1.4548e-01,  7.6969e-02,  5.7326e-02,\n",
       "           3.3645e-01, -3.2787e-01, -2.0793e-02,  3.9083e-01,  1.3202e-01,\n",
       "           5.6701e-01, -6.9840e-01, -3.2511e-01,  1.6771e-01,  3.4031e-01,\n",
       "          -1.4345e-03,  5.5335e-01, -7.0797e-02,  5.7491e-01,  8.6270e-01,\n",
       "          -1.0808e+00, -1.2846e-01, -6.1143e-01,  3.9097e-01,  1.5975e-01,\n",
       "          -1.1853e-01, -1.0755e-01, -3.1730e-01, -2.9697e-01,  4.5344e-01,\n",
       "           7.5598e-01, -1.6393e+00,  2.3277e-01,  7.7570e-01, -3.5250e-01,\n",
       "          -6.3455e-02, -1.5563e-02,  5.3659e-01,  2.3435e-01,  6.5379e-02,\n",
       "           7.1031e-02,  3.1827e-01, -3.9154e-01, -5.0151e-01,  2.5212e-01,\n",
       "           6.0703e-01,  1.5304e-02, -1.0166e-01,  1.0758e-01, -4.4199e-02,\n",
       "          -6.4099e-02, -4.1236e-01, -3.0499e-01,  1.1604e-01, -8.4817e-02,\n",
       "           8.0187e-02, -5.3104e-01,  1.6169e-02,  3.1144e-01, -3.2944e-01,\n",
       "           4.3597e-01,  2.7496e-01,  6.6116e-01, -3.0402e-01, -3.2066e-02,\n",
       "           7.7424e-02,  3.5768e-01, -6.6071e-02,  4.2891e-01,  2.7328e-01,\n",
       "          -3.6563e-01,  4.2873e-01, -1.7872e-01, -2.4964e-01, -2.6899e-02,\n",
       "          -1.1214e-01,  1.3644e-01,  1.1943e-01, -5.5254e-01,  3.6508e-01,\n",
       "          -2.0795e-02,  8.4246e-02, -2.2015e-01,  2.2516e-01,  1.4497e-01,\n",
       "          -2.3540e-01,  1.0273e+00, -1.6595e-01,  4.3518e-01, -6.4540e-01,\n",
       "           1.3220e-02, -6.5209e-01, -2.4056e-01,  1.3999e-01, -2.0009e-01,\n",
       "          -4.4039e-01, -1.5340e-01, -1.6267e-01,  2.5339e-01,  4.7559e-01,\n",
       "           5.7232e-01, -3.6762e-01, -6.0975e-01, -4.2002e-01, -1.6752e-01,\n",
       "           3.3373e-01, -1.6584e-01,  5.6919e-01, -1.3775e-01,  4.5406e-01,\n",
       "          -6.1251e-01, -2.4841e-01,  1.6482e-01,  3.2892e-01, -2.2834e-01,\n",
       "           1.0532e+00,  1.3165e-01,  1.8188e-01,  4.1314e-01,  3.1658e-01,\n",
       "           2.2315e-01,  6.5021e-01, -5.0070e-03, -6.6784e-01,  2.3574e-01,\n",
       "           3.7198e-01, -1.5434e-01,  2.9062e-02, -3.7374e-01,  3.6224e-01,\n",
       "           2.2120e-01,  5.5194e-01,  7.3431e-01]], device='mps:0'),\n",
       " tensor([[-0.3253, -0.5724, -0.7419,  ...,  0.2650,  0.4150,  0.7864],\n",
       "         [-0.5159, -0.4882, -0.8141,  ...,  0.1242,  0.1460,  0.7260],\n",
       "         [-0.0509, -0.3662, -0.7399,  ...,  0.1469, -0.3549,  0.7809]],\n",
       "        device='mps:0'),\n",
       " tensor([[-0.3253, -0.5724, -0.7419,  ...,  0.2650,  0.4150,  0.7864],\n",
       "         [-0.5159, -0.4882, -0.8141,  ...,  0.1242,  0.1460,  0.7260],\n",
       "         [-0.0509, -0.3662, -0.7399,  ...,  0.1469, -0.3549,  0.7809]],\n",
       "        device='mps:0'),\n",
       " tensor([[-9.8608e-02, -6.2949e-01, -5.4737e-01, -4.6335e-01, -3.7454e-01,\n",
       "           1.0248e-01, -2.8097e-01,  4.4707e-01, -1.3869e-01,  5.9022e-01,\n",
       "           3.3979e-01,  3.6518e-02, -7.9829e-02, -1.9712e-01, -4.0961e-01,\n",
       "          -5.4994e-01,  1.7904e-01,  2.6107e-01,  2.6766e-01, -1.3636e-01,\n",
       "           9.8026e-01,  3.6199e-01, -8.5372e-02,  1.8939e-01, -6.9872e-01,\n",
       "           3.9986e-01,  4.5490e-02,  8.0399e-02,  4.6963e-01, -6.9729e-01,\n",
       "          -2.0828e-01,  5.4662e-01, -1.2219e-01,  1.1649e+00, -1.8525e-01,\n",
       "          -2.5129e-01,  4.1644e-01, -2.6874e-01,  8.0973e-01,  4.0156e-02,\n",
       "           6.2909e-01, -1.2839e-01,  2.5974e-01,  2.1204e-01,  2.5654e-01,\n",
       "          -1.6577e-02,  1.0736e-01, -6.2638e-01,  5.4556e-01, -5.1867e-01,\n",
       "          -2.1843e-01, -1.5953e-01,  3.6604e-01,  4.8748e-01, -6.7281e-02,\n",
       "           3.5607e-02,  3.0541e-01,  1.6232e-01, -2.5763e-01,  2.0386e-01,\n",
       "           1.3393e-01,  6.9371e-01,  5.7586e-02, -1.9525e-01, -3.0287e-01,\n",
       "           3.0687e-01, -5.2685e-01, -7.0972e-01, -3.4711e-01, -1.5503e-01,\n",
       "          -6.3145e-02,  3.6453e-01, -2.3968e-01, -8.3683e-02,  7.8001e-02,\n",
       "          -8.8895e-02,  3.9835e-02, -1.7901e-01, -2.7637e-01, -9.8806e-03,\n",
       "           6.4246e-02,  3.2661e-01, -1.9744e-01, -4.1654e-01, -2.1876e-01,\n",
       "           3.0475e-01,  7.8369e-01,  6.3690e-02,  2.5669e-01,  8.5968e-02,\n",
       "           3.1750e-01,  9.4549e-02, -3.4083e-01, -4.9471e-01,  3.1401e-01,\n",
       "          -1.3255e-01,  4.9667e-01, -1.4940e-01, -6.5687e-02, -4.7700e-02,\n",
       "           3.7095e-01,  7.2304e-02,  1.7323e-03,  1.1325e-01, -6.6663e-02,\n",
       "          -4.8542e-01,  1.4204e-01, -2.5931e-01, -6.6520e-01, -1.1522e-01,\n",
       "          -7.9637e-02,  1.3277e-01,  1.5561e-02, -3.1430e-01,  2.6189e-01,\n",
       "          -7.3846e-02,  1.9145e-01, -1.3642e-02, -3.3838e-01,  3.7293e-01,\n",
       "           2.5609e-01, -4.0848e-01,  2.2958e-02,  1.6951e-01,  4.6388e-01,\n",
       "          -2.2394e-01,  6.0049e-01, -2.8694e-01,  2.5392e-01,  4.8866e-01,\n",
       "           8.4207e-02, -6.0898e-02, -9.6990e-02, -1.7751e-01, -3.9547e-01,\n",
       "          -4.2915e-02,  7.3082e-02, -3.2798e-01,  1.0024e-01,  2.1027e-02,\n",
       "          -1.5484e-01, -1.5575e-01, -7.1363e-01, -3.2433e-01,  1.6862e-01,\n",
       "          -2.9423e-01,  2.9391e-01, -7.2839e-02, -5.6183e-01, -3.6233e-01,\n",
       "          -8.1793e-01, -4.5420e-01,  5.0598e-01,  3.1144e-02, -1.4616e-01,\n",
       "          -4.4906e-02,  1.9702e-01,  1.8656e-01, -1.9001e-02,  3.6581e-01,\n",
       "           4.5356e-01, -1.7970e-01,  5.5773e-02, -4.7383e-01,  5.1199e-02,\n",
       "           3.6512e-02, -1.1536e-01, -1.4678e-01, -1.8904e-01,  8.2812e-02,\n",
       "           1.8413e-01, -2.0916e-01, -2.0209e-02, -4.3147e-01,  1.4387e-01,\n",
       "          -9.5740e-02,  8.7516e-02,  2.8605e-01, -3.1460e-01, -3.4357e-01,\n",
       "          -4.4457e-01, -2.0415e+00,  1.3037e-01, -1.6708e-01, -1.9640e-01,\n",
       "          -7.8661e-01,  5.6989e-01,  4.0038e-01,  2.3491e-01,  1.2055e-02,\n",
       "          -4.1441e-01, -1.3256e-01, -2.2065e-01,  2.0346e-01, -1.5937e-01,\n",
       "           4.1104e-01,  3.6856e-01,  2.1222e-01, -8.2505e-02, -5.3417e-02,\n",
       "          -3.3993e-01,  1.9027e-01,  6.6698e-02,  4.3789e-02,  1.3752e-02,\n",
       "           8.6470e-02, -3.4516e-01,  1.5865e-01,  6.3413e-01, -3.2035e-01,\n",
       "          -2.6413e-01, -1.5432e-01, -3.1116e-01, -1.1825e-01, -2.1011e-01,\n",
       "           7.5833e-01, -1.0075e-01, -1.9090e-01,  4.1462e-01,  2.9186e-01,\n",
       "          -2.0576e-01,  2.6479e-02, -1.7243e-01, -4.6025e-01, -8.4004e-02,\n",
       "           8.3166e-01, -2.6809e-01, -1.8435e-01, -1.3576e-01, -2.8072e-01,\n",
       "           9.9627e-02, -3.9657e-01,  7.0865e-01,  1.3131e-01, -2.4437e-01,\n",
       "          -1.3250e-01, -4.9085e-01, -3.6081e-02, -6.5221e-02,  2.7934e-01,\n",
       "          -3.2480e-01, -4.4413e-01, -6.2940e-01,  9.8728e-02,  4.2580e-01,\n",
       "           1.0494e-01, -2.2581e-01,  5.8588e-01,  3.7420e-01,  5.5104e-01,\n",
       "          -7.6877e-02,  1.8477e-03,  4.0139e-01,  4.0454e-02, -3.1153e-01,\n",
       "           4.0137e-02,  1.1839e-01, -6.1671e-02,  2.8916e-01, -5.0007e-01,\n",
       "           2.1460e-01,  1.3048e-01,  4.2285e-02,  4.0869e-01, -3.3156e-01,\n",
       "           3.3759e-01, -6.4578e-04, -4.2352e-01, -2.2792e-01,  3.1587e-01,\n",
       "          -3.4318e-01,  4.7882e-01, -1.1497e-01, -1.3533e-01, -6.5793e-01,\n",
       "          -3.0023e-01,  9.9027e-02, -5.2265e-01,  4.0820e-02, -8.5922e-01,\n",
       "          -1.4457e-01,  4.3467e-01, -1.1843e-01,  2.8682e-01, -4.3268e-01,\n",
       "           2.7079e-01,  1.1150e-01,  1.0583e-01, -1.8427e-01, -4.8035e-01,\n",
       "           6.4368e-03,  3.5471e-01, -1.9635e-01,  6.7501e-02,  1.8728e-02,\n",
       "           1.4821e-01,  7.4914e-02, -1.4411e-01,  1.8116e-02, -3.9952e-01,\n",
       "           1.2334e-01,  2.7083e-01, -6.4203e-02, -1.1696e-01, -2.5127e-01,\n",
       "           8.1457e-02,  5.7341e-02,  4.5826e-01, -1.5574e-01, -4.7792e-01,\n",
       "           1.3602e-02,  5.6159e-01,  4.2917e-02,  9.5691e-02, -4.1476e-02,\n",
       "          -1.3511e+00, -3.8201e-01,  1.3646e-01, -2.3684e-01, -1.6250e-01,\n",
       "           1.4728e-01, -4.5766e-02, -1.0990e-01,  1.3434e-01, -3.3727e-01,\n",
       "           4.9913e-01, -7.4794e-03, -9.4830e-02,  2.6318e-01, -2.9595e-01,\n",
       "          -2.6245e-01, -5.5022e-02, -5.5963e-02, -1.9048e-01, -2.1108e-01,\n",
       "           1.0760e-01,  2.4895e-01, -1.6610e-02,  2.9324e-01, -4.2315e-01,\n",
       "           6.9357e-01,  2.0403e-02, -3.6256e-01,  1.8028e-01, -4.6817e-01,\n",
       "          -2.2470e-01,  6.2020e-01, -2.1111e-01,  1.7508e-01,  7.0164e-02,\n",
       "           2.2583e-01, -1.6263e-01, -2.5922e-01, -1.5555e-01,  5.6515e-01,\n",
       "           9.7645e-02,  3.8631e-01, -3.1465e-01,  4.6445e-01,  6.0830e-01,\n",
       "          -4.3876e-01,  2.7575e-01, -3.0324e-01, -3.5509e-01, -1.4900e-01,\n",
       "           3.7937e-01,  2.4503e-01, -2.4568e-01,  1.2754e-01,  2.1491e-01,\n",
       "           2.4189e-01, -2.8123e-01, -1.4375e-01,  1.5623e-01, -1.6800e-02,\n",
       "          -1.6159e-01, -5.9860e-01,  4.8387e-02, -4.0540e-01, -3.7598e-01,\n",
       "          -5.5542e-02, -9.6940e-02,  2.3018e-01, -3.4557e-01, -2.8109e-01,\n",
       "          -4.1082e-01,  1.5328e-01,  7.9392e-01, -8.2566e-02,  1.3355e-02,\n",
       "          -2.4290e-01,  1.3134e-01, -3.9568e-01,  5.6959e-01,  2.4837e-01,\n",
       "          -5.6746e-02,  5.0092e-01, -5.9799e-01,  3.2502e-01,  1.6325e-01,\n",
       "          -2.5296e-01,  7.9720e-02, -1.1265e-01,  4.2524e-01, -1.9241e-01,\n",
       "          -2.6666e-01,  1.8088e-01,  2.9568e-01,  8.0022e-02,  3.3680e-02,\n",
       "           8.9946e-02,  7.7279e-01, -5.6155e-01, -1.1355e-01, -3.2412e-02,\n",
       "           4.6990e-01, -1.4868e-01,  1.7842e-01, -2.5184e-01, -4.5494e-01,\n",
       "           2.8096e-01,  7.6220e-03,  3.0259e-02,  1.5848e-01, -1.1470e+01,\n",
       "           1.7761e-02,  7.9956e-01, -5.9407e-01,  3.9142e-02, -7.9952e-04,\n",
       "           1.4375e-01, -2.8145e-01,  3.6361e-03, -2.5851e-01, -3.7282e-01,\n",
       "           1.6253e-01,  5.8747e-02,  6.8827e-02, -1.9252e-01, -1.2168e-01,\n",
       "          -4.4456e-02, -9.0133e-01,  1.4013e-01, -2.9238e-01,  3.1633e-01,\n",
       "          -4.6944e-01,  3.5652e-01,  1.0775e-01,  4.8818e-02, -3.3732e-01,\n",
       "          -6.3534e-02, -4.1684e-01,  2.3683e-01,  1.0364e-01, -1.6207e-01,\n",
       "          -1.3703e-01, -1.5828e-01,  4.6034e-01, -1.8993e-01, -4.8947e-01,\n",
       "           1.5862e-01, -7.0142e-02,  4.5043e-01, -1.6798e-02, -2.3410e-01,\n",
       "           5.0011e-02, -4.1456e-01,  3.0195e-01,  6.4668e-02,  5.0457e-01,\n",
       "           4.7572e-01,  6.3490e-01,  1.0739e-01, -2.1848e-01,  4.9891e-02,\n",
       "          -8.9429e-01,  3.3171e-01, -9.8824e-01, -2.6344e-01,  3.6779e-02,\n",
       "           7.4622e-01, -4.6007e-01,  9.9323e-03,  8.2971e-02, -3.9312e-01,\n",
       "           2.4107e-01, -5.5413e-02,  1.0686e-01, -6.2270e-01, -1.1555e-01,\n",
       "           3.6365e-01,  1.0164e-01,  1.3567e-01, -3.5856e-01, -1.3223e-02,\n",
       "          -4.3049e-02,  5.9870e-01,  1.5647e-01, -2.5151e-01,  1.0429e-01,\n",
       "          -8.0721e-03,  2.3224e-01,  6.2748e-01,  4.7218e-02,  1.3164e-01,\n",
       "           1.7459e-01,  5.1672e-01, -5.4105e-01, -4.7941e-01, -1.2530e-01,\n",
       "           6.3722e-02,  2.8461e-01,  8.3486e-02,  3.6163e-01, -3.9872e-01,\n",
       "          -3.1281e-01, -7.8561e-02,  3.1281e-02,  5.0728e-02,  1.2098e+00,\n",
       "          -5.8519e-01, -2.2122e-01, -1.0256e-01, -9.1470e-02, -1.3571e-01,\n",
       "          -3.9828e-02, -3.3651e-01, -4.2370e-01,  3.3182e-01,  1.1019e-01,\n",
       "           1.7328e-01,  1.5282e-02,  3.9299e-01,  1.9321e-01, -5.2099e-01,\n",
       "           2.8514e-01,  1.4470e-01,  4.3958e-01, -1.8846e-01, -4.1365e-01,\n",
       "          -1.0991e-01,  3.3157e-01, -3.8281e-01, -3.4973e-01, -1.0983e-01,\n",
       "           3.0412e-02,  5.9583e-02, -5.8809e-02, -3.7567e-01,  2.5850e-01,\n",
       "          -6.2094e-01, -9.3514e-03, -4.2726e-01, -1.1183e-01,  4.7884e-01,\n",
       "           1.2318e-01, -3.6857e-02,  1.1137e-01, -4.2935e-01, -6.7978e-01,\n",
       "          -6.4684e-01, -7.4975e-02,  5.2269e-02, -3.7956e-01, -1.6842e-02,\n",
       "          -1.8671e-01,  6.6050e-02, -3.6548e-01,  1.8633e-01, -8.2184e-02,\n",
       "           7.5332e-01,  2.3533e-01, -1.8486e-01,  1.2410e-01, -1.4424e-01,\n",
       "          -3.6447e-01, -2.1959e-01, -2.4676e-01,  2.2621e-01, -5.0366e-01,\n",
       "          -1.3080e-01,  1.8011e-01,  1.1438e-01,  6.4878e-03, -3.4448e-01,\n",
       "          -1.2932e-01,  2.0570e-01, -1.3707e-01,  3.0997e-01,  1.7373e-01,\n",
       "           2.0329e-01, -6.4335e-02, -4.6287e-01,  8.7786e-02,  4.5179e-01,\n",
       "          -1.1348e-01, -2.7396e-01,  3.8658e-02, -7.0830e-01,  2.1125e-01,\n",
       "           3.8231e-01,  8.2746e-01, -2.5998e-01, -5.3996e-01,  4.6996e-01,\n",
       "           6.2219e-01, -4.6785e-01, -5.7024e-02, -2.1133e-01,  1.2442e-01,\n",
       "          -1.0488e-01, -3.4183e-01,  2.6033e-01, -2.5177e-01,  4.2375e-01,\n",
       "          -9.9348e-02,  2.0811e-01, -1.9413e-01,  4.1807e-01,  9.7647e-02,\n",
       "          -2.0238e-01, -3.5246e-02, -6.2741e-01, -4.3863e-01,  2.3873e-01,\n",
       "           2.6324e-01, -1.5522e-01, -2.3160e-02, -1.2275e-01,  1.1149e-01,\n",
       "           4.8271e-01,  8.5393e-01,  8.1221e-02,  1.4584e-01,  3.8317e-01,\n",
       "           2.3356e-01, -2.7630e-01,  5.9953e-02,  2.9270e-01,  4.0766e-01,\n",
       "           3.5259e-01,  5.3145e-01, -3.4176e-01,  7.2203e-02, -1.6344e-01,\n",
       "           2.0465e-01, -5.7619e-01, -7.6698e-01, -1.3126e-01,  4.3082e-01,\n",
       "           1.5541e-02,  2.9468e-01, -2.7851e-01,  3.4408e-01,  6.3093e-01,\n",
       "          -1.1306e+00, -5.9181e-02, -3.9846e-02, -9.3994e-02, -6.2591e-02,\n",
       "          -5.4897e-01, -2.2413e-01, -3.5234e-01, -2.4418e-01,  5.2852e-01,\n",
       "           5.8811e-01, -8.5286e-01,  1.6085e-01,  2.2953e-01, -1.5005e-01,\n",
       "           9.4328e-02, -9.1644e-02,  3.2863e-02,  5.0134e-01,  1.5175e-01,\n",
       "          -5.7932e-01,  4.5778e-01,  1.5595e-01, -4.3637e-01,  2.6165e-01,\n",
       "           4.1776e-01, -1.8687e-02,  1.5952e-01,  2.2085e-01, -4.9751e-01,\n",
       "          -1.0141e-02, -4.6200e-01, -1.0629e-01,  1.0751e-01, -6.7537e-02,\n",
       "           2.4815e-02, -4.9794e-01, -6.6236e-02,  3.4484e-01,  2.9505e-03,\n",
       "           5.0621e-01,  6.0668e-01,  7.6526e-01, -5.6484e-02, -2.1306e-01,\n",
       "           3.3852e-01,  2.5943e-01, -9.1834e-02,  5.2227e-01, -2.1885e-02,\n",
       "          -1.6835e-01,  3.1658e-01,  2.3124e-02, -2.2045e-01,  1.7294e-01,\n",
       "          -2.5761e-01,  2.6423e-01,  1.0870e-01, -4.9724e-01,  3.6477e-01,\n",
       "           3.7087e-01,  1.5082e-01, -3.0923e-01,  1.8171e-01,  3.9828e-01,\n",
       "          -3.1100e-01,  6.8356e-01,  1.6607e-04,  2.2973e-01,  6.3476e-02,\n",
       "          -1.1506e-01, -5.5647e-01, -3.1374e-01,  2.1509e-01,  9.2105e-02,\n",
       "          -5.0547e-01, -9.5287e-02,  1.5106e-01, -4.9544e-01,  7.8060e-01,\n",
       "           6.2385e-01, -3.7763e-01, -3.5664e-01, -7.1127e-02,  1.1295e-01,\n",
       "           1.2269e-01, -3.3879e-01,  2.9572e-01,  1.1312e-01,  1.0076e-01,\n",
       "          -1.0219e+00, -2.9326e-01,  2.4530e-01,  4.9147e-02, -4.2291e-01,\n",
       "           5.9939e-01,  7.9206e-02,  5.0359e-02,  3.8428e-01,  3.0534e-01,\n",
       "           1.4161e-01,  4.3910e-01,  7.0775e-02, -2.1375e-01,  6.5311e-02,\n",
       "           2.5600e-01, -3.0970e-01, -3.2766e-01, -6.3757e-01,  5.3684e-01,\n",
       "           1.5068e-01,  8.4160e-01,  6.5313e-01]], device='mps:0'),\n",
       " tensor([[-8.5352e-02, -5.4200e-01, -7.7812e-01, -5.7796e-01, -1.5778e-01,\n",
       "          -3.1081e-01, -7.3408e-01,  3.2368e-01,  2.0013e-01,  4.4995e-01,\n",
       "           2.2526e-01, -1.0538e-01, -5.5232e-02, -2.4839e-01, -6.6787e-01,\n",
       "          -1.6160e-01, -1.5623e-02,  3.7049e-01,  2.8087e-01,  1.0797e-01,\n",
       "           7.8579e-01,  6.2412e-01, -3.7101e-02,  6.6066e-02, -4.5260e-01,\n",
       "           1.9056e-02, -7.1114e-02,  3.1682e-01,  1.1692e-01, -3.5315e-01,\n",
       "           3.6763e-02,  5.2388e-01, -3.2580e-01,  1.0498e+00, -2.6193e-01,\n",
       "          -1.6594e-01,  2.4664e-01, -4.0875e-01,  8.7636e-02, -8.6682e-02,\n",
       "           7.0373e-01, -1.1147e-01, -9.7933e-02,  2.6443e-01,  2.2999e-01,\n",
       "           8.0347e-02,  2.1640e-01, -6.0267e-01,  6.1556e-01, -4.5145e-01,\n",
       "          -4.4746e-01, -1.3975e-01,  1.6852e-01,  2.0371e-01, -5.4485e-01,\n",
       "          -3.0285e-01,  1.6084e-01,  4.2909e-01, -3.3298e-01,  4.8815e-02,\n",
       "          -2.7658e-01,  2.7909e-01, -8.3807e-02,  1.1149e-01,  1.3796e-01,\n",
       "           2.3395e-01, -4.9556e-01, -5.6528e-01, -3.9083e-01, -1.1753e-02,\n",
       "           7.5521e-02,  2.5580e-01,  1.3387e-01, -2.5937e-01,  2.7396e-01,\n",
       "          -4.2638e-01,  3.4863e-01, -1.0096e-01, -5.2071e-01,  2.3043e-01,\n",
       "           2.4880e-02,  3.0409e-01, -6.9937e-02, -7.0969e-02,  3.0273e-01,\n",
       "          -6.8895e-02,  3.0005e-01,  1.1177e-01,  2.7575e-01,  1.0562e-01,\n",
       "           5.0528e-01,  1.3105e-01, -9.4473e-02, -1.8671e-01,  3.1863e-01,\n",
       "           4.0521e-02,  4.5060e-01, -3.9926e-01, -1.7475e-01, -3.3047e-01,\n",
       "          -1.0012e-01,  2.7895e-01,  1.4967e-01,  1.7029e-01, -4.4736e-01,\n",
       "          -4.6142e-01,  3.1370e-02, -2.2620e-01, -6.7977e-01, -1.3792e-01,\n",
       "          -3.1635e-01, -3.8664e-02,  4.1532e-01, -7.8583e-01, -1.9493e-01,\n",
       "          -2.1144e-03, -1.5099e-01,  3.2943e-02, -2.1022e-01,  3.7144e-01,\n",
       "           2.8866e-01, -4.0348e-01,  1.6325e-01, -7.3804e-03,  1.7723e-01,\n",
       "          -5.7855e-02,  4.2565e-01, -2.8615e-01,  2.9822e-01,  3.6789e-01,\n",
       "           2.5791e-01, -1.4381e-01, -3.0211e-01, -1.4992e-02,  6.6634e-02,\n",
       "           1.8575e-01,  1.9342e-01, -2.0710e-01, -4.4704e-01,  3.0695e-02,\n",
       "           2.4582e-01, -1.9266e-01, -9.8029e-02, -2.7398e-01, -4.7516e-02,\n",
       "          -3.0406e-01, -1.2753e-01,  1.1766e-03, -2.2356e-01, -4.0935e-01,\n",
       "          -6.5169e-01, -5.0401e-01,  3.3093e-01, -1.6200e-01,  3.6555e-03,\n",
       "          -6.2522e-02,  6.1639e-02, -5.5247e-02, -4.9399e-02,  3.6935e-01,\n",
       "           8.0886e-01, -2.5862e-01,  2.0670e-01, -7.2115e-01, -1.6116e-01,\n",
       "           1.1985e-01,  3.5486e-01, -2.5503e-03, -1.6637e-01,  1.5381e-02,\n",
       "           3.5972e-01, -1.9360e-01,  9.2272e-02, -5.7615e-01, -4.2992e-02,\n",
       "          -8.0818e-01,  4.2762e-01,  2.8893e-01, -1.9466e-01,  2.6110e-01,\n",
       "           3.6573e-01, -1.3442e+00,  1.8319e-01, -3.1841e-01,  2.1174e-01,\n",
       "          -2.4252e-01,  5.4984e-01,  7.5211e-02,  2.6002e-01, -1.8583e-01,\n",
       "           2.6708e-02, -3.5249e-02,  2.2844e-01,  1.4773e-01,  1.4270e-01,\n",
       "           2.1856e-01,  5.6483e-01,  1.8901e-01, -1.8068e-01,  3.0144e-01,\n",
       "          -7.6221e-01,  3.0027e-01,  2.5500e-01, -3.7544e-01,  1.7256e-01,\n",
       "           1.7151e-01, -2.7184e-01,  1.5192e-01,  1.0585e-02,  1.6881e-01,\n",
       "          -2.1968e-01, -3.3298e-01, -5.3596e-01, -2.7891e-01, -4.9125e-01,\n",
       "           5.0738e-01, -1.8172e-01, -7.5449e-02,  4.5839e-01,  3.2365e-02,\n",
       "          -6.6940e-02,  7.7261e-02,  1.0513e-02, -3.4296e-01, -1.9313e-01,\n",
       "           4.4596e-01, -6.0057e-01,  1.2037e-02, -2.6242e-01, -2.4613e-01,\n",
       "          -9.0221e-02, -1.0401e+00,  3.8546e-01,  1.4715e-01,  9.0113e-02,\n",
       "          -4.6145e-01, -3.2563e-01,  6.3219e-02, -4.2712e-01, -1.7140e-01,\n",
       "          -4.4322e-01, -4.8773e-01, -7.8624e-01,  1.3412e-01, -3.5413e-02,\n",
       "           5.1760e-03, -1.8266e-01,  7.6615e-01,  4.1392e-01,  5.0923e-01,\n",
       "          -1.7534e-01, -1.0341e-01,  6.3013e-01, -2.2275e-01, -3.7527e-01,\n",
       "           2.3318e-01, -3.9765e-02, -1.9610e-01,  3.4616e-01, -4.0966e-01,\n",
       "          -3.3229e-01, -2.6185e-01,  5.2190e-01,  4.2875e-01, -2.3258e-01,\n",
       "           2.0577e-01,  2.5147e-02, -5.0031e-02, -3.4358e-01,  7.2709e-01,\n",
       "          -1.4892e-01, -1.5922e-01, -7.5681e-02, -4.4348e-01, -5.8341e-01,\n",
       "          -4.2543e-01,  3.7295e-01,  5.8986e-02,  1.4956e-01, -9.1695e-01,\n",
       "          -4.5394e-01,  5.4734e-01,  1.3683e-01,  2.4949e-01, -1.0118e-01,\n",
       "           2.2672e-01, -1.9583e-01,  2.8479e-01, -1.1349e-01, -4.9296e-01,\n",
       "           2.6143e-02,  2.9665e-01, -3.5374e-01,  4.1312e-01,  4.7999e-01,\n",
       "          -2.5795e-01,  4.5549e-01,  1.3802e-01, -1.2006e-01, -3.3263e-01,\n",
       "          -8.4937e-02, -9.0306e-02, -4.9521e-01, -7.1183e-03,  2.8916e-01,\n",
       "           2.2150e-01,  2.1499e-01,  6.6135e-01, -3.0387e-01, -5.7622e-01,\n",
       "           2.3336e-01,  1.0641e+00, -1.5066e-01, -8.3620e-02, -2.0546e-01,\n",
       "          -8.7831e-01, -2.2693e-01, -2.4303e-01, -4.3907e-01, -6.9420e-01,\n",
       "          -1.3449e-02, -1.1460e-01, -8.6383e-02, -1.8492e-01, -2.7096e-01,\n",
       "           8.4130e-02,  6.0825e-01,  3.2889e-01, -1.2376e-02, -6.2738e-02,\n",
       "          -2.6125e-01, -3.0602e-02, -4.5178e-01, -2.9390e-01, -3.7293e-01,\n",
       "           5.7350e-02,  1.7854e-01, -4.1448e-03, -1.7033e-01, -9.5383e-02,\n",
       "           4.1239e-01, -4.7842e-01, -6.0362e-01,  1.6072e-01, -3.3995e-01,\n",
       "          -3.0753e-01, -1.4007e-02, -1.7644e-02, -2.4196e-01,  2.8517e-01,\n",
       "           1.5960e-01, -4.4645e-01, -2.0627e-01, -2.8362e-01,  4.2421e-01,\n",
       "           4.4599e-01,  2.1915e-01, -5.7873e-01,  6.4842e-01,  4.4109e-01,\n",
       "          -5.6096e-01,  1.2320e-01,  4.7418e-02, -1.4745e-02, -1.6421e-01,\n",
       "           3.6527e-01,  2.8106e-01, -4.6431e-01,  9.7280e-02,  1.2928e-01,\n",
       "           6.7904e-02, -7.5789e-02, -8.7562e-02,  1.1003e-01, -2.8401e-01,\n",
       "          -4.3692e-02, -4.5243e-01,  1.5744e-01, -1.2885e-01, -2.0683e-01,\n",
       "           3.5938e-01,  1.2324e-01, -1.0681e-01, -2.9855e-01, -4.4098e-01,\n",
       "          -1.3604e-01, -1.3777e-02,  5.3113e-01, -7.8424e-02,  1.2442e-01,\n",
       "           1.9316e-01, -7.8938e-02, -1.9188e-01,  2.1065e-01,  1.3372e-01,\n",
       "          -3.9275e-01,  3.9536e-01, -7.3037e-02,  2.4218e-01,  4.9963e-01,\n",
       "          -2.0822e-01, -7.3981e-02,  1.2108e-01,  3.9341e-01, -1.5549e-02,\n",
       "          -4.2864e-01,  5.4742e-02,  3.7807e-01, -4.6137e-01,  8.6060e-02,\n",
       "           2.8605e-03,  2.8014e-01, -4.3735e-01,  1.0646e-01, -7.2431e-02,\n",
       "          -4.5125e-02,  3.3560e-01, -2.6489e-01, -5.5373e-01, -2.7962e-01,\n",
       "           6.8647e-01,  8.0663e-02,  4.3946e-01,  5.3241e-02, -1.1450e+01,\n",
       "           2.0675e-01,  8.0788e-01, -4.1559e-01, -2.2247e-01, -6.2910e-02,\n",
       "           5.0113e-02, -1.3081e-01,  1.6629e-01, -2.6866e-01, -2.2984e-01,\n",
       "          -2.3449e-01,  1.0001e-01,  1.6765e-01, -5.0028e-01,  1.2798e-02,\n",
       "           8.8846e-02, -4.9069e-01,  3.9219e-01, -4.5027e-02,  6.0990e-01,\n",
       "          -5.2500e-01,  3.7675e-01,  1.6112e-01, -1.5963e-01, -2.8692e-01,\n",
       "          -2.4230e-01, -1.1168e-03,  1.9452e-01,  5.2854e-01, -1.4434e-01,\n",
       "           2.1902e-01,  1.2143e-01,  1.8858e-01, -4.4710e-01, -5.5420e-01,\n",
       "           1.5818e-01,  3.6149e-02,  3.9156e-01,  2.3105e-01, -4.9722e-01,\n",
       "           1.8943e-01, -4.0721e-01,  6.7182e-01,  3.1204e-01,  5.6034e-01,\n",
       "           6.6944e-01,  1.0038e+00,  1.7837e-01, -8.9114e-02, -2.0705e-01,\n",
       "          -9.6647e-01,  2.5393e-01, -4.5632e-01,  4.2474e-02, -5.1028e-02,\n",
       "           4.4600e-01, -5.2107e-01,  8.7713e-02, -4.1395e-01, -4.6562e-02,\n",
       "           5.8872e-01, -5.6692e-02,  3.0562e-01, -4.7314e-01, -4.9347e-02,\n",
       "           1.6830e-01, -5.0050e-01,  2.0847e-01, -9.4661e-02, -1.4571e-01,\n",
       "          -2.6074e-01,  4.1500e-01,  2.7828e-01,  3.0769e-02,  1.0117e-01,\n",
       "          -6.7049e-02, -3.5102e-01,  1.3130e+00,  1.8026e-01, -8.3608e-02,\n",
       "           3.2971e-01,  3.1204e-01, -3.3668e-01, -6.7137e-01, -1.0869e-01,\n",
       "          -2.3707e-01, -1.0732e-01,  1.1802e-01,  6.8057e-01, -9.0723e-02,\n",
       "          -6.9405e-01, -3.8265e-01,  1.5802e-01,  3.1593e-01,  5.2334e-01,\n",
       "          -5.8925e-01,  9.7967e-02, -9.8275e-02, -1.4223e-01, -7.6482e-01,\n",
       "          -9.0303e-02, -4.4373e-01, -7.6004e-01,  4.5764e-01, -1.9799e-01,\n",
       "           1.7446e-01, -1.1122e-01,  2.3015e-01,  4.4774e-01, -3.3697e-01,\n",
       "           2.1506e-01,  2.2755e-01,  2.2820e-01, -3.6082e-01,  1.2364e-02,\n",
       "           5.1235e-01,  7.5418e-01, -2.2875e-01, -5.8241e-01, -2.7791e-01,\n",
       "           2.1160e-01,  3.5360e-01,  2.5121e-01, -6.7250e-01, -2.0682e-01,\n",
       "          -7.7326e-01,  3.7586e-02,  3.8890e-02,  2.0816e-01,  5.7487e-01,\n",
       "          -9.3914e-02, -3.8960e-02,  3.0020e-01, -7.2613e-01, -3.0762e-01,\n",
       "          -6.5403e-01, -7.9473e-02,  1.3408e-01, -5.4293e-01,  2.8682e-02,\n",
       "          -1.6787e-01, -1.7593e-02, -2.0197e-01,  1.5522e-01, -1.2918e-01,\n",
       "           5.2075e-01,  1.0777e-01,  1.7315e-02,  1.0249e-01, -9.8586e-02,\n",
       "          -3.7786e-01, -2.5546e-01, -1.3040e-02, -1.0968e-01, -1.2975e-01,\n",
       "          -2.3268e-02,  1.5497e-01, -7.1419e-02, -1.9069e-01,  6.7980e-03,\n",
       "           1.8708e-01, -3.2900e-01, -2.0931e-01,  5.4163e-01,  2.3161e-01,\n",
       "          -9.2599e-02,  2.8866e-02, -3.5488e-03, -2.0607e-01, -4.5318e-04,\n",
       "           2.0993e-01, -1.7446e-01,  2.4448e-01, -6.5602e-01, -3.9577e-01,\n",
       "          -2.4689e-01,  1.3741e+00, -2.0766e-01, -5.7796e-01,  1.3199e-01,\n",
       "           6.3343e-02, -1.1972e-01,  3.8811e-02, -6.3372e-02,  1.8325e-01,\n",
       "          -2.2137e-01, -1.1759e-01,  1.6447e-01, -5.6970e-02,  2.3669e-01,\n",
       "           1.8544e-01, -1.0250e-01,  7.2186e-02,  4.7748e-01,  6.0721e-01,\n",
       "          -1.8922e-01, -6.8663e-02, -2.8000e-01,  1.0304e-02,  7.4415e-02,\n",
       "          -3.1569e-01,  2.4387e-01,  5.0984e-02, -1.3208e-01,  2.5820e-01,\n",
       "           2.9143e-01,  8.3826e-01, -1.5165e-01,  1.0731e-01,  5.6227e-01,\n",
       "          -1.5529e-01,  3.4492e-01,  3.3387e-01,  4.0610e-01,  1.8138e-01,\n",
       "           2.2465e-01, -1.5377e-01, -2.8119e-01,  2.1958e-02,  2.1571e-01,\n",
       "           2.3993e-01, -5.9998e-01, -6.5879e-01,  9.6888e-03,  6.5424e-01,\n",
       "          -8.2534e-02,  3.2619e-01, -2.3275e-01,  6.7169e-01,  7.1879e-01,\n",
       "          -7.4376e-01, -1.3999e-02, -9.5961e-02,  1.7473e-01, -4.7105e-02,\n",
       "          -5.7567e-01,  2.9411e-01,  4.7941e-02, -3.4961e-01,  2.9884e-01,\n",
       "           6.8643e-01, -6.2143e-01,  1.5435e-01,  4.8193e-01, -5.5294e-01,\n",
       "           1.2798e-01, -2.3599e-01,  2.7546e-01,  4.9687e-01,  5.9794e-02,\n",
       "          -1.6564e-01,  2.6867e-01, -5.0242e-01, -5.0401e-01, -2.9049e-03,\n",
       "           7.8640e-01, -2.2475e-02,  1.9234e-01, -1.4340e-01, -1.9674e-01,\n",
       "          -7.0773e-02, -4.3288e-01, -2.4667e-01, -8.7161e-02, -1.7671e-01,\n",
       "           7.1086e-02, -1.1793e-01,  1.5111e-01,  3.2377e-01, -3.6495e-01,\n",
       "           3.8476e-01,  7.0222e-01,  4.2708e-01,  7.3167e-02, -9.6236e-02,\n",
       "          -1.8105e-01,  1.2102e-01, -1.6058e-02,  5.0655e-01, -2.5079e-01,\n",
       "          -1.9586e-01,  1.0872e-01,  2.4934e-01, -3.6776e-01,  2.7902e-01,\n",
       "          -6.3027e-01,  1.4778e-01, -9.7151e-02, -3.2902e-01,  3.2394e-01,\n",
       "           2.1580e-01,  3.7318e-01, -7.3496e-01,  3.3905e-01,  4.0642e-01,\n",
       "           1.5006e-01,  3.8549e-01,  1.8061e-02,  2.3574e-01,  6.1547e-02,\n",
       "           1.1846e-01, -4.8146e-01, -1.7332e-01, -2.0315e-01,  2.2019e-01,\n",
       "          -3.5745e-01, -1.5918e-01,  7.2762e-01,  4.1113e-01,  7.1035e-01,\n",
       "           5.4919e-01,  8.4720e-02, -6.0074e-01,  3.3356e-03,  1.4229e-01,\n",
       "           1.1458e-02, -3.4140e-01,  8.8579e-02, -1.4029e-02,  6.4853e-02,\n",
       "          -7.3825e-01, -5.5251e-01,  1.4212e-01,  3.5648e-01,  3.0075e-02,\n",
       "           5.9818e-01, -1.3093e-01,  2.6761e-01,  3.3328e-02,  2.7566e-01,\n",
       "           1.1119e-01,  7.9991e-01,  1.3070e-01, -6.7800e-01,  9.9035e-02,\n",
       "           1.6132e-01, -8.0572e-02,  2.7606e-01, -5.6287e-01,  3.5673e-01,\n",
       "           6.6479e-02,  1.0070e+00,  5.5202e-01]], device='mps:0'),\n",
       " tensor([[-6.1750e-01, -4.2829e-01, -2.3934e-01, -9.4116e-01, -3.4548e-01,\n",
       "          -6.5496e-02, -6.3874e-01,  9.7828e-01, -9.5892e-02,  4.0096e-01,\n",
       "           1.4919e-01, -5.1603e-03,  2.0980e-01,  1.3680e-01, -6.9506e-01,\n",
       "           1.6740e-01,  4.6094e-01,  2.1159e-01,  3.5951e-01, -4.4329e-02,\n",
       "           7.9832e-01,  3.4245e-01,  1.8019e-01, -2.5205e-01, -4.5876e-01,\n",
       "           5.8994e-01, -2.4628e-01, -9.2014e-02,  2.0509e-01, -8.1949e-01,\n",
       "          -1.3586e-01, -4.9757e-02, -3.6281e-01,  8.7602e-01,  1.1948e-01,\n",
       "          -2.7310e-01, -1.5276e-01, -2.7996e-01,  2.3494e-01, -6.3637e-02,\n",
       "           3.7814e-01,  2.6592e-02, -2.9347e-01,  3.1386e-01, -1.6421e-01,\n",
       "          -1.4057e-01,  1.4047e-02, -7.8429e-01,  1.3666e-02,  1.8955e-01,\n",
       "           5.8851e-02, -4.7205e-02,  1.9231e-01,  3.7030e-01, -1.0617e+00,\n",
       "          -2.6348e-01,  2.7603e-01,  2.9849e-01, -1.7932e-01, -5.7784e-02,\n",
       "          -1.9989e-01,  9.0201e-01,  1.7949e-01, -2.7822e-01,  1.7824e-01,\n",
       "          -4.4899e-01, -5.4601e-01, -6.0781e-01, -1.0682e-01, -1.1967e-01,\n",
       "          -1.0002e-01,  9.7528e-02,  1.4009e-01, -2.9385e-01,  2.3170e-01,\n",
       "           2.1813e-01,  4.7024e-02, -2.7652e-02, -4.3158e-01, -2.6546e-01,\n",
       "           3.6947e-01,  3.4425e-02, -1.4730e-02,  2.0326e-01, -5.8744e-01,\n",
       "          -2.0226e-01,  3.7959e-01,  1.5703e-01,  2.5914e-01,  2.2662e-03,\n",
       "           5.9967e-01,  1.2024e-02,  8.0339e-02, -4.5369e-01,  5.8427e-01,\n",
       "          -5.4955e-01,  1.0329e-01, -1.6007e-01,  4.9532e-02, -5.2088e-01,\n",
       "          -9.8735e-02,  2.4494e-01, -1.8085e-01,  1.6053e-01, -4.6592e-01,\n",
       "          -3.2983e-02,  2.4514e-01,  3.0468e-01, -4.9183e-01,  4.2685e-02,\n",
       "          -6.5134e-03, -2.9791e-01,  7.3694e-02, -6.6407e-01, -8.5795e-01,\n",
       "           5.8354e-01,  1.8428e-02, -4.5240e-01, -2.3930e-01,  7.4443e-01,\n",
       "           2.9738e-01, -2.3201e-01, -5.3072e-02, -3.5548e-01,  2.0259e-01,\n",
       "          -1.8518e-01,  2.5689e-01, -2.7018e-01,  3.8439e-01,  5.5026e-01,\n",
       "           3.1581e-01,  2.8489e-01, -1.8327e-01,  2.7637e-01,  1.4465e-01,\n",
       "           3.6947e-02, -6.6614e-02, -1.5929e-01,  1.5180e-01,  1.2041e-01,\n",
       "          -2.9580e-01, -3.2999e-01,  5.4746e-03,  1.1182e-01, -3.3279e-01,\n",
       "          -1.0202e-01,  1.8680e-01,  3.4808e-01, -1.4092e-01, -2.5403e-01,\n",
       "          -7.8984e-01, -4.2328e-01,  1.0217e-01, -1.8560e-01, -1.7135e-01,\n",
       "          -4.1911e-01,  3.8121e-01, -6.6660e-02,  4.7638e-01,  3.4189e-01,\n",
       "           6.9978e-01, -1.5311e-01,  5.9776e-02, -5.6412e-01,  3.5197e-01,\n",
       "           3.8270e-02, -6.1110e-01, -1.7247e-01,  1.5680e-01, -4.1813e-04,\n",
       "           6.8195e-01,  7.0979e-02,  4.0124e-01,  1.2070e-01, -3.2553e-01,\n",
       "           8.2089e-03,  1.9890e-01, -1.2864e-01, -1.4297e-01,  7.3766e-01,\n",
       "          -5.5188e-01, -2.3029e+00,  1.0721e-01, -3.5844e-01, -3.6529e-01,\n",
       "          -1.8135e-01,  3.7255e-01,  3.3772e-02,  4.9966e-01,  1.5486e-01,\n",
       "           3.7598e-02,  1.6729e-01, -2.8025e-02,  3.6851e-01, -8.4399e-02,\n",
       "           2.6751e-01,  3.3341e-01,  1.2988e-01, -2.3298e-01, -1.0956e-01,\n",
       "          -3.9227e-01,  2.4031e-01, -9.8916e-02,  8.6010e-01,  2.3260e-01,\n",
       "           2.0943e-01, -4.3657e-03, -4.7262e-02,  4.8964e-01,  4.3099e-01,\n",
       "          -6.6541e-02, -1.6585e-01, -4.9522e-01,  4.6269e-03, -4.1279e-02,\n",
       "          -1.2963e-02, -4.5611e-01,  1.3753e-01,  4.2841e-01, -1.2416e-01,\n",
       "          -1.8354e-01,  1.5525e-01, -4.7715e-02, -3.1517e-01, -2.6492e-02,\n",
       "           6.8453e-01,  1.9080e-01, -2.2994e-01,  1.2377e-01, -1.1896e-01,\n",
       "           3.7006e-02, -4.6975e-01,  7.0044e-01,  6.1147e-01,  1.5585e-01,\n",
       "           1.6214e-02, -6.9943e-01,  2.3699e-01, -3.1574e-01, -4.7647e-03,\n",
       "          -1.5027e-01, -1.5731e-01, -4.2986e-01, -5.7285e-02, -5.5854e-02,\n",
       "           4.0115e-01,  7.9943e-02,  4.6653e-01,  1.1837e+00,  2.7973e-01,\n",
       "           1.9184e-02,  5.6097e-01,  2.0801e-01,  4.2492e-01, -4.0180e-01,\n",
       "           3.3082e-01,  8.8825e-02, -6.3896e-01,  2.7188e-01, -4.1593e-01,\n",
       "           4.4963e-02, -4.1432e-02, -3.7926e-01,  1.4208e-01, -2.1276e-02,\n",
       "           3.4237e-01,  3.3517e-01,  1.5186e-01,  1.6838e-01,  2.9411e-01,\n",
       "          -3.1322e-01,  5.9715e-01, -6.0809e-02, -1.1127e-01, -6.3492e-01,\n",
       "          -7.6732e-01,  9.5206e-02,  4.4743e-01, -1.2880e-01, -6.1279e-01,\n",
       "           1.0462e-01, -2.3866e-01, -1.2257e-01, -3.5064e-02,  2.1191e-01,\n",
       "           2.7455e-01,  5.0687e-02, -9.5436e-02,  6.4791e-02, -4.6950e-01,\n",
       "          -4.9277e-01,  4.9140e-02, -1.5758e-01, -1.3063e-01,  5.3918e-01,\n",
       "          -7.9415e-02,  1.1575e-01,  5.7221e-01,  1.9799e-02, -4.5976e-01,\n",
       "          -2.5640e-01,  1.4802e-01,  2.8667e-01, -7.2082e-02,  2.1981e-01,\n",
       "           3.4119e-01,  2.6467e-02,  5.1649e-01,  1.4823e-01, -2.1046e-01,\n",
       "          -2.7239e-01,  5.1173e-01, -1.5755e-01,  6.1403e-02, -1.2527e-01,\n",
       "          -8.7207e-01, -5.6150e-02, -7.8413e-02, -1.6833e-01, -5.4440e-01,\n",
       "           2.6566e-01,  1.4351e-01,  1.0561e-01,  2.2350e-01, -1.5984e-01,\n",
       "           7.0123e-01,  6.5285e-01, -3.3100e-01, -1.6943e-01, -4.0104e-01,\n",
       "           9.3922e-02, -6.8374e-02, -2.9329e-01, -1.4009e-02,  9.5368e-03,\n",
       "          -2.2765e-01, -1.2516e-01, -1.0805e-01,  2.8921e-01, -1.8876e-01,\n",
       "           5.2862e-01,  1.2189e-01,  8.3649e-02,  4.3873e-01,  2.0612e-02,\n",
       "           1.3071e-01,  3.6515e-01, -4.2117e-01,  1.2870e-01,  4.0424e-01,\n",
       "           4.0578e-01, -1.0551e+00, -8.3291e-01, -1.4936e-01,  3.6111e-01,\n",
       "           2.2860e-01,  2.5189e-01,  1.6999e-02, -7.2215e-03,  8.9455e-02,\n",
       "          -2.3402e-01,  3.5135e-01, -6.6415e-02, -5.7412e-01,  3.0561e-01,\n",
       "           2.3567e-01,  5.3263e-01, -2.7377e-01, -5.0579e-01, -6.5233e-02,\n",
       "           2.4616e-01, -3.6381e-01,  1.1392e-01,  2.3480e-01,  4.5466e-01,\n",
       "          -3.6015e-01, -5.6164e-01, -7.0282e-02, -7.3986e-01, -1.3578e-01,\n",
       "           3.2483e-01, -2.1437e-01,  2.7178e-02, -2.4758e-01,  1.5231e-01,\n",
       "          -1.7925e-02, -1.8879e-01,  6.0757e-01,  2.0522e-01, -2.5855e-01,\n",
       "           1.1981e-01, -2.3892e-01, -1.9475e-01,  5.0091e-01, -5.0226e-01,\n",
       "          -3.0348e-01, -5.0960e-02, -1.5349e-01, -3.2418e-01,  3.2181e-01,\n",
       "          -4.1389e-01, -1.4861e-01,  1.6748e-01, -6.4736e-02, -1.3068e-01,\n",
       "          -3.5233e-01,  1.4268e-01, -1.2707e-01, -4.9883e-02, -1.7654e-02,\n",
       "          -1.3353e-01,  6.0900e-02,  1.3308e-01,  6.4538e-02,  1.4388e-01,\n",
       "           5.3696e-01, -1.1464e-01,  2.7582e-03, -2.5900e-01, -2.2469e-01,\n",
       "           5.2498e-01, -9.4973e-02, -3.3117e-01,  5.4928e-01, -1.1156e+01,\n",
       "           1.0386e-01,  7.2631e-01, -5.5808e-01,  5.5128e-02, -2.8539e-01,\n",
       "           2.1085e-01, -3.0009e-01,  7.9101e-02,  2.4967e-01, -2.8460e-01,\n",
       "          -1.2277e-01, -2.7364e-01, -2.3193e-01, -4.0872e-01, -3.9742e-01,\n",
       "          -4.7229e-01, -5.6789e-01,  1.1772e-01, -1.7663e-01, -1.0996e-01,\n",
       "          -5.7036e-01, -1.9379e-01,  2.9696e-01, -4.9559e-03, -5.1069e-01,\n",
       "           5.9541e-02, -2.2948e-01,  2.6399e-01,  4.6126e-02,  2.2045e-01,\n",
       "          -1.9949e-02,  1.4108e-01,  2.0669e-01, -4.4946e-02, -4.5100e-01,\n",
       "          -4.4853e-01,  2.2129e-01, -1.5955e-01, -2.2267e-01, -5.4017e-02,\n",
       "          -3.3838e-01, -2.7689e-01,  1.3447e-01,  3.7594e-01,  6.2596e-01,\n",
       "           1.1595e+00,  5.6208e-01,  2.9245e-01, -1.0625e-01, -8.5438e-02,\n",
       "          -1.0762e+00,  5.5900e-01, -3.7318e-01,  8.1787e-02, -2.5119e-01,\n",
       "           6.6201e-01, -4.9906e-01,  5.4685e-01, -4.0754e-01, -2.5500e-01,\n",
       "           3.7179e-01,  3.0194e-01,  7.1104e-03,  2.8239e-02, -2.7432e-01,\n",
       "           4.0257e-01, -2.6094e-01, -3.6915e-02, -3.7819e-01,  4.0081e-01,\n",
       "          -3.9472e-01,  5.0770e-01, -5.6534e-02,  1.2945e-01,  1.8567e-01,\n",
       "          -3.5001e-01,  1.6610e-01,  1.0337e+00,  1.7414e-01,  4.1071e-01,\n",
       "           1.7420e-01,  5.1160e-01,  5.3064e-02, -8.8958e-01,  1.3691e-01,\n",
       "           2.0502e-01,  2.0789e-01, -6.7096e-02,  5.7131e-01, -2.5517e-02,\n",
       "          -6.5181e-01, -2.3094e-01, -3.2169e-01,  4.3108e-01,  6.5352e-01,\n",
       "          -2.2597e-01, -4.4343e-01,  2.3894e-01, -7.9776e-03, -5.3395e-01,\n",
       "           9.0074e-02, -2.2320e-01, -5.2792e-01,  8.7275e-02, -5.8914e-01,\n",
       "           1.4985e-01, -1.2428e-01,  5.4238e-02,  6.6887e-01, -4.0159e-01,\n",
       "          -1.6113e-01,  3.6602e-02, -4.4532e-02, -4.1270e-01, -4.9520e-01,\n",
       "          -7.7083e-02,  4.1073e-01, -1.9776e-01, -2.7976e-01,  7.0636e-02,\n",
       "          -3.0673e-01,  2.2584e-01, -1.7377e-01, -6.4935e-01,  3.2879e-01,\n",
       "          -6.0415e-01,  2.0578e-01,  2.5360e-01, -2.2507e-01,  4.9017e-01,\n",
       "           1.1337e-01, -8.1817e-02,  1.9397e-01, -9.9369e-02, -7.8554e-01,\n",
       "          -3.2573e-01, -2.7299e-02,  3.2662e-01, -4.2844e-01,  1.5776e-01,\n",
       "           1.5022e-01, -4.5752e-02, -2.2574e-01,  2.7904e-01,  1.0174e-01,\n",
       "           3.7762e-01,  3.9846e-01, -5.5006e-01,  1.1727e-01, -2.9867e-01,\n",
       "          -2.9262e-02, -6.1174e-02,  1.7952e-01,  2.3654e-01, -7.0764e-01,\n",
       "          -1.5224e-01, -2.6434e-01, -3.0965e-01, -2.1650e-01, -2.5120e-01,\n",
       "           2.0680e-01,  1.4038e-01, -1.5897e-02, -1.1308e-01,  5.8077e-01,\n",
       "           4.0814e-02, -1.1167e-01, -4.5997e-01,  3.1443e-01,  4.3178e-01,\n",
       "           2.1794e-01,  7.5875e-02, -1.2830e-01, -5.5532e-01, -1.1922e-02,\n",
       "          -4.3644e-01,  9.1081e-01, -7.3906e-01, -7.1970e-01,  6.6086e-02,\n",
       "           1.4951e-02, -8.8259e-04, -2.3450e-01, -5.0265e-01,  8.1298e-02,\n",
       "          -2.1017e-01, -1.4891e-01,  6.1550e-03, -1.0607e-02,  1.6299e-01,\n",
       "           4.0742e-01,  3.6838e-01,  1.5951e-01, -3.3938e-01, -4.4081e-01,\n",
       "          -6.0244e-01,  1.4567e-02, -1.6795e-01,  2.9330e-01, -2.6752e-01,\n",
       "          -3.2661e-01,  1.6043e-02, -9.1892e-03, -2.2272e-01,  5.2876e-01,\n",
       "           1.2427e-01,  1.6396e+00, -3.7287e-01, -4.4551e-01,  2.9018e-01,\n",
       "           7.2306e-02, -2.7073e-02, -2.8130e-01,  2.5074e-01,  9.5447e-02,\n",
       "           4.0194e-02, -3.6304e-01,  1.5023e-02, -2.0558e-02,  3.2773e-01,\n",
       "           3.2056e-01, -1.0703e+00, -2.0238e-01, -1.7993e-01,  5.5664e-01,\n",
       "           3.2388e-02,  6.3185e-01, -2.2305e-01,  3.0577e-01,  4.6106e-01,\n",
       "          -8.6667e-01, -2.3000e-01, -1.6129e-01,  2.7295e-01, -1.0583e-01,\n",
       "          -9.8735e-02, -1.6178e-01, -3.7165e-02, -1.6936e-01,  3.9538e-01,\n",
       "           7.4751e-01, -7.8426e-01,  7.4081e-01,  7.0467e-01, -9.8234e-02,\n",
       "           5.6815e-01,  9.4187e-02,  3.2641e-01, -1.4659e-01,  2.2669e-02,\n",
       "           2.6200e-01,  1.8542e-01, -6.3165e-01, -2.3962e-01,  2.4473e-01,\n",
       "           4.0158e-01, -1.1943e-01,  1.2108e-01,  7.9292e-02,  3.0960e-02,\n",
       "          -3.7691e-01, -1.0145e-01, -2.4552e-01, -3.9904e-01,  1.1356e-01,\n",
       "           1.4603e-02, -1.5001e-01, -8.6997e-02,  4.7709e-01, -1.7580e-01,\n",
       "           7.5067e-01,  4.2910e-01,  3.7145e-01, -3.5103e-01,  8.9950e-04,\n",
       "          -5.7565e-02,  6.0671e-01, -3.1670e-01,  7.7385e-01,  6.3337e-02,\n",
       "          -2.3140e-01,  2.7251e-01,  6.0547e-03,  1.7829e-01,  2.9892e-01,\n",
       "          -2.3519e-01,  1.2702e-01, -3.0718e-01, -2.3711e-01,  2.8330e-01,\n",
       "          -1.8640e-01,  1.1542e-01, -7.2270e-01, -6.3602e-03,  3.9857e-01,\n",
       "          -3.2051e-01,  4.9873e-01, -2.3074e-01,  6.1026e-01, -4.3509e-01,\n",
       "           3.2669e-02, -4.7470e-01, -4.3275e-01, -2.8508e-01,  5.1854e-02,\n",
       "          -8.6027e-01, -1.9453e-01,  1.7684e-01,  3.7405e-02,  2.1707e-01,\n",
       "           7.5155e-01, -3.3794e-01, -2.4771e-01, -2.0386e-01,  4.0190e-02,\n",
       "           1.8299e-01, -8.2071e-02, -3.5586e-01,  2.1161e-01, -3.5607e-01,\n",
       "          -1.0520e+00, -2.3117e-01,  8.9990e-02,  3.8419e-01, -3.5853e-01,\n",
       "           1.1570e+00,  2.5132e-02, -2.0226e-01,  6.7253e-01,  2.0808e-01,\n",
       "           6.4157e-01,  6.0323e-01,  1.3628e-03, -4.1259e-01, -5.4109e-02,\n",
       "          -1.5877e-01, -3.2881e-01,  1.5757e-01, -1.8346e-01,  5.0556e-01,\n",
       "           5.4802e-01, -2.0806e-02,  5.4756e-01]], device='mps:0'),\n",
       " tensor([[-2.5017e-01, -4.0951e-01, -5.8150e-01, -8.1915e-01, -1.5052e-01,\n",
       "          -5.7906e-02, -7.1361e-01,  6.0564e-01,  1.0555e-01,  4.8171e-01,\n",
       "           1.2954e-01,  5.3878e-01,  9.5785e-02, -9.3411e-02, -1.8779e-01,\n",
       "          -4.7836e-01,  9.6318e-02,  3.5652e-01,  6.7559e-01, -6.7480e-02,\n",
       "           2.3792e-01,  5.7466e-01, -2.4463e-01,  6.8683e-04, -7.7319e-01,\n",
       "           2.6964e-01, -1.0216e-01,  2.9489e-01,  1.8117e-01, -6.2981e-01,\n",
       "           1.2420e-02,  4.5940e-01,  3.9376e-02,  1.3512e+00, -5.7086e-01,\n",
       "          -1.1859e-01,  8.3756e-02, -4.0226e-01, -2.7221e-01,  5.6352e-02,\n",
       "           6.4333e-01, -3.9972e-01,  1.1714e-01,  3.6047e-01, -8.5964e-02,\n",
       "           9.2013e-02,  1.6169e-01, -3.9988e-01,  2.4502e-01, -8.4166e-01,\n",
       "          -3.5078e-02, -1.9502e-01,  5.1588e-01,  6.4750e-01, -5.8013e-01,\n",
       "          -2.2288e-01,  3.0069e-01, -2.9519e-02, -2.1658e-02,  1.7266e-01,\n",
       "           7.2473e-02,  3.8579e-01, -2.9588e-01,  2.3163e-01,  3.2240e-01,\n",
       "           2.3911e-01, -3.2364e-01, -4.1691e-01, -5.2935e-01, -7.6943e-02,\n",
       "          -1.9683e-02,  2.0523e-02,  1.3996e-01, -2.4681e-01,  4.6097e-01,\n",
       "          -6.9722e-01, -4.3019e-02, -9.3010e-02, -2.4177e-01,  2.6989e-01,\n",
       "          -5.8863e-02,  1.9175e-01,  2.8619e-01,  1.0675e-01,  1.1802e-01,\n",
       "           6.9944e-02,  3.2182e-01,  6.9156e-03,  4.2364e-02,  3.5143e-01,\n",
       "          -1.1314e-01, -8.2441e-02, -2.9962e-01, -3.6263e-01,  2.8866e-01,\n",
       "           1.7554e-02,  3.0347e-01, -2.1027e-01,  1.9062e-01, -3.1363e-01,\n",
       "           1.4216e-01,  1.5208e-01, -1.2738e-01,  5.6671e-02, -3.3835e-01,\n",
       "          -4.6835e-01, -1.7200e-03, -2.9526e-01, -4.8128e-01,  2.2380e-01,\n",
       "          -1.5512e-01, -2.8188e-01, -4.0428e-02, -4.1174e-01, -7.7818e-02,\n",
       "          -8.7161e-02,  2.2507e-01, -1.4841e-01, -3.8843e-01, -3.9220e-02,\n",
       "           2.1652e-01, -2.4084e-01,  1.3679e-01,  2.5592e-01,  1.4687e-01,\n",
       "          -9.6053e-02,  6.1971e-01, -7.3813e-03,  2.8383e-01,  2.7330e-01,\n",
       "           1.0690e-01, -3.8049e-02, -2.1126e-01, -2.2528e-01,  9.1181e-02,\n",
       "          -5.4375e-02,  1.5200e-01, -1.9005e-01, -2.1040e-01, -1.5400e-01,\n",
       "           3.1047e-01, -3.3086e-01, -6.0249e-01, -4.7440e-01, -1.6223e-01,\n",
       "          -1.7671e-01, -9.9660e-02, -4.0213e-02, -3.3531e-01, -2.7298e-01,\n",
       "          -5.8918e-01, -3.7714e-01,  8.1131e-01, -4.1900e-02, -3.3844e-01,\n",
       "          -1.4593e-01,  2.2210e-01,  2.9186e-01, -2.9020e-02,  3.5399e-01,\n",
       "           7.9036e-01, -9.9190e-02,  1.8008e-01, -5.2091e-01, -3.6723e-01,\n",
       "           5.1628e-02,  2.6380e-01, -2.5755e-01,  5.5200e-01,  4.3711e-02,\n",
       "           8.6171e-02, -6.9663e-02,  2.1518e-01, -3.8215e-01, -1.7097e-02,\n",
       "          -5.6698e-01,  3.7750e-01,  3.0867e-01, -3.9823e-01, -1.1851e-01,\n",
       "          -4.5334e-03, -1.1324e+00,  7.4096e-02, -7.8959e-01,  3.3459e-01,\n",
       "          -2.6717e-01,  1.7047e-01,  2.5466e-01,  2.7189e-01, -4.4905e-02,\n",
       "           2.0896e-01,  3.1014e-01,  2.7877e-01,  3.9540e-01, -5.4134e-02,\n",
       "           2.7154e-01,  2.5538e-01,  4.5461e-01,  4.1841e-01,  1.8964e-01,\n",
       "          -3.5379e-01,  3.2341e-01,  3.1097e-01,  1.2088e-01,  1.7186e-01,\n",
       "           8.8999e-02, -1.8873e-01,  2.1297e-01,  1.9525e-01, -1.6276e-01,\n",
       "          -2.3764e-01, -2.7365e-02, -4.8983e-01, -4.4757e-02, -4.4449e-01,\n",
       "           2.6005e-01, -4.5452e-01, -8.6925e-02,  3.2782e-01,  2.4457e-01,\n",
       "           1.6024e-02,  2.0624e-01,  1.7821e-01, -2.4205e-01,  2.0102e-01,\n",
       "           3.7136e-01, -3.3120e-01, -1.9295e-02, -2.8387e-01, -2.7960e-01,\n",
       "           2.3741e-01, -5.3431e-01,  6.8730e-01,  3.7042e-01,  3.5529e-02,\n",
       "           1.4587e-01, -2.9238e-01, -3.5315e-02, -5.4176e-01,  3.5152e-01,\n",
       "          -2.9335e-01, -3.9253e-01, -5.4683e-01,  2.0628e-02,  3.9753e-01,\n",
       "           2.4895e-01, -7.0621e-02,  7.3737e-01,  4.1805e-01,  4.4346e-01,\n",
       "          -4.0927e-01, -5.2784e-02,  3.3155e-01, -7.8749e-02, -4.5406e-01,\n",
       "           7.5405e-02, -1.4444e-01, -5.8281e-01,  2.5771e-01, -3.0080e-01,\n",
       "          -1.3368e-01, -1.8064e-01,  3.6852e-01,  1.5025e-01, -1.8232e-01,\n",
       "          -3.5693e-01,  3.6421e-01, -1.4678e-01, -3.9140e-01,  4.9325e-01,\n",
       "          -3.5614e-01,  1.0838e-01,  7.3604e-02, -3.8048e-01, -6.4232e-01,\n",
       "          -2.8738e-01,  2.2486e-01,  1.9225e-01,  1.2100e-02, -6.7703e-01,\n",
       "          -4.1673e-01,  3.1469e-01,  2.2671e-01,  3.0906e-01,  3.2315e-01,\n",
       "           5.3000e-01, -3.3875e-01, -1.4106e-01,  1.6106e-01, -3.9088e-01,\n",
       "          -1.2240e-01, -3.7879e-02, -1.7759e-01,  3.0291e-01,  2.8769e-01,\n",
       "           1.0548e-01,  2.0494e-01,  4.5473e-01, -2.4699e-01, -3.6108e-01,\n",
       "          -2.1697e-02,  2.1922e-01, -9.0812e-02, -7.6728e-02,  2.3934e-01,\n",
       "          -3.7446e-02,  5.5793e-01,  6.7969e-01,  3.9771e-02, -6.3612e-01,\n",
       "           2.1718e-01,  5.1793e-01, -2.7114e-01, -6.9072e-02,  6.3773e-02,\n",
       "          -9.9311e-01, -2.1655e-01, -1.4389e-01, -2.1025e-01, -6.6402e-01,\n",
       "           6.3178e-02,  3.0749e-01,  2.2138e-02, -1.5104e-01, -3.7173e-01,\n",
       "           1.1851e-01,  7.1422e-02, -7.2859e-02,  4.1952e-01, -2.8977e-01,\n",
       "          -3.6422e-01, -2.3505e-01, -2.2530e-01, -5.4622e-01, -4.0241e-01,\n",
       "           3.8418e-01,  3.1143e-01,  1.4944e-01, -2.5240e-01, -5.1746e-01,\n",
       "           4.9996e-01, -6.6463e-01, -7.6214e-02,  2.5059e-01, -4.6959e-01,\n",
       "           5.1906e-02,  4.7303e-01,  1.5847e-01,  1.1851e-01,  7.4993e-02,\n",
       "           1.1224e-01, -3.2333e-01, -5.1588e-01, -1.2636e-01,  6.0520e-01,\n",
       "           3.7206e-01,  3.5409e-01, -1.7350e-01,  3.8754e-01,  4.4354e-01,\n",
       "          -2.9183e-01,  3.8213e-01,  3.8743e-02,  9.9960e-02, -7.9370e-02,\n",
       "           1.7404e-01,  3.5814e-01, -3.4480e-01,  1.8377e-01,  1.9510e-01,\n",
       "          -2.8304e-01, -2.8955e-01,  3.8318e-02,  1.9648e-01, -1.8365e-01,\n",
       "          -2.8061e-01, -2.0338e-01,  5.8693e-02, -3.2039e-01, -4.1471e-01,\n",
       "           2.7876e-01,  5.7883e-06,  9.1288e-02, -3.8857e-01, -4.4405e-01,\n",
       "          -2.1133e-01, -2.6352e-01,  6.7573e-01,  2.2590e-01, -1.8288e-01,\n",
       "           2.0238e-01, -1.1906e-01, -3.4727e-01,  4.7423e-01,  1.2089e-02,\n",
       "          -4.4990e-01,  4.5459e-01, -3.4447e-01,  1.0089e-01,  5.4845e-01,\n",
       "          -1.6523e-01, -1.2093e-01,  1.2572e-01,  9.6750e-02,  2.6504e-01,\n",
       "          -3.8030e-01,  1.4793e-01, -4.1372e-02, -3.3413e-01,  3.1185e-01,\n",
       "           2.8734e-01,  1.8983e-01, -1.9734e-01, -1.8947e-02,  5.2032e-03,\n",
       "           2.2424e-01, -4.9497e-02, -3.9028e-01, -4.9272e-01, -4.9339e-01,\n",
       "           2.1281e-01,  2.0365e-01,  5.2513e-01,  3.4500e-01, -1.1443e+01,\n",
       "           9.6007e-02,  7.8981e-01, -4.2356e-01, -6.7589e-02, -6.8774e-02,\n",
       "           1.2418e-02, -1.3310e-01,  4.2161e-01, -2.8039e-01, -4.3426e-01,\n",
       "          -3.3968e-01, -1.1461e-01,  1.8448e-01, -3.2366e-01, -2.5863e-01,\n",
       "          -1.4210e-02, -7.0924e-01,  4.1055e-01, -3.0025e-01,  4.9784e-01,\n",
       "          -2.1084e-01, -1.0316e-01,  1.3025e-01, -3.3956e-01, -7.1239e-03,\n",
       "          -8.0531e-05, -1.2103e-01,  1.6311e-01,  4.0719e-01, -1.6586e-03,\n",
       "           5.9737e-02, -3.4043e-02,  6.1076e-02, -2.8657e-01, -6.3780e-01,\n",
       "           3.0268e-03, -7.0640e-02,  3.4159e-01, -1.7460e-02, -6.2844e-01,\n",
       "          -2.0958e-01, -2.6812e-01,  3.2146e-01,  2.6108e-01,  9.1483e-01,\n",
       "           1.4543e-01,  9.6944e-01,  1.9305e-01,  1.7947e-01,  9.1374e-03,\n",
       "          -7.4038e-01, -2.9897e-02, -4.7864e-01, -4.3871e-03,  1.1894e-01,\n",
       "           3.8950e-01, -7.7016e-01,  2.3447e-01, -3.4404e-01,  7.7855e-03,\n",
       "           5.0308e-01,  4.2977e-02,  2.6893e-01, -5.9209e-02, -9.0202e-02,\n",
       "           1.3097e-01, -7.4020e-03,  1.2681e-01, -2.1330e-01, -2.5184e-01,\n",
       "          -1.3095e-01,  6.2058e-01,  3.9195e-01, -6.4228e-01,  3.1402e-01,\n",
       "           7.7158e-02, -2.0360e-01,  1.6796e+00, -7.6025e-02, -7.7009e-03,\n",
       "           2.9526e-02,  3.0702e-01, -5.3223e-01, -7.5727e-01,  3.8641e-01,\n",
       "          -1.0468e-01,  6.8159e-02,  2.0797e-01,  4.5057e-01, -1.0299e-01,\n",
       "          -5.1246e-01, -5.0775e-02,  1.2430e-01, -1.1008e-01,  5.9569e-01,\n",
       "          -7.1823e-01, -1.7416e-01, -1.0218e-01,  3.1661e-01, -8.9731e-01,\n",
       "          -6.0705e-02, -6.3190e-01, -3.8042e-01,  5.1712e-01, -2.7966e-01,\n",
       "           1.2145e-01, -9.5618e-04, -6.3254e-02,  5.6354e-01, -2.1322e-01,\n",
       "           8.9682e-02,  2.0787e-01,  9.1625e-02, -1.7563e-01,  3.9689e-02,\n",
       "           4.1475e-01,  4.5235e-01, -4.2447e-01, -6.3043e-01, -4.4245e-02,\n",
       "           1.4456e-01,  1.3719e-01,  4.0748e-01, -5.2132e-01,  3.5309e-01,\n",
       "          -7.9024e-01, -1.4151e-01,  1.3741e-01, -7.6973e-02,  3.6780e-01,\n",
       "           2.0252e-01,  6.3192e-02,  1.8927e-01, -7.4875e-01, -5.9445e-01,\n",
       "          -6.6186e-01, -1.9190e-01,  2.8286e-01, -5.2302e-01, -1.2020e-01,\n",
       "          -1.0853e-01, -1.6859e-02, -1.0828e-01, -5.9862e-02,  1.5334e-02,\n",
       "           1.9237e-01,  2.1922e-01, -1.4902e-01, -1.0578e-01, -1.1191e-01,\n",
       "          -5.9242e-01, -1.1421e-01, -1.2388e-01, -8.3739e-02, -5.3014e-01,\n",
       "          -1.0848e-01,  2.3412e-02, -2.1524e-01, -4.6817e-02,  3.3456e-01,\n",
       "          -3.2766e-02, -3.4286e-02, -5.2196e-01,  6.8360e-01,  3.9405e-01,\n",
       "           5.9487e-02, -2.4667e-01, -5.2080e-01, -1.0784e-01,  2.5922e-01,\n",
       "           1.6638e-01, -5.6197e-02, -1.9007e-01, -2.4435e-01, -4.5099e-01,\n",
       "           9.4639e-02,  7.6377e-01, -5.3703e-02, -6.3629e-01,  2.9742e-01,\n",
       "           5.5223e-01, -1.6000e-01, -1.7826e-01,  1.2050e-02, -1.0996e-01,\n",
       "          -3.2473e-01, -6.2915e-01,  7.6956e-01, -1.5525e-01, -5.3467e-02,\n",
       "           2.2598e-01,  1.4828e-01,  1.5698e-03,  4.1873e-01, -1.2031e-01,\n",
       "          -3.2586e-01,  3.3714e-02, -7.1654e-01,  3.8757e-01,  2.4388e-02,\n",
       "          -2.8522e-01,  7.6943e-01,  2.4115e-01, -3.2780e-02, -6.6488e-02,\n",
       "           1.7696e-01,  7.1225e-01, -2.8266e-01,  6.2572e-02,  5.3779e-01,\n",
       "           1.1822e-01, -1.1897e-01,  3.8856e-02,  4.4299e-02,  3.4177e-02,\n",
       "           3.5400e-01, -9.2593e-02, -3.7978e-01, -4.2728e-02,  3.8138e-01,\n",
       "           3.7934e-01, -5.2837e-01, -7.6276e-01, -4.4565e-03,  7.0754e-01,\n",
       "          -3.2074e-01,  2.5711e-01, -4.4881e-02,  3.4053e-01,  1.4002e-01,\n",
       "          -9.0975e-01, -1.5314e-01, -6.0086e-02,  2.1246e-01, -7.2451e-02,\n",
       "          -1.8029e-01, -1.0542e-01, -2.6903e-01, -1.5699e-01,  6.7497e-01,\n",
       "           2.3387e-01, -7.5683e-01,  7.1208e-02,  5.9180e-01, -5.8435e-01,\n",
       "           4.6497e-01, -2.6567e-01, -5.2158e-02,  3.9212e-01,  1.1812e-01,\n",
       "           6.8218e-02,  1.5088e-01, -4.0419e-01, -4.1748e-01,  1.2428e-01,\n",
       "           9.1968e-01, -8.9010e-02,  1.2188e-01, -4.0636e-01, -2.1431e-01,\n",
       "          -5.1038e-02, -3.5373e-01, -2.9258e-01,  3.5844e-01,  1.5803e-01,\n",
       "          -4.9462e-02, -1.3566e-01, -2.2133e-01,  3.1935e-01,  1.2503e-01,\n",
       "           3.1279e-01,  5.6876e-01,  6.2725e-01,  4.1402e-02, -2.5295e-01,\n",
       "          -7.1905e-02,  1.3912e-01, -2.0489e-01,  5.2657e-01, -1.9732e-01,\n",
       "          -3.4752e-01,  3.6192e-02,  3.0442e-01, -5.6352e-01, -5.4941e-02,\n",
       "          -4.4111e-01,  3.7568e-01, -1.8274e-01,  4.3298e-02,  2.7316e-01,\n",
       "           1.7231e-02,  3.5561e-02, -2.5664e-01,  5.8479e-01,  4.0112e-01,\n",
       "           1.5171e-01,  4.5987e-01,  1.5098e-01,  2.1056e-03, -4.9280e-01,\n",
       "           2.1283e-01, -9.8891e-01, -1.0972e-01, -3.1833e-02, -1.8014e-01,\n",
       "          -4.2845e-01, -5.1729e-02,  1.5384e-01,  1.8842e-01,  7.4014e-01,\n",
       "           7.6174e-01, -2.9544e-01, -5.0204e-01, -1.4508e-01,  1.2075e-01,\n",
       "           2.5420e-01, -1.9170e-01,  3.3792e-01,  8.8430e-02,  2.3056e-01,\n",
       "          -5.1295e-01, -4.6026e-01, -6.0762e-02,  3.3283e-01, -1.7930e-01,\n",
       "           5.5485e-01, -2.0945e-01, -9.8476e-02,  1.4952e-01,  1.5685e-01,\n",
       "           1.8627e-01,  7.3261e-01,  4.1859e-01, -6.4092e-01,  7.6128e-02,\n",
       "           2.0045e-04, -3.0167e-01,  8.1407e-02, -4.3988e-01,  2.9292e-01,\n",
       "           1.2903e-01,  5.3485e-01,  6.5348e-01]], device='mps:0'),\n",
       " tensor([[-0.2614, -0.5826, -0.2952,  ...,  0.1299,  0.5419,  0.6152],\n",
       "         [-0.4844, -0.3417, -0.3675,  ...,  0.1999,  0.4052,  0.4802]],\n",
       "        device='mps:0'),\n",
       " tensor([[-0.3670, -0.8917, -0.7430,  ...,  0.1222,  0.8454,  0.5251],\n",
       "         [-0.2804, -0.8977, -0.7813,  ...,  0.0441,  0.7521,  0.7393],\n",
       "         [-0.2571, -0.6116, -0.7777,  ...,  0.1132,  1.1327,  0.5774]],\n",
       "        device='mps:0'),\n",
       " tensor([[-2.2717e-01, -2.7302e-01, -6.0181e-01, -8.5654e-01, -1.2571e-01,\n",
       "          -3.5271e-01, -6.5153e-01,  8.3764e-01,  5.0578e-02,  6.2638e-01,\n",
       "           8.9650e-02,  4.8298e-01,  1.5202e-02,  7.3042e-02,  1.3883e-02,\n",
       "          -4.0089e-01, -3.8173e-01,  3.8149e-01,  9.9877e-01,  2.2204e-01,\n",
       "           4.2766e-01,  5.4752e-01, -1.8749e-01, -2.3981e-01, -7.1847e-01,\n",
       "           3.1600e-01,  1.8717e-02,  2.8436e-01,  2.0175e-01, -9.0480e-01,\n",
       "           2.4377e-01,  3.1054e-01,  1.8070e-01,  9.7200e-01, -2.0415e-01,\n",
       "          -4.1478e-01,  3.1837e-01, -7.1732e-01, -2.1319e-01, -8.2044e-02,\n",
       "           7.1693e-01,  1.0383e-01,  9.4291e-02,  2.0223e-01,  1.2632e-01,\n",
       "           1.5914e-01,  7.1290e-02, -1.4311e-01,  1.1851e-01, -9.7241e-01,\n",
       "          -5.7139e-02, -3.9734e-02,  7.0548e-01,  5.2461e-01, -7.3747e-01,\n",
       "          -1.1568e-01,  4.8688e-01,  2.5040e-01, -1.0020e-01,  1.5830e-01,\n",
       "          -8.9869e-02,  5.1234e-01, -2.1234e-01, -2.2522e-02,  2.6172e-01,\n",
       "           9.2027e-02, -4.6505e-01, -3.1759e-01, -4.5826e-01, -8.0433e-03,\n",
       "          -2.5181e-01,  4.6269e-01,  3.3633e-01, -3.9249e-01,  4.8752e-01,\n",
       "          -3.4595e-01,  1.2786e-01, -1.7649e-01, -5.0429e-01,  2.3194e-01,\n",
       "           1.6750e-01,  4.6914e-01,  1.4446e-01, -6.0506e-02, -7.9935e-02,\n",
       "          -1.0678e-01,  3.5835e-01,  1.5729e-01, -4.8014e-03,  7.1077e-02,\n",
       "           1.3981e-01, -9.9378e-02, -3.5737e-01, -9.1909e-02,  3.6021e-01,\n",
       "          -5.7258e-01,  2.7439e-01, -1.9577e-01,  3.8045e-02, -3.2958e-01,\n",
       "          -1.1391e-01,  2.4214e-01, -2.1107e-01,  2.2154e-01, -1.1380e-01,\n",
       "          -4.2739e-01,  2.5288e-01, -2.6972e-01, -5.8125e-01,  2.0968e-01,\n",
       "           1.0506e-01, -1.9827e-01, -1.7347e-02, -3.0237e-01, -2.9077e-01,\n",
       "           9.6362e-02,  1.9512e-01, -1.4366e-01, -2.1843e-01,  3.0800e-01,\n",
       "           1.4898e-01, -2.0374e-01, -1.7362e-01,  4.5321e-02, -6.5337e-02,\n",
       "          -3.3705e-01,  7.6587e-01,  8.3439e-02,  3.2863e-01,  2.8733e-01,\n",
       "           1.9077e-01,  1.6046e-01, -2.5924e-01, -7.2796e-02,  1.2824e-02,\n",
       "           1.9408e-01,  1.1055e-01, -3.9923e-01, -9.2941e-03,  1.7271e-01,\n",
       "           3.4020e-02, -2.4442e-01, -3.0432e-01, -1.8281e-01, -1.1009e-01,\n",
       "          -1.2536e-01,  9.9658e-02, -2.1388e-01, -5.4799e-02, -9.9604e-02,\n",
       "          -3.6038e-01, -6.1866e-01,  3.1311e-01, -4.1408e-02, -1.3373e-01,\n",
       "          -5.0103e-01,  9.4623e-02,  8.1636e-02,  9.7390e-02,  9.4125e-02,\n",
       "           1.2069e+00,  1.1763e-01,  3.3951e-01, -3.7124e-01, -4.1355e-01,\n",
       "          -1.8756e-02,  1.7005e-02, -3.0622e-01,  2.0387e-01,  4.7638e-02,\n",
       "           9.0825e-02, -6.4422e-03,  3.0895e-01, -7.0727e-02,  3.2299e-01,\n",
       "          -5.1035e-01,  5.9706e-01,  4.0296e-01, -1.5621e-01,  1.9248e-01,\n",
       "           4.1224e-02, -1.1810e+00,  1.1052e-02, -4.4094e-01,  1.5505e-01,\n",
       "          -5.8768e-02,  4.7948e-01,  2.5033e-01,  1.1854e-01, -1.5608e-02,\n",
       "           1.1388e-01,  3.8044e-01, -4.8714e-02,  2.7445e-01, -1.1333e-01,\n",
       "           3.2253e-01,  4.7746e-01,  5.6719e-01,  4.7582e-01, -3.9061e-02,\n",
       "          -4.9948e-01,  2.2380e-01,  1.5382e-01,  3.2683e-02,  3.6899e-01,\n",
       "          -2.5654e-03, -3.3135e-01, -1.0014e-01,  6.3539e-02, -2.7668e-02,\n",
       "          -3.0951e-01, -5.6530e-02, -6.1820e-01,  7.0682e-02, -4.2689e-01,\n",
       "           2.0664e-01, -1.5044e-01, -5.0726e-02,  4.7845e-01, -1.1923e-02,\n",
       "          -2.1850e-01,  2.9529e-01,  2.1139e-01, -2.7043e-01,  8.2472e-02,\n",
       "           4.0991e-01, -2.6044e-02,  5.9584e-02, -7.0890e-02, -1.5798e-01,\n",
       "           9.0615e-02, -6.7261e-01,  8.5116e-01,  1.6113e-01,  5.4400e-02,\n",
       "           1.2543e-01, -1.9401e-01,  9.1348e-02, -3.6870e-01,  3.2411e-01,\n",
       "          -3.6197e-01, -5.5697e-01, -6.6817e-01,  4.7312e-02,  1.2388e-01,\n",
       "           1.2510e-01, -2.5340e-01,  5.3352e-01,  5.0904e-01,  4.4488e-01,\n",
       "          -4.1766e-01, -1.1635e-01,  6.3818e-02, -8.1595e-03, -2.8025e-01,\n",
       "           2.3313e-01, -2.8311e-01, -7.0234e-01,  3.3389e-01, -1.6131e-01,\n",
       "           7.1695e-02, -7.8972e-02,  2.4941e-01,  2.1270e-01, -3.0694e-01,\n",
       "          -4.7729e-01,  3.5769e-01,  1.0930e-01, -4.1332e-01,  6.0194e-01,\n",
       "          -3.6031e-01,  2.2402e-01, -1.4453e-01, -3.3666e-01, -4.9519e-01,\n",
       "          -4.8349e-01,  2.7944e-02,  7.4915e-02, -2.4876e-01, -9.5608e-01,\n",
       "          -5.1778e-01,  4.2040e-01,  6.7138e-02,  6.9008e-02,  3.9415e-01,\n",
       "           3.8460e-01, -1.6499e-01,  2.7297e-02,  5.0415e-02, -1.8782e-01,\n",
       "          -4.9658e-02, -1.4824e-02, -2.2973e-01,  3.5559e-01,  4.3997e-01,\n",
       "           4.0542e-01,  3.0206e-01,  4.1686e-01, -1.8253e-01, -1.5494e-01,\n",
       "          -6.5838e-02,  6.7138e-02, -5.5045e-01, -2.4250e-01,  5.5037e-01,\n",
       "          -4.9501e-02,  5.4009e-01,  5.7184e-01,  2.5778e-01, -5.5419e-01,\n",
       "           2.1070e-01,  4.1244e-01, -2.3179e-01, -3.2351e-01,  3.3485e-01,\n",
       "          -8.4385e-01, -2.0555e-01,  3.2408e-02,  4.1960e-02, -3.6385e-01,\n",
       "           1.3844e-01,  3.6764e-01,  5.6891e-02, -1.3171e-01, -4.5433e-01,\n",
       "           5.0970e-01,  1.0212e-01,  3.5791e-02,  2.0244e-01, -3.3433e-01,\n",
       "          -8.6988e-02, -2.7966e-01, -1.8644e-01, -2.8909e-01, -2.9978e-01,\n",
       "           1.8666e-01,  7.4564e-02, -1.7243e-01, -2.0442e-01, -1.4553e-01,\n",
       "           3.5750e-01, -3.9184e-01, -4.2478e-01,  1.5349e-01, -2.0347e-01,\n",
       "          -1.7331e-01,  2.0739e-01, -2.9410e-01,  1.7163e-02,  1.7707e-01,\n",
       "           6.8914e-02, -2.9028e-01, -2.3587e-01, -1.0476e-01,  6.2999e-01,\n",
       "           4.4829e-01,  4.6236e-01, -2.7856e-01,  4.8862e-01,  1.9237e-01,\n",
       "          -2.2399e-01,  2.8042e-01, -4.3249e-02, -2.9295e-01, -3.8750e-01,\n",
       "           1.1223e-01,  4.3341e-01, -4.6060e-01,  1.1895e-01,  1.3917e-02,\n",
       "           8.9540e-02,  2.5481e-02, -2.2514e-01,  2.0132e-02, -1.0714e-01,\n",
       "          -3.0235e-01, -5.3475e-01,  1.7260e-01, -2.6659e-01, -2.3956e-01,\n",
       "           2.4620e-01,  8.3081e-02,  2.0021e-01, -2.3758e-01, -3.8331e-01,\n",
       "          -1.3272e-01, -6.9849e-02,  7.4554e-01,  4.6181e-02, -2.7416e-01,\n",
       "           2.4149e-01, -3.1745e-01, -2.1556e-01,  4.3037e-02, -4.0241e-01,\n",
       "          -4.5970e-01,  3.1018e-01, -2.1995e-01, -3.5326e-02,  4.5199e-01,\n",
       "          -2.4465e-01,  9.6050e-02, -1.5673e-01,  3.5403e-01,  9.0403e-02,\n",
       "          -4.6445e-01,  6.9774e-02,  2.2956e-01, -4.1718e-01,  3.3612e-01,\n",
       "           1.4700e-01,  4.0744e-01, -3.5080e-01, -1.6451e-03, -1.2415e-01,\n",
       "           2.0199e-01, -9.2008e-02, -3.7973e-01, -6.2503e-01, -3.8259e-01,\n",
       "           3.0653e-01,  4.9866e-01,  2.2824e-01,  4.0167e-01, -1.1256e+01,\n",
       "           3.1084e-01,  5.8849e-01, -5.3192e-01, -1.9577e-01, -9.8556e-02,\n",
       "           1.3651e-01,  7.0608e-02,  4.4634e-01, -5.0260e-01, -3.9525e-01,\n",
       "          -5.3885e-01, -2.0246e-01, -1.8836e-02, -3.6126e-01, -2.3081e-01,\n",
       "           7.2144e-02, -8.1524e-01,  4.4907e-01, -2.0450e-01,  1.9910e-01,\n",
       "          -1.1690e-02,  2.5559e-02,  1.0292e-01, -4.6065e-01, -1.9328e-02,\n",
       "           2.6624e-03, -8.2043e-02,  2.0747e-01,  4.7433e-01,  2.8398e-01,\n",
       "           5.9589e-02,  4.2495e-02, -2.0339e-01, -3.1901e-01, -4.6892e-01,\n",
       "          -2.2812e-01, -1.8286e-01, -1.5271e-01, -2.5336e-01, -5.6598e-01,\n",
       "          -8.2102e-01, -1.8349e-01,  4.0510e-01,  1.6226e-01,  7.4686e-01,\n",
       "           1.6349e-01,  7.8221e-01,  2.4770e-01,  3.1391e-02, -1.1747e-01,\n",
       "          -9.5922e-01,  1.8255e-02, -2.7873e-01, -5.1774e-01, -1.5369e-01,\n",
       "           4.8173e-01, -8.7572e-01,  3.3231e-01, -6.1631e-01, -7.3267e-02,\n",
       "           4.0848e-01,  2.0625e-01,  1.3357e-01, -2.0258e-01, -2.3780e-01,\n",
       "          -2.2324e-01, -1.9947e-01,  4.7445e-02, -1.8636e-01, -9.3720e-02,\n",
       "          -2.3776e-01,  5.4805e-01,  3.8466e-01, -2.0714e-01, -2.9304e-02,\n",
       "          -1.4520e-01, -3.6933e-01,  2.3732e+00,  8.1616e-02,  1.1870e-01,\n",
       "          -8.3821e-03,  4.4196e-01, -7.7242e-01, -7.2176e-01,  3.2259e-01,\n",
       "          -2.4966e-01,  4.0482e-01,  3.2415e-01,  2.1768e-01, -1.8780e-01,\n",
       "          -7.6924e-01, -3.0599e-01,  1.4089e-01,  1.2372e-01,  7.4673e-01,\n",
       "          -5.9262e-01, -4.2443e-02, -2.2946e-03,  2.0315e-01, -7.0646e-01,\n",
       "          -1.0937e-01, -4.5651e-01, -5.2140e-01,  3.0557e-01, -4.0541e-01,\n",
       "           1.0728e-01,  2.6007e-01, -2.7122e-01,  3.7101e-01, -4.5228e-01,\n",
       "           3.7925e-01,  4.9418e-01, -4.0276e-02,  3.0903e-02,  9.2051e-02,\n",
       "           6.9419e-01,  4.1433e-01, -3.8692e-01, -5.0032e-01,  6.7535e-03,\n",
       "           1.1810e-01,  2.2211e-01,  1.4978e-01, -3.1475e-01,  2.8136e-01,\n",
       "          -5.7135e-01, -1.7940e-01,  4.1815e-01,  1.3802e-01, -7.6783e-02,\n",
       "           3.3353e-02, -8.7527e-02,  4.8980e-01, -6.1702e-01, -5.1431e-01,\n",
       "          -8.1973e-01, -1.7623e-01,  4.2617e-01, -1.8986e-01,  2.5278e-03,\n",
       "          -4.1550e-01,  1.5920e-03, -1.1054e-01,  6.4886e-02,  1.5067e-01,\n",
       "           1.4876e-01,  2.7651e-01, -3.3909e-01,  6.9115e-02, -8.0341e-02,\n",
       "          -5.9108e-01, -2.5989e-02, -3.5374e-01, -1.7907e-01, -5.8507e-01,\n",
       "          -3.8331e-01, -6.5274e-02, -2.0200e-01,  6.3456e-02,  7.8083e-02,\n",
       "           2.4726e-01, -8.5035e-02, -4.9981e-01,  5.4679e-01,  7.0909e-01,\n",
       "           1.7255e-01, -4.0332e-01, -4.1311e-01, -2.0700e-01,  2.2032e-01,\n",
       "           2.0169e-01,  6.7456e-02,  8.3577e-02, -4.8579e-01, -5.3866e-01,\n",
       "          -1.3078e-01,  8.3069e-01,  6.6552e-02, -6.2269e-01,  3.5287e-01,\n",
       "           8.2926e-01, -4.2999e-03,  1.0470e-01, -1.4769e-01, -1.0905e-02,\n",
       "          -6.8002e-01, -4.5242e-01,  5.4889e-01, -3.3043e-01,  8.8498e-02,\n",
       "           2.6919e-01, -1.4538e-01, -1.3520e-02,  5.5307e-01, -6.7633e-03,\n",
       "          -2.9054e-01, -9.5423e-02, -3.5432e-01,  4.6172e-01,  2.5282e-01,\n",
       "          -1.7983e-01,  5.4675e-01,  2.4048e-01, -9.1556e-02, -2.2891e-01,\n",
       "           2.6058e-01,  8.2644e-01, -3.8717e-01, -5.3324e-02,  7.3629e-01,\n",
       "           1.8070e-01, -1.8857e-01,  2.4060e-01,  8.1942e-02, -1.1491e-01,\n",
       "           1.6143e-01,  8.3733e-02, -1.9284e-01,  7.0935e-02,  2.9230e-01,\n",
       "           4.1264e-01, -6.1457e-01, -7.9276e-01, -1.8001e-01,  9.8722e-01,\n",
       "          -2.7242e-03,  6.4823e-01, -4.8079e-02,  5.8855e-01,  5.8653e-01,\n",
       "          -1.0060e+00,  3.5877e-02, -4.4649e-01,  3.5878e-01,  2.1654e-01,\n",
       "          -9.3323e-02,  1.2367e-02, -1.8079e-01, -4.4975e-01,  6.3889e-01,\n",
       "           3.8220e-01, -9.3123e-01, -4.7922e-02,  7.6774e-01, -4.0160e-01,\n",
       "           5.2869e-01, -4.1642e-02,  5.3272e-02,  2.3037e-01,  2.5443e-01,\n",
       "          -1.2764e-01,  2.5209e-01, -6.3695e-01, -5.7503e-01, -1.6376e-01,\n",
       "           7.4119e-01,  2.1508e-01,  1.3206e-01, -2.0906e-01, -1.0338e-01,\n",
       "          -8.3088e-02, -5.7457e-01, -1.7732e-01,  1.3004e-01,  1.0950e-02,\n",
       "           4.7226e-02, -1.6124e-01, -1.4109e-01,  3.0634e-01, -1.4438e-01,\n",
       "           3.1713e-01,  3.4791e-01,  6.6171e-01, -9.7189e-02, -2.0056e-02,\n",
       "          -2.9369e-01,  2.2148e-01, -2.1916e-01,  3.8658e-01,  7.1078e-03,\n",
       "          -2.4818e-01,  2.1654e-01,  7.3740e-02, -5.2780e-01,  1.8656e-01,\n",
       "          -4.1056e-01,  6.6987e-01, -6.0303e-02, -2.8762e-01,  2.8514e-01,\n",
       "          -2.2890e-02,  6.2806e-02, -2.6061e-01,  3.9024e-01,  7.6619e-02,\n",
       "           1.2862e-01,  7.3918e-01, -3.9964e-02, -2.6331e-02, -7.6659e-01,\n",
       "           1.5460e-01, -6.7370e-01, -1.6320e-01,  1.2318e-01, -2.7990e-01,\n",
       "          -3.3913e-01, -1.3037e-01,  1.1407e-01,  1.0009e-01,  7.2576e-01,\n",
       "           8.8744e-01, -3.7540e-01, -7.0123e-01, -2.9989e-01,  8.9747e-02,\n",
       "           6.0510e-02, -9.0180e-02,  8.4200e-03,  2.3967e-02,  4.7692e-01,\n",
       "          -6.5268e-01, -2.3930e-01, -2.4694e-01,  3.1092e-01, -2.4689e-01,\n",
       "           3.5951e-01, -3.2686e-01,  9.0535e-02,  3.0054e-01,  2.6831e-01,\n",
       "           1.4601e-01,  5.2618e-01,  1.8094e-01, -4.6085e-01,  3.8492e-02,\n",
       "          -1.6138e-01, -4.2823e-01,  9.7934e-02, -4.9977e-01,  4.5377e-01,\n",
       "           7.1934e-02,  4.4028e-01,  7.9590e-01]], device='mps:0'),\n",
       " tensor([[-2.5187e-01, -4.0054e-01, -5.8007e-01, -8.3055e-01, -1.2868e-01,\n",
       "          -2.8804e-02, -7.1917e-01,  6.6701e-01,  1.2709e-01,  5.0855e-01,\n",
       "           1.2570e-01,  5.1916e-01,  5.1676e-02, -1.1521e-01, -1.6481e-01,\n",
       "          -4.9820e-01,  5.6222e-02,  3.7338e-01,  6.7187e-01, -2.4111e-02,\n",
       "           2.2844e-01,  5.7979e-01, -2.6323e-01, -1.2386e-02, -8.3013e-01,\n",
       "           2.9613e-01, -1.1576e-01,  2.2409e-01,  1.7718e-01, -6.3092e-01,\n",
       "           4.0999e-02,  4.7417e-01,  2.3068e-02,  1.3497e+00, -5.6080e-01,\n",
       "          -1.3230e-01,  8.2213e-02, -3.9453e-01, -2.3010e-01,  3.7283e-02,\n",
       "           6.1164e-01, -4.1315e-01,  1.3635e-01,  4.0254e-01, -6.1085e-02,\n",
       "           8.9076e-02,  1.3899e-01, -3.7744e-01,  2.5892e-01, -8.3961e-01,\n",
       "          -4.8318e-02, -2.0669e-01,  4.9144e-01,  6.2615e-01, -6.3176e-01,\n",
       "          -2.2975e-01,  2.7419e-01, -3.7716e-02, -3.3020e-02,  1.5349e-01,\n",
       "           9.7980e-02,  3.8582e-01, -2.9995e-01,  2.3257e-01,  3.3441e-01,\n",
       "           2.4458e-01, -3.0392e-01, -4.8889e-01, -5.4078e-01, -5.0111e-02,\n",
       "           9.1859e-03,  2.2490e-02,  1.4846e-01, -2.3272e-01,  4.1973e-01,\n",
       "          -7.1214e-01, -6.4354e-02, -8.9125e-02, -2.5973e-01,  2.9246e-01,\n",
       "          -4.5549e-02,  1.9719e-01,  3.1010e-01,  1.0268e-01,  1.2457e-01,\n",
       "           4.3419e-02,  2.7215e-01,  5.2932e-04,  4.7373e-02,  3.7533e-01,\n",
       "          -1.3375e-01, -8.1722e-02, -2.8313e-01, -3.5814e-01,  3.1229e-01,\n",
       "           4.0598e-02,  2.9907e-01, -1.9557e-01,  1.8579e-01, -3.2474e-01,\n",
       "           1.5588e-01,  1.7768e-01, -1.8083e-01,  5.7286e-02, -3.5172e-01,\n",
       "          -4.5475e-01,  2.3589e-02, -2.8569e-01, -4.8285e-01,  2.2697e-01,\n",
       "          -1.4211e-01, -3.0205e-01, -4.0922e-02, -3.8935e-01, -3.1524e-02,\n",
       "          -7.5164e-02,  2.6013e-01, -1.3230e-01, -3.4806e-01, -8.3033e-02,\n",
       "           2.0734e-01, -2.2808e-01,  1.6077e-01,  2.6882e-01,  1.2232e-01,\n",
       "          -9.6705e-02,  6.5251e-01,  1.8950e-02,  3.1674e-01,  2.5763e-01,\n",
       "           1.1699e-01, -6.5768e-02, -1.9745e-01, -2.3154e-01,  7.5988e-02,\n",
       "          -7.6941e-02,  1.0687e-01, -1.8182e-01, -2.2790e-01, -1.8508e-01,\n",
       "           3.1771e-01, -3.3097e-01, -6.0893e-01, -4.9260e-01, -1.6098e-01,\n",
       "          -1.6326e-01, -2.4804e-02, -8.2473e-03, -3.3693e-01, -3.1956e-01,\n",
       "          -6.7443e-01, -3.6717e-01,  8.1041e-01, -1.6831e-02, -3.3356e-01,\n",
       "          -1.6499e-01,  2.3208e-01,  3.1666e-01, -4.5959e-02,  3.7161e-01,\n",
       "           7.7250e-01, -4.7670e-02,  1.6786e-01, -5.1827e-01, -3.8195e-01,\n",
       "           4.8335e-02,  2.6658e-01, -2.7054e-01,  6.1879e-01,  4.8352e-02,\n",
       "           7.5627e-02, -6.6170e-02,  2.7430e-01, -3.2704e-01, -1.2324e-02,\n",
       "          -5.7573e-01,  3.7750e-01,  2.9539e-01, -3.9892e-01, -6.6567e-02,\n",
       "          -2.0834e-02, -1.2060e+00,  9.6964e-02, -8.3040e-01,  3.1236e-01,\n",
       "          -2.3110e-01,  1.7963e-01,  2.3389e-01,  2.9545e-01, -6.5320e-02,\n",
       "           2.5404e-01,  2.8725e-01,  2.5687e-01,  3.3609e-01, -3.4924e-02,\n",
       "           2.8019e-01,  2.3924e-01,  5.0790e-01,  4.1073e-01,  1.4769e-01,\n",
       "          -3.5440e-01,  2.9683e-01,  3.5781e-01,  9.0598e-02,  1.6954e-01,\n",
       "           9.9825e-02, -2.1636e-01,  2.1091e-01,  1.7501e-01, -1.6618e-01,\n",
       "          -2.2406e-01, -3.1100e-02, -4.6818e-01, -5.6655e-02, -4.8467e-01,\n",
       "           3.2557e-01, -5.1983e-01, -1.1912e-01,  3.4831e-01,  2.4340e-01,\n",
       "          -1.4896e-02,  1.8319e-01,  1.7196e-01, -2.4338e-01,  2.1735e-01,\n",
       "           3.7449e-01, -3.4618e-01, -3.9804e-02, -3.2197e-01, -2.9367e-01,\n",
       "           2.4357e-01, -5.6244e-01,  6.6635e-01,  4.0574e-01,  6.0165e-02,\n",
       "           1.5119e-01, -2.7920e-01, -5.1638e-02, -5.7408e-01,  3.5471e-01,\n",
       "          -3.2130e-01, -3.4955e-01, -5.4801e-01,  6.4487e-02,  4.2773e-01,\n",
       "           2.7565e-01, -6.4915e-02,  7.7868e-01,  4.2408e-01,  4.2813e-01,\n",
       "          -4.3003e-01, -1.5896e-02,  4.0377e-01, -4.1405e-02, -4.6059e-01,\n",
       "           6.7116e-02, -1.4962e-01, -5.1643e-01,  2.8000e-01, -2.6598e-01,\n",
       "          -1.2047e-01, -1.4913e-01,  3.3682e-01,  1.5365e-01, -2.1964e-01,\n",
       "          -3.8461e-01,  3.9587e-01, -1.4349e-01, -4.1723e-01,  4.7123e-01,\n",
       "          -3.5025e-01,  1.5209e-01,  1.0694e-01, -3.8441e-01, -6.2072e-01,\n",
       "          -3.2235e-01,  2.3090e-01,  1.5585e-01,  3.4799e-02, -6.6306e-01,\n",
       "          -4.0571e-01,  3.1197e-01,  2.4601e-01,  2.9976e-01,  2.9840e-01,\n",
       "           5.7427e-01, -3.0944e-01, -1.6702e-01,  1.4982e-01, -4.1377e-01,\n",
       "          -1.5704e-01, -3.3228e-02, -1.9995e-01,  2.9569e-01,  2.4768e-01,\n",
       "           1.1600e-01,  2.0367e-01,  4.3482e-01, -2.5272e-01, -3.8673e-01,\n",
       "          -7.1543e-02,  2.0161e-01, -9.3629e-02, -5.8306e-02,  2.3951e-01,\n",
       "          -6.3095e-02,  5.9650e-01,  7.2143e-01,  7.7439e-02, -6.4006e-01,\n",
       "           1.7696e-01,  5.2502e-01, -2.7253e-01, -4.0343e-02,  9.6365e-02,\n",
       "          -1.0344e+00, -2.1331e-01, -1.2883e-01, -2.0181e-01, -6.5352e-01,\n",
       "           8.8115e-02,  3.4863e-01,  4.6833e-03, -1.7864e-01, -3.0947e-01,\n",
       "           1.0448e-01,  4.8724e-02, -4.8980e-02,  4.3534e-01, -2.8901e-01,\n",
       "          -3.4797e-01, -2.3438e-01, -2.1532e-01, -5.4201e-01, -3.9968e-01,\n",
       "           3.5914e-01,  2.8129e-01,  1.1986e-01, -2.6486e-01, -5.1832e-01,\n",
       "           4.5577e-01, -6.8169e-01, -1.4632e-01,  2.4840e-01, -4.7702e-01,\n",
       "           1.4650e-02,  4.8990e-01,  1.4999e-01,  1.6336e-01,  6.5258e-02,\n",
       "           7.8557e-02, -3.3903e-01, -5.0030e-01, -1.5000e-01,  6.2323e-01,\n",
       "           3.2283e-01,  3.3506e-01, -1.7236e-01,  3.9096e-01,  4.5167e-01,\n",
       "          -2.9582e-01,  4.1670e-01,  1.0930e-02,  7.7439e-02, -6.0414e-02,\n",
       "           1.9048e-01,  3.4370e-01, -3.6842e-01,  2.0145e-01,  1.7929e-01,\n",
       "          -3.1634e-01, -3.2149e-01,  1.5691e-02,  2.0642e-01, -1.8580e-01,\n",
       "          -3.0875e-01, -2.0361e-01,  9.9698e-02, -3.4856e-01, -4.3104e-01,\n",
       "           2.9161e-01, -3.6543e-02,  7.0495e-02, -4.0357e-01, -4.3462e-01,\n",
       "          -2.0181e-01, -2.6055e-01,  6.8681e-01,  2.1064e-01, -1.9621e-01,\n",
       "           1.8720e-01, -1.0907e-01, -3.6330e-01,  4.7847e-01,  2.0162e-02,\n",
       "          -4.8754e-01,  4.5482e-01, -3.4822e-01,  1.0627e-01,  5.5756e-01,\n",
       "          -1.6769e-01, -1.0840e-01,  1.1008e-01,  1.0820e-01,  2.9026e-01,\n",
       "          -3.9884e-01,  1.5516e-01, -6.3476e-02, -3.1885e-01,  3.1820e-01,\n",
       "           2.6411e-01,  2.2781e-01, -1.7137e-01, -3.3426e-02,  3.5677e-02,\n",
       "           2.8782e-01, -6.0853e-02, -3.9888e-01, -4.7347e-01, -4.6132e-01,\n",
       "           1.2832e-01,  1.8927e-01,  5.6207e-01,  3.4144e-01, -1.1394e+01,\n",
       "           1.2938e-01,  7.8031e-01, -4.0713e-01, -8.2819e-02, -5.0320e-02,\n",
       "           1.0976e-02, -1.3708e-01,  4.2955e-01, -2.7734e-01, -4.4105e-01,\n",
       "          -3.5727e-01, -1.3308e-01,  1.5454e-01, -3.2189e-01, -2.6945e-01,\n",
       "          -1.4686e-02, -7.0935e-01,  4.0963e-01, -3.0286e-01,  5.0298e-01,\n",
       "          -2.3511e-01, -8.3798e-02,  1.5816e-01, -3.5851e-01, -1.3439e-02,\n",
       "           5.9766e-02, -1.6033e-01,  1.4651e-01,  4.2486e-01, -5.2072e-02,\n",
       "           3.3244e-02, -4.1258e-03,  4.8461e-02, -2.5723e-01, -5.9999e-01,\n",
       "           1.9296e-02, -1.4141e-01,  3.3422e-01, -4.4133e-02, -7.1265e-01,\n",
       "          -2.2533e-01, -2.7418e-01,  3.1407e-01,  2.7021e-01,  9.0598e-01,\n",
       "           2.2955e-01,  1.0686e+00,  1.5022e-01,  1.7748e-01,  2.6230e-02,\n",
       "          -7.8764e-01, -3.3269e-02, -4.1907e-01,  2.3280e-02,  1.0114e-01,\n",
       "           3.6576e-01, -7.9367e-01,  2.5251e-01, -3.5521e-01,  3.6487e-02,\n",
       "           4.9094e-01,  3.1768e-02,  2.7700e-01, -3.2298e-02, -7.8123e-02,\n",
       "           1.2089e-01, -2.8544e-02,  1.1098e-01, -2.2583e-01, -2.4084e-01,\n",
       "          -1.0177e-01,  5.9888e-01,  3.9592e-01, -7.2211e-01,  3.6112e-01,\n",
       "           4.9258e-02, -1.8622e-01,  1.4556e+00, -4.8573e-02,  1.8361e-02,\n",
       "           1.9800e-03,  2.9604e-01, -5.0236e-01, -7.3516e-01,  3.6511e-01,\n",
       "          -1.0211e-01,  5.6427e-02,  1.0127e-01,  4.8148e-01, -9.7481e-02,\n",
       "          -4.7940e-01, -1.8593e-03,  1.0556e-01, -1.3380e-01,  5.4710e-01,\n",
       "          -7.2824e-01, -1.6963e-01, -1.1223e-01,  3.1723e-01, -8.4054e-01,\n",
       "          -6.1126e-02, -7.0210e-01, -3.9328e-01,  5.0586e-01, -2.5896e-01,\n",
       "           1.5852e-01, -2.7443e-02, -4.3439e-02,  5.5574e-01, -2.1489e-01,\n",
       "           7.1555e-02,  2.2813e-01,  8.4995e-02, -1.6733e-01,  3.9349e-02,\n",
       "           4.1630e-01,  4.9571e-01, -4.2740e-01, -6.3911e-01, -7.6901e-03,\n",
       "           1.7180e-01,  1.4933e-01,  3.4019e-01, -5.3463e-01,  3.6431e-01,\n",
       "          -7.9162e-01, -1.8769e-01,  7.2949e-02, -4.3273e-02,  3.6175e-01,\n",
       "           2.3925e-01,  7.2038e-02,  2.2256e-01, -7.7756e-01, -6.4498e-01,\n",
       "          -6.5932e-01, -2.1445e-01,  2.6143e-01, -5.4236e-01, -8.3654e-02,\n",
       "          -1.1033e-01, -4.8837e-03, -5.0292e-02, -6.6102e-02,  1.1939e-02,\n",
       "           1.3632e-01,  2.3587e-01, -1.3023e-01, -1.3178e-01, -9.2464e-02,\n",
       "          -5.5566e-01, -1.1753e-01, -9.9591e-02, -1.2505e-01, -5.6783e-01,\n",
       "          -1.3797e-01,  3.3075e-02, -2.5373e-01, -6.8124e-02,  3.5608e-01,\n",
       "          -2.9832e-02, -2.7856e-02, -4.8850e-01,  7.2846e-01,  4.0279e-01,\n",
       "           5.2088e-02, -2.8298e-01, -5.5189e-01, -1.0567e-01,  2.9592e-01,\n",
       "           1.9555e-01, -6.7007e-02, -1.7709e-01, -2.4145e-01, -4.2522e-01,\n",
       "           1.4302e-01,  7.1147e-01, -6.6296e-02, -6.8115e-01,  2.6621e-01,\n",
       "           5.8223e-01, -1.2882e-01, -2.4432e-01,  2.0983e-02, -9.3284e-02,\n",
       "          -3.1851e-01, -6.0317e-01,  7.8821e-01, -1.1246e-01, -5.8638e-02,\n",
       "           2.2427e-01,  2.0359e-01, -9.2588e-03,  4.0209e-01, -1.5770e-01,\n",
       "          -3.5434e-01, -1.6477e-03, -7.3600e-01,  4.3257e-01,  6.1666e-02,\n",
       "          -3.1489e-01,  7.6082e-01,  2.7260e-01, -6.0593e-02, -4.5779e-02,\n",
       "           2.2863e-01,  7.9941e-01, -2.4636e-01,  6.4653e-02,  5.3779e-01,\n",
       "           8.8110e-02, -1.1229e-01,  3.9580e-02,  4.1665e-02,  4.5410e-02,\n",
       "           3.7012e-01, -3.2145e-02, -4.1564e-01, -6.1194e-02,  3.8924e-01,\n",
       "           3.9295e-01, -5.4610e-01, -7.3487e-01,  2.2333e-02,  6.8551e-01,\n",
       "          -2.9054e-01,  2.3654e-01, -3.2708e-02,  3.0880e-01,  1.2001e-01,\n",
       "          -9.2160e-01, -1.5016e-01, -1.0056e-01,  2.8229e-01, -2.5482e-02,\n",
       "          -1.6139e-01, -1.4727e-01, -2.3686e-01, -1.6208e-01,  6.8308e-01,\n",
       "           2.1991e-01, -6.9160e-01,  3.4058e-02,  5.7281e-01, -6.0551e-01,\n",
       "           4.6052e-01, -2.4577e-01, -3.4302e-02,  3.6435e-01,  1.3117e-01,\n",
       "           4.9067e-02,  1.7856e-01, -4.2138e-01, -4.0598e-01,  1.7824e-01,\n",
       "           9.1672e-01, -7.6613e-02,  1.1166e-01, -3.9384e-01, -1.7182e-01,\n",
       "          -7.1855e-02, -3.1913e-01, -3.1286e-01,  3.7185e-01,  2.0529e-01,\n",
       "          -3.2552e-02, -1.3271e-01, -2.2992e-01,  3.0816e-01,  1.1821e-01,\n",
       "           3.2536e-01,  5.8187e-01,  6.4368e-01, -1.0077e-02, -2.4936e-01,\n",
       "          -6.1235e-02,  1.4568e-01, -1.9524e-01,  5.2704e-01, -2.0627e-01,\n",
       "          -3.3453e-01, -3.8069e-03,  3.0134e-01, -6.1681e-01, -5.8725e-02,\n",
       "          -4.5426e-01,  4.2361e-01, -1.6955e-01,  3.7868e-02,  2.8833e-01,\n",
       "           4.0522e-02,  5.7925e-02, -2.2651e-01,  5.5807e-01,  4.4511e-01,\n",
       "           1.6609e-01,  4.5662e-01,  1.6104e-01,  4.5272e-02, -5.1597e-01,\n",
       "           2.1978e-01, -1.0213e+00, -9.4260e-02, -4.6640e-02, -1.5401e-01,\n",
       "          -4.4224e-01, -2.8126e-02,  1.6814e-01,  1.8538e-01,  6.9597e-01,\n",
       "           7.2967e-01, -3.0871e-01, -4.5290e-01, -1.8510e-01,  1.5304e-01,\n",
       "           2.7705e-01, -1.9517e-01,  3.5971e-01,  9.6103e-02,  2.4432e-01,\n",
       "          -5.0967e-01, -4.3165e-01, -4.2252e-02,  3.1411e-01, -2.0171e-01,\n",
       "           5.7118e-01, -2.0164e-01, -1.3863e-01,  9.0832e-02,  1.3770e-01,\n",
       "           1.6639e-01,  7.9664e-01,  4.3416e-01, -6.3425e-01,  8.7469e-02,\n",
       "          -1.3980e-02, -2.9988e-01,  2.5983e-02, -4.3577e-01,  2.8858e-01,\n",
       "           1.2984e-01,  4.9280e-01,  6.7504e-01]], device='mps:0'),\n",
       " tensor([[-0.1672, -0.5249, -0.4540,  ...,  0.3771,  0.2016,  0.5340],\n",
       "         [-0.3292, -0.2590, -0.5016,  ..., -0.0148,  0.3509,  0.1607]],\n",
       "        device='mps:0'),\n",
       " tensor([[-5.6844e-01, -2.5393e-01, -5.0750e-01, -8.7244e-01, -3.2844e-01,\n",
       "          -3.6569e-01, -9.0136e-01,  7.7320e-01,  8.1617e-02,  4.9008e-01,\n",
       "           9.4442e-02,  1.7855e-01,  2.8632e-01, -1.8449e-01, -5.2880e-01,\n",
       "          -3.7395e-01,  1.3400e-01,  3.8131e-01,  3.5838e-01,  1.0985e-01,\n",
       "           5.8069e-01,  1.8646e-01, -1.6628e-01,  6.8736e-02, -6.3904e-01,\n",
       "           2.5407e-01,  2.1383e-01,  3.1362e-02,  1.9298e-01, -8.6934e-01,\n",
       "           1.5436e-01,  4.0132e-01,  1.4270e-01,  9.5617e-01, -2.6119e-01,\n",
       "          -1.3691e-01,  7.3087e-02, -4.0375e-01,  5.6589e-01,  4.6931e-01,\n",
       "           2.6016e-01, -4.5020e-01,  6.3035e-03,  1.8017e-01, -1.3233e-01,\n",
       "           7.1244e-02, -6.5158e-02, -6.6997e-01,  4.3002e-01, -6.2951e-01,\n",
       "          -4.0576e-01, -1.7096e-01,  6.4477e-01,  4.3165e-01, -9.5134e-01,\n",
       "          -5.1421e-01,  3.6650e-01,  3.7034e-01, -6.0353e-01,  2.5449e-01,\n",
       "          -5.6246e-02,  7.4130e-01,  3.2109e-01, -3.4573e-03,  4.6800e-01,\n",
       "           2.8329e-01, -2.3211e-01, -6.1568e-01, -1.7770e-01, -1.6986e-01,\n",
       "          -1.4760e-02, -1.5805e-01, -4.9675e-01, -1.8398e-01,  5.6503e-01,\n",
       "           1.4111e-01,  1.8994e-01,  6.2489e-01, -3.7451e-01,  1.5743e-01,\n",
       "           1.1599e-01,  1.8623e-01,  9.5090e-02,  1.9397e-02,  2.6867e-01,\n",
       "           3.0676e-01,  6.6873e-01,  1.7461e-01, -4.1899e-02,  7.6356e-03,\n",
       "           4.0632e-01, -1.3571e-01, -1.9968e-01, -9.0196e-01,  3.4882e-01,\n",
       "          -2.3752e-01,  2.9844e-01, -3.2413e-01,  2.0613e-01, -1.9357e-01,\n",
       "           2.7677e-01,  4.7619e-01, -3.1541e-01, -2.9064e-02, -3.1801e-01,\n",
       "          -7.2487e-01,  2.4777e-01,  1.3619e-01, -4.8521e-01,  2.4988e-01,\n",
       "           6.3011e-02, -2.1547e-01,  4.0180e-02, -1.5635e-01, -2.3985e-01,\n",
       "          -1.3972e-01,  2.8368e-01, -1.5412e-01, -6.8293e-01,  3.4060e-01,\n",
       "           1.7510e-01, -3.9404e-01, -3.4872e-01, -7.5550e-02,  2.1179e-01,\n",
       "          -3.4693e-01,  4.3310e-01,  5.8511e-02,  2.1442e-01,  7.8144e-01,\n",
       "          -2.3282e-03,  1.7204e-01, -1.7537e-01, -2.0757e-01, -3.1992e-01,\n",
       "           1.2913e-01,  1.1688e-01, -9.7004e-02,  7.2642e-02,  8.6313e-02,\n",
       "          -2.5738e-01, -2.4418e-01, -7.5146e-01, -5.0822e-01, -2.4911e-01,\n",
       "          -4.5583e-01,  2.1359e-01, -4.9226e-02, -4.0862e-02, -4.1612e-01,\n",
       "          -7.7786e-01, -5.3849e-01,  7.3452e-01,  2.2917e-02, -5.7738e-02,\n",
       "          -3.4818e-01,  1.8976e-01,  3.5786e-01, -6.8795e-02,  5.2628e-01,\n",
       "           4.0225e-01,  2.8456e-01,  1.1433e-01, -5.6339e-01, -1.9941e-01,\n",
       "           8.1964e-02, -2.7762e-01, -3.5982e-01,  1.1957e-01,  3.0120e-01,\n",
       "           7.2749e-02, -1.9488e-01,  1.1217e-01, -6.4809e-01, -1.6623e-01,\n",
       "          -6.0746e-01,  4.2744e-01,  2.4433e-01, -4.7229e-01, -1.0579e-01,\n",
       "          -3.3341e-01, -1.6541e+00,  2.9410e-01, -7.4445e-01, -2.2061e-01,\n",
       "          -9.1557e-01,  2.5267e-02,  2.7081e-01,  2.3996e-01,  1.7481e-01,\n",
       "           3.0564e-02,  1.9829e-01, -2.6402e-02,  3.0950e-01,  1.3367e-01,\n",
       "           2.1250e-01,  4.2880e-01, -3.5856e-02, -3.1493e-01,  1.7774e-01,\n",
       "          -2.9625e-01,  3.1158e-01, -2.8723e-02,  1.6269e-01,  3.5764e-01,\n",
       "           3.8199e-01, -6.8891e-02,  2.9418e-01,  1.9944e-01, -2.8504e-01,\n",
       "           7.7662e-02, -2.0543e-01, -3.6897e-01,  1.2753e-02, -1.4946e-01,\n",
       "           5.7360e-02, -4.4640e-01,  6.0071e-03,  2.8395e-01,  1.4853e-01,\n",
       "          -1.1382e-01,  2.8209e-01, -2.7212e-01, -4.6728e-01,  5.0678e-02,\n",
       "           5.0747e-01, -2.5028e-01,  4.1382e-01, -4.3106e-01,  1.1839e-01,\n",
       "          -2.8436e-01, -4.3520e-01,  5.3732e-01,  6.0607e-01,  2.4345e-01,\n",
       "          -1.8731e-01, -3.4659e-01,  3.4032e-01, -1.7321e-01, -3.5382e-01,\n",
       "          -2.7126e-01, -3.0751e-01, -6.6050e-02, -1.4941e-01,  6.6002e-01,\n",
       "           1.4670e-01, -3.1667e-02,  1.0202e+00,  6.9238e-01,  2.0488e-01,\n",
       "          -4.7313e-02, -1.3113e-01,  3.3900e-01,  2.8843e-01, -4.0147e-01,\n",
       "          -2.2174e-01, -4.5826e-02, -5.3387e-01,  5.6696e-01, -2.8629e-01,\n",
       "           3.1854e-01,  8.9703e-02,  9.8121e-02,  7.2426e-01,  4.0746e-01,\n",
       "          -5.5773e-01, -1.0385e-02, -3.8020e-02, -2.2596e-01, -8.8121e-02,\n",
       "           5.8970e-04,  2.4156e-01, -4.0000e-01, -2.9996e-01, -7.2638e-01,\n",
       "          -2.3492e-01, -2.1963e-01,  1.6584e-01,  3.0846e-01, -6.2015e-01,\n",
       "          -2.4234e-01,  3.6705e-01, -2.8687e-01,  2.0502e-01,  2.5558e-01,\n",
       "           4.2903e-01, -3.4778e-02, -9.3549e-02, -2.0382e-01, -7.2037e-02,\n",
       "          -7.4568e-02,  1.1676e-01, -2.9530e-01,  2.0986e-02,  5.1684e-03,\n",
       "          -1.1136e-01,  4.3742e-01,  3.4734e-01,  7.8739e-02, -6.1562e-01,\n",
       "           4.4871e-02, -1.5161e-01, -2.5925e-01, -2.2698e-01, -3.9915e-02,\n",
       "          -3.3285e-01,  3.5655e-01,  8.4436e-01, -4.9563e-02, -2.6343e-01,\n",
       "           1.9915e-01,  5.1512e-01, -1.3016e-01,  1.4426e-01, -2.5021e-02,\n",
       "          -9.0834e-01, -2.5524e-01, -1.9994e-01, -1.2356e-01, -4.6987e-01,\n",
       "           1.1757e-01,  6.0037e-01, -2.2855e-01, -3.6448e-01, -6.8986e-01,\n",
       "           4.8317e-01,  1.6688e-01,  3.3535e-01,  1.3286e-01, -2.9904e-01,\n",
       "          -4.7916e-02, -8.2334e-01,  1.2087e-01, -5.1199e-01, -4.3309e-01,\n",
       "           2.3617e-01,  2.8534e-01,  4.3335e-02,  3.0027e-01, -4.8130e-01,\n",
       "           6.8084e-01, -2.2633e-01, -1.7373e-01,  3.8647e-01, -3.8197e-01,\n",
       "          -2.4056e-01,  5.5692e-01, -1.3181e-01,  3.6477e-01,  5.5141e-02,\n",
       "           5.7557e-01, -2.1858e-01, -5.1305e-01, -1.2626e-01,  4.3394e-01,\n",
       "           2.2980e-01,  3.8016e-01, -2.2147e-01,  6.9901e-02,  2.4774e-01,\n",
       "          -3.1906e-01,  5.5489e-01, -3.6133e-01, -4.8930e-01,  1.1730e-01,\n",
       "           3.6468e-01,  1.4321e-01, -3.3744e-01, -1.9847e-01,  5.3661e-01,\n",
       "           6.1608e-02, -2.2447e-02,  3.8687e-01,  4.1707e-01,  7.5413e-02,\n",
       "          -4.1578e-01, -5.8536e-01,  1.2953e-01, -3.9358e-01, -2.0976e-01,\n",
       "           3.5414e-01, -2.8840e-01,  2.8654e-01, -3.9533e-01, -4.0960e-01,\n",
       "          -4.1113e-01,  1.3799e-01,  6.9650e-01,  2.6831e-01,  7.2333e-02,\n",
       "           2.6272e-01, -4.3981e-02, -2.9518e-01,  5.3852e-01, -1.2421e-01,\n",
       "          -3.4740e-01,  3.1586e-01, -2.1424e-01,  5.4590e-02,  2.9407e-01,\n",
       "          -2.2166e-01,  1.6948e-01,  2.2397e-01,  4.9950e-01, -3.4496e-01,\n",
       "          -2.7747e-01, -1.1121e-01, -2.1886e-01, -1.9180e-01,  1.6433e-01,\n",
       "           1.2031e-01,  5.6576e-01, -1.2635e-01, -7.3580e-02,  5.6376e-01,\n",
       "           1.4494e-02, -2.4303e-01,  1.4364e-01, -5.4605e-01, -4.0354e-01,\n",
       "           8.3474e-03,  1.2952e-01,  1.1459e-01,  3.5272e-01, -1.0831e+01,\n",
       "          -1.4639e-01,  7.7977e-01, -4.0763e-01, -9.5451e-03,  6.6180e-02,\n",
       "           1.1330e-01, -1.9354e-02,  2.2326e-01, -2.0652e-02, -6.5077e-01,\n",
       "          -4.5718e-01, -2.8242e-01,  1.1227e-01,  1.4893e-01, -3.7762e-01,\n",
       "           3.6570e-02, -7.7520e-01, -2.6453e-02, -5.8563e-01,  1.1856e-01,\n",
       "          -3.1038e-01,  1.1832e-01,  2.6041e-01, -4.3650e-01, -1.8852e-01,\n",
       "           1.1934e-01, -1.7524e-01,  9.0805e-02,  8.6730e-02, -1.5353e-01,\n",
       "          -1.3314e-01, -8.9663e-02,  1.7702e-01, -1.7022e-01, -7.7946e-01,\n",
       "          -4.5533e-02, -1.0732e-02,  5.5088e-01, -1.8145e-01, -7.5737e-01,\n",
       "          -5.5589e-01, -7.2691e-01,  4.4681e-01,  4.6576e-01,  8.4337e-01,\n",
       "           7.3694e-01,  7.8225e-01,  2.8821e-01,  1.1654e-01, -7.8781e-03,\n",
       "          -4.5251e-01,  7.6623e-02, -5.4234e-01, -2.0190e-01, -1.6733e-01,\n",
       "           3.1524e-01, -7.4778e-01,  5.4153e-01, -3.3615e-01, -2.3183e-01,\n",
       "           2.1240e-01,  3.8045e-02,  2.9980e-01, -1.5400e-01, -5.7548e-01,\n",
       "           6.9551e-01, -8.1846e-02,  8.7726e-02, -3.6636e-01,  1.2998e-01,\n",
       "          -1.1565e-01,  4.2252e-01,  4.9964e-01, -4.2858e-01,  5.0750e-01,\n",
       "          -2.9582e-02, -3.5795e-01,  1.5328e+00, -2.8105e-01,  3.4889e-01,\n",
       "          -3.9615e-02,  1.9863e-01, -5.1251e-01, -7.2759e-01,  5.4114e-02,\n",
       "           7.7376e-03, -5.2419e-02,  5.1108e-01,  3.0878e-01, -3.2255e-01,\n",
       "          -7.7520e-01,  1.9643e-01,  1.9483e-01,  4.1738e-01,  8.3253e-01,\n",
       "          -3.7144e-01, -2.2433e-01,  1.3851e-01,  4.9568e-02, -9.1221e-01,\n",
       "           1.5130e-03, -4.3752e-01, -3.9226e-01,  4.8852e-01,  4.2343e-02,\n",
       "           5.0267e-01,  1.8224e-01, -5.7171e-03,  5.0501e-01, -4.4502e-01,\n",
       "           3.0443e-01, -2.1046e-01,  1.4119e-01, -4.2433e-01,  1.0247e-01,\n",
       "           3.6354e-02,  1.8873e-01, -2.9864e-01, -4.8747e-01,  1.9256e-02,\n",
       "           5.4759e-03,  1.6032e-01,  4.2622e-01, -5.6251e-01,  5.0070e-01,\n",
       "          -4.8306e-01, -6.8047e-03,  1.4265e-01, -2.3568e-01,  7.4557e-01,\n",
       "          -1.8767e-01, -3.2433e-01, -1.5232e-01, -7.4690e-01, -5.0003e-01,\n",
       "          -5.5172e-01, -1.4172e-01,  1.9040e-01, -9.0813e-01, -1.5122e-02,\n",
       "           1.7257e-01,  3.3071e-01, -3.2582e-01,  2.3588e-01,  3.5528e-01,\n",
       "          -2.3571e-02,  5.9669e-01, -4.8350e-03,  2.2136e-01, -1.7667e-01,\n",
       "          -5.2245e-01,  5.5999e-03,  1.7072e-01,  1.7848e-02, -1.0474e+00,\n",
       "          -2.8334e-01, -8.6200e-02, -4.5816e-01,  3.3449e-01,  2.6597e-02,\n",
       "          -7.7005e-02, -1.1522e-01, -4.6150e-01,  4.5910e-01,  5.7866e-01,\n",
       "           2.6352e-01, -1.0573e-02, -2.1847e-01,  6.8300e-02,  4.7001e-01,\n",
       "           3.7492e-02, -2.0095e-01, -3.8934e-01, -3.0347e-01,  2.2958e-01,\n",
       "           3.3929e-01,  5.1629e-01, -4.2074e-01, -7.4347e-01,  2.6759e-01,\n",
       "           5.2226e-01,  1.6757e-01, -1.0236e-01, -2.4258e-01, -2.6495e-01,\n",
       "          -2.6086e-01, -2.0432e-01,  4.2878e-01, -2.3748e-01,  1.3720e-01,\n",
       "           1.2240e-01,  6.0109e-01,  3.4140e-01,  4.1248e-01, -7.8067e-01,\n",
       "          -2.4939e-01, -2.0628e-02, -9.6719e-01,  1.4958e-01,  1.0040e-01,\n",
       "          -9.1728e-02,  4.3231e-01, -1.2503e-01, -1.5137e-01,  2.4529e-01,\n",
       "           5.3670e-01,  6.4405e-01,  1.3705e-01,  2.8725e-01,  8.6653e-01,\n",
       "           4.5630e-01,  7.6329e-02,  5.2303e-02,  3.7178e-01,  5.7313e-02,\n",
       "           2.7836e-01, -8.3007e-02, -3.8735e-01, -1.3449e-01,  1.5554e-01,\n",
       "           5.8080e-01, -1.1659e+00, -4.9298e-01, -2.5056e-01,  3.4759e-01,\n",
       "          -4.3638e-01,  1.2396e-01, -5.5119e-02,  3.2727e-01,  7.0494e-01,\n",
       "          -1.2319e+00, -4.9664e-01,  4.3808e-01,  2.5547e-01,  6.9514e-03,\n",
       "          -3.5297e-01,  1.8096e-01, -5.0733e-01, -2.1436e-01,  6.9160e-01,\n",
       "           3.2753e-01, -8.7155e-01,  3.8160e-02,  5.9753e-01, -5.1593e-01,\n",
       "           1.8231e-01,  1.0520e-02,  1.2073e-01, -3.5920e-02,  2.1339e-01,\n",
       "          -1.0837e-01,  3.6255e-01, -5.6160e-01, -4.2000e-01, -4.6695e-02,\n",
       "           5.9002e-01,  9.8017e-02,  2.4896e-01, -1.9997e-01,  3.5892e-01,\n",
       "          -2.7904e-01, -3.0208e-01, -2.4714e-01,  2.8074e-02, -2.1321e-01,\n",
       "           2.8785e-01, -1.3190e-01, -1.8593e-01,  2.6220e-01, -1.4457e-01,\n",
       "           5.4373e-01,  4.6093e-01,  6.4338e-01,  9.0278e-02, -1.7838e-01,\n",
       "           2.4395e-01,  6.3504e-02, -3.0764e-01,  5.1212e-01,  1.4861e-02,\n",
       "          -4.7605e-01,  1.3018e-01,  6.3130e-02,  5.3125e-02, -1.4310e-01,\n",
       "          -1.8210e-01,  2.8353e-01,  4.7433e-03, -3.4418e-01,  2.0436e-01,\n",
       "           3.2605e-01,  2.2278e-01, -5.5433e-01,  2.7091e-01,  3.4276e-01,\n",
       "          -1.6297e-02,  1.6658e-02, -1.3971e-01,  1.2699e-01, -6.8731e-01,\n",
       "           2.1912e-01, -7.5659e-01, -2.1822e-01,  2.2174e-01, -1.2384e-01,\n",
       "          -4.9953e-01,  3.5551e-02, -1.5131e-01,  3.5066e-03,  8.9862e-01,\n",
       "           8.9985e-01, -1.2377e-01, -4.9015e-01, -1.2054e-01,  2.6347e-01,\n",
       "          -5.6200e-02, -2.3647e-01,  2.6229e-01, -5.1455e-02, -8.3190e-02,\n",
       "          -1.2290e+00, -2.8710e-01,  2.0861e-01,  7.9545e-02, -1.8979e-01,\n",
       "           1.0801e+00, -2.4190e-01,  1.2983e-01,  6.1443e-01,  2.0776e-01,\n",
       "           3.0221e-01,  4.1798e-01,  1.9241e-02, -3.1289e-01, -1.1601e-01,\n",
       "           1.7249e-01, -6.4133e-02, -1.7759e-01, -5.8779e-01,  2.5980e-01,\n",
       "           7.9414e-02,  4.5242e-01,  6.2170e-01]], device='mps:0'),\n",
       " tensor([[-2.1164e-01, -5.0199e-01, -3.6347e-01, -4.3150e-01, -3.0235e-01,\n",
       "           1.4287e-01, -2.1504e-01,  3.5374e-01, -1.6641e-01,  8.7638e-01,\n",
       "           1.8797e-01, -4.8590e-02, -8.3879e-02, -3.6532e-01, -8.5629e-01,\n",
       "          -5.5877e-01,  4.3222e-01,  5.9843e-01,  6.2961e-01,  4.8914e-01,\n",
       "           5.9952e-01,  3.6446e-01, -7.7855e-02, -1.2031e-01, -8.0130e-01,\n",
       "           4.6873e-01, -1.9059e-01,  1.7383e-01,  3.7135e-01, -6.9845e-01,\n",
       "          -2.3926e-01,  3.6487e-01, -9.9332e-02,  1.0880e+00, -2.4081e-01,\n",
       "          -3.5048e-01,  3.1765e-01, -9.9221e-01,  6.1385e-01, -5.5277e-02,\n",
       "           6.9279e-01,  9.8489e-02,  7.5006e-02,  1.6411e-01,  2.9388e-01,\n",
       "          -2.5501e-01, -2.1415e-02, -5.4804e-01, -1.0120e-01, -3.6309e-01,\n",
       "           3.4990e-02,  1.4336e-01,  5.2420e-01,  5.7319e-01, -3.2484e-01,\n",
       "          -1.6109e-01, -3.2739e-02,  9.6116e-02, -3.2239e-01,  2.3708e-01,\n",
       "          -2.3932e-01,  5.1576e-01,  2.5251e-01, -1.5248e-01, -9.5999e-02,\n",
       "           1.8892e-01, -2.6685e-01, -6.7739e-01, -2.7298e-01,  9.2939e-02,\n",
       "          -4.8149e-02,  5.0634e-01, -1.5515e-01, -2.6953e-01,  3.6435e-01,\n",
       "           1.3303e-01, -1.4162e-02, -1.8521e-01, -5.9583e-01, -1.8286e-01,\n",
       "          -2.8189e-02,  2.5063e-01, -2.8633e-01, -5.2884e-01,  4.3032e-02,\n",
       "           3.0496e-01,  9.4705e-01, -1.8129e-02,  1.2148e-01, -2.5945e-01,\n",
       "           4.1877e-01, -2.5901e-01, -6.5867e-01, -4.1669e-01,  1.6457e-01,\n",
       "          -2.6165e-01,  2.0734e-01, -4.6761e-01, -4.0110e-02, -1.6509e-01,\n",
       "           1.6259e-01,  1.7397e-01, -5.7886e-02,  2.4347e-01, -3.9437e-01,\n",
       "          -6.0696e-01,  2.6213e-01, -7.3358e-02, -7.5732e-01, -2.3183e-02,\n",
       "          -6.1886e-02,  3.2554e-01,  1.4182e-01, -1.1500e-01, -1.4344e-01,\n",
       "          -3.2843e-01,  3.5556e-01,  5.5440e-02, -1.8305e-01,  7.3335e-01,\n",
       "           4.5324e-01, -5.6244e-01, -6.9024e-02,  1.0657e-01,  3.4334e-01,\n",
       "          -4.5030e-01,  5.3498e-01, -1.2508e-01,  3.9350e-01,  3.2923e-01,\n",
       "           1.7361e-01,  2.0504e-02, -6.1116e-01, -2.0452e-01, -7.8060e-02,\n",
       "           2.1248e-01,  1.7352e-01, -7.7867e-01,  4.1630e-01,  3.7605e-01,\n",
       "          -2.9791e-01, -2.0558e-01, -7.6535e-01,  6.9050e-03, -5.2704e-02,\n",
       "          -3.5276e-01,  2.9703e-01, -3.7523e-01, -5.1002e-02, -3.6091e-01,\n",
       "          -7.6495e-01, -6.4796e-01,  3.4252e-01, -2.5838e-01,  4.8264e-02,\n",
       "          -3.0934e-01,  3.6745e-01,  1.4535e-02, -7.6654e-02,  3.8071e-01,\n",
       "           4.8692e-01,  1.0424e-01, -5.8377e-02, -3.4022e-01, -1.3403e-01,\n",
       "           2.3029e-01, -6.1057e-02, -3.0189e-01, -5.7991e-01,  3.5376e-01,\n",
       "           1.9147e-01, -9.2409e-02,  1.6333e-01, -2.0574e-01,  3.0072e-01,\n",
       "          -3.7907e-01,  4.7062e-01,  6.1226e-02, -2.0280e-01, -8.2743e-02,\n",
       "          -4.9539e-01, -1.9036e+00,  6.7255e-02,  3.1448e-02, -2.1904e-01,\n",
       "          -5.0515e-01,  6.8068e-01,  5.6385e-01,  2.0610e-01,  3.5489e-02,\n",
       "          -9.0016e-02, -1.8436e-02, -2.2400e-01,  1.3376e-01, -1.1536e-01,\n",
       "           6.7652e-01,  5.9594e-01,  2.3776e-01, -4.0682e-02, -9.8899e-03,\n",
       "          -4.3319e-01, -4.6077e-02,  3.3930e-01, -1.4825e-01,  3.7672e-01,\n",
       "           1.3757e-01, -6.0119e-01, -5.7103e-02,  2.0548e-01, -1.9587e-01,\n",
       "          -1.1861e-01, -2.6950e-01, -3.9009e-01, -2.4835e-01, -9.0794e-02,\n",
       "           6.3473e-01, -1.5657e-01, -3.6079e-01,  4.8789e-01,  1.4287e-01,\n",
       "          -2.2633e-01,  2.3778e-01, -1.4133e-01, -4.9724e-01, -8.2866e-02,\n",
       "           6.4847e-01, -1.7190e-01, -1.5587e-01, -2.4403e-01, -2.1675e-02,\n",
       "           2.7423e-01, -2.6949e-01,  6.6105e-01,  3.5597e-01, -2.2402e-01,\n",
       "          -2.9264e-01, -2.1755e-01,  1.6998e-01,  1.7183e-02,  1.6433e-01,\n",
       "          -2.9235e-01, -4.5733e-01, -4.6883e-01,  2.0160e-01,  7.1870e-02,\n",
       "           1.3287e-01, -3.2406e-01,  5.0062e-01,  3.9770e-01,  2.9884e-01,\n",
       "          -2.3111e-02, -6.0877e-02,  5.2024e-02, -1.5689e-02, -2.0009e-01,\n",
       "           3.2274e-02,  9.8560e-02, -3.9523e-01,  2.4767e-01, -6.4572e-01,\n",
       "           5.6471e-01,  1.7589e-01, -3.5772e-02,  3.1147e-01, -5.0922e-02,\n",
       "          -2.8004e-01,  4.0813e-02, -1.0498e-01, -2.8317e-01,  2.2155e-01,\n",
       "          -1.8434e-01,  4.7662e-01, -3.2263e-01, -3.4537e-01, -3.7603e-01,\n",
       "          -8.2284e-01, -8.6316e-02, -3.0073e-01, -1.7669e-02, -7.4501e-01,\n",
       "          -1.3632e-01,  4.8093e-01, -1.4908e-01, -7.9510e-03,  8.0060e-02,\n",
       "           1.9503e-01,  2.9085e-01,  2.1612e-01, -1.5080e-01, -4.3815e-01,\n",
       "          -3.5928e-01,  2.5028e-01, -2.2012e-01, -7.6203e-02,  1.0154e-01,\n",
       "           3.9247e-01,  1.0550e-02, -1.3150e-01, -3.8953e-02, -4.1974e-01,\n",
       "          -1.1836e-01,  1.9576e-01, -1.3775e-02, -2.0590e-01, -1.0793e-01,\n",
       "          -6.3769e-02,  1.5837e-01,  3.3957e-01, -3.9771e-01, -5.2691e-01,\n",
       "           2.4631e-02,  4.2702e-01,  3.9002e-02, -1.2383e-01,  1.0996e-02,\n",
       "          -1.1857e+00, -3.2585e-01,  2.1437e-01, -3.9943e-02, -2.1696e-01,\n",
       "           1.2375e-01,  2.2335e-01, -2.7201e-01,  2.8650e-02, -4.3022e-01,\n",
       "           6.4560e-01,  5.9447e-01,  8.1510e-02, -8.1084e-02, -4.5627e-01,\n",
       "          -2.3529e-01, -5.2890e-01,  1.5207e-01, -3.3075e-01, -2.7980e-01,\n",
       "           1.8828e-01,  1.5287e-01, -9.6553e-02,  3.4694e-01, -3.5785e-01,\n",
       "           5.4324e-01, -2.1139e-01, -1.5974e-01,  4.1744e-01, -5.1872e-01,\n",
       "          -2.9157e-01,  4.7355e-01, -6.0852e-01,  6.5128e-02,  3.1234e-01,\n",
       "           2.4635e-01, -5.1119e-03, -1.7666e-01, -1.7074e-01,  7.9613e-01,\n",
       "           1.9544e-01,  2.2557e-01, -4.1420e-01,  5.6192e-01,  4.6488e-01,\n",
       "          -3.3269e-01,  2.1453e-01, -3.1480e-01, -6.3563e-01, -4.8264e-01,\n",
       "           1.8440e-01,  1.5772e-01, -5.3797e-01,  2.4946e-02, -2.8903e-02,\n",
       "           3.3621e-01, -1.8438e-01, -1.2973e-01,  4.1829e-02,  6.7045e-02,\n",
       "          -2.6214e-01, -3.7517e-01,  2.6053e-01, -1.5399e-01,  1.7613e-02,\n",
       "          -2.7438e-01, -1.8051e-01,  3.7426e-01, -2.2339e-01, -2.7966e-01,\n",
       "          -4.3494e-01, -6.4853e-02,  9.3244e-01,  1.7437e-01, -1.4962e-01,\n",
       "          -5.3449e-02, -7.2893e-02, -4.5650e-01,  2.8403e-01, -3.5159e-01,\n",
       "           6.5228e-02,  4.9551e-01, -1.8386e-01, -1.4532e-02,  4.2098e-01,\n",
       "          -2.1605e-01,  1.3820e-02, -1.4076e-01,  2.5538e-01, -5.1844e-02,\n",
       "          -2.1238e-01,  7.6361e-02,  3.6692e-01, -1.6734e-02,  3.2605e-02,\n",
       "           1.0564e-01,  4.6154e-01, -3.5060e-01, -1.6965e-01, -4.4517e-01,\n",
       "           2.6765e-01, -2.1179e-01,  1.1620e-01, -3.4351e-01, -7.5707e-01,\n",
       "           3.8427e-01,  6.5987e-03,  1.5184e-01,  2.2350e-01, -1.1243e+01,\n",
       "          -7.2998e-02,  4.5811e-01, -5.0205e-01,  1.4574e-01, -2.3360e-01,\n",
       "           1.6702e-01, -2.2336e-01,  1.8278e-01, -1.4158e-01, -2.4529e-01,\n",
       "          -2.8834e-01,  4.0501e-02,  9.3999e-03, -1.7507e-01, -1.0539e-01,\n",
       "          -1.5120e-01, -8.4580e-01,  1.0760e-01, -2.8698e-01, -7.8388e-02,\n",
       "          -4.6446e-01,  8.5946e-02,  2.3805e-02,  7.7194e-03, -2.5750e-01,\n",
       "           2.9652e-01, -4.2940e-01,  3.1088e-01,  1.7418e-01,  2.4059e-02,\n",
       "          -7.8492e-02,  3.7246e-02,  3.4124e-01,  5.5757e-02, -3.4474e-01,\n",
       "          -9.8970e-02, -1.7341e-01,  2.0238e-02, -4.1107e-01, -1.8311e-01,\n",
       "          -3.2301e-01, -2.9449e-01,  3.0021e-01,  3.5813e-01,  6.4494e-01,\n",
       "           7.2062e-01,  5.9546e-01,  4.8496e-01, -1.9777e-01, -2.5093e-02,\n",
       "          -5.3472e-01,  2.4667e-01, -1.8485e-01, -7.3473e-01, -2.7798e-01,\n",
       "           4.7836e-01, -7.0740e-01,  3.4484e-01, -2.1192e-01, -4.3616e-01,\n",
       "           4.7106e-01,  8.1483e-02, -1.7725e-01, -5.5864e-01, -1.8186e-01,\n",
       "           2.0076e-01,  5.5754e-02,  8.5448e-02, -2.9783e-01, -7.5098e-02,\n",
       "          -1.0308e-01,  8.6132e-01, -3.8616e-02, -3.0106e-01, -2.7492e-01,\n",
       "           1.4988e-03,  2.0280e-02,  1.4021e+00, -1.9859e-02,  1.5510e-01,\n",
       "           1.6306e-01,  6.1504e-01, -8.4608e-01, -6.0316e-01, -1.6412e-01,\n",
       "          -2.4606e-01,  1.1442e-01,  3.4642e-01,  3.0945e-01, -2.0665e-01,\n",
       "          -6.5170e-01, -2.3452e-01,  8.9758e-02,  2.8417e-01,  1.2012e+00,\n",
       "          -4.8443e-01, -2.0203e-01,  1.6203e-01,  6.9688e-02, -2.9897e-01,\n",
       "           2.2230e-02, -1.0128e-01, -1.4376e-01,  2.6128e-01, -3.4564e-02,\n",
       "           5.9390e-02,  1.9536e-01,  1.5295e-01,  2.1922e-01, -5.3916e-01,\n",
       "           2.8766e-01, -2.2008e-01,  3.7340e-01, -1.7818e-01, -2.1340e-01,\n",
       "           2.7822e-01,  3.5267e-01, -2.5944e-01, -1.1418e-01, -1.8346e-01,\n",
       "           3.4791e-01,  1.6112e-01, -1.1817e-01, -5.6008e-02,  5.5091e-01,\n",
       "          -4.6460e-01, -3.8385e-02,  3.1343e-02, -2.9745e-02,  2.8967e-01,\n",
       "           4.0218e-03, -2.4371e-01,  9.7295e-02, -5.1014e-01, -8.3873e-01,\n",
       "          -8.8805e-01, -1.7447e-01,  2.0244e-02, -4.2506e-01,  2.3899e-02,\n",
       "          -2.2517e-01,  1.5976e-01, -3.1562e-01,  2.2873e-01, -4.9506e-02,\n",
       "           3.6740e-01,  2.2498e-01, -1.2316e-01,  3.3195e-01, -2.6782e-01,\n",
       "          -1.4146e-01, -1.1204e-01, -3.2108e-01,  6.4574e-02, -4.4640e-01,\n",
       "          -6.4782e-01, -1.1248e-01,  2.2522e-01,  3.3155e-01, -3.6732e-01,\n",
       "           9.5610e-02,  2.0224e-01, -1.0420e-01,  2.8846e-01,  5.1453e-01,\n",
       "           2.6878e-01, -1.7201e-01, -2.9534e-01, -1.0592e-02,  2.7615e-01,\n",
       "           1.9941e-02,  9.9655e-03,  2.3579e-01, -3.0125e-01, -8.7342e-02,\n",
       "          -2.0303e-03,  1.1773e+00,  1.0228e-02, -3.1486e-01,  2.7290e-01,\n",
       "           1.1227e+00, -4.0275e-01,  3.4864e-01, -2.9779e-01,  3.0414e-01,\n",
       "          -3.4952e-01, -4.8596e-01,  5.9648e-01, -1.1177e-01,  4.7048e-01,\n",
       "           6.1000e-02,  1.4448e-01, -4.2568e-02,  3.7892e-01,  2.7698e-01,\n",
       "          -1.5939e-01, -3.3432e-03, -3.7601e-01, -3.4491e-01,  7.0163e-01,\n",
       "           1.3023e-01, -1.7129e-01,  6.5916e-02, -7.3735e-02, -1.0876e-01,\n",
       "           4.2259e-01,  9.8389e-01, -2.2004e-01, -2.4490e-01,  1.0090e+00,\n",
       "           2.1434e-01,  4.6065e-02,  3.4556e-01,  3.8758e-01,  2.8829e-01,\n",
       "           5.1958e-01,  2.7808e-01, -1.7199e-01,  1.1222e-01, -3.5000e-03,\n",
       "           5.0731e-01, -9.5693e-01, -7.1400e-01,  1.2642e-02,  3.5231e-01,\n",
       "          -3.8758e-02,  4.7774e-01, -3.4981e-01,  4.5720e-01,  1.0893e+00,\n",
       "          -1.1283e+00, -7.0549e-02, -2.4802e-01,  2.4639e-01,  1.1226e-01,\n",
       "          -1.8910e-01,  6.9420e-03, -2.2180e-01, -2.7019e-01,  5.3197e-01,\n",
       "           6.2496e-01, -1.2156e+00, -6.5308e-02,  3.8662e-01, -2.8134e-01,\n",
       "           1.8050e-01,  9.3360e-02,  1.3443e-01,  7.7310e-02,  2.7618e-01,\n",
       "          -2.9509e-01,  2.4784e-01, -1.1349e-01, -2.5243e-01,  2.0353e-01,\n",
       "           6.0546e-01,  9.3773e-03,  3.6004e-02,  2.1999e-01, -3.3325e-01,\n",
       "          -1.9916e-02, -5.7197e-01, -1.6419e-01,  1.3720e-01, -2.9474e-01,\n",
       "           1.1582e-01, -6.4346e-01, -1.5691e-01,  8.0253e-02, -8.9000e-02,\n",
       "           3.6944e-01,  4.3394e-01,  8.3629e-01, -6.0632e-03,  7.2745e-02,\n",
       "           5.1547e-01,  2.1420e-01, -1.3456e-01,  5.6869e-01,  3.1237e-01,\n",
       "          -4.4696e-01,  2.9997e-01, -9.3167e-02, -7.1597e-03,  1.3402e-01,\n",
       "          -2.2884e-01,  2.7238e-01,  1.8838e-01, -5.8412e-01,  1.9155e-01,\n",
       "           6.3684e-02,  9.7163e-02, -3.7305e-01,  3.5550e-01,  4.2871e-01,\n",
       "          -3.5865e-01,  7.7271e-01,  3.5168e-02,  3.5651e-01, -2.8366e-01,\n",
       "          -1.1581e-01, -4.4653e-01, -2.7504e-01,  9.7959e-02,  1.9642e-02,\n",
       "          -4.5514e-01, -5.1352e-02, -7.7884e-02, -3.2687e-01,  5.0312e-01,\n",
       "           8.8312e-01, -3.5546e-01, -6.5984e-01, -3.0001e-01,  1.1395e-02,\n",
       "           5.3912e-02, -3.6198e-03,  5.0024e-01, -2.6261e-01,  3.6769e-01,\n",
       "          -9.1786e-01, -1.5061e-01,  1.7033e-01,  3.1392e-01, -3.1852e-01,\n",
       "           7.6093e-01,  8.4939e-02,  1.9163e-01,  3.5373e-01,  5.0366e-01,\n",
       "          -2.2057e-02,  6.1177e-01,  7.3480e-02, -2.0580e-02,  6.2282e-02,\n",
       "           2.2181e-01, -4.1530e-01, -1.2175e-01, -7.5463e-01,  4.8336e-01,\n",
       "           1.6953e-01,  5.5130e-01,  7.0354e-01]], device='mps:0'),\n",
       " tensor([[-2.5187e-01, -4.0054e-01, -5.8007e-01, -8.3055e-01, -1.2868e-01,\n",
       "          -2.8804e-02, -7.1917e-01,  6.6701e-01,  1.2709e-01,  5.0855e-01,\n",
       "           1.2570e-01,  5.1916e-01,  5.1676e-02, -1.1521e-01, -1.6481e-01,\n",
       "          -4.9820e-01,  5.6222e-02,  3.7338e-01,  6.7187e-01, -2.4111e-02,\n",
       "           2.2844e-01,  5.7979e-01, -2.6323e-01, -1.2386e-02, -8.3013e-01,\n",
       "           2.9613e-01, -1.1576e-01,  2.2409e-01,  1.7718e-01, -6.3092e-01,\n",
       "           4.0999e-02,  4.7417e-01,  2.3068e-02,  1.3497e+00, -5.6080e-01,\n",
       "          -1.3230e-01,  8.2213e-02, -3.9453e-01, -2.3010e-01,  3.7283e-02,\n",
       "           6.1164e-01, -4.1315e-01,  1.3635e-01,  4.0254e-01, -6.1085e-02,\n",
       "           8.9076e-02,  1.3899e-01, -3.7744e-01,  2.5892e-01, -8.3961e-01,\n",
       "          -4.8318e-02, -2.0669e-01,  4.9144e-01,  6.2615e-01, -6.3176e-01,\n",
       "          -2.2975e-01,  2.7419e-01, -3.7716e-02, -3.3020e-02,  1.5349e-01,\n",
       "           9.7980e-02,  3.8582e-01, -2.9995e-01,  2.3257e-01,  3.3441e-01,\n",
       "           2.4458e-01, -3.0392e-01, -4.8889e-01, -5.4078e-01, -5.0111e-02,\n",
       "           9.1859e-03,  2.2490e-02,  1.4846e-01, -2.3272e-01,  4.1973e-01,\n",
       "          -7.1214e-01, -6.4354e-02, -8.9125e-02, -2.5973e-01,  2.9246e-01,\n",
       "          -4.5549e-02,  1.9719e-01,  3.1010e-01,  1.0268e-01,  1.2457e-01,\n",
       "           4.3419e-02,  2.7215e-01,  5.2932e-04,  4.7373e-02,  3.7533e-01,\n",
       "          -1.3375e-01, -8.1722e-02, -2.8313e-01, -3.5814e-01,  3.1229e-01,\n",
       "           4.0598e-02,  2.9907e-01, -1.9557e-01,  1.8579e-01, -3.2474e-01,\n",
       "           1.5588e-01,  1.7768e-01, -1.8083e-01,  5.7286e-02, -3.5172e-01,\n",
       "          -4.5475e-01,  2.3589e-02, -2.8569e-01, -4.8285e-01,  2.2697e-01,\n",
       "          -1.4211e-01, -3.0205e-01, -4.0922e-02, -3.8935e-01, -3.1524e-02,\n",
       "          -7.5164e-02,  2.6013e-01, -1.3230e-01, -3.4806e-01, -8.3033e-02,\n",
       "           2.0734e-01, -2.2808e-01,  1.6077e-01,  2.6882e-01,  1.2232e-01,\n",
       "          -9.6705e-02,  6.5251e-01,  1.8950e-02,  3.1674e-01,  2.5763e-01,\n",
       "           1.1699e-01, -6.5768e-02, -1.9745e-01, -2.3154e-01,  7.5988e-02,\n",
       "          -7.6941e-02,  1.0687e-01, -1.8182e-01, -2.2790e-01, -1.8508e-01,\n",
       "           3.1771e-01, -3.3097e-01, -6.0893e-01, -4.9260e-01, -1.6098e-01,\n",
       "          -1.6326e-01, -2.4804e-02, -8.2473e-03, -3.3693e-01, -3.1956e-01,\n",
       "          -6.7443e-01, -3.6717e-01,  8.1041e-01, -1.6831e-02, -3.3356e-01,\n",
       "          -1.6499e-01,  2.3208e-01,  3.1666e-01, -4.5959e-02,  3.7161e-01,\n",
       "           7.7250e-01, -4.7670e-02,  1.6786e-01, -5.1827e-01, -3.8195e-01,\n",
       "           4.8335e-02,  2.6658e-01, -2.7054e-01,  6.1879e-01,  4.8352e-02,\n",
       "           7.5627e-02, -6.6170e-02,  2.7430e-01, -3.2704e-01, -1.2324e-02,\n",
       "          -5.7573e-01,  3.7750e-01,  2.9539e-01, -3.9892e-01, -6.6567e-02,\n",
       "          -2.0834e-02, -1.2060e+00,  9.6964e-02, -8.3040e-01,  3.1236e-01,\n",
       "          -2.3110e-01,  1.7963e-01,  2.3389e-01,  2.9545e-01, -6.5320e-02,\n",
       "           2.5404e-01,  2.8725e-01,  2.5687e-01,  3.3609e-01, -3.4924e-02,\n",
       "           2.8019e-01,  2.3924e-01,  5.0790e-01,  4.1073e-01,  1.4769e-01,\n",
       "          -3.5440e-01,  2.9683e-01,  3.5781e-01,  9.0598e-02,  1.6954e-01,\n",
       "           9.9825e-02, -2.1636e-01,  2.1091e-01,  1.7501e-01, -1.6618e-01,\n",
       "          -2.2406e-01, -3.1100e-02, -4.6818e-01, -5.6655e-02, -4.8467e-01,\n",
       "           3.2557e-01, -5.1983e-01, -1.1912e-01,  3.4831e-01,  2.4340e-01,\n",
       "          -1.4896e-02,  1.8319e-01,  1.7196e-01, -2.4338e-01,  2.1735e-01,\n",
       "           3.7449e-01, -3.4618e-01, -3.9804e-02, -3.2197e-01, -2.9367e-01,\n",
       "           2.4357e-01, -5.6244e-01,  6.6635e-01,  4.0574e-01,  6.0165e-02,\n",
       "           1.5119e-01, -2.7920e-01, -5.1638e-02, -5.7408e-01,  3.5471e-01,\n",
       "          -3.2130e-01, -3.4955e-01, -5.4801e-01,  6.4487e-02,  4.2773e-01,\n",
       "           2.7565e-01, -6.4915e-02,  7.7868e-01,  4.2408e-01,  4.2813e-01,\n",
       "          -4.3003e-01, -1.5896e-02,  4.0377e-01, -4.1405e-02, -4.6059e-01,\n",
       "           6.7116e-02, -1.4962e-01, -5.1643e-01,  2.8000e-01, -2.6598e-01,\n",
       "          -1.2047e-01, -1.4913e-01,  3.3682e-01,  1.5365e-01, -2.1964e-01,\n",
       "          -3.8461e-01,  3.9587e-01, -1.4349e-01, -4.1723e-01,  4.7123e-01,\n",
       "          -3.5025e-01,  1.5209e-01,  1.0694e-01, -3.8441e-01, -6.2072e-01,\n",
       "          -3.2235e-01,  2.3090e-01,  1.5585e-01,  3.4799e-02, -6.6306e-01,\n",
       "          -4.0571e-01,  3.1197e-01,  2.4601e-01,  2.9976e-01,  2.9840e-01,\n",
       "           5.7427e-01, -3.0944e-01, -1.6702e-01,  1.4982e-01, -4.1377e-01,\n",
       "          -1.5704e-01, -3.3228e-02, -1.9995e-01,  2.9569e-01,  2.4768e-01,\n",
       "           1.1600e-01,  2.0367e-01,  4.3482e-01, -2.5272e-01, -3.8673e-01,\n",
       "          -7.1543e-02,  2.0161e-01, -9.3629e-02, -5.8306e-02,  2.3951e-01,\n",
       "          -6.3095e-02,  5.9650e-01,  7.2143e-01,  7.7439e-02, -6.4006e-01,\n",
       "           1.7696e-01,  5.2502e-01, -2.7253e-01, -4.0343e-02,  9.6365e-02,\n",
       "          -1.0344e+00, -2.1331e-01, -1.2883e-01, -2.0181e-01, -6.5352e-01,\n",
       "           8.8115e-02,  3.4863e-01,  4.6833e-03, -1.7864e-01, -3.0947e-01,\n",
       "           1.0448e-01,  4.8724e-02, -4.8980e-02,  4.3534e-01, -2.8901e-01,\n",
       "          -3.4797e-01, -2.3438e-01, -2.1532e-01, -5.4201e-01, -3.9968e-01,\n",
       "           3.5914e-01,  2.8129e-01,  1.1986e-01, -2.6486e-01, -5.1832e-01,\n",
       "           4.5577e-01, -6.8169e-01, -1.4632e-01,  2.4840e-01, -4.7702e-01,\n",
       "           1.4650e-02,  4.8990e-01,  1.4999e-01,  1.6336e-01,  6.5258e-02,\n",
       "           7.8557e-02, -3.3903e-01, -5.0030e-01, -1.5000e-01,  6.2323e-01,\n",
       "           3.2283e-01,  3.3506e-01, -1.7236e-01,  3.9096e-01,  4.5167e-01,\n",
       "          -2.9582e-01,  4.1670e-01,  1.0930e-02,  7.7439e-02, -6.0414e-02,\n",
       "           1.9048e-01,  3.4370e-01, -3.6842e-01,  2.0145e-01,  1.7929e-01,\n",
       "          -3.1634e-01, -3.2149e-01,  1.5691e-02,  2.0642e-01, -1.8580e-01,\n",
       "          -3.0875e-01, -2.0361e-01,  9.9698e-02, -3.4856e-01, -4.3104e-01,\n",
       "           2.9161e-01, -3.6543e-02,  7.0495e-02, -4.0357e-01, -4.3462e-01,\n",
       "          -2.0181e-01, -2.6055e-01,  6.8681e-01,  2.1064e-01, -1.9621e-01,\n",
       "           1.8720e-01, -1.0907e-01, -3.6330e-01,  4.7847e-01,  2.0162e-02,\n",
       "          -4.8754e-01,  4.5482e-01, -3.4822e-01,  1.0627e-01,  5.5756e-01,\n",
       "          -1.6769e-01, -1.0840e-01,  1.1008e-01,  1.0820e-01,  2.9026e-01,\n",
       "          -3.9884e-01,  1.5516e-01, -6.3476e-02, -3.1885e-01,  3.1820e-01,\n",
       "           2.6411e-01,  2.2781e-01, -1.7137e-01, -3.3426e-02,  3.5677e-02,\n",
       "           2.8782e-01, -6.0853e-02, -3.9888e-01, -4.7347e-01, -4.6132e-01,\n",
       "           1.2832e-01,  1.8927e-01,  5.6207e-01,  3.4144e-01, -1.1394e+01,\n",
       "           1.2938e-01,  7.8031e-01, -4.0713e-01, -8.2819e-02, -5.0320e-02,\n",
       "           1.0976e-02, -1.3708e-01,  4.2955e-01, -2.7734e-01, -4.4105e-01,\n",
       "          -3.5727e-01, -1.3308e-01,  1.5454e-01, -3.2189e-01, -2.6945e-01,\n",
       "          -1.4686e-02, -7.0935e-01,  4.0963e-01, -3.0286e-01,  5.0298e-01,\n",
       "          -2.3511e-01, -8.3798e-02,  1.5816e-01, -3.5851e-01, -1.3439e-02,\n",
       "           5.9766e-02, -1.6033e-01,  1.4651e-01,  4.2486e-01, -5.2072e-02,\n",
       "           3.3244e-02, -4.1258e-03,  4.8461e-02, -2.5723e-01, -5.9999e-01,\n",
       "           1.9296e-02, -1.4141e-01,  3.3422e-01, -4.4133e-02, -7.1265e-01,\n",
       "          -2.2533e-01, -2.7418e-01,  3.1407e-01,  2.7021e-01,  9.0598e-01,\n",
       "           2.2955e-01,  1.0686e+00,  1.5022e-01,  1.7748e-01,  2.6230e-02,\n",
       "          -7.8764e-01, -3.3269e-02, -4.1907e-01,  2.3280e-02,  1.0114e-01,\n",
       "           3.6576e-01, -7.9367e-01,  2.5251e-01, -3.5521e-01,  3.6487e-02,\n",
       "           4.9094e-01,  3.1768e-02,  2.7700e-01, -3.2298e-02, -7.8123e-02,\n",
       "           1.2089e-01, -2.8544e-02,  1.1098e-01, -2.2583e-01, -2.4084e-01,\n",
       "          -1.0177e-01,  5.9888e-01,  3.9592e-01, -7.2211e-01,  3.6112e-01,\n",
       "           4.9258e-02, -1.8622e-01,  1.4556e+00, -4.8573e-02,  1.8361e-02,\n",
       "           1.9800e-03,  2.9604e-01, -5.0236e-01, -7.3516e-01,  3.6511e-01,\n",
       "          -1.0211e-01,  5.6427e-02,  1.0127e-01,  4.8148e-01, -9.7481e-02,\n",
       "          -4.7940e-01, -1.8593e-03,  1.0556e-01, -1.3380e-01,  5.4710e-01,\n",
       "          -7.2824e-01, -1.6963e-01, -1.1223e-01,  3.1723e-01, -8.4054e-01,\n",
       "          -6.1126e-02, -7.0210e-01, -3.9328e-01,  5.0586e-01, -2.5896e-01,\n",
       "           1.5852e-01, -2.7443e-02, -4.3439e-02,  5.5574e-01, -2.1489e-01,\n",
       "           7.1555e-02,  2.2813e-01,  8.4995e-02, -1.6733e-01,  3.9349e-02,\n",
       "           4.1630e-01,  4.9571e-01, -4.2740e-01, -6.3911e-01, -7.6901e-03,\n",
       "           1.7180e-01,  1.4933e-01,  3.4019e-01, -5.3463e-01,  3.6431e-01,\n",
       "          -7.9162e-01, -1.8769e-01,  7.2949e-02, -4.3273e-02,  3.6175e-01,\n",
       "           2.3925e-01,  7.2038e-02,  2.2256e-01, -7.7756e-01, -6.4498e-01,\n",
       "          -6.5932e-01, -2.1445e-01,  2.6143e-01, -5.4236e-01, -8.3654e-02,\n",
       "          -1.1033e-01, -4.8837e-03, -5.0292e-02, -6.6102e-02,  1.1939e-02,\n",
       "           1.3632e-01,  2.3587e-01, -1.3023e-01, -1.3178e-01, -9.2464e-02,\n",
       "          -5.5566e-01, -1.1753e-01, -9.9591e-02, -1.2505e-01, -5.6783e-01,\n",
       "          -1.3797e-01,  3.3075e-02, -2.5373e-01, -6.8124e-02,  3.5608e-01,\n",
       "          -2.9832e-02, -2.7856e-02, -4.8850e-01,  7.2846e-01,  4.0279e-01,\n",
       "           5.2088e-02, -2.8298e-01, -5.5189e-01, -1.0567e-01,  2.9592e-01,\n",
       "           1.9555e-01, -6.7007e-02, -1.7709e-01, -2.4145e-01, -4.2522e-01,\n",
       "           1.4302e-01,  7.1147e-01, -6.6296e-02, -6.8115e-01,  2.6621e-01,\n",
       "           5.8223e-01, -1.2882e-01, -2.4432e-01,  2.0983e-02, -9.3284e-02,\n",
       "          -3.1851e-01, -6.0317e-01,  7.8821e-01, -1.1246e-01, -5.8638e-02,\n",
       "           2.2427e-01,  2.0359e-01, -9.2588e-03,  4.0209e-01, -1.5770e-01,\n",
       "          -3.5434e-01, -1.6477e-03, -7.3600e-01,  4.3257e-01,  6.1666e-02,\n",
       "          -3.1489e-01,  7.6082e-01,  2.7260e-01, -6.0593e-02, -4.5779e-02,\n",
       "           2.2863e-01,  7.9941e-01, -2.4636e-01,  6.4653e-02,  5.3779e-01,\n",
       "           8.8110e-02, -1.1229e-01,  3.9580e-02,  4.1665e-02,  4.5410e-02,\n",
       "           3.7012e-01, -3.2145e-02, -4.1564e-01, -6.1194e-02,  3.8924e-01,\n",
       "           3.9295e-01, -5.4610e-01, -7.3487e-01,  2.2333e-02,  6.8551e-01,\n",
       "          -2.9054e-01,  2.3654e-01, -3.2708e-02,  3.0880e-01,  1.2001e-01,\n",
       "          -9.2160e-01, -1.5016e-01, -1.0056e-01,  2.8229e-01, -2.5482e-02,\n",
       "          -1.6139e-01, -1.4727e-01, -2.3686e-01, -1.6208e-01,  6.8308e-01,\n",
       "           2.1991e-01, -6.9160e-01,  3.4058e-02,  5.7281e-01, -6.0551e-01,\n",
       "           4.6052e-01, -2.4577e-01, -3.4302e-02,  3.6435e-01,  1.3117e-01,\n",
       "           4.9067e-02,  1.7856e-01, -4.2138e-01, -4.0598e-01,  1.7824e-01,\n",
       "           9.1672e-01, -7.6613e-02,  1.1166e-01, -3.9384e-01, -1.7182e-01,\n",
       "          -7.1855e-02, -3.1913e-01, -3.1286e-01,  3.7185e-01,  2.0529e-01,\n",
       "          -3.2552e-02, -1.3271e-01, -2.2992e-01,  3.0816e-01,  1.1821e-01,\n",
       "           3.2536e-01,  5.8187e-01,  6.4368e-01, -1.0077e-02, -2.4936e-01,\n",
       "          -6.1235e-02,  1.4568e-01, -1.9524e-01,  5.2704e-01, -2.0627e-01,\n",
       "          -3.3453e-01, -3.8069e-03,  3.0134e-01, -6.1681e-01, -5.8725e-02,\n",
       "          -4.5426e-01,  4.2361e-01, -1.6955e-01,  3.7868e-02,  2.8833e-01,\n",
       "           4.0522e-02,  5.7925e-02, -2.2651e-01,  5.5807e-01,  4.4511e-01,\n",
       "           1.6609e-01,  4.5662e-01,  1.6104e-01,  4.5272e-02, -5.1597e-01,\n",
       "           2.1978e-01, -1.0213e+00, -9.4260e-02, -4.6640e-02, -1.5401e-01,\n",
       "          -4.4224e-01, -2.8126e-02,  1.6814e-01,  1.8538e-01,  6.9597e-01,\n",
       "           7.2967e-01, -3.0871e-01, -4.5290e-01, -1.8510e-01,  1.5304e-01,\n",
       "           2.7705e-01, -1.9517e-01,  3.5971e-01,  9.6103e-02,  2.4432e-01,\n",
       "          -5.0967e-01, -4.3165e-01, -4.2252e-02,  3.1411e-01, -2.0171e-01,\n",
       "           5.7118e-01, -2.0164e-01, -1.3863e-01,  9.0832e-02,  1.3770e-01,\n",
       "           1.6639e-01,  7.9664e-01,  4.3416e-01, -6.3425e-01,  8.7469e-02,\n",
       "          -1.3980e-02, -2.9988e-01,  2.5983e-02, -4.3577e-01,  2.8858e-01,\n",
       "           1.2984e-01,  4.9280e-01,  6.7504e-01]], device='mps:0'),\n",
       " tensor([[ 7.8665e-02, -3.3319e-01, -5.5660e-01, -5.8965e-01, -4.6930e-01,\n",
       "          -1.1946e-01, -4.5759e-01,  6.5134e-01,  1.4655e-01,  5.9324e-01,\n",
       "           3.8617e-01,  5.1988e-01, -8.6744e-02, -4.0391e-02, -8.0315e-01,\n",
       "          -3.4874e-01, -1.0696e-01,  2.4962e-01,  4.9079e-01,  4.1632e-01,\n",
       "           7.9087e-01,  2.9980e-01, -8.4391e-02, -1.0321e-01, -7.6051e-01,\n",
       "           3.8608e-01,  2.4686e-01,  8.7815e-02,  7.6643e-02, -8.2045e-01,\n",
       "          -3.6573e-01,  4.6250e-01,  1.1736e-01,  8.3404e-01, -8.5751e-02,\n",
       "          -2.6677e-01,  1.8969e-01, -2.5406e-01,  3.9238e-01, -1.6812e-01,\n",
       "           6.7438e-01, -1.0123e-01,  5.0173e-01, -6.9290e-02,  4.7742e-01,\n",
       "          -4.2209e-02,  1.8865e-01, -7.1643e-01,  7.9959e-02, -4.2033e-01,\n",
       "          -1.9924e-01, -2.5431e-01,  2.0197e-01,  2.5903e-01, -2.9010e-01,\n",
       "          -4.9244e-01,  9.3847e-02,  1.9423e-01, -1.0342e-01, -2.1171e-01,\n",
       "          -4.2479e-01,  8.4228e-01, -3.5377e-01,  4.3697e-02, -1.7282e-01,\n",
       "           6.7371e-02,  2.0693e-01, -6.5276e-01, -6.2630e-01, -1.0272e-01,\n",
       "          -1.3679e-01,  1.2663e-01, -1.3620e-02, -2.5428e-01,  6.3548e-01,\n",
       "          -3.5617e-01,  1.6481e-01,  2.7260e-03, -4.1286e-01, -2.4351e-02,\n",
       "           1.7587e-01,  2.5331e-01,  1.2718e-01, -1.0982e-01, -5.1255e-01,\n",
       "           1.5244e-01,  4.7846e-01,  6.2648e-01,  1.8116e-01, -2.6247e-01,\n",
       "           2.8822e-01,  1.7173e-01, -3.4797e-01, -1.5619e-01,  5.4757e-01,\n",
       "          -1.7455e-01,  1.9786e-01, -3.1488e-01,  2.4265e-01, -4.8853e-01,\n",
       "           2.0382e-01,  3.1247e-01, -8.9069e-02,  3.8070e-01, -3.5779e-01,\n",
       "          -9.2581e-01,  4.3997e-01, -1.9861e-01, -5.5595e-01, -1.9809e-01,\n",
       "          -1.5441e-01, -1.6270e-01,  2.9113e-01, -2.9993e-01, -1.5597e-01,\n",
       "           8.5736e-02,  1.8774e-01, -1.0887e-01, -4.8866e-01,  3.0041e-01,\n",
       "           3.7222e-01, -3.4038e-01, -7.7277e-02, -1.3002e-03,  1.0354e-01,\n",
       "           1.3987e-01,  2.0961e-01, -1.3358e-01,  5.4239e-01,  4.0550e-01,\n",
       "           1.3351e-03,  3.3908e-01, -4.4286e-01, -1.5048e-02,  1.3303e-01,\n",
       "           2.1415e-01, -2.0294e-01, -1.6943e-01,  6.1144e-01,  1.7822e-01,\n",
       "          -1.9778e-01, -4.4103e-02, -7.0901e-02, -1.9709e-01,  2.2189e-01,\n",
       "          -2.5595e-02,  1.2318e-02, -1.4898e-01, -2.1898e-01, -8.5230e-02,\n",
       "          -5.4516e-01, -8.4766e-01,  3.2739e-01, -2.5227e-01,  2.0967e-01,\n",
       "          -2.8692e-01,  3.8498e-01,  8.0672e-02,  2.1787e-01,  3.7474e-01,\n",
       "           3.3423e-01, -9.3499e-02, -8.7517e-03, -2.8127e-01, -4.4604e-02,\n",
       "           3.4312e-01, -4.3486e-02, -4.0885e-01, -2.4033e-01,  2.6412e-03,\n",
       "           1.4530e-02, -4.5320e-01,  1.1431e-01, -3.2538e-01,  2.0005e-02,\n",
       "          -2.9642e-01,  6.3012e-03,  3.2302e-02, -4.4854e-01,  1.9007e-01,\n",
       "          -5.6012e-01, -1.6311e+00,  4.7305e-01,  5.1473e-02, -3.7522e-01,\n",
       "          -4.8794e-01,  3.7521e-01,  3.5346e-01,  3.8928e-01, -2.2185e-02,\n",
       "          -2.5600e-01,  1.9699e-02,  1.9123e-01,  5.9596e-01, -3.3575e-01,\n",
       "           4.6698e-01,  4.4459e-01, -1.4062e-01,  9.7715e-03,  1.6836e-01,\n",
       "          -5.3519e-01,  5.9995e-02,  1.0718e-01, -6.0225e-01,  1.8609e-01,\n",
       "           3.5584e-02,  1.1801e-01, -2.0560e-02,  1.1724e-01,  1.1455e-01,\n",
       "          -4.7473e-02, -2.2635e-01,  2.9356e-02, -2.1258e-01, -3.2194e-01,\n",
       "           6.5296e-01, -3.0155e-02,  1.5382e-01,  5.1350e-01, -2.2872e-02,\n",
       "          -2.2212e-01,  7.3842e-02, -1.0983e-01, -7.0626e-02, -3.2899e-01,\n",
       "           9.9057e-02,  6.7015e-02, -8.4346e-02, -3.1247e-01, -3.1246e-01,\n",
       "          -2.6931e-01, -3.6587e-01,  8.3386e-01,  2.5966e-01, -3.9696e-01,\n",
       "          -2.1199e-01, -2.7683e-01, -2.0610e-01,  1.9071e-01,  2.4327e-01,\n",
       "          -1.4497e-01, -3.9168e-01, -6.3954e-01,  9.7886e-02,  1.0352e-01,\n",
       "           9.2401e-02, -9.5361e-02,  4.3674e-01,  5.9194e-01,  4.4844e-01,\n",
       "          -3.2206e-02, -2.2028e-01,  3.6476e-01,  3.3383e-01, -8.0178e-02,\n",
       "           1.8134e-01,  9.4323e-02, -1.4522e-01,  4.4022e-01, -3.3692e-01,\n",
       "           2.3695e-01,  8.0931e-02,  2.8936e-02,  4.5324e-01, -3.3168e-01,\n",
       "          -1.3309e-01,  3.4396e-01, -3.6911e-01,  6.4961e-02,  1.9793e-01,\n",
       "          -2.9062e-01,  3.7940e-01,  2.3827e-01, -3.3976e-01, -7.8926e-01,\n",
       "          -2.1933e-01,  4.9726e-01, -3.1672e-03, -1.2541e-01, -9.5543e-01,\n",
       "          -1.8762e-01, -3.3156e-03,  9.2503e-02, -6.4099e-02, -4.0994e-01,\n",
       "           1.2834e-01,  3.9177e-01, -5.0183e-01, -8.7568e-02, -6.1278e-01,\n",
       "           6.9972e-02,  2.5074e-01, -2.8204e-01,  3.7159e-01,  3.0243e-01,\n",
       "          -2.4428e-01,  2.8653e-01,  7.5605e-02,  1.1205e-01, -3.2914e-01,\n",
       "          -8.2701e-02, -1.7332e-01, -7.7504e-02,  1.8209e-01,  4.2936e-02,\n",
       "           2.9511e-01,  3.2389e-01,  3.8615e-01, -7.4800e-01, -6.1691e-01,\n",
       "          -2.8275e-01,  6.6626e-01, -9.6199e-02, -5.5451e-01, -2.2432e-01,\n",
       "          -9.9932e-01, -7.7640e-02, -3.0153e-01, -1.8693e-01, -5.6794e-01,\n",
       "           3.2395e-01,  1.5117e-01, -3.3903e-01,  5.8060e-02, -1.8543e-01,\n",
       "           3.9446e-01,  5.7926e-01, -4.6635e-01, -4.7312e-02, -1.8110e-01,\n",
       "          -7.0000e-01, -1.6876e-01,  9.6926e-03, -1.4039e-01, -2.6525e-01,\n",
       "           1.4828e-01,  8.2458e-03, -3.1826e-02, -3.1465e-02, -4.7696e-01,\n",
       "           4.8306e-01, -7.8942e-02, -8.0065e-02, -6.8700e-02, -4.6410e-01,\n",
       "           7.7896e-01,  5.3654e-01, -2.3148e-01, -2.5147e-01, -1.1381e-01,\n",
       "          -2.4422e-01, -1.8979e-02, -2.6958e-01, -7.2210e-02,  6.7415e-01,\n",
       "           5.6705e-01,  1.3674e-01, -3.2257e-01,  5.3214e-01,  4.5663e-01,\n",
       "          -4.6946e-01,  4.6575e-01, -1.0605e-01, -2.4862e-01,  5.9298e-02,\n",
       "           1.4316e-01,  1.8748e-01, -4.5911e-01,  1.2969e-01,  1.3473e-01,\n",
       "           2.4729e-01, -5.4253e-01,  1.5897e-01, -1.3442e-02,  6.4844e-01,\n",
       "          -4.0979e-01, -4.1103e-01,  2.6310e-01, -5.7506e-01, -1.4183e-01,\n",
       "           1.1065e-01,  2.7836e-01,  1.3613e-01, -1.2344e-03, -2.2089e-01,\n",
       "          -4.2060e-01,  4.6985e-01,  1.0382e+00,  4.9191e-01, -1.1880e-01,\n",
       "           2.7936e-01, -1.7533e-01, -2.2582e-01,  5.2580e-01, -3.0311e-01,\n",
       "          -2.6708e-01,  7.1871e-01, -6.8122e-01,  1.5384e-01,  4.6540e-01,\n",
       "          -3.0831e-01, -2.4656e-01, -3.2853e-01, -2.3824e-01, -2.7128e-01,\n",
       "          -4.1425e-01,  2.6325e-01, -3.0651e-01, -5.8545e-03, -5.0543e-02,\n",
       "           1.9667e-01,  7.0085e-02, -1.3518e-01,  5.5212e-02, -3.5810e-01,\n",
       "           4.2658e-02, -5.1103e-02,  1.3118e-01,  1.1394e-01, -2.7586e-01,\n",
       "           5.3951e-01,  1.0687e-01,  5.8638e-02,  3.7220e-01, -1.1108e+01,\n",
       "           8.8746e-02,  3.3313e-01, -2.9655e-01, -1.2104e-01, -9.0374e-02,\n",
       "           1.4931e-01, -1.8101e-01,  1.0486e-01, -3.0765e-01, -2.0723e-01,\n",
       "          -1.2570e-01,  2.4640e-01,  2.5130e-01,  2.2168e-01, -3.4182e-01,\n",
       "          -1.6470e-01, -9.1021e-01,  1.6431e-01,  2.0580e-01, -3.6130e-02,\n",
       "          -2.9418e-01,  1.5450e-01, -7.8763e-03,  1.3719e-02,  2.8503e-01,\n",
       "          -3.2059e-01, -8.4449e-02,  3.6967e-01,  3.7890e-01,  2.7910e-01,\n",
       "           1.1647e-01,  2.1658e-01,  5.1481e-02, -4.6570e-01, -5.1569e-01,\n",
       "           1.0081e-01, -6.3294e-02, -1.6165e-01,  2.9209e-01, -3.6465e-01,\n",
       "          -2.3637e-01, -3.2496e-01,  3.9977e-01,  4.1944e-01,  8.8848e-01,\n",
       "           9.2062e-01,  4.7680e-01,  5.0283e-01,  1.4722e-01, -5.4062e-02,\n",
       "          -7.3410e-01,  3.3623e-02, -7.9774e-01, -5.3108e-02, -1.3550e-01,\n",
       "           6.7615e-01, -3.5227e-01, -9.0720e-02, -4.1440e-01, -2.6125e-01,\n",
       "           2.4691e-01,  4.5322e-01, -1.4443e-01, -8.2877e-01,  5.2842e-02,\n",
       "           2.2913e-01, -3.6616e-02,  1.3933e-01, -2.1148e-02,  1.1532e-01,\n",
       "           1.8874e-02,  4.7326e-01,  2.0297e-01, -1.6483e-01,  4.1378e-01,\n",
       "           2.5998e-01, -6.2633e-03,  1.2631e+00, -7.9458e-02, -1.4828e-01,\n",
       "          -1.2580e-01,  2.2602e-02, -5.4676e-01, -7.3130e-01, -1.8601e-01,\n",
       "           3.1138e-01,  3.4070e-02,  1.2881e-01,  1.2845e-01, -2.6901e-01,\n",
       "          -1.0709e+00, -2.1814e-01,  3.4137e-01,  5.9279e-01,  9.6023e-01,\n",
       "          -8.9445e-01, -1.3622e-01,  2.3819e-01,  6.5576e-02, -6.3389e-01,\n",
       "           3.7269e-02, -3.6797e-01, -1.8228e-01,  1.0887e-01,  3.8440e-02,\n",
       "          -6.1551e-02,  9.8307e-02, -2.9388e-02,  2.0752e-01, -4.1445e-01,\n",
       "           2.3701e-01,  4.4028e-01,  1.2296e-01, -1.2347e-01,  1.6682e-01,\n",
       "           7.1984e-02,  7.3137e-01, -8.2626e-02, -6.5632e-01, -3.7716e-02,\n",
       "          -1.0649e-02,  1.5745e-01, -8.1279e-02,  4.8702e-02,  5.2501e-01,\n",
       "          -3.5379e-01, -7.3176e-02,  2.9441e-01,  3.2810e-01,  4.1030e-01,\n",
       "           6.5463e-02, -6.3273e-01, -1.7626e-02, -8.8577e-01, -9.4784e-01,\n",
       "          -7.1769e-01,  1.8862e-01,  1.2131e-03, -1.6380e-01, -3.8259e-01,\n",
       "          -1.8723e-02,  2.6946e-01, -3.9179e-01,  6.9487e-02,  2.3106e-01,\n",
       "           7.5315e-01,  3.0494e-01, -3.1046e-01,  3.7063e-01, -2.1554e-01,\n",
       "          -5.9244e-01, -3.7050e-01, -1.7002e-01,  7.9902e-02, -3.8246e-01,\n",
       "          -5.5448e-01, -1.4587e-01, -4.0493e-01, -3.0235e-01,  3.4117e-01,\n",
       "          -3.8965e-01,  6.2961e-02, -1.3742e-01,  3.6226e-01,  3.7745e-01,\n",
       "           2.5163e-01, -6.4800e-02,  9.0067e-02, -1.9536e-02,  5.0779e-01,\n",
       "          -8.2760e-02,  2.2725e-01, -4.8690e-01, -6.5028e-02,  7.8279e-02,\n",
       "           4.7451e-01,  8.5143e-01,  1.1698e-02, -1.3860e-01,  1.3725e-01,\n",
       "           4.1962e-01, -2.4000e-01,  5.8429e-02,  1.3686e-01, -4.2022e-01,\n",
       "           1.6675e-02, -6.0083e-01,  2.4538e-01, -3.3244e-02,  3.8033e-01,\n",
       "          -1.9480e-01, -1.0302e-01, -9.5638e-02,  6.1240e-01, -7.3715e-02,\n",
       "          -3.1113e-01, -2.5702e-01, -3.2976e-01,  4.1188e-03,  2.8295e-01,\n",
       "          -1.1001e-01, -2.8407e-02, -2.9404e-01, -2.0039e-01,  1.7325e-01,\n",
       "          -4.8451e-02,  1.0851e+00,  2.0519e-01, -6.9019e-01,  6.2926e-01,\n",
       "           3.4115e-01, -4.9966e-01,  1.0309e-01, -3.3142e-01,  3.3400e-01,\n",
       "           4.1797e-01, -1.2241e-02,  1.6279e-01,  2.2284e-01,  6.1621e-02,\n",
       "           5.4012e-01, -6.8948e-01, -5.9747e-01,  8.5144e-02,  2.1068e-01,\n",
       "          -2.6200e-01,  5.0605e-01,  3.2392e-02,  5.1016e-01,  7.9295e-01,\n",
       "          -1.2491e+00, -2.1483e-01, -6.5459e-01,  4.1457e-01,  1.2664e-01,\n",
       "          -2.8600e-01, -9.1929e-02, -4.4405e-01, -4.8891e-01,  3.4095e-01,\n",
       "           9.6014e-01, -1.3052e+00,  1.0671e-01,  4.2998e-01,  2.0346e-01,\n",
       "           2.0621e-01, -3.8512e-01,  3.3660e-01, -3.0197e-01,  1.4346e-01,\n",
       "          -2.9912e-01,  3.6593e-01, -6.6939e-01, -7.8222e-01, -1.3041e-03,\n",
       "           4.4873e-01,  1.9597e-01,  1.3924e-01,  5.5806e-03,  1.9202e-01,\n",
       "          -1.3225e-01, -9.4832e-02, -9.5299e-02,  4.6108e-01, -2.0293e-01,\n",
       "           1.8000e-01, -4.9486e-01,  8.6916e-02,  2.5310e-01,  6.8945e-02,\n",
       "           2.4457e-03,  4.5603e-01,  9.4221e-01, -2.2006e-02, -1.0648e-01,\n",
       "           4.3890e-01,  5.1716e-01, -1.8882e-01,  6.7642e-01, -3.1450e-01,\n",
       "          -2.7973e-01,  5.7423e-01, -3.3632e-02, -3.4042e-01,  3.1184e-01,\n",
       "          -2.3210e-01,  2.5750e-01,  1.5642e-01, -1.8786e-01,  4.9289e-02,\n",
       "           1.3476e-01, -4.2449e-02, -5.0464e-01,  8.7000e-02,  2.9638e-01,\n",
       "          -3.9737e-02,  9.0737e-01, -2.9828e-01,  5.2502e-01, -5.4668e-01,\n",
       "           1.1240e-01, -5.7810e-01, -1.9621e-01, -3.5303e-01, -3.1059e-01,\n",
       "          -5.0994e-01, -1.6831e-01,  1.5122e-01,  3.3280e-01,  9.1844e-01,\n",
       "           4.4060e-01, -4.7953e-01, -4.6967e-01, -1.0615e-01,  3.8615e-01,\n",
       "          -9.7167e-02, -1.0981e-01,  3.1683e-01,  1.5071e-01,  4.4850e-01,\n",
       "          -5.9487e-01, -4.0341e-01, -4.0880e-02,  3.4785e-01, -2.0486e-01,\n",
       "           6.1918e-01, -5.8090e-02,  2.8287e-01,  2.0884e-01,  1.3691e-01,\n",
       "          -1.4738e-01,  2.0868e-01, -2.8917e-01, -5.5648e-01, -9.5854e-02,\n",
       "           4.5684e-02, -5.6348e-01, -2.9590e-02, -6.8213e-01,  1.5402e-01,\n",
       "           4.2150e-01,  8.7482e-01,  5.2121e-01]], device='mps:0'),\n",
       " tensor([[-1.9310e-01, -3.5515e-01, -6.2940e-01, -8.4105e-01, -8.5561e-02,\n",
       "          -3.2906e-01, -7.5040e-01,  8.7674e-01,  1.1943e-01,  6.7736e-01,\n",
       "           1.2762e-01,  3.8323e-01, -3.7411e-02,  5.8575e-02,  2.9880e-02,\n",
       "          -3.8629e-01, -4.1060e-01,  3.8795e-01,  1.0454e+00,  2.4482e-01,\n",
       "           4.5164e-01,  5.7226e-01, -1.8375e-01, -2.9432e-01, -7.4052e-01,\n",
       "           4.1961e-01, -4.0209e-02,  2.3742e-01,  2.1512e-01, -8.6620e-01,\n",
       "           2.6521e-01,  3.1235e-01,  1.3069e-01,  9.2503e-01, -2.0903e-01,\n",
       "          -4.1057e-01,  3.0817e-01, -6.3766e-01, -1.1961e-01, -8.8263e-02,\n",
       "           5.8318e-01,  1.1480e-01,  9.5236e-02,  2.6219e-01,  1.2687e-01,\n",
       "           1.2006e-01,  4.1832e-02, -1.6541e-01,  1.9309e-01, -9.1180e-01,\n",
       "          -1.2672e-02, -2.2981e-02,  6.4307e-01,  5.5530e-01, -8.0451e-01,\n",
       "          -4.6715e-02,  4.6039e-01,  1.6087e-01, -4.4392e-02,  1.6024e-01,\n",
       "          -4.1390e-02,  4.6187e-01, -2.5777e-01, -1.2183e-02,  2.7744e-01,\n",
       "           2.6801e-02, -4.3739e-01, -4.4309e-01, -4.9228e-01,  4.0272e-02,\n",
       "          -2.7421e-01,  4.6532e-01,  3.9616e-01, -3.0882e-01,  4.7591e-01,\n",
       "          -4.3520e-01,  8.2006e-02, -2.4188e-01, -5.1214e-01,  3.1857e-01,\n",
       "           2.7660e-01,  4.6339e-01,  1.5695e-01, -1.0280e-01, -7.7339e-02,\n",
       "          -9.1150e-02,  2.7582e-01,  1.6120e-01, -4.4269e-02,  1.2639e-01,\n",
       "           1.0771e-01, -8.1740e-02, -3.6745e-01, -5.8679e-02,  4.0319e-01,\n",
       "          -5.5498e-01,  2.5282e-01, -2.0929e-01,  7.7797e-02, -3.1471e-01,\n",
       "          -8.4671e-02,  2.4582e-01, -2.8649e-01,  2.3472e-01, -1.5337e-01,\n",
       "          -4.4763e-01,  2.9938e-01, -2.8207e-01, -5.0550e-01,  2.4128e-01,\n",
       "           1.1305e-01, -2.7976e-01, -1.0460e-02, -2.7464e-01, -2.2799e-01,\n",
       "           1.4561e-01,  2.1012e-01, -1.1759e-01, -1.4006e-01,  1.5538e-01,\n",
       "           2.3098e-01, -2.2767e-01, -2.2287e-01,  2.5929e-02, -9.5446e-02,\n",
       "          -3.8560e-01,  7.3205e-01,  1.0769e-01,  3.6212e-01,  2.7250e-01,\n",
       "           2.2165e-01,  1.1468e-01, -3.2616e-01, -2.1783e-01, -1.9856e-02,\n",
       "           2.4676e-01,  1.0762e-01, -4.2803e-01, -1.7435e-01,  1.7655e-01,\n",
       "           9.6565e-02, -1.9297e-01, -3.6869e-01, -1.6427e-01, -9.1871e-02,\n",
       "          -9.8492e-02,  9.3032e-02, -2.0423e-01, -9.9723e-02, -1.2264e-01,\n",
       "          -4.3692e-01, -6.3384e-01,  3.4207e-01, -4.3163e-02, -1.6860e-01,\n",
       "          -5.1066e-01,  1.5503e-01,  1.0696e-01,  1.3972e-01,  8.3105e-02,\n",
       "           1.1689e+00,  8.3552e-02,  2.9380e-01, -3.5841e-01, -3.9904e-01,\n",
       "          -4.1345e-03,  1.2314e-02, -3.0537e-01,  3.8979e-01, -7.5801e-03,\n",
       "           1.1506e-01, -7.3730e-02,  4.2951e-01,  2.5190e-02,  2.8841e-01,\n",
       "          -4.4273e-01,  5.9431e-01,  4.3663e-01, -1.2336e-01,  2.7896e-01,\n",
       "           1.4778e-01, -1.3970e+00,  3.2735e-02, -5.0015e-01,  1.6754e-01,\n",
       "          -2.4490e-02,  4.3812e-01,  2.4399e-01,  9.6387e-02, -3.2682e-02,\n",
       "           2.0048e-01,  3.4727e-01, -2.1625e-02,  2.5552e-01, -1.3387e-01,\n",
       "           3.4935e-01,  5.3010e-01,  5.2875e-01,  5.0255e-01, -9.2919e-02,\n",
       "          -5.6734e-01,  2.2084e-01,  2.6949e-01, -1.6006e-02,  2.7270e-01,\n",
       "          -3.6497e-02, -4.1440e-01, -8.7973e-02,  4.5342e-02, -6.1910e-02,\n",
       "          -3.2075e-01, -1.8537e-02, -5.9956e-01,  1.2385e-01, -5.1626e-01,\n",
       "           2.7322e-01, -2.3562e-01, -3.3350e-02,  5.0609e-01,  1.5604e-02,\n",
       "          -1.7961e-01,  2.5360e-01,  1.9935e-01, -2.7500e-01,  4.4282e-02,\n",
       "           4.1341e-01, -5.1330e-02, -1.1561e-01, -2.0195e-01, -1.6525e-01,\n",
       "           1.1873e-01, -5.8523e-01,  9.1471e-01,  1.7274e-01,  8.7345e-02,\n",
       "           1.8165e-01, -2.0204e-01,  4.1921e-02, -4.4040e-01,  3.4525e-01,\n",
       "          -2.8979e-01, -5.4944e-01, -7.2578e-01,  1.4339e-01,  2.4035e-01,\n",
       "           1.9520e-01, -2.7262e-01,  6.0193e-01,  5.1746e-01,  4.6084e-01,\n",
       "          -4.5864e-01,  7.6388e-03,  2.3083e-01,  8.0994e-02, -2.4717e-01,\n",
       "           1.8398e-01, -2.4454e-01, -6.5511e-01,  3.5541e-01, -1.1763e-01,\n",
       "          -2.7872e-02, -4.9772e-02,  3.0532e-01,  2.7125e-01, -4.2138e-01,\n",
       "          -5.3525e-01,  4.9060e-01,  4.6353e-02, -3.7293e-01,  5.7259e-01,\n",
       "          -3.8268e-01,  2.9331e-01, -8.7763e-02, -2.8026e-01, -4.6287e-01,\n",
       "          -5.7868e-01,  9.9226e-03,  2.5605e-02, -3.1397e-01, -8.9768e-01,\n",
       "          -5.0234e-01,  3.8046e-01,  1.4740e-01,  1.4815e-01,  3.8801e-01,\n",
       "           4.5559e-01, -1.2018e-01,  6.1804e-02,  1.0680e-01, -2.0468e-01,\n",
       "          -7.6783e-02, -7.0120e-03, -2.5639e-01,  3.2149e-01,  3.9085e-01,\n",
       "           3.7070e-01,  2.7144e-01,  4.5358e-01, -2.2443e-01, -1.2908e-01,\n",
       "          -1.3356e-01,  4.4455e-02, -6.1335e-01, -1.9535e-01,  5.7352e-01,\n",
       "          -4.0393e-02,  5.8247e-01,  5.8468e-01,  2.1858e-01, -4.3458e-01,\n",
       "           1.9537e-01,  5.0041e-01, -2.1632e-01, -3.6628e-01,  3.6734e-01,\n",
       "          -8.4672e-01, -2.0723e-01,  7.5676e-02,  9.2304e-02, -3.2858e-01,\n",
       "           1.7791e-01,  4.4238e-01,  4.8505e-02, -1.9864e-01, -3.4075e-01,\n",
       "           3.9043e-01,  1.0411e-01,  8.8079e-02,  2.2021e-01, -3.9035e-01,\n",
       "          -1.4092e-01, -2.2162e-01, -2.0899e-01, -2.8542e-01, -2.3019e-01,\n",
       "           1.9613e-01,  4.9861e-02, -1.9292e-01, -1.8447e-01, -1.8354e-01,\n",
       "           3.9952e-01, -4.8315e-01, -4.8140e-01,  2.0873e-01, -1.9842e-01,\n",
       "          -1.2302e-01,  1.8274e-01, -2.5562e-01,  1.0475e-01,  1.5596e-01,\n",
       "           5.5716e-02, -3.3033e-01, -2.1296e-01, -1.0287e-01,  7.3788e-01,\n",
       "           3.6373e-01,  3.6197e-01, -3.7295e-01,  5.8145e-01,  2.2941e-01,\n",
       "          -2.6161e-01,  3.1626e-01,  1.5985e-03, -2.7853e-01, -3.5925e-01,\n",
       "           3.6359e-02,  3.6860e-01, -4.9975e-01,  2.0173e-01, -4.3638e-02,\n",
       "          -4.9154e-03, -5.8638e-02, -2.3888e-01,  6.9152e-02, -5.9390e-03,\n",
       "          -3.5268e-01, -5.2562e-01,  1.5340e-01, -2.9764e-01, -3.2900e-01,\n",
       "           1.9582e-01,  3.8513e-02,  1.3993e-01, -3.4773e-01, -3.2641e-01,\n",
       "          -1.0950e-01, -7.7979e-02,  7.1367e-01,  1.2609e-01, -2.6930e-01,\n",
       "           2.1552e-01, -2.7061e-01, -2.6595e-01,  1.4622e-01, -3.6977e-01,\n",
       "          -5.1898e-01,  3.7545e-01, -2.8342e-01, -7.4792e-02,  4.7030e-01,\n",
       "          -2.5475e-01,  1.5864e-01, -1.9851e-01,  3.2684e-01,  1.8240e-01,\n",
       "          -5.1549e-01,  1.1637e-01,  2.3197e-01, -4.4603e-01,  3.7599e-01,\n",
       "           8.1823e-02,  3.7736e-01, -2.3803e-01, -2.6549e-02, -8.5456e-02,\n",
       "           3.0498e-01, -1.4696e-01, -3.7374e-01, -6.5123e-01, -3.3511e-01,\n",
       "           2.1759e-01,  4.6516e-01,  2.4871e-01,  2.7187e-01, -1.1201e+01,\n",
       "           3.9915e-01,  5.9661e-01, -4.5479e-01, -5.1873e-02, -9.2336e-02,\n",
       "           1.2530e-01,  8.1300e-02,  4.8211e-01, -5.0019e-01, -3.7562e-01,\n",
       "          -5.5959e-01, -2.3205e-01, -7.2150e-02, -3.6729e-01, -2.2183e-01,\n",
       "           1.1704e-01, -8.5217e-01,  5.1261e-01, -2.0741e-01,  2.2689e-01,\n",
       "          -4.2181e-02, -1.1318e-02,  1.4871e-01, -4.8775e-01,  3.4635e-02,\n",
       "          -2.2489e-03, -1.1761e-01,  9.9104e-02,  4.4946e-01,  2.2776e-01,\n",
       "           2.8692e-02,  7.8599e-02, -2.1515e-01, -3.3685e-01, -3.7855e-01,\n",
       "          -2.9439e-01, -1.9770e-01, -1.2196e-01, -2.0003e-01, -5.9691e-01,\n",
       "          -8.3629e-01, -2.2328e-01,  3.9853e-01,  1.9711e-01,  7.5107e-01,\n",
       "           5.5481e-01,  9.2018e-01,  1.9608e-01,  9.9100e-04, -1.2666e-01,\n",
       "          -9.7408e-01,  7.0969e-02, -2.4854e-01, -4.0012e-01, -1.6147e-01,\n",
       "           4.8928e-01, -8.6302e-01,  3.4311e-01, -5.9079e-01,  2.6207e-02,\n",
       "           4.0531e-01,  1.8459e-01,  2.1992e-01, -9.5999e-02, -2.3044e-01,\n",
       "          -3.0278e-01, -1.3702e-01, -5.9022e-02, -2.5114e-01, -1.2737e-01,\n",
       "          -2.5891e-01,  5.0555e-01,  3.5438e-01, -2.8749e-01, -3.9309e-02,\n",
       "          -1.1568e-01, -3.1045e-01,  1.9329e+00,  1.2062e-01,  1.6726e-01,\n",
       "          -4.2158e-02,  4.6910e-01, -8.1837e-01, -7.0912e-01,  3.0471e-01,\n",
       "          -2.3814e-01,  4.2634e-01,  2.4494e-01,  2.3095e-01, -1.7888e-01,\n",
       "          -8.0222e-01, -2.6667e-01,  1.5777e-01,  1.0748e-01,  7.1191e-01,\n",
       "          -6.5700e-01, -8.5292e-02, -2.1365e-02,  2.1271e-01, -4.8840e-01,\n",
       "          -1.4869e-01, -5.8890e-01, -5.4039e-01,  2.7521e-01, -4.6152e-01,\n",
       "           1.3853e-01,  3.2709e-01, -2.2909e-01,  4.0038e-01, -4.5563e-01,\n",
       "           3.4688e-01,  4.5553e-01, -1.7329e-02,  2.2751e-02, -3.3371e-02,\n",
       "           7.0430e-01,  3.4745e-01, -4.5575e-01, -5.2940e-01,  3.0918e-02,\n",
       "           1.1775e-01,  2.1379e-01,  9.5586e-02, -3.6899e-01,  2.6988e-01,\n",
       "          -5.6571e-01, -1.4278e-01,  4.3926e-01,  1.4755e-01, -4.6461e-02,\n",
       "           1.4042e-01, -1.7399e-02,  4.8012e-01, -7.1944e-01, -6.2705e-01,\n",
       "          -8.2767e-01, -1.4704e-01,  4.4293e-01, -1.7528e-01,  2.2186e-02,\n",
       "          -4.5141e-01, -2.8808e-02, -5.6246e-02,  1.6717e-02,  5.2251e-02,\n",
       "           1.0042e-01,  2.3943e-01, -2.8971e-01, -2.2727e-02, -1.0481e-01,\n",
       "          -5.7892e-01, -3.1107e-02, -3.6378e-01, -2.5755e-01, -6.1797e-01,\n",
       "          -3.8754e-01, -3.6484e-02, -1.4533e-01, -3.6982e-02,  1.1748e-01,\n",
       "           2.2142e-01, -5.9427e-02, -4.8068e-01,  5.4573e-01,  7.3312e-01,\n",
       "           1.8012e-01, -5.1392e-01, -4.9430e-01, -1.9693e-01,  2.0377e-01,\n",
       "           2.3330e-01,  1.1983e-01,  5.5191e-02, -4.7194e-01, -5.7558e-01,\n",
       "          -1.2499e-01,  7.8875e-01,  3.3228e-02, -6.5929e-01,  3.5362e-01,\n",
       "           6.9088e-01, -9.5645e-03, -2.2830e-02, -1.5599e-01, -9.1710e-02,\n",
       "          -5.8736e-01, -4.3154e-01,  5.0625e-01, -2.4295e-01,  1.1049e-01,\n",
       "           1.9728e-01, -7.2524e-02, -1.0005e-01,  5.3588e-01, -9.5062e-02,\n",
       "          -2.7950e-01, -8.6410e-02, -3.9340e-01,  5.1576e-01,  2.7424e-01,\n",
       "          -1.9549e-01,  5.6173e-01,  2.3240e-01, -1.2735e-01, -1.4786e-01,\n",
       "           2.3027e-01,  9.9244e-01, -4.0137e-01, -3.3598e-02,  8.1366e-01,\n",
       "           1.0305e-01, -2.1392e-01,  1.5834e-01, -3.8121e-02, -8.8192e-02,\n",
       "           2.0739e-01,  8.7651e-02, -2.5020e-01,  2.3226e-03,  3.5963e-01,\n",
       "           4.2640e-01, -6.7331e-01, -7.4985e-01, -1.9206e-01,  1.0446e+00,\n",
       "           6.1729e-02,  6.4496e-01, -5.7890e-02,  5.4964e-01,  6.1821e-01,\n",
       "          -9.8557e-01,  7.6061e-02, -4.3122e-01,  3.9366e-01,  2.8689e-01,\n",
       "          -6.8691e-02, -4.0052e-02, -2.6205e-01, -4.6167e-01,  6.5670e-01,\n",
       "           3.9456e-01, -8.8600e-01, -1.0295e-01,  7.0002e-01, -5.3098e-01,\n",
       "           4.6289e-01, -4.0711e-02,  1.1266e-01,  2.3777e-01,  3.0915e-01,\n",
       "          -5.2257e-02,  2.6316e-01, -7.0602e-01, -5.3072e-01, -1.5684e-01,\n",
       "           7.0533e-01,  2.6845e-01,  1.0638e-01, -2.0033e-01, -1.2678e-01,\n",
       "          -1.1182e-01, -5.4481e-01, -2.0128e-01,  2.5046e-01,  3.6011e-02,\n",
       "           5.4912e-02, -1.1575e-01, -1.2845e-01,  3.6003e-01, -1.3823e-01,\n",
       "           3.0107e-01,  2.9338e-01,  7.1578e-01, -1.6396e-01, -3.3664e-02,\n",
       "          -3.0881e-01,  2.5137e-01, -2.2238e-01,  3.6637e-01, -1.5392e-02,\n",
       "          -1.8491e-01,  1.7782e-01,  9.6676e-02, -5.6080e-01,  1.4189e-01,\n",
       "          -4.6153e-01,  6.8162e-01, -7.7716e-02, -2.5072e-01,  3.0992e-01,\n",
       "          -6.8037e-03,  6.0858e-02, -1.6750e-01,  3.8481e-01,  8.8749e-02,\n",
       "           1.6619e-01,  7.4941e-01,  4.3112e-02,  4.8636e-02, -7.9033e-01,\n",
       "           1.3545e-01, -6.8932e-01, -8.9275e-02,  6.2460e-02, -2.7035e-01,\n",
       "          -3.1990e-01, -1.4526e-01,  1.2194e-01,  9.2611e-02,  6.9980e-01,\n",
       "           8.2720e-01, -3.8795e-01, -6.7418e-01, -3.4328e-01,  1.4535e-01,\n",
       "           9.3662e-02, -1.3789e-01,  2.4402e-02,  7.1184e-02,  4.8766e-01,\n",
       "          -6.0241e-01, -1.7335e-01, -2.8791e-01,  2.4133e-01, -2.5475e-01,\n",
       "           4.0923e-01, -2.8531e-01,  1.0002e-01,  2.2536e-01,  3.2184e-01,\n",
       "           3.9911e-02,  5.4407e-01,  2.0999e-01, -5.2641e-01,  5.0776e-02,\n",
       "          -1.8387e-01, -4.2868e-01,  1.9461e-02, -3.9030e-01,  4.1012e-01,\n",
       "           8.4722e-02,  4.1263e-01,  8.6386e-01]], device='mps:0'),\n",
       " tensor([[-3.3768e-01, -4.6481e-01, -6.2210e-01, -8.1416e-01, -2.9414e-01,\n",
       "          -1.9479e-01, -4.0387e-01,  9.9047e-02, -2.8047e-01,  4.2683e-01,\n",
       "           1.7397e-01,  4.4599e-01,  2.9501e-01,  2.2080e-01, -9.4703e-02,\n",
       "           1.4239e-01,  2.5765e-01,  6.0775e-01,  1.9895e-01,  6.3488e-02,\n",
       "           5.8876e-01,  3.1990e-01,  8.0984e-02,  1.3602e-01, -3.0591e-01,\n",
       "          -8.4479e-03, -3.7276e-01,  1.9821e-02,  2.3866e-01, -5.2653e-01,\n",
       "          -1.6822e-01,  2.8794e-01,  2.4843e-02,  8.4539e-01, -4.8419e-02,\n",
       "           2.0212e-02,  1.2750e-01, -5.0192e-01,  9.9579e-01, -1.2761e-01,\n",
       "           5.4403e-01,  7.8283e-02,  1.0369e-01,  8.8915e-02,  1.8765e-01,\n",
       "          -1.6261e-01, -3.6062e-02, -1.0706e+00,  1.2329e-01, -3.4555e-01,\n",
       "           7.2268e-02, -4.7083e-02,  4.6360e-01,  5.1868e-01, -6.7019e-01,\n",
       "           1.6261e-01,  3.2792e-02,  2.2995e-01, -2.1694e-01,  1.1038e-01,\n",
       "           7.4558e-03,  5.6900e-01,  6.7344e-02, -1.4505e-01,  2.8295e-01,\n",
       "          -5.2370e-01, -3.7244e-01, -5.0279e-01, -8.7523e-02, -1.8651e-01,\n",
       "          -3.1916e-02,  1.5664e-01, -3.6497e-02, -2.2045e-01, -6.3025e-02,\n",
       "          -3.2649e-01,  4.2967e-02,  1.5797e-01, -5.8107e-01,  3.8411e-01,\n",
       "           3.9565e-01,  2.3745e-02,  5.4289e-02, -1.1359e-01, -7.1719e-02,\n",
       "          -1.6827e-01,  3.9099e-02,  1.8779e-01,  3.7147e-01,  4.8397e-02,\n",
       "           2.5522e-01,  4.7280e-02, -5.6758e-02, -2.4723e-01,  4.6075e-02,\n",
       "          -4.7519e-02,  3.1875e-01, -5.5995e-02,  1.9624e-01, -2.6799e-01,\n",
       "           9.7410e-02,  1.4688e-01,  5.9810e-02,  2.7196e-01, -4.8962e-01,\n",
       "          -3.2098e-01, -4.2786e-02, -8.0849e-02, -8.8601e-01, -3.6456e-02,\n",
       "           1.9674e-01, -1.1486e-01,  2.7040e-02, -4.2395e-01, -6.3180e-01,\n",
       "           3.0049e-01, -2.0457e-01, -9.6426e-03, -1.7570e-01,  9.9059e-01,\n",
       "           4.3490e-01, -3.4696e-01, -1.1263e-01, -2.5470e-01,  6.0527e-01,\n",
       "          -1.0639e-01,  4.7009e-01, -4.2398e-01,  2.4011e-01,  2.8650e-01,\n",
       "           5.3017e-02,  1.3740e-01, -3.7720e-01, -8.4838e-02,  1.9698e-01,\n",
       "           4.8240e-02,  1.5980e-01, -4.0447e-01,  9.3066e-02,  5.4071e-02,\n",
       "           5.1962e-02, -3.8386e-01, -3.4960e-01, -1.9138e-01, -1.1688e-01,\n",
       "          -4.1908e-02, -1.1926e-01, -1.8461e-01,  2.1052e-01, -3.5103e-01,\n",
       "          -9.6035e-01, -5.3399e-01,  3.0840e-01, -1.5417e-01,  1.2151e-01,\n",
       "          -2.4952e-01,  4.1441e-01,  4.3859e-01,  6.2603e-02,  1.5746e-01,\n",
       "           2.1348e-01, -2.1238e-01,  3.7273e-01, -4.6776e-01,  2.3838e-01,\n",
       "          -8.7344e-02, -3.3696e-01, -1.8624e-01,  3.1928e-01, -2.5442e-03,\n",
       "           6.2356e-01, -7.3209e-02, -4.8009e-02, -4.0704e-01,  2.5262e-01,\n",
       "           5.9241e-02,  4.8445e-01,  3.8952e-01, -4.4201e-01, -3.8018e-01,\n",
       "          -1.5241e-01, -1.7678e+00,  1.6758e-01, -4.3604e-01, -7.4550e-01,\n",
       "          -5.1156e-01,  3.5686e-01,  2.9583e-01,  1.5882e-01, -2.5447e-01,\n",
       "          -1.5313e-01, -2.5278e-01,  3.6020e-01,  5.1506e-01,  1.8800e-01,\n",
       "           3.9236e-01,  5.9554e-01, -1.5338e-01,  2.1861e-01,  1.3129e-01,\n",
       "          -4.3524e-01,  2.6882e-01,  2.8859e-01,  3.9034e-02,  9.9718e-02,\n",
       "           3.3906e-01, -1.6364e-01,  3.9568e-02,  4.1276e-01,  3.7972e-01,\n",
       "          -1.5094e-01, -2.5921e-01, -3.5837e-01,  2.4099e-03, -3.7432e-01,\n",
       "           2.0811e-01, -2.7629e-01, -7.7800e-02,  5.3125e-01,  1.9003e-01,\n",
       "           1.3113e-01, -1.7225e-01, -7.4317e-02, -2.4554e-01,  3.0840e-02,\n",
       "           4.2718e-01, -9.9421e-03, -4.2773e-02,  1.3128e-01, -2.3733e-01,\n",
       "           1.5342e-01, -6.1789e-01,  5.7677e-01,  3.0020e-01, -7.6470e-03,\n",
       "          -3.2925e-01, -2.1087e-01,  2.9031e-01, -1.0637e-01, -3.4515e-01,\n",
       "          -1.7183e-01, -6.9551e-01, -6.2138e-01, -1.4339e-01,  1.1568e-01,\n",
       "           1.2025e-01, -1.1472e-01,  3.9308e-01,  4.5737e-01,  1.6433e-01,\n",
       "           7.6075e-02,  7.1991e-02,  3.8295e-01,  4.9459e-01, -2.8362e-01,\n",
       "           6.0418e-02,  1.1870e-01, -2.8562e-01,  1.5454e-01, -4.8709e-01,\n",
       "          -3.9105e-01, -1.2181e-01, -4.9165e-02,  3.3770e-01,  8.9816e-02,\n",
       "           5.1637e-01,  3.4205e-01, -6.0390e-02, -2.5048e-01,  1.7446e-01,\n",
       "          -5.2153e-01,  2.8290e-01,  2.4784e-01, -2.7977e-01, -6.5182e-01,\n",
       "          -1.1408e+00,  9.4018e-02,  2.5095e-01, -4.5646e-02, -7.6627e-01,\n",
       "          -5.9897e-02,  7.8919e-02, -1.8916e-01, -2.0356e-01, -1.4033e-01,\n",
       "           7.8545e-02,  3.4461e-05,  2.4067e-01, -5.4801e-02, -2.5837e-01,\n",
       "          -8.2165e-02,  3.5642e-01, -2.3314e-01,  8.6870e-02,  2.1156e-01,\n",
       "           8.2423e-02,  1.1425e-01,  5.6183e-01,  7.4203e-04, -3.0863e-01,\n",
       "          -4.5701e-01,  4.0755e-01,  1.2221e-01, -3.8973e-01,  1.4748e-01,\n",
       "           2.8642e-01,  9.4576e-02,  5.4392e-01, -1.6586e-01, -4.0480e-01,\n",
       "           1.7483e-01,  1.0660e+00,  1.2018e-02,  1.1238e-01, -3.5450e-01,\n",
       "          -8.0777e-01, -1.0377e-01, -5.5358e-01, -5.3291e-02, -6.3820e-01,\n",
       "           3.0563e-01, -8.4184e-02, -1.2414e-01,  2.8581e-01, -2.6980e-01,\n",
       "          -2.7250e-01,  5.0646e-01,  2.6721e-01,  3.1125e-02, -1.3256e-01,\n",
       "           1.3422e-01, -4.9375e-01, -3.4143e-01, -2.2365e-01,  4.2542e-01,\n",
       "           3.8589e-01,  2.2889e-01,  1.0212e-01, -4.7329e-02, -2.6458e-01,\n",
       "           6.1189e-01, -1.2357e-01,  2.0030e-01,  4.7322e-01, -4.0161e-01,\n",
       "          -1.6498e-01,  1.8401e-01, -2.0616e-01, -2.6173e-01,  1.7437e-01,\n",
       "           3.2921e-01, -4.6093e-01, -8.4930e-01,  5.3896e-02,  8.9570e-01,\n",
       "           5.4385e-01,  4.6203e-01, -2.2302e-01,  1.4864e-01,  3.4476e-01,\n",
       "          -4.8196e-01,  1.2962e-02, -1.0839e-01, -2.2991e-01,  5.6279e-02,\n",
       "           4.6510e-01,  1.2802e-01, -7.8708e-02,  5.5343e-02,  2.2674e-01,\n",
       "           3.6804e-01, -4.2439e-01, -1.7114e-01,  1.3951e-01,  5.4100e-02,\n",
       "          -1.3904e-01, -4.8960e-01,  3.2708e-01, -8.5187e-01, -3.0676e-01,\n",
       "          -5.8296e-02, -1.6362e-01, -4.9688e-02, -6.7789e-02, -2.0386e-01,\n",
       "          -9.3709e-03, -3.5436e-01,  3.6808e-01, -3.5364e-02, -8.1521e-02,\n",
       "           3.3219e-01,  1.1997e-01, -2.2318e-01,  4.9466e-01,  9.5236e-02,\n",
       "          -2.1645e-01,  6.7533e-02, -5.7884e-01,  1.5346e-01,  3.4215e-02,\n",
       "          -4.9110e-01,  8.0507e-02,  2.3801e-01,  3.1598e-01, -2.8124e-01,\n",
       "          -3.4395e-01,  6.5722e-02,  4.3754e-01,  2.0573e-01,  1.6585e-01,\n",
       "          -1.9877e-01,  5.4861e-01, -6.3763e-02,  4.4029e-01, -1.5478e-01,\n",
       "           1.0711e-01, -7.7308e-02, -2.0106e-01, -1.0772e-01, -3.2712e-01,\n",
       "           4.5347e-01,  2.2656e-01, -2.8464e-01,  2.2987e-01, -1.1357e+01,\n",
       "           1.9262e-01,  7.7206e-01, -4.3318e-01,  4.2479e-02,  1.0594e-02,\n",
       "           1.5041e-01, -3.4225e-01,  2.2543e-01, -2.3058e-01, -2.6677e-01,\n",
       "          -2.2076e-01, -1.5078e-01, -3.1470e-02, -2.5320e-01, -4.6015e-01,\n",
       "          -1.4622e-01, -6.4136e-01,  2.1944e-01, -9.9293e-02,  2.8629e-01,\n",
       "          -7.6091e-01,  3.8348e-01,  3.2011e-01, -2.5654e-01, -3.1559e-01,\n",
       "          -6.1707e-01, -3.5252e-01,  2.9561e-02, -6.3981e-02, -1.6087e-01,\n",
       "           2.3747e-01, -9.0179e-02,  5.1059e-01,  2.8978e-02, -6.9979e-01,\n",
       "          -2.5860e-02,  2.8272e-02,  3.8298e-01, -2.5762e-01, -4.8326e-02,\n",
       "          -2.8105e-03,  7.2552e-02,  4.2049e-01,  3.3683e-01,  6.4375e-01,\n",
       "           1.0586e+00,  5.7250e-01, -7.7384e-02, -3.6332e-01,  3.6696e-02,\n",
       "          -6.7007e-01,  4.5624e-01, -1.1492e+00,  2.1429e-01, -1.1567e-01,\n",
       "           3.4463e-01, -4.9085e-01, -2.0521e-01, -1.5873e-01, -1.9828e-01,\n",
       "           2.0482e-01,  3.5390e-02,  1.3564e-01, -1.5909e-01, -5.1653e-01,\n",
       "           5.3072e-01,  9.0672e-02,  7.7840e-02, -2.5712e-02, -1.9669e-01,\n",
       "           4.2710e-02,  3.0120e-01,  1.4872e-02,  2.3216e-01,  7.5144e-01,\n",
       "          -5.1070e-02, -7.7279e-02,  8.9092e-01, -1.4556e-01,  9.1886e-02,\n",
       "           3.8353e-01,  4.8072e-02, -3.4254e-01, -3.5937e-01, -2.1221e-01,\n",
       "          -1.6800e-01,  4.6926e-01, -1.1843e-01,  8.6201e-01, -6.8128e-02,\n",
       "          -1.6726e-01, -1.5050e-01, -2.1292e-01,  5.1506e-01,  7.7494e-01,\n",
       "          -2.5575e-01,  2.2707e-01,  1.4611e-01, -1.1343e-01, -8.4562e-01,\n",
       "          -1.4919e-01,  6.7335e-02, -5.4559e-01,  1.9294e-02, -4.0234e-01,\n",
       "           7.9680e-02,  4.5716e-02,  1.3253e-01,  5.9934e-01, -3.0746e-01,\n",
       "           1.2101e-01,  1.3462e-01,  6.0853e-02, -3.4922e-01, -5.2223e-01,\n",
       "           1.8566e-01,  7.0380e-01, -3.3715e-01, -6.6149e-01, -4.6630e-02,\n",
       "          -9.8133e-02, -6.0713e-02, -1.6625e-01, -2.4037e-01,  2.8465e-01,\n",
       "          -4.7341e-01,  2.0610e-01,  1.3507e-01, -1.7528e-01,  6.2010e-01,\n",
       "           2.7111e-01,  2.2351e-01,  5.8859e-01, -5.0638e-01, -4.3878e-01,\n",
       "          -3.9291e-01, -1.1346e-01,  6.9041e-01,  1.6982e-01,  2.5938e-01,\n",
       "           2.3190e-01,  1.2365e-01, -3.8919e-01,  4.9317e-01, -2.4364e-01,\n",
       "           1.8233e-01,  5.4943e-01,  5.8668e-02, -6.8532e-02, -2.1071e-01,\n",
       "          -2.5934e-01, -2.3999e-01, -7.3814e-02, -3.8973e-01, -1.7544e-01,\n",
       "          -2.5720e-01,  1.0004e-01, -1.4603e-01, -4.6020e-01, -5.6324e-02,\n",
       "           3.7852e-01, -9.6518e-02, -2.1719e-01,  2.8895e-01,  3.4693e-01,\n",
       "           1.7226e-02, -5.7595e-01, -2.2642e-01, -1.2500e-01, -1.2374e-01,\n",
       "           4.6828e-01, -8.7919e-02, -4.8554e-01, -8.9963e-01, -3.7558e-01,\n",
       "          -1.9939e-01,  1.2977e+00, -4.5006e-01, -3.2144e-01,  3.2487e-01,\n",
       "          -1.0346e-01,  4.9148e-03, -7.8642e-03, -7.5161e-01,  6.0072e-02,\n",
       "          -1.0634e-01, -3.2259e-01,  4.6652e-01, -3.7058e-01,  4.1556e-01,\n",
       "           5.0854e-01,  9.7597e-02, -1.6986e-01,  4.0686e-01, -2.0387e-01,\n",
       "          -3.7267e-01, -1.4208e-01, -4.0737e-01, -1.6164e-01,  1.2207e-01,\n",
       "          -3.4907e-01,  2.6476e-01, -2.7378e-01, -5.5332e-01,  2.8389e-01,\n",
       "           1.8953e-01,  1.2957e+00,  2.2064e-02, -1.2364e-01,  1.7208e-01,\n",
       "           1.3218e-01,  5.8423e-01,  3.0917e-01, -1.3161e-01,  4.2529e-01,\n",
       "           4.8068e-01, -1.5222e-01, -5.5609e-01,  2.3531e-01,  1.6932e-01,\n",
       "           4.4220e-01, -6.2344e-01, -2.1842e-01, -1.6518e-01,  4.1796e-01,\n",
       "           1.1371e-01,  7.8666e-02, -6.8582e-02,  3.9284e-01,  6.3759e-01,\n",
       "          -1.1938e+00,  2.1497e-01,  2.2229e-01, -2.0684e-01, -1.2715e-01,\n",
       "          -4.0257e-01, -2.2954e-01, -1.3534e-02, -3.3223e-01,  2.7377e-01,\n",
       "           6.6125e-01, -1.0914e+00,  2.9638e-01,  4.3755e-01, -9.0971e-02,\n",
       "           2.1417e-01,  1.1765e-03,  4.5157e-01,  3.2632e-01,  5.1999e-02,\n",
       "           3.4399e-01,  4.8163e-02, -9.4013e-02, -4.5627e-01,  1.3804e-01,\n",
       "           4.8141e-01,  4.1808e-02,  1.9851e-02,  1.9692e-01, -6.7097e-02,\n",
       "          -6.3515e-02,  1.6088e-01, -4.8925e-01, -2.0541e-01,  1.2683e-01,\n",
       "           4.8502e-02, -7.2628e-01,  4.6910e-03,  3.4048e-01, -1.9755e-01,\n",
       "           4.5357e-01,  6.7692e-01,  3.7126e-01, -1.9078e-01,  1.5138e-01,\n",
       "          -3.2037e-01,  2.1722e-01,  1.8209e-02,  6.7763e-01, -5.0815e-01,\n",
       "          -2.9356e-01, -5.1337e-02,  2.0091e-01, -3.6379e-01,  2.2450e-01,\n",
       "          -2.0343e-01,  2.6630e-02, -2.0492e-01, -1.8590e-01,  3.1568e-01,\n",
       "          -7.0606e-02,  2.6384e-04, -9.9399e-01,  2.2373e-01,  1.3999e-02,\n",
       "          -2.2803e-01,  7.3298e-01,  9.6779e-02,  2.6014e-01, -3.9676e-02,\n",
       "           3.5154e-02, -7.4621e-01, -5.6404e-01,  2.7920e-01,  1.0175e-01,\n",
       "          -1.6896e-01, -2.3039e-01, -4.9923e-02,  9.9302e-02,  5.6410e-01,\n",
       "           4.5749e-01, -3.5735e-01, -6.4311e-01, -3.6350e-01,  8.2885e-02,\n",
       "           1.5985e-01, -2.7533e-01,  2.6099e-01,  5.0535e-01, -1.6657e-01,\n",
       "          -1.0631e+00, -1.9577e-01, -6.0604e-02,  9.8545e-02, -2.6191e-01,\n",
       "           7.9551e-01, -5.0371e-01,  3.3424e-02,  1.2208e-01,  3.2285e-02,\n",
       "           3.2121e-01,  6.3031e-01, -9.6469e-03, -5.9842e-01, -1.9643e-01,\n",
       "           1.9220e-01, -2.9221e-01, -2.2856e-01, -4.2978e-01,  3.5499e-01,\n",
       "           1.9024e-01, -2.0146e-02,  6.6552e-01]], device='mps:0'),\n",
       " tensor([[-0.0932, -0.6293, -0.4879,  ...,  0.3041, -0.1958,  0.5576],\n",
       "         [-0.1242, -0.5179, -0.4093,  ...,  0.3691, -0.0529,  0.5134],\n",
       "         [-0.1672, -0.5249, -0.4540,  ...,  0.3771,  0.2016,  0.5340]],\n",
       "        device='mps:0'),\n",
       " tensor([[-2.8441e-01, -5.5394e-01, -5.3487e-01, -8.6137e-01, -1.6500e-01,\n",
       "          -1.5627e-01, -6.9154e-01,  6.2842e-01, -3.2176e-01,  7.4840e-01,\n",
       "           2.7524e-01,  6.2658e-01,  4.8185e-02,  8.1655e-02, -5.4587e-01,\n",
       "          -1.8390e-01, -8.5546e-02,  3.8717e-01,  2.1180e-01,  6.1308e-01,\n",
       "           8.1107e-01,  6.8047e-01, -1.9768e-01,  2.1503e-01, -5.0498e-01,\n",
       "           3.5649e-01, -6.9646e-02,  6.8591e-01,  2.3685e-01, -4.8309e-01,\n",
       "           4.5488e-02,  5.1755e-01, -7.6515e-02,  1.6821e+00, -1.0095e-01,\n",
       "          -2.5680e-01,  2.3473e-01, -5.4682e-01, -4.1870e-01, -4.8584e-01,\n",
       "           6.0923e-01, -3.5607e-01,  1.1066e-02,  1.0019e-01, -1.2748e-01,\n",
       "           4.3426e-01,  3.3175e-01, -4.5028e-01,  2.1806e-01, -8.6036e-01,\n",
       "          -6.0268e-01, -1.1943e-01,  1.3884e-01,  7.5491e-01, -7.6410e-01,\n",
       "          -7.1493e-01,  8.1095e-02, -1.2005e-01, -1.3680e-01,  3.9091e-02,\n",
       "           2.6227e-01,  2.0679e-01, -8.1044e-02,  6.1526e-02,  1.7546e-01,\n",
       "           1.7241e-01, -1.9290e-01, -7.4059e-01, -4.7855e-01,  7.6078e-02,\n",
       "           9.5219e-02,  1.1330e-01,  1.1899e-01, -1.9241e-01,  3.2532e-01,\n",
       "          -2.3076e-01,  2.7391e-01, -3.3521e-01, -1.3154e-01, -7.0176e-02,\n",
       "          -4.4051e-02,  8.9766e-02,  1.9541e-01, -1.9095e-01, -1.4401e-01,\n",
       "           1.1033e-02,  5.0526e-01,  5.0223e-02, -8.2932e-02, -4.1653e-02,\n",
       "           1.5756e-02,  1.8359e-02, -6.3429e-01, -4.0049e-01,  1.9855e-01,\n",
       "           7.6685e-03,  3.6200e-01, -3.5243e-01,  1.8487e-01, -2.3329e-01,\n",
       "           2.0495e-01, -1.3106e-01, -1.0602e-01,  5.0891e-02, -6.4400e-01,\n",
       "          -5.0268e-01,  1.7121e-01,  1.3393e-05, -5.4945e-01,  1.4619e-02,\n",
       "          -4.0735e-01, -6.1942e-01,  3.2930e-02, -2.3743e-01, -5.5993e-02,\n",
       "           1.5373e-01,  6.8113e-02, -4.6111e-04,  3.8857e-03, -7.4266e-02,\n",
       "           5.8578e-01, -4.1646e-02,  1.6548e-01,  9.9639e-02,  2.6984e-01,\n",
       "          -3.2575e-01,  6.6620e-01, -3.1006e-01,  2.6532e-01,  4.3677e-01,\n",
       "          -1.2551e-01,  2.2294e-03, -3.9518e-01, -9.1430e-02,  3.8994e-01,\n",
       "           3.5504e-01, -2.5205e-01, -1.6596e-01, -3.2194e-01,  1.2660e-01,\n",
       "          -1.1306e-01, -1.9384e-01, -1.2373e-01, -4.7651e-01, -9.7027e-02,\n",
       "          -2.5137e-01, -1.8891e-01,  9.8524e-03, -1.3107e-02, -5.6441e-01,\n",
       "          -3.7041e-01, -1.0094e+00,  8.9048e-01, -7.7338e-02,  1.0971e-01,\n",
       "          -5.2245e-02,  1.3917e-01,  1.3157e-01,  2.4190e-01,  3.4615e-01,\n",
       "           4.0161e-01, -3.0681e-01,  1.7669e-01, -6.3478e-01, -2.1620e-01,\n",
       "          -1.6021e-02,  5.5562e-02, -4.2343e-01,  4.4272e-01,  1.4258e-01,\n",
       "           6.6604e-02, -3.1665e-01,  9.0541e-02, -2.5514e-01,  2.3988e-01,\n",
       "          -5.2097e-01,  1.3542e-01,  8.6542e-03,  9.9876e-02, -2.3345e-01,\n",
       "           1.6611e-01, -1.2452e+00,  8.5164e-02, -1.3329e-01, -3.9477e-02,\n",
       "          -3.4149e-01,  3.9552e-01,  3.1089e-01,  3.2796e-01, -1.5308e-01,\n",
       "          -8.1520e-02, -7.1932e-02,  9.9920e-02,  2.7144e-02, -2.8755e-01,\n",
       "           4.6967e-01,  6.6779e-01,  4.0208e-01, -1.6045e-01,  9.8117e-02,\n",
       "           2.1663e-02,  2.2509e-01, -6.3057e-02,  1.4737e-01,  3.4920e-01,\n",
       "           1.6982e-01, -1.9862e-01, -5.2368e-02,  3.0330e-01, -4.0755e-02,\n",
       "          -2.1991e-01, -2.3430e-02, -4.2824e-01, -1.0580e-01, -3.9187e-01,\n",
       "           1.2879e-01, -3.9608e-01, -1.2370e-01,  2.9763e-01, -1.3954e-01,\n",
       "          -1.9156e-01,  4.9995e-02,  1.7278e-01, -1.4398e-01,  1.1770e-01,\n",
       "           1.3569e-01, -3.2259e-01,  9.6180e-02, -3.1440e-01, -1.8272e-01,\n",
       "           5.7590e-02, -4.0861e-01,  7.0641e-01,  2.7090e-01, -2.1076e-01,\n",
       "           2.2706e-01, -3.7797e-01,  7.5726e-02, -1.9180e-02,  1.7069e-01,\n",
       "          -5.9430e-03, -3.4191e-01, -9.1396e-01,  2.0747e-01,  6.7531e-02,\n",
       "           2.8930e-01, -2.4971e-02,  4.7363e-01,  8.3110e-01,  1.9939e-01,\n",
       "          -3.0489e-02, -1.6327e-01,  4.5326e-01,  3.1228e-01, -2.1691e-01,\n",
       "          -2.6165e-02, -9.7939e-02, -8.4730e-01,  3.6313e-02, -4.6692e-01,\n",
       "          -2.0889e-02,  2.2189e-01, -3.3192e-03,  2.0753e-01, -3.7387e-01,\n",
       "           6.3320e-01,  4.4671e-01,  9.7346e-02, -3.8640e-01,  6.3860e-01,\n",
       "           1.0362e-01,  3.3258e-01, -3.3726e-01, -1.6538e-01, -5.7175e-01,\n",
       "          -6.9858e-01, -1.9525e-01,  4.4452e-01,  7.1309e-02, -4.3032e-01,\n",
       "          -1.7021e-02,  1.0086e-01,  4.6269e-02, -3.2002e-01,  4.1130e-02,\n",
       "           1.1790e-01, -1.4409e-01, -2.3553e-02, -1.8172e-01, -3.0541e-01,\n",
       "          -4.0109e-01,  3.2059e-01, -4.5628e-01, -1.9039e-02,  6.3341e-01,\n",
       "          -5.6343e-02, -2.5236e-01,  1.3631e-01, -7.7882e-03, -1.5985e-01,\n",
       "           7.7204e-02,  4.9073e-01, -1.7577e-01, -3.1271e-01,  3.3961e-01,\n",
       "          -6.8220e-02,  4.1486e-01,  3.8705e-02, -5.9873e-01, -2.8172e-01,\n",
       "           8.1087e-02,  4.2650e-01, -6.3146e-02, -1.9707e-01, -1.0664e-01,\n",
       "          -7.9628e-01, -3.1771e-01, -1.7417e-01, -3.7691e-01, -4.4016e-01,\n",
       "           1.1354e-02,  2.5387e-01, -1.6713e-01,  3.0450e-02, -2.3705e-01,\n",
       "           3.6587e-01,  9.3493e-01, -3.4931e-01,  4.2838e-02, -4.9836e-01,\n",
       "           4.5254e-02, -8.1734e-02, -6.2819e-02, -3.3827e-01, -4.0564e-01,\n",
       "           3.5635e-01,  1.3258e-01,  2.5235e-01,  2.0424e-02, -2.3089e-01,\n",
       "           3.3129e-01, -3.3075e-01, -8.9622e-02,  7.1975e-01, -2.8549e-01,\n",
       "           4.0563e-02,  2.8078e-01, -6.6874e-01, -8.5723e-02,  3.4916e-01,\n",
       "           1.6608e-01, -6.5247e-02, -8.3725e-01, -7.6890e-03,  4.5139e-01,\n",
       "           3.8668e-01,  1.8128e-01,  2.0962e-01,  5.4342e-01,  3.6529e-01,\n",
       "          -4.0580e-01,  3.7468e-01, -1.1954e-01, -1.3160e-01,  4.5812e-02,\n",
       "           1.4795e-01,  1.4958e-01, -4.0168e-01,  3.3391e-01, -1.7011e-01,\n",
       "           2.4076e-02, -4.0941e-01, -1.8527e-01, -9.4000e-03,  3.9568e-01,\n",
       "          -2.4590e-02, -4.3180e-01,  1.3466e-01, -3.7448e-01, -2.4752e-01,\n",
       "           9.1886e-02,  7.0489e-02,  3.2154e-01, -4.9037e-01, -2.5280e-01,\n",
       "          -1.1939e-01, -3.0043e-01,  5.9537e-01,  1.4481e-02, -1.3875e-01,\n",
       "           3.3885e-01, -2.4500e-01, -1.9853e-01,  3.8153e-01,  3.6541e-02,\n",
       "          -3.2929e-01,  1.7709e-01, -2.1721e-01,  2.6912e-02,  7.2715e-01,\n",
       "          -3.6499e-01, -1.2932e-02, -2.0360e-02,  2.0395e-01,  3.4879e-01,\n",
       "          -2.8039e-01, -1.2410e-01,  1.7326e-01, -4.9986e-01, -4.2984e-02,\n",
       "          -1.2823e-01,  3.1628e-01, -2.6186e-01, -1.3695e-01, -4.3339e-01,\n",
       "           1.3805e-02,  3.3696e-02, -1.2055e-01,  1.0068e-01, -2.8598e-01,\n",
       "           6.6533e-01, -2.4381e-01, -2.6331e-01,  1.3605e-01, -1.1418e+01,\n",
       "           1.1377e-01,  8.4308e-01,  1.6280e-01,  2.3584e-01, -2.9657e-01,\n",
       "           1.0418e-01,  1.1050e-01,  7.0321e-02, -3.2792e-01,  1.4004e-01,\n",
       "          -8.6370e-02,  6.6247e-03,  3.9086e-01, -3.6419e-01,  9.2481e-02,\n",
       "          -1.1838e-01, -1.0034e+00,  3.8269e-02,  2.4191e-01,  4.5080e-01,\n",
       "          -2.9394e-01,  2.1662e-02,  3.1407e-01, -3.1812e-01, -6.7698e-02,\n",
       "          -3.1501e-01, -9.0094e-02,  2.0232e-01,  1.2938e-01, -5.3503e-02,\n",
       "           1.2833e-02, -8.4343e-02,  4.0612e-01, -2.2802e-02, -4.8456e-01,\n",
       "          -1.5634e-01,  6.4033e-02,  3.7287e-02, -5.1122e-01, -4.6959e-01,\n",
       "           1.6187e-01, -2.7521e-01,  1.5326e-01,  3.0423e-01,  5.8338e-01,\n",
       "           6.4433e-01,  8.7874e-01,  9.8589e-02,  6.4743e-03, -1.2715e-01,\n",
       "          -8.6394e-01,  1.4528e-01, -6.1263e-01,  3.3415e-01, -2.5418e-01,\n",
       "           6.9162e-01, -5.7891e-01,  3.3339e-01, -4.9050e-01, -2.3375e-01,\n",
       "           4.9885e-01,  2.1698e-01,  2.7781e-02, -5.2102e-01, -9.8122e-02,\n",
       "           3.4905e-01,  8.7095e-03,  2.4822e-01, -4.3775e-01, -6.8448e-02,\n",
       "           2.5354e-02,  7.5355e-01,  6.8922e-02,  1.5600e-01,  4.6976e-02,\n",
       "          -2.9262e-01, -1.1298e-01,  1.9105e+00, -4.5921e-02,  9.8367e-02,\n",
       "           1.5618e-01,  5.6595e-01, -3.6775e-01, -8.2060e-01, -9.5813e-03,\n",
       "          -1.5806e-01, -4.0744e-02,  6.1770e-02,  5.3651e-01, -4.0122e-01,\n",
       "          -2.8241e-01, -4.2715e-02,  2.2891e-01,  8.1120e-02,  9.9514e-01,\n",
       "          -5.0647e-01, -1.3592e-01,  6.5777e-03,  6.3368e-01, -4.4815e-01,\n",
       "           2.5587e-02, -3.4930e-01, -5.6310e-02, -1.7721e-02, -3.8494e-02,\n",
       "          -1.2066e-01, -3.6498e-02,  6.6482e-02,  3.8879e-01, -5.0006e-02,\n",
       "          -1.0725e-01, -1.0326e-01, -4.3712e-02, -3.7134e-01, -1.2494e-01,\n",
       "           1.8692e-01,  3.5026e-01, -5.6838e-01, -5.4588e-01, -3.2157e-01,\n",
       "          -3.7195e-02,  2.8179e-01,  5.1425e-02, -4.7998e-01,  5.6484e-01,\n",
       "          -6.6180e-01, -1.0760e-01,  2.6979e-01,  2.1886e-01,  4.8424e-01,\n",
       "           2.3962e-01, -2.7152e-01,  4.6382e-01, -5.7533e-01, -3.5601e-01,\n",
       "          -7.6686e-01,  7.1204e-03,  3.3090e-01, -3.4110e-01,  7.7304e-02,\n",
       "           1.5510e-01, -4.8632e-02,  1.7306e-01,  1.1810e-01,  1.6733e-01,\n",
       "          -2.1232e-02, -1.6458e-02, -5.1332e-01,  2.7794e-01, -6.9124e-02,\n",
       "          -4.3402e-01,  3.8644e-02, -3.1164e-01, -1.5215e-01, -1.1196e-01,\n",
       "          -2.3435e-02,  7.5629e-02, -4.2071e-01, -1.6949e-01,  1.5603e-01,\n",
       "           1.2218e-01, -1.6835e-01, -3.5913e-01,  4.8729e-01,  9.7509e-02,\n",
       "          -5.5797e-02, -2.0874e-01, -3.8294e-01, -1.4694e-01,  1.5783e-01,\n",
       "           6.2652e-02, -6.3195e-02, -5.2337e-01, -4.0067e-01, -2.3889e-01,\n",
       "          -5.7472e-01,  1.0858e+00, -9.2285e-02, -5.5553e-01, -7.3773e-02,\n",
       "           3.4282e-01, -2.2721e-01,  2.4674e-01,  1.7079e-01,  3.8558e-01,\n",
       "          -3.3978e-01, -3.9431e-01,  6.1823e-03, -2.5758e-02,  3.9101e-01,\n",
       "           3.4840e-01,  5.9696e-01,  4.9011e-01,  2.1357e-01, -1.1559e-01,\n",
       "          -5.7487e-01,  3.2738e-01, -3.6646e-01, -3.9739e-01,  3.8141e-01,\n",
       "          -9.0689e-02,  2.4697e-02,  2.8430e-01, -3.6602e-01,  1.8617e-01,\n",
       "           6.4427e-01,  7.4935e-01, -6.0195e-02, -3.8977e-01,  5.6978e-01,\n",
       "           2.1737e-01, -1.3383e-01,  1.7105e-01,  4.2291e-01,  1.2353e-01,\n",
       "           3.0668e-01,  1.2484e-01, -2.9104e-01, -9.1789e-02,  1.9966e-01,\n",
       "           3.2000e-01, -4.7134e-01, -5.8480e-01,  6.0997e-02,  1.4006e-01,\n",
       "          -1.2669e-01,  4.8179e-01, -5.6792e-02,  3.6236e-01,  3.2851e-01,\n",
       "          -1.1371e+00,  1.4501e-01, -1.7041e-01,  1.8232e-02, -1.7073e-01,\n",
       "          -1.1745e-01, -4.2501e-01, -2.3848e-01, -2.9930e-01,  2.4263e-01,\n",
       "           2.5979e-01, -6.7001e-01, -6.7164e-02,  3.6919e-01, -2.9916e-01,\n",
       "           4.2863e-01, -4.9112e-01,  4.9717e-01,  6.8122e-01, -3.4430e-01,\n",
       "           2.7410e-01, -8.4230e-02, -6.1617e-01, -3.5603e-01,  2.0665e-01,\n",
       "           5.3170e-01, -2.7458e-02,  1.2537e-01, -2.1460e-01, -5.2783e-01,\n",
       "          -5.7901e-02, -2.4859e-01,  1.4412e-01, -3.5078e-01,  3.9901e-02,\n",
       "          -1.7903e-01, -3.2260e-01, -2.7667e-02,  4.1629e-01,  1.4599e-01,\n",
       "           2.2399e-01,  4.8948e-01,  6.3414e-01,  2.3314e-02, -2.6792e-01,\n",
       "          -9.9237e-02,  3.6744e-01, -5.3180e-02,  9.9155e-01, -7.8306e-02,\n",
       "          -1.8306e-01, -2.1731e-02,  8.6113e-03, -1.2811e-01,  3.4870e-01,\n",
       "          -2.2924e-01,  2.5329e-02, -1.6893e-01, -6.6031e-02,  2.5295e-01,\n",
       "           1.3629e-01,  3.7671e-02, -4.3246e-01,  2.0870e-01,  4.7896e-01,\n",
       "          -2.3456e-02,  4.0253e-01, -1.3764e-01,  4.6894e-01, -4.5455e-01,\n",
       "          -2.9003e-01, -5.5458e-01,  2.4767e-01,  2.0208e-01, -8.0991e-02,\n",
       "          -5.6740e-01, -2.9935e-01,  3.7720e-01, -1.4502e-01,  6.8787e-01,\n",
       "           7.6811e-01,  3.6459e-02, -5.9048e-01, -4.3508e-02, -1.0246e-01,\n",
       "          -1.8006e-01,  4.1084e-01,  1.0845e-01, -2.2925e-01, -1.1876e-01,\n",
       "          -3.1507e-01, -9.5207e-02, -3.5142e-01,  3.5755e-01, -9.2964e-02,\n",
       "           6.9756e-01,  1.4460e-01,  7.2438e-02,  3.5404e-01,  3.9432e-01,\n",
       "          -2.9369e-01,  7.8359e-01,  2.2095e-01, -4.6333e-01,  1.8300e-01,\n",
       "          -7.5266e-02, -2.9379e-01,  6.5779e-01, -7.0699e-01,  2.5311e-01,\n",
       "           1.8474e-01,  8.1891e-01,  4.0170e-01]], device='mps:0'),\n",
       " tensor([[-1.7881e-01, -6.6499e-01, -6.6250e-01, -3.8064e-01, -5.3327e-01,\n",
       "          -3.3321e-01, -3.9274e-01,  6.4222e-01,  8.8745e-02,  7.6129e-01,\n",
       "           8.2894e-02,  3.2500e-03,  1.0414e-01, -4.3917e-01, -5.0308e-01,\n",
       "          -3.5619e-02,  2.7375e-02,  2.0976e-01, -4.1405e-02, -1.2046e-01,\n",
       "           5.6496e-01,  2.5051e-01, -1.4612e-01,  2.8763e-04, -8.1412e-01,\n",
       "           8.2624e-02, -7.9140e-02,  5.6947e-02,  1.5093e-01, -1.0158e+00,\n",
       "          -1.4126e-01,  6.9705e-01,  1.8335e-01,  9.9361e-01, -1.6907e-01,\n",
       "          -1.9637e-01, -1.8171e-02, -6.3389e-01,  3.5983e-01, -5.7950e-01,\n",
       "           5.2398e-01,  5.7851e-02,  4.3936e-01,  2.7855e-01,  5.7436e-01,\n",
       "           2.7038e-02,  4.9720e-01, -8.0865e-01,  4.8602e-01, -5.4393e-01,\n",
       "          -1.9932e-01,  2.0532e-01,  3.6735e-01,  3.7939e-01, -6.7269e-01,\n",
       "          -3.8448e-01,  3.0598e-01,  4.6057e-01, -3.3349e-01,  4.5421e-01,\n",
       "          -1.9411e-01,  4.1896e-01, -1.0249e-01,  9.7633e-02,  3.8970e-01,\n",
       "           1.2227e-01, -3.9585e-01, -3.5980e-01, -4.4311e-01, -8.7480e-02,\n",
       "          -1.0813e-01,  4.8131e-01, -8.8099e-02, -4.0425e-01,  3.8511e-01,\n",
       "          -2.1998e-01, -6.7930e-02, -4.2645e-01, -2.3321e-01, -2.7112e-01,\n",
       "           1.4015e-01,  3.0284e-01, -1.5922e-01, -2.6777e-01,  1.0429e-01,\n",
       "           4.4270e-02,  3.5869e-01,  3.5654e-01,  2.3767e-01, -8.3747e-02,\n",
       "           6.2927e-01,  2.1842e-01, -4.6764e-01, -4.1219e-02,  3.9541e-01,\n",
       "          -3.6046e-01,  3.3566e-01, -1.4502e-01,  5.9178e-02, -3.6167e-01,\n",
       "           1.1733e-01,  1.2207e-01,  1.1345e-01, -3.2544e-02, -3.8636e-01,\n",
       "          -4.7332e-01,  1.5984e-01, -2.5880e-02, -6.6205e-01, -1.8990e-01,\n",
       "          -5.3871e-02, -1.1326e-01,  3.0193e-01, -5.6963e-01, -1.6666e-01,\n",
       "          -1.6559e-01,  4.9283e-02, -2.0125e-01, -1.9125e-01,  6.4155e-01,\n",
       "           1.8228e-01, -3.1892e-01,  2.6347e-02,  2.3126e-01,  5.6642e-01,\n",
       "          -4.2079e-01,  4.6093e-01, -3.1190e-01,  4.2326e-01,  2.3812e-01,\n",
       "           4.1860e-02, -8.6547e-02, -5.1200e-01,  8.5429e-02,  1.1612e-01,\n",
       "          -2.8648e-03,  4.6419e-01, -2.9893e-01,  2.9562e-01, -8.3866e-02,\n",
       "           5.0456e-02, -1.9448e-01, -4.9607e-01,  2.8199e-01, -2.4901e-01,\n",
       "          -1.2040e-01, -2.6371e-01, -2.1398e-01, -3.0484e-01, -2.7289e-01,\n",
       "          -5.7166e-01, -5.3834e-01,  1.4023e-01, -4.4331e-01,  2.5294e-01,\n",
       "          -4.1103e-01,  1.3692e-01,  5.3187e-02,  1.5592e-01,  1.9806e-01,\n",
       "           7.7432e-01, -2.4055e-01,  1.3431e-01, -7.2081e-01, -2.3984e-01,\n",
       "           3.9477e-02, -2.4731e-01, -1.4424e-01,  9.7922e-02, -9.1248e-02,\n",
       "           2.4398e-01, -2.6499e-01, -2.2886e-02, -4.2431e-01,  9.1756e-02,\n",
       "          -4.9843e-01,  3.4078e-01,  1.2757e-01,  4.7824e-03, -2.1576e-01,\n",
       "           2.5370e-01, -1.9613e+00, -2.2385e-01, -5.9731e-02, -1.1305e-02,\n",
       "          -3.5355e-01,  4.3698e-01,  1.7009e-01,  3.5529e-01, -1.3524e-01,\n",
       "          -1.3622e-01, -4.7054e-02,  2.0711e-01,  1.6899e-01,  1.9397e-01,\n",
       "           3.7645e-01,  2.5456e-01,  2.2458e-02,  4.7247e-01,  3.3227e-01,\n",
       "          -6.0882e-01,  2.5836e-01,  3.3588e-02,  4.5596e-02,  2.1447e-01,\n",
       "           7.2657e-02, -3.3287e-01,  1.6170e-01,  2.3826e-03,  2.5118e-01,\n",
       "          -1.7539e-01,  6.0773e-02, -1.8447e-01, -2.3425e-02, -4.2134e-01,\n",
       "           9.8857e-01,  5.9320e-02, -2.9135e-01,  4.9833e-01,  2.6168e-01,\n",
       "          -2.5824e-02,  1.8254e-01,  1.4133e-01, -1.0630e-01, -1.7261e-01,\n",
       "           5.0381e-01, -5.3053e-01, -7.3947e-02, -5.2683e-02, -3.2975e-01,\n",
       "          -4.3404e-02, -6.1902e-01,  5.3025e-01, -1.3200e-01, -5.9127e-02,\n",
       "          -4.3892e-01, -1.9437e-01, -3.5632e-02, -1.8477e-01,  8.7910e-02,\n",
       "          -3.4236e-01, -4.0984e-01, -8.4569e-01,  3.3418e-01,  1.0155e-01,\n",
       "           1.6664e-02, -5.6753e-02,  7.7968e-01,  3.5342e-01,  5.2727e-01,\n",
       "          -9.3960e-02, -1.6759e-01,  3.5326e-01, -1.5666e-01, -1.6530e-01,\n",
       "           1.0741e-01, -8.6389e-02, -3.5363e-01,  2.2799e-01, -1.3908e-01,\n",
       "          -2.2285e-01, -6.1717e-02,  2.2306e-01,  6.5221e-01, -2.2085e-01,\n",
       "          -1.1405e-01,  2.5111e-01, -4.2348e-01, -4.1188e-01,  3.3966e-01,\n",
       "           2.7539e-02,  2.9845e-01,  9.3035e-03, -5.1141e-01, -4.8960e-01,\n",
       "          -8.1643e-01, -7.8155e-02, -5.5546e-01,  3.5066e-01, -7.3430e-01,\n",
       "          -3.9063e-01,  3.7374e-01, -1.0391e-01, -2.3707e-01, -2.5418e-01,\n",
       "           2.7433e-01,  7.9325e-02,  3.4982e-01, -1.1193e-01, -4.6939e-01,\n",
       "          -3.7914e-02,  1.9008e-01, -1.7318e-01, -4.2519e-02,  2.8500e-01,\n",
       "           2.1981e-01,  3.5977e-01, -7.4351e-02,  8.8829e-03, -1.3848e-01,\n",
       "          -2.6508e-01, -3.1089e-01, -2.0899e-01, -2.0691e-01, -1.4235e-02,\n",
       "           2.9113e-01,  8.8310e-02,  1.2159e-01, -8.0157e-01, -6.9137e-01,\n",
       "           1.3713e-01,  5.8320e-01, -1.1509e-01, -2.4777e-02, -2.9416e-01,\n",
       "          -8.3339e-01, -5.3353e-01, -4.2137e-01, -1.1556e-01,  6.8745e-02,\n",
       "           3.0161e-02,  3.0395e-01, -2.1864e-01,  1.3720e-02, -4.6266e-01,\n",
       "           8.7280e-02,  7.2535e-01,  4.3113e-01, -6.1671e-02, -1.6691e-01,\n",
       "          -1.8219e-01, -2.9237e-01, -1.8709e-01, -1.9109e-01, -1.6803e-01,\n",
       "           3.9350e-01,  2.5613e-01, -4.7081e-01, -1.5263e-01, -1.9351e-01,\n",
       "           5.8723e-01, -4.2586e-01, -3.4744e-01,  4.6033e-01, -2.8083e-01,\n",
       "          -4.3844e-01,  1.4553e-01, -7.0824e-01,  1.3150e-01,  2.0868e-01,\n",
       "           1.3445e-01, -5.0152e-01, -6.4479e-02, -2.3183e-01,  6.6183e-01,\n",
       "           6.0709e-01,  4.7509e-01, -5.1437e-01,  4.8166e-01,  1.7959e-01,\n",
       "          -3.7184e-01,  1.6443e-01, -1.3860e-01, -6.0997e-02, -2.7216e-01,\n",
       "           2.1730e-01,  2.7472e-01, -9.3557e-01,  3.2701e-02, -2.1228e-02,\n",
       "           1.0753e-01, -1.1664e-01, -2.4880e-01,  3.8395e-01, -4.6315e-02,\n",
       "          -3.8857e-01, -4.0990e-01,  2.4548e-01, -1.5234e-01, -1.2619e-01,\n",
       "           9.7980e-02,  9.3011e-02, -3.2575e-04, -3.8029e-01, -3.3237e-02,\n",
       "          -2.2592e-01,  1.5538e-01,  5.3955e-01,  2.1249e-01,  9.0557e-02,\n",
       "           8.8931e-02, -1.8152e-01, -1.2858e-01,  8.3907e-01,  2.0467e-01,\n",
       "          -2.2701e-01,  2.2395e-01,  1.5182e-01,  8.6087e-02,  6.2861e-01,\n",
       "          -3.6691e-01, -6.1110e-01, -1.6525e-01, -3.1184e-01, -1.9935e-01,\n",
       "          -2.6486e-01,  1.9804e-01,  4.9872e-01,  1.1543e-01,  2.7607e-01,\n",
       "          -1.0731e-01,  6.9656e-01, -3.7347e-01, -1.7082e-01, -1.9464e-01,\n",
       "           1.9630e-01, -4.6176e-02,  1.8200e-01,  7.9702e-02, -5.0252e-01,\n",
       "           5.4793e-01,  2.9551e-01,  3.4946e-01,  2.2991e-01, -1.1237e+01,\n",
       "           2.6936e-01,  6.8986e-01, -4.1645e-01,  2.1306e-01,  1.3717e-01,\n",
       "           2.6754e-01, -1.5058e-01,  3.9110e-01, -3.6518e-01, -1.9322e-01,\n",
       "          -1.4267e-02, -9.0570e-02,  1.6878e-01, -4.2624e-01,  1.9730e-01,\n",
       "          -8.6384e-02, -8.2251e-01,  3.6977e-01, -5.1687e-01,  4.5512e-01,\n",
       "          -3.1217e-01,  1.1655e-01,  3.4111e-01, -3.0825e-01, -4.7427e-01,\n",
       "          -2.4587e-01, -4.0949e-01, -1.7549e-02,  5.1358e-01, -1.3010e-01,\n",
       "          -1.3692e-01, -2.0493e-01,  2.6636e-01, -7.7549e-02, -5.0297e-01,\n",
       "           1.6286e-01, -3.7741e-01,  1.6891e-01,  2.7314e-01, -2.2475e-01,\n",
       "           2.7264e-01, -1.8618e-01,  6.2691e-01,  3.2158e-01,  6.9787e-01,\n",
       "           1.2940e+00,  6.2874e-01,  7.6760e-02, -7.0461e-01,  1.4454e-01,\n",
       "          -8.8903e-01,  4.4866e-02, -5.8726e-01,  3.0389e-01, -2.6599e-01,\n",
       "           4.7461e-01, -5.2863e-01,  1.8746e-01, -2.2374e-01, -1.5165e-01,\n",
       "           3.8849e-01, -1.1103e-01,  2.3465e-01, -2.8148e-01, -1.9925e-01,\n",
       "           3.4080e-01, -2.1195e-01,  5.8670e-02,  2.1167e-01, -1.9601e-01,\n",
       "          -3.2403e-01,  5.5136e-01,  2.6864e-01, -6.9482e-01,  3.5431e-01,\n",
       "           1.5319e-02, -6.1560e-02,  6.0383e-01, -2.5716e-03,  1.1389e-02,\n",
       "           3.2092e-01,  2.3878e-01, -2.4750e-01, -7.5506e-01, -2.5234e-01,\n",
       "          -1.2507e-01,  3.7695e-01,  2.7633e-01,  3.2790e-01, -1.8578e-01,\n",
       "          -2.8098e-01, -3.2515e-01,  2.8549e-02,  2.2211e-01,  8.3931e-01,\n",
       "          -8.0461e-01,  5.9492e-01,  2.4154e-01, -1.4834e-01, -3.0438e-01,\n",
       "           3.1440e-02, -1.9296e-01, -2.4580e-01, -1.3283e-02, -6.0118e-01,\n",
       "           1.3162e-01,  2.7573e-01,  9.2772e-02,  4.3551e-01, -9.0290e-02,\n",
       "           3.0309e-01,  2.4787e-01,  1.2524e-01, -3.8656e-01,  4.6497e-01,\n",
       "           7.6853e-01,  5.8528e-01, -4.0657e-01, -9.2728e-02, -3.7330e-01,\n",
       "           3.6552e-01,  4.8926e-01, -2.2179e-01, -4.6177e-01,  3.9527e-01,\n",
       "          -7.9518e-01, -1.7349e-01,  7.9719e-02,  2.2911e-01,  3.1825e-01,\n",
       "          -7.5460e-02, -3.2906e-02,  7.3511e-02, -1.6751e-02, -2.9370e-01,\n",
       "          -1.0637e+00, -8.2644e-03,  3.1170e-01,  1.6263e-01,  4.0983e-01,\n",
       "          -1.8292e-01, -1.1000e-01, -9.8246e-02,  7.7828e-01,  7.9833e-02,\n",
       "           7.6720e-01,  1.2426e-01, -1.2785e-01,  2.0530e-01, -1.9425e-01,\n",
       "          -3.3785e-01, -6.3818e-02, -2.2519e-01, -4.2234e-01, -3.2943e-01,\n",
       "          -1.2599e-02, -3.0892e-02, -7.5682e-02, -2.0463e-01,  4.5123e-01,\n",
       "          -3.5485e-01, -2.1077e-01, -2.6175e-01,  2.8029e-01,  2.4365e-01,\n",
       "          -7.3326e-02, -1.3816e-01, -1.7891e-01, -3.4311e-02,  1.2186e-01,\n",
       "           2.5513e-01, -1.2254e-01, -6.5715e-02, -4.4834e-01, -2.8756e-02,\n",
       "           8.2372e-02,  1.2578e+00,  5.7883e-02, -5.4068e-01,  3.6677e-02,\n",
       "           5.6974e-01, -2.1652e-01,  1.9408e-01,  9.3412e-03, -3.8117e-02,\n",
       "          -4.3350e-01,  5.6836e-02,  4.7410e-01,  1.3686e-01,  3.1628e-01,\n",
       "           1.7785e-01,  1.2885e-01, -3.7541e-01,  3.9096e-01,  2.2465e-01,\n",
       "          -5.9301e-01, -8.3481e-02, -5.2051e-01, -1.8008e-01,  8.1286e-01,\n",
       "           1.1081e-01, -2.3766e-01, -2.7706e-01, -1.3523e-01,  1.8892e-03,\n",
       "           4.8490e-02,  1.1167e+00,  2.2097e-02, -9.8702e-02,  5.5235e-01,\n",
       "           1.3700e-02, -2.6737e-01,  2.5178e-01,  1.3226e-01,  1.0149e-01,\n",
       "           3.2855e-01, -1.1443e-01, -3.9333e-01,  3.9083e-01,  3.3593e-01,\n",
       "           4.3467e-01, -9.3114e-01, -6.8096e-01, -1.1458e-01,  4.1579e-01,\n",
       "           4.5144e-01,  4.4243e-01, -6.7248e-02,  3.0855e-01,  9.5992e-01,\n",
       "          -1.1511e+00, -2.3692e-01,  1.4995e-01,  3.1128e-01, -2.2936e-01,\n",
       "          -3.0648e-01, -1.0800e-01, -4.0567e-02, -5.7230e-01, -7.2048e-04,\n",
       "           9.4527e-01, -8.4556e-01,  2.1704e-04,  4.8766e-01, -8.8265e-01,\n",
       "          -2.4326e-02, -2.0923e-01,  9.7365e-02,  3.8820e-01,  4.8486e-02,\n",
       "          -3.5302e-01,  2.0555e-01, -3.7666e-01, -4.9277e-01,  4.7216e-01,\n",
       "           4.9199e-01,  3.2269e-02, -1.1347e-02,  1.6769e-01, -3.4306e-01,\n",
       "           5.0071e-02, -4.0336e-01, -2.2702e-01, -3.3931e-02, -6.2439e-02,\n",
       "           1.1386e-01, -2.8328e-01,  2.5743e-01,  1.7562e-01, -6.2979e-01,\n",
       "           1.5787e-01,  5.2472e-01,  9.3248e-01, -1.2232e-01, -3.2383e-01,\n",
       "           3.4564e-01, -2.2530e-02, -2.0972e-01,  5.1485e-01, -1.2655e-01,\n",
       "          -2.2417e-01,  2.7310e-01,  6.3726e-02, -3.0126e-01, -4.9312e-02,\n",
       "          -2.9878e-01,  1.1766e-01, -2.4869e-01, -5.0406e-01,  1.5218e-01,\n",
       "          -2.6539e-01,  2.1298e-01, -7.7145e-01,  1.8711e-02,  1.5044e-01,\n",
       "           6.2167e-02,  6.0212e-01,  1.5526e-01,  3.6168e-01,  5.2232e-02,\n",
       "           1.6797e-01, -5.1133e-01, -3.1723e-01,  4.8653e-02,  1.9746e-01,\n",
       "          -4.9054e-01,  7.5386e-02, -1.6015e-01,  6.3129e-01,  6.1743e-01,\n",
       "           4.1096e-01,  6.7871e-02, -6.2704e-01, -1.2870e-01, -2.1494e-01,\n",
       "           2.6222e-01,  3.1724e-03,  3.8271e-01, -4.0272e-01,  9.1250e-02,\n",
       "          -5.6564e-01, -2.4590e-01,  8.9155e-02,  2.9186e-01, -2.2387e-01,\n",
       "           5.5824e-01, -9.6183e-02,  3.2262e-01,  4.0279e-01,  1.8794e-01,\n",
       "           1.4928e-01,  8.4672e-01,  2.3639e-01, -4.2857e-01,  9.7638e-02,\n",
       "           6.5397e-01, -2.8493e-01, -4.9998e-02, -7.0296e-01,  1.9278e-01,\n",
       "          -6.7980e-02,  8.3353e-01,  2.1805e-01]], device='mps:0'),\n",
       " tensor([[-2.5187e-01, -4.0054e-01, -5.8007e-01, -8.3055e-01, -1.2868e-01,\n",
       "          -2.8804e-02, -7.1917e-01,  6.6701e-01,  1.2709e-01,  5.0855e-01,\n",
       "           1.2570e-01,  5.1916e-01,  5.1676e-02, -1.1521e-01, -1.6481e-01,\n",
       "          -4.9820e-01,  5.6222e-02,  3.7338e-01,  6.7187e-01, -2.4111e-02,\n",
       "           2.2844e-01,  5.7979e-01, -2.6323e-01, -1.2386e-02, -8.3013e-01,\n",
       "           2.9613e-01, -1.1576e-01,  2.2409e-01,  1.7718e-01, -6.3092e-01,\n",
       "           4.0999e-02,  4.7417e-01,  2.3068e-02,  1.3497e+00, -5.6080e-01,\n",
       "          -1.3230e-01,  8.2213e-02, -3.9453e-01, -2.3010e-01,  3.7283e-02,\n",
       "           6.1164e-01, -4.1315e-01,  1.3635e-01,  4.0254e-01, -6.1085e-02,\n",
       "           8.9076e-02,  1.3899e-01, -3.7744e-01,  2.5892e-01, -8.3961e-01,\n",
       "          -4.8318e-02, -2.0669e-01,  4.9144e-01,  6.2615e-01, -6.3176e-01,\n",
       "          -2.2975e-01,  2.7419e-01, -3.7716e-02, -3.3020e-02,  1.5349e-01,\n",
       "           9.7980e-02,  3.8582e-01, -2.9995e-01,  2.3257e-01,  3.3441e-01,\n",
       "           2.4458e-01, -3.0392e-01, -4.8889e-01, -5.4078e-01, -5.0111e-02,\n",
       "           9.1859e-03,  2.2490e-02,  1.4846e-01, -2.3272e-01,  4.1973e-01,\n",
       "          -7.1214e-01, -6.4354e-02, -8.9125e-02, -2.5973e-01,  2.9246e-01,\n",
       "          -4.5549e-02,  1.9719e-01,  3.1010e-01,  1.0268e-01,  1.2457e-01,\n",
       "           4.3419e-02,  2.7215e-01,  5.2932e-04,  4.7373e-02,  3.7533e-01,\n",
       "          -1.3375e-01, -8.1722e-02, -2.8313e-01, -3.5814e-01,  3.1229e-01,\n",
       "           4.0598e-02,  2.9907e-01, -1.9557e-01,  1.8579e-01, -3.2474e-01,\n",
       "           1.5588e-01,  1.7768e-01, -1.8083e-01,  5.7286e-02, -3.5172e-01,\n",
       "          -4.5475e-01,  2.3589e-02, -2.8569e-01, -4.8285e-01,  2.2697e-01,\n",
       "          -1.4211e-01, -3.0205e-01, -4.0922e-02, -3.8935e-01, -3.1524e-02,\n",
       "          -7.5164e-02,  2.6013e-01, -1.3230e-01, -3.4806e-01, -8.3033e-02,\n",
       "           2.0734e-01, -2.2808e-01,  1.6077e-01,  2.6882e-01,  1.2232e-01,\n",
       "          -9.6705e-02,  6.5251e-01,  1.8950e-02,  3.1674e-01,  2.5763e-01,\n",
       "           1.1699e-01, -6.5768e-02, -1.9745e-01, -2.3154e-01,  7.5988e-02,\n",
       "          -7.6941e-02,  1.0687e-01, -1.8182e-01, -2.2790e-01, -1.8508e-01,\n",
       "           3.1771e-01, -3.3097e-01, -6.0893e-01, -4.9260e-01, -1.6098e-01,\n",
       "          -1.6326e-01, -2.4804e-02, -8.2473e-03, -3.3693e-01, -3.1956e-01,\n",
       "          -6.7443e-01, -3.6717e-01,  8.1041e-01, -1.6831e-02, -3.3356e-01,\n",
       "          -1.6499e-01,  2.3208e-01,  3.1666e-01, -4.5959e-02,  3.7161e-01,\n",
       "           7.7250e-01, -4.7670e-02,  1.6786e-01, -5.1827e-01, -3.8195e-01,\n",
       "           4.8335e-02,  2.6658e-01, -2.7054e-01,  6.1879e-01,  4.8352e-02,\n",
       "           7.5627e-02, -6.6170e-02,  2.7430e-01, -3.2704e-01, -1.2324e-02,\n",
       "          -5.7573e-01,  3.7750e-01,  2.9539e-01, -3.9892e-01, -6.6567e-02,\n",
       "          -2.0834e-02, -1.2060e+00,  9.6964e-02, -8.3040e-01,  3.1236e-01,\n",
       "          -2.3110e-01,  1.7963e-01,  2.3389e-01,  2.9545e-01, -6.5320e-02,\n",
       "           2.5404e-01,  2.8725e-01,  2.5687e-01,  3.3609e-01, -3.4924e-02,\n",
       "           2.8019e-01,  2.3924e-01,  5.0790e-01,  4.1073e-01,  1.4769e-01,\n",
       "          -3.5440e-01,  2.9683e-01,  3.5781e-01,  9.0598e-02,  1.6954e-01,\n",
       "           9.9825e-02, -2.1636e-01,  2.1091e-01,  1.7501e-01, -1.6618e-01,\n",
       "          -2.2406e-01, -3.1100e-02, -4.6818e-01, -5.6655e-02, -4.8467e-01,\n",
       "           3.2557e-01, -5.1983e-01, -1.1912e-01,  3.4831e-01,  2.4340e-01,\n",
       "          -1.4896e-02,  1.8319e-01,  1.7196e-01, -2.4338e-01,  2.1735e-01,\n",
       "           3.7449e-01, -3.4618e-01, -3.9804e-02, -3.2197e-01, -2.9367e-01,\n",
       "           2.4357e-01, -5.6244e-01,  6.6635e-01,  4.0574e-01,  6.0165e-02,\n",
       "           1.5119e-01, -2.7920e-01, -5.1638e-02, -5.7408e-01,  3.5471e-01,\n",
       "          -3.2130e-01, -3.4955e-01, -5.4801e-01,  6.4487e-02,  4.2773e-01,\n",
       "           2.7565e-01, -6.4915e-02,  7.7868e-01,  4.2408e-01,  4.2813e-01,\n",
       "          -4.3003e-01, -1.5896e-02,  4.0377e-01, -4.1405e-02, -4.6059e-01,\n",
       "           6.7116e-02, -1.4962e-01, -5.1643e-01,  2.8000e-01, -2.6598e-01,\n",
       "          -1.2047e-01, -1.4913e-01,  3.3682e-01,  1.5365e-01, -2.1964e-01,\n",
       "          -3.8461e-01,  3.9587e-01, -1.4349e-01, -4.1723e-01,  4.7123e-01,\n",
       "          -3.5025e-01,  1.5209e-01,  1.0694e-01, -3.8441e-01, -6.2072e-01,\n",
       "          -3.2235e-01,  2.3090e-01,  1.5585e-01,  3.4799e-02, -6.6306e-01,\n",
       "          -4.0571e-01,  3.1197e-01,  2.4601e-01,  2.9976e-01,  2.9840e-01,\n",
       "           5.7427e-01, -3.0944e-01, -1.6702e-01,  1.4982e-01, -4.1377e-01,\n",
       "          -1.5704e-01, -3.3228e-02, -1.9995e-01,  2.9569e-01,  2.4768e-01,\n",
       "           1.1600e-01,  2.0367e-01,  4.3482e-01, -2.5272e-01, -3.8673e-01,\n",
       "          -7.1543e-02,  2.0161e-01, -9.3629e-02, -5.8306e-02,  2.3951e-01,\n",
       "          -6.3095e-02,  5.9650e-01,  7.2143e-01,  7.7439e-02, -6.4006e-01,\n",
       "           1.7696e-01,  5.2502e-01, -2.7253e-01, -4.0343e-02,  9.6365e-02,\n",
       "          -1.0344e+00, -2.1331e-01, -1.2883e-01, -2.0181e-01, -6.5352e-01,\n",
       "           8.8115e-02,  3.4863e-01,  4.6833e-03, -1.7864e-01, -3.0947e-01,\n",
       "           1.0448e-01,  4.8724e-02, -4.8980e-02,  4.3534e-01, -2.8901e-01,\n",
       "          -3.4797e-01, -2.3438e-01, -2.1532e-01, -5.4201e-01, -3.9968e-01,\n",
       "           3.5914e-01,  2.8129e-01,  1.1986e-01, -2.6486e-01, -5.1832e-01,\n",
       "           4.5577e-01, -6.8169e-01, -1.4632e-01,  2.4840e-01, -4.7702e-01,\n",
       "           1.4650e-02,  4.8990e-01,  1.4999e-01,  1.6336e-01,  6.5258e-02,\n",
       "           7.8557e-02, -3.3903e-01, -5.0030e-01, -1.5000e-01,  6.2323e-01,\n",
       "           3.2283e-01,  3.3506e-01, -1.7236e-01,  3.9096e-01,  4.5167e-01,\n",
       "          -2.9582e-01,  4.1670e-01,  1.0930e-02,  7.7439e-02, -6.0414e-02,\n",
       "           1.9048e-01,  3.4370e-01, -3.6842e-01,  2.0145e-01,  1.7929e-01,\n",
       "          -3.1634e-01, -3.2149e-01,  1.5691e-02,  2.0642e-01, -1.8580e-01,\n",
       "          -3.0875e-01, -2.0361e-01,  9.9698e-02, -3.4856e-01, -4.3104e-01,\n",
       "           2.9161e-01, -3.6543e-02,  7.0495e-02, -4.0357e-01, -4.3462e-01,\n",
       "          -2.0181e-01, -2.6055e-01,  6.8681e-01,  2.1064e-01, -1.9621e-01,\n",
       "           1.8720e-01, -1.0907e-01, -3.6330e-01,  4.7847e-01,  2.0162e-02,\n",
       "          -4.8754e-01,  4.5482e-01, -3.4822e-01,  1.0627e-01,  5.5756e-01,\n",
       "          -1.6769e-01, -1.0840e-01,  1.1008e-01,  1.0820e-01,  2.9026e-01,\n",
       "          -3.9884e-01,  1.5516e-01, -6.3476e-02, -3.1885e-01,  3.1820e-01,\n",
       "           2.6411e-01,  2.2781e-01, -1.7137e-01, -3.3426e-02,  3.5677e-02,\n",
       "           2.8782e-01, -6.0853e-02, -3.9888e-01, -4.7347e-01, -4.6132e-01,\n",
       "           1.2832e-01,  1.8927e-01,  5.6207e-01,  3.4144e-01, -1.1394e+01,\n",
       "           1.2938e-01,  7.8031e-01, -4.0713e-01, -8.2819e-02, -5.0320e-02,\n",
       "           1.0976e-02, -1.3708e-01,  4.2955e-01, -2.7734e-01, -4.4105e-01,\n",
       "          -3.5727e-01, -1.3308e-01,  1.5454e-01, -3.2189e-01, -2.6945e-01,\n",
       "          -1.4686e-02, -7.0935e-01,  4.0963e-01, -3.0286e-01,  5.0298e-01,\n",
       "          -2.3511e-01, -8.3798e-02,  1.5816e-01, -3.5851e-01, -1.3439e-02,\n",
       "           5.9766e-02, -1.6033e-01,  1.4651e-01,  4.2486e-01, -5.2072e-02,\n",
       "           3.3244e-02, -4.1258e-03,  4.8461e-02, -2.5723e-01, -5.9999e-01,\n",
       "           1.9296e-02, -1.4141e-01,  3.3422e-01, -4.4133e-02, -7.1265e-01,\n",
       "          -2.2533e-01, -2.7418e-01,  3.1407e-01,  2.7021e-01,  9.0598e-01,\n",
       "           2.2955e-01,  1.0686e+00,  1.5022e-01,  1.7748e-01,  2.6230e-02,\n",
       "          -7.8764e-01, -3.3269e-02, -4.1907e-01,  2.3280e-02,  1.0114e-01,\n",
       "           3.6576e-01, -7.9367e-01,  2.5251e-01, -3.5521e-01,  3.6487e-02,\n",
       "           4.9094e-01,  3.1768e-02,  2.7700e-01, -3.2298e-02, -7.8123e-02,\n",
       "           1.2089e-01, -2.8544e-02,  1.1098e-01, -2.2583e-01, -2.4084e-01,\n",
       "          -1.0177e-01,  5.9888e-01,  3.9592e-01, -7.2211e-01,  3.6112e-01,\n",
       "           4.9258e-02, -1.8622e-01,  1.4556e+00, -4.8573e-02,  1.8361e-02,\n",
       "           1.9800e-03,  2.9604e-01, -5.0236e-01, -7.3516e-01,  3.6511e-01,\n",
       "          -1.0211e-01,  5.6427e-02,  1.0127e-01,  4.8148e-01, -9.7481e-02,\n",
       "          -4.7940e-01, -1.8593e-03,  1.0556e-01, -1.3380e-01,  5.4710e-01,\n",
       "          -7.2824e-01, -1.6963e-01, -1.1223e-01,  3.1723e-01, -8.4054e-01,\n",
       "          -6.1126e-02, -7.0210e-01, -3.9328e-01,  5.0586e-01, -2.5896e-01,\n",
       "           1.5852e-01, -2.7443e-02, -4.3439e-02,  5.5574e-01, -2.1489e-01,\n",
       "           7.1555e-02,  2.2813e-01,  8.4995e-02, -1.6733e-01,  3.9349e-02,\n",
       "           4.1630e-01,  4.9571e-01, -4.2740e-01, -6.3911e-01, -7.6901e-03,\n",
       "           1.7180e-01,  1.4933e-01,  3.4019e-01, -5.3463e-01,  3.6431e-01,\n",
       "          -7.9162e-01, -1.8769e-01,  7.2949e-02, -4.3273e-02,  3.6175e-01,\n",
       "           2.3925e-01,  7.2038e-02,  2.2256e-01, -7.7756e-01, -6.4498e-01,\n",
       "          -6.5932e-01, -2.1445e-01,  2.6143e-01, -5.4236e-01, -8.3654e-02,\n",
       "          -1.1033e-01, -4.8837e-03, -5.0292e-02, -6.6102e-02,  1.1939e-02,\n",
       "           1.3632e-01,  2.3587e-01, -1.3023e-01, -1.3178e-01, -9.2464e-02,\n",
       "          -5.5566e-01, -1.1753e-01, -9.9591e-02, -1.2505e-01, -5.6783e-01,\n",
       "          -1.3797e-01,  3.3075e-02, -2.5373e-01, -6.8124e-02,  3.5608e-01,\n",
       "          -2.9832e-02, -2.7856e-02, -4.8850e-01,  7.2846e-01,  4.0279e-01,\n",
       "           5.2088e-02, -2.8298e-01, -5.5189e-01, -1.0567e-01,  2.9592e-01,\n",
       "           1.9555e-01, -6.7007e-02, -1.7709e-01, -2.4145e-01, -4.2522e-01,\n",
       "           1.4302e-01,  7.1147e-01, -6.6296e-02, -6.8115e-01,  2.6621e-01,\n",
       "           5.8223e-01, -1.2882e-01, -2.4432e-01,  2.0983e-02, -9.3284e-02,\n",
       "          -3.1851e-01, -6.0317e-01,  7.8821e-01, -1.1246e-01, -5.8638e-02,\n",
       "           2.2427e-01,  2.0359e-01, -9.2588e-03,  4.0209e-01, -1.5770e-01,\n",
       "          -3.5434e-01, -1.6477e-03, -7.3600e-01,  4.3257e-01,  6.1666e-02,\n",
       "          -3.1489e-01,  7.6082e-01,  2.7260e-01, -6.0593e-02, -4.5779e-02,\n",
       "           2.2863e-01,  7.9941e-01, -2.4636e-01,  6.4653e-02,  5.3779e-01,\n",
       "           8.8110e-02, -1.1229e-01,  3.9580e-02,  4.1665e-02,  4.5410e-02,\n",
       "           3.7012e-01, -3.2145e-02, -4.1564e-01, -6.1194e-02,  3.8924e-01,\n",
       "           3.9295e-01, -5.4610e-01, -7.3487e-01,  2.2333e-02,  6.8551e-01,\n",
       "          -2.9054e-01,  2.3654e-01, -3.2708e-02,  3.0880e-01,  1.2001e-01,\n",
       "          -9.2160e-01, -1.5016e-01, -1.0056e-01,  2.8229e-01, -2.5482e-02,\n",
       "          -1.6139e-01, -1.4727e-01, -2.3686e-01, -1.6208e-01,  6.8308e-01,\n",
       "           2.1991e-01, -6.9160e-01,  3.4058e-02,  5.7281e-01, -6.0551e-01,\n",
       "           4.6052e-01, -2.4577e-01, -3.4302e-02,  3.6435e-01,  1.3117e-01,\n",
       "           4.9067e-02,  1.7856e-01, -4.2138e-01, -4.0598e-01,  1.7824e-01,\n",
       "           9.1672e-01, -7.6613e-02,  1.1166e-01, -3.9384e-01, -1.7182e-01,\n",
       "          -7.1855e-02, -3.1913e-01, -3.1286e-01,  3.7185e-01,  2.0529e-01,\n",
       "          -3.2552e-02, -1.3271e-01, -2.2992e-01,  3.0816e-01,  1.1821e-01,\n",
       "           3.2536e-01,  5.8187e-01,  6.4368e-01, -1.0077e-02, -2.4936e-01,\n",
       "          -6.1235e-02,  1.4568e-01, -1.9524e-01,  5.2704e-01, -2.0627e-01,\n",
       "          -3.3453e-01, -3.8069e-03,  3.0134e-01, -6.1681e-01, -5.8725e-02,\n",
       "          -4.5426e-01,  4.2361e-01, -1.6955e-01,  3.7868e-02,  2.8833e-01,\n",
       "           4.0522e-02,  5.7925e-02, -2.2651e-01,  5.5807e-01,  4.4511e-01,\n",
       "           1.6609e-01,  4.5662e-01,  1.6104e-01,  4.5272e-02, -5.1597e-01,\n",
       "           2.1978e-01, -1.0213e+00, -9.4260e-02, -4.6640e-02, -1.5401e-01,\n",
       "          -4.4224e-01, -2.8126e-02,  1.6814e-01,  1.8538e-01,  6.9597e-01,\n",
       "           7.2967e-01, -3.0871e-01, -4.5290e-01, -1.8510e-01,  1.5304e-01,\n",
       "           2.7705e-01, -1.9517e-01,  3.5971e-01,  9.6103e-02,  2.4432e-01,\n",
       "          -5.0967e-01, -4.3165e-01, -4.2252e-02,  3.1411e-01, -2.0171e-01,\n",
       "           5.7118e-01, -2.0164e-01, -1.3863e-01,  9.0832e-02,  1.3770e-01,\n",
       "           1.6639e-01,  7.9664e-01,  4.3416e-01, -6.3425e-01,  8.7469e-02,\n",
       "          -1.3980e-02, -2.9988e-01,  2.5983e-02, -4.3577e-01,  2.8858e-01,\n",
       "           1.2984e-01,  4.9280e-01,  6.7504e-01]], device='mps:0'),\n",
       " tensor([[-1.8397e-02, -5.5900e-01, -5.9151e-01, -7.0983e-01, -3.9801e-02,\n",
       "           5.0283e-02, -8.5554e-01,  8.1456e-01,  1.8582e-01,  4.7998e-01,\n",
       "           3.0498e-01,  4.7630e-01, -1.7639e-01, -2.4322e-01, -1.9694e-01,\n",
       "          -4.3248e-01, -1.3683e-01,  3.0316e-01,  6.4005e-01,  1.6635e-01,\n",
       "           4.1864e-01,  7.4014e-01, -2.5322e-01, -7.4617e-02, -9.3935e-01,\n",
       "           5.0430e-01, -1.0427e-01,  3.4593e-01,  3.7857e-01, -7.7233e-01,\n",
       "          -3.7240e-02,  5.0138e-01,  8.8280e-02,  1.2000e+00, -4.4433e-01,\n",
       "          -2.1968e-01,  1.6880e-01, -3.9279e-01, -1.2924e-01, -4.6366e-02,\n",
       "           4.0677e-01, -4.2265e-02,  1.5338e-01,  3.2903e-01,  5.5836e-02,\n",
       "           1.0690e-01,  7.8375e-02, -2.6304e-01,  4.6112e-01, -5.8695e-01,\n",
       "          -9.1804e-02, -1.8489e-01,  4.0597e-01,  5.9383e-01, -5.3826e-01,\n",
       "          -1.3850e-01,  3.6875e-01, -1.2025e-01,  6.2779e-02,  3.3101e-01,\n",
       "           1.2588e-01,  4.0455e-01, -4.0883e-01,  3.1469e-01,  1.5752e-01,\n",
       "           1.4436e-01, -4.7305e-01, -2.9332e-01, -5.0567e-01,  1.7753e-01,\n",
       "          -1.1451e-01,  1.3657e-01,  2.1789e-01, -1.5231e-01,  3.0572e-01,\n",
       "          -7.9337e-01,  1.6839e-02, -2.4873e-01, -3.9202e-01,  1.6154e-01,\n",
       "           5.0833e-02,  2.8643e-01,  3.7861e-01,  9.9442e-02, -2.5894e-02,\n",
       "          -7.5196e-02,  1.5190e-01, -3.4771e-02,  1.2341e-01,  3.5029e-01,\n",
       "          -2.4287e-01,  9.4031e-02, -2.5132e-01, -1.2549e-01,  2.3921e-01,\n",
       "          -3.0685e-01,  3.5697e-01, -6.2040e-03,  1.0202e-01, -4.1880e-01,\n",
       "           1.0358e-01,  2.2140e-01, -1.5578e-01,  2.6553e-01, -4.4714e-01,\n",
       "          -5.0547e-01,  6.7794e-02, -5.0917e-01, -4.8789e-01,  8.9702e-02,\n",
       "          -5.4593e-02, -4.2078e-01,  1.2731e-02, -5.0266e-01,  1.2467e-01,\n",
       "           1.8527e-01,  2.2345e-01, -1.6992e-01, -6.2753e-02, -1.6622e-01,\n",
       "           2.2589e-01, -3.9778e-01, -2.2770e-02,  4.0188e-01,  1.8489e-01,\n",
       "          -8.7239e-02,  7.9938e-01,  1.0494e-01,  6.8345e-01,  4.2101e-01,\n",
       "           2.0098e-01, -3.0215e-01, -1.4711e-01, -3.3265e-01,  1.4523e-01,\n",
       "          -1.5426e-01, -4.8677e-02, -4.2050e-01, -7.3853e-02, -1.5696e-01,\n",
       "           2.4718e-01, -2.6320e-01, -6.1759e-01, -1.6365e-01, -3.9926e-02,\n",
       "          -1.8310e-01, -7.8133e-02, -1.0172e-02, -2.8698e-01, -3.1247e-01,\n",
       "          -6.6499e-01, -5.0803e-01,  6.2812e-01,  5.1929e-02, -3.2029e-01,\n",
       "          -1.5759e-01,  3.5836e-01,  3.0616e-01, -1.5648e-01,  1.8902e-01,\n",
       "           9.5187e-01, -6.3152e-02,  2.4292e-01, -3.5443e-01, -3.1514e-01,\n",
       "           1.6567e-01,  3.7383e-01, -1.0765e-01,  7.5076e-01,  1.0396e-02,\n",
       "           7.9330e-02, -2.3338e-01,  4.9007e-01, -8.9086e-03,  1.4820e-01,\n",
       "          -5.3516e-01,  3.4590e-01,  3.6768e-01, -4.1754e-01, -6.5055e-02,\n",
       "          -1.0726e-01, -1.3689e+00, -5.5657e-02, -6.0142e-01,  3.5763e-01,\n",
       "          -1.5198e-01,  3.5871e-01,  3.6502e-02,  2.6092e-01, -1.2097e-01,\n",
       "           2.2338e-01,  6.9889e-02,  1.0755e-01,  4.3265e-01, -2.3260e-01,\n",
       "           4.7668e-01,  1.7535e-01,  5.5829e-01,  4.3960e-01,  4.5589e-02,\n",
       "          -2.4661e-01,  2.8007e-01,  4.6320e-01,  3.6792e-02,  2.7090e-02,\n",
       "           6.6532e-03, -3.5439e-01,  6.5522e-02,  1.9667e-01, -1.0572e-01,\n",
       "          -3.1452e-01,  2.5437e-02, -4.6720e-01, -7.5342e-02, -8.0284e-01,\n",
       "           4.5054e-01, -5.1920e-01, -1.4066e-01,  3.0640e-01,  2.7043e-01,\n",
       "          -2.6553e-01, -2.1551e-03,  8.5449e-02, -2.3071e-01,  2.5589e-02,\n",
       "           4.6658e-01, -2.5938e-01, -1.4069e-01, -3.5176e-01, -5.0289e-01,\n",
       "           2.6279e-01, -5.4909e-01,  5.9870e-01,  3.0598e-01,  9.0110e-02,\n",
       "           1.4201e-01, -2.7538e-01, -2.7767e-01, -6.8502e-01,  5.1046e-01,\n",
       "          -3.2019e-01, -2.7014e-01, -8.6864e-01,  2.0478e-01,  2.7148e-01,\n",
       "           1.4012e-01, -1.0396e-01,  6.9525e-01,  5.7645e-01,  4.9793e-01,\n",
       "          -3.6396e-01,  6.0697e-02,  5.2583e-01,  1.6622e-02, -3.6447e-01,\n",
       "           2.8916e-02, -1.7388e-01, -3.7601e-01,  1.8869e-01, -2.6812e-01,\n",
       "          -1.2790e-01, -1.0321e-01,  4.2238e-01,  1.9667e-01, -4.8508e-01,\n",
       "          -3.5373e-01,  3.8730e-01, -7.3424e-02, -4.5641e-01,  5.4340e-01,\n",
       "          -4.4982e-01,  2.5810e-01,  2.0317e-02, -2.8028e-01, -5.5779e-01,\n",
       "          -3.4907e-01,  3.7665e-01, -6.2679e-02, -1.9370e-02, -9.5313e-01,\n",
       "          -4.1100e-01,  4.7459e-01,  5.2493e-01,  4.2481e-01,  1.7428e-01,\n",
       "           6.5610e-01, -1.6712e-01,  1.2203e-02,  2.0552e-01, -2.7874e-01,\n",
       "           6.4356e-02, -5.4126e-02, -3.6535e-01,  3.7684e-01,  2.1599e-01,\n",
       "           2.1935e-01,  1.3219e-01,  3.5756e-01, -2.6433e-01, -2.1475e-01,\n",
       "          -1.9854e-01,  3.0282e-01, -4.1554e-01, -2.0591e-01,  2.6175e-01,\n",
       "          -4.1004e-02,  6.3973e-01,  6.9361e-01,  1.9275e-01, -5.1530e-01,\n",
       "           9.9732e-02,  6.2349e-01, -1.2927e-01, -1.8212e-01,  1.9792e-01,\n",
       "          -1.1896e+00, -5.2166e-02,  5.7590e-02, -1.1912e-01, -4.6221e-01,\n",
       "           1.3118e-01,  3.8028e-01,  6.9439e-03, -3.5827e-01, -1.9357e-01,\n",
       "           3.9114e-02, -8.5654e-02,  2.2130e-01,  3.1362e-01, -3.1410e-01,\n",
       "          -2.1214e-01,  6.7474e-02, -2.4429e-01, -4.2376e-01, -1.6034e-01,\n",
       "           3.0053e-01,  8.9154e-02, -1.3900e-01, -2.3191e-01, -4.7268e-01,\n",
       "           4.9668e-01, -7.4048e-01, -4.6899e-01,  1.0082e-01, -2.7626e-01,\n",
       "          -1.2000e-01,  3.9909e-01,  2.8280e-01,  4.5709e-02,  1.0417e-01,\n",
       "          -3.9023e-02, -2.4469e-01, -2.6800e-01, -2.0944e-01,  6.7439e-01,\n",
       "           6.8014e-02,  2.0854e-01, -2.8372e-01,  6.6688e-01,  4.9043e-01,\n",
       "          -3.3248e-01,  2.9732e-01, -5.7393e-02, -5.1787e-03, -1.6043e-01,\n",
       "           2.2288e-01,  3.9082e-01, -4.4713e-01,  2.6695e-01, -4.1292e-02,\n",
       "          -1.7052e-01, -4.0118e-01, -2.6754e-01,  1.8445e-01, -2.1661e-01,\n",
       "          -2.9513e-01, -2.9925e-01,  5.0105e-02, -4.4345e-01, -5.9217e-01,\n",
       "           1.8615e-01, -9.1439e-03, -1.9278e-03, -3.2853e-01, -2.4541e-01,\n",
       "          -1.7885e-01, -2.0987e-01,  5.3866e-01,  1.7160e-01, -1.6553e-01,\n",
       "           2.0951e-02, -1.9867e-01, -3.7789e-01,  4.3467e-01, -1.7638e-01,\n",
       "          -3.5996e-01,  5.2837e-01, -2.3028e-01,  7.2141e-02,  4.4889e-01,\n",
       "          -1.9039e-01, -1.0885e-01, -9.8003e-02,  2.7088e-01,  3.3127e-01,\n",
       "          -5.6967e-01,  2.1359e-01,  2.1919e-01, -3.1523e-01,  1.7867e-01,\n",
       "           6.1767e-02,  4.6907e-01, -3.2588e-01, -5.3380e-02,  1.0278e-01,\n",
       "           2.7067e-01,  2.4165e-03, -3.8506e-01, -4.9246e-01, -3.0467e-01,\n",
       "           1.7111e-01,  3.0296e-01,  4.4574e-01,  2.5823e-01, -1.1255e+01,\n",
       "           2.4336e-01,  6.9271e-01, -4.9053e-01, -1.2989e-01, -3.3043e-02,\n",
       "          -1.0521e-01, -6.9750e-02,  5.0818e-01, -5.2611e-01, -1.4942e-01,\n",
       "          -2.8021e-01,  2.5571e-02,  1.1122e-01, -3.9366e-01, -2.5079e-01,\n",
       "          -4.4592e-02, -7.6365e-01,  5.5994e-01, -2.6226e-01,  6.0572e-01,\n",
       "          -2.8297e-01,  4.7066e-02,  1.2289e-01, -2.0602e-01,  1.2742e-01,\n",
       "           2.2007e-03, -1.8192e-01,  5.4452e-02,  4.4671e-01, -9.3956e-02,\n",
       "           1.8081e-02,  1.2952e-02,  9.5677e-02, -2.5469e-01, -5.0936e-01,\n",
       "           1.5826e-01, -2.0568e-01,  1.8578e-01, -6.0542e-02, -8.4718e-01,\n",
       "          -1.5640e-01, -3.0531e-01,  3.0107e-01,  1.9453e-01,  8.0267e-01,\n",
       "           3.8682e-01,  1.0124e+00,  3.0388e-02,  2.2759e-01,  9.8215e-02,\n",
       "          -9.0434e-01,  1.1933e-01, -4.4343e-01,  3.4923e-02, -1.2300e-01,\n",
       "           4.2095e-01, -6.6707e-01,  1.9347e-01, -3.0029e-01, -3.3081e-02,\n",
       "           3.0682e-01,  1.2690e-01,  2.5428e-01, -1.9726e-01, -8.4250e-02,\n",
       "           3.2927e-02, -1.4302e-01,  6.4765e-02, -2.7036e-01, -1.9057e-01,\n",
       "          -1.4872e-01,  6.9440e-01,  3.8364e-01, -5.6122e-01,  1.2327e-01,\n",
       "           4.8368e-02, -8.3600e-02,  1.2667e+00,  2.0243e-01, -3.9277e-02,\n",
       "          -6.7848e-02,  2.0670e-01, -2.6686e-01, -5.2173e-01,  2.6031e-01,\n",
       "          -3.4210e-02,  4.3204e-01, -2.0327e-01,  4.8577e-01, -9.7261e-02,\n",
       "          -4.7156e-01, -1.4949e-01,  4.2674e-02, -2.6000e-01,  6.3245e-01,\n",
       "          -8.5472e-01, -3.8644e-01, -1.6367e-01,  2.6593e-01, -3.8661e-01,\n",
       "          -1.4511e-01, -5.7614e-01, -4.7919e-01,  2.8725e-01, -2.8430e-01,\n",
       "           1.3855e-01,  9.7249e-02, -9.8408e-03,  3.7915e-01, -2.3129e-01,\n",
       "           3.7073e-02,  3.8618e-01,  9.0936e-02, -2.3310e-01,  1.4091e-01,\n",
       "           3.2870e-01,  3.8495e-01, -3.5475e-01, -6.8410e-01, -1.5838e-02,\n",
       "           1.1015e-01,  2.3482e-01,  2.4509e-01, -5.9923e-01,  1.5746e-01,\n",
       "          -7.9947e-01, -4.8978e-02,  2.9114e-02,  5.4950e-02,  6.3014e-02,\n",
       "           2.1978e-01,  1.4066e-01,  4.3519e-01, -7.2049e-01, -7.1625e-01,\n",
       "          -5.3655e-01, -9.3943e-02,  3.8849e-01, -5.2684e-01, -1.4554e-02,\n",
       "          -2.4298e-01, -2.0394e-02, -8.4217e-02, -1.4763e-01, -3.9119e-02,\n",
       "           3.9112e-01,  2.5864e-01, -2.8849e-01, -2.0472e-01, -1.2929e-01,\n",
       "          -7.5019e-01, -2.7171e-01, -2.1829e-01, -2.1919e-01, -4.5287e-01,\n",
       "          -2.0229e-02,  1.0121e-01, -2.6627e-01, -1.6134e-01,  2.3627e-01,\n",
       "          -2.3806e-02, -1.6541e-01, -4.9895e-01,  8.0662e-01,  3.8342e-01,\n",
       "           1.1708e-01, -4.1172e-01, -7.8428e-01, -2.0402e-03,  2.4013e-01,\n",
       "           1.3697e-01, -2.4204e-01, -1.1694e-01, -3.6040e-01, -3.7498e-01,\n",
       "           1.3777e-01,  8.8516e-01, -9.3729e-02, -6.0930e-01,  2.1534e-01,\n",
       "           5.4458e-01, -1.4209e-01, -3.3991e-01, -8.5746e-02, -6.2793e-02,\n",
       "          -3.8532e-01, -4.4344e-01,  7.1320e-01, -2.4854e-02,  8.8018e-02,\n",
       "          -9.7726e-03,  1.3246e-01, -2.9818e-01,  3.8730e-01, -1.7579e-02,\n",
       "          -2.4867e-01,  2.0602e-02, -5.5142e-01,  3.5220e-01,  1.2030e-01,\n",
       "          -2.6099e-01,  6.8594e-01,  1.2404e-01, -1.6017e-01,  9.1465e-03,\n",
       "           1.1258e-01,  1.0505e+00, -2.0379e-01, -7.1049e-02,  5.4910e-01,\n",
       "           1.5276e-02, -3.0699e-01,  1.7677e-01, -1.9922e-01,  1.7149e-02,\n",
       "           2.9131e-01,  1.7992e-01, -5.0304e-01, -2.9537e-02,  2.4166e-01,\n",
       "           3.7296e-01, -5.9616e-01, -8.5231e-01, -2.3356e-03,  9.1691e-01,\n",
       "           3.9524e-02,  3.7298e-01, -8.9075e-02,  3.2319e-01, -7.3738e-02,\n",
       "          -1.0420e+00, -5.8071e-02, -3.1155e-01,  6.2655e-01,  7.4657e-02,\n",
       "          -2.6495e-01, -1.1524e-01, -3.7052e-02, -3.1653e-01,  6.9879e-01,\n",
       "           2.7761e-01, -6.6215e-01,  1.9533e-01,  2.5184e-01, -5.4590e-01,\n",
       "           4.2987e-01, -1.7895e-01,  5.3415e-02,  3.0307e-01,  2.4678e-01,\n",
       "           7.8113e-02,  2.3742e-01, -5.0592e-01, -3.4696e-01,  2.6606e-01,\n",
       "           7.5168e-01,  2.1540e-02,  2.1279e-01, -2.1443e-01, -2.2686e-01,\n",
       "          -8.6139e-02, -3.2840e-01, -3.8100e-01,  3.4170e-01,  1.9817e-01,\n",
       "          -1.7627e-02, -3.3496e-01, -1.2618e-01,  4.7393e-01,  2.6262e-02,\n",
       "           2.2961e-01,  6.6964e-01,  5.2598e-01, -9.7399e-02, -1.3545e-01,\n",
       "          -2.2904e-01,  2.2499e-01, -1.1806e-01,  4.2967e-01, -2.5691e-01,\n",
       "          -7.2061e-02,  1.1091e-01,  3.6410e-01, -6.5146e-01,  1.2030e-01,\n",
       "          -5.8035e-01,  4.9541e-01, -4.6906e-02, -5.6658e-02,  3.2847e-01,\n",
       "           1.8553e-01,  1.8786e-01, -1.4526e-02,  4.1536e-01,  2.1257e-01,\n",
       "           1.2718e-01,  7.7636e-01,  1.6321e-01,  1.0842e-01, -5.5073e-01,\n",
       "           1.4515e-01, -8.8269e-01, -5.8233e-02,  2.6008e-02,  8.4348e-02,\n",
       "          -4.6196e-01, -7.7963e-02,  1.1676e-01,  3.9761e-02,  6.8945e-01,\n",
       "           7.2201e-01, -3.6659e-01, -5.2974e-01, -2.9030e-01,  8.8388e-02,\n",
       "           1.9298e-01, -6.1853e-02,  3.5846e-01,  4.1948e-01,  3.6854e-01,\n",
       "          -5.3499e-01, -3.2735e-01, -1.7090e-01,  2.9738e-01, -3.3778e-01,\n",
       "           4.4912e-01, -1.1625e-01, -1.2445e-01, -3.6748e-02,  1.8229e-01,\n",
       "           1.3532e-01,  5.6113e-01,  4.9286e-01, -6.2605e-01,  1.3557e-01,\n",
       "          -9.3864e-02, -3.1531e-01, -9.0355e-02, -2.9414e-01,  3.1074e-01,\n",
       "           2.1612e-01,  3.4336e-01,  9.3422e-01]], device='mps:0'),\n",
       " tensor([[-0.0278, -0.7575, -0.5307,  ...,  0.0801,  0.9419,  0.7043],\n",
       "         [-0.0111, -0.5057, -0.4670,  ...,  0.2365, -0.0712,  0.4388],\n",
       "         [-0.3292, -0.2590, -0.5016,  ..., -0.0148,  0.3509,  0.1607]],\n",
       "        device='mps:0'),\n",
       " tensor([[-0.1932, -0.4948, -0.6750,  ...,  0.2888,  0.2386,  0.7024],\n",
       "         [-0.0583, -0.6757, -0.8037,  ...,  0.1694,  0.7852,  0.5016]],\n",
       "        device='mps:0'),\n",
       " tensor([[-2.1557e-01, -3.4302e-01, -5.3454e-01, -8.8363e-01, -6.0228e-02,\n",
       "          -3.9555e-01, -8.3520e-01,  4.2932e-01,  4.9146e-02,  3.3053e-01,\n",
       "           1.1544e-01,  4.9200e-01,  1.7665e-01,  4.1852e-02, -3.5439e-01,\n",
       "          -3.9834e-02,  2.2093e-01,  3.5368e-01,  3.0812e-01,  2.4063e-01,\n",
       "           4.6833e-01,  4.8268e-01, -1.1595e-01,  1.7440e-02, -8.2687e-01,\n",
       "           1.7049e-01,  9.8705e-03,  3.3937e-01,  2.8929e-01, -3.9326e-01,\n",
       "          -1.2633e-01,  2.4473e-01, -1.4527e-01,  1.5918e+00, -9.4493e-02,\n",
       "          -7.4941e-02,  1.3339e-01, -5.9292e-01,  1.0284e-01, -9.0184e-03,\n",
       "           5.2707e-01, -2.3729e-01,  1.2671e-01,  3.9361e-01,  2.2218e-01,\n",
       "           1.2874e-01,  1.3737e-01, -2.3741e-01,  2.3237e-01, -6.2971e-01,\n",
       "          -1.1962e-01, -2.5275e-01,  5.5227e-01,  5.1938e-01, -1.1180e+00,\n",
       "          -5.9156e-01,  3.6817e-01,  2.0554e-01, -3.6088e-01,  2.2772e-02,\n",
       "          -5.7591e-02,  2.7615e-01,  2.1165e-02,  3.2266e-01,  1.8788e-02,\n",
       "           3.7221e-01, -2.1358e-01, -6.3098e-01, -5.3598e-01, -4.7268e-01,\n",
       "           1.4808e-02,  1.0864e-01,  1.4762e-01, -2.1165e-01,  3.3026e-01,\n",
       "          -3.7047e-01,  8.7954e-02, -1.7560e-01, -1.6127e-01,  5.7955e-03,\n",
       "          -7.2312e-02,  2.2787e-01,  1.2575e-01, -1.1595e-01,  6.9053e-02,\n",
       "           1.8631e-01,  1.5945e-01,  1.3073e-01,  1.1838e-01, -3.0293e-02,\n",
       "          -3.7750e-02,  2.0769e-03, -3.5665e-01, -2.6140e-01,  3.0009e-01,\n",
       "          -8.8599e-02,  2.5662e-01, -3.8743e-01,  7.2947e-02, -6.4576e-02,\n",
       "           4.3362e-02,  5.5173e-02, -8.0903e-02,  2.4139e-01, -2.8904e-01,\n",
       "          -6.1432e-01,  1.7252e-01, -1.0068e-01, -8.1571e-01,  1.5752e-01,\n",
       "          -1.2342e-01, -4.1377e-01,  2.0792e-02, -3.1927e-01, -2.3805e-01,\n",
       "          -8.8598e-02, -1.9253e-01,  1.1267e-01, -2.4556e-01,  3.1407e-01,\n",
       "           3.2215e-01, -2.2466e-01,  2.5970e-01, -9.4584e-02,  1.8047e-01,\n",
       "          -2.7668e-01,  2.8358e-01, -1.3223e-02,  2.9831e-01,  2.9925e-01,\n",
       "           1.0013e-01,  8.0034e-02, -2.3931e-01, -3.9883e-02,  1.3174e-01,\n",
       "           2.0585e-01,  2.1442e-01, -2.3368e-01,  1.7978e-02,  6.2208e-02,\n",
       "           2.0043e-01, -1.3184e-02, -2.2168e-01, -1.6869e-01, -2.2059e-01,\n",
       "          -1.7857e-01, -1.4671e-01,  5.8244e-02, -3.5367e-01, -4.0686e-01,\n",
       "          -6.0569e-01, -5.0516e-01,  6.7528e-01, -7.9958e-02, -1.1554e-01,\n",
       "          -1.6982e-01,  4.6091e-01, -4.3617e-02,  3.4217e-03,  1.7970e-01,\n",
       "           7.1405e-01,  2.7194e-02,  1.6640e-01, -4.7625e-01, -2.0394e-01,\n",
       "           2.7045e-01, -3.0122e-01, -2.0669e-01, -5.5958e-02,  1.4700e-01,\n",
       "           1.8914e-01, -1.0384e-01,  2.8586e-01,  1.8930e-02,  2.3565e-01,\n",
       "          -7.3573e-01,  5.1941e-01,  2.9407e-01, -2.3449e-01,  6.8919e-02,\n",
       "          -1.5530e-01, -1.2998e+00,  2.1974e-01, -7.6685e-01,  1.8615e-01,\n",
       "          -1.3486e-01,  1.8778e-01,  3.4968e-01,  3.4987e-01, -2.3343e-02,\n",
       "           1.8538e-01,  1.8583e-01,  2.9852e-01,  3.8594e-01, -1.0652e-01,\n",
       "           3.3285e-01,  3.9289e-01,  2.6937e-01,  2.5171e-01,  3.0164e-01,\n",
       "          -6.5569e-01,  3.8671e-01,  3.1694e-01,  4.6357e-01,  1.0670e-01,\n",
       "           1.4295e-01,  1.5333e-01,  1.2618e-01,  2.3161e-01,  1.3248e-02,\n",
       "          -3.2979e-01, -1.5194e-01, -5.2744e-01, -1.0320e-02, -1.3782e-01,\n",
       "           3.7245e-01, -1.9296e-01,  1.4482e-01,  7.2240e-01,  3.0177e-01,\n",
       "          -5.9742e-02,  3.0168e-01,  1.2129e-02, -2.5151e-01,  7.9077e-02,\n",
       "           4.2383e-01, -1.5031e-01,  1.5279e-01, -4.6546e-01, -1.8859e-01,\n",
       "           3.7013e-01, -3.5821e-01,  6.3186e-01,  4.6392e-01,  2.1484e-02,\n",
       "           1.9729e-01, -3.8772e-01, -2.4794e-02, -1.3292e-01,  4.2606e-01,\n",
       "          -3.2235e-01, -4.4201e-01, -4.6025e-01,  1.2488e-01,  5.1963e-01,\n",
       "           2.7507e-01,  1.8342e-02,  4.0816e-01,  5.8068e-01,  4.3071e-01,\n",
       "          -2.1999e-01, -1.9365e-01,  3.5483e-01,  9.1989e-02, -4.9810e-01,\n",
       "           3.0726e-01, -1.4577e-02, -5.0317e-01,  3.5902e-01, -5.3025e-01,\n",
       "          -5.5532e-02, -5.9393e-02,  4.5818e-01,  4.0688e-01, -5.1826e-01,\n",
       "          -3.4261e-02,  4.0496e-01, -1.3117e-01, -3.0446e-01,  3.2324e-01,\n",
       "          -3.4511e-01,  3.1175e-01,  1.6773e-01, -5.5057e-01, -5.7871e-01,\n",
       "          -4.2803e-01, -2.4415e-01, -8.8823e-02,  1.4967e-02, -7.8610e-01,\n",
       "          -2.8046e-01,  2.0665e-01, -5.1283e-03, -8.6086e-02,  5.7148e-02,\n",
       "           2.8681e-01, -2.0106e-01, -9.2440e-02, -1.5680e-01, -6.0368e-01,\n",
       "          -1.5160e-01, -1.0235e-01, -3.0615e-01,  3.1186e-01,  8.6952e-02,\n",
       "           7.5664e-02,  1.9435e-01,  2.7866e-01, -6.4456e-02, -4.0782e-01,\n",
       "          -1.2747e-01, -2.3504e-02, -3.1469e-01, -2.1963e-01,  1.4944e-01,\n",
       "          -3.3305e-02,  2.7266e-01,  8.1616e-01, -3.8768e-01, -5.5249e-01,\n",
       "           1.4606e-01,  8.3344e-01, -2.9283e-01, -9.5223e-02, -1.1903e-01,\n",
       "          -9.2530e-01, -3.7072e-01, -1.9451e-01, -2.9672e-01, -4.8813e-01,\n",
       "           1.0980e-01, -1.7958e-01, -1.1784e-01, -1.8766e-01, -3.3024e-01,\n",
       "           3.9487e-01,  6.7966e-01, -1.1788e-01,  3.2904e-01, -3.4598e-01,\n",
       "          -3.1677e-01, -9.7884e-02, -4.3956e-01, -5.7858e-01, -2.7805e-01,\n",
       "           2.5600e-01,  3.4740e-01,  2.1232e-02, -6.8528e-02, -3.8176e-01,\n",
       "           5.3752e-01, -4.1048e-01,  3.0514e-02,  3.4183e-01, -4.6879e-01,\n",
       "           1.0197e-01,  4.2102e-01, -1.3648e-01,  1.4493e-01, -5.2486e-02,\n",
       "           1.1283e-01, -3.4785e-01, -4.6263e-01,  1.3005e-02,  6.9578e-01,\n",
       "           3.8538e-01,  1.9462e-01, -2.2188e-01,  4.6105e-01,  4.0659e-01,\n",
       "          -5.2163e-01,  5.0398e-01,  1.0944e-01, -2.3511e-01, -4.7327e-01,\n",
       "           2.1476e-01,  3.1016e-01, -3.7805e-01,  1.6156e-01,  1.3281e-01,\n",
       "          -1.4074e-01, -2.5158e-01, -1.6374e-01,  4.7408e-01,  4.5011e-01,\n",
       "          -1.2427e-01, -4.3088e-01,  1.1413e-01, -5.0812e-01, -2.0407e-01,\n",
       "           2.0856e-01, -1.2761e-01,  2.1117e-01, -4.1330e-01, -3.9195e-01,\n",
       "          -3.3138e-01, -1.6117e-01,  7.0991e-01,  3.2262e-01, -3.7908e-01,\n",
       "           1.0666e-01, -2.6697e-01, -3.6592e-01,  4.0021e-01,  9.3521e-02,\n",
       "          -2.9539e-01,  3.4456e-01, -4.6624e-01, -3.8041e-02, -1.1845e-03,\n",
       "          -1.4180e-01, -5.5793e-02, -5.7221e-02, -5.7840e-02,  3.9790e-01,\n",
       "          -2.7578e-01, -4.4203e-03, -2.5783e-02, -9.3828e-02,  2.5156e-01,\n",
       "           1.9253e-01,  1.4145e-01, -4.2573e-01,  3.0091e-02,  2.0523e-01,\n",
       "           1.9929e-01, -1.9835e-01,  5.9970e-02, -3.5347e-01, -3.4772e-01,\n",
       "           4.7067e-01,  4.5500e-01,  2.8334e-01,  2.9687e-01, -1.1389e+01,\n",
       "          -1.2259e-01,  5.9564e-01, -4.0830e-01, -2.9001e-01,  8.8802e-02,\n",
       "          -4.6091e-02, -2.1754e-01,  1.8197e-01, -1.2609e-01, -6.0791e-01,\n",
       "          -5.4574e-01, -2.0938e-01,  2.8982e-01, -3.4635e-01, -4.4516e-01,\n",
       "           1.3640e-01, -8.0581e-01,  1.5269e-01, -2.0424e-01,  5.1419e-01,\n",
       "          -3.2018e-01,  1.7545e-03,  2.7812e-01, -2.5248e-01, -2.5937e-02,\n",
       "           1.0715e-01, -3.0715e-03,  5.4778e-01,  3.6972e-01,  9.6707e-02,\n",
       "           2.7368e-01, -3.1309e-01, -1.4697e-01, -3.1938e-01, -5.8170e-01,\n",
       "          -5.9212e-02, -3.0166e-01,  3.7046e-01,  2.0678e-01, -5.3534e-01,\n",
       "          -3.2953e-01, -3.5016e-01,  2.2099e-01,  2.9859e-01,  9.2455e-01,\n",
       "           6.2430e-01,  1.0096e+00,  3.2782e-01,  6.1630e-02, -1.0016e-01,\n",
       "          -7.7906e-01,  2.3107e-01, -2.1678e-01, -3.8133e-01, -1.3936e-02,\n",
       "           2.9604e-01, -4.9503e-01,  1.0404e-01, -4.3391e-01, -9.8243e-02,\n",
       "           6.7254e-01,  2.0503e-01,  2.2486e-01, -6.3817e-02, -1.8371e-01,\n",
       "           1.5121e-01, -5.9580e-02,  1.9754e-01, -1.2314e-01, -2.3528e-01,\n",
       "           3.7950e-02,  3.5330e-01,  3.8059e-02, -6.4610e-01,  2.7982e-01,\n",
       "          -5.7172e-02, -2.8368e-01,  1.8774e+00,  2.4910e-01,  9.0204e-02,\n",
       "           1.3192e-01,  1.3170e-01, -5.0931e-01, -8.2021e-01,  4.1625e-02,\n",
       "          -1.1771e-01,  4.8621e-02,  3.3109e-01,  3.1045e-01, -3.9686e-01,\n",
       "          -8.3503e-01,  8.8458e-03,  2.6393e-01,  3.8405e-01,  6.6810e-01,\n",
       "          -7.3522e-01, -2.7382e-02,  1.6784e-01, -5.0340e-03, -5.4275e-01,\n",
       "           7.3825e-02, -3.0804e-01, -1.8592e-01,  4.3392e-01, -4.9519e-01,\n",
       "           9.3395e-02, -9.9280e-02, -3.3757e-01,  3.8242e-01, -3.0130e-01,\n",
       "           1.6792e-01,  1.3893e-01,  2.3818e-02, -3.0528e-01,  3.1779e-03,\n",
       "           2.5971e-01,  6.3491e-01, -4.7380e-01, -3.5711e-01, -1.2010e-01,\n",
       "           2.6295e-01,  1.3416e-01,  3.5675e-01, -4.7070e-01,  5.1508e-01,\n",
       "          -7.6393e-01, -5.4993e-02, -1.5303e-02,  2.6561e-02,  3.8799e-01,\n",
       "          -1.0307e-01,  3.6686e-02,  4.1945e-01, -6.9520e-01, -6.0791e-01,\n",
       "          -5.1056e-01, -6.3155e-02,  6.9854e-02, -2.9517e-01,  4.0342e-02,\n",
       "          -1.3685e-01, -4.2188e-02, -1.4064e-01,  2.0590e-01,  5.1737e-03,\n",
       "           3.2727e-02,  3.1345e-01,  4.1379e-03,  1.0171e-01, -8.6417e-02,\n",
       "          -5.1170e-01,  1.9350e-01, -5.6769e-02,  1.1530e-01, -4.9557e-01,\n",
       "          -4.6112e-01, -9.4338e-02, -6.6168e-02, -5.0070e-01,  1.3697e-01,\n",
       "          -1.0336e-01, -3.4211e-02, -2.9169e-01,  4.1859e-01,  4.4790e-01,\n",
       "           1.7379e-01, -3.4664e-01, -1.1183e-01, -2.1416e-01,  2.6304e-01,\n",
       "           1.7069e-01,  4.0728e-01, -2.9340e-01, -7.2700e-01, -1.3937e-01,\n",
       "          -4.5551e-02,  8.0850e-01, -6.0790e-02, -4.7980e-01,  1.0876e-01,\n",
       "           2.9569e-01, -2.2702e-01, -2.5532e-01,  2.9731e-02,  6.5002e-02,\n",
       "          -1.6618e-01, -7.1052e-01,  5.5661e-01,  8.3718e-02,  1.9604e-01,\n",
       "           4.0477e-01,  1.5657e-01,  1.3425e-01,  7.1581e-01, -2.3745e-01,\n",
       "          -6.2046e-01, -2.6741e-01, -7.9407e-01,  1.7069e-01,  1.4890e-01,\n",
       "          -6.3217e-02,  4.8177e-01,  1.0144e-01, -8.9269e-02, -6.0832e-02,\n",
       "           4.0118e-01,  1.1128e+00, -1.2449e-01,  2.6457e-01,  8.0649e-01,\n",
       "           3.0339e-01, -4.5361e-02,  2.0610e-01,  1.6160e-01,  2.6019e-01,\n",
       "           3.8251e-01, -2.0437e-01, -3.2468e-01, -9.1600e-02,  2.0393e-01,\n",
       "           5.3296e-01, -4.6040e-01, -6.9971e-01, -3.2672e-01,  6.8627e-01,\n",
       "          -3.8964e-01,  2.6576e-01, -2.7233e-01,  3.1561e-01,  4.6440e-01,\n",
       "          -8.5009e-01,  7.0507e-03, -1.8646e-01, -1.7795e-02, -1.9321e-03,\n",
       "          -3.6371e-01,  1.5946e-01, -4.0737e-01, -2.0887e-01,  5.2176e-01,\n",
       "           2.4735e-01, -8.0778e-01,  2.0780e-01,  8.5386e-01, -5.5973e-01,\n",
       "           3.2365e-01, -3.9568e-01, -1.8308e-01,  4.0536e-01,  1.4305e-01,\n",
       "          -1.1124e-01,  2.2893e-01, -4.9468e-01, -5.9884e-01, -4.8032e-02,\n",
       "           9.5249e-01, -4.7501e-02,  3.9376e-02, -2.3031e-01, -7.3316e-02,\n",
       "          -2.5871e-01, -3.7012e-01, -3.0622e-01,  1.4150e-01, -9.1555e-02,\n",
       "           7.6066e-02, -1.6942e-02, -1.9552e-01,  3.3901e-01, -2.0967e-01,\n",
       "           2.8537e-01,  3.9990e-01,  9.0599e-01, -3.4863e-02, -3.4870e-01,\n",
       "          -3.8202e-02,  5.6629e-02, -1.1396e-01,  3.8273e-01, -3.0769e-01,\n",
       "          -2.9748e-01,  1.1371e-03,  7.0561e-02, -4.7597e-01,  1.8526e-02,\n",
       "          -3.9717e-01,  1.2734e-01,  6.2316e-02, -1.6889e-01,  4.9766e-01,\n",
       "           3.5772e-02, -5.1320e-03, -2.3181e-01,  3.6451e-01,  3.9618e-01,\n",
       "          -6.6660e-02,  5.4402e-01,  2.1621e-01,  1.7946e-01, -1.1713e-01,\n",
       "           7.7112e-02, -1.0672e+00, -1.3640e-01, -6.8354e-02, -3.5718e-01,\n",
       "          -3.6618e-01, -3.4641e-01,  3.2972e-01,  4.2438e-01,  5.2024e-01,\n",
       "           4.7299e-01,  2.4308e-02, -6.1962e-01, -2.7716e-01,  9.3200e-02,\n",
       "           2.0286e-01, -2.2097e-01,  3.1465e-01,  6.9508e-02,  3.2527e-01,\n",
       "          -4.7197e-01, -4.3926e-01,  1.9456e-02,  3.8353e-01,  7.0162e-02,\n",
       "           9.9717e-01, -6.1842e-02, -1.4776e-01,  1.2019e-01,  7.2328e-02,\n",
       "           1.6920e-01,  7.5268e-01, -3.0741e-02, -6.7453e-01, -2.4769e-02,\n",
       "           2.5609e-04, -3.2638e-01,  3.4564e-01, -8.4256e-01,  1.0561e-01,\n",
       "           2.7495e-01,  4.3042e-01,  4.8383e-01]], device='mps:0'),\n",
       " tensor([[-6.2147e-01, -2.7909e-01, -5.8427e-01, -9.9002e-01, -2.9554e-01,\n",
       "          -4.7538e-01, -9.6206e-01,  6.4104e-01,  6.1091e-02,  4.6500e-01,\n",
       "           7.3369e-02,  4.9601e-02,  2.7944e-01, -3.0968e-01, -5.6239e-01,\n",
       "          -3.4283e-01,  7.8243e-02,  4.1117e-01,  4.2289e-01,  1.6033e-01,\n",
       "           5.7747e-01,  1.6247e-01, -1.4951e-01, -7.9650e-02, -7.4487e-01,\n",
       "           2.2784e-01,  8.2380e-02,  3.5021e-02,  1.7593e-01, -9.8723e-01,\n",
       "           8.9511e-02,  5.7493e-01,  1.7106e-01,  8.4890e-01, -1.5101e-01,\n",
       "          -7.8906e-02,  5.6625e-02, -2.7923e-01,  4.3349e-01,  3.3798e-01,\n",
       "           9.8549e-02, -3.9615e-01,  1.9667e-02,  2.8072e-01, -4.1116e-02,\n",
       "           7.7358e-02,  7.7976e-02, -5.9648e-01,  4.8639e-01, -7.3545e-01,\n",
       "          -4.3894e-01, -3.5320e-01,  5.7964e-01,  3.1114e-01, -9.2843e-01,\n",
       "          -5.6147e-01,  3.0711e-01,  2.8316e-01, -5.0974e-01,  1.6772e-01,\n",
       "           3.9450e-02,  7.9119e-01,  3.0022e-01,  1.7023e-03,  6.3901e-01,\n",
       "           3.0306e-01, -3.2876e-01, -4.8707e-01, -2.1729e-01, -1.8232e-01,\n",
       "          -1.1278e-01,  3.5085e-02, -4.9720e-01, -3.6740e-01,  4.9187e-01,\n",
       "           1.6060e-01,  2.9186e-01,  4.6514e-01, -4.3775e-01,  2.0532e-01,\n",
       "           1.0188e-01,  3.4046e-02,  1.4445e-01,  7.7073e-02,  1.9767e-01,\n",
       "           6.1967e-02,  6.4615e-01,  2.5781e-02, -4.9378e-03,  1.9408e-01,\n",
       "           3.7651e-01, -1.4380e-01,  6.5951e-02, -8.3155e-01,  2.7386e-01,\n",
       "          -3.1128e-01,  4.6411e-01, -2.1227e-01,  1.5833e-01, -1.4959e-01,\n",
       "           3.3854e-01,  4.9563e-01, -2.8463e-01, -3.5598e-02, -4.1844e-01,\n",
       "          -5.4820e-01,  2.2177e-01,  1.7948e-01, -3.7892e-01,  2.7753e-01,\n",
       "           1.2244e-01, -1.1488e-01,  2.0865e-02, -1.7921e-01, -2.4966e-01,\n",
       "          -3.6704e-02,  1.6839e-01, -2.7970e-01, -5.6369e-01,  4.7227e-01,\n",
       "           2.6570e-01, -2.7753e-01, -8.1921e-02,  2.4128e-02,  3.5865e-01,\n",
       "          -3.2498e-01,  4.4390e-01,  1.1966e-01,  2.9490e-01,  4.7950e-01,\n",
       "          -3.0856e-02,  7.7384e-02, -2.5856e-01,  1.1367e-02, -2.7871e-01,\n",
       "           2.2823e-01,  1.1682e-01, -2.3971e-01,  1.1832e-02,  3.8988e-02,\n",
       "          -4.0128e-01, -1.5978e-01, -7.9301e-01, -5.1297e-01, -1.5607e-01,\n",
       "          -3.8031e-01,  9.5759e-02, -2.5969e-04,  2.4645e-02, -3.5630e-01,\n",
       "          -7.6330e-01, -6.9062e-01,  8.1372e-01,  4.1005e-02, -4.8626e-02,\n",
       "          -2.5762e-01, -3.0426e-02,  2.3866e-01, -2.7289e-02,  4.6353e-01,\n",
       "           4.3548e-01,  1.9302e-01,  1.0895e-01, -5.9339e-01, -2.1304e-01,\n",
       "           4.8371e-02, -2.2364e-01, -3.8808e-01,  9.5696e-02,  2.8964e-01,\n",
       "           2.5241e-01, -1.1163e-03, -2.7771e-02, -7.5562e-01, -1.4601e-01,\n",
       "          -5.7426e-01,  3.2970e-01,  4.3319e-01, -4.0065e-01, -1.6685e-01,\n",
       "          -1.0298e-01, -1.5471e+00,  2.1071e-01, -8.2421e-01, -2.5399e-01,\n",
       "          -8.1985e-01,  6.0274e-02,  2.4573e-01,  1.4394e-01,  2.1386e-01,\n",
       "           2.0622e-01,  1.7473e-01,  5.8743e-02,  3.3387e-01,  2.2759e-01,\n",
       "           7.8413e-02,  5.2019e-01, -6.0536e-02, -9.8967e-02,  2.1916e-01,\n",
       "          -2.8847e-01,  4.3442e-01, -1.9413e-01,  1.3818e-01,  3.1818e-01,\n",
       "           4.0064e-01, -3.8991e-01,  3.3452e-01,  7.1941e-02, -3.1507e-01,\n",
       "          -6.6971e-02, -1.7744e-01, -4.2581e-01, -9.5990e-02, -3.6681e-01,\n",
       "          -5.7182e-02, -4.5243e-01,  8.7616e-02,  3.1282e-01,  1.6453e-01,\n",
       "          -9.4469e-02,  2.9940e-01, -2.3357e-01, -4.0181e-01, -1.9304e-02,\n",
       "           5.0003e-01, -1.9707e-01,  3.9355e-01, -2.9817e-01,  1.0271e-01,\n",
       "          -3.4203e-01, -6.8618e-01,  5.7258e-01,  8.3060e-01,  2.6556e-01,\n",
       "          -3.0777e-01, -3.3172e-01,  2.2363e-01, -1.5463e-01, -3.9629e-01,\n",
       "          -3.3078e-01, -4.1735e-01, -2.8823e-01, -1.3198e-01,  4.7375e-01,\n",
       "           1.0095e-01, -1.0612e-01,  1.0587e+00,  6.0503e-01,  1.7057e-01,\n",
       "          -1.2056e-01, -1.3545e-01,  2.4273e-01,  4.2754e-01, -4.0979e-01,\n",
       "          -2.3499e-01,  8.0182e-05, -6.0867e-01,  5.5855e-01, -1.0272e-01,\n",
       "           1.5502e-01,  1.0790e-01,  9.8681e-02,  5.6712e-01,  3.2353e-01,\n",
       "          -3.5060e-01, -3.6505e-02,  1.5962e-01, -2.5410e-01, -5.3648e-02,\n",
       "          -1.4294e-01,  2.0909e-01, -6.6258e-01, -2.4928e-01, -7.1921e-01,\n",
       "          -3.0484e-01, -1.8090e-01,  3.2368e-01,  3.8572e-01, -7.5476e-01,\n",
       "          -2.2074e-01,  2.4454e-01, -2.7909e-01,  2.5780e-01,  2.5031e-01,\n",
       "           3.6506e-01, -1.6624e-01, -3.8172e-02, -7.9168e-02, -5.4046e-02,\n",
       "          -8.9020e-02,  3.0304e-01, -2.0984e-01,  6.4244e-02,  7.4377e-02,\n",
       "          -1.9079e-01,  4.2329e-01,  2.4846e-01, -1.5845e-02, -3.6455e-01,\n",
       "          -7.2769e-02, -3.4083e-01, -2.9112e-01, -3.4985e-01, -6.1713e-02,\n",
       "          -1.0931e-01,  3.1988e-01,  7.6290e-01,  7.8416e-02, -1.7579e-01,\n",
       "           2.2704e-01,  6.2818e-01, -1.4475e-01, -1.1080e-02, -1.2639e-01,\n",
       "          -1.0949e+00, -1.1506e-01, -1.4284e-01, -2.0997e-01, -4.1901e-01,\n",
       "           1.6316e-01,  4.8597e-01, -2.9807e-01, -2.5226e-01, -6.1721e-01,\n",
       "           3.5488e-01,  1.7183e-01,  2.4734e-01,  7.1970e-02, -1.1558e-01,\n",
       "           3.6128e-03, -8.2397e-01,  1.9348e-01, -4.9155e-01, -2.0193e-01,\n",
       "           3.0291e-01,  2.8698e-01, -1.3359e-03,  2.1475e-01, -6.3376e-01,\n",
       "           7.5167e-01, -2.7316e-01, -1.3171e-01,  4.7497e-01, -5.2777e-01,\n",
       "          -1.3179e-01,  4.5365e-01, -1.0714e-01,  3.3660e-01, -8.5164e-03,\n",
       "           5.5819e-01, -2.8265e-01, -6.3096e-01, -1.5259e-01,  4.1793e-01,\n",
       "           2.0545e-01,  5.2061e-01, -3.0179e-01, -6.4133e-02,  2.3854e-01,\n",
       "          -3.9864e-01,  5.8216e-01, -3.2775e-01, -3.8716e-01,  4.2894e-02,\n",
       "           3.6800e-01,  2.3399e-01, -3.5745e-01, -1.0782e-01,  4.1811e-01,\n",
       "           7.1613e-04, -1.5130e-01,  3.3448e-01,  4.0361e-01,  5.2881e-02,\n",
       "          -3.7063e-01, -4.6386e-01,  2.1292e-01, -4.3723e-01, -7.7599e-02,\n",
       "           2.5724e-01, -1.6890e-01,  3.0597e-01, -2.6166e-01, -3.4578e-01,\n",
       "          -3.3132e-01,  2.3619e-01,  5.5639e-01,  4.0263e-01,  2.6231e-01,\n",
       "           2.8771e-01, -6.5572e-02, -1.8660e-01,  5.0480e-01, -2.1051e-01,\n",
       "          -3.5671e-01,  4.9652e-01, -2.1034e-01,  4.9551e-02,  2.4658e-01,\n",
       "          -1.7487e-01,  1.9092e-01,  3.1132e-01,  4.7956e-01, -2.8902e-01,\n",
       "          -3.7743e-01, -3.9740e-02, -3.4959e-02, -1.9822e-01,  1.9328e-01,\n",
       "           1.3177e-01,  5.9879e-01, -1.1406e-01,  8.4461e-02,  7.0430e-01,\n",
       "          -9.8085e-02, -1.2380e-01,  1.5140e-01, -5.1596e-01, -4.3391e-01,\n",
       "           1.7553e-01,  1.6226e-01,  1.0138e-01,  3.5805e-01, -1.0833e+01,\n",
       "          -1.6207e-01,  9.0578e-01, -6.0310e-01,  2.4933e-02,  1.6747e-01,\n",
       "           7.1939e-02, -1.1283e-01,  3.5721e-01, -1.6514e-01, -7.0795e-01,\n",
       "          -5.0718e-01, -9.1159e-02,  6.9236e-02,  1.3412e-01, -3.8980e-01,\n",
       "          -2.3735e-02, -7.9766e-01, -1.6670e-01, -5.2474e-01,  2.9513e-01,\n",
       "          -1.2206e-01,  1.4791e-01,  2.7908e-01, -2.5309e-01, -1.9447e-01,\n",
       "          -1.0511e-01, -1.6679e-01,  4.4689e-02,  7.2576e-02, -1.6973e-01,\n",
       "          -2.1179e-01, -1.1415e-01,  1.4888e-01, -3.2851e-02, -8.0626e-01,\n",
       "          -1.2724e-01, -7.5413e-02,  7.2363e-01, -2.1558e-01, -7.4533e-01,\n",
       "          -4.2988e-01, -5.3624e-01,  4.6139e-01,  4.1852e-01,  8.1323e-01,\n",
       "           8.9064e-01,  8.1672e-01,  2.7935e-01,  1.8352e-01, -5.4103e-02,\n",
       "          -3.9133e-01,  1.2840e-01, -5.0677e-01, -2.7266e-01, -2.9463e-01,\n",
       "           3.1125e-01, -6.7750e-01,  4.4689e-01, -2.7814e-01, -2.4217e-01,\n",
       "           2.4110e-01, -1.9155e-02,  2.7564e-01, -1.0594e-01, -6.4239e-01,\n",
       "           7.9492e-01, -2.1349e-02,  7.9448e-02, -3.4257e-01,  1.0813e-01,\n",
       "          -2.5206e-01,  6.6180e-01,  4.5828e-01, -2.2729e-01,  6.0107e-01,\n",
       "          -6.7045e-02, -3.3716e-01,  1.7347e+00, -4.4232e-01,  2.0456e-01,\n",
       "          -1.4020e-01,  1.5561e-01, -6.5670e-01, -7.4457e-01,  5.0137e-02,\n",
       "           5.1606e-02,  5.1494e-02,  5.3269e-01,  2.8032e-01, -2.3415e-01,\n",
       "          -7.1568e-01,  2.3439e-01,  1.5848e-01,  4.7977e-01,  7.7819e-01,\n",
       "          -2.7915e-01, -3.1101e-01,  5.3442e-02, -3.2881e-02, -1.1440e+00,\n",
       "          -5.5691e-02, -3.6070e-01, -3.4941e-01,  4.0776e-01,  1.2756e-01,\n",
       "           4.3550e-01,  1.9093e-01,  1.3858e-02,  6.1001e-01, -4.1931e-01,\n",
       "           2.8392e-01, -3.4018e-01,  2.2048e-01, -4.1684e-01,  6.3885e-02,\n",
       "           2.6338e-02,  3.2704e-01, -2.1298e-01, -6.4053e-01, -1.3787e-01,\n",
       "          -4.5131e-02,  2.0702e-01,  3.3411e-01, -6.2706e-01,  4.8566e-01,\n",
       "          -5.5122e-01, -1.2985e-01, -5.2469e-02, -2.5593e-01,  7.1763e-01,\n",
       "          -2.7623e-01, -2.3047e-01, -1.9115e-02, -6.0895e-01, -3.6121e-01,\n",
       "          -3.9694e-01, -7.7886e-02,  4.4520e-01, -8.2197e-01, -1.1430e-01,\n",
       "          -9.0592e-03,  3.1419e-01, -3.3115e-01,  3.2193e-01,  3.1982e-01,\n",
       "          -8.4345e-02,  5.6772e-01,  1.0522e-01,  2.6664e-01, -2.1187e-01,\n",
       "          -5.2860e-01, -2.8810e-02,  2.3729e-01, -4.1434e-02, -9.5386e-01,\n",
       "          -2.2150e-01,  2.9814e-02, -3.4193e-01,  3.2673e-01,  1.2839e-02,\n",
       "          -6.6403e-02, -1.8773e-01, -5.7201e-01,  5.0832e-01,  6.1644e-01,\n",
       "           1.1682e-01, -7.7501e-02, -1.1131e-01,  5.7220e-02,  4.7815e-01,\n",
       "           6.3466e-02, -8.8508e-02, -4.8506e-01, -2.0456e-01,  2.2971e-01,\n",
       "           2.1680e-01,  6.1788e-01, -2.7839e-01, -6.1943e-01,  4.2313e-01,\n",
       "           3.7894e-01,  1.6323e-01,  1.8677e-02, -2.2962e-01, -2.8187e-01,\n",
       "          -3.9409e-01, -6.5763e-02,  5.0300e-01, -2.1894e-01,  8.7850e-02,\n",
       "           2.8566e-01,  5.6102e-01,  2.0975e-01,  3.4104e-01, -5.8759e-01,\n",
       "          -2.7717e-01,  7.2183e-03, -7.7288e-01,  9.8352e-03,  2.7177e-02,\n",
       "          -5.4430e-02,  2.8733e-01, -7.9284e-02, -8.8908e-02,  4.4477e-01,\n",
       "           4.4546e-01,  7.2034e-01,  8.2220e-02,  4.9640e-02,  7.0942e-01,\n",
       "           3.8687e-01,  1.8191e-01,  6.5891e-02,  4.4234e-01,  1.2782e-01,\n",
       "           2.0089e-01,  5.6057e-02, -3.7844e-01, -6.9491e-03,  6.2278e-02,\n",
       "           5.9645e-01, -1.2927e+00, -6.1757e-01, -1.6875e-01,  4.3174e-01,\n",
       "          -4.7422e-01,  1.6580e-01, -8.6364e-02,  4.1136e-01,  5.9741e-01,\n",
       "          -1.2876e+00, -5.4781e-01,  2.5280e-01,  2.0265e-01, -2.1166e-02,\n",
       "          -5.8650e-01,  1.2355e-01, -3.4895e-01, -2.2824e-01,  5.8641e-01,\n",
       "           3.9472e-01, -7.2788e-01,  1.4399e-01,  6.0641e-01, -5.7154e-01,\n",
       "           1.8083e-01,  6.9447e-02,  1.5791e-01,  8.5714e-02,  8.6819e-03,\n",
       "           3.1758e-02,  3.1329e-01, -5.3736e-01, -4.8835e-01, -5.3950e-02,\n",
       "           4.6537e-01,  7.9753e-02,  1.8423e-01, -1.7508e-01,  4.0627e-01,\n",
       "          -3.3120e-01, -2.0646e-01, -2.2260e-01, -6.0527e-02,  4.7484e-02,\n",
       "           4.4993e-01, -7.0835e-02, -1.1495e-01,  3.8020e-01, -2.4916e-01,\n",
       "           4.3580e-01,  3.7290e-01,  5.3130e-01,  3.7457e-02, -2.8417e-01,\n",
       "           1.6727e-01,  9.6342e-02, -3.8526e-01,  5.5867e-01, -8.9730e-02,\n",
       "          -4.1383e-01,  1.2647e-01,  1.1172e-02, -9.5989e-02, -1.9809e-01,\n",
       "          -1.1288e-01,  2.4349e-01,  2.2980e-02, -3.6165e-01,  2.0880e-01,\n",
       "           1.7864e-01,  1.7051e-01, -8.0907e-01,  2.8740e-01,  3.1193e-01,\n",
       "          -1.6036e-01,  1.0301e-01, -5.6301e-02,  1.8537e-01, -6.3504e-01,\n",
       "           2.2279e-01, -7.1048e-01, -2.6776e-01,  3.7713e-01, -1.1157e-01,\n",
       "          -5.0124e-01, -9.8763e-02, -2.5023e-01,  1.0370e-01,  1.0287e+00,\n",
       "           6.9883e-01, -4.0463e-02, -6.1774e-01, -1.5172e-01,  2.5662e-01,\n",
       "          -2.2699e-02, -2.9047e-01,  4.1420e-01, -8.0642e-02,  4.1008e-02,\n",
       "          -1.1916e+00, -2.4969e-01,  4.9257e-02,  1.0534e-01, -1.4916e-01,\n",
       "           9.3032e-01, -4.1688e-01,  3.1935e-01,  7.7488e-01,  1.8214e-01,\n",
       "           3.6172e-01,  4.8894e-01,  6.2518e-02, -3.5421e-01, -1.3332e-01,\n",
       "           3.1418e-01, -2.5181e-01, -1.8999e-01, -5.0514e-01,  3.1585e-01,\n",
       "           3.3923e-02,  4.4809e-01,  6.5872e-01]], device='mps:0'),\n",
       " tensor([[-1.2655e-01, -7.4520e-01, -7.3208e-01, -2.8851e-01, -4.0142e-01,\n",
       "          -1.1156e-01, -1.7252e-01,  5.3901e-01,  7.7116e-02,  7.3915e-01,\n",
       "           1.7537e-01,  1.7667e-01, -2.6012e-02, -1.4756e-01, -3.8676e-01,\n",
       "          -3.7443e-01, -3.3357e-02,  2.8249e-01, -1.9368e-01, -2.2135e-01,\n",
       "           6.0315e-01,  4.3964e-01, -2.4971e-01,  8.9217e-02, -6.7105e-01,\n",
       "          -4.0270e-02,  1.1560e-01,  9.3162e-02,  8.8618e-02, -7.6946e-01,\n",
       "          -1.3872e-01,  6.3594e-01,  1.0944e-01,  1.1640e+00, -2.1243e-01,\n",
       "          -2.0144e-01,  6.2456e-02, -3.6353e-01,  3.5678e-01, -5.0662e-01,\n",
       "           6.4547e-01,  9.5906e-02,  1.0728e-01,  4.9590e-01,  3.0941e-01,\n",
       "          -9.9768e-02,  3.8522e-01, -9.3185e-01,  5.9163e-01, -5.7836e-01,\n",
       "          -4.5920e-01,  2.5531e-01,  3.5370e-01,  2.4685e-01, -5.3818e-01,\n",
       "          -3.4115e-01,  4.4585e-01,  4.0757e-01, -3.7288e-01,  2.8399e-01,\n",
       "          -2.7777e-01,  4.8607e-01, -9.5103e-02,  1.2683e-01,  3.3310e-01,\n",
       "           2.5259e-01, -3.6833e-01, -4.1241e-01, -3.6330e-01, -1.6269e-01,\n",
       "          -1.7272e-02,  3.9606e-01, -9.4364e-02, -4.2791e-01,  2.8922e-01,\n",
       "          -2.7780e-01, -2.6952e-02, -3.1996e-01, -5.3741e-01, -1.5066e-01,\n",
       "           2.3701e-01,  4.1333e-01,  1.2565e-01, -2.6415e-01,  1.6644e-01,\n",
       "           5.9443e-02,  3.9176e-01,  4.9960e-01,  2.5199e-01, -1.6932e-01,\n",
       "           7.0013e-01,  1.1491e-01, -4.5725e-01, -3.0624e-01,  4.2559e-01,\n",
       "          -1.8026e-01,  4.8127e-01, -2.2585e-01, -1.7340e-01, -2.0582e-01,\n",
       "           2.9714e-02,  1.4234e-01,  1.3911e-01, -1.5366e-01, -1.9418e-01,\n",
       "          -4.5523e-01,  9.2212e-02, -1.0325e-01, -7.3849e-01, -3.2521e-01,\n",
       "          -2.1782e-01,  1.0146e-01,  2.7080e-01, -6.0420e-01, -4.1818e-02,\n",
       "           5.6733e-03,  1.6661e-01, -2.6615e-02, -2.2333e-01,  5.1674e-01,\n",
       "           2.3316e-01, -3.5313e-01, -1.5852e-01,  2.8876e-01,  4.5897e-01,\n",
       "          -5.7108e-01,  7.3468e-01, -3.4321e-01,  3.1998e-01,  4.8956e-01,\n",
       "           9.4281e-02,  8.1907e-02, -4.1177e-01, -1.0418e-01,  2.1088e-01,\n",
       "          -6.6784e-02,  4.1300e-01, -4.0179e-01,  1.9421e-02, -3.8683e-02,\n",
       "           1.6348e-01, -1.7412e-01, -4.3610e-01, -1.1717e-02, -1.1581e-01,\n",
       "          -1.6510e-01, -1.8304e-01, -7.2298e-02, -2.9771e-01, -5.0788e-01,\n",
       "          -5.4121e-01, -5.5751e-01,  1.0337e-01, -2.3788e-01,  1.0912e-01,\n",
       "          -3.0113e-01,  2.7625e-01,  2.0260e-01, -8.1330e-03,  1.9222e-01,\n",
       "           7.9019e-01, -2.1016e-01,  2.6126e-01, -8.5382e-01, -7.1905e-02,\n",
       "          -4.8229e-02, -4.6440e-02, -1.7337e-01,  8.4039e-03,  4.1042e-02,\n",
       "           3.5682e-01, -2.8487e-01, -9.8864e-02, -4.7375e-01,  1.5659e-01,\n",
       "          -5.6279e-01,  4.2615e-01,  1.3419e-01, -4.2396e-02, -1.4768e-01,\n",
       "           9.7335e-02, -1.8931e+00, -1.5465e-01, -2.6779e-02, -1.4878e-01,\n",
       "          -4.4806e-01,  3.9898e-01,  3.0941e-01,  3.3377e-01, -1.3627e-01,\n",
       "          -6.6707e-01,  9.2625e-02,  2.3287e-02,  1.6853e-01,  1.2024e-01,\n",
       "           4.5454e-01,  4.6832e-01,  9.4002e-03,  2.9393e-02,  3.2991e-01,\n",
       "          -6.1158e-01,  2.3558e-01,  4.2741e-02, -2.0119e-01,  7.1801e-02,\n",
       "           1.7236e-01, -2.4868e-01,  1.9310e-01,  1.3628e-01,  1.6636e-01,\n",
       "          -2.4036e-01, -2.2848e-01, -1.5175e-01, -2.2029e-02, -4.4728e-01,\n",
       "           1.2618e+00,  4.6907e-02, -1.5209e-01,  4.7050e-01,  2.5533e-01,\n",
       "          -1.1167e-01, -4.1617e-02,  8.5119e-02, -2.4530e-01, -9.3295e-02,\n",
       "           6.0232e-01, -5.8611e-01, -9.9963e-02, -1.4179e-01, -2.6297e-01,\n",
       "          -1.9659e-02, -6.8554e-01,  6.7727e-01, -2.9161e-01,  2.3121e-02,\n",
       "          -2.3610e-01, -5.6084e-02, -3.2733e-03, -2.4264e-01,  1.1758e-01,\n",
       "          -5.1666e-01, -5.9031e-01, -8.7811e-01,  2.3734e-01,  2.1112e-01,\n",
       "           3.2798e-02, -2.9125e-01,  8.7574e-01,  4.0826e-01,  5.6717e-01,\n",
       "          -2.9647e-02, -2.5191e-01,  3.9790e-01, -2.7886e-01, -1.5353e-01,\n",
       "          -3.6532e-02, -1.0843e-01, -2.4601e-01,  4.9387e-01, -1.4118e-01,\n",
       "          -1.1985e-01, -1.7838e-01,  2.8209e-01,  7.1989e-01, -1.3719e-01,\n",
       "           1.5138e-01,  1.4714e-01, -4.7592e-01, -2.4229e-01,  4.8534e-01,\n",
       "          -9.8380e-02,  1.7773e-01, -2.5375e-01, -5.9610e-01, -5.0816e-01,\n",
       "          -4.2394e-01,  3.6723e-02, -4.1870e-01,  4.6948e-01, -7.8169e-01,\n",
       "          -4.6660e-01,  2.7831e-01, -1.4913e-01,  9.1576e-02, -2.5437e-01,\n",
       "           4.2342e-01,  1.6156e-01,  3.1737e-01, -2.6485e-01, -5.5156e-01,\n",
       "          -5.8013e-02,  2.6101e-01, -6.4054e-02, -2.2062e-02,  1.4868e-01,\n",
       "          -1.6116e-02,  3.8746e-01, -2.7389e-02, -5.7911e-02, -2.5913e-01,\n",
       "           2.2320e-02, -1.8898e-01, -1.6678e-01, -1.5143e-01,  1.2601e-01,\n",
       "           1.4057e-02,  1.0538e-01,  1.8138e-01, -5.9508e-01, -7.2537e-01,\n",
       "           1.5597e-02,  5.9317e-01, -1.4668e-01, -1.0549e-01, -2.0807e-01,\n",
       "          -8.5347e-01, -4.2241e-01, -2.0503e-01, -2.0471e-01, -4.2933e-02,\n",
       "           1.1471e-01,  1.5247e-01, -8.3550e-02, -2.9233e-02, -4.9752e-01,\n",
       "           2.7221e-01,  3.8839e-01,  6.6993e-01,  1.1004e-01, -1.4645e-01,\n",
       "          -2.8724e-01, -2.9969e-01, -2.9549e-01, -6.2650e-03, -2.0145e-01,\n",
       "           2.8245e-01,  2.1621e-01, -5.0797e-01, -2.7820e-01, -8.5465e-02,\n",
       "           5.3849e-01, -5.6729e-01, -5.8864e-01,  1.3343e-01, -3.1772e-01,\n",
       "          -6.6511e-01,  1.3537e-01, -6.8609e-01, -7.7676e-02,  1.6837e-01,\n",
       "           3.0898e-01, -3.5734e-01,  1.5134e-01, -3.4125e-01,  7.0115e-01,\n",
       "           5.0160e-01,  4.2641e-01, -4.4835e-01,  3.9933e-01,  4.5206e-01,\n",
       "          -5.1036e-01,  1.6760e-02, -2.3424e-01, -1.6616e-01, -1.3416e-01,\n",
       "           1.4652e-01,  3.2764e-01, -6.9834e-01,  1.5715e-01,  3.1416e-01,\n",
       "           5.0575e-02, -1.9972e-01, -3.9472e-01,  2.8528e-01, -3.0174e-01,\n",
       "          -1.8285e-01, -3.6691e-01,  1.1957e-01, -1.7406e-01, -3.2604e-01,\n",
       "           2.0167e-01,  7.2607e-02, -6.4363e-02, -2.7870e-01, -1.4528e-01,\n",
       "          -2.5803e-01,  3.0054e-01,  7.2737e-01,  7.0952e-02,  2.7148e-01,\n",
       "           1.0595e-01, -5.6026e-02, -1.7841e-01,  4.6604e-01,  5.7957e-02,\n",
       "          -2.9957e-01,  2.5715e-01, -4.8969e-02,  2.3108e-01,  7.1128e-01,\n",
       "          -3.2841e-01, -4.9112e-01,  9.4450e-02, -1.2226e-01, -2.4214e-01,\n",
       "          -3.2170e-01,  2.9251e-01,  3.9917e-01,  1.2972e-01,  7.2342e-02,\n",
       "          -2.2402e-01,  6.4680e-01, -3.7128e-01, -2.2534e-01, -2.1277e-01,\n",
       "           1.5045e-01, -2.3928e-02, -3.7103e-02, -5.8279e-02, -3.2639e-01,\n",
       "           3.5380e-01,  2.3950e-01,  4.7152e-01,  1.3765e-01, -1.1137e+01,\n",
       "           2.3304e-01,  8.5987e-01, -5.3979e-01,  2.0917e-01,  1.1097e-01,\n",
       "           3.8024e-01, -2.2548e-01,  1.9945e-01, -2.2175e-01, -2.0747e-01,\n",
       "           1.0544e-01, -1.0698e-01,  2.6923e-01, -5.2697e-01,  2.6734e-01,\n",
       "          -9.4496e-02, -6.8301e-01,  4.8033e-01, -3.7921e-01,  5.5370e-01,\n",
       "          -4.9093e-01,  3.7755e-01,  2.4613e-01, -2.4723e-01, -5.5835e-01,\n",
       "          -1.1016e-01, -5.9247e-01,  2.0945e-01,  6.5333e-01, -1.2674e-01,\n",
       "           7.5986e-02,  1.2314e-01,  3.9375e-01, -2.7636e-01, -7.2495e-01,\n",
       "           3.3798e-01, -2.5869e-01,  2.7356e-01,  2.8488e-01, -3.2794e-01,\n",
       "           1.7858e-01, -3.0247e-01,  6.9807e-01,  2.2586e-01,  7.0655e-01,\n",
       "           1.0020e+00,  7.1939e-01, -3.5037e-02, -6.7434e-01,  1.0608e-01,\n",
       "          -7.8312e-01,  4.9115e-02, -7.6924e-01,  1.9184e-01, -1.7609e-01,\n",
       "           4.8911e-01, -5.8598e-01,  8.3167e-02,  9.0617e-02, -5.0307e-03,\n",
       "           1.6894e-01, -1.6678e-01,  2.9473e-01, -2.3873e-01, -1.1305e-01,\n",
       "           3.5025e-01, -1.2861e-01,  2.4635e-01,  1.6825e-01, -4.8248e-01,\n",
       "          -3.1677e-01,  4.1630e-01,  9.9644e-02, -6.3425e-01,  2.9325e-01,\n",
       "           3.9251e-02, -2.3113e-02,  5.3637e-01, -4.7048e-02, -1.3664e-01,\n",
       "           3.8503e-01,  4.1584e-01, -1.0700e-01, -5.4898e-01, -3.8213e-01,\n",
       "          -1.1865e-01,  7.3741e-02,  2.9014e-01,  5.7460e-01, -3.2655e-01,\n",
       "          -3.7755e-01, -3.0112e-01,  1.5459e-01, -1.8642e-03,  1.0263e+00,\n",
       "          -7.1321e-01,  5.3698e-01,  3.1682e-01, -2.2188e-01, -4.7782e-01,\n",
       "          -4.2439e-02, -1.6548e-01, -3.3178e-01,  1.4035e-01, -3.9319e-01,\n",
       "           5.2460e-02,  3.1456e-01,  1.7819e-01,  4.4850e-01, -1.3422e-01,\n",
       "           3.7151e-01,  2.0540e-01,  1.0344e-01, -2.9791e-01,  1.8818e-01,\n",
       "           3.0356e-01,  7.1717e-01, -4.2188e-01, -1.9580e-01, -4.0693e-01,\n",
       "           7.0743e-02,  5.6798e-01,  4.0898e-02, -3.8230e-01,  9.8394e-02,\n",
       "          -7.2955e-01, -9.2036e-02, -2.4510e-01, -3.1327e-02,  4.7172e-01,\n",
       "           1.5715e-01, -2.1307e-02,  2.0088e-02, -2.1966e-01, -3.9113e-01,\n",
       "          -9.2195e-01, -1.6279e-01,  1.2524e-01, -4.8603e-02,  2.1508e-01,\n",
       "          -8.5762e-02, -1.4127e-01, -3.4117e-02,  6.5544e-01, -1.5354e-02,\n",
       "           7.1977e-01,  2.2196e-01, -1.3854e-01,  1.5714e-01, -6.1457e-02,\n",
       "          -4.3722e-01, -2.1751e-01, -2.9514e-01, -1.8853e-01, -2.8292e-01,\n",
       "           6.0776e-02,  1.7421e-01,  1.5628e-01, -3.5800e-01,  4.9556e-01,\n",
       "           5.3650e-03, -1.7080e-01, -1.1970e-01,  6.6134e-02,  1.4497e-01,\n",
       "           1.0977e-01, -3.4114e-01, -2.4418e-01, -9.5924e-02,  7.2654e-02,\n",
       "           2.3113e-01, -4.7076e-01,  3.5924e-02, -5.8183e-01,  1.4300e-01,\n",
       "           1.8358e-01,  1.3775e+00, -3.3167e-02, -5.4050e-01,  1.5789e-01,\n",
       "           6.0053e-01, -3.3132e-01,  1.5013e-01, -1.6961e-02,  2.0315e-01,\n",
       "          -4.3932e-01, -1.6432e-01,  3.2769e-01, -9.7204e-02,  2.7980e-01,\n",
       "           6.6650e-03,  2.8495e-01, -4.1759e-01,  3.4674e-01,  1.1297e-01,\n",
       "          -3.7993e-01, -1.3553e-01, -5.6543e-01, -3.4098e-02,  6.2812e-01,\n",
       "           3.0632e-02, -1.5477e-01, -2.2456e-01,  4.4538e-03, -1.7448e-01,\n",
       "           4.5311e-01,  1.2193e+00,  2.2650e-02,  4.3392e-02,  4.1220e-01,\n",
       "           1.3233e-02, -3.5122e-01,  2.5082e-01,  7.7446e-02,  6.5709e-02,\n",
       "           4.1433e-01,  2.8429e-01, -3.9413e-01,  3.2738e-01,  1.8144e-01,\n",
       "          -1.2196e-02, -8.6635e-01, -7.4945e-01, -1.8081e-01,  2.9443e-01,\n",
       "           2.3014e-01,  3.1770e-01, -6.9946e-02,  5.6758e-01,  8.6579e-01,\n",
       "          -1.0240e+00, -3.6591e-01,  2.1984e-01,  1.1269e-01, -1.3523e-01,\n",
       "          -2.8316e-01,  3.6406e-02, -6.6481e-02, -4.8127e-01,  2.5986e-01,\n",
       "           7.4845e-01, -8.2243e-01,  1.0108e-01,  3.3324e-01, -4.8112e-01,\n",
       "          -4.6681e-02,  1.0242e-02,  2.7858e-02,  3.4290e-01,  4.9584e-02,\n",
       "          -5.8812e-01,  5.7447e-01, -4.3146e-01, -5.1602e-01,  3.3867e-01,\n",
       "           6.1310e-01, -1.7420e-02, -1.5538e-02,  2.5800e-01, -1.9679e-01,\n",
       "           1.0622e-01, -3.8051e-01, -3.0642e-01, -4.1422e-02, -3.3001e-01,\n",
       "           4.2411e-02, -1.9060e-01,  3.5958e-01,  1.5858e-01, -6.1822e-01,\n",
       "           2.4092e-01,  7.2764e-01,  8.2494e-01, -4.0542e-02, -4.1354e-01,\n",
       "           4.3800e-01, -4.4073e-02, -3.2619e-01,  4.6795e-01, -3.7749e-02,\n",
       "          -1.7259e-01,  3.5766e-01,  1.9563e-01, -2.8715e-01, -1.5703e-01,\n",
       "          -2.8124e-01,  2.8023e-01, -6.0972e-02, -3.8436e-01,  2.7954e-01,\n",
       "          -1.8757e-01,  3.1244e-01, -6.0115e-01,  1.0007e-01,  3.1193e-01,\n",
       "          -6.1689e-02,  7.0562e-01,  1.3887e-01,  4.2675e-01, -1.3264e-01,\n",
       "           1.2326e-01, -4.0292e-01, -4.7626e-01,  1.4668e-02,  1.1085e-01,\n",
       "          -3.0529e-01,  2.2641e-01,  1.2448e-01,  3.0816e-02,  7.6298e-01,\n",
       "           7.3448e-01, -4.3314e-02, -6.5195e-01,  4.6754e-02,  2.6127e-02,\n",
       "           1.4634e-01, -1.0814e-01,  3.7987e-01, -2.9931e-01, -7.7688e-02,\n",
       "          -8.8698e-01, -3.9418e-01,  1.6635e-01,  2.5230e-01, -2.3809e-01,\n",
       "           5.2611e-01, -1.3072e-01,  2.8637e-01,  3.2407e-01,  7.9897e-02,\n",
       "           3.1870e-01,  9.1416e-01,  2.1795e-01, -5.4236e-01,  2.1027e-01,\n",
       "           5.3772e-01, -1.5431e-01, -3.0445e-01, -8.9030e-01,  2.3249e-01,\n",
       "           1.3295e-02,  7.6987e-01,  1.5070e-01]], device='mps:0'),\n",
       " tensor([[-0.2744, -0.4852, -0.4880,  ...,  0.2064,  0.2977,  0.6606],\n",
       "         [ 0.0506, -0.6230, -0.6446,  ..., -0.0875,  0.4018,  0.3101],\n",
       "         [-0.4643, -0.3750, -0.2078,  ...,  0.2835,  0.5093,  0.4110]],\n",
       "        device='mps:0'),\n",
       " tensor([[-8.7564e-02, -4.9833e-01, -6.3687e-01, -4.0578e-01, -6.9840e-01,\n",
       "          -2.3387e-01, -4.1180e-01,  5.2317e-01, -2.2035e-01,  4.5067e-01,\n",
       "           2.5062e-01,  1.1995e-01,  1.5031e-01, -9.6288e-02, -6.0084e-01,\n",
       "          -3.4361e-01,  3.9723e-01,  3.9226e-01,  6.5111e-01,  1.1035e-01,\n",
       "           6.7427e-01,  3.9429e-01, -1.4784e-01, -1.1417e-01, -8.9684e-01,\n",
       "           2.6734e-01, -1.5040e-01,  3.5751e-02,  3.8249e-01, -6.5691e-01,\n",
       "          -5.0431e-01,  1.2997e-01, -9.6623e-02,  7.5944e-01, -3.4214e-01,\n",
       "          -4.9353e-03,  1.6918e-01, -5.5750e-01,  1.5409e-01,  3.5742e-02,\n",
       "           5.0410e-01,  2.7796e-02,  8.9746e-02,  4.2259e-01,  5.2340e-01,\n",
       "          -4.3294e-03, -9.1370e-02, -7.8080e-01,  1.3439e-01, -2.8322e-01,\n",
       "           2.7167e-01,  2.8308e-02,  4.6509e-01,  4.3443e-01, -6.3205e-01,\n",
       "          -7.4689e-01,  3.7258e-01,  1.9066e-01, -4.4268e-01,  3.4560e-01,\n",
       "          -2.7884e-01,  7.1737e-01,  1.3666e-01,  4.6901e-02, -1.7654e-01,\n",
       "           3.4580e-02, -1.9115e-01, -9.7322e-01, -2.5911e-01, -2.2553e-01,\n",
       "          -2.1928e-01,  2.6272e-01, -2.6864e-01, -1.8710e-01,  1.7180e-01,\n",
       "          -2.5149e-01,  2.3745e-01, -5.6538e-02, -3.8011e-01, -1.4267e-01,\n",
       "           3.4435e-01,  1.6998e-01,  8.6814e-02, -4.4090e-02, -4.8772e-01,\n",
       "           3.1845e-02,  5.0133e-01,  4.0723e-01,  5.2782e-01, -1.2693e-01,\n",
       "           2.5539e-01,  1.3344e-01, -1.3033e-01, -3.9779e-01,  4.8072e-01,\n",
       "          -4.2228e-01,  3.0926e-01, -1.4101e-01,  9.2363e-02, -4.1453e-01,\n",
       "           8.9056e-02,  3.0370e-01, -9.2494e-02,  2.4348e-01, -4.8023e-01,\n",
       "          -7.0432e-01,  1.1504e-01, -1.5078e-01, -9.5519e-01, -1.0935e-01,\n",
       "          -1.5210e-01, -1.6603e-01,  4.3343e-02, -4.3308e-01, -4.3246e-01,\n",
       "          -4.6076e-02,  1.8719e-02, -1.8741e-01, -3.9340e-01,  6.6873e-01,\n",
       "           2.9672e-01, -3.1147e-01,  3.1526e-01,  2.5915e-01,  2.0141e-01,\n",
       "          -1.5187e-01,  1.4409e-01, -7.6357e-02,  3.8308e-01,  3.8767e-01,\n",
       "           7.4158e-02,  2.2964e-01, -2.2839e-01, -1.3054e-01,  1.2166e-01,\n",
       "           3.0174e-02,  3.2912e-02, -2.6636e-01,  5.1718e-02,  4.3690e-02,\n",
       "          -2.3695e-01,  4.5720e-02, -2.9789e-01,  2.4778e-01, -1.8620e-01,\n",
       "           4.8947e-02,  1.3339e-02,  6.9357e-02, -2.6009e-01, -3.5804e-01,\n",
       "          -4.8208e-01, -3.2343e-01,  4.1121e-01, -9.7802e-02, -2.6906e-01,\n",
       "          -8.0661e-02,  5.2934e-01, -7.8499e-02, -9.2217e-02,  4.6220e-01,\n",
       "           5.3264e-01, -7.7533e-02,  2.0590e-01, -2.3845e-01, -1.0281e-01,\n",
       "           5.3442e-01, -1.3821e-01, -2.9975e-01, -1.4752e-01,  1.7769e-01,\n",
       "           2.7041e-01, -1.5525e-01,  9.3563e-02, -1.5307e-01, -4.8001e-02,\n",
       "          -7.1748e-01,  6.3281e-01,  2.5560e-01, -3.1297e-01,  8.3186e-02,\n",
       "          -5.3646e-01, -1.8162e+00,  2.9583e-01, -8.9802e-03,  8.3883e-02,\n",
       "          -5.2191e-01,  3.3534e-01,  5.6150e-01,  1.9227e-01, -3.0995e-01,\n",
       "           5.6134e-02,  6.0282e-03,  2.6009e-01,  7.5598e-01, -3.0579e-01,\n",
       "           3.5314e-01,  5.4756e-01,  1.8529e-01,  1.0187e-01, -5.7044e-02,\n",
       "          -5.4487e-01,  1.0087e-01, -1.5430e-01,  7.5219e-02,  8.3591e-02,\n",
       "           1.9317e-01,  7.9363e-02, -2.4227e-02,  4.5380e-01,  1.7686e-01,\n",
       "          -7.4333e-02, -3.2795e-01, -3.6432e-01, -4.7966e-01, -4.5083e-02,\n",
       "           4.7412e-01, -1.1604e-01, -3.1275e-01,  4.4881e-01,  1.4713e-01,\n",
       "          -1.7880e-01,  1.0187e-01, -2.8504e-01, -1.5873e-01, -2.2084e-01,\n",
       "           4.1350e-01,  1.1884e-02,  3.8026e-02, -1.2162e-01, -2.9140e-01,\n",
       "           1.4320e-01, -3.8984e-01,  6.8835e-01,  9.8232e-02, -3.3128e-02,\n",
       "          -4.6651e-01, -3.3263e-01, -1.1748e-01, -5.5279e-02,  1.1751e-01,\n",
       "          -2.1558e-01, -4.4945e-01, -7.9196e-01, -2.1067e-01,  2.1134e-01,\n",
       "           2.0527e-01, -1.4779e-01,  7.6655e-01,  4.5252e-01,  2.0021e-01,\n",
       "          -1.6759e-02, -1.4259e-01,  3.8714e-01, -1.0633e-01, -1.5823e-01,\n",
       "           4.4625e-01,  1.0733e-01, -2.0841e-01,  2.7224e-01, -3.4130e-01,\n",
       "          -7.5523e-02, -2.1376e-01,  1.2725e-01,  3.8114e-01, -2.1832e-01,\n",
       "           2.6804e-01,  4.8200e-01, -4.2946e-01,  3.6150e-03,  6.9857e-02,\n",
       "          -2.4165e-01,  4.9451e-01,  2.8256e-01, -2.5111e-01, -7.1497e-01,\n",
       "          -3.9098e-01,  1.8738e-01, -8.5326e-02,  1.1588e-01, -7.5625e-01,\n",
       "          -1.9516e-02,  1.8532e-01, -5.0850e-02,  3.8660e-02, -1.6982e-01,\n",
       "           8.9387e-02,  1.5403e-01, -1.1319e-01,  1.1929e-01, -6.0965e-01,\n",
       "          -1.1577e-01,  8.1052e-02, -2.6227e-01,  3.9117e-01, -2.0472e-02,\n",
       "          -1.3146e-01, -7.8082e-02,  7.4697e-02,  1.3624e-01, -2.2372e-01,\n",
       "          -1.9221e-01, -1.9927e-01, -3.8998e-01, -1.7622e-01, -4.4131e-01,\n",
       "           2.0261e-02, -6.5790e-02,  4.1060e-01, -5.8363e-01, -7.0746e-01,\n",
       "          -4.0332e-02,  8.3122e-01,  1.5539e-01, -1.1409e-01, -2.9864e-01,\n",
       "          -1.2446e+00, -1.2758e-03, -1.1275e-01, -5.0157e-02, -3.7661e-01,\n",
       "           2.6208e-01, -4.4367e-01, -7.4600e-02,  6.0106e-02, -3.2547e-01,\n",
       "           4.6508e-01,  4.2918e-01, -9.5908e-02, -1.4445e-01, -2.2066e-01,\n",
       "          -1.6484e-01, -4.7839e-03, -1.0723e-02, -2.1120e-01, -8.7370e-02,\n",
       "           4.7734e-02,  2.2848e-01, -1.0237e-01,  9.7299e-03, -2.7817e-01,\n",
       "           4.6833e-01, -8.1631e-02,  2.3962e-01,  7.8523e-02, -1.6546e-01,\n",
       "          -1.4914e-01,  6.0792e-01, -4.8988e-01, -5.5599e-03, -2.8925e-01,\n",
       "          -1.7773e-02, -3.1432e-01, -2.4843e-01,  2.6644e-01,  6.2187e-01,\n",
       "           2.0591e-01,  2.5202e-01, -3.1135e-01,  6.1628e-01,  3.6461e-01,\n",
       "          -2.1338e-01,  4.0945e-01,  1.4651e-01, -1.6433e-01, -1.7420e-01,\n",
       "           1.2906e-01,  1.3202e-01, -5.1943e-01, -3.1207e-01,  4.5083e-02,\n",
       "           3.1362e-01, -5.4666e-01,  1.3951e-01,  2.8671e-01,  4.6681e-01,\n",
       "          -4.5926e-01, -3.2972e-01,  4.9980e-01, -4.3622e-01, -3.5347e-01,\n",
       "           7.8199e-02, -1.9888e-02,  9.9291e-02,  1.0423e-01, -1.7312e-01,\n",
       "          -3.7282e-01,  1.4114e-01,  8.3825e-01,  3.3371e-01, -1.1001e-01,\n",
       "           7.0772e-02, -2.4292e-01, -3.1195e-01,  1.6386e-01, -1.4070e-01,\n",
       "          -3.7851e-02,  2.1954e-01, -2.0071e-01,  1.7762e-01,  2.7308e-01,\n",
       "          -3.4342e-01, -1.7359e-01, -1.0640e-01, -4.0226e-01, -1.6770e-01,\n",
       "          -2.8967e-01,  2.2717e-01,  1.0074e-01, -3.2637e-02, -2.0053e-01,\n",
       "           2.7228e-01,  1.3624e-01, -4.3032e-01,  4.6722e-01,  1.1571e-01,\n",
       "           1.3363e-01, -7.6380e-02, -9.0105e-02, -1.7606e-01, -2.1443e-01,\n",
       "           7.4092e-01,  3.7268e-01,  3.4188e-01,  5.9585e-01, -1.1430e+01,\n",
       "          -3.2854e-01,  7.0622e-01, -5.0632e-01, -2.8414e-01,  1.7379e-02,\n",
       "           3.0317e-02, -1.3094e-01, -1.2223e-01, -1.7213e-01, -2.0848e-01,\n",
       "           3.8581e-02, -1.8417e-01, -1.5127e-02,  1.6480e-01, -3.4073e-01,\n",
       "          -2.7777e-01, -5.6942e-01,  3.7622e-01, -1.0005e-02,  4.3531e-01,\n",
       "          -1.6429e-01,  3.5614e-01,  1.9622e-01, -1.5602e-01, -3.3314e-02,\n",
       "          -2.0365e-02, -2.9865e-01,  3.1557e-01,  2.8572e-01, -9.2891e-02,\n",
       "           1.4331e-01,  2.7301e-01,  2.4403e-02, -2.2509e-01, -3.7899e-01,\n",
       "           3.2023e-02, -2.3634e-01,  5.6497e-02, -1.1343e-01, -3.9160e-01,\n",
       "          -3.5643e-01, -5.3921e-01,  3.7870e-01,  3.0318e-01,  9.2623e-01,\n",
       "           1.1131e+00,  3.8894e-01,  4.2962e-01,  2.0895e-02,  4.2812e-02,\n",
       "          -7.3146e-01,  2.7751e-01, -5.6560e-01,  6.9131e-02, -1.3600e-01,\n",
       "           5.0392e-01, -7.3013e-01,  4.8388e-02,  2.2433e-02, -3.9560e-01,\n",
       "           5.7490e-01,  1.0766e-01,  1.4034e-01, -1.6424e-01, -2.5920e-01,\n",
       "           6.4803e-01,  1.1666e-01,  1.7487e-01, -1.0874e-01,  1.6091e-01,\n",
       "          -3.9845e-01,  4.5722e-01,  2.5292e-01, -4.0623e-04,  1.0912e-01,\n",
       "           7.7109e-02, -8.3452e-02,  9.0745e-01, -4.4577e-02, -2.4422e-01,\n",
       "           1.2623e-01,  5.6754e-01, -3.3746e-01, -9.9758e-01, -9.3581e-02,\n",
       "           2.2376e-01,  4.1335e-02,  2.3630e-02,  5.5075e-01, -1.1704e-01,\n",
       "          -6.3826e-01, -2.6895e-01, -5.6567e-02,  2.9303e-01,  3.1220e-01,\n",
       "          -5.9542e-01, -1.2874e-01,  2.4996e-01,  5.7697e-02, -6.5912e-01,\n",
       "          -9.0239e-02, -1.8286e-01, -1.3062e-01,  2.9323e-01, -1.3426e-01,\n",
       "           1.1384e-02, -1.7566e-01,  1.4161e-01,  2.7246e-01, -4.5931e-01,\n",
       "           3.1196e-01, -1.4102e-02, -7.9098e-02, -5.6663e-02, -7.7202e-02,\n",
       "           2.1152e-01,  6.2387e-01, -1.6157e-01, -4.4824e-01, -2.0598e-01,\n",
       "          -8.3521e-02,  5.4951e-02, -6.6286e-02, -2.8166e-01,  5.6667e-01,\n",
       "          -3.6566e-01,  2.0546e-01,  3.9367e-01,  9.2395e-02,  3.1025e-01,\n",
       "           5.9569e-02,  3.4229e-01,  5.0153e-01, -7.4760e-01, -9.4850e-01,\n",
       "          -5.2820e-01,  2.6921e-01,  1.0290e-02, -2.7223e-01,  2.3821e-02,\n",
       "           1.5258e-01,  2.2258e-01, -2.5784e-01,  1.4523e-01,  3.1170e-01,\n",
       "           4.5654e-01,  6.1220e-01, -1.4783e-01, -2.1438e-01, -3.9943e-01,\n",
       "          -6.0836e-01, -1.8491e-01,  1.3476e-01,  1.6914e-01, -3.3037e-01,\n",
       "          -1.4396e-01,  1.1995e-02, -2.5094e-01,  1.9505e-01,  1.5258e-01,\n",
       "          -3.6240e-01, -2.3974e-01,  7.5298e-02, -1.7455e-01,  3.7739e-01,\n",
       "           2.8511e-01, -2.7377e-01, -1.7243e-01, -1.6147e-01,  3.5121e-01,\n",
       "           2.3123e-01,  1.5227e-02, -3.0671e-01, -9.3039e-02, -1.5211e-01,\n",
       "           1.5022e-01,  8.6500e-01, -2.0458e-01, -3.2117e-01,  2.8759e-01,\n",
       "           2.3440e-01, -7.9811e-02, -1.6529e-01, -1.6782e-01, -1.7306e-01,\n",
       "           7.7893e-03, -8.2331e-01,  5.0536e-01,  1.2359e-01,  4.7159e-01,\n",
       "           2.7540e-01,  6.8203e-02, -1.2741e-01,  3.6222e-01, -4.0652e-01,\n",
       "          -3.7754e-01, -1.2237e-01, -4.5658e-01,  4.1031e-02, -1.6036e-01,\n",
       "          -1.5228e-01,  7.5494e-02, -1.4731e-01, -1.5987e-01,  3.5461e-01,\n",
       "          -1.4282e-01,  1.1264e+00,  1.9878e-01, -2.6831e-01,  4.8278e-01,\n",
       "           4.2749e-01, -2.9024e-01,  9.0734e-02,  1.5853e-01,  2.5162e-01,\n",
       "           3.8973e-01, -1.0037e-01, -3.3038e-02,  2.1620e-01,  1.9310e-01,\n",
       "           6.9697e-01, -8.1216e-01, -4.0955e-01, -9.4589e-03,  2.8265e-01,\n",
       "          -1.0637e-01,  2.2458e-01, -8.9556e-02,  2.3982e-01,  6.4468e-01,\n",
       "          -1.2239e+00, -2.4918e-01, -3.3600e-01,  9.0556e-01,  6.8938e-02,\n",
       "          -1.2904e-01, -5.2787e-02, -6.6009e-01, -3.8467e-01,  2.0051e-01,\n",
       "           5.6018e-01, -1.1897e+00,  1.8207e-01,  5.1902e-01, -3.0900e-01,\n",
       "           1.3867e-01, -6.1281e-01,  3.8051e-01, -1.0249e-01,  4.5109e-01,\n",
       "           1.2145e-01,  3.8773e-01, -2.7366e-01, -5.6277e-01,  1.5809e-01,\n",
       "           4.7588e-01,  1.1298e-01, -9.5715e-02,  6.8718e-02,  7.3584e-01,\n",
       "          -1.9387e-01, -5.5478e-01, -2.9804e-01,  8.9795e-02, -5.3699e-02,\n",
       "           1.4660e-01, -2.7834e-01,  1.4446e-02,  1.8556e-01, -2.7090e-02,\n",
       "           4.4350e-01,  7.5971e-01,  7.8064e-01, -2.9624e-01, -3.3911e-01,\n",
       "           4.3568e-01,  2.0869e-01, -1.1570e-01,  4.9662e-01, -1.5444e-01,\n",
       "          -4.9718e-01,  7.1934e-01,  1.5821e-01, -1.3144e-01,  1.8528e-01,\n",
       "          -1.9851e-01,  1.2533e-01,  1.9548e-01, -2.5137e-01,  5.9767e-02,\n",
       "          -4.1842e-02,  2.2554e-02, -2.1785e-01,  3.7487e-02,  2.5822e-01,\n",
       "          -5.3973e-01,  8.7751e-01, -3.2580e-01,  6.7522e-01, -5.3727e-01,\n",
       "           2.9079e-01, -6.8812e-01, -5.0529e-01, -1.7472e-01, -4.3135e-01,\n",
       "          -2.9410e-01, -1.5732e-01,  9.2664e-02,  2.0313e-01,  6.3815e-01,\n",
       "           1.3223e-01, -2.2851e-01, -5.4095e-01, -4.2331e-01,  2.9837e-01,\n",
       "           4.7109e-02, -1.0500e-01,  6.3951e-02,  2.3236e-01,  1.1879e-01,\n",
       "          -7.5560e-01, -6.7449e-01, -8.7288e-03,  2.8159e-01, -2.6361e-01,\n",
       "           8.5595e-01, -7.4605e-02, -1.2331e-01,  8.7453e-02,  1.0628e-01,\n",
       "           3.7378e-01,  6.2921e-01,  9.7905e-02, -3.7329e-01, -8.9811e-02,\n",
       "          -6.4163e-02, -3.8641e-01,  3.8222e-02, -5.3569e-01,  5.6306e-01,\n",
       "           3.7212e-01,  3.5028e-01,  5.7574e-01]], device='mps:0'),\n",
       " tensor([[-8.7564e-02, -4.9833e-01, -6.3687e-01, -4.0578e-01, -6.9840e-01,\n",
       "          -2.3387e-01, -4.1180e-01,  5.2317e-01, -2.2035e-01,  4.5067e-01,\n",
       "           2.5062e-01,  1.1995e-01,  1.5031e-01, -9.6288e-02, -6.0084e-01,\n",
       "          -3.4361e-01,  3.9723e-01,  3.9226e-01,  6.5111e-01,  1.1035e-01,\n",
       "           6.7427e-01,  3.9429e-01, -1.4784e-01, -1.1417e-01, -8.9684e-01,\n",
       "           2.6734e-01, -1.5040e-01,  3.5751e-02,  3.8249e-01, -6.5691e-01,\n",
       "          -5.0431e-01,  1.2997e-01, -9.6623e-02,  7.5944e-01, -3.4214e-01,\n",
       "          -4.9353e-03,  1.6918e-01, -5.5750e-01,  1.5409e-01,  3.5742e-02,\n",
       "           5.0410e-01,  2.7796e-02,  8.9746e-02,  4.2259e-01,  5.2340e-01,\n",
       "          -4.3294e-03, -9.1370e-02, -7.8080e-01,  1.3439e-01, -2.8322e-01,\n",
       "           2.7167e-01,  2.8308e-02,  4.6509e-01,  4.3443e-01, -6.3205e-01,\n",
       "          -7.4689e-01,  3.7258e-01,  1.9066e-01, -4.4268e-01,  3.4560e-01,\n",
       "          -2.7884e-01,  7.1737e-01,  1.3666e-01,  4.6901e-02, -1.7654e-01,\n",
       "           3.4580e-02, -1.9115e-01, -9.7322e-01, -2.5911e-01, -2.2553e-01,\n",
       "          -2.1928e-01,  2.6272e-01, -2.6864e-01, -1.8710e-01,  1.7180e-01,\n",
       "          -2.5149e-01,  2.3745e-01, -5.6538e-02, -3.8011e-01, -1.4267e-01,\n",
       "           3.4435e-01,  1.6998e-01,  8.6814e-02, -4.4090e-02, -4.8772e-01,\n",
       "           3.1845e-02,  5.0133e-01,  4.0723e-01,  5.2782e-01, -1.2693e-01,\n",
       "           2.5539e-01,  1.3344e-01, -1.3033e-01, -3.9779e-01,  4.8072e-01,\n",
       "          -4.2228e-01,  3.0926e-01, -1.4101e-01,  9.2363e-02, -4.1453e-01,\n",
       "           8.9056e-02,  3.0370e-01, -9.2494e-02,  2.4348e-01, -4.8023e-01,\n",
       "          -7.0432e-01,  1.1504e-01, -1.5078e-01, -9.5519e-01, -1.0935e-01,\n",
       "          -1.5210e-01, -1.6603e-01,  4.3343e-02, -4.3308e-01, -4.3246e-01,\n",
       "          -4.6076e-02,  1.8719e-02, -1.8741e-01, -3.9340e-01,  6.6873e-01,\n",
       "           2.9672e-01, -3.1147e-01,  3.1526e-01,  2.5915e-01,  2.0141e-01,\n",
       "          -1.5187e-01,  1.4409e-01, -7.6357e-02,  3.8308e-01,  3.8767e-01,\n",
       "           7.4158e-02,  2.2964e-01, -2.2839e-01, -1.3054e-01,  1.2166e-01,\n",
       "           3.0174e-02,  3.2912e-02, -2.6636e-01,  5.1718e-02,  4.3690e-02,\n",
       "          -2.3695e-01,  4.5720e-02, -2.9789e-01,  2.4778e-01, -1.8620e-01,\n",
       "           4.8947e-02,  1.3339e-02,  6.9357e-02, -2.6009e-01, -3.5804e-01,\n",
       "          -4.8208e-01, -3.2343e-01,  4.1121e-01, -9.7802e-02, -2.6906e-01,\n",
       "          -8.0661e-02,  5.2934e-01, -7.8499e-02, -9.2217e-02,  4.6220e-01,\n",
       "           5.3264e-01, -7.7533e-02,  2.0590e-01, -2.3845e-01, -1.0281e-01,\n",
       "           5.3442e-01, -1.3821e-01, -2.9975e-01, -1.4752e-01,  1.7769e-01,\n",
       "           2.7041e-01, -1.5525e-01,  9.3563e-02, -1.5307e-01, -4.8001e-02,\n",
       "          -7.1748e-01,  6.3281e-01,  2.5560e-01, -3.1297e-01,  8.3186e-02,\n",
       "          -5.3646e-01, -1.8162e+00,  2.9583e-01, -8.9802e-03,  8.3883e-02,\n",
       "          -5.2191e-01,  3.3534e-01,  5.6150e-01,  1.9227e-01, -3.0995e-01,\n",
       "           5.6134e-02,  6.0282e-03,  2.6009e-01,  7.5598e-01, -3.0579e-01,\n",
       "           3.5314e-01,  5.4756e-01,  1.8529e-01,  1.0187e-01, -5.7044e-02,\n",
       "          -5.4487e-01,  1.0087e-01, -1.5430e-01,  7.5219e-02,  8.3591e-02,\n",
       "           1.9317e-01,  7.9363e-02, -2.4227e-02,  4.5380e-01,  1.7686e-01,\n",
       "          -7.4333e-02, -3.2795e-01, -3.6432e-01, -4.7966e-01, -4.5083e-02,\n",
       "           4.7412e-01, -1.1604e-01, -3.1275e-01,  4.4881e-01,  1.4713e-01,\n",
       "          -1.7880e-01,  1.0187e-01, -2.8504e-01, -1.5873e-01, -2.2084e-01,\n",
       "           4.1350e-01,  1.1884e-02,  3.8026e-02, -1.2162e-01, -2.9140e-01,\n",
       "           1.4320e-01, -3.8984e-01,  6.8835e-01,  9.8232e-02, -3.3128e-02,\n",
       "          -4.6651e-01, -3.3263e-01, -1.1748e-01, -5.5279e-02,  1.1751e-01,\n",
       "          -2.1558e-01, -4.4945e-01, -7.9196e-01, -2.1067e-01,  2.1134e-01,\n",
       "           2.0527e-01, -1.4779e-01,  7.6655e-01,  4.5252e-01,  2.0021e-01,\n",
       "          -1.6759e-02, -1.4259e-01,  3.8714e-01, -1.0633e-01, -1.5823e-01,\n",
       "           4.4625e-01,  1.0733e-01, -2.0841e-01,  2.7224e-01, -3.4130e-01,\n",
       "          -7.5523e-02, -2.1376e-01,  1.2725e-01,  3.8114e-01, -2.1832e-01,\n",
       "           2.6804e-01,  4.8200e-01, -4.2946e-01,  3.6150e-03,  6.9857e-02,\n",
       "          -2.4165e-01,  4.9451e-01,  2.8256e-01, -2.5111e-01, -7.1497e-01,\n",
       "          -3.9098e-01,  1.8738e-01, -8.5326e-02,  1.1588e-01, -7.5625e-01,\n",
       "          -1.9516e-02,  1.8532e-01, -5.0850e-02,  3.8660e-02, -1.6982e-01,\n",
       "           8.9387e-02,  1.5403e-01, -1.1319e-01,  1.1929e-01, -6.0965e-01,\n",
       "          -1.1577e-01,  8.1052e-02, -2.6227e-01,  3.9117e-01, -2.0472e-02,\n",
       "          -1.3146e-01, -7.8082e-02,  7.4697e-02,  1.3624e-01, -2.2372e-01,\n",
       "          -1.9221e-01, -1.9927e-01, -3.8998e-01, -1.7622e-01, -4.4131e-01,\n",
       "           2.0261e-02, -6.5790e-02,  4.1060e-01, -5.8363e-01, -7.0746e-01,\n",
       "          -4.0332e-02,  8.3122e-01,  1.5539e-01, -1.1409e-01, -2.9864e-01,\n",
       "          -1.2446e+00, -1.2758e-03, -1.1275e-01, -5.0157e-02, -3.7661e-01,\n",
       "           2.6208e-01, -4.4367e-01, -7.4600e-02,  6.0106e-02, -3.2547e-01,\n",
       "           4.6508e-01,  4.2918e-01, -9.5908e-02, -1.4445e-01, -2.2066e-01,\n",
       "          -1.6484e-01, -4.7839e-03, -1.0723e-02, -2.1120e-01, -8.7370e-02,\n",
       "           4.7734e-02,  2.2848e-01, -1.0237e-01,  9.7299e-03, -2.7817e-01,\n",
       "           4.6833e-01, -8.1631e-02,  2.3962e-01,  7.8523e-02, -1.6546e-01,\n",
       "          -1.4914e-01,  6.0792e-01, -4.8988e-01, -5.5599e-03, -2.8925e-01,\n",
       "          -1.7773e-02, -3.1432e-01, -2.4843e-01,  2.6644e-01,  6.2187e-01,\n",
       "           2.0591e-01,  2.5202e-01, -3.1135e-01,  6.1628e-01,  3.6461e-01,\n",
       "          -2.1338e-01,  4.0945e-01,  1.4651e-01, -1.6433e-01, -1.7420e-01,\n",
       "           1.2906e-01,  1.3202e-01, -5.1943e-01, -3.1207e-01,  4.5083e-02,\n",
       "           3.1362e-01, -5.4666e-01,  1.3951e-01,  2.8671e-01,  4.6681e-01,\n",
       "          -4.5926e-01, -3.2972e-01,  4.9980e-01, -4.3622e-01, -3.5347e-01,\n",
       "           7.8199e-02, -1.9888e-02,  9.9291e-02,  1.0423e-01, -1.7312e-01,\n",
       "          -3.7282e-01,  1.4114e-01,  8.3825e-01,  3.3371e-01, -1.1001e-01,\n",
       "           7.0772e-02, -2.4292e-01, -3.1195e-01,  1.6386e-01, -1.4070e-01,\n",
       "          -3.7851e-02,  2.1954e-01, -2.0071e-01,  1.7762e-01,  2.7308e-01,\n",
       "          -3.4342e-01, -1.7359e-01, -1.0640e-01, -4.0226e-01, -1.6770e-01,\n",
       "          -2.8967e-01,  2.2717e-01,  1.0074e-01, -3.2637e-02, -2.0053e-01,\n",
       "           2.7228e-01,  1.3624e-01, -4.3032e-01,  4.6722e-01,  1.1571e-01,\n",
       "           1.3363e-01, -7.6380e-02, -9.0105e-02, -1.7606e-01, -2.1443e-01,\n",
       "           7.4092e-01,  3.7268e-01,  3.4188e-01,  5.9585e-01, -1.1430e+01,\n",
       "          -3.2854e-01,  7.0622e-01, -5.0632e-01, -2.8414e-01,  1.7379e-02,\n",
       "           3.0317e-02, -1.3094e-01, -1.2223e-01, -1.7213e-01, -2.0848e-01,\n",
       "           3.8581e-02, -1.8417e-01, -1.5127e-02,  1.6480e-01, -3.4073e-01,\n",
       "          -2.7777e-01, -5.6942e-01,  3.7622e-01, -1.0005e-02,  4.3531e-01,\n",
       "          -1.6429e-01,  3.5614e-01,  1.9622e-01, -1.5602e-01, -3.3314e-02,\n",
       "          -2.0365e-02, -2.9865e-01,  3.1557e-01,  2.8572e-01, -9.2891e-02,\n",
       "           1.4331e-01,  2.7301e-01,  2.4403e-02, -2.2509e-01, -3.7899e-01,\n",
       "           3.2023e-02, -2.3634e-01,  5.6497e-02, -1.1343e-01, -3.9160e-01,\n",
       "          -3.5643e-01, -5.3921e-01,  3.7870e-01,  3.0318e-01,  9.2623e-01,\n",
       "           1.1131e+00,  3.8894e-01,  4.2962e-01,  2.0895e-02,  4.2812e-02,\n",
       "          -7.3146e-01,  2.7751e-01, -5.6560e-01,  6.9131e-02, -1.3600e-01,\n",
       "           5.0392e-01, -7.3013e-01,  4.8388e-02,  2.2433e-02, -3.9560e-01,\n",
       "           5.7490e-01,  1.0766e-01,  1.4034e-01, -1.6424e-01, -2.5920e-01,\n",
       "           6.4803e-01,  1.1666e-01,  1.7487e-01, -1.0874e-01,  1.6091e-01,\n",
       "          -3.9845e-01,  4.5722e-01,  2.5292e-01, -4.0623e-04,  1.0912e-01,\n",
       "           7.7109e-02, -8.3452e-02,  9.0745e-01, -4.4577e-02, -2.4422e-01,\n",
       "           1.2623e-01,  5.6754e-01, -3.3746e-01, -9.9758e-01, -9.3581e-02,\n",
       "           2.2376e-01,  4.1335e-02,  2.3630e-02,  5.5075e-01, -1.1704e-01,\n",
       "          -6.3826e-01, -2.6895e-01, -5.6567e-02,  2.9303e-01,  3.1220e-01,\n",
       "          -5.9542e-01, -1.2874e-01,  2.4996e-01,  5.7697e-02, -6.5912e-01,\n",
       "          -9.0239e-02, -1.8286e-01, -1.3062e-01,  2.9323e-01, -1.3426e-01,\n",
       "           1.1384e-02, -1.7566e-01,  1.4161e-01,  2.7246e-01, -4.5931e-01,\n",
       "           3.1196e-01, -1.4102e-02, -7.9098e-02, -5.6663e-02, -7.7202e-02,\n",
       "           2.1152e-01,  6.2387e-01, -1.6157e-01, -4.4824e-01, -2.0598e-01,\n",
       "          -8.3521e-02,  5.4951e-02, -6.6286e-02, -2.8166e-01,  5.6667e-01,\n",
       "          -3.6566e-01,  2.0546e-01,  3.9367e-01,  9.2395e-02,  3.1025e-01,\n",
       "           5.9569e-02,  3.4229e-01,  5.0153e-01, -7.4760e-01, -9.4850e-01,\n",
       "          -5.2820e-01,  2.6921e-01,  1.0290e-02, -2.7223e-01,  2.3821e-02,\n",
       "           1.5258e-01,  2.2258e-01, -2.5784e-01,  1.4523e-01,  3.1170e-01,\n",
       "           4.5654e-01,  6.1220e-01, -1.4783e-01, -2.1438e-01, -3.9943e-01,\n",
       "          -6.0836e-01, -1.8491e-01,  1.3476e-01,  1.6914e-01, -3.3037e-01,\n",
       "          -1.4396e-01,  1.1995e-02, -2.5094e-01,  1.9505e-01,  1.5258e-01,\n",
       "          -3.6240e-01, -2.3974e-01,  7.5298e-02, -1.7455e-01,  3.7739e-01,\n",
       "           2.8511e-01, -2.7377e-01, -1.7243e-01, -1.6147e-01,  3.5121e-01,\n",
       "           2.3123e-01,  1.5227e-02, -3.0671e-01, -9.3039e-02, -1.5211e-01,\n",
       "           1.5022e-01,  8.6500e-01, -2.0458e-01, -3.2117e-01,  2.8759e-01,\n",
       "           2.3440e-01, -7.9811e-02, -1.6529e-01, -1.6782e-01, -1.7306e-01,\n",
       "           7.7893e-03, -8.2331e-01,  5.0536e-01,  1.2359e-01,  4.7159e-01,\n",
       "           2.7540e-01,  6.8203e-02, -1.2741e-01,  3.6222e-01, -4.0652e-01,\n",
       "          -3.7754e-01, -1.2237e-01, -4.5658e-01,  4.1031e-02, -1.6036e-01,\n",
       "          -1.5228e-01,  7.5494e-02, -1.4731e-01, -1.5987e-01,  3.5461e-01,\n",
       "          -1.4282e-01,  1.1264e+00,  1.9878e-01, -2.6831e-01,  4.8278e-01,\n",
       "           4.2749e-01, -2.9024e-01,  9.0734e-02,  1.5853e-01,  2.5162e-01,\n",
       "           3.8973e-01, -1.0037e-01, -3.3038e-02,  2.1620e-01,  1.9310e-01,\n",
       "           6.9697e-01, -8.1216e-01, -4.0955e-01, -9.4589e-03,  2.8265e-01,\n",
       "          -1.0637e-01,  2.2458e-01, -8.9556e-02,  2.3982e-01,  6.4468e-01,\n",
       "          -1.2239e+00, -2.4918e-01, -3.3600e-01,  9.0556e-01,  6.8938e-02,\n",
       "          -1.2904e-01, -5.2787e-02, -6.6009e-01, -3.8467e-01,  2.0051e-01,\n",
       "           5.6018e-01, -1.1897e+00,  1.8207e-01,  5.1902e-01, -3.0900e-01,\n",
       "           1.3867e-01, -6.1281e-01,  3.8051e-01, -1.0249e-01,  4.5109e-01,\n",
       "           1.2145e-01,  3.8773e-01, -2.7366e-01, -5.6277e-01,  1.5809e-01,\n",
       "           4.7588e-01,  1.1298e-01, -9.5715e-02,  6.8718e-02,  7.3584e-01,\n",
       "          -1.9387e-01, -5.5478e-01, -2.9804e-01,  8.9795e-02, -5.3699e-02,\n",
       "           1.4660e-01, -2.7834e-01,  1.4446e-02,  1.8556e-01, -2.7090e-02,\n",
       "           4.4350e-01,  7.5971e-01,  7.8064e-01, -2.9624e-01, -3.3911e-01,\n",
       "           4.3568e-01,  2.0869e-01, -1.1570e-01,  4.9662e-01, -1.5444e-01,\n",
       "          -4.9718e-01,  7.1934e-01,  1.5821e-01, -1.3144e-01,  1.8528e-01,\n",
       "          -1.9851e-01,  1.2533e-01,  1.9548e-01, -2.5137e-01,  5.9767e-02,\n",
       "          -4.1842e-02,  2.2554e-02, -2.1785e-01,  3.7487e-02,  2.5822e-01,\n",
       "          -5.3973e-01,  8.7751e-01, -3.2580e-01,  6.7522e-01, -5.3727e-01,\n",
       "           2.9079e-01, -6.8812e-01, -5.0529e-01, -1.7472e-01, -4.3135e-01,\n",
       "          -2.9410e-01, -1.5732e-01,  9.2664e-02,  2.0313e-01,  6.3815e-01,\n",
       "           1.3223e-01, -2.2851e-01, -5.4095e-01, -4.2331e-01,  2.9837e-01,\n",
       "           4.7109e-02, -1.0500e-01,  6.3951e-02,  2.3236e-01,  1.1879e-01,\n",
       "          -7.5560e-01, -6.7449e-01, -8.7288e-03,  2.8159e-01, -2.6361e-01,\n",
       "           8.5595e-01, -7.4605e-02, -1.2331e-01,  8.7453e-02,  1.0628e-01,\n",
       "           3.7378e-01,  6.2921e-01,  9.7905e-02, -3.7329e-01, -8.9811e-02,\n",
       "          -6.4163e-02, -3.8641e-01,  3.8222e-02, -5.3569e-01,  5.6306e-01,\n",
       "           3.7212e-01,  3.5028e-01,  5.7574e-01]], device='mps:0'),\n",
       " tensor([[-0.0441, -0.5132, -0.4175,  ...,  0.2929,  0.4426,  0.8593],\n",
       "         [-0.1979, -0.6462, -0.8826,  ...,  0.1613,  1.0377,  0.5252],\n",
       "         [-0.1390, -0.8191, -0.6195,  ...,  0.2847,  0.9152,  0.4824],\n",
       "         [-0.5904, -0.3106, -0.3112,  ..., -0.0533,  0.5494,  0.2181]],\n",
       "        device='mps:0'),\n",
       " tensor([[-2.9010e-01, -6.1957e-01, -5.3252e-01, -6.8058e-01, -3.8540e-01,\n",
       "          -2.5920e-01, -5.6468e-01,  4.7697e-01,  5.5986e-02,  4.8572e-01,\n",
       "           1.4230e-01,  4.2920e-01,  7.5656e-02, -5.3480e-01, -7.8066e-01,\n",
       "          -1.7367e-01, -4.2145e-02,  4.7396e-01,  4.9203e-01,  4.3783e-01,\n",
       "           4.9367e-01,  2.5580e-01,  4.9838e-02, -6.3360e-02, -7.2440e-02,\n",
       "           2.9724e-01, -1.2900e-01, -1.0538e-01,  2.7006e-01, -4.9755e-01,\n",
       "           1.2963e-01,  3.0458e-01, -2.4084e-01,  9.3265e-01, -1.3130e-01,\n",
       "          -2.7353e-01,  2.2170e-01, -3.1771e-01,  6.7320e-01, -8.3518e-02,\n",
       "           5.6291e-01,  7.0301e-02,  2.4665e-01,  2.1534e-01,  1.3720e-01,\n",
       "          -9.5776e-02,  3.8511e-01, -5.9836e-01,  5.6981e-01, -3.7355e-01,\n",
       "          -4.9664e-01, -7.8985e-02,  4.6334e-01,  3.7650e-01, -6.3224e-01,\n",
       "          -3.2435e-01,  1.2941e-01,  2.7905e-01, -7.9634e-01,  2.5356e-01,\n",
       "          -4.3589e-01,  6.5367e-01,  5.2369e-01,  4.4911e-02,  3.5073e-02,\n",
       "          -3.6499e-02, -5.0215e-01, -8.3866e-03, -2.7929e-01, -2.3914e-03,\n",
       "           1.8180e-01,  3.7017e-01,  2.3163e-01,  4.7902e-02,  1.0601e-01,\n",
       "          -2.4618e-01,  3.1890e-01, -6.7766e-02, -2.3527e-01, -2.7011e-01,\n",
       "          -3.0667e-01,  4.3615e-01, -2.2818e-01, -4.3849e-02, -7.7917e-02,\n",
       "          -1.1039e-01,  3.1709e-01,  1.8746e-01,  1.7004e-01, -1.0816e-01,\n",
       "           2.8932e-01,  6.8871e-02, -5.3485e-01, -3.4023e-02,  2.0021e-01,\n",
       "          -1.5600e-01,  4.4421e-01, -1.4836e-01, -1.7977e-01, -3.8913e-01,\n",
       "           9.0064e-03,  1.3680e-01,  3.3588e-03,  5.6446e-01, -6.3781e-01,\n",
       "          -5.5530e-01,  2.4531e-01, -1.3051e-02, -7.2014e-01, -2.8969e-01,\n",
       "          -2.8808e-01, -2.0507e-01,  3.4981e-01, -4.9070e-01, -2.0738e-01,\n",
       "           1.9217e-01, -3.2868e-01, -2.7286e-01, -2.2569e-01,  2.3617e-01,\n",
       "           1.0002e-01, -5.4420e-01,  2.2756e-01, -3.1355e-01,  1.6779e-01,\n",
       "          -1.9251e-01,  3.9798e-01,  8.4412e-02,  1.0834e-01,  2.4657e-01,\n",
       "          -1.9617e-02,  2.9028e-01, -3.9095e-01, -5.8515e-02, -5.8318e-03,\n",
       "           3.5136e-01, -7.7349e-02, -3.4979e-01,  8.8003e-02, -2.5363e-01,\n",
       "          -6.7111e-02, -9.5711e-03, -6.7933e-02,  6.0075e-02,  2.3497e-01,\n",
       "          -3.9338e-01,  2.8650e-01, -3.7323e-01, -4.7569e-01, -2.5373e-01,\n",
       "          -2.8909e-01, -5.7194e-01,  1.6154e-01, -1.7777e-01,  1.6431e-01,\n",
       "          -4.3530e-02, -9.0980e-02, -8.2733e-02, -7.9678e-02,  3.5044e-01,\n",
       "           5.7532e-01,  5.2204e-02,  1.9072e-01, -3.4752e-01,  8.5055e-02,\n",
       "           1.9740e-02, -3.8324e-01, -2.8255e-01, -3.0150e-01, -1.9923e-02,\n",
       "           4.6514e-02, -8.2893e-02,  1.2876e-01, -7.2722e-01,  1.0956e-02,\n",
       "          -8.0064e-01,  7.2770e-02,  2.1490e-01,  2.4822e-01, -2.5903e-02,\n",
       "           1.4095e-01, -6.9657e-01,  9.3281e-02, -2.6073e-01, -1.0677e-02,\n",
       "          -4.8298e-01,  1.8196e-01, -8.0584e-02,  1.1325e-01, -3.2090e-03,\n",
       "          -1.4917e-02,  1.9273e-01,  1.5670e-01,  3.7036e-01,  4.6552e-02,\n",
       "           2.8188e-01,  7.9062e-01,  2.3313e-01, -1.6198e-01,  2.0879e-01,\n",
       "          -6.7540e-01, -6.3993e-03, -1.2308e-01,  3.5417e-01,  2.6572e-01,\n",
       "           1.9922e-01, -2.4609e-01,  1.6613e-01, -9.5365e-02, -1.3182e-01,\n",
       "          -3.7934e-02, -2.0546e-02, -5.6228e-01,  1.5114e-02, -6.7662e-03,\n",
       "           3.0533e-01, -3.8752e-01, -2.0671e-01,  5.4223e-01,  2.8816e-01,\n",
       "          -2.3952e-01, -1.8250e-01,  1.3161e-01, -4.1205e-01, -2.4551e-01,\n",
       "           2.9597e-01, -2.1472e-01,  1.2703e-01, -2.7597e-01, -2.8534e-01,\n",
       "           2.6444e-01, -8.2785e-01,  4.0556e-01,  1.9104e-01, -1.5415e-01,\n",
       "          -1.3444e-01, -3.7506e-01,  4.0178e-01, -4.0546e-01,  2.7252e-02,\n",
       "          -4.0308e-01, -4.2405e-01, -4.7496e-01, -2.6540e-01,  1.6277e-01,\n",
       "           2.5917e-01,  2.7457e-03,  9.6960e-01,  3.8031e-01,  3.5530e-01,\n",
       "           7.5266e-02, -3.0331e-01,  3.6905e-01,  6.3150e-02, -2.1985e-01,\n",
       "          -5.1483e-02, -3.6905e-02, -1.0317e-02,  1.6402e-01, -1.2728e-01,\n",
       "          -2.6453e-01,  2.8104e-02,  4.1058e-01,  5.3833e-01, -1.8131e-01,\n",
       "           4.0797e-01, -1.1053e-01, -8.3978e-02, -1.1285e-01,  1.7238e-01,\n",
       "          -5.9035e-01,  2.6747e-01,  1.4732e-01, -4.7405e-01, -8.0684e-01,\n",
       "          -3.3377e-01, -1.1058e-01, -1.4831e-01,  6.3654e-02, -7.1810e-01,\n",
       "          -2.9669e-01, -2.0977e-01, -2.0191e-01, -7.4206e-02, -8.1818e-02,\n",
       "           1.1084e-01, -3.3327e-02, -1.1844e-01, -1.9429e-01, -4.7331e-01,\n",
       "          -1.8902e-01,  1.1900e-01, -2.6122e-01,  4.5223e-02,  2.1944e-01,\n",
       "          -6.4984e-01,  2.2420e-01, -1.8365e-01,  2.2561e-01, -3.0575e-01,\n",
       "           1.2504e-02, -5.2143e-02, -1.7364e-01,  6.8545e-03,  1.3588e-01,\n",
       "           2.1228e-01,  3.0848e-01,  6.3187e-01, -2.5032e-01, -8.6856e-01,\n",
       "          -6.4641e-02,  4.1439e-01, -3.5311e-01, -1.1088e-01, -7.6554e-02,\n",
       "          -1.1242e+00, -1.5927e-01, -4.1754e-01, -3.1629e-01, -4.2671e-01,\n",
       "           1.0607e-01,  6.6074e-02, -1.7105e-02, -1.1384e-01, -3.0160e-01,\n",
       "           2.8167e-01,  2.5948e-01, -2.2689e-01,  1.7803e-01, -4.8608e-01,\n",
       "           1.2111e-01, -2.2373e-01, -4.2205e-01, -3.5975e-01, -7.7795e-02,\n",
       "          -1.0124e-01,  1.3847e-01,  2.2982e-01, -7.1403e-02, -4.8133e-03,\n",
       "           6.6604e-01, -1.9740e-01, -4.3671e-01,  3.7247e-01, -1.7236e-01,\n",
       "          -1.5817e-01,  4.1080e-01, -3.1071e-01,  4.4119e-02,  4.6548e-01,\n",
       "           2.7215e-01, -2.9492e-01, -3.4096e-01, -2.3799e-01,  3.1392e-01,\n",
       "           6.5454e-01,  5.2817e-01, -2.1552e-01,  3.5172e-01,  2.3129e-01,\n",
       "          -5.2431e-01,  3.3714e-01, -4.9455e-01, -2.7535e-01,  1.4021e-01,\n",
       "           1.8712e-01,  3.4267e-01, -3.5445e-01, -3.1550e-02, -1.4270e-02,\n",
       "           8.5163e-02, -1.1288e-01,  1.9061e-01,  4.1144e-01,  1.4486e-01,\n",
       "           1.0596e-01, -4.5375e-01,  4.2614e-01,  2.1837e-01, -2.8284e-01,\n",
       "          -1.0150e-02, -1.0218e-01, -6.5806e-02, -3.2101e-01,  1.1850e-01,\n",
       "          -1.3360e-01,  1.4006e-01,  8.5114e-01, -1.1626e-02, -1.0235e-01,\n",
       "          -1.2712e-01, -2.6088e-01, -2.1290e-01,  4.9592e-01, -1.4744e-01,\n",
       "          -3.9086e-01,  3.0225e-01, -6.6896e-02,  8.6624e-02,  5.7996e-01,\n",
       "          -3.6061e-01,  7.8320e-02,  2.0648e-01,  2.7312e-01, -2.0361e-02,\n",
       "          -3.7402e-01,  2.7560e-01,  3.8222e-01, -4.3437e-02,  1.6337e-01,\n",
       "          -3.7014e-01,  2.2490e-01,  4.1065e-02, -5.7577e-02, -4.5775e-02,\n",
       "           4.4360e-02, -2.4948e-01,  4.0893e-01,  2.9017e-02, -3.0709e-01,\n",
       "           7.6785e-01,  3.7996e-01,  7.0028e-01,  4.5193e-01, -1.1764e+01,\n",
       "           5.9830e-01,  5.1803e-01, -4.2747e-01,  1.3230e-01,  9.4847e-02,\n",
       "           4.1827e-01, -4.1758e-01,  2.5423e-01, -9.8194e-02, -4.1669e-02,\n",
       "          -3.3154e-01,  4.2399e-02, -8.9367e-02, -2.6707e-01, -1.6104e-01,\n",
       "           1.1012e-01, -6.6198e-01, -9.5877e-03, -1.2330e-01,  5.7198e-01,\n",
       "          -2.3666e-01, -5.2115e-02,  3.4532e-01, -1.5749e-01, -4.0187e-01,\n",
       "           3.2795e-01,  5.6605e-02,  2.2294e-01,  1.6879e-01, -5.5681e-02,\n",
       "           1.8155e-01, -1.6787e-01,  2.8771e-01, -1.6991e-01, -4.7845e-01,\n",
       "          -2.3854e-02, -6.1838e-02,  3.3673e-01, -3.5982e-01, -2.0412e-01,\n",
       "           3.3125e-02, -4.6054e-01,  4.6159e-01,  4.5703e-01,  2.3605e-01,\n",
       "           6.5223e-01,  7.8115e-01,  1.3101e-01, -1.1651e-01, -2.1335e-01,\n",
       "          -6.5353e-01,  4.5344e-01, -6.1024e-01, -1.6712e-01,  9.0306e-02,\n",
       "           6.2439e-01, -7.4442e-01,  2.6844e-01, -4.0254e-01, -2.2013e-01,\n",
       "           2.6627e-02, -1.2812e-01, -1.9330e-01, -3.1340e-01, -3.8544e-02,\n",
       "           3.0440e-01, -2.7250e-01,  7.8314e-02, -3.0328e-01,  5.3241e-01,\n",
       "          -1.5585e-01,  3.2214e-01,  1.0980e-01,  1.5839e-03,  1.8571e-01,\n",
       "          -1.1503e-01, -2.7215e-01,  2.1883e+00, -1.9780e-01,  1.2181e-01,\n",
       "           2.8113e-01, -5.2273e-02, -4.0506e-01, -4.9381e-01, -1.2224e-01,\n",
       "           1.7626e-02,  1.0639e-01,  5.4420e-01,  4.4815e-01, -9.7375e-02,\n",
       "          -2.5646e-01, -9.9784e-02, -9.6022e-02,  2.0906e-01,  4.6525e-01,\n",
       "          -5.3025e-01, -1.7328e-02,  2.4033e-01,  5.6998e-02, -4.8167e-01,\n",
       "          -6.5008e-02, -1.4846e-01, -2.4457e-01,  2.4718e-01,  3.6014e-01,\n",
       "           1.5060e-01, -3.7862e-02,  9.1108e-02, -1.0797e-01, -3.4441e-01,\n",
       "           5.2706e-01,  1.3708e-01, -3.8440e-02, -2.1701e-01, -2.5961e-01,\n",
       "           1.8518e-01,  6.3245e-01, -3.2645e-01, -3.4484e-01, -3.2959e-01,\n",
       "           1.5071e-02,  4.0212e-01,  4.9419e-02, -2.7837e-01,  3.9017e-01,\n",
       "          -5.9142e-01,  2.9673e-02, -2.9349e-01, -3.0998e-01,  4.4960e-01,\n",
       "           2.7789e-01, -6.4994e-02,  7.2055e-02, -3.0683e-01, -3.3547e-01,\n",
       "          -7.0446e-01, -2.6249e-01,  3.6067e-01, -6.0299e-01, -4.1016e-02,\n",
       "          -7.8054e-02,  1.2687e-01, -2.3417e-01,  3.9866e-01,  8.0168e-02,\n",
       "           6.5879e-01,  2.8352e-01,  2.6828e-02,  2.6984e-01,  7.7099e-02,\n",
       "          -1.4093e-01,  2.4527e-02, -3.9913e-02, -9.2053e-02, -5.9055e-01,\n",
       "          -5.9824e-01, -6.0020e-02,  8.5387e-02,  1.0105e-01, -7.5289e-02,\n",
       "           3.4278e-01,  4.4742e-02, -1.0267e-01,  6.0924e-02,  2.7522e-01,\n",
       "           7.9723e-02, -4.2483e-01, -1.8822e-01, -1.3406e-01,  4.7392e-02,\n",
       "           1.9432e-01, -3.6488e-02, -9.0974e-02, -1.6542e-01, -2.9621e-01,\n",
       "          -1.8251e-01,  6.6585e-01, -1.0238e-01, -3.9659e-01,  9.4436e-02,\n",
       "           5.2900e-01, -4.4700e-02, -1.0493e-01,  3.6194e-02,  9.2471e-02,\n",
       "          -3.2804e-01, -1.2649e-01,  3.7345e-01,  6.9938e-02,  3.2545e-01,\n",
       "           5.1171e-01,  5.4275e-02, -8.2490e-03,  1.9834e-01,  2.8770e-01,\n",
       "          -3.2254e-01, -1.8521e-01, -2.3696e-01,  1.8022e-01,  1.0272e-01,\n",
       "          -3.2412e-01,  6.2188e-02,  1.1498e-01,  3.7973e-02, -2.0941e-02,\n",
       "           2.4457e-01,  7.0749e-01, -1.7038e-01,  2.2357e-01,  7.0129e-01,\n",
       "           2.4521e-01,  1.6563e-01,  3.6712e-01,  2.1002e-01,  3.2866e-01,\n",
       "           6.4307e-01, -4.0556e-01, -3.4277e-01, -4.6066e-01,  1.5520e-01,\n",
       "           3.8282e-01, -4.1653e-01, -3.3137e-01, -1.8182e-01,  4.1840e-01,\n",
       "          -1.6800e-01,  2.6400e-01, -1.4409e-01,  3.7494e-01,  5.8489e-01,\n",
       "          -9.9579e-01,  2.5245e-01, -2.0085e-01,  2.9761e-01, -4.4682e-02,\n",
       "          -1.4281e-01, -1.0511e-01, -2.8579e-01, -2.3102e-01,  1.4189e-01,\n",
       "           3.8831e-01, -5.8581e-01, -1.1642e-01,  6.6722e-01, -3.4964e-01,\n",
       "          -9.6030e-03, -1.2642e-02,  3.0522e-02,  2.9520e-01,  1.0219e-01,\n",
       "          -2.1225e-01,  5.3823e-02, -1.8327e-01, -5.8603e-01,  2.5020e-01,\n",
       "           5.1520e-01,  1.4231e-01,  1.3160e-01,  1.9361e-01, -2.7998e-01,\n",
       "          -3.3588e-02, -4.1263e-01, -1.9093e-01, -4.7414e-01,  9.8468e-03,\n",
       "           3.4997e-02,  8.9377e-02,  2.3931e-01,  3.8261e-01, -3.0983e-01,\n",
       "           2.2599e-01,  5.9045e-01,  4.4813e-01,  2.0745e-01,  5.9690e-04,\n",
       "          -3.1501e-02,  2.1184e-01, -1.7436e-01,  2.1178e-01,  1.6601e-01,\n",
       "          -4.3246e-01,  2.4233e-01, -6.5211e-02, -3.5530e-01,  1.0731e-01,\n",
       "          -4.0044e-01,  1.1447e-01, -1.6124e-01, -2.6491e-01,  3.2480e-01,\n",
       "           1.3574e-01,  2.8446e-01, -7.5792e-01,  2.4965e-01,  3.1164e-01,\n",
       "           2.0489e-01,  2.1610e-01, -7.6969e-03,  3.5717e-01, -9.8022e-02,\n",
       "           3.7309e-02, -6.8710e-01, -6.7351e-02,  8.7986e-02,  1.4950e-01,\n",
       "          -1.9625e-01, -3.5192e-01,  5.5415e-01, -9.9453e-02,  4.6196e-01,\n",
       "           3.2529e-01,  4.1679e-02, -1.1738e+00, -2.8396e-01,  1.4260e-01,\n",
       "          -1.7033e-01, -2.2917e-01,  1.8981e-01, -2.5386e-01,  2.8040e-01,\n",
       "          -8.2290e-01, -6.2827e-02,  2.3686e-02,  3.3901e-01, -2.6604e-01,\n",
       "           6.2188e-01, -3.0344e-01,  1.8954e-01,  2.0463e-01,  1.9480e-01,\n",
       "           3.5704e-01,  6.3107e-01, -1.9795e-01, -7.1055e-01, -3.0159e-01,\n",
       "           1.6718e-01,  1.4217e-01, -3.4165e-02, -8.1922e-01,  3.2177e-01,\n",
       "           3.1805e-01,  3.9594e-01,  4.8856e-01]], device='mps:0'),\n",
       " tensor([[-6.1750e-01, -4.2829e-01, -2.3934e-01, -9.4116e-01, -3.4548e-01,\n",
       "          -6.5496e-02, -6.3874e-01,  9.7828e-01, -9.5892e-02,  4.0096e-01,\n",
       "           1.4919e-01, -5.1603e-03,  2.0980e-01,  1.3680e-01, -6.9506e-01,\n",
       "           1.6740e-01,  4.6094e-01,  2.1159e-01,  3.5951e-01, -4.4329e-02,\n",
       "           7.9832e-01,  3.4245e-01,  1.8019e-01, -2.5205e-01, -4.5876e-01,\n",
       "           5.8994e-01, -2.4628e-01, -9.2014e-02,  2.0509e-01, -8.1949e-01,\n",
       "          -1.3586e-01, -4.9757e-02, -3.6281e-01,  8.7602e-01,  1.1948e-01,\n",
       "          -2.7310e-01, -1.5276e-01, -2.7996e-01,  2.3494e-01, -6.3637e-02,\n",
       "           3.7814e-01,  2.6592e-02, -2.9347e-01,  3.1386e-01, -1.6421e-01,\n",
       "          -1.4057e-01,  1.4047e-02, -7.8429e-01,  1.3666e-02,  1.8955e-01,\n",
       "           5.8851e-02, -4.7205e-02,  1.9231e-01,  3.7030e-01, -1.0617e+00,\n",
       "          -2.6348e-01,  2.7603e-01,  2.9849e-01, -1.7932e-01, -5.7784e-02,\n",
       "          -1.9989e-01,  9.0201e-01,  1.7949e-01, -2.7822e-01,  1.7824e-01,\n",
       "          -4.4899e-01, -5.4601e-01, -6.0781e-01, -1.0682e-01, -1.1967e-01,\n",
       "          -1.0002e-01,  9.7528e-02,  1.4009e-01, -2.9385e-01,  2.3170e-01,\n",
       "           2.1813e-01,  4.7024e-02, -2.7652e-02, -4.3158e-01, -2.6546e-01,\n",
       "           3.6947e-01,  3.4425e-02, -1.4730e-02,  2.0326e-01, -5.8744e-01,\n",
       "          -2.0226e-01,  3.7959e-01,  1.5703e-01,  2.5914e-01,  2.2662e-03,\n",
       "           5.9967e-01,  1.2024e-02,  8.0339e-02, -4.5369e-01,  5.8427e-01,\n",
       "          -5.4955e-01,  1.0329e-01, -1.6007e-01,  4.9532e-02, -5.2088e-01,\n",
       "          -9.8735e-02,  2.4494e-01, -1.8085e-01,  1.6053e-01, -4.6592e-01,\n",
       "          -3.2983e-02,  2.4514e-01,  3.0468e-01, -4.9183e-01,  4.2685e-02,\n",
       "          -6.5134e-03, -2.9791e-01,  7.3694e-02, -6.6407e-01, -8.5795e-01,\n",
       "           5.8354e-01,  1.8428e-02, -4.5240e-01, -2.3930e-01,  7.4443e-01,\n",
       "           2.9738e-01, -2.3201e-01, -5.3072e-02, -3.5548e-01,  2.0259e-01,\n",
       "          -1.8518e-01,  2.5689e-01, -2.7018e-01,  3.8439e-01,  5.5026e-01,\n",
       "           3.1581e-01,  2.8489e-01, -1.8327e-01,  2.7637e-01,  1.4465e-01,\n",
       "           3.6947e-02, -6.6614e-02, -1.5929e-01,  1.5180e-01,  1.2041e-01,\n",
       "          -2.9580e-01, -3.2999e-01,  5.4746e-03,  1.1182e-01, -3.3279e-01,\n",
       "          -1.0202e-01,  1.8680e-01,  3.4808e-01, -1.4092e-01, -2.5403e-01,\n",
       "          -7.8984e-01, -4.2328e-01,  1.0217e-01, -1.8560e-01, -1.7135e-01,\n",
       "          -4.1911e-01,  3.8121e-01, -6.6660e-02,  4.7638e-01,  3.4189e-01,\n",
       "           6.9978e-01, -1.5311e-01,  5.9776e-02, -5.6412e-01,  3.5197e-01,\n",
       "           3.8270e-02, -6.1110e-01, -1.7247e-01,  1.5680e-01, -4.1813e-04,\n",
       "           6.8195e-01,  7.0979e-02,  4.0124e-01,  1.2070e-01, -3.2553e-01,\n",
       "           8.2089e-03,  1.9890e-01, -1.2864e-01, -1.4297e-01,  7.3766e-01,\n",
       "          -5.5188e-01, -2.3029e+00,  1.0721e-01, -3.5844e-01, -3.6529e-01,\n",
       "          -1.8135e-01,  3.7255e-01,  3.3772e-02,  4.9966e-01,  1.5486e-01,\n",
       "           3.7598e-02,  1.6729e-01, -2.8025e-02,  3.6851e-01, -8.4399e-02,\n",
       "           2.6751e-01,  3.3341e-01,  1.2988e-01, -2.3298e-01, -1.0956e-01,\n",
       "          -3.9227e-01,  2.4031e-01, -9.8916e-02,  8.6010e-01,  2.3260e-01,\n",
       "           2.0943e-01, -4.3657e-03, -4.7262e-02,  4.8964e-01,  4.3099e-01,\n",
       "          -6.6541e-02, -1.6585e-01, -4.9522e-01,  4.6269e-03, -4.1279e-02,\n",
       "          -1.2963e-02, -4.5611e-01,  1.3753e-01,  4.2841e-01, -1.2416e-01,\n",
       "          -1.8354e-01,  1.5525e-01, -4.7715e-02, -3.1517e-01, -2.6492e-02,\n",
       "           6.8453e-01,  1.9080e-01, -2.2994e-01,  1.2377e-01, -1.1896e-01,\n",
       "           3.7006e-02, -4.6975e-01,  7.0044e-01,  6.1147e-01,  1.5585e-01,\n",
       "           1.6214e-02, -6.9943e-01,  2.3699e-01, -3.1574e-01, -4.7647e-03,\n",
       "          -1.5027e-01, -1.5731e-01, -4.2986e-01, -5.7285e-02, -5.5854e-02,\n",
       "           4.0115e-01,  7.9943e-02,  4.6653e-01,  1.1837e+00,  2.7973e-01,\n",
       "           1.9184e-02,  5.6097e-01,  2.0801e-01,  4.2492e-01, -4.0180e-01,\n",
       "           3.3082e-01,  8.8825e-02, -6.3896e-01,  2.7188e-01, -4.1593e-01,\n",
       "           4.4963e-02, -4.1432e-02, -3.7926e-01,  1.4208e-01, -2.1276e-02,\n",
       "           3.4237e-01,  3.3517e-01,  1.5186e-01,  1.6838e-01,  2.9411e-01,\n",
       "          -3.1322e-01,  5.9715e-01, -6.0809e-02, -1.1127e-01, -6.3492e-01,\n",
       "          -7.6732e-01,  9.5206e-02,  4.4743e-01, -1.2880e-01, -6.1279e-01,\n",
       "           1.0462e-01, -2.3866e-01, -1.2257e-01, -3.5064e-02,  2.1191e-01,\n",
       "           2.7455e-01,  5.0687e-02, -9.5436e-02,  6.4791e-02, -4.6950e-01,\n",
       "          -4.9277e-01,  4.9140e-02, -1.5758e-01, -1.3063e-01,  5.3918e-01,\n",
       "          -7.9415e-02,  1.1575e-01,  5.7221e-01,  1.9799e-02, -4.5976e-01,\n",
       "          -2.5640e-01,  1.4802e-01,  2.8667e-01, -7.2082e-02,  2.1981e-01,\n",
       "           3.4119e-01,  2.6467e-02,  5.1649e-01,  1.4823e-01, -2.1046e-01,\n",
       "          -2.7239e-01,  5.1173e-01, -1.5755e-01,  6.1403e-02, -1.2527e-01,\n",
       "          -8.7207e-01, -5.6150e-02, -7.8413e-02, -1.6833e-01, -5.4440e-01,\n",
       "           2.6566e-01,  1.4351e-01,  1.0561e-01,  2.2350e-01, -1.5984e-01,\n",
       "           7.0123e-01,  6.5285e-01, -3.3100e-01, -1.6943e-01, -4.0104e-01,\n",
       "           9.3922e-02, -6.8374e-02, -2.9329e-01, -1.4009e-02,  9.5368e-03,\n",
       "          -2.2765e-01, -1.2516e-01, -1.0805e-01,  2.8921e-01, -1.8876e-01,\n",
       "           5.2862e-01,  1.2189e-01,  8.3649e-02,  4.3873e-01,  2.0612e-02,\n",
       "           1.3071e-01,  3.6515e-01, -4.2117e-01,  1.2870e-01,  4.0424e-01,\n",
       "           4.0578e-01, -1.0551e+00, -8.3291e-01, -1.4936e-01,  3.6111e-01,\n",
       "           2.2860e-01,  2.5189e-01,  1.6999e-02, -7.2215e-03,  8.9455e-02,\n",
       "          -2.3402e-01,  3.5135e-01, -6.6415e-02, -5.7412e-01,  3.0561e-01,\n",
       "           2.3567e-01,  5.3263e-01, -2.7377e-01, -5.0579e-01, -6.5233e-02,\n",
       "           2.4616e-01, -3.6381e-01,  1.1392e-01,  2.3480e-01,  4.5466e-01,\n",
       "          -3.6015e-01, -5.6164e-01, -7.0282e-02, -7.3986e-01, -1.3578e-01,\n",
       "           3.2483e-01, -2.1437e-01,  2.7178e-02, -2.4758e-01,  1.5231e-01,\n",
       "          -1.7925e-02, -1.8879e-01,  6.0757e-01,  2.0522e-01, -2.5855e-01,\n",
       "           1.1981e-01, -2.3892e-01, -1.9475e-01,  5.0091e-01, -5.0226e-01,\n",
       "          -3.0348e-01, -5.0960e-02, -1.5349e-01, -3.2418e-01,  3.2181e-01,\n",
       "          -4.1389e-01, -1.4861e-01,  1.6748e-01, -6.4736e-02, -1.3068e-01,\n",
       "          -3.5233e-01,  1.4268e-01, -1.2707e-01, -4.9883e-02, -1.7654e-02,\n",
       "          -1.3353e-01,  6.0900e-02,  1.3308e-01,  6.4538e-02,  1.4388e-01,\n",
       "           5.3696e-01, -1.1464e-01,  2.7582e-03, -2.5900e-01, -2.2469e-01,\n",
       "           5.2498e-01, -9.4973e-02, -3.3117e-01,  5.4928e-01, -1.1156e+01,\n",
       "           1.0386e-01,  7.2631e-01, -5.5808e-01,  5.5128e-02, -2.8539e-01,\n",
       "           2.1085e-01, -3.0009e-01,  7.9101e-02,  2.4967e-01, -2.8460e-01,\n",
       "          -1.2277e-01, -2.7364e-01, -2.3193e-01, -4.0872e-01, -3.9742e-01,\n",
       "          -4.7229e-01, -5.6789e-01,  1.1772e-01, -1.7663e-01, -1.0996e-01,\n",
       "          -5.7036e-01, -1.9379e-01,  2.9696e-01, -4.9559e-03, -5.1069e-01,\n",
       "           5.9541e-02, -2.2948e-01,  2.6399e-01,  4.6126e-02,  2.2045e-01,\n",
       "          -1.9949e-02,  1.4108e-01,  2.0669e-01, -4.4946e-02, -4.5100e-01,\n",
       "          -4.4853e-01,  2.2129e-01, -1.5955e-01, -2.2267e-01, -5.4017e-02,\n",
       "          -3.3838e-01, -2.7689e-01,  1.3447e-01,  3.7594e-01,  6.2596e-01,\n",
       "           1.1595e+00,  5.6208e-01,  2.9245e-01, -1.0625e-01, -8.5438e-02,\n",
       "          -1.0762e+00,  5.5900e-01, -3.7318e-01,  8.1787e-02, -2.5119e-01,\n",
       "           6.6201e-01, -4.9906e-01,  5.4685e-01, -4.0754e-01, -2.5500e-01,\n",
       "           3.7179e-01,  3.0194e-01,  7.1104e-03,  2.8239e-02, -2.7432e-01,\n",
       "           4.0257e-01, -2.6094e-01, -3.6915e-02, -3.7819e-01,  4.0081e-01,\n",
       "          -3.9472e-01,  5.0770e-01, -5.6534e-02,  1.2945e-01,  1.8567e-01,\n",
       "          -3.5001e-01,  1.6610e-01,  1.0337e+00,  1.7414e-01,  4.1071e-01,\n",
       "           1.7420e-01,  5.1160e-01,  5.3064e-02, -8.8958e-01,  1.3691e-01,\n",
       "           2.0502e-01,  2.0789e-01, -6.7096e-02,  5.7131e-01, -2.5517e-02,\n",
       "          -6.5181e-01, -2.3094e-01, -3.2169e-01,  4.3108e-01,  6.5352e-01,\n",
       "          -2.2597e-01, -4.4343e-01,  2.3894e-01, -7.9776e-03, -5.3395e-01,\n",
       "           9.0074e-02, -2.2320e-01, -5.2792e-01,  8.7275e-02, -5.8914e-01,\n",
       "           1.4985e-01, -1.2428e-01,  5.4238e-02,  6.6887e-01, -4.0159e-01,\n",
       "          -1.6113e-01,  3.6602e-02, -4.4532e-02, -4.1270e-01, -4.9520e-01,\n",
       "          -7.7083e-02,  4.1073e-01, -1.9776e-01, -2.7976e-01,  7.0636e-02,\n",
       "          -3.0673e-01,  2.2584e-01, -1.7377e-01, -6.4935e-01,  3.2879e-01,\n",
       "          -6.0415e-01,  2.0578e-01,  2.5360e-01, -2.2507e-01,  4.9017e-01,\n",
       "           1.1337e-01, -8.1817e-02,  1.9397e-01, -9.9369e-02, -7.8554e-01,\n",
       "          -3.2573e-01, -2.7299e-02,  3.2662e-01, -4.2844e-01,  1.5776e-01,\n",
       "           1.5022e-01, -4.5752e-02, -2.2574e-01,  2.7904e-01,  1.0174e-01,\n",
       "           3.7762e-01,  3.9846e-01, -5.5006e-01,  1.1727e-01, -2.9867e-01,\n",
       "          -2.9262e-02, -6.1174e-02,  1.7952e-01,  2.3654e-01, -7.0764e-01,\n",
       "          -1.5224e-01, -2.6434e-01, -3.0965e-01, -2.1650e-01, -2.5120e-01,\n",
       "           2.0680e-01,  1.4038e-01, -1.5897e-02, -1.1308e-01,  5.8077e-01,\n",
       "           4.0814e-02, -1.1167e-01, -4.5997e-01,  3.1443e-01,  4.3178e-01,\n",
       "           2.1794e-01,  7.5875e-02, -1.2830e-01, -5.5532e-01, -1.1922e-02,\n",
       "          -4.3644e-01,  9.1081e-01, -7.3906e-01, -7.1970e-01,  6.6086e-02,\n",
       "           1.4951e-02, -8.8259e-04, -2.3450e-01, -5.0265e-01,  8.1298e-02,\n",
       "          -2.1017e-01, -1.4891e-01,  6.1550e-03, -1.0607e-02,  1.6299e-01,\n",
       "           4.0742e-01,  3.6838e-01,  1.5951e-01, -3.3938e-01, -4.4081e-01,\n",
       "          -6.0244e-01,  1.4567e-02, -1.6795e-01,  2.9330e-01, -2.6752e-01,\n",
       "          -3.2661e-01,  1.6043e-02, -9.1892e-03, -2.2272e-01,  5.2876e-01,\n",
       "           1.2427e-01,  1.6396e+00, -3.7287e-01, -4.4551e-01,  2.9018e-01,\n",
       "           7.2306e-02, -2.7073e-02, -2.8130e-01,  2.5074e-01,  9.5447e-02,\n",
       "           4.0194e-02, -3.6304e-01,  1.5023e-02, -2.0558e-02,  3.2773e-01,\n",
       "           3.2056e-01, -1.0703e+00, -2.0238e-01, -1.7993e-01,  5.5664e-01,\n",
       "           3.2388e-02,  6.3185e-01, -2.2305e-01,  3.0577e-01,  4.6106e-01,\n",
       "          -8.6667e-01, -2.3000e-01, -1.6129e-01,  2.7295e-01, -1.0583e-01,\n",
       "          -9.8735e-02, -1.6178e-01, -3.7165e-02, -1.6936e-01,  3.9538e-01,\n",
       "           7.4751e-01, -7.8426e-01,  7.4081e-01,  7.0467e-01, -9.8234e-02,\n",
       "           5.6815e-01,  9.4187e-02,  3.2641e-01, -1.4659e-01,  2.2669e-02,\n",
       "           2.6200e-01,  1.8542e-01, -6.3165e-01, -2.3962e-01,  2.4473e-01,\n",
       "           4.0158e-01, -1.1943e-01,  1.2108e-01,  7.9292e-02,  3.0960e-02,\n",
       "          -3.7691e-01, -1.0145e-01, -2.4552e-01, -3.9904e-01,  1.1356e-01,\n",
       "           1.4603e-02, -1.5001e-01, -8.6997e-02,  4.7709e-01, -1.7580e-01,\n",
       "           7.5067e-01,  4.2910e-01,  3.7145e-01, -3.5103e-01,  8.9950e-04,\n",
       "          -5.7565e-02,  6.0671e-01, -3.1670e-01,  7.7385e-01,  6.3337e-02,\n",
       "          -2.3140e-01,  2.7251e-01,  6.0547e-03,  1.7829e-01,  2.9892e-01,\n",
       "          -2.3519e-01,  1.2702e-01, -3.0718e-01, -2.3711e-01,  2.8330e-01,\n",
       "          -1.8640e-01,  1.1542e-01, -7.2270e-01, -6.3602e-03,  3.9857e-01,\n",
       "          -3.2051e-01,  4.9873e-01, -2.3074e-01,  6.1026e-01, -4.3509e-01,\n",
       "           3.2669e-02, -4.7470e-01, -4.3275e-01, -2.8508e-01,  5.1854e-02,\n",
       "          -8.6027e-01, -1.9453e-01,  1.7684e-01,  3.7405e-02,  2.1707e-01,\n",
       "           7.5155e-01, -3.3794e-01, -2.4771e-01, -2.0386e-01,  4.0190e-02,\n",
       "           1.8299e-01, -8.2071e-02, -3.5586e-01,  2.1161e-01, -3.5607e-01,\n",
       "          -1.0520e+00, -2.3117e-01,  8.9990e-02,  3.8419e-01, -3.5853e-01,\n",
       "           1.1570e+00,  2.5132e-02, -2.0226e-01,  6.7253e-01,  2.0808e-01,\n",
       "           6.4157e-01,  6.0323e-01,  1.3628e-03, -4.1259e-01, -5.4109e-02,\n",
       "          -1.5877e-01, -3.2881e-01,  1.5757e-01, -1.8346e-01,  5.0556e-01,\n",
       "           5.4802e-01, -2.0806e-02,  5.4756e-01]], device='mps:0'),\n",
       " tensor([[-2.8441e-01, -5.5394e-01, -5.3487e-01, -8.6137e-01, -1.6500e-01,\n",
       "          -1.5627e-01, -6.9154e-01,  6.2842e-01, -3.2176e-01,  7.4840e-01,\n",
       "           2.7524e-01,  6.2658e-01,  4.8185e-02,  8.1655e-02, -5.4587e-01,\n",
       "          -1.8390e-01, -8.5546e-02,  3.8717e-01,  2.1180e-01,  6.1308e-01,\n",
       "           8.1107e-01,  6.8047e-01, -1.9768e-01,  2.1503e-01, -5.0498e-01,\n",
       "           3.5649e-01, -6.9646e-02,  6.8591e-01,  2.3685e-01, -4.8309e-01,\n",
       "           4.5488e-02,  5.1755e-01, -7.6515e-02,  1.6821e+00, -1.0095e-01,\n",
       "          -2.5680e-01,  2.3473e-01, -5.4682e-01, -4.1870e-01, -4.8584e-01,\n",
       "           6.0923e-01, -3.5607e-01,  1.1066e-02,  1.0019e-01, -1.2748e-01,\n",
       "           4.3426e-01,  3.3175e-01, -4.5028e-01,  2.1806e-01, -8.6036e-01,\n",
       "          -6.0268e-01, -1.1943e-01,  1.3884e-01,  7.5491e-01, -7.6410e-01,\n",
       "          -7.1493e-01,  8.1095e-02, -1.2005e-01, -1.3680e-01,  3.9091e-02,\n",
       "           2.6227e-01,  2.0679e-01, -8.1044e-02,  6.1526e-02,  1.7546e-01,\n",
       "           1.7241e-01, -1.9290e-01, -7.4059e-01, -4.7855e-01,  7.6078e-02,\n",
       "           9.5219e-02,  1.1330e-01,  1.1899e-01, -1.9241e-01,  3.2532e-01,\n",
       "          -2.3076e-01,  2.7391e-01, -3.3521e-01, -1.3154e-01, -7.0176e-02,\n",
       "          -4.4051e-02,  8.9766e-02,  1.9541e-01, -1.9095e-01, -1.4401e-01,\n",
       "           1.1033e-02,  5.0526e-01,  5.0223e-02, -8.2932e-02, -4.1653e-02,\n",
       "           1.5756e-02,  1.8359e-02, -6.3429e-01, -4.0049e-01,  1.9855e-01,\n",
       "           7.6685e-03,  3.6200e-01, -3.5243e-01,  1.8487e-01, -2.3329e-01,\n",
       "           2.0495e-01, -1.3106e-01, -1.0602e-01,  5.0891e-02, -6.4400e-01,\n",
       "          -5.0268e-01,  1.7121e-01,  1.3393e-05, -5.4945e-01,  1.4619e-02,\n",
       "          -4.0735e-01, -6.1942e-01,  3.2930e-02, -2.3743e-01, -5.5993e-02,\n",
       "           1.5373e-01,  6.8113e-02, -4.6111e-04,  3.8857e-03, -7.4266e-02,\n",
       "           5.8578e-01, -4.1646e-02,  1.6548e-01,  9.9639e-02,  2.6984e-01,\n",
       "          -3.2575e-01,  6.6620e-01, -3.1006e-01,  2.6532e-01,  4.3677e-01,\n",
       "          -1.2551e-01,  2.2294e-03, -3.9518e-01, -9.1430e-02,  3.8994e-01,\n",
       "           3.5504e-01, -2.5205e-01, -1.6596e-01, -3.2194e-01,  1.2660e-01,\n",
       "          -1.1306e-01, -1.9384e-01, -1.2373e-01, -4.7651e-01, -9.7027e-02,\n",
       "          -2.5137e-01, -1.8891e-01,  9.8524e-03, -1.3107e-02, -5.6441e-01,\n",
       "          -3.7041e-01, -1.0094e+00,  8.9048e-01, -7.7338e-02,  1.0971e-01,\n",
       "          -5.2245e-02,  1.3917e-01,  1.3157e-01,  2.4190e-01,  3.4615e-01,\n",
       "           4.0161e-01, -3.0681e-01,  1.7669e-01, -6.3478e-01, -2.1620e-01,\n",
       "          -1.6021e-02,  5.5562e-02, -4.2343e-01,  4.4272e-01,  1.4258e-01,\n",
       "           6.6604e-02, -3.1665e-01,  9.0541e-02, -2.5514e-01,  2.3988e-01,\n",
       "          -5.2097e-01,  1.3542e-01,  8.6542e-03,  9.9876e-02, -2.3345e-01,\n",
       "           1.6611e-01, -1.2452e+00,  8.5164e-02, -1.3329e-01, -3.9477e-02,\n",
       "          -3.4149e-01,  3.9552e-01,  3.1089e-01,  3.2796e-01, -1.5308e-01,\n",
       "          -8.1520e-02, -7.1932e-02,  9.9920e-02,  2.7144e-02, -2.8755e-01,\n",
       "           4.6967e-01,  6.6779e-01,  4.0208e-01, -1.6045e-01,  9.8117e-02,\n",
       "           2.1663e-02,  2.2509e-01, -6.3057e-02,  1.4737e-01,  3.4920e-01,\n",
       "           1.6982e-01, -1.9862e-01, -5.2368e-02,  3.0330e-01, -4.0755e-02,\n",
       "          -2.1991e-01, -2.3430e-02, -4.2824e-01, -1.0580e-01, -3.9187e-01,\n",
       "           1.2879e-01, -3.9608e-01, -1.2370e-01,  2.9763e-01, -1.3954e-01,\n",
       "          -1.9156e-01,  4.9995e-02,  1.7278e-01, -1.4398e-01,  1.1770e-01,\n",
       "           1.3569e-01, -3.2259e-01,  9.6180e-02, -3.1440e-01, -1.8272e-01,\n",
       "           5.7590e-02, -4.0861e-01,  7.0641e-01,  2.7090e-01, -2.1076e-01,\n",
       "           2.2706e-01, -3.7797e-01,  7.5726e-02, -1.9180e-02,  1.7069e-01,\n",
       "          -5.9430e-03, -3.4191e-01, -9.1396e-01,  2.0747e-01,  6.7531e-02,\n",
       "           2.8930e-01, -2.4971e-02,  4.7363e-01,  8.3110e-01,  1.9939e-01,\n",
       "          -3.0489e-02, -1.6327e-01,  4.5326e-01,  3.1228e-01, -2.1691e-01,\n",
       "          -2.6165e-02, -9.7939e-02, -8.4730e-01,  3.6313e-02, -4.6692e-01,\n",
       "          -2.0889e-02,  2.2189e-01, -3.3192e-03,  2.0753e-01, -3.7387e-01,\n",
       "           6.3320e-01,  4.4671e-01,  9.7346e-02, -3.8640e-01,  6.3860e-01,\n",
       "           1.0362e-01,  3.3258e-01, -3.3726e-01, -1.6538e-01, -5.7175e-01,\n",
       "          -6.9858e-01, -1.9525e-01,  4.4452e-01,  7.1309e-02, -4.3032e-01,\n",
       "          -1.7021e-02,  1.0086e-01,  4.6269e-02, -3.2002e-01,  4.1130e-02,\n",
       "           1.1790e-01, -1.4409e-01, -2.3553e-02, -1.8172e-01, -3.0541e-01,\n",
       "          -4.0109e-01,  3.2059e-01, -4.5628e-01, -1.9039e-02,  6.3341e-01,\n",
       "          -5.6343e-02, -2.5236e-01,  1.3631e-01, -7.7882e-03, -1.5985e-01,\n",
       "           7.7204e-02,  4.9073e-01, -1.7577e-01, -3.1271e-01,  3.3961e-01,\n",
       "          -6.8220e-02,  4.1486e-01,  3.8705e-02, -5.9873e-01, -2.8172e-01,\n",
       "           8.1087e-02,  4.2650e-01, -6.3146e-02, -1.9707e-01, -1.0664e-01,\n",
       "          -7.9628e-01, -3.1771e-01, -1.7417e-01, -3.7691e-01, -4.4016e-01,\n",
       "           1.1354e-02,  2.5387e-01, -1.6713e-01,  3.0450e-02, -2.3705e-01,\n",
       "           3.6587e-01,  9.3493e-01, -3.4931e-01,  4.2838e-02, -4.9836e-01,\n",
       "           4.5254e-02, -8.1734e-02, -6.2819e-02, -3.3827e-01, -4.0564e-01,\n",
       "           3.5635e-01,  1.3258e-01,  2.5235e-01,  2.0424e-02, -2.3089e-01,\n",
       "           3.3129e-01, -3.3075e-01, -8.9622e-02,  7.1975e-01, -2.8549e-01,\n",
       "           4.0563e-02,  2.8078e-01, -6.6874e-01, -8.5723e-02,  3.4916e-01,\n",
       "           1.6608e-01, -6.5247e-02, -8.3725e-01, -7.6890e-03,  4.5139e-01,\n",
       "           3.8668e-01,  1.8128e-01,  2.0962e-01,  5.4342e-01,  3.6529e-01,\n",
       "          -4.0580e-01,  3.7468e-01, -1.1954e-01, -1.3160e-01,  4.5812e-02,\n",
       "           1.4795e-01,  1.4958e-01, -4.0168e-01,  3.3391e-01, -1.7011e-01,\n",
       "           2.4076e-02, -4.0941e-01, -1.8527e-01, -9.4000e-03,  3.9568e-01,\n",
       "          -2.4590e-02, -4.3180e-01,  1.3466e-01, -3.7448e-01, -2.4752e-01,\n",
       "           9.1886e-02,  7.0489e-02,  3.2154e-01, -4.9037e-01, -2.5280e-01,\n",
       "          -1.1939e-01, -3.0043e-01,  5.9537e-01,  1.4481e-02, -1.3875e-01,\n",
       "           3.3885e-01, -2.4500e-01, -1.9853e-01,  3.8153e-01,  3.6541e-02,\n",
       "          -3.2929e-01,  1.7709e-01, -2.1721e-01,  2.6912e-02,  7.2715e-01,\n",
       "          -3.6499e-01, -1.2932e-02, -2.0360e-02,  2.0395e-01,  3.4879e-01,\n",
       "          -2.8039e-01, -1.2410e-01,  1.7326e-01, -4.9986e-01, -4.2984e-02,\n",
       "          -1.2823e-01,  3.1628e-01, -2.6186e-01, -1.3695e-01, -4.3339e-01,\n",
       "           1.3805e-02,  3.3696e-02, -1.2055e-01,  1.0068e-01, -2.8598e-01,\n",
       "           6.6533e-01, -2.4381e-01, -2.6331e-01,  1.3605e-01, -1.1418e+01,\n",
       "           1.1377e-01,  8.4308e-01,  1.6280e-01,  2.3584e-01, -2.9657e-01,\n",
       "           1.0418e-01,  1.1050e-01,  7.0321e-02, -3.2792e-01,  1.4004e-01,\n",
       "          -8.6370e-02,  6.6247e-03,  3.9086e-01, -3.6419e-01,  9.2481e-02,\n",
       "          -1.1838e-01, -1.0034e+00,  3.8269e-02,  2.4191e-01,  4.5080e-01,\n",
       "          -2.9394e-01,  2.1662e-02,  3.1407e-01, -3.1812e-01, -6.7698e-02,\n",
       "          -3.1501e-01, -9.0094e-02,  2.0232e-01,  1.2938e-01, -5.3503e-02,\n",
       "           1.2833e-02, -8.4343e-02,  4.0612e-01, -2.2802e-02, -4.8456e-01,\n",
       "          -1.5634e-01,  6.4033e-02,  3.7287e-02, -5.1122e-01, -4.6959e-01,\n",
       "           1.6187e-01, -2.7521e-01,  1.5326e-01,  3.0423e-01,  5.8338e-01,\n",
       "           6.4433e-01,  8.7874e-01,  9.8589e-02,  6.4743e-03, -1.2715e-01,\n",
       "          -8.6394e-01,  1.4528e-01, -6.1263e-01,  3.3415e-01, -2.5418e-01,\n",
       "           6.9162e-01, -5.7891e-01,  3.3339e-01, -4.9050e-01, -2.3375e-01,\n",
       "           4.9885e-01,  2.1698e-01,  2.7781e-02, -5.2102e-01, -9.8122e-02,\n",
       "           3.4905e-01,  8.7095e-03,  2.4822e-01, -4.3775e-01, -6.8448e-02,\n",
       "           2.5354e-02,  7.5355e-01,  6.8922e-02,  1.5600e-01,  4.6976e-02,\n",
       "          -2.9262e-01, -1.1298e-01,  1.9105e+00, -4.5921e-02,  9.8367e-02,\n",
       "           1.5618e-01,  5.6595e-01, -3.6775e-01, -8.2060e-01, -9.5813e-03,\n",
       "          -1.5806e-01, -4.0744e-02,  6.1770e-02,  5.3651e-01, -4.0122e-01,\n",
       "          -2.8241e-01, -4.2715e-02,  2.2891e-01,  8.1120e-02,  9.9514e-01,\n",
       "          -5.0647e-01, -1.3592e-01,  6.5777e-03,  6.3368e-01, -4.4815e-01,\n",
       "           2.5587e-02, -3.4930e-01, -5.6310e-02, -1.7721e-02, -3.8494e-02,\n",
       "          -1.2066e-01, -3.6498e-02,  6.6482e-02,  3.8879e-01, -5.0006e-02,\n",
       "          -1.0725e-01, -1.0326e-01, -4.3712e-02, -3.7134e-01, -1.2494e-01,\n",
       "           1.8692e-01,  3.5026e-01, -5.6838e-01, -5.4588e-01, -3.2157e-01,\n",
       "          -3.7195e-02,  2.8179e-01,  5.1425e-02, -4.7998e-01,  5.6484e-01,\n",
       "          -6.6180e-01, -1.0760e-01,  2.6979e-01,  2.1886e-01,  4.8424e-01,\n",
       "           2.3962e-01, -2.7152e-01,  4.6382e-01, -5.7533e-01, -3.5601e-01,\n",
       "          -7.6686e-01,  7.1204e-03,  3.3090e-01, -3.4110e-01,  7.7304e-02,\n",
       "           1.5510e-01, -4.8632e-02,  1.7306e-01,  1.1810e-01,  1.6733e-01,\n",
       "          -2.1232e-02, -1.6458e-02, -5.1332e-01,  2.7794e-01, -6.9124e-02,\n",
       "          -4.3402e-01,  3.8644e-02, -3.1164e-01, -1.5215e-01, -1.1196e-01,\n",
       "          -2.3435e-02,  7.5629e-02, -4.2071e-01, -1.6949e-01,  1.5603e-01,\n",
       "           1.2218e-01, -1.6835e-01, -3.5913e-01,  4.8729e-01,  9.7509e-02,\n",
       "          -5.5797e-02, -2.0874e-01, -3.8294e-01, -1.4694e-01,  1.5783e-01,\n",
       "           6.2652e-02, -6.3195e-02, -5.2337e-01, -4.0067e-01, -2.3889e-01,\n",
       "          -5.7472e-01,  1.0858e+00, -9.2285e-02, -5.5553e-01, -7.3773e-02,\n",
       "           3.4282e-01, -2.2721e-01,  2.4674e-01,  1.7079e-01,  3.8558e-01,\n",
       "          -3.3978e-01, -3.9431e-01,  6.1823e-03, -2.5758e-02,  3.9101e-01,\n",
       "           3.4840e-01,  5.9696e-01,  4.9011e-01,  2.1357e-01, -1.1559e-01,\n",
       "          -5.7487e-01,  3.2738e-01, -3.6646e-01, -3.9739e-01,  3.8141e-01,\n",
       "          -9.0689e-02,  2.4697e-02,  2.8430e-01, -3.6602e-01,  1.8617e-01,\n",
       "           6.4427e-01,  7.4935e-01, -6.0195e-02, -3.8977e-01,  5.6978e-01,\n",
       "           2.1737e-01, -1.3383e-01,  1.7105e-01,  4.2291e-01,  1.2353e-01,\n",
       "           3.0668e-01,  1.2484e-01, -2.9104e-01, -9.1789e-02,  1.9966e-01,\n",
       "           3.2000e-01, -4.7134e-01, -5.8480e-01,  6.0997e-02,  1.4006e-01,\n",
       "          -1.2669e-01,  4.8179e-01, -5.6792e-02,  3.6236e-01,  3.2851e-01,\n",
       "          -1.1371e+00,  1.4501e-01, -1.7041e-01,  1.8232e-02, -1.7073e-01,\n",
       "          -1.1745e-01, -4.2501e-01, -2.3848e-01, -2.9930e-01,  2.4263e-01,\n",
       "           2.5979e-01, -6.7001e-01, -6.7164e-02,  3.6919e-01, -2.9916e-01,\n",
       "           4.2863e-01, -4.9112e-01,  4.9717e-01,  6.8122e-01, -3.4430e-01,\n",
       "           2.7410e-01, -8.4230e-02, -6.1617e-01, -3.5603e-01,  2.0665e-01,\n",
       "           5.3170e-01, -2.7458e-02,  1.2537e-01, -2.1460e-01, -5.2783e-01,\n",
       "          -5.7901e-02, -2.4859e-01,  1.4412e-01, -3.5078e-01,  3.9901e-02,\n",
       "          -1.7903e-01, -3.2260e-01, -2.7667e-02,  4.1629e-01,  1.4599e-01,\n",
       "           2.2399e-01,  4.8948e-01,  6.3414e-01,  2.3314e-02, -2.6792e-01,\n",
       "          -9.9237e-02,  3.6744e-01, -5.3180e-02,  9.9155e-01, -7.8306e-02,\n",
       "          -1.8306e-01, -2.1731e-02,  8.6113e-03, -1.2811e-01,  3.4870e-01,\n",
       "          -2.2924e-01,  2.5329e-02, -1.6893e-01, -6.6031e-02,  2.5295e-01,\n",
       "           1.3629e-01,  3.7671e-02, -4.3246e-01,  2.0870e-01,  4.7896e-01,\n",
       "          -2.3456e-02,  4.0253e-01, -1.3764e-01,  4.6894e-01, -4.5455e-01,\n",
       "          -2.9003e-01, -5.5458e-01,  2.4767e-01,  2.0208e-01, -8.0991e-02,\n",
       "          -5.6740e-01, -2.9935e-01,  3.7720e-01, -1.4502e-01,  6.8787e-01,\n",
       "           7.6811e-01,  3.6459e-02, -5.9048e-01, -4.3508e-02, -1.0246e-01,\n",
       "          -1.8006e-01,  4.1084e-01,  1.0845e-01, -2.2925e-01, -1.1876e-01,\n",
       "          -3.1507e-01, -9.5207e-02, -3.5142e-01,  3.5755e-01, -9.2964e-02,\n",
       "           6.9756e-01,  1.4460e-01,  7.2438e-02,  3.5404e-01,  3.9432e-01,\n",
       "          -2.9369e-01,  7.8359e-01,  2.2095e-01, -4.6333e-01,  1.8300e-01,\n",
       "          -7.5266e-02, -2.9379e-01,  6.5779e-01, -7.0699e-01,  2.5311e-01,\n",
       "           1.8474e-01,  8.1891e-01,  4.0170e-01]], device='mps:0'),\n",
       " tensor([[-0.2744, -0.4852, -0.4880,  ...,  0.2064,  0.2977,  0.6606],\n",
       "         [ 0.0506, -0.6230, -0.6446,  ..., -0.0875,  0.4018,  0.3101],\n",
       "         [-0.4643, -0.3750, -0.2078,  ...,  0.2835,  0.5093,  0.4110]],\n",
       "        device='mps:0'),\n",
       " tensor([[-0.0889, -0.8979, -0.5569,  ..., -0.0338,  0.5361,  0.3565],\n",
       "         [ 0.0370, -0.6937, -0.3180,  ...,  0.1541,  0.3871,  0.4194],\n",
       "         [-0.1689, -0.4450, -0.6101,  ...,  0.4169,  0.1256,  0.6819]],\n",
       "        device='mps:0'),\n",
       " tensor([[-0.3664, -0.4919, -0.5589,  ...,  0.3709,  0.5427,  0.4708],\n",
       "         [-0.2592,  0.0925, -0.6365,  ..., -0.1192,  0.3528,  0.0108]],\n",
       "        device='mps:0'),\n",
       " tensor([[-1.2655e-01, -7.4520e-01, -7.3208e-01, -2.8851e-01, -4.0142e-01,\n",
       "          -1.1156e-01, -1.7252e-01,  5.3901e-01,  7.7116e-02,  7.3915e-01,\n",
       "           1.7537e-01,  1.7667e-01, -2.6012e-02, -1.4756e-01, -3.8676e-01,\n",
       "          -3.7443e-01, -3.3357e-02,  2.8249e-01, -1.9368e-01, -2.2135e-01,\n",
       "           6.0315e-01,  4.3964e-01, -2.4971e-01,  8.9217e-02, -6.7105e-01,\n",
       "          -4.0270e-02,  1.1560e-01,  9.3162e-02,  8.8618e-02, -7.6946e-01,\n",
       "          -1.3872e-01,  6.3594e-01,  1.0944e-01,  1.1640e+00, -2.1243e-01,\n",
       "          -2.0144e-01,  6.2456e-02, -3.6353e-01,  3.5678e-01, -5.0662e-01,\n",
       "           6.4547e-01,  9.5906e-02,  1.0728e-01,  4.9590e-01,  3.0941e-01,\n",
       "          -9.9768e-02,  3.8522e-01, -9.3185e-01,  5.9163e-01, -5.7836e-01,\n",
       "          -4.5920e-01,  2.5531e-01,  3.5370e-01,  2.4685e-01, -5.3818e-01,\n",
       "          -3.4115e-01,  4.4585e-01,  4.0757e-01, -3.7288e-01,  2.8399e-01,\n",
       "          -2.7777e-01,  4.8607e-01, -9.5103e-02,  1.2683e-01,  3.3310e-01,\n",
       "           2.5259e-01, -3.6833e-01, -4.1241e-01, -3.6330e-01, -1.6269e-01,\n",
       "          -1.7272e-02,  3.9606e-01, -9.4364e-02, -4.2791e-01,  2.8922e-01,\n",
       "          -2.7780e-01, -2.6952e-02, -3.1996e-01, -5.3741e-01, -1.5066e-01,\n",
       "           2.3701e-01,  4.1333e-01,  1.2565e-01, -2.6415e-01,  1.6644e-01,\n",
       "           5.9443e-02,  3.9176e-01,  4.9960e-01,  2.5199e-01, -1.6932e-01,\n",
       "           7.0013e-01,  1.1491e-01, -4.5725e-01, -3.0624e-01,  4.2559e-01,\n",
       "          -1.8026e-01,  4.8127e-01, -2.2585e-01, -1.7340e-01, -2.0582e-01,\n",
       "           2.9714e-02,  1.4234e-01,  1.3911e-01, -1.5366e-01, -1.9418e-01,\n",
       "          -4.5523e-01,  9.2212e-02, -1.0325e-01, -7.3849e-01, -3.2521e-01,\n",
       "          -2.1782e-01,  1.0146e-01,  2.7080e-01, -6.0420e-01, -4.1818e-02,\n",
       "           5.6733e-03,  1.6661e-01, -2.6615e-02, -2.2333e-01,  5.1674e-01,\n",
       "           2.3316e-01, -3.5313e-01, -1.5852e-01,  2.8876e-01,  4.5897e-01,\n",
       "          -5.7108e-01,  7.3468e-01, -3.4321e-01,  3.1998e-01,  4.8956e-01,\n",
       "           9.4281e-02,  8.1907e-02, -4.1177e-01, -1.0418e-01,  2.1088e-01,\n",
       "          -6.6784e-02,  4.1300e-01, -4.0179e-01,  1.9421e-02, -3.8683e-02,\n",
       "           1.6348e-01, -1.7412e-01, -4.3610e-01, -1.1717e-02, -1.1581e-01,\n",
       "          -1.6510e-01, -1.8304e-01, -7.2298e-02, -2.9771e-01, -5.0788e-01,\n",
       "          -5.4121e-01, -5.5751e-01,  1.0337e-01, -2.3788e-01,  1.0912e-01,\n",
       "          -3.0113e-01,  2.7625e-01,  2.0260e-01, -8.1330e-03,  1.9222e-01,\n",
       "           7.9019e-01, -2.1016e-01,  2.6126e-01, -8.5382e-01, -7.1905e-02,\n",
       "          -4.8229e-02, -4.6440e-02, -1.7337e-01,  8.4039e-03,  4.1042e-02,\n",
       "           3.5682e-01, -2.8487e-01, -9.8864e-02, -4.7375e-01,  1.5659e-01,\n",
       "          -5.6279e-01,  4.2615e-01,  1.3419e-01, -4.2396e-02, -1.4768e-01,\n",
       "           9.7335e-02, -1.8931e+00, -1.5465e-01, -2.6779e-02, -1.4878e-01,\n",
       "          -4.4806e-01,  3.9898e-01,  3.0941e-01,  3.3377e-01, -1.3627e-01,\n",
       "          -6.6707e-01,  9.2625e-02,  2.3287e-02,  1.6853e-01,  1.2024e-01,\n",
       "           4.5454e-01,  4.6832e-01,  9.4002e-03,  2.9393e-02,  3.2991e-01,\n",
       "          -6.1158e-01,  2.3558e-01,  4.2741e-02, -2.0119e-01,  7.1801e-02,\n",
       "           1.7236e-01, -2.4868e-01,  1.9310e-01,  1.3628e-01,  1.6636e-01,\n",
       "          -2.4036e-01, -2.2848e-01, -1.5175e-01, -2.2029e-02, -4.4728e-01,\n",
       "           1.2618e+00,  4.6907e-02, -1.5209e-01,  4.7050e-01,  2.5533e-01,\n",
       "          -1.1167e-01, -4.1617e-02,  8.5119e-02, -2.4530e-01, -9.3295e-02,\n",
       "           6.0232e-01, -5.8611e-01, -9.9963e-02, -1.4179e-01, -2.6297e-01,\n",
       "          -1.9659e-02, -6.8554e-01,  6.7727e-01, -2.9161e-01,  2.3121e-02,\n",
       "          -2.3610e-01, -5.6084e-02, -3.2733e-03, -2.4264e-01,  1.1758e-01,\n",
       "          -5.1666e-01, -5.9031e-01, -8.7811e-01,  2.3734e-01,  2.1112e-01,\n",
       "           3.2798e-02, -2.9125e-01,  8.7574e-01,  4.0826e-01,  5.6717e-01,\n",
       "          -2.9647e-02, -2.5191e-01,  3.9790e-01, -2.7886e-01, -1.5353e-01,\n",
       "          -3.6532e-02, -1.0843e-01, -2.4601e-01,  4.9387e-01, -1.4118e-01,\n",
       "          -1.1985e-01, -1.7838e-01,  2.8209e-01,  7.1989e-01, -1.3719e-01,\n",
       "           1.5138e-01,  1.4714e-01, -4.7592e-01, -2.4229e-01,  4.8534e-01,\n",
       "          -9.8380e-02,  1.7773e-01, -2.5375e-01, -5.9610e-01, -5.0816e-01,\n",
       "          -4.2394e-01,  3.6723e-02, -4.1870e-01,  4.6948e-01, -7.8169e-01,\n",
       "          -4.6660e-01,  2.7831e-01, -1.4913e-01,  9.1576e-02, -2.5437e-01,\n",
       "           4.2342e-01,  1.6156e-01,  3.1737e-01, -2.6485e-01, -5.5156e-01,\n",
       "          -5.8013e-02,  2.6101e-01, -6.4054e-02, -2.2062e-02,  1.4868e-01,\n",
       "          -1.6116e-02,  3.8746e-01, -2.7389e-02, -5.7911e-02, -2.5913e-01,\n",
       "           2.2320e-02, -1.8898e-01, -1.6678e-01, -1.5143e-01,  1.2601e-01,\n",
       "           1.4057e-02,  1.0538e-01,  1.8138e-01, -5.9508e-01, -7.2537e-01,\n",
       "           1.5597e-02,  5.9317e-01, -1.4668e-01, -1.0549e-01, -2.0807e-01,\n",
       "          -8.5347e-01, -4.2241e-01, -2.0503e-01, -2.0471e-01, -4.2933e-02,\n",
       "           1.1471e-01,  1.5247e-01, -8.3550e-02, -2.9233e-02, -4.9752e-01,\n",
       "           2.7221e-01,  3.8839e-01,  6.6993e-01,  1.1004e-01, -1.4645e-01,\n",
       "          -2.8724e-01, -2.9969e-01, -2.9549e-01, -6.2650e-03, -2.0145e-01,\n",
       "           2.8245e-01,  2.1621e-01, -5.0797e-01, -2.7820e-01, -8.5465e-02,\n",
       "           5.3849e-01, -5.6729e-01, -5.8864e-01,  1.3343e-01, -3.1772e-01,\n",
       "          -6.6511e-01,  1.3537e-01, -6.8609e-01, -7.7676e-02,  1.6837e-01,\n",
       "           3.0898e-01, -3.5734e-01,  1.5134e-01, -3.4125e-01,  7.0115e-01,\n",
       "           5.0160e-01,  4.2641e-01, -4.4835e-01,  3.9933e-01,  4.5206e-01,\n",
       "          -5.1036e-01,  1.6760e-02, -2.3424e-01, -1.6616e-01, -1.3416e-01,\n",
       "           1.4652e-01,  3.2764e-01, -6.9834e-01,  1.5715e-01,  3.1416e-01,\n",
       "           5.0575e-02, -1.9972e-01, -3.9472e-01,  2.8528e-01, -3.0174e-01,\n",
       "          -1.8285e-01, -3.6691e-01,  1.1957e-01, -1.7406e-01, -3.2604e-01,\n",
       "           2.0167e-01,  7.2607e-02, -6.4363e-02, -2.7870e-01, -1.4528e-01,\n",
       "          -2.5803e-01,  3.0054e-01,  7.2737e-01,  7.0952e-02,  2.7148e-01,\n",
       "           1.0595e-01, -5.6026e-02, -1.7841e-01,  4.6604e-01,  5.7957e-02,\n",
       "          -2.9957e-01,  2.5715e-01, -4.8969e-02,  2.3108e-01,  7.1128e-01,\n",
       "          -3.2841e-01, -4.9112e-01,  9.4450e-02, -1.2226e-01, -2.4214e-01,\n",
       "          -3.2170e-01,  2.9251e-01,  3.9917e-01,  1.2972e-01,  7.2342e-02,\n",
       "          -2.2402e-01,  6.4680e-01, -3.7128e-01, -2.2534e-01, -2.1277e-01,\n",
       "           1.5045e-01, -2.3928e-02, -3.7103e-02, -5.8279e-02, -3.2639e-01,\n",
       "           3.5380e-01,  2.3950e-01,  4.7152e-01,  1.3765e-01, -1.1137e+01,\n",
       "           2.3304e-01,  8.5987e-01, -5.3979e-01,  2.0917e-01,  1.1097e-01,\n",
       "           3.8024e-01, -2.2548e-01,  1.9945e-01, -2.2175e-01, -2.0747e-01,\n",
       "           1.0544e-01, -1.0698e-01,  2.6923e-01, -5.2697e-01,  2.6734e-01,\n",
       "          -9.4496e-02, -6.8301e-01,  4.8033e-01, -3.7921e-01,  5.5370e-01,\n",
       "          -4.9093e-01,  3.7755e-01,  2.4613e-01, -2.4723e-01, -5.5835e-01,\n",
       "          -1.1016e-01, -5.9247e-01,  2.0945e-01,  6.5333e-01, -1.2674e-01,\n",
       "           7.5986e-02,  1.2314e-01,  3.9375e-01, -2.7636e-01, -7.2495e-01,\n",
       "           3.3798e-01, -2.5869e-01,  2.7356e-01,  2.8488e-01, -3.2794e-01,\n",
       "           1.7858e-01, -3.0247e-01,  6.9807e-01,  2.2586e-01,  7.0655e-01,\n",
       "           1.0020e+00,  7.1939e-01, -3.5037e-02, -6.7434e-01,  1.0608e-01,\n",
       "          -7.8312e-01,  4.9115e-02, -7.6924e-01,  1.9184e-01, -1.7609e-01,\n",
       "           4.8911e-01, -5.8598e-01,  8.3167e-02,  9.0617e-02, -5.0307e-03,\n",
       "           1.6894e-01, -1.6678e-01,  2.9473e-01, -2.3873e-01, -1.1305e-01,\n",
       "           3.5025e-01, -1.2861e-01,  2.4635e-01,  1.6825e-01, -4.8248e-01,\n",
       "          -3.1677e-01,  4.1630e-01,  9.9644e-02, -6.3425e-01,  2.9325e-01,\n",
       "           3.9251e-02, -2.3113e-02,  5.3637e-01, -4.7048e-02, -1.3664e-01,\n",
       "           3.8503e-01,  4.1584e-01, -1.0700e-01, -5.4898e-01, -3.8213e-01,\n",
       "          -1.1865e-01,  7.3741e-02,  2.9014e-01,  5.7460e-01, -3.2655e-01,\n",
       "          -3.7755e-01, -3.0112e-01,  1.5459e-01, -1.8642e-03,  1.0263e+00,\n",
       "          -7.1321e-01,  5.3698e-01,  3.1682e-01, -2.2188e-01, -4.7782e-01,\n",
       "          -4.2439e-02, -1.6548e-01, -3.3178e-01,  1.4035e-01, -3.9319e-01,\n",
       "           5.2460e-02,  3.1456e-01,  1.7819e-01,  4.4850e-01, -1.3422e-01,\n",
       "           3.7151e-01,  2.0540e-01,  1.0344e-01, -2.9791e-01,  1.8818e-01,\n",
       "           3.0356e-01,  7.1717e-01, -4.2188e-01, -1.9580e-01, -4.0693e-01,\n",
       "           7.0743e-02,  5.6798e-01,  4.0898e-02, -3.8230e-01,  9.8394e-02,\n",
       "          -7.2955e-01, -9.2036e-02, -2.4510e-01, -3.1327e-02,  4.7172e-01,\n",
       "           1.5715e-01, -2.1307e-02,  2.0088e-02, -2.1966e-01, -3.9113e-01,\n",
       "          -9.2195e-01, -1.6279e-01,  1.2524e-01, -4.8603e-02,  2.1508e-01,\n",
       "          -8.5762e-02, -1.4127e-01, -3.4117e-02,  6.5544e-01, -1.5354e-02,\n",
       "           7.1977e-01,  2.2196e-01, -1.3854e-01,  1.5714e-01, -6.1457e-02,\n",
       "          -4.3722e-01, -2.1751e-01, -2.9514e-01, -1.8853e-01, -2.8292e-01,\n",
       "           6.0776e-02,  1.7421e-01,  1.5628e-01, -3.5800e-01,  4.9556e-01,\n",
       "           5.3650e-03, -1.7080e-01, -1.1970e-01,  6.6134e-02,  1.4497e-01,\n",
       "           1.0977e-01, -3.4114e-01, -2.4418e-01, -9.5924e-02,  7.2654e-02,\n",
       "           2.3113e-01, -4.7076e-01,  3.5924e-02, -5.8183e-01,  1.4300e-01,\n",
       "           1.8358e-01,  1.3775e+00, -3.3167e-02, -5.4050e-01,  1.5789e-01,\n",
       "           6.0053e-01, -3.3132e-01,  1.5013e-01, -1.6961e-02,  2.0315e-01,\n",
       "          -4.3932e-01, -1.6432e-01,  3.2769e-01, -9.7204e-02,  2.7980e-01,\n",
       "           6.6650e-03,  2.8495e-01, -4.1759e-01,  3.4674e-01,  1.1297e-01,\n",
       "          -3.7993e-01, -1.3553e-01, -5.6543e-01, -3.4098e-02,  6.2812e-01,\n",
       "           3.0632e-02, -1.5477e-01, -2.2456e-01,  4.4538e-03, -1.7448e-01,\n",
       "           4.5311e-01,  1.2193e+00,  2.2650e-02,  4.3392e-02,  4.1220e-01,\n",
       "           1.3233e-02, -3.5122e-01,  2.5082e-01,  7.7446e-02,  6.5709e-02,\n",
       "           4.1433e-01,  2.8429e-01, -3.9413e-01,  3.2738e-01,  1.8144e-01,\n",
       "          -1.2196e-02, -8.6635e-01, -7.4945e-01, -1.8081e-01,  2.9443e-01,\n",
       "           2.3014e-01,  3.1770e-01, -6.9946e-02,  5.6758e-01,  8.6579e-01,\n",
       "          -1.0240e+00, -3.6591e-01,  2.1984e-01,  1.1269e-01, -1.3523e-01,\n",
       "          -2.8316e-01,  3.6406e-02, -6.6481e-02, -4.8127e-01,  2.5986e-01,\n",
       "           7.4845e-01, -8.2243e-01,  1.0108e-01,  3.3324e-01, -4.8112e-01,\n",
       "          -4.6681e-02,  1.0242e-02,  2.7858e-02,  3.4290e-01,  4.9584e-02,\n",
       "          -5.8812e-01,  5.7447e-01, -4.3146e-01, -5.1602e-01,  3.3867e-01,\n",
       "           6.1310e-01, -1.7420e-02, -1.5538e-02,  2.5800e-01, -1.9679e-01,\n",
       "           1.0622e-01, -3.8051e-01, -3.0642e-01, -4.1422e-02, -3.3001e-01,\n",
       "           4.2411e-02, -1.9060e-01,  3.5958e-01,  1.5858e-01, -6.1822e-01,\n",
       "           2.4092e-01,  7.2764e-01,  8.2494e-01, -4.0542e-02, -4.1354e-01,\n",
       "           4.3800e-01, -4.4073e-02, -3.2619e-01,  4.6795e-01, -3.7749e-02,\n",
       "          -1.7259e-01,  3.5766e-01,  1.9563e-01, -2.8715e-01, -1.5703e-01,\n",
       "          -2.8124e-01,  2.8023e-01, -6.0972e-02, -3.8436e-01,  2.7954e-01,\n",
       "          -1.8757e-01,  3.1244e-01, -6.0115e-01,  1.0007e-01,  3.1193e-01,\n",
       "          -6.1689e-02,  7.0562e-01,  1.3887e-01,  4.2675e-01, -1.3264e-01,\n",
       "           1.2326e-01, -4.0292e-01, -4.7626e-01,  1.4668e-02,  1.1085e-01,\n",
       "          -3.0529e-01,  2.2641e-01,  1.2448e-01,  3.0816e-02,  7.6298e-01,\n",
       "           7.3448e-01, -4.3314e-02, -6.5195e-01,  4.6754e-02,  2.6127e-02,\n",
       "           1.4634e-01, -1.0814e-01,  3.7987e-01, -2.9931e-01, -7.7688e-02,\n",
       "          -8.8698e-01, -3.9418e-01,  1.6635e-01,  2.5230e-01, -2.3809e-01,\n",
       "           5.2611e-01, -1.3072e-01,  2.8637e-01,  3.2407e-01,  7.9897e-02,\n",
       "           3.1870e-01,  9.1416e-01,  2.1795e-01, -5.4236e-01,  2.1027e-01,\n",
       "           5.3772e-01, -1.5431e-01, -3.0445e-01, -8.9030e-01,  2.3249e-01,\n",
       "           1.3295e-02,  7.6987e-01,  1.5070e-01]], device='mps:0'),\n",
       " tensor([[-3.0175e-01, -6.6980e-01, -5.8058e-01, -1.0169e+00, -4.7769e-01,\n",
       "          -1.2967e-03, -6.6155e-01,  2.1546e-01,  1.0829e-01,  2.8188e-01,\n",
       "           2.6802e-01, -7.4553e-03,  8.8102e-02,  1.2687e-02, -2.8827e-01,\n",
       "           2.5617e-01, -2.6220e-01,  2.1438e-01,  7.7033e-02, -3.7727e-02,\n",
       "           8.1075e-01,  3.6768e-01, -1.6212e-01, -2.0573e-01, -3.3086e-01,\n",
       "           4.5605e-01, -1.3148e-01,  4.8222e-02,  3.5668e-01, -7.9078e-01,\n",
       "          -2.5202e-01,  1.1725e-01, -2.9749e-01,  1.0378e+00, -1.1594e-01,\n",
       "          -3.3333e-01, -1.7748e-01, -4.3787e-01,  2.1301e-01, -5.0568e-02,\n",
       "           4.5755e-01, -1.4778e-01, -2.4105e-01,  2.0005e-01, -1.2606e-02,\n",
       "           9.5042e-02,  3.7443e-01, -8.8176e-01,  3.3878e-01, -1.1266e-01,\n",
       "           1.8889e-01, -3.4865e-01,  5.0318e-01,  2.1380e-01, -8.5959e-01,\n",
       "           5.2285e-02,  2.7071e-01, -4.0013e-02, -8.2586e-01,  1.2128e-01,\n",
       "          -4.6670e-01,  5.9963e-01, -1.1233e-01,  1.0830e-01,  1.9111e-01,\n",
       "          -1.8788e-01, -5.8713e-01,  1.3197e-01, -2.4455e-01, -5.4474e-02,\n",
       "          -5.4218e-03,  6.5278e-03,  1.1025e-02,  5.1945e-02, -1.4253e-01,\n",
       "          -4.2182e-01, -1.2333e-01, -1.5679e-01, -4.8684e-01, -2.8943e-01,\n",
       "           1.1703e-01,  2.2252e-01, -9.0303e-02, -2.5004e-01,  1.6458e-01,\n",
       "          -2.2152e-02, -1.1224e-01,  2.6949e-01,  1.3030e-01,  8.0149e-02,\n",
       "           3.0522e-01,  2.0928e-01, -1.5859e-02, -2.0004e-01,  5.0817e-01,\n",
       "          -3.4280e-01,  5.8038e-01, -1.4811e-01, -1.0314e-01, -1.4254e-01,\n",
       "          -8.5900e-02,  1.6763e-01,  2.0530e-01,  3.5774e-01, -1.2954e-01,\n",
       "          -2.2287e-01, -1.0609e-01, -1.5258e-01, -6.7798e-01, -3.9055e-02,\n",
       "          -2.1100e-01, -3.5712e-01, -7.7076e-02, -6.5853e-01, -4.8448e-01,\n",
       "           3.3116e-01,  8.6550e-02, -3.0879e-01, -6.8888e-02,  6.9461e-01,\n",
       "           4.5933e-01, -4.6570e-01,  9.2314e-02, -2.0174e-01,  3.6437e-01,\n",
       "          -2.6446e-01,  3.1500e-01, -2.6301e-01,  1.6895e-01,  3.9658e-01,\n",
       "           2.4446e-01,  1.7780e-01, -2.5048e-01, -9.5926e-02, -4.5958e-03,\n",
       "          -2.0377e-01,  3.0765e-02, -1.7716e-01,  2.5017e-01, -1.6240e-01,\n",
       "          -1.1815e-01, -1.9138e-01, -3.8586e-01, -2.9267e-01,  2.0262e-02,\n",
       "          -3.8745e-01,  1.7052e-01,  2.0249e-01,  2.3481e-01,  2.0374e-01,\n",
       "          -5.9123e-01, -3.7090e-01,  2.6279e-01, -1.5877e-01, -1.7998e-01,\n",
       "          -2.7334e-01,  4.8190e-01,  1.1280e-01,  3.4257e-02,  4.9299e-01,\n",
       "           4.8370e-01, -3.4823e-01,  3.8388e-01, -5.6908e-01,  4.7970e-01,\n",
       "          -8.0979e-02,  2.2367e-02, -3.4129e-01,  6.6072e-01, -2.2073e-01,\n",
       "           4.2082e-01,  4.4552e-02, -5.8999e-02, -4.2000e-01,  2.1493e-01,\n",
       "           3.2119e-01,  5.5215e-01,  2.8715e-01, -2.9960e-01, -5.3479e-02,\n",
       "           5.6045e-02, -1.5361e+00,  1.7164e-01, -9.1554e-01, -2.4568e-01,\n",
       "          -2.9564e-01,  4.7877e-01, -1.0528e-01,  5.3352e-01,  2.0833e-01,\n",
       "           1.5799e-01,  2.2647e-01,  4.1759e-01,  6.8457e-01,  1.3707e-01,\n",
       "           3.0360e-01,  6.7200e-01,  5.1151e-01,  1.5972e-01, -9.9524e-02,\n",
       "          -8.4006e-01,  4.5848e-01, -2.4269e-01,  4.0564e-01, -4.0343e-01,\n",
       "           2.8765e-01, -4.5223e-01,  2.4525e-01,  5.3122e-01,  6.8031e-02,\n",
       "          -1.8708e-01, -3.3998e-01, -5.5491e-01, -1.4993e-01, -5.1624e-01,\n",
       "          -6.4144e-02, -3.5381e-01, -1.2625e-01,  2.2048e-01,  2.4657e-01,\n",
       "           2.5758e-01, -1.0877e-01,  8.6218e-02, -4.4103e-01, -2.5500e-01,\n",
       "           4.7314e-01, -4.2033e-02, -2.8710e-01, -2.4144e-01, -2.3728e-01,\n",
       "           2.8834e-02, -6.8555e-01,  5.9320e-01,  5.4095e-01,  2.8945e-01,\n",
       "          -3.7894e-01, -2.6274e-01,  3.3479e-01, -8.1051e-02, -1.9290e-01,\n",
       "          -3.2112e-01, -2.8164e-01, -6.3114e-01,  3.0389e-02,  3.2483e-01,\n",
       "           4.5472e-01,  2.2502e-02,  5.9776e-01,  9.2431e-01,  6.2616e-01,\n",
       "          -6.3152e-02,  4.1419e-01,  3.5893e-01,  4.0719e-01, -3.6893e-01,\n",
       "           2.3675e-01,  2.5578e-02, -3.0394e-01,  1.8930e-01, -3.4051e-01,\n",
       "          -5.4999e-01, -3.0810e-01,  1.8355e-01,  3.9312e-01, -4.0218e-01,\n",
       "           1.3738e-01,  3.6324e-01, -9.4898e-02, -2.0045e-01,  2.3112e-01,\n",
       "          -4.5769e-01,  2.7052e-01, -3.8193e-02, -2.4676e-01, -7.8228e-01,\n",
       "          -8.0561e-01,  1.7671e-01,  4.0383e-01,  2.7239e-02, -6.2392e-01,\n",
       "           4.0250e-01, -7.7847e-02, -1.6572e-01,  5.7036e-02, -2.0384e-01,\n",
       "           5.1000e-01, -2.4829e-01, -4.3042e-02,  2.3431e-01, -5.3414e-02,\n",
       "          -2.6091e-01,  2.8407e-01, -2.1272e-01, -1.4648e-01,  2.0817e-01,\n",
       "           1.1829e-01,  2.4851e-01,  2.9531e-01, -1.8469e-01, -1.0712e-01,\n",
       "          -1.4378e-01,  1.4497e-01, -3.1745e-01, -1.1662e-01,  2.7385e-01,\n",
       "           4.4367e-01, -8.9016e-02,  5.4896e-01, -2.2052e-01, -4.3292e-01,\n",
       "          -5.1639e-02,  7.8045e-01, -2.5523e-01,  4.0706e-01, -4.3595e-01,\n",
       "          -1.0067e+00, -2.3247e-01, -2.4278e-01, -1.6148e-01, -4.9052e-01,\n",
       "           3.3004e-01, -2.3283e-01,  9.8914e-02, -2.6420e-02, -2.8987e-01,\n",
       "           3.0162e-01,  6.6029e-01,  4.9695e-01, -3.8492e-02, -2.6121e-01,\n",
       "          -9.2459e-02, -1.8654e-01, -6.9434e-01, -2.8268e-01, -3.3254e-01,\n",
       "           2.9086e-01,  5.4987e-01, -9.4563e-03, -3.4511e-02, -2.2907e-01,\n",
       "           6.7731e-01,  4.2296e-02, -2.2174e-01,  4.3139e-01, -2.6503e-01,\n",
       "          -3.1949e-02,  4.2142e-01,  2.4983e-01,  1.4853e-01,  2.7700e-02,\n",
       "           3.8850e-01, -2.4014e-01, -5.2190e-01,  2.2986e-02,  7.3507e-01,\n",
       "           2.4180e-01,  4.1704e-01, -1.2385e-01,  6.9499e-01,  1.1516e-01,\n",
       "          -4.0155e-01,  1.9317e-01, -3.4905e-01, -9.7611e-02, -4.1392e-02,\n",
       "           2.1507e-01,  1.7865e-01, -4.6424e-01, -1.4887e-01, -3.2423e-01,\n",
       "           5.4446e-01, -4.4284e-01,  1.0868e-01,  6.0624e-01,  9.5384e-02,\n",
       "          -8.9992e-02, -5.1526e-01,  4.1846e-02, -9.2054e-01, -4.2575e-01,\n",
       "           1.3709e-03, -6.5500e-02,  3.1680e-01, -1.4904e-01, -6.4729e-02,\n",
       "          -9.6847e-03, -2.5828e-01,  6.6431e-01,  2.4458e-01,  2.2317e-02,\n",
       "           2.4482e-02, -4.2979e-01, -1.6413e-01,  7.3132e-01, -2.8516e-01,\n",
       "          -2.0146e-01,  2.4408e-01, -1.7036e-01,  7.5170e-02,  9.1916e-01,\n",
       "          -5.4459e-01, -1.2919e-01,  1.6822e-01, -9.2317e-02, -4.5496e-01,\n",
       "          -2.6936e-01,  3.7161e-01,  3.1036e-01, -2.3544e-01,  9.3051e-03,\n",
       "          -1.9547e-01, -4.8565e-02, -1.3911e-01,  1.4568e-01,  4.0195e-01,\n",
       "           3.8214e-01, -3.3720e-01, -9.5011e-02, -2.3665e-01, -1.5221e-01,\n",
       "           5.8983e-01,  4.1325e-02,  5.0294e-02,  4.4798e-01, -1.1088e+01,\n",
       "           3.1967e-01,  8.8400e-01, -4.8213e-01, -3.6244e-01,  1.7139e-01,\n",
       "           2.1689e-01, -3.2129e-01,  5.4136e-01, -3.4162e-01, -4.5191e-01,\n",
       "          -3.6412e-01, -2.6337e-01, -5.2896e-02, -5.3673e-01, -2.6602e-01,\n",
       "          -4.5073e-02, -6.6599e-01,  4.9827e-01, -5.6950e-02,  4.1197e-01,\n",
       "          -7.3141e-01,  2.9915e-01,  2.4368e-01, -2.5628e-01, -5.1343e-01,\n",
       "          -4.1890e-01,  5.6851e-02, -5.3908e-02,  1.5332e-01, -1.4587e-02,\n",
       "          -7.8227e-02, -1.4075e-01,  3.3179e-01, -1.0998e-01, -6.0080e-01,\n",
       "          -2.6099e-01, -1.5141e-01,  3.0497e-01,  3.4405e-01, -1.6240e-02,\n",
       "           1.4509e-01, -6.8089e-02,  4.0110e-01,  3.9231e-01,  8.2129e-01,\n",
       "           1.4654e+00,  3.3271e-01,  1.3907e-01,  1.6580e-01, -3.7775e-01,\n",
       "          -7.0736e-01,  3.5929e-01, -8.6010e-01,  1.6285e-02,  1.0245e-01,\n",
       "           4.0998e-01, -5.2030e-01,  1.8141e-01, -1.0184e-01, -2.9502e-01,\n",
       "           4.7893e-01, -4.8131e-02,  3.6590e-01,  6.5942e-02, -2.0825e-01,\n",
       "           3.6483e-01,  9.8140e-03,  1.0428e-01, -2.6566e-01,  7.0489e-02,\n",
       "          -1.0515e-01,  9.0629e-01,  5.6924e-02,  2.2025e-01,  3.5505e-01,\n",
       "           1.9906e-01,  5.6627e-02,  9.7982e-01,  1.4887e-02,  2.3439e-02,\n",
       "           2.4081e-01,  1.2057e-01, -8.2163e-01, -6.6525e-01, -3.0672e-01,\n",
       "          -1.9990e-01,  2.9884e-01,  2.1077e-01,  4.5599e-01,  1.0191e-01,\n",
       "          -2.4030e-01, -4.7529e-01, -8.4384e-02,  3.4405e-01,  2.7867e-01,\n",
       "          -8.1985e-01, -2.7656e-01,  3.8855e-01,  5.9943e-03, -3.3150e-01,\n",
       "          -1.0185e-01,  3.9454e-02, -6.9953e-01,  2.8364e-01, -8.8777e-01,\n",
       "           4.4082e-01, -1.1780e-01,  1.2180e-01,  3.3622e-01, -2.0828e-01,\n",
       "           1.4357e-01,  2.4272e-01,  1.5712e-01, -1.2949e-01, -3.2449e-01,\n",
       "          -6.2829e-02,  5.6284e-01, -3.0667e-01, -6.1434e-01,  7.8437e-02,\n",
       "          -4.1754e-01, -3.2532e-02,  2.0604e-01, -4.0847e-01,  2.9599e-01,\n",
       "          -9.2509e-01,  5.7991e-01,  5.3225e-01,  1.3132e-01,  4.2593e-01,\n",
       "           1.0221e-01,  2.7063e-01,  3.5230e-01, -3.5712e-01, -6.5350e-01,\n",
       "          -4.5414e-01,  1.1263e-01,  4.4292e-01, -4.3713e-01, -6.4580e-02,\n",
       "          -2.5887e-01, -1.8874e-02, -3.6240e-01,  4.6295e-01, -4.5580e-01,\n",
       "           5.0356e-01,  2.7012e-01, -3.2513e-01,  8.5783e-02, -9.8713e-02,\n",
       "          -3.8531e-02, -3.5799e-02, -9.5866e-02, -2.5031e-01, -6.3330e-01,\n",
       "          -2.7211e-01,  6.7283e-02,  1.7325e-01, -4.6052e-01, -4.7682e-01,\n",
       "          -2.3481e-01, -3.3311e-01, -3.1033e-01,  5.6504e-01,  4.5932e-01,\n",
       "           3.9209e-03, -2.8593e-01, -8.9424e-02,  6.8840e-02,  1.4673e-01,\n",
       "           5.1449e-01,  1.8069e-02, -1.3468e-01, -1.1048e+00, -2.6467e-01,\n",
       "          -2.6220e-01,  1.1618e+00, -5.0380e-01, -2.3770e-01,  1.6040e-01,\n",
       "          -3.5048e-01, -1.3186e-01,  3.4729e-01, -4.3105e-01, -2.8086e-01,\n",
       "          -3.8445e-02,  3.7695e-02,  4.9161e-01, -1.4846e-01,  4.3220e-01,\n",
       "           3.6677e-01,  1.2157e-01, -2.9789e-01,  1.2610e-01, -5.6793e-01,\n",
       "          -5.4133e-01,  1.5364e-01, -3.2761e-01,  2.5915e-01, -1.4858e-01,\n",
       "          -4.2085e-01,  4.1585e-02, -2.8757e-01,  3.6606e-02,  6.1003e-01,\n",
       "           1.4671e-01,  1.3806e+00, -3.4816e-01, -7.2524e-02,  2.0323e-02,\n",
       "           1.1091e-02,  2.8904e-01, -4.7353e-02, -2.8690e-01,  4.2239e-01,\n",
       "           3.8130e-01, -5.1135e-01, -2.1610e-02,  2.4837e-01,  8.4575e-02,\n",
       "           1.5806e-01, -7.0628e-01, -5.6971e-01, -4.5014e-01,  8.3924e-01,\n",
       "           4.2194e-01,  6.0600e-02, -1.3581e-01,  3.6786e-01,  6.1553e-01,\n",
       "          -7.2307e-01,  1.2407e-01, -5.6063e-01,  3.4624e-01, -2.8673e-02,\n",
       "          -3.8840e-02, -1.4214e-01, -4.1968e-01, -2.4922e-01,  5.1180e-01,\n",
       "           8.8041e-01, -8.0026e-01,  2.7926e-01,  2.8429e-01, -6.0185e-01,\n",
       "           2.7400e-01, -3.0303e-01,  4.9336e-01,  2.6875e-01, -1.5132e-01,\n",
       "           7.2616e-02,  3.0180e-01, -8.4330e-02, -4.9152e-01,  4.2695e-02,\n",
       "           6.6296e-01, -1.1219e-01,  1.9635e-01, -7.4417e-02, -4.0446e-01,\n",
       "           1.0495e-01,  3.4093e-02, -4.6593e-01, -1.8069e-01,  2.7051e-01,\n",
       "          -1.6951e-01, -5.1150e-01,  3.0621e-02,  3.2784e-01, -9.6532e-02,\n",
       "           3.0489e-01,  6.1836e-01,  1.8625e-01,  2.8380e-01, -2.6720e-01,\n",
       "          -1.7008e-01,  2.8282e-01, -3.1094e-01,  4.9413e-01, -2.5900e-01,\n",
       "          -2.8052e-01,  5.1395e-01, -5.1076e-02, -2.3065e-01,  1.3898e-01,\n",
       "          -1.9188e-01,  1.2284e-01, -2.0852e-01, -8.5311e-02,  1.3929e-01,\n",
       "           1.5890e-01, -2.0449e-01, -3.2401e-01, -1.5393e-01,  2.0152e-01,\n",
       "          -3.0676e-01,  6.9643e-01, -5.2115e-02,  2.7670e-01, -3.2876e-01,\n",
       "           3.2702e-01, -6.1801e-01, -3.5319e-01,  3.7352e-01,  4.6685e-03,\n",
       "          -4.5457e-01, -7.3337e-01,  3.5639e-01,  1.9663e-01,  6.8855e-01,\n",
       "           4.1244e-01, -4.6492e-02, -4.1667e-01, -3.7578e-01,  4.9655e-02,\n",
       "          -5.1082e-02, -1.9918e-01,  4.2887e-02,  2.6552e-01, -3.0464e-03,\n",
       "          -7.6106e-01, -5.7014e-01,  5.2966e-02,  3.2732e-01, -1.7900e-01,\n",
       "           7.9234e-01, -4.4085e-01,  6.0370e-02,  6.0840e-01,  4.8676e-01,\n",
       "           1.1522e-01,  7.3154e-01,  1.1035e-01, -7.9930e-01,  5.9814e-04,\n",
       "          -1.4466e-01, -2.9386e-01,  2.4061e-01, -9.9708e-02,  2.6837e-01,\n",
       "           3.2979e-01,  1.6883e-01,  7.6565e-01]], device='mps:0'),\n",
       " tensor([[-9.8608e-02, -6.2949e-01, -5.4737e-01, -4.6335e-01, -3.7454e-01,\n",
       "           1.0248e-01, -2.8097e-01,  4.4707e-01, -1.3869e-01,  5.9022e-01,\n",
       "           3.3979e-01,  3.6518e-02, -7.9829e-02, -1.9712e-01, -4.0961e-01,\n",
       "          -5.4994e-01,  1.7904e-01,  2.6107e-01,  2.6766e-01, -1.3636e-01,\n",
       "           9.8026e-01,  3.6199e-01, -8.5372e-02,  1.8939e-01, -6.9872e-01,\n",
       "           3.9986e-01,  4.5490e-02,  8.0399e-02,  4.6963e-01, -6.9729e-01,\n",
       "          -2.0828e-01,  5.4662e-01, -1.2219e-01,  1.1649e+00, -1.8525e-01,\n",
       "          -2.5129e-01,  4.1644e-01, -2.6874e-01,  8.0973e-01,  4.0156e-02,\n",
       "           6.2909e-01, -1.2839e-01,  2.5974e-01,  2.1204e-01,  2.5654e-01,\n",
       "          -1.6577e-02,  1.0736e-01, -6.2638e-01,  5.4556e-01, -5.1867e-01,\n",
       "          -2.1843e-01, -1.5953e-01,  3.6604e-01,  4.8748e-01, -6.7281e-02,\n",
       "           3.5607e-02,  3.0541e-01,  1.6232e-01, -2.5763e-01,  2.0386e-01,\n",
       "           1.3393e-01,  6.9371e-01,  5.7586e-02, -1.9525e-01, -3.0287e-01,\n",
       "           3.0687e-01, -5.2685e-01, -7.0972e-01, -3.4711e-01, -1.5503e-01,\n",
       "          -6.3145e-02,  3.6453e-01, -2.3968e-01, -8.3683e-02,  7.8001e-02,\n",
       "          -8.8895e-02,  3.9835e-02, -1.7901e-01, -2.7637e-01, -9.8806e-03,\n",
       "           6.4246e-02,  3.2661e-01, -1.9744e-01, -4.1654e-01, -2.1876e-01,\n",
       "           3.0475e-01,  7.8369e-01,  6.3690e-02,  2.5669e-01,  8.5968e-02,\n",
       "           3.1750e-01,  9.4549e-02, -3.4083e-01, -4.9471e-01,  3.1401e-01,\n",
       "          -1.3255e-01,  4.9667e-01, -1.4940e-01, -6.5687e-02, -4.7700e-02,\n",
       "           3.7095e-01,  7.2304e-02,  1.7323e-03,  1.1325e-01, -6.6663e-02,\n",
       "          -4.8542e-01,  1.4204e-01, -2.5931e-01, -6.6520e-01, -1.1522e-01,\n",
       "          -7.9637e-02,  1.3277e-01,  1.5561e-02, -3.1430e-01,  2.6189e-01,\n",
       "          -7.3846e-02,  1.9145e-01, -1.3642e-02, -3.3838e-01,  3.7293e-01,\n",
       "           2.5609e-01, -4.0848e-01,  2.2958e-02,  1.6951e-01,  4.6388e-01,\n",
       "          -2.2394e-01,  6.0049e-01, -2.8694e-01,  2.5392e-01,  4.8866e-01,\n",
       "           8.4207e-02, -6.0898e-02, -9.6990e-02, -1.7751e-01, -3.9547e-01,\n",
       "          -4.2915e-02,  7.3082e-02, -3.2798e-01,  1.0024e-01,  2.1027e-02,\n",
       "          -1.5484e-01, -1.5575e-01, -7.1363e-01, -3.2433e-01,  1.6862e-01,\n",
       "          -2.9423e-01,  2.9391e-01, -7.2839e-02, -5.6183e-01, -3.6233e-01,\n",
       "          -8.1793e-01, -4.5420e-01,  5.0598e-01,  3.1144e-02, -1.4616e-01,\n",
       "          -4.4906e-02,  1.9702e-01,  1.8656e-01, -1.9001e-02,  3.6581e-01,\n",
       "           4.5356e-01, -1.7970e-01,  5.5773e-02, -4.7383e-01,  5.1199e-02,\n",
       "           3.6512e-02, -1.1536e-01, -1.4678e-01, -1.8904e-01,  8.2812e-02,\n",
       "           1.8413e-01, -2.0916e-01, -2.0209e-02, -4.3147e-01,  1.4387e-01,\n",
       "          -9.5740e-02,  8.7516e-02,  2.8605e-01, -3.1460e-01, -3.4357e-01,\n",
       "          -4.4457e-01, -2.0415e+00,  1.3037e-01, -1.6708e-01, -1.9640e-01,\n",
       "          -7.8661e-01,  5.6989e-01,  4.0038e-01,  2.3491e-01,  1.2055e-02,\n",
       "          -4.1441e-01, -1.3256e-01, -2.2065e-01,  2.0346e-01, -1.5937e-01,\n",
       "           4.1104e-01,  3.6856e-01,  2.1222e-01, -8.2505e-02, -5.3417e-02,\n",
       "          -3.3993e-01,  1.9027e-01,  6.6698e-02,  4.3789e-02,  1.3752e-02,\n",
       "           8.6470e-02, -3.4516e-01,  1.5865e-01,  6.3413e-01, -3.2035e-01,\n",
       "          -2.6413e-01, -1.5432e-01, -3.1116e-01, -1.1825e-01, -2.1011e-01,\n",
       "           7.5833e-01, -1.0075e-01, -1.9090e-01,  4.1462e-01,  2.9186e-01,\n",
       "          -2.0576e-01,  2.6479e-02, -1.7243e-01, -4.6025e-01, -8.4004e-02,\n",
       "           8.3166e-01, -2.6809e-01, -1.8435e-01, -1.3576e-01, -2.8072e-01,\n",
       "           9.9627e-02, -3.9657e-01,  7.0865e-01,  1.3131e-01, -2.4437e-01,\n",
       "          -1.3250e-01, -4.9085e-01, -3.6081e-02, -6.5221e-02,  2.7934e-01,\n",
       "          -3.2480e-01, -4.4413e-01, -6.2940e-01,  9.8728e-02,  4.2580e-01,\n",
       "           1.0494e-01, -2.2581e-01,  5.8588e-01,  3.7420e-01,  5.5104e-01,\n",
       "          -7.6877e-02,  1.8477e-03,  4.0139e-01,  4.0454e-02, -3.1153e-01,\n",
       "           4.0137e-02,  1.1839e-01, -6.1671e-02,  2.8916e-01, -5.0007e-01,\n",
       "           2.1460e-01,  1.3048e-01,  4.2285e-02,  4.0869e-01, -3.3156e-01,\n",
       "           3.3759e-01, -6.4578e-04, -4.2352e-01, -2.2792e-01,  3.1587e-01,\n",
       "          -3.4318e-01,  4.7882e-01, -1.1497e-01, -1.3533e-01, -6.5793e-01,\n",
       "          -3.0023e-01,  9.9027e-02, -5.2265e-01,  4.0820e-02, -8.5922e-01,\n",
       "          -1.4457e-01,  4.3467e-01, -1.1843e-01,  2.8682e-01, -4.3268e-01,\n",
       "           2.7079e-01,  1.1150e-01,  1.0583e-01, -1.8427e-01, -4.8035e-01,\n",
       "           6.4368e-03,  3.5471e-01, -1.9635e-01,  6.7501e-02,  1.8728e-02,\n",
       "           1.4821e-01,  7.4914e-02, -1.4411e-01,  1.8116e-02, -3.9952e-01,\n",
       "           1.2334e-01,  2.7083e-01, -6.4203e-02, -1.1696e-01, -2.5127e-01,\n",
       "           8.1457e-02,  5.7341e-02,  4.5826e-01, -1.5574e-01, -4.7792e-01,\n",
       "           1.3602e-02,  5.6159e-01,  4.2917e-02,  9.5691e-02, -4.1476e-02,\n",
       "          -1.3511e+00, -3.8201e-01,  1.3646e-01, -2.3684e-01, -1.6250e-01,\n",
       "           1.4728e-01, -4.5766e-02, -1.0990e-01,  1.3434e-01, -3.3727e-01,\n",
       "           4.9913e-01, -7.4794e-03, -9.4830e-02,  2.6318e-01, -2.9595e-01,\n",
       "          -2.6245e-01, -5.5022e-02, -5.5963e-02, -1.9048e-01, -2.1108e-01,\n",
       "           1.0760e-01,  2.4895e-01, -1.6610e-02,  2.9324e-01, -4.2315e-01,\n",
       "           6.9357e-01,  2.0403e-02, -3.6256e-01,  1.8028e-01, -4.6817e-01,\n",
       "          -2.2470e-01,  6.2020e-01, -2.1111e-01,  1.7508e-01,  7.0164e-02,\n",
       "           2.2583e-01, -1.6263e-01, -2.5922e-01, -1.5555e-01,  5.6515e-01,\n",
       "           9.7645e-02,  3.8631e-01, -3.1465e-01,  4.6445e-01,  6.0830e-01,\n",
       "          -4.3876e-01,  2.7575e-01, -3.0324e-01, -3.5509e-01, -1.4900e-01,\n",
       "           3.7937e-01,  2.4503e-01, -2.4568e-01,  1.2754e-01,  2.1491e-01,\n",
       "           2.4189e-01, -2.8123e-01, -1.4375e-01,  1.5623e-01, -1.6800e-02,\n",
       "          -1.6159e-01, -5.9860e-01,  4.8387e-02, -4.0540e-01, -3.7598e-01,\n",
       "          -5.5542e-02, -9.6940e-02,  2.3018e-01, -3.4557e-01, -2.8109e-01,\n",
       "          -4.1082e-01,  1.5328e-01,  7.9392e-01, -8.2566e-02,  1.3355e-02,\n",
       "          -2.4290e-01,  1.3134e-01, -3.9568e-01,  5.6959e-01,  2.4837e-01,\n",
       "          -5.6746e-02,  5.0092e-01, -5.9799e-01,  3.2502e-01,  1.6325e-01,\n",
       "          -2.5296e-01,  7.9720e-02, -1.1265e-01,  4.2524e-01, -1.9241e-01,\n",
       "          -2.6666e-01,  1.8088e-01,  2.9568e-01,  8.0022e-02,  3.3680e-02,\n",
       "           8.9946e-02,  7.7279e-01, -5.6155e-01, -1.1355e-01, -3.2412e-02,\n",
       "           4.6990e-01, -1.4868e-01,  1.7842e-01, -2.5184e-01, -4.5494e-01,\n",
       "           2.8096e-01,  7.6220e-03,  3.0259e-02,  1.5848e-01, -1.1470e+01,\n",
       "           1.7761e-02,  7.9956e-01, -5.9407e-01,  3.9142e-02, -7.9952e-04,\n",
       "           1.4375e-01, -2.8145e-01,  3.6361e-03, -2.5851e-01, -3.7282e-01,\n",
       "           1.6253e-01,  5.8747e-02,  6.8827e-02, -1.9252e-01, -1.2168e-01,\n",
       "          -4.4456e-02, -9.0133e-01,  1.4013e-01, -2.9238e-01,  3.1633e-01,\n",
       "          -4.6944e-01,  3.5652e-01,  1.0775e-01,  4.8818e-02, -3.3732e-01,\n",
       "          -6.3534e-02, -4.1684e-01,  2.3683e-01,  1.0364e-01, -1.6207e-01,\n",
       "          -1.3703e-01, -1.5828e-01,  4.6034e-01, -1.8993e-01, -4.8947e-01,\n",
       "           1.5862e-01, -7.0142e-02,  4.5043e-01, -1.6798e-02, -2.3410e-01,\n",
       "           5.0011e-02, -4.1456e-01,  3.0195e-01,  6.4668e-02,  5.0457e-01,\n",
       "           4.7572e-01,  6.3490e-01,  1.0739e-01, -2.1848e-01,  4.9891e-02,\n",
       "          -8.9429e-01,  3.3171e-01, -9.8824e-01, -2.6344e-01,  3.6779e-02,\n",
       "           7.4622e-01, -4.6007e-01,  9.9323e-03,  8.2971e-02, -3.9312e-01,\n",
       "           2.4107e-01, -5.5413e-02,  1.0686e-01, -6.2270e-01, -1.1555e-01,\n",
       "           3.6365e-01,  1.0164e-01,  1.3567e-01, -3.5856e-01, -1.3223e-02,\n",
       "          -4.3049e-02,  5.9870e-01,  1.5647e-01, -2.5151e-01,  1.0429e-01,\n",
       "          -8.0721e-03,  2.3224e-01,  6.2748e-01,  4.7218e-02,  1.3164e-01,\n",
       "           1.7459e-01,  5.1672e-01, -5.4105e-01, -4.7941e-01, -1.2530e-01,\n",
       "           6.3722e-02,  2.8461e-01,  8.3486e-02,  3.6163e-01, -3.9872e-01,\n",
       "          -3.1281e-01, -7.8561e-02,  3.1281e-02,  5.0728e-02,  1.2098e+00,\n",
       "          -5.8519e-01, -2.2122e-01, -1.0256e-01, -9.1470e-02, -1.3571e-01,\n",
       "          -3.9828e-02, -3.3651e-01, -4.2370e-01,  3.3182e-01,  1.1019e-01,\n",
       "           1.7328e-01,  1.5282e-02,  3.9299e-01,  1.9321e-01, -5.2099e-01,\n",
       "           2.8514e-01,  1.4470e-01,  4.3958e-01, -1.8846e-01, -4.1365e-01,\n",
       "          -1.0991e-01,  3.3157e-01, -3.8281e-01, -3.4973e-01, -1.0983e-01,\n",
       "           3.0412e-02,  5.9583e-02, -5.8809e-02, -3.7567e-01,  2.5850e-01,\n",
       "          -6.2094e-01, -9.3514e-03, -4.2726e-01, -1.1183e-01,  4.7884e-01,\n",
       "           1.2318e-01, -3.6857e-02,  1.1137e-01, -4.2935e-01, -6.7978e-01,\n",
       "          -6.4684e-01, -7.4975e-02,  5.2269e-02, -3.7956e-01, -1.6842e-02,\n",
       "          -1.8671e-01,  6.6050e-02, -3.6548e-01,  1.8633e-01, -8.2184e-02,\n",
       "           7.5332e-01,  2.3533e-01, -1.8486e-01,  1.2410e-01, -1.4424e-01,\n",
       "          -3.6447e-01, -2.1959e-01, -2.4676e-01,  2.2621e-01, -5.0366e-01,\n",
       "          -1.3080e-01,  1.8011e-01,  1.1438e-01,  6.4878e-03, -3.4448e-01,\n",
       "          -1.2932e-01,  2.0570e-01, -1.3707e-01,  3.0997e-01,  1.7373e-01,\n",
       "           2.0329e-01, -6.4335e-02, -4.6287e-01,  8.7786e-02,  4.5179e-01,\n",
       "          -1.1348e-01, -2.7396e-01,  3.8658e-02, -7.0830e-01,  2.1125e-01,\n",
       "           3.8231e-01,  8.2746e-01, -2.5998e-01, -5.3996e-01,  4.6996e-01,\n",
       "           6.2219e-01, -4.6785e-01, -5.7024e-02, -2.1133e-01,  1.2442e-01,\n",
       "          -1.0488e-01, -3.4183e-01,  2.6033e-01, -2.5177e-01,  4.2375e-01,\n",
       "          -9.9348e-02,  2.0811e-01, -1.9413e-01,  4.1807e-01,  9.7647e-02,\n",
       "          -2.0238e-01, -3.5246e-02, -6.2741e-01, -4.3863e-01,  2.3873e-01,\n",
       "           2.6324e-01, -1.5522e-01, -2.3160e-02, -1.2275e-01,  1.1149e-01,\n",
       "           4.8271e-01,  8.5393e-01,  8.1221e-02,  1.4584e-01,  3.8317e-01,\n",
       "           2.3356e-01, -2.7630e-01,  5.9953e-02,  2.9270e-01,  4.0766e-01,\n",
       "           3.5259e-01,  5.3145e-01, -3.4176e-01,  7.2203e-02, -1.6344e-01,\n",
       "           2.0465e-01, -5.7619e-01, -7.6698e-01, -1.3126e-01,  4.3082e-01,\n",
       "           1.5541e-02,  2.9468e-01, -2.7851e-01,  3.4408e-01,  6.3093e-01,\n",
       "          -1.1306e+00, -5.9181e-02, -3.9846e-02, -9.3994e-02, -6.2591e-02,\n",
       "          -5.4897e-01, -2.2413e-01, -3.5234e-01, -2.4418e-01,  5.2852e-01,\n",
       "           5.8811e-01, -8.5286e-01,  1.6085e-01,  2.2953e-01, -1.5005e-01,\n",
       "           9.4328e-02, -9.1644e-02,  3.2863e-02,  5.0134e-01,  1.5175e-01,\n",
       "          -5.7932e-01,  4.5778e-01,  1.5595e-01, -4.3637e-01,  2.6165e-01,\n",
       "           4.1776e-01, -1.8687e-02,  1.5952e-01,  2.2085e-01, -4.9751e-01,\n",
       "          -1.0141e-02, -4.6200e-01, -1.0629e-01,  1.0751e-01, -6.7537e-02,\n",
       "           2.4815e-02, -4.9794e-01, -6.6236e-02,  3.4484e-01,  2.9505e-03,\n",
       "           5.0621e-01,  6.0668e-01,  7.6526e-01, -5.6484e-02, -2.1306e-01,\n",
       "           3.3852e-01,  2.5943e-01, -9.1834e-02,  5.2227e-01, -2.1885e-02,\n",
       "          -1.6835e-01,  3.1658e-01,  2.3124e-02, -2.2045e-01,  1.7294e-01,\n",
       "          -2.5761e-01,  2.6423e-01,  1.0870e-01, -4.9724e-01,  3.6477e-01,\n",
       "           3.7087e-01,  1.5082e-01, -3.0923e-01,  1.8171e-01,  3.9828e-01,\n",
       "          -3.1100e-01,  6.8356e-01,  1.6607e-04,  2.2973e-01,  6.3476e-02,\n",
       "          -1.1506e-01, -5.5647e-01, -3.1374e-01,  2.1509e-01,  9.2105e-02,\n",
       "          -5.0547e-01, -9.5287e-02,  1.5106e-01, -4.9544e-01,  7.8060e-01,\n",
       "           6.2385e-01, -3.7763e-01, -3.5664e-01, -7.1127e-02,  1.1295e-01,\n",
       "           1.2269e-01, -3.3879e-01,  2.9572e-01,  1.1312e-01,  1.0076e-01,\n",
       "          -1.0219e+00, -2.9326e-01,  2.4530e-01,  4.9147e-02, -4.2291e-01,\n",
       "           5.9939e-01,  7.9206e-02,  5.0359e-02,  3.8428e-01,  3.0534e-01,\n",
       "           1.4161e-01,  4.3910e-01,  7.0775e-02, -2.1375e-01,  6.5311e-02,\n",
       "           2.5600e-01, -3.0970e-01, -3.2766e-01, -6.3757e-01,  5.3684e-01,\n",
       "           1.5068e-01,  8.4160e-01,  6.5313e-01]], device='mps:0'),\n",
       " tensor([[-0.1230, -0.2743, -0.7632,  ...,  0.0595,  0.8762,  0.7366],\n",
       "         [-0.2143, -0.4630, -0.6023,  ...,  0.1281,  0.4524,  0.6160]],\n",
       "        device='mps:0'),\n",
       " tensor([[-0.1672, -0.5249, -0.4540,  ...,  0.3771,  0.2016,  0.5340],\n",
       "         [-0.3292, -0.2590, -0.5016,  ..., -0.0148,  0.3509,  0.1607]],\n",
       "        device='mps:0'),\n",
       " tensor([[-0.0111, -0.5057, -0.4670,  ...,  0.2365, -0.0712,  0.4388],\n",
       "         [-0.3292, -0.2590, -0.5016,  ..., -0.0148,  0.3509,  0.1607]],\n",
       "        device='mps:0'),\n",
       " tensor([[-6.3660e-02, -7.0359e-01, -8.1569e-02, -8.8158e-01,  3.4994e-02,\n",
       "          -4.5083e-01, -6.3052e-01,  2.3072e-01, -5.7101e-02,  8.1769e-01,\n",
       "           1.2302e-01,  1.8531e-01,  4.5536e-01,  2.3235e-01, -6.4351e-01,\n",
       "          -2.3035e-01, -5.4593e-02,  3.8878e-01,  6.5353e-02,  4.1796e-01,\n",
       "           9.6914e-02,  5.7851e-01,  4.6976e-02, -1.5480e-01, -7.6759e-01,\n",
       "           1.6043e-01,  1.1824e-01, -1.7810e-01, -1.4366e-01, -2.8344e-01,\n",
       "          -3.8239e-02,  3.6106e-01,  4.2168e-02,  1.2776e+00, -1.9196e-01,\n",
       "          -3.7363e-01,  3.3735e-01,  4.3436e-02,  4.1075e-01, -8.2654e-01,\n",
       "           5.3940e-01, -1.5442e-01,  2.5672e-01,  1.4481e-01,  3.3940e-01,\n",
       "          -1.0255e-01, -1.3390e-01, -3.8077e-01,  5.0081e-01, -6.1703e-01,\n",
       "          -1.7319e-01, -2.7674e-01, -1.6836e-01,  3.1066e-01, -5.6382e-01,\n",
       "          -3.9552e-01,  1.2492e-01,  1.4951e-01,  1.6566e-02,  4.6492e-02,\n",
       "          -4.5406e-02,  1.1688e-01, -1.6788e-01,  1.2560e-01,  1.9688e-01,\n",
       "          -2.8222e-01, -6.3916e-01, -7.6971e-01, -7.5577e-01, -3.2710e-01,\n",
       "          -1.1027e-01,  1.7113e-01,  7.8641e-02, -2.7047e-01, -1.2197e-02,\n",
       "           3.2937e-02,  8.9621e-02, -3.8588e-01, -1.9903e-01,  3.8849e-03,\n",
       "          -6.6784e-02,  2.4118e-01,  2.6614e-01,  8.1746e-02, -3.5837e-01,\n",
       "          -1.4970e-01,  2.8098e-01,  1.8501e-01,  5.5596e-03,  2.7966e-01,\n",
       "           5.2068e-02, -2.1917e-01, -1.4163e-01, -2.9491e-01,  4.7070e-01,\n",
       "           5.5243e-02,  1.5085e-01, -4.7138e-01,  1.1693e-02, -2.3916e-01,\n",
       "           9.3257e-03, -1.2050e-01,  1.6442e-02,  3.2374e-01, -4.7997e-01,\n",
       "          -2.0281e-01,  1.1067e-01, -1.0554e-01, -4.8690e-01,  2.1337e-01,\n",
       "          -1.4453e-01, -5.2103e-01, -1.2767e-02, -5.8552e-01, -3.8935e-01,\n",
       "           1.9762e-02,  3.1461e-01,  4.2663e-01, -1.7660e-01,  5.3154e-01,\n",
       "           3.1281e-01,  1.2339e-01,  9.4934e-02,  2.0953e-01,  3.6854e-02,\n",
       "          -5.1513e-02,  7.4764e-01, -1.6507e-01,  8.0369e-02,  3.3128e-01,\n",
       "          -1.3148e-01, -4.6809e-02, -5.8858e-01, -2.1626e-01,  3.8663e-01,\n",
       "           1.4215e-01, -1.1366e-01, -2.7202e-01, -2.7648e-01, -1.0333e-01,\n",
       "           2.7028e-01,  2.8396e-01, -6.1453e-01, -4.0374e-01, -1.1232e-01,\n",
       "          -1.6071e-01, -1.1362e-01, -7.9456e-02, -2.9762e-01, -4.7183e-01,\n",
       "          -7.5342e-01, -5.2162e-01,  8.4199e-01,  5.0192e-02, -2.0557e-01,\n",
       "          -2.0806e-01,  5.4474e-01, -1.3480e-01,  3.3857e-01,  1.4177e-01,\n",
       "           8.8030e-01, -1.9183e-01,  8.1270e-02, -7.1075e-01, -5.8633e-02,\n",
       "          -2.4276e-01,  2.1721e-01, -4.5495e-01,  4.8535e-01,  3.0062e-01,\n",
       "           3.2224e-01, -1.8344e-01,  3.9356e-01, -4.2401e-01,  3.5388e-01,\n",
       "          -2.2512e-02,  2.3472e-01,  2.0299e-01, -3.4573e-01,  3.4537e-02,\n",
       "           9.7004e-02, -1.2335e+00,  2.0063e-01, -1.0631e-01, -5.4442e-01,\n",
       "          -2.8806e-01, -2.6192e-01,  1.7322e-01,  3.3236e-01, -3.4066e-01,\n",
       "          -1.5959e-01,  3.6463e-01,  1.4078e-01,  3.6416e-01, -2.1339e-01,\n",
       "          -2.1711e-01,  5.5270e-01,  7.7588e-01, -1.1373e-01, -9.3698e-02,\n",
       "          -2.1684e-01,  3.1208e-01, -1.7494e-01, -1.2096e-01, -2.1082e-01,\n",
       "           3.0381e-01, -3.2752e-01, -1.3203e-01,  9.1041e-02,  2.4047e-01,\n",
       "          -2.3222e-01, -3.8511e-01, -4.2054e-01, -1.1736e-01, -3.0369e-01,\n",
       "           4.7052e-01, -4.5912e-01, -1.4761e-01,  4.5544e-01,  3.8487e-01,\n",
       "          -1.7952e-01,  9.7157e-02,  4.0572e-01,  3.1537e-02,  2.4354e-02,\n",
       "           5.3066e-01, -1.6041e-01, -2.2142e-01,  1.5688e-02, -1.5091e-01,\n",
       "          -2.2892e-01, -8.4899e-01,  5.3104e-01,  1.2316e-01, -1.7696e-01,\n",
       "          -9.6078e-02, -2.3571e-01,  1.6756e-01, -2.3050e-01,  5.1070e-01,\n",
       "           1.1539e-01, -3.5134e-01, -6.6779e-01, -3.6547e-02,  7.0389e-04,\n",
       "           3.9037e-02, -7.5231e-02,  6.1386e-01,  7.4134e-01,  1.0742e-01,\n",
       "          -4.2193e-01,  1.9701e-01,  7.8800e-01,  3.3339e-01, -4.3084e-01,\n",
       "          -1.5554e-02,  1.4011e-01, -7.3580e-01,  2.3123e-01, -2.0928e-01,\n",
       "          -3.6079e-01, -9.6173e-02,  8.2354e-03,  4.7417e-01, -2.4306e-01,\n",
       "           1.0534e-01,  5.4846e-01, -2.3196e-01, -3.5417e-03,  4.2123e-01,\n",
       "          -4.6901e-01,  2.6778e-01, -3.0972e-02, -5.1125e-01, -5.2076e-01,\n",
       "          -5.3294e-01, -2.5733e-02,  5.3655e-01, -2.3351e-02, -6.0166e-01,\n",
       "          -1.0269e-01, -2.2004e-01,  1.4001e-01, -1.6176e-01, -4.8441e-01,\n",
       "           1.4930e-01, -1.7611e-01, -1.6430e-01, -4.6899e-01, -4.0677e-01,\n",
       "          -4.4721e-01, -8.3116e-02, -3.9755e-01, -4.8332e-02,  7.5163e-01,\n",
       "           3.1077e-01,  3.8663e-01,  3.5406e-01,  3.5435e-01, -2.6022e-01,\n",
       "          -1.7493e-01,  4.9319e-01,  1.3410e-01, -2.2369e-01,  5.9610e-01,\n",
       "           3.6790e-01,  4.2034e-01,  9.6918e-01, -3.8569e-01, -3.0960e-01,\n",
       "           9.2053e-02,  9.6189e-01,  4.5396e-03, -8.8250e-01, -1.0678e-01,\n",
       "          -1.5245e+00,  1.9568e-02,  2.5555e-01,  3.2868e-02, -4.2972e-01,\n",
       "          -5.9892e-02,  4.4172e-01, -4.0122e-02,  1.0152e-01, -1.6307e-02,\n",
       "           1.5665e-01,  6.5560e-01, -2.3379e-02,  4.0088e-03, -1.2796e-01,\n",
       "          -4.2723e-01, -7.0946e-02, -5.2741e-01, -9.8233e-02,  2.4980e-01,\n",
       "           3.0833e-01,  4.0444e-01,  1.6377e-02, -3.7455e-01, -1.3981e-01,\n",
       "           2.1881e-01, -2.4181e-01, -2.8994e-01,  3.3207e-02,  5.9594e-02,\n",
       "           1.8066e-01,  3.8795e-01, -2.9973e-01, -3.7632e-01,  2.7444e-01,\n",
       "           3.3607e-02, -2.8521e-01, -3.4620e-01, -1.3035e-01,  7.3802e-01,\n",
       "           3.2815e-01, -1.8282e-01, -9.0580e-02,  3.4452e-01,  9.7667e-02,\n",
       "          -3.3571e-01,  8.5954e-02, -4.8382e-02, -9.6304e-02,  5.7429e-01,\n",
       "           2.1334e-01,  4.0127e-01, -1.8178e-01,  8.4365e-02,  1.7224e-01,\n",
       "           3.1980e-02, -7.0960e-01, -1.0994e-01,  1.9835e-01,  5.1552e-01,\n",
       "          -2.7715e-01, -4.1003e-01, -6.2859e-02, -9.9340e-01, -3.6220e-01,\n",
       "          -2.6464e-02, -2.7296e-01,  2.8657e-01, -3.4627e-01, -5.9818e-01,\n",
       "           1.1096e-01, -3.6553e-01,  6.8905e-01,  1.3755e-02,  1.4406e-02,\n",
       "           5.4772e-01,  1.5788e-01, -2.6293e-01,  5.3009e-01,  1.2232e-01,\n",
       "          -2.4067e-01,  1.4507e-01, -7.4799e-01,  1.6170e-02,  2.1078e-01,\n",
       "          -3.0292e-01, -9.5226e-02,  4.2249e-01,  6.4682e-01,  2.8987e-01,\n",
       "          -3.3913e-01,  1.3462e-01,  1.8931e-01, -4.9877e-01, -7.5750e-03,\n",
       "           1.8786e-01,  2.6535e-01, -2.5993e-01,  3.8056e-01, -2.3739e-01,\n",
       "           1.7358e-01,  6.4413e-02, -1.5831e-01, -4.2978e-02, -5.3697e-01,\n",
       "           4.2401e-01, -1.1300e-01,  2.7115e-01,  4.0954e-01, -1.1155e+01,\n",
       "           1.5285e-01,  1.0540e+00, -1.2958e-01, -3.0586e-02, -2.5433e-01,\n",
       "           3.7116e-01, -4.4917e-01,  2.1879e-01, -1.7120e-01, -4.2764e-01,\n",
       "          -4.8415e-01,  6.0616e-02,  3.0047e-01, -3.7222e-01,  2.3391e-02,\n",
       "           9.1454e-02, -8.1570e-01,  2.2156e-01,  1.2135e-01,  2.2099e-01,\n",
       "          -6.2769e-02, -1.4374e-01,  1.0675e-01, -1.5945e-01, -1.4285e-01,\n",
       "          -1.7716e-01, -2.6887e-01,  2.5599e-01,  5.0296e-01, -8.9852e-02,\n",
       "           4.6643e-01,  1.4933e-01, -2.4000e-02, -2.5029e-01, -7.8371e-01,\n",
       "          -2.8378e-01, -1.9129e-01,  3.4727e-01,  1.7059e-01, -2.3286e-02,\n",
       "          -1.4311e-01, -4.0817e-01,  1.4556e-01,  3.1240e-01,  6.8233e-01,\n",
       "           7.6034e-01,  8.4683e-01,  4.1967e-01, -4.0086e-01, -1.2573e-01,\n",
       "          -8.5509e-01,  5.0727e-02, -7.5983e-01,  2.5698e-01, -1.8777e-01,\n",
       "           5.7542e-01, -5.5664e-01, -1.7341e-01, -2.6495e-01,  3.7254e-02,\n",
       "           3.9540e-01,  3.9478e-02, -2.3732e-01, -2.1306e-01, -1.8839e-01,\n",
       "           2.7860e-01,  3.6534e-02, -1.0641e-01, -6.4241e-01, -3.7915e-01,\n",
       "           7.2197e-03,  3.2114e-01, -2.5012e-02,  3.7460e-01,  5.1288e-01,\n",
       "          -1.5627e-01, -4.7910e-03,  1.7505e+00, -4.7461e-01, -1.2464e-01,\n",
       "          -2.7146e-01,  4.4714e-01, -1.5916e-01, -3.8716e-01,  7.2920e-02,\n",
       "           5.9804e-02,  5.7889e-01,  3.1373e-01,  7.5964e-01, -2.8989e-01,\n",
       "          -4.7750e-01, -2.6761e-02,  3.5545e-01,  1.0502e-01,  1.0012e+00,\n",
       "          -2.2541e-01, -3.7270e-01,  2.0248e-01,  5.6284e-01, -6.0411e-01,\n",
       "          -3.1328e-02, -3.4438e-01, -3.3282e-01, -2.6075e-01, -2.6158e-02,\n",
       "          -1.7032e-02,  5.1753e-02,  2.7162e-01,  5.7080e-01, -3.9706e-01,\n",
       "           3.8989e-01,  1.9004e-01, -1.8248e-01, -2.7241e-01, -8.7440e-01,\n",
       "           5.5249e-02,  7.9455e-01, -3.2301e-01, -7.5664e-01, -1.5510e-01,\n",
       "          -1.5542e-01,  5.0583e-02,  8.7218e-02, -2.5632e-01,  5.3853e-01,\n",
       "          -8.4337e-01, -1.4451e-01,  1.8345e-01, -4.6297e-01,  4.7605e-01,\n",
       "           4.1288e-01,  6.2445e-02,  3.4487e-02, -6.7027e-01, -2.1880e-01,\n",
       "          -8.5919e-01, -1.0029e-01,  5.7802e-01, -2.3881e-01, -1.5432e-01,\n",
       "           4.5230e-01, -1.2899e-01, -1.2159e-01,  1.8253e-01, -1.0790e-01,\n",
       "           5.7591e-02,  1.0994e-01, -1.3841e-01,  2.2322e-01,  1.6592e-03,\n",
       "           2.3268e-01, -5.9201e-01, -2.6329e-01, -1.2813e-01, -4.8841e-01,\n",
       "          -2.6965e-01,  1.4213e-01, -4.4177e-01,  1.3307e-01,  2.9584e-01,\n",
       "           3.1391e-01,  3.3033e-02,  4.4921e-02,  5.0014e-01,  2.9735e-01,\n",
       "          -1.0271e-01, -3.4976e-01, -5.8183e-02, -3.3582e-01,  1.2506e-01,\n",
       "           1.6146e-01, -2.2645e-01, -2.5220e-01, -4.4534e-01, -4.5335e-01,\n",
       "          -2.7473e-01,  1.2781e+00, -2.6287e-01, -3.3188e-01,  2.8896e-01,\n",
       "           4.4421e-01, -1.0096e-01,  1.9864e-01,  1.8132e-01,  3.1285e-01,\n",
       "          -3.0726e-01, -1.6866e-01,  5.1756e-01, -1.7423e-01,  2.3199e-01,\n",
       "           1.1482e-01,  4.3287e-02, -9.0979e-02,  7.7368e-02,  9.9691e-02,\n",
       "          -5.3722e-01,  1.6507e-01, -1.7745e-01,  2.0629e-01,  2.2885e-01,\n",
       "          -2.1714e-01, -2.4027e-01,  1.4243e-01, -2.7966e-01,  9.2641e-02,\n",
       "           1.4310e-01,  9.6745e-01, -5.6540e-02, -2.7167e-01,  6.4677e-01,\n",
       "           2.6552e-01,  1.4555e-01, -1.3182e-01,  2.0353e-01,  1.7392e-01,\n",
       "           3.9072e-01,  5.5611e-01, -5.9553e-01,  5.5038e-02,  6.4210e-02,\n",
       "           1.7920e-01, -4.9438e-01, -2.1127e-01,  3.5170e-01,  2.1303e-01,\n",
       "          -8.9566e-02,  3.2137e-01,  3.8199e-02,  2.5288e-01,  3.7599e-01,\n",
       "          -8.5903e-01,  8.6963e-02, -9.3896e-02, -4.3068e-01, -1.0781e-01,\n",
       "          -4.4690e-01, -3.1473e-01,  8.5380e-02, -4.2354e-01,  3.1221e-01,\n",
       "           5.4388e-01, -1.1241e+00,  1.4731e-01,  5.9514e-01, -4.0355e-01,\n",
       "           2.7757e-01,  9.2658e-02,  4.0264e-01,  1.9387e-01, -3.7439e-01,\n",
       "           6.6266e-02,  1.1098e-01, -4.2346e-01, -6.7246e-01,  2.0327e-01,\n",
       "           7.5403e-01,  1.1349e-01, -2.0150e-01, -2.1585e-01, -4.2441e-03,\n",
       "          -3.5713e-01, -3.9603e-01, -2.4378e-01, -1.6930e-02,  4.0383e-01,\n",
       "          -3.4128e-01, -1.2380e-01, -2.7945e-01,  2.1798e-01,  3.0289e-01,\n",
       "           3.6812e-01,  5.8350e-01,  7.8301e-01,  5.9435e-02, -2.6345e-01,\n",
       "          -2.1735e-01,  7.5820e-01,  6.5724e-03,  7.3939e-01, -1.1760e-01,\n",
       "          -3.3008e-01, -4.1604e-02,  1.1412e-02,  9.5438e-02, -2.5193e-01,\n",
       "          -1.5892e-01,  5.2617e-01, -3.5438e-02, -1.9908e-02,  4.3814e-01,\n",
       "          -1.5645e-01,  1.2825e-01, -4.9445e-01,  1.1428e-01,  3.3915e-01,\n",
       "           1.5803e-02,  5.6122e-01,  1.6382e-01,  4.6431e-01, -6.9501e-01,\n",
       "           8.6815e-02, -5.1235e-01, -1.1081e-01, -1.6269e-01, -2.9335e-01,\n",
       "          -4.1511e-01, -1.9989e-01,  1.9354e-01,  7.9081e-02,  5.2719e-01,\n",
       "           5.9205e-01, -1.5609e-01, -5.9961e-01, -1.8604e-01,  9.4833e-02,\n",
       "           4.2820e-01, -1.3857e-01,  6.8642e-01, -1.2007e-01,  1.5244e-02,\n",
       "          -7.4171e-01, -7.9910e-02,  5.1878e-02,  1.0624e-01, -6.0799e-02,\n",
       "           4.3445e-01, -1.4526e-01,  5.0166e-01,  9.7796e-02,  3.3521e-01,\n",
       "           3.5235e-01,  6.9826e-01,  1.4090e-01, -7.6857e-01, -1.9484e-01,\n",
       "           2.7389e-01, -1.6172e-01,  2.4140e-01, -8.3217e-01,  1.2475e-01,\n",
       "           2.5083e-01,  7.8436e-01,  4.3375e-01]], device='mps:0'),\n",
       " tensor([[-5.9179e-03, -4.1998e-01, -5.0229e-01, -7.3511e-01, -4.5834e-02,\n",
       "          -1.2884e-01, -7.8341e-01,  5.7507e-01,  2.6684e-01,  6.2232e-01,\n",
       "           3.2579e-01,  4.9793e-01, -2.5659e-01, -2.6551e-01, -2.1439e-01,\n",
       "          -2.0187e-01, -1.1339e-02,  1.3112e-01,  8.2812e-01,  4.3376e-01,\n",
       "           3.7652e-01,  6.6596e-01, -4.4789e-02, -2.2572e-01, -1.0817e+00,\n",
       "           3.5554e-01, -1.2044e-01,  2.4959e-01,  4.2387e-01, -7.2993e-01,\n",
       "           1.9636e-02,  8.5046e-01,  5.0634e-03,  1.3825e+00, -3.0430e-01,\n",
       "          -2.6026e-01,  1.9773e-01, -5.0296e-01, -2.3360e-02, -3.6061e-01,\n",
       "           3.5498e-01,  2.9586e-01,  2.6590e-01,  2.7359e-01,  1.3341e-01,\n",
       "           8.4401e-03, -7.7328e-02, -1.5596e-01, -5.7719e-02, -4.8710e-01,\n",
       "          -2.4084e-01, -2.9721e-01,  4.1457e-01,  2.6215e-01, -6.5225e-01,\n",
       "          -3.5546e-01,  3.2303e-01, -4.0267e-02,  3.6625e-01,  7.6335e-02,\n",
       "          -2.1397e-01,  4.9766e-01, -3.4709e-01,  1.0594e-01,  4.4920e-02,\n",
       "          -8.2730e-03, -4.2580e-01, -5.2447e-01, -3.9813e-01,  1.3732e-01,\n",
       "           4.3316e-02,  4.0985e-01,  1.1400e-01, -2.7066e-01,  9.6689e-02,\n",
       "          -3.9833e-01,  1.5953e-01, -4.2664e-01, -3.4535e-01,  1.3154e-01,\n",
       "           3.4932e-01,  2.8238e-01,  3.7452e-01, -5.1145e-01,  1.8465e-01,\n",
       "          -9.9706e-03,  3.2697e-01,  7.8465e-02,  6.7475e-02,  3.3342e-01,\n",
       "          -3.9254e-01,  2.1753e-01, -3.8316e-01, -2.1620e-01,  2.1515e-01,\n",
       "          -1.3452e-02, -9.9024e-02, -1.1232e-01,  1.2102e-01, -2.8773e-01,\n",
       "           5.0472e-02,  6.7391e-02, -1.2135e-02,  9.1128e-02, -6.0833e-01,\n",
       "          -9.2680e-01,  2.6504e-01, -5.3766e-01, -5.8675e-01, -2.3417e-02,\n",
       "          -3.0442e-01, -2.2282e-01, -4.2731e-02, -3.2884e-01,  9.8244e-03,\n",
       "           3.2566e-01,  3.7217e-01,  1.1683e-01,  4.5423e-02,  1.1131e-01,\n",
       "           3.7113e-01, -1.3920e-01, -5.3773e-02,  1.5054e-01,  1.5030e-02,\n",
       "          -2.6383e-01,  7.0332e-01,  1.5532e-02,  6.3081e-01,  5.3005e-01,\n",
       "           2.0902e-01,  3.4219e-02, -5.1521e-01, -2.3368e-01,  5.7438e-01,\n",
       "           4.4736e-02,  1.4489e-01, -2.3467e-01, -7.9414e-02,  1.9532e-02,\n",
       "           9.5931e-02,  1.1702e-02, -2.8280e-01, -3.5387e-01, -2.1288e-01,\n",
       "          -3.7261e-01, -9.6307e-03, -1.2334e-01, -4.9358e-01, -2.2327e-01,\n",
       "          -1.0160e+00, -5.7005e-01,  6.8261e-01, -1.0601e-01, -2.2433e-01,\n",
       "          -4.3243e-01,  3.4931e-01,  1.4140e-01, -1.2379e-01,  4.2336e-02,\n",
       "           8.3263e-01, -1.7134e-01,  2.3027e-01, -6.4725e-01, -3.4350e-01,\n",
       "           1.5296e-01, -1.1479e-01, -8.9936e-02,  3.2910e-01, -1.9953e-01,\n",
       "          -7.3352e-02, -4.1008e-01,  4.5700e-01,  2.7683e-01,  7.9286e-02,\n",
       "          -2.7927e-01,  7.1797e-02,  2.5334e-01, -2.9401e-01,  1.2271e-01,\n",
       "          -3.1419e-01, -1.3565e+00,  3.5876e-01, -5.5651e-01, -1.0625e-01,\n",
       "          -1.1253e-01,  2.4134e-01,  2.2978e-01,  3.0874e-01, -1.5871e-01,\n",
       "           2.8289e-01,  2.8636e-01, -7.5058e-03,  4.0906e-01, -2.7163e-01,\n",
       "           5.0389e-01,  5.3369e-01,  5.4312e-01,  2.5197e-01, -5.9460e-02,\n",
       "          -4.3183e-01,  3.0183e-02,  4.5516e-01,  2.1362e-01,  2.5551e-01,\n",
       "          -5.4497e-02, -1.9279e-01,  1.1550e-01,  5.9524e-02, -1.2564e-01,\n",
       "          -2.4360e-01,  2.0566e-01, -4.0348e-01,  2.1063e-01, -5.4406e-01,\n",
       "           8.3748e-01, -2.4489e-01,  1.3712e-01,  4.1594e-01,  4.2844e-01,\n",
       "          -2.5722e-01,  2.2741e-01,  3.1293e-01, -3.2591e-01, -1.9496e-01,\n",
       "           2.5404e-01, -2.0384e-01,  1.7646e-02, -1.1906e-01, -4.9191e-01,\n",
       "           3.4672e-01, -6.9751e-01,  7.0778e-01,  1.5412e-01, -2.1010e-01,\n",
       "           2.5728e-01,  1.2580e-02,  2.3543e-02, -3.9308e-01,  7.1432e-01,\n",
       "          -7.5540e-03, -3.8965e-01, -5.5894e-01,  2.3736e-01,  3.7278e-01,\n",
       "           1.8975e-01, -2.3810e-01,  2.6549e-01,  1.0390e+00,  3.5405e-01,\n",
       "          -2.6470e-01, -1.0869e-01,  3.3740e-01,  7.8907e-02, -2.4339e-01,\n",
       "          -8.4001e-02, -4.0234e-02, -5.8585e-01,  2.3966e-01, -1.4812e-01,\n",
       "           3.7812e-02,  5.1495e-02,  1.3404e-01, -4.2029e-02, -5.4586e-01,\n",
       "          -4.3942e-01,  3.6477e-01, -1.9639e-02, -1.5518e-01,  2.5486e-01,\n",
       "          -3.8681e-01,  2.2038e-01, -1.5652e-01, -5.3802e-01, -3.6853e-01,\n",
       "          -7.3431e-01,  3.3854e-01, -6.2589e-02, -4.1500e-02, -8.5635e-01,\n",
       "          -2.8142e-02,  2.2216e-01,  4.0020e-01,  4.4861e-02,  1.7195e-01,\n",
       "           5.1286e-01, -1.8116e-02, -1.5263e-01, -9.7538e-02, -4.1126e-01,\n",
       "           3.9568e-03, -7.2533e-02, -3.9278e-01,  3.5378e-01,  5.0853e-01,\n",
       "           5.0836e-01,  1.3292e-01,  1.3550e-01, -4.0132e-01, -1.5570e-01,\n",
       "          -2.1862e-01,  1.0511e-01, -3.2763e-01, -2.0517e-01,  5.9571e-01,\n",
       "          -1.4691e-01,  6.8865e-01,  3.6335e-01, -1.3471e-01, -3.8796e-01,\n",
       "          -6.8184e-03,  9.6956e-01, -2.0389e-01, -5.0201e-01,  1.4010e-01,\n",
       "          -1.0105e+00,  1.2340e-02,  1.2697e-01, -2.2955e-01, -4.4250e-01,\n",
       "           2.7106e-01,  3.0947e-01,  2.4060e-01, -1.9036e-01, -1.2164e-01,\n",
       "           3.1995e-01,  1.3781e-01,  1.7765e-01,  4.9361e-02, -4.1579e-01,\n",
       "          -5.8501e-01, -3.1560e-01, -1.5635e-01, -2.1049e-01, -1.4030e-01,\n",
       "           1.7130e-01,  2.7772e-01, -2.6169e-01, -3.2172e-01, -4.5906e-01,\n",
       "           4.9661e-01, -6.3282e-01, -1.4469e-01,  3.7213e-01, -2.8031e-01,\n",
       "           1.9196e-02,  4.4427e-01, -5.1320e-03, -2.2620e-01, -4.6519e-02,\n",
       "           6.4414e-02, -1.7558e-01, -6.4138e-01, -8.3765e-02,  9.3572e-01,\n",
       "           3.9943e-01, -9.6437e-02, -4.6571e-01,  9.2186e-01,  5.3711e-01,\n",
       "          -3.7461e-01,  4.3365e-01, -2.8922e-01, -1.7582e-02, -1.4850e-01,\n",
       "           1.7613e-02,  5.5633e-01, -4.6312e-01,  2.8343e-01, -8.9008e-02,\n",
       "           2.7891e-01, -2.5001e-01, -3.9489e-01,  1.7236e-01,  3.9758e-01,\n",
       "          -7.1633e-02, -4.8238e-01,  9.6436e-02, -4.0680e-01, -6.1606e-01,\n",
       "           1.1973e-01,  1.5466e-01,  2.4401e-01, -4.0135e-01, -4.1566e-01,\n",
       "          -1.2345e-01, -1.3914e-01,  6.8320e-01,  6.2336e-01, -3.4908e-01,\n",
       "           1.6728e-02,  1.6246e-02, -2.0175e-01,  1.1255e-01, -3.1550e-01,\n",
       "          -3.0856e-01,  3.8366e-01, -2.0778e-01,  1.9936e-02,  2.3041e-01,\n",
       "          -1.2817e-01, -2.5494e-01, -1.5960e-01,  4.8729e-01,  3.7323e-02,\n",
       "          -6.1480e-01,  1.4031e-01,  1.3266e-01, -3.9378e-01,  9.9719e-02,\n",
       "          -4.5911e-01,  6.4388e-01, -1.4027e-01,  1.0091e-01, -2.4921e-01,\n",
       "           9.7454e-02, -6.0955e-02, -1.0206e-01, -3.2019e-01, -5.8134e-01,\n",
       "           7.5182e-02,  2.3898e-01,  4.5021e-02,  2.8507e-01, -1.0951e+01,\n",
       "           3.2031e-01,  7.3847e-01, -4.0939e-01,  2.1200e-02,  4.7001e-02,\n",
       "           3.0199e-02,  1.5382e-01,  4.5248e-01, -5.6591e-01, -1.4675e-01,\n",
       "          -5.7198e-01,  1.3142e-01,  2.6745e-02, -5.4259e-01, -3.3899e-01,\n",
       "           1.2458e-01, -8.0577e-01,  3.0871e-01, -3.6979e-02,  7.5919e-01,\n",
       "          -4.5548e-01,  4.8188e-02,  2.2068e-01, -9.2066e-02, -3.5275e-01,\n",
       "           4.3650e-01, -3.4514e-01,  5.4580e-02,  7.1542e-01,  4.8183e-01,\n",
       "           1.7668e-01, -1.8943e-01,  1.5003e-01, -2.0156e-01, -4.5687e-01,\n",
       "           1.3850e-01, -2.1326e-01,  2.8060e-01, -3.1921e-02, -6.0316e-01,\n",
       "          -6.3345e-01,  1.4686e-01,  2.1502e-02,  1.6453e-01,  1.0198e+00,\n",
       "           5.4993e-01,  8.2084e-01,  4.4525e-02,  1.9908e-01,  1.2260e-01,\n",
       "          -9.3512e-01,  1.7722e-01, -4.9416e-01, -3.5862e-01, -4.1841e-02,\n",
       "           5.5988e-01, -3.6568e-01,  2.7622e-01, -6.0760e-01, -1.9435e-01,\n",
       "           2.5756e-01,  4.3839e-01,  1.6251e-01, -3.2975e-01, -1.6902e-01,\n",
       "          -3.9093e-01, -1.7876e-01,  4.8608e-02, -2.4983e-01, -3.1353e-01,\n",
       "           2.0635e-01,  4.3233e-01,  2.0248e-01, -6.4319e-01,  8.1634e-02,\n",
       "           1.1927e-01, -8.6306e-02,  1.5479e+00, -1.6392e-02,  1.7733e-01,\n",
       "          -3.4350e-01,  3.8821e-01, -3.2625e-01, -2.8232e-01, -2.6330e-01,\n",
       "          -1.1935e-02,  4.9256e-01,  7.5063e-02,  2.6494e-01, -8.4415e-02,\n",
       "          -8.4915e-01,  5.8002e-02,  2.3653e-01, -1.7916e-03,  6.8876e-01,\n",
       "          -1.1109e+00, -3.6235e-01,  1.2135e-01,  1.6393e-01, -5.9199e-01,\n",
       "           7.8315e-02, -4.9017e-01, -3.9993e-01,  3.5848e-01, -2.9180e-01,\n",
       "           1.1229e-01,  2.3286e-01, -3.2920e-01,  4.7045e-01, -2.3591e-01,\n",
       "           1.4967e-01,  4.6493e-01, -1.8001e-01, -2.0520e-01, -1.8734e-01,\n",
       "           4.3635e-01,  1.8579e-01, -3.1763e-01, -3.3345e-01,  1.2208e-01,\n",
       "           8.8785e-02,  1.1038e-01,  1.0926e-01, -2.1657e-01,  4.4801e-02,\n",
       "          -5.9242e-01,  1.5288e-01,  2.9498e-01,  1.2231e-01, -1.6932e-01,\n",
       "           3.3164e-01, -1.4902e-01,  2.3193e-01, -7.8418e-01, -6.2392e-01,\n",
       "          -7.3446e-01,  2.4548e-02,  2.5995e-01, -4.6791e-01, -2.2832e-01,\n",
       "          -4.2331e-02, -4.3234e-02, -1.2609e-01, -6.0443e-02, -6.0897e-02,\n",
       "           2.4085e-01,  1.3954e-01, -5.4115e-02,  4.2286e-02, -1.3539e-01,\n",
       "          -6.1854e-01, -1.0088e-01, -4.3220e-01,  7.2607e-02, -8.0827e-01,\n",
       "          -1.4735e-01,  1.3311e-01, -3.3301e-01, -4.7597e-01, -1.9556e-01,\n",
       "          -1.7268e-01, -1.3905e-01, -4.5889e-01,  8.2485e-01,  3.6239e-01,\n",
       "           3.8835e-01, -6.8989e-01, -2.0199e-01, -9.5641e-02,  2.3690e-01,\n",
       "           1.4716e-01,  3.1331e-01, -1.4866e-01, -6.9249e-01, -3.3463e-01,\n",
       "           4.4794e-02,  1.2233e+00, -8.2630e-02, -3.8256e-01,  3.2334e-01,\n",
       "           6.0254e-01, -6.3181e-02, -2.0718e-01, -8.6649e-02, -2.3812e-01,\n",
       "          -4.3302e-01, -6.0007e-01,  4.7863e-01, -1.0627e-01,  1.6231e-01,\n",
       "           1.6757e-01,  1.3009e-01, -5.2305e-01,  2.0813e-01,  9.2187e-02,\n",
       "          -2.9430e-01,  2.8867e-02, -6.0197e-01,  3.3827e-01,  4.0971e-01,\n",
       "          -2.8967e-01,  6.2108e-01,  1.3831e-01, -9.9204e-02, -1.8949e-01,\n",
       "           2.8030e-01,  1.5022e+00, -4.0663e-01, -1.0181e-01,  6.8177e-01,\n",
       "           2.6357e-01, -1.1246e-01,  3.8322e-02, -2.9326e-01,  1.9138e-02,\n",
       "           1.7858e-01, -2.5671e-02, -1.9799e-01,  4.3190e-02,  1.1839e-01,\n",
       "           3.4818e-01, -5.3023e-01, -5.6732e-01,  3.2249e-02,  6.9645e-01,\n",
       "          -1.3175e-01,  5.2807e-01,  2.2614e-02,  5.2902e-01,  1.1627e-01,\n",
       "          -9.5217e-01, -1.8160e-01, -6.6925e-01,  2.4877e-01,  2.1840e-01,\n",
       "          -2.6392e-01, -9.8631e-02, -2.0202e-01, -4.4310e-01,  6.9456e-01,\n",
       "           3.3309e-01, -1.0688e+00,  2.4605e-01,  7.0267e-01, -5.7585e-01,\n",
       "           1.5509e-01, -5.8364e-03,  2.4980e-01,  3.2985e-01,  2.6066e-01,\n",
       "           2.1541e-02,  2.6025e-01, -7.9271e-01, -5.5193e-01,  1.9207e-01,\n",
       "           6.2585e-01, -3.7250e-02,  8.8015e-02, -8.7509e-02, -9.6755e-02,\n",
       "           8.0280e-03, -2.0554e-01, -2.5968e-01,  2.6199e-01,  6.5840e-02,\n",
       "           6.4533e-03, -3.9846e-01,  5.4653e-02,  5.3571e-01, -1.3358e-01,\n",
       "           3.4129e-01,  3.6741e-01,  4.4826e-01, -2.5150e-01, -1.6433e-01,\n",
       "          -2.3263e-01,  4.0145e-01, -2.0016e-01,  3.8369e-01, -7.3335e-03,\n",
       "          -1.5047e-01,  2.0897e-01,  6.0362e-02, -5.6280e-01,  3.7811e-02,\n",
       "          -2.9643e-01,  2.6309e-01,  1.2131e-01, -3.6922e-01,  4.3408e-01,\n",
       "           6.9698e-02,  1.0633e-01,  1.9228e-01,  3.0956e-01,  1.2188e-01,\n",
       "          -7.2635e-02,  1.0159e+00,  2.2724e-02,  2.5605e-01, -8.3094e-01,\n",
       "          -2.1731e-02, -8.2871e-01, -4.3163e-02,  1.3480e-01, -2.3576e-01,\n",
       "          -4.9716e-01, -6.0834e-02,  5.8365e-02,  8.0970e-02,  7.0218e-01,\n",
       "           8.4348e-01, -4.2623e-01, -5.4025e-01, -4.4331e-01,  1.3954e-01,\n",
       "           4.1983e-01, -2.4263e-01,  6.6070e-01,  3.0840e-01,  4.6929e-01,\n",
       "          -6.0568e-01, -2.8666e-01, -2.9106e-01,  3.4693e-01, -2.4023e-01,\n",
       "           9.0032e-01,  9.8950e-02,  6.2100e-02,  9.0467e-02,  5.0067e-02,\n",
       "           1.0893e-01,  6.0983e-01,  6.8657e-02, -8.6657e-01,  1.8328e-01,\n",
       "           8.2756e-02, -1.4925e-01, -9.0346e-02, -1.8081e-01,  3.1474e-01,\n",
       "           3.4100e-01,  1.3909e-01,  9.9544e-01]], device='mps:0'),\n",
       " tensor([[-5.7009e-01, -7.0616e-01, -7.7654e-01, -9.8719e-01, -3.6567e-01,\n",
       "          -2.5630e-01, -5.9328e-01,  7.3188e-01, -1.3917e-01,  1.4983e-01,\n",
       "           3.0276e-01,  3.9456e-02,  2.9582e-01,  9.8687e-02, -3.7405e-01,\n",
       "           1.4522e-01,  3.7094e-01,  1.9387e-01,  2.4392e-01, -3.1031e-02,\n",
       "           1.0322e+00,  1.3952e-01, -1.1627e-01, -8.6595e-02, -3.9327e-01,\n",
       "           1.7716e-01, -4.1484e-01, -9.9995e-02,  3.3992e-01, -6.6918e-01,\n",
       "          -3.7707e-01,  8.9693e-02, -3.6732e-01,  9.2610e-01, -2.2768e-01,\n",
       "          -2.7508e-01,  2.9704e-02, -6.0159e-01,  6.4213e-01, -2.1046e-01,\n",
       "           2.2804e-01,  1.1963e-01, -1.8568e-01,  3.5767e-01,  8.6975e-02,\n",
       "           1.8528e-01, -2.3806e-02, -7.3258e-01,  7.6715e-02, -3.5348e-01,\n",
       "          -9.4658e-03, -8.4049e-02,  3.7771e-01,  3.6464e-01, -7.4922e-01,\n",
       "           2.2046e-02,  1.7758e-01,  3.7807e-02, -4.0473e-01,  2.5494e-01,\n",
       "           7.9635e-02,  7.1502e-01,  4.7596e-01,  2.1112e-01, -2.6314e-02,\n",
       "          -2.8790e-01, -6.1648e-01, -4.8644e-01, -3.2656e-01, -1.9755e-01,\n",
       "          -3.6959e-02,  1.4623e-02,  1.7640e-01, -1.1145e-01, -2.9496e-01,\n",
       "          -1.6810e-01,  2.4455e-01, -2.2076e-01, -5.6330e-01, -1.2318e-01,\n",
       "           3.4677e-01, -8.0865e-02, -4.3615e-02,  6.3321e-02, -1.6216e-01,\n",
       "          -7.3429e-03,  1.7331e-01, -2.7363e-02,  4.1080e-01,  1.5982e-01,\n",
       "           6.7130e-01, -1.5193e-01,  5.6972e-02, -4.4391e-01,  4.4627e-01,\n",
       "          -2.2024e-01,  3.5710e-01, -3.9368e-02,  7.9424e-02, -3.1228e-01,\n",
       "          -7.3658e-02,  3.0712e-01,  5.5042e-02,  9.0599e-02, -2.4678e-01,\n",
       "          -1.6671e-01, -9.8450e-02, -6.3440e-02, -5.5606e-01,  1.3837e-02,\n",
       "          -2.3955e-02, -4.6652e-01,  1.0598e-01, -5.1628e-01, -5.4694e-01,\n",
       "           3.0904e-01, -4.0919e-01, -4.0399e-01, -5.7538e-02,  1.2257e+00,\n",
       "           4.3965e-01, -3.6422e-01, -8.6200e-02, -1.2102e-02,  5.3759e-01,\n",
       "          -2.3719e-01,  1.6814e-01, -2.3635e-01,  2.2341e-01,  1.3253e-01,\n",
       "           5.6146e-02,  2.4904e-01, -4.5169e-01,  6.7593e-02,  1.9881e-01,\n",
       "          -1.4378e-01, -4.6994e-02, -3.6101e-01, -1.5688e-01, -1.6449e-01,\n",
       "           8.1545e-03, -1.2412e-01, -2.9320e-01,  8.9240e-02,  1.1247e-01,\n",
       "          -1.1443e-01, -1.2694e-01,  1.1758e-01,  8.4576e-02, -3.8067e-01,\n",
       "          -7.1266e-01, -1.2814e-01, -9.7195e-03,  7.3999e-02, -1.5344e-01,\n",
       "          -3.2034e-01,  5.3839e-01,  1.4508e-01,  9.2065e-02,  1.7652e-01,\n",
       "           4.2707e-01, -5.0711e-02,  3.9180e-01, -2.2431e-01,  2.9667e-01,\n",
       "           1.2127e-02, -5.7919e-01, -1.6867e-01,  2.3637e-01,  1.2929e-01,\n",
       "           5.1821e-01,  2.6464e-01, -9.2810e-03, -5.3712e-01, -3.8710e-02,\n",
       "           2.5342e-01,  6.1029e-01,  3.9373e-01, -2.4661e-01, -8.3068e-02,\n",
       "          -2.3387e-01, -1.9842e+00,  1.8765e-01, -5.7396e-01, -5.8766e-01,\n",
       "          -1.8809e-01,  2.2427e-01,  3.3606e-01,  4.3544e-01, -3.2097e-02,\n",
       "          -8.1912e-02,  3.7852e-03,  4.4665e-01,  7.6598e-01,  6.3606e-02,\n",
       "           4.9322e-01,  4.7313e-01,  2.1259e-03,  8.1333e-02, -2.6024e-02,\n",
       "          -5.4774e-01,  1.6095e-01,  9.9883e-02,  3.0353e-01, -7.8005e-02,\n",
       "           2.4050e-01, -2.6641e-01,  6.5831e-02,  4.7276e-01,  2.6703e-01,\n",
       "          -1.9159e-01, -3.1873e-01, -2.6361e-01, -1.4752e-01, -4.0831e-01,\n",
       "           1.4471e-01, -5.0283e-01, -8.6340e-02,  3.9664e-01,  1.1666e-01,\n",
       "           1.2062e-01, -2.6169e-01, -2.7931e-01, -2.6284e-01, -1.8121e-01,\n",
       "           4.4782e-01, -1.6591e-01, -1.1408e-01, -1.4700e-01, -2.4898e-01,\n",
       "          -7.5977e-02, -7.6446e-01,  3.9213e-01,  1.6836e-01,  3.2082e-01,\n",
       "          -1.6143e-01, -3.6315e-01,  3.4564e-01, -3.9865e-02, -3.3263e-02,\n",
       "          -1.0186e-01, -3.2834e-01, -5.4156e-01, -2.2352e-01, -2.5976e-03,\n",
       "           1.9511e-01,  5.5767e-02,  5.1024e-01,  7.9949e-01,  2.1341e-01,\n",
       "           1.3877e-02,  1.4214e-01,  4.7758e-01,  3.1735e-01, -5.4252e-01,\n",
       "           8.6942e-02,  3.5649e-02, -5.1474e-01,  3.1354e-01, -5.3173e-01,\n",
       "          -2.1116e-01, -8.2136e-02, -5.6981e-02,  6.2094e-01, -3.5166e-01,\n",
       "           1.9877e-01,  1.1213e-01,  7.3899e-03, -6.8394e-02,  3.8196e-01,\n",
       "          -6.1278e-01,  6.5257e-01, -5.8941e-03, -1.0083e-01, -7.3893e-01,\n",
       "          -5.8777e-01,  3.4423e-01,  2.2066e-01, -1.0786e-01, -4.4334e-01,\n",
       "           1.2171e-01, -2.0711e-01, -2.2283e-01, -2.0115e-02, -2.6458e-01,\n",
       "           3.5551e-01, -8.8462e-02,  4.5402e-02,  1.4040e-01, -2.9907e-01,\n",
       "          -7.8327e-01,  1.2181e-01, -1.1090e-01,  1.7142e-02,  2.1471e-01,\n",
       "           2.0621e-01,  1.7010e-01,  3.9476e-01, -1.0212e-03, -2.6620e-01,\n",
       "          -2.3867e-01, -5.8177e-02, -2.1834e-01, -1.9893e-01,  1.0638e-01,\n",
       "           6.3627e-01, -1.7385e-01,  4.9350e-01, -8.2697e-02, -6.4181e-01,\n",
       "           2.6336e-01,  9.4825e-01, -2.3364e-02,  2.1056e-01, -3.0474e-01,\n",
       "          -1.2817e+00, -1.9741e-01, -3.2547e-01, -2.5819e-01, -6.1732e-01,\n",
       "           3.6977e-01, -1.8343e-01, -2.6915e-03, -2.1333e-02, -1.8511e-01,\n",
       "           8.1074e-02,  5.9297e-01,  3.2062e-01,  1.3399e-01, -2.5531e-01,\n",
       "           2.6919e-03, -2.9727e-01, -2.3054e-01, -2.4627e-01,  8.1521e-02,\n",
       "           1.2796e-01,  2.8664e-01, -6.2438e-02,  1.6109e-01, -3.2997e-01,\n",
       "           6.4002e-01, -1.4559e-01,  4.0607e-02,  3.1353e-01, -2.0117e-01,\n",
       "          -1.5002e-01,  3.9336e-01, -2.3244e-01,  2.5442e-01,  4.1024e-02,\n",
       "           3.1503e-01, -5.2437e-01, -4.2163e-01,  1.7930e-01,  3.7983e-01,\n",
       "           4.1380e-01,  4.6159e-01, -1.8063e-01,  3.2263e-01,  2.2435e-01,\n",
       "          -3.4466e-01,  3.1789e-01, -3.8466e-01, -7.4143e-02,  7.9789e-02,\n",
       "           2.0190e-01,  1.2090e-01, -2.1209e-01, -8.3214e-02, -7.1386e-02,\n",
       "           3.4720e-01, -7.5740e-01,  3.9949e-02,  3.4476e-01, -2.4606e-01,\n",
       "          -1.6581e-01, -5.7513e-01,  1.6394e-01, -8.6867e-01, -4.3655e-01,\n",
       "          -2.5662e-02, -2.1618e-01,  5.8749e-02, -1.4333e-01, -7.8117e-02,\n",
       "          -1.2008e-02, -2.0377e-01,  4.1521e-01,  3.1926e-01,  7.9173e-02,\n",
       "           1.8330e-01, -1.0445e-01, -1.9646e-01,  2.9591e-01, -4.8153e-01,\n",
       "          -2.7900e-01,  7.7318e-02, -4.3749e-01,  1.6298e-02,  1.3116e-01,\n",
       "          -4.9610e-01, -2.0398e-01,  1.9597e-01, -4.4736e-02, -2.4319e-01,\n",
       "          -2.9726e-01,  1.5671e-02,  5.4157e-01, -1.6570e-01,  5.4771e-02,\n",
       "           1.0225e-02,  4.3404e-01, -1.1906e-01,  2.6383e-01,  2.0142e-01,\n",
       "          -1.4425e-01, -2.2780e-01, -1.2713e-01, -1.1441e-01, -3.1085e-01,\n",
       "           4.7883e-01,  3.3386e-02,  1.9083e-02,  3.1080e-01, -1.1331e+01,\n",
       "          -8.5964e-02,  8.3270e-01, -5.4294e-01, -1.3088e-01, -7.6311e-02,\n",
       "           7.2404e-02, -3.2318e-01,  1.1873e-01, -6.8110e-02, -3.2078e-01,\n",
       "          -3.0743e-01,  1.0373e-01, -2.4920e-01, -5.4285e-01, -2.6106e-01,\n",
       "          -3.6598e-01, -3.6439e-01,  1.7656e-01, -1.0455e-01,  2.5356e-01,\n",
       "          -7.1738e-01,  2.9476e-01,  3.6538e-01, -3.8715e-01, -5.8834e-01,\n",
       "          -5.4509e-01, -1.8663e-01,  1.6010e-01, -2.7008e-02, -2.6430e-01,\n",
       "           1.1522e-01, -1.2302e-01,  6.2389e-01, -1.9110e-02, -6.5540e-01,\n",
       "           5.7875e-02, -1.3937e-02,  5.6140e-01, -2.8702e-01,  6.9802e-02,\n",
       "          -2.1433e-01,  6.7109e-02,  3.5213e-01,  4.2667e-01,  7.1800e-01,\n",
       "           1.2492e+00,  5.3842e-01,  2.1806e-01,  9.8474e-02, -2.5419e-01,\n",
       "          -6.4352e-01,  5.1851e-01, -8.4375e-01,  7.5178e-02, -1.6625e-01,\n",
       "           5.2268e-01, -4.6953e-01, -2.0117e-01, -1.6704e-01, -9.1179e-02,\n",
       "           2.3032e-01,  1.8785e-01,  1.5994e-01, -6.2593e-02, -5.4343e-01,\n",
       "           6.4759e-01,  4.2481e-02,  8.4811e-02, -1.2425e-01, -7.4098e-02,\n",
       "          -2.5627e-01,  7.6494e-01,  2.2170e-02,  2.6402e-01,  4.4012e-01,\n",
       "           7.6412e-02,  2.0867e-01,  9.6449e-01, -8.8395e-02,  1.2303e-01,\n",
       "           5.2642e-01, -1.5003e-01, -3.0321e-01, -7.4052e-01,  4.0759e-02,\n",
       "          -3.2565e-02,  3.0725e-01,  2.0922e-01,  6.6175e-01, -2.0000e-01,\n",
       "          -2.2939e-01, -5.0617e-01, -3.1379e-01,  4.7640e-01,  8.9183e-01,\n",
       "          -4.5251e-01,  1.3039e-01, -1.5541e-01, -3.2147e-02, -5.3570e-01,\n",
       "          -6.7508e-02, -9.9119e-02, -5.7476e-01,  2.0181e-01, -2.8577e-01,\n",
       "           3.3467e-02,  1.6321e-01,  2.8508e-01,  5.7019e-01, -1.8868e-01,\n",
       "           8.2621e-02,  9.7819e-02,  1.3603e-01, -3.9973e-01, -7.5787e-02,\n",
       "          -3.6801e-02,  5.9613e-01, -2.1823e-02, -5.9709e-01,  1.5603e-01,\n",
       "          -1.9537e-01,  5.3122e-02,  6.8860e-02, -1.2172e-01,  4.9706e-01,\n",
       "          -7.0828e-01,  2.7900e-01,  3.2827e-01, -5.9399e-02,  7.8449e-01,\n",
       "           1.6671e-01,  1.8468e-01,  7.4671e-01, -3.8967e-01, -6.2083e-01,\n",
       "          -5.0571e-01, -7.6695e-02,  5.1384e-01, -4.2861e-01,  2.3148e-01,\n",
       "          -9.8841e-02,  1.8745e-01, -2.0415e-01,  3.5488e-01,  4.2969e-02,\n",
       "           3.2306e-01,  5.5474e-01, -1.0544e-02,  7.8583e-02, -1.8719e-01,\n",
       "          -3.1561e-01,  2.5043e-02, -4.9659e-02, -6.0277e-02, -3.0451e-01,\n",
       "          -5.3916e-02,  1.4698e-01, -5.0053e-01, -2.0416e-01, -2.3241e-01,\n",
       "           1.4052e-01, -2.4150e-01,  1.0012e-02,  3.3739e-02,  4.0760e-01,\n",
       "           1.1741e-01, -3.1554e-01, -3.7425e-01,  1.8211e-02,  1.0846e-01,\n",
       "           2.3058e-01, -1.6928e-02, -2.4001e-01, -8.9093e-01, -3.0032e-01,\n",
       "           2.8391e-02,  9.4187e-01, -5.3980e-01, -3.8487e-01,  3.6706e-01,\n",
       "           1.3515e-02, -2.5033e-01, -1.8306e-01, -6.2051e-01,  1.0878e-01,\n",
       "          -1.7889e-01, -1.9702e-01,  5.2566e-01,  2.5948e-03,  3.7097e-01,\n",
       "           5.0790e-01,  1.0635e-01, -1.1689e-02, -4.5858e-02, -4.5605e-01,\n",
       "          -6.9939e-01, -1.3259e-02, -3.2999e-01,  1.7426e-01, -3.2910e-01,\n",
       "          -2.4418e-01,  6.4175e-02, -1.5387e-02, -5.6779e-02,  7.2820e-01,\n",
       "           6.1920e-02,  1.2944e+00, -2.9247e-01,  5.6791e-04,  2.0640e-01,\n",
       "           1.7730e-01,  2.9784e-01,  2.7438e-01, -1.7091e-01,  8.2879e-02,\n",
       "           3.8866e-01, -2.1344e-01, -3.9032e-01,  2.1816e-01,  2.2756e-01,\n",
       "           3.9593e-01, -5.9076e-01, -2.7565e-01, -2.6625e-01,  6.4058e-01,\n",
       "           3.5704e-01, -3.3174e-02, -3.0307e-01,  2.8794e-01,  6.3242e-01,\n",
       "          -7.7053e-01,  2.6025e-01,  3.3501e-01,  1.2954e-01, -1.7979e-01,\n",
       "          -1.2486e-01, -3.4262e-02, -2.9787e-01, -3.1994e-01,  6.1839e-01,\n",
       "           7.7700e-01, -8.2809e-01,  5.0716e-01,  4.3278e-01, -8.7074e-02,\n",
       "           4.6662e-01,  7.1283e-02,  3.2832e-01,  3.8450e-01,  3.9792e-01,\n",
       "           3.7343e-01,  1.7883e-01, -2.9576e-01, -5.1922e-01,  2.4912e-01,\n",
       "           5.2804e-01,  1.6244e-01,  3.1933e-02,  8.2444e-02, -6.1699e-02,\n",
       "          -8.0348e-02,  4.0183e-02, -5.1247e-01, -2.8335e-01,  1.0346e-01,\n",
       "          -8.7673e-02, -2.7936e-01,  7.7286e-02,  3.0930e-01, -5.4603e-01,\n",
       "           5.2040e-01,  6.8617e-01,  4.1754e-01,  7.3055e-02, -1.0444e-01,\n",
       "          -2.2831e-01,  3.0040e-01, -2.4528e-01,  7.4886e-01, -1.0915e-01,\n",
       "          -8.1949e-02, -2.7973e-02,  1.1720e-01, -1.3769e-01,  3.4946e-01,\n",
       "          -3.0301e-01,  1.4474e-01, -2.4709e-01, -2.4159e-01,  1.9357e-01,\n",
       "          -3.3037e-02,  1.1210e-01, -6.5210e-01,  3.3298e-02,  2.3861e-01,\n",
       "          -4.1704e-01,  4.2070e-01, -1.8508e-01,  4.2032e-01,  5.0210e-02,\n",
       "           1.6898e-01, -6.4937e-01, -3.5741e-01,  1.6026e-01,  3.3484e-01,\n",
       "          -4.2869e-01, -5.2079e-01,  1.2549e-01,  1.3522e-01,  5.9171e-01,\n",
       "           3.5069e-01, -1.8784e-02, -4.7269e-01, -2.7303e-01,  2.5930e-01,\n",
       "           2.2028e-01, -3.3033e-01,  2.4769e-02, -1.2633e-01, -3.9334e-02,\n",
       "          -8.1783e-01, -1.6359e-01, -1.9666e-02, -4.6624e-04, -3.0209e-01,\n",
       "           8.2335e-01, -2.5908e-01,  9.1975e-02,  2.1194e-01,  1.1198e-01,\n",
       "           5.5523e-01,  7.0247e-01,  1.5578e-01, -6.2660e-01, -3.0862e-01,\n",
       "           7.5352e-03, -1.8532e-01, -2.1019e-01, -5.5806e-01,  4.0257e-01,\n",
       "           4.8804e-01, -7.9040e-02,  7.2806e-01]], device='mps:0'),\n",
       " tensor([[-9.2086e-02, -4.8197e-01, -2.0190e-01, -5.9037e-01, -1.1071e-01,\n",
       "          -2.9195e-01, -6.2197e-01,  5.0371e-01,  6.0618e-02,  6.8046e-01,\n",
       "           1.8329e-01,  5.3310e-01,  7.0717e-02,  3.7933e-01,  1.0348e-01,\n",
       "          -2.0846e-01, -1.9767e-01,  9.1832e-02,  5.9667e-01,  4.3009e-01,\n",
       "           3.6746e-01,  4.7944e-01, -1.9285e-01,  9.9497e-02, -7.6796e-01,\n",
       "           1.4208e-03,  1.3929e-01,  2.9815e-01,  2.5351e-01, -6.4524e-01,\n",
       "           1.4849e-01,  2.7661e-01, -2.0397e-01,  1.2183e+00, -1.2711e-01,\n",
       "          -4.0149e-01,  4.1354e-01, -3.3954e-01, -4.0150e-01, -2.6590e-01,\n",
       "           2.9271e-01, -6.8292e-02,  1.3947e-01, -2.5900e-01,  4.8931e-02,\n",
       "           4.6504e-01,  2.4977e-01, -1.7196e-01, -2.5199e-01, -6.3593e-01,\n",
       "          -4.0991e-01, -2.6780e-01,  3.4882e-01,  6.0937e-01, -9.8201e-01,\n",
       "          -5.6981e-01,  3.0391e-01, -1.5436e-01, -2.5300e-02,  8.2326e-02,\n",
       "          -5.4760e-02,  7.6222e-02,  5.0445e-02, -2.2423e-03,  3.3069e-01,\n",
       "          -2.2869e-02, -5.9596e-01, -5.7643e-01, -6.5190e-01, -6.2289e-02,\n",
       "          -2.2884e-01,  6.9878e-02,  3.2056e-01,  6.8141e-02, -3.3550e-02,\n",
       "          -2.2956e-01,  2.6993e-01,  5.9545e-02, -2.2806e-01, -8.7753e-02,\n",
       "          -1.6420e-01,  1.2161e-01,  6.0158e-01, -3.5439e-02, -3.2038e-01,\n",
       "          -6.0751e-02,  1.6030e-01, -9.9748e-03, -1.8588e-01,  3.8051e-01,\n",
       "           1.5305e-01,  1.8608e-01, -1.9030e-01, -2.1511e-01,  2.7049e-01,\n",
       "          -3.3092e-01,  5.0754e-01, -3.3147e-02,  1.3429e-01, -4.4257e-01,\n",
       "           3.2620e-02,  5.1491e-02, -9.0885e-02, -1.0599e-01, -2.6693e-01,\n",
       "          -4.5221e-01,  1.2064e-01, -2.5379e-01, -5.7395e-01,  8.8280e-02,\n",
       "          -2.2220e-02, -6.1123e-01, -2.6881e-03, -3.5817e-01,  2.4677e-02,\n",
       "           2.8344e-01,  1.5156e-01, -1.9539e-01, -1.3489e-01, -1.8903e-01,\n",
       "           1.4010e-02, -8.9530e-02,  2.6635e-01,  3.8004e-01,  1.2670e-01,\n",
       "          -1.4141e-01,  5.7842e-01, -1.5685e-01,  6.6243e-01,  6.5568e-01,\n",
       "          -1.6469e-03, -1.2322e-01, -3.4073e-01,  7.7581e-02,  4.6341e-01,\n",
       "           1.4431e-01, -2.5550e-01, -5.6231e-02, -5.4074e-02, -1.4377e-01,\n",
       "           5.0345e-02, -4.2292e-02, -4.2503e-01, -1.8331e-01,  9.1019e-02,\n",
       "           3.2221e-02, -4.0931e-02, -6.1281e-02,  8.8780e-02, -5.4016e-01,\n",
       "          -5.0061e-01, -5.9786e-01,  6.9657e-01,  2.2519e-01,  9.3603e-03,\n",
       "          -1.2679e-01,  5.2388e-01,  1.3619e-01, -1.1223e-01, -6.0731e-02,\n",
       "           1.2542e+00,  1.6560e-01,  1.4037e-01, -5.9317e-01, -1.9529e-01,\n",
       "           3.0616e-02, -1.5834e-01, -3.7651e-01,  7.6939e-01, -1.6199e-01,\n",
       "          -1.1112e-01, -3.7701e-01,  6.3450e-01,  4.6705e-02,  4.2139e-01,\n",
       "          -5.3575e-01,  4.0143e-01,  7.9570e-02, -2.0458e-01, -1.9048e-01,\n",
       "          -3.7309e-02, -1.1431e+00,  4.0618e-01, -2.0471e-01,  3.4391e-02,\n",
       "          -3.0971e-01, -6.2289e-02, -5.8693e-02,  1.1649e-01, -2.5613e-01,\n",
       "          -3.0149e-03,  1.9632e-01, -6.0814e-02,  6.7580e-01, -4.5352e-01,\n",
       "           4.1356e-01,  5.3317e-01,  7.5066e-01,  2.5426e-02,  3.3828e-02,\n",
       "           1.2101e-01,  2.1460e-02,  1.6664e-01,  2.3279e-01,  1.5729e-01,\n",
       "           3.4417e-01, -1.9893e-01, -5.7670e-02,  2.3358e-01,  5.7086e-02,\n",
       "          -2.8666e-01,  8.3690e-02, -3.7913e-01,  1.5502e-01, -2.0760e-01,\n",
       "           1.9538e-01, -6.4413e-01,  7.1710e-03,  3.6479e-01,  3.2722e-01,\n",
       "          -2.1529e-01, -1.4754e-01,  8.1077e-02, -7.5570e-02,  8.0352e-02,\n",
       "           1.9215e-01, -1.4712e-01,  1.1957e-01, -3.5424e-01,  2.9991e-02,\n",
       "           1.5137e-01, -1.1430e+00,  5.9113e-01,  2.9865e-01, -1.1323e-01,\n",
       "           3.6156e-01, -1.6498e-01, -4.8607e-02, -4.5083e-01,  4.1302e-01,\n",
       "          -7.7397e-02, -3.5692e-01, -7.7575e-01,  1.7005e-02,  1.4733e-01,\n",
       "           2.1131e-01, -1.1405e-01,  3.0299e-01,  8.4730e-01,  4.8139e-01,\n",
       "           1.0263e-01,  3.1459e-01,  5.9412e-01,  4.6380e-01, -1.6320e-01,\n",
       "          -1.2687e-01,  1.4503e-01, -7.1869e-01,  2.5899e-01, -3.8433e-01,\n",
       "          -1.5324e-01,  1.5400e-01,  2.9448e-01,  2.0072e-01, -2.7627e-01,\n",
       "           7.0039e-02,  4.6548e-01,  6.3781e-02, -3.4242e-01,  2.3966e-01,\n",
       "          -3.0607e-01,  4.4744e-01,  4.8999e-01, -1.9652e-01, -6.9931e-01,\n",
       "          -4.2576e-01,  1.0886e-01,  6.1474e-03, -1.2977e-01, -7.5840e-01,\n",
       "          -3.3383e-01,  8.7323e-03,  1.6064e-01, -2.0014e-01, -3.0644e-01,\n",
       "           3.2277e-01,  9.2511e-02, -3.8239e-02,  3.4460e-02, -2.4561e-01,\n",
       "          -4.4961e-01,  1.5630e-01, -3.0474e-01, -1.7796e-03,  4.0967e-01,\n",
       "           8.1471e-02, -8.9804e-03,  3.1713e-01, -2.3490e-02, -3.7222e-01,\n",
       "          -5.3801e-01,  5.2750e-02, -9.3251e-02, -1.2875e-01,  3.5134e-01,\n",
       "           1.0197e-02,  4.9450e-01,  4.3517e-01, -2.9942e-01, -2.9967e-01,\n",
       "          -2.6640e-01,  4.7598e-01, -1.6244e-01, -6.7480e-01,  1.3557e-01,\n",
       "          -9.3883e-01, -1.0624e-01,  3.2658e-02, -1.9918e-01, -1.6936e-01,\n",
       "           2.2003e-01,  3.3626e-01, -1.9524e-01, -1.0085e-02,  1.9694e-01,\n",
       "           1.9283e-01,  5.9212e-01, -1.8077e-01,  2.4326e-01, -5.2738e-01,\n",
       "          -3.2773e-02,  3.1968e-02, -4.8828e-01, -2.1142e-01,  2.1905e-02,\n",
       "           3.0562e-01,  1.5043e-02,  8.4870e-02, -3.4227e-01, -2.7563e-01,\n",
       "           3.9054e-01, -5.1638e-01, -4.1854e-02,  3.9404e-01,  1.1151e-01,\n",
       "           2.9527e-01,  1.1752e-01, -5.1686e-01,  2.4414e-01,  4.5157e-01,\n",
       "          -1.0332e-01,  6.2938e-02, -4.6901e-01, -2.0049e-01,  5.2640e-01,\n",
       "           2.9032e-01,  1.6498e-01, -2.4439e-02,  4.2961e-01,  4.8219e-01,\n",
       "          -2.9220e-01,  1.8311e-01, -5.5779e-01, -1.0355e-01,  2.5274e-01,\n",
       "           3.0454e-01,  4.4024e-01, -3.8690e-01,  2.3873e-01,  1.5434e-03,\n",
       "          -2.9249e-01, -2.8351e-01, -1.5356e-01,  8.9383e-02,  2.3062e-01,\n",
       "          -3.2934e-01, -3.3479e-01,  7.7064e-02, -5.2949e-01, -5.2586e-01,\n",
       "           8.5486e-02, -4.5315e-02,  1.4318e-01, -3.6044e-01, -4.3693e-01,\n",
       "          -1.5502e-01, -2.8389e-01,  6.9712e-01,  1.4931e-01,  1.3445e-02,\n",
       "           4.4259e-01,  5.0597e-02, -2.7951e-01,  1.3873e-01,  3.3306e-01,\n",
       "          -4.5019e-01,  2.9834e-01, -4.4544e-01,  2.2065e-01,  6.1727e-01,\n",
       "          -4.9585e-01,  5.1503e-02, -6.0808e-02,  1.7306e-01,  5.5549e-01,\n",
       "          -3.1273e-01, -5.8293e-02,  4.1096e-01, -4.9367e-01, -8.8170e-02,\n",
       "          -4.7055e-01,  9.7305e-01, -7.2546e-01,  1.5546e-01,  1.9316e-01,\n",
       "          -1.3104e-01, -1.9746e-01, -7.9436e-02,  1.5146e-01, -6.3497e-01,\n",
       "           3.5841e-01,  5.2547e-02,  1.5338e-01,  2.4367e-01, -1.1353e+01,\n",
       "           4.3187e-01,  6.7366e-01, -1.5212e-01, -1.4854e-01, -4.5326e-01,\n",
       "           5.1407e-02,  5.4346e-02,  3.8320e-01, -2.1948e-01,  6.5007e-02,\n",
       "          -1.0885e-01,  3.7192e-02, -5.6402e-02, -3.9486e-01, -5.0000e-02,\n",
       "          -5.7723e-02, -9.1414e-01, -9.2503e-03,  2.4729e-01,  4.1000e-01,\n",
       "          -1.4716e-01, -1.7361e-01,  2.1433e-01, -2.3525e-01, -7.3932e-02,\n",
       "          -2.5339e-01, -2.9121e-01,  7.9080e-02,  3.1555e-01, -1.8443e-01,\n",
       "           2.1625e-01,  1.4238e-01,  4.0122e-01, -2.5806e-01, -6.4264e-01,\n",
       "          -3.5594e-01,  2.0172e-01,  1.6474e-01, -5.4002e-03, -6.6306e-01,\n",
       "          -2.3261e-02,  1.4776e-01,  2.1766e-01,  2.6323e-01,  6.6509e-01,\n",
       "           8.8013e-01,  9.3747e-01,  3.7971e-01,  2.6990e-01, -9.5566e-02,\n",
       "          -1.0740e+00,  3.3014e-01, -7.1743e-01,  2.5806e-01,  1.1859e-02,\n",
       "           7.8259e-01, -3.6444e-01, -1.2713e-01, -1.3643e-01,  9.2640e-02,\n",
       "           1.7708e-01,  2.2257e-01,  2.0212e-01, -4.0000e-01,  1.2667e-01,\n",
       "           1.6114e-01, -7.2043e-02,  9.8674e-02, -4.2609e-01, -7.0089e-01,\n",
       "           1.2058e-01,  1.3170e-01,  1.5195e-01,  5.4804e-02,  9.5234e-02,\n",
       "          -3.4067e-01, -1.3180e-01,  1.4742e+00,  5.2120e-02, -2.0701e-02,\n",
       "          -1.2223e-01,  3.0441e-01, -2.6966e-01, -5.9000e-01,  1.5465e-01,\n",
       "          -1.7642e-01,  4.8292e-02, -2.0952e-01,  4.5616e-01, -4.9056e-01,\n",
       "          -1.4577e-01, -4.0523e-02,  3.4910e-01, -1.3023e-01,  8.2165e-01,\n",
       "          -4.9583e-01, -5.0404e-01,  4.4172e-02,  8.1729e-01, -3.0123e-01,\n",
       "          -3.2238e-01, -3.2496e-01, -3.7698e-01,  9.4086e-02, -1.4079e-02,\n",
       "          -2.4737e-01, -9.0646e-03,  3.0770e-02,  2.6239e-01,  8.1398e-02,\n",
       "           1.6357e-01,  3.2699e-01, -2.4127e-01, -3.8549e-01, -5.9168e-01,\n",
       "           1.2756e-02,  7.0303e-01, -4.5263e-01, -5.0067e-01, -2.4108e-01,\n",
       "           6.7335e-02,  6.9207e-02,  9.0022e-02, -7.3652e-01,  2.9660e-01,\n",
       "          -7.3014e-01, -9.1366e-02,  1.2519e-01, -2.2486e-02,  4.2864e-01,\n",
       "           3.2578e-01, -4.0876e-02,  1.1157e-01, -2.6828e-01, -6.5998e-01,\n",
       "          -9.0688e-01, -2.6323e-01,  5.8857e-01, -2.6846e-01,  1.9569e-01,\n",
       "           8.1888e-02, -1.3698e-01, -6.2709e-02, -6.4857e-02, -8.1550e-02,\n",
       "           7.9889e-02,  2.8959e-01, -5.4208e-01,  1.5989e-01,  4.0693e-02,\n",
       "          -5.6043e-01, -4.7732e-01, -2.6937e-01, -2.5716e-01, -2.5350e-01,\n",
       "          -3.2670e-02,  7.7460e-02, -7.1199e-01, -5.3613e-01,  2.7511e-01,\n",
       "           5.2260e-01, -2.8028e-02, -3.5211e-01,  5.9553e-01,  2.8115e-01,\n",
       "          -4.6250e-02, -6.3656e-01, -3.9283e-01, -2.1312e-01,  1.1390e-01,\n",
       "           7.6532e-02, -2.9874e-01, -6.0972e-01, -5.9176e-01, -3.7891e-01,\n",
       "          -2.3669e-01,  1.1123e+00, -1.6722e-01, -4.0755e-01,  2.0202e-01,\n",
       "           5.0988e-01, -1.7647e-01, -9.2337e-02,  2.7465e-01,  1.4562e-01,\n",
       "          -3.7469e-01, -2.3356e-01,  4.4695e-01, -8.9821e-02,  4.3503e-02,\n",
       "           3.6879e-01,  2.5913e-01, -4.4974e-01,  5.9566e-02, -2.3253e-01,\n",
       "          -6.1541e-01, -2.1743e-02, -2.0581e-01, -1.7544e-01,  3.0243e-02,\n",
       "          -2.5324e-01, -1.5516e-02,  4.3533e-01, -3.0562e-01, -1.0153e-01,\n",
       "           3.5366e-01,  1.0228e+00,  8.6149e-02, -9.0167e-02,  7.6040e-01,\n",
       "           4.8343e-01, -3.0315e-01,  2.8885e-02, -2.4901e-02,  4.1424e-02,\n",
       "           2.7037e-01,  3.2325e-01, -4.3462e-01, -1.2305e-01,  2.7722e-01,\n",
       "           6.1122e-01, -2.7832e-01, -4.0056e-01,  7.2018e-02,  4.2132e-01,\n",
       "          -5.0209e-01,  2.5734e-01, -9.1727e-02,  2.3432e-01, -5.5404e-03,\n",
       "          -9.0284e-01, -1.0773e-01, -2.0128e-01, -3.1759e-02, -4.7760e-02,\n",
       "          -3.3718e-01, -3.8096e-01, -7.1311e-02, -4.7060e-01,  4.5336e-01,\n",
       "           4.0753e-01, -8.8587e-01,  2.5150e-01,  4.4642e-01, -5.1075e-01,\n",
       "           4.4724e-01, -1.5802e-01,  2.7291e-01,  5.3564e-01, -1.8041e-01,\n",
       "          -6.4373e-02, -9.9363e-02, -5.8072e-01, -5.4219e-01,  2.7334e-01,\n",
       "           6.4579e-01, -9.8579e-03,  2.8085e-01, -1.6729e-01, -2.2467e-02,\n",
       "          -3.5824e-01, -1.3291e-01, -4.9040e-02, -1.0446e-01,  1.9719e-01,\n",
       "          -4.1679e-02, -1.4224e-01, -6.0728e-02,  3.3595e-01, -2.9911e-01,\n",
       "           2.0374e-01,  5.3520e-01,  3.5164e-01, -6.0028e-02, -1.7886e-01,\n",
       "           2.2633e-03,  6.0170e-01, -2.3362e-01,  8.7256e-01, -1.4766e-01,\n",
       "           9.6258e-03, -3.6608e-02,  5.0968e-02, -2.8862e-01,  1.7793e-01,\n",
       "          -4.1569e-01,  1.0496e-01, -4.4817e-02, -1.9413e-01,  5.9895e-01,\n",
       "          -1.1704e-01,  1.0670e-01, -8.6500e-02,  6.3739e-02,  3.7120e-01,\n",
       "           6.9825e-02,  8.8895e-01,  2.0992e-02,  3.0082e-01, -5.1239e-01,\n",
       "          -2.8862e-02, -7.0158e-01,  9.4725e-02, -3.9760e-02, -8.0033e-02,\n",
       "          -5.0277e-01,  7.9309e-02,  4.1400e-01,  5.8569e-04,  8.2560e-01,\n",
       "           6.2905e-01,  1.1662e-02, -3.9824e-01, -2.7094e-01,  6.5250e-02,\n",
       "           2.0714e-01,  1.9529e-01,  1.7528e-01, -2.0141e-02,  6.6970e-03,\n",
       "          -2.8417e-01, -2.8270e-01, -3.3956e-01,  1.6385e-01, -2.5470e-01,\n",
       "           4.2842e-01,  7.3024e-02,  1.7428e-01,  5.0469e-01,  2.5169e-01,\n",
       "          -4.7908e-03,  4.6980e-01,  1.8980e-01, -8.4330e-01, -1.2716e-01,\n",
       "           1.4561e-01, -2.1943e-01,  4.1072e-01, -6.4133e-01,  2.3010e-01,\n",
       "           5.2675e-01,  5.8504e-01,  5.5829e-01]], device='mps:0'),\n",
       " tensor([[-0.2485, -0.5584, -0.4741,  ...,  0.2258,  0.4312,  0.7559],\n",
       "         [-0.2427, -0.6267, -0.1893,  ...,  0.2335,  0.0749,  0.3544],\n",
       "         [-0.3292, -0.2590, -0.5016,  ..., -0.0148,  0.3509,  0.1607]],\n",
       "        device='mps:0'),\n",
       " tensor([[-0.0832, -0.7572, -0.9735,  ...,  0.2331,  0.1512,  0.6988],\n",
       "         [-0.2411, -0.5328, -0.3614,  ..., -0.0271,  0.5279,  0.4270],\n",
       "         [-0.1689, -0.4450, -0.6101,  ...,  0.4169,  0.1256,  0.6819]],\n",
       "        device='mps:0'),\n",
       " tensor([[-1.6463e-01, -6.2987e-01, -5.9491e-01, -8.7478e-01, -2.1700e-01,\n",
       "          -2.0336e-01, -8.2936e-01,  4.8916e-01,  2.4250e-01,  3.8665e-01,\n",
       "           1.1729e-01,  3.1057e-01,  3.8316e-02, -1.2584e-01, -2.9434e-01,\n",
       "          -2.6972e-01,  1.2207e-01,  4.2729e-01,  6.4028e-01, -6.6002e-02,\n",
       "           4.4404e-01,  5.9261e-01, -2.6728e-01, -6.5947e-03, -6.7512e-01,\n",
       "           2.2431e-01, -1.7137e-02,  4.1736e-01,  1.3046e-01, -6.0291e-01,\n",
       "           2.0874e-01,  4.7479e-01, -8.7799e-02,  1.3142e+00, -3.8592e-01,\n",
       "          -2.0139e-01,  8.5795e-02, -3.2643e-01, -1.5136e-01,  1.7652e-01,\n",
       "           6.5120e-01, -2.4499e-01, -1.0717e-01,  4.6042e-01,  2.2148e-02,\n",
       "           3.1235e-02,  1.8368e-01, -3.5098e-01,  4.9288e-01, -6.4433e-01,\n",
       "          -1.3215e-01, -1.2080e-01,  3.4523e-01,  3.6739e-01, -6.3042e-01,\n",
       "          -3.5579e-01,  3.1771e-01,  1.2236e-01, -2.0105e-02,  5.2927e-02,\n",
       "           1.4721e-02,  1.9159e-01, -3.1728e-01,  3.5220e-01,  1.0779e-01,\n",
       "           1.6108e-01, -4.2660e-01, -6.0987e-01, -5.8009e-01, -1.1271e-01,\n",
       "           5.6932e-02,  1.3088e-01,  2.4465e-01, -1.1914e-01,  3.9873e-01,\n",
       "          -8.6018e-01,  1.5567e-01, -3.7925e-02, -2.1723e-01,  2.6124e-01,\n",
       "          -2.0660e-01,  1.4472e-01,  2.7131e-01,  1.0551e-01,  2.3004e-01,\n",
       "           1.2352e-01,  3.8836e-01, -2.2385e-02,  8.0606e-02,  3.7966e-01,\n",
       "           9.5459e-03, -1.6299e-02, -4.9791e-02, -2.7074e-01,  1.4597e-01,\n",
       "           1.0283e-02,  3.7803e-01, -3.7144e-01, -2.2207e-02, -3.3699e-01,\n",
       "           1.4116e-01,  1.5095e-01, -2.6062e-02,  2.5036e-01, -5.4525e-01,\n",
       "          -4.5018e-01, -4.9731e-02, -2.5837e-01, -5.4973e-01,  1.4000e-01,\n",
       "          -3.0289e-01, -1.6754e-01,  6.9134e-02, -5.3704e-01,  1.6621e-01,\n",
       "          -1.4252e-01,  6.7407e-02, -2.7345e-02, -3.6052e-01, -5.3335e-02,\n",
       "           1.6791e-01, -3.6935e-01,  2.2958e-01,  7.7249e-02,  7.6346e-02,\n",
       "          -1.5792e-02,  6.3841e-01,  6.7344e-04,  2.3178e-01,  4.5671e-01,\n",
       "           1.3759e-01, -1.4423e-01, -2.4303e-01, -2.5499e-01, -1.1037e-01,\n",
       "           9.4719e-02,  2.4416e-01, -1.1443e-01, -4.2458e-01, -1.7138e-01,\n",
       "           3.8709e-01, -2.2040e-01, -4.7081e-01, -2.9592e-01, -2.4868e-01,\n",
       "          -3.2852e-01, -2.7708e-01, -1.1309e-01, -3.4374e-01, -5.5468e-01,\n",
       "          -6.0400e-01, -4.2181e-01,  8.4459e-01, -1.9217e-01, -2.6822e-01,\n",
       "          -1.8096e-01,  5.9869e-02,  1.5524e-01,  9.7654e-02,  4.8522e-01,\n",
       "           7.0164e-01, -3.0016e-01,  1.6184e-01, -5.3280e-01, -2.9208e-01,\n",
       "           2.0326e-01,  2.1472e-01, -6.0359e-02,  3.2878e-01,  2.2210e-02,\n",
       "           1.8229e-01, -1.2904e-01,  4.6832e-01, -2.7730e-01,  2.5480e-02,\n",
       "          -7.4897e-01,  5.3443e-01,  3.0671e-01, -3.5066e-01,  4.1661e-02,\n",
       "           8.9642e-02, -1.2211e+00,  7.0465e-02, -9.3871e-01,  5.1877e-01,\n",
       "          -7.7421e-02,  1.7014e-01,  2.1269e-01,  2.9861e-01, -1.2347e-01,\n",
       "           2.7480e-01,  1.3661e-01,  3.1198e-01,  3.5359e-01, -2.7697e-03,\n",
       "           2.2140e-01,  3.0792e-01,  4.0054e-01,  2.3283e-01,  2.2982e-01,\n",
       "          -5.4856e-01,  3.4385e-01,  3.7915e-01, -2.9584e-01,  2.8055e-01,\n",
       "           5.2847e-02, -3.6500e-01,  1.8934e-01, -4.8190e-02,  8.7815e-02,\n",
       "          -2.7963e-01, -2.3524e-01, -5.7314e-01, -1.7597e-01, -4.7477e-01,\n",
       "           2.0949e-01, -5.3230e-01, -2.6017e-01,  3.1297e-01,  1.8246e-01,\n",
       "          -2.9728e-02,  2.1614e-01,  1.6065e-01, -3.0667e-01,  1.5972e-01,\n",
       "           4.9853e-01, -5.3870e-01,  1.5897e-02, -1.6202e-01, -3.1372e-01,\n",
       "           2.6124e-01, -6.7454e-01,  4.9924e-01,  2.2036e-01,  2.4722e-01,\n",
       "           1.4398e-01, -2.0770e-01,  5.2168e-04, -5.4831e-01,  1.6056e-01,\n",
       "          -3.7938e-01, -3.8020e-01, -6.3491e-01, -2.3992e-03,  2.6465e-01,\n",
       "           1.2630e-01, -5.3641e-02,  6.8822e-01,  5.0221e-01,  4.9185e-01,\n",
       "          -3.7572e-01, -1.0667e-01,  4.9007e-01, -1.8869e-02, -4.9092e-01,\n",
       "           3.0384e-01, -6.0874e-02, -3.9836e-01,  2.9423e-01, -3.4384e-01,\n",
       "          -2.1663e-01, -1.5437e-01,  4.1186e-01,  2.7257e-01, -2.6826e-01,\n",
       "          -3.4368e-01,  8.1981e-02, -7.8358e-03, -2.9391e-01,  6.3611e-01,\n",
       "          -2.3484e-01,  1.6300e-02, -1.2838e-01, -3.2362e-01, -6.5488e-01,\n",
       "          -3.5663e-01,  3.2779e-01,  2.0480e-01,  1.2334e-01, -7.1848e-01,\n",
       "          -4.0515e-01,  4.5434e-01,  3.4615e-01,  2.4363e-01,  1.9003e-01,\n",
       "           5.2387e-01, -4.9199e-01,  5.6070e-02,  1.0032e-01, -4.2104e-01,\n",
       "          -5.2305e-02,  8.2874e-03, -5.0788e-01,  3.6136e-01,  3.6076e-01,\n",
       "          -4.1847e-02,  3.3360e-01,  4.9221e-01, -1.6382e-01, -4.5178e-01,\n",
       "          -4.8624e-02,  3.4233e-01, -3.5644e-01,  3.5331e-02,  1.6714e-01,\n",
       "           1.2276e-02,  5.5694e-01,  6.1463e-01, -4.5377e-02, -5.5626e-01,\n",
       "           3.5769e-01,  6.4229e-01, -1.5019e-01, -2.2446e-01,  2.1123e-02,\n",
       "          -8.1552e-01, -6.7352e-02, -1.6750e-01, -3.8302e-01, -8.4950e-01,\n",
       "          -6.2288e-02,  1.6949e-01, -4.6078e-02, -1.6614e-01, -5.0993e-01,\n",
       "           2.4136e-01,  2.9618e-01,  1.0102e-01,  2.4892e-01, -1.5047e-01,\n",
       "          -4.7821e-01, -2.0784e-01, -3.1303e-01, -5.7959e-01, -6.2894e-01,\n",
       "           2.7068e-01,  3.2294e-01,  5.1956e-02, -2.1313e-01, -2.2139e-01,\n",
       "           3.9853e-01, -5.6407e-01, -2.7864e-01,  3.1518e-01, -3.8869e-01,\n",
       "          -7.4271e-02,  3.1308e-01,  4.2101e-01, -4.4025e-02, -4.6946e-02,\n",
       "          -7.8398e-03, -5.0189e-01, -5.5504e-01, -2.6592e-01,  5.7629e-01,\n",
       "           4.7239e-01,  2.6343e-01, -4.3606e-01,  3.2872e-01,  4.9554e-01,\n",
       "          -3.3007e-01,  2.9133e-01,  2.1073e-01,  4.6633e-02, -2.4488e-01,\n",
       "           2.9313e-01,  5.3606e-01, -3.6562e-01,  5.3967e-02,  1.9392e-01,\n",
       "          -2.2816e-01, -2.1657e-01, -1.8747e-01,  2.7192e-01, -2.8113e-01,\n",
       "          -3.1234e-01, -1.8871e-01,  1.7532e-02, -2.0115e-01, -3.4979e-01,\n",
       "           3.4833e-01,  8.5896e-02, -7.9513e-02, -3.5707e-01, -4.1229e-01,\n",
       "          -1.6507e-01, -3.2432e-01,  6.9834e-01,  1.5338e-01, -1.4228e-01,\n",
       "           3.3665e-01, -9.0682e-02, -4.0066e-01,  4.6967e-01, -7.0365e-03,\n",
       "          -5.2839e-01,  5.5518e-01, -1.2325e-01,  1.0326e-01,  3.9681e-01,\n",
       "          -1.4070e-01,  7.0161e-02,  1.2451e-01,  3.8194e-01,  3.6549e-01,\n",
       "          -2.7507e-01,  6.2256e-02,  8.2569e-02, -3.8922e-01,  3.9837e-01,\n",
       "           3.2644e-01, -6.7971e-02, -2.3194e-01,  1.2435e-01, -6.1024e-02,\n",
       "           2.0518e-01,  9.0217e-02, -4.8852e-01, -5.8374e-01, -3.3657e-01,\n",
       "           3.2339e-01,  1.4435e-01,  5.4899e-01,  2.5475e-01, -1.1235e+01,\n",
       "           7.4443e-02,  8.9649e-01, -4.5782e-01, -3.3704e-01, -1.4448e-01,\n",
       "          -8.0281e-02, -9.8865e-02,  4.0282e-01, -2.1118e-01, -4.8351e-01,\n",
       "          -4.0287e-01,  1.0217e-02,  1.0951e-01, -4.5019e-01, -2.9744e-01,\n",
       "           2.6881e-02, -5.6494e-01,  4.1648e-01, -1.6170e-01,  5.4023e-01,\n",
       "          -3.9491e-01,  5.9613e-02,  1.6446e-01, -3.5262e-01,  1.1604e-01,\n",
       "           1.5545e-02, -1.4863e-02,  2.6880e-01,  6.0715e-01, -1.1844e-01,\n",
       "           2.1563e-01, -2.9074e-02, -1.7177e-02, -4.3625e-01, -4.2844e-01,\n",
       "           2.0475e-01, -1.3481e-01,  7.2580e-01,  2.8171e-02, -8.1600e-01,\n",
       "          -1.8184e-01, -3.3844e-01,  2.4657e-01,  3.6242e-01,  8.3710e-01,\n",
       "           5.3237e-01,  1.1983e+00,  1.4850e-01,  1.8626e-01, -9.1346e-02,\n",
       "          -8.3781e-01,  1.3912e-01, -2.7175e-01,  1.1582e-03,  1.7560e-01,\n",
       "           2.4348e-01, -7.4361e-01,  2.2386e-01, -4.8452e-01,  2.7767e-02,\n",
       "           6.6164e-01,  5.2310e-02,  3.8033e-01, -2.4301e-01, -1.5607e-01,\n",
       "           2.0409e-01, -8.6334e-02,  2.8390e-01, -3.5716e-01, -3.8284e-01,\n",
       "          -1.9444e-01,  6.1009e-01,  3.7326e-01, -1.8706e-01,  2.6314e-01,\n",
       "           9.4880e-02, -2.5121e-01,  1.6192e+00,  7.9465e-02, -8.4777e-03,\n",
       "           1.8613e-01,  2.9204e-01, -4.3477e-01, -7.4140e-01,  1.5682e-01,\n",
       "          -1.4927e-01, -8.4614e-03,  1.8699e-01,  3.9803e-01, -1.2880e-01,\n",
       "          -6.7369e-01, -1.5170e-01,  2.0668e-01,  4.3763e-02,  5.5442e-01,\n",
       "          -7.2413e-01, -1.9327e-01, -2.1508e-01,  2.6601e-01, -1.0039e+00,\n",
       "           3.4616e-02, -6.7629e-01, -4.5550e-01,  5.7226e-01, -1.8857e-01,\n",
       "           1.3537e-01, -1.8299e-01,  9.0334e-02,  6.8122e-01, -1.5635e-01,\n",
       "           1.2207e-01, -4.6982e-02,  1.9227e-01, -2.8020e-01, -5.9959e-02,\n",
       "           3.9894e-01,  3.5282e-01, -4.0664e-01, -6.4802e-01, -5.4346e-02,\n",
       "           2.7406e-01, -5.4682e-02,  5.7432e-01, -6.0554e-01,  1.0472e-01,\n",
       "          -7.6609e-01,  1.7642e-03,  1.4169e-01, -5.0839e-02,  3.7317e-01,\n",
       "           7.9992e-02, -6.6711e-02,  2.9936e-01, -7.3280e-01, -4.3822e-01,\n",
       "          -6.8158e-01, -3.4103e-01,  2.5181e-01, -6.5217e-01, -1.9971e-01,\n",
       "          -1.9098e-01, -7.6230e-02, -3.4876e-02, -7.4858e-03, -4.6451e-02,\n",
       "           2.9607e-01,  1.2213e-01, -1.9058e-01, -1.0918e-01, -6.7776e-02,\n",
       "          -6.7437e-01, -7.0749e-02, -4.5643e-02, -2.1728e-02, -3.0115e-01,\n",
       "          -1.3757e-01, -3.1506e-03, -4.1711e-01, -2.3814e-01,  1.8353e-01,\n",
       "          -6.8982e-02, -6.8373e-02, -5.3997e-01,  8.1370e-01,  3.8834e-01,\n",
       "           2.7003e-02, -1.3485e-01, -3.8264e-01, -1.8808e-01,  2.4116e-01,\n",
       "           2.8589e-01, -9.1656e-02,  2.1301e-02, -3.9404e-01, -5.7616e-01,\n",
       "           8.4876e-02,  7.9514e-01, -1.3994e-01, -7.3410e-01,  1.6504e-01,\n",
       "           3.3562e-01, -2.2904e-01, -2.1350e-01,  1.3330e-01,  5.1994e-02,\n",
       "          -1.8567e-01, -5.3589e-01,  5.9451e-01, -1.2079e-01, -5.7221e-03,\n",
       "           1.9094e-01, -3.8488e-02, -1.5186e-02,  5.7142e-01,  2.3102e-01,\n",
       "          -3.6051e-01, -1.3530e-02, -6.3962e-01,  3.2304e-01, -1.0206e-01,\n",
       "          -3.2911e-01,  8.8233e-01,  1.4349e-01, -3.9878e-02, -5.8033e-02,\n",
       "           2.8933e-01,  7.6854e-01, -6.1313e-02,  1.5558e-01,  5.2002e-01,\n",
       "           1.4286e-01,  3.0896e-01,  2.1588e-01,  2.6079e-01,  9.8689e-02,\n",
       "           1.8055e-01, -2.0737e-01, -2.8836e-01, -1.5873e-02,  3.2246e-01,\n",
       "           3.3397e-01, -5.1201e-01, -6.6558e-01,  1.1189e-01,  5.2264e-01,\n",
       "          -2.3232e-02,  2.9175e-01, -2.6605e-01,  3.8845e-01,  4.5679e-01,\n",
       "          -8.3972e-01, -1.5682e-01, -1.2207e-02,  1.2224e-01, -1.0263e-01,\n",
       "          -3.7550e-01,  1.2262e-01, -2.1591e-01, -1.0958e-01,  6.7103e-01,\n",
       "           4.5117e-01, -7.5916e-01,  2.0372e-02,  7.2381e-01, -4.4436e-01,\n",
       "           3.1189e-01, -4.7181e-01,  1.3724e-01,  3.9517e-01,  2.6571e-01,\n",
       "           1.0533e-01,  1.8197e-01, -6.3467e-01, -4.1184e-01,  1.5987e-01,\n",
       "           9.4693e-01, -1.0792e-01,  1.2216e-01, -3.7231e-01, -2.3794e-01,\n",
       "          -1.1578e-01, -3.7518e-01, -3.3161e-01,  2.0421e-01, -7.0193e-02,\n",
       "          -2.3596e-04, -4.7337e-02, -1.7414e-01,  2.4538e-01,  7.4256e-02,\n",
       "           1.4345e-01,  6.9872e-01,  5.6507e-01, -2.2021e-02, -1.6569e-01,\n",
       "          -2.6327e-01,  9.2934e-02, -1.3871e-01,  3.6364e-01, -3.2259e-01,\n",
       "          -2.4633e-01, -6.5084e-02,  2.7859e-01, -6.3065e-01,  1.4644e-01,\n",
       "          -5.8094e-01,  3.1744e-01, -1.5674e-01, -1.6243e-02,  2.5572e-01,\n",
       "           1.1499e-01,  1.5172e-01, -3.1603e-01,  5.0661e-01,  4.5451e-01,\n",
       "           1.7379e-01,  3.8553e-01,  5.8656e-02,  9.9802e-02, -3.1322e-01,\n",
       "           1.0911e-01, -9.5036e-01, -1.3614e-01, -2.3175e-01,  7.2286e-02,\n",
       "          -2.7482e-01, -5.7624e-02,  4.9664e-01,  3.4990e-01,  8.4452e-01,\n",
       "           6.1083e-01, -1.3134e-01, -5.5092e-01, -1.1634e-01,  1.5570e-01,\n",
       "           3.0229e-01, -1.8740e-01,  4.1225e-01,  3.7789e-02,  1.7598e-01,\n",
       "          -4.7409e-01, -4.6931e-01, -1.6163e-02,  3.9329e-01,  4.6267e-03,\n",
       "           7.3195e-01, -1.6405e-01, -2.5740e-01,  3.0336e-02,  1.4914e-01,\n",
       "          -5.6990e-03,  9.9034e-01,  2.8344e-01, -5.6885e-01,  1.7748e-01,\n",
       "          -1.2519e-01, -2.1606e-01,  2.7562e-01, -4.1079e-01,  2.2364e-01,\n",
       "           8.3334e-02,  6.8286e-01,  7.2325e-01]], device='mps:0'),\n",
       " tensor([[-2.1374e-01, -4.1789e-01, -6.4773e-01, -9.3585e-01, -3.4490e-01,\n",
       "          -3.8145e-01, -7.7210e-01,  4.5968e-01, -1.1754e-01,  4.5442e-01,\n",
       "           4.2445e-02,  2.1857e-01,  1.5448e-01, -8.7442e-02, -2.3614e-01,\n",
       "          -3.6181e-01,  2.4301e-01,  5.3184e-01,  7.7685e-01,  1.7059e-01,\n",
       "           4.8546e-01,  3.9211e-01, -8.5595e-02, -4.2957e-02, -8.3557e-01,\n",
       "           3.4112e-01, -1.3769e-01,  2.3915e-01,  1.3328e-01, -6.8400e-01,\n",
       "          -5.7738e-02,  1.5067e-01, -9.8993e-02,  1.1729e+00, -3.7669e-01,\n",
       "          -1.1761e-01,  5.2211e-02, -6.3374e-01, -2.6589e-01,  3.4828e-02,\n",
       "           5.5268e-01, -3.1840e-01, -1.1159e-01,  4.2679e-01,  3.3644e-02,\n",
       "           5.9168e-02,  7.7695e-02, -3.5325e-01,  8.3465e-02, -4.3828e-01,\n",
       "           1.6898e-01, -3.4962e-02,  5.0655e-01,  5.6027e-01, -1.1157e+00,\n",
       "          -6.4546e-01,  4.3292e-01,  1.1349e-01, -1.9004e-01,  2.7023e-01,\n",
       "          -1.8749e-01,  3.0694e-01, -7.9324e-04,  3.7088e-01,  5.7529e-02,\n",
       "           1.9124e-01, -3.1539e-01, -7.2613e-01, -6.0047e-01, -2.5681e-01,\n",
       "           2.5155e-02,  2.0554e-01,  2.0169e-01, -1.7413e-01,  1.7997e-01,\n",
       "          -6.3532e-01,  1.9691e-01, -3.3564e-02, -2.1323e-01,  8.1501e-02,\n",
       "          -1.8476e-02, -1.0946e-02,  2.5954e-01, -2.7559e-02, -9.6152e-03,\n",
       "           2.0052e-01,  3.0141e-01,  1.2187e-01,  2.7567e-01,  2.5565e-01,\n",
       "           9.8598e-02, -6.5957e-03, -1.0956e-01, -2.9580e-01,  3.5533e-01,\n",
       "          -1.3507e-01,  3.3655e-01, -2.2593e-01, -1.0639e-01, -3.3594e-01,\n",
       "           9.3589e-02,  1.9999e-01, -2.0325e-01,  2.4985e-01, -4.6360e-01,\n",
       "          -5.4171e-01, -1.6627e-02, -2.3153e-01, -7.2886e-01,  1.0187e-01,\n",
       "          -1.8923e-01, -3.0166e-01, -1.2648e-01, -3.9433e-01, -2.5180e-01,\n",
       "          -2.1369e-01, -1.9414e-01, -1.1336e-02, -3.7676e-01,  3.6846e-01,\n",
       "           2.1950e-01, -1.4221e-01,  2.8269e-01,  8.0385e-02, -1.9975e-02,\n",
       "          -2.2541e-01,  4.2704e-01,  1.3970e-01,  1.7302e-01,  2.7546e-01,\n",
       "           6.6832e-02,  1.4232e-01, -1.4682e-01, -1.9766e-01,  2.2698e-02,\n",
       "           1.2296e-02,  2.5792e-01, -1.9773e-01, -4.4254e-01, -1.7442e-01,\n",
       "           2.3555e-01, -1.1806e-01, -3.5986e-01, -1.9027e-01, -1.7552e-01,\n",
       "          -2.3501e-01, -1.6710e-01, -2.5847e-02, -2.7861e-01, -4.5325e-01,\n",
       "          -4.2872e-01, -3.2483e-01,  6.5757e-01, -1.4811e-01, -3.5766e-01,\n",
       "          -2.1211e-01,  2.8451e-01,  7.2395e-02, -4.3301e-02,  3.7433e-01,\n",
       "           8.0563e-01, -6.4416e-02,  9.4677e-02, -3.1575e-01, -2.8272e-01,\n",
       "           2.9598e-01, -8.0853e-02, -1.5320e-01,  1.7030e-01,  2.2874e-03,\n",
       "           2.4934e-01, -1.6088e-02,  3.7246e-01,  7.0127e-02,  3.8139e-02,\n",
       "          -7.6008e-01,  6.9187e-01,  3.1800e-01, -2.4973e-01,  5.4468e-02,\n",
       "          -2.7664e-01, -1.1622e+00,  2.1320e-01, -8.0119e-01,  3.2994e-01,\n",
       "          -2.0356e-01,  3.0482e-01,  4.2133e-01,  1.2820e-01, -1.0745e-01,\n",
       "           4.0450e-01,  3.5326e-01,  4.1690e-01,  4.6778e-01, -1.6175e-01,\n",
       "           2.8374e-01,  3.9215e-01,  3.8519e-01,  3.8804e-01,  1.4029e-01,\n",
       "          -6.4597e-01,  3.1031e-01,  2.6258e-01,  1.5265e-01,  2.1252e-01,\n",
       "           9.2459e-02, -6.8764e-02,  1.6159e-01,  1.9082e-01,  9.6511e-02,\n",
       "          -1.6731e-01, -2.6168e-01, -6.4529e-01, -1.9831e-01, -2.8915e-01,\n",
       "           3.5401e-01, -3.8008e-01, -2.3934e-01,  4.2013e-01,  2.4468e-01,\n",
       "          -8.1316e-02,  1.8481e-01, -6.2412e-02, -2.4457e-01, -1.7044e-03,\n",
       "           5.1820e-01, -2.7648e-01,  1.3263e-01, -1.5809e-01, -2.1962e-01,\n",
       "           3.7146e-01, -7.2545e-01,  7.0767e-01,  2.1966e-01,  2.4327e-01,\n",
       "           1.3273e-01, -2.8462e-01,  5.2727e-02, -4.2804e-01,  2.3817e-01,\n",
       "          -3.1911e-01, -4.5932e-01, -5.1262e-01, -5.4876e-02,  3.9605e-01,\n",
       "           2.7480e-01, -1.0011e-01,  6.1763e-01,  6.9664e-01,  3.9389e-01,\n",
       "          -2.8424e-01, -2.6906e-01,  4.5971e-01,  2.9208e-02, -5.2151e-01,\n",
       "           5.0230e-01, -1.2128e-01, -5.0275e-01,  1.4813e-01, -4.3880e-01,\n",
       "          -1.8980e-01, -1.0810e-01,  3.1013e-01,  2.2142e-01, -3.2660e-01,\n",
       "          -1.5045e-01,  2.2084e-01, -1.0605e-01, -3.6994e-01,  2.6772e-01,\n",
       "          -3.3683e-01,  2.0308e-01,  1.8159e-01, -5.3247e-01, -6.1613e-01,\n",
       "          -4.2299e-01,  1.2798e-01,  2.4644e-01,  7.2552e-02, -6.6007e-01,\n",
       "          -3.2566e-01,  4.4547e-01,  1.8553e-02,  5.7761e-02,  1.9840e-01,\n",
       "           4.9248e-01, -3.2706e-01, -1.3739e-01,  3.0310e-01, -4.8194e-01,\n",
       "          -1.4166e-01, -7.0686e-02, -3.2982e-01,  3.3963e-01,  2.3046e-01,\n",
       "          -9.5350e-02,  1.5568e-01,  2.2227e-01, -5.9483e-02, -3.8127e-01,\n",
       "           3.6000e-02,  1.3694e-01, -2.8310e-01, -1.4701e-01, -7.6747e-02,\n",
       "          -2.6607e-02,  4.0325e-01,  4.8121e-01, -4.0247e-02, -6.4933e-01,\n",
       "           2.3979e-01,  8.3477e-01, -6.5032e-02, -1.8632e-01,  3.7088e-03,\n",
       "          -8.3181e-01, -7.5899e-02, -1.5869e-03, -2.8185e-01, -7.0652e-01,\n",
       "           1.5215e-01, -2.4246e-01, -2.9745e-02, -8.7331e-02, -5.7316e-01,\n",
       "           4.3513e-01,  4.0459e-01, -4.1687e-02,  2.8568e-01, -1.7510e-01,\n",
       "          -3.5387e-01, -2.5231e-01, -2.4050e-01, -4.4412e-01, -4.8730e-01,\n",
       "           7.7798e-02,  3.6803e-01,  8.6129e-02, -2.3415e-01, -2.7398e-01,\n",
       "           3.8864e-01, -4.3325e-01, -3.0761e-02,  2.3625e-01, -3.9621e-01,\n",
       "           4.6639e-02,  5.2722e-01, -6.6328e-02,  5.0478e-02, -2.3796e-01,\n",
       "           5.7397e-02, -2.8028e-01, -5.3788e-01,  2.1951e-02,  6.2575e-01,\n",
       "           4.0348e-01,  2.4496e-01, -2.7057e-01,  5.1802e-01,  3.5148e-01,\n",
       "          -4.5261e-01,  4.8778e-01,  3.0578e-01, -8.7017e-02, -3.2478e-01,\n",
       "           9.6757e-02,  4.0303e-01, -3.9010e-01, -1.4361e-01,  6.3006e-02,\n",
       "          -3.4007e-02, -4.1749e-01,  2.2563e-02,  3.5636e-01, -3.2140e-02,\n",
       "          -3.2070e-01, -2.5156e-01,  1.4719e-01, -3.2250e-01, -3.5637e-01,\n",
       "           3.2552e-01,  1.7828e-02, -1.7372e-02, -3.0453e-01, -1.8319e-01,\n",
       "          -2.2816e-01, -1.3252e-01,  6.7040e-01,  4.6758e-01, -3.6865e-01,\n",
       "           1.6871e-01, -1.1599e-01, -4.2776e-01,  3.6264e-01, -1.7600e-02,\n",
       "          -5.1208e-01,  3.6987e-01, -1.1112e-01,  4.8165e-02,  3.7687e-01,\n",
       "          -2.2437e-01,  6.5071e-02,  1.0887e-01,  1.4991e-01,  2.6639e-01,\n",
       "          -2.8520e-01,  1.0783e-01,  1.0486e-02, -2.8781e-01,  2.5575e-01,\n",
       "           3.1945e-01,  6.5802e-02, -3.4772e-01,  3.7791e-01,  2.1509e-02,\n",
       "           1.7009e-01, -4.3126e-03, -2.4388e-01, -5.5055e-01, -4.9075e-01,\n",
       "           5.1662e-01,  4.1648e-01,  4.8007e-01,  4.4126e-01, -1.1266e+01,\n",
       "          -2.1080e-01,  7.9493e-01, -5.4635e-01, -4.7009e-01, -1.1475e-01,\n",
       "          -2.1251e-02, -1.8203e-01,  2.7993e-01, -1.6074e-01, -4.8186e-01,\n",
       "          -3.9433e-01, -1.5353e-01,  1.2933e-01, -2.6278e-01, -4.3479e-01,\n",
       "          -1.1155e-01, -6.1596e-01,  3.4689e-01, -1.5671e-01,  6.5727e-01,\n",
       "          -4.2190e-01,  4.8057e-02,  2.5571e-01, -4.2364e-01, -5.2303e-02,\n",
       "           6.0043e-02,  4.3702e-03,  3.2885e-01,  5.2785e-01,  3.6712e-02,\n",
       "           2.2000e-01, -1.3352e-01,  1.2091e-02, -2.4410e-01, -4.0329e-01,\n",
       "           5.0767e-03, -2.9720e-01,  5.5485e-01, -7.9165e-03, -8.0791e-01,\n",
       "          -4.6572e-01, -3.3362e-01,  3.1478e-01,  2.5853e-01,  1.0088e+00,\n",
       "           5.7860e-01,  8.1256e-01,  2.6285e-01,  3.9684e-02,  5.3377e-02,\n",
       "          -7.7745e-01,  1.9334e-01, -3.6117e-01, -2.5251e-01,  1.7019e-01,\n",
       "           4.8390e-01, -9.0470e-01,  1.6182e-01, -3.6102e-01, -1.6399e-01,\n",
       "           6.0990e-01,  1.0735e-01,  2.2624e-01, -1.4571e-01, -2.8517e-01,\n",
       "           3.5492e-01, -1.0143e-02,  2.4149e-01, -2.8186e-01, -3.9726e-01,\n",
       "          -2.5761e-01,  6.0326e-01,  3.4197e-01, -2.7532e-01,  1.2347e-01,\n",
       "           5.0979e-02, -2.4549e-01,  1.9444e+00,  3.0058e-02, -9.2630e-03,\n",
       "           2.8775e-01,  3.9145e-01, -4.6076e-01, -9.9389e-01,  1.1364e-01,\n",
       "           5.8445e-02,  8.2622e-02,  2.0715e-01,  4.7907e-01, -1.2979e-01,\n",
       "          -7.2391e-01, -2.3142e-02,  8.3385e-02,  1.9624e-01,  2.7048e-01,\n",
       "          -6.9974e-01, -3.8831e-02, -2.3404e-03,  2.2731e-01, -8.7517e-01,\n",
       "           8.2326e-02, -5.9250e-01, -2.4369e-01,  6.1885e-01, -1.3374e-01,\n",
       "           1.6656e-02, -1.8601e-01, -3.0624e-02,  5.3508e-01, -2.3770e-01,\n",
       "           1.5526e-01, -1.0903e-01, -4.3648e-03, -1.7314e-01,  4.9110e-02,\n",
       "           4.2204e-01,  5.2262e-01, -2.6088e-01, -5.2956e-01, -3.0345e-02,\n",
       "           9.0790e-02, -1.3192e-01,  5.0779e-01, -4.4656e-01,  5.2441e-01,\n",
       "          -6.9319e-01,  1.1125e-01,  2.7735e-01, -2.7792e-02,  2.8212e-01,\n",
       "          -1.6937e-02,  2.1732e-01,  4.9735e-01, -7.9085e-01, -7.1376e-01,\n",
       "          -6.6204e-01, -2.6182e-02,  8.7541e-02, -4.8449e-01, -1.1029e-03,\n",
       "           3.1675e-02,  1.6356e-01,  7.3569e-03, -8.6134e-02,  1.6984e-01,\n",
       "           8.0499e-02,  3.9390e-01, -1.8564e-01, -4.7513e-02, -1.6594e-01,\n",
       "          -6.2910e-01,  7.6598e-02, -6.5037e-03,  9.9224e-02, -4.7010e-01,\n",
       "          -2.1034e-01,  7.8415e-02, -3.5908e-01, -1.8333e-01,  2.0042e-01,\n",
       "          -1.3797e-01, -1.2811e-02, -3.7519e-01,  4.5804e-01,  5.4104e-01,\n",
       "           2.4576e-01, -3.6135e-01, -1.2394e-01, -2.8791e-01,  3.4703e-01,\n",
       "           2.5953e-01,  1.3830e-01, -6.4463e-02, -3.2856e-01, -4.8153e-01,\n",
       "           1.3464e-01,  7.8122e-01, -2.7804e-02, -6.3580e-01,  3.2466e-01,\n",
       "           4.1010e-01, -7.9924e-02, -3.1801e-01,  4.0212e-02, -1.2595e-01,\n",
       "          -3.3189e-01, -6.8359e-01,  7.1652e-01,  4.1259e-02,  1.2982e-01,\n",
       "           4.7270e-01, -8.2137e-02, -1.4601e-01,  5.8005e-01, -3.1975e-01,\n",
       "          -4.8246e-01, -1.2714e-01, -5.8561e-01,  3.1878e-01, -9.5162e-02,\n",
       "          -4.3742e-01,  6.5288e-01,  1.4492e-01,  1.5018e-02,  1.2098e-01,\n",
       "           2.1061e-01,  9.0740e-01, -2.4010e-01,  1.9673e-02,  6.6308e-01,\n",
       "           3.8213e-01,  2.6478e-01,  5.5004e-02,  2.6593e-01,  1.9846e-01,\n",
       "           3.1756e-01, -2.3130e-01, -1.7685e-01,  5.1132e-02,  2.9127e-01,\n",
       "           5.6388e-01, -5.8258e-01, -5.2930e-01,  3.0171e-02,  6.0979e-01,\n",
       "          -1.2362e-01,  1.9453e-01, -2.8567e-01,  2.5614e-01,  4.7527e-01,\n",
       "          -1.0728e+00, -2.2717e-01, -2.4568e-01,  3.2974e-01,  2.7485e-02,\n",
       "          -1.7628e-01,  1.2252e-01, -4.2565e-01, -8.1192e-02,  5.5772e-01,\n",
       "           2.0563e-01, -7.8044e-01,  1.2406e-01,  9.3490e-01, -4.7844e-01,\n",
       "           3.3257e-01, -4.2001e-01,  9.6496e-02,  1.5913e-01,  4.6652e-01,\n",
       "           2.1216e-01,  1.9599e-01, -4.4662e-01, -5.5881e-01,  1.4481e-01,\n",
       "           9.7615e-01, -5.9284e-02, -1.8270e-02, -2.4029e-01,  1.6455e-01,\n",
       "          -1.2898e-01, -3.9030e-01, -4.3616e-01,  1.3861e-01, -7.8671e-02,\n",
       "          -1.5589e-02, -8.3941e-02, -3.0303e-02,  1.3432e-01,  7.0680e-02,\n",
       "           1.9925e-01,  6.7346e-01,  7.2547e-01, -8.1051e-02, -3.7135e-01,\n",
       "          -1.0065e-01,  9.4265e-02, -2.5947e-01,  3.7618e-01, -2.8229e-01,\n",
       "          -3.0379e-01,  1.7423e-01,  2.6034e-01, -4.8834e-01,  2.1460e-01,\n",
       "          -3.0686e-01,  2.2544e-01, -1.2043e-01, -9.2938e-02,  2.2098e-01,\n",
       "          -1.1805e-01,  8.2422e-02, -1.8304e-01,  4.0287e-01,  4.6594e-01,\n",
       "          -1.8534e-01,  5.7513e-01,  5.1620e-03,  2.0391e-01, -4.4542e-01,\n",
       "           1.8728e-01, -1.1704e+00, -3.1036e-01,  8.8097e-03, -1.5700e-01,\n",
       "          -3.0110e-01, -1.0371e-01,  1.3871e-01,  4.7985e-01,  5.6383e-01,\n",
       "           3.9672e-01, -2.4277e-02, -7.5231e-01, -3.1351e-01,  3.1751e-01,\n",
       "           2.0978e-01, -2.0918e-01,  3.9677e-01,  8.8142e-02,  1.2955e-01,\n",
       "          -5.1388e-01, -5.5430e-01, -1.8601e-01,  3.4525e-01,  3.4589e-02,\n",
       "           9.1057e-01, -1.1112e-01, -3.0146e-01,  6.8609e-02,  2.6769e-02,\n",
       "           2.6257e-01,  8.8180e-01,  1.0982e-01, -5.5444e-01, -3.0190e-02,\n",
       "          -2.0610e-01, -2.2405e-01,  6.8093e-02, -6.7339e-01,  4.9373e-01,\n",
       "           2.7116e-01,  3.5715e-01,  6.9394e-01]], device='mps:0'),\n",
       " tensor([[-0.0247, -0.4426, -0.3895,  ...,  0.2060, -0.0660,  0.6121],\n",
       "         [-0.2327, -0.6282, -0.5041,  ...,  0.5289,  0.6556,  0.5584]],\n",
       "        device='mps:0'),\n",
       " tensor([[-2.5187e-01, -4.0054e-01, -5.8007e-01, -8.3055e-01, -1.2868e-01,\n",
       "          -2.8804e-02, -7.1917e-01,  6.6701e-01,  1.2709e-01,  5.0855e-01,\n",
       "           1.2570e-01,  5.1916e-01,  5.1676e-02, -1.1521e-01, -1.6481e-01,\n",
       "          -4.9820e-01,  5.6222e-02,  3.7338e-01,  6.7187e-01, -2.4111e-02,\n",
       "           2.2844e-01,  5.7979e-01, -2.6323e-01, -1.2386e-02, -8.3013e-01,\n",
       "           2.9613e-01, -1.1576e-01,  2.2409e-01,  1.7718e-01, -6.3092e-01,\n",
       "           4.0999e-02,  4.7417e-01,  2.3068e-02,  1.3497e+00, -5.6080e-01,\n",
       "          -1.3230e-01,  8.2213e-02, -3.9453e-01, -2.3010e-01,  3.7283e-02,\n",
       "           6.1164e-01, -4.1315e-01,  1.3635e-01,  4.0254e-01, -6.1085e-02,\n",
       "           8.9076e-02,  1.3899e-01, -3.7744e-01,  2.5892e-01, -8.3961e-01,\n",
       "          -4.8318e-02, -2.0669e-01,  4.9144e-01,  6.2615e-01, -6.3176e-01,\n",
       "          -2.2975e-01,  2.7419e-01, -3.7716e-02, -3.3020e-02,  1.5349e-01,\n",
       "           9.7980e-02,  3.8582e-01, -2.9995e-01,  2.3257e-01,  3.3441e-01,\n",
       "           2.4458e-01, -3.0392e-01, -4.8889e-01, -5.4078e-01, -5.0111e-02,\n",
       "           9.1859e-03,  2.2490e-02,  1.4846e-01, -2.3272e-01,  4.1973e-01,\n",
       "          -7.1214e-01, -6.4354e-02, -8.9125e-02, -2.5973e-01,  2.9246e-01,\n",
       "          -4.5549e-02,  1.9719e-01,  3.1010e-01,  1.0268e-01,  1.2457e-01,\n",
       "           4.3419e-02,  2.7215e-01,  5.2932e-04,  4.7373e-02,  3.7533e-01,\n",
       "          -1.3375e-01, -8.1722e-02, -2.8313e-01, -3.5814e-01,  3.1229e-01,\n",
       "           4.0598e-02,  2.9907e-01, -1.9557e-01,  1.8579e-01, -3.2474e-01,\n",
       "           1.5588e-01,  1.7768e-01, -1.8083e-01,  5.7286e-02, -3.5172e-01,\n",
       "          -4.5475e-01,  2.3589e-02, -2.8569e-01, -4.8285e-01,  2.2697e-01,\n",
       "          -1.4211e-01, -3.0205e-01, -4.0922e-02, -3.8935e-01, -3.1524e-02,\n",
       "          -7.5164e-02,  2.6013e-01, -1.3230e-01, -3.4806e-01, -8.3033e-02,\n",
       "           2.0734e-01, -2.2808e-01,  1.6077e-01,  2.6882e-01,  1.2232e-01,\n",
       "          -9.6705e-02,  6.5251e-01,  1.8950e-02,  3.1674e-01,  2.5763e-01,\n",
       "           1.1699e-01, -6.5768e-02, -1.9745e-01, -2.3154e-01,  7.5988e-02,\n",
       "          -7.6941e-02,  1.0687e-01, -1.8182e-01, -2.2790e-01, -1.8508e-01,\n",
       "           3.1771e-01, -3.3097e-01, -6.0893e-01, -4.9260e-01, -1.6098e-01,\n",
       "          -1.6326e-01, -2.4804e-02, -8.2473e-03, -3.3693e-01, -3.1956e-01,\n",
       "          -6.7443e-01, -3.6717e-01,  8.1041e-01, -1.6831e-02, -3.3356e-01,\n",
       "          -1.6499e-01,  2.3208e-01,  3.1666e-01, -4.5959e-02,  3.7161e-01,\n",
       "           7.7250e-01, -4.7670e-02,  1.6786e-01, -5.1827e-01, -3.8195e-01,\n",
       "           4.8335e-02,  2.6658e-01, -2.7054e-01,  6.1879e-01,  4.8352e-02,\n",
       "           7.5627e-02, -6.6170e-02,  2.7430e-01, -3.2704e-01, -1.2324e-02,\n",
       "          -5.7573e-01,  3.7750e-01,  2.9539e-01, -3.9892e-01, -6.6567e-02,\n",
       "          -2.0834e-02, -1.2060e+00,  9.6964e-02, -8.3040e-01,  3.1236e-01,\n",
       "          -2.3110e-01,  1.7963e-01,  2.3389e-01,  2.9545e-01, -6.5320e-02,\n",
       "           2.5404e-01,  2.8725e-01,  2.5687e-01,  3.3609e-01, -3.4924e-02,\n",
       "           2.8019e-01,  2.3924e-01,  5.0790e-01,  4.1073e-01,  1.4769e-01,\n",
       "          -3.5440e-01,  2.9683e-01,  3.5781e-01,  9.0598e-02,  1.6954e-01,\n",
       "           9.9825e-02, -2.1636e-01,  2.1091e-01,  1.7501e-01, -1.6618e-01,\n",
       "          -2.2406e-01, -3.1100e-02, -4.6818e-01, -5.6655e-02, -4.8467e-01,\n",
       "           3.2557e-01, -5.1983e-01, -1.1912e-01,  3.4831e-01,  2.4340e-01,\n",
       "          -1.4896e-02,  1.8319e-01,  1.7196e-01, -2.4338e-01,  2.1735e-01,\n",
       "           3.7449e-01, -3.4618e-01, -3.9804e-02, -3.2197e-01, -2.9367e-01,\n",
       "           2.4357e-01, -5.6244e-01,  6.6635e-01,  4.0574e-01,  6.0165e-02,\n",
       "           1.5119e-01, -2.7920e-01, -5.1638e-02, -5.7408e-01,  3.5471e-01,\n",
       "          -3.2130e-01, -3.4955e-01, -5.4801e-01,  6.4487e-02,  4.2773e-01,\n",
       "           2.7565e-01, -6.4915e-02,  7.7868e-01,  4.2408e-01,  4.2813e-01,\n",
       "          -4.3003e-01, -1.5896e-02,  4.0377e-01, -4.1405e-02, -4.6059e-01,\n",
       "           6.7116e-02, -1.4962e-01, -5.1643e-01,  2.8000e-01, -2.6598e-01,\n",
       "          -1.2047e-01, -1.4913e-01,  3.3682e-01,  1.5365e-01, -2.1964e-01,\n",
       "          -3.8461e-01,  3.9587e-01, -1.4349e-01, -4.1723e-01,  4.7123e-01,\n",
       "          -3.5025e-01,  1.5209e-01,  1.0694e-01, -3.8441e-01, -6.2072e-01,\n",
       "          -3.2235e-01,  2.3090e-01,  1.5585e-01,  3.4799e-02, -6.6306e-01,\n",
       "          -4.0571e-01,  3.1197e-01,  2.4601e-01,  2.9976e-01,  2.9840e-01,\n",
       "           5.7427e-01, -3.0944e-01, -1.6702e-01,  1.4982e-01, -4.1377e-01,\n",
       "          -1.5704e-01, -3.3228e-02, -1.9995e-01,  2.9569e-01,  2.4768e-01,\n",
       "           1.1600e-01,  2.0367e-01,  4.3482e-01, -2.5272e-01, -3.8673e-01,\n",
       "          -7.1543e-02,  2.0161e-01, -9.3629e-02, -5.8306e-02,  2.3951e-01,\n",
       "          -6.3095e-02,  5.9650e-01,  7.2143e-01,  7.7439e-02, -6.4006e-01,\n",
       "           1.7696e-01,  5.2502e-01, -2.7253e-01, -4.0343e-02,  9.6365e-02,\n",
       "          -1.0344e+00, -2.1331e-01, -1.2883e-01, -2.0181e-01, -6.5352e-01,\n",
       "           8.8115e-02,  3.4863e-01,  4.6833e-03, -1.7864e-01, -3.0947e-01,\n",
       "           1.0448e-01,  4.8724e-02, -4.8980e-02,  4.3534e-01, -2.8901e-01,\n",
       "          -3.4797e-01, -2.3438e-01, -2.1532e-01, -5.4201e-01, -3.9968e-01,\n",
       "           3.5914e-01,  2.8129e-01,  1.1986e-01, -2.6486e-01, -5.1832e-01,\n",
       "           4.5577e-01, -6.8169e-01, -1.4632e-01,  2.4840e-01, -4.7702e-01,\n",
       "           1.4650e-02,  4.8990e-01,  1.4999e-01,  1.6336e-01,  6.5258e-02,\n",
       "           7.8557e-02, -3.3903e-01, -5.0030e-01, -1.5000e-01,  6.2323e-01,\n",
       "           3.2283e-01,  3.3506e-01, -1.7236e-01,  3.9096e-01,  4.5167e-01,\n",
       "          -2.9582e-01,  4.1670e-01,  1.0930e-02,  7.7439e-02, -6.0414e-02,\n",
       "           1.9048e-01,  3.4370e-01, -3.6842e-01,  2.0145e-01,  1.7929e-01,\n",
       "          -3.1634e-01, -3.2149e-01,  1.5691e-02,  2.0642e-01, -1.8580e-01,\n",
       "          -3.0875e-01, -2.0361e-01,  9.9698e-02, -3.4856e-01, -4.3104e-01,\n",
       "           2.9161e-01, -3.6543e-02,  7.0495e-02, -4.0357e-01, -4.3462e-01,\n",
       "          -2.0181e-01, -2.6055e-01,  6.8681e-01,  2.1064e-01, -1.9621e-01,\n",
       "           1.8720e-01, -1.0907e-01, -3.6330e-01,  4.7847e-01,  2.0162e-02,\n",
       "          -4.8754e-01,  4.5482e-01, -3.4822e-01,  1.0627e-01,  5.5756e-01,\n",
       "          -1.6769e-01, -1.0840e-01,  1.1008e-01,  1.0820e-01,  2.9026e-01,\n",
       "          -3.9884e-01,  1.5516e-01, -6.3476e-02, -3.1885e-01,  3.1820e-01,\n",
       "           2.6411e-01,  2.2781e-01, -1.7137e-01, -3.3426e-02,  3.5677e-02,\n",
       "           2.8782e-01, -6.0853e-02, -3.9888e-01, -4.7347e-01, -4.6132e-01,\n",
       "           1.2832e-01,  1.8927e-01,  5.6207e-01,  3.4144e-01, -1.1394e+01,\n",
       "           1.2938e-01,  7.8031e-01, -4.0713e-01, -8.2819e-02, -5.0320e-02,\n",
       "           1.0976e-02, -1.3708e-01,  4.2955e-01, -2.7734e-01, -4.4105e-01,\n",
       "          -3.5727e-01, -1.3308e-01,  1.5454e-01, -3.2189e-01, -2.6945e-01,\n",
       "          -1.4686e-02, -7.0935e-01,  4.0963e-01, -3.0286e-01,  5.0298e-01,\n",
       "          -2.3511e-01, -8.3798e-02,  1.5816e-01, -3.5851e-01, -1.3439e-02,\n",
       "           5.9766e-02, -1.6033e-01,  1.4651e-01,  4.2486e-01, -5.2072e-02,\n",
       "           3.3244e-02, -4.1258e-03,  4.8461e-02, -2.5723e-01, -5.9999e-01,\n",
       "           1.9296e-02, -1.4141e-01,  3.3422e-01, -4.4133e-02, -7.1265e-01,\n",
       "          -2.2533e-01, -2.7418e-01,  3.1407e-01,  2.7021e-01,  9.0598e-01,\n",
       "           2.2955e-01,  1.0686e+00,  1.5022e-01,  1.7748e-01,  2.6230e-02,\n",
       "          -7.8764e-01, -3.3269e-02, -4.1907e-01,  2.3280e-02,  1.0114e-01,\n",
       "           3.6576e-01, -7.9367e-01,  2.5251e-01, -3.5521e-01,  3.6487e-02,\n",
       "           4.9094e-01,  3.1768e-02,  2.7700e-01, -3.2298e-02, -7.8123e-02,\n",
       "           1.2089e-01, -2.8544e-02,  1.1098e-01, -2.2583e-01, -2.4084e-01,\n",
       "          -1.0177e-01,  5.9888e-01,  3.9592e-01, -7.2211e-01,  3.6112e-01,\n",
       "           4.9258e-02, -1.8622e-01,  1.4556e+00, -4.8573e-02,  1.8361e-02,\n",
       "           1.9800e-03,  2.9604e-01, -5.0236e-01, -7.3516e-01,  3.6511e-01,\n",
       "          -1.0211e-01,  5.6427e-02,  1.0127e-01,  4.8148e-01, -9.7481e-02,\n",
       "          -4.7940e-01, -1.8593e-03,  1.0556e-01, -1.3380e-01,  5.4710e-01,\n",
       "          -7.2824e-01, -1.6963e-01, -1.1223e-01,  3.1723e-01, -8.4054e-01,\n",
       "          -6.1126e-02, -7.0210e-01, -3.9328e-01,  5.0586e-01, -2.5896e-01,\n",
       "           1.5852e-01, -2.7443e-02, -4.3439e-02,  5.5574e-01, -2.1489e-01,\n",
       "           7.1555e-02,  2.2813e-01,  8.4995e-02, -1.6733e-01,  3.9349e-02,\n",
       "           4.1630e-01,  4.9571e-01, -4.2740e-01, -6.3911e-01, -7.6901e-03,\n",
       "           1.7180e-01,  1.4933e-01,  3.4019e-01, -5.3463e-01,  3.6431e-01,\n",
       "          -7.9162e-01, -1.8769e-01,  7.2949e-02, -4.3273e-02,  3.6175e-01,\n",
       "           2.3925e-01,  7.2038e-02,  2.2256e-01, -7.7756e-01, -6.4498e-01,\n",
       "          -6.5932e-01, -2.1445e-01,  2.6143e-01, -5.4236e-01, -8.3654e-02,\n",
       "          -1.1033e-01, -4.8837e-03, -5.0292e-02, -6.6102e-02,  1.1939e-02,\n",
       "           1.3632e-01,  2.3587e-01, -1.3023e-01, -1.3178e-01, -9.2464e-02,\n",
       "          -5.5566e-01, -1.1753e-01, -9.9591e-02, -1.2505e-01, -5.6783e-01,\n",
       "          -1.3797e-01,  3.3075e-02, -2.5373e-01, -6.8124e-02,  3.5608e-01,\n",
       "          -2.9832e-02, -2.7856e-02, -4.8850e-01,  7.2846e-01,  4.0279e-01,\n",
       "           5.2088e-02, -2.8298e-01, -5.5189e-01, -1.0567e-01,  2.9592e-01,\n",
       "           1.9555e-01, -6.7007e-02, -1.7709e-01, -2.4145e-01, -4.2522e-01,\n",
       "           1.4302e-01,  7.1147e-01, -6.6296e-02, -6.8115e-01,  2.6621e-01,\n",
       "           5.8223e-01, -1.2882e-01, -2.4432e-01,  2.0983e-02, -9.3284e-02,\n",
       "          -3.1851e-01, -6.0317e-01,  7.8821e-01, -1.1246e-01, -5.8638e-02,\n",
       "           2.2427e-01,  2.0359e-01, -9.2588e-03,  4.0209e-01, -1.5770e-01,\n",
       "          -3.5434e-01, -1.6477e-03, -7.3600e-01,  4.3257e-01,  6.1666e-02,\n",
       "          -3.1489e-01,  7.6082e-01,  2.7260e-01, -6.0593e-02, -4.5779e-02,\n",
       "           2.2863e-01,  7.9941e-01, -2.4636e-01,  6.4653e-02,  5.3779e-01,\n",
       "           8.8110e-02, -1.1229e-01,  3.9580e-02,  4.1665e-02,  4.5410e-02,\n",
       "           3.7012e-01, -3.2145e-02, -4.1564e-01, -6.1194e-02,  3.8924e-01,\n",
       "           3.9295e-01, -5.4610e-01, -7.3487e-01,  2.2333e-02,  6.8551e-01,\n",
       "          -2.9054e-01,  2.3654e-01, -3.2708e-02,  3.0880e-01,  1.2001e-01,\n",
       "          -9.2160e-01, -1.5016e-01, -1.0056e-01,  2.8229e-01, -2.5482e-02,\n",
       "          -1.6139e-01, -1.4727e-01, -2.3686e-01, -1.6208e-01,  6.8308e-01,\n",
       "           2.1991e-01, -6.9160e-01,  3.4058e-02,  5.7281e-01, -6.0551e-01,\n",
       "           4.6052e-01, -2.4577e-01, -3.4302e-02,  3.6435e-01,  1.3117e-01,\n",
       "           4.9067e-02,  1.7856e-01, -4.2138e-01, -4.0598e-01,  1.7824e-01,\n",
       "           9.1672e-01, -7.6613e-02,  1.1166e-01, -3.9384e-01, -1.7182e-01,\n",
       "          -7.1855e-02, -3.1913e-01, -3.1286e-01,  3.7185e-01,  2.0529e-01,\n",
       "          -3.2552e-02, -1.3271e-01, -2.2992e-01,  3.0816e-01,  1.1821e-01,\n",
       "           3.2536e-01,  5.8187e-01,  6.4368e-01, -1.0077e-02, -2.4936e-01,\n",
       "          -6.1235e-02,  1.4568e-01, -1.9524e-01,  5.2704e-01, -2.0627e-01,\n",
       "          -3.3453e-01, -3.8069e-03,  3.0134e-01, -6.1681e-01, -5.8725e-02,\n",
       "          -4.5426e-01,  4.2361e-01, -1.6955e-01,  3.7868e-02,  2.8833e-01,\n",
       "           4.0522e-02,  5.7925e-02, -2.2651e-01,  5.5807e-01,  4.4511e-01,\n",
       "           1.6609e-01,  4.5662e-01,  1.6104e-01,  4.5272e-02, -5.1597e-01,\n",
       "           2.1978e-01, -1.0213e+00, -9.4260e-02, -4.6640e-02, -1.5401e-01,\n",
       "          -4.4224e-01, -2.8126e-02,  1.6814e-01,  1.8538e-01,  6.9597e-01,\n",
       "           7.2967e-01, -3.0871e-01, -4.5290e-01, -1.8510e-01,  1.5304e-01,\n",
       "           2.7705e-01, -1.9517e-01,  3.5971e-01,  9.6103e-02,  2.4432e-01,\n",
       "          -5.0967e-01, -4.3165e-01, -4.2252e-02,  3.1411e-01, -2.0171e-01,\n",
       "           5.7118e-01, -2.0164e-01, -1.3863e-01,  9.0832e-02,  1.3770e-01,\n",
       "           1.6639e-01,  7.9664e-01,  4.3416e-01, -6.3425e-01,  8.7469e-02,\n",
       "          -1.3980e-02, -2.9988e-01,  2.5983e-02, -4.3577e-01,  2.8858e-01,\n",
       "           1.2984e-01,  4.9280e-01,  6.7504e-01]], device='mps:0'),\n",
       " tensor([[-0.4654, -0.4115, -0.4370,  ...,  0.1625,  0.4338,  0.5908],\n",
       "         [-0.0932, -0.6293, -0.4879,  ...,  0.3041, -0.1958,  0.5576],\n",
       "         [-0.3292, -0.2590, -0.5016,  ..., -0.0148,  0.3509,  0.1607]],\n",
       "        device='mps:0'),\n",
       " tensor([[-0.1549, -0.3944, -0.3535,  ..., -0.0686,  0.6273,  0.3374],\n",
       "         [-0.1689, -0.4450, -0.6101,  ...,  0.4169,  0.1256,  0.6819]],\n",
       "        device='mps:0'),\n",
       " tensor([[-1.4191e-01, -5.7117e-01, -2.5963e-01, -6.2762e-01,  1.8690e-01,\n",
       "          -4.8415e-01, -7.0672e-01,  2.6386e-01,  7.8063e-02,  5.2931e-01,\n",
       "           1.7205e-01,  6.5085e-01,  4.6658e-01,  2.0496e-01, -2.9833e-01,\n",
       "           2.7841e-02, -4.9046e-01,  4.0150e-01,  2.5282e-01,  4.6598e-01,\n",
       "           4.4895e-01,  4.6437e-01,  8.5144e-02, -1.4949e-02, -2.7077e-01,\n",
       "           1.1115e-01,  3.3853e-01,  7.4613e-02,  4.0008e-01, -3.0969e-01,\n",
       "          -8.2900e-02,  3.1361e-01, -2.3361e-01,  1.3660e+00,  2.8383e-01,\n",
       "          -3.4895e-01,  4.8506e-01, -3.5628e-01,  1.4101e-02,  1.9825e-01,\n",
       "           1.6234e-01, -1.6385e-01,  4.2942e-01,  9.8734e-02,  4.4068e-01,\n",
       "           4.1728e-01,  1.1496e-02, -1.6335e-02,  6.2744e-01, -8.6088e-01,\n",
       "          -3.9067e-01, -2.9033e-01,  4.2922e-01,  1.6184e-01, -7.7540e-01,\n",
       "          -4.7287e-01,  1.7166e-01,  7.2946e-02, -3.6678e-01, -1.0704e-01,\n",
       "           1.0332e-01,  2.4561e-01, -3.6130e-01,  1.1566e-01,  7.6144e-02,\n",
       "           1.7556e-01, -4.0411e-01, -2.2707e-01, -5.7348e-01, -4.3768e-01,\n",
       "          -2.9709e-01, -8.7413e-02,  9.2447e-02,  1.5493e-01,  9.2028e-02,\n",
       "           4.4956e-02,  2.2206e-01, -1.0631e-01, -3.9048e-01,  5.4582e-03,\n",
       "          -1.8283e-01,  7.2747e-02, -2.1705e-01, -2.8527e-01,  4.6721e-02,\n",
       "          -1.7768e-01,  6.2793e-01, -4.2805e-03, -3.4667e-02, -1.9655e-01,\n",
       "          -8.9683e-02, -6.9616e-02, -4.4604e-01, -4.3986e-01,  4.0534e-01,\n",
       "          -7.9715e-02,  5.2844e-01, -2.1523e-01,  4.5859e-01, -8.7458e-02,\n",
       "           4.2212e-02,  1.5429e-02, -1.0076e-01,  2.9828e-01, -3.4672e-01,\n",
       "          -6.5824e-01,  9.5742e-02, -2.6553e-01, -4.3468e-01,  4.6238e-01,\n",
       "          -3.4474e-02, -5.9156e-01,  2.0930e-01, -3.0534e-01, -1.6509e-01,\n",
       "          -2.6767e-02,  1.5357e-01,  2.2946e-01, -2.8139e-01, -1.2924e-01,\n",
       "           1.9138e-01, -5.3732e-02,  2.5670e-01,  1.2173e-01,  1.0701e-01,\n",
       "          -1.4286e-01,  4.2937e-01,  1.3105e-01,  3.3197e-01,  6.2723e-01,\n",
       "          -1.5572e-01, -2.1007e-01, -4.9067e-01,  2.7276e-01,  1.5842e-01,\n",
       "           9.9442e-02, -6.9899e-03, -2.9685e-01,  1.5236e-03,  1.7725e-01,\n",
       "           2.1036e-01,  1.9798e-02, -1.8854e-01, -4.3814e-01, -3.4087e-02,\n",
       "          -1.3268e-01,  1.6974e-01, -5.0428e-01,  8.9264e-02, -3.3561e-01,\n",
       "          -6.6202e-01, -6.9329e-01,  6.4444e-01,  7.6009e-02, -2.2303e-01,\n",
       "          -1.5480e-01,  8.6707e-01, -9.4542e-02,  1.4899e-01, -1.5821e-01,\n",
       "           7.2160e-01, -2.8223e-02,  1.0445e-01, -5.5515e-01, -1.1034e-01,\n",
       "           2.1820e-01, -4.7835e-01, -3.9155e-01,  9.0172e-03,  5.5840e-02,\n",
       "           2.2724e-01, -5.7789e-01,  4.0496e-01,  8.5320e-02,  4.2506e-01,\n",
       "          -2.4958e-01,  1.0799e-01,  7.8592e-02, -2.9876e-01,  1.4534e-02,\n",
       "          -8.5775e-02, -1.1841e+00,  6.6866e-01, -2.6827e-01, -4.3532e-01,\n",
       "          -2.2421e-01, -3.6203e-02, -7.0547e-02,  2.7107e-01, -1.5923e-01,\n",
       "           1.3617e-02,  3.1912e-02,  5.4666e-02,  3.1360e-01,  5.9500e-02,\n",
       "           2.8280e-01,  2.9356e-01,  5.6154e-01,  2.0320e-01,  1.9784e-01,\n",
       "          -3.1223e-03,  4.7676e-02,  2.4303e-01,  4.7998e-01,  1.1158e-01,\n",
       "           3.7483e-01,  1.5164e-01, -3.0397e-02, -4.8314e-03, -1.4559e-02,\n",
       "          -3.3596e-01, -5.3268e-02, -5.8902e-01,  1.6342e-01, -1.6660e-01,\n",
       "           3.7742e-01, -4.1482e-01,  2.9030e-01,  5.1222e-01,  2.4465e-01,\n",
       "          -1.0100e-01,  1.3341e-01,  1.8157e-01, -2.9669e-01,  8.5645e-02,\n",
       "           4.1250e-01, -2.0842e-01,  1.4131e-01, -3.7209e-01, -1.4735e-01,\n",
       "           1.9754e-01, -3.6301e-01,  3.2693e-01,  4.6797e-01, -1.8766e-01,\n",
       "           6.3647e-02, -4.1837e-01,  2.1002e-01,  1.6335e-01,  3.1965e-01,\n",
       "          -3.5939e-01, -4.5096e-01, -5.4349e-01, -1.4165e-01,  2.1962e-01,\n",
       "           2.4792e-01,  1.4454e-01,  2.6803e-01,  7.7155e-01,  5.7742e-01,\n",
       "          -8.1498e-02,  5.7583e-02,  5.5285e-01, -2.3055e-02, -5.8053e-01,\n",
       "          -3.7132e-01,  1.0471e-01, -5.3310e-01,  6.2290e-01, -2.6100e-01,\n",
       "          -2.6612e-01, -1.1176e-01,  3.6584e-01,  4.9336e-01, -5.9010e-01,\n",
       "          -3.2976e-01,  2.8255e-01, -2.1204e-01, -3.9069e-01,  4.3052e-01,\n",
       "          -2.4144e-01,  4.1822e-01, -6.1721e-01, -2.8838e-01, -5.7570e-01,\n",
       "          -4.9766e-01, -2.6042e-01, -4.5289e-02, -6.8305e-03, -6.9149e-01,\n",
       "           1.3545e-01, -4.5284e-01, -1.8016e-01,  4.1357e-03, -5.4376e-01,\n",
       "           3.5491e-01,  6.1575e-02, -4.7555e-02, -5.3063e-01, -3.3309e-01,\n",
       "          -5.5609e-01,  1.3654e-01, -6.1715e-01, -1.5268e-01,  3.4378e-02,\n",
       "           3.5914e-01, -3.3306e-02,  4.2985e-01, -2.1995e-02, -4.4225e-01,\n",
       "           7.4970e-02,  1.8088e-01, -2.3089e-02, -4.2849e-02,  4.9334e-01,\n",
       "           3.6282e-01,  4.7160e-01,  7.4137e-01, -6.0795e-01, -3.8332e-01,\n",
       "          -1.1933e-01,  6.9790e-01, -2.2389e-01, -4.1830e-01, -1.9719e-01,\n",
       "          -1.0167e+00, -4.6248e-01, -2.7672e-01, -2.6361e-01, -1.8640e-01,\n",
       "           7.1259e-02,  4.8961e-01, -9.1116e-02, -1.1026e-01,  9.3946e-02,\n",
       "           4.5147e-01,  7.0083e-01, -1.4646e-01,  4.6532e-01, -6.3453e-01,\n",
       "          -2.0848e-01, -4.0562e-01, -6.1915e-01, -5.0838e-01, -1.9118e-01,\n",
       "           4.3393e-01,  6.0402e-02,  3.5361e-01, -1.1776e-01, -3.8055e-01,\n",
       "           4.8460e-01, -4.9873e-01, -3.0362e-01,  3.9851e-01, -2.6902e-02,\n",
       "           7.9015e-01, -4.6489e-03, -1.2765e-01, -9.0233e-02,  2.7778e-01,\n",
       "           1.6551e-01,  3.0273e-02, -2.1089e-01, -4.6859e-01,  7.0594e-01,\n",
       "           2.6151e-01, -2.2831e-01, -1.8779e-01,  4.8284e-01,  2.5610e-01,\n",
       "          -4.3803e-01,  1.9928e-01, -5.6217e-01, -1.6419e-01, -7.7369e-02,\n",
       "           4.4707e-01,  4.3415e-01, -2.6229e-01, -4.2786e-02, -2.1371e-01,\n",
       "           4.0757e-02, -4.7898e-01, -7.7829e-02,  5.3684e-02,  9.3311e-01,\n",
       "           1.6204e-02, -1.9546e-01,  7.1554e-02, -7.5881e-01, -3.3648e-01,\n",
       "          -4.8898e-02, -1.3441e-01,  1.5306e-01, -6.9413e-01, -5.3821e-01,\n",
       "          -3.2347e-01, -1.1826e-01,  6.6101e-01,  3.0510e-01,  2.0383e-01,\n",
       "           1.6480e-01,  1.1809e-02, -2.1590e-01,  8.2088e-01,  1.1295e-01,\n",
       "          -6.7932e-02,  4.1160e-01, -2.0316e-01, -9.7770e-02,  7.8292e-02,\n",
       "          -4.6149e-01,  6.3820e-03,  1.0049e-01,  2.4748e-01,  2.1586e-01,\n",
       "          -1.2926e-01, -1.9905e-01,  2.2433e-02, -4.2003e-01, -5.5113e-02,\n",
       "          -3.3722e-01,  6.6492e-01, -5.1988e-01, -1.3381e-01,  2.7804e-01,\n",
       "           3.2395e-01, -2.7638e-01,  9.1919e-02,  2.3831e-01, -6.5861e-01,\n",
       "           5.6790e-01,  6.6541e-02,  6.5929e-02,  4.2031e-01, -1.1221e+01,\n",
       "           3.8089e-01,  7.5675e-01, -3.4673e-01,  1.1622e-02,  1.1055e-01,\n",
       "           1.6467e-01, -1.5814e-01,  1.6681e-01, -1.5758e-01, -3.8873e-01,\n",
       "          -5.3402e-01,  2.8953e-02,  4.5415e-01, -4.1418e-01, -1.5477e-01,\n",
       "           7.9981e-02, -8.1513e-01,  1.0305e-01, -1.8363e-01,  3.8477e-01,\n",
       "          -3.1635e-02, -4.2593e-01,  3.6502e-01, -2.3423e-01, -1.3281e-02,\n",
       "           8.4111e-02, -2.0818e-01,  4.0318e-01,  3.7487e-01, -3.1379e-01,\n",
       "          -6.5374e-02, -3.5834e-02,  7.2242e-02, -2.9741e-01, -8.8016e-01,\n",
       "           1.9326e-04, -3.6108e-01,  1.2794e-01,  2.4294e-01, -2.6707e-01,\n",
       "          -3.8973e-02, -2.6584e-01,  2.6746e-01,  6.0525e-01,  6.9771e-01,\n",
       "           4.5712e-01,  1.3514e+00,  2.4702e-01, -2.7079e-01, -4.4883e-01,\n",
       "          -8.7737e-01,  2.5039e-01, -6.5193e-01, -5.7930e-02, -2.2160e-01,\n",
       "           6.1585e-01,  1.8394e-02,  5.6367e-02, -3.2999e-01, -1.7392e-01,\n",
       "           2.0911e-01,  3.0032e-01, -2.1092e-01, -4.6258e-01, -2.2452e-01,\n",
       "           5.8605e-02, -1.2605e-01,  3.0806e-02, -2.6410e-01, -2.5455e-01,\n",
       "           1.7928e-01,  1.1167e-01,  1.8730e-01, -3.8375e-01,  7.0007e-01,\n",
       "          -4.9415e-02, -2.2813e-01,  2.0878e+00, -2.4215e-01,  2.3210e-01,\n",
       "          -1.3825e-01,  1.4035e-01, -3.5666e-01, -6.0754e-01, -6.0965e-02,\n",
       "          -1.2072e-01, -7.8956e-02,  5.4356e-01,  4.2430e-01, -4.4410e-01,\n",
       "          -5.1805e-01,  4.2305e-01,  2.5639e-01, -5.9848e-03,  9.5397e-01,\n",
       "          -5.9387e-01, -4.4543e-01,  2.7163e-01,  4.0814e-01, -1.4397e-01,\n",
       "          -9.3779e-02, -2.4239e-01, -5.0947e-01,  1.9145e-01, -4.1707e-01,\n",
       "           3.2687e-01, -3.3021e-02, -1.4471e-01,  1.6991e-01, -7.8047e-04,\n",
       "           3.2643e-01,  3.2956e-01,  4.0115e-02, -3.7227e-01, -6.2576e-01,\n",
       "           2.7757e-02,  5.6018e-01, -7.1769e-02, -5.0635e-01, -3.2333e-01,\n",
       "          -1.2749e-01,  3.0330e-01,  3.6201e-01, -5.7101e-01,  3.3613e-01,\n",
       "          -6.9750e-01,  1.1197e-01, -6.9550e-02, -7.9329e-02,  6.7265e-01,\n",
       "          -5.4104e-02, -3.6115e-01,  1.4840e-01, -4.8547e-01, -9.2507e-02,\n",
       "          -4.9154e-01, -2.0145e-01,  2.9390e-01, -3.3508e-01,  6.8345e-02,\n",
       "          -7.2285e-02, -3.1343e-01, -3.9993e-01,  2.5855e-01, -4.4205e-02,\n",
       "           2.0277e-01,  3.3402e-01, -2.9117e-01,  1.0683e-01, -8.4976e-02,\n",
       "          -2.6810e-01, -1.7248e-01, -2.5605e-01,  1.8496e-01, -2.7045e-01,\n",
       "          -1.6481e-01,  3.8406e-02, -1.1533e-01, -7.1775e-01,  1.0923e-01,\n",
       "           4.5976e-01,  6.8804e-02, -3.1421e-01,  7.3897e-01,  1.5768e-01,\n",
       "          -5.2355e-02, -1.9201e-01, -3.6413e-01, -4.2443e-02,  2.9670e-01,\n",
       "           3.2154e-02,  2.1736e-02, -7.0163e-01, -3.5630e-01, -2.8292e-01,\n",
       "          -1.9350e-01,  1.0498e+00, -6.2561e-01, -4.7296e-01,  3.6463e-01,\n",
       "           1.3803e-01, -1.8497e-01,  1.6363e-01,  1.5009e-01,  1.6497e-02,\n",
       "           6.5562e-02, -4.2787e-01,  5.8274e-01, -8.3323e-02,  2.6232e-01,\n",
       "           2.0252e-01,  2.2265e-01,  1.7818e-01,  4.0072e-01, -6.7360e-02,\n",
       "          -4.0217e-01, -2.1849e-01, -3.0100e-01, -2.3583e-01,  1.0169e-01,\n",
       "           1.4331e-01, -4.7894e-02,  3.2724e-01, -2.3968e-01, -1.4987e-01,\n",
       "           5.6955e-01,  1.0642e+00, -1.3710e-01,  2.6185e-01,  9.6393e-01,\n",
       "           5.3939e-01, -5.9268e-02,  2.1713e-01,  5.9828e-02,  2.0883e-01,\n",
       "           3.3136e-01,  1.5199e-01, -1.8688e-01,  2.8912e-02,  3.3256e-01,\n",
       "           3.2531e-01, -1.2198e-01, -4.2586e-01, -2.0672e-01,  2.2934e-01,\n",
       "          -1.4890e-01,  2.9022e-01, -4.6803e-01,  1.9248e-01,  2.5145e-01,\n",
       "          -8.0670e-01,  1.4524e-02,  1.5851e-01, -5.1995e-01, -1.7736e-01,\n",
       "          -5.9858e-01, -2.6668e-01, -4.0838e-01, -4.3805e-01,  3.7160e-01,\n",
       "           2.7308e-01, -9.4773e-01,  1.4005e-01,  7.9628e-01, -6.8679e-01,\n",
       "           4.7228e-01, -5.5946e-02,  4.7293e-02,  8.1861e-01, -1.4509e-01,\n",
       "          -2.8407e-01,  3.0312e-01, -3.3000e-01, -4.7746e-01,  5.4529e-02,\n",
       "           7.1361e-01,  9.6475e-02,  1.3604e-01, -1.8759e-02, -2.3359e-01,\n",
       "          -3.1430e-01, -7.3693e-02, -5.3574e-02,  6.2187e-02,  2.7009e-01,\n",
       "          -9.3928e-02,  7.4676e-02, -1.6488e-01,  4.6861e-01, -4.2255e-01,\n",
       "           4.0172e-01,  3.5165e-01,  7.6506e-01,  4.7174e-02, -5.7496e-01,\n",
       "           2.7294e-01,  2.9759e-01, -2.9162e-01,  5.4819e-01, -5.0439e-02,\n",
       "          -3.4166e-01, -2.2522e-01,  1.4009e-01, -1.8045e-01, -1.7275e-02,\n",
       "          -5.8443e-01,  2.3140e-01, -1.4980e-01, -2.6508e-01,  6.8420e-01,\n",
       "           7.7737e-02,  1.6041e-01, -4.9511e-01,  1.1047e-01,  3.9060e-01,\n",
       "          -1.2590e-01,  4.8917e-01,  4.7049e-02,  2.8891e-01, -3.8427e-01,\n",
       "          -5.6655e-02, -5.0064e-01, -1.2101e-01,  2.4307e-01, -4.2843e-01,\n",
       "          -3.2115e-01, -4.1234e-01,  4.0868e-01,  3.2634e-02,  5.7541e-01,\n",
       "           6.7289e-01,  8.3674e-02, -7.1376e-01, -3.7180e-01,  1.8954e-01,\n",
       "           8.0114e-02,  4.8183e-02,  4.0815e-02, -1.5285e-01,  2.0654e-01,\n",
       "          -3.9544e-01, -4.9615e-01, -4.8431e-02,  5.3588e-02, -1.6501e-01,\n",
       "           6.8773e-01, -8.4472e-02, -1.7768e-01,  4.6965e-01,  1.8737e-01,\n",
       "           2.2292e-01,  5.8660e-01, -2.8026e-01, -4.6687e-01, -4.0024e-01,\n",
       "           2.0158e-01, -3.9701e-01,  3.8131e-01, -5.6214e-01,  1.3224e-01,\n",
       "           2.3115e-01,  2.6599e-01,  3.2796e-01]], device='mps:0'),\n",
       " tensor([[-6.1750e-01, -4.2829e-01, -2.3934e-01, -9.4116e-01, -3.4548e-01,\n",
       "          -6.5496e-02, -6.3874e-01,  9.7828e-01, -9.5892e-02,  4.0096e-01,\n",
       "           1.4919e-01, -5.1603e-03,  2.0980e-01,  1.3680e-01, -6.9506e-01,\n",
       "           1.6740e-01,  4.6094e-01,  2.1159e-01,  3.5951e-01, -4.4329e-02,\n",
       "           7.9832e-01,  3.4245e-01,  1.8019e-01, -2.5205e-01, -4.5876e-01,\n",
       "           5.8994e-01, -2.4628e-01, -9.2014e-02,  2.0509e-01, -8.1949e-01,\n",
       "          -1.3586e-01, -4.9757e-02, -3.6281e-01,  8.7602e-01,  1.1948e-01,\n",
       "          -2.7310e-01, -1.5276e-01, -2.7996e-01,  2.3494e-01, -6.3637e-02,\n",
       "           3.7814e-01,  2.6592e-02, -2.9347e-01,  3.1386e-01, -1.6421e-01,\n",
       "          -1.4057e-01,  1.4047e-02, -7.8429e-01,  1.3666e-02,  1.8955e-01,\n",
       "           5.8851e-02, -4.7205e-02,  1.9231e-01,  3.7030e-01, -1.0617e+00,\n",
       "          -2.6348e-01,  2.7603e-01,  2.9849e-01, -1.7932e-01, -5.7784e-02,\n",
       "          -1.9989e-01,  9.0201e-01,  1.7949e-01, -2.7822e-01,  1.7824e-01,\n",
       "          -4.4899e-01, -5.4601e-01, -6.0781e-01, -1.0682e-01, -1.1967e-01,\n",
       "          -1.0002e-01,  9.7528e-02,  1.4009e-01, -2.9385e-01,  2.3170e-01,\n",
       "           2.1813e-01,  4.7024e-02, -2.7652e-02, -4.3158e-01, -2.6546e-01,\n",
       "           3.6947e-01,  3.4425e-02, -1.4730e-02,  2.0326e-01, -5.8744e-01,\n",
       "          -2.0226e-01,  3.7959e-01,  1.5703e-01,  2.5914e-01,  2.2662e-03,\n",
       "           5.9967e-01,  1.2024e-02,  8.0339e-02, -4.5369e-01,  5.8427e-01,\n",
       "          -5.4955e-01,  1.0329e-01, -1.6007e-01,  4.9532e-02, -5.2088e-01,\n",
       "          -9.8735e-02,  2.4494e-01, -1.8085e-01,  1.6053e-01, -4.6592e-01,\n",
       "          -3.2983e-02,  2.4514e-01,  3.0468e-01, -4.9183e-01,  4.2685e-02,\n",
       "          -6.5134e-03, -2.9791e-01,  7.3694e-02, -6.6407e-01, -8.5795e-01,\n",
       "           5.8354e-01,  1.8428e-02, -4.5240e-01, -2.3930e-01,  7.4443e-01,\n",
       "           2.9738e-01, -2.3201e-01, -5.3072e-02, -3.5548e-01,  2.0259e-01,\n",
       "          -1.8518e-01,  2.5689e-01, -2.7018e-01,  3.8439e-01,  5.5026e-01,\n",
       "           3.1581e-01,  2.8489e-01, -1.8327e-01,  2.7637e-01,  1.4465e-01,\n",
       "           3.6947e-02, -6.6614e-02, -1.5929e-01,  1.5180e-01,  1.2041e-01,\n",
       "          -2.9580e-01, -3.2999e-01,  5.4746e-03,  1.1182e-01, -3.3279e-01,\n",
       "          -1.0202e-01,  1.8680e-01,  3.4808e-01, -1.4092e-01, -2.5403e-01,\n",
       "          -7.8984e-01, -4.2328e-01,  1.0217e-01, -1.8560e-01, -1.7135e-01,\n",
       "          -4.1911e-01,  3.8121e-01, -6.6660e-02,  4.7638e-01,  3.4189e-01,\n",
       "           6.9978e-01, -1.5311e-01,  5.9776e-02, -5.6412e-01,  3.5197e-01,\n",
       "           3.8270e-02, -6.1110e-01, -1.7247e-01,  1.5680e-01, -4.1813e-04,\n",
       "           6.8195e-01,  7.0979e-02,  4.0124e-01,  1.2070e-01, -3.2553e-01,\n",
       "           8.2089e-03,  1.9890e-01, -1.2864e-01, -1.4297e-01,  7.3766e-01,\n",
       "          -5.5188e-01, -2.3029e+00,  1.0721e-01, -3.5844e-01, -3.6529e-01,\n",
       "          -1.8135e-01,  3.7255e-01,  3.3772e-02,  4.9966e-01,  1.5486e-01,\n",
       "           3.7598e-02,  1.6729e-01, -2.8025e-02,  3.6851e-01, -8.4399e-02,\n",
       "           2.6751e-01,  3.3341e-01,  1.2988e-01, -2.3298e-01, -1.0956e-01,\n",
       "          -3.9227e-01,  2.4031e-01, -9.8916e-02,  8.6010e-01,  2.3260e-01,\n",
       "           2.0943e-01, -4.3657e-03, -4.7262e-02,  4.8964e-01,  4.3099e-01,\n",
       "          -6.6541e-02, -1.6585e-01, -4.9522e-01,  4.6269e-03, -4.1279e-02,\n",
       "          -1.2963e-02, -4.5611e-01,  1.3753e-01,  4.2841e-01, -1.2416e-01,\n",
       "          -1.8354e-01,  1.5525e-01, -4.7715e-02, -3.1517e-01, -2.6492e-02,\n",
       "           6.8453e-01,  1.9080e-01, -2.2994e-01,  1.2377e-01, -1.1896e-01,\n",
       "           3.7006e-02, -4.6975e-01,  7.0044e-01,  6.1147e-01,  1.5585e-01,\n",
       "           1.6214e-02, -6.9943e-01,  2.3699e-01, -3.1574e-01, -4.7647e-03,\n",
       "          -1.5027e-01, -1.5731e-01, -4.2986e-01, -5.7285e-02, -5.5854e-02,\n",
       "           4.0115e-01,  7.9943e-02,  4.6653e-01,  1.1837e+00,  2.7973e-01,\n",
       "           1.9184e-02,  5.6097e-01,  2.0801e-01,  4.2492e-01, -4.0180e-01,\n",
       "           3.3082e-01,  8.8825e-02, -6.3896e-01,  2.7188e-01, -4.1593e-01,\n",
       "           4.4963e-02, -4.1432e-02, -3.7926e-01,  1.4208e-01, -2.1276e-02,\n",
       "           3.4237e-01,  3.3517e-01,  1.5186e-01,  1.6838e-01,  2.9411e-01,\n",
       "          -3.1322e-01,  5.9715e-01, -6.0809e-02, -1.1127e-01, -6.3492e-01,\n",
       "          -7.6732e-01,  9.5206e-02,  4.4743e-01, -1.2880e-01, -6.1279e-01,\n",
       "           1.0462e-01, -2.3866e-01, -1.2257e-01, -3.5064e-02,  2.1191e-01,\n",
       "           2.7455e-01,  5.0687e-02, -9.5436e-02,  6.4791e-02, -4.6950e-01,\n",
       "          -4.9277e-01,  4.9140e-02, -1.5758e-01, -1.3063e-01,  5.3918e-01,\n",
       "          -7.9415e-02,  1.1575e-01,  5.7221e-01,  1.9799e-02, -4.5976e-01,\n",
       "          -2.5640e-01,  1.4802e-01,  2.8667e-01, -7.2082e-02,  2.1981e-01,\n",
       "           3.4119e-01,  2.6467e-02,  5.1649e-01,  1.4823e-01, -2.1046e-01,\n",
       "          -2.7239e-01,  5.1173e-01, -1.5755e-01,  6.1403e-02, -1.2527e-01,\n",
       "          -8.7207e-01, -5.6150e-02, -7.8413e-02, -1.6833e-01, -5.4440e-01,\n",
       "           2.6566e-01,  1.4351e-01,  1.0561e-01,  2.2350e-01, -1.5984e-01,\n",
       "           7.0123e-01,  6.5285e-01, -3.3100e-01, -1.6943e-01, -4.0104e-01,\n",
       "           9.3922e-02, -6.8374e-02, -2.9329e-01, -1.4009e-02,  9.5368e-03,\n",
       "          -2.2765e-01, -1.2516e-01, -1.0805e-01,  2.8921e-01, -1.8876e-01,\n",
       "           5.2862e-01,  1.2189e-01,  8.3649e-02,  4.3873e-01,  2.0612e-02,\n",
       "           1.3071e-01,  3.6515e-01, -4.2117e-01,  1.2870e-01,  4.0424e-01,\n",
       "           4.0578e-01, -1.0551e+00, -8.3291e-01, -1.4936e-01,  3.6111e-01,\n",
       "           2.2860e-01,  2.5189e-01,  1.6999e-02, -7.2215e-03,  8.9455e-02,\n",
       "          -2.3402e-01,  3.5135e-01, -6.6415e-02, -5.7412e-01,  3.0561e-01,\n",
       "           2.3567e-01,  5.3263e-01, -2.7377e-01, -5.0579e-01, -6.5233e-02,\n",
       "           2.4616e-01, -3.6381e-01,  1.1392e-01,  2.3480e-01,  4.5466e-01,\n",
       "          -3.6015e-01, -5.6164e-01, -7.0282e-02, -7.3986e-01, -1.3578e-01,\n",
       "           3.2483e-01, -2.1437e-01,  2.7178e-02, -2.4758e-01,  1.5231e-01,\n",
       "          -1.7925e-02, -1.8879e-01,  6.0757e-01,  2.0522e-01, -2.5855e-01,\n",
       "           1.1981e-01, -2.3892e-01, -1.9475e-01,  5.0091e-01, -5.0226e-01,\n",
       "          -3.0348e-01, -5.0960e-02, -1.5349e-01, -3.2418e-01,  3.2181e-01,\n",
       "          -4.1389e-01, -1.4861e-01,  1.6748e-01, -6.4736e-02, -1.3068e-01,\n",
       "          -3.5233e-01,  1.4268e-01, -1.2707e-01, -4.9883e-02, -1.7654e-02,\n",
       "          -1.3353e-01,  6.0900e-02,  1.3308e-01,  6.4538e-02,  1.4388e-01,\n",
       "           5.3696e-01, -1.1464e-01,  2.7582e-03, -2.5900e-01, -2.2469e-01,\n",
       "           5.2498e-01, -9.4973e-02, -3.3117e-01,  5.4928e-01, -1.1156e+01,\n",
       "           1.0386e-01,  7.2631e-01, -5.5808e-01,  5.5128e-02, -2.8539e-01,\n",
       "           2.1085e-01, -3.0009e-01,  7.9101e-02,  2.4967e-01, -2.8460e-01,\n",
       "          -1.2277e-01, -2.7364e-01, -2.3193e-01, -4.0872e-01, -3.9742e-01,\n",
       "          -4.7229e-01, -5.6789e-01,  1.1772e-01, -1.7663e-01, -1.0996e-01,\n",
       "          -5.7036e-01, -1.9379e-01,  2.9696e-01, -4.9559e-03, -5.1069e-01,\n",
       "           5.9541e-02, -2.2948e-01,  2.6399e-01,  4.6126e-02,  2.2045e-01,\n",
       "          -1.9949e-02,  1.4108e-01,  2.0669e-01, -4.4946e-02, -4.5100e-01,\n",
       "          -4.4853e-01,  2.2129e-01, -1.5955e-01, -2.2267e-01, -5.4017e-02,\n",
       "          -3.3838e-01, -2.7689e-01,  1.3447e-01,  3.7594e-01,  6.2596e-01,\n",
       "           1.1595e+00,  5.6208e-01,  2.9245e-01, -1.0625e-01, -8.5438e-02,\n",
       "          -1.0762e+00,  5.5900e-01, -3.7318e-01,  8.1787e-02, -2.5119e-01,\n",
       "           6.6201e-01, -4.9906e-01,  5.4685e-01, -4.0754e-01, -2.5500e-01,\n",
       "           3.7179e-01,  3.0194e-01,  7.1104e-03,  2.8239e-02, -2.7432e-01,\n",
       "           4.0257e-01, -2.6094e-01, -3.6915e-02, -3.7819e-01,  4.0081e-01,\n",
       "          -3.9472e-01,  5.0770e-01, -5.6534e-02,  1.2945e-01,  1.8567e-01,\n",
       "          -3.5001e-01,  1.6610e-01,  1.0337e+00,  1.7414e-01,  4.1071e-01,\n",
       "           1.7420e-01,  5.1160e-01,  5.3064e-02, -8.8958e-01,  1.3691e-01,\n",
       "           2.0502e-01,  2.0789e-01, -6.7096e-02,  5.7131e-01, -2.5517e-02,\n",
       "          -6.5181e-01, -2.3094e-01, -3.2169e-01,  4.3108e-01,  6.5352e-01,\n",
       "          -2.2597e-01, -4.4343e-01,  2.3894e-01, -7.9776e-03, -5.3395e-01,\n",
       "           9.0074e-02, -2.2320e-01, -5.2792e-01,  8.7275e-02, -5.8914e-01,\n",
       "           1.4985e-01, -1.2428e-01,  5.4238e-02,  6.6887e-01, -4.0159e-01,\n",
       "          -1.6113e-01,  3.6602e-02, -4.4532e-02, -4.1270e-01, -4.9520e-01,\n",
       "          -7.7083e-02,  4.1073e-01, -1.9776e-01, -2.7976e-01,  7.0636e-02,\n",
       "          -3.0673e-01,  2.2584e-01, -1.7377e-01, -6.4935e-01,  3.2879e-01,\n",
       "          -6.0415e-01,  2.0578e-01,  2.5360e-01, -2.2507e-01,  4.9017e-01,\n",
       "           1.1337e-01, -8.1817e-02,  1.9397e-01, -9.9369e-02, -7.8554e-01,\n",
       "          -3.2573e-01, -2.7299e-02,  3.2662e-01, -4.2844e-01,  1.5776e-01,\n",
       "           1.5022e-01, -4.5752e-02, -2.2574e-01,  2.7904e-01,  1.0174e-01,\n",
       "           3.7762e-01,  3.9846e-01, -5.5006e-01,  1.1727e-01, -2.9867e-01,\n",
       "          -2.9262e-02, -6.1174e-02,  1.7952e-01,  2.3654e-01, -7.0764e-01,\n",
       "          -1.5224e-01, -2.6434e-01, -3.0965e-01, -2.1650e-01, -2.5120e-01,\n",
       "           2.0680e-01,  1.4038e-01, -1.5897e-02, -1.1308e-01,  5.8077e-01,\n",
       "           4.0814e-02, -1.1167e-01, -4.5997e-01,  3.1443e-01,  4.3178e-01,\n",
       "           2.1794e-01,  7.5875e-02, -1.2830e-01, -5.5532e-01, -1.1922e-02,\n",
       "          -4.3644e-01,  9.1081e-01, -7.3906e-01, -7.1970e-01,  6.6086e-02,\n",
       "           1.4951e-02, -8.8259e-04, -2.3450e-01, -5.0265e-01,  8.1298e-02,\n",
       "          -2.1017e-01, -1.4891e-01,  6.1550e-03, -1.0607e-02,  1.6299e-01,\n",
       "           4.0742e-01,  3.6838e-01,  1.5951e-01, -3.3938e-01, -4.4081e-01,\n",
       "          -6.0244e-01,  1.4567e-02, -1.6795e-01,  2.9330e-01, -2.6752e-01,\n",
       "          -3.2661e-01,  1.6043e-02, -9.1892e-03, -2.2272e-01,  5.2876e-01,\n",
       "           1.2427e-01,  1.6396e+00, -3.7287e-01, -4.4551e-01,  2.9018e-01,\n",
       "           7.2306e-02, -2.7073e-02, -2.8130e-01,  2.5074e-01,  9.5447e-02,\n",
       "           4.0194e-02, -3.6304e-01,  1.5023e-02, -2.0558e-02,  3.2773e-01,\n",
       "           3.2056e-01, -1.0703e+00, -2.0238e-01, -1.7993e-01,  5.5664e-01,\n",
       "           3.2388e-02,  6.3185e-01, -2.2305e-01,  3.0577e-01,  4.6106e-01,\n",
       "          -8.6667e-01, -2.3000e-01, -1.6129e-01,  2.7295e-01, -1.0583e-01,\n",
       "          -9.8735e-02, -1.6178e-01, -3.7165e-02, -1.6936e-01,  3.9538e-01,\n",
       "           7.4751e-01, -7.8426e-01,  7.4081e-01,  7.0467e-01, -9.8234e-02,\n",
       "           5.6815e-01,  9.4187e-02,  3.2641e-01, -1.4659e-01,  2.2669e-02,\n",
       "           2.6200e-01,  1.8542e-01, -6.3165e-01, -2.3962e-01,  2.4473e-01,\n",
       "           4.0158e-01, -1.1943e-01,  1.2108e-01,  7.9292e-02,  3.0960e-02,\n",
       "          -3.7691e-01, -1.0145e-01, -2.4552e-01, -3.9904e-01,  1.1356e-01,\n",
       "           1.4603e-02, -1.5001e-01, -8.6997e-02,  4.7709e-01, -1.7580e-01,\n",
       "           7.5067e-01,  4.2910e-01,  3.7145e-01, -3.5103e-01,  8.9950e-04,\n",
       "          -5.7565e-02,  6.0671e-01, -3.1670e-01,  7.7385e-01,  6.3337e-02,\n",
       "          -2.3140e-01,  2.7251e-01,  6.0547e-03,  1.7829e-01,  2.9892e-01,\n",
       "          -2.3519e-01,  1.2702e-01, -3.0718e-01, -2.3711e-01,  2.8330e-01,\n",
       "          -1.8640e-01,  1.1542e-01, -7.2270e-01, -6.3602e-03,  3.9857e-01,\n",
       "          -3.2051e-01,  4.9873e-01, -2.3074e-01,  6.1026e-01, -4.3509e-01,\n",
       "           3.2669e-02, -4.7470e-01, -4.3275e-01, -2.8508e-01,  5.1854e-02,\n",
       "          -8.6027e-01, -1.9453e-01,  1.7684e-01,  3.7405e-02,  2.1707e-01,\n",
       "           7.5155e-01, -3.3794e-01, -2.4771e-01, -2.0386e-01,  4.0190e-02,\n",
       "           1.8299e-01, -8.2071e-02, -3.5586e-01,  2.1161e-01, -3.5607e-01,\n",
       "          -1.0520e+00, -2.3117e-01,  8.9990e-02,  3.8419e-01, -3.5853e-01,\n",
       "           1.1570e+00,  2.5132e-02, -2.0226e-01,  6.7253e-01,  2.0808e-01,\n",
       "           6.4157e-01,  6.0323e-01,  1.3628e-03, -4.1259e-01, -5.4109e-02,\n",
       "          -1.5877e-01, -3.2881e-01,  1.5757e-01, -1.8346e-01,  5.0556e-01,\n",
       "           5.4802e-01, -2.0806e-02,  5.4756e-01]], device='mps:0'),\n",
       " tensor([[-9.8608e-02, -6.2949e-01, -5.4737e-01, -4.6335e-01, -3.7454e-01,\n",
       "           1.0248e-01, -2.8097e-01,  4.4707e-01, -1.3869e-01,  5.9022e-01,\n",
       "           3.3979e-01,  3.6518e-02, -7.9829e-02, -1.9712e-01, -4.0961e-01,\n",
       "          -5.4994e-01,  1.7904e-01,  2.6107e-01,  2.6766e-01, -1.3636e-01,\n",
       "           9.8026e-01,  3.6199e-01, -8.5372e-02,  1.8939e-01, -6.9872e-01,\n",
       "           3.9986e-01,  4.5490e-02,  8.0399e-02,  4.6963e-01, -6.9729e-01,\n",
       "          -2.0828e-01,  5.4662e-01, -1.2219e-01,  1.1649e+00, -1.8525e-01,\n",
       "          -2.5129e-01,  4.1644e-01, -2.6874e-01,  8.0973e-01,  4.0156e-02,\n",
       "           6.2909e-01, -1.2839e-01,  2.5974e-01,  2.1204e-01,  2.5654e-01,\n",
       "          -1.6577e-02,  1.0736e-01, -6.2638e-01,  5.4556e-01, -5.1867e-01,\n",
       "          -2.1843e-01, -1.5953e-01,  3.6604e-01,  4.8748e-01, -6.7281e-02,\n",
       "           3.5607e-02,  3.0541e-01,  1.6232e-01, -2.5763e-01,  2.0386e-01,\n",
       "           1.3393e-01,  6.9371e-01,  5.7586e-02, -1.9525e-01, -3.0287e-01,\n",
       "           3.0687e-01, -5.2685e-01, -7.0972e-01, -3.4711e-01, -1.5503e-01,\n",
       "          -6.3145e-02,  3.6453e-01, -2.3968e-01, -8.3683e-02,  7.8001e-02,\n",
       "          -8.8895e-02,  3.9835e-02, -1.7901e-01, -2.7637e-01, -9.8806e-03,\n",
       "           6.4246e-02,  3.2661e-01, -1.9744e-01, -4.1654e-01, -2.1876e-01,\n",
       "           3.0475e-01,  7.8369e-01,  6.3690e-02,  2.5669e-01,  8.5968e-02,\n",
       "           3.1750e-01,  9.4549e-02, -3.4083e-01, -4.9471e-01,  3.1401e-01,\n",
       "          -1.3255e-01,  4.9667e-01, -1.4940e-01, -6.5687e-02, -4.7700e-02,\n",
       "           3.7095e-01,  7.2304e-02,  1.7323e-03,  1.1325e-01, -6.6663e-02,\n",
       "          -4.8542e-01,  1.4204e-01, -2.5931e-01, -6.6520e-01, -1.1522e-01,\n",
       "          -7.9637e-02,  1.3277e-01,  1.5561e-02, -3.1430e-01,  2.6189e-01,\n",
       "          -7.3846e-02,  1.9145e-01, -1.3642e-02, -3.3838e-01,  3.7293e-01,\n",
       "           2.5609e-01, -4.0848e-01,  2.2958e-02,  1.6951e-01,  4.6388e-01,\n",
       "          -2.2394e-01,  6.0049e-01, -2.8694e-01,  2.5392e-01,  4.8866e-01,\n",
       "           8.4207e-02, -6.0898e-02, -9.6990e-02, -1.7751e-01, -3.9547e-01,\n",
       "          -4.2915e-02,  7.3082e-02, -3.2798e-01,  1.0024e-01,  2.1027e-02,\n",
       "          -1.5484e-01, -1.5575e-01, -7.1363e-01, -3.2433e-01,  1.6862e-01,\n",
       "          -2.9423e-01,  2.9391e-01, -7.2839e-02, -5.6183e-01, -3.6233e-01,\n",
       "          -8.1793e-01, -4.5420e-01,  5.0598e-01,  3.1144e-02, -1.4616e-01,\n",
       "          -4.4906e-02,  1.9702e-01,  1.8656e-01, -1.9001e-02,  3.6581e-01,\n",
       "           4.5356e-01, -1.7970e-01,  5.5773e-02, -4.7383e-01,  5.1199e-02,\n",
       "           3.6512e-02, -1.1536e-01, -1.4678e-01, -1.8904e-01,  8.2812e-02,\n",
       "           1.8413e-01, -2.0916e-01, -2.0209e-02, -4.3147e-01,  1.4387e-01,\n",
       "          -9.5740e-02,  8.7516e-02,  2.8605e-01, -3.1460e-01, -3.4357e-01,\n",
       "          -4.4457e-01, -2.0415e+00,  1.3037e-01, -1.6708e-01, -1.9640e-01,\n",
       "          -7.8661e-01,  5.6989e-01,  4.0038e-01,  2.3491e-01,  1.2055e-02,\n",
       "          -4.1441e-01, -1.3256e-01, -2.2065e-01,  2.0346e-01, -1.5937e-01,\n",
       "           4.1104e-01,  3.6856e-01,  2.1222e-01, -8.2505e-02, -5.3417e-02,\n",
       "          -3.3993e-01,  1.9027e-01,  6.6698e-02,  4.3789e-02,  1.3752e-02,\n",
       "           8.6470e-02, -3.4516e-01,  1.5865e-01,  6.3413e-01, -3.2035e-01,\n",
       "          -2.6413e-01, -1.5432e-01, -3.1116e-01, -1.1825e-01, -2.1011e-01,\n",
       "           7.5833e-01, -1.0075e-01, -1.9090e-01,  4.1462e-01,  2.9186e-01,\n",
       "          -2.0576e-01,  2.6479e-02, -1.7243e-01, -4.6025e-01, -8.4004e-02,\n",
       "           8.3166e-01, -2.6809e-01, -1.8435e-01, -1.3576e-01, -2.8072e-01,\n",
       "           9.9627e-02, -3.9657e-01,  7.0865e-01,  1.3131e-01, -2.4437e-01,\n",
       "          -1.3250e-01, -4.9085e-01, -3.6081e-02, -6.5221e-02,  2.7934e-01,\n",
       "          -3.2480e-01, -4.4413e-01, -6.2940e-01,  9.8728e-02,  4.2580e-01,\n",
       "           1.0494e-01, -2.2581e-01,  5.8588e-01,  3.7420e-01,  5.5104e-01,\n",
       "          -7.6877e-02,  1.8477e-03,  4.0139e-01,  4.0454e-02, -3.1153e-01,\n",
       "           4.0137e-02,  1.1839e-01, -6.1671e-02,  2.8916e-01, -5.0007e-01,\n",
       "           2.1460e-01,  1.3048e-01,  4.2285e-02,  4.0869e-01, -3.3156e-01,\n",
       "           3.3759e-01, -6.4578e-04, -4.2352e-01, -2.2792e-01,  3.1587e-01,\n",
       "          -3.4318e-01,  4.7882e-01, -1.1497e-01, -1.3533e-01, -6.5793e-01,\n",
       "          -3.0023e-01,  9.9027e-02, -5.2265e-01,  4.0820e-02, -8.5922e-01,\n",
       "          -1.4457e-01,  4.3467e-01, -1.1843e-01,  2.8682e-01, -4.3268e-01,\n",
       "           2.7079e-01,  1.1150e-01,  1.0583e-01, -1.8427e-01, -4.8035e-01,\n",
       "           6.4368e-03,  3.5471e-01, -1.9635e-01,  6.7501e-02,  1.8728e-02,\n",
       "           1.4821e-01,  7.4914e-02, -1.4411e-01,  1.8116e-02, -3.9952e-01,\n",
       "           1.2334e-01,  2.7083e-01, -6.4203e-02, -1.1696e-01, -2.5127e-01,\n",
       "           8.1457e-02,  5.7341e-02,  4.5826e-01, -1.5574e-01, -4.7792e-01,\n",
       "           1.3602e-02,  5.6159e-01,  4.2917e-02,  9.5691e-02, -4.1476e-02,\n",
       "          -1.3511e+00, -3.8201e-01,  1.3646e-01, -2.3684e-01, -1.6250e-01,\n",
       "           1.4728e-01, -4.5766e-02, -1.0990e-01,  1.3434e-01, -3.3727e-01,\n",
       "           4.9913e-01, -7.4794e-03, -9.4830e-02,  2.6318e-01, -2.9595e-01,\n",
       "          -2.6245e-01, -5.5022e-02, -5.5963e-02, -1.9048e-01, -2.1108e-01,\n",
       "           1.0760e-01,  2.4895e-01, -1.6610e-02,  2.9324e-01, -4.2315e-01,\n",
       "           6.9357e-01,  2.0403e-02, -3.6256e-01,  1.8028e-01, -4.6817e-01,\n",
       "          -2.2470e-01,  6.2020e-01, -2.1111e-01,  1.7508e-01,  7.0164e-02,\n",
       "           2.2583e-01, -1.6263e-01, -2.5922e-01, -1.5555e-01,  5.6515e-01,\n",
       "           9.7645e-02,  3.8631e-01, -3.1465e-01,  4.6445e-01,  6.0830e-01,\n",
       "          -4.3876e-01,  2.7575e-01, -3.0324e-01, -3.5509e-01, -1.4900e-01,\n",
       "           3.7937e-01,  2.4503e-01, -2.4568e-01,  1.2754e-01,  2.1491e-01,\n",
       "           2.4189e-01, -2.8123e-01, -1.4375e-01,  1.5623e-01, -1.6800e-02,\n",
       "          -1.6159e-01, -5.9860e-01,  4.8387e-02, -4.0540e-01, -3.7598e-01,\n",
       "          -5.5542e-02, -9.6940e-02,  2.3018e-01, -3.4557e-01, -2.8109e-01,\n",
       "          -4.1082e-01,  1.5328e-01,  7.9392e-01, -8.2566e-02,  1.3355e-02,\n",
       "          -2.4290e-01,  1.3134e-01, -3.9568e-01,  5.6959e-01,  2.4837e-01,\n",
       "          -5.6746e-02,  5.0092e-01, -5.9799e-01,  3.2502e-01,  1.6325e-01,\n",
       "          -2.5296e-01,  7.9720e-02, -1.1265e-01,  4.2524e-01, -1.9241e-01,\n",
       "          -2.6666e-01,  1.8088e-01,  2.9568e-01,  8.0022e-02,  3.3680e-02,\n",
       "           8.9946e-02,  7.7279e-01, -5.6155e-01, -1.1355e-01, -3.2412e-02,\n",
       "           4.6990e-01, -1.4868e-01,  1.7842e-01, -2.5184e-01, -4.5494e-01,\n",
       "           2.8096e-01,  7.6220e-03,  3.0259e-02,  1.5848e-01, -1.1470e+01,\n",
       "           1.7761e-02,  7.9956e-01, -5.9407e-01,  3.9142e-02, -7.9952e-04,\n",
       "           1.4375e-01, -2.8145e-01,  3.6361e-03, -2.5851e-01, -3.7282e-01,\n",
       "           1.6253e-01,  5.8747e-02,  6.8827e-02, -1.9252e-01, -1.2168e-01,\n",
       "          -4.4456e-02, -9.0133e-01,  1.4013e-01, -2.9238e-01,  3.1633e-01,\n",
       "          -4.6944e-01,  3.5652e-01,  1.0775e-01,  4.8818e-02, -3.3732e-01,\n",
       "          -6.3534e-02, -4.1684e-01,  2.3683e-01,  1.0364e-01, -1.6207e-01,\n",
       "          -1.3703e-01, -1.5828e-01,  4.6034e-01, -1.8993e-01, -4.8947e-01,\n",
       "           1.5862e-01, -7.0142e-02,  4.5043e-01, -1.6798e-02, -2.3410e-01,\n",
       "           5.0011e-02, -4.1456e-01,  3.0195e-01,  6.4668e-02,  5.0457e-01,\n",
       "           4.7572e-01,  6.3490e-01,  1.0739e-01, -2.1848e-01,  4.9891e-02,\n",
       "          -8.9429e-01,  3.3171e-01, -9.8824e-01, -2.6344e-01,  3.6779e-02,\n",
       "           7.4622e-01, -4.6007e-01,  9.9323e-03,  8.2971e-02, -3.9312e-01,\n",
       "           2.4107e-01, -5.5413e-02,  1.0686e-01, -6.2270e-01, -1.1555e-01,\n",
       "           3.6365e-01,  1.0164e-01,  1.3567e-01, -3.5856e-01, -1.3223e-02,\n",
       "          -4.3049e-02,  5.9870e-01,  1.5647e-01, -2.5151e-01,  1.0429e-01,\n",
       "          -8.0721e-03,  2.3224e-01,  6.2748e-01,  4.7218e-02,  1.3164e-01,\n",
       "           1.7459e-01,  5.1672e-01, -5.4105e-01, -4.7941e-01, -1.2530e-01,\n",
       "           6.3722e-02,  2.8461e-01,  8.3486e-02,  3.6163e-01, -3.9872e-01,\n",
       "          -3.1281e-01, -7.8561e-02,  3.1281e-02,  5.0728e-02,  1.2098e+00,\n",
       "          -5.8519e-01, -2.2122e-01, -1.0256e-01, -9.1470e-02, -1.3571e-01,\n",
       "          -3.9828e-02, -3.3651e-01, -4.2370e-01,  3.3182e-01,  1.1019e-01,\n",
       "           1.7328e-01,  1.5282e-02,  3.9299e-01,  1.9321e-01, -5.2099e-01,\n",
       "           2.8514e-01,  1.4470e-01,  4.3958e-01, -1.8846e-01, -4.1365e-01,\n",
       "          -1.0991e-01,  3.3157e-01, -3.8281e-01, -3.4973e-01, -1.0983e-01,\n",
       "           3.0412e-02,  5.9583e-02, -5.8809e-02, -3.7567e-01,  2.5850e-01,\n",
       "          -6.2094e-01, -9.3514e-03, -4.2726e-01, -1.1183e-01,  4.7884e-01,\n",
       "           1.2318e-01, -3.6857e-02,  1.1137e-01, -4.2935e-01, -6.7978e-01,\n",
       "          -6.4684e-01, -7.4975e-02,  5.2269e-02, -3.7956e-01, -1.6842e-02,\n",
       "          -1.8671e-01,  6.6050e-02, -3.6548e-01,  1.8633e-01, -8.2184e-02,\n",
       "           7.5332e-01,  2.3533e-01, -1.8486e-01,  1.2410e-01, -1.4424e-01,\n",
       "          -3.6447e-01, -2.1959e-01, -2.4676e-01,  2.2621e-01, -5.0366e-01,\n",
       "          -1.3080e-01,  1.8011e-01,  1.1438e-01,  6.4878e-03, -3.4448e-01,\n",
       "          -1.2932e-01,  2.0570e-01, -1.3707e-01,  3.0997e-01,  1.7373e-01,\n",
       "           2.0329e-01, -6.4335e-02, -4.6287e-01,  8.7786e-02,  4.5179e-01,\n",
       "          -1.1348e-01, -2.7396e-01,  3.8658e-02, -7.0830e-01,  2.1125e-01,\n",
       "           3.8231e-01,  8.2746e-01, -2.5998e-01, -5.3996e-01,  4.6996e-01,\n",
       "           6.2219e-01, -4.6785e-01, -5.7024e-02, -2.1133e-01,  1.2442e-01,\n",
       "          -1.0488e-01, -3.4183e-01,  2.6033e-01, -2.5177e-01,  4.2375e-01,\n",
       "          -9.9348e-02,  2.0811e-01, -1.9413e-01,  4.1807e-01,  9.7647e-02,\n",
       "          -2.0238e-01, -3.5246e-02, -6.2741e-01, -4.3863e-01,  2.3873e-01,\n",
       "           2.6324e-01, -1.5522e-01, -2.3160e-02, -1.2275e-01,  1.1149e-01,\n",
       "           4.8271e-01,  8.5393e-01,  8.1221e-02,  1.4584e-01,  3.8317e-01,\n",
       "           2.3356e-01, -2.7630e-01,  5.9953e-02,  2.9270e-01,  4.0766e-01,\n",
       "           3.5259e-01,  5.3145e-01, -3.4176e-01,  7.2203e-02, -1.6344e-01,\n",
       "           2.0465e-01, -5.7619e-01, -7.6698e-01, -1.3126e-01,  4.3082e-01,\n",
       "           1.5541e-02,  2.9468e-01, -2.7851e-01,  3.4408e-01,  6.3093e-01,\n",
       "          -1.1306e+00, -5.9181e-02, -3.9846e-02, -9.3994e-02, -6.2591e-02,\n",
       "          -5.4897e-01, -2.2413e-01, -3.5234e-01, -2.4418e-01,  5.2852e-01,\n",
       "           5.8811e-01, -8.5286e-01,  1.6085e-01,  2.2953e-01, -1.5005e-01,\n",
       "           9.4328e-02, -9.1644e-02,  3.2863e-02,  5.0134e-01,  1.5175e-01,\n",
       "          -5.7932e-01,  4.5778e-01,  1.5595e-01, -4.3637e-01,  2.6165e-01,\n",
       "           4.1776e-01, -1.8687e-02,  1.5952e-01,  2.2085e-01, -4.9751e-01,\n",
       "          -1.0141e-02, -4.6200e-01, -1.0629e-01,  1.0751e-01, -6.7537e-02,\n",
       "           2.4815e-02, -4.9794e-01, -6.6236e-02,  3.4484e-01,  2.9505e-03,\n",
       "           5.0621e-01,  6.0668e-01,  7.6526e-01, -5.6484e-02, -2.1306e-01,\n",
       "           3.3852e-01,  2.5943e-01, -9.1834e-02,  5.2227e-01, -2.1885e-02,\n",
       "          -1.6835e-01,  3.1658e-01,  2.3124e-02, -2.2045e-01,  1.7294e-01,\n",
       "          -2.5761e-01,  2.6423e-01,  1.0870e-01, -4.9724e-01,  3.6477e-01,\n",
       "           3.7087e-01,  1.5082e-01, -3.0923e-01,  1.8171e-01,  3.9828e-01,\n",
       "          -3.1100e-01,  6.8356e-01,  1.6607e-04,  2.2973e-01,  6.3476e-02,\n",
       "          -1.1506e-01, -5.5647e-01, -3.1374e-01,  2.1509e-01,  9.2105e-02,\n",
       "          -5.0547e-01, -9.5287e-02,  1.5106e-01, -4.9544e-01,  7.8060e-01,\n",
       "           6.2385e-01, -3.7763e-01, -3.5664e-01, -7.1127e-02,  1.1295e-01,\n",
       "           1.2269e-01, -3.3879e-01,  2.9572e-01,  1.1312e-01,  1.0076e-01,\n",
       "          -1.0219e+00, -2.9326e-01,  2.4530e-01,  4.9147e-02, -4.2291e-01,\n",
       "           5.9939e-01,  7.9206e-02,  5.0359e-02,  3.8428e-01,  3.0534e-01,\n",
       "           1.4161e-01,  4.3910e-01,  7.0775e-02, -2.1375e-01,  6.5311e-02,\n",
       "           2.5600e-01, -3.0970e-01, -3.2766e-01, -6.3757e-01,  5.3684e-01,\n",
       "           1.5068e-01,  8.4160e-01,  6.5313e-01]], device='mps:0'),\n",
       " tensor([[-1.1097e-02, -5.0567e-01, -4.6699e-01, -6.4640e-01, -3.9556e-01,\n",
       "          -2.5266e-01, -5.5990e-01,  3.0801e-01, -1.2536e-01,  1.4189e-01,\n",
       "           2.9133e-01,  5.3879e-01, -3.6706e-01, -2.8859e-01, -4.8290e-01,\n",
       "          -5.3699e-02, -6.1741e-01, -1.6918e-02,  3.4690e-01,  3.0491e-01,\n",
       "           5.3718e-01,  1.8073e-01, -4.1129e-01,  2.9149e-01, -3.7959e-01,\n",
       "           6.6561e-02, -3.4985e-01, -3.1546e-01,  2.6210e-01,  2.8012e-01,\n",
       "           5.3618e-02,  7.7692e-01, -2.4307e-02,  1.0962e+00, -1.8413e-01,\n",
       "          -2.1778e-01,  1.5359e-01,  2.3320e-01,  8.6439e-01, -6.2204e-01,\n",
       "           5.6395e-01,  3.6489e-01,  2.7586e-02, -4.0063e-02,  2.7845e-01,\n",
       "           3.5451e-01,  1.4567e-01, -5.7913e-01, -2.7161e-01, -3.4037e-01,\n",
       "          -1.2979e-01, -1.5164e-01,  5.0058e-01, -2.3292e-01, -7.9546e-01,\n",
       "          -1.9253e-01,  1.1288e-02,  2.6982e-01,  1.1117e-01,  2.3591e-01,\n",
       "           2.8900e-02,  7.9425e-01, -1.9339e-01, -3.2966e-03,  2.0559e-01,\n",
       "           2.5502e-01, -7.6243e-01, -1.2873e-01, -7.2822e-01,  1.1548e-01,\n",
       "           2.1413e-01,  3.5162e-01,  1.0500e-01, -2.6816e-01, -1.9644e-01,\n",
       "           6.0243e-02,  1.6956e-01,  1.8610e-01, -2.0035e-01,  3.5967e-02,\n",
       "           2.1140e-01,  4.4140e-01,  4.5980e-01,  5.9922e-02,  1.6117e-01,\n",
       "           5.6960e-02,  2.0080e-01, -3.9869e-02, -3.1738e-02,  1.7672e-01,\n",
       "           1.1984e-01, -9.9577e-02, -3.7483e-01, -4.3956e-01, -9.5205e-02,\n",
       "          -3.0408e-01,  3.2250e-02, -2.9469e-01, -2.3954e-01, -3.3261e-01,\n",
       "           2.5982e-01,  9.2762e-02,  3.2750e-01, -9.3082e-02, -6.6514e-01,\n",
       "          -3.5248e-01,  4.0832e-01,  3.4586e-03, -2.8129e-01, -3.8448e-01,\n",
       "          -6.6713e-02, -4.9346e-01,  1.2765e-01, -4.4650e-01,  1.9893e-01,\n",
       "           1.0835e-01,  1.8686e-01, -3.2670e-01,  6.5172e-02,  8.8101e-02,\n",
       "           6.1068e-02, -5.0938e-01,  2.7421e-01,  1.8326e-01,  9.3046e-02,\n",
       "           2.2489e-01,  6.8015e-01, -1.6210e-01,  4.3454e-01,  3.5722e-01,\n",
       "           3.1169e-01,  1.6221e-01, -4.1686e-01, -5.0259e-01, -7.5302e-02,\n",
       "           1.8488e-02, -5.9908e-01, -5.7442e-01,  6.1723e-02, -2.2868e-01,\n",
       "           1.6627e-01, -7.3083e-02, -5.1768e-01, -5.4307e-01, -3.1494e-01,\n",
       "          -3.0772e-01, -2.0517e-02, -3.2393e-01, -4.8159e-01, -2.5398e-01,\n",
       "          -6.1867e-01, -7.2891e-01,  1.9443e-01,  1.5164e-01,  2.7714e-01,\n",
       "          -4.3163e-01,  5.4580e-01, -3.0612e-01, -4.8554e-01,  3.7642e-01,\n",
       "           4.3432e-01, -5.0330e-01,  2.9526e-01, -6.3499e-01,  5.1171e-02,\n",
       "           2.0480e-01,  2.5291e-02, -3.4147e-02,  4.2807e-01,  1.1727e-01,\n",
       "           3.9831e-01,  2.7152e-02,  1.2476e-01, -2.9187e-01,  1.5015e-01,\n",
       "          -4.5427e-02, -3.2147e-01, -1.1176e-01,  1.7279e-02, -5.2967e-02,\n",
       "           5.4936e-01, -1.7306e+00, -1.7852e-01,  7.0060e-02, -2.7460e-01,\n",
       "          -1.3173e-01,  2.1473e-01,  5.0407e-03,  3.2283e-01, -3.5674e-01,\n",
       "           8.3698e-02,  8.7128e-02,  8.4654e-02, -1.0725e-01, -5.0983e-02,\n",
       "           2.9101e-01,  7.5929e-01,  4.0793e-01,  9.7463e-02, -2.6657e-01,\n",
       "          -6.0645e-01,  4.1554e-01,  1.9081e-01,  2.0730e-01,  3.9000e-01,\n",
       "           6.7931e-02, -6.9343e-01,  2.9674e-01,  1.9859e-01,  3.7002e-01,\n",
       "           2.5546e-02, -5.8496e-02, -4.8962e-01, -2.4682e-01, -4.3705e-01,\n",
       "           2.7759e-01, -1.3436e-01, -2.5922e-01,  6.3173e-01,  1.9273e-01,\n",
       "          -5.3531e-02,  3.9653e-01,  1.8733e-01, -4.1041e-01,  4.1633e-01,\n",
       "           5.2814e-01,  7.9114e-02,  1.4573e-01, -2.8055e-01, -1.6673e-01,\n",
       "          -3.1532e-01, -4.4420e-01,  4.4703e-01,  8.3941e-01, -7.2084e-03,\n",
       "           5.5343e-01, -3.5757e-01,  5.1174e-01, -3.6424e-01, -6.7779e-03,\n",
       "          -4.6486e-02,  6.8700e-02, -3.3988e-01,  1.6284e-01,  1.5551e-01,\n",
       "           1.0340e-01,  1.9709e-01,  8.5425e-01,  4.6494e-01,  1.4684e-01,\n",
       "           4.1031e-01,  2.5417e-02,  3.2080e-01,  7.8684e-01, -3.0254e-01,\n",
       "           3.4990e-01, -7.1578e-03, -4.1842e-01,  5.6461e-01, -3.5139e-01,\n",
       "           1.4248e-01, -2.3020e-01,  2.2987e-01,  1.7866e-01, -2.7854e-01,\n",
       "           2.1426e-01,  6.5273e-01,  9.5860e-02, -3.9283e-01,  7.3417e-01,\n",
       "          -4.0214e-01,  7.1262e-01, -3.0138e-02, -2.6992e-02, -1.0120e-01,\n",
       "          -7.4577e-01, -6.8569e-01,  1.4294e-01,  3.0808e-01, -7.6332e-01,\n",
       "          -3.4295e-01,  1.2380e-01,  1.3255e-01,  3.5068e-01,  2.4156e-01,\n",
       "           4.6144e-01, -1.7016e-01, -4.7185e-02, -4.4024e-01, -4.6015e-01,\n",
       "           1.1917e-01,  7.3311e-02,  9.5311e-02, -9.8199e-02,  9.2587e-02,\n",
       "           6.1067e-01,  2.4297e-01,  2.6799e-01,  1.4946e-01, -3.4610e-01,\n",
       "          -2.7723e-02,  1.7135e-01,  3.7874e-01, -6.1900e-01, -2.1807e-01,\n",
       "           5.3543e-02,  8.3164e-02,  6.6826e-01, -2.1832e-01, -1.1546e-01,\n",
       "          -1.2450e-01,  5.9739e-01, -2.2108e-01,  6.8002e-02,  1.0269e-01,\n",
       "          -6.8586e-01,  2.7619e-01, -5.9939e-02, -8.5297e-02, -5.2861e-01,\n",
       "           2.2997e-01, -5.7787e-02, -1.6068e-01, -7.3269e-03, -7.7601e-02,\n",
       "           2.2454e-01,  1.3406e-01,  2.8044e-01, -5.6270e-01, -4.2092e-01,\n",
       "           1.1781e-01, -3.2365e-01, -6.6016e-02, -4.5861e-01,  3.6766e-01,\n",
       "           1.3890e-01,  4.3829e-01, -1.0837e-01, -2.8576e-02, -1.3190e-01,\n",
       "           5.2866e-01, -6.2780e-01,  1.5573e-02,  4.8525e-01, -3.5595e-01,\n",
       "          -1.8139e-01,  4.7644e-01, -2.8011e-01,  4.9664e-02,  2.2982e-01,\n",
       "           3.7738e-01, -1.1530e-01, -4.7105e-01, -5.5247e-02,  9.3761e-01,\n",
       "           4.8980e-01,  2.0874e-02, -6.3820e-01,  1.2560e-01,  4.2066e-01,\n",
       "          -3.4219e-01,  1.0777e-01, -3.7280e-01, -2.3519e-01,  8.6456e-02,\n",
       "           3.3260e-02,  5.0344e-01, -5.8998e-02,  1.9741e-01,  4.6778e-01,\n",
       "           1.7011e-01, -6.9331e-01, -3.0557e-01,  6.3124e-01,  5.4670e-01,\n",
       "          -1.2247e-01, -5.4832e-01,  1.1480e-01, -6.4563e-01,  3.1198e-02,\n",
       "          -3.8623e-03, -3.5755e-01,  2.9653e-01, -1.4962e-01, -2.5683e-01,\n",
       "          -2.0003e-01, -5.3748e-01,  4.4973e-03,  1.0291e-01, -6.9595e-02,\n",
       "           4.1738e-01, -7.9610e-03, -3.1154e-01,  1.0058e+00, -2.7292e-01,\n",
       "          -3.1356e-02,  7.4139e-01, -5.4234e-01,  1.0457e-02,  2.4642e-01,\n",
       "          -1.3433e-01, -1.2416e-02,  2.4323e-02,  1.0660e+00,  1.1515e-05,\n",
       "           3.6402e-02, -2.0346e-01,  4.2203e-01, -4.7659e-01,  1.9394e-01,\n",
       "          -4.9958e-01,  4.8310e-01, -6.9399e-02,  2.4253e-01, -8.2329e-01,\n",
       "           4.3257e-01, -3.0604e-01, -3.2040e-01, -6.3922e-01,  5.0359e-02,\n",
       "          -5.0711e-02,  6.6524e-02, -5.8707e-02,  6.0787e-01, -1.0953e+01,\n",
       "           1.0946e-01,  8.4108e-01, -3.2618e-02,  1.3653e-01,  4.3336e-01,\n",
       "           1.2158e-01, -4.6684e-01,  7.2509e-01, -2.3242e-01, -6.6092e-03,\n",
       "          -3.5861e-01, -1.2444e-01, -1.3819e-01, -6.3519e-01,  4.0758e-01,\n",
       "          -1.3255e-01, -5.9114e-01,  2.8540e-01, -1.6945e-01,  2.4978e-01,\n",
       "          -2.3651e-01,  1.5671e-01,  5.9352e-01, -6.1010e-01, -3.3943e-01,\n",
       "          -1.1958e-01, -2.9991e-01,  5.1723e-01,  4.0217e-01, -2.3672e-01,\n",
       "           2.9096e-01,  3.3529e-02,  2.9455e-01, -5.5694e-01, -4.8133e-01,\n",
       "           4.3369e-01, -3.9950e-01,  2.1957e-01, -9.9943e-01, -3.3265e-01,\n",
       "          -1.8082e-01, -5.0951e-01, -2.7870e-02,  5.3812e-01,  7.5524e-01,\n",
       "           1.2821e+00,  1.1534e+00,  9.8198e-02,  4.6362e-01, -1.0523e-01,\n",
       "          -6.6935e-01,  5.1619e-02, -5.6525e-01, -4.1224e-01,  8.7196e-02,\n",
       "           2.2385e-01, -5.4352e-01,  4.7854e-01, -5.9897e-01,  1.8912e-01,\n",
       "           1.4329e-01,  7.9852e-02,  1.7368e-01, -5.1897e-01, -5.5039e-01,\n",
       "          -1.3282e-01,  8.1201e-02,  8.6324e-02, -6.7881e-01,  1.1881e-01,\n",
       "           1.7006e-01,  6.8197e-01, -2.5397e-01, -4.4268e-01,  4.1137e-01,\n",
       "          -2.2746e-03,  9.3809e-02,  1.3158e+00, -2.6036e-01, -1.7413e-01,\n",
       "          -4.8428e-02, -2.3194e-01, -3.3484e-01, -3.9656e-01, -4.5613e-01,\n",
       "          -2.5020e-02,  3.6573e-01,  5.1948e-01,  2.6062e-01, -2.3329e-02,\n",
       "          -3.0054e-01, -4.1445e-02,  5.1977e-02,  3.3862e-01,  9.6772e-01,\n",
       "          -6.6887e-01,  2.6280e-01, -1.9148e-01,  9.2178e-02,  2.2223e-01,\n",
       "          -5.7106e-01, -1.6871e-02,  2.9434e-03,  1.3975e-02,  5.7281e-01,\n",
       "           1.9705e-01,  2.6453e-01,  3.1190e-02,  2.2859e-01, -4.1682e-01,\n",
       "          -4.4044e-01,  2.9125e-04, -1.3190e-01, -5.8294e-01, -4.1034e-01,\n",
       "          -1.3514e-01,  1.3712e-01, -2.8585e-01, -5.3585e-01, -2.5012e-01,\n",
       "           1.4192e-01,  2.5900e-03,  4.9952e-02, -3.3857e-01,  4.7379e-01,\n",
       "          -7.6918e-01, -2.1429e-01, -2.7578e-01, -3.8495e-03,  1.4633e-01,\n",
       "           2.5508e-01, -5.1518e-01, -2.7838e-01, -1.0123e-01, -9.2314e-01,\n",
       "          -8.4489e-01,  2.2927e-02,  7.4620e-01, -1.1960e-01,  1.1861e-01,\n",
       "           8.5428e-02, -3.1671e-03, -3.7708e-01,  2.5194e-01, -2.4644e-01,\n",
       "           2.8149e-01,  4.3265e-01, -4.9583e-01,  2.9773e-02, -1.2792e-01,\n",
       "          -1.8992e-02, -1.0887e-01, -8.8647e-02, -5.4131e-01, -6.1371e-01,\n",
       "          -4.1197e-01,  1.3736e-01, -2.2747e-01,  4.1516e-01,  6.8072e-02,\n",
       "           1.2582e-01, -2.0826e-01,  9.8224e-02,  8.2540e-01,  6.0197e-01,\n",
       "           2.3014e-01, -3.2920e-01, -7.7666e-01,  3.4140e-02,  4.1642e-02,\n",
       "          -1.6450e-01,  2.1317e-01, -1.6384e-01, -3.0657e-01, -1.0112e-01,\n",
       "           1.2620e-02,  1.0002e+00, -2.0790e-01, -1.9155e-01,  1.0507e-01,\n",
       "           7.4626e-01, -6.0834e-01, -5.7221e-01,  3.7612e-01,  5.0418e-02,\n",
       "          -5.6831e-01, -1.9450e-01,  4.0315e-01, -1.7111e-01,  8.9921e-02,\n",
       "           3.7572e-01,  1.7821e-01, -1.8612e-01,  2.2711e-01, -6.7291e-02,\n",
       "          -1.7001e-01,  3.2731e-01, -1.6533e-01, -7.7254e-02,  4.0384e-01,\n",
       "           9.9517e-02,  1.5584e-01, -2.5262e-01, -4.8444e-01, -2.7025e-01,\n",
       "           1.5439e-01,  1.1110e+00, -1.7989e-01, -3.2527e-01,  6.6881e-01,\n",
       "          -3.0956e-01,  3.8928e-01,  4.8478e-01, -2.2611e-01, -5.0807e-02,\n",
       "           5.8049e-01,  3.0288e-01, -7.3236e-01,  2.2417e-01,  3.1670e-01,\n",
       "          -1.7335e-01, -5.3095e-01, -3.2838e-01, -2.1330e-01, -3.5189e-02,\n",
       "          -3.6356e-01,  5.2132e-01,  1.2255e-01,  3.9253e-01, -1.0843e-01,\n",
       "          -8.7207e-01,  4.7185e-01,  2.6128e-01,  8.0250e-01, -6.2568e-03,\n",
       "          -4.4880e-01, -3.5541e-01, -3.5790e-01, -6.6453e-02,  3.6543e-01,\n",
       "           2.0116e-01, -1.0333e+00,  4.6071e-02,  4.0507e-01, -3.9812e-02,\n",
       "           4.7083e-01, -1.4254e-01,  1.9634e-01,  4.7891e-01, -4.3918e-02,\n",
       "          -2.1856e-01,  1.3161e-01, -3.3027e-02, -2.7190e-01,  4.2861e-01,\n",
       "           4.6755e-01,  1.6062e-02,  2.5454e-02,  6.4881e-02, -4.2427e-01,\n",
       "          -3.4980e-01,  7.7064e-02, -1.7966e-01, -3.5675e-01, -1.4295e-01,\n",
       "          -4.0726e-01, -5.1772e-01, -6.6164e-02, -1.1357e-01,  1.3233e-01,\n",
       "           3.1032e-01,  1.3036e-01,  3.8431e-01, -2.3708e-01, -1.2541e-01,\n",
       "          -2.7977e-01,  4.5653e-01, -2.7595e-01,  8.0808e-01,  2.0566e-01,\n",
       "          -1.4423e-01,  2.0720e-01,  2.1514e-01, -2.9046e-01,  1.3028e-01,\n",
       "          -2.7828e-01,  2.8810e-01, -5.7798e-01, -3.4110e-01, -5.9710e-02,\n",
       "          -7.7948e-02, -1.5049e-01, -2.8505e-01, -1.0179e-01,  1.0608e-01,\n",
       "          -9.9043e-02,  9.7868e-01, -4.1759e-01,  6.3833e-01, -1.0342e-01,\n",
       "          -1.1350e-01, -7.5089e-01,  3.9397e-01,  3.6376e-01, -1.4798e-01,\n",
       "          -5.6434e-01, -2.1653e-01, -8.9158e-02, -4.2284e-01,  8.8621e-01,\n",
       "           5.8157e-01, -7.4673e-01, -3.7246e-01, -1.5919e-01, -2.4113e-01,\n",
       "          -4.7389e-02,  6.9820e-02, -3.2710e-01, -1.3295e-02, -1.6580e-01,\n",
       "          -9.8715e-01,  2.7641e-01, -3.6473e-02,  4.9205e-01, -2.5466e-01,\n",
       "           7.6979e-01, -8.6017e-02, -1.2749e-01,  1.1690e+00,  1.1169e-01,\n",
       "          -6.1844e-02,  5.9531e-01, -1.6837e-01, -6.2973e-01, -7.0295e-02,\n",
       "           4.4239e-02, -3.1098e-01,  8.7670e-02, -4.7880e-01,  2.9353e-01,\n",
       "           2.3648e-01, -7.1226e-02,  4.3877e-01]], device='mps:0'),\n",
       " tensor([[ 0.0069, -0.5770, -0.5384,  ...,  0.3485,  0.5202,  0.6808],\n",
       "         [-0.2434, -0.3725, -0.3338,  ...,  0.2009,  0.3829,  0.4365],\n",
       "         [-0.1212, -0.7440, -0.6361,  ...,  0.1977,  0.2087,  0.3996]],\n",
       "        device='mps:0'),\n",
       " tensor([[-6.1750e-01, -4.2829e-01, -2.3934e-01, -9.4116e-01, -3.4548e-01,\n",
       "          -6.5496e-02, -6.3874e-01,  9.7828e-01, -9.5892e-02,  4.0096e-01,\n",
       "           1.4919e-01, -5.1603e-03,  2.0980e-01,  1.3680e-01, -6.9506e-01,\n",
       "           1.6740e-01,  4.6094e-01,  2.1159e-01,  3.5951e-01, -4.4329e-02,\n",
       "           7.9832e-01,  3.4245e-01,  1.8019e-01, -2.5205e-01, -4.5876e-01,\n",
       "           5.8994e-01, -2.4628e-01, -9.2014e-02,  2.0509e-01, -8.1949e-01,\n",
       "          -1.3586e-01, -4.9757e-02, -3.6281e-01,  8.7602e-01,  1.1948e-01,\n",
       "          -2.7310e-01, -1.5276e-01, -2.7996e-01,  2.3494e-01, -6.3637e-02,\n",
       "           3.7814e-01,  2.6592e-02, -2.9347e-01,  3.1386e-01, -1.6421e-01,\n",
       "          -1.4057e-01,  1.4047e-02, -7.8429e-01,  1.3666e-02,  1.8955e-01,\n",
       "           5.8851e-02, -4.7205e-02,  1.9231e-01,  3.7030e-01, -1.0617e+00,\n",
       "          -2.6348e-01,  2.7603e-01,  2.9849e-01, -1.7932e-01, -5.7784e-02,\n",
       "          -1.9989e-01,  9.0201e-01,  1.7949e-01, -2.7822e-01,  1.7824e-01,\n",
       "          -4.4899e-01, -5.4601e-01, -6.0781e-01, -1.0682e-01, -1.1967e-01,\n",
       "          -1.0002e-01,  9.7528e-02,  1.4009e-01, -2.9385e-01,  2.3170e-01,\n",
       "           2.1813e-01,  4.7024e-02, -2.7652e-02, -4.3158e-01, -2.6546e-01,\n",
       "           3.6947e-01,  3.4425e-02, -1.4730e-02,  2.0326e-01, -5.8744e-01,\n",
       "          -2.0226e-01,  3.7959e-01,  1.5703e-01,  2.5914e-01,  2.2662e-03,\n",
       "           5.9967e-01,  1.2024e-02,  8.0339e-02, -4.5369e-01,  5.8427e-01,\n",
       "          -5.4955e-01,  1.0329e-01, -1.6007e-01,  4.9532e-02, -5.2088e-01,\n",
       "          -9.8735e-02,  2.4494e-01, -1.8085e-01,  1.6053e-01, -4.6592e-01,\n",
       "          -3.2983e-02,  2.4514e-01,  3.0468e-01, -4.9183e-01,  4.2685e-02,\n",
       "          -6.5134e-03, -2.9791e-01,  7.3694e-02, -6.6407e-01, -8.5795e-01,\n",
       "           5.8354e-01,  1.8428e-02, -4.5240e-01, -2.3930e-01,  7.4443e-01,\n",
       "           2.9738e-01, -2.3201e-01, -5.3072e-02, -3.5548e-01,  2.0259e-01,\n",
       "          -1.8518e-01,  2.5689e-01, -2.7018e-01,  3.8439e-01,  5.5026e-01,\n",
       "           3.1581e-01,  2.8489e-01, -1.8327e-01,  2.7637e-01,  1.4465e-01,\n",
       "           3.6947e-02, -6.6614e-02, -1.5929e-01,  1.5180e-01,  1.2041e-01,\n",
       "          -2.9580e-01, -3.2999e-01,  5.4746e-03,  1.1182e-01, -3.3279e-01,\n",
       "          -1.0202e-01,  1.8680e-01,  3.4808e-01, -1.4092e-01, -2.5403e-01,\n",
       "          -7.8984e-01, -4.2328e-01,  1.0217e-01, -1.8560e-01, -1.7135e-01,\n",
       "          -4.1911e-01,  3.8121e-01, -6.6660e-02,  4.7638e-01,  3.4189e-01,\n",
       "           6.9978e-01, -1.5311e-01,  5.9776e-02, -5.6412e-01,  3.5197e-01,\n",
       "           3.8270e-02, -6.1110e-01, -1.7247e-01,  1.5680e-01, -4.1813e-04,\n",
       "           6.8195e-01,  7.0979e-02,  4.0124e-01,  1.2070e-01, -3.2553e-01,\n",
       "           8.2089e-03,  1.9890e-01, -1.2864e-01, -1.4297e-01,  7.3766e-01,\n",
       "          -5.5188e-01, -2.3029e+00,  1.0721e-01, -3.5844e-01, -3.6529e-01,\n",
       "          -1.8135e-01,  3.7255e-01,  3.3772e-02,  4.9966e-01,  1.5486e-01,\n",
       "           3.7598e-02,  1.6729e-01, -2.8025e-02,  3.6851e-01, -8.4399e-02,\n",
       "           2.6751e-01,  3.3341e-01,  1.2988e-01, -2.3298e-01, -1.0956e-01,\n",
       "          -3.9227e-01,  2.4031e-01, -9.8916e-02,  8.6010e-01,  2.3260e-01,\n",
       "           2.0943e-01, -4.3657e-03, -4.7262e-02,  4.8964e-01,  4.3099e-01,\n",
       "          -6.6541e-02, -1.6585e-01, -4.9522e-01,  4.6269e-03, -4.1279e-02,\n",
       "          -1.2963e-02, -4.5611e-01,  1.3753e-01,  4.2841e-01, -1.2416e-01,\n",
       "          -1.8354e-01,  1.5525e-01, -4.7715e-02, -3.1517e-01, -2.6492e-02,\n",
       "           6.8453e-01,  1.9080e-01, -2.2994e-01,  1.2377e-01, -1.1896e-01,\n",
       "           3.7006e-02, -4.6975e-01,  7.0044e-01,  6.1147e-01,  1.5585e-01,\n",
       "           1.6214e-02, -6.9943e-01,  2.3699e-01, -3.1574e-01, -4.7647e-03,\n",
       "          -1.5027e-01, -1.5731e-01, -4.2986e-01, -5.7285e-02, -5.5854e-02,\n",
       "           4.0115e-01,  7.9943e-02,  4.6653e-01,  1.1837e+00,  2.7973e-01,\n",
       "           1.9184e-02,  5.6097e-01,  2.0801e-01,  4.2492e-01, -4.0180e-01,\n",
       "           3.3082e-01,  8.8825e-02, -6.3896e-01,  2.7188e-01, -4.1593e-01,\n",
       "           4.4963e-02, -4.1432e-02, -3.7926e-01,  1.4208e-01, -2.1276e-02,\n",
       "           3.4237e-01,  3.3517e-01,  1.5186e-01,  1.6838e-01,  2.9411e-01,\n",
       "          -3.1322e-01,  5.9715e-01, -6.0809e-02, -1.1127e-01, -6.3492e-01,\n",
       "          -7.6732e-01,  9.5206e-02,  4.4743e-01, -1.2880e-01, -6.1279e-01,\n",
       "           1.0462e-01, -2.3866e-01, -1.2257e-01, -3.5064e-02,  2.1191e-01,\n",
       "           2.7455e-01,  5.0687e-02, -9.5436e-02,  6.4791e-02, -4.6950e-01,\n",
       "          -4.9277e-01,  4.9140e-02, -1.5758e-01, -1.3063e-01,  5.3918e-01,\n",
       "          -7.9415e-02,  1.1575e-01,  5.7221e-01,  1.9799e-02, -4.5976e-01,\n",
       "          -2.5640e-01,  1.4802e-01,  2.8667e-01, -7.2082e-02,  2.1981e-01,\n",
       "           3.4119e-01,  2.6467e-02,  5.1649e-01,  1.4823e-01, -2.1046e-01,\n",
       "          -2.7239e-01,  5.1173e-01, -1.5755e-01,  6.1403e-02, -1.2527e-01,\n",
       "          -8.7207e-01, -5.6150e-02, -7.8413e-02, -1.6833e-01, -5.4440e-01,\n",
       "           2.6566e-01,  1.4351e-01,  1.0561e-01,  2.2350e-01, -1.5984e-01,\n",
       "           7.0123e-01,  6.5285e-01, -3.3100e-01, -1.6943e-01, -4.0104e-01,\n",
       "           9.3922e-02, -6.8374e-02, -2.9329e-01, -1.4009e-02,  9.5368e-03,\n",
       "          -2.2765e-01, -1.2516e-01, -1.0805e-01,  2.8921e-01, -1.8876e-01,\n",
       "           5.2862e-01,  1.2189e-01,  8.3649e-02,  4.3873e-01,  2.0612e-02,\n",
       "           1.3071e-01,  3.6515e-01, -4.2117e-01,  1.2870e-01,  4.0424e-01,\n",
       "           4.0578e-01, -1.0551e+00, -8.3291e-01, -1.4936e-01,  3.6111e-01,\n",
       "           2.2860e-01,  2.5189e-01,  1.6999e-02, -7.2215e-03,  8.9455e-02,\n",
       "          -2.3402e-01,  3.5135e-01, -6.6415e-02, -5.7412e-01,  3.0561e-01,\n",
       "           2.3567e-01,  5.3263e-01, -2.7377e-01, -5.0579e-01, -6.5233e-02,\n",
       "           2.4616e-01, -3.6381e-01,  1.1392e-01,  2.3480e-01,  4.5466e-01,\n",
       "          -3.6015e-01, -5.6164e-01, -7.0282e-02, -7.3986e-01, -1.3578e-01,\n",
       "           3.2483e-01, -2.1437e-01,  2.7178e-02, -2.4758e-01,  1.5231e-01,\n",
       "          -1.7925e-02, -1.8879e-01,  6.0757e-01,  2.0522e-01, -2.5855e-01,\n",
       "           1.1981e-01, -2.3892e-01, -1.9475e-01,  5.0091e-01, -5.0226e-01,\n",
       "          -3.0348e-01, -5.0960e-02, -1.5349e-01, -3.2418e-01,  3.2181e-01,\n",
       "          -4.1389e-01, -1.4861e-01,  1.6748e-01, -6.4736e-02, -1.3068e-01,\n",
       "          -3.5233e-01,  1.4268e-01, -1.2707e-01, -4.9883e-02, -1.7654e-02,\n",
       "          -1.3353e-01,  6.0900e-02,  1.3308e-01,  6.4538e-02,  1.4388e-01,\n",
       "           5.3696e-01, -1.1464e-01,  2.7582e-03, -2.5900e-01, -2.2469e-01,\n",
       "           5.2498e-01, -9.4973e-02, -3.3117e-01,  5.4928e-01, -1.1156e+01,\n",
       "           1.0386e-01,  7.2631e-01, -5.5808e-01,  5.5128e-02, -2.8539e-01,\n",
       "           2.1085e-01, -3.0009e-01,  7.9101e-02,  2.4967e-01, -2.8460e-01,\n",
       "          -1.2277e-01, -2.7364e-01, -2.3193e-01, -4.0872e-01, -3.9742e-01,\n",
       "          -4.7229e-01, -5.6789e-01,  1.1772e-01, -1.7663e-01, -1.0996e-01,\n",
       "          -5.7036e-01, -1.9379e-01,  2.9696e-01, -4.9559e-03, -5.1069e-01,\n",
       "           5.9541e-02, -2.2948e-01,  2.6399e-01,  4.6126e-02,  2.2045e-01,\n",
       "          -1.9949e-02,  1.4108e-01,  2.0669e-01, -4.4946e-02, -4.5100e-01,\n",
       "          -4.4853e-01,  2.2129e-01, -1.5955e-01, -2.2267e-01, -5.4017e-02,\n",
       "          -3.3838e-01, -2.7689e-01,  1.3447e-01,  3.7594e-01,  6.2596e-01,\n",
       "           1.1595e+00,  5.6208e-01,  2.9245e-01, -1.0625e-01, -8.5438e-02,\n",
       "          -1.0762e+00,  5.5900e-01, -3.7318e-01,  8.1787e-02, -2.5119e-01,\n",
       "           6.6201e-01, -4.9906e-01,  5.4685e-01, -4.0754e-01, -2.5500e-01,\n",
       "           3.7179e-01,  3.0194e-01,  7.1104e-03,  2.8239e-02, -2.7432e-01,\n",
       "           4.0257e-01, -2.6094e-01, -3.6915e-02, -3.7819e-01,  4.0081e-01,\n",
       "          -3.9472e-01,  5.0770e-01, -5.6534e-02,  1.2945e-01,  1.8567e-01,\n",
       "          -3.5001e-01,  1.6610e-01,  1.0337e+00,  1.7414e-01,  4.1071e-01,\n",
       "           1.7420e-01,  5.1160e-01,  5.3064e-02, -8.8958e-01,  1.3691e-01,\n",
       "           2.0502e-01,  2.0789e-01, -6.7096e-02,  5.7131e-01, -2.5517e-02,\n",
       "          -6.5181e-01, -2.3094e-01, -3.2169e-01,  4.3108e-01,  6.5352e-01,\n",
       "          -2.2597e-01, -4.4343e-01,  2.3894e-01, -7.9776e-03, -5.3395e-01,\n",
       "           9.0074e-02, -2.2320e-01, -5.2792e-01,  8.7275e-02, -5.8914e-01,\n",
       "           1.4985e-01, -1.2428e-01,  5.4238e-02,  6.6887e-01, -4.0159e-01,\n",
       "          -1.6113e-01,  3.6602e-02, -4.4532e-02, -4.1270e-01, -4.9520e-01,\n",
       "          -7.7083e-02,  4.1073e-01, -1.9776e-01, -2.7976e-01,  7.0636e-02,\n",
       "          -3.0673e-01,  2.2584e-01, -1.7377e-01, -6.4935e-01,  3.2879e-01,\n",
       "          -6.0415e-01,  2.0578e-01,  2.5360e-01, -2.2507e-01,  4.9017e-01,\n",
       "           1.1337e-01, -8.1817e-02,  1.9397e-01, -9.9369e-02, -7.8554e-01,\n",
       "          -3.2573e-01, -2.7299e-02,  3.2662e-01, -4.2844e-01,  1.5776e-01,\n",
       "           1.5022e-01, -4.5752e-02, -2.2574e-01,  2.7904e-01,  1.0174e-01,\n",
       "           3.7762e-01,  3.9846e-01, -5.5006e-01,  1.1727e-01, -2.9867e-01,\n",
       "          -2.9262e-02, -6.1174e-02,  1.7952e-01,  2.3654e-01, -7.0764e-01,\n",
       "          -1.5224e-01, -2.6434e-01, -3.0965e-01, -2.1650e-01, -2.5120e-01,\n",
       "           2.0680e-01,  1.4038e-01, -1.5897e-02, -1.1308e-01,  5.8077e-01,\n",
       "           4.0814e-02, -1.1167e-01, -4.5997e-01,  3.1443e-01,  4.3178e-01,\n",
       "           2.1794e-01,  7.5875e-02, -1.2830e-01, -5.5532e-01, -1.1922e-02,\n",
       "          -4.3644e-01,  9.1081e-01, -7.3906e-01, -7.1970e-01,  6.6086e-02,\n",
       "           1.4951e-02, -8.8259e-04, -2.3450e-01, -5.0265e-01,  8.1298e-02,\n",
       "          -2.1017e-01, -1.4891e-01,  6.1550e-03, -1.0607e-02,  1.6299e-01,\n",
       "           4.0742e-01,  3.6838e-01,  1.5951e-01, -3.3938e-01, -4.4081e-01,\n",
       "          -6.0244e-01,  1.4567e-02, -1.6795e-01,  2.9330e-01, -2.6752e-01,\n",
       "          -3.2661e-01,  1.6043e-02, -9.1892e-03, -2.2272e-01,  5.2876e-01,\n",
       "           1.2427e-01,  1.6396e+00, -3.7287e-01, -4.4551e-01,  2.9018e-01,\n",
       "           7.2306e-02, -2.7073e-02, -2.8130e-01,  2.5074e-01,  9.5447e-02,\n",
       "           4.0194e-02, -3.6304e-01,  1.5023e-02, -2.0558e-02,  3.2773e-01,\n",
       "           3.2056e-01, -1.0703e+00, -2.0238e-01, -1.7993e-01,  5.5664e-01,\n",
       "           3.2388e-02,  6.3185e-01, -2.2305e-01,  3.0577e-01,  4.6106e-01,\n",
       "          -8.6667e-01, -2.3000e-01, -1.6129e-01,  2.7295e-01, -1.0583e-01,\n",
       "          -9.8735e-02, -1.6178e-01, -3.7165e-02, -1.6936e-01,  3.9538e-01,\n",
       "           7.4751e-01, -7.8426e-01,  7.4081e-01,  7.0467e-01, -9.8234e-02,\n",
       "           5.6815e-01,  9.4187e-02,  3.2641e-01, -1.4659e-01,  2.2669e-02,\n",
       "           2.6200e-01,  1.8542e-01, -6.3165e-01, -2.3962e-01,  2.4473e-01,\n",
       "           4.0158e-01, -1.1943e-01,  1.2108e-01,  7.9292e-02,  3.0960e-02,\n",
       "          -3.7691e-01, -1.0145e-01, -2.4552e-01, -3.9904e-01,  1.1356e-01,\n",
       "           1.4603e-02, -1.5001e-01, -8.6997e-02,  4.7709e-01, -1.7580e-01,\n",
       "           7.5067e-01,  4.2910e-01,  3.7145e-01, -3.5103e-01,  8.9950e-04,\n",
       "          -5.7565e-02,  6.0671e-01, -3.1670e-01,  7.7385e-01,  6.3337e-02,\n",
       "          -2.3140e-01,  2.7251e-01,  6.0547e-03,  1.7829e-01,  2.9892e-01,\n",
       "          -2.3519e-01,  1.2702e-01, -3.0718e-01, -2.3711e-01,  2.8330e-01,\n",
       "          -1.8640e-01,  1.1542e-01, -7.2270e-01, -6.3602e-03,  3.9857e-01,\n",
       "          -3.2051e-01,  4.9873e-01, -2.3074e-01,  6.1026e-01, -4.3509e-01,\n",
       "           3.2669e-02, -4.7470e-01, -4.3275e-01, -2.8508e-01,  5.1854e-02,\n",
       "          -8.6027e-01, -1.9453e-01,  1.7684e-01,  3.7405e-02,  2.1707e-01,\n",
       "           7.5155e-01, -3.3794e-01, -2.4771e-01, -2.0386e-01,  4.0190e-02,\n",
       "           1.8299e-01, -8.2071e-02, -3.5586e-01,  2.1161e-01, -3.5607e-01,\n",
       "          -1.0520e+00, -2.3117e-01,  8.9990e-02,  3.8419e-01, -3.5853e-01,\n",
       "           1.1570e+00,  2.5132e-02, -2.0226e-01,  6.7253e-01,  2.0808e-01,\n",
       "           6.4157e-01,  6.0323e-01,  1.3628e-03, -4.1259e-01, -5.4109e-02,\n",
       "          -1.5877e-01, -3.2881e-01,  1.5757e-01, -1.8346e-01,  5.0556e-01,\n",
       "           5.4802e-01, -2.0806e-02,  5.4756e-01]], device='mps:0'),\n",
       " tensor([[ 4.7106e-03, -4.1941e-01, -4.9094e-01, -8.5242e-01, -5.7896e-02,\n",
       "          -7.7640e-02, -9.2377e-01,  3.4477e-01,  2.6829e-01,  5.3953e-01,\n",
       "           4.2638e-01,  6.9053e-01, -2.2860e-01, -4.7452e-02, -2.6848e-01,\n",
       "          -3.7506e-02, -5.6876e-02,  2.2354e-01,  8.4389e-01,  4.7237e-01,\n",
       "           2.5999e-01,  6.3958e-01, -7.9325e-03, -1.2462e-01, -8.4050e-01,\n",
       "           5.0987e-01, -7.6141e-02,  2.7592e-01,  1.0701e-01, -6.5476e-01,\n",
       "           1.4336e-02,  6.6116e-01, -9.9763e-02,  1.3975e+00, -2.1596e-01,\n",
       "          -1.1600e-01,  2.6595e-01, -3.6289e-01,  3.5959e-02, -2.0357e-01,\n",
       "           2.2580e-01,  1.8401e-01,  1.4554e-01,  2.7283e-01,  4.4744e-02,\n",
       "           9.6394e-02,  6.5960e-02, -2.6560e-01, -8.6185e-02, -5.8277e-01,\n",
       "          -2.1472e-01, -3.8237e-01,  3.5499e-01,  2.4389e-01, -5.2132e-01,\n",
       "          -4.4865e-01,  3.5555e-01, -1.2621e-02,  8.3075e-02,  4.1382e-02,\n",
       "          -2.0682e-01,  4.7834e-01, -2.4181e-01,  1.2139e-01,  1.2806e-01,\n",
       "           1.9160e-01, -4.2520e-01, -4.1385e-01, -4.0163e-01,  3.2918e-02,\n",
       "           8.1299e-02,  2.9461e-01,  1.1216e-01, -1.9333e-01,  4.3844e-01,\n",
       "          -3.4267e-01,  3.5242e-01, -4.2356e-01, -4.4045e-01,  1.3743e-01,\n",
       "           1.9119e-01,  2.7547e-01,  4.5287e-01, -3.8337e-01,  9.2916e-02,\n",
       "          -1.4152e-01,  2.6484e-01,  9.0211e-02,  1.1440e-01,  1.2222e-01,\n",
       "          -3.4360e-01,  2.6312e-01, -2.6097e-01, -1.7055e-01,  2.0914e-01,\n",
       "          -6.0062e-04, -8.2452e-02, -1.3102e-02,  1.2713e-01, -2.7348e-01,\n",
       "           1.5023e-01,  2.4396e-02,  7.8356e-02,  2.9417e-01, -4.7802e-01,\n",
       "          -7.4697e-01,  2.3313e-01, -4.4759e-01, -5.5595e-01, -7.0881e-02,\n",
       "          -2.6995e-01, -3.6214e-01,  2.2038e-01, -2.9553e-01, -1.7661e-01,\n",
       "           1.7625e-01,  1.4715e-01,  3.2423e-01,  5.1341e-02,  2.2535e-01,\n",
       "           1.7458e-01, -2.8183e-02,  2.0469e-02,  8.2624e-02,  5.0804e-02,\n",
       "          -1.2911e-01,  5.6342e-01,  7.8041e-02,  4.9365e-01,  5.6345e-01,\n",
       "           1.3255e-01, -5.1905e-02, -5.0769e-01, -1.9219e-01,  4.6027e-01,\n",
       "           4.5581e-02, -1.1714e-01, -1.7161e-01,  7.6093e-02, -5.0147e-03,\n",
       "           7.4886e-02,  9.5607e-02, -1.3946e-01, -3.3946e-01, -2.3415e-01,\n",
       "          -2.1947e-01, -3.1592e-02, -2.2009e-01, -4.6517e-01, -3.7663e-01,\n",
       "          -8.9297e-01, -5.9349e-01,  5.0128e-01, -7.8229e-02, -2.4801e-01,\n",
       "          -2.6828e-01,  2.8932e-01,  2.0871e-03, -1.6385e-01, -1.7172e-02,\n",
       "           8.3851e-01, -3.3183e-02,  1.4295e-01, -4.8564e-01, -1.3618e-01,\n",
       "           1.1155e-01, -8.7636e-02, -6.7902e-02,  9.3916e-02, -1.7767e-01,\n",
       "          -4.4016e-02, -4.3730e-01,  4.6307e-01,  3.5201e-01, -5.3423e-02,\n",
       "          -5.2030e-01,  2.3772e-01,  1.1044e-01, -2.4995e-01,  2.8489e-01,\n",
       "          -2.1323e-01, -1.3440e+00,  4.1769e-01, -5.1067e-01, -1.0874e-01,\n",
       "          -1.4999e-01,  1.3837e-01,  2.1748e-01,  2.7935e-01, -3.8229e-02,\n",
       "           2.3746e-01,  6.3912e-02,  9.0981e-02,  4.8112e-01, -2.9771e-01,\n",
       "           4.8620e-01,  6.1948e-01,  4.4750e-01,  1.3375e-01, -8.5458e-02,\n",
       "          -4.0804e-01,  4.3960e-02,  5.1964e-01,  5.1491e-02,  2.3200e-01,\n",
       "           1.1652e-01, -3.0576e-02,  4.2867e-02, -3.4450e-04, -5.7195e-02,\n",
       "          -8.2024e-02,  5.2917e-02, -4.2911e-01,  1.5869e-01, -3.0869e-01,\n",
       "           6.5872e-01, -2.2781e-01,  1.2178e-01,  4.3070e-01,  3.2056e-01,\n",
       "          -3.1265e-01,  5.7879e-02,  2.1855e-01, -3.1844e-01, -1.5740e-01,\n",
       "           1.7974e-01, -1.1942e-01,  6.3240e-02, -2.5219e-01, -3.8078e-01,\n",
       "           2.1412e-01, -7.7649e-01,  5.4351e-01,  3.9384e-01, -2.3813e-01,\n",
       "           2.2883e-01,  7.8883e-02,  6.7749e-02, -2.9917e-01,  6.4534e-01,\n",
       "          -4.5187e-02, -3.7436e-01, -6.5600e-01,  2.6769e-01,  4.5929e-01,\n",
       "           2.6044e-01, -2.0244e-01,  4.3998e-01,  1.0077e+00,  3.9294e-01,\n",
       "          -1.2800e-01, -1.0068e-01,  3.9667e-01,  2.7371e-01, -1.3869e-01,\n",
       "           1.1542e-01, -1.2394e-03, -4.2192e-01,  2.0386e-01, -2.3549e-01,\n",
       "           3.8388e-02,  1.3436e-01,  2.3176e-01,  1.8170e-01, -6.2149e-01,\n",
       "          -2.6880e-01,  2.6981e-01,  2.7788e-02, -1.2023e-01,  2.1874e-01,\n",
       "          -4.3361e-01,  1.3683e-01, -4.2239e-02, -4.8579e-01, -4.3249e-01,\n",
       "          -5.9333e-01,  1.0596e-01, -6.9370e-02,  1.0710e-01, -8.4596e-01,\n",
       "          -2.3107e-01,  1.5895e-01,  1.9080e-01, -7.3362e-02,  1.6495e-03,\n",
       "           4.7469e-01,  1.6424e-03, -3.2406e-01, -1.7181e-01, -5.5972e-01,\n",
       "          -4.2510e-04, -8.6993e-03, -4.9937e-01,  3.1318e-01,  4.8266e-01,\n",
       "           2.0864e-01,  7.2338e-02,  1.2634e-01, -3.5677e-01, -2.9247e-01,\n",
       "          -6.6798e-02,  9.9149e-02, -2.4478e-01, -2.0565e-01,  4.8294e-01,\n",
       "          -5.1945e-02,  5.4654e-01,  4.0830e-01, -2.4307e-01, -3.4900e-01,\n",
       "          -1.2492e-01,  8.8614e-01, -1.4160e-01, -3.8257e-01, -4.7604e-02,\n",
       "          -8.8060e-01, -6.8497e-02,  1.2713e-01, -3.2929e-01, -4.2561e-01,\n",
       "           2.2956e-01,  1.4308e-01,  1.2843e-01, -1.7442e-01, -1.5852e-01,\n",
       "           2.9617e-01,  5.5989e-02,  7.4913e-02,  9.6663e-02, -3.8825e-01,\n",
       "          -3.7432e-01, -1.1160e-01, -2.9162e-01, -2.5878e-01, -2.2579e-01,\n",
       "           9.9708e-02,  3.3106e-01, -3.0041e-01, -2.2164e-01, -2.8097e-01,\n",
       "           5.7990e-01, -4.9004e-01, -9.5992e-02,  3.6044e-01, -1.1088e-01,\n",
       "           1.1428e-01,  2.7934e-01,  3.0352e-02, -1.0071e-01,  7.2974e-02,\n",
       "          -1.0785e-01, -1.2025e-01, -5.9826e-01, -7.2036e-02,  8.4290e-01,\n",
       "           3.8202e-01, -1.0483e-01, -2.9528e-01,  7.6021e-01,  5.4842e-01,\n",
       "          -3.2585e-01,  3.1097e-01, -4.6237e-01, -2.7704e-02,  5.0187e-02,\n",
       "           4.9144e-02,  3.6313e-01, -4.1027e-01,  2.2532e-01, -6.4469e-02,\n",
       "           3.3218e-01, -2.1826e-01, -3.0748e-01,  2.0243e-01,  3.0794e-01,\n",
       "           8.6260e-02, -5.4064e-01,  2.2432e-01, -2.0803e-01, -6.1036e-01,\n",
       "           7.1932e-02,  1.2862e-01,  1.1259e-01, -3.5435e-01, -4.1218e-01,\n",
       "          -2.4804e-02, -2.4546e-01,  6.5560e-01,  4.4862e-01, -3.3106e-01,\n",
       "           1.2477e-01, -1.1871e-01, -2.3242e-01,  2.6513e-02, -1.5140e-01,\n",
       "          -3.0969e-01,  2.9817e-01, -5.6540e-02,  2.4887e-02,  1.4554e-01,\n",
       "          -1.3003e-01, -2.6701e-01, -1.2804e-01,  2.5223e-01,  7.4374e-03,\n",
       "          -5.0053e-01,  7.6731e-02,  8.0046e-02, -4.0010e-01,  1.4470e-01,\n",
       "          -4.7883e-01,  5.1486e-01, -1.0019e-01,  2.0890e-01, -1.9388e-01,\n",
       "           1.5729e-01, -6.5654e-02,  3.2473e-03, -3.7791e-01, -5.3187e-01,\n",
       "           1.4683e-01,  3.3781e-01,  5.4420e-02,  3.4663e-01, -1.1452e+01,\n",
       "           3.3264e-01,  7.1303e-01, -4.5656e-01,  3.4650e-02, -4.6086e-02,\n",
       "          -6.0171e-02,  9.3568e-02,  4.1468e-01, -4.3236e-01, -1.0951e-01,\n",
       "          -4.9330e-01, -3.8692e-03,  7.8325e-02, -4.9033e-01, -2.8479e-01,\n",
       "           1.6226e-01, -7.7342e-01,  1.9402e-01,  4.4864e-02,  7.1306e-01,\n",
       "          -4.2247e-01,  3.7957e-02,  2.0496e-01,  1.7212e-03, -2.2060e-01,\n",
       "           3.1218e-01, -3.7371e-01,  2.1894e-01,  6.6627e-01,  3.4563e-01,\n",
       "           1.6280e-01, -6.6086e-02,  1.0281e-02, -2.1151e-01, -3.9323e-01,\n",
       "           3.1149e-02, -2.5334e-01,  2.1942e-01, -1.5381e-01, -5.4335e-01,\n",
       "          -3.1733e-01,  9.6942e-03,  9.8059e-02,  2.2875e-01,  8.5948e-01,\n",
       "           5.9198e-01,  8.4326e-01,  2.4368e-02,  2.0298e-01,  2.3055e-01,\n",
       "          -1.0217e+00,  3.1466e-01, -3.4589e-01, -3.6958e-01, -4.2980e-02,\n",
       "           6.0170e-01, -5.2458e-01,  3.2160e-01, -4.7946e-01, -5.7355e-02,\n",
       "           1.9954e-01,  3.5435e-01,  1.3418e-01, -3.3579e-01,  2.2173e-02,\n",
       "          -1.2233e-01, -2.4074e-01,  2.1308e-02, -2.5025e-01, -3.2662e-01,\n",
       "           1.9822e-01,  2.5117e-01,  9.9216e-02, -3.5924e-01,  2.0084e-01,\n",
       "           1.7458e-01, -1.7840e-01,  1.6051e+00, -2.8908e-02,  1.3880e-01,\n",
       "          -3.7285e-01,  3.4582e-01, -1.6244e-01, -3.7832e-01, -2.8879e-01,\n",
       "          -5.8067e-02,  2.1590e-01,  1.9165e-01,  2.5980e-01, -2.2919e-01,\n",
       "          -8.1663e-01, -1.3447e-02,  1.5095e-01,  2.2529e-02,  5.9888e-01,\n",
       "          -9.7709e-01, -5.2015e-01, -2.8585e-02,  4.6551e-02, -6.6477e-01,\n",
       "          -2.0331e-02, -3.6428e-01, -4.5137e-01,  4.2283e-01, -1.5141e-01,\n",
       "           4.3807e-02,  1.5978e-01, -3.9194e-01,  4.1381e-01, -2.0019e-01,\n",
       "           6.9091e-02,  4.0127e-01, -2.3541e-01, -2.3667e-01, -2.7041e-01,\n",
       "           3.4175e-01,  3.1803e-01, -2.9001e-01, -3.1588e-01, -1.0783e-01,\n",
       "          -1.1220e-02,  1.4895e-01,  2.8598e-01, -2.4347e-01,  1.2915e-01,\n",
       "          -6.5541e-01,  1.9826e-01,  1.7410e-01,  9.7079e-02,  1.8030e-02,\n",
       "           2.1805e-01, -1.9102e-01,  3.3727e-01, -6.8711e-01, -5.5677e-01,\n",
       "          -7.6166e-01,  6.7296e-02,  1.8449e-01, -4.0965e-01, -1.8972e-01,\n",
       "          -3.4631e-02, -5.9509e-02, -9.9908e-02, -5.0347e-02, -3.8204e-02,\n",
       "           6.6269e-02,  1.9679e-01, -2.4339e-01,  1.1468e-01, -9.2751e-02,\n",
       "          -4.3834e-01, -2.6637e-02, -4.3017e-01,  3.4911e-02, -7.1296e-01,\n",
       "          -1.5614e-01,  1.5205e-01, -3.1634e-01, -4.6368e-01, -9.0703e-02,\n",
       "           8.8195e-02,  4.5774e-03, -3.2571e-01,  7.8373e-01,  2.4774e-01,\n",
       "           3.2254e-01, -6.1778e-01, -4.9940e-02, -8.7095e-02,  2.7779e-01,\n",
       "           2.1450e-01,  3.5564e-01, -2.7069e-01, -6.1656e-01, -3.4760e-01,\n",
       "          -3.1564e-02,  1.0647e+00, -1.6482e-01, -8.6384e-02,  1.6291e-01,\n",
       "           5.7321e-01, -5.1239e-02, -1.8714e-01,  4.6004e-02, -2.5052e-01,\n",
       "          -2.4016e-01, -6.5861e-01,  5.2069e-01, -9.4601e-02,  1.6365e-01,\n",
       "           2.0746e-01,  8.0333e-02, -4.3395e-01,  1.3653e-01, -1.4862e-01,\n",
       "          -2.6067e-01, -8.7276e-03, -5.2681e-01,  2.0789e-01,  3.0375e-01,\n",
       "          -3.1365e-01,  6.2910e-01, -5.0593e-03, -1.5613e-01, -2.1493e-01,\n",
       "           3.5637e-01,  1.5407e+00, -2.5025e-01, -5.7082e-02,  8.1650e-01,\n",
       "           3.2050e-01, -3.4545e-02,  1.8150e-01, -2.0241e-01,  1.2894e-01,\n",
       "           2.0990e-01, -2.4531e-02, -1.4187e-01,  8.3439e-02,  2.1700e-01,\n",
       "           3.5118e-01, -5.7450e-01, -4.8490e-01, -6.5003e-02,  7.6167e-01,\n",
       "          -1.0032e-01,  4.1886e-01, -2.8729e-02,  5.4428e-01,  2.8883e-01,\n",
       "          -1.0013e+00, -1.9527e-01, -7.2901e-01,  2.4713e-01,  2.4550e-01,\n",
       "          -3.1615e-01, -1.2261e-01, -1.1275e-01, -4.5085e-01,  7.4131e-01,\n",
       "           3.6743e-01, -9.4643e-01,  1.3686e-01,  7.7621e-01, -6.3941e-01,\n",
       "           2.0834e-01, -1.3773e-01,  1.0551e-01,  1.9138e-01,  1.6381e-01,\n",
       "          -1.7485e-02,  1.5586e-01, -7.5839e-01, -6.6616e-01,  2.3659e-01,\n",
       "           6.1048e-01, -1.5588e-02,  2.2563e-01, -9.1773e-02, -2.7541e-01,\n",
       "          -1.5944e-02, -2.8195e-01, -2.7864e-01,  1.4402e-01, -4.6706e-02,\n",
       "           7.1363e-02, -2.8455e-01,  6.5256e-04,  3.5724e-01, -5.2783e-02,\n",
       "           3.1628e-01,  2.2761e-01,  3.9839e-01, -1.3871e-01, -1.6273e-01,\n",
       "          -1.6455e-01,  3.8519e-01, -1.9922e-01,  4.5993e-01, -4.6767e-02,\n",
       "          -2.1463e-01,  7.6549e-02,  5.5340e-03, -3.8076e-01,  1.0963e-01,\n",
       "          -4.6892e-01,  1.2923e-01,  4.2717e-02, -3.9521e-01,  3.0516e-01,\n",
       "           1.8785e-01,  1.8347e-01, -1.0060e-01,  1.8034e-01,  2.3171e-01,\n",
       "          -1.7351e-02,  9.1230e-01,  1.2952e-02,  1.8612e-01, -7.7900e-01,\n",
       "           1.3868e-02, -7.3447e-01,  1.4680e-02,  7.7392e-02, -2.4240e-01,\n",
       "          -4.4166e-01, -1.9410e-01,  3.6350e-01, -4.7865e-03,  7.0288e-01,\n",
       "           7.3995e-01, -3.9561e-01, -5.5688e-01, -3.6418e-01,  2.6278e-01,\n",
       "           2.2536e-01, -1.3132e-01,  5.1888e-01,  2.8392e-01,  3.8333e-01,\n",
       "          -6.9680e-01, -2.0138e-01, -3.0282e-01,  3.8130e-01, -1.3952e-01,\n",
       "           7.1014e-01,  1.4858e-01,  3.9051e-02,  1.6615e-01, -6.8612e-03,\n",
       "           9.2560e-02,  5.5296e-01, -1.3564e-01, -7.6109e-01,  1.1630e-02,\n",
       "           1.1547e-02, -1.6926e-01,  1.7780e-01, -4.1095e-01,  3.3747e-01,\n",
       "           4.7554e-01,  2.7973e-01,  9.1879e-01]], device='mps:0'),\n",
       " tensor([[-3.9397e-01, -5.8754e-01, -5.9991e-01, -1.0209e+00, -2.2381e-01,\n",
       "          -5.4571e-01, -6.3667e-01,  1.4316e-01, -6.7069e-02,  4.2305e-01,\n",
       "           2.5500e-01,  1.1010e-01,  1.2499e-01, -3.4692e-01, -8.4136e-01,\n",
       "          -3.0231e-01, -5.5806e-02,  5.8592e-01,  4.6595e-01,  1.9724e-01,\n",
       "           2.7358e-01,  4.5363e-01,  4.9668e-02, -2.3262e-01, -4.0742e-01,\n",
       "           8.4079e-02, -9.7280e-02, -2.9584e-01,  3.0343e-01, -5.3450e-01,\n",
       "          -3.2588e-02,  5.8770e-01, -1.5001e-01,  8.5157e-01, -1.8297e-01,\n",
       "          -2.4053e-01,  1.5278e-01,  7.8407e-02,  1.3225e-02, -5.4663e-01,\n",
       "           3.2274e-01,  3.6458e-01,  2.4693e-01,  2.5213e-01,  8.9671e-02,\n",
       "           2.0399e-02,  1.6903e-01, -3.2989e-01, -8.4320e-02, -3.4846e-01,\n",
       "          -4.5784e-01, -3.0563e-01,  2.3854e-01,  1.3667e-01, -2.4936e-01,\n",
       "          -6.2630e-01,  8.1114e-02,  2.3255e-01, -2.9494e-01,  2.2963e-01,\n",
       "          -1.4217e-01,  5.7903e-01, -1.9241e-01,  1.6640e-04,  6.6146e-01,\n",
       "           4.2239e-02, -5.2901e-01, -4.4167e-01, -6.7224e-01, -9.5478e-02,\n",
       "          -7.6849e-02,  3.7466e-01, -8.0453e-02, -5.3082e-01,  3.6840e-01,\n",
       "          -3.9636e-01,  1.7244e-01, -7.1807e-02, -4.6623e-01,  8.7896e-02,\n",
       "          -8.2733e-03, -2.5271e-02,  1.4184e-01,  1.4437e-02, -4.0503e-03,\n",
       "          -3.2594e-01,  4.7511e-01, -7.8056e-02,  9.9280e-03,  4.4952e-02,\n",
       "           4.8101e-01, -1.8154e-01,  2.6985e-01, -1.0899e-01,  4.3475e-01,\n",
       "          -1.9624e-01,  3.8182e-01, -3.0496e-01,  8.4196e-02, -5.7506e-03,\n",
       "           2.2105e-01,  2.0789e-01,  7.8474e-02,  2.8903e-01, -3.3472e-01,\n",
       "          -1.1899e-01, -5.4698e-02,  9.4588e-02, -3.4137e-01, -1.3079e-01,\n",
       "          -1.4821e-01, -1.1151e-01,  3.9656e-01, -7.8772e-01, -1.5523e-01,\n",
       "           1.8061e-01, -5.4146e-02, -1.7820e-02, -6.2665e-02,  8.0945e-01,\n",
       "           5.1137e-01, -1.2890e-01,  3.9475e-01,  2.4560e-01,  5.3352e-02,\n",
       "          -5.2399e-02,  6.7089e-01, -1.5675e-01, -2.4677e-02, -4.1786e-02,\n",
       "           6.7251e-02, -3.2128e-02, -4.0666e-01,  3.1175e-01,  1.1769e-01,\n",
       "           5.5493e-02, -3.1016e-01, -4.7178e-01,  2.6316e-02, -1.1430e-01,\n",
       "          -1.9792e-01,  5.5865e-02, -6.0096e-01, -2.1372e-01,  3.4587e-01,\n",
       "          -1.9636e-01, -2.9922e-01, -2.8082e-01, -5.2154e-02, -5.3538e-01,\n",
       "          -3.8346e-01, -4.2304e-01,  4.1160e-01, -9.1325e-02, -1.4490e-02,\n",
       "          -1.5862e-01,  1.6883e-01, -1.8114e-01,  4.3102e-02,  1.4746e-01,\n",
       "           1.0005e+00, -1.8746e-01,  6.3623e-02, -4.8477e-01, -1.0066e-01,\n",
       "           1.9676e-01,  1.5027e-01, -5.8017e-01, -9.8609e-02,  1.8951e-01,\n",
       "           4.1085e-01,  1.2291e-01,  7.6470e-02, -7.3612e-01, -2.1366e-01,\n",
       "          -6.3271e-01,  2.7304e-01,  1.4748e-01,  6.2648e-02,  6.8649e-02,\n",
       "           1.8561e-01, -1.5106e+00,  1.6431e-01, -1.1563e-01, -1.5884e-01,\n",
       "          -1.4299e-01, -1.5129e-01,  1.6056e-01,  3.3077e-01,  7.1602e-02,\n",
       "           4.9264e-01,  4.2774e-02,  3.4327e-01,  2.0058e-01, -9.6204e-02,\n",
       "           2.5018e-01,  5.2594e-01, -1.7884e-01, -1.1478e-01,  9.3591e-02,\n",
       "          -8.1371e-01,  5.9829e-01,  2.4647e-01, -1.6184e-01, -2.6339e-01,\n",
       "           9.5700e-02, -5.9060e-02,  1.9811e-01, -3.7550e-01, -2.8787e-01,\n",
       "           9.0026e-02, -2.1901e-01, -4.2776e-01, -3.2280e-01, -4.6982e-01,\n",
       "           3.0142e-01,  5.5336e-02, -3.8533e-02,  6.0511e-01,  3.0958e-01,\n",
       "          -3.5776e-02, -1.7143e-01, -3.0816e-01, -1.4957e-01, -4.7905e-01,\n",
       "           4.4108e-01, -3.5459e-01, -2.0781e-01, -3.0759e-01, -2.6653e-01,\n",
       "          -2.8305e-01, -6.7437e-01,  6.0670e-01,  6.1556e-01, -7.8038e-02,\n",
       "          -2.7803e-01, -1.8543e-01, -1.5926e-01, -3.4601e-01,  9.8443e-02,\n",
       "          -3.4757e-01, -6.8495e-01, -9.4679e-01,  1.5514e-01,  2.9009e-01,\n",
       "           1.6537e-01,  7.0723e-02,  8.5321e-01,  2.1733e-01,  3.3220e-01,\n",
       "          -1.2598e-01, -1.8712e-01,  4.8793e-01,  2.0452e-01,  6.3766e-03,\n",
       "           1.6965e-01,  6.1583e-02, -2.2119e-01,  6.4229e-01, -2.8977e-01,\n",
       "          -1.4015e-01, -8.9861e-02,  3.6417e-01,  5.2150e-01, -5.5715e-01,\n",
       "           1.0281e-01,  5.0592e-01, -1.0955e-01, -2.4569e-01,  4.8861e-01,\n",
       "          -6.9963e-01,  1.6861e-01, -9.2736e-02, -5.7621e-01, -7.1155e-01,\n",
       "          -5.4468e-01, -2.4693e-01,  2.8281e-01,  2.4789e-01, -5.3505e-01,\n",
       "          -8.0113e-01, -4.1626e-01,  2.1271e-01,  2.5913e-01, -1.5838e-01,\n",
       "           1.3565e-01, -1.8049e-01, -3.5232e-05, -3.9343e-01, -5.5641e-01,\n",
       "          -4.1142e-01,  4.8638e-01, -3.2481e-03,  4.0689e-01,  2.3181e-01,\n",
       "          -4.8118e-01,  3.9941e-01,  6.2593e-02, -3.0571e-01, -1.7305e-01,\n",
       "          -9.9898e-02, -6.3966e-01,  3.0362e-01, -4.2328e-01, -2.5860e-01,\n",
       "           4.3865e-01,  1.6314e-01,  8.6761e-01,  2.2151e-01, -4.1072e-01,\n",
       "           8.1011e-02,  6.6462e-01, -8.8975e-02,  6.5978e-02, -1.7952e-01,\n",
       "          -1.0381e+00, -2.2985e-03,  4.2575e-01, -4.5715e-01, -2.2831e-01,\n",
       "           9.3804e-02,  2.9542e-01, -3.1978e-01, -9.9867e-02,  2.6597e-02,\n",
       "           5.3852e-02,  5.9216e-01, -1.6561e-02,  2.3636e-01,  5.4684e-02,\n",
       "           3.6796e-02, -3.0057e-01, -9.3449e-02, -1.7464e-01,  9.9690e-02,\n",
       "           3.1742e-01,  3.8880e-01, -7.3808e-02, -2.7145e-01, -2.6520e-01,\n",
       "           6.2614e-01, -3.2440e-01,  1.2341e-02,  2.2455e-01, -5.8581e-01,\n",
       "          -1.4410e-02,  1.1270e-01, -2.2233e-01,  9.0239e-02,  4.5411e-03,\n",
       "           3.1665e-01, -4.3095e-01, -3.4773e-01, -1.4808e-01,  4.7653e-01,\n",
       "           3.4301e-01,  1.9878e-01, -4.2432e-02, -1.7934e-01,  3.8365e-01,\n",
       "          -4.2077e-01,  3.9937e-01, -9.1985e-02, -1.0747e-01, -1.3523e-01,\n",
       "           1.9581e-01,  1.5733e-01, -4.0589e-01,  3.8866e-01,  1.5324e-01,\n",
       "          -1.7556e-01, -6.3346e-01, -1.4938e-01,  2.4216e-01,  1.9864e-01,\n",
       "          -1.4221e-01, -2.9883e-01,  4.7038e-02, -2.5441e-01,  2.0366e-01,\n",
       "           4.9525e-02,  1.6182e-01,  3.7674e-02, -4.3137e-01, -2.8021e-01,\n",
       "          -1.9835e-01,  7.0996e-02,  4.3602e-01,  6.8023e-01, -2.0368e-02,\n",
       "           2.0340e-01, -1.4691e-01, -3.0174e-01,  3.4229e-01,  1.7431e-01,\n",
       "          -2.2504e-01,  4.0925e-01,  1.2129e-01, -7.8254e-02,  1.3691e-01,\n",
       "          -1.4993e-01, -1.6211e-02,  5.1528e-01,  1.1942e-01, -3.4513e-01,\n",
       "          -4.4716e-01,  1.0144e-01,  8.8505e-02, -4.3818e-01,  3.1505e-01,\n",
       "           9.5420e-02,  2.7531e-01, -2.7314e-01, -1.6410e-02,  1.0115e-01,\n",
       "           7.5193e-03,  4.1968e-02, -1.4619e-01, -1.5567e-01, -4.0147e-01,\n",
       "           7.5740e-01,  3.8378e-02,  4.6443e-01,  3.2515e-01, -1.1445e+01,\n",
       "          -6.8750e-02,  8.0976e-01, -5.1263e-01,  1.7799e-01,  1.1749e-01,\n",
       "          -2.1766e-02, -4.6545e-01,  4.5397e-01, -4.7748e-01, -4.8139e-01,\n",
       "          -3.4229e-01, -1.5462e-01,  1.6770e-01, -5.9584e-01,  1.0455e-01,\n",
       "           1.0663e-01, -6.6095e-01, -3.8485e-02, -2.1366e-01,  5.2334e-01,\n",
       "           2.1822e-01, -4.5051e-02,  2.7393e-01,  1.4896e-01, -1.1201e-01,\n",
       "          -9.0058e-02, -1.2686e-01,  8.8981e-02,  3.2861e-01, -2.5550e-01,\n",
       "           4.9136e-02,  2.3928e-01,  5.0325e-02, -2.6798e-01, -6.7340e-01,\n",
       "          -2.1371e-02, -2.0758e-01,  1.7928e-01,  3.2153e-01, -9.8842e-02,\n",
       "           2.0603e-01, -4.0346e-01,  3.0606e-01,  3.4622e-01,  5.7354e-01,\n",
       "           1.1494e+00,  9.3442e-01,  3.3771e-01,  5.4582e-02, -1.4097e-01,\n",
       "          -5.2019e-01,  7.2429e-02, -1.8160e-01, -1.5299e-02, -2.5842e-01,\n",
       "           3.6047e-01, -5.7707e-01, -5.8438e-02, -2.7191e-01,  4.4450e-02,\n",
       "           4.6311e-01,  6.6502e-02,  3.2627e-01, -4.1991e-01,  1.9460e-01,\n",
       "           5.5903e-01,  3.0840e-01, -5.2072e-02, -1.1001e-01, -2.4986e-01,\n",
       "          -3.6967e-01,  5.3333e-01,  3.7653e-01, -1.7773e-01,  3.0359e-01,\n",
       "          -2.1684e-01, -1.2044e-01,  9.3760e-01, -2.3563e-01, -1.2147e-03,\n",
       "          -5.9003e-02,  6.0994e-02, -6.2274e-01, -4.8742e-01,  6.3268e-01,\n",
       "           1.0667e-01, -1.4816e-01,  3.7111e-01,  4.2599e-01, -3.4455e-02,\n",
       "          -4.6923e-01, -3.9377e-02,  5.5886e-02,  2.3746e-01,  9.6473e-01,\n",
       "          -2.1715e-01,  1.0804e-01,  4.3493e-02, -5.5779e-02, -7.0763e-01,\n",
       "          -1.3828e-01, -5.7291e-02, -2.1402e-01,  1.9307e-01,  1.2900e-01,\n",
       "          -1.7660e-02,  1.4703e-01,  1.2816e-01,  3.0486e-01, -4.0239e-01,\n",
       "           1.0895e-01,  1.7743e-01,  3.8580e-01, -2.6512e-01, -5.0749e-01,\n",
       "           2.8117e-01,  6.9438e-01, -2.7867e-01, -7.6340e-01, -6.9310e-01,\n",
       "          -7.3788e-02,  3.6170e-01,  3.1623e-02, -3.2547e-01,  4.2984e-01,\n",
       "          -8.2274e-01, -3.8586e-01, -1.0837e-01, -4.0446e-02,  4.2974e-01,\n",
       "           5.2723e-01,  3.4303e-01,  2.3735e-01, -6.3121e-01, -4.0922e-01,\n",
       "          -5.2601e-01, -2.2909e-02,  4.6535e-01, -1.2204e-01, -1.4371e-01,\n",
       "          -9.7379e-02, -1.7265e-01,  2.2648e-01,  2.6228e-01, -1.6687e-01,\n",
       "           2.4207e-01,  1.7419e-01,  2.0454e-01,  2.9944e-01, -9.8931e-02,\n",
       "          -2.2267e-01, -2.9219e-01,  9.2346e-02, -2.1581e-01, -6.0678e-01,\n",
       "          -4.3264e-01,  1.2105e-01,  2.7746e-01,  5.4919e-01,  4.1905e-01,\n",
       "           1.0736e-02, -8.4986e-02, -9.7186e-02,  4.0545e-01,  3.7267e-01,\n",
       "          -7.8875e-02, -5.6764e-01,  4.4652e-02, -1.5333e-02,  1.1400e-01,\n",
       "           3.1456e-01,  2.2887e-01, -4.9542e-01, -4.9041e-01, -1.1799e-01,\n",
       "           1.7159e-01,  1.2586e+00, -1.0523e-01, -5.0350e-01,  2.8381e-01,\n",
       "           6.0846e-02, -3.1417e-01,  2.6816e-02, -1.4901e-01,  3.3145e-01,\n",
       "          -3.9705e-01, -1.1834e-01,  2.4701e-01,  1.1823e-02,  5.1991e-02,\n",
       "           8.9940e-01,  1.6130e-01, -5.9953e-02,  4.0490e-01,  1.0172e-01,\n",
       "          -4.4280e-01,  1.5337e-01,  4.7192e-02, -3.8074e-02,  6.5233e-02,\n",
       "          -4.5520e-02,  9.5724e-02, -2.6755e-02,  1.3836e-01,  2.1174e-02,\n",
       "           2.4398e-01,  1.3382e+00, -2.7362e-01, -5.1204e-01,  5.2613e-01,\n",
       "          -1.9888e-01, -2.5798e-02,  1.5342e-01,  2.4630e-02,  1.9425e-01,\n",
       "           3.9728e-01,  3.3086e-01, -6.6419e-01, -1.2310e-01,  4.2633e-01,\n",
       "           3.0917e-01, -6.6187e-01, -5.5753e-01, -8.7010e-02,  2.3525e-01,\n",
       "          -3.7869e-01,  3.2610e-01, -1.1042e-01,  5.7094e-01,  8.9085e-02,\n",
       "          -9.7340e-01,  1.3484e-01, -3.6426e-01,  6.3413e-02, -1.2397e-01,\n",
       "          -6.6789e-01, -1.8784e-01,  1.8560e-01, -1.8658e-01,  1.7858e-01,\n",
       "           4.6074e-01, -7.3904e-01,  8.3079e-02,  3.9996e-01, -9.8751e-02,\n",
       "           3.3658e-01,  8.4697e-02, -1.3720e-01,  1.6087e-01, -1.2859e-01,\n",
       "          -7.0053e-02,  8.2939e-02, -5.1675e-01, -3.5989e-01,  1.4397e-01,\n",
       "           4.4453e-01,  1.0210e-01,  1.3304e-01, -1.7109e-01, -8.5205e-02,\n",
       "          -4.4336e-01, -1.7980e-01, -3.3560e-01,  7.3131e-02, -4.8724e-02,\n",
       "           1.3312e-01,  4.5701e-02, -1.7737e-02,  2.9470e-01, -1.8890e-01,\n",
       "           2.6184e-01,  5.4665e-01,  7.4225e-01,  9.4411e-02, -3.0112e-01,\n",
       "           3.5951e-02,  4.8005e-01, -2.0072e-01,  5.7703e-01, -1.6700e-01,\n",
       "          -4.3428e-01,  1.5619e-01,  2.1772e-01, -2.1557e-01, -9.6990e-02,\n",
       "          -2.4256e-01,  3.1268e-01, -2.7071e-01, -1.1641e-01,  4.4281e-01,\n",
       "          -2.4893e-01,  1.1798e-01, -1.0669e+00,  3.4089e-01,  4.1410e-01,\n",
       "          -1.2461e-01,  6.3808e-01,  1.7492e-01,  7.9649e-01, -1.5367e-01,\n",
       "           4.3770e-01, -5.3451e-01, -2.4963e-01, -1.9112e-01, -9.9515e-02,\n",
       "          -4.4446e-01, -2.5209e-01,  1.7343e-01,  3.3069e-01,  6.2923e-01,\n",
       "           2.9326e-01, -1.3923e-01, -4.5999e-01, -8.5201e-05,  1.6167e-01,\n",
       "           5.8151e-02, -3.2017e-01,  5.3471e-01, -2.5290e-01,  3.6736e-02,\n",
       "          -6.8753e-01, -2.9496e-01, -1.3839e-01,  1.2597e-01, -3.7557e-02,\n",
       "           5.2499e-01, -3.9799e-01,  4.3656e-01,  4.8355e-01,  2.1227e-01,\n",
       "           3.9756e-01,  5.3926e-01, -1.7333e-01, -5.8337e-01, -2.9268e-01,\n",
       "           2.1503e-01, -2.7369e-01,  2.6546e-01, -1.0342e+00,  4.9280e-01,\n",
       "           5.7834e-02,  6.7945e-01,  3.7643e-01]], device='mps:0'),\n",
       " tensor([[-0.0111, -0.5057, -0.4670,  ...,  0.2365, -0.0712,  0.4388],\n",
       "         [-0.3292, -0.2590, -0.5016,  ..., -0.0148,  0.3509,  0.1607]],\n",
       "        device='mps:0'),\n",
       " tensor([[-0.0279, -0.7322, -0.5137,  ...,  0.0454,  0.9329,  0.5506],\n",
       "         [-0.3024, -0.3250, -0.4098,  ...,  0.1233,  1.0168,  0.9023]],\n",
       "        device='mps:0'),\n",
       " tensor([[-0.0153, -0.4994, -0.5617,  ...,  0.2217,  0.2081,  0.4534],\n",
       "         [-0.2528, -0.5673, -0.4982,  ..., -0.1000,  0.8731,  0.3391]],\n",
       "        device='mps:0'),\n",
       " tensor([[-0.0800, -0.7033, -0.3539,  ...,  0.2254,  0.8639,  0.6267],\n",
       "         [-0.2234, -0.3239, -0.6407,  ...,  0.4012,  0.7267,  0.4836]],\n",
       "        device='mps:0'),\n",
       " tensor([[-0.0153, -0.4994, -0.5617,  ...,  0.2217,  0.2081,  0.4534],\n",
       "         [-0.2249, -0.6726, -0.4102,  ...,  0.1923,  0.3393,  0.4531],\n",
       "         [-0.3472, -0.0364, -0.8333,  ...,  0.0891,  0.2595,  0.5087],\n",
       "         [-0.3555, -0.7435, -0.5910,  ...,  0.2763,  1.3491,  0.2523]],\n",
       "        device='mps:0'),\n",
       " tensor([[-6.1750e-01, -4.2829e-01, -2.3934e-01, -9.4116e-01, -3.4548e-01,\n",
       "          -6.5496e-02, -6.3874e-01,  9.7828e-01, -9.5892e-02,  4.0096e-01,\n",
       "           1.4919e-01, -5.1603e-03,  2.0980e-01,  1.3680e-01, -6.9506e-01,\n",
       "           1.6740e-01,  4.6094e-01,  2.1159e-01,  3.5951e-01, -4.4329e-02,\n",
       "           7.9832e-01,  3.4245e-01,  1.8019e-01, -2.5205e-01, -4.5876e-01,\n",
       "           5.8994e-01, -2.4628e-01, -9.2014e-02,  2.0509e-01, -8.1949e-01,\n",
       "          -1.3586e-01, -4.9757e-02, -3.6281e-01,  8.7602e-01,  1.1948e-01,\n",
       "          -2.7310e-01, -1.5276e-01, -2.7996e-01,  2.3494e-01, -6.3637e-02,\n",
       "           3.7814e-01,  2.6592e-02, -2.9347e-01,  3.1386e-01, -1.6421e-01,\n",
       "          -1.4057e-01,  1.4047e-02, -7.8429e-01,  1.3666e-02,  1.8955e-01,\n",
       "           5.8851e-02, -4.7205e-02,  1.9231e-01,  3.7030e-01, -1.0617e+00,\n",
       "          -2.6348e-01,  2.7603e-01,  2.9849e-01, -1.7932e-01, -5.7784e-02,\n",
       "          -1.9989e-01,  9.0201e-01,  1.7949e-01, -2.7822e-01,  1.7824e-01,\n",
       "          -4.4899e-01, -5.4601e-01, -6.0781e-01, -1.0682e-01, -1.1967e-01,\n",
       "          -1.0002e-01,  9.7528e-02,  1.4009e-01, -2.9385e-01,  2.3170e-01,\n",
       "           2.1813e-01,  4.7024e-02, -2.7652e-02, -4.3158e-01, -2.6546e-01,\n",
       "           3.6947e-01,  3.4425e-02, -1.4730e-02,  2.0326e-01, -5.8744e-01,\n",
       "          -2.0226e-01,  3.7959e-01,  1.5703e-01,  2.5914e-01,  2.2662e-03,\n",
       "           5.9967e-01,  1.2024e-02,  8.0339e-02, -4.5369e-01,  5.8427e-01,\n",
       "          -5.4955e-01,  1.0329e-01, -1.6007e-01,  4.9532e-02, -5.2088e-01,\n",
       "          -9.8735e-02,  2.4494e-01, -1.8085e-01,  1.6053e-01, -4.6592e-01,\n",
       "          -3.2983e-02,  2.4514e-01,  3.0468e-01, -4.9183e-01,  4.2685e-02,\n",
       "          -6.5134e-03, -2.9791e-01,  7.3694e-02, -6.6407e-01, -8.5795e-01,\n",
       "           5.8354e-01,  1.8428e-02, -4.5240e-01, -2.3930e-01,  7.4443e-01,\n",
       "           2.9738e-01, -2.3201e-01, -5.3072e-02, -3.5548e-01,  2.0259e-01,\n",
       "          -1.8518e-01,  2.5689e-01, -2.7018e-01,  3.8439e-01,  5.5026e-01,\n",
       "           3.1581e-01,  2.8489e-01, -1.8327e-01,  2.7637e-01,  1.4465e-01,\n",
       "           3.6947e-02, -6.6614e-02, -1.5929e-01,  1.5180e-01,  1.2041e-01,\n",
       "          -2.9580e-01, -3.2999e-01,  5.4746e-03,  1.1182e-01, -3.3279e-01,\n",
       "          -1.0202e-01,  1.8680e-01,  3.4808e-01, -1.4092e-01, -2.5403e-01,\n",
       "          -7.8984e-01, -4.2328e-01,  1.0217e-01, -1.8560e-01, -1.7135e-01,\n",
       "          -4.1911e-01,  3.8121e-01, -6.6660e-02,  4.7638e-01,  3.4189e-01,\n",
       "           6.9978e-01, -1.5311e-01,  5.9776e-02, -5.6412e-01,  3.5197e-01,\n",
       "           3.8270e-02, -6.1110e-01, -1.7247e-01,  1.5680e-01, -4.1813e-04,\n",
       "           6.8195e-01,  7.0979e-02,  4.0124e-01,  1.2070e-01, -3.2553e-01,\n",
       "           8.2089e-03,  1.9890e-01, -1.2864e-01, -1.4297e-01,  7.3766e-01,\n",
       "          -5.5188e-01, -2.3029e+00,  1.0721e-01, -3.5844e-01, -3.6529e-01,\n",
       "          -1.8135e-01,  3.7255e-01,  3.3772e-02,  4.9966e-01,  1.5486e-01,\n",
       "           3.7598e-02,  1.6729e-01, -2.8025e-02,  3.6851e-01, -8.4399e-02,\n",
       "           2.6751e-01,  3.3341e-01,  1.2988e-01, -2.3298e-01, -1.0956e-01,\n",
       "          -3.9227e-01,  2.4031e-01, -9.8916e-02,  8.6010e-01,  2.3260e-01,\n",
       "           2.0943e-01, -4.3657e-03, -4.7262e-02,  4.8964e-01,  4.3099e-01,\n",
       "          -6.6541e-02, -1.6585e-01, -4.9522e-01,  4.6269e-03, -4.1279e-02,\n",
       "          -1.2963e-02, -4.5611e-01,  1.3753e-01,  4.2841e-01, -1.2416e-01,\n",
       "          -1.8354e-01,  1.5525e-01, -4.7715e-02, -3.1517e-01, -2.6492e-02,\n",
       "           6.8453e-01,  1.9080e-01, -2.2994e-01,  1.2377e-01, -1.1896e-01,\n",
       "           3.7006e-02, -4.6975e-01,  7.0044e-01,  6.1147e-01,  1.5585e-01,\n",
       "           1.6214e-02, -6.9943e-01,  2.3699e-01, -3.1574e-01, -4.7647e-03,\n",
       "          -1.5027e-01, -1.5731e-01, -4.2986e-01, -5.7285e-02, -5.5854e-02,\n",
       "           4.0115e-01,  7.9943e-02,  4.6653e-01,  1.1837e+00,  2.7973e-01,\n",
       "           1.9184e-02,  5.6097e-01,  2.0801e-01,  4.2492e-01, -4.0180e-01,\n",
       "           3.3082e-01,  8.8825e-02, -6.3896e-01,  2.7188e-01, -4.1593e-01,\n",
       "           4.4963e-02, -4.1432e-02, -3.7926e-01,  1.4208e-01, -2.1276e-02,\n",
       "           3.4237e-01,  3.3517e-01,  1.5186e-01,  1.6838e-01,  2.9411e-01,\n",
       "          -3.1322e-01,  5.9715e-01, -6.0809e-02, -1.1127e-01, -6.3492e-01,\n",
       "          -7.6732e-01,  9.5206e-02,  4.4743e-01, -1.2880e-01, -6.1279e-01,\n",
       "           1.0462e-01, -2.3866e-01, -1.2257e-01, -3.5064e-02,  2.1191e-01,\n",
       "           2.7455e-01,  5.0687e-02, -9.5436e-02,  6.4791e-02, -4.6950e-01,\n",
       "          -4.9277e-01,  4.9140e-02, -1.5758e-01, -1.3063e-01,  5.3918e-01,\n",
       "          -7.9415e-02,  1.1575e-01,  5.7221e-01,  1.9799e-02, -4.5976e-01,\n",
       "          -2.5640e-01,  1.4802e-01,  2.8667e-01, -7.2082e-02,  2.1981e-01,\n",
       "           3.4119e-01,  2.6467e-02,  5.1649e-01,  1.4823e-01, -2.1046e-01,\n",
       "          -2.7239e-01,  5.1173e-01, -1.5755e-01,  6.1403e-02, -1.2527e-01,\n",
       "          -8.7207e-01, -5.6150e-02, -7.8413e-02, -1.6833e-01, -5.4440e-01,\n",
       "           2.6566e-01,  1.4351e-01,  1.0561e-01,  2.2350e-01, -1.5984e-01,\n",
       "           7.0123e-01,  6.5285e-01, -3.3100e-01, -1.6943e-01, -4.0104e-01,\n",
       "           9.3922e-02, -6.8374e-02, -2.9329e-01, -1.4009e-02,  9.5368e-03,\n",
       "          -2.2765e-01, -1.2516e-01, -1.0805e-01,  2.8921e-01, -1.8876e-01,\n",
       "           5.2862e-01,  1.2189e-01,  8.3649e-02,  4.3873e-01,  2.0612e-02,\n",
       "           1.3071e-01,  3.6515e-01, -4.2117e-01,  1.2870e-01,  4.0424e-01,\n",
       "           4.0578e-01, -1.0551e+00, -8.3291e-01, -1.4936e-01,  3.6111e-01,\n",
       "           2.2860e-01,  2.5189e-01,  1.6999e-02, -7.2215e-03,  8.9455e-02,\n",
       "          -2.3402e-01,  3.5135e-01, -6.6415e-02, -5.7412e-01,  3.0561e-01,\n",
       "           2.3567e-01,  5.3263e-01, -2.7377e-01, -5.0579e-01, -6.5233e-02,\n",
       "           2.4616e-01, -3.6381e-01,  1.1392e-01,  2.3480e-01,  4.5466e-01,\n",
       "          -3.6015e-01, -5.6164e-01, -7.0282e-02, -7.3986e-01, -1.3578e-01,\n",
       "           3.2483e-01, -2.1437e-01,  2.7178e-02, -2.4758e-01,  1.5231e-01,\n",
       "          -1.7925e-02, -1.8879e-01,  6.0757e-01,  2.0522e-01, -2.5855e-01,\n",
       "           1.1981e-01, -2.3892e-01, -1.9475e-01,  5.0091e-01, -5.0226e-01,\n",
       "          -3.0348e-01, -5.0960e-02, -1.5349e-01, -3.2418e-01,  3.2181e-01,\n",
       "          -4.1389e-01, -1.4861e-01,  1.6748e-01, -6.4736e-02, -1.3068e-01,\n",
       "          -3.5233e-01,  1.4268e-01, -1.2707e-01, -4.9883e-02, -1.7654e-02,\n",
       "          -1.3353e-01,  6.0900e-02,  1.3308e-01,  6.4538e-02,  1.4388e-01,\n",
       "           5.3696e-01, -1.1464e-01,  2.7582e-03, -2.5900e-01, -2.2469e-01,\n",
       "           5.2498e-01, -9.4973e-02, -3.3117e-01,  5.4928e-01, -1.1156e+01,\n",
       "           1.0386e-01,  7.2631e-01, -5.5808e-01,  5.5128e-02, -2.8539e-01,\n",
       "           2.1085e-01, -3.0009e-01,  7.9101e-02,  2.4967e-01, -2.8460e-01,\n",
       "          -1.2277e-01, -2.7364e-01, -2.3193e-01, -4.0872e-01, -3.9742e-01,\n",
       "          -4.7229e-01, -5.6789e-01,  1.1772e-01, -1.7663e-01, -1.0996e-01,\n",
       "          -5.7036e-01, -1.9379e-01,  2.9696e-01, -4.9559e-03, -5.1069e-01,\n",
       "           5.9541e-02, -2.2948e-01,  2.6399e-01,  4.6126e-02,  2.2045e-01,\n",
       "          -1.9949e-02,  1.4108e-01,  2.0669e-01, -4.4946e-02, -4.5100e-01,\n",
       "          -4.4853e-01,  2.2129e-01, -1.5955e-01, -2.2267e-01, -5.4017e-02,\n",
       "          -3.3838e-01, -2.7689e-01,  1.3447e-01,  3.7594e-01,  6.2596e-01,\n",
       "           1.1595e+00,  5.6208e-01,  2.9245e-01, -1.0625e-01, -8.5438e-02,\n",
       "          -1.0762e+00,  5.5900e-01, -3.7318e-01,  8.1787e-02, -2.5119e-01,\n",
       "           6.6201e-01, -4.9906e-01,  5.4685e-01, -4.0754e-01, -2.5500e-01,\n",
       "           3.7179e-01,  3.0194e-01,  7.1104e-03,  2.8239e-02, -2.7432e-01,\n",
       "           4.0257e-01, -2.6094e-01, -3.6915e-02, -3.7819e-01,  4.0081e-01,\n",
       "          -3.9472e-01,  5.0770e-01, -5.6534e-02,  1.2945e-01,  1.8567e-01,\n",
       "          -3.5001e-01,  1.6610e-01,  1.0337e+00,  1.7414e-01,  4.1071e-01,\n",
       "           1.7420e-01,  5.1160e-01,  5.3064e-02, -8.8958e-01,  1.3691e-01,\n",
       "           2.0502e-01,  2.0789e-01, -6.7096e-02,  5.7131e-01, -2.5517e-02,\n",
       "          -6.5181e-01, -2.3094e-01, -3.2169e-01,  4.3108e-01,  6.5352e-01,\n",
       "          -2.2597e-01, -4.4343e-01,  2.3894e-01, -7.9776e-03, -5.3395e-01,\n",
       "           9.0074e-02, -2.2320e-01, -5.2792e-01,  8.7275e-02, -5.8914e-01,\n",
       "           1.4985e-01, -1.2428e-01,  5.4238e-02,  6.6887e-01, -4.0159e-01,\n",
       "          -1.6113e-01,  3.6602e-02, -4.4532e-02, -4.1270e-01, -4.9520e-01,\n",
       "          -7.7083e-02,  4.1073e-01, -1.9776e-01, -2.7976e-01,  7.0636e-02,\n",
       "          -3.0673e-01,  2.2584e-01, -1.7377e-01, -6.4935e-01,  3.2879e-01,\n",
       "          -6.0415e-01,  2.0578e-01,  2.5360e-01, -2.2507e-01,  4.9017e-01,\n",
       "           1.1337e-01, -8.1817e-02,  1.9397e-01, -9.9369e-02, -7.8554e-01,\n",
       "          -3.2573e-01, -2.7299e-02,  3.2662e-01, -4.2844e-01,  1.5776e-01,\n",
       "           1.5022e-01, -4.5752e-02, -2.2574e-01,  2.7904e-01,  1.0174e-01,\n",
       "           3.7762e-01,  3.9846e-01, -5.5006e-01,  1.1727e-01, -2.9867e-01,\n",
       "          -2.9262e-02, -6.1174e-02,  1.7952e-01,  2.3654e-01, -7.0764e-01,\n",
       "          -1.5224e-01, -2.6434e-01, -3.0965e-01, -2.1650e-01, -2.5120e-01,\n",
       "           2.0680e-01,  1.4038e-01, -1.5897e-02, -1.1308e-01,  5.8077e-01,\n",
       "           4.0814e-02, -1.1167e-01, -4.5997e-01,  3.1443e-01,  4.3178e-01,\n",
       "           2.1794e-01,  7.5875e-02, -1.2830e-01, -5.5532e-01, -1.1922e-02,\n",
       "          -4.3644e-01,  9.1081e-01, -7.3906e-01, -7.1970e-01,  6.6086e-02,\n",
       "           1.4951e-02, -8.8259e-04, -2.3450e-01, -5.0265e-01,  8.1298e-02,\n",
       "          -2.1017e-01, -1.4891e-01,  6.1550e-03, -1.0607e-02,  1.6299e-01,\n",
       "           4.0742e-01,  3.6838e-01,  1.5951e-01, -3.3938e-01, -4.4081e-01,\n",
       "          -6.0244e-01,  1.4567e-02, -1.6795e-01,  2.9330e-01, -2.6752e-01,\n",
       "          -3.2661e-01,  1.6043e-02, -9.1892e-03, -2.2272e-01,  5.2876e-01,\n",
       "           1.2427e-01,  1.6396e+00, -3.7287e-01, -4.4551e-01,  2.9018e-01,\n",
       "           7.2306e-02, -2.7073e-02, -2.8130e-01,  2.5074e-01,  9.5447e-02,\n",
       "           4.0194e-02, -3.6304e-01,  1.5023e-02, -2.0558e-02,  3.2773e-01,\n",
       "           3.2056e-01, -1.0703e+00, -2.0238e-01, -1.7993e-01,  5.5664e-01,\n",
       "           3.2388e-02,  6.3185e-01, -2.2305e-01,  3.0577e-01,  4.6106e-01,\n",
       "          -8.6667e-01, -2.3000e-01, -1.6129e-01,  2.7295e-01, -1.0583e-01,\n",
       "          -9.8735e-02, -1.6178e-01, -3.7165e-02, -1.6936e-01,  3.9538e-01,\n",
       "           7.4751e-01, -7.8426e-01,  7.4081e-01,  7.0467e-01, -9.8234e-02,\n",
       "           5.6815e-01,  9.4187e-02,  3.2641e-01, -1.4659e-01,  2.2669e-02,\n",
       "           2.6200e-01,  1.8542e-01, -6.3165e-01, -2.3962e-01,  2.4473e-01,\n",
       "           4.0158e-01, -1.1943e-01,  1.2108e-01,  7.9292e-02,  3.0960e-02,\n",
       "          -3.7691e-01, -1.0145e-01, -2.4552e-01, -3.9904e-01,  1.1356e-01,\n",
       "           1.4603e-02, -1.5001e-01, -8.6997e-02,  4.7709e-01, -1.7580e-01,\n",
       "           7.5067e-01,  4.2910e-01,  3.7145e-01, -3.5103e-01,  8.9950e-04,\n",
       "          -5.7565e-02,  6.0671e-01, -3.1670e-01,  7.7385e-01,  6.3337e-02,\n",
       "          -2.3140e-01,  2.7251e-01,  6.0547e-03,  1.7829e-01,  2.9892e-01,\n",
       "          -2.3519e-01,  1.2702e-01, -3.0718e-01, -2.3711e-01,  2.8330e-01,\n",
       "          -1.8640e-01,  1.1542e-01, -7.2270e-01, -6.3602e-03,  3.9857e-01,\n",
       "          -3.2051e-01,  4.9873e-01, -2.3074e-01,  6.1026e-01, -4.3509e-01,\n",
       "           3.2669e-02, -4.7470e-01, -4.3275e-01, -2.8508e-01,  5.1854e-02,\n",
       "          -8.6027e-01, -1.9453e-01,  1.7684e-01,  3.7405e-02,  2.1707e-01,\n",
       "           7.5155e-01, -3.3794e-01, -2.4771e-01, -2.0386e-01,  4.0190e-02,\n",
       "           1.8299e-01, -8.2071e-02, -3.5586e-01,  2.1161e-01, -3.5607e-01,\n",
       "          -1.0520e+00, -2.3117e-01,  8.9990e-02,  3.8419e-01, -3.5853e-01,\n",
       "           1.1570e+00,  2.5132e-02, -2.0226e-01,  6.7253e-01,  2.0808e-01,\n",
       "           6.4157e-01,  6.0323e-01,  1.3628e-03, -4.1259e-01, -5.4109e-02,\n",
       "          -1.5877e-01, -3.2881e-01,  1.5757e-01, -1.8346e-01,  5.0556e-01,\n",
       "           5.4802e-01, -2.0806e-02,  5.4756e-01]], device='mps:0'),\n",
       " tensor([[-6.1750e-01, -4.2829e-01, -2.3934e-01, -9.4116e-01, -3.4548e-01,\n",
       "          -6.5496e-02, -6.3874e-01,  9.7828e-01, -9.5892e-02,  4.0096e-01,\n",
       "           1.4919e-01, -5.1603e-03,  2.0980e-01,  1.3680e-01, -6.9506e-01,\n",
       "           1.6740e-01,  4.6094e-01,  2.1159e-01,  3.5951e-01, -4.4329e-02,\n",
       "           7.9832e-01,  3.4245e-01,  1.8019e-01, -2.5205e-01, -4.5876e-01,\n",
       "           5.8994e-01, -2.4628e-01, -9.2014e-02,  2.0509e-01, -8.1949e-01,\n",
       "          -1.3586e-01, -4.9757e-02, -3.6281e-01,  8.7602e-01,  1.1948e-01,\n",
       "          -2.7310e-01, -1.5276e-01, -2.7996e-01,  2.3494e-01, -6.3637e-02,\n",
       "           3.7814e-01,  2.6592e-02, -2.9347e-01,  3.1386e-01, -1.6421e-01,\n",
       "          -1.4057e-01,  1.4047e-02, -7.8429e-01,  1.3666e-02,  1.8955e-01,\n",
       "           5.8851e-02, -4.7205e-02,  1.9231e-01,  3.7030e-01, -1.0617e+00,\n",
       "          -2.6348e-01,  2.7603e-01,  2.9849e-01, -1.7932e-01, -5.7784e-02,\n",
       "          -1.9989e-01,  9.0201e-01,  1.7949e-01, -2.7822e-01,  1.7824e-01,\n",
       "          -4.4899e-01, -5.4601e-01, -6.0781e-01, -1.0682e-01, -1.1967e-01,\n",
       "          -1.0002e-01,  9.7528e-02,  1.4009e-01, -2.9385e-01,  2.3170e-01,\n",
       "           2.1813e-01,  4.7024e-02, -2.7652e-02, -4.3158e-01, -2.6546e-01,\n",
       "           3.6947e-01,  3.4425e-02, -1.4730e-02,  2.0326e-01, -5.8744e-01,\n",
       "          -2.0226e-01,  3.7959e-01,  1.5703e-01,  2.5914e-01,  2.2662e-03,\n",
       "           5.9967e-01,  1.2024e-02,  8.0339e-02, -4.5369e-01,  5.8427e-01,\n",
       "          -5.4955e-01,  1.0329e-01, -1.6007e-01,  4.9532e-02, -5.2088e-01,\n",
       "          -9.8735e-02,  2.4494e-01, -1.8085e-01,  1.6053e-01, -4.6592e-01,\n",
       "          -3.2983e-02,  2.4514e-01,  3.0468e-01, -4.9183e-01,  4.2685e-02,\n",
       "          -6.5134e-03, -2.9791e-01,  7.3694e-02, -6.6407e-01, -8.5795e-01,\n",
       "           5.8354e-01,  1.8428e-02, -4.5240e-01, -2.3930e-01,  7.4443e-01,\n",
       "           2.9738e-01, -2.3201e-01, -5.3072e-02, -3.5548e-01,  2.0259e-01,\n",
       "          -1.8518e-01,  2.5689e-01, -2.7018e-01,  3.8439e-01,  5.5026e-01,\n",
       "           3.1581e-01,  2.8489e-01, -1.8327e-01,  2.7637e-01,  1.4465e-01,\n",
       "           3.6947e-02, -6.6614e-02, -1.5929e-01,  1.5180e-01,  1.2041e-01,\n",
       "          -2.9580e-01, -3.2999e-01,  5.4746e-03,  1.1182e-01, -3.3279e-01,\n",
       "          -1.0202e-01,  1.8680e-01,  3.4808e-01, -1.4092e-01, -2.5403e-01,\n",
       "          -7.8984e-01, -4.2328e-01,  1.0217e-01, -1.8560e-01, -1.7135e-01,\n",
       "          -4.1911e-01,  3.8121e-01, -6.6660e-02,  4.7638e-01,  3.4189e-01,\n",
       "           6.9978e-01, -1.5311e-01,  5.9776e-02, -5.6412e-01,  3.5197e-01,\n",
       "           3.8270e-02, -6.1110e-01, -1.7247e-01,  1.5680e-01, -4.1813e-04,\n",
       "           6.8195e-01,  7.0979e-02,  4.0124e-01,  1.2070e-01, -3.2553e-01,\n",
       "           8.2089e-03,  1.9890e-01, -1.2864e-01, -1.4297e-01,  7.3766e-01,\n",
       "          -5.5188e-01, -2.3029e+00,  1.0721e-01, -3.5844e-01, -3.6529e-01,\n",
       "          -1.8135e-01,  3.7255e-01,  3.3772e-02,  4.9966e-01,  1.5486e-01,\n",
       "           3.7598e-02,  1.6729e-01, -2.8025e-02,  3.6851e-01, -8.4399e-02,\n",
       "           2.6751e-01,  3.3341e-01,  1.2988e-01, -2.3298e-01, -1.0956e-01,\n",
       "          -3.9227e-01,  2.4031e-01, -9.8916e-02,  8.6010e-01,  2.3260e-01,\n",
       "           2.0943e-01, -4.3657e-03, -4.7262e-02,  4.8964e-01,  4.3099e-01,\n",
       "          -6.6541e-02, -1.6585e-01, -4.9522e-01,  4.6269e-03, -4.1279e-02,\n",
       "          -1.2963e-02, -4.5611e-01,  1.3753e-01,  4.2841e-01, -1.2416e-01,\n",
       "          -1.8354e-01,  1.5525e-01, -4.7715e-02, -3.1517e-01, -2.6492e-02,\n",
       "           6.8453e-01,  1.9080e-01, -2.2994e-01,  1.2377e-01, -1.1896e-01,\n",
       "           3.7006e-02, -4.6975e-01,  7.0044e-01,  6.1147e-01,  1.5585e-01,\n",
       "           1.6214e-02, -6.9943e-01,  2.3699e-01, -3.1574e-01, -4.7647e-03,\n",
       "          -1.5027e-01, -1.5731e-01, -4.2986e-01, -5.7285e-02, -5.5854e-02,\n",
       "           4.0115e-01,  7.9943e-02,  4.6653e-01,  1.1837e+00,  2.7973e-01,\n",
       "           1.9184e-02,  5.6097e-01,  2.0801e-01,  4.2492e-01, -4.0180e-01,\n",
       "           3.3082e-01,  8.8825e-02, -6.3896e-01,  2.7188e-01, -4.1593e-01,\n",
       "           4.4963e-02, -4.1432e-02, -3.7926e-01,  1.4208e-01, -2.1276e-02,\n",
       "           3.4237e-01,  3.3517e-01,  1.5186e-01,  1.6838e-01,  2.9411e-01,\n",
       "          -3.1322e-01,  5.9715e-01, -6.0809e-02, -1.1127e-01, -6.3492e-01,\n",
       "          -7.6732e-01,  9.5206e-02,  4.4743e-01, -1.2880e-01, -6.1279e-01,\n",
       "           1.0462e-01, -2.3866e-01, -1.2257e-01, -3.5064e-02,  2.1191e-01,\n",
       "           2.7455e-01,  5.0687e-02, -9.5436e-02,  6.4791e-02, -4.6950e-01,\n",
       "          -4.9277e-01,  4.9140e-02, -1.5758e-01, -1.3063e-01,  5.3918e-01,\n",
       "          -7.9415e-02,  1.1575e-01,  5.7221e-01,  1.9799e-02, -4.5976e-01,\n",
       "          -2.5640e-01,  1.4802e-01,  2.8667e-01, -7.2082e-02,  2.1981e-01,\n",
       "           3.4119e-01,  2.6467e-02,  5.1649e-01,  1.4823e-01, -2.1046e-01,\n",
       "          -2.7239e-01,  5.1173e-01, -1.5755e-01,  6.1403e-02, -1.2527e-01,\n",
       "          -8.7207e-01, -5.6150e-02, -7.8413e-02, -1.6833e-01, -5.4440e-01,\n",
       "           2.6566e-01,  1.4351e-01,  1.0561e-01,  2.2350e-01, -1.5984e-01,\n",
       "           7.0123e-01,  6.5285e-01, -3.3100e-01, -1.6943e-01, -4.0104e-01,\n",
       "           9.3922e-02, -6.8374e-02, -2.9329e-01, -1.4009e-02,  9.5368e-03,\n",
       "          -2.2765e-01, -1.2516e-01, -1.0805e-01,  2.8921e-01, -1.8876e-01,\n",
       "           5.2862e-01,  1.2189e-01,  8.3649e-02,  4.3873e-01,  2.0612e-02,\n",
       "           1.3071e-01,  3.6515e-01, -4.2117e-01,  1.2870e-01,  4.0424e-01,\n",
       "           4.0578e-01, -1.0551e+00, -8.3291e-01, -1.4936e-01,  3.6111e-01,\n",
       "           2.2860e-01,  2.5189e-01,  1.6999e-02, -7.2215e-03,  8.9455e-02,\n",
       "          -2.3402e-01,  3.5135e-01, -6.6415e-02, -5.7412e-01,  3.0561e-01,\n",
       "           2.3567e-01,  5.3263e-01, -2.7377e-01, -5.0579e-01, -6.5233e-02,\n",
       "           2.4616e-01, -3.6381e-01,  1.1392e-01,  2.3480e-01,  4.5466e-01,\n",
       "          -3.6015e-01, -5.6164e-01, -7.0282e-02, -7.3986e-01, -1.3578e-01,\n",
       "           3.2483e-01, -2.1437e-01,  2.7178e-02, -2.4758e-01,  1.5231e-01,\n",
       "          -1.7925e-02, -1.8879e-01,  6.0757e-01,  2.0522e-01, -2.5855e-01,\n",
       "           1.1981e-01, -2.3892e-01, -1.9475e-01,  5.0091e-01, -5.0226e-01,\n",
       "          -3.0348e-01, -5.0960e-02, -1.5349e-01, -3.2418e-01,  3.2181e-01,\n",
       "          -4.1389e-01, -1.4861e-01,  1.6748e-01, -6.4736e-02, -1.3068e-01,\n",
       "          -3.5233e-01,  1.4268e-01, -1.2707e-01, -4.9883e-02, -1.7654e-02,\n",
       "          -1.3353e-01,  6.0900e-02,  1.3308e-01,  6.4538e-02,  1.4388e-01,\n",
       "           5.3696e-01, -1.1464e-01,  2.7582e-03, -2.5900e-01, -2.2469e-01,\n",
       "           5.2498e-01, -9.4973e-02, -3.3117e-01,  5.4928e-01, -1.1156e+01,\n",
       "           1.0386e-01,  7.2631e-01, -5.5808e-01,  5.5128e-02, -2.8539e-01,\n",
       "           2.1085e-01, -3.0009e-01,  7.9101e-02,  2.4967e-01, -2.8460e-01,\n",
       "          -1.2277e-01, -2.7364e-01, -2.3193e-01, -4.0872e-01, -3.9742e-01,\n",
       "          -4.7229e-01, -5.6789e-01,  1.1772e-01, -1.7663e-01, -1.0996e-01,\n",
       "          -5.7036e-01, -1.9379e-01,  2.9696e-01, -4.9559e-03, -5.1069e-01,\n",
       "           5.9541e-02, -2.2948e-01,  2.6399e-01,  4.6126e-02,  2.2045e-01,\n",
       "          -1.9949e-02,  1.4108e-01,  2.0669e-01, -4.4946e-02, -4.5100e-01,\n",
       "          -4.4853e-01,  2.2129e-01, -1.5955e-01, -2.2267e-01, -5.4017e-02,\n",
       "          -3.3838e-01, -2.7689e-01,  1.3447e-01,  3.7594e-01,  6.2596e-01,\n",
       "           1.1595e+00,  5.6208e-01,  2.9245e-01, -1.0625e-01, -8.5438e-02,\n",
       "          -1.0762e+00,  5.5900e-01, -3.7318e-01,  8.1787e-02, -2.5119e-01,\n",
       "           6.6201e-01, -4.9906e-01,  5.4685e-01, -4.0754e-01, -2.5500e-01,\n",
       "           3.7179e-01,  3.0194e-01,  7.1104e-03,  2.8239e-02, -2.7432e-01,\n",
       "           4.0257e-01, -2.6094e-01, -3.6915e-02, -3.7819e-01,  4.0081e-01,\n",
       "          -3.9472e-01,  5.0770e-01, -5.6534e-02,  1.2945e-01,  1.8567e-01,\n",
       "          -3.5001e-01,  1.6610e-01,  1.0337e+00,  1.7414e-01,  4.1071e-01,\n",
       "           1.7420e-01,  5.1160e-01,  5.3064e-02, -8.8958e-01,  1.3691e-01,\n",
       "           2.0502e-01,  2.0789e-01, -6.7096e-02,  5.7131e-01, -2.5517e-02,\n",
       "          -6.5181e-01, -2.3094e-01, -3.2169e-01,  4.3108e-01,  6.5352e-01,\n",
       "          -2.2597e-01, -4.4343e-01,  2.3894e-01, -7.9776e-03, -5.3395e-01,\n",
       "           9.0074e-02, -2.2320e-01, -5.2792e-01,  8.7275e-02, -5.8914e-01,\n",
       "           1.4985e-01, -1.2428e-01,  5.4238e-02,  6.6887e-01, -4.0159e-01,\n",
       "          -1.6113e-01,  3.6602e-02, -4.4532e-02, -4.1270e-01, -4.9520e-01,\n",
       "          -7.7083e-02,  4.1073e-01, -1.9776e-01, -2.7976e-01,  7.0636e-02,\n",
       "          -3.0673e-01,  2.2584e-01, -1.7377e-01, -6.4935e-01,  3.2879e-01,\n",
       "          -6.0415e-01,  2.0578e-01,  2.5360e-01, -2.2507e-01,  4.9017e-01,\n",
       "           1.1337e-01, -8.1817e-02,  1.9397e-01, -9.9369e-02, -7.8554e-01,\n",
       "          -3.2573e-01, -2.7299e-02,  3.2662e-01, -4.2844e-01,  1.5776e-01,\n",
       "           1.5022e-01, -4.5752e-02, -2.2574e-01,  2.7904e-01,  1.0174e-01,\n",
       "           3.7762e-01,  3.9846e-01, -5.5006e-01,  1.1727e-01, -2.9867e-01,\n",
       "          -2.9262e-02, -6.1174e-02,  1.7952e-01,  2.3654e-01, -7.0764e-01,\n",
       "          -1.5224e-01, -2.6434e-01, -3.0965e-01, -2.1650e-01, -2.5120e-01,\n",
       "           2.0680e-01,  1.4038e-01, -1.5897e-02, -1.1308e-01,  5.8077e-01,\n",
       "           4.0814e-02, -1.1167e-01, -4.5997e-01,  3.1443e-01,  4.3178e-01,\n",
       "           2.1794e-01,  7.5875e-02, -1.2830e-01, -5.5532e-01, -1.1922e-02,\n",
       "          -4.3644e-01,  9.1081e-01, -7.3906e-01, -7.1970e-01,  6.6086e-02,\n",
       "           1.4951e-02, -8.8259e-04, -2.3450e-01, -5.0265e-01,  8.1298e-02,\n",
       "          -2.1017e-01, -1.4891e-01,  6.1550e-03, -1.0607e-02,  1.6299e-01,\n",
       "           4.0742e-01,  3.6838e-01,  1.5951e-01, -3.3938e-01, -4.4081e-01,\n",
       "          -6.0244e-01,  1.4567e-02, -1.6795e-01,  2.9330e-01, -2.6752e-01,\n",
       "          -3.2661e-01,  1.6043e-02, -9.1892e-03, -2.2272e-01,  5.2876e-01,\n",
       "           1.2427e-01,  1.6396e+00, -3.7287e-01, -4.4551e-01,  2.9018e-01,\n",
       "           7.2306e-02, -2.7073e-02, -2.8130e-01,  2.5074e-01,  9.5447e-02,\n",
       "           4.0194e-02, -3.6304e-01,  1.5023e-02, -2.0558e-02,  3.2773e-01,\n",
       "           3.2056e-01, -1.0703e+00, -2.0238e-01, -1.7993e-01,  5.5664e-01,\n",
       "           3.2388e-02,  6.3185e-01, -2.2305e-01,  3.0577e-01,  4.6106e-01,\n",
       "          -8.6667e-01, -2.3000e-01, -1.6129e-01,  2.7295e-01, -1.0583e-01,\n",
       "          -9.8735e-02, -1.6178e-01, -3.7165e-02, -1.6936e-01,  3.9538e-01,\n",
       "           7.4751e-01, -7.8426e-01,  7.4081e-01,  7.0467e-01, -9.8234e-02,\n",
       "           5.6815e-01,  9.4187e-02,  3.2641e-01, -1.4659e-01,  2.2669e-02,\n",
       "           2.6200e-01,  1.8542e-01, -6.3165e-01, -2.3962e-01,  2.4473e-01,\n",
       "           4.0158e-01, -1.1943e-01,  1.2108e-01,  7.9292e-02,  3.0960e-02,\n",
       "          -3.7691e-01, -1.0145e-01, -2.4552e-01, -3.9904e-01,  1.1356e-01,\n",
       "           1.4603e-02, -1.5001e-01, -8.6997e-02,  4.7709e-01, -1.7580e-01,\n",
       "           7.5067e-01,  4.2910e-01,  3.7145e-01, -3.5103e-01,  8.9950e-04,\n",
       "          -5.7565e-02,  6.0671e-01, -3.1670e-01,  7.7385e-01,  6.3337e-02,\n",
       "          -2.3140e-01,  2.7251e-01,  6.0547e-03,  1.7829e-01,  2.9892e-01,\n",
       "          -2.3519e-01,  1.2702e-01, -3.0718e-01, -2.3711e-01,  2.8330e-01,\n",
       "          -1.8640e-01,  1.1542e-01, -7.2270e-01, -6.3602e-03,  3.9857e-01,\n",
       "          -3.2051e-01,  4.9873e-01, -2.3074e-01,  6.1026e-01, -4.3509e-01,\n",
       "           3.2669e-02, -4.7470e-01, -4.3275e-01, -2.8508e-01,  5.1854e-02,\n",
       "          -8.6027e-01, -1.9453e-01,  1.7684e-01,  3.7405e-02,  2.1707e-01,\n",
       "           7.5155e-01, -3.3794e-01, -2.4771e-01, -2.0386e-01,  4.0190e-02,\n",
       "           1.8299e-01, -8.2071e-02, -3.5586e-01,  2.1161e-01, -3.5607e-01,\n",
       "          -1.0520e+00, -2.3117e-01,  8.9990e-02,  3.8419e-01, -3.5853e-01,\n",
       "           1.1570e+00,  2.5132e-02, -2.0226e-01,  6.7253e-01,  2.0808e-01,\n",
       "           6.4157e-01,  6.0323e-01,  1.3628e-03, -4.1259e-01, -5.4109e-02,\n",
       "          -1.5877e-01, -3.2881e-01,  1.5757e-01, -1.8346e-01,  5.0556e-01,\n",
       "           5.4802e-01, -2.0806e-02,  5.4756e-01]], device='mps:0'),\n",
       " tensor([[-1.5781e-01, -5.0992e-01, -5.2580e-01, -7.0710e-01, -1.4161e-01,\n",
       "          -1.6547e-02, -6.4323e-01,  5.8570e-01,  1.8530e-01,  5.0470e-01,\n",
       "           6.9798e-02,  5.0401e-01, -2.2203e-02, -7.6359e-02,  1.2330e-01,\n",
       "          -4.7902e-01, -3.9285e-01,  1.6326e-01,  8.1238e-01,  2.9375e-01,\n",
       "           2.7931e-01,  6.1244e-01, -3.5493e-01, -1.5226e-01, -8.2967e-01,\n",
       "           2.5615e-01,  8.3514e-02,  3.0973e-01,  4.1873e-01, -8.6655e-01,\n",
       "           1.6462e-01,  4.5239e-01, -5.8220e-02,  1.1117e+00, -3.6446e-01,\n",
       "          -4.5364e-01,  2.2506e-01, -7.7945e-01, -1.1660e-01,  9.3723e-02,\n",
       "           5.6723e-01,  1.7321e-01,  1.3774e-01,  2.3506e-01,  6.3253e-02,\n",
       "           3.4456e-01,  1.8056e-02,  2.7214e-02,  1.7971e-01, -6.6470e-01,\n",
       "          -3.4960e-01, -6.1410e-02,  4.1376e-01,  6.0163e-01, -5.4180e-01,\n",
       "          -1.0748e-01,  6.5859e-01,  6.1187e-02,  2.3336e-01,  2.8437e-01,\n",
       "          -4.1412e-02,  3.8336e-01, -2.3617e-01,  2.6498e-01,  2.2513e-01,\n",
       "           1.9179e-02, -5.3687e-01, -3.1205e-01, -5.7096e-01,  2.0575e-01,\n",
       "          -4.7793e-02,  2.7945e-01,  2.0980e-01, -1.6856e-01,  3.6934e-01,\n",
       "          -5.3071e-01,  8.4366e-02, -6.8787e-02, -3.4101e-01,  1.8018e-01,\n",
       "           1.7697e-01,  4.4777e-01,  4.2679e-01, -1.6426e-01, -4.9577e-02,\n",
       "           1.5999e-02,  2.1008e-01,  5.3013e-02,  7.6437e-02,  8.2946e-02,\n",
       "          -8.9483e-02, -5.8380e-02, -2.2370e-01,  8.2717e-02,  1.2760e-01,\n",
       "          -5.0948e-01,  3.0188e-01, -2.1457e-01,  6.4903e-02, -3.2624e-01,\n",
       "           9.2878e-02, -2.2163e-02, -3.9446e-01,  2.4425e-01, -1.8535e-01,\n",
       "          -4.6015e-01,  2.5296e-01, -3.5769e-01, -5.1183e-01,  1.4507e-01,\n",
       "          -1.4680e-02, -1.5266e-01, -2.3045e-01, -4.6554e-01,  3.2003e-02,\n",
       "           2.6113e-01,  2.8067e-01, -1.8721e-01, -2.4708e-01,  2.4327e-01,\n",
       "           9.3129e-02, -2.4554e-01, -6.6139e-02,  5.0763e-01,  8.6338e-02,\n",
       "          -2.5970e-01,  8.9385e-01,  2.1686e-01,  6.8913e-01,  5.5281e-01,\n",
       "           2.4155e-01, -2.6915e-01, -1.8724e-01, -2.2260e-01,  1.4744e-01,\n",
       "           2.7082e-02,  2.0385e-01, -2.6759e-01, -1.0453e-01,  3.4967e-02,\n",
       "           2.8031e-01, -1.4464e-01, -4.3931e-01, -1.4683e-01,  1.0016e-01,\n",
       "          -2.3232e-01, -1.5207e-01, -2.2894e-01, -2.6596e-01, -1.2399e-01,\n",
       "          -6.3435e-01, -4.2272e-01,  5.5937e-01,  1.5478e-01, -2.9002e-01,\n",
       "          -3.7936e-01,  2.7465e-01,  1.8570e-01, -6.5997e-02,  1.7489e-01,\n",
       "           1.1557e+00,  2.5073e-01,  1.4654e-01, -2.4056e-01, -5.5174e-01,\n",
       "           2.8521e-01,  7.7835e-02, -3.2071e-01,  5.8224e-01, -1.1559e-01,\n",
       "          -1.8615e-01,  5.3328e-02,  7.9291e-01,  1.0865e-01,  2.9654e-01,\n",
       "          -6.7051e-01,  5.7451e-01,  1.6863e-01, -2.1418e-01,  1.9789e-01,\n",
       "          -7.2173e-02, -1.0774e+00,  4.3750e-02, -5.6154e-01,  3.4103e-01,\n",
       "          -1.9562e-01,  2.9922e-01,  1.2636e-01,  1.8478e-01, -3.4550e-02,\n",
       "           2.3402e-02,  3.9286e-01, -2.1711e-01,  2.2115e-01, -2.2336e-01,\n",
       "           2.7407e-01,  4.0094e-01,  5.5743e-01,  2.4645e-01, -2.3108e-01,\n",
       "          -2.3382e-01,  2.9999e-01,  3.0993e-01,  6.0755e-02,  2.8265e-01,\n",
       "          -5.4359e-02, -2.9089e-01, -1.2384e-01,  3.1278e-02, -7.8755e-02,\n",
       "          -3.6363e-01, -3.6966e-02, -7.2378e-01, -9.6473e-02, -3.8647e-01,\n",
       "           2.7957e-01, -2.9475e-01, -2.4662e-01,  4.1676e-01,  2.0345e-01,\n",
       "          -2.7736e-01,  1.6306e-01,  3.8752e-02, -1.7066e-01,  1.7130e-01,\n",
       "           4.2787e-01, -7.8545e-02,  1.8394e-01, -6.8328e-02, -3.2753e-01,\n",
       "           2.0181e-01, -6.3700e-01,  7.9564e-01,  1.1177e-01,  1.8687e-02,\n",
       "           2.6190e-01, -2.9208e-01, -2.1664e-01, -6.1600e-01,  3.6885e-01,\n",
       "          -3.1443e-01, -4.3459e-01, -3.5100e-01,  1.9732e-01,  1.9968e-01,\n",
       "           2.2480e-01, -1.6236e-01,  8.4536e-01,  3.2556e-01,  5.5046e-01,\n",
       "          -2.5113e-01,  1.0679e-02,  2.4304e-01,  1.2561e-01, -3.3538e-01,\n",
       "           8.8028e-02, -3.3142e-01, -5.3094e-01,  1.2902e-01, -8.6394e-02,\n",
       "           1.0718e-02,  5.0505e-02,  2.4938e-01,  2.0843e-01, -4.8641e-01,\n",
       "          -5.8416e-01,  4.6299e-01,  1.3440e-01, -3.8833e-01,  5.6609e-01,\n",
       "          -3.2050e-01,  1.4517e-01,  1.7377e-01, -1.9703e-01, -5.8215e-01,\n",
       "          -4.4394e-01,  3.1996e-01, -4.3420e-02, -2.3261e-01, -1.0205e+00,\n",
       "          -4.5040e-01,  4.3744e-01,  1.8882e-01,  1.9652e-01,  2.0742e-01,\n",
       "           6.8175e-01, -2.6957e-03, -1.3534e-01, -8.2438e-02, -2.9796e-01,\n",
       "          -1.6681e-01, -1.6167e-01, -3.8973e-01,  2.0079e-01,  4.4172e-01,\n",
       "           2.4041e-01,  2.5392e-01,  3.4537e-01, -2.5310e-01, -1.2667e-01,\n",
       "          -5.3792e-02,  1.9521e-01, -6.3764e-01, -1.9579e-01,  1.9362e-01,\n",
       "          -1.5552e-01,  7.1322e-01,  6.3208e-01,  3.5991e-02, -6.8564e-01,\n",
       "           2.0996e-02,  5.0123e-01, -1.3038e-01, -2.8951e-01,  4.2505e-01,\n",
       "          -9.7612e-01, -7.1284e-02,  2.2609e-01,  7.0211e-02, -4.3095e-01,\n",
       "           2.1175e-01,  2.4446e-01,  1.1383e-01, -8.8363e-02, -2.2051e-01,\n",
       "           3.0479e-01,  1.1060e-01,  1.0715e-01,  3.0169e-01, -3.8249e-01,\n",
       "          -8.7919e-02, -2.5846e-01, -7.3490e-02, -5.3722e-01, -4.2831e-01,\n",
       "           2.0623e-01, -2.6238e-02, -2.6921e-01, -4.2697e-01, -1.0382e-01,\n",
       "           4.8947e-01, -6.0245e-01, -8.3064e-02,  3.2129e-01,  1.2692e-02,\n",
       "          -1.9074e-01,  4.9721e-01, -1.4804e-01,  6.6244e-02,  9.7425e-03,\n",
       "           2.8007e-02, -2.1384e-01,  1.4408e-02, -4.8289e-02,  6.8610e-01,\n",
       "           3.8773e-01,  2.9117e-01, -2.2263e-01,  5.9408e-01,  2.6366e-01,\n",
       "          -2.1527e-01,  3.2937e-01, -2.6726e-01, -1.5779e-01, -4.0238e-01,\n",
       "           2.4534e-01,  4.8729e-01, -5.0954e-01,  1.9696e-01,  1.6698e-01,\n",
       "          -1.3545e-01, -2.6171e-02, -3.9172e-01,  1.6433e-01, -1.2916e-01,\n",
       "          -3.7194e-01, -5.4959e-01,  1.6118e-01, -2.8359e-01, -6.1355e-01,\n",
       "           4.3530e-01,  7.2215e-02,  1.6596e-01, -2.9144e-01, -4.3534e-01,\n",
       "          -2.1953e-01, -3.6469e-01,  8.3734e-01,  2.9694e-01, -3.2911e-01,\n",
       "           1.4232e-01, -2.0388e-01, -2.8853e-01,  2.9172e-01, -5.7740e-01,\n",
       "          -2.3860e-01,  3.8303e-01, -1.0529e-01,  1.2966e-01,  2.9424e-01,\n",
       "          -1.4396e-01, -3.0486e-02,  1.7053e-02,  4.9249e-01,  3.3606e-01,\n",
       "          -4.3229e-01,  1.8079e-01,  3.3487e-01, -3.6621e-01,  1.6318e-01,\n",
       "           7.7443e-02,  4.4454e-01, -4.6277e-01,  2.3695e-02, -3.6825e-02,\n",
       "           4.2013e-01, -3.0293e-01, -3.5290e-01, -6.0332e-01, -3.6233e-01,\n",
       "           1.0989e-01,  3.2839e-01,  3.6213e-01,  2.5186e-01, -1.1030e+01,\n",
       "           3.1711e-01,  7.3241e-01, -5.4285e-01, -2.7618e-01, -1.2090e-01,\n",
       "           9.9222e-02,  2.2094e-02,  2.7858e-01, -3.6664e-01, -2.3071e-01,\n",
       "          -4.1633e-01, -7.0163e-02, -1.3666e-02, -3.6821e-01, -2.4362e-01,\n",
       "           5.4209e-04, -1.0914e+00,  2.7345e-01, -3.4846e-01,  2.6811e-01,\n",
       "          -7.0889e-02, -8.4898e-02,  5.2085e-02, -4.6668e-01,  1.0174e-03,\n",
       "           2.5913e-01, -1.7198e-01,  2.2442e-01,  5.0384e-01,  2.2575e-01,\n",
       "           2.4121e-01,  4.0186e-02, -8.8747e-02, -3.9107e-01, -5.0520e-01,\n",
       "          -1.3486e-01, -2.3363e-01, -1.5591e-01, -3.5417e-01, -8.5191e-01,\n",
       "          -6.1727e-01, -3.1118e-01,  2.5997e-01,  4.2820e-01,  7.9170e-01,\n",
       "           2.5497e-01,  8.0057e-01,  1.7291e-01,  4.2108e-01,  2.4568e-03,\n",
       "          -1.0609e+00, -8.9845e-02, -2.0113e-01, -4.2582e-01, -6.6911e-02,\n",
       "           4.1607e-01, -1.0703e+00,  3.2149e-01, -5.5763e-01, -6.7813e-03,\n",
       "           3.3469e-01,  2.9976e-01,  1.8783e-01, -3.9465e-01, -2.2432e-01,\n",
       "          -2.7091e-01, -1.9850e-01,  2.2442e-01, -1.5522e-01, -2.6296e-01,\n",
       "          -1.4519e-01,  5.7515e-01,  5.1605e-02, -3.3781e-01, -2.1173e-01,\n",
       "           1.9103e-01, -2.8293e-01,  1.9011e+00,  5.8388e-02,  4.4762e-02,\n",
       "          -3.0451e-01,  4.7782e-01, -4.3705e-01, -4.5586e-01,  1.2000e-01,\n",
       "          -7.1133e-02,  3.1204e-01,  2.6026e-01,  2.3921e-01, -2.0121e-01,\n",
       "          -7.3120e-01, -4.4785e-02,  1.3814e-01,  8.7033e-02,  9.5182e-01,\n",
       "          -9.9429e-01, -4.2905e-01, -9.3967e-02,  6.0855e-01, -4.1387e-01,\n",
       "           9.6510e-02, -5.1339e-01, -5.1399e-01,  2.6513e-01, -2.8502e-01,\n",
       "           1.1389e-01,  3.3005e-01, -2.4264e-01,  3.9821e-01, -4.0572e-01,\n",
       "           6.0944e-03,  3.5082e-01, -1.9862e-01, -2.7852e-01, -1.4716e-01,\n",
       "           2.7089e-01,  5.6084e-01, -4.1438e-01, -4.3248e-01,  4.7478e-02,\n",
       "           2.2978e-01,  1.1056e-01,  4.0999e-01, -2.7706e-01,  4.3286e-01,\n",
       "          -6.6180e-01, -1.1328e-01,  1.3712e-01, -1.5367e-01, -1.8240e-01,\n",
       "           2.2588e-01,  6.3350e-02,  4.0825e-01, -6.1725e-01, -6.3658e-01,\n",
       "          -8.0743e-01, -7.5102e-02,  2.7082e-01, -2.8056e-01,  8.2971e-03,\n",
       "          -4.3073e-01, -1.9313e-01,  9.3965e-02, -2.7665e-01,  2.4826e-01,\n",
       "           1.6863e-02,  2.5478e-01, -5.0768e-01, -7.8010e-02, -3.2678e-02,\n",
       "          -7.3908e-01, -1.3772e-01, -4.0208e-01, -2.1995e-01, -5.8911e-01,\n",
       "          -5.0210e-01, -2.0471e-01, -3.7798e-01,  1.3923e-01,  2.2849e-01,\n",
       "           1.1320e-01, -1.0638e-01, -4.2935e-01,  8.7038e-01,  5.5785e-01,\n",
       "           2.3351e-01, -5.1414e-01, -7.5266e-01, -2.7597e-01,  3.0834e-01,\n",
       "           2.2888e-01, -9.4850e-02,  6.0818e-02, -3.6472e-01, -5.7167e-01,\n",
       "          -2.1715e-02,  6.7660e-01, -3.1861e-02, -5.3267e-01,  2.7331e-01,\n",
       "           9.3668e-01, -4.0132e-02,  5.1511e-02, -7.5189e-02, -1.4426e-01,\n",
       "          -5.9898e-01, -4.3118e-01,  7.6098e-01, -3.5298e-01, -3.7740e-02,\n",
       "           1.7773e-01,  1.4428e-01,  1.3363e-01,  2.9029e-01, -1.9853e-01,\n",
       "          -1.3984e-01, -1.3821e-01, -6.5803e-01,  4.1368e-01,  1.3674e-01,\n",
       "          -2.5315e-01,  6.9148e-01,  3.4142e-01, -3.5088e-01, -1.9319e-01,\n",
       "           3.4658e-01,  9.5337e-01, -2.8288e-01, -2.4996e-01,  6.3519e-01,\n",
       "           1.9774e-01, -2.9483e-01,  2.1986e-01,  1.0498e-01, -2.0473e-01,\n",
       "          -8.7072e-03,  4.8959e-01, -3.0988e-01, -1.1451e-01,  2.0239e-01,\n",
       "           4.5341e-01, -6.7135e-01, -7.6438e-01, -2.7376e-02,  6.4987e-01,\n",
       "          -1.8359e-01,  6.0993e-01, -2.0156e-01,  5.0500e-01,  2.1994e-01,\n",
       "          -9.9608e-01, -2.6803e-02, -4.8150e-01,  5.7726e-01,  2.0684e-01,\n",
       "          -1.0299e-01,  3.8453e-02, -2.8648e-01, -2.6735e-01,  5.6056e-01,\n",
       "           1.6144e-01, -6.1590e-01, -3.8190e-02,  5.9326e-01, -2.5955e-01,\n",
       "           6.2992e-01, -1.2607e-01,  3.2073e-02, -1.4413e-01,  2.7854e-01,\n",
       "           7.6557e-02,  1.5883e-01, -7.7913e-01, -4.4130e-01,  1.1568e-01,\n",
       "           8.1227e-01,  1.5364e-01,  1.7341e-01, -2.4724e-01,  2.2413e-01,\n",
       "          -9.9613e-02, -6.5001e-01, -8.6631e-02,  1.8095e-01,  6.2217e-02,\n",
       "           1.9002e-01, -1.6283e-01, -3.3140e-01,  4.2762e-01, -2.2792e-01,\n",
       "           1.5089e-01,  6.3875e-01,  6.4098e-01, -1.3167e-01,  2.8385e-02,\n",
       "          -3.3151e-01,  3.1497e-01, -1.1413e-01,  1.4675e-01, -2.7362e-02,\n",
       "          -2.5461e-01,  2.3888e-01,  3.2932e-01, -4.6269e-01,  1.8259e-01,\n",
       "          -5.4074e-01,  5.7914e-01, -5.8193e-02, -3.5020e-01,  2.1871e-01,\n",
       "           1.8065e-01,  2.2335e-01, -4.0101e-02,  2.1823e-01,  1.3830e-01,\n",
       "           8.7891e-02,  8.2575e-01,  1.9083e-01,  1.0306e-01, -6.6881e-01,\n",
       "           1.7689e-01, -8.6517e-01,  6.7618e-02, -1.9413e-01, -1.8497e-01,\n",
       "          -4.7084e-01, -8.2079e-02,  2.5657e-01,  1.2803e-01,  7.5084e-01,\n",
       "           1.0202e+00, -2.9167e-01, -7.1120e-01, -4.9722e-01,  1.1108e-01,\n",
       "           2.5412e-01,  1.0216e-01,  3.1019e-02,  3.7573e-01,  6.2195e-01,\n",
       "          -6.6838e-01, -7.0439e-02, -2.8948e-01,  3.6012e-01, -3.3450e-01,\n",
       "           4.1078e-01, -2.4111e-01, -1.7442e-01,  1.7711e-01,  1.0893e-01,\n",
       "           4.1415e-01,  6.6806e-01,  3.2305e-01, -3.8370e-01,  1.1077e-01,\n",
       "          -2.6189e-01, -4.0737e-01, -1.6271e-02, -5.1631e-01,  3.1841e-01,\n",
       "           2.3918e-01,  5.3457e-01,  9.1401e-01]], device='mps:0'),\n",
       " tensor([[-0.0513, -0.4731, -0.7129,  ...,  0.2473,  0.8054,  0.7577],\n",
       "         [-0.1544, -0.5675, -0.3689,  ..., -0.0941,  0.4334,  0.5743],\n",
       "         [-0.0676, -0.9284, -0.4982,  ..., -0.0947,  0.1492,  0.4318],\n",
       "         [-0.3088, -0.4400, -0.4987,  ...,  0.0387,  0.3656,  0.3273],\n",
       "         [-0.1683, -0.5138, -0.5756,  ...,  0.0116,  0.6081,  0.5018],\n",
       "         [ 0.0071, -0.6636, -0.6233,  ...,  0.2248,  0.5202,  0.6931]],\n",
       "        device='mps:0'),\n",
       " tensor([[-1.8820e-01, -5.5322e-01, -5.8603e-01, -6.7243e-01, -1.6545e-01,\n",
       "          -3.2276e-01, -4.2466e-01, -1.6232e-01, -1.9667e-01,  9.3293e-01,\n",
       "           1.6115e-01,  3.8046e-01, -5.7266e-02,  6.0431e-02, -3.1797e-01,\n",
       "           2.8137e-01,  8.3394e-01,  2.2954e-01,  2.0276e-01,  2.8926e-01,\n",
       "           3.5989e-01,  3.5079e-01,  2.9294e-01,  1.2363e-01, -4.1078e-01,\n",
       "          -2.1224e-01, -2.1678e-01, -2.6079e-01,  5.3462e-01, -4.1145e-01,\n",
       "          -1.5701e-01,  6.6504e-01,  3.9750e-02,  7.8349e-01, -1.2944e-01,\n",
       "           8.3003e-03,  2.4287e-02, -2.0974e-01,  9.4160e-01, -2.7919e-01,\n",
       "           6.0713e-01,  4.1603e-02,  2.0751e-01,  9.0747e-02,  2.1682e-01,\n",
       "          -4.0078e-02,  5.3810e-02, -7.7264e-01,  4.1105e-01, -2.8407e-02,\n",
       "          -2.0875e-01,  4.2690e-02,  6.3482e-01, -4.9974e-02, -5.3218e-01,\n",
       "          -2.6560e-01, -8.5375e-02,  3.2165e-01, -3.1154e-01, -2.2122e-01,\n",
       "          -1.3866e-01,  4.2382e-01, -1.1064e-01, -4.1011e-01, -2.9316e-01,\n",
       "          -2.3992e-01, -1.4067e-01, -9.4070e-01,  8.3380e-02, -2.0454e-01,\n",
       "           2.7542e-01,  4.4664e-01, -1.8579e-02, -4.2696e-01, -9.8204e-02,\n",
       "          -2.8681e-01,  1.6587e-01, -1.0216e-01, -7.0937e-01,  1.4381e-01,\n",
       "           4.8932e-01,  1.7797e-01,  8.2572e-02, -6.9653e-01,  2.9692e-02,\n",
       "           1.8314e-02,  4.1505e-01,  1.7759e-01,  3.6052e-01,  1.9446e-02,\n",
       "           2.6236e-01,  1.2449e-01, -4.8694e-01, -1.6652e-01,  1.2496e-01,\n",
       "          -7.7987e-02, -4.7261e-02, -1.0998e-01, -6.2739e-02, -3.0714e-01,\n",
       "          -6.9493e-02,  1.5824e-01,  1.5395e-01,  2.1533e-01, -3.6975e-01,\n",
       "          -4.8403e-01,  4.0363e-01, -3.7410e-01, -5.8849e-01, -1.8725e-01,\n",
       "           4.4365e-02, -1.4813e-01,  1.2395e-01, -2.2811e-01, -5.4170e-01,\n",
       "           1.7038e-01,  1.9835e-01,  3.2346e-01, -2.2454e-01,  7.1839e-01,\n",
       "           5.5667e-01, -4.9921e-02, -1.4970e-01, -1.0688e-01,  7.6861e-02,\n",
       "          -2.3505e-01,  5.7505e-01, -3.7668e-01,  1.8314e-01,  2.2411e-01,\n",
       "          -7.8894e-02,  1.2647e-01, -5.1478e-01,  1.3416e-01,  4.0108e-01,\n",
       "           3.8876e-01,  1.8893e-01, -4.0053e-01,  6.5394e-02,  2.8460e-01,\n",
       "           2.3132e-01, -2.9483e-01, -5.4404e-01, -4.9885e-01, -6.5666e-02,\n",
       "          -8.1615e-02, -2.2142e-01, -3.3296e-01, -2.3709e-01, -3.3953e-01,\n",
       "          -9.0174e-01, -6.6945e-01,  4.6240e-01, -2.9788e-01,  4.1650e-01,\n",
       "          -2.1544e-01,  3.5330e-01,  1.8047e-01, -3.4421e-01, -5.4293e-03,\n",
       "           4.5466e-01, -3.0943e-01,  2.2156e-01, -5.4001e-01,  3.3083e-01,\n",
       "          -1.0857e-01, -6.2650e-01, -1.7225e-01,  1.4664e-01, -1.0052e-01,\n",
       "          -7.4095e-02, -1.8374e-01,  3.2984e-01, -2.8468e-01,  7.8814e-02,\n",
       "          -2.1150e-01, -1.3598e-01,  2.0190e-01, -2.2285e-01, -3.2013e-01,\n",
       "          -1.8735e-01, -1.8636e+00,  3.7526e-01, -2.8986e-01, -7.1700e-01,\n",
       "          -5.8791e-01,  1.8058e-01,  3.0768e-01,  5.1919e-02, -2.3336e-01,\n",
       "          -4.3389e-01, -1.2689e-01, -2.0500e-01,  7.7166e-02, -1.0324e-02,\n",
       "           3.9754e-02,  7.2874e-01,  1.1389e-01, -1.7728e-01,  2.4622e-01,\n",
       "          -8.2868e-01,  6.6258e-02, -1.6339e-01,  1.4594e-02,  2.4032e-01,\n",
       "           2.1204e-01,  2.2616e-01,  1.5082e-02,  4.0856e-01,  2.6779e-01,\n",
       "          -5.9731e-02,  6.6131e-03, -1.7273e-01,  4.1647e-01, -1.4084e-01,\n",
       "           7.0950e-01, -2.9621e-01,  2.1683e-01,  7.6637e-01,  8.2811e-02,\n",
       "          -4.1094e-02,  1.1654e-01,  1.4952e-01, -3.5278e-01, -1.2196e-01,\n",
       "           3.1662e-01, -1.8344e-01,  5.7820e-02,  1.4079e-01, -2.7388e-01,\n",
       "          -1.7229e-01, -9.6992e-01,  4.8435e-01,  3.3197e-01, -3.1700e-01,\n",
       "          -3.6487e-01,  4.7528e-02,  6.2610e-01, -9.9821e-03, -3.0335e-01,\n",
       "          -1.2590e-01, -6.0504e-01, -5.9538e-01, -7.6554e-02,  8.6442e-02,\n",
       "          -6.8488e-03, -2.4464e-01,  2.4599e-01,  5.1044e-01,  2.9528e-01,\n",
       "           1.8019e-01,  1.7929e-02,  4.3011e-01,  6.4860e-01, -4.4134e-01,\n",
       "           1.1897e-01,  5.0062e-01, -8.0373e-02,  5.7094e-02, -3.0564e-01,\n",
       "          -4.8290e-01,  1.3149e-01,  4.8761e-02,  1.5030e-01,  7.6497e-02,\n",
       "           4.1589e-01,  3.9938e-01, -1.3077e-01, -4.4873e-01, -3.6717e-01,\n",
       "          -3.4260e-01,  1.3087e-01, -1.7108e-03, -6.0247e-01, -4.4915e-01,\n",
       "          -9.3511e-01, -1.2090e-01,  2.0786e-01, -3.3639e-02, -9.2810e-01,\n",
       "           1.4189e-02,  2.7474e-01, -3.2014e-01, -2.4398e-01,  1.8180e-01,\n",
       "          -1.0077e-01,  6.6739e-02,  7.4738e-02, -3.5180e-01, -4.4143e-01,\n",
       "           3.7802e-02,  2.5481e-01, -1.1550e-01,  3.7247e-01,  5.0041e-01,\n",
       "          -3.4729e-01,  4.2402e-01,  3.1100e-01, -1.1526e-01, -3.0080e-01,\n",
       "          -3.0684e-01,  3.3255e-01,  1.2127e-01, -1.0515e-01,  4.9488e-01,\n",
       "          -1.1672e-01,  2.5311e-01,  6.0280e-01, -8.0549e-01, -4.2388e-01,\n",
       "           9.8496e-03,  9.1802e-01, -2.0401e-02, -7.2090e-02, -2.9725e-01,\n",
       "          -6.7349e-01, -1.4526e-02, -2.7709e-01, -1.5915e-01, -4.9253e-01,\n",
       "           4.9242e-01,  9.5654e-02, -1.8553e-01, -2.7360e-03, -2.8998e-01,\n",
       "           4.8236e-01,  4.2413e-01,  1.4952e-01, -3.2406e-02, -3.2791e-01,\n",
       "          -1.7711e-01, -5.5577e-01, -3.7369e-01, -1.1625e-01, -1.2550e-01,\n",
       "           5.4559e-01,  2.8132e-01,  1.6962e-01, -7.5636e-02, -6.4554e-01,\n",
       "           5.5653e-01, -4.2541e-02,  1.6437e-01,  5.8836e-01, -4.4367e-01,\n",
       "           3.7472e-02,  2.5403e-01, -1.6017e-01, -2.0907e-01,  1.4633e-01,\n",
       "           3.4590e-01, -4.7802e-02, -6.3576e-01, -8.5078e-02,  9.3202e-01,\n",
       "           3.9292e-01,  1.6623e-01, -4.3080e-01,  2.4293e-01,  3.0721e-01,\n",
       "          -4.2984e-01, -7.9686e-02, -1.9869e-01, -2.7917e-01, -8.8013e-02,\n",
       "           3.2101e-01,  4.3052e-01, -1.5070e-01, -1.2228e-01,  2.0489e-01,\n",
       "           3.0597e-02,  2.2097e-02, -1.8611e-01, -7.2410e-03,  6.5900e-01,\n",
       "          -2.0787e-01, -4.7221e-01,  4.4904e-01, -6.7167e-01, -1.5659e-01,\n",
       "          -7.5371e-02, -3.0595e-02, -1.8847e-01, -1.5406e-01, -2.4712e-01,\n",
       "          -2.2923e-01,  2.7438e-02,  6.8617e-01,  3.2027e-01, -1.2938e-01,\n",
       "           1.5307e-01,  2.2175e-01, -1.2626e-01,  7.2378e-01,  2.2572e-01,\n",
       "          -1.8592e-01,  2.3615e-01, -6.9134e-01,  8.5598e-02,  1.6357e-01,\n",
       "          -4.1352e-01, -3.5142e-01,  3.9786e-02,  7.8673e-01, -1.8162e-02,\n",
       "          -6.2733e-01, -4.1998e-02,  1.8839e-01,  2.9326e-02,  2.0055e-01,\n",
       "          -3.7057e-01,  5.5612e-01,  4.7868e-02,  3.3724e-01, -4.8301e-01,\n",
       "           7.3917e-02, -1.8208e-01,  1.0081e-01,  2.3507e-01, -5.8684e-01,\n",
       "           6.8846e-01,  1.8171e-02,  3.0868e-01,  3.6056e-01, -1.0981e+01,\n",
       "           1.9548e-01,  6.2176e-01, -4.0806e-01,  8.9273e-02,  8.8253e-02,\n",
       "           1.1848e-01, -7.4201e-02,  3.0092e-01, -4.0314e-01,  2.6531e-02,\n",
       "          -3.8864e-01, -6.0716e-02,  2.1701e-01, -9.2364e-02, -4.0688e-01,\n",
       "           2.1582e-01, -7.2858e-01,  2.3026e-02,  1.4132e-01,  2.2466e-01,\n",
       "          -4.1171e-01,  2.0438e-01,  3.1403e-01, -2.7418e-01, -4.0017e-01,\n",
       "           8.7779e-04, -3.6673e-01,  7.5996e-02,  2.7240e-01,  2.1456e-01,\n",
       "           3.1656e-01, -6.1420e-02,  8.7563e-01, -1.6723e-02, -4.0798e-01,\n",
       "           1.7506e-02, -3.0325e-02,  2.3538e-01,  5.6081e-02,  5.0976e-02,\n",
       "          -4.0188e-01, -3.9291e-02,  1.5695e-01,  3.2895e-01,  6.7455e-01,\n",
       "           1.2158e+00,  5.6031e-01,  2.4425e-02, -4.1029e-01,  2.7569e-01,\n",
       "          -5.6252e-01,  3.3242e-01, -5.8870e-01,  1.5083e-01, -1.5237e-01,\n",
       "           2.3013e-01, -3.8972e-01,  1.8945e-01, -3.2615e-01, -5.7273e-01,\n",
       "           8.2255e-02,  3.2073e-01,  8.5880e-02, -6.0694e-01, -2.3082e-01,\n",
       "           1.2251e-01, -3.7285e-02, -2.8389e-02, -3.7912e-01,  2.1318e-01,\n",
       "           3.5148e-01, -4.6965e-01,  1.5702e-01, -3.0825e-01,  5.8103e-01,\n",
       "          -4.2275e-01,  1.1750e-02,  8.5507e-01, -2.0872e-01,  1.9382e-01,\n",
       "           1.2210e-01,  4.0394e-02, -5.6742e-01, -7.7297e-02, -5.1800e-01,\n",
       "          -1.5532e-01, -3.9270e-02,  1.5687e-02,  7.1719e-01, -3.9348e-01,\n",
       "          -3.9643e-01,  1.8460e-01, -2.1787e-01,  5.2026e-01,  8.6583e-01,\n",
       "          -6.4814e-01,  8.9131e-02,  3.9549e-01,  6.2723e-03, -9.5954e-01,\n",
       "           2.3032e-01,  3.3410e-02, -4.4449e-01,  7.8823e-03, -4.5750e-02,\n",
       "           1.0772e-01, -1.1281e-01, -3.0088e-01,  4.5831e-01, -2.2711e-01,\n",
       "           2.0132e-01,  3.4893e-02, -3.0519e-02, -3.0273e-01, -1.0689e+00,\n",
       "          -4.1833e-02,  5.2289e-01, -5.0164e-01, -3.9944e-01,  2.6955e-01,\n",
       "          -1.3588e-01,  3.5972e-03, -1.8982e-01, -1.6044e-01,  2.0773e-01,\n",
       "           5.4384e-02,  1.8866e-01,  2.9435e-01, -3.0142e-02,  3.3127e-01,\n",
       "           9.2193e-02, -3.6075e-01,  3.3493e-01, -6.6182e-01, -6.4631e-01,\n",
       "          -6.9078e-01, -3.1695e-01,  4.4485e-01, -7.3665e-02, -6.1765e-02,\n",
       "           6.7367e-01, -1.7916e-01, -4.9952e-01,  3.0591e-01,  2.9925e-01,\n",
       "           2.2893e-01,  5.5048e-01, -2.4656e-01,  2.8411e-01, -3.7066e-02,\n",
       "           6.4284e-02, -3.1816e-01, -4.4429e-01, -2.9599e-01, -1.6653e-01,\n",
       "          -1.6863e-01, -1.1019e-01, -5.9013e-01, -8.3780e-02, -3.0230e-01,\n",
       "           3.5633e-01, -1.3958e-03, -3.0046e-01,  6.0403e-01,  5.3443e-01,\n",
       "           1.7596e-01, -6.1340e-01,  1.3219e-02,  2.1672e-02, -1.1161e-02,\n",
       "           5.1506e-01,  2.8152e-02, -7.7703e-01, -8.4371e-01, -2.4783e-01,\n",
       "          -1.9881e-01,  1.3098e+00, -2.1223e-01, -1.9688e-01,  8.0602e-02,\n",
       "          -1.5591e-01,  4.0742e-01,  8.5768e-03, -2.2756e-01,  3.0834e-01,\n",
       "           1.7988e-02, -6.4420e-01,  6.1896e-02, -3.1963e-01,  3.0598e-01,\n",
       "           2.6186e-01,  1.0479e-01, -3.7684e-01,  2.7401e-01, -1.2157e-03,\n",
       "          -3.1018e-01, -1.7041e-01, -1.4505e-01, -2.5746e-01,  7.2362e-01,\n",
       "           4.3944e-02,  6.0032e-01, -6.3478e-02,  2.2753e-01, -2.7276e-01,\n",
       "           4.2720e-01,  1.5991e+00, -2.9401e-02, -4.2660e-01,  6.7875e-01,\n",
       "           3.1775e-01,  3.1570e-01,  2.5361e-01, -1.8306e-01,  3.2815e-01,\n",
       "           3.3132e-01, -1.4334e-02, -2.3533e-01,  4.0416e-01,  1.6370e-01,\n",
       "           7.4526e-01, -3.8402e-01, -1.7872e-01, -9.7541e-03,  1.7489e-01,\n",
       "          -1.7116e-01,  5.4773e-01, -1.6502e-02,  3.2928e-01,  5.0696e-01,\n",
       "          -9.8529e-01, -5.1904e-02,  1.3333e-01, -2.2445e-01, -1.4725e-01,\n",
       "          -7.3278e-01, -3.3817e-01, -1.2718e-01, -3.8663e-01,  1.7473e-01,\n",
       "           6.7160e-01, -1.5736e+00,  4.9636e-01,  6.3931e-01, -5.2644e-01,\n",
       "           3.2770e-01,  1.7166e-01,  4.7610e-01,  2.0851e-01,  1.5689e-01,\n",
       "          -1.6886e-01,  2.6337e-01, -6.5471e-01, -3.9883e-01,  1.0937e-01,\n",
       "           7.1385e-01,  1.0299e-01, -1.2860e-01,  3.1453e-01, -1.8828e-01,\n",
       "          -6.0618e-02,  1.9600e-01, -3.4978e-01, -1.2914e-01,  7.3132e-02,\n",
       "           2.8797e-02, -8.1082e-01,  1.2588e-01,  3.5356e-01, -7.0832e-02,\n",
       "           3.8920e-01,  8.5013e-02,  1.1705e-01, -3.6104e-01, -4.1991e-03,\n",
       "          -2.7827e-02,  5.4501e-01, -1.3692e-02,  6.1384e-01, -2.1803e-01,\n",
       "          -4.0495e-01, -1.3093e-01, -3.4807e-02, -7.8228e-01,  2.0931e-01,\n",
       "          -2.0884e-01, -2.6999e-02, -4.5935e-02, -2.3986e-01,  5.9544e-01,\n",
       "          -1.8135e-02,  1.9538e-02, -5.2174e-01,  3.0249e-01,  2.7618e-01,\n",
       "          -1.0090e-02,  1.1895e+00, -5.9010e-02,  3.4057e-01, -6.3226e-01,\n",
       "          -5.5014e-02, -8.4602e-01, -1.6363e-01,  2.1247e-01, -4.9331e-01,\n",
       "          -1.0017e-01, -2.8914e-01, -4.2424e-01,  1.9530e-01,  7.9963e-01,\n",
       "           5.6256e-01, -3.0861e-01, -4.0141e-01, -4.2430e-01, -3.6983e-03,\n",
       "           4.6512e-01, -9.5207e-02,  7.1806e-01,  4.5355e-01, -1.2693e-01,\n",
       "          -9.3142e-01, -5.2621e-01,  2.1018e-01,  4.2585e-01, -3.0087e-01,\n",
       "           1.1394e+00,  1.4304e-02, -3.4133e-01,  2.8189e-01,  1.8562e-01,\n",
       "           4.2663e-01,  5.4107e-01, -2.7467e-01, -7.5543e-01,  1.3963e-01,\n",
       "           6.7393e-01, -1.3836e-01, -7.9744e-02, -6.9770e-01,  3.6309e-01,\n",
       "           1.4811e-01,  1.3678e-01,  5.2915e-01]], device='mps:0'),\n",
       " tensor([[-0.2258, -0.7099, -0.3778,  ...,  0.3292,  0.7059,  0.7494],\n",
       "         [-0.0932, -0.6293, -0.4879,  ...,  0.3041, -0.1958,  0.5576],\n",
       "         [-0.3292, -0.2590, -0.5016,  ..., -0.0148,  0.3509,  0.1607]],\n",
       "        device='mps:0'),\n",
       " tensor([[-0.3402, -0.6569, -0.3442,  ...,  0.2599,  0.7381,  0.5866],\n",
       "         [-0.0238, -0.5721, -0.8340,  ...,  0.0483,  0.8204,  0.5286]],\n",
       "        device='mps:0'),\n",
       " tensor([[-3.0855e-01, -8.1179e-01, -2.2475e-01, -1.0202e+00, -1.2604e-01,\n",
       "          -1.7336e-01, -5.7341e-01,  4.1989e-01,  9.4276e-02,  9.4136e-02,\n",
       "           2.2167e-01,  1.0626e-01, -1.2932e-01, -8.2596e-02, -1.0037e+00,\n",
       "          -2.1709e-01, -6.2486e-01,  8.8827e-02,  2.7018e-01,  7.7706e-01,\n",
       "           5.1988e-01,  2.3047e-01, -8.1455e-02, -9.7506e-02, -7.4013e-01,\n",
       "           3.5217e-01, -1.6234e-01,  1.7674e-01,  2.6675e-01, -3.4875e-01,\n",
       "          -7.3235e-02,  4.7696e-01, -2.9592e-01,  8.2859e-01, -1.0140e-01,\n",
       "          -3.0455e-01,  9.8932e-02, -3.4365e-01,  6.5861e-01, -4.0813e-01,\n",
       "           2.5783e-01, -1.5124e-02,  1.4746e-01,  6.2785e-02,  3.8825e-02,\n",
       "          -1.4582e-01,  5.5868e-02, -3.9838e-01, -2.2679e-01, -1.5478e-01,\n",
       "           1.1179e-01,  2.1258e-02,  4.1128e-01,  4.6110e-01, -8.0856e-03,\n",
       "          -3.9480e-01,  1.2036e-01,  1.9370e-01, -2.3575e-01,  2.5765e-01,\n",
       "          -3.8181e-01,  8.8333e-01,  3.9113e-01, -8.7849e-02,  3.4229e-01,\n",
       "           3.6113e-02, -2.9526e-01, -5.4803e-01, -4.8511e-01,  1.2902e-01,\n",
       "           5.2382e-02,  1.3842e-01,  2.4680e-01, -3.1720e-01,  4.7976e-01,\n",
       "           8.4799e-02,  1.6137e-01, -1.2349e-01, -6.0568e-01, -9.1476e-02,\n",
       "           2.9950e-01, -8.4846e-02,  3.3713e-01, -2.4135e-02, -2.1264e-01,\n",
       "          -1.5468e-01,  2.6548e-01,  1.6198e-01,  2.1175e-01, -1.5286e-02,\n",
       "           3.9710e-01,  2.8269e-02, -6.6345e-02, -4.8443e-01,  1.2234e-01,\n",
       "          -6.6041e-02,  3.1102e-01, -3.7728e-01, -1.4195e-01, -4.3358e-01,\n",
       "           1.6451e-01,  3.2336e-01, -4.0994e-02,  4.0121e-01, -6.6497e-01,\n",
       "          -5.9882e-01,  7.3968e-02,  2.8082e-01, -6.7694e-01, -1.6059e-01,\n",
       "          -6.1282e-02,  5.0204e-02,  8.3287e-02, -3.2085e-01, -8.8887e-02,\n",
       "           2.0767e-01, -4.0037e-01,  1.1030e-01,  1.7806e-01,  7.7353e-01,\n",
       "           5.0987e-01, -4.6032e-01,  9.6902e-02,  9.5484e-02,  2.4345e-01,\n",
       "          -1.4969e-01,  4.0365e-01,  1.2683e-02,  6.7275e-01,  4.0627e-01,\n",
       "           1.4862e-01,  6.9763e-02, -3.8122e-01, -4.3848e-01, -2.8998e-02,\n",
       "          -1.3522e-01,  1.5099e-01, -7.2227e-01,  4.2154e-01,  7.0089e-02,\n",
       "          -1.3813e-01,  3.0549e-01, -2.9617e-01, -7.7176e-02, -1.0704e-01,\n",
       "          -1.8309e-01,  6.8896e-03, -6.5759e-02, -3.0588e-01, -2.5429e-01,\n",
       "          -3.2357e-01, -8.3219e-01,  3.7551e-01, -7.9799e-02,  2.0221e-01,\n",
       "          -4.0343e-01,  6.4243e-01,  1.7710e-02, -2.7932e-02, -2.8966e-02,\n",
       "           6.7149e-01,  1.0538e-02,  2.9439e-01, -2.7324e-01, -8.8069e-02,\n",
       "           2.0286e-01, -2.9393e-01, -9.2310e-02,  1.2824e-01,  6.7058e-02,\n",
       "           2.9614e-01, -1.8304e-01,  1.6753e-01, -2.4729e-01,  2.4346e-01,\n",
       "          -2.0808e-01,  3.5501e-01, -3.0988e-02, -2.3653e-01,  1.5852e-01,\n",
       "           5.3579e-02, -1.7586e+00,  2.3947e-01, -8.6657e-02, -1.3043e-01,\n",
       "          -9.8979e-02,  1.8413e-01,  2.9083e-01,  1.9451e-01, -1.5242e-01,\n",
       "           3.3732e-02,  2.1215e-01,  2.1063e-01,  2.0090e-01, -4.2621e-02,\n",
       "           3.9713e-01,  6.5675e-01,  2.3050e-01,  1.5143e-01, -7.4119e-02,\n",
       "          -5.8705e-01,  1.5523e-01,  4.1150e-01,  3.1379e-01,  3.1536e-01,\n",
       "           2.3829e-01, -3.3101e-01,  2.0491e-01,  1.5893e-01,  1.0608e-01,\n",
       "           9.1054e-02, -2.4556e-01, -3.9617e-01, -1.4280e-01, -2.0344e-01,\n",
       "           1.5480e-01, -3.0242e-01, -2.1508e-01,  3.9734e-01,  1.4958e-01,\n",
       "          -2.0924e-01,  2.5299e-01,  8.6183e-03, -1.2773e-01, -1.5181e-01,\n",
       "           5.1319e-01, -2.4999e-01,  1.3232e-01, -2.6446e-01, -3.4122e-01,\n",
       "           1.0862e-01, -5.1978e-01,  9.0444e-01,  1.4701e-01, -4.5464e-01,\n",
       "          -2.2118e-01, -3.9554e-01,  2.7506e-01,  1.1258e-01, -7.3870e-02,\n",
       "          -1.4794e-01, -1.4506e-01, -4.7730e-01,  1.9289e-01,  1.7138e-01,\n",
       "           3.7552e-01,  3.7850e-02,  5.2675e-01,  4.9594e-01,  4.9494e-01,\n",
       "          -2.3805e-01, -2.1427e-01,  5.7157e-01,  2.9248e-01, -1.3156e-01,\n",
       "           9.6560e-02, -1.3418e-01, -4.2601e-01,  3.3081e-01, -6.2632e-01,\n",
       "           3.0925e-01, -3.2398e-02,  1.2460e-01,  4.4237e-01,  8.2625e-02,\n",
       "          -6.9166e-03,  1.0197e-02, -1.9287e-01, -3.7185e-01,  5.5287e-01,\n",
       "          -2.4285e-01,  4.8020e-01,  4.9236e-01, -6.8721e-01, -4.7936e-01,\n",
       "          -8.4352e-01, -1.2012e-01,  2.7656e-02, -1.3229e-01, -6.3859e-01,\n",
       "          -1.5503e-01,  3.9624e-02, -1.3877e-02, -7.3112e-02,  2.6137e-01,\n",
       "           5.6699e-01,  1.1844e-01, -1.7418e-01, -2.4684e-01, -4.4985e-01,\n",
       "          -1.8662e-01, -7.0934e-02, -1.6096e-01,  1.5922e-01,  1.5279e-01,\n",
       "          -1.8183e-01,  2.5638e-02,  6.0046e-02,  2.0820e-01, -1.6062e-01,\n",
       "          -3.6242e-01, -1.5776e-01, -2.3484e-01, -4.2714e-01, -1.5907e-01,\n",
       "           2.6608e-01,  2.0170e-01,  5.5239e-01,  5.0467e-02, -2.5673e-01,\n",
       "          -2.4499e-02,  2.7058e-01, -1.4327e-01, -5.3251e-01,  1.0001e-01,\n",
       "          -1.2465e+00,  1.6170e-01, -2.4204e-01, -7.5852e-02, -4.7097e-01,\n",
       "           1.4778e-01,  2.0794e-01, -2.2422e-01, -8.5152e-02, -4.9650e-01,\n",
       "           8.4130e-01,  6.5322e-01,  1.0956e-01,  1.2090e-02, -3.1300e-01,\n",
       "           1.9399e-03, -6.2738e-01, -6.5597e-02, -1.3262e-01,  2.6829e-01,\n",
       "           3.3541e-02,  1.4481e-01, -5.3644e-02, -8.2436e-02, -2.2029e-01,\n",
       "           2.5692e-01, -4.6781e-01,  1.5208e-01,  3.9622e-01, -3.4249e-01,\n",
       "           1.4894e-01,  4.1276e-01, -6.1095e-01, -2.4045e-01,  2.7816e-01,\n",
       "           2.4468e-01, -4.1079e-01, -4.1355e-01,  3.1398e-02,  3.3494e-01,\n",
       "           5.1038e-01, -5.9354e-03, -4.0925e-01,  3.4717e-01,  3.9085e-01,\n",
       "          -4.6282e-01,  8.5203e-02,  2.6018e-01, -3.9848e-01, -1.4736e-01,\n",
       "           2.1545e-01,  3.1876e-02, -5.3557e-01,  1.4587e-02, -8.0968e-02,\n",
       "           2.7254e-01, -4.5776e-01, -2.9100e-01,  2.1015e-01,  4.7869e-01,\n",
       "          -2.6196e-01, -6.8840e-01,  2.7823e-01, -4.9868e-01, -1.4379e-01,\n",
       "          -1.6855e-01, -2.4020e-01,  1.6901e-01,  1.5457e-01, -4.9775e-01,\n",
       "          -8.3055e-02, -9.9441e-02,  5.8705e-01,  7.3100e-01, -5.3460e-01,\n",
       "           3.1729e-01, -2.3862e-01, -6.1459e-01,  6.0057e-01, -5.3665e-01,\n",
       "          -2.6939e-01,  2.8216e-01, -2.6159e-01, -1.1950e-01,  5.0670e-01,\n",
       "          -3.1431e-01, -2.4603e-01,  2.7503e-01,  3.1390e-01,  9.7072e-02,\n",
       "          -5.1686e-01,  8.0266e-02,  1.5053e-01, -1.7726e-01,  4.2073e-01,\n",
       "           9.7919e-02,  2.5612e-01, -3.4652e-01,  4.4437e-01, -5.1821e-01,\n",
       "          -2.0899e-02, -3.1622e-01, -9.7144e-02,  2.2790e-01, -3.3110e-01,\n",
       "           3.5769e-01,  1.6178e-01,  4.1852e-02,  2.7648e-01, -1.1391e+01,\n",
       "           1.9078e-01,  5.6440e-01, -7.1916e-01, -1.9422e-01, -2.1765e-01,\n",
       "           6.6553e-01, -3.0532e-01,  2.9878e-01, -3.9702e-01, -2.5976e-01,\n",
       "          -4.2122e-01, -6.5427e-02,  1.4699e-01, -4.0905e-01, -4.4877e-02,\n",
       "          -3.1118e-01, -1.0571e+00,  2.2849e-02,  1.3100e-01, -1.1036e-01,\n",
       "           5.0282e-03, -2.7446e-03,  2.3212e-01,  5.8937e-03, -1.5064e-01,\n",
       "          -2.4650e-01, -8.0066e-02,  3.2989e-01,  1.3123e-01,  9.5475e-03,\n",
       "           2.2105e-01,  5.0224e-02, -4.0655e-02, -8.9038e-02, -2.6007e-01,\n",
       "           6.8495e-02,  6.0737e-02,  2.6159e-02, -7.0163e-01, -3.6932e-01,\n",
       "          -4.9766e-01, -4.6205e-01,  2.1906e-01,  4.4679e-01,  7.8106e-01,\n",
       "           1.5987e+00,  8.2906e-01,  7.8526e-02,  1.0597e-01, -1.1034e-01,\n",
       "          -3.7191e-01,  1.9042e-01, -1.0822e-01, -2.7262e-02, -3.0221e-01,\n",
       "           2.4172e-01, -9.1402e-01,  1.7741e-01, -3.1460e-01, -3.8991e-01,\n",
       "           3.5424e-01,  3.3579e-01, -1.2647e-01, -3.4389e-01, -5.7171e-01,\n",
       "           3.8539e-01,  1.6782e-01, -2.6260e-02, -5.5287e-02,  4.0880e-01,\n",
       "          -3.5947e-01,  6.7098e-01,  3.6750e-02, -3.4888e-01, -1.4385e-01,\n",
       "           4.8871e-02,  5.2934e-02,  1.3672e+00, -2.9531e-01, -1.0166e-01,\n",
       "           2.3942e-02,  3.7099e-01, -2.8096e-01, -4.6566e-01,  4.1967e-01,\n",
       "          -4.3187e-02,  8.8658e-02,  1.6894e-01,  2.2740e-01, -1.2672e-02,\n",
       "          -8.1239e-01, -8.0687e-02, -1.2550e-01,  3.9118e-01,  9.3490e-01,\n",
       "          -6.8148e-01,  3.4722e-02,  2.0540e-01,  9.0745e-04, -4.4686e-01,\n",
       "          -1.6672e-01, -4.5728e-02, -3.3620e-01, -3.1753e-03, -4.1709e-01,\n",
       "          -4.3794e-02,  1.7046e-01, -9.9999e-02,  5.1683e-01, -6.2680e-01,\n",
       "           1.2128e-01,  5.2484e-02,  8.1570e-02, -3.6155e-01, -1.2660e-02,\n",
       "           3.7741e-01,  6.4686e-01, -1.0938e-01, -2.7898e-01, -2.1201e-01,\n",
       "           1.0396e-01,  6.3991e-03,  7.4379e-02, -2.5973e-01,  2.8947e-01,\n",
       "          -6.1108e-01, -2.8985e-01,  5.1368e-02, -3.2827e-03,  2.3241e-02,\n",
       "           1.8584e-02, -5.3897e-01,  3.6400e-01, -3.2112e-01, -6.3143e-01,\n",
       "          -9.7100e-01, -2.2077e-01, -2.4016e-02, -3.1826e-01, -4.1043e-01,\n",
       "          -1.0984e-01,  4.6880e-03,  1.6210e-02,  1.9340e-01, -4.8602e-02,\n",
       "           2.5376e-01,  1.2962e-01, -1.2063e-01,  1.7568e-01, -1.0460e-01,\n",
       "          -1.2011e-01, -2.5323e-01, -9.1673e-02, -5.6517e-02, -5.7411e-01,\n",
       "          -7.4443e-01, -7.2219e-02, -1.5611e-01,  1.9500e-02, -2.4199e-01,\n",
       "           2.8195e-02, -2.6314e-01,  5.6062e-02,  2.1686e-01,  4.8255e-01,\n",
       "           1.8861e-01, -8.6256e-01, -8.1050e-03,  7.4207e-02,  2.8489e-01,\n",
       "           1.3621e-01,  1.5313e-01, -3.6861e-01, -1.8177e-01, -3.8214e-01,\n",
       "           1.0454e-01,  1.1149e+00,  8.3709e-03, -4.2996e-01,  1.4694e-01,\n",
       "           7.3450e-01, -1.7128e-02, -4.1052e-02, -7.9043e-02,  1.4908e-01,\n",
       "          -1.4954e-01, -4.7344e-01,  5.0339e-01,  1.0046e-01,  2.9176e-01,\n",
       "           7.4660e-01,  5.4710e-02,  4.3326e-02,  3.1106e-01,  2.1028e-01,\n",
       "          -2.2689e-01,  3.1935e-01, -3.6739e-01,  3.9624e-01,  1.5704e-01,\n",
       "          -2.2700e-01,  1.1531e-01, -2.7234e-01, -1.0716e-01,  8.8371e-02,\n",
       "           1.6709e-01,  1.5184e+00, -2.0111e-01, -5.9224e-01,  6.6605e-01,\n",
       "          -2.1598e-01,  4.2741e-01,  5.5500e-01, -1.4962e-01,  2.9101e-01,\n",
       "           3.0731e-01, -1.4312e-01, -2.8876e-01,  4.7703e-01,  1.6214e-01,\n",
       "           3.1376e-01, -1.1406e+00, -1.7642e-01, -3.6796e-02,  7.3335e-01,\n",
       "           7.8514e-02,  4.1366e-01,  1.9424e-01,  5.2248e-01,  7.8549e-01,\n",
       "          -1.1011e+00,  2.3370e-01, -4.2985e-01,  6.2974e-01,  1.3390e-01,\n",
       "          -2.9164e-01, -1.2276e-01, -2.2452e-01, -1.6232e-01,  7.0307e-02,\n",
       "           3.3178e-01, -1.2158e+00, -9.8158e-02,  4.5132e-01,  1.9518e-01,\n",
       "           2.2965e-01, -1.9284e-01,  5.8892e-01,  7.4989e-02, -6.3416e-02,\n",
       "          -9.5876e-02,  2.0852e-01, -4.6612e-01, -1.9113e-01,  3.5412e-01,\n",
       "           6.3764e-01,  2.1086e-01, -7.1880e-02, -2.8793e-02, -1.3534e-01,\n",
       "          -1.3625e-01, -2.1309e-01, -2.3898e-02, -3.8850e-03, -3.9534e-01,\n",
       "          -8.1321e-02, -4.0423e-01, -5.2088e-01,  1.6922e-01, -2.6253e-01,\n",
       "           1.5015e-01,  4.0068e-01,  9.3850e-01, -8.3437e-04,  1.6460e-01,\n",
       "          -3.9125e-01,  4.2966e-01, -2.6744e-01,  5.1850e-01, -4.2858e-02,\n",
       "          -9.9159e-02,  2.5804e-01,  3.4216e-01,  1.3307e-01,  2.6829e-01,\n",
       "          -2.1920e-01,  3.6879e-01, -2.2424e-01, -2.5498e-01,  3.1192e-01,\n",
       "          -1.7267e-01,  1.6017e-01, -3.7315e-01,  1.4248e-01,  7.5563e-02,\n",
       "          -2.2030e-01,  7.5743e-01, -5.1219e-02,  6.8977e-01, -1.0856e+00,\n",
       "           7.1363e-02, -8.3908e-01,  1.5210e-01, -4.0897e-01,  4.4041e-02,\n",
       "          -4.9725e-01, -1.9316e-02,  1.6025e-02,  1.6086e-01,  5.6211e-01,\n",
       "           4.6843e-01, -1.4194e-01, -6.2611e-01, -1.5645e-01,  1.6809e-01,\n",
       "           2.3640e-02, -9.3483e-02,  2.8561e-01, -1.4954e-01,  4.3094e-01,\n",
       "          -3.5045e-01,  9.4879e-02, -1.7097e-03, -3.3614e-02, -3.8821e-01,\n",
       "           7.1436e-01,  8.2203e-02, -1.1916e-01,  4.1724e-01,  5.4348e-02,\n",
       "           6.3345e-01,  7.4298e-01,  6.0054e-02, -4.6779e-01, -2.0175e-02,\n",
       "          -6.0227e-02, -1.5731e-01,  4.3378e-02, -3.9580e-01,  5.1294e-01,\n",
       "           1.4066e-02,  3.7564e-01,  7.1131e-01]], device='mps:0'),\n",
       " tensor([[-0.3188, -0.6431, -0.4880,  ...,  0.2684,  0.4995,  0.4356],\n",
       "         [-0.5999, -0.5807, -0.4463,  ...,  0.1744,  0.6115,  0.2542]],\n",
       "        device='mps:0'),\n",
       " tensor([[-0.2841, -0.9073, -0.8214,  ...,  0.1888,  0.6933,  0.5187],\n",
       "         [-0.2643, -0.8360, -0.8564,  ...,  0.1610,  0.5643,  0.6620],\n",
       "         [-0.2170, -0.9206, -0.6892,  ...,  0.0033,  0.6907,  0.6555]],\n",
       "        device='mps:0'),\n",
       " tensor([[-2.7585e-02, -3.9063e-01, -4.8392e-01, -6.2694e-01, -5.9994e-01,\n",
       "          -3.2353e-01, -4.6797e-01,  4.3803e-01,  1.6166e-02,  3.2834e-01,\n",
       "           3.2125e-01,  3.9826e-01, -5.7934e-02, -5.2507e-02, -7.1311e-01,\n",
       "          -3.4204e-01, -4.5561e-02,  1.6796e-01,  4.6251e-01,  5.9007e-01,\n",
       "           8.6259e-01,  2.0896e-01, -7.0482e-02, -6.2825e-02, -9.5342e-01,\n",
       "           3.5864e-01, -9.0412e-02,  1.3053e-01,  1.9639e-01, -8.9935e-01,\n",
       "          -2.2307e-01,  4.1714e-01,  2.1464e-01,  7.8188e-01, -2.1905e-02,\n",
       "          -3.1625e-01,  2.9418e-01, -5.9480e-01,  4.3011e-01, -1.7024e-01,\n",
       "           2.6176e-01,  2.8424e-02,  2.1719e-01,  6.3108e-02,  4.1889e-01,\n",
       "          -7.8141e-02,  5.5180e-02, -5.5545e-01,  2.0304e-02, -2.8300e-01,\n",
       "           7.9487e-02, -1.6844e-01,  2.0530e-01,  2.6901e-01, -4.9948e-01,\n",
       "          -8.4785e-01,  3.6903e-01,  1.6899e-01, -1.5039e-01,  1.3569e-01,\n",
       "          -3.0465e-01,  9.1226e-01,  1.0661e-02,  7.9637e-02, -2.3900e-02,\n",
       "          -4.7938e-02, -5.7584e-03, -7.2791e-01, -6.8840e-01, -3.3221e-01,\n",
       "          -9.9254e-02,  2.9794e-01,  1.0683e-01, -1.2494e-01,  5.4996e-01,\n",
       "          -2.1630e-01,  5.7656e-02,  2.1396e-02, -2.5048e-01, -7.4785e-02,\n",
       "           8.2719e-02,  1.5800e-01,  4.7315e-01,  1.2002e-01, -6.3504e-01,\n",
       "           1.7343e-02,  5.1519e-01,  5.7378e-01,  2.5232e-01, -2.5374e-01,\n",
       "           3.1294e-01,  8.9331e-02, -2.3767e-01, -5.0756e-01,  3.4777e-01,\n",
       "          -2.2199e-01,  1.3958e-01, -2.6880e-01,  9.9738e-02, -3.8764e-01,\n",
       "           4.9480e-02,  4.5284e-01, -1.2413e-01,  4.9302e-01, -4.6188e-01,\n",
       "          -8.8358e-01,  3.9797e-01, -2.6192e-01, -7.7036e-01, -1.9703e-01,\n",
       "          -2.0517e-01, -1.7220e-01,  3.7803e-01, -4.0832e-01, -4.3464e-02,\n",
       "          -8.0410e-03,  7.3579e-02, -9.3704e-02, -4.8022e-01,  1.4676e-01,\n",
       "           1.7159e-01, -4.3731e-01, -6.6994e-02, -1.6144e-01,  6.7626e-02,\n",
       "           2.6470e-01,  1.9059e-01,  4.7652e-02,  4.1379e-01,  7.1148e-01,\n",
       "           8.7890e-02,  3.6182e-01, -4.0594e-01, -2.9439e-01,  1.4099e-01,\n",
       "           2.0904e-01, -1.6663e-01, -1.6108e-01,  1.9942e-01,  8.1561e-02,\n",
       "          -1.8159e-01,  6.8842e-02, -1.0493e-01, -8.2199e-02, -1.6785e-01,\n",
       "          -2.2175e-01,  3.4212e-03, -4.5861e-01, -2.2856e-01, -7.9578e-02,\n",
       "          -4.4475e-01, -6.0378e-01,  3.0636e-01, -1.5638e-01,  5.5722e-02,\n",
       "          -3.2574e-01,  6.8842e-01, -1.8695e-01,  1.7264e-01,  3.2921e-01,\n",
       "           5.2182e-01, -4.4519e-02,  1.3245e-01, -2.2792e-01,  1.0131e-01,\n",
       "           5.7357e-01, -2.2805e-01, -2.0069e-01, -2.4643e-01,  2.0931e-01,\n",
       "           2.1424e-01, -2.5321e-01,  1.6542e-01, -1.4293e-01, -1.8163e-01,\n",
       "          -3.6534e-01,  1.3546e-01,  8.6802e-02, -3.0924e-01,  3.6737e-01,\n",
       "          -6.3740e-01, -1.5787e+00,  2.0753e-01,  2.5194e-02, -9.2997e-03,\n",
       "          -4.7837e-01,  2.4673e-01,  4.4305e-01,  2.8305e-01, -7.9240e-02,\n",
       "          -7.7790e-02,  1.7237e-01,  1.8192e-01,  7.8611e-01, -1.4974e-01,\n",
       "           4.8738e-01,  2.6263e-01, -7.1910e-02,  1.0931e-01,  8.1195e-02,\n",
       "          -5.0097e-01,  1.3100e-01,  9.6385e-02, -3.0994e-01,  1.3286e-01,\n",
       "           1.0469e-01,  3.4373e-01,  6.0348e-02, -4.5915e-04,  1.3794e-01,\n",
       "          -5.1532e-02, -4.7747e-01, -1.7420e-01, -3.2822e-01, -4.3764e-01,\n",
       "           5.0447e-01,  2.8563e-02, -5.4969e-02,  5.2073e-01,  7.7781e-02,\n",
       "          -5.0994e-01,  1.2393e-01, -1.9492e-01, -1.6875e-01, -4.7041e-01,\n",
       "           4.0358e-01,  1.7517e-01, -1.5965e-02, -2.1508e-01, -2.6550e-01,\n",
       "          -2.0331e-01, -5.6667e-01,  7.9018e-01,  2.2286e-01, -1.5437e-01,\n",
       "          -6.9695e-02, -3.9947e-01, -5.1292e-02, -6.0367e-03,  4.0598e-01,\n",
       "          -1.9960e-01, -3.0949e-01, -6.0505e-01, -1.7184e-02,  1.1309e-01,\n",
       "           1.8549e-01, -9.8427e-02,  6.0618e-01,  5.8489e-01,  3.7472e-01,\n",
       "          -8.4345e-03, -1.7504e-01,  4.0710e-01,  3.0915e-01, -1.1014e-01,\n",
       "           4.6230e-01,  6.9809e-02, -3.5735e-01,  3.9675e-01, -3.9907e-01,\n",
       "           3.2025e-01, -1.2463e-02,  3.1433e-01,  4.2985e-01, -5.6252e-01,\n",
       "           3.1427e-02,  3.9044e-01, -5.1340e-01,  6.6708e-02,  4.2297e-01,\n",
       "          -3.8823e-01,  5.4112e-01,  3.3595e-01, -4.2628e-01, -7.0030e-01,\n",
       "          -3.2555e-01,  2.2480e-01, -1.8813e-02, -5.0181e-02, -9.6294e-01,\n",
       "          -2.0085e-01, -4.8655e-03,  7.3543e-02,  2.8233e-01, -3.2052e-01,\n",
       "           4.3347e-02,  2.7998e-01, -4.7371e-01, -3.6505e-04, -6.2940e-01,\n",
       "           7.5457e-02, -2.2679e-02, -1.7923e-01,  2.8217e-01,  2.0162e-01,\n",
       "          -1.8444e-01,  2.2269e-01,  1.7834e-01,  6.3999e-02, -1.7335e-01,\n",
       "          -9.4714e-02, -2.7266e-01,  1.7423e-01, -8.2702e-02, -1.1982e-01,\n",
       "           2.4916e-01,  1.1777e-01,  5.9543e-01, -5.4731e-01, -8.1465e-01,\n",
       "          -1.9489e-01,  7.3686e-01, -5.9451e-02, -5.3675e-01, -3.8031e-02,\n",
       "          -1.1669e+00, -3.4933e-02, -2.4355e-02, -1.4537e-01, -4.4292e-01,\n",
       "           1.4030e-01, -1.7063e-02, -3.3736e-01,  3.5385e-02, -3.4979e-01,\n",
       "           3.1063e-01,  5.2338e-01, -1.3676e-01, -1.9578e-01, -1.6752e-01,\n",
       "          -5.9986e-01, -2.4249e-02, -1.5233e-01, -1.9184e-01,  1.4099e-01,\n",
       "          -1.6351e-01,  2.0890e-01, -1.6961e-01,  1.9157e-02, -2.5337e-01,\n",
       "           3.5094e-01, -4.1486e-01, -1.6244e-01, -2.0385e-02, -1.3084e-01,\n",
       "           5.0858e-01,  5.7896e-01, -4.0386e-01, -3.5639e-01, -9.2947e-02,\n",
       "          -1.0375e-01, -8.9376e-02, -1.0165e-01,  9.8993e-02,  6.8878e-01,\n",
       "           4.3962e-01,  6.3167e-02, -3.1828e-01,  4.9797e-01,  2.7982e-01,\n",
       "          -4.2389e-01,  4.8925e-01,  5.7260e-02, -1.7166e-01, -4.4168e-02,\n",
       "           1.8468e-01,  1.7170e-01, -4.1089e-01, -4.4255e-02,  7.7227e-02,\n",
       "           3.7392e-01, -5.8107e-01,  1.7846e-01,  7.7756e-02,  6.8104e-01,\n",
       "          -2.8369e-01, -4.5798e-01,  3.6711e-01, -3.4862e-01, -2.7103e-01,\n",
       "           1.5438e-01,  1.2889e-02,  2.5843e-01,  1.8045e-01, -2.1640e-01,\n",
       "          -4.0361e-01,  2.7081e-01,  6.8058e-01,  5.2178e-01, -1.3472e-01,\n",
       "           2.6887e-01, -1.0213e-01, -4.3342e-01,  3.6082e-01, -2.3058e-01,\n",
       "          -1.4555e-01,  6.0044e-01, -5.2817e-01,  2.5213e-02,  2.1936e-01,\n",
       "          -2.5746e-01, -1.5081e-01, -2.3014e-01, -1.0220e-01, -2.8282e-01,\n",
       "          -1.8656e-01,  1.4348e-01, -1.6440e-01, -1.6072e-01, -1.3545e-01,\n",
       "           2.5879e-01,  9.9682e-03, -2.1455e-01,  4.6070e-01, -1.8461e-01,\n",
       "           1.4510e-02, -2.7844e-01, -8.6247e-02, -2.2763e-01, -3.0940e-01,\n",
       "           6.9381e-01,  5.0866e-01,  1.6968e-01,  3.7570e-01, -1.1146e+01,\n",
       "          -2.3398e-02,  4.1214e-01, -3.7811e-01, -1.8202e-01, -1.6373e-01,\n",
       "           5.7596e-02, -2.7925e-01,  2.1064e-01, -1.9062e-01, -3.2644e-01,\n",
       "          -8.0995e-02,  1.8994e-01,  2.1452e-01,  1.0059e-01, -4.3806e-01,\n",
       "          -1.5227e-01, -7.6489e-01,  1.5811e-01,  6.5734e-02,  1.8605e-01,\n",
       "          -2.1370e-01,  1.7442e-01,  1.0329e-01, -6.6665e-02,  3.7471e-01,\n",
       "          -2.9368e-01, -7.1669e-02,  4.8388e-01,  4.7795e-01,  1.8866e-01,\n",
       "           3.0877e-01,  3.9043e-01,  4.3791e-02, -4.7252e-01, -5.3060e-01,\n",
       "           1.8056e-01, -4.0949e-01, -1.7332e-02,  1.7099e-01, -3.7020e-01,\n",
       "          -3.2586e-01, -4.7394e-01,  1.8432e-01,  2.0607e-01,  8.7370e-01,\n",
       "           1.2178e+00,  6.0666e-01,  7.4852e-01,  2.8982e-01,  1.0596e-02,\n",
       "          -7.7573e-01,  1.2724e-01, -8.3036e-01, -2.5422e-01, -2.3218e-01,\n",
       "           6.6593e-01, -5.5107e-01, -1.3553e-01, -3.4404e-01, -3.2567e-01,\n",
       "           5.8688e-01,  3.9671e-01, -1.3806e-01, -4.5212e-01, -1.1934e-01,\n",
       "           4.5729e-01, -1.2005e-01,  1.2274e-01, -1.2652e-01, -2.4808e-02,\n",
       "          -2.8217e-01,  6.7522e-01,  2.7834e-01,  2.3072e-01,  5.7324e-01,\n",
       "           4.2928e-01, -8.5029e-02,  1.4821e+00,  1.6614e-02, -2.2189e-01,\n",
       "           6.3856e-02,  2.7369e-02, -4.5352e-01, -7.5801e-01,  3.6667e-02,\n",
       "           4.5797e-01,  2.8320e-01,  1.2206e-01,  1.9248e-01, -1.6230e-01,\n",
       "          -9.7125e-01, -8.7838e-02,  4.6221e-01,  3.6752e-01,  5.7972e-01,\n",
       "          -7.8714e-01, -2.7145e-01,  1.3428e-01, -3.4683e-02, -5.5119e-01,\n",
       "          -5.1915e-02, -3.5800e-01, -2.4701e-01,  1.7706e-01, -1.2351e-01,\n",
       "          -5.5259e-02,  1.7111e-01,  8.8861e-02,  2.5978e-01, -4.7084e-01,\n",
       "           1.8148e-01,  1.5408e-01, -9.6168e-02, -3.0115e-01,  2.2387e-01,\n",
       "          -1.2030e-02,  6.2757e-01, -1.8211e-01, -6.4111e-01, -1.2862e-01,\n",
       "          -1.5051e-01,  2.4377e-02,  5.9725e-02, -5.3722e-02,  5.1353e-01,\n",
       "          -4.2552e-01,  4.6220e-02,  4.5854e-01,  1.7832e-01,  1.6200e-01,\n",
       "           5.6954e-02, -1.6630e-01,  3.7959e-01, -9.2428e-01, -8.8487e-01,\n",
       "          -7.0678e-01,  2.9951e-01,  2.7297e-02, -8.6083e-02, -8.8577e-02,\n",
       "           1.4680e-03,  2.4025e-01, -3.0700e-01, -6.4113e-02,  1.6911e-01,\n",
       "           7.5654e-01,  3.9296e-01, -2.4504e-01,  2.6033e-01, -3.5946e-01,\n",
       "          -7.3982e-01, -3.9643e-01, -1.2665e-03,  6.4547e-02, -4.6022e-01,\n",
       "          -3.3000e-01, -1.9966e-02, -3.0000e-01, -1.3829e-01,  2.5884e-01,\n",
       "          -4.8107e-01,  1.8384e-02,  9.5085e-02,  2.1568e-01,  3.7857e-01,\n",
       "           2.7080e-01, -1.1128e-01,  1.6599e-01, -2.0464e-01,  3.6914e-01,\n",
       "          -9.7721e-02,  7.6962e-02, -7.9609e-02, -1.6315e-01, -1.4009e-01,\n",
       "           3.1846e-01,  8.4803e-01, -1.4726e-01, -3.0295e-01,  3.1487e-01,\n",
       "           4.4041e-01, -2.9085e-01, -3.8815e-01, -1.2236e-02, -4.8580e-01,\n",
       "          -1.5981e-01, -5.9993e-01,  5.0106e-01,  9.3278e-02,  4.5909e-01,\n",
       "           1.6842e-01, -3.2108e-02, -2.1053e-01,  4.3192e-01, -4.4227e-02,\n",
       "          -2.7030e-01,  9.9698e-02, -6.1341e-01, -7.6237e-02,  2.8433e-01,\n",
       "          -1.2990e-01,  1.3680e-02, -2.5672e-01, -1.1860e-01,  1.6678e-01,\n",
       "          -2.4925e-02,  9.6310e-01,  2.2345e-01, -4.6542e-01,  8.4530e-01,\n",
       "           3.5355e-01, -3.5086e-01,  1.0186e-01, -1.4445e-01,  2.5474e-01,\n",
       "           2.9116e-01, -1.9611e-01,  1.6441e-01,  1.5138e-01,  1.4441e-01,\n",
       "           6.2718e-01, -7.6169e-01, -4.9410e-01,  3.5547e-03,  2.6145e-01,\n",
       "          -2.5507e-01,  3.0624e-01, -1.6345e-01,  4.5258e-01,  6.9829e-01,\n",
       "          -1.1490e+00, -1.7989e-02, -8.1588e-01,  5.4204e-01,  2.5138e-01,\n",
       "          -2.7122e-01,  7.9965e-02, -4.2721e-01, -4.1029e-01,  3.8285e-01,\n",
       "           8.4930e-01, -1.2949e+00,  3.8823e-02,  6.4987e-01,  6.5989e-02,\n",
       "           2.0621e-01, -5.6308e-01,  3.9333e-01, -3.5643e-01,  2.6246e-01,\n",
       "          -3.7271e-01,  4.2999e-01, -4.2654e-01, -8.3588e-01,  1.6338e-02,\n",
       "           5.0275e-01,  1.6767e-01, -3.3477e-02, -2.7644e-02,  5.1787e-01,\n",
       "          -2.0877e-01, -3.5865e-01, -2.4842e-01,  3.6456e-01, -3.4185e-01,\n",
       "           1.9222e-01, -4.3717e-01,  6.0184e-02,  2.6684e-01,  3.2882e-02,\n",
       "           1.3684e-01,  5.3412e-01,  1.1407e+00, -2.1076e-01, -2.6513e-01,\n",
       "           1.0499e-01,  4.8446e-01, -2.0606e-02,  4.1878e-01, -2.7832e-01,\n",
       "          -2.7091e-01,  5.0238e-01,  1.9189e-01, -2.5506e-01,  3.8003e-01,\n",
       "          -1.5617e-01,  7.8315e-02,  8.6939e-02, -2.2307e-01,  7.6787e-02,\n",
       "           6.9763e-02,  7.0773e-02, -3.1665e-01,  1.2147e-01,  8.3487e-02,\n",
       "          -1.0643e-01,  9.9131e-01, -3.8230e-01,  5.1800e-01, -5.5296e-01,\n",
       "           2.3252e-01, -6.6094e-01, -8.0300e-02, -3.4554e-01, -4.1414e-01,\n",
       "          -5.7083e-01, -1.9552e-01,  2.0992e-01,  4.1346e-01,  1.0948e+00,\n",
       "           1.4533e-01, -2.7794e-01, -4.5667e-01, -2.6876e-01,  2.2832e-01,\n",
       "          -1.0974e-01, -2.2846e-01,  2.2012e-01,  3.0845e-01,  4.6276e-01,\n",
       "          -8.6517e-01, -3.8296e-01, -2.9797e-02,  2.4665e-01,  2.6298e-02,\n",
       "           7.4106e-01, -7.4733e-02,  2.5578e-01,  2.9200e-02, -1.2527e-01,\n",
       "           1.8653e-01,  3.0950e-01, -3.5835e-01, -4.1046e-01, -1.3589e-01,\n",
       "          -1.6078e-01, -5.4545e-01,  2.0254e-01, -7.9081e-01,  3.4068e-01,\n",
       "           5.8229e-01,  7.6797e-01,  4.8783e-01]], device='mps:0'),\n",
       " tensor([[-0.3402, -0.6569, -0.3442,  ...,  0.2599,  0.7381,  0.5866],\n",
       "         [-0.0238, -0.5721, -0.8340,  ...,  0.0483,  0.8204,  0.5286]],\n",
       "        device='mps:0'),\n",
       " tensor([[-5.0498e-01, -8.3430e-01, -3.9035e-01, -9.1489e-01, -1.1412e-01,\n",
       "          -2.7416e-01, -4.3774e-01,  2.5720e-01, -5.7506e-02,  6.3564e-01,\n",
       "           6.5203e-03,  6.8992e-01,  3.5941e-01, -3.6631e-01, -4.3334e-01,\n",
       "           7.6561e-02,  3.1089e-02,  6.3020e-01, -1.1658e-01,  2.2404e-01,\n",
       "          -3.8032e-03,  4.0520e-01,  1.2402e-01, -4.4674e-02, -3.8886e-01,\n",
       "           7.8080e-02, -1.3295e-02,  1.3022e-01,  4.2596e-01, -4.4068e-01,\n",
       "          -2.1245e-01,  4.9611e-01,  4.7371e-02,  1.1117e+00, -1.2598e-01,\n",
       "          -9.8290e-02,  2.6801e-03,  5.5021e-02,  1.1317e+00, -2.1021e-01,\n",
       "           9.7613e-01,  4.2367e-02,  3.2481e-01,  3.1318e-01,  1.1178e-01,\n",
       "           1.5871e-01,  2.2295e-01, -5.2789e-01,  4.9178e-01, -6.5542e-01,\n",
       "          -7.2335e-02,  2.9676e-02,  8.5483e-01,  3.5055e-01, -1.8184e-01,\n",
       "          -3.3106e-01, -3.9493e-02,  5.5298e-01, -4.3481e-01, -3.8420e-01,\n",
       "           1.0441e-01,  2.5314e-01, -2.3852e-02, -5.6775e-01,  2.9291e-01,\n",
       "          -6.7428e-02, -2.6110e-01, -7.4196e-01, -1.1125e-01, -4.9292e-01,\n",
       "          -1.4316e-01,  5.2645e-01, -1.9067e-01, -4.3503e-01, -1.5450e-02,\n",
       "          -2.5972e-01, -3.2056e-02,  2.7675e-01, -4.4073e-01,  5.4834e-02,\n",
       "           2.7119e-01,  1.1247e-01,  1.5753e-01, -2.6417e-01,  1.2303e-01,\n",
       "           8.7106e-02,  4.1770e-01,  1.1914e-01,  3.2932e-01, -3.8937e-01,\n",
       "           4.4896e-01, -3.2582e-01, -3.5644e-01, -2.9270e-02,  9.4901e-02,\n",
       "          -2.1365e-01,  1.8169e-01, -3.6902e-01,  1.4165e-01, -2.8327e-01,\n",
       "          -6.2157e-02,  2.0843e-01, -2.0174e-01,  9.3673e-02, -9.8631e-01,\n",
       "          -4.7330e-01,  3.0113e-01, -2.3394e-01, -1.0695e+00,  5.8272e-02,\n",
       "           2.1289e-01, -1.8140e-01,  1.4225e-01, -4.2522e-01, -1.1608e-01,\n",
       "           1.1409e-01,  3.9181e-01,  2.6035e-01, -2.2665e-01,  3.8286e-01,\n",
       "           5.6682e-01, -1.1255e-01,  4.9790e-01, -3.3856e-01, -1.5757e-01,\n",
       "          -1.7013e-01,  6.8726e-01, -3.5049e-01,  5.3609e-02,  1.5391e-01,\n",
       "           2.5182e-01,  4.2858e-01, -2.2705e-01, -3.1657e-02,  2.7917e-01,\n",
       "          -1.3912e-02,  3.8807e-01, -3.9680e-01,  5.5233e-01,  4.8544e-01,\n",
       "          -7.8200e-02, -3.4814e-01, -1.5414e-01, -2.3080e-01, -4.6004e-02,\n",
       "          -1.5516e-01, -1.8065e-01, -1.3248e-01, -4.4226e-01, -5.7306e-01,\n",
       "          -1.1466e+00, -5.3731e-01,  5.3914e-01, -3.3543e-01,  2.9201e-01,\n",
       "          -3.3228e-02,  4.8136e-01,  8.9990e-02,  1.7251e-01,  2.0960e-01,\n",
       "           2.6287e-01, -2.8768e-01,  1.5631e-01, -6.2519e-01,  1.8984e-01,\n",
       "          -2.5619e-02, -3.1903e-01, -2.2077e-01,  1.5795e-01,  8.7012e-02,\n",
       "           4.0058e-01, -1.6539e-01,  3.0273e-01, -5.7401e-02,  2.9327e-01,\n",
       "          -3.8626e-01,  2.3773e-01,  2.7607e-01, -3.3245e-01, -3.2944e-01,\n",
       "           2.1244e-01, -1.6600e+00,  7.9328e-02, -3.7093e-01, -3.3959e-01,\n",
       "          -3.6501e-01,  2.8491e-01,  9.2842e-02,  3.0860e-01, -3.0728e-02,\n",
       "           9.6153e-02, -9.8047e-02, -6.4813e-02,  3.3754e-01,  2.5509e-01,\n",
       "           3.5820e-01,  3.9334e-01,  2.2849e-01, -2.8284e-01,  2.0519e-02,\n",
       "          -1.6563e-01,  3.4198e-01,  2.2820e-01,  4.9440e-01,  4.3702e-02,\n",
       "           3.5380e-01,  2.3838e-02,  2.3342e-01,  1.2794e-02,  1.5748e-01,\n",
       "          -2.1518e-01,  5.0471e-02, -3.6959e-01, -1.9120e-01, -3.1928e-01,\n",
       "           3.3143e-01, -2.0152e-01, -3.2353e-02,  7.1980e-01,  1.1265e-01,\n",
       "          -1.0564e-01,  1.0483e-01,  6.6842e-02, -2.7214e-01,  2.7487e-01,\n",
       "           2.8932e-01, -1.9609e-01,  2.0110e-01,  1.4699e-01, -2.2294e-01,\n",
       "          -3.9402e-01, -1.0314e-01,  4.7628e-01,  1.9314e-01, -3.6184e-01,\n",
       "          -1.2636e-01,  2.4543e-01,  2.9551e-01, -3.6204e-02,  1.2832e-01,\n",
       "          -3.3515e-01, -5.1892e-01, -5.0637e-01, -1.5227e-01, -1.6639e-01,\n",
       "           2.2838e-01,  6.0260e-02,  3.7215e-01,  4.9226e-01,  2.7541e-01,\n",
       "          -1.8866e-01,  4.5569e-02,  3.6625e-01,  2.2889e-02, -3.3835e-01,\n",
       "           6.8587e-02,  2.3077e-01, -4.1075e-01,  5.0442e-01, -3.2305e-01,\n",
       "          -4.0844e-01, -2.6284e-01,  5.6305e-01,  1.4596e-01, -2.1857e-01,\n",
       "           6.3686e-01,  5.3464e-01, -1.9691e-01, -3.8720e-01,  4.0936e-01,\n",
       "          -5.2927e-01,  2.4122e-01, -1.4720e-01, -5.1626e-01, -7.4321e-01,\n",
       "          -9.9810e-01, -3.0766e-01,  1.0697e-02,  1.6052e-01, -5.8938e-01,\n",
       "          -2.0105e-01, -2.0384e-01,  7.6087e-02, -1.6304e-01, -1.5542e-01,\n",
       "          -1.4917e-01, -7.9686e-02,  1.0195e-01, -7.1382e-01, -4.1796e-01,\n",
       "          -1.5102e-01,  2.5616e-01, -3.1012e-01, -4.0970e-02, -8.7416e-02,\n",
       "           3.3362e-02,  5.4762e-02,  6.5671e-01, -2.0411e-01, -3.3907e-01,\n",
       "          -8.2102e-02,  3.8275e-01,  9.9005e-03, -3.2851e-01,  4.3266e-01,\n",
       "           3.4133e-02,  2.6331e-01,  1.0049e+00, -9.5262e-01, -2.3921e-01,\n",
       "           1.0991e-01,  6.8675e-01, -1.9712e-01, -1.0031e-01, -2.9344e-01,\n",
       "          -1.1575e+00, -2.8320e-01, -4.8900e-01, -1.0401e-02, -5.0837e-01,\n",
       "           1.9963e-01, -8.1579e-02, -9.9880e-02, -2.6693e-02, -4.1748e-01,\n",
       "           5.5774e-01,  2.2867e-01,  5.2563e-03,  1.2742e-01, -5.2474e-01,\n",
       "          -3.6656e-02, -4.9525e-01, -2.5308e-01, -2.1267e-01,  3.0289e-01,\n",
       "           5.0530e-01,  2.3128e-01, -1.1752e-01,  1.3310e-02, -7.0013e-01,\n",
       "           4.9008e-01,  2.6466e-02,  4.2356e-01,  1.9737e-01, -3.4074e-01,\n",
       "          -2.3351e-02,  3.6226e-03, -7.1503e-02, -2.7102e-01,  2.1903e-01,\n",
       "           4.3581e-01, -1.7988e-01, -5.2385e-01, -1.8639e-01,  7.1591e-01,\n",
       "           2.8199e-01,  6.2237e-02, -1.3031e-01, -1.9746e-01,  6.8108e-01,\n",
       "          -2.2562e-01,  7.3283e-02,  1.3033e-01, -2.8248e-01,  5.9952e-02,\n",
       "           3.1840e-01,  3.1062e-01, -2.5137e-01,  1.7018e-01,  2.9365e-01,\n",
       "          -3.2135e-02,  3.5834e-01,  5.1948e-02, -9.7705e-02,  1.9903e-01,\n",
       "          -1.0620e-01, -4.2971e-01,  6.7083e-02, -5.4237e-01, -2.3910e-01,\n",
       "           2.7401e-01,  9.1861e-02,  1.8405e-01, -4.2254e-01, -4.9825e-01,\n",
       "          -1.2168e-01, -3.1183e-01,  6.4339e-01,  3.9879e-01,  5.2750e-02,\n",
       "           4.3463e-01,  2.9434e-02, -4.6825e-01,  4.4217e-01,  1.5157e-01,\n",
       "          -3.8754e-01,  1.2438e-01, -4.5914e-01, -7.2491e-03, -1.5113e-01,\n",
       "          -4.0070e-01, -2.0301e-01,  2.5994e-01,  4.4374e-01,  9.6775e-02,\n",
       "          -6.6223e-01,  1.0952e-01,  3.8876e-01, -5.1184e-02,  1.3753e-01,\n",
       "           4.2270e-01,  3.1513e-01, -3.7393e-01,  9.0421e-02,  1.8767e-02,\n",
       "           2.8987e-01,  1.2819e-01, -3.0523e-01, -1.0405e-01, -2.0640e-01,\n",
       "           4.4737e-01,  2.2101e-01, -1.2718e-01,  2.1141e-01, -1.0942e+01,\n",
       "           1.1050e-01,  8.5544e-01, -5.2140e-01,  2.2041e-01,  2.1215e-01,\n",
       "          -2.4208e-02, -1.8694e-01,  2.2605e-02, -4.3637e-01, -2.7965e-01,\n",
       "          -5.4986e-01,  5.4270e-02,  8.6123e-02, -2.8027e-01, -3.2054e-01,\n",
       "           1.3498e-01, -7.1350e-01,  1.4063e-01, -4.1788e-01,  8.8659e-02,\n",
       "          -2.1505e-01,  6.9406e-02,  1.9795e-01, -1.2782e-01, -6.8821e-02,\n",
       "          -5.4184e-01, -5.6486e-01,  3.9797e-01,  6.9700e-02,  2.1218e-02,\n",
       "           2.1987e-01, -2.0349e-01,  2.4712e-01, -1.9144e-01, -8.5480e-01,\n",
       "          -1.1313e-01, -3.5680e-01,  6.8929e-01,  6.3573e-02, -1.4380e-01,\n",
       "          -4.7867e-01, -6.0029e-01,  2.3333e-01,  1.4410e-01,  7.1819e-01,\n",
       "           6.4070e-01,  6.5214e-01, -1.2080e-01, -3.9954e-01, -3.0842e-01,\n",
       "          -7.8224e-01,  4.1596e-01, -6.1544e-01,  3.5987e-02, -4.7014e-01,\n",
       "           2.2199e-01, -4.7411e-01, -1.4963e-02, -2.9178e-01, -3.6009e-01,\n",
       "          -1.2351e-01,  2.7617e-01, -1.4225e-01, -2.8036e-01, -2.9837e-01,\n",
       "           4.0815e-01,  8.5596e-02,  3.3934e-02, -1.4840e-01,  1.7110e-01,\n",
       "           2.9119e-01,  1.6177e-02,  4.2397e-02, -3.8482e-01,  6.5238e-01,\n",
       "          -2.2675e-01, -9.9727e-02,  1.3553e+00, -2.4277e-01,  1.8430e-01,\n",
       "          -1.1621e-01,  2.8511e-01, -9.0702e-02, -3.9775e-01,  3.7083e-02,\n",
       "          -7.4916e-02,  2.9671e-01, -3.3179e-02,  8.8350e-01, -2.8382e-01,\n",
       "          -3.1901e-01,  2.1219e-02,  3.0826e-01,  6.1871e-01,  7.9330e-01,\n",
       "          -2.8696e-01,  7.0674e-02,  5.2887e-01,  1.8046e-01, -5.7500e-01,\n",
       "           4.6738e-01,  1.0499e-02, -1.3274e-01,  1.8639e-01, -4.5617e-01,\n",
       "           5.3939e-02,  1.0650e-01, -2.2543e-01,  5.7118e-01, -3.7568e-01,\n",
       "           1.9199e-01,  3.6890e-01, -1.8205e-01, -5.5368e-01, -3.5906e-01,\n",
       "          -9.8929e-02,  6.9423e-01, -3.8591e-01, -4.6618e-01, -3.2401e-01,\n",
       "          -1.1994e-01,  5.9138e-02, -1.9806e-01, -5.0667e-01,  1.5740e-01,\n",
       "          -2.6202e-01, -8.5835e-02, -5.1441e-02, -5.9778e-01,  3.7869e-01,\n",
       "           5.1501e-01, -1.7064e-01,  3.0995e-01, -7.5817e-01, -6.9876e-01,\n",
       "          -3.1059e-01, -5.5557e-01,  3.7969e-01,  2.8174e-01, -2.2300e-02,\n",
       "           2.5717e-01, -8.0659e-02, -3.5102e-01,  6.4568e-01,  1.4336e-01,\n",
       "           3.2292e-02,  5.8862e-01, -2.9361e-01,  6.7094e-02, -1.2895e-02,\n",
       "           6.8876e-03,  1.4091e-01, -3.0665e-02, -1.9424e-01, -4.4549e-01,\n",
       "          -2.1396e-01, -1.6803e-02, -2.2997e-01, -1.7670e-01, -1.1848e-01,\n",
       "           4.8663e-01, -9.4333e-02, -2.0714e-01,  3.1667e-01,  5.0458e-01,\n",
       "          -1.0933e-01, -4.4459e-01, -2.8909e-01, -6.9437e-02, -2.7063e-01,\n",
       "           4.8359e-01, -7.3025e-02, -4.2779e-01, -6.4551e-01, -3.7373e-01,\n",
       "          -1.4449e-01,  1.3109e+00, -5.0114e-01, -5.5120e-01,  2.2273e-01,\n",
       "           3.3985e-01,  1.8496e-01, -1.4316e-03, -6.5401e-01,  2.4435e-01,\n",
       "          -1.2867e-01, -1.2352e+00,  4.2873e-01, -4.5792e-01,  4.4269e-01,\n",
       "           4.3300e-01,  3.1017e-01,  4.4895e-03,  5.1496e-01,  2.1579e-01,\n",
       "          -2.8015e-01,  5.0963e-02, -3.5120e-01, -1.6739e-01,  2.6150e-01,\n",
       "           1.9626e-02,  2.2152e-01,  2.0477e-01,  2.5651e-02, -1.9651e-01,\n",
       "           3.6080e-01,  1.5881e+00,  5.5113e-03, -9.4346e-02,  7.4588e-01,\n",
       "           5.4827e-02,  1.1709e-01,  1.8641e-01,  8.0501e-02,  3.3152e-01,\n",
       "           4.4495e-01, -3.5697e-01, -5.6868e-01,  6.2070e-02,  2.6122e-02,\n",
       "           7.7607e-01, -5.4258e-01, -3.3469e-01, -4.9154e-02,  3.1235e-01,\n",
       "           1.7489e-01,  6.9757e-01, -3.8729e-01,  3.9715e-01,  4.1846e-01,\n",
       "          -1.0181e+00,  5.9794e-02,  2.9904e-01, -1.2651e-01, -3.5070e-01,\n",
       "          -5.5685e-01, -3.7330e-01,  1.0249e-01, -1.6943e-01,  4.5193e-01,\n",
       "           3.5801e-01, -1.4088e+00,  3.6898e-01,  7.0805e-01, -5.5302e-01,\n",
       "           2.0951e-01,  2.5262e-01,  1.3621e-01,  1.0122e-01, -2.2263e-01,\n",
       "           2.3919e-02,  2.7688e-01, -3.9357e-01, -3.0543e-01,  2.3163e-01,\n",
       "           7.1254e-01, -1.4827e-01, -5.0516e-02,  7.0887e-03, -1.2506e-01,\n",
       "          -8.9531e-02,  7.0557e-02,  9.3028e-04, -1.3082e-02, -1.1285e-01,\n",
       "           8.9753e-02, -5.1435e-01,  1.4277e-01,  4.1980e-01,  3.5882e-02,\n",
       "           5.7599e-01,  3.1479e-01,  4.9543e-01, -3.1575e-01, -2.4668e-01,\n",
       "          -3.3838e-02,  1.8182e-01,  2.4110e-01,  6.1363e-01, -1.6195e-01,\n",
       "          -5.1345e-01, -4.7131e-01, -4.4178e-02, -9.1696e-01, -4.3819e-02,\n",
       "          -7.5012e-02,  4.0870e-01,  9.4285e-02,  4.2724e-02,  5.6717e-01,\n",
       "          -2.1835e-01,  3.8417e-01, -6.1932e-01,  4.5013e-01,  1.7334e-01,\n",
       "          -8.1640e-03,  5.0540e-01,  3.0623e-01,  6.6729e-01, -1.0436e-01,\n",
       "           1.3161e-01, -1.0386e+00, -3.1218e-01,  1.1245e-01, -2.7864e-01,\n",
       "          -2.3984e-01, -2.8272e-01,  8.3016e-02,  1.5926e-01,  5.2437e-01,\n",
       "           6.8580e-01, -1.7808e-01, -6.7300e-01, -2.7378e-01,  8.5540e-02,\n",
       "           4.1591e-02, -4.3482e-02,  6.4563e-01, -8.1492e-02,  1.9653e-01,\n",
       "          -7.2482e-01, -2.4167e-01,  1.0127e-01,  6.0521e-02, -3.1888e-01,\n",
       "           6.7486e-01, -9.8470e-02, -2.6940e-01,  4.7329e-01,  2.6759e-01,\n",
       "           5.5313e-01,  7.5563e-01,  1.4888e-01, -5.4077e-01, -3.2189e-01,\n",
       "           2.1114e-01, -4.6477e-01,  3.0196e-02, -4.3202e-01, -1.3964e-01,\n",
       "           1.7101e-01,  8.2667e-01,  2.6104e-02]], device='mps:0'),\n",
       " tensor([[-0.3088, -0.4400, -0.4987,  ...,  0.0387,  0.3656,  0.3273],\n",
       "         [-0.0676, -0.9284, -0.4982,  ..., -0.0947,  0.1492,  0.4318],\n",
       "         [-0.2109, -0.4681, -0.4863,  ..., -0.0135,  0.2572,  0.4091],\n",
       "         [-0.0708, -0.4298, -0.7432,  ..., -0.1173,  0.7779,  0.6586],\n",
       "         [-0.0899, -0.3298, -0.3177,  ...,  0.0618,  0.7647,  0.8215],\n",
       "         [-0.1683, -0.5138, -0.5756,  ...,  0.0116,  0.6081,  0.5018]],\n",
       "        device='mps:0'),\n",
       " tensor([[-1.3658e-01, -7.2004e-01, -3.8330e-01, -7.4309e-01, -1.7934e-01,\n",
       "          -3.8045e-01, -6.8014e-01,  2.9567e-01,  2.3060e-02,  3.0878e-01,\n",
       "           3.8456e-01, -3.1401e-03, -1.5520e-01, -4.1630e-01, -6.4982e-01,\n",
       "          -1.2037e-01, -4.3994e-01,  1.8804e-01,  5.0569e-01,  6.3625e-02,\n",
       "           4.1044e-01,  1.7070e-01, -4.2049e-01, -4.5982e-02, -8.3850e-01,\n",
       "           1.0721e-01, -1.6637e-01,  1.2959e-01,  4.6130e-01, -2.6540e-01,\n",
       "           4.0644e-02,  8.4762e-01, -2.0383e-01,  7.4056e-01, -2.3701e-02,\n",
       "          -2.8074e-01,  2.3608e-01, -6.8979e-02,  4.1513e-01, -5.1397e-01,\n",
       "           3.3928e-01,  2.9963e-01,  1.1707e-01,  1.8467e-01,  5.8919e-02,\n",
       "           1.0651e-01,  1.5200e-01, -4.9116e-01, -2.9413e-01, -6.2907e-01,\n",
       "          -1.6100e-01, -5.0626e-01,  2.8668e-01,  3.7997e-02, -3.8425e-01,\n",
       "          -1.5569e-01, -6.3167e-03,  6.2569e-02, -1.0106e-01,  2.5507e-01,\n",
       "          -1.7289e-01,  8.3448e-01, -1.0140e-01, -1.3287e-01,  1.6572e-01,\n",
       "           2.8399e-01, -7.2920e-01, -5.4025e-01, -6.1323e-01, -1.6790e-02,\n",
       "           1.1838e-01,  5.6699e-01,  2.9383e-01, -4.5808e-01,  2.9809e-01,\n",
       "           1.6672e-01,  1.8205e-01, -1.2244e-01, -3.3693e-01,  2.6879e-01,\n",
       "           3.5816e-01,  2.9181e-01,  2.2766e-01,  3.8161e-01,  2.4256e-02,\n",
       "          -1.6610e-01,  2.7694e-01, -5.9415e-02,  3.0153e-03,  5.2020e-01,\n",
       "          -2.2154e-02, -4.3709e-02,  2.5771e-02, -2.9494e-01,  1.5614e-02,\n",
       "          -2.4701e-01,  1.7982e-01, -2.8907e-01, -1.3067e-01, -9.4758e-02,\n",
       "           2.0237e-01,  2.2990e-01,  3.4809e-01, -1.6829e-01, -7.3735e-01,\n",
       "          -3.2851e-01,  2.1011e-01, -2.0863e-01, -2.9046e-02, -6.9622e-02,\n",
       "          -2.7943e-01,  3.9711e-01,  1.2421e-01, -4.7763e-01, -3.7005e-02,\n",
       "           2.8785e-01,  1.6272e-01, -2.1691e-01,  8.6807e-02,  5.1086e-01,\n",
       "           1.2050e-01, -1.8756e-01,  3.9372e-01,  3.6098e-01,  4.0806e-01,\n",
       "           1.6165e-01,  6.7846e-01, -7.0129e-02,  3.7155e-01,  8.6475e-02,\n",
       "           3.5897e-02,  1.9094e-01, -3.2169e-01, -1.2867e-01,  3.7321e-02,\n",
       "           1.7902e-01, -3.1057e-01, -4.4242e-01,  2.6746e-01, -1.4233e-01,\n",
       "          -3.4416e-02,  1.4208e-02, -4.8332e-01, -3.5355e-01, -3.0960e-01,\n",
       "          -1.4852e-01,  3.5615e-02,  1.2947e-01, -4.8434e-01, -1.8932e-01,\n",
       "          -9.1167e-01, -6.9889e-01,  3.7673e-01,  3.0169e-02,  2.2899e-02,\n",
       "          -3.1873e-01,  3.0694e-02, -1.7544e-01, -1.9162e-01,  1.2956e-01,\n",
       "           8.6811e-01, -4.6363e-01, -1.2713e-02, -5.3759e-01, -2.2240e-02,\n",
       "          -3.3459e-02,  3.3827e-01, -1.3432e-01,  3.7591e-01,  1.1842e-01,\n",
       "           3.4281e-01, -1.4093e-02,  9.4730e-02, -7.1993e-01,  5.5381e-02,\n",
       "          -1.8622e-01, -3.6213e-02,  3.4609e-01, -1.3627e-01, -8.3681e-02,\n",
       "           1.4523e-01, -1.6407e+00,  1.2792e-01, -3.8443e-01,  1.2444e-03,\n",
       "           4.5813e-03,  2.1904e-01,  2.7758e-02,  3.0577e-01, -1.7336e-01,\n",
       "           1.6349e-01,  1.2697e-01,  3.5724e-01,  3.7835e-01,  9.8142e-02,\n",
       "           1.0975e-01,  5.5124e-01,  3.7929e-01,  1.7135e-01, -1.3624e-01,\n",
       "          -3.7788e-01,  4.3084e-01,  2.9508e-02,  2.7728e-01, -4.0794e-02,\n",
       "           8.2884e-02, -7.9664e-01,  1.4187e-01,  1.6091e-01, -1.3534e-01,\n",
       "          -2.3877e-01, -2.4176e-01, -4.7996e-01, -2.6547e-01, -7.6870e-01,\n",
       "           1.9697e-01, -2.5621e-01, -5.9981e-02,  4.6699e-01,  9.8445e-02,\n",
       "          -1.7313e-01,  3.1950e-01,  1.0127e-01, -2.9914e-01,  3.9552e-02,\n",
       "           6.5842e-01, -9.1478e-02, -7.6754e-02, -2.5907e-01, -4.2674e-01,\n",
       "          -8.0895e-04, -7.2356e-01,  6.8940e-01,  6.2095e-01, -6.7129e-02,\n",
       "           2.4276e-01, -4.7590e-01,  1.3059e-02, -2.2680e-01,  1.2042e-01,\n",
       "          -3.3737e-01, -9.4175e-02, -6.8716e-01,  2.2920e-01, -1.0109e-01,\n",
       "           4.9011e-02,  5.1113e-02,  8.7115e-01,  5.0334e-01,  3.5005e-01,\n",
       "          -1.0220e-02,  1.0576e-01,  2.6229e-01,  5.1028e-01, -2.0958e-01,\n",
       "           3.0916e-01,  1.3630e-02, -5.5574e-01,  4.9004e-01, -3.9509e-02,\n",
       "          -8.0429e-02, -2.9151e-01, -9.5875e-02,  1.7270e-01, -3.4711e-01,\n",
       "           5.7568e-04,  2.5288e-01,  1.5347e-01, -3.9601e-01,  5.1048e-01,\n",
       "          -6.0307e-01,  4.1440e-01,  1.7972e-02, -1.9281e-01, -4.7589e-01,\n",
       "          -6.4891e-01, -4.2089e-02,  1.2998e-01,  2.1240e-01, -8.1773e-01,\n",
       "          -4.4934e-01,  4.4751e-01,  2.4999e-01,  5.4039e-01,  9.2877e-02,\n",
       "           3.1589e-01, -2.4505e-01,  6.4219e-02, -2.4947e-01, -5.7786e-01,\n",
       "          -1.6888e-01,  1.5375e-01, -1.5514e-01,  1.4496e-01,  2.7022e-01,\n",
       "           3.6892e-01,  2.6109e-01,  1.0943e-01, -3.1132e-01, -5.4186e-02,\n",
       "          -1.0380e-01, -1.6623e-01,  2.3547e-01, -4.7562e-01, -5.0199e-02,\n",
       "           1.1640e-01,  1.7743e-01,  5.8191e-01, -1.1180e-02, -3.3915e-01,\n",
       "          -1.9415e-02,  7.2368e-01, -1.9118e-01, -3.6879e-01, -6.6528e-02,\n",
       "          -8.7979e-01,  1.2723e-01,  1.4928e-01, -2.7599e-01, -3.6529e-01,\n",
       "           3.7826e-01,  7.9039e-02, -3.1087e-01,  9.1582e-02, -8.4462e-03,\n",
       "           2.4943e-01,  2.7349e-01,  1.7312e-01, -2.2854e-01, -9.9767e-02,\n",
       "           3.5899e-02, -2.9388e-01, -4.9091e-02, -2.9612e-01,  2.4740e-01,\n",
       "           1.6521e-01,  3.1355e-01, -9.0397e-02, -1.0587e-01, -3.4698e-01,\n",
       "           7.5515e-01, -4.6022e-01, -1.3047e-01,  4.0354e-01, -2.5454e-01,\n",
       "          -1.3874e-01,  5.4456e-01,  7.9616e-02,  2.0368e-01, -4.2198e-02,\n",
       "           2.4793e-01, -3.8116e-01, -4.6186e-01, -2.0234e-01,  6.3649e-01,\n",
       "           2.7713e-01,  1.5581e-01, -5.9734e-01,  7.2512e-02,  4.0877e-01,\n",
       "          -2.8638e-01,  5.2757e-03, -3.1838e-01, -1.9772e-01, -1.0500e-01,\n",
       "           7.7012e-02,  5.9539e-01, -2.6025e-01,  3.7129e-01,  2.4539e-01,\n",
       "           3.0308e-01, -7.9641e-01, -8.9019e-02,  3.0482e-01,  1.7226e-01,\n",
       "           1.4320e-01, -3.3307e-01,  1.4532e-01, -4.2544e-01,  7.2827e-02,\n",
       "          -7.8315e-02, -4.6815e-03,  2.7044e-01, -2.6706e-01, -5.0976e-01,\n",
       "          -3.6289e-02, -1.3232e-01,  2.5810e-01,  3.4357e-01,  3.9499e-02,\n",
       "           4.3395e-01,  7.5727e-02, -3.4729e-01,  6.1395e-01,  9.4205e-02,\n",
       "          -1.9527e-01,  5.9673e-01, -5.9105e-01,  6.4177e-02,  7.5219e-02,\n",
       "          -1.3312e-01, -7.7515e-03,  1.3334e-01,  6.9486e-01,  7.3457e-02,\n",
       "          -1.7528e-01,  3.6388e-02,  3.3640e-01, -3.1542e-01,  2.0556e-01,\n",
       "          -1.4207e-01,  6.7624e-01, -1.8250e-01,  3.3105e-01, -2.0446e-01,\n",
       "           7.8527e-02, -2.0799e-01, -3.8950e-01, -1.9306e-01, -4.2489e-01,\n",
       "           6.8372e-02,  1.6023e-01,  3.1647e-02,  9.0926e-02, -1.1443e+01,\n",
       "          -1.7748e-01,  1.0051e+00, -4.6308e-01,  1.3497e-01,  2.5131e-01,\n",
       "           9.4560e-02, -4.9836e-01,  5.6586e-01, -5.5486e-01, -2.1277e-01,\n",
       "          -2.6997e-01,  1.3353e-01, -4.0421e-02, -5.8458e-01,  1.4588e-01,\n",
       "          -1.9817e-01, -6.2459e-01,  5.0689e-03,  1.8525e-02,  5.2404e-01,\n",
       "          -4.0634e-02,  1.7985e-01,  2.0101e-01, -1.0182e-01, -5.9618e-01,\n",
       "          -5.0901e-01, -5.4053e-01,  2.7497e-01,  4.0158e-01, -2.2603e-01,\n",
       "           2.5726e-02,  1.8820e-02,  7.5300e-02, -2.2335e-01, -6.2264e-01,\n",
       "           4.3428e-01, -3.5600e-01,  2.9236e-01, -4.3385e-01, -1.4777e-01,\n",
       "          -3.0363e-01, -3.8641e-02,  2.1949e-01,  2.0872e-01,  7.7476e-01,\n",
       "           1.1174e+00,  1.0684e+00,  1.9515e-01,  4.0956e-01,  6.0520e-02,\n",
       "          -6.9356e-01,  2.7198e-01, -6.5252e-01, -2.0687e-01, -1.8451e-01,\n",
       "           2.6959e-01, -4.6477e-01,  1.6281e-01, -3.0746e-01, -9.8721e-02,\n",
       "           5.7212e-01,  1.2026e-01,  5.8415e-02, -4.4683e-01, -5.5638e-01,\n",
       "           3.8744e-01, -9.7460e-02,  7.1511e-02, -4.7225e-01, -1.5401e-02,\n",
       "          -5.6358e-02,  6.6453e-01,  8.0328e-02, -2.0444e-01,  1.6306e-01,\n",
       "          -1.6026e-01,  2.9896e-01,  1.0237e+00, -2.4467e-01, -6.9611e-02,\n",
       "           2.6812e-03,  1.6142e-01, -6.6336e-01, -3.3776e-01, -3.1943e-03,\n",
       "           6.6028e-02,  3.8672e-01,  2.9945e-01,  4.8848e-01,  9.4608e-03,\n",
       "          -3.4301e-01, -2.3573e-02,  4.3770e-02,  3.3596e-01,  7.7098e-01,\n",
       "          -5.5531e-01, -4.8606e-02, -3.5622e-01, -5.7964e-02, -5.4890e-01,\n",
       "          -2.7115e-01, -4.8269e-03, -2.1979e-01,  1.8583e-02,  2.2503e-01,\n",
       "          -3.7268e-02,  2.5130e-01,  1.1708e-01,  3.4896e-01, -4.0606e-01,\n",
       "          -1.0782e-01,  3.1663e-02,  1.3516e-01, -5.8568e-01, -2.5149e-01,\n",
       "           1.1728e-01,  4.5830e-01, -1.6867e-01, -7.0266e-01, -3.0060e-01,\n",
       "           2.7788e-01,  1.3333e-01, -1.8817e-01, -2.6603e-01,  3.3324e-01,\n",
       "          -1.0194e+00, -3.9171e-01, -2.3012e-01,  8.3917e-02,  1.5329e-01,\n",
       "           4.4158e-02, -1.2463e-01, -1.5915e-01, -2.0432e-01, -4.0679e-01,\n",
       "          -4.9983e-01,  1.3457e-01,  6.5432e-01, -2.8831e-01, -2.3293e-01,\n",
       "          -9.6918e-02,  1.7240e-01, -2.5179e-01,  2.5951e-01, -2.6553e-01,\n",
       "           1.8068e-01,  1.6136e-01, -3.1105e-02,  7.9154e-02,  7.2528e-02,\n",
       "          -1.6682e-01, -1.4969e-01,  1.3068e-01, -1.8243e-01, -5.9356e-01,\n",
       "          -1.0338e-01,  3.0147e-01, -2.4436e-01,  2.1206e-01, -1.8683e-01,\n",
       "           1.2639e-01, -2.0630e-02, -3.3936e-01,  6.7322e-01,  4.4175e-01,\n",
       "          -1.0447e-03, -3.0852e-01, -5.0209e-01,  7.8390e-02,  1.6508e-01,\n",
       "          -1.3424e-01,  1.1818e-01, -1.0929e-01, -2.8721e-01, -2.5900e-01,\n",
       "           7.0498e-02,  1.2940e+00,  6.6313e-03, -3.3928e-01,  5.4605e-01,\n",
       "           6.5236e-01, -3.1265e-01,  6.1466e-02, -1.4284e-01,  2.2917e-01,\n",
       "          -5.8073e-01, -2.0410e-01,  5.1649e-01, -3.9372e-01,  2.2967e-01,\n",
       "           3.1110e-01,  3.7503e-02, -1.8977e-01, -4.1015e-02,  1.8945e-01,\n",
       "          -5.2648e-01,  1.3648e-01, -3.4216e-01,  2.4664e-02,  3.1453e-01,\n",
       "           9.2418e-02,  4.8282e-02, -2.4841e-02, -1.6866e-02,  5.8356e-02,\n",
       "           1.5565e-01,  1.2252e+00, -2.6450e-01, -1.9896e-01,  3.2370e-01,\n",
       "          -3.9012e-01,  6.3679e-01,  1.2602e-01, -7.5775e-02,  1.6396e-01,\n",
       "           3.6642e-01,  9.2990e-02, -4.5701e-01,  3.8094e-01,  1.4621e-01,\n",
       "          -1.0685e-01, -7.2663e-01, -4.8291e-01, -4.0190e-02,  5.0384e-01,\n",
       "          -3.7069e-01,  4.0312e-01, -1.0823e-01,  7.3065e-01,  1.0199e-01,\n",
       "          -9.6118e-01,  8.3651e-02, -4.0857e-01,  3.5361e-01, -6.7545e-03,\n",
       "          -6.6724e-01, -3.3350e-01,  1.2482e-01, -2.7411e-01,  3.3525e-01,\n",
       "           3.8690e-01, -9.7039e-01, -1.0704e-02,  3.9717e-01, -9.7281e-02,\n",
       "           2.8225e-01, -6.1232e-02,  1.4523e-01,  4.4320e-01, -1.7392e-01,\n",
       "          -1.0947e-01,  1.9066e-01,  7.8303e-03, -3.5862e-01,  1.2799e-01,\n",
       "           4.0711e-01, -2.4405e-03,  7.0412e-03, -3.5824e-01,  1.1475e-01,\n",
       "          -3.9083e-01, -5.4962e-03, -3.9482e-01, -7.6378e-02,  5.2778e-01,\n",
       "          -4.7143e-02, -2.4720e-01, -1.1573e-02,  3.1051e-01, -1.2812e-01,\n",
       "           4.6348e-01,  3.3011e-01,  6.2273e-01, -2.1340e-01, -8.2499e-02,\n",
       "          -2.8012e-01,  3.4202e-01, -1.9070e-01,  6.8851e-01, -8.2991e-02,\n",
       "          -2.5176e-01,  2.9337e-01,  4.6769e-02, -5.0837e-01, -8.5408e-02,\n",
       "          -1.4292e-01,  1.5081e-01, -1.6286e-01, -2.0966e-01,  3.1069e-01,\n",
       "          -1.1842e-01,  4.6435e-03, -5.6936e-01,  2.7244e-01, -1.6435e-02,\n",
       "          -2.7305e-01,  1.1041e+00, -1.8266e-01,  2.8146e-01, -6.8321e-02,\n",
       "           1.3233e-01, -8.1700e-01, -1.2216e-01,  2.3936e-01, -9.4041e-02,\n",
       "          -5.5379e-01, -9.5600e-02,  2.2174e-02, -3.5821e-02,  7.3317e-01,\n",
       "           4.6286e-01, -5.2593e-01, -3.7340e-01, -6.3182e-02, -1.1937e-01,\n",
       "           6.3590e-02, -4.7719e-01,  2.3609e-01, -3.5884e-01, -8.9827e-02,\n",
       "          -7.4291e-01, -1.5549e-01, -1.5277e-01,  1.0932e-01, -1.1947e-01,\n",
       "           7.8608e-01, -1.4095e-01,  2.4257e-01,  8.1484e-01,  1.7692e-01,\n",
       "           3.5248e-01,  7.2602e-01,  1.8920e-01, -5.7397e-01, -7.8195e-02,\n",
       "           3.3864e-01, -3.1246e-01,  1.3923e-01, -3.9887e-01,  5.6093e-01,\n",
       "           9.0758e-02,  3.4504e-01,  7.9037e-01]], device='mps:0'),\n",
       " tensor([[-6.1750e-01, -4.2829e-01, -2.3934e-01, -9.4116e-01, -3.4548e-01,\n",
       "          -6.5496e-02, -6.3874e-01,  9.7828e-01, -9.5892e-02,  4.0096e-01,\n",
       "           1.4919e-01, -5.1603e-03,  2.0980e-01,  1.3680e-01, -6.9506e-01,\n",
       "           1.6740e-01,  4.6094e-01,  2.1159e-01,  3.5951e-01, -4.4329e-02,\n",
       "           7.9832e-01,  3.4245e-01,  1.8019e-01, -2.5205e-01, -4.5876e-01,\n",
       "           5.8994e-01, -2.4628e-01, -9.2014e-02,  2.0509e-01, -8.1949e-01,\n",
       "          -1.3586e-01, -4.9757e-02, -3.6281e-01,  8.7602e-01,  1.1948e-01,\n",
       "          -2.7310e-01, -1.5276e-01, -2.7996e-01,  2.3494e-01, -6.3637e-02,\n",
       "           3.7814e-01,  2.6592e-02, -2.9347e-01,  3.1386e-01, -1.6421e-01,\n",
       "          -1.4057e-01,  1.4047e-02, -7.8429e-01,  1.3666e-02,  1.8955e-01,\n",
       "           5.8851e-02, -4.7205e-02,  1.9231e-01,  3.7030e-01, -1.0617e+00,\n",
       "          -2.6348e-01,  2.7603e-01,  2.9849e-01, -1.7932e-01, -5.7784e-02,\n",
       "          -1.9989e-01,  9.0201e-01,  1.7949e-01, -2.7822e-01,  1.7824e-01,\n",
       "          -4.4899e-01, -5.4601e-01, -6.0781e-01, -1.0682e-01, -1.1967e-01,\n",
       "          -1.0002e-01,  9.7528e-02,  1.4009e-01, -2.9385e-01,  2.3170e-01,\n",
       "           2.1813e-01,  4.7024e-02, -2.7652e-02, -4.3158e-01, -2.6546e-01,\n",
       "           3.6947e-01,  3.4425e-02, -1.4730e-02,  2.0326e-01, -5.8744e-01,\n",
       "          -2.0226e-01,  3.7959e-01,  1.5703e-01,  2.5914e-01,  2.2662e-03,\n",
       "           5.9967e-01,  1.2024e-02,  8.0339e-02, -4.5369e-01,  5.8427e-01,\n",
       "          -5.4955e-01,  1.0329e-01, -1.6007e-01,  4.9532e-02, -5.2088e-01,\n",
       "          -9.8735e-02,  2.4494e-01, -1.8085e-01,  1.6053e-01, -4.6592e-01,\n",
       "          -3.2983e-02,  2.4514e-01,  3.0468e-01, -4.9183e-01,  4.2685e-02,\n",
       "          -6.5134e-03, -2.9791e-01,  7.3694e-02, -6.6407e-01, -8.5795e-01,\n",
       "           5.8354e-01,  1.8428e-02, -4.5240e-01, -2.3930e-01,  7.4443e-01,\n",
       "           2.9738e-01, -2.3201e-01, -5.3072e-02, -3.5548e-01,  2.0259e-01,\n",
       "          -1.8518e-01,  2.5689e-01, -2.7018e-01,  3.8439e-01,  5.5026e-01,\n",
       "           3.1581e-01,  2.8489e-01, -1.8327e-01,  2.7637e-01,  1.4465e-01,\n",
       "           3.6947e-02, -6.6614e-02, -1.5929e-01,  1.5180e-01,  1.2041e-01,\n",
       "          -2.9580e-01, -3.2999e-01,  5.4746e-03,  1.1182e-01, -3.3279e-01,\n",
       "          -1.0202e-01,  1.8680e-01,  3.4808e-01, -1.4092e-01, -2.5403e-01,\n",
       "          -7.8984e-01, -4.2328e-01,  1.0217e-01, -1.8560e-01, -1.7135e-01,\n",
       "          -4.1911e-01,  3.8121e-01, -6.6660e-02,  4.7638e-01,  3.4189e-01,\n",
       "           6.9978e-01, -1.5311e-01,  5.9776e-02, -5.6412e-01,  3.5197e-01,\n",
       "           3.8270e-02, -6.1110e-01, -1.7247e-01,  1.5680e-01, -4.1813e-04,\n",
       "           6.8195e-01,  7.0979e-02,  4.0124e-01,  1.2070e-01, -3.2553e-01,\n",
       "           8.2089e-03,  1.9890e-01, -1.2864e-01, -1.4297e-01,  7.3766e-01,\n",
       "          -5.5188e-01, -2.3029e+00,  1.0721e-01, -3.5844e-01, -3.6529e-01,\n",
       "          -1.8135e-01,  3.7255e-01,  3.3772e-02,  4.9966e-01,  1.5486e-01,\n",
       "           3.7598e-02,  1.6729e-01, -2.8025e-02,  3.6851e-01, -8.4399e-02,\n",
       "           2.6751e-01,  3.3341e-01,  1.2988e-01, -2.3298e-01, -1.0956e-01,\n",
       "          -3.9227e-01,  2.4031e-01, -9.8916e-02,  8.6010e-01,  2.3260e-01,\n",
       "           2.0943e-01, -4.3657e-03, -4.7262e-02,  4.8964e-01,  4.3099e-01,\n",
       "          -6.6541e-02, -1.6585e-01, -4.9522e-01,  4.6269e-03, -4.1279e-02,\n",
       "          -1.2963e-02, -4.5611e-01,  1.3753e-01,  4.2841e-01, -1.2416e-01,\n",
       "          -1.8354e-01,  1.5525e-01, -4.7715e-02, -3.1517e-01, -2.6492e-02,\n",
       "           6.8453e-01,  1.9080e-01, -2.2994e-01,  1.2377e-01, -1.1896e-01,\n",
       "           3.7006e-02, -4.6975e-01,  7.0044e-01,  6.1147e-01,  1.5585e-01,\n",
       "           1.6214e-02, -6.9943e-01,  2.3699e-01, -3.1574e-01, -4.7647e-03,\n",
       "          -1.5027e-01, -1.5731e-01, -4.2986e-01, -5.7285e-02, -5.5854e-02,\n",
       "           4.0115e-01,  7.9943e-02,  4.6653e-01,  1.1837e+00,  2.7973e-01,\n",
       "           1.9184e-02,  5.6097e-01,  2.0801e-01,  4.2492e-01, -4.0180e-01,\n",
       "           3.3082e-01,  8.8825e-02, -6.3896e-01,  2.7188e-01, -4.1593e-01,\n",
       "           4.4963e-02, -4.1432e-02, -3.7926e-01,  1.4208e-01, -2.1276e-02,\n",
       "           3.4237e-01,  3.3517e-01,  1.5186e-01,  1.6838e-01,  2.9411e-01,\n",
       "          -3.1322e-01,  5.9715e-01, -6.0809e-02, -1.1127e-01, -6.3492e-01,\n",
       "          -7.6732e-01,  9.5206e-02,  4.4743e-01, -1.2880e-01, -6.1279e-01,\n",
       "           1.0462e-01, -2.3866e-01, -1.2257e-01, -3.5064e-02,  2.1191e-01,\n",
       "           2.7455e-01,  5.0687e-02, -9.5436e-02,  6.4791e-02, -4.6950e-01,\n",
       "          -4.9277e-01,  4.9140e-02, -1.5758e-01, -1.3063e-01,  5.3918e-01,\n",
       "          -7.9415e-02,  1.1575e-01,  5.7221e-01,  1.9799e-02, -4.5976e-01,\n",
       "          -2.5640e-01,  1.4802e-01,  2.8667e-01, -7.2082e-02,  2.1981e-01,\n",
       "           3.4119e-01,  2.6467e-02,  5.1649e-01,  1.4823e-01, -2.1046e-01,\n",
       "          -2.7239e-01,  5.1173e-01, -1.5755e-01,  6.1403e-02, -1.2527e-01,\n",
       "          -8.7207e-01, -5.6150e-02, -7.8413e-02, -1.6833e-01, -5.4440e-01,\n",
       "           2.6566e-01,  1.4351e-01,  1.0561e-01,  2.2350e-01, -1.5984e-01,\n",
       "           7.0123e-01,  6.5285e-01, -3.3100e-01, -1.6943e-01, -4.0104e-01,\n",
       "           9.3922e-02, -6.8374e-02, -2.9329e-01, -1.4009e-02,  9.5368e-03,\n",
       "          -2.2765e-01, -1.2516e-01, -1.0805e-01,  2.8921e-01, -1.8876e-01,\n",
       "           5.2862e-01,  1.2189e-01,  8.3649e-02,  4.3873e-01,  2.0612e-02,\n",
       "           1.3071e-01,  3.6515e-01, -4.2117e-01,  1.2870e-01,  4.0424e-01,\n",
       "           4.0578e-01, -1.0551e+00, -8.3291e-01, -1.4936e-01,  3.6111e-01,\n",
       "           2.2860e-01,  2.5189e-01,  1.6999e-02, -7.2215e-03,  8.9455e-02,\n",
       "          -2.3402e-01,  3.5135e-01, -6.6415e-02, -5.7412e-01,  3.0561e-01,\n",
       "           2.3567e-01,  5.3263e-01, -2.7377e-01, -5.0579e-01, -6.5233e-02,\n",
       "           2.4616e-01, -3.6381e-01,  1.1392e-01,  2.3480e-01,  4.5466e-01,\n",
       "          -3.6015e-01, -5.6164e-01, -7.0282e-02, -7.3986e-01, -1.3578e-01,\n",
       "           3.2483e-01, -2.1437e-01,  2.7178e-02, -2.4758e-01,  1.5231e-01,\n",
       "          -1.7925e-02, -1.8879e-01,  6.0757e-01,  2.0522e-01, -2.5855e-01,\n",
       "           1.1981e-01, -2.3892e-01, -1.9475e-01,  5.0091e-01, -5.0226e-01,\n",
       "          -3.0348e-01, -5.0960e-02, -1.5349e-01, -3.2418e-01,  3.2181e-01,\n",
       "          -4.1389e-01, -1.4861e-01,  1.6748e-01, -6.4736e-02, -1.3068e-01,\n",
       "          -3.5233e-01,  1.4268e-01, -1.2707e-01, -4.9883e-02, -1.7654e-02,\n",
       "          -1.3353e-01,  6.0900e-02,  1.3308e-01,  6.4538e-02,  1.4388e-01,\n",
       "           5.3696e-01, -1.1464e-01,  2.7582e-03, -2.5900e-01, -2.2469e-01,\n",
       "           5.2498e-01, -9.4973e-02, -3.3117e-01,  5.4928e-01, -1.1156e+01,\n",
       "           1.0386e-01,  7.2631e-01, -5.5808e-01,  5.5128e-02, -2.8539e-01,\n",
       "           2.1085e-01, -3.0009e-01,  7.9101e-02,  2.4967e-01, -2.8460e-01,\n",
       "          -1.2277e-01, -2.7364e-01, -2.3193e-01, -4.0872e-01, -3.9742e-01,\n",
       "          -4.7229e-01, -5.6789e-01,  1.1772e-01, -1.7663e-01, -1.0996e-01,\n",
       "          -5.7036e-01, -1.9379e-01,  2.9696e-01, -4.9559e-03, -5.1069e-01,\n",
       "           5.9541e-02, -2.2948e-01,  2.6399e-01,  4.6126e-02,  2.2045e-01,\n",
       "          -1.9949e-02,  1.4108e-01,  2.0669e-01, -4.4946e-02, -4.5100e-01,\n",
       "          -4.4853e-01,  2.2129e-01, -1.5955e-01, -2.2267e-01, -5.4017e-02,\n",
       "          -3.3838e-01, -2.7689e-01,  1.3447e-01,  3.7594e-01,  6.2596e-01,\n",
       "           1.1595e+00,  5.6208e-01,  2.9245e-01, -1.0625e-01, -8.5438e-02,\n",
       "          -1.0762e+00,  5.5900e-01, -3.7318e-01,  8.1787e-02, -2.5119e-01,\n",
       "           6.6201e-01, -4.9906e-01,  5.4685e-01, -4.0754e-01, -2.5500e-01,\n",
       "           3.7179e-01,  3.0194e-01,  7.1104e-03,  2.8239e-02, -2.7432e-01,\n",
       "           4.0257e-01, -2.6094e-01, -3.6915e-02, -3.7819e-01,  4.0081e-01,\n",
       "          -3.9472e-01,  5.0770e-01, -5.6534e-02,  1.2945e-01,  1.8567e-01,\n",
       "          -3.5001e-01,  1.6610e-01,  1.0337e+00,  1.7414e-01,  4.1071e-01,\n",
       "           1.7420e-01,  5.1160e-01,  5.3064e-02, -8.8958e-01,  1.3691e-01,\n",
       "           2.0502e-01,  2.0789e-01, -6.7096e-02,  5.7131e-01, -2.5517e-02,\n",
       "          -6.5181e-01, -2.3094e-01, -3.2169e-01,  4.3108e-01,  6.5352e-01,\n",
       "          -2.2597e-01, -4.4343e-01,  2.3894e-01, -7.9776e-03, -5.3395e-01,\n",
       "           9.0074e-02, -2.2320e-01, -5.2792e-01,  8.7275e-02, -5.8914e-01,\n",
       "           1.4985e-01, -1.2428e-01,  5.4238e-02,  6.6887e-01, -4.0159e-01,\n",
       "          -1.6113e-01,  3.6602e-02, -4.4532e-02, -4.1270e-01, -4.9520e-01,\n",
       "          -7.7083e-02,  4.1073e-01, -1.9776e-01, -2.7976e-01,  7.0636e-02,\n",
       "          -3.0673e-01,  2.2584e-01, -1.7377e-01, -6.4935e-01,  3.2879e-01,\n",
       "          -6.0415e-01,  2.0578e-01,  2.5360e-01, -2.2507e-01,  4.9017e-01,\n",
       "           1.1337e-01, -8.1817e-02,  1.9397e-01, -9.9369e-02, -7.8554e-01,\n",
       "          -3.2573e-01, -2.7299e-02,  3.2662e-01, -4.2844e-01,  1.5776e-01,\n",
       "           1.5022e-01, -4.5752e-02, -2.2574e-01,  2.7904e-01,  1.0174e-01,\n",
       "           3.7762e-01,  3.9846e-01, -5.5006e-01,  1.1727e-01, -2.9867e-01,\n",
       "          -2.9262e-02, -6.1174e-02,  1.7952e-01,  2.3654e-01, -7.0764e-01,\n",
       "          -1.5224e-01, -2.6434e-01, -3.0965e-01, -2.1650e-01, -2.5120e-01,\n",
       "           2.0680e-01,  1.4038e-01, -1.5897e-02, -1.1308e-01,  5.8077e-01,\n",
       "           4.0814e-02, -1.1167e-01, -4.5997e-01,  3.1443e-01,  4.3178e-01,\n",
       "           2.1794e-01,  7.5875e-02, -1.2830e-01, -5.5532e-01, -1.1922e-02,\n",
       "          -4.3644e-01,  9.1081e-01, -7.3906e-01, -7.1970e-01,  6.6086e-02,\n",
       "           1.4951e-02, -8.8259e-04, -2.3450e-01, -5.0265e-01,  8.1298e-02,\n",
       "          -2.1017e-01, -1.4891e-01,  6.1550e-03, -1.0607e-02,  1.6299e-01,\n",
       "           4.0742e-01,  3.6838e-01,  1.5951e-01, -3.3938e-01, -4.4081e-01,\n",
       "          -6.0244e-01,  1.4567e-02, -1.6795e-01,  2.9330e-01, -2.6752e-01,\n",
       "          -3.2661e-01,  1.6043e-02, -9.1892e-03, -2.2272e-01,  5.2876e-01,\n",
       "           1.2427e-01,  1.6396e+00, -3.7287e-01, -4.4551e-01,  2.9018e-01,\n",
       "           7.2306e-02, -2.7073e-02, -2.8130e-01,  2.5074e-01,  9.5447e-02,\n",
       "           4.0194e-02, -3.6304e-01,  1.5023e-02, -2.0558e-02,  3.2773e-01,\n",
       "           3.2056e-01, -1.0703e+00, -2.0238e-01, -1.7993e-01,  5.5664e-01,\n",
       "           3.2388e-02,  6.3185e-01, -2.2305e-01,  3.0577e-01,  4.6106e-01,\n",
       "          -8.6667e-01, -2.3000e-01, -1.6129e-01,  2.7295e-01, -1.0583e-01,\n",
       "          -9.8735e-02, -1.6178e-01, -3.7165e-02, -1.6936e-01,  3.9538e-01,\n",
       "           7.4751e-01, -7.8426e-01,  7.4081e-01,  7.0467e-01, -9.8234e-02,\n",
       "           5.6815e-01,  9.4187e-02,  3.2641e-01, -1.4659e-01,  2.2669e-02,\n",
       "           2.6200e-01,  1.8542e-01, -6.3165e-01, -2.3962e-01,  2.4473e-01,\n",
       "           4.0158e-01, -1.1943e-01,  1.2108e-01,  7.9292e-02,  3.0960e-02,\n",
       "          -3.7691e-01, -1.0145e-01, -2.4552e-01, -3.9904e-01,  1.1356e-01,\n",
       "           1.4603e-02, -1.5001e-01, -8.6997e-02,  4.7709e-01, -1.7580e-01,\n",
       "           7.5067e-01,  4.2910e-01,  3.7145e-01, -3.5103e-01,  8.9950e-04,\n",
       "          -5.7565e-02,  6.0671e-01, -3.1670e-01,  7.7385e-01,  6.3337e-02,\n",
       "          -2.3140e-01,  2.7251e-01,  6.0547e-03,  1.7829e-01,  2.9892e-01,\n",
       "          -2.3519e-01,  1.2702e-01, -3.0718e-01, -2.3711e-01,  2.8330e-01,\n",
       "          -1.8640e-01,  1.1542e-01, -7.2270e-01, -6.3602e-03,  3.9857e-01,\n",
       "          -3.2051e-01,  4.9873e-01, -2.3074e-01,  6.1026e-01, -4.3509e-01,\n",
       "           3.2669e-02, -4.7470e-01, -4.3275e-01, -2.8508e-01,  5.1854e-02,\n",
       "          -8.6027e-01, -1.9453e-01,  1.7684e-01,  3.7405e-02,  2.1707e-01,\n",
       "           7.5155e-01, -3.3794e-01, -2.4771e-01, -2.0386e-01,  4.0190e-02,\n",
       "           1.8299e-01, -8.2071e-02, -3.5586e-01,  2.1161e-01, -3.5607e-01,\n",
       "          -1.0520e+00, -2.3117e-01,  8.9990e-02,  3.8419e-01, -3.5853e-01,\n",
       "           1.1570e+00,  2.5132e-02, -2.0226e-01,  6.7253e-01,  2.0808e-01,\n",
       "           6.4157e-01,  6.0323e-01,  1.3628e-03, -4.1259e-01, -5.4109e-02,\n",
       "          -1.5877e-01, -3.2881e-01,  1.5757e-01, -1.8346e-01,  5.0556e-01,\n",
       "           5.4802e-01, -2.0806e-02,  5.4756e-01]], device='mps:0'),\n",
       " tensor([[-0.4654, -0.4115, -0.4370,  ...,  0.1625,  0.4338,  0.5908],\n",
       "         [-0.1242, -0.5179, -0.4093,  ...,  0.3691, -0.0529,  0.5134],\n",
       "         [-0.0029, -0.4260, -0.6820,  ...,  0.1889,  0.1934,  0.5362],\n",
       "         [-0.3113, -0.0324, -0.3558,  ...,  0.2697,  1.0230,  0.2478]],\n",
       "        device='mps:0'),\n",
       " tensor([[-0.4152, -0.4002, -0.8252,  ...,  0.0077,  0.4768,  0.3012],\n",
       "         [-0.1377, -0.6164, -0.4381,  ...,  0.2714,  0.0332,  0.5972],\n",
       "         [-0.3292, -0.2590, -0.5016,  ..., -0.0148,  0.3509,  0.1607]],\n",
       "        device='mps:0'),\n",
       " tensor([[-0.1230, -0.2743, -0.7632,  ...,  0.0595,  0.8762,  0.7366],\n",
       "         [-0.4240, -0.5232, -0.6954,  ...,  0.1532,  0.3221,  0.6263]],\n",
       "        device='mps:0'),\n",
       " tensor([[-0.0111, -0.5057, -0.4670,  ...,  0.2365, -0.0712,  0.4388],\n",
       "         [-0.3292, -0.2590, -0.5016,  ..., -0.0148,  0.3509,  0.1607]],\n",
       "        device='mps:0'),\n",
       " tensor([[-4.3893e-01, -8.3642e-01, -7.1425e-01, -6.5047e-01, -2.4178e-01,\n",
       "          -4.6821e-01, -8.5016e-01,  2.8734e-01,  6.1245e-03,  4.7705e-01,\n",
       "           1.8053e-01, -9.3699e-02,  7.3992e-02, -2.8377e-01, -4.8785e-01,\n",
       "          -1.6906e-01,  5.3340e-02,  2.8278e-01,  3.4594e-01,  1.2597e-01,\n",
       "           4.7705e-01,  3.6448e-01, -1.1967e-01, -1.3955e-01, -8.8263e-01,\n",
       "           1.3639e-01, -3.9954e-01,  5.0085e-02,  5.1721e-01, -8.0331e-01,\n",
       "          -4.3557e-02,  8.1389e-01, -6.1013e-02,  4.3767e-01,  6.7594e-02,\n",
       "          -2.6332e-01,  6.2141e-02, -3.4019e-02,  4.1672e-01, -3.0260e-01,\n",
       "           1.5589e-01,  1.6038e-01,  2.2164e-01,  2.0284e-01,  3.2346e-01,\n",
       "           2.3643e-01,  3.0533e-01, -7.1457e-01,  6.7604e-02, -5.5334e-01,\n",
       "          -1.7853e-01, -3.8055e-01,  3.8199e-02,  2.0691e-01, -4.2637e-01,\n",
       "          -2.7415e-01,  7.2675e-02,  1.3679e-01, -1.8338e-02,  6.8582e-02,\n",
       "           2.0545e-01,  7.0538e-01, -2.3705e-02, -5.0261e-02,  2.4114e-01,\n",
       "           2.4270e-01, -7.2049e-01,  1.1884e-01, -5.4320e-01, -1.7641e-01,\n",
       "          -1.5927e-01,  3.1720e-01, -4.7420e-02, -7.3685e-01,  1.3818e-01,\n",
       "          -1.7407e-01,  3.8365e-01, -3.7939e-01, -4.8563e-01,  5.6187e-02,\n",
       "           2.3671e-01,  1.6582e-01,  1.0588e-01, -2.8038e-02, -3.1122e-01,\n",
       "          -2.6558e-01,  4.5674e-01, -3.2075e-02,  1.8907e-01,  4.5659e-01,\n",
       "           4.3184e-01, -8.8547e-02,  5.3856e-01, -1.9776e-01,  1.8667e-01,\n",
       "          -1.5302e-01,  2.5068e-01, -7.6263e-03,  9.1955e-02, -1.8667e-01,\n",
       "           9.4288e-02,  1.8748e-01,  9.0473e-02, -6.7667e-02, -5.8063e-01,\n",
       "          -7.9106e-02,  1.8662e-01,  9.2470e-02, -5.5641e-01, -1.5187e-01,\n",
       "          -2.2848e-02,  2.8509e-01,  2.6112e-01, -4.9551e-01, -2.7398e-01,\n",
       "           3.2360e-01,  4.3537e-02, -1.7931e-01, -6.7884e-02,  4.0515e-01,\n",
       "           2.7276e-01, -3.3059e-01,  5.4339e-01,  3.0180e-01,  6.2495e-01,\n",
       "          -6.8242e-02,  5.6603e-01, -9.9196e-02,  3.1988e-01, -6.8019e-02,\n",
       "          -2.5483e-01, -1.9302e-02, -4.3479e-01,  4.6920e-02,  7.1548e-02,\n",
       "          -2.0760e-02, -7.1394e-02, -4.5493e-01, -1.2793e-01, -6.2671e-02,\n",
       "          -4.2447e-01,  4.3916e-02, -5.4860e-01, -3.7674e-01,  1.9036e-01,\n",
       "          -1.7667e-01, -7.4348e-02,  1.8304e-01, -1.9135e-01, -3.5758e-01,\n",
       "          -5.4880e-01, -7.5304e-01,  7.1890e-01,  2.0143e-02,  3.2871e-02,\n",
       "          -1.9521e-01, -1.4608e-01,  7.4876e-02,  1.4261e-01,  1.9806e-01,\n",
       "           7.7951e-01, -4.0883e-01,  1.5286e-01, -3.0666e-01, -2.0031e-02,\n",
       "           2.1676e-01,  1.0152e-01, -3.1967e-01,  2.6728e-01,  6.9500e-02,\n",
       "           3.4679e-01, -1.9730e-02, -2.2406e-01, -5.9829e-01, -4.8645e-02,\n",
       "          -3.0804e-02,  1.1942e-01,  4.6125e-01, -8.1087e-02, -1.9101e-01,\n",
       "           1.5211e-01, -2.0106e+00,  1.2182e-01, -7.2234e-01, -4.3590e-02,\n",
       "          -3.9394e-01,  3.1319e-01,  2.2222e-01,  2.6987e-01,  5.0735e-02,\n",
       "           3.4632e-01, -1.1002e-01,  1.0658e-01,  3.1964e-01, -6.6156e-02,\n",
       "          -1.3234e-02,  6.6971e-01,  1.3766e-01,  4.1761e-01,  1.0054e-01,\n",
       "          -4.6834e-01,  5.8398e-01, -1.7865e-01, -4.6977e-02,  3.0210e-02,\n",
       "           2.0593e-01, -7.7930e-01,  4.2740e-01,  1.4737e-01, -2.4888e-01,\n",
       "          -1.7118e-01, -1.1380e-01, -3.1463e-01, -3.7178e-01, -2.8931e-01,\n",
       "           2.3472e-01, -3.0439e-01,  6.3104e-03,  4.2049e-01,  2.1528e-01,\n",
       "          -1.2180e-01, -3.4649e-02, -1.0384e-01, -1.6283e-01, -1.9948e-01,\n",
       "           5.8393e-01, -1.7795e-01, -2.1110e-02, -9.9328e-02, -2.8517e-01,\n",
       "           5.5152e-02, -8.3049e-01,  5.7992e-01,  6.9090e-01, -1.3461e-01,\n",
       "          -3.2525e-01, -1.5881e-01, -3.5938e-02, -2.7682e-01, -1.0331e-01,\n",
       "          -4.0413e-01, -5.4710e-01, -9.8030e-01,  1.4477e-01, -1.3794e-01,\n",
       "          -1.2906e-02, -1.7625e-01,  6.7218e-01,  5.6365e-01,  4.4101e-01,\n",
       "          -1.9362e-01,  8.7707e-03,  2.5015e-01,  4.3796e-01, -2.5293e-01,\n",
       "          -4.6742e-02,  8.0662e-02, -4.4831e-01,  3.5058e-01, -1.4293e-01,\n",
       "          -3.1042e-01, -1.5392e-02, -1.1276e-01,  2.7397e-01, -5.3243e-01,\n",
       "           7.6849e-01,  1.0806e-01,  1.1219e-01, -3.3838e-01,  2.7502e-01,\n",
       "          -5.8890e-01,  2.0731e-01, -5.7186e-01, -4.9591e-01, -6.4592e-01,\n",
       "          -8.1292e-01,  4.9423e-02,  3.2367e-01,  1.9947e-01, -8.3975e-01,\n",
       "          -5.0017e-01, -2.5605e-01,  1.7995e-01,  1.5852e-01, -1.7728e-01,\n",
       "           1.7897e-01, -2.9005e-01, -1.2990e-01,  1.7407e-01, -5.4526e-01,\n",
       "          -3.3903e-01,  5.4079e-01, -1.3237e-01,  3.1943e-01,  3.9817e-01,\n",
       "          -1.7940e-01,  2.1675e-01,  6.5074e-02, -2.7863e-01,  1.0319e-01,\n",
       "          -1.9745e-01, -2.0514e-01,  1.8769e-01, -1.9579e-01, -8.1690e-02,\n",
       "           3.9170e-01,  1.8162e-01,  5.6731e-01,  1.3514e-01, -2.1562e-01,\n",
       "           1.6761e-03,  6.7443e-01, -1.8189e-01, -2.3955e-02, -2.4598e-01,\n",
       "          -1.4564e+00, -5.0160e-02, -3.6548e-01, -3.7287e-01, -2.6733e-01,\n",
       "           3.7964e-01,  1.6499e-02, -1.0041e-01,  3.4866e-02, -2.2881e-04,\n",
       "          -7.5443e-02,  4.0788e-01, -2.2596e-01,  1.1531e-01, -1.8164e-02,\n",
       "          -7.3570e-02, -8.2556e-02, -1.2367e-01, -1.2268e-01,  1.8088e-01,\n",
       "           2.5295e-01,  1.5921e-01,  1.3451e-01, -2.8206e-01, -4.5800e-01,\n",
       "           9.6162e-01, -3.1689e-01,  1.3593e-01,  3.6500e-01, -5.7709e-01,\n",
       "           2.0278e-01,  3.0122e-01, -1.8312e-01,  2.0394e-01, -1.6884e-01,\n",
       "           1.3857e-01, -3.2317e-01, -6.3262e-01, -1.8689e-01,  6.4501e-01,\n",
       "           2.1724e-01,  5.7068e-01, -3.6922e-01, -3.2259e-02,  3.4829e-01,\n",
       "          -6.2176e-01,  3.6953e-01, -2.7257e-02, -7.8081e-03, -1.2123e-01,\n",
       "           5.6308e-02,  4.6457e-01, -4.0361e-01,  3.6660e-01, -4.4230e-02,\n",
       "          -8.1044e-02, -6.1276e-01, -1.2371e-01,  4.8504e-02,  9.9777e-02,\n",
       "           1.8905e-01, -2.8110e-01,  4.5324e-02, -6.0131e-01, -1.7716e-01,\n",
       "          -6.5173e-02,  5.4484e-02,  2.0270e-01, -1.4404e-01, -3.9139e-01,\n",
       "          -3.8103e-01,  2.3744e-01,  2.4605e-01,  7.8101e-01,  1.9952e-01,\n",
       "           3.2746e-01,  3.0836e-05, -1.3668e-01,  3.5899e-01, -3.0206e-02,\n",
       "          -2.6745e-01,  6.7999e-01, -1.3744e-01,  1.9332e-01,  1.3934e-01,\n",
       "          -2.6236e-01, -1.9039e-01,  5.4959e-02,  1.2849e-01,  1.3018e-01,\n",
       "          -4.3710e-01,  3.1806e-01,  5.6189e-01, -2.4479e-01, -9.9674e-02,\n",
       "          -3.1560e-02,  8.6508e-01, -1.1294e-02,  3.2283e-01,  2.3051e-01,\n",
       "           2.2071e-01, -5.6498e-02,  1.6954e-01,  7.3032e-02, -6.3491e-01,\n",
       "           4.6975e-01,  1.7686e-01,  1.3641e-01,  9.7463e-02, -1.1441e+01,\n",
       "           1.6115e-01,  9.8166e-01, -6.1998e-01,  7.6963e-02,  1.6237e-01,\n",
       "          -3.4056e-02, -4.2333e-01,  2.5567e-01, -6.6207e-01, -3.3554e-01,\n",
       "          -1.5640e-01,  1.6454e-01,  9.7124e-02, -4.7393e-01, -1.6605e-01,\n",
       "          -3.2256e-01, -7.6598e-01, -2.1737e-01,  3.0102e-02,  5.6025e-01,\n",
       "          -4.9848e-02,  1.8195e-01,  2.4852e-01,  6.4182e-02, -4.4133e-01,\n",
       "          -6.2554e-01, -3.2056e-01, -8.7605e-02,  1.7029e-02,  1.3947e-01,\n",
       "          -9.3363e-02,  6.0682e-02,  1.1994e-01,  2.1695e-02, -5.7067e-01,\n",
       "          -2.4708e-01, -1.8822e-01,  3.7337e-01,  1.2066e-02, -2.0759e-01,\n",
       "           1.2805e-01,  9.4967e-02,  4.5766e-01,  8.7136e-02,  6.9273e-01,\n",
       "           1.0622e+00,  8.9702e-01, -1.6410e-02,  8.2039e-02, -7.1796e-02,\n",
       "          -6.5028e-01,  3.0527e-01, -6.8426e-01, -1.1257e-01, -1.3796e-01,\n",
       "           5.3974e-01, -2.8893e-01,  1.3197e-01,  1.8808e-02, -1.8310e-01,\n",
       "           1.5580e-01, -5.3432e-02,  2.0198e-01, -2.9335e-01, -2.9055e-01,\n",
       "           5.5764e-01,  8.1481e-02,  5.6508e-02, -7.6543e-02, -7.3920e-02,\n",
       "          -2.4715e-01,  8.2028e-01,  5.6641e-02,  2.1549e-01,  2.2821e-01,\n",
       "          -2.4230e-01,  1.4977e-01,  7.2989e-01, -3.4340e-01, -1.2304e-01,\n",
       "           4.5849e-02,  1.4249e-01, -8.1244e-01, -8.4482e-01,  1.0869e-01,\n",
       "           1.0017e-01,  2.7895e-01,  3.7070e-01,  3.9057e-01, -1.4714e-01,\n",
       "          -1.9120e-01, -5.8890e-02, -1.4704e-02,  3.1907e-01,  4.7006e-01,\n",
       "          -1.0671e-01, -1.3450e-01, -1.9889e-01, -1.4310e-01, -7.9072e-01,\n",
       "          -1.1242e-01, -1.9501e-01, -1.3221e-01,  1.4454e-01, -2.0447e-01,\n",
       "          -4.3879e-02,  2.7153e-01,  3.0435e-01,  4.5489e-01, -1.6971e-01,\n",
       "          -1.8977e-02, -2.2956e-01,  4.0886e-01, -2.4201e-01, -2.4806e-01,\n",
       "           6.6141e-02,  7.4474e-01, -2.9456e-01, -6.8309e-01, -3.2309e-01,\n",
       "          -7.5136e-02,  4.0794e-01, -1.5002e-01, -4.1268e-01,  1.3313e-01,\n",
       "          -6.9902e-01, -3.2851e-01, -2.2412e-01,  1.7217e-01,  4.3843e-01,\n",
       "           3.3872e-01,  7.1752e-02,  3.5039e-01, -6.0239e-02, -3.1152e-01,\n",
       "          -5.4982e-01,  1.6474e-01,  5.3477e-01, -3.2579e-01, -1.4422e-01,\n",
       "          -2.9923e-01,  1.8962e-01, -2.0012e-01,  3.8822e-01, -1.0360e-01,\n",
       "           3.5957e-01,  3.0636e-01,  1.0951e-01,  1.6140e-01, -7.7761e-02,\n",
       "          -3.6255e-01, -1.3840e-01, -3.1073e-02, -2.3684e-01, -2.4263e-01,\n",
       "          -8.3673e-02,  3.2691e-01,  1.3861e-01,  5.5616e-02, -4.6237e-02,\n",
       "          -1.0679e-01, -5.6120e-02, -3.5739e-01,  4.1254e-01,  3.5298e-01,\n",
       "          -2.9939e-02, -3.2833e-01, -2.0567e-01,  4.4117e-02,  2.1890e-01,\n",
       "           1.8395e-01,  7.3882e-02, -4.2237e-01, -2.2655e-01, -5.2365e-02,\n",
       "           7.8551e-02,  1.0656e+00, -3.0533e-01, -3.1606e-01,  4.3515e-01,\n",
       "           2.8656e-01, -2.4544e-01,  4.5316e-02, -1.5213e-01,  1.7535e-01,\n",
       "          -2.9850e-01, -6.5694e-03,  4.0275e-01, -1.8633e-01,  3.4687e-01,\n",
       "           3.5960e-01,  2.0786e-02, -1.9508e-01,  5.5870e-02,  2.4668e-01,\n",
       "          -6.4399e-01,  1.4322e-01, -2.1666e-01, -3.1450e-01,  2.2059e-01,\n",
       "          -1.2512e-01, -2.1697e-01, -1.6908e-01,  2.6648e-01,  4.5133e-01,\n",
       "           3.7059e-01,  1.4688e+00, -1.6580e-01, -3.7369e-01,  2.9542e-01,\n",
       "          -1.2106e-01, -8.0145e-02,  5.1626e-02, -5.4733e-02,  4.0264e-01,\n",
       "           3.1412e-01,  1.8444e-01, -3.9237e-01,  2.4295e-01,  8.4188e-02,\n",
       "           2.7580e-01, -7.9190e-01, -5.6408e-01,  1.5078e-01,  6.8241e-01,\n",
       "          -9.2218e-02,  4.0866e-01, -2.0246e-01,  6.0937e-01,  3.5157e-01,\n",
       "          -9.0843e-01,  2.7436e-02, -2.0056e-01,  2.6305e-02, -2.7466e-01,\n",
       "          -8.5844e-01, -4.1246e-01,  1.9467e-01, -3.3795e-01,  2.0586e-01,\n",
       "           4.9487e-01, -4.4447e-01,  3.7711e-01,  2.9692e-01, -5.0355e-01,\n",
       "          -2.3662e-02,  1.4733e-01,  8.7750e-02,  4.4271e-01, -2.5670e-01,\n",
       "           3.5255e-02,  2.5796e-02, -1.0522e-01, -4.9771e-01,  3.5518e-01,\n",
       "           3.9843e-01, -9.3510e-02,  3.1672e-02,  1.2752e-01, -3.2974e-02,\n",
       "          -1.5314e-01, -9.5101e-03, -1.6265e-01,  6.3879e-03,  4.0280e-01,\n",
       "           2.7129e-01, -6.8549e-02,  2.8482e-02,  3.4789e-01, -3.0519e-01,\n",
       "           5.4380e-02,  4.4457e-01,  6.5059e-01, -3.1864e-01, -2.6643e-01,\n",
       "           2.9482e-02,  4.4152e-01, -3.4081e-01,  6.0846e-01, -2.0128e-01,\n",
       "          -1.5510e-01,  2.6330e-01, -7.9246e-02, -7.2553e-01,  4.1120e-04,\n",
       "          -1.9464e-01,  1.1808e-01, -8.1188e-02, -2.1540e-01,  3.8268e-01,\n",
       "          -2.3934e-02,  6.2175e-02, -9.6714e-01,  7.0050e-02,  2.9975e-01,\n",
       "          -2.9598e-01,  6.6966e-01,  2.9537e-01,  3.9950e-01, -4.3592e-02,\n",
       "           1.7007e-01, -5.5347e-01, -2.6863e-01,  2.0554e-01,  1.3720e-01,\n",
       "          -3.3198e-01, -2.7782e-01,  8.9175e-02,  1.3489e-01,  7.1948e-01,\n",
       "           2.6332e-01, -9.8623e-02, -4.7869e-01, -1.4986e-01, -3.1686e-02,\n",
       "          -3.7722e-02, -2.3903e-01,  4.6162e-01, -1.3987e-01,  2.5558e-01,\n",
       "          -6.7818e-01, -1.2329e-01, -7.0457e-02,  2.2930e-01, -1.0551e-01,\n",
       "           5.2744e-01, -5.0530e-01,  6.4910e-01,  7.2124e-01,  2.9144e-01,\n",
       "           3.0397e-01,  4.2049e-01,  2.8500e-01, -5.9819e-01, -8.0852e-02,\n",
       "           5.4096e-01, -6.9373e-01, -7.5145e-02, -5.2384e-01,  2.5228e-01,\n",
       "           3.4507e-02,  6.9940e-01,  6.5773e-01]], device='mps:0'),\n",
       " tensor([[ 1.1277e-01, -5.1182e-01, -5.1456e-01, -9.9081e-02, -3.0116e-01,\n",
       "           3.3007e-03, -4.0947e-01,  4.0427e-01, -5.8056e-02,  7.3590e-01,\n",
       "           2.2527e-01,  8.2791e-02, -3.8724e-02, -7.5126e-02, -1.5840e-01,\n",
       "          -7.3986e-01,  3.8029e-02,  3.1506e-01,  2.4049e-01,  1.5258e-01,\n",
       "           8.4897e-01,  3.7601e-01,  3.7846e-02,  1.5338e-01, -8.9169e-01,\n",
       "           2.4549e-02,  1.7534e-01, -3.8121e-01,  3.0474e-01, -9.4057e-01,\n",
       "          -3.1590e-02,  4.1782e-01, -3.6549e-01,  9.4881e-01, -2.1749e-01,\n",
       "          -2.3738e-01,  2.7768e-01,  1.1423e-01,  1.5727e-03,  8.7533e-02,\n",
       "           6.3608e-01, -3.9554e-01,  3.5800e-01,  1.7469e-01,  5.4999e-01,\n",
       "          -9.7317e-02,  1.1774e-02, -5.5940e-01,  4.2788e-01, -2.4478e-01,\n",
       "          -8.0416e-02, -1.6983e-01,  3.4295e-01,  4.9191e-01, -6.8602e-01,\n",
       "          -5.2333e-01, -6.2734e-02,  1.8164e-01, -2.1619e-01,  2.9696e-01,\n",
       "           1.3140e-01,  5.4967e-01,  1.4880e-01,  8.9218e-02, -9.8997e-02,\n",
       "           2.7962e-01, -6.9163e-01, -1.1292e-01, -4.6817e-01, -1.7222e-01,\n",
       "           7.9175e-02,  3.5231e-01, -2.4967e-01, -7.3913e-02,  7.4744e-01,\n",
       "          -3.1754e-01,  3.1779e-01, -1.0437e-01, -5.2653e-01,  2.5923e-01,\n",
       "           5.3624e-01,  2.2665e-01, -3.8162e-01, -4.3491e-01, -2.9896e-01,\n",
       "           3.1739e-01,  3.6286e-01,  2.1409e-01,  4.1664e-01,  2.3004e-01,\n",
       "           6.9510e-01,  1.1862e-01, -9.0323e-01, -1.9789e-01,  3.0084e-01,\n",
       "          -2.8430e-01,  3.6802e-01, -1.0774e-01,  1.9963e-01,  2.4580e-01,\n",
       "           9.4402e-02,  1.0678e-01, -1.2301e-01,  3.3043e-01, -7.3431e-01,\n",
       "          -3.8387e-01,  2.7538e-01,  7.5278e-02, -7.1095e-01, -5.3743e-02,\n",
       "          -1.3470e-01,  2.0865e-01,  1.6720e-01, -4.9918e-01, -2.6117e-01,\n",
       "           8.9412e-02, -1.5191e-01,  3.4209e-02, -4.2780e-01,  3.6346e-01,\n",
       "           2.9865e-01, -4.8785e-01,  1.3818e-01,  1.4371e-01,  4.4715e-01,\n",
       "          -4.1340e-01,  3.8903e-01, -2.5446e-01,  2.2889e-01,  8.2835e-01,\n",
       "          -1.2485e-01,  2.6252e-02, -4.0830e-02, -1.7487e-01, -2.8440e-01,\n",
       "          -3.4081e-01,  2.2259e-01, -7.0968e-01, -2.4259e-01,  1.8416e-01,\n",
       "           2.6748e-01, -3.7059e-02, -4.8123e-01, -1.1090e-01,  2.2158e-01,\n",
       "          -2.0902e-01, -3.0049e-01,  7.2718e-02, -2.7801e-01, -3.0796e-01,\n",
       "          -1.0773e+00, -6.4338e-01,  3.3934e-01,  1.0304e-01, -8.3425e-02,\n",
       "          -3.2985e-02,  2.6094e-01,  1.2604e-01, -1.1825e-01,  2.1861e-02,\n",
       "           1.4631e-01,  1.5870e-01,  4.0601e-01, -7.2045e-01, -2.6279e-01,\n",
       "           5.1222e-01, -6.9927e-02, -1.1101e-01, -3.8262e-01,  1.1619e-01,\n",
       "           9.3115e-02, -4.6417e-01,  1.4190e-02, -5.3166e-01,  2.5739e-01,\n",
       "          -6.9917e-01,  2.6343e-01,  7.8697e-02,  4.6975e-02, -5.2005e-01,\n",
       "          -5.1269e-01, -1.6613e+00,  4.4969e-02, -1.1536e-01,  4.2832e-01,\n",
       "          -6.4780e-01,  6.8155e-01,  4.9667e-01,  1.3957e-01, -1.0725e-01,\n",
       "          -1.1613e-01, -2.0495e-01, -1.3738e-01,  4.8684e-01, -1.1487e-01,\n",
       "           4.7349e-01,  6.1281e-01,  3.6342e-01,  1.0458e-01,  3.8659e-02,\n",
       "          -5.1090e-01,  3.2178e-01, -6.3780e-02,  6.5283e-02, -4.0803e-03,\n",
       "           5.2045e-01, -1.5508e-01, -5.5974e-02,  4.7542e-01, -9.0455e-02,\n",
       "           2.1110e-02, -5.0237e-01, -2.0562e-01, -1.7373e-01,  3.3842e-02,\n",
       "           6.1372e-01,  9.7513e-02, -1.1752e-01,  6.5636e-01,  6.4063e-01,\n",
       "          -4.3595e-02, -1.1716e-01, -2.8002e-01, -4.5596e-01, -3.9645e-01,\n",
       "           4.4838e-01, -4.1673e-01, -3.8331e-02, -1.3722e-01, -2.2336e-03,\n",
       "           2.9366e-01, -7.7720e-01,  7.8321e-01,  2.7313e-01,  1.2754e-01,\n",
       "          -2.9649e-01, -3.8133e-01, -8.9398e-02,  1.1919e-01,  8.8294e-02,\n",
       "           8.6567e-02, -6.9762e-01, -4.4501e-01,  2.6233e-01, -1.7112e-02,\n",
       "           2.9408e-02, -4.8063e-01,  9.5151e-01,  3.2366e-02,  2.1866e-01,\n",
       "          -8.1848e-02, -2.3095e-01,  2.8605e-01,  5.2168e-02, -4.2972e-01,\n",
       "          -1.1705e-01, -5.8341e-02, -3.5230e-01,  3.8333e-01, -5.6102e-01,\n",
       "          -2.8672e-01,  4.4379e-02,  2.3252e-01,  3.8893e-01, -1.8287e-02,\n",
       "           3.8934e-01, -9.1873e-02, -5.2896e-01, -2.6387e-01,  3.8154e-01,\n",
       "          -2.8519e-01,  3.6897e-01, -4.2110e-01, -2.0175e-01, -6.3157e-01,\n",
       "          -7.1356e-01,  3.0949e-01, -4.4762e-01,  2.3137e-01, -9.6326e-01,\n",
       "          -4.6904e-01,  1.5038e-01, -3.9511e-02,  7.9127e-02, -2.8288e-01,\n",
       "           3.9424e-01,  2.3657e-01,  1.0638e-01, -1.3625e-02, -6.8624e-01,\n",
       "          -1.7361e-01,  5.1713e-01, -3.0739e-01,  3.0249e-01,  3.2178e-01,\n",
       "          -1.5219e-01, -2.7008e-01,  2.0505e-01, -6.8380e-02, -1.3222e-01,\n",
       "          -2.4847e-01, -2.3051e-01,  7.3440e-02, -2.9512e-01, -4.0589e-01,\n",
       "          -1.9683e-01,  3.5653e-01,  3.1357e-01, -1.9608e-01, -4.0936e-01,\n",
       "          -3.1297e-02,  6.9414e-01, -8.3245e-02,  2.4383e-01, -1.2362e-01,\n",
       "          -1.4305e+00, -4.0833e-01, -1.8540e-02, -1.3066e-01, -2.5434e-01,\n",
       "           2.5155e-01,  6.0078e-02, -4.4774e-02, -2.3862e-02, -3.9032e-01,\n",
       "           4.2100e-01,  9.6530e-02, -5.5141e-01,  1.4950e-01, -2.0711e-01,\n",
       "          -1.2827e-01, -1.0196e-01,  5.8219e-02,  1.1231e-02, -1.1398e-01,\n",
       "           1.6417e-01, -2.6502e-02,  2.1900e-01,  8.7820e-02, -3.6880e-01,\n",
       "           5.6536e-01, -4.0455e-01, -9.2397e-02,  2.2993e-01, -6.7463e-01,\n",
       "          -5.8048e-01,  3.5097e-01,  2.4469e-01, -1.4927e-01,  8.5738e-02,\n",
       "           2.5571e-01, -1.1733e-01, -1.8922e-01,  1.1565e-01,  5.0862e-01,\n",
       "           1.6608e-01,  1.5372e-01, -3.3662e-01,  6.7240e-01,  6.1452e-01,\n",
       "          -6.7974e-01,  3.0024e-01, -8.6431e-02, -7.1191e-02, -7.8438e-02,\n",
       "           2.5682e-01,  2.9384e-01, -2.6061e-01,  3.1474e-01,  1.1061e-01,\n",
       "           8.3275e-02, -1.8821e-02, -4.8232e-02,  1.8014e-01,  6.5893e-01,\n",
       "          -6.6205e-02, -4.0549e-01,  3.5343e-01, -4.1343e-01, -2.6268e-01,\n",
       "          -7.5984e-02, -3.7208e-02,  6.5495e-02, -3.1240e-01, -3.4070e-01,\n",
       "          -4.4867e-01, -4.5409e-02,  5.9015e-01, -1.3339e-01, -1.2243e-01,\n",
       "           6.4085e-03,  1.1937e-01, -3.9825e-01,  4.0693e-01, -1.6292e-01,\n",
       "          -6.7273e-03,  1.6511e-01, -4.9061e-01,  5.9405e-01,  2.2158e-01,\n",
       "          -1.9118e-01,  7.7159e-02,  3.7849e-01,  2.9315e-01,  7.3291e-02,\n",
       "          -6.2785e-01, -1.9814e-01,  1.2483e-01,  8.7468e-02,  1.3790e-01,\n",
       "           5.2048e-01,  8.7935e-01, -4.8997e-01,  1.1694e-01, -4.3027e-02,\n",
       "           1.9913e-01, -2.3995e-01,  3.2426e-01,  2.5199e-02, -6.5157e-01,\n",
       "           5.7959e-01, -5.5682e-02,  1.2983e-01, -8.3751e-04, -1.1148e+01,\n",
       "          -2.4697e-02,  6.7893e-01, -5.5838e-01,  2.0324e-01,  1.4167e-01,\n",
       "           2.7728e-01, -3.5224e-01,  2.1719e-04, -1.7645e-01, -4.1684e-01,\n",
       "          -2.9137e-01,  1.3225e-01,  2.1139e-01, -2.9905e-01,  8.3671e-02,\n",
       "          -6.6111e-02, -7.4170e-01,  4.0852e-01,  6.7087e-02,  3.1935e-01,\n",
       "          -8.0690e-02,  3.8597e-01, -1.1090e-01, -1.3520e-02, -3.8995e-01,\n",
       "           1.1861e-01,  2.4902e-02,  5.2355e-02,  3.9441e-02, -2.2822e-01,\n",
       "           2.3118e-01,  4.8091e-02,  4.5015e-01, -2.8330e-01, -7.0757e-01,\n",
       "          -1.9601e-01, -1.5070e-02,  1.7927e-02, -1.7013e-02, -2.8357e-01,\n",
       "           6.6553e-02, -5.8297e-01,  2.2333e-01,  1.1373e-01,  8.5199e-01,\n",
       "           9.8358e-01,  6.7080e-01,  7.6940e-02, -5.4386e-02, -1.6916e-02,\n",
       "          -5.6321e-01,  1.6130e-01, -8.1082e-01, -1.3062e-01, -2.1928e-01,\n",
       "           7.0352e-01, -3.6822e-01,  1.1476e-01,  2.6222e-01, -3.4481e-01,\n",
       "           2.0382e-01,  3.7136e-02,  3.1790e-01, -4.3110e-01, -5.0590e-01,\n",
       "           5.2232e-01, -7.6970e-03,  1.8870e-01, -5.3201e-01,  2.1314e-01,\n",
       "          -3.1748e-01,  2.6052e-01,  1.8880e-01, -5.2874e-01, -2.0760e-01,\n",
       "          -4.9469e-01,  1.3613e-01,  8.4277e-01,  1.4568e-01,  7.7079e-02,\n",
       "           3.6946e-01,  2.0022e-01, -4.6711e-01, -7.1561e-01, -4.9342e-01,\n",
       "           2.4731e-01, -2.3498e-01,  3.4844e-01,  5.7382e-01, -7.8887e-02,\n",
       "          -4.7290e-01,  1.1404e-01,  1.6852e-01,  2.4731e-01,  6.4383e-01,\n",
       "          -3.0459e-01, -4.1848e-02, -2.9485e-02,  9.0120e-02, -3.1542e-01,\n",
       "          -3.7731e-01, -3.2564e-01, -1.7041e-01,  4.5536e-01, -4.5159e-02,\n",
       "           3.8940e-01,  2.2186e-01,  5.9770e-02,  1.9004e-01, -8.8353e-01,\n",
       "           3.4893e-01, -2.2614e-01,  3.7178e-01, -1.8391e-01, -4.1715e-01,\n",
       "           2.0961e-01,  5.8875e-01, -2.8693e-01, -4.0336e-01, -2.5849e-01,\n",
       "           2.3852e-01,  1.3124e-01, -1.3747e-01, -8.3652e-01,  1.2572e-01,\n",
       "          -6.3280e-01, -8.3095e-02, -2.7914e-02,  5.4686e-01,  8.0290e-01,\n",
       "           2.7624e-01, -6.7492e-02,  8.6376e-02, -3.8386e-01, -6.1207e-01,\n",
       "          -6.3800e-01,  1.3163e-02, -1.3512e-01, -7.8098e-01,  2.1135e-02,\n",
       "           2.8876e-02, -2.0300e-02,  5.8427e-02, -2.3546e-01,  1.7729e-01,\n",
       "           4.8705e-01,  3.6141e-01, -1.2408e-01,  5.9419e-02, -3.7922e-01,\n",
       "          -6.2190e-01, -8.6745e-02,  7.7854e-02,  1.6172e-01, -5.1278e-01,\n",
       "          -2.9601e-01,  1.1747e-01, -3.1322e-02,  2.7528e-02, -1.9899e-01,\n",
       "           1.3669e-01, -2.2503e-01, -1.2000e-02,  2.7866e-01,  4.6635e-01,\n",
       "           1.1528e-02, -2.7282e-01, -4.3857e-01, -1.8906e-02,  1.5881e-01,\n",
       "           2.4698e-01, -2.2711e-01, -3.0300e-01, -3.8354e-01,  2.0940e-01,\n",
       "           2.9705e-01,  1.0639e+00, -4.8833e-01, -4.0899e-01,  1.9583e-01,\n",
       "           3.5230e-01, -1.0565e-01, -2.8806e-01, -3.5628e-01,  2.0647e-01,\n",
       "           1.1352e-01, -8.0433e-02, -3.4162e-02, -7.7830e-02,  5.1550e-01,\n",
       "           1.6482e-01,  4.7041e-01,  3.0757e-01,  4.8310e-01,  1.9579e-01,\n",
       "          -1.9577e-01, -8.1087e-02, -6.1276e-01, -5.7265e-01,  1.7477e-01,\n",
       "          -3.4696e-02,  1.7945e-01,  2.9475e-01, -5.4680e-01,  3.5253e-01,\n",
       "           3.9298e-01,  1.0477e+00,  3.8225e-02, -1.9548e-01,  4.0182e-01,\n",
       "           3.8969e-01, -4.5758e-01, -5.5331e-02,  3.1775e-01,  9.0078e-02,\n",
       "           4.2217e-01,  2.6969e-01, -2.5045e-01, -2.6832e-01,  1.3747e-01,\n",
       "           3.6461e-01, -5.7917e-01, -3.7681e-01, -1.5167e-02,  5.5110e-01,\n",
       "          -2.4375e-01,  2.1840e-01,  4.5848e-02,  3.7804e-01,  3.7463e-01,\n",
       "          -1.0699e+00,  1.6807e-01, -3.2041e-01,  2.7435e-01, -2.4271e-02,\n",
       "          -7.8250e-01, -3.4543e-02, -1.9427e-01, -1.1596e-01,  3.3735e-01,\n",
       "           4.1202e-01, -7.9805e-01,  3.3972e-01,  2.3964e-01, -3.8364e-01,\n",
       "           4.7577e-02, -1.7531e-01,  2.4049e-01,  3.3544e-01,  3.2362e-02,\n",
       "          -1.4892e-01,  1.8644e-01, -1.9447e-01, -8.0775e-01,  5.7712e-01,\n",
       "           6.0096e-01, -2.1005e-01,  1.4164e-01,  1.1858e-01, -8.4184e-02,\n",
       "          -3.2385e-01, -3.8343e-01, -1.2416e-01,  1.6063e-01,  9.6145e-02,\n",
       "          -9.8046e-02, -4.7037e-01, -4.0440e-02,  3.8794e-01, -3.9096e-01,\n",
       "           4.3372e-01,  3.4804e-01,  5.3842e-01, -2.8332e-01, -4.2575e-02,\n",
       "           1.7443e-01,  5.7542e-01, -1.1207e-01,  3.8094e-01, -3.8443e-01,\n",
       "          -8.6311e-03,  4.5158e-01, -1.4772e-01,  7.1035e-03,  2.3732e-01,\n",
       "          -5.0991e-01,  2.9103e-01,  1.0851e-01, -5.6297e-01,  5.5767e-01,\n",
       "           5.0509e-01, -7.6896e-02, -4.0047e-01,  1.5744e-01,  2.9919e-01,\n",
       "           3.9453e-02,  7.8684e-01,  2.8627e-01,  3.8282e-01,  2.7073e-01,\n",
       "          -6.7450e-02, -5.4737e-01, -3.2683e-01, -5.6482e-03,  2.6335e-01,\n",
       "           1.1920e-01, -3.6386e-01,  3.2594e-01,  1.8073e-01,  5.2975e-01,\n",
       "           6.1907e-01, -1.8066e-01, -9.0544e-01, -5.6162e-01,  9.6928e-02,\n",
       "          -1.8944e-01, -1.1782e-01, -1.4313e-01,  5.4460e-02,  1.6554e-01,\n",
       "          -7.7022e-01, -5.7397e-01, -1.9516e-01,  2.1820e-01, -3.5735e-01,\n",
       "           6.6382e-01,  4.6434e-02, -3.8504e-02,  3.3171e-01,  3.8933e-01,\n",
       "           2.1859e-01,  3.6118e-01, -5.0593e-02, -5.7660e-01,  1.3136e-02,\n",
       "           9.6468e-02, -4.2191e-01, -1.8386e-01, -4.0805e-01,  5.0626e-01,\n",
       "           3.3973e-01,  9.6001e-01,  5.3169e-01]], device='mps:0'),\n",
       " tensor([[-0.1672, -0.5249, -0.4540,  ...,  0.3771,  0.2016,  0.5340],\n",
       "         [-0.2188, -0.5875, -0.1788,  ..., -0.1233,  0.2447, -0.1722]],\n",
       "        device='mps:0'),\n",
       " tensor([[-0.4566, -0.4704, -0.6432,  ...,  0.0311,  0.9323,  0.0764],\n",
       "         [-0.2105, -0.6075, -0.1881,  ...,  0.1049,  0.5933,  0.7102],\n",
       "         [-0.4615, -0.5586, -0.3419,  ...,  0.2423,  0.2570,  0.4881],\n",
       "         [-0.2628, -0.8360, -0.4189,  ..., -0.0075,  0.5654,  0.4481]],\n",
       "        device='mps:0'),\n",
       " tensor([[-2.2115e-01, -4.0785e-01, -5.7008e-01, -8.4110e-01, -1.8063e-01,\n",
       "          -6.4874e-02, -8.1797e-01,  4.8747e-01,  1.3876e-01,  5.1025e-01,\n",
       "           1.3308e-01,  4.5201e-01,  1.2124e-01, -2.2697e-02, -3.0274e-01,\n",
       "          -2.7463e-01,  1.9110e-01,  3.0301e-01,  5.1059e-01, -4.2661e-02,\n",
       "           2.0994e-01,  5.2801e-01, -8.4736e-02, -2.8641e-02, -6.8592e-01,\n",
       "           2.9997e-01, -7.5231e-02,  8.7288e-02,  2.4767e-01, -3.5634e-01,\n",
       "           1.3398e-02,  4.1260e-01, -1.3038e-01,  1.5028e+00, -3.4582e-01,\n",
       "           8.9252e-03,  1.2309e-01, -3.5245e-01,  2.9210e-02,  8.3731e-02,\n",
       "           5.9744e-01, -3.4789e-01,  9.0473e-02,  4.4851e-01, -8.2971e-02,\n",
       "           2.0044e-01,  2.3719e-01, -3.5455e-01,  2.5407e-01, -9.2699e-01,\n",
       "          -2.0832e-01, -2.7803e-01,  5.0140e-01,  5.6522e-01, -5.5595e-01,\n",
       "          -2.6412e-01,  2.5578e-01,  7.9181e-02, -4.0664e-01,  1.7214e-01,\n",
       "           5.3818e-02,  4.7333e-01, -1.7820e-01,  2.1154e-01,  2.0118e-01,\n",
       "           3.0639e-01, -2.2901e-01, -4.6961e-01, -5.2300e-01, -2.1040e-01,\n",
       "          -6.5956e-02,  2.9151e-02,  5.5275e-02, -2.6878e-01,  5.1983e-01,\n",
       "          -4.5297e-01,  6.7164e-02, -2.5771e-01, -1.3175e-01,  2.1727e-01,\n",
       "          -7.8943e-02,  2.4914e-01,  1.9719e-01,  1.9665e-01,  1.5592e-01,\n",
       "           2.6325e-02,  1.3043e-01,  6.5136e-02,  3.4640e-02,  2.9002e-01,\n",
       "          -2.1763e-01, -7.0018e-02, -2.9948e-01, -2.1331e-01,  2.5006e-01,\n",
       "           1.1040e-01,  2.2652e-01, -1.7122e-01,  2.2869e-01, -1.1475e-01,\n",
       "           1.3421e-01, -2.5993e-02, -1.1727e-01, -6.2681e-03, -2.7773e-01,\n",
       "          -4.5485e-01,  1.3560e-01, -2.8137e-01, -5.5872e-01,  1.5983e-01,\n",
       "          -2.2923e-01, -3.3457e-01,  4.2271e-02, -4.2907e-01,  1.4274e-02,\n",
       "          -1.2454e-01,  1.0327e-01, -3.2173e-02, -1.8872e-01,  5.5090e-03,\n",
       "           2.3560e-01, -1.5391e-01,  1.3128e-01,  1.7807e-01,  2.4997e-01,\n",
       "          -2.5083e-02,  5.3017e-01,  2.9607e-02,  1.8693e-01,  3.3842e-01,\n",
       "           1.3163e-01, -1.7996e-01, -2.2352e-01, -6.3588e-02,  6.6907e-03,\n",
       "           3.6669e-02,  4.0967e-02, -1.5772e-01,  8.7341e-02, -5.9198e-02,\n",
       "           1.8809e-01, -2.1040e-01, -6.5251e-01, -4.9339e-01, -1.3446e-01,\n",
       "          -2.3077e-01, -6.3704e-02, -5.1876e-02, -4.8677e-01, -3.4766e-01,\n",
       "          -7.5440e-01, -3.9442e-01,  6.9877e-01, -1.6336e-01, -2.0169e-01,\n",
       "          -3.6948e-04,  2.8772e-01,  1.5802e-01, -1.0174e-01,  2.7112e-01,\n",
       "           7.4345e-01, -1.1905e-01,  4.1394e-02, -4.8582e-01, -2.5506e-01,\n",
       "           6.4343e-02,  7.3559e-02, -3.7165e-01,  1.4194e-01,  1.1692e-01,\n",
       "           8.7149e-02, -1.9535e-01,  1.8104e-01, -2.8488e-01,  1.1084e-01,\n",
       "          -6.4854e-01,  3.3547e-01,  2.6510e-01, -2.0923e-01, -1.2559e-01,\n",
       "           3.1449e-02, -1.3046e+00,  1.1282e-01, -6.7069e-01,  1.9442e-01,\n",
       "          -1.7050e-01,  1.4410e-01,  3.1129e-01,  2.6171e-01, -2.4990e-02,\n",
       "           2.6322e-01,  1.2807e-01,  1.7444e-01,  3.1132e-01, -6.1163e-02,\n",
       "           1.5101e-01,  2.9770e-01,  3.8045e-01,  2.5646e-01,  1.2933e-01,\n",
       "          -3.9422e-01,  3.2834e-01,  3.6373e-01,  1.8673e-01,  1.8082e-01,\n",
       "           1.4619e-01, -1.4171e-01,  1.3173e-01,  5.8809e-02, -1.9359e-01,\n",
       "          -2.2467e-01,  5.8982e-02, -4.6934e-01, -1.6353e-02, -2.7697e-01,\n",
       "           1.9581e-01, -3.3842e-01,  1.0616e-03,  5.1638e-01,  2.7234e-01,\n",
       "           6.1655e-02,  2.4680e-01,  9.9117e-02, -3.2177e-01,  2.4660e-01,\n",
       "           2.5621e-01, -3.2565e-01,  6.3398e-02, -4.0649e-01, -2.5452e-01,\n",
       "           2.8386e-01, -6.1659e-01,  5.2713e-01,  5.2735e-01,  1.1534e-02,\n",
       "           4.4759e-02, -3.7464e-01,  3.4684e-02, -4.2542e-01,  3.8320e-01,\n",
       "          -2.7552e-01, -4.4780e-01, -5.7812e-01,  7.0197e-02,  5.0748e-01,\n",
       "           2.8194e-01,  1.5510e-02,  7.2880e-01,  3.2546e-01,  3.6210e-01,\n",
       "          -3.0822e-01, -4.0243e-02,  3.8499e-01,  1.0456e-01, -3.8188e-01,\n",
       "           1.9314e-02, -3.2356e-02, -4.2720e-01,  2.7195e-01, -2.7270e-01,\n",
       "          -1.4005e-01, -1.0777e-01,  3.6784e-01,  3.4209e-01, -3.7914e-01,\n",
       "          -1.3150e-01,  3.0634e-01, -1.0394e-01, -4.3525e-01,  2.9672e-01,\n",
       "          -4.4497e-01,  1.8905e-01,  1.2563e-01, -3.6693e-01, -5.7399e-01,\n",
       "          -3.2053e-01, -6.9191e-02, -1.7400e-01,  1.1770e-01, -7.2510e-01,\n",
       "          -4.5610e-01,  4.9362e-02,  1.0284e-01,  1.0720e-01,  2.9626e-01,\n",
       "           3.9273e-01, -2.3658e-01, -1.7200e-01, -1.3320e-01, -5.5932e-01,\n",
       "          -8.2797e-02,  5.6099e-02, -3.1822e-01,  3.1942e-01,  2.0091e-01,\n",
       "           1.4024e-02,  1.0330e-01,  4.1367e-01, -2.9776e-01, -3.0643e-01,\n",
       "          -4.9441e-02,  5.0167e-02, -2.3929e-01, -1.0467e-01,  2.5486e-01,\n",
       "           5.0751e-02,  4.1962e-01,  9.0461e-01,  5.9333e-02, -5.0320e-01,\n",
       "           1.3741e-01,  6.1045e-01, -3.5701e-01,  6.5412e-03, -1.2161e-01,\n",
       "          -1.0206e+00, -2.4188e-01, -1.5891e-01, -2.9731e-01, -5.3073e-01,\n",
       "           1.0215e-01,  2.2052e-01,  1.2428e-02, -1.5334e-01, -3.7618e-01,\n",
       "           2.0821e-01,  2.2134e-01, -1.1328e-01,  3.9144e-01, -3.0264e-01,\n",
       "          -2.6101e-01, -2.3275e-01, -3.4418e-01, -4.6901e-01, -4.2015e-01,\n",
       "           3.5280e-01,  2.2192e-01,  1.2231e-01, -1.6684e-01, -4.1457e-01,\n",
       "           4.8600e-01, -5.5057e-01, -1.4724e-02,  2.6035e-01, -4.3419e-01,\n",
       "          -1.6682e-01,  5.0294e-01, -4.1691e-02,  2.1870e-01,  1.6419e-01,\n",
       "           1.6186e-01, -3.7523e-01, -5.2219e-01, -2.5891e-01,  6.5956e-01,\n",
       "           2.7140e-01,  3.6756e-01, -2.2686e-01,  2.0260e-01,  5.0353e-01,\n",
       "          -2.8142e-01,  4.8572e-01, -1.2124e-01, -4.3083e-02, -1.7088e-01,\n",
       "           3.3554e-01,  3.0027e-01, -2.2149e-01,  3.4676e-01,  2.4224e-01,\n",
       "          -2.9572e-01, -1.1043e-01, -3.0894e-02,  3.4511e-01,  2.3173e-02,\n",
       "          -9.9270e-02, -2.3328e-01,  1.7864e-01, -2.5785e-01, -3.1786e-01,\n",
       "           2.5711e-01, -1.5047e-01,  1.3902e-01, -4.1619e-01, -2.7322e-01,\n",
       "          -1.9672e-01, -2.2865e-01,  7.9684e-01,  1.7149e-01, -2.7634e-01,\n",
       "           2.1428e-01, -2.1926e-01, -3.9203e-01,  5.0543e-01,  1.1788e-01,\n",
       "          -4.1701e-01,  4.2716e-01, -3.4446e-01,  1.0070e-01,  2.3280e-01,\n",
       "          -7.9724e-02, -1.1698e-01,  4.7742e-02,  2.7417e-01,  3.7730e-01,\n",
       "          -3.7167e-01,  3.7586e-03, -6.2826e-02, -2.0672e-01,  2.4468e-01,\n",
       "           1.7035e-01,  3.5660e-01, -1.9704e-01, -1.7483e-01,  8.3473e-02,\n",
       "           3.6290e-01, -1.1784e-01,  4.9871e-03, -4.0691e-01, -4.5798e-01,\n",
       "           3.6629e-01,  1.7803e-01,  4.8482e-01,  2.1583e-01, -1.1684e+01,\n",
       "           3.3027e-02,  7.5940e-01, -3.1420e-01,  7.9339e-02,  3.6269e-02,\n",
       "           1.4228e-03, -3.1587e-01,  3.3568e-01, -7.5273e-02, -4.2220e-01,\n",
       "          -3.2106e-01, -3.3083e-01,  2.4085e-01, -2.4356e-01, -1.1312e-01,\n",
       "           1.8976e-01, -7.2268e-01,  2.2771e-01, -2.3753e-01,  5.5671e-01,\n",
       "          -2.7443e-01, -9.3054e-02,  2.5924e-01, -2.0426e-01, -4.4191e-02,\n",
       "           2.1536e-01, -4.5893e-02,  2.2346e-01,  2.4036e-01, -4.1898e-02,\n",
       "           1.4362e-01, -1.2398e-01,  7.8931e-03, -1.7247e-01, -6.1442e-01,\n",
       "          -1.2422e-01, -3.0776e-01,  2.1553e-01, -8.6393e-02, -5.8819e-01,\n",
       "           2.4543e-02, -2.8082e-01,  2.6703e-01,  4.0834e-01,  7.3936e-01,\n",
       "           5.1571e-01,  1.1077e+00,  6.6908e-02,  1.9403e-01,  1.6779e-01,\n",
       "          -7.3383e-01,  7.2103e-02, -3.0086e-01,  5.4596e-02, -4.9264e-04,\n",
       "           2.9902e-01, -6.3275e-01,  3.2030e-01, -3.3935e-01,  1.0018e-01,\n",
       "           5.2465e-01, -3.2176e-02,  2.2974e-01, -1.6183e-01, -1.1412e-01,\n",
       "           1.6192e-01, -5.0214e-02,  1.2932e-01, -1.5990e-01, -2.4795e-01,\n",
       "           1.1974e-02,  5.1514e-01,  1.9103e-01, -1.0085e+00,  3.3074e-01,\n",
       "           1.1479e-01, -1.1113e-01,  1.3344e+00,  6.0265e-02,  2.0453e-02,\n",
       "          -7.2403e-02,  2.7630e-01, -4.8136e-01, -5.5133e-01,  3.3743e-01,\n",
       "          -1.7901e-01, -7.6146e-02,  2.4472e-01,  3.9879e-01, -2.0777e-01,\n",
       "          -4.8103e-01,  4.6200e-02,  1.6913e-01,  2.0664e-02,  5.9595e-01,\n",
       "          -8.0440e-01, -1.7446e-01, -2.9740e-02,  1.2390e-01, -6.9377e-01,\n",
       "          -2.8243e-03, -4.3000e-01, -3.3049e-01,  5.1123e-01, -2.6437e-01,\n",
       "           1.1945e-01, -1.1062e-01, -2.2750e-01,  4.4296e-01, -2.7944e-01,\n",
       "           1.6202e-02,  3.6161e-01,  5.1586e-02, -1.8236e-01, -8.4289e-02,\n",
       "           2.7268e-01,  5.5625e-01, -5.0114e-01, -4.0332e-01, -7.6952e-02,\n",
       "           1.3843e-01,  2.5794e-01,  3.1399e-01, -3.8615e-01,  3.8565e-01,\n",
       "          -8.2569e-01, -1.3949e-01,  8.5260e-02,  9.9086e-02,  4.9717e-01,\n",
       "           1.0666e-01,  1.1684e-02,  1.1151e-01, -5.6289e-01, -4.8569e-01,\n",
       "          -7.2403e-01, -1.9239e-01,  1.2706e-01, -2.3196e-01, -2.0672e-02,\n",
       "          -1.6149e-01, -4.2470e-02,  3.3131e-02,  2.0050e-01, -7.2901e-02,\n",
       "           2.0855e-01,  1.6585e-01, -4.7449e-02, -1.1259e-01, -5.3955e-02,\n",
       "          -4.2408e-01,  5.7576e-02, -8.2128e-02, -1.1104e-01, -6.4200e-01,\n",
       "          -2.0823e-01,  2.4172e-02,  8.9488e-02, -2.8029e-01,  1.9129e-01,\n",
       "           1.5103e-01,  1.4656e-01, -2.6516e-01,  6.7193e-01,  3.0966e-01,\n",
       "          -1.5960e-02, -2.6234e-01, -5.4246e-01, -4.5675e-02,  1.9935e-01,\n",
       "           2.9931e-01,  2.5905e-01, -3.2107e-01, -3.3201e-01, -3.0591e-01,\n",
       "           7.5813e-03,  6.9124e-01, -1.5810e-01, -5.0761e-01,  1.0113e-01,\n",
       "           5.7340e-01, -1.7424e-01, -3.3546e-01,  1.3036e-01,  7.7503e-02,\n",
       "          -2.6064e-01, -5.5278e-01,  6.4369e-01, -1.1135e-01,  4.1509e-02,\n",
       "           3.1584e-01,  2.0766e-01, -1.1206e-01,  5.9170e-01, -5.2692e-03,\n",
       "          -4.6380e-01, -5.2345e-02, -8.1214e-01,  3.1456e-01,  1.6943e-01,\n",
       "          -1.6479e-01,  7.6410e-01,  2.3705e-01, -3.1335e-01, -4.2930e-02,\n",
       "           4.3377e-01,  9.6831e-01, -1.0729e-01,  1.9369e-01,  6.5521e-01,\n",
       "           3.6739e-02, -2.1334e-01,  1.8032e-01,  3.7207e-02,  1.0767e-01,\n",
       "           4.5342e-01, -2.6050e-01, -4.4642e-01, -1.0399e-01,  3.6934e-01,\n",
       "           4.3893e-01, -4.5900e-01, -6.9414e-01, -1.5317e-01,  5.3691e-01,\n",
       "          -3.3230e-01,  2.9391e-01, -7.8549e-02,  3.0547e-01,  1.3460e-01,\n",
       "          -9.1134e-01, -1.1964e-01,  8.9327e-02,  4.7779e-02,  2.3483e-02,\n",
       "          -3.5310e-01, -2.0503e-01, -2.0414e-01, -1.2068e-01,  6.7507e-01,\n",
       "           2.2571e-01, -7.5395e-01,  8.1098e-02,  6.0859e-01, -7.0330e-01,\n",
       "           2.6161e-01, -3.0057e-01, -1.0814e-01,  3.7312e-01,  7.1977e-02,\n",
       "          -1.8065e-01,  1.2742e-01, -4.2657e-01, -4.1132e-01,  1.5504e-01,\n",
       "           8.8180e-01, -9.8331e-02,  1.9444e-01, -3.1352e-01, -2.6970e-01,\n",
       "          -2.9879e-01, -4.2308e-01, -3.1602e-01,  2.4105e-01,  1.0052e-01,\n",
       "          -1.3343e-02, -8.0468e-02, -3.3635e-01,  3.7359e-01,  8.4713e-02,\n",
       "           3.8641e-01,  3.9878e-01,  5.7910e-01,  2.4004e-02, -2.8970e-01,\n",
       "          -1.5561e-02,  1.0451e-01, -1.4209e-01,  4.8604e-01, -2.7948e-01,\n",
       "          -4.8096e-01,  1.9265e-02,  7.9697e-02, -5.4979e-01, -1.1590e-01,\n",
       "          -4.8090e-01,  2.4789e-01, -1.4125e-01,  1.2204e-02,  3.8391e-01,\n",
       "           1.8131e-01,  9.0521e-02, -2.9486e-01,  5.2355e-01,  5.0011e-01,\n",
       "           9.4086e-02,  5.7952e-01,  2.4944e-01,  9.6160e-02, -3.1777e-01,\n",
       "           2.8294e-01, -9.6606e-01, -1.5936e-01, -1.0321e-01, -2.2222e-01,\n",
       "          -3.4218e-01, -1.4505e-01,  1.7743e-01,  1.7824e-01,  6.2117e-01,\n",
       "           5.8925e-01, -2.2997e-01, -6.0209e-01, -1.7097e-01,  1.1398e-01,\n",
       "           1.4756e-01, -1.2023e-01,  2.0127e-01,  1.4290e-01,  3.1449e-01,\n",
       "          -5.6819e-01, -3.6697e-01,  4.5629e-02,  1.8809e-01, -1.1195e-01,\n",
       "           7.5055e-01, -2.0544e-01, -1.4336e-01, -4.5771e-02,  8.9154e-02,\n",
       "           2.1665e-01,  6.8106e-01,  2.5200e-01, -6.4751e-01, -1.0427e-02,\n",
       "           6.3059e-02, -3.0872e-01,  1.5917e-01, -6.3080e-01,  1.8315e-01,\n",
       "           2.0125e-01,  4.6589e-01,  5.2440e-01]], device='mps:0'),\n",
       " tensor([[-0.0029, -0.4260, -0.6820,  ...,  0.1889,  0.1934,  0.5362],\n",
       "         [-0.2258, -0.7099, -0.3778,  ...,  0.3292,  0.7059,  0.7494],\n",
       "         [-0.3627, -0.4382, -0.8203,  ...,  0.1541,  0.1866,  0.3640],\n",
       "         [-0.3216, -0.9058, -0.4610,  ..., -0.0331,  0.2960,  0.0087]],\n",
       "        device='mps:0'),\n",
       " tensor([[-0.2744, -0.4852, -0.4880,  ...,  0.2064,  0.2977,  0.6606],\n",
       "         [ 0.0506, -0.6230, -0.6446,  ..., -0.0875,  0.4018,  0.3101],\n",
       "         [-0.3308, -0.4627, -0.2900,  ...,  0.2546,  0.1334,  0.6058]],\n",
       "        device='mps:0'),\n",
       " tensor([[-6.1750e-01, -4.2829e-01, -2.3934e-01, -9.4116e-01, -3.4548e-01,\n",
       "          -6.5496e-02, -6.3874e-01,  9.7828e-01, -9.5892e-02,  4.0096e-01,\n",
       "           1.4919e-01, -5.1603e-03,  2.0980e-01,  1.3680e-01, -6.9506e-01,\n",
       "           1.6740e-01,  4.6094e-01,  2.1159e-01,  3.5951e-01, -4.4329e-02,\n",
       "           7.9832e-01,  3.4245e-01,  1.8019e-01, -2.5205e-01, -4.5876e-01,\n",
       "           5.8994e-01, -2.4628e-01, -9.2014e-02,  2.0509e-01, -8.1949e-01,\n",
       "          -1.3586e-01, -4.9757e-02, -3.6281e-01,  8.7602e-01,  1.1948e-01,\n",
       "          -2.7310e-01, -1.5276e-01, -2.7996e-01,  2.3494e-01, -6.3637e-02,\n",
       "           3.7814e-01,  2.6592e-02, -2.9347e-01,  3.1386e-01, -1.6421e-01,\n",
       "          -1.4057e-01,  1.4047e-02, -7.8429e-01,  1.3666e-02,  1.8955e-01,\n",
       "           5.8851e-02, -4.7205e-02,  1.9231e-01,  3.7030e-01, -1.0617e+00,\n",
       "          -2.6348e-01,  2.7603e-01,  2.9849e-01, -1.7932e-01, -5.7784e-02,\n",
       "          -1.9989e-01,  9.0201e-01,  1.7949e-01, -2.7822e-01,  1.7824e-01,\n",
       "          -4.4899e-01, -5.4601e-01, -6.0781e-01, -1.0682e-01, -1.1967e-01,\n",
       "          -1.0002e-01,  9.7528e-02,  1.4009e-01, -2.9385e-01,  2.3170e-01,\n",
       "           2.1813e-01,  4.7024e-02, -2.7652e-02, -4.3158e-01, -2.6546e-01,\n",
       "           3.6947e-01,  3.4425e-02, -1.4730e-02,  2.0326e-01, -5.8744e-01,\n",
       "          -2.0226e-01,  3.7959e-01,  1.5703e-01,  2.5914e-01,  2.2662e-03,\n",
       "           5.9967e-01,  1.2024e-02,  8.0339e-02, -4.5369e-01,  5.8427e-01,\n",
       "          -5.4955e-01,  1.0329e-01, -1.6007e-01,  4.9532e-02, -5.2088e-01,\n",
       "          -9.8735e-02,  2.4494e-01, -1.8085e-01,  1.6053e-01, -4.6592e-01,\n",
       "          -3.2983e-02,  2.4514e-01,  3.0468e-01, -4.9183e-01,  4.2685e-02,\n",
       "          -6.5134e-03, -2.9791e-01,  7.3694e-02, -6.6407e-01, -8.5795e-01,\n",
       "           5.8354e-01,  1.8428e-02, -4.5240e-01, -2.3930e-01,  7.4443e-01,\n",
       "           2.9738e-01, -2.3201e-01, -5.3072e-02, -3.5548e-01,  2.0259e-01,\n",
       "          -1.8518e-01,  2.5689e-01, -2.7018e-01,  3.8439e-01,  5.5026e-01,\n",
       "           3.1581e-01,  2.8489e-01, -1.8327e-01,  2.7637e-01,  1.4465e-01,\n",
       "           3.6947e-02, -6.6614e-02, -1.5929e-01,  1.5180e-01,  1.2041e-01,\n",
       "          -2.9580e-01, -3.2999e-01,  5.4746e-03,  1.1182e-01, -3.3279e-01,\n",
       "          -1.0202e-01,  1.8680e-01,  3.4808e-01, -1.4092e-01, -2.5403e-01,\n",
       "          -7.8984e-01, -4.2328e-01,  1.0217e-01, -1.8560e-01, -1.7135e-01,\n",
       "          -4.1911e-01,  3.8121e-01, -6.6660e-02,  4.7638e-01,  3.4189e-01,\n",
       "           6.9978e-01, -1.5311e-01,  5.9776e-02, -5.6412e-01,  3.5197e-01,\n",
       "           3.8270e-02, -6.1110e-01, -1.7247e-01,  1.5680e-01, -4.1813e-04,\n",
       "           6.8195e-01,  7.0979e-02,  4.0124e-01,  1.2070e-01, -3.2553e-01,\n",
       "           8.2089e-03,  1.9890e-01, -1.2864e-01, -1.4297e-01,  7.3766e-01,\n",
       "          -5.5188e-01, -2.3029e+00,  1.0721e-01, -3.5844e-01, -3.6529e-01,\n",
       "          -1.8135e-01,  3.7255e-01,  3.3772e-02,  4.9966e-01,  1.5486e-01,\n",
       "           3.7598e-02,  1.6729e-01, -2.8025e-02,  3.6851e-01, -8.4399e-02,\n",
       "           2.6751e-01,  3.3341e-01,  1.2988e-01, -2.3298e-01, -1.0956e-01,\n",
       "          -3.9227e-01,  2.4031e-01, -9.8916e-02,  8.6010e-01,  2.3260e-01,\n",
       "           2.0943e-01, -4.3657e-03, -4.7262e-02,  4.8964e-01,  4.3099e-01,\n",
       "          -6.6541e-02, -1.6585e-01, -4.9522e-01,  4.6269e-03, -4.1279e-02,\n",
       "          -1.2963e-02, -4.5611e-01,  1.3753e-01,  4.2841e-01, -1.2416e-01,\n",
       "          -1.8354e-01,  1.5525e-01, -4.7715e-02, -3.1517e-01, -2.6492e-02,\n",
       "           6.8453e-01,  1.9080e-01, -2.2994e-01,  1.2377e-01, -1.1896e-01,\n",
       "           3.7006e-02, -4.6975e-01,  7.0044e-01,  6.1147e-01,  1.5585e-01,\n",
       "           1.6214e-02, -6.9943e-01,  2.3699e-01, -3.1574e-01, -4.7647e-03,\n",
       "          -1.5027e-01, -1.5731e-01, -4.2986e-01, -5.7285e-02, -5.5854e-02,\n",
       "           4.0115e-01,  7.9943e-02,  4.6653e-01,  1.1837e+00,  2.7973e-01,\n",
       "           1.9184e-02,  5.6097e-01,  2.0801e-01,  4.2492e-01, -4.0180e-01,\n",
       "           3.3082e-01,  8.8825e-02, -6.3896e-01,  2.7188e-01, -4.1593e-01,\n",
       "           4.4963e-02, -4.1432e-02, -3.7926e-01,  1.4208e-01, -2.1276e-02,\n",
       "           3.4237e-01,  3.3517e-01,  1.5186e-01,  1.6838e-01,  2.9411e-01,\n",
       "          -3.1322e-01,  5.9715e-01, -6.0809e-02, -1.1127e-01, -6.3492e-01,\n",
       "          -7.6732e-01,  9.5206e-02,  4.4743e-01, -1.2880e-01, -6.1279e-01,\n",
       "           1.0462e-01, -2.3866e-01, -1.2257e-01, -3.5064e-02,  2.1191e-01,\n",
       "           2.7455e-01,  5.0687e-02, -9.5436e-02,  6.4791e-02, -4.6950e-01,\n",
       "          -4.9277e-01,  4.9140e-02, -1.5758e-01, -1.3063e-01,  5.3918e-01,\n",
       "          -7.9415e-02,  1.1575e-01,  5.7221e-01,  1.9799e-02, -4.5976e-01,\n",
       "          -2.5640e-01,  1.4802e-01,  2.8667e-01, -7.2082e-02,  2.1981e-01,\n",
       "           3.4119e-01,  2.6467e-02,  5.1649e-01,  1.4823e-01, -2.1046e-01,\n",
       "          -2.7239e-01,  5.1173e-01, -1.5755e-01,  6.1403e-02, -1.2527e-01,\n",
       "          -8.7207e-01, -5.6150e-02, -7.8413e-02, -1.6833e-01, -5.4440e-01,\n",
       "           2.6566e-01,  1.4351e-01,  1.0561e-01,  2.2350e-01, -1.5984e-01,\n",
       "           7.0123e-01,  6.5285e-01, -3.3100e-01, -1.6943e-01, -4.0104e-01,\n",
       "           9.3922e-02, -6.8374e-02, -2.9329e-01, -1.4009e-02,  9.5368e-03,\n",
       "          -2.2765e-01, -1.2516e-01, -1.0805e-01,  2.8921e-01, -1.8876e-01,\n",
       "           5.2862e-01,  1.2189e-01,  8.3649e-02,  4.3873e-01,  2.0612e-02,\n",
       "           1.3071e-01,  3.6515e-01, -4.2117e-01,  1.2870e-01,  4.0424e-01,\n",
       "           4.0578e-01, -1.0551e+00, -8.3291e-01, -1.4936e-01,  3.6111e-01,\n",
       "           2.2860e-01,  2.5189e-01,  1.6999e-02, -7.2215e-03,  8.9455e-02,\n",
       "          -2.3402e-01,  3.5135e-01, -6.6415e-02, -5.7412e-01,  3.0561e-01,\n",
       "           2.3567e-01,  5.3263e-01, -2.7377e-01, -5.0579e-01, -6.5233e-02,\n",
       "           2.4616e-01, -3.6381e-01,  1.1392e-01,  2.3480e-01,  4.5466e-01,\n",
       "          -3.6015e-01, -5.6164e-01, -7.0282e-02, -7.3986e-01, -1.3578e-01,\n",
       "           3.2483e-01, -2.1437e-01,  2.7178e-02, -2.4758e-01,  1.5231e-01,\n",
       "          -1.7925e-02, -1.8879e-01,  6.0757e-01,  2.0522e-01, -2.5855e-01,\n",
       "           1.1981e-01, -2.3892e-01, -1.9475e-01,  5.0091e-01, -5.0226e-01,\n",
       "          -3.0348e-01, -5.0960e-02, -1.5349e-01, -3.2418e-01,  3.2181e-01,\n",
       "          -4.1389e-01, -1.4861e-01,  1.6748e-01, -6.4736e-02, -1.3068e-01,\n",
       "          -3.5233e-01,  1.4268e-01, -1.2707e-01, -4.9883e-02, -1.7654e-02,\n",
       "          -1.3353e-01,  6.0900e-02,  1.3308e-01,  6.4538e-02,  1.4388e-01,\n",
       "           5.3696e-01, -1.1464e-01,  2.7582e-03, -2.5900e-01, -2.2469e-01,\n",
       "           5.2498e-01, -9.4973e-02, -3.3117e-01,  5.4928e-01, -1.1156e+01,\n",
       "           1.0386e-01,  7.2631e-01, -5.5808e-01,  5.5128e-02, -2.8539e-01,\n",
       "           2.1085e-01, -3.0009e-01,  7.9101e-02,  2.4967e-01, -2.8460e-01,\n",
       "          -1.2277e-01, -2.7364e-01, -2.3193e-01, -4.0872e-01, -3.9742e-01,\n",
       "          -4.7229e-01, -5.6789e-01,  1.1772e-01, -1.7663e-01, -1.0996e-01,\n",
       "          -5.7036e-01, -1.9379e-01,  2.9696e-01, -4.9559e-03, -5.1069e-01,\n",
       "           5.9541e-02, -2.2948e-01,  2.6399e-01,  4.6126e-02,  2.2045e-01,\n",
       "          -1.9949e-02,  1.4108e-01,  2.0669e-01, -4.4946e-02, -4.5100e-01,\n",
       "          -4.4853e-01,  2.2129e-01, -1.5955e-01, -2.2267e-01, -5.4017e-02,\n",
       "          -3.3838e-01, -2.7689e-01,  1.3447e-01,  3.7594e-01,  6.2596e-01,\n",
       "           1.1595e+00,  5.6208e-01,  2.9245e-01, -1.0625e-01, -8.5438e-02,\n",
       "          -1.0762e+00,  5.5900e-01, -3.7318e-01,  8.1787e-02, -2.5119e-01,\n",
       "           6.6201e-01, -4.9906e-01,  5.4685e-01, -4.0754e-01, -2.5500e-01,\n",
       "           3.7179e-01,  3.0194e-01,  7.1104e-03,  2.8239e-02, -2.7432e-01,\n",
       "           4.0257e-01, -2.6094e-01, -3.6915e-02, -3.7819e-01,  4.0081e-01,\n",
       "          -3.9472e-01,  5.0770e-01, -5.6534e-02,  1.2945e-01,  1.8567e-01,\n",
       "          -3.5001e-01,  1.6610e-01,  1.0337e+00,  1.7414e-01,  4.1071e-01,\n",
       "           1.7420e-01,  5.1160e-01,  5.3064e-02, -8.8958e-01,  1.3691e-01,\n",
       "           2.0502e-01,  2.0789e-01, -6.7096e-02,  5.7131e-01, -2.5517e-02,\n",
       "          -6.5181e-01, -2.3094e-01, -3.2169e-01,  4.3108e-01,  6.5352e-01,\n",
       "          -2.2597e-01, -4.4343e-01,  2.3894e-01, -7.9776e-03, -5.3395e-01,\n",
       "           9.0074e-02, -2.2320e-01, -5.2792e-01,  8.7275e-02, -5.8914e-01,\n",
       "           1.4985e-01, -1.2428e-01,  5.4238e-02,  6.6887e-01, -4.0159e-01,\n",
       "          -1.6113e-01,  3.6602e-02, -4.4532e-02, -4.1270e-01, -4.9520e-01,\n",
       "          -7.7083e-02,  4.1073e-01, -1.9776e-01, -2.7976e-01,  7.0636e-02,\n",
       "          -3.0673e-01,  2.2584e-01, -1.7377e-01, -6.4935e-01,  3.2879e-01,\n",
       "          -6.0415e-01,  2.0578e-01,  2.5360e-01, -2.2507e-01,  4.9017e-01,\n",
       "           1.1337e-01, -8.1817e-02,  1.9397e-01, -9.9369e-02, -7.8554e-01,\n",
       "          -3.2573e-01, -2.7299e-02,  3.2662e-01, -4.2844e-01,  1.5776e-01,\n",
       "           1.5022e-01, -4.5752e-02, -2.2574e-01,  2.7904e-01,  1.0174e-01,\n",
       "           3.7762e-01,  3.9846e-01, -5.5006e-01,  1.1727e-01, -2.9867e-01,\n",
       "          -2.9262e-02, -6.1174e-02,  1.7952e-01,  2.3654e-01, -7.0764e-01,\n",
       "          -1.5224e-01, -2.6434e-01, -3.0965e-01, -2.1650e-01, -2.5120e-01,\n",
       "           2.0680e-01,  1.4038e-01, -1.5897e-02, -1.1308e-01,  5.8077e-01,\n",
       "           4.0814e-02, -1.1167e-01, -4.5997e-01,  3.1443e-01,  4.3178e-01,\n",
       "           2.1794e-01,  7.5875e-02, -1.2830e-01, -5.5532e-01, -1.1922e-02,\n",
       "          -4.3644e-01,  9.1081e-01, -7.3906e-01, -7.1970e-01,  6.6086e-02,\n",
       "           1.4951e-02, -8.8259e-04, -2.3450e-01, -5.0265e-01,  8.1298e-02,\n",
       "          -2.1017e-01, -1.4891e-01,  6.1550e-03, -1.0607e-02,  1.6299e-01,\n",
       "           4.0742e-01,  3.6838e-01,  1.5951e-01, -3.3938e-01, -4.4081e-01,\n",
       "          -6.0244e-01,  1.4567e-02, -1.6795e-01,  2.9330e-01, -2.6752e-01,\n",
       "          -3.2661e-01,  1.6043e-02, -9.1892e-03, -2.2272e-01,  5.2876e-01,\n",
       "           1.2427e-01,  1.6396e+00, -3.7287e-01, -4.4551e-01,  2.9018e-01,\n",
       "           7.2306e-02, -2.7073e-02, -2.8130e-01,  2.5074e-01,  9.5447e-02,\n",
       "           4.0194e-02, -3.6304e-01,  1.5023e-02, -2.0558e-02,  3.2773e-01,\n",
       "           3.2056e-01, -1.0703e+00, -2.0238e-01, -1.7993e-01,  5.5664e-01,\n",
       "           3.2388e-02,  6.3185e-01, -2.2305e-01,  3.0577e-01,  4.6106e-01,\n",
       "          -8.6667e-01, -2.3000e-01, -1.6129e-01,  2.7295e-01, -1.0583e-01,\n",
       "          -9.8735e-02, -1.6178e-01, -3.7165e-02, -1.6936e-01,  3.9538e-01,\n",
       "           7.4751e-01, -7.8426e-01,  7.4081e-01,  7.0467e-01, -9.8234e-02,\n",
       "           5.6815e-01,  9.4187e-02,  3.2641e-01, -1.4659e-01,  2.2669e-02,\n",
       "           2.6200e-01,  1.8542e-01, -6.3165e-01, -2.3962e-01,  2.4473e-01,\n",
       "           4.0158e-01, -1.1943e-01,  1.2108e-01,  7.9292e-02,  3.0960e-02,\n",
       "          -3.7691e-01, -1.0145e-01, -2.4552e-01, -3.9904e-01,  1.1356e-01,\n",
       "           1.4603e-02, -1.5001e-01, -8.6997e-02,  4.7709e-01, -1.7580e-01,\n",
       "           7.5067e-01,  4.2910e-01,  3.7145e-01, -3.5103e-01,  8.9950e-04,\n",
       "          -5.7565e-02,  6.0671e-01, -3.1670e-01,  7.7385e-01,  6.3337e-02,\n",
       "          -2.3140e-01,  2.7251e-01,  6.0547e-03,  1.7829e-01,  2.9892e-01,\n",
       "          -2.3519e-01,  1.2702e-01, -3.0718e-01, -2.3711e-01,  2.8330e-01,\n",
       "          -1.8640e-01,  1.1542e-01, -7.2270e-01, -6.3602e-03,  3.9857e-01,\n",
       "          -3.2051e-01,  4.9873e-01, -2.3074e-01,  6.1026e-01, -4.3509e-01,\n",
       "           3.2669e-02, -4.7470e-01, -4.3275e-01, -2.8508e-01,  5.1854e-02,\n",
       "          -8.6027e-01, -1.9453e-01,  1.7684e-01,  3.7405e-02,  2.1707e-01,\n",
       "           7.5155e-01, -3.3794e-01, -2.4771e-01, -2.0386e-01,  4.0190e-02,\n",
       "           1.8299e-01, -8.2071e-02, -3.5586e-01,  2.1161e-01, -3.5607e-01,\n",
       "          -1.0520e+00, -2.3117e-01,  8.9990e-02,  3.8419e-01, -3.5853e-01,\n",
       "           1.1570e+00,  2.5132e-02, -2.0226e-01,  6.7253e-01,  2.0808e-01,\n",
       "           6.4157e-01,  6.0323e-01,  1.3628e-03, -4.1259e-01, -5.4109e-02,\n",
       "          -1.5877e-01, -3.2881e-01,  1.5757e-01, -1.8346e-01,  5.0556e-01,\n",
       "           5.4802e-01, -2.0806e-02,  5.4756e-01]], device='mps:0'),\n",
       " tensor([[-0.2278, -0.8884, -0.2710,  ...,  0.1133,  0.6210,  0.2187],\n",
       "         [-0.2870, -0.5518, -0.8176,  ...,  0.0813,  0.4395,  0.6942],\n",
       "         [-0.2571, -0.6116, -0.7777,  ...,  0.1132,  1.1327,  0.5774]],\n",
       "        device='mps:0'),\n",
       " tensor([[-7.7321e-01, -6.2687e-01, -1.1323e-01, -1.0519e+00, -3.0578e-01,\n",
       "          -1.2828e-01, -7.9462e-01,  8.9870e-01,  2.7503e-01,  3.4266e-01,\n",
       "           9.7406e-02,  4.7614e-01,  3.1134e-01, -2.2559e-01, -6.6291e-01,\n",
       "          -9.0009e-02,  1.0545e-01,  3.9034e-01, -4.6425e-02,  1.4528e-01,\n",
       "           3.3829e-01,  4.1183e-01,  8.9642e-02, -2.1223e-01, -6.6662e-01,\n",
       "           3.7453e-01,  1.9571e-02,  2.9559e-01,  2.6897e-01, -6.3539e-01,\n",
       "          -1.6031e-01,  1.0746e-01, -4.3636e-01,  1.0615e+00,  4.7424e-02,\n",
       "          -3.0549e-01,  7.3814e-02, -2.0105e-02,  2.9871e-01, -1.6277e-01,\n",
       "           6.0938e-01,  1.4016e-01, -1.9665e-01,  3.5034e-01,  2.5624e-02,\n",
       "           1.4546e-01,  1.1702e-01, -5.0723e-01,  8.2079e-02, -3.4846e-01,\n",
       "           1.5886e-02, -1.7777e-01,  5.0070e-01,  4.3733e-01, -8.2345e-01,\n",
       "          -4.0863e-01,  2.0093e-01,  3.0476e-01, -2.6527e-01, -3.1017e-01,\n",
       "          -7.0817e-02,  6.6140e-01,  1.3297e-01, -4.2067e-01,  3.1706e-01,\n",
       "          -2.5013e-01, -4.3836e-01, -7.4553e-01, -2.6208e-01, -3.2976e-01,\n",
       "          -2.1258e-01,  3.6001e-01,  5.4873e-03, -2.4650e-01,  1.5868e-01,\n",
       "           1.9072e-01,  1.1015e-01,  1.4375e-01, -5.1134e-01, -2.9988e-01,\n",
       "           2.7337e-01,  7.7376e-02,  1.0861e-01, -1.6924e-01, -2.2513e-01,\n",
       "          -1.3933e-01,  3.2449e-01, -1.1193e-01,  9.7566e-02, -2.7596e-01,\n",
       "           5.0466e-01, -1.6304e-01, -7.7904e-02, -1.7741e-01,  3.9799e-01,\n",
       "          -3.5735e-01,  3.7119e-01, -3.8631e-01,  6.7408e-02, -3.3830e-01,\n",
       "          -1.0091e-01,  3.1574e-01, -4.6741e-01,  7.5676e-02, -8.2078e-01,\n",
       "          -2.0141e-01,  3.9638e-01,  3.3758e-02, -7.9220e-01,  3.8280e-01,\n",
       "           6.6784e-02, -2.2760e-01,  1.3458e-01, -6.3937e-01, -3.7279e-01,\n",
       "           2.9848e-01,  1.3611e-01, -1.9894e-01, -2.7687e-01,  3.2662e-01,\n",
       "           5.9498e-01, -1.6441e-01,  3.9762e-01, -3.2657e-01, -1.2225e-01,\n",
       "          -3.5970e-01,  5.1959e-01, -2.5859e-01,  2.8066e-01,  5.1872e-01,\n",
       "           3.0119e-01,  4.4430e-01, -3.3990e-01,  1.8205e-01,  2.2975e-01,\n",
       "          -4.7178e-02, -7.8001e-02, -2.0479e-01,  3.7065e-01,  3.4234e-01,\n",
       "          -3.9794e-01, -1.4807e-01,  9.9227e-02,  3.5483e-02, -1.7338e-01,\n",
       "          -1.2012e-01,  1.1952e-01,  3.4514e-01, -4.2500e-01, -4.6954e-01,\n",
       "          -1.0979e+00, -4.1312e-01,  3.0553e-01, -2.7741e-01, -8.1543e-02,\n",
       "          -3.6753e-01,  4.5371e-01, -9.1060e-02,  4.1548e-01,  2.4899e-01,\n",
       "           4.1687e-01, -2.0890e-01,  1.8333e-01, -6.6740e-01,  1.6327e-01,\n",
       "           8.8034e-02, -5.9759e-01, -1.6030e-01,  6.7873e-02,  5.4095e-02,\n",
       "           5.5723e-01, -3.6537e-02,  3.4373e-01,  2.4610e-01, -9.1039e-02,\n",
       "          -8.5804e-02,  3.8944e-01, -2.2669e-02, -1.9318e-01,  4.4997e-01,\n",
       "          -3.9328e-01, -1.6641e+00,  2.1061e-02, -3.6660e-01, -2.7273e-01,\n",
       "          -1.8075e-01,  1.6633e-01, -1.1210e-01,  7.2187e-01,  1.2032e-01,\n",
       "           2.0667e-01,  1.0435e-01,  9.2067e-02,  6.2763e-01,  7.4261e-02,\n",
       "           6.9905e-01,  2.8081e-01,  2.5431e-01, -4.0320e-01, -1.2291e-01,\n",
       "          -1.1816e-01,  3.2573e-01,  1.6575e-01,  7.6783e-01,  1.1286e-01,\n",
       "           4.0734e-01, -2.6539e-02,  1.0816e-01,  1.8229e-01,  2.4724e-01,\n",
       "          -1.0367e-01, -9.7266e-02, -5.1488e-01, -2.0181e-01, -2.5129e-01,\n",
       "           4.4661e-03, -4.1517e-01,  1.1589e-01,  6.5213e-01,  1.7437e-01,\n",
       "          -2.7106e-01,  1.8904e-01, -5.9420e-02, -3.4779e-01,  8.9500e-02,\n",
       "           3.8460e-01, -4.4867e-02, -8.0748e-02,  3.3476e-02, -1.6668e-01,\n",
       "          -3.8619e-01, -1.3992e-01,  7.1544e-01,  4.5934e-01, -1.7661e-02,\n",
       "           1.5771e-01, -1.7183e-01,  1.9816e-02, -2.2863e-01,  2.3332e-01,\n",
       "          -3.1734e-01, -2.2687e-01, -2.5791e-01, -4.3372e-02, -2.8484e-01,\n",
       "           4.1624e-01,  2.0144e-01,  3.5344e-01,  1.0077e+00,  3.5414e-01,\n",
       "          -9.7598e-02,  4.7834e-01,  8.9590e-02,  2.0440e-02, -4.2401e-01,\n",
       "           1.1981e-01,  3.5031e-02, -6.0313e-01,  5.7843e-01, -3.4234e-01,\n",
       "          -8.4432e-02, -2.1038e-01,  1.6453e-01,  2.0682e-01, -1.3867e-01,\n",
       "           2.1268e-01,  4.1669e-01, -7.3291e-02, -3.1340e-03,  5.7382e-01,\n",
       "          -6.0676e-01,  5.1837e-01, -2.3748e-01, -1.5117e-01, -6.6025e-01,\n",
       "          -9.0444e-01, -2.6525e-01,  2.1241e-01,  2.2547e-02, -4.1267e-01,\n",
       "          -7.9495e-02, -2.0898e-01,  1.0978e-01,  5.4731e-02, -1.8526e-01,\n",
       "           1.5659e-01, -3.2859e-03,  2.8002e-02, -2.7618e-01, -3.6427e-01,\n",
       "          -3.5254e-01, -3.4823e-03, -3.1560e-01, -1.3512e-01,  1.3209e-01,\n",
       "           3.7439e-01,  3.3561e-02,  6.3626e-01, -1.6080e-01, -4.8787e-01,\n",
       "          -4.7525e-02,  1.1387e-01,  3.4072e-01, -2.7442e-01,  3.2350e-01,\n",
       "           2.9190e-01,  1.0970e-01,  6.8334e-01, -3.5350e-01, -3.7647e-01,\n",
       "          -1.4029e-01,  6.0914e-01, -2.1377e-01,  6.0311e-02, -1.9619e-01,\n",
       "          -1.2343e+00, -3.1049e-01, -1.5309e-01, -5.7898e-02, -5.8084e-01,\n",
       "           2.8740e-01, -7.5467e-02,  1.9024e-02,  2.8401e-04, -3.2542e-01,\n",
       "           8.3520e-01,  5.0400e-01, -1.0683e-01,  1.6830e-02, -3.7057e-01,\n",
       "          -1.8329e-01, -2.7434e-01, -1.4090e-01, -1.5826e-01,  2.4642e-01,\n",
       "           2.9841e-02,  3.7530e-02, -1.8980e-01,  2.4125e-01, -5.1315e-01,\n",
       "           5.4343e-01,  4.2777e-02,  2.3395e-01,  3.7638e-01, -1.2805e-01,\n",
       "           3.4027e-02,  2.8299e-01, -2.0204e-01, -1.4811e-01,  3.2584e-01,\n",
       "           3.3372e-01, -8.8138e-01, -5.1898e-01, -2.6369e-01,  5.1767e-01,\n",
       "           1.5673e-01, -2.2051e-02, -3.1030e-02,  4.3244e-02,  2.9392e-01,\n",
       "          -1.3387e-01,  4.0949e-01, -2.7277e-01, -4.0577e-01,  4.2893e-01,\n",
       "           1.7759e-01,  4.2048e-01, -2.6752e-01, -1.3286e-01,  2.8595e-02,\n",
       "           8.4550e-02, -1.5209e-01,  2.2520e-01,  1.2180e-01,  5.5296e-01,\n",
       "          -2.5708e-01, -5.5714e-01, -8.1882e-02, -7.8622e-01, -3.2853e-01,\n",
       "           4.1102e-01,  5.8244e-02,  3.4419e-01, -3.3313e-01, -2.1719e-01,\n",
       "          -2.8936e-02, -2.3847e-01,  7.6228e-01,  4.9511e-01, -1.4818e-01,\n",
       "           1.7166e-01, -1.4684e-01, -4.4306e-01,  3.9785e-01, -4.2509e-01,\n",
       "          -4.0821e-01,  2.2267e-01, -2.1832e-01, -2.0668e-01,  1.1743e-01,\n",
       "          -2.9511e-01, -2.5025e-01,  2.6822e-01, -1.3802e-02, -1.7570e-02,\n",
       "          -4.8570e-01,  2.8937e-01,  1.8084e-01, -2.1500e-01, -1.4267e-01,\n",
       "           5.4312e-02,  1.6443e-01, -1.6687e-01, -1.6463e-01,  2.7348e-01,\n",
       "           3.0534e-01,  2.5477e-02, -1.0301e-01, -2.3927e-01, -6.3120e-02,\n",
       "           2.3914e-01,  4.1707e-02, -4.1425e-01,  3.3727e-01, -1.1116e+01,\n",
       "           1.2218e-01,  6.9571e-01, -3.6401e-01,  2.1130e-01, -7.0414e-02,\n",
       "           1.3285e-01, -2.2291e-01,  2.5198e-02, -1.1632e-01, -3.0651e-01,\n",
       "          -4.3426e-01,  7.5216e-02, -1.3469e-01, -4.8990e-01, -2.6878e-01,\n",
       "          -2.2074e-01, -6.8259e-01,  7.3342e-02, -3.6132e-01,  7.0090e-02,\n",
       "          -3.7839e-01, -7.7298e-02,  1.8289e-01,  1.6034e-01, -2.5914e-01,\n",
       "          -2.1086e-01, -3.3923e-01,  4.2900e-01, -2.7880e-02,  2.3269e-01,\n",
       "          -4.7916e-02,  5.3773e-02, -7.7165e-02, -3.3629e-01, -7.1667e-01,\n",
       "          -2.8900e-01,  8.0228e-02,  3.1313e-02, -2.1438e-02, -1.1770e-01,\n",
       "          -5.4993e-01, -4.5587e-01,  2.6023e-01,  2.0470e-01,  6.8350e-01,\n",
       "           5.7704e-01,  5.5665e-01,  1.0701e-01, -6.1474e-03, -3.3204e-01,\n",
       "          -1.1085e+00,  5.3164e-01, -4.3781e-01, -1.5395e-01, -5.0115e-01,\n",
       "           5.8213e-01, -5.0845e-01,  2.1009e-01, -3.5810e-01, -2.5793e-01,\n",
       "           2.5942e-01,  3.4370e-01, -1.3864e-01, -1.1606e-02, -2.2584e-01,\n",
       "           3.6303e-01, -2.6268e-01,  1.4398e-01, -8.9721e-02,  2.9169e-01,\n",
       "          -2.0596e-01,  3.4394e-01, -1.6067e-01, -1.2093e-01,  3.3433e-01,\n",
       "          -2.6407e-01,  6.8775e-03,  1.5540e+00, -2.9635e-01,  2.0449e-01,\n",
       "          -7.3665e-02,  4.3047e-01, -8.1535e-02, -8.2743e-01,  1.7751e-01,\n",
       "           2.3274e-01,  2.9549e-01,  2.6318e-02,  6.4920e-01, -5.3756e-02,\n",
       "          -5.3068e-01, -1.3656e-01,  1.6124e-01,  5.2893e-01,  8.1857e-01,\n",
       "          -1.5484e-01, -4.4539e-01,  2.9168e-01,  4.9415e-02, -3.7554e-01,\n",
       "           4.2280e-01, -3.2148e-01, -1.5666e-01,  1.7558e-01, -9.8340e-01,\n",
       "           8.3267e-02,  9.0138e-02, -1.6199e-02,  6.3190e-01, -2.3218e-01,\n",
       "           4.9231e-02,  2.2644e-01, -1.1056e-01, -4.9860e-01, -1.7494e-01,\n",
       "          -1.5822e-01,  6.7329e-01, -1.6298e-01, -3.4921e-01, -6.6686e-02,\n",
       "          -8.6842e-02,  2.1009e-01, -5.7610e-02, -5.9890e-01,  2.5005e-01,\n",
       "          -5.1103e-01,  8.9459e-02,  1.1475e-03, -3.7856e-01,  5.2312e-01,\n",
       "           4.5287e-01, -1.8703e-01,  1.6514e-01, -4.8700e-01, -8.3266e-01,\n",
       "          -2.6425e-01, -3.7145e-01,  2.0828e-01, -1.7779e-01,  2.0028e-01,\n",
       "          -3.6363e-02, -1.1121e-03, -2.2617e-01,  2.9850e-01,  2.0573e-01,\n",
       "           1.8990e-01,  4.3124e-01, -4.4911e-01,  2.9413e-01, -7.2638e-02,\n",
       "          -1.0421e-01,  8.6963e-02,  1.3743e-01,  8.8458e-02, -8.0368e-01,\n",
       "          -2.5689e-01, -2.4321e-02, -2.0982e-01, -1.5366e-01, -3.5494e-01,\n",
       "           4.0547e-01, -1.0922e-01, -4.1155e-02, -2.5667e-01,  4.8447e-01,\n",
       "          -5.8231e-03, -3.1673e-01, -3.1390e-01,  8.7978e-02,  1.9225e-01,\n",
       "           4.0525e-01,  4.4407e-02, -2.8018e-01, -7.5905e-01, -1.9487e-01,\n",
       "          -2.7877e-01,  1.0223e+00, -5.4829e-01, -6.1067e-01,  1.7788e-01,\n",
       "           5.2950e-01,  1.3334e-01, -9.4812e-02, -5.8920e-01,  1.4832e-01,\n",
       "          -1.8615e-01, -7.6224e-01,  1.6335e-01, -2.0802e-01,  3.1236e-01,\n",
       "           4.9497e-01,  5.9210e-01,  1.5360e-01, -3.6832e-01, -2.5469e-02,\n",
       "          -5.3736e-01,  7.1845e-02, -2.0300e-01,  1.8285e-01, -2.8790e-01,\n",
       "          -2.5898e-01, -3.1613e-01,  1.9037e-01,  2.2181e-01,  2.5568e-01,\n",
       "           2.9786e-01,  1.6084e+00, -2.4429e-01, -2.6113e-01,  5.5912e-01,\n",
       "           7.1617e-02,  1.2381e-02, -2.0995e-01,  2.8344e-01,  1.1128e-01,\n",
       "           2.7900e-01, -3.2030e-01, -1.8270e-01, -9.7887e-02,  1.0752e-01,\n",
       "           3.9105e-01, -9.6947e-01, -2.4648e-01, -1.5894e-01,  6.2180e-01,\n",
       "           1.3602e-01,  7.6289e-01, -4.6845e-01,  4.1095e-01,  3.9791e-01,\n",
       "          -9.1237e-01, -1.5661e-01,  5.9917e-02,  2.4799e-01, -2.6718e-01,\n",
       "          -1.7562e-01, -2.0997e-01, -1.3624e-02, -1.3422e-01,  6.6881e-01,\n",
       "           4.0836e-01, -8.1924e-01,  5.8551e-01,  6.4786e-01,  1.4682e-02,\n",
       "           4.1732e-01,  1.2533e-01,  1.2440e-01, -1.2395e-01, -2.2446e-01,\n",
       "           2.6778e-01,  2.1486e-01, -4.4833e-01, -2.6121e-01,  2.3863e-01,\n",
       "           4.5013e-01, -2.0176e-01,  1.2656e-01, -2.1820e-01, -7.8500e-02,\n",
       "          -3.5896e-01, -1.3699e-01, -2.3390e-01, -2.6974e-01, -6.3138e-02,\n",
       "           1.1217e-01, -5.7928e-02,  1.6826e-01,  3.8812e-01, -1.0711e-01,\n",
       "           7.3091e-01,  3.8353e-01,  5.1848e-01, -2.5938e-01, -2.3318e-01,\n",
       "           6.7012e-02,  3.9322e-01, -9.9816e-02,  8.8660e-01,  8.9010e-02,\n",
       "          -4.1422e-01, -4.3339e-02, -7.1976e-02, -2.8267e-01,  1.3812e-01,\n",
       "          -1.0586e-01,  4.0994e-01,  2.7567e-02, -9.2159e-02,  3.2189e-01,\n",
       "          -2.5161e-01,  3.3879e-01, -7.5494e-01,  1.8975e-01,  3.7319e-01,\n",
       "          -2.8221e-01,  4.5308e-01, -4.9486e-02,  6.7949e-01, -2.8115e-01,\n",
       "           1.5331e-01, -7.8774e-01, -2.8907e-01, -1.2508e-01,  6.7031e-02,\n",
       "          -8.5788e-01, -2.5180e-01,  2.9192e-01,  1.6326e-01,  9.9452e-02,\n",
       "           8.2615e-01, -7.9478e-02, -3.5233e-01, -2.3304e-01,  1.8114e-01,\n",
       "           5.4662e-02, -6.6287e-02,  2.8314e-01,  5.8864e-03,  9.5743e-02,\n",
       "          -9.3001e-01, -3.1547e-01,  1.6295e-01,  2.2357e-01, -4.2469e-01,\n",
       "           8.9244e-01,  1.4141e-01,  9.7363e-02,  7.5260e-01,  2.7591e-01,\n",
       "           7.8104e-01,  6.6230e-01,  2.0610e-01, -6.6219e-01, -2.7237e-01,\n",
       "           3.6859e-02, -5.0094e-01,  1.4547e-01, -1.6921e-01,  1.6438e-01,\n",
       "           4.4235e-01,  5.6421e-01,  4.2962e-01]], device='mps:0'),\n",
       " tensor([[-0.2744, -0.4852, -0.4880,  ...,  0.2064,  0.2977,  0.6606],\n",
       "         [ 0.0506, -0.6230, -0.6446,  ..., -0.0875,  0.4018,  0.3101],\n",
       "         [-0.1156, -0.5318, -0.5096,  ...,  0.0343,  1.0385,  0.3071],\n",
       "         [-0.0921, -0.4820, -0.2019,  ...,  0.5267,  0.5850,  0.5583]],\n",
       "        device='mps:0'),\n",
       " tensor([[-1.6775e-01, -6.8467e-01, -4.6240e-01, -6.2843e-01, -3.9495e-01,\n",
       "          -3.9917e-01, -5.4355e-01,  5.4865e-01,  1.3103e-01,  5.6839e-01,\n",
       "           2.6130e-01, -2.0962e-01,  8.7134e-02, -4.2508e-01, -1.1259e+00,\n",
       "          -4.0118e-01, -1.2003e-01,  4.3551e-01,  6.2357e-01,  2.3777e-02,\n",
       "           4.4145e-01,  3.3499e-01,  8.6753e-03, -6.7884e-02, -1.1449e+00,\n",
       "           1.2940e-01, -5.5384e-02, -2.4320e-01,  2.2523e-01, -4.9751e-01,\n",
       "           1.5223e-01,  3.5871e-01, -1.3111e-01,  6.7471e-01, -1.4615e-01,\n",
       "          -1.4558e-01,  2.7899e-01, -1.6278e-01,  2.3716e-01, -5.5373e-01,\n",
       "           5.2803e-01,  1.4758e-01,  5.2443e-01, -9.7775e-02,  4.4514e-01,\n",
       "          -2.8706e-01,  2.2957e-01, -5.1555e-01, -3.8627e-01, -3.8054e-01,\n",
       "          -5.6257e-02, -4.2809e-02,  2.7718e-01,  1.9014e-01, -1.2658e-01,\n",
       "          -5.2030e-01,  2.3704e-01,  2.3062e-01, -1.8140e-01,  3.7363e-01,\n",
       "          -1.0103e-01,  5.3080e-01,  1.0050e-01, -1.2315e-01,  1.5246e-01,\n",
       "           7.8711e-02, -8.3909e-01, -3.4588e-01, -7.3304e-01,  7.6002e-02,\n",
       "           1.0113e-01,  2.6671e-01,  1.6092e-01, -3.3802e-01,  5.1274e-01,\n",
       "          -2.3913e-01,  1.4218e-01, -8.3222e-02, -3.9295e-01, -3.6155e-02,\n",
       "          -1.5273e-02,  2.0092e-03, -4.7164e-02, -1.0424e-01, -8.6827e-02,\n",
       "          -6.6989e-02,  4.2445e-01,  3.0233e-02,  1.1516e-01,  1.1137e-01,\n",
       "           3.5738e-01,  7.4092e-02, -9.7100e-02,  1.4715e-01,  3.6847e-01,\n",
       "          -1.9348e-01,  2.2950e-01, -2.1839e-01,  1.1676e-01,  4.2205e-02,\n",
       "          -1.6555e-01,  1.1806e-01,  1.1238e-01,  1.9819e-01, -4.0698e-01,\n",
       "          -2.6168e-01,  4.0036e-02,  1.0612e-01, -4.5339e-01, -9.2106e-02,\n",
       "          -1.5640e-01, -1.2348e-02,  2.7362e-01, -4.9681e-01, -9.4014e-02,\n",
       "           8.3650e-02,  2.6133e-02, -1.9768e-01, -1.4199e-01,  9.6219e-01,\n",
       "           3.0732e-01, -4.7622e-01,  1.9929e-01,  2.3703e-01,  7.3568e-02,\n",
       "           4.5486e-02,  3.3111e-01, -1.2294e-01,  5.7693e-02,  2.1185e-01,\n",
       "           7.4215e-02, -1.2710e-02, -1.2557e-01,  2.5217e-02, -1.6844e-01,\n",
       "           4.3339e-02,  3.0044e-02, -3.9602e-01,  2.3371e-02, -3.2178e-01,\n",
       "           2.0043e-01,  6.0985e-02, -3.7217e-01, -1.4981e-01,  1.0443e-01,\n",
       "          -1.4403e-01, -1.8353e-01, -2.2752e-02, -1.8669e-01, -2.7572e-01,\n",
       "          -7.0899e-01, -4.3979e-01,  2.1239e-01, -6.6087e-02,  1.4982e-01,\n",
       "          -2.1730e-01, -7.5071e-02, -3.5156e-01, -1.2395e-01, -1.6650e-01,\n",
       "           8.5436e-01,  1.0713e-02,  8.6766e-03, -3.2075e-01, -2.5513e-01,\n",
       "           4.0048e-01,  8.3263e-03, -4.3222e-01, -4.4798e-01,  2.6765e-01,\n",
       "           1.1410e-01, -1.7137e-01,  1.9100e-01, -7.6796e-01, -2.4297e-02,\n",
       "          -4.3352e-01,  9.0911e-02,  2.6381e-01, -4.3563e-02, -4.4751e-02,\n",
       "           9.0044e-02, -1.4519e+00,  4.6423e-01, -2.7812e-01, -3.5013e-03,\n",
       "          -2.5040e-01,  1.0617e-01,  1.7299e-01,  7.3191e-02,  9.2541e-02,\n",
       "           2.7298e-02,  2.9707e-01,  1.3514e-01,  1.2494e-01, -2.6242e-02,\n",
       "          -9.8827e-03,  6.4622e-01,  2.7052e-01,  1.2942e-02,  3.4764e-02,\n",
       "          -8.0154e-01,  3.7398e-01,  5.3662e-02, -2.1793e-01, -1.5494e-01,\n",
       "           2.3554e-01, -2.0060e-01,  2.3569e-01,  2.8662e-02,  7.9935e-02,\n",
       "           1.0461e-01, -2.8143e-01, -3.1740e-01, -3.7778e-01, -3.1273e-01,\n",
       "           5.8081e-01,  1.2864e-02, -2.4817e-01,  7.7225e-01,  2.8668e-01,\n",
       "          -2.2498e-01, -6.9382e-02, -2.7833e-01, -1.6190e-01, -2.9646e-01,\n",
       "           4.0291e-01, -4.0986e-01, -1.2818e-01,  4.1290e-02, -2.7051e-01,\n",
       "          -1.7305e-01, -9.9300e-01,  7.2195e-01,  3.5281e-01, -1.9333e-01,\n",
       "          -3.7544e-01, -4.1496e-01, -1.2443e-01, -6.1768e-01,  2.4076e-01,\n",
       "          -1.5725e-01, -4.2336e-01, -7.0479e-01,  2.1704e-01, -6.0610e-02,\n",
       "           2.2143e-01, -4.9496e-03,  9.2260e-01,  1.0551e-01,  1.5034e-01,\n",
       "           5.7897e-02, -5.7938e-02,  4.7647e-01, -1.6482e-02, -1.4465e-01,\n",
       "           3.8200e-02, -1.0322e-01, -5.1114e-01,  4.9331e-01, -7.6119e-02,\n",
       "          -3.7043e-01, -1.7415e-01,  4.9343e-01,  6.0900e-01, -5.8898e-01,\n",
       "          -2.3420e-02,  2.1231e-01, -1.1803e-01, -1.9825e-01,  5.2943e-01,\n",
       "          -6.0210e-01,  2.0953e-01,  3.8590e-01, -4.2173e-01, -5.9212e-01,\n",
       "          -7.4985e-01, -5.3439e-02, -5.6003e-02, -5.3705e-02, -8.3447e-01,\n",
       "          -8.0635e-01,  1.7041e-01,  8.1555e-02,  8.6731e-02, -8.0385e-02,\n",
       "           2.8623e-01,  1.3261e-01,  1.0035e-01, -4.1355e-01, -6.6492e-01,\n",
       "          -3.1600e-01,  2.8262e-01, -4.3167e-03,  4.1512e-01,  3.2453e-01,\n",
       "          -2.4008e-01,  2.0111e-01, -6.6329e-02,  1.3810e-02, -1.6885e-01,\n",
       "          -3.2525e-01, -5.4355e-01,  3.3401e-01, -1.9264e-01, -1.5619e-01,\n",
       "           3.3020e-01,  2.3854e-01,  6.9506e-01,  1.1450e-01, -6.1155e-01,\n",
       "           1.2800e-01,  5.3585e-01, -2.1037e-01,  1.1454e-01, -1.2711e-01,\n",
       "          -1.0308e+00, -2.1998e-01,  6.3633e-02, -2.7082e-02, -3.4748e-01,\n",
       "           2.5876e-01,  4.9304e-02, -2.6435e-02, -1.8258e-01, -9.4219e-02,\n",
       "          -2.3571e-02,  1.0782e-01, -1.4531e-01, -1.6647e-01, -5.3906e-03,\n",
       "          -3.0429e-02,  1.1591e-01, -2.9279e-02,  4.6029e-02, -8.0295e-02,\n",
       "           2.0754e-01,  3.9851e-01,  4.4347e-02,  6.6271e-02, -1.1965e-01,\n",
       "           7.5141e-01, -3.6162e-01, -2.7389e-01,  4.1950e-01, -3.1850e-01,\n",
       "          -2.1294e-01,  2.2791e-01, -2.4970e-01, -6.1170e-02,  1.4935e-01,\n",
       "           3.0893e-02, -1.7900e-01, -3.1399e-01, -1.7342e-02,  4.6295e-01,\n",
       "           4.7814e-01,  4.0390e-01, -3.7302e-01,  3.5379e-01,  3.2886e-01,\n",
       "          -4.2786e-01,  2.2149e-01, -3.0591e-02,  1.0104e-02, -1.5927e-01,\n",
       "          -5.9777e-02,  3.5616e-02, -6.2499e-01, -5.1112e-02,  2.0680e-01,\n",
       "          -1.9941e-01, -2.7536e-01,  1.4131e-01,  3.9896e-01,  5.2850e-01,\n",
       "          -3.1048e-01, -3.3807e-01,  5.0098e-01, -4.3628e-01, -2.4004e-01,\n",
       "           1.3444e-01,  2.2818e-01,  9.7980e-02, -2.5080e-01, -3.5257e-01,\n",
       "          -2.3947e-01,  1.2978e-01,  4.8407e-01,  3.5517e-01, -1.9502e-01,\n",
       "           2.9172e-01,  1.7791e-03, -3.5221e-01,  8.1999e-01, -1.0998e-01,\n",
       "          -3.0789e-01,  2.2361e-01,  1.1567e-01,  1.5607e-01,  2.4740e-01,\n",
       "          -1.2514e-01,  1.2963e-02,  5.8849e-01, -6.1364e-02, -3.3783e-01,\n",
       "          -2.2101e-01,  3.3202e-01,  2.3596e-01, -5.5765e-01,  2.0740e-01,\n",
       "           1.7694e-01,  6.5778e-01, -6.1531e-01,  2.2386e-02,  5.6559e-02,\n",
       "           9.4653e-02, -2.2771e-01,  1.8566e-01, -2.0581e-01, -4.8417e-01,\n",
       "           4.6155e-01,  3.2637e-01,  4.4399e-01,  2.2008e-01, -1.1463e+01,\n",
       "           2.7768e-01,  6.2243e-01, -3.3910e-01,  2.5190e-01,  7.2681e-02,\n",
       "           1.9278e-01, -6.3625e-01,  2.7277e-01, -3.3113e-01, -3.2017e-01,\n",
       "          -2.3619e-01,  2.1677e-01,  3.5827e-02, -6.7433e-01,  2.1863e-01,\n",
       "          -9.2716e-03, -5.9539e-01,  7.3826e-02, -1.0371e-01,  1.6714e-01,\n",
       "           2.4188e-01,  3.9763e-02,  1.8994e-01, -4.8700e-02,  7.6276e-04,\n",
       "          -8.8207e-02,  4.9762e-02,  1.9121e-01,  3.5938e-01, -1.0015e-02,\n",
       "           6.6401e-02,  1.9620e-01,  1.7784e-01, -1.8367e-01, -5.5335e-01,\n",
       "          -1.0899e-02, -1.9569e-02,  2.3629e-01,  1.2778e-01,  2.4573e-02,\n",
       "          -5.9155e-02, -3.6091e-01,  1.0465e-01,  1.0681e-01,  4.9374e-01,\n",
       "           1.1143e+00,  1.0143e+00,  2.6795e-01, -2.2704e-03, -4.8226e-02,\n",
       "          -7.5747e-01,  5.2961e-02, -7.5655e-01,  9.4186e-03, -1.7591e-01,\n",
       "           6.1078e-01, -8.0254e-01, -1.5983e-01, -1.5745e-01, -7.0910e-02,\n",
       "           4.1870e-01,  5.2737e-02,  3.4372e-01, -1.7429e-01, -2.8577e-02,\n",
       "           6.4345e-01, -2.8779e-01,  1.4011e-01, -2.6120e-01, -3.2775e-02,\n",
       "          -2.4950e-01,  4.0921e-01,  4.7130e-01, -3.8052e-01,  1.7237e-01,\n",
       "          -1.7362e-01, -2.9310e-01,  1.3959e+00, -1.4388e-01,  4.2106e-02,\n",
       "           1.0294e-01,  3.2895e-02, -3.6623e-01, -7.2756e-01,  9.8002e-02,\n",
       "           2.2572e-01,  9.4129e-02,  4.0972e-01,  5.5837e-01,  7.1110e-02,\n",
       "          -5.6962e-01, -1.1069e-01,  1.1786e-01,  2.7185e-01,  7.3090e-01,\n",
       "          -2.8468e-01,  1.6942e-01, -6.6954e-02,  6.1695e-02, -4.3238e-01,\n",
       "          -3.9644e-03, -7.6315e-02, -1.2762e-01,  2.8357e-01,  3.1216e-01,\n",
       "           7.8601e-02,  2.9440e-01,  7.4304e-02,  2.6293e-01, -5.8672e-01,\n",
       "           2.2019e-01, -7.7207e-02,  3.5123e-01, -3.3948e-01, -4.8105e-01,\n",
       "           3.3133e-01,  8.1181e-01, -4.3390e-01, -4.9829e-01, -4.9288e-01,\n",
       "           3.9379e-01,  3.4972e-01,  5.1170e-02, -4.5988e-01,  3.8380e-01,\n",
       "          -6.4552e-01, -2.9128e-01,  1.5482e-01,  4.7335e-01,  4.9025e-01,\n",
       "           3.3854e-01,  2.0783e-01,  2.7450e-01, -3.7184e-01, -6.1417e-01,\n",
       "          -8.0681e-01,  1.9249e-01,  1.9237e-01, -6.4662e-01,  9.1618e-02,\n",
       "           5.6487e-03, -2.1072e-01,  3.8881e-01, -7.3228e-02, -1.7115e-01,\n",
       "           2.5014e-01,  2.9470e-01,  1.0630e-01,  4.4519e-01, -3.7599e-01,\n",
       "          -4.3749e-01, -3.2824e-01, -6.1801e-03, -2.1875e-02, -5.1506e-01,\n",
       "          -2.4613e-01,  9.2511e-02, -2.8334e-01,  6.8517e-01,  5.1130e-01,\n",
       "          -6.3541e-02, -1.9933e-01, -1.4332e-02,  1.1810e-01,  4.2542e-01,\n",
       "          -1.4137e-01, -5.1487e-01,  1.0778e-01, -3.5166e-02,  1.2072e-01,\n",
       "           1.4611e-01,  1.4475e-01, -3.2919e-01, -4.0249e-01, -2.0278e-01,\n",
       "           1.7032e-01,  1.4412e+00, -2.4287e-01, -6.8618e-01,  2.6251e-01,\n",
       "           6.3342e-01, -2.2238e-01, -7.8270e-03, -2.6317e-01,  1.1929e-01,\n",
       "          -4.6211e-01, -6.7070e-02,  4.6727e-01, -4.7729e-03,  2.6474e-01,\n",
       "           8.0024e-01,  1.7420e-01, -1.3003e-01,  5.5504e-01,  2.9652e-01,\n",
       "          -5.1945e-01, -4.7457e-02, -3.5783e-01, -3.1719e-01,  1.2294e-01,\n",
       "           5.6594e-02,  2.0617e-01,  4.4335e-02,  8.0010e-02,  2.3212e-01,\n",
       "           1.2164e-02,  1.1719e+00, -2.6145e-01, -5.2089e-01,  3.8535e-01,\n",
       "           5.0800e-02, -3.4169e-01, -3.3246e-02,  2.8896e-01,  9.9269e-02,\n",
       "           2.0667e-01,  2.5749e-01, -5.8632e-01, -1.7963e-01,  3.2156e-01,\n",
       "           5.1856e-01, -6.0853e-01, -4.0730e-01, -1.2521e-01,  1.5171e-01,\n",
       "          -1.7883e-01,  3.4600e-01, -1.8822e-03,  4.7550e-01,  4.6158e-01,\n",
       "          -1.0124e+00,  1.7155e-01, -2.7750e-01,  9.8742e-02, -2.3244e-01,\n",
       "          -7.1061e-01, -2.2397e-01, -1.2769e-02, -4.2894e-01,  1.3636e-01,\n",
       "           6.1285e-01, -5.9278e-01,  2.4386e-02,  4.9134e-01,  2.4951e-02,\n",
       "           1.7252e-02, -1.8054e-01,  2.6580e-01,  1.3981e-01, -8.4594e-02,\n",
       "          -3.9872e-01,  9.4823e-02, -3.5642e-01, -7.2435e-01,  2.1116e-01,\n",
       "           5.4477e-01, -1.2251e-02,  3.5983e-02, -2.7943e-02, -9.7076e-03,\n",
       "          -4.0837e-01, -3.6851e-01, -2.7686e-01,  1.7933e-01, -2.0758e-02,\n",
       "           1.4522e-01,  2.4120e-01,  1.4282e-01,  2.9538e-02, -4.8499e-01,\n",
       "           2.9304e-01,  3.3458e-01,  6.5611e-01,  5.3831e-02, -3.4853e-02,\n",
       "          -1.3095e-01,  4.7663e-01, -3.2797e-01,  5.6841e-01, -2.5825e-01,\n",
       "          -3.5391e-01,  1.1571e-02,  1.2601e-01,  2.0209e-02, -1.0222e-01,\n",
       "          -3.9038e-01,  1.6596e-01, -9.1921e-02, -1.1361e-01,  4.9847e-01,\n",
       "          -5.9259e-02,  1.8868e-01, -7.9471e-01,  2.4381e-01,  2.6846e-01,\n",
       "           2.1476e-01,  6.5131e-01,  5.2754e-02,  6.9129e-01,  8.0590e-02,\n",
       "           4.4271e-01, -4.0643e-01, -1.9919e-01, -2.2528e-01,  1.4780e-01,\n",
       "          -4.7269e-01, -1.6423e-01,  4.1678e-01,  6.9114e-01,  5.1587e-01,\n",
       "           1.7231e-01, -1.1425e-01, -5.3244e-01, -2.0237e-01,  2.0468e-01,\n",
       "           1.6388e-01, -4.0348e-01,  2.5892e-01, -3.9944e-01,  9.5351e-02,\n",
       "          -8.0325e-01, -4.8229e-01, -5.6736e-02, -1.6782e-02, -1.9341e-01,\n",
       "           4.7158e-01, -3.6474e-01,  3.7512e-01,  3.1138e-01,  2.2187e-02,\n",
       "           1.9742e-01,  3.3707e-01, -6.8835e-02, -7.5202e-01, -1.4140e-01,\n",
       "           4.1467e-01, -1.4825e-01, -4.2918e-02, -7.2920e-01,  3.3551e-01,\n",
       "           2.6394e-01,  9.0548e-01,  4.1784e-01]], device='mps:0'),\n",
       " tensor([[-1.1706e-01, -7.0106e-01, -7.9217e-01, -2.1583e-01, -3.1166e-01,\n",
       "          -2.8355e-01, -3.4567e-01,  4.0393e-01, -1.1804e-01,  5.4759e-01,\n",
       "           4.1226e-01,  2.0949e-01,  9.5251e-02, -1.3826e-01, -7.9470e-01,\n",
       "          -4.8312e-01, -1.5518e-01,  2.5737e-01,  2.4814e-01,  2.9108e-01,\n",
       "           7.3972e-01,  5.0511e-01,  6.0171e-02,  3.6346e-01, -4.1964e-02,\n",
       "           3.8580e-01, -1.8099e-01,  2.3888e-02,  1.8432e-01, -7.9439e-01,\n",
       "          -2.4389e-01,  7.4834e-01,  5.4349e-02,  8.3754e-01, -4.8390e-01,\n",
       "          -4.4903e-01,  1.1414e-01, -4.6245e-01,  3.5820e-01, -1.3907e-01,\n",
       "           5.5263e-01, -5.8565e-01,  2.7313e-01,  2.3991e-01,  5.3969e-01,\n",
       "           2.0473e-01,  2.9116e-01, -1.0207e+00, -4.0495e-02, -3.3621e-01,\n",
       "          -3.9837e-01, -1.9232e-03,  2.1783e-01,  3.2423e-01,  3.1602e-02,\n",
       "          -6.8885e-01,  1.0310e-01,  2.9676e-01, -4.3978e-01,  1.0526e-01,\n",
       "          -3.2826e-02,  4.8139e-01, -1.3490e-01,  9.9928e-03,  1.2996e-02,\n",
       "           5.1658e-01, -9.9598e-02, -1.0144e-01, -4.6363e-01,  1.0906e-01,\n",
       "           7.0235e-02, -1.2620e-01, -1.9459e-03, -4.6745e-01,  3.2583e-01,\n",
       "          -3.9323e-01,  3.5434e-01, -8.0407e-02, -7.7064e-01, -1.1719e-04,\n",
       "           4.6049e-02,  4.3184e-01, -3.2108e-01, -3.2218e-01, -2.4087e-01,\n",
       "          -2.4773e-02,  6.8821e-01,  4.6940e-01,  3.6845e-01, -7.6702e-01,\n",
       "           2.8364e-02,  3.1111e-01, -4.6413e-01, -3.5557e-01,  1.8758e-01,\n",
       "           1.3118e-01,  3.9860e-01, -1.1522e-01,  5.2986e-02, -5.0511e-01,\n",
       "           1.1891e-01,  1.1898e-01,  1.2919e-01,  2.4281e-01, -8.3598e-01,\n",
       "          -6.8520e-01,  1.5934e-01, -3.6955e-02, -6.7581e-01, -3.3584e-01,\n",
       "          -1.7493e-01, -6.8527e-02,  3.4881e-01, -6.1386e-01, -4.1954e-01,\n",
       "          -2.2622e-02,  8.4003e-02,  7.9894e-02, -3.8332e-01,  6.4779e-01,\n",
       "           1.2236e-01, -4.6632e-01,  4.3548e-02,  1.4969e-01,  1.7021e-01,\n",
       "           7.8819e-03,  3.4826e-01,  4.6045e-03,  4.0729e-01,  1.3528e-01,\n",
       "          -1.3939e-01,  3.1375e-02, -4.5257e-01, -2.7150e-01,  9.7140e-02,\n",
       "          -6.2868e-02,  8.1241e-02, -6.1446e-01, -6.2215e-01, -2.1417e-01,\n",
       "           1.3131e-01, -1.4980e-01, -3.1211e-01, -2.2033e-02,  2.8488e-01,\n",
       "           9.7253e-03, -3.3631e-01, -1.0679e-01, -7.4138e-01, -6.5663e-01,\n",
       "          -3.5332e-01, -6.6598e-01,  2.8780e-01, -1.6477e-01, -4.8988e-02,\n",
       "          -6.9212e-02,  2.8284e-01, -2.4413e-02, -3.0904e-02,  1.1480e-01,\n",
       "           4.6464e-01, -1.0373e-01,  3.3672e-01, -6.9175e-01, -1.6181e-01,\n",
       "           5.7155e-01, -2.1920e-01,  7.7340e-03,  2.9213e-01,  3.1567e-01,\n",
       "           2.1278e-01, -3.9742e-01,  1.8116e-01, -7.3796e-01,  1.1576e-01,\n",
       "          -5.1827e-01,  3.8189e-01,  1.4910e-01, -2.7729e-01,  4.7723e-02,\n",
       "           1.4738e-01, -1.8916e+00, -9.5913e-03, -3.3740e-01,  2.0957e-01,\n",
       "          -5.1156e-01,  7.0578e-01,  2.5700e-01,  2.9267e-01, -2.5075e-01,\n",
       "           1.7090e-01, -1.6370e-01,  8.7745e-02,  2.9401e-01,  1.2231e-02,\n",
       "           4.2903e-01,  6.0398e-01, -1.6030e-01, -1.0909e-01,  1.2779e-01,\n",
       "          -6.1174e-01,  1.0128e-01,  1.1363e-01,  2.3658e-01,  4.3866e-01,\n",
       "           1.7882e-01,  8.5515e-02,  6.5620e-02,  2.8393e-01,  1.6879e-01,\n",
       "           4.6136e-04, -2.4417e-01, -2.2348e-01, -2.2281e-01, -1.9176e-01,\n",
       "           7.1657e-01,  5.7767e-02, -2.3827e-01,  3.5492e-01,  3.0532e-01,\n",
       "          -2.3048e-01, -2.2995e-01,  1.0448e-01, -2.0061e-01, -1.5078e-01,\n",
       "           4.2029e-01, -1.9282e-01, -1.7735e-01, -1.1696e-01, -2.9148e-01,\n",
       "           8.9053e-02, -5.2330e-01,  4.1973e-01,  3.9196e-01, -1.3138e-01,\n",
       "          -4.3678e-01, -5.3365e-02,  3.1835e-01, -5.2033e-01,  7.5302e-02,\n",
       "          -1.5355e-01, -4.8655e-01, -7.9203e-01,  1.1256e-01,  1.3019e-01,\n",
       "           1.0731e-01, -3.7604e-01,  1.0320e+00,  4.3715e-01,  3.9474e-01,\n",
       "          -1.4364e-01, -2.3255e-01,  5.2188e-01, -2.6717e-02, -2.8720e-01,\n",
       "           1.7692e-01, -1.3180e-01, -6.2328e-02,  2.2360e-01, -2.7184e-01,\n",
       "           5.4297e-02,  1.1288e-01,  1.7916e-01,  6.2805e-01, -5.0657e-01,\n",
       "          -3.7437e-02,  4.4402e-01, -3.9909e-01, -4.5587e-01,  1.2416e-01,\n",
       "          -1.3825e-01,  3.1496e-01, -4.2519e-02, -6.5030e-01, -6.8944e-01,\n",
       "          -3.5719e-01,  2.6889e-01, -8.2794e-02,  4.1825e-01, -8.8982e-01,\n",
       "          -1.0076e-01,  7.0080e-01, -2.1790e-01,  2.4114e-01, -2.9759e-01,\n",
       "           3.8891e-01,  1.9288e-01, -1.5281e-01, -2.2075e-01, -6.4388e-01,\n",
       "           6.7232e-02, -8.9829e-03, -2.4822e-01,  4.8120e-01,  3.2063e-01,\n",
       "           2.2351e-01,  5.4348e-01,  4.7212e-02,  6.6773e-02, -5.5275e-01,\n",
       "           8.8638e-02, -8.6996e-02, -5.2798e-01, -2.8856e-01, -2.5898e-01,\n",
       "           2.2209e-01,  9.9438e-02,  5.5104e-01, -6.3432e-01, -6.5884e-01,\n",
       "           1.6786e-01,  5.5967e-01, -1.2316e-01,  1.7082e-01, -1.3721e-01,\n",
       "          -1.5344e+00, -1.0518e-01, -6.3249e-01, -1.1557e-01, -6.3906e-01,\n",
       "           1.6676e-01, -1.1420e-01, -3.4081e-02, -3.0450e-01, -1.5952e-01,\n",
       "           3.4711e-02,  2.9585e-01,  4.3278e-02,  8.9418e-02,  5.8747e-02,\n",
       "          -2.8650e-01,  2.0821e-01, -1.7420e-01, -2.9687e-01, -4.1802e-01,\n",
       "           8.1701e-02,  4.9813e-01, -1.9607e-02, -2.6745e-01, -1.7166e-01,\n",
       "           6.8435e-01, -4.7763e-01, -2.7406e-01,  3.1347e-01, -3.1970e-01,\n",
       "          -8.5895e-02,  3.0318e-01, -1.9024e-01, -2.0489e-01, -2.1476e-01,\n",
       "          -1.1528e-01, -2.3369e-01, -3.7160e-01,  5.1082e-02,  4.9826e-01,\n",
       "           3.7503e-01,  3.4247e-01, -6.8006e-01,  4.4663e-01,  4.5761e-01,\n",
       "          -6.6881e-01,  1.9125e-01,  3.5294e-01, -1.2421e-01,  1.0002e-02,\n",
       "           1.7459e-01,  3.5887e-01, -4.1018e-01,  1.5499e-02,  1.6458e-01,\n",
       "           8.9114e-02, -2.9113e-01, -2.2279e-01,  1.9396e-01,  2.2108e-01,\n",
       "           2.8004e-03, -5.3497e-01,  5.4792e-01, -1.7216e-01, -6.2724e-01,\n",
       "           3.9916e-02,  1.6771e-01, -1.3539e-01, -2.4143e-01, -3.9902e-01,\n",
       "          -4.6778e-01,  2.3708e-01,  5.2003e-01,  4.1117e-01, -1.7314e-01,\n",
       "           9.1063e-03, -5.7126e-02, -1.4317e-01,  5.7048e-01, -2.1047e-02,\n",
       "          -4.8709e-02,  2.5779e-01,  3.3104e-01,  2.1715e-01,  3.3608e-01,\n",
       "          -2.5549e-01, -4.2823e-01,  1.6286e-01,  6.3834e-02, -2.2438e-01,\n",
       "          -3.6461e-01,  3.5778e-02,  1.2948e-01, -3.3185e-01, -2.3471e-01,\n",
       "          -2.7489e-03,  7.1621e-01, -2.7641e-01,  4.1391e-02, -7.8449e-01,\n",
       "           1.3653e-01,  4.2945e-02,  1.4682e-01, -7.9368e-02, -4.2973e-01,\n",
       "           5.2631e-01,  5.1587e-01,  2.9225e-01,  1.2334e-01, -1.1085e+01,\n",
       "           4.0370e-01,  3.8768e-01, -3.8783e-01,  1.4448e-01, -6.0022e-02,\n",
       "           1.8547e-01,  4.5848e-03, -2.1632e-01, -5.9039e-01,  9.1478e-02,\n",
       "           4.4868e-02,  2.5676e-02,  8.7736e-02, -4.1841e-01,  3.1124e-02,\n",
       "          -4.6433e-01, -5.7741e-01,  2.6011e-01, -3.2253e-02,  5.7645e-01,\n",
       "          -6.2887e-02,  1.8610e-01,  1.3448e-01, -1.6144e-01, -1.9398e-02,\n",
       "           4.0461e-02, -5.2220e-01,  1.2185e-01,  4.4506e-01, -6.4249e-02,\n",
       "           6.3196e-02,  1.9590e-01,  4.8324e-01, -1.8829e-01, -4.0467e-01,\n",
       "           3.4881e-01,  7.2491e-03,  2.6047e-01,  1.3439e-01, -2.9041e-01,\n",
       "           1.0263e-01, -7.3466e-01,  4.0929e-01, -1.0528e-02,  7.9633e-01,\n",
       "           1.0485e+00,  9.0708e-01,  2.2950e-01, -2.0993e-01, -2.2005e-02,\n",
       "          -4.6459e-01,  2.7797e-01, -9.3534e-01,  3.2815e-02, -1.1923e-01,\n",
       "           8.5204e-01, -3.7397e-01, -3.3006e-03, -1.7954e-01, -3.1173e-01,\n",
       "           1.7166e-01,  7.7113e-02,  3.0161e-01, -4.1905e-01, -4.1816e-01,\n",
       "           3.1321e-01, -4.9247e-02,  4.0457e-01, -9.5947e-02,  2.3702e-01,\n",
       "          -1.0356e-01,  4.3462e-01,  3.1189e-01, -2.4360e-01,  3.1439e-01,\n",
       "           1.9403e-01, -7.9570e-02,  8.7262e-01, -1.5332e-01, -1.1546e-01,\n",
       "           1.9210e-01,  2.3149e-01, -1.9112e-01, -8.2297e-01, -4.6500e-01,\n",
       "          -6.2964e-02,  3.0936e-02,  3.5310e-01,  3.5346e-01, -1.8343e-01,\n",
       "          -7.7538e-01, -8.7575e-02, -3.4769e-02,  3.7815e-01,  1.1063e+00,\n",
       "          -5.4711e-01,  2.5710e-01, -8.7183e-02, -2.6962e-01, -5.5476e-01,\n",
       "          -2.6118e-02, -1.1649e-01, -4.2367e-01,  4.5194e-01,  1.0561e-01,\n",
       "          -1.9447e-01,  1.7589e-01,  9.0475e-02,  3.2920e-01, -3.1693e-01,\n",
       "           2.3848e-01, -2.4059e-01,  3.2758e-01, -3.4113e-01,  1.0722e-02,\n",
       "           1.5653e-01,  4.1592e-01, -3.5785e-01, -6.2384e-01, -3.9344e-01,\n",
       "          -1.7680e-01,  2.6081e-01,  2.3646e-01, -3.5757e-01,  3.8666e-02,\n",
       "          -6.0910e-01, -1.4685e-01,  4.0903e-02,  2.2681e-01,  3.9131e-01,\n",
       "          -5.5185e-03,  1.4314e-01,  2.6407e-01, -8.1609e-01, -6.8889e-01,\n",
       "          -5.7029e-01, -2.2290e-01, -1.6667e-01, -4.2561e-01, -2.4176e-02,\n",
       "           2.9642e-01, -2.2794e-01, -1.3423e-01,  8.2764e-02,  2.2028e-01,\n",
       "           5.3398e-01,  4.8534e-01, -9.6911e-02, -8.1266e-02, -1.8191e-01,\n",
       "          -6.1937e-01, -2.0188e-01, -3.0536e-01, -7.8819e-02, -2.0203e-01,\n",
       "           1.3021e-01,  2.2207e-01, -2.9004e-01,  4.3733e-01,  3.5416e-01,\n",
       "           5.5238e-02, -2.4922e-01, -2.6525e-01,  1.5753e-01,  6.3989e-02,\n",
       "           2.5667e-01, -2.3879e-01, -4.0272e-01,  1.8737e-01,  3.8423e-01,\n",
       "           7.6236e-02, -1.3753e-01, -5.3187e-01, -1.7217e-02, -5.1450e-02,\n",
       "           5.5346e-01,  1.3569e+00, -3.6419e-01, -2.8408e-01,  2.4599e-01,\n",
       "           5.0316e-01, -7.0082e-02,  1.2618e-01,  1.2429e-01,  4.6297e-01,\n",
       "          -1.3266e-01, -5.8621e-01,  3.2474e-01, -1.4995e-01,  3.9678e-01,\n",
       "           7.6500e-01, -3.5816e-01,  2.8911e-01,  5.7230e-01, -5.1575e-02,\n",
       "           2.0946e-02,  1.5821e-02, -3.4155e-01, -4.8573e-01,  1.5243e-01,\n",
       "           1.0969e-01,  2.7802e-01, -4.3792e-01, -9.1986e-03,  1.7005e-01,\n",
       "           4.1533e-01,  1.0548e+00,  3.3369e-01, -2.5255e-01,  5.0153e-01,\n",
       "           3.1229e-01, -2.8408e-01,  4.3746e-01, -4.3546e-02,  5.9819e-02,\n",
       "           2.7205e-01,  1.7542e-01, -2.2623e-01, -4.8755e-02,  2.8393e-01,\n",
       "           6.8261e-01, -3.3171e-01, -5.4402e-01, -1.8923e-01,  2.9884e-01,\n",
       "           2.9022e-01,  4.5789e-01, -2.0424e-01,  8.0293e-01,  8.7222e-01,\n",
       "          -1.0173e+00, -1.3236e-01, -3.1961e-02,  4.8256e-01, -1.4128e-01,\n",
       "          -7.4893e-01, -9.6035e-02, -2.9280e-01, -3.6747e-01, -2.4918e-01,\n",
       "           6.3461e-01, -7.2958e-01,  3.1096e-01,  1.3678e-01, -2.3126e-01,\n",
       "           1.9824e-01, -1.9285e-01,  9.9017e-02,  3.6835e-01,  1.7626e-01,\n",
       "          -4.7942e-02,  3.1388e-01, -2.7108e-01, -7.5125e-01,  3.3443e-01,\n",
       "           5.9485e-01,  2.8476e-01,  4.9416e-02,  3.4706e-03, -1.8891e-01,\n",
       "           4.1464e-02, -5.6569e-01, -1.5282e-01,  1.8840e-01, -6.5101e-01,\n",
       "           9.9979e-02, -4.4264e-01,  3.3773e-01,  3.2576e-01, -2.2491e-01,\n",
       "           2.7974e-01,  5.2177e-01,  5.5562e-01, -4.2523e-01, -3.3576e-01,\n",
       "          -7.9958e-03,  3.9071e-01, -4.0576e-01,  5.9055e-01, -9.5322e-02,\n",
       "          -2.3789e-01,  2.3314e-01,  3.3516e-01,  2.4167e-01,  2.8250e-01,\n",
       "          -4.7842e-01,  2.4585e-01, -3.3078e-01, -3.3582e-01,  2.0642e-01,\n",
       "           2.4732e-01,  9.2941e-02, -5.9431e-01, -6.6541e-02,  1.4761e-01,\n",
       "           1.2533e-02,  8.2460e-01, -2.9932e-01,  3.7919e-01, -1.1796e-01,\n",
       "           3.0246e-01, -4.4121e-01, -2.3152e-01,  2.3292e-02, -2.5554e-01,\n",
       "          -3.7249e-01, -1.9555e-01,  5.2162e-01,  3.3216e-01,  7.2856e-01,\n",
       "           6.3543e-01, -1.9513e-01, -5.4686e-01, -2.0602e-01,  3.1193e-02,\n",
       "          -1.0029e-01,  2.1102e-01,  2.3073e-01, -2.5515e-02,  2.1278e-01,\n",
       "          -8.6160e-01, -3.6834e-01, -2.2794e-02,  4.3833e-01, -2.4664e-01,\n",
       "           3.8398e-01, -2.4560e-02, -6.3413e-02,  4.9332e-01,  2.8422e-01,\n",
       "           7.9676e-02,  7.2428e-01, -1.0864e-02, -7.5991e-01,  1.4489e-01,\n",
       "           1.8423e-01, -3.0388e-01, -5.4155e-02, -5.1997e-01,  3.8511e-01,\n",
       "           7.8159e-02,  4.5142e-01,  4.6548e-01]], device='mps:0'),\n",
       " tensor([[-1.8125e-01, -5.3941e-01, -6.3927e-01, -1.0098e+00, -2.8783e-01,\n",
       "          -2.1882e-01, -1.0281e+00, -1.7340e-01,  2.4031e-01,  2.7760e-01,\n",
       "           3.0857e-01, -7.7927e-02, -3.6156e-02, -4.2549e-01, -5.2216e-01,\n",
       "           1.8134e-01, -6.8568e-01,  2.2781e-01,  2.1846e-01,  3.0694e-01,\n",
       "           5.0518e-01,  4.5827e-01, -1.2953e-01, -6.7867e-02, -3.8156e-01,\n",
       "           4.8648e-01, -1.4061e-01, -1.0568e-01,  4.1649e-01, -5.6227e-01,\n",
       "          -1.1822e-01,  4.1637e-01, -2.8890e-01,  1.0232e+00, -2.7726e-01,\n",
       "          -1.9397e-01, -1.5722e-02, -2.9039e-01,  7.2799e-01, -1.7010e-01,\n",
       "           7.7617e-01,  1.9518e-02,  1.0304e-01,  1.6628e-01, -8.2880e-02,\n",
       "          -3.5757e-02,  1.3635e-01, -6.0027e-01,  3.1174e-01, -5.5584e-03,\n",
       "           8.5816e-02, -3.5499e-01,  8.5637e-01,  6.0647e-02, -1.0362e+00,\n",
       "          -2.5411e-01,  2.8172e-03,  3.8684e-01, -5.2876e-01,  1.0923e-06,\n",
       "          -4.5613e-01,  7.5637e-01, -1.4170e-01,  2.5820e-02, -6.0593e-02,\n",
       "           2.1621e-01, -2.6360e-01,  1.5498e-02, -1.9825e-01, -5.0399e-02,\n",
       "           1.7821e-01,  1.1281e-01,  2.3618e-01, -2.9955e-01,  2.2628e-01,\n",
       "          -3.0808e-01,  1.9821e-01, -1.8619e-01, -4.1080e-01, -3.2766e-01,\n",
       "           8.1010e-02,  4.1627e-01,  7.2625e-03, -2.5551e-01,  3.7653e-01,\n",
       "           1.0477e-01,  1.8571e-01,  3.9259e-01,  1.2738e-01, -1.3645e-01,\n",
       "           1.7898e-01,  3.8670e-01, -6.6799e-01, -1.5790e-01,  3.8008e-01,\n",
       "          -5.4039e-01,  3.7737e-01, -4.0705e-01, -2.4421e-01, -1.6000e-02,\n",
       "           2.2870e-02,  2.8459e-01,  1.3710e-01,  4.7693e-01, -6.2880e-01,\n",
       "          -2.5019e-01,  4.8294e-02, -1.4122e-01, -8.3285e-01, -4.7727e-02,\n",
       "          -1.8876e-01, -6.8212e-01,  8.8418e-02, -5.0235e-01, -2.4874e-01,\n",
       "           1.2123e-01,  2.9727e-01, -6.3440e-03,  4.3220e-02,  6.7753e-01,\n",
       "           5.0907e-01, -6.1834e-01,  6.9245e-02, -1.2055e-01,  1.9358e-01,\n",
       "          -2.7235e-01,  2.1309e-02,  3.1925e-02,  3.8671e-01,  4.3912e-01,\n",
       "           1.2180e-01,  2.4186e-01, -2.3732e-01,  4.5976e-02, -9.5561e-02,\n",
       "           1.5169e-01,  8.4453e-02, -5.2913e-01,  4.6545e-01, -5.4024e-02,\n",
       "          -7.8597e-02, -1.2819e-01, -2.3233e-01, -4.9233e-02, -3.4019e-01,\n",
       "          -3.2730e-01,  1.5719e-01,  6.3312e-02,  1.8003e-01,  1.3730e-01,\n",
       "          -4.1650e-01, -4.6239e-01,  3.8184e-01, -2.2120e-01, -4.8124e-02,\n",
       "          -1.7404e-01,  2.9756e-01,  6.8767e-02, -3.8523e-01,  3.7724e-01,\n",
       "           6.5312e-01, -2.5044e-01,  2.4868e-01, -7.5985e-01,  5.6022e-02,\n",
       "          -8.2676e-02,  4.6790e-01, -1.6298e-01,  4.1384e-01, -1.6411e-01,\n",
       "           3.7614e-01, -1.1070e-01,  1.2000e-01, -3.7964e-01,  2.6834e-01,\n",
       "          -1.9826e-01,  5.5964e-01,  4.1540e-01, -1.7262e-01,  1.9281e-01,\n",
       "           3.3436e-01, -1.6120e+00,  2.9266e-01, -1.1052e+00, -8.4049e-03,\n",
       "          -2.8378e-01,  4.5338e-01, -1.0799e-01,  6.0614e-01,  3.3641e-01,\n",
       "           2.0493e-01,  4.8331e-01,  3.0620e-01,  7.6288e-01,  6.7834e-02,\n",
       "           2.3403e-01,  7.8490e-01,  5.0176e-01, -1.2374e-01, -1.5611e-01,\n",
       "          -1.2346e+00,  3.4243e-01, -3.7400e-01,  3.7436e-01, -2.7116e-01,\n",
       "           4.0014e-01, -1.8428e-01,  2.6521e-01,  3.9437e-01, -8.8023e-02,\n",
       "          -1.0504e-01, -3.1978e-01, -5.5941e-01, -2.2290e-01, -3.4764e-01,\n",
       "          -3.0162e-01, -1.8807e-01, -2.2434e-01,  4.4897e-01,  3.3236e-01,\n",
       "           1.5925e-01,  1.3939e-01,  7.5576e-02, -2.0424e-01, -4.1669e-01,\n",
       "           3.4301e-01,  3.6863e-02, -2.1983e-01, -5.9604e-01, -1.9368e-01,\n",
       "           1.3661e-01, -3.9137e-01,  5.5508e-01,  7.5245e-01,  5.4772e-02,\n",
       "          -2.2472e-01, -9.5399e-02,  5.3804e-01,  4.3691e-02, -2.4827e-01,\n",
       "          -2.7490e-01, -1.7186e-01, -6.0788e-01,  1.6213e-01,  2.3377e-01,\n",
       "           3.6288e-01,  1.1392e-01,  8.1961e-01,  8.8090e-01,  7.9032e-01,\n",
       "           2.7554e-02,  9.1280e-02,  2.5115e-01,  2.9210e-01, -3.0370e-01,\n",
       "           8.3534e-02,  1.4907e-01, -2.3136e-01,  2.3131e-01, -3.9355e-01,\n",
       "          -5.9320e-01, -8.6518e-02,  1.1780e-01,  6.2063e-01, -3.7438e-01,\n",
       "           5.4174e-02,  4.8206e-01, -2.8925e-01, -2.5506e-01,  3.0643e-01,\n",
       "          -2.5924e-01,  3.2578e-02, -2.5601e-01, -1.6621e-01, -6.7339e-01,\n",
       "          -8.2424e-01, -7.7856e-02,  1.4707e-01,  1.0011e-01, -7.4542e-01,\n",
       "          -1.0964e-02,  1.7475e-01, -3.6762e-02,  8.3423e-02, -2.2309e-01,\n",
       "           4.9886e-01, -3.7746e-01,  6.7744e-02, -7.1699e-02, -2.4615e-01,\n",
       "           1.1049e-01,  2.3905e-02, -2.9412e-01, -1.2295e-01,  1.4983e-01,\n",
       "          -2.7230e-01,  3.0218e-01,  2.1994e-01, -1.4860e-01, -6.8041e-04,\n",
       "          -6.2754e-02,  6.1179e-02, -1.0572e-01, -1.1839e-01,  3.4980e-01,\n",
       "           1.7629e-01,  6.2785e-02,  8.3437e-01, -6.5658e-01, -4.6387e-01,\n",
       "          -3.7774e-02,  4.0610e-01, -3.2336e-01,  5.4144e-02, -3.7941e-01,\n",
       "          -8.5244e-01, -3.5678e-01, -6.6477e-02, -2.9339e-02, -5.5914e-01,\n",
       "           1.9270e-01, -2.2854e-01,  1.5188e-01, -2.8985e-01, -3.5161e-01,\n",
       "           5.6875e-01,  7.5265e-01,  2.2035e-01, -2.4067e-01, -3.0883e-01,\n",
       "          -1.4852e-01,  5.6358e-02, -5.5831e-01, -6.5955e-01, -3.7322e-01,\n",
       "           3.2970e-01,  7.5011e-01, -1.0687e-01, -1.5464e-01, -1.9335e-01,\n",
       "           7.5880e-01, -2.4774e-01, -1.2847e-01,  4.5503e-01, -3.1264e-01,\n",
       "          -1.8458e-01,  3.5380e-01,  3.2619e-01,  3.2099e-01, -7.3481e-03,\n",
       "           3.9081e-01, -2.2947e-02, -4.5570e-01,  7.5956e-02,  7.8667e-01,\n",
       "           1.2923e-01,  2.2185e-01, -2.8903e-01,  7.1499e-01,  1.8670e-01,\n",
       "          -9.5475e-02,  2.7283e-01, -3.8227e-01, -2.2453e-01, -1.8390e-01,\n",
       "           1.6456e-01,  3.1698e-01, -4.5740e-01,  3.4663e-01, -1.8621e-01,\n",
       "           2.1728e-01, -4.9834e-01,  7.7680e-02,  4.5806e-01,  3.8794e-01,\n",
       "           5.6253e-02, -4.3948e-01,  3.8953e-01, -8.4529e-01, -2.0620e-01,\n",
       "          -6.9546e-02, -5.7147e-02,  1.9121e-01, -4.3970e-01, -4.4468e-01,\n",
       "          -2.0808e-01, -1.7028e-01,  7.9077e-01,  2.2017e-01,  4.4573e-02,\n",
       "           1.9766e-02, -3.9475e-01, -3.9663e-01,  6.8009e-01, -3.6712e-01,\n",
       "          -9.3259e-02,  2.8212e-01,  1.0940e-01,  6.5341e-02,  7.5319e-01,\n",
       "          -4.2010e-01, -7.6439e-02,  3.6539e-01, -4.1567e-01, -4.2418e-01,\n",
       "          -3.9862e-01,  2.4993e-01,  6.3705e-02, -3.2124e-01,  2.3731e-01,\n",
       "           1.4922e-01, -6.8350e-02, -3.1705e-02, -2.2497e-02,  1.7119e-01,\n",
       "           1.5659e-01, -4.3346e-01, -1.4841e-01, -2.4025e-01, -3.7714e-01,\n",
       "           6.5428e-01,  1.9340e-01,  1.0309e-01,  5.0786e-01, -1.0892e+01,\n",
       "           2.9662e-01,  6.9372e-01, -3.7809e-01, -2.1889e-01,  5.1872e-01,\n",
       "           4.0801e-01, -3.3957e-01,  3.8911e-01, -4.6351e-01, -5.3969e-01,\n",
       "          -4.5127e-01, -1.3306e-01,  8.0967e-03, -6.1188e-01, -2.7709e-01,\n",
       "           1.1543e-01, -8.9459e-01,  4.2263e-01, -3.3380e-02,  6.9925e-01,\n",
       "          -3.4596e-01,  2.6562e-01,  1.6402e-01, -3.2485e-02, -2.0233e-01,\n",
       "           2.5420e-01, -2.7514e-01,  3.0105e-01,  2.3100e-01,  8.1241e-02,\n",
       "          -2.9693e-01, -1.6632e-01, -9.2963e-02, -5.2695e-01, -6.8050e-01,\n",
       "          -4.4837e-02, -8.6848e-02,  2.1930e-01,  2.9385e-01, -2.7491e-02,\n",
       "          -3.0097e-01, -2.1279e-01,  9.0686e-02,  4.2268e-01,  9.6398e-01,\n",
       "           1.9275e+00,  4.6355e-01,  2.2661e-01,  1.1332e-01, -4.3098e-01,\n",
       "          -2.6632e-01,  2.8910e-01, -4.5070e-01,  2.7557e-01,  1.7609e-01,\n",
       "           3.5182e-01, -5.0869e-01,  3.7245e-01, -2.5337e-01, -2.7457e-01,\n",
       "           5.3877e-01,  4.9399e-02,  1.2313e-01,  5.2775e-02, -2.8543e-01,\n",
       "           2.7016e-01, -3.3746e-01,  2.4167e-01, -3.7608e-01,  2.6903e-01,\n",
       "          -1.8995e-01,  8.6108e-01,  1.5673e-02,  1.6334e-01,  6.3780e-02,\n",
       "           2.1611e-02, -1.4699e-01,  9.3771e-01,  2.9691e-02,  2.0411e-01,\n",
       "           9.7656e-02,  5.0048e-03, -8.6828e-01, -6.0111e-01, -4.8738e-01,\n",
       "          -5.4427e-02, -1.1216e-01,  4.5284e-01,  3.7178e-01, -3.2015e-02,\n",
       "          -6.7463e-01, -3.1682e-01,  1.3176e-01,  5.6663e-01,  2.0259e-01,\n",
       "          -8.5370e-01, -5.9560e-01,  2.8762e-01,  5.4299e-02, -4.5511e-01,\n",
       "          -3.1251e-02,  7.2466e-03, -3.8234e-01,  4.6630e-01, -9.0379e-01,\n",
       "           8.7347e-02, -1.9629e-01, -2.3685e-01,  2.2633e-01, -4.1747e-01,\n",
       "           1.4283e-01,  1.9851e-01,  2.4983e-01, -1.8039e-01, -3.4200e-01,\n",
       "           2.7873e-01,  5.0424e-01, -3.3787e-01, -3.0782e-01, -2.1623e-01,\n",
       "           2.8449e-03,  2.2240e-01,  2.3333e-01, -4.9540e-01,  4.7009e-01,\n",
       "          -9.2407e-01,  4.4949e-01,  3.2892e-01,  1.8150e-01,  2.7510e-01,\n",
       "          -3.7568e-01, -2.0037e-01,  2.6296e-01, -4.6461e-01, -4.3094e-01,\n",
       "          -3.6326e-01,  2.0760e-01,  2.2593e-01, -5.6026e-01, -4.6590e-01,\n",
       "          -3.4866e-01,  9.1570e-03, -4.5285e-01,  2.4233e-01, -2.2839e-01,\n",
       "           3.2850e-01,  2.3210e-01, -3.8039e-01,  3.1926e-01, -1.1399e-01,\n",
       "          -8.6524e-02,  5.4909e-02, -2.0074e-01, -1.1373e-01, -5.5656e-01,\n",
       "          -3.3827e-01,  5.0743e-02,  1.9318e-02,  7.9585e-02, -7.4153e-01,\n",
       "           7.0116e-03, -3.4823e-01, -3.7502e-01,  8.7393e-01,  4.6544e-01,\n",
       "           2.2184e-01, -4.6954e-01,  1.1614e-01, -9.9412e-02,  1.9880e-01,\n",
       "           2.5492e-01,  4.4476e-01, -8.0296e-02, -7.9232e-01, -3.1004e-01,\n",
       "          -1.8270e-01,  7.7770e-01, -2.7541e-01, -2.9676e-01,  9.2858e-02,\n",
       "          -4.8191e-01, -1.6215e-01,  3.2633e-01, -1.8703e-01, -1.8444e-01,\n",
       "           5.6220e-02, -1.8540e-01,  4.9326e-01, -1.1413e-01,  3.8113e-01,\n",
       "          -9.4738e-02,  3.2597e-01, -2.5507e-02,  4.8348e-01, -4.2563e-01,\n",
       "          -5.7592e-01,  9.1995e-02, -3.8031e-01,  2.8746e-01, -7.2663e-02,\n",
       "          -8.4291e-02,  4.7406e-02, -2.3045e-01, -3.7923e-02,  2.9133e-01,\n",
       "           4.6604e-02,  1.4119e+00, -5.3291e-01,  3.8299e-03,  3.1469e-01,\n",
       "           7.8734e-02,  2.9609e-01, -6.5136e-03, -3.1731e-01,  4.1293e-01,\n",
       "           3.8461e-01, -4.4902e-01, -3.5394e-01,  4.4121e-01,  1.1087e-01,\n",
       "           1.9829e-01, -9.1236e-01, -7.0067e-01, -3.6629e-01,  6.1876e-01,\n",
       "          -2.3047e-01,  3.2566e-02, -1.5090e-01,  5.2744e-01,  7.4904e-01,\n",
       "          -6.5487e-01,  9.9778e-02, -5.9713e-01,  9.3523e-01, -1.9166e-01,\n",
       "          -1.5790e-01, -1.8765e-02, -6.8289e-01, -1.5308e-01,  3.8234e-01,\n",
       "           7.2352e-01, -1.2231e+00,  8.6156e-02,  3.1197e-01, -6.2155e-01,\n",
       "           1.9322e-01, -3.2950e-01,  3.6156e-01,  1.3380e-01,  2.4767e-02,\n",
       "          -2.4486e-01,  2.5362e-01, -5.6344e-01, -6.5031e-01,  4.8332e-02,\n",
       "           6.9805e-01,  6.5346e-02, -1.4268e-02, -1.3936e-01, -7.3708e-01,\n",
       "          -1.8298e-01,  3.6978e-02, -2.4582e-01, -2.4035e-01,  1.9283e-01,\n",
       "          -3.7349e-02, -5.4113e-01,  7.8672e-02,  3.0387e-01, -1.4832e-02,\n",
       "           2.9087e-01,  4.3360e-01,  6.3519e-01,  6.7011e-02, -5.1458e-01,\n",
       "           1.0903e-01,  2.2647e-01, -2.8595e-01,  4.3698e-01, -2.4263e-01,\n",
       "          -3.2825e-01,  6.5951e-01, -1.1096e-01, -4.7134e-01, -2.6599e-02,\n",
       "          -7.2647e-02,  4.3989e-01, -7.3078e-03, -3.0432e-01,  1.4181e-01,\n",
       "           2.5430e-01, -5.2127e-03, -4.5927e-01, -1.8771e-02,  5.4225e-02,\n",
       "          -7.6366e-02,  7.5391e-01,  1.2748e-01,  4.0979e-01, -3.6831e-01,\n",
       "           1.4192e-01, -9.9577e-01,  2.2156e-02,  8.1579e-02, -6.1962e-02,\n",
       "          -3.6303e-01, -8.6250e-01,  3.8323e-01,  3.8475e-01,  6.8750e-01,\n",
       "           4.5516e-01, -2.4845e-01, -2.3551e-01, -2.9733e-01,  4.4536e-02,\n",
       "          -6.2662e-02,  1.4700e-02,  2.2100e-01,  9.8156e-02,  1.6497e-01,\n",
       "          -4.1744e-01, -4.6911e-01,  1.6927e-01,  4.4333e-01, -7.4590e-02,\n",
       "           8.1072e-01, -3.2187e-01,  3.3466e-01,  6.7860e-01,  4.6642e-01,\n",
       "           9.1481e-02,  5.3016e-01,  2.2833e-01, -1.0606e+00,  2.3997e-01,\n",
       "           8.0776e-02, -9.7066e-02,  3.6242e-01, -1.8799e-01,  1.6469e-01,\n",
       "           3.5002e-01,  2.5631e-01,  7.7674e-01]], device='mps:0'),\n",
       " tensor([[-0.0986, -0.6295, -0.5474,  ...,  0.1507,  0.8416,  0.6531],\n",
       "         [-0.2176, -0.4739, -0.7415,  ...,  0.0314,  0.6988,  0.3850]],\n",
       "        device='mps:0'),\n",
       " tensor([[-0.0892, -0.4743, -0.6687,  ...,  0.3833,  0.3221,  0.6478],\n",
       "         [-0.0583, -0.6757, -0.8037,  ...,  0.1694,  0.7852,  0.5016]],\n",
       "        device='mps:0'),\n",
       " tensor([[-1.7159e-01, -6.5047e-01, -4.6892e-01, -7.7307e-01, -5.7357e-01,\n",
       "          -1.9739e-01, -4.5961e-01,  2.6979e-01, -1.5224e-01,  1.4185e-01,\n",
       "           3.3978e-01,  3.3370e-01, -2.4403e-01, -5.9609e-03, -3.1489e-01,\n",
       "          -1.4180e-01, -1.7653e-01,  6.5161e-03,  4.6293e-01,  1.7347e-01,\n",
       "           5.5889e-01,  1.8253e-01, -3.5867e-01,  1.9117e-01, -5.9602e-01,\n",
       "           2.9500e-02, -5.0375e-01, -5.4658e-02,  3.2622e-01, -1.1037e-01,\n",
       "          -1.0703e-01,  5.1767e-01, -6.6235e-02,  1.0792e+00, -1.2428e-01,\n",
       "          -2.5799e-01,  1.0123e-01, -1.4654e-01,  5.2091e-01, -5.6293e-01,\n",
       "           4.4805e-01,  2.5727e-01, -1.4242e-01,  1.2044e-01,  2.3191e-01,\n",
       "           2.0883e-01,  2.2718e-01, -7.7344e-01, -3.4872e-01, -5.0666e-01,\n",
       "           1.3562e-02, -2.0815e-01,  4.3230e-01,  3.3282e-02, -8.3201e-01,\n",
       "          -1.5258e-01,  1.2702e-01,  1.1798e-01, -2.1948e-01,  2.6948e-01,\n",
       "          -5.3377e-02,  8.5041e-01, -9.2169e-02,  9.7170e-02,  1.7776e-01,\n",
       "           8.6154e-02, -7.0459e-01, -2.4800e-01, -5.7381e-01,  8.3544e-02,\n",
       "           1.3922e-01,  1.6696e-01,  4.5450e-02, -3.4133e-01, -2.0937e-01,\n",
       "           1.3208e-01,  1.3343e-01, -4.3265e-02, -4.7486e-01, -9.5042e-02,\n",
       "           3.4916e-01,  2.8492e-01,  3.7865e-01,  1.3408e-01,  2.5013e-02,\n",
       "           3.9855e-02,  2.4166e-01,  5.3213e-02,  1.3812e-01,  2.5194e-01,\n",
       "           1.9194e-01, -9.9673e-02, -2.3794e-01, -3.5270e-01, -1.0700e-01,\n",
       "          -3.0383e-01,  3.0599e-01, -1.9025e-01, -3.1865e-02, -2.3253e-01,\n",
       "           4.2720e-01,  2.8724e-01,  2.0753e-01, -5.2877e-02, -7.0237e-01,\n",
       "          -3.1930e-01,  1.3017e-01, -3.7113e-02, -4.8714e-01, -1.6248e-01,\n",
       "          -1.9357e-01, -2.1185e-01,  2.2623e-02, -5.3685e-01, -1.2083e-01,\n",
       "           1.2992e-01, -2.1886e-03, -3.4598e-01,  8.9042e-02,  6.7334e-01,\n",
       "           1.8372e-01, -5.4353e-01,  1.4977e-01,  1.9256e-01,  3.6322e-01,\n",
       "           3.9484e-02,  4.4994e-01, -2.9496e-01,  2.8164e-01,  3.7980e-01,\n",
       "           2.5548e-01,  3.5259e-01, -3.5812e-01, -2.0636e-01, -1.4060e-01,\n",
       "           3.1377e-02, -3.5639e-01, -5.5987e-01,  2.4253e-01, -2.8923e-01,\n",
       "           6.9884e-02, -4.8735e-03, -4.3275e-01, -3.4602e-01, -3.1855e-01,\n",
       "          -2.9919e-01,  2.8710e-02, -1.2665e-01, -2.5756e-01, -1.8886e-01,\n",
       "          -7.5295e-01, -6.2517e-01,  1.3672e-01,  2.5578e-02,  4.8213e-02,\n",
       "          -4.9841e-01,  3.9719e-01, -1.0320e-01, -3.9466e-01,  3.4641e-01,\n",
       "           4.2729e-01, -5.0812e-01,  4.6166e-01, -5.2250e-01,  1.3222e-01,\n",
       "           1.7357e-01, -2.6362e-01, -5.3655e-02,  4.4478e-01,  6.6407e-02,\n",
       "           5.3273e-01,  1.8553e-01,  6.4902e-02, -3.7514e-01,  2.0584e-01,\n",
       "           1.0791e-01, -2.2305e-02,  1.3956e-01, -1.3657e-01, -9.2297e-02,\n",
       "           2.0061e-01, -1.5784e+00, -4.3704e-02, -8.1728e-02, -4.1353e-01,\n",
       "          -2.0652e-01,  1.3446e-01,  1.5144e-01,  3.9993e-01, -3.6618e-01,\n",
       "           1.1774e-01, -4.5520e-02,  2.9456e-01,  4.0551e-01,  1.7826e-01,\n",
       "           3.1747e-01,  7.1402e-01,  1.8038e-01,  2.2965e-01, -1.9335e-01,\n",
       "          -6.7158e-01,  3.6547e-01,  2.1628e-01,  3.2718e-01,  1.2909e-01,\n",
       "           1.2426e-01, -6.0367e-01,  2.6312e-01,  4.3107e-01,  2.2555e-01,\n",
       "          -7.3752e-03, -3.1894e-01, -4.6796e-01, -2.3385e-01, -5.1302e-01,\n",
       "           2.0488e-01, -2.4528e-01, -1.4014e-01,  5.6595e-01,  9.3785e-02,\n",
       "           6.8394e-02,  2.8835e-01, -9.7830e-03, -3.9971e-01,  1.3898e-01,\n",
       "           6.0062e-01,  6.2234e-02,  6.2612e-02, -3.4351e-01, -2.5360e-01,\n",
       "          -2.9887e-01, -6.4861e-01,  5.9931e-01,  5.5294e-01,  1.1954e-01,\n",
       "           2.9507e-01, -3.1858e-01,  3.1088e-01, -2.4976e-01, -3.5447e-02,\n",
       "          -1.1823e-01, -2.6050e-02, -3.8145e-01,  2.3045e-02,  1.1682e-01,\n",
       "           8.4044e-02,  1.3324e-01,  7.2705e-01,  6.6443e-01,  1.6017e-01,\n",
       "           3.5661e-01,  1.8067e-01,  2.5884e-01,  7.5619e-01, -2.8028e-01,\n",
       "           3.0379e-01, -1.1318e-01, -5.2433e-01,  4.6763e-01, -3.6890e-01,\n",
       "           1.6221e-01, -1.6347e-01,  2.0841e-03,  3.0808e-01, -3.7264e-01,\n",
       "           1.2310e-01,  4.8302e-01,  1.7607e-01, -4.3742e-01,  6.8671e-01,\n",
       "          -4.2232e-01,  7.4595e-01,  1.5351e-01, -4.9302e-02, -2.8704e-01,\n",
       "          -9.4881e-01, -2.3991e-01,  3.1303e-01,  1.5389e-01, -8.4223e-01,\n",
       "          -2.2968e-01,  3.5899e-01, -4.7876e-02,  3.4716e-01, -3.8491e-02,\n",
       "           4.3019e-01, -1.2758e-01, -1.3954e-02, -1.7301e-01, -3.4912e-01,\n",
       "          -1.8023e-01,  1.6746e-01, -5.8060e-02, -1.0679e-01,  1.2608e-01,\n",
       "           4.6723e-01,  1.0312e-01,  2.8722e-01,  6.4884e-02, -2.0756e-01,\n",
       "          -2.0162e-01, -3.8638e-02,  3.0154e-01, -6.5958e-01, -1.3738e-01,\n",
       "           2.6170e-01, -3.6213e-02,  5.9721e-01, -1.0533e-01, -3.6909e-01,\n",
       "           1.2765e-03,  8.2945e-01, -8.9379e-02,  2.0624e-01, -1.9657e-02,\n",
       "          -8.9562e-01,  1.5446e-01, -5.0867e-02, -1.0947e-01, -4.9132e-01,\n",
       "           3.0926e-01, -1.6417e-01, -1.4995e-01, -2.7832e-03, -5.7823e-02,\n",
       "           1.0907e-01,  3.4243e-01,  4.6750e-01, -4.0586e-01, -2.6991e-01,\n",
       "           1.7525e-01, -4.7475e-01, -1.3052e-01, -3.7328e-01,  2.6785e-01,\n",
       "           1.2886e-01,  4.5299e-01, -1.0578e-01,  2.4979e-02, -2.7891e-01,\n",
       "           6.2462e-01, -4.3759e-01,  1.1515e-01,  5.0476e-01, -3.2657e-01,\n",
       "          -2.9824e-01,  3.7560e-01, -1.9198e-01,  1.8900e-01,  1.2198e-01,\n",
       "           4.0765e-01, -2.8466e-01, -4.9751e-01,  1.3282e-02,  7.9118e-01,\n",
       "           5.3408e-01,  2.2380e-01, -7.2697e-01,  2.9264e-01,  2.8161e-01,\n",
       "          -4.2776e-01,  1.2326e-01, -2.7723e-01, -1.6902e-01,  3.1018e-02,\n",
       "           5.6207e-02,  2.7873e-01, -7.4539e-02, -2.1744e-02,  2.1758e-01,\n",
       "           5.0702e-01, -7.8891e-01, -2.4744e-01,  5.6740e-01,  2.5977e-01,\n",
       "           8.5632e-02, -6.4480e-01,  1.8811e-01, -7.9084e-01, -7.8004e-02,\n",
       "          -4.1582e-02, -1.8918e-01,  2.7330e-01, -3.8566e-02, -1.9932e-01,\n",
       "           5.9549e-02, -4.7566e-01,  4.4088e-02,  1.7123e-01, -1.4061e-01,\n",
       "           3.9526e-01, -9.0444e-02, -2.9539e-01,  9.6742e-01, -2.8209e-01,\n",
       "          -1.4511e-01,  4.1500e-01, -5.5024e-01,  1.1629e-02,  2.2414e-01,\n",
       "          -2.3043e-01, -4.2548e-02,  1.4894e-01,  9.0645e-01, -1.3648e-01,\n",
       "          -5.1201e-02, -8.4130e-02,  5.0899e-01, -3.0259e-01,  1.5145e-01,\n",
       "          -3.3726e-01,  5.7827e-01, -1.5673e-01,  3.2224e-01, -4.8520e-01,\n",
       "           1.7041e-01, -3.4058e-01, -3.4397e-01, -6.0091e-01, -2.7220e-02,\n",
       "           3.3564e-02,  7.4224e-02, -7.0578e-02,  5.1454e-01, -1.1204e+01,\n",
       "           1.0280e-01,  8.5940e-01, -2.0909e-01,  6.2486e-02,  4.3630e-01,\n",
       "           6.8040e-02, -4.1050e-01,  5.6518e-01, -3.5846e-01, -1.5531e-01,\n",
       "          -1.6727e-01,  2.9257e-02, -9.0033e-02, -5.9051e-01,  1.7671e-01,\n",
       "          -2.9676e-01, -6.0503e-01,  3.4807e-01, -1.5596e-01,  4.4483e-01,\n",
       "          -3.8384e-01,  3.7654e-01,  4.7517e-01, -3.2096e-01, -6.4504e-01,\n",
       "          -4.6052e-01, -4.1239e-01,  3.6120e-01,  2.2476e-01, -2.9389e-01,\n",
       "           1.7052e-01, -3.2415e-02,  5.2330e-01, -2.8162e-01, -5.3914e-01,\n",
       "           5.3090e-01, -2.7322e-01,  3.3299e-01, -8.0288e-01, -1.2644e-01,\n",
       "          -2.1055e-02, -7.4782e-02,  6.6511e-02,  5.0478e-01,  8.2748e-01,\n",
       "           1.4175e+00,  9.6365e-01,  1.0888e-01,  4.4945e-01, -1.1501e-01,\n",
       "          -7.8258e-01,  2.6661e-01, -7.9946e-01, -2.7458e-01,  4.6589e-02,\n",
       "           4.4600e-01, -6.6713e-01,  1.7541e-01, -3.9320e-01,  6.2547e-02,\n",
       "           2.3300e-01,  8.6906e-02,  2.3635e-01, -3.5361e-01, -6.7642e-01,\n",
       "           2.6204e-01,  2.8101e-02,  6.2602e-02, -4.7074e-01, -1.2272e-01,\n",
       "           3.5490e-02,  7.3836e-01, -4.9196e-02, -2.8705e-01,  5.4050e-01,\n",
       "           2.8242e-02,  1.4599e-01,  1.2602e+00, -2.6538e-01, -7.0624e-02,\n",
       "           1.6753e-01, -1.3377e-01, -3.3530e-01, -3.1165e-01, -3.4813e-01,\n",
       "          -8.0899e-03,  6.1544e-01,  3.9284e-01,  5.0541e-01,  7.2390e-02,\n",
       "          -3.4672e-01, -1.0037e-01, -4.3728e-02,  4.1288e-01,  8.3799e-01,\n",
       "          -5.2679e-01,  1.0762e-01, -1.5560e-01, -6.6085e-02, -1.7208e-01,\n",
       "          -5.6240e-01, -7.8968e-02, -3.4958e-01,  5.4192e-02,  6.7743e-02,\n",
       "           1.1280e-01,  2.4702e-01,  3.0419e-01,  2.8995e-01, -3.6962e-01,\n",
       "          -3.6434e-01, -1.7675e-01, -1.0300e-01, -5.0338e-01, -2.8418e-01,\n",
       "          -1.6072e-01,  2.7227e-01, -1.8020e-01, -7.3206e-01, -2.1556e-01,\n",
       "           1.8144e-01, -1.0423e-01,  4.5888e-02, -7.6308e-02,  5.2157e-01,\n",
       "          -7.8070e-01,  1.7355e-01, -4.8350e-02,  1.0537e-01,  3.4751e-01,\n",
       "           3.4353e-01, -7.1171e-02,  1.8005e-01, -1.8422e-01, -5.7355e-01,\n",
       "          -7.4101e-01,  1.0339e-01,  7.6731e-01, -1.7519e-01,  1.9002e-01,\n",
       "           1.2513e-01,  8.8551e-02, -4.1736e-01,  2.3265e-01, -3.6784e-01,\n",
       "           2.2908e-01,  3.2876e-01, -1.4699e-01,  7.1711e-02, -1.8989e-01,\n",
       "          -1.8792e-01, -1.3460e-01,  1.0369e-02, -4.1293e-01, -6.4693e-01,\n",
       "          -2.3785e-01,  2.3227e-01, -1.4456e-01,  6.8693e-02, -9.5862e-02,\n",
       "           6.4348e-02, -2.6669e-01, -6.3863e-02,  7.0364e-01,  5.5976e-01,\n",
       "           1.4996e-01, -2.9408e-01, -6.4316e-01,  2.0870e-02, -1.1603e-02,\n",
       "           5.1370e-02,  1.4113e-01,  1.0587e-01, -7.6472e-01, -8.2909e-02,\n",
       "          -1.5996e-01,  1.2037e+00, -2.8130e-01, -2.8684e-01,  2.5980e-01,\n",
       "           5.2501e-01, -4.5725e-01, -3.1406e-01, -1.8582e-02, -1.6877e-01,\n",
       "          -5.1608e-01, -1.8773e-01,  3.6167e-01, -3.7508e-01,  2.5738e-01,\n",
       "           5.3673e-01,  1.0730e-01, -1.5087e-01,  8.5621e-02, -1.4998e-01,\n",
       "          -3.6484e-01,  1.8381e-01, -3.7362e-01, -1.6737e-01,  7.5027e-02,\n",
       "          -3.0103e-02, -4.1076e-02, -1.7187e-01, -4.3327e-01,  1.4190e-01,\n",
       "           1.6772e-01,  1.1909e+00, -2.0945e-01, -1.7193e-01,  4.6823e-01,\n",
       "          -1.6816e-01,  5.5216e-01,  2.9719e-01, -2.0956e-01,  6.5382e-02,\n",
       "           5.0906e-01,  2.6612e-01, -6.4469e-01,  2.3646e-01,  1.8520e-01,\n",
       "          -1.3365e-01, -6.7735e-01, -2.6910e-01, -3.1256e-01,  3.2793e-01,\n",
       "           6.5165e-02,  1.0986e-01, -8.8632e-02,  4.2956e-01,  2.5570e-01,\n",
       "          -9.3287e-01,  3.5987e-01,  2.4357e-01,  4.3485e-01,  4.7925e-03,\n",
       "          -4.7468e-01, -2.2638e-01, -2.6400e-01, -1.9986e-01,  5.0004e-01,\n",
       "           4.6015e-01, -9.8469e-01, -1.1135e-01,  3.3994e-01, -5.7210e-02,\n",
       "           4.8980e-01, -1.2018e-01,  2.2646e-01,  5.3403e-01,  6.0998e-02,\n",
       "           5.5359e-02,  2.2500e-01,  9.5417e-02, -3.9015e-01,  3.1351e-01,\n",
       "           4.1853e-01, -5.9112e-02,  7.5738e-02,  5.6640e-02, -9.4292e-02,\n",
       "          -2.8567e-01, -9.8340e-02, -3.7998e-01, -4.2104e-01,  5.8898e-02,\n",
       "          -2.7289e-01, -5.1226e-01,  1.3072e-01,  2.3095e-01, -2.1044e-01,\n",
       "           5.3698e-01,  4.2677e-01,  4.4744e-01, -2.5148e-01, -4.5781e-02,\n",
       "          -4.8679e-01,  3.0969e-01, -2.3851e-01,  7.9092e-01, -3.0350e-02,\n",
       "          -1.4857e-01,  3.2221e-01,  2.2732e-01, -1.5367e-01,  2.1795e-01,\n",
       "          -1.8111e-01,  2.1058e-01, -4.1506e-01, -3.0724e-01,  2.4939e-02,\n",
       "          -6.8988e-02, -2.6676e-01, -4.5556e-01, -3.8178e-02,  1.2695e-01,\n",
       "          -3.5145e-01,  9.0709e-01, -3.4333e-01,  4.4069e-01,  8.4794e-03,\n",
       "          -9.4187e-02, -1.0052e+00,  3.6713e-02,  4.7556e-01, -1.8721e-02,\n",
       "          -5.8769e-01, -3.7404e-01, -1.5168e-01, -2.0319e-01,  7.8235e-01,\n",
       "           4.2380e-01, -5.3370e-01, -5.7441e-01, -7.1838e-02,  7.2351e-02,\n",
       "           2.5231e-02, -2.6629e-01, -8.6414e-02, -3.4329e-02, -1.2315e-01,\n",
       "          -9.9100e-01,  7.8933e-02, -1.6397e-01,  2.7466e-01, -2.6690e-01,\n",
       "           9.4324e-01, -3.1653e-01, -1.7136e-02,  9.9673e-01,  3.2777e-02,\n",
       "           2.2927e-01,  7.6203e-01, -5.1125e-02, -8.6275e-01, -7.5736e-02,\n",
       "          -7.7563e-02, -2.7339e-01, -1.6368e-03, -3.7510e-01,  4.9680e-01,\n",
       "           3.6803e-01, -1.0470e-01,  6.3944e-01]], device='mps:0'),\n",
       " tensor([[-1.9586e-01, -6.7376e-01, -5.8137e-01, -3.6831e-01, -3.1885e-02,\n",
       "          -3.5125e-01, -5.5109e-01,  4.3960e-02,  1.4900e-01,  5.1506e-01,\n",
       "           1.4517e-01, -1.3691e-01,  4.7542e-02, -3.8053e-01, -4.6936e-01,\n",
       "          -4.7868e-01, -2.5590e-02,  5.5842e-01,  5.0056e-01,  3.9158e-01,\n",
       "           5.0458e-01,  7.1126e-01, -1.8067e-01, -5.8518e-03, -3.0800e-01,\n",
       "           1.1927e-01, -2.2374e-01,  3.0469e-01,  2.4866e-01, -6.6601e-01,\n",
       "           7.3954e-02,  4.9656e-01, -1.3839e-01,  9.1732e-01, -6.8926e-02,\n",
       "          -3.4130e-01,  4.8362e-02, -9.0161e-01,  4.4485e-01,  2.1938e-01,\n",
       "           8.7234e-01, -8.0124e-02,  1.6246e-01,  1.5784e-01,  4.3713e-01,\n",
       "           1.0264e-01,  2.4015e-01, -3.9350e-01,  6.3392e-02, -2.1689e-01,\n",
       "          -8.4904e-02,  3.8853e-01,  2.1457e-01,  7.2199e-01, -6.6364e-01,\n",
       "          -4.7429e-01,  1.7673e-01,  4.7546e-01, -1.3892e-01,  6.3759e-02,\n",
       "          -3.8133e-01,  1.1183e-01,  2.1317e-01,  1.2243e-01,  2.5850e-01,\n",
       "          -1.1669e-01, -4.1596e-01, -7.3153e-01, -4.4022e-01, -5.9944e-02,\n",
       "          -6.3243e-02,  3.6108e-01,  4.0144e-02, -2.7102e-01,  3.0692e-01,\n",
       "          -1.6929e-01,  7.1054e-02, -1.0428e-01, -6.0340e-01, -2.8575e-02,\n",
       "           9.0223e-02,  7.1587e-02, -8.0062e-02, -3.0140e-01,  5.4579e-02,\n",
       "           1.5143e-01,  8.6892e-01,  1.8659e-01, -1.5329e-01, -4.8342e-01,\n",
       "           4.8005e-01,  7.0052e-02, -2.8747e-01, -3.9927e-01,  3.1416e-01,\n",
       "          -1.8626e-01,  2.8739e-01, -6.4190e-01, -2.8369e-01, -4.6659e-01,\n",
       "           3.7846e-02,  9.7574e-02,  1.7546e-01,  1.5625e-01, -6.4597e-01,\n",
       "          -8.2407e-01,  3.2287e-01, -2.3124e-02, -8.6067e-01, -6.7556e-02,\n",
       "          -1.7390e-01,  9.1788e-02,  2.7555e-01, -3.6940e-01, -7.0410e-01,\n",
       "          -2.4999e-01,  1.2037e-01,  6.6484e-02, -1.9850e-01,  4.2256e-01,\n",
       "           3.9243e-01, -2.7410e-01, -7.7516e-02, -1.9357e-01,  1.2149e-01,\n",
       "          -3.9063e-01,  4.9112e-01, -1.7384e-01,  5.0415e-01,  1.3875e-01,\n",
       "           3.7615e-02,  3.9250e-01, -5.4878e-01, -4.0548e-01,  3.3174e-01,\n",
       "           2.5510e-01,  5.3859e-01, -8.0895e-01,  1.2816e-02,  3.1514e-01,\n",
       "          -1.3197e-01, -1.7707e-01, -2.2192e-01,  1.9425e-01,  1.6699e-01,\n",
       "          -1.7984e-02, -8.3330e-02, -4.3247e-01, -2.9574e-01, -1.6543e-01,\n",
       "          -6.9195e-01, -5.8082e-01,  1.3703e-01, -2.1977e-01,  2.6797e-01,\n",
       "          -1.3512e-01,  3.4779e-01, -1.2140e-02, -1.7509e-01, -7.8446e-02,\n",
       "           8.0217e-01,  1.7803e-01,  3.8684e-01, -4.6983e-01, -3.6110e-01,\n",
       "           3.7669e-01, -8.5725e-02, -3.2975e-01, -5.5969e-01,  6.4457e-02,\n",
       "           2.5491e-01, -2.7136e-01,  7.3124e-02, -7.8653e-02,  4.2842e-01,\n",
       "          -3.8840e-01,  6.4820e-01,  1.4484e-01,  7.2134e-02, -6.1187e-02,\n",
       "           2.0331e-01, -1.8544e+00,  6.0725e-01, -3.6162e-01, -4.1025e-02,\n",
       "          -4.1010e-01,  5.1321e-01,  9.3415e-02,  2.5189e-01,  9.4956e-03,\n",
       "          -1.1720e-01, -1.0197e-01, -1.5516e-01,  6.1468e-02, -7.1343e-02,\n",
       "           6.1913e-01,  8.4803e-01,  2.8666e-01, -6.9386e-02,  8.9387e-03,\n",
       "          -5.5901e-01, -6.3292e-02,  4.2877e-01, -5.5774e-01,  5.1061e-01,\n",
       "           2.1718e-01, -6.1685e-01, -1.3469e-01,  9.9906e-02,  2.1670e-01,\n",
       "          -1.6243e-01, -1.2952e-01, -6.9373e-01, -2.3665e-01, -2.0022e-01,\n",
       "           4.1130e-01, -1.0151e-01, -1.6121e-01,  3.2218e-01,  3.7147e-03,\n",
       "          -5.1342e-01, -1.2887e-01, -2.5112e-02, -4.0478e-01,  3.4648e-02,\n",
       "           3.4462e-01,  6.3997e-03,  8.3026e-02, -2.9837e-01, -6.0131e-02,\n",
       "           1.7599e-01, -4.5955e-01,  5.0715e-01,  5.3140e-02, -2.8734e-01,\n",
       "          -2.9007e-01, -1.1010e-01,  1.5493e-01, -2.7648e-01, -1.7760e-02,\n",
       "          -3.0385e-01, -6.9151e-01, -3.6895e-01,  2.3569e-01, -2.0671e-01,\n",
       "           1.5371e-01, -2.1338e-01,  6.9496e-01,  4.8426e-01,  5.9068e-01,\n",
       "           1.0624e-02,  2.4402e-02,  1.4580e-01, -2.3709e-01, -2.7518e-01,\n",
       "           9.8451e-02, -9.8464e-02, -1.8838e-01,  2.0932e-01, -3.0163e-01,\n",
       "          -5.1143e-02,  1.6486e-01,  2.6624e-01,  5.3079e-01,  4.2198e-02,\n",
       "          -3.4745e-01,  1.0741e-01, -2.6828e-01, -1.0029e-01, -1.4408e-03,\n",
       "          -2.3476e-01,  1.8302e-01, -6.5645e-02, -5.2210e-01, -4.2349e-01,\n",
       "          -1.1735e+00,  3.2735e-01, -1.8061e-01,  2.0324e-01, -7.8600e-01,\n",
       "          -2.5737e-01,  3.5595e-01,  1.3235e-01, -2.7339e-01, -1.0362e-01,\n",
       "           1.1996e-01,  1.8997e-01, -2.6115e-03, -3.9249e-01, -4.6319e-01,\n",
       "          -2.0424e-01,  8.1901e-02, -3.1418e-01, -7.9847e-03,  5.1835e-01,\n",
       "          -6.1223e-02,  2.9408e-01,  1.5579e-01,  1.7720e-01, -2.5392e-01,\n",
       "          -3.3543e-01,  8.8861e-02, -2.8193e-01, -1.4132e-01,  1.8438e-01,\n",
       "           1.6948e-01,  3.6807e-01,  6.7321e-01, -6.7627e-01, -6.0454e-01,\n",
       "          -1.4923e-01,  9.3722e-01, -1.8162e-02, -6.1467e-01,  4.7559e-02,\n",
       "          -1.1123e+00, -4.4161e-01, -6.2904e-03,  8.1383e-02, -3.5580e-01,\n",
       "           8.2529e-02, -2.9690e-02, -2.4470e-01, -1.5359e-01, -9.5958e-02,\n",
       "           2.9432e-01,  4.2656e-01,  4.9051e-01,  5.3402e-02,  6.5663e-03,\n",
       "          -4.3958e-01, -3.8392e-01, -3.8936e-02, -3.6456e-01, -1.6014e-01,\n",
       "           9.4097e-02,  3.1200e-01, -4.2426e-01,  1.0201e-01, -1.5972e-01,\n",
       "           5.3763e-01, -7.6060e-01, -1.5905e-01,  6.7963e-01, -2.5139e-01,\n",
       "          -2.1292e-01,  1.6935e-01, -6.9145e-01, -2.0301e-01,  1.0008e-01,\n",
       "          -2.0661e-02, -2.0487e-01,  1.0106e-01, -4.4830e-01,  8.7909e-01,\n",
       "           5.7729e-01,  2.9026e-01, -6.0838e-01,  2.4193e-01,  5.4640e-01,\n",
       "          -1.2268e-01,  1.3994e-01, -3.9152e-01, -3.7977e-01, -2.6423e-01,\n",
       "           1.6288e-01,  3.3118e-01, -4.4958e-01,  6.2040e-02, -1.1650e-01,\n",
       "           3.2411e-01, -2.8725e-01, -6.7442e-02,  4.3661e-02,  2.2568e-01,\n",
       "          -2.9599e-01, -1.8273e-01,  3.7389e-01, -6.3163e-02, -8.6979e-02,\n",
       "           1.3565e-01, -7.4249e-02,  1.5131e-01, -1.1550e-01, -6.7155e-01,\n",
       "          -2.5888e-01,  1.1687e-01,  8.3207e-01,  4.0817e-01, -6.4637e-02,\n",
       "           1.4757e-01,  2.6837e-03, -5.4951e-01,  8.5010e-02, -3.3668e-01,\n",
       "          -7.6246e-02,  2.3955e-01, -2.2258e-01,  2.7303e-01,  1.2708e-01,\n",
       "          -3.2367e-01, -1.1912e-01,  3.1075e-02,  6.3804e-01, -5.5515e-02,\n",
       "          -2.6353e-01,  3.0753e-01,  5.2133e-01, -1.7702e-01, -1.7752e-01,\n",
       "           3.4185e-01,  5.0366e-01, -2.1222e-01, -2.2527e-02, -1.3083e-02,\n",
       "           1.2723e-01,  9.7063e-02,  1.4060e-01, -3.0156e-01, -6.3698e-01,\n",
       "           4.2016e-01,  4.1585e-01,  3.0243e-01,  3.1987e-01, -1.1075e+01,\n",
       "           2.6402e-01,  5.5083e-01, -3.3032e-01, -1.6431e-01, -3.7877e-01,\n",
       "           3.6626e-01,  1.9117e-02, -5.0434e-02, -1.0789e-01, -9.2796e-02,\n",
       "          -3.1480e-01,  1.7597e-01,  7.8334e-02, -4.8162e-01,  2.3042e-01,\n",
       "          -3.4346e-01, -5.8014e-01,  1.6257e-01, -3.0014e-01,  7.5987e-02,\n",
       "          -1.1028e-01, -1.2053e-01,  9.3234e-02, -2.6230e-01, -1.9484e-01,\n",
       "           5.1784e-02, -2.8943e-01, -2.5037e-03,  1.5901e-01,  9.8368e-03,\n",
       "           2.0789e-01,  2.0218e-01,  1.7613e-01, -2.8700e-01, -3.7949e-01,\n",
       "          -3.7002e-02, -7.7929e-02,  2.2784e-01, -1.3108e-01, -3.7703e-01,\n",
       "          -2.1417e-01, -2.5680e-01,  6.2219e-01,  5.6286e-01,  6.2453e-01,\n",
       "           5.4226e-01,  3.5794e-01,  3.1441e-01, -2.2561e-01,  2.7233e-03,\n",
       "          -6.7006e-01,  2.5564e-01, -3.7015e-01,  1.4988e-01, -9.6536e-02,\n",
       "           4.3298e-01, -7.5117e-01, -1.9969e-02, -6.4571e-01, -2.8995e-01,\n",
       "           2.9827e-01,  1.3389e-01, -1.8937e-01, -6.0027e-01, -9.1235e-02,\n",
       "           3.6937e-01, -2.0585e-01,  1.8278e-01,  5.3460e-02, -3.5340e-01,\n",
       "          -2.5454e-02,  8.0660e-01, -1.8170e-01, -8.3134e-03, -1.0515e-01,\n",
       "           2.8360e-01, -1.5941e-01,  1.4767e+00, -5.7822e-02, -1.5655e-01,\n",
       "           7.7644e-02,  1.9597e-01, -3.9431e-01, -8.5143e-01, -6.3357e-01,\n",
       "          -3.0324e-01,  2.6987e-01,  1.0532e-01,  4.1609e-01, -4.0792e-01,\n",
       "          -7.5260e-01, -2.9677e-01,  1.0457e-01,  3.5611e-01,  9.8002e-01,\n",
       "          -5.3149e-01, -2.2568e-01,  1.8465e-01,  6.7174e-02, -5.8742e-01,\n",
       "           1.3636e-01, -4.9267e-01, -2.6657e-01,  2.9855e-01,  4.3406e-02,\n",
       "          -6.3079e-02,  2.9678e-01,  4.7071e-01,  2.5276e-01, -5.7991e-01,\n",
       "           2.9085e-01, -4.0187e-01,  8.5911e-03, -3.5017e-01, -4.7881e-02,\n",
       "          -2.3550e-02,  9.6569e-01, -1.1859e-01, -9.3630e-02, -2.5539e-01,\n",
       "           1.5896e-01,  1.6118e-01, -5.4551e-02, -1.4577e-01,  3.9770e-01,\n",
       "          -3.8675e-01,  4.9045e-02, -5.1378e-02, -5.8571e-02,  1.6441e-01,\n",
       "           1.8953e-01, -3.7445e-01,  8.1786e-02, -4.9538e-01, -7.6180e-01,\n",
       "          -7.8019e-01, -2.1269e-01,  1.4675e-01, -2.2356e-01, -2.1930e-01,\n",
       "          -4.3530e-01,  9.3870e-02, -2.4733e-01, -4.7006e-02, -1.3707e-01,\n",
       "           4.2336e-01,  4.2713e-01, -1.0010e-01,  2.3438e-01, -2.1446e-01,\n",
       "          -8.0445e-02,  3.2583e-02, -1.3154e-01, -3.1497e-01, -2.7671e-01,\n",
       "          -5.2373e-01, -1.2672e-01, -2.4327e-01,  1.1048e-01, -6.2738e-02,\n",
       "           3.0180e-01,  1.5072e-01, -1.2386e-01,  3.8071e-01,  3.8793e-01,\n",
       "           4.6035e-02, -3.2356e-01, -2.2749e-01, -2.0151e-01, -1.3695e-01,\n",
       "           2.1114e-01, -9.6806e-02,  9.9673e-02, -1.1505e-01, -2.7153e-01,\n",
       "          -1.0688e-01,  1.1554e+00,  8.6899e-02, -4.7659e-01,  3.1226e-01,\n",
       "           7.5998e-01, -3.7035e-01,  3.5939e-01,  2.1399e-01,  9.6751e-02,\n",
       "          -3.8212e-01, -2.5310e-01,  5.0789e-01, -1.9154e-01,  3.3013e-01,\n",
       "          -2.5993e-01,  3.0374e-01, -3.2034e-01,  4.7911e-01,  3.1191e-01,\n",
       "          -3.2152e-01, -1.4411e-01, -2.6343e-01, -2.9175e-01,  5.4151e-01,\n",
       "           1.1651e-01, -1.4960e-01,  1.8880e-01, -4.8362e-01, -1.1601e-01,\n",
       "           2.0317e-01,  9.3444e-01, -1.9270e-01,  7.9645e-02,  7.1531e-01,\n",
       "           4.4834e-01,  4.1694e-02,  7.5855e-01,  3.5950e-01,  1.3650e-01,\n",
       "           6.4710e-01,  3.1917e-01, -4.1701e-01,  2.9267e-01,  2.7159e-02,\n",
       "           7.0074e-01, -1.0259e+00, -6.4983e-01,  3.2089e-01,  1.3833e-01,\n",
       "          -2.3133e-01,  2.3971e-01, -5.2752e-01,  3.3886e-01,  1.0188e+00,\n",
       "          -9.4899e-01, -5.8372e-02, -2.4649e-01,  3.8717e-01, -3.4773e-02,\n",
       "          -1.1490e-01,  4.4677e-02, -2.2170e-01, -2.2977e-01,  2.9405e-01,\n",
       "           6.8051e-01, -8.5808e-01,  9.5243e-05,  5.4084e-01, -3.8127e-01,\n",
       "           2.8837e-02,  2.4372e-02, -9.0057e-02, -2.2438e-01,  1.5400e-01,\n",
       "           1.0404e-01,  9.5543e-02, -7.0823e-01, -4.5920e-01,  3.3985e-01,\n",
       "           6.0610e-01,  1.7077e-01,  8.8439e-02,  2.3720e-01, -1.7875e-02,\n",
       "           1.4648e-01, -5.1775e-01,  6.4462e-03,  2.7650e-01, -2.3409e-01,\n",
       "           1.9346e-01, -2.7019e-01, -7.6656e-02,  3.6097e-01, -5.8428e-01,\n",
       "           3.5668e-01,  4.2129e-01,  8.0508e-01, -4.5319e-02,  3.7277e-02,\n",
       "           4.9329e-01,  2.7265e-01, -6.7218e-03,  5.6663e-01,  1.6328e-01,\n",
       "          -2.2609e-01,  8.2648e-02,  1.8222e-01, -3.3125e-01,  4.0799e-02,\n",
       "          -4.8981e-01,  4.7307e-01,  2.7059e-01, -4.6375e-01,  1.5353e-01,\n",
       "          -1.0247e-01,  2.9916e-01, -3.6217e-01,  2.1707e-01,  2.1310e-01,\n",
       "           8.4569e-02,  6.1327e-01,  2.5972e-03,  6.0739e-01, -2.4081e-02,\n",
       "          -2.7594e-03, -4.5581e-01, -1.0912e-01,  5.0979e-02, -5.9407e-02,\n",
       "          -4.8583e-01, -2.9065e-02,  1.3636e-01, -5.3551e-02,  7.9340e-01,\n",
       "           7.9800e-01, -1.4132e-01, -9.2188e-01, -3.9866e-01,  8.1523e-02,\n",
       "           3.3660e-02,  8.6028e-02,  4.5728e-01,  2.2171e-02,  6.2237e-01,\n",
       "          -7.2397e-01, -8.2883e-02, -1.0853e-01,  7.4981e-02, -2.9833e-01,\n",
       "           8.0847e-01,  1.5056e-01,  6.6523e-01,  2.2333e-01,  3.9076e-01,\n",
       "           2.5331e-01,  7.7623e-01,  2.0236e-01, -4.6575e-01,  1.6824e-01,\n",
       "           3.3768e-01, -3.2486e-01,  5.9517e-02, -9.6471e-01,  1.5167e-01,\n",
       "           2.9492e-01,  8.9514e-01,  6.4954e-01]], device='mps:0'),\n",
       " tensor([[-1.4118e-01, -8.1866e-01, -6.6768e-01, -5.8272e-01, -7.4533e-01,\n",
       "          -1.0872e-01, -4.2529e-01,  5.4076e-01, -2.0069e-01,  6.9191e-01,\n",
       "           2.1856e-01, -2.9021e-01,  1.0845e-01, -1.0501e-01, -3.5841e-01,\n",
       "          -4.4773e-01,  3.7788e-01,  6.4187e-01,  8.0332e-01,  5.2519e-01,\n",
       "           6.3099e-01,  2.2629e-01,  7.0447e-02, -1.0115e-01, -9.6678e-01,\n",
       "           2.7320e-01, -2.4950e-01,  5.6213e-02,  4.6211e-01, -6.5372e-01,\n",
       "          -1.7625e-01,  9.5124e-02, -1.9624e-01,  1.0680e+00, -2.1779e-01,\n",
       "          -3.4961e-02,  9.7265e-02, -4.3186e-01,  2.8171e-02, -2.2504e-01,\n",
       "           5.0537e-01,  7.1985e-02,  1.5082e-01,  3.9765e-01,  2.8522e-01,\n",
       "          -1.1614e-01, -2.4742e-02, -5.1837e-01, -3.8135e-01,  6.2490e-02,\n",
       "           3.6495e-01, -2.0041e-02,  5.5445e-01,  3.9742e-01, -8.8173e-01,\n",
       "          -5.9067e-01,  1.8513e-01,  3.1285e-01, -5.1840e-01,  2.1931e-01,\n",
       "          -8.6257e-02,  6.0339e-01,  2.7296e-01,  2.9265e-01, -1.6994e-01,\n",
       "           8.3861e-02, -6.3527e-01, -7.8200e-01, -3.2520e-01, -1.4800e-02,\n",
       "          -1.3077e-02,  3.7612e-01, -1.2320e-01, -5.5198e-02,  1.8563e-01,\n",
       "          -4.3212e-01,  1.7958e-01, -2.6132e-01, -6.3146e-01, -4.4011e-01,\n",
       "           3.1155e-01, -3.1417e-02, -8.7707e-02, -2.0670e-01, -2.4692e-01,\n",
       "           1.8901e-02,  4.0735e-01,  2.6244e-01,  4.6577e-01, -8.5428e-02,\n",
       "           3.7719e-01,  1.3949e-02, -3.2830e-01, -4.6496e-01,  5.5782e-01,\n",
       "          -4.3886e-01,  3.1458e-01, -1.8966e-01, -1.6686e-01, -3.4537e-01,\n",
       "           6.2564e-02,  4.5698e-01, -4.7824e-02,  1.3155e-01, -5.4042e-01,\n",
       "          -5.5356e-01,  7.7436e-02, -8.2108e-02, -8.8666e-01, -3.6047e-02,\n",
       "           1.1830e-02, -3.9258e-01,  3.8996e-02, -7.3362e-01, -4.3803e-01,\n",
       "          -3.2792e-02, -1.4332e-01, -8.9048e-02, -2.5082e-01,  6.9120e-01,\n",
       "           3.1478e-01, -4.1761e-01,  2.4740e-01,  1.8184e-01,  1.3432e-01,\n",
       "          -3.7251e-01,  8.9812e-02,  1.0900e-01,  2.4498e-01,  5.2173e-01,\n",
       "          -5.4707e-02,  1.5244e-01, -4.2285e-02, -1.8762e-01,  1.4393e-02,\n",
       "          -2.0325e-01,  2.0935e-01, -5.6719e-01, -2.6926e-02,  1.4711e-01,\n",
       "          -1.5281e-01,  8.6822e-02, -2.6749e-01,  2.3441e-01,  1.0892e-01,\n",
       "          -1.6362e-01,  1.8785e-01, -6.9545e-02, -3.0359e-01, -2.8148e-01,\n",
       "          -5.0902e-01, -1.7342e-01,  4.7444e-01, -8.7535e-02, -1.9097e-01,\n",
       "          -1.2992e-01,  5.5769e-01,  3.6204e-02, -3.5186e-01,  1.1400e-01,\n",
       "           7.7808e-01,  1.6660e-02,  2.7360e-01, -3.0130e-01,  2.4758e-02,\n",
       "           4.2536e-01, -9.8611e-02, -2.4581e-01,  8.3814e-02,  1.1565e-01,\n",
       "           1.7171e-01, -2.8836e-01,  1.8686e-01,  3.5228e-01,  1.7245e-02,\n",
       "          -5.1560e-01,  4.7496e-01,  3.3456e-01, -1.9728e-01,  9.3937e-02,\n",
       "          -3.2681e-01, -1.7903e+00,  2.8797e-02, -1.1589e-01,  8.6640e-02,\n",
       "          -2.6816e-01,  5.5108e-01,  3.2525e-01,  1.4973e-01, -3.1566e-01,\n",
       "           1.9595e-01, -1.9999e-01,  2.6502e-01,  4.8757e-01, -7.2031e-02,\n",
       "           4.4222e-01,  5.4897e-01,  3.0902e-01,  1.8193e-03, -2.0038e-01,\n",
       "          -4.5807e-01,  1.4870e-01, -4.6683e-02,  8.3306e-01,  1.3398e-01,\n",
       "           1.8273e-01, -1.7806e-01, -1.5140e-01,  3.5061e-01,  1.1319e-01,\n",
       "           5.3394e-02, -3.7204e-01, -5.5731e-01, -2.3347e-01, -2.1546e-02,\n",
       "           6.4577e-01, -1.2358e-01, -3.7358e-01,  2.1855e-01,  3.3167e-01,\n",
       "          -2.8512e-01, -1.0783e-01, -2.3532e-01, -4.3781e-01, -3.0194e-01,\n",
       "           3.5691e-01, -4.0058e-02,  8.2864e-02, -1.5450e-01, -8.3801e-02,\n",
       "           3.4793e-01, -5.0082e-01,  6.5784e-01,  3.0088e-01, -2.0872e-02,\n",
       "          -3.2207e-01, -3.8911e-01,  1.0680e-01, -2.4616e-01,  1.8899e-01,\n",
       "          -2.3369e-01, -6.9284e-01, -5.7782e-01, -9.0146e-02,  3.6348e-01,\n",
       "           3.2497e-01, -1.7004e-01,  8.2506e-01,  4.9043e-01,  9.6295e-02,\n",
       "          -1.8057e-01, -2.8104e-01,  2.6695e-01, -2.1707e-02, -1.1640e-01,\n",
       "           2.7808e-01,  1.4639e-01, -3.4252e-01,  3.9232e-01, -6.9619e-01,\n",
       "          -1.2893e-01,  4.9038e-03,  2.3227e-02,  3.3243e-01, -3.7384e-01,\n",
       "           3.4160e-01,  2.1035e-01, -5.5189e-01, -8.1697e-02, -1.9315e-02,\n",
       "          -4.7822e-01,  6.6352e-01,  4.3595e-01, -5.7196e-01, -4.5719e-01,\n",
       "          -7.9380e-01, -6.4615e-02, -1.8747e-01, -9.5414e-03, -7.5489e-01,\n",
       "           3.0372e-02,  3.2626e-02,  1.0833e-01, -4.4854e-02,  4.0351e-02,\n",
       "           1.1551e-01,  2.5097e-01, -1.3682e-01,  2.3162e-01, -4.8715e-01,\n",
       "          -1.6390e-01,  1.2843e-01, -2.1729e-01,  3.4919e-01,  5.3390e-02,\n",
       "          -3.5024e-02, -2.8070e-01,  5.4286e-02,  1.8593e-01, -1.5187e-01,\n",
       "          -1.3481e-01, -1.9629e-01,  3.4102e-03, -2.9399e-01, -3.3748e-01,\n",
       "           1.3321e-01,  5.2086e-02,  7.3395e-01, -6.6318e-02, -6.7602e-01,\n",
       "           6.3017e-02,  7.3181e-01,  1.2679e-02,  9.9954e-02, -2.1889e-01,\n",
       "          -1.2600e+00,  2.3212e-02,  8.9271e-02, -4.7344e-02, -2.5120e-01,\n",
       "           3.9324e-01, -3.1864e-01,  2.5629e-01, -5.9862e-02, -4.7719e-01,\n",
       "           6.1584e-01,  2.6777e-01,  1.6637e-01, -2.5101e-01, -1.5642e-01,\n",
       "          -8.5578e-02, -2.6567e-02, -1.2913e-01, -5.4907e-02,  1.4252e-01,\n",
       "           1.1470e-01,  4.4051e-01,  1.1200e-01,  9.7281e-02, -1.8845e-01,\n",
       "           4.9384e-01, -1.2369e-01, -6.5141e-02,  3.1713e-01, -3.2356e-01,\n",
       "          -3.4323e-01,  5.5658e-01, -2.9561e-01, -1.3373e-01, -2.0171e-01,\n",
       "           3.5496e-01,  6.6842e-02, -3.7806e-01,  4.0306e-01,  4.2915e-01,\n",
       "           4.1471e-01,  3.3650e-01, -1.1996e-01,  7.4167e-01,  5.3443e-01,\n",
       "          -3.0667e-01,  3.6224e-01,  3.0166e-01, -2.0855e-01, -3.8233e-01,\n",
       "           1.0083e-01,  2.6424e-01, -5.8997e-01, -2.2376e-01, -1.9655e-01,\n",
       "           3.0244e-01, -6.1579e-01,  2.8523e-03,  2.4017e-01,  4.8558e-01,\n",
       "          -4.1616e-01, -3.9832e-01,  5.4117e-01, -5.3484e-01, -5.2851e-01,\n",
       "           1.2709e-01, -2.7020e-02,  6.6262e-02, -1.5637e-01, -2.0800e-01,\n",
       "          -4.0297e-01, -2.3778e-03,  7.9226e-01,  4.3455e-01, -1.0657e-01,\n",
       "          -1.7951e-01, -3.5392e-02, -3.3086e-01,  1.1435e-01, -7.5484e-02,\n",
       "          -9.2757e-02,  8.5425e-02,  5.9885e-02, -1.2980e-02,  2.5307e-02,\n",
       "          -2.9873e-01, -1.4998e-01, -8.8520e-03, -1.3534e-01, -1.6629e-01,\n",
       "          -3.2118e-01,  7.2667e-02,  3.8770e-01, -1.2099e-01, -1.4338e-01,\n",
       "           2.4918e-01,  2.2725e-01, -2.7903e-01,  4.9473e-01,  2.2756e-01,\n",
       "           2.9600e-01, -3.9611e-01,  1.4021e-02, -3.4656e-01, -7.7216e-01,\n",
       "           7.4803e-01,  5.0223e-01,  4.8307e-01,  4.1463e-01, -1.1142e+01,\n",
       "          -7.4926e-02,  5.6245e-01, -6.0574e-01, -2.4055e-01,  3.9995e-02,\n",
       "           7.4353e-02, -4.5752e-01, -8.1501e-02, -1.5328e-01, -3.5183e-01,\n",
       "          -4.5180e-02, -1.7125e-01, -5.9471e-03, -1.0367e-01, -2.0764e-01,\n",
       "          -3.4308e-01, -7.2023e-01,  3.0390e-01,  4.4333e-02,  4.6822e-01,\n",
       "          -8.2627e-02,  1.9246e-01,  6.9727e-02, -7.8803e-02, -4.4940e-01,\n",
       "           3.1124e-01, -2.2892e-01,  1.1979e-01,  3.3936e-01, -1.1533e-01,\n",
       "           3.6332e-01,  1.3334e-01,  1.1599e-01, -1.8327e-01, -4.0537e-01,\n",
       "          -9.9170e-02, -8.4306e-02,  1.9789e-01,  6.6646e-02, -3.0922e-01,\n",
       "          -4.2498e-01, -2.6404e-01,  2.9661e-01,  8.1614e-02,  9.3388e-01,\n",
       "           1.1278e+00,  5.8402e-01,  2.4676e-01,  5.9932e-02,  2.4197e-02,\n",
       "          -8.3963e-01,  4.2992e-01, -8.4381e-01,  6.6087e-02,  3.4621e-02,\n",
       "           6.8160e-01, -7.9866e-01,  1.1168e-01, -3.3093e-02, -3.8494e-01,\n",
       "           4.7472e-01,  1.7768e-01,  3.0235e-01, -1.4484e-01, -4.3288e-01,\n",
       "           5.5576e-01,  2.8336e-01,  2.1888e-01, -1.8637e-01,  1.4173e-01,\n",
       "          -4.4906e-01,  4.9090e-01,  4.1943e-01, -3.9117e-01,  2.2552e-02,\n",
       "           6.9365e-02,  4.6035e-02,  9.6285e-01,  8.2094e-02, -1.2289e-01,\n",
       "           2.9154e-01,  5.4841e-01, -3.3728e-01, -9.9225e-01,  1.6876e-01,\n",
       "           2.2088e-01,  1.5360e-01,  1.7151e-01,  6.7610e-01, -7.9779e-02,\n",
       "          -7.1568e-01, -1.0876e-01, -8.5508e-02,  2.8875e-01,  2.3050e-01,\n",
       "          -4.3793e-01, -1.3268e-02,  4.0889e-01,  6.3000e-02, -5.0623e-01,\n",
       "          -1.7621e-01, -1.0381e-01, -1.7020e-01,  4.4362e-01,  5.6214e-02,\n",
       "          -1.2821e-03,  3.2447e-02,  1.1834e-01,  2.3824e-01, -5.7673e-01,\n",
       "           3.5964e-01, -1.6218e-01, -2.0342e-01, -1.3661e-01, -4.2474e-01,\n",
       "           7.2598e-02,  3.3097e-01, -2.6179e-01, -3.2545e-01, -2.9021e-01,\n",
       "           4.5414e-02, -1.1004e-01,  4.9933e-02, -4.2918e-01,  5.7254e-01,\n",
       "          -6.0805e-01,  2.2872e-01,  1.6300e-01,  2.1273e-02,  2.1783e-01,\n",
       "          -5.9678e-02,  8.5997e-01,  7.6284e-01, -3.4141e-01, -1.0826e+00,\n",
       "          -6.6888e-01,  8.8948e-02, -9.8412e-02, -2.5053e-01,  8.6729e-02,\n",
       "           1.6330e-01,  1.8220e-01,  9.2999e-02,  7.9844e-02,  2.3901e-02,\n",
       "           4.2125e-01,  5.6852e-01, -5.7824e-02, -2.7052e-01, -3.8393e-01,\n",
       "          -4.6945e-01, -1.1722e-01,  4.9212e-03,  1.1485e-01, -3.2376e-01,\n",
       "          -6.4865e-02, -9.0373e-02, -3.2354e-01,  1.5722e-01, -3.4709e-02,\n",
       "          -1.1456e-01, -2.7543e-01,  1.6215e-01, -5.3730e-02,  3.2402e-01,\n",
       "           3.2326e-01, -4.1004e-01, -5.5539e-01, -9.1584e-02,  2.0239e-01,\n",
       "           6.5420e-02,  1.1355e-01, -1.0432e-01, -4.1652e-01, -9.5264e-02,\n",
       "           5.9585e-02,  1.0428e+00, -2.2224e-01, -5.9878e-01,  2.6695e-01,\n",
       "           5.9710e-01, -1.7186e-01, -2.3105e-01, -3.0377e-01,  1.1350e-01,\n",
       "          -1.4897e-01, -6.3624e-01,  5.7686e-01,  2.2071e-01,  4.5051e-01,\n",
       "           6.3864e-01, -3.1602e-02, -3.9243e-01,  4.3123e-01, -6.7434e-01,\n",
       "          -3.4723e-01, -1.7286e-01, -1.7582e-01,  1.4988e-03,  8.2203e-03,\n",
       "          -2.0154e-01, -3.2555e-02,  3.3242e-02,  2.3875e-02,  2.8456e-01,\n",
       "           9.8807e-02,  1.4234e+00, -2.0718e-01, -2.6344e-01,  6.3536e-01,\n",
       "           4.4016e-01, -2.2223e-01,  7.9798e-02,  3.0405e-01,  2.0940e-01,\n",
       "           4.3632e-01, -1.4323e-01, -1.9700e-01,  1.8931e-01,  1.5960e-01,\n",
       "           6.0947e-01, -8.0504e-01, -3.3671e-01, -6.6813e-03,  4.6907e-01,\n",
       "           1.5955e-01,  1.1589e-01, -4.1848e-01,  1.9966e-01,  5.6717e-01,\n",
       "          -1.0005e+00, -9.1059e-02, -3.1611e-01,  5.6642e-01,  1.6407e-01,\n",
       "          -2.1739e-01, -7.3597e-02, -4.1403e-01, -2.7418e-01,  3.2445e-01,\n",
       "           3.1302e-01, -1.0470e+00,  3.0078e-01,  4.2546e-01, -5.2656e-01,\n",
       "           1.7092e-01, -2.4094e-01,  4.2164e-01,  2.7871e-02,  2.5469e-01,\n",
       "           7.2777e-02,  3.0160e-01, -1.5313e-01, -5.6084e-01,  8.6770e-02,\n",
       "           6.2224e-01, -4.0171e-02, -2.6454e-01,  4.0724e-02,  1.7978e-01,\n",
       "          -9.7563e-02, -5.6400e-01, -2.3658e-01, -7.0725e-02, -2.9230e-01,\n",
       "          -1.0324e-01, -3.2389e-01,  1.3957e-01,  2.0752e-01, -3.1946e-01,\n",
       "           5.7227e-01,  7.1750e-01,  7.3615e-01, -3.5994e-01, -5.3154e-02,\n",
       "           1.1900e-01,  1.1680e-01, -1.5827e-01,  4.0103e-01, -1.3437e-01,\n",
       "          -5.2635e-01,  4.6891e-01,  1.2511e-01, -7.2186e-02,  6.6215e-02,\n",
       "          -2.8752e-01,  1.1836e-01,  2.1279e-01, -3.0245e-01,  3.0853e-01,\n",
       "          -5.0685e-02, -6.8187e-02, -2.8807e-01,  2.0488e-01,  1.9952e-01,\n",
       "          -6.1963e-01,  9.8433e-01, -2.2187e-01,  8.0780e-01, -2.5095e-01,\n",
       "           4.1225e-01, -9.3567e-01, -6.8732e-01,  3.7883e-02, -2.3009e-01,\n",
       "          -4.7944e-01,  5.5653e-02,  1.6975e-01,  3.1872e-01,  4.2902e-01,\n",
       "           3.3244e-01, -1.9091e-01, -7.9585e-01, -4.7994e-01,  2.5034e-01,\n",
       "           4.2626e-02, -1.9667e-01,  1.1235e-01, -1.8347e-01,  1.7011e-01,\n",
       "          -6.4038e-01, -7.0414e-01, -1.3937e-01,  1.8303e-01, -3.0078e-01,\n",
       "           8.5164e-01, -1.6648e-01, -1.7843e-01,  3.9779e-01,  1.1668e-01,\n",
       "           4.2171e-01,  5.2192e-01, -3.4142e-02, -6.1750e-01, -1.5650e-01,\n",
       "          -1.1124e-01, -2.3702e-01, -7.2387e-02, -3.4932e-01,  7.1984e-01,\n",
       "           3.0178e-01, -2.1098e-02,  7.0361e-01]], device='mps:0'),\n",
       " tensor([[-0.2612, -0.6486, -0.3036,  ...,  0.4929,  0.9962,  0.8028],\n",
       "         [-0.0932, -0.6293, -0.4879,  ...,  0.3041, -0.1958,  0.5576],\n",
       "         [-0.3187, -0.5736, -0.4291,  ...,  0.1316,  0.7618,  0.5839],\n",
       "         [-0.0626, -0.6265, -0.7028,  ...,  0.1217,  0.4033,  0.6177]],\n",
       "        device='mps:0'),\n",
       " tensor([[-0.1230, -0.2743, -0.7632,  ...,  0.0595,  0.8762,  0.7366],\n",
       "         [-0.2143, -0.4630, -0.6023,  ...,  0.1281,  0.4524,  0.6160]],\n",
       "        device='mps:0'),\n",
       " tensor([[-0.4654, -0.4115, -0.4370,  ...,  0.1625,  0.4338,  0.5908],\n",
       "         [-0.2894, -0.5557, -0.6093,  ...,  0.4140, -0.0586,  0.6268]],\n",
       "        device='mps:0'),\n",
       " tensor([[-0.2744, -0.4852, -0.4880,  ...,  0.2064,  0.2977,  0.6606],\n",
       "         [ 0.0506, -0.6230, -0.6446,  ..., -0.0875,  0.4018,  0.3101],\n",
       "         [-0.4643, -0.3750, -0.2078,  ...,  0.2835,  0.5093,  0.4110]],\n",
       "        device='mps:0'),\n",
       " tensor([[-0.2925, -0.5382, -0.3957,  ...,  0.2774,  1.0308,  0.5077],\n",
       "         [-0.4615, -0.5586, -0.3419,  ...,  0.2423,  0.2570,  0.4881],\n",
       "         [-0.1649, -0.5331, -0.2723,  ...,  0.1723,  0.7889,  0.2604]],\n",
       "        device='mps:0'),\n",
       " tensor([[-0.1705, -0.4939, -0.5141,  ...,  0.1882,  0.5383,  0.4465],\n",
       "         [-0.3627, -0.6522, -0.5099,  ..., -0.0153,  0.7033,  0.4959]],\n",
       "        device='mps:0'),\n",
       " tensor([[-0.4654, -0.4115, -0.4370,  ...,  0.1625,  0.4338,  0.5908],\n",
       "         [-0.0932, -0.6293, -0.4879,  ...,  0.3041, -0.1958,  0.5576],\n",
       "         [-0.3292, -0.2590, -0.5016,  ..., -0.0148,  0.3509,  0.1607],\n",
       "         [-0.1979, -0.6462, -0.8826,  ...,  0.1613,  1.0377,  0.5252],\n",
       "         [-0.2330, -0.4411, -0.4565,  ...,  0.0257,  0.5397,  0.4576],\n",
       "         [-0.0238, -0.5721, -0.8340,  ...,  0.0483,  0.8204,  0.5286]],\n",
       "        device='mps:0'),\n",
       " tensor([[-0.2834, -0.5749, -0.7048,  ...,  0.2683,  0.0067,  0.6095],\n",
       "         [-0.0791, -0.4449, -0.8793,  ..., -0.0130,  0.9666,  0.4780]],\n",
       "        device='mps:0'),\n",
       " tensor([[-0.2744, -0.4852, -0.4880,  ...,  0.2064,  0.2977,  0.6606],\n",
       "         [ 0.0506, -0.6230, -0.6446,  ..., -0.0875,  0.4018,  0.3101],\n",
       "         [-0.1156, -0.5318, -0.5096,  ...,  0.0343,  1.0385,  0.3071]],\n",
       "        device='mps:0'),\n",
       " tensor([[-5.0499e-01, -8.2118e-01, -7.6476e-01, -9.9111e-01, -2.2208e-01,\n",
       "          -3.7576e-01, -8.3059e-01,  7.2052e-01, -1.4017e-01,  3.5809e-01,\n",
       "           3.0761e-01, -6.7034e-02,  1.8358e-01, -3.7617e-01, -7.7348e-01,\n",
       "           1.0369e-01,  2.6740e-01,  3.4853e-01,  1.9143e-01,  1.3776e-01,\n",
       "           1.0954e+00,  2.5168e-01, -2.1948e-01, -1.9226e-02, -2.1005e-01,\n",
       "           3.3398e-01, -2.5711e-01, -2.1323e-02,  4.8683e-01, -5.1680e-01,\n",
       "          -3.5026e-01,  2.7639e-01, -2.3706e-01,  9.0062e-01, -3.1924e-01,\n",
       "          -2.0994e-01,  2.9109e-02, -3.9039e-01,  1.0164e+00, -3.7797e-01,\n",
       "           3.2967e-01,  1.1019e-01, -9.2718e-02,  4.5330e-01, -1.7620e-01,\n",
       "           2.0263e-01,  3.5313e-02, -5.8204e-01,  3.7849e-01, -2.0688e-01,\n",
       "           5.3304e-02, -3.8936e-02,  1.9863e-01,  1.5987e-01, -4.0145e-01,\n",
       "          -1.7207e-01,  1.4759e-01,  1.2515e-01, -2.9789e-01,  3.3777e-01,\n",
       "           2.4530e-01,  6.2695e-01,  6.1321e-01,  1.5577e-01, -8.1977e-02,\n",
       "           1.1718e-01, -4.2723e-01, -3.6440e-01, -3.1637e-01, -2.9041e-02,\n",
       "          -1.0737e-01,  2.1853e-01,  2.5210e-01, -6.6237e-02, -5.4993e-02,\n",
       "          -1.2783e-01,  3.5607e-01, -3.9984e-01, -2.1136e-01, -3.2119e-01,\n",
       "           2.0123e-01, -6.5197e-02,  7.1507e-02,  8.9401e-02, -2.8954e-01,\n",
       "           5.3259e-02,  4.0642e-01, -6.2177e-03,  4.0600e-01,  2.3899e-01,\n",
       "           6.1420e-01, -1.1346e-03, -1.9578e-01, -5.5979e-01,  4.8135e-01,\n",
       "          -2.9909e-01,  1.6371e-01,  1.5952e-02, -2.0245e-01, -1.9607e-01,\n",
       "          -1.4520e-01,  1.0480e-01,  2.7178e-01,  2.2719e-01, -4.0203e-01,\n",
       "          -2.4808e-01,  3.9359e-02, -1.7717e-02, -2.5256e-01, -1.9136e-01,\n",
       "          -2.3922e-01, -8.2918e-01,  9.9130e-02, -2.0802e-01, -1.6772e-01,\n",
       "           6.0337e-02, -1.6807e-02, -2.8291e-01, -1.4691e-01,  9.1852e-01,\n",
       "           3.4002e-01, -4.2743e-01, -4.4005e-02,  1.8030e-01,  3.6192e-01,\n",
       "          -1.4272e-01,  1.0279e-01,  6.7054e-02,  1.4259e-01,  6.4850e-03,\n",
       "          -3.9632e-03,  1.5072e-02, -3.3513e-01, -5.3438e-02,  8.4307e-02,\n",
       "          -2.1348e-01, -1.5182e-01, -4.5125e-01, -2.6114e-01, -5.1525e-02,\n",
       "           4.2808e-02, -5.8480e-02, -1.4795e-01,  2.1627e-01,  9.6090e-02,\n",
       "          -2.2532e-01, -3.4083e-02, -8.2514e-02, -1.1592e-01, -5.7377e-01,\n",
       "          -7.5406e-01, -2.7138e-01,  8.9766e-02,  3.1435e-03, -2.4122e-01,\n",
       "          -1.0017e-01,  3.5841e-01,  6.8310e-02, -1.0358e-01,  1.2201e-01,\n",
       "           7.3328e-01, -2.3584e-02,  2.4787e-01, -2.4832e-01,  2.7459e-01,\n",
       "           4.9058e-02, -3.1684e-01, -2.3513e-01,  1.5011e-01,  2.2946e-01,\n",
       "           3.2071e-01,  3.5543e-02, -1.3977e-01, -3.6370e-01, -1.4976e-01,\n",
       "          -2.1753e-01,  4.5554e-01,  2.1842e-01, -1.0398e-01, -9.2348e-02,\n",
       "           1.6418e-01, -2.2758e+00, -9.3984e-02, -5.2606e-01, -1.9985e-01,\n",
       "          -9.6371e-02,  2.0678e-01,  3.1453e-01,  4.2260e-01,  7.8929e-02,\n",
       "          -1.5924e-01,  4.0727e-02,  2.3542e-01,  2.5893e-01, -1.3442e-01,\n",
       "           3.2944e-01,  3.6709e-01,  1.7511e-01, -1.6218e-01, -1.1249e-01,\n",
       "          -4.6750e-01,  2.6147e-01, -2.2091e-01,  2.2022e-01, -1.9018e-01,\n",
       "           1.3925e-01, -2.2811e-01,  1.2006e-01,  3.1067e-01,  5.5013e-02,\n",
       "          -2.9813e-01, -7.8955e-02, -1.6280e-01, -9.2514e-02, -4.5145e-01,\n",
       "           3.7726e-01, -5.1478e-01, -1.6254e-01,  3.0691e-01,  3.5791e-01,\n",
       "           6.8748e-03, -1.9329e-01, -1.5428e-01, -1.1209e-01, -2.3499e-01,\n",
       "           4.9073e-01, -1.0652e-01, -3.5901e-02, -2.0743e-01, -3.1444e-01,\n",
       "          -3.3275e-01, -5.0506e-01,  2.2455e-02,  2.8326e-01,  1.4413e-01,\n",
       "          -6.3720e-02, -3.4474e-01,  4.2638e-01, -1.4870e-01,  3.6848e-01,\n",
       "          -7.3186e-02, -3.2688e-01, -5.9034e-01, -3.2002e-02, -1.5678e-01,\n",
       "           1.8389e-01,  5.3850e-02,  7.7393e-01,  7.5819e-01,  2.6452e-01,\n",
       "          -1.8290e-01, -2.1666e-01,  5.6468e-01,  3.3020e-01, -4.0355e-01,\n",
       "          -9.7584e-02,  9.9611e-02, -5.2543e-01,  1.6918e-01, -5.4871e-01,\n",
       "          -1.9601e-01,  1.2074e-02, -7.8768e-02,  7.7149e-01, -2.3055e-01,\n",
       "           2.5854e-01,  3.1322e-01,  5.0345e-02, -3.6707e-02,  3.3077e-01,\n",
       "          -5.7419e-01,  4.3544e-01, -6.9989e-01, -6.9566e-02, -7.2799e-01,\n",
       "          -1.7726e-01,  2.8663e-01,  1.0227e-01,  8.1414e-02, -1.8401e-01,\n",
       "          -7.2238e-02, -4.2614e-01, -2.0355e-02,  1.7551e-03, -1.1307e-01,\n",
       "           4.8675e-01, -9.7687e-02,  4.4365e-03, -9.7756e-02, -4.0924e-01,\n",
       "          -7.1178e-01, -1.1796e-01, -2.1478e-01,  2.5580e-02,  2.3718e-01,\n",
       "          -1.4455e-01,  2.1396e-01,  2.2999e-01, -1.0994e-01, -3.0278e-01,\n",
       "           9.1120e-02,  2.6824e-01, -3.6815e-01,  1.6179e-02,  1.4575e-01,\n",
       "           4.6853e-01,  3.0262e-02,  4.7232e-01, -3.5807e-01, -5.4407e-01,\n",
       "           1.2575e-01,  5.5043e-01, -1.0481e-01, -5.3063e-02, -3.6354e-01,\n",
       "          -1.3162e+00, -1.3949e-01, -1.2235e-01, -3.4999e-01, -6.1979e-01,\n",
       "           2.8955e-01,  5.6526e-02, -1.0247e-01, -1.6665e-01, -1.4275e-01,\n",
       "           5.7253e-01,  4.5124e-01,  1.7847e-01, -7.8841e-03, -3.8598e-01,\n",
       "          -1.2221e-01, -6.7433e-02, -1.1928e-01, -2.4118e-01, -2.7369e-01,\n",
       "           1.2587e-01,  2.4923e-01, -1.1084e-01,  2.1207e-01, -2.4417e-01,\n",
       "           8.2663e-01, -1.5478e-01, -7.2327e-02,  1.8439e-01, -6.8310e-03,\n",
       "           8.2059e-03,  5.2645e-01, -3.1181e-01,  4.4534e-01,  2.3651e-02,\n",
       "           3.6965e-01, -3.5857e-01, -2.2888e-01,  1.0166e-01,  2.1807e-01,\n",
       "           2.8014e-01,  4.6191e-01, -1.8992e-02,  4.7233e-02,  4.1863e-01,\n",
       "          -5.1255e-02,  2.8384e-01, -3.0820e-01, -1.3203e-01,  6.8816e-02,\n",
       "           1.5021e-01,  4.6049e-01, -3.3287e-01,  1.2259e-01,  8.8817e-02,\n",
       "           8.1573e-03, -5.8031e-01,  1.1577e-01,  1.9187e-01, -3.3888e-01,\n",
       "          -2.4153e-01, -4.3900e-01,  1.0707e-01, -2.7217e-01, -4.7575e-01,\n",
       "           1.3247e-02, -1.3627e-01, -1.2613e-01, -3.6432e-01, -4.2053e-02,\n",
       "          -1.5487e-01,  2.5240e-02,  5.2792e-01,  3.7057e-01,  2.5148e-01,\n",
       "           2.7848e-02, -5.9623e-02, -8.2294e-02,  3.2609e-01, -4.5340e-01,\n",
       "          -3.4224e-01,  1.5613e-01, -1.8602e-01, -1.4289e-01,  1.9604e-01,\n",
       "          -4.2570e-01, -1.7430e-01,  2.9003e-01, -2.8501e-01, -1.1644e-01,\n",
       "          -2.9964e-01,  3.4197e-02,  3.5784e-01, -3.6475e-01, -1.1161e-01,\n",
       "           6.5399e-02,  1.1129e-01,  4.2631e-02,  3.7924e-02,  1.9837e-01,\n",
       "           2.3634e-01, -2.6113e-01,  4.1472e-02,  6.0414e-02, -5.8610e-01,\n",
       "           4.5357e-01, -2.2471e-02,  4.0601e-01,  4.5555e-01, -1.1677e+01,\n",
       "           2.4060e-02,  8.2959e-01, -5.8000e-01, -1.0574e-01, -2.0170e-02,\n",
       "           1.4278e-01, -2.6154e-01, -3.8369e-02, -5.9455e-02, -2.1355e-01,\n",
       "          -2.8797e-01,  1.7472e-01, -2.6190e-01, -5.1078e-01, -1.0883e-01,\n",
       "          -1.0900e-01, -3.8206e-01,  2.2803e-02, -1.9187e-02,  5.0204e-01,\n",
       "          -4.4898e-01, -2.4423e-02,  3.2469e-01, -5.1136e-01, -4.9156e-01,\n",
       "          -1.9703e-01, -1.5075e-01,  1.6015e-01,  1.5421e-01, -1.7222e-01,\n",
       "           1.3746e-01, -7.5832e-02,  7.5240e-01, -1.6071e-01, -6.6729e-01,\n",
       "          -3.0539e-02, -7.3103e-02,  6.2437e-01, -2.0149e-01, -3.7953e-02,\n",
       "          -8.9020e-02, -2.7983e-01,  3.4676e-01,  3.8771e-01,  6.3784e-01,\n",
       "           1.3177e+00,  5.8790e-01,  8.3876e-02, -1.3047e-02, -2.0939e-01,\n",
       "          -3.7902e-01,  5.3067e-01, -5.6746e-01,  2.0000e-01, -2.0569e-01,\n",
       "           3.4175e-01, -2.6147e-01,  1.6235e-01, -2.1068e-01, -7.8708e-02,\n",
       "           2.3301e-02,  1.9356e-01, -8.1988e-02, -1.7377e-01, -4.7459e-01,\n",
       "           5.3907e-01,  1.9831e-01,  9.5512e-02, -1.9663e-01,  2.8935e-01,\n",
       "          -2.4737e-01,  7.7494e-01, -1.8325e-01, -1.2487e-01,  1.3500e-01,\n",
       "           2.4646e-02,  3.2845e-01,  8.0136e-01, -1.9316e-01,  1.7048e-01,\n",
       "           2.9851e-01, -6.6635e-02, -1.3317e-01, -9.7612e-01,  1.5381e-01,\n",
       "           1.2703e-01,  1.6142e-01,  3.0555e-01,  3.2260e-01, -3.5256e-01,\n",
       "          -2.6958e-01, -3.9930e-01, -4.5577e-01,  2.7216e-01,  6.9771e-01,\n",
       "          -6.3930e-01,  1.2298e-01, -1.0115e-01,  2.0395e-01, -4.2934e-01,\n",
       "          -1.8940e-02, -4.4405e-02, -2.5874e-01,  3.0569e-01,  4.8620e-02,\n",
       "          -1.7500e-01,  2.6737e-01,  1.9949e-01,  4.8625e-01, -1.6580e-01,\n",
       "           8.3688e-02, -6.3539e-02,  1.2593e-01, -4.1402e-01,  2.2518e-01,\n",
       "          -8.6732e-02,  2.5005e-01, -1.6151e-01, -4.4646e-01,  1.2838e-01,\n",
       "          -4.7393e-01,  3.8960e-01,  5.1097e-02, -3.9022e-01,  5.7822e-01,\n",
       "          -6.8153e-01,  1.4112e-02,  1.2359e-01, -2.2274e-01,  4.7807e-01,\n",
       "           2.7588e-02, -9.7603e-02,  4.0834e-01, -2.6970e-01, -5.9091e-01,\n",
       "          -3.8007e-01, -9.5179e-02,  3.4895e-01, -5.7683e-01,  2.2027e-04,\n",
       "          -2.4414e-01,  1.8949e-01,  8.2815e-03,  2.9267e-01,  3.6721e-01,\n",
       "           2.1274e-01,  5.6545e-01, -3.8355e-01, -9.4197e-03, -2.0765e-01,\n",
       "          -4.2716e-01,  3.0035e-02, -8.0637e-02,  1.5717e-01, -3.7136e-01,\n",
       "           1.2824e-01, -1.1185e-01, -4.8148e-01,  4.0851e-01, -1.9760e-01,\n",
       "           1.7425e-02, -1.2714e-01,  3.2671e-03,  1.5392e-02,  4.7235e-01,\n",
       "           9.3373e-02, -2.0066e-01, -5.4365e-01,  2.3947e-01,  3.2468e-01,\n",
       "          -7.0077e-02, -4.5814e-02, -2.4914e-01, -4.9515e-01, -4.4614e-01,\n",
       "           6.8583e-03,  8.1458e-01, -5.1925e-01, -3.5888e-01,  2.8117e-03,\n",
       "           2.1593e-01, -5.6665e-01, -2.4153e-01, -6.5445e-02,  2.5790e-01,\n",
       "          -4.7191e-02, -5.8328e-02,  5.0681e-01,  1.6928e-01,  1.7456e-01,\n",
       "           2.8192e-01,  1.9917e-01,  6.6443e-02, -4.3980e-02, -5.2623e-01,\n",
       "          -5.8723e-01,  2.1584e-01, -1.1913e-01,  3.6668e-01, -3.6422e-02,\n",
       "           2.3185e-02,  2.4575e-01,  1.9591e-02, -2.2886e-01,  4.9226e-01,\n",
       "           1.8390e-02,  1.4175e+00, -3.7917e-01, -4.8482e-02,  3.7596e-01,\n",
       "           3.2168e-01,  3.7958e-01,  2.6174e-01, -7.1994e-02,  1.3641e-02,\n",
       "           3.4322e-01, -3.9770e-01, -3.0779e-01,  1.4465e-01,  2.7631e-01,\n",
       "           4.8218e-01, -6.2836e-01, -5.1703e-01, -9.4442e-02,  6.3327e-01,\n",
       "           2.7487e-01,  2.4536e-01, -3.4829e-01,  2.8105e-01,  3.5931e-01,\n",
       "          -5.4584e-01,  3.3140e-02,  3.2824e-01,  3.4182e-01, -3.8515e-01,\n",
       "          -9.3333e-02, -2.0818e-01, -3.8104e-01, -4.1073e-01,  4.9265e-01,\n",
       "           5.8374e-01, -7.5236e-01,  5.4250e-01,  3.5005e-01, -3.4503e-01,\n",
       "           5.2865e-01,  2.4811e-01,  1.9624e-01,  2.5359e-01,  2.9997e-01,\n",
       "           2.7055e-01, -1.2446e-02, -7.5242e-01, -5.8186e-01,  2.8154e-01,\n",
       "           4.6194e-01,  2.7620e-01,  5.2315e-02,  1.8080e-01, -2.9711e-01,\n",
       "          -1.5962e-01, -2.6482e-02, -4.0249e-01, -3.5321e-01,  1.2878e-01,\n",
       "          -7.1990e-03, -2.3516e-01,  6.7248e-02,  3.5380e-01, -4.6050e-01,\n",
       "           4.1108e-01,  6.3458e-01,  5.0468e-01,  5.6329e-02, -2.4788e-01,\n",
       "           1.8025e-01,  2.8834e-01, -3.5066e-01,  6.1979e-01,  1.5862e-01,\n",
       "          -3.4737e-02, -1.9491e-01,  4.4387e-02, -3.5646e-01,  1.5167e-01,\n",
       "          -4.3336e-01,  8.4305e-02, -2.6107e-01, -3.0157e-01,  2.6187e-01,\n",
       "           1.4201e-01,  2.6568e-01, -2.8816e-01,  2.3580e-01,  3.4657e-01,\n",
       "          -1.7782e-01,  2.9806e-02, -8.2361e-02,  6.6998e-01, -1.1944e-01,\n",
       "           1.7894e-01, -5.2566e-01, -2.1699e-01, -1.4722e-02,  3.3346e-01,\n",
       "          -4.0042e-01, -4.7312e-01,  1.3881e-01, -4.4104e-02,  7.4059e-01,\n",
       "           5.9152e-01,  7.9884e-02, -4.4644e-01, -4.1708e-01, -4.8653e-03,\n",
       "           1.9330e-01,  7.9955e-02,  1.3001e-01, -4.2598e-01, -1.7623e-01,\n",
       "          -4.8887e-01,  6.2351e-02,  5.1720e-03, -9.0473e-03, -2.1174e-01,\n",
       "           6.7404e-01, -9.3255e-02, -1.7077e-02,  3.1175e-01,  1.2898e-01,\n",
       "           4.7163e-01,  7.0009e-01,  2.1411e-01, -3.0927e-01, -2.6054e-01,\n",
       "           1.1847e-01, -7.0442e-02, -1.1602e-01, -5.3186e-01,  3.5968e-01,\n",
       "           3.3682e-01,  2.1570e-01,  4.5981e-01]], device='mps:0'),\n",
       " tensor([[-0.0111, -0.5057, -0.4670,  ...,  0.2365, -0.0712,  0.4388],\n",
       "         [-0.3034, -0.2152, -0.5753,  ...,  0.3475,  0.5612,  0.4483],\n",
       "         [-0.1136, -0.6122, -0.7569,  ...,  0.3204,  0.5077,  0.6029],\n",
       "         [-0.2225, -0.2765, -0.5902,  ...,  0.2728,  0.4876,  0.6150]],\n",
       "        device='mps:0'),\n",
       " tensor([[-2.2581e-01, -7.0992e-01, -3.7782e-01,  ...,  3.2917e-01,\n",
       "           7.0594e-01,  7.4939e-01],\n",
       "         [-4.2358e-01, -3.3872e-01, -7.8313e-01,  ...,  2.4723e-01,\n",
       "          -4.3970e-02,  2.5120e-01],\n",
       "         [-5.8045e-01, -6.2733e-01, -6.9340e-01,  ..., -2.0918e-04,\n",
       "           1.9936e-01,  1.3886e-01]], device='mps:0'),\n",
       " tensor([[-0.4057, -0.4476, -0.6056,  ...,  0.1203,  0.3041,  0.4450],\n",
       "         [-0.3150, -0.6909, -0.7037,  ...,  0.2289,  0.5866,  0.3764],\n",
       "         [-0.0543, -0.3578, -0.3144,  ..., -0.0321,  0.1085,  0.2146]],\n",
       "        device='mps:0'),\n",
       " tensor([[-0.1959, -0.3163, -0.6089,  ..., -0.2600,  0.1989,  0.5143],\n",
       "         [-0.1682, -0.3560, -0.3949,  ..., -0.0628,  0.8556,  0.6643],\n",
       "         [-0.2795, -0.5983, -0.5215,  ..., -0.1949,  0.3647,  0.4091],\n",
       "         [-0.0915, -0.5960, -0.2936,  ..., -0.0535,  0.3212,  0.5493]],\n",
       "        device='mps:0'),\n",
       " tensor([[-4.6085e-01, -5.6598e-01, -4.7905e-01, -8.5908e-01, -1.9677e-01,\n",
       "          -2.7844e-01, -3.9924e-01,  5.4953e-01, -3.0268e-02,  7.4797e-01,\n",
       "           2.1281e-01,  3.9587e-01,  2.1136e-01,  2.5003e-01, -4.3589e-01,\n",
       "          -3.7419e-01, -6.1137e-02,  3.9874e-01,  5.9988e-01,  1.9960e-01,\n",
       "           4.5840e-01,  6.3510e-01,  3.0337e-02,  2.3858e-02, -5.6276e-01,\n",
       "           2.1006e-01, -4.4613e-02, -1.6415e-01, -4.9882e-03, -4.0270e-01,\n",
       "          -4.5716e-02,  4.0370e-01, -3.2697e-02,  1.3700e+00, -3.2375e-01,\n",
       "          -9.2441e-02,  2.2424e-01, -6.5136e-01, -1.1246e-01, -1.9557e-01,\n",
       "           5.8739e-01, -5.3447e-01,  5.1574e-01,  2.2677e-01, -8.9916e-02,\n",
       "           1.2738e-01,  1.2672e-01, -6.0578e-01,  3.5202e-01, -8.1745e-01,\n",
       "          -2.1870e-01, -3.3698e-01,  2.9759e-01,  4.3902e-01, -7.1074e-01,\n",
       "          -7.4847e-01,  3.4802e-01,  8.2786e-02, -1.7442e-01, -1.0195e-01,\n",
       "           4.7052e-02,  2.5087e-01, -2.6245e-01,  1.1440e-01,  2.1786e-01,\n",
       "           5.4505e-02, -4.4809e-01, -5.2473e-01, -4.0794e-01, -5.7997e-02,\n",
       "          -1.6587e-01, -1.4832e-02,  9.8164e-03, -3.0879e-01,  1.3333e-01,\n",
       "          -1.6956e-01,  4.1027e-01, -1.0822e-02, -8.3891e-02,  2.5998e-01,\n",
       "          -5.2630e-02,  2.1468e-01,  1.2133e-01, -5.4095e-02,  8.8400e-02,\n",
       "           1.0823e-01,  6.3232e-01,  7.5784e-02,  1.1399e-01,  1.0357e-01,\n",
       "           9.2074e-02, -1.2842e-01, -4.1320e-01, -2.7649e-01,  4.3822e-01,\n",
       "           7.9239e-02,  5.4806e-01, -1.9019e-01,  3.9732e-01, -1.3411e-01,\n",
       "           3.7790e-02,  7.5584e-03, -3.7831e-01,  7.7874e-02, -5.4834e-01,\n",
       "          -4.6631e-01, -1.5365e-02, -3.1839e-01, -1.6352e-01,  3.1730e-01,\n",
       "          -8.1131e-02, -3.7504e-01, -4.6459e-02, -4.8291e-01, -1.1624e-01,\n",
       "           2.9144e-02,  6.1295e-01,  6.2542e-03, -5.9695e-01, -4.0296e-03,\n",
       "           2.5992e-01, -1.2384e-01,  2.5329e-01,  3.6165e-01,  2.6847e-01,\n",
       "          -8.3828e-02,  6.6351e-01, -2.5962e-01,  5.0360e-02,  4.0460e-01,\n",
       "          -1.3409e-01,  4.4781e-02, -3.7196e-01,  1.7832e-01,  1.2138e-01,\n",
       "          -6.1874e-02, -8.9533e-02, -8.0099e-03, -4.3593e-01, -2.4428e-01,\n",
       "           4.3712e-01, -1.5073e-01, -5.7252e-01, -4.8309e-01,  8.3866e-02,\n",
       "          -4.4766e-01,  6.2422e-02, -2.0149e-01, -9.8514e-02, -8.2141e-01,\n",
       "          -7.6775e-01, -2.5766e-01,  9.9257e-01,  6.0452e-02, -7.9639e-02,\n",
       "          -1.5844e-01,  3.1293e-01,  1.9918e-01,  1.2511e-01,  2.4980e-01,\n",
       "           6.1547e-01, -7.0023e-02, -1.4212e-01, -6.0066e-01, -1.1289e-01,\n",
       "           3.2711e-02,  1.8869e-01, -7.0938e-01,  4.4123e-01,  2.9051e-01,\n",
       "           1.4780e-01, -1.8910e-03,  4.7690e-02, -2.4427e-01, -8.2654e-02,\n",
       "          -3.2776e-01,  2.1080e-01,  6.4702e-02, -3.0334e-01, -1.4278e-01,\n",
       "           1.1192e-01, -9.8848e-01,  2.6880e-01, -5.3645e-01, -3.1004e-01,\n",
       "          -3.3529e-01,  6.7010e-02,  1.4224e-01,  3.0220e-01, -1.7899e-01,\n",
       "           2.6049e-01,  3.6234e-01,  8.7504e-02,  2.5372e-01, -2.7836e-01,\n",
       "           1.5111e-01,  5.6457e-01,  4.8572e-01,  1.7727e-01,  2.3084e-01,\n",
       "          -1.0665e-01,  2.0910e-01, -1.7008e-01,  1.5777e-01,  2.4308e-01,\n",
       "           2.3336e-01, -1.4744e-01,  1.9621e-01,  1.8334e-01, -1.8269e-01,\n",
       "           3.5084e-03, -1.5500e-01, -2.7924e-01,  3.8995e-02, -1.7156e-01,\n",
       "           2.1621e-01, -4.0519e-01, -1.2547e-01,  4.7102e-01,  9.5210e-03,\n",
       "           1.7430e-01,  1.3316e-01,  2.0852e-01, -5.9368e-02,  8.5292e-02,\n",
       "           4.4079e-01, -3.2208e-01,  2.9633e-02, -7.8950e-02, -1.0051e-02,\n",
       "           1.8517e-01, -6.5257e-01,  5.1740e-01,  5.9245e-01,  9.0054e-03,\n",
       "           4.1102e-04, -2.5228e-01,  2.2338e-01,  2.1197e-02,  2.3983e-01,\n",
       "          -7.2773e-02, -4.6780e-01, -5.3510e-01, -2.3036e-02,  1.6148e-01,\n",
       "           4.7408e-01, -1.5164e-01,  6.3939e-01,  7.5126e-01,  3.6086e-01,\n",
       "          -3.8436e-02,  9.0272e-02,  3.4723e-01,  2.1050e-02, -3.7326e-01,\n",
       "          -1.7129e-01,  3.3129e-02, -7.6492e-01,  3.4443e-01, -4.4240e-01,\n",
       "          -3.1812e-01, -8.7727e-02, -1.1722e-02,  2.5150e-01, -8.3473e-02,\n",
       "           2.1067e-01,  1.4719e-01, -1.4373e-01, -3.7343e-01,  5.3310e-01,\n",
       "          -1.9823e-01, -4.8460e-02,  1.5910e-02, -1.9880e-01, -7.5846e-01,\n",
       "          -3.8654e-01,  2.0406e-01,  5.2358e-01, -1.3009e-01, -3.8709e-01,\n",
       "          -3.5118e-01, -2.0951e-01, -7.1406e-02, -1.5900e-01, -3.0233e-01,\n",
       "           5.5465e-01, -3.1143e-01, -2.2156e-01, -1.0853e-01, -4.2147e-01,\n",
       "          -7.7633e-01,  1.8128e-01, -4.7821e-01, -1.9199e-01,  2.5214e-01,\n",
       "          -5.8255e-02,  1.5899e-01,  3.1991e-01, -1.7758e-03, -4.7469e-01,\n",
       "           1.2872e-01,  2.7448e-01,  1.7947e-01,  7.3083e-02,  3.9061e-01,\n",
       "           1.6625e-01,  3.1105e-01,  4.4880e-01, -3.4159e-01, -4.9985e-01,\n",
       "          -5.7731e-02,  3.6785e-01, -4.8684e-02, -2.1406e-01, -6.0050e-02,\n",
       "          -1.2460e+00, -3.9759e-01, -5.5277e-01, -1.2431e-01, -7.2411e-01,\n",
       "          -9.0872e-02,  8.0127e-01,  1.0125e-01,  1.0000e-01, -6.3863e-02,\n",
       "           2.8965e-01,  5.3090e-01, -2.6124e-01,  5.5496e-01, -5.4189e-01,\n",
       "          -1.7191e-01, -2.0941e-01, -4.2012e-01, -5.3785e-01, -3.9543e-01,\n",
       "           4.7790e-01,  1.4253e-01,  2.3105e-01, -4.2973e-01, -3.5944e-01,\n",
       "           4.6942e-01, -5.7701e-01, -6.5543e-02,  3.6169e-01, -3.0868e-01,\n",
       "           6.1009e-01,  3.7238e-01,  7.9041e-02,  8.8123e-02,  4.6535e-01,\n",
       "           1.4157e-01, -1.2979e-01, -4.0201e-01, -2.3154e-01,  4.7492e-01,\n",
       "           4.2551e-01,  2.7060e-01,  1.5096e-01,  3.6610e-01,  3.7772e-01,\n",
       "          -2.3276e-01,  4.2517e-01, -3.2658e-01, -2.5678e-01,  2.5610e-01,\n",
       "           4.4119e-02,  2.5241e-01, -3.7062e-01,  2.0118e-01,  4.2635e-02,\n",
       "          -2.8901e-01, -5.0459e-01,  4.4605e-01,  2.2239e-02,  1.5838e-01,\n",
       "          -1.1164e-02, -3.3406e-01,  1.6751e-01, -5.9902e-01, -7.7234e-02,\n",
       "           5.7444e-02, -1.8535e-01,  2.3314e-01, -4.3627e-01, -4.6482e-01,\n",
       "          -1.9816e-01, -1.5221e-01,  7.2863e-01,  1.4465e-01,  2.7785e-01,\n",
       "           2.5842e-01, -1.4953e-02, -1.0439e-01,  9.2671e-01,  3.7663e-01,\n",
       "          -2.0995e-01,  3.9613e-01, -1.2324e-01,  1.2153e-01,  5.4355e-01,\n",
       "          -3.2331e-01, -5.8238e-02,  7.7621e-02,  2.8884e-03,  2.9717e-01,\n",
       "          -2.6449e-01, -2.2209e-01,  2.1764e-01, -5.2520e-01, -8.2954e-02,\n",
       "          -1.3805e-01,  4.0687e-01, -4.6113e-01,  8.4609e-02, -9.0349e-02,\n",
       "           2.2492e-01,  1.6974e-02, -9.4221e-02,  2.7133e-01, -4.9879e-01,\n",
       "           2.2579e-01, -2.3743e-01,  1.5776e-01,  4.8922e-01, -1.1327e+01,\n",
       "           1.2724e-01,  9.7062e-01, -3.4920e-01,  3.6731e-03, -2.2031e-01,\n",
       "           1.2371e-01, -1.7631e-01,  1.1934e-01, -1.1724e-01, -2.1382e-01,\n",
       "          -4.3127e-01, -2.4335e-01,  1.7253e-01, -1.7401e-01, -1.9536e-01,\n",
       "           1.0384e-01, -7.8900e-01,  1.3594e-01,  1.9294e-02,  3.5358e-01,\n",
       "          -1.6056e-01, -3.9059e-01,  2.8299e-01, -3.6627e-01,  2.0941e-02,\n",
       "          -1.9911e-01, -3.7387e-01,  1.2779e-01,  3.7582e-01, -2.8435e-01,\n",
       "           7.3935e-02,  1.1672e-01,  4.4529e-01, -1.4392e-01, -6.2762e-01,\n",
       "           6.2533e-03, -5.4407e-02,  2.1153e-01, -2.1291e-02, -4.4602e-01,\n",
       "          -1.1469e-01, -4.3203e-01,  2.2598e-01,  2.6279e-01,  7.9330e-01,\n",
       "           3.5046e-01,  1.1303e+00,  1.9889e-01, -6.9955e-02, -1.2767e-01,\n",
       "          -7.2047e-01, -1.0885e-02, -9.1850e-01,  1.7892e-01, -6.2279e-03,\n",
       "           6.8186e-01, -2.9705e-01,  2.5443e-01, -4.9705e-01, -1.2247e-01,\n",
       "           2.6117e-01, -1.6420e-02,  1.3313e-01, -2.3200e-01, -3.7867e-01,\n",
       "           2.9743e-01,  1.9180e-02,  6.5850e-02, -3.8961e-01, -1.2099e-01,\n",
       "           8.4635e-03,  4.3935e-01,  2.1527e-01,  1.1176e-01,  4.6371e-01,\n",
       "          -1.3381e-02, -3.1079e-01,  1.7809e+00, -5.3046e-01,  1.3773e-01,\n",
       "          -2.9690e-01,  5.1763e-01, -2.9244e-01, -9.5253e-01,  3.5530e-01,\n",
       "          -1.2168e-01, -1.8395e-01,  4.8426e-01,  3.5767e-01, -2.1522e-01,\n",
       "          -3.2732e-01,  9.5037e-02,  3.9325e-02, -1.5961e-01,  6.4938e-01,\n",
       "          -4.9284e-01, -3.4690e-01, -1.9555e-02,  3.5515e-01, -8.3101e-01,\n",
       "          -1.1084e-01, -2.5220e-01, -1.2003e-01,  2.7971e-01, -7.2330e-02,\n",
       "           8.0915e-02,  3.5807e-01, -1.4922e-02,  5.5670e-01, -1.0536e-01,\n",
       "           6.4813e-02,  1.5170e-01,  9.9478e-02, -2.9023e-01, -6.6693e-01,\n",
       "           2.9312e-01,  4.3020e-01, -1.7304e-01, -7.2907e-01, -2.6906e-01,\n",
       "          -9.3472e-02, -1.0835e-01,  2.6828e-01, -6.0349e-01,  5.3369e-01,\n",
       "          -8.2665e-01, -2.6401e-01,  3.7914e-01, -6.7971e-02,  5.5148e-01,\n",
       "           1.5188e-01, -4.6685e-01,  3.5182e-03, -5.0041e-01, -1.2172e-01,\n",
       "          -6.6561e-01, -3.6334e-01,  2.9170e-01, -4.2732e-01,  1.4000e-01,\n",
       "           5.7038e-02,  1.8283e-03, -1.1758e-01,  1.0728e-01,  4.3731e-02,\n",
       "          -2.2838e-02,  1.8879e-01, -2.7191e-02,  2.0306e-01,  8.0934e-03,\n",
       "          -1.1525e-01, -2.0689e-01, -1.9230e-01,  2.3312e-02, -5.6068e-01,\n",
       "           3.2356e-02, -1.3991e-01, -2.7658e-01, -1.8078e-01,  3.1749e-01,\n",
       "           2.0447e-01, -1.9427e-02, -4.3378e-01,  6.8637e-01,  1.7971e-01,\n",
       "          -1.4630e-01, -5.5637e-02, -3.1297e-01, -2.3608e-01,  4.2160e-01,\n",
       "           1.2628e-01, -2.7365e-01, -4.6687e-01, -9.4563e-02, -5.3696e-01,\n",
       "           3.0637e-02,  7.5039e-01, -4.1316e-01, -4.9318e-01,  3.9994e-01,\n",
       "           5.1087e-01, -2.0418e-01, -7.3282e-03,  1.3982e-01, -1.7102e-01,\n",
       "          -3.9689e-02, -1.9561e-01,  5.2645e-01, -1.7516e-02,  8.8630e-02,\n",
       "           3.6140e-01,  4.2569e-01,  2.8952e-01,  2.3311e-02, -2.8699e-01,\n",
       "          -5.2594e-01,  2.7717e-02, -2.7859e-01, -1.3369e-01, -3.8327e-02,\n",
       "          -1.1528e-01,  3.7106e-02,  2.2557e-01, -4.2270e-01,  2.0205e-01,\n",
       "           6.3685e-01,  5.9527e-01, -7.6784e-02,  9.0474e-03,  3.6283e-01,\n",
       "           3.9017e-01, -1.5335e-01, -2.0636e-02,  2.1500e-01,  5.1101e-03,\n",
       "           4.7950e-01, -5.5823e-02, -2.8330e-01, -1.6631e-01,  3.7316e-01,\n",
       "           2.2824e-01, -4.4384e-01, -5.0080e-01,  7.3268e-02,  4.1967e-01,\n",
       "          -3.9255e-01,  3.8660e-01, -7.1888e-02,  1.4818e-01,  1.4184e-01,\n",
       "          -9.7030e-01, -1.1201e-01,  1.5502e-01, -4.4592e-01, -2.6117e-01,\n",
       "          -1.8934e-01, -3.7052e-01, -3.4551e-01, -2.8399e-01,  4.8215e-01,\n",
       "           5.6143e-01, -7.7874e-01,  2.2511e-01,  6.6661e-01, -3.9697e-01,\n",
       "           5.7709e-01, -1.7770e-01,  3.0882e-01,  8.7605e-01, -1.7209e-01,\n",
       "          -3.2216e-02,  5.6146e-02,  3.7658e-02, -3.4814e-01,  1.8533e-02,\n",
       "           7.5564e-01,  1.8990e-02, -6.5024e-02, -4.6416e-01, -1.3060e-01,\n",
       "          -2.3557e-01, -2.0165e-01, -2.4314e-01,  1.7898e-01,  5.6543e-01,\n",
       "          -1.6474e-01, -1.7708e-01, -2.9691e-01,  1.7943e-01, -5.7723e-02,\n",
       "           2.9573e-01,  5.7022e-01,  7.4429e-01,  1.6145e-01, -7.1755e-01,\n",
       "          -4.8328e-02,  5.1962e-01, -3.8446e-01,  7.9507e-01, -2.4289e-01,\n",
       "          -4.9076e-01, -7.6733e-02,  2.0380e-01, -2.0812e-01, -1.7483e-01,\n",
       "          -4.0346e-01,  1.9765e-01, -3.2374e-01, -1.8385e-01,  3.9374e-01,\n",
       "           8.7942e-02, -2.0366e-02, -7.7012e-01,  1.8608e-01,  6.7520e-01,\n",
       "           1.1803e-01,  1.2351e-01,  1.2800e-01,  7.1826e-03, -2.3228e-01,\n",
       "          -1.5764e-01, -5.7117e-01, -8.8729e-02,  1.3367e-01, -9.5451e-02,\n",
       "          -5.6603e-01, -1.1645e-01,  9.6016e-02, -2.0297e-02,  7.0205e-01,\n",
       "           8.0210e-01, -2.1464e-01, -4.7329e-01, -2.9963e-01, -3.0494e-02,\n",
       "           1.3864e-01, -5.7732e-03,  1.9613e-01, -3.6748e-01, -6.2195e-02,\n",
       "          -2.3292e-01, -3.3981e-01,  1.5453e-01,  4.3557e-01, -9.7357e-02,\n",
       "           5.3869e-01, -1.0649e-01,  1.0034e-01,  5.0421e-01,  2.3945e-01,\n",
       "           1.2849e-01,  7.1368e-01,  2.6139e-01, -3.5124e-01, -2.2475e-01,\n",
       "           2.1641e-01, -1.3495e-01,  2.4522e-01, -5.5745e-01,  3.6797e-01,\n",
       "           5.6888e-02,  5.9251e-01,  3.1864e-01]], device='mps:0'),\n",
       " tensor([[-0.2750, -0.5265, -0.6212,  ...,  0.2518,  0.2805,  0.5264],\n",
       "         [-0.2035, -0.5998, -0.8688,  ...,  0.5074,  0.4200,  0.7821],\n",
       "         [-0.0695, -0.4081, -0.4380,  ...,  0.2948,  0.7431,  0.8256]],\n",
       "        device='mps:0'),\n",
       " tensor([[-0.3670, -0.8917, -0.7430,  ...,  0.1222,  0.8454,  0.5251],\n",
       "         [-0.1819, -0.8335, -0.7050,  ..., -0.0304,  0.4907,  0.5283]],\n",
       "        device='mps:0'),\n",
       " tensor([[ 2.0208e-02, -7.6503e-01, -5.5929e-01, -2.5692e-01, -5.6472e-01,\n",
       "           5.7194e-02, -2.9899e-01,  6.1910e-01,  5.2920e-03,  4.5897e-01,\n",
       "           4.0233e-01,  2.0171e-01,  1.7813e-02, -1.9294e-01, -5.8388e-01,\n",
       "          -3.3605e-01,  3.3281e-01,  4.2457e-01,  3.0132e-01, -6.8317e-02,\n",
       "           7.2122e-01,  4.3789e-01, -3.0936e-01, -1.1078e-02, -6.4128e-01,\n",
       "           3.0362e-01,  8.5775e-02,  1.2002e-01,  3.6188e-01, -4.3962e-01,\n",
       "          -4.3361e-01,  5.5624e-01, -4.7431e-02,  6.8908e-01, -3.6465e-01,\n",
       "          -1.6057e-01,  1.2906e-01, -2.5347e-01,  3.7705e-01,  1.3642e-01,\n",
       "           7.8618e-01, -1.3841e-01,  1.9413e-01,  3.8705e-01,  5.8566e-01,\n",
       "           5.5818e-02,  5.9252e-02, -9.8198e-01,  4.3724e-01, -4.2433e-01,\n",
       "          -8.7132e-02, -1.9352e-01,  2.9384e-01,  4.3466e-01,  4.1407e-02,\n",
       "          -3.8051e-01,  1.3777e-01,  2.5721e-01, -3.6372e-01,  1.6510e-01,\n",
       "          -7.3787e-02,  8.1903e-01, -1.2874e-01, -6.0702e-02, -5.3394e-02,\n",
       "           4.8926e-02, -1.0550e-01, -7.6464e-01, -3.0272e-01, -1.4867e-01,\n",
       "          -2.1655e-01,  1.6044e-01, -3.2349e-01, -1.7165e-01,  4.2270e-01,\n",
       "          -3.6935e-01,  1.4055e-01, -4.9128e-02, -4.9496e-01, -1.3233e-02,\n",
       "           4.0388e-01,  4.8051e-01,  3.8214e-02, -2.0929e-02, -4.3727e-01,\n",
       "          -6.0656e-02,  6.7806e-01,  4.1573e-01,  3.6109e-01, -6.7068e-03,\n",
       "           4.0578e-01,  1.5997e-01, -1.4478e-02, -3.7510e-01,  3.7012e-01,\n",
       "          -2.0713e-01,  3.9778e-01, -1.8583e-01,  6.5082e-02, -9.5916e-02,\n",
       "           7.1551e-02,  1.8106e-01,  1.3193e-01,  1.7613e-01, -5.0597e-01,\n",
       "          -6.4350e-01,  2.0210e-01, -4.2119e-02, -8.4742e-01, -8.1491e-02,\n",
       "          -2.3964e-01,  2.2045e-01,  3.4319e-01, -4.2882e-01, -1.0789e-01,\n",
       "           1.2355e-01,  3.2049e-01, -3.3026e-01, -5.3377e-01,  3.8314e-01,\n",
       "           2.0406e-01, -5.1428e-01,  1.4590e-01,  2.3334e-01,  3.6936e-01,\n",
       "          -5.1731e-02,  2.7681e-01, -3.6767e-01,  4.4541e-01,  4.9211e-01,\n",
       "          -2.3287e-03,  1.9927e-01, -2.3671e-01, -2.1461e-01, -2.8364e-02,\n",
       "          -1.9802e-02,  1.1374e-01, -1.8721e-01,  3.1578e-01,  7.9393e-02,\n",
       "          -1.7306e-01, -1.4675e-01, -4.4787e-01,  6.8730e-02, -8.9550e-02,\n",
       "          -7.4923e-03, -6.0417e-02,  1.3191e-01, -2.0985e-01, -3.4197e-01,\n",
       "          -8.3526e-01, -6.6114e-01,  2.9746e-01,  1.6620e-02, -2.6681e-02,\n",
       "          -8.1852e-03,  4.1030e-01,  1.2823e-01,  2.0095e-01,  4.6233e-01,\n",
       "           3.9786e-01, -1.6751e-01,  2.4818e-01, -5.1207e-01, -6.5876e-02,\n",
       "           3.7497e-01,  2.1087e-01, -2.8988e-01, -8.3998e-02,  2.4633e-01,\n",
       "           2.3304e-01, -4.3286e-01, -1.2071e-01, -7.9654e-01, -3.2950e-02,\n",
       "          -6.0529e-01,  2.8070e-01,  2.2814e-01, -4.3748e-01,  9.8900e-03,\n",
       "          -4.4120e-01, -2.0305e+00,  2.0706e-01, -2.1848e-01,  1.2624e-01,\n",
       "          -6.4365e-01,  3.0940e-01,  3.7443e-01,  3.4252e-01, -1.5505e-01,\n",
       "          -3.8439e-01, -5.8677e-03,  1.1463e-01,  6.6717e-01, -1.3443e-01,\n",
       "           3.3261e-01,  6.4918e-01,  1.0890e-02, -2.5317e-01, -7.8681e-03,\n",
       "          -3.3116e-01,  1.3954e-01, -1.1193e-01, -4.6732e-01,  1.5745e-02,\n",
       "           2.0312e-01,  1.5326e-01,  2.1860e-02,  4.0112e-01,  2.8193e-03,\n",
       "          -9.9428e-02, -3.1422e-01, -5.0077e-03, -4.3295e-01, -8.7508e-02,\n",
       "           3.7966e-01, -1.5067e-01, -1.0608e-01,  3.9688e-01,  1.2532e-01,\n",
       "          -1.3816e-01,  1.1189e-01, -1.4272e-01, -5.7000e-02, -1.6625e-01,\n",
       "           2.8259e-01, -1.1423e-01, -1.0816e-01, -2.9287e-01, -4.0375e-01,\n",
       "          -3.0225e-03, -2.9997e-01,  7.1459e-01, -5.2832e-02, -2.1241e-01,\n",
       "          -4.9738e-01, -2.7527e-01, -9.5959e-02, -4.1868e-02,  1.8233e-01,\n",
       "          -1.9273e-01, -4.2171e-01, -8.5361e-01, -8.9961e-02,  8.8740e-02,\n",
       "          -4.7861e-02, -2.6897e-01,  9.0623e-01,  1.7193e-01,  2.4860e-01,\n",
       "          -3.4133e-02,  3.6571e-02,  4.1085e-01,  2.5951e-02, -1.6019e-01,\n",
       "           1.0218e-01, -2.8330e-02, -8.7010e-02,  4.1328e-01, -2.0672e-01,\n",
       "           2.8802e-02, -1.2390e-01, -4.0113e-02,  5.2599e-01, -1.0086e-01,\n",
       "           2.7946e-01,  4.4659e-01, -4.4787e-01,  1.7640e-01,  3.1108e-01,\n",
       "          -2.8588e-01,  3.4071e-01,  2.1494e-02, -1.5333e-01, -7.8185e-01,\n",
       "          -2.5145e-01,  4.3187e-01, -1.5070e-01,  1.5915e-01, -7.8193e-01,\n",
       "          -2.8628e-02,  2.9829e-01, -9.5702e-02,  2.2307e-01, -3.3996e-01,\n",
       "           1.0336e-01,  2.7664e-01,  5.4592e-02, -1.8774e-01, -6.2797e-01,\n",
       "           2.4925e-02,  2.6298e-01, -3.6244e-01,  3.6625e-01, -2.3536e-02,\n",
       "          -2.7597e-01,  6.5678e-02,  3.4875e-01, -5.6993e-03, -3.0233e-01,\n",
       "          -1.8564e-01, -1.5399e-01, -4.3299e-01, -8.5505e-02, -1.9394e-01,\n",
       "           1.6030e-01, -1.8583e-02,  4.4168e-01, -7.7571e-01, -6.2004e-01,\n",
       "          -8.2860e-02,  6.8730e-01,  1.1407e-01, -2.6381e-01, -3.6458e-01,\n",
       "          -1.4420e+00, -2.3726e-01, -4.1471e-01, -1.6035e-01, -5.0801e-01,\n",
       "           2.4845e-01,  5.6050e-03, -7.0122e-02,  5.1329e-02, -1.9641e-01,\n",
       "           2.2804e-01,  4.2249e-01, -1.4574e-01, -5.9133e-02, -4.7953e-02,\n",
       "          -2.9291e-01,  5.4336e-02, -1.0803e-01, -2.3218e-01, -7.5940e-02,\n",
       "           8.5886e-02,  1.0303e-01, -2.5835e-01, -5.4286e-02, -3.2694e-01,\n",
       "           6.3005e-01, -9.8360e-02,  2.2665e-01,  3.2024e-02, -3.4603e-01,\n",
       "          -1.4111e-01,  5.2638e-01, -2.6100e-01, -9.5505e-02, -3.1647e-02,\n",
       "          -8.1588e-04, -3.7655e-01, -6.9738e-02, -5.0863e-02,  5.1015e-01,\n",
       "           3.5030e-01,  2.1533e-01, -3.5168e-01,  4.2025e-01,  6.7044e-01,\n",
       "          -1.8195e-01,  2.5602e-01,  2.4438e-02, -1.7030e-01,  8.4277e-03,\n",
       "           2.9253e-01,  1.8083e-01, -5.4752e-01,  9.4749e-02,  3.7275e-01,\n",
       "           2.9185e-01, -3.6818e-01, -6.6197e-02,  2.1315e-01,  2.7459e-01,\n",
       "          -3.4370e-01, -3.8059e-01,  3.9311e-01, -3.9890e-01, -3.4777e-01,\n",
       "           4.3958e-02,  9.1178e-02,  1.5579e-01,  7.0696e-02, -5.7060e-01,\n",
       "          -4.3794e-01,  2.4488e-01,  8.7743e-01,  1.2242e-01,  1.5704e-01,\n",
       "           3.1480e-01, -1.1315e-01, -2.4633e-01,  3.3256e-01, -1.6159e-01,\n",
       "          -7.7163e-02,  3.1051e-01, -4.6952e-01,  4.0089e-01,  3.4175e-01,\n",
       "          -1.6735e-01, -2.5179e-01, -1.0563e-01, -3.9040e-01, -1.1559e-01,\n",
       "          -2.1741e-01,  3.9607e-01,  4.9480e-02,  1.2715e-01, -8.5624e-02,\n",
       "           2.8878e-01,  1.4682e-01, -2.8650e-01,  8.1864e-02, -7.2852e-02,\n",
       "           1.0194e-01,  1.1781e-01, -8.8097e-02, -4.7460e-02, -5.2011e-02,\n",
       "           3.8932e-01,  1.0028e-01,  2.4817e-01,  2.0441e-01, -1.1597e+01,\n",
       "           6.8626e-02,  7.5693e-01, -4.7819e-01, -1.7697e-02,  4.0950e-02,\n",
       "           1.6838e-01, -2.1576e-01, -6.7861e-02, -2.9902e-01, -2.2121e-01,\n",
       "           1.6634e-02,  2.9037e-02,  1.5752e-01, -2.8156e-02, -9.0802e-02,\n",
       "          -1.4198e-01, -6.5549e-01,  5.0503e-01, -4.5816e-02,  1.5790e-01,\n",
       "          -2.2957e-01,  3.6382e-01, -5.2319e-02, -1.6315e-01, -1.3850e-01,\n",
       "          -3.3683e-01, -4.1339e-01,  2.7810e-01,  1.4482e-01, -1.1726e-01,\n",
       "          -3.9508e-02,  3.4742e-01,  1.4369e-01, -4.0219e-01, -5.6183e-01,\n",
       "           2.0400e-01, -1.8716e-02,  4.9096e-02,  4.0898e-02, -1.9458e-01,\n",
       "           1.6506e-01, -6.7758e-01,  3.8554e-01,  1.8458e-01,  7.8416e-01,\n",
       "           7.3742e-01,  7.0307e-01,  2.3267e-01, -4.3792e-02,  2.8482e-02,\n",
       "          -8.0305e-01,  1.6911e-01, -8.0406e-01,  1.9478e-01, -8.5476e-02,\n",
       "           4.9424e-01, -6.4510e-01, -6.0906e-02,  8.3130e-02, -3.2829e-01,\n",
       "           2.5847e-01, -7.8931e-02,  2.7519e-01, -2.7523e-01, -1.2579e-01,\n",
       "           3.5699e-01,  1.1202e-01,  2.9189e-01, -1.4989e-02,  1.9941e-01,\n",
       "          -2.1878e-01,  5.6106e-01,  2.6796e-01, -2.9849e-01,  1.1820e-01,\n",
       "           2.0249e-02,  7.7441e-02,  5.5225e-01, -8.4660e-02, -4.6368e-02,\n",
       "           7.2658e-02,  3.4458e-01, -2.6370e-01, -7.9845e-01, -8.8288e-02,\n",
       "           2.4457e-01, -7.1626e-02,  6.4396e-02,  5.0609e-01, -3.0357e-01,\n",
       "          -2.6946e-01, -4.2302e-01,  1.9476e-02,  1.8042e-01,  7.1453e-01,\n",
       "          -5.8034e-01, -1.0122e-01, -1.6808e-02, -8.1182e-02, -7.3097e-01,\n",
       "          -8.3426e-02, -1.9399e-01, -2.7188e-01,  2.1695e-01, -2.7433e-01,\n",
       "          -7.3079e-03,  4.6410e-02,  2.2299e-01,  3.8007e-01, -5.8152e-01,\n",
       "           5.3584e-01,  3.9769e-02,  1.8878e-01, -1.4119e-01,  1.2564e-01,\n",
       "           1.4194e-01,  5.4991e-01, -2.6273e-01, -5.8466e-01, -2.5381e-01,\n",
       "          -1.5310e-01,  3.0415e-01, -1.5312e-01, -3.1279e-01,  1.1880e-01,\n",
       "          -2.5092e-01, -3.4404e-03,  8.2193e-02,  7.3047e-02,  4.6305e-01,\n",
       "           8.4067e-02,  6.7405e-02,  6.4410e-02, -5.4696e-01, -8.2122e-01,\n",
       "          -5.6732e-01,  1.5895e-01,  1.4352e-01, -2.9913e-01, -1.5565e-01,\n",
       "          -8.1726e-02,  7.9281e-02, -4.6112e-01,  2.3830e-01,  3.3623e-02,\n",
       "           7.5235e-01,  4.5941e-01, -1.1610e-01, -1.2501e-01, -2.4152e-01,\n",
       "          -4.7697e-01, -2.4586e-01,  8.6539e-03,  3.4751e-02, -3.2841e-01,\n",
       "          -3.3981e-01,  2.1232e-01, -2.0264e-01,  9.3159e-02,  2.3010e-01,\n",
       "          -2.8576e-01, -3.4762e-01, -2.4335e-01, -1.8679e-02,  3.3081e-01,\n",
       "           1.3124e-01, -5.1549e-02, -3.5166e-01,  9.7943e-02,  3.3827e-01,\n",
       "           9.2638e-02, -2.7596e-01, -3.1185e-01, -1.7647e-01, -3.0932e-02,\n",
       "           2.4132e-01,  8.9103e-01, -2.2223e-01, -2.5462e-01,  2.4756e-01,\n",
       "           1.2559e-01, -1.6163e-01,  2.4328e-01, -2.1901e-01, -8.2403e-02,\n",
       "          -3.1274e-02, -6.0327e-01,  2.3566e-01,  1.2174e-02,  3.7033e-01,\n",
       "           1.0918e-02,  1.0173e-01,  4.5553e-02,  3.5172e-01, -6.4380e-02,\n",
       "          -1.3083e-01, -7.2966e-02, -5.9665e-01, -1.2796e-01, -1.2882e-02,\n",
       "           1.6433e-02,  1.0146e-02, -2.8245e-01, -2.4282e-01,  2.5145e-01,\n",
       "          -4.8975e-02,  1.0930e+00,  3.7531e-01,  4.5117e-02,  1.7778e-01,\n",
       "           1.8941e-01, -4.0156e-01,  2.1312e-01, -3.3692e-02,  2.1484e-01,\n",
       "           3.7423e-01,  2.5027e-01, -2.1613e-01,  8.1767e-02,  1.1973e-01,\n",
       "           3.6036e-01, -7.3182e-01, -6.4988e-01, -6.2273e-02,  4.1248e-01,\n",
       "          -8.2303e-02,  3.2196e-01,  2.8974e-02,  5.7252e-01,  6.7178e-01,\n",
       "          -1.1460e+00, -2.3901e-01, -2.7814e-01,  7.7240e-01, -2.6478e-02,\n",
       "          -4.1822e-01, -1.6920e-01, -5.3889e-01, -4.6541e-01,  3.1793e-01,\n",
       "           6.9987e-01, -1.0613e+00,  1.6571e-01,  1.2003e-01, -1.9229e-01,\n",
       "           5.4059e-02, -5.7781e-01,  3.3046e-01,  1.5885e-02,  2.1187e-01,\n",
       "          -9.0080e-02,  5.4108e-01, -4.1372e-01, -5.0692e-01,  1.5216e-01,\n",
       "           3.9384e-01,  4.3704e-02,  4.0912e-02,  2.8431e-02,  3.3223e-01,\n",
       "          -1.3847e-01, -1.8566e-01, -2.6675e-01,  3.5006e-01, -4.9763e-02,\n",
       "           2.1789e-01, -5.1372e-01, -7.2430e-02,  2.6108e-01, -3.8436e-02,\n",
       "           3.6857e-01,  6.1013e-01,  6.8394e-01, -9.4953e-02, -2.8380e-01,\n",
       "           4.8577e-01,  1.5567e-01, -7.3465e-02,  6.4399e-01, -1.2834e-01,\n",
       "          -4.3102e-01,  6.4595e-01,  2.9300e-02, -2.5687e-01,  4.9953e-02,\n",
       "          -1.7909e-01,  3.3106e-01,  2.3613e-01, -1.9502e-01,  9.3905e-02,\n",
       "           1.3858e-01,  1.2522e-01, -3.8871e-01,  8.9062e-02,  1.8340e-01,\n",
       "          -3.6279e-01,  7.6027e-01, -1.6448e-01,  5.0500e-01, -2.9734e-01,\n",
       "           1.8890e-01, -3.0684e-01, -4.4585e-01, -3.3397e-01, -2.8147e-01,\n",
       "          -2.8851e-01, -1.6761e-01,  1.7526e-01, -9.1573e-02,  7.4284e-01,\n",
       "           3.8549e-01, -4.0562e-01, -3.0646e-01, -2.1099e-01,  1.9009e-01,\n",
       "           1.6331e-01, -1.7484e-01,  7.9365e-02,  2.1902e-01,  1.8608e-01,\n",
       "          -9.0702e-01, -4.3824e-01,  3.0154e-01,  3.2570e-01, -3.4959e-01,\n",
       "           5.0241e-01, -9.3615e-02,  1.9549e-01,  1.3200e-01,  1.4179e-01,\n",
       "           1.4406e-01,  5.4799e-01,  2.3142e-01, -3.4907e-01,  9.9233e-02,\n",
       "           1.8399e-01, -4.6496e-01,  7.9736e-02, -5.1373e-01,  2.3767e-01,\n",
       "           2.7593e-01,  8.8341e-01,  5.3475e-01]], device='mps:0'),\n",
       " tensor([[-2.3318e-01, -3.1999e-01, -6.3947e-01, -1.4512e-01, -2.4559e-01,\n",
       "          -4.3963e-01, -6.2153e-01,  7.1316e-01, -1.0264e-01,  2.8570e-01,\n",
       "          -6.6171e-02,  5.3188e-01,  1.1373e-01, -3.5027e-01, -1.5733e-01,\n",
       "          -2.6968e-01,  3.1279e-01, -3.0835e-01,  1.3411e-01,  3.8034e-01,\n",
       "           6.6849e-01,  3.3958e-01, -2.1964e-01,  3.1616e-01, -1.0069e+00,\n",
       "           4.0108e-01, -4.0708e-01,  1.9637e-01,  4.7499e-01, -8.1670e-01,\n",
       "          -6.6263e-02,  1.0011e+00, -3.5135e-01,  1.2075e+00,  4.6158e-03,\n",
       "          -3.5229e-01,  1.8809e-01,  7.3711e-02,  5.7828e-01, -1.2332e-01,\n",
       "           3.0632e-01, -2.1826e-01,  9.2529e-02,  3.2571e-01,  1.9089e-01,\n",
       "           3.2696e-01,  9.9352e-02, -6.1875e-01,  2.3549e-01, -2.7471e-01,\n",
       "          -1.5862e-01, -2.2745e-01,  5.0525e-01,  3.4709e-01, -5.8422e-01,\n",
       "          -1.3780e-01,  1.9325e-01,  1.5525e-01,  1.6976e-01,  2.5799e-01,\n",
       "           1.5466e-01,  6.1222e-01, -2.3405e-01, -5.5491e-02, -6.6217e-02,\n",
       "           1.1766e-01, -3.2164e-01, -2.6232e-01, -3.0685e-02, -2.3217e-01,\n",
       "          -1.0649e-01,  1.3231e-01,  9.1720e-02, -6.8261e-01,  2.8728e-01,\n",
       "           3.2949e-02,  9.4688e-02,  5.4894e-02, -4.5696e-01,  1.0404e-01,\n",
       "           2.0108e-01,  2.3645e-01,  2.1632e-01, -4.7950e-01, -4.1727e-01,\n",
       "           3.6365e-01, -3.7615e-01,  4.7918e-03,  3.2961e-01,  1.3703e-01,\n",
       "           3.1118e-01,  3.3635e-01, -2.0809e-01, -4.1538e-01, -8.2125e-03,\n",
       "          -3.1717e-01, -3.2345e-01, -2.7925e-01,  1.8867e-02, -4.4465e-02,\n",
       "           2.2636e-01, -6.6523e-02,  2.1672e-02, -3.3539e-01, -5.1798e-01,\n",
       "          -3.1764e-01,  3.4576e-01,  3.4802e-01, -7.4788e-01, -4.6319e-01,\n",
       "           2.0971e-02,  1.5688e-01,  1.7084e-01, -4.2955e-01, -7.9106e-02,\n",
       "           6.6253e-02, -3.5997e-02, -1.5555e-01, -8.0212e-02,  2.3600e-02,\n",
       "           1.4675e-01, -5.0447e-01,  1.3082e-01,  1.5799e-01,  8.8291e-01,\n",
       "          -2.3569e-01,  4.2791e-01, -3.4875e-01,  5.3631e-01,  4.6739e-01,\n",
       "           1.6243e-01,  1.1779e-01, -3.5955e-01, -3.4842e-01,  3.3075e-01,\n",
       "          -1.2031e-01, -3.5873e-02, -6.6550e-01, -7.6523e-02, -4.5640e-02,\n",
       "          -3.6119e-01, -1.5833e-01, -1.7573e-01, -4.7085e-01, -3.5040e-01,\n",
       "           7.6037e-02,  1.2995e-01, -2.5590e-01, -5.4639e-01, -1.0075e-01,\n",
       "          -1.1939e-01, -1.0931e+00,  3.6296e-01,  9.9374e-02, -6.3201e-02,\n",
       "          -3.8951e-01,  3.5105e-01,  1.6536e-01,  2.1304e-01,  2.2474e-01,\n",
       "           5.4382e-01, -1.0947e-02,  4.8908e-01, -5.8352e-01,  7.7905e-03,\n",
       "           7.4841e-02, -2.9156e-02,  7.4045e-02,  3.8673e-01, -1.1405e-02,\n",
       "           1.1394e-01,  3.8914e-01,  3.9227e-01, -1.7772e-01,  4.1629e-01,\n",
       "           7.1460e-02,  1.9663e-01, -1.3957e-01, -2.5709e-02, -4.6763e-01,\n",
       "          -1.9968e-02, -1.6712e+00, -3.2762e-01, -7.3450e-01,  1.0580e-01,\n",
       "          -5.1334e-01,  5.2879e-01, -3.3399e-02,  2.1460e-01, -2.1768e-01,\n",
       "          -2.8026e-01, -5.5857e-02, -1.6111e-03,  1.9543e-02,  1.4039e-01,\n",
       "           1.5657e-01,  2.5621e-01,  4.7433e-02,  1.0436e-01, -2.6073e-02,\n",
       "           3.4952e-04,  2.5721e-01, -1.4943e-02, -2.6811e-02,  1.9037e-01,\n",
       "           3.3211e-01, -9.7619e-02,  4.9361e-01,  4.8402e-01, -3.6945e-02,\n",
       "           1.2284e-01, -2.1412e-01, -2.3068e-01, -2.6231e-01,  2.1454e-01,\n",
       "           6.6512e-01, -5.2231e-01, -2.4810e-01,  5.4029e-01,  1.2828e-01,\n",
       "          -8.0274e-02, -1.4104e-01,  2.0249e-01, -2.1460e-01,  1.6446e-01,\n",
       "           6.5999e-01, -6.7879e-01,  6.5489e-02, -4.2470e-01, -1.3921e-01,\n",
       "          -9.5336e-03, -7.8010e-01,  8.2078e-01,  2.2333e-01,  1.3127e-02,\n",
       "           2.5417e-03, -4.2410e-01,  4.3397e-01,  2.1061e-02,  1.9388e-01,\n",
       "          -4.5130e-01, -2.7494e-01, -6.7797e-01,  3.7286e-02, -5.5879e-02,\n",
       "           3.7354e-01, -4.1118e-02,  5.6330e-01,  2.4143e-01,  2.7345e-01,\n",
       "           1.2724e-01, -4.6010e-01, -2.3288e-02,  4.7093e-01,  1.3865e-01,\n",
       "          -3.4798e-01,  3.9846e-02, -7.3836e-02,  3.4511e-01, -5.0107e-01,\n",
       "          -5.9768e-01,  3.4899e-03,  4.1435e-01,  2.4032e-01, -3.9888e-01,\n",
       "           6.4264e-01,  5.4252e-01,  6.4257e-02, -4.0095e-01,  2.3821e-01,\n",
       "          -2.0487e-01,  3.8772e-01, -1.0460e-01, -3.7015e-01, -2.5067e-01,\n",
       "          -9.8605e-01, -2.8258e-01, -5.3820e-01,  4.1507e-01, -7.0921e-01,\n",
       "          -6.0121e-01, -1.8404e-01,  3.1877e-01,  1.9131e-01,  1.6533e-01,\n",
       "           1.8953e-01,  2.1953e-01, -2.9388e-01,  7.0939e-02, -6.3159e-01,\n",
       "          -3.2382e-01,  3.9359e-01, -1.9864e-01, -1.3646e-02,  4.3648e-01,\n",
       "           2.0843e-01, -2.5129e-01,  6.4528e-01, -8.0851e-02, -3.6877e-01,\n",
       "          -3.3123e-01, -4.3133e-01,  4.6254e-01, -9.5473e-02, -3.6253e-01,\n",
       "          -4.0854e-01,  1.1156e-01,  3.2628e-01, -1.2798e-01, -2.9998e-01,\n",
       "          -1.8048e-01,  6.2732e-01, -2.9985e-01, -9.9983e-03,  1.1286e-02,\n",
       "          -1.2050e+00, -1.9698e-01, -3.8003e-01,  7.1372e-02,  1.5323e-01,\n",
       "           2.6922e-01, -3.7501e-01, -3.6443e-01,  9.8982e-02,  1.0275e-01,\n",
       "           3.8848e-01,  7.8303e-01,  1.6253e-01, -8.1986e-03, -1.0373e-01,\n",
       "           2.6427e-01, -2.8514e-01, -8.4587e-02, -3.6368e-01, -2.4336e-01,\n",
       "           1.1484e-01, -1.3721e-01, -2.9370e-01, -1.0059e-01, -2.3841e-01,\n",
       "           1.0609e+00, -4.7744e-01, -1.7116e-01,  3.7407e-01, -2.5641e-01,\n",
       "          -4.7743e-01,  1.8411e-02, -3.9257e-01,  2.5574e-01, -4.5308e-02,\n",
       "           7.5180e-02, -3.2243e-01, -3.5874e-01,  1.9703e-01,  6.9039e-01,\n",
       "           4.3068e-01,  2.4349e-01, -4.0191e-05,  7.9324e-01,  6.0710e-02,\n",
       "          -7.1937e-01,  3.1934e-01,  3.9175e-01, -3.3962e-01, -2.5170e-01,\n",
       "           4.7298e-02,  2.3791e-01, -4.3999e-01, -2.2705e-01,  1.1602e-01,\n",
       "           1.5044e-01,  2.5983e-01, -4.8513e-01,  2.5025e-01,  2.0291e-01,\n",
       "          -2.7070e-01, -6.1462e-01,  1.4305e-01, -5.1986e-01, -2.7465e-01,\n",
       "          -3.0949e-01,  1.1123e-01,  3.4607e-01, -3.1317e-01,  1.0940e-01,\n",
       "          -3.8488e-01, -1.0657e-01,  4.5140e-01,  1.1228e-01, -1.1009e-01,\n",
       "           4.1526e-01,  1.3665e-02, -3.5945e-01,  1.4893e-01, -1.0124e-01,\n",
       "          -3.5592e-01, -1.8571e-01, -2.3753e-01,  5.8953e-01,  4.1991e-01,\n",
       "          -3.9919e-01, -2.8201e-03,  3.7576e-02,  2.0575e-01,  4.4127e-01,\n",
       "          -2.6794e-01,  3.3321e-01,  1.4315e-01,  1.5701e-01,  6.7117e-02,\n",
       "          -2.0004e-01,  7.4379e-01, -6.6496e-01,  3.7001e-02, -2.8319e-01,\n",
       "           4.2191e-01, -2.5786e-01, -2.9513e-01, -2.9084e-01, -8.4197e-01,\n",
       "           4.8969e-01, -2.3568e-01,  3.0665e-01,  3.5256e-01, -1.1001e+01,\n",
       "           1.7779e-01,  8.8398e-01, -1.7339e-01,  2.9703e-01, -2.8770e-01,\n",
       "           2.1226e-01, -1.9430e-01,  1.5012e-01, -1.6678e-01, -2.0006e-02,\n",
       "           3.4744e-02, -7.3555e-02,  1.8184e-01, -4.4705e-01,  1.9303e-03,\n",
       "          -6.1772e-01, -6.9338e-01,  4.1267e-01, -4.4878e-01, -7.6131e-02,\n",
       "          -7.8907e-01,  4.8027e-01,  2.0851e-01, -4.0593e-01, -6.2601e-01,\n",
       "          -1.8793e-01, -1.1166e-01,  4.2808e-01, -1.1133e-02,  1.4598e-01,\n",
       "          -1.7241e-02,  1.0903e-01,  5.9023e-01,  3.1443e-02, -1.2130e-01,\n",
       "           2.5123e-02,  3.4859e-01,  5.3211e-01, -2.8928e-01, -6.8703e-01,\n",
       "           6.5016e-02, -3.2120e-01,  4.1826e-01,  4.8033e-01,  7.2677e-01,\n",
       "           1.6077e+00,  9.0959e-01, -2.8731e-02,  1.9514e-01, -2.2615e-01,\n",
       "          -8.9577e-01,  3.7030e-01, -3.8066e-01,  3.1361e-01, -3.1611e-02,\n",
       "           1.1830e-01, -8.9552e-01, -4.3094e-02, -1.0836e-01, -2.1140e-01,\n",
       "           4.2716e-01,  1.2022e-01, -2.1579e-01, -5.2365e-02, -2.1106e-01,\n",
       "           2.3641e-01, -2.0893e-02,  2.5890e-01, -3.7405e-01, -1.2255e-02,\n",
       "           1.8523e-01,  5.5774e-01, -1.0157e-02, -8.6216e-01,  1.0923e-01,\n",
       "          -2.8093e-01,  8.4693e-02,  1.3344e+00, -1.6099e-01,  1.0836e-01,\n",
       "           2.9932e-01, -1.0924e-01, -5.9046e-01, -5.2077e-01, -5.1795e-01,\n",
       "          -1.2358e-02, -1.7448e-01,  1.1186e-01,  6.5064e-01, -1.8070e-01,\n",
       "          -4.5454e-01,  2.1444e-02,  3.4162e-01,  4.7253e-01,  5.9399e-01,\n",
       "          -2.7525e-01, -4.4022e-02,  1.9038e-01,  6.1598e-01, -5.1523e-01,\n",
       "          -7.4799e-02, -4.6180e-01, -4.6019e-01,  4.6552e-01, -2.4443e-01,\n",
       "          -8.5948e-02,  1.9989e-01,  1.7312e-02,  5.7908e-01, -3.4545e-01,\n",
       "          -4.0621e-02, -1.5137e-01,  1.9965e-02, -3.6173e-01,  8.4336e-02,\n",
       "          -4.0749e-01,  9.4392e-01, -4.3883e-01, -3.3117e-01,  7.2361e-02,\n",
       "           2.9447e-01,  3.6329e-01,  2.4443e-02, -5.2966e-01,  1.2136e-01,\n",
       "          -2.7041e-01, -2.4410e-01,  9.2973e-02,  3.2810e-01,  6.3050e-01,\n",
       "           6.8833e-01, -6.1736e-01,  3.4011e-01,  1.3864e-01, -6.0570e-01,\n",
       "          -9.7881e-01, -2.2210e-02,  3.7253e-01, -5.1386e-01,  3.8272e-01,\n",
       "           1.4649e-01,  1.2780e-02, -1.7710e-01,  1.7895e-01,  2.0832e-01,\n",
       "           3.0462e-01,  2.3153e-01, -2.1694e-01,  9.5572e-02, -3.6258e-01,\n",
       "          -4.5244e-01,  2.1933e-01,  2.0012e-01, -1.0232e-01,  2.6348e-02,\n",
       "          -6.1503e-01, -3.3977e-02,  1.3372e-01, -3.3339e-01,  2.9996e-01,\n",
       "           7.1031e-03, -1.5094e-01, -1.4912e-01,  2.5813e-01,  8.6768e-03,\n",
       "           5.6604e-02, -2.1406e-01, -7.5657e-01, -3.6225e-02,  2.6334e-01,\n",
       "           5.0223e-01, -1.7343e-02, -5.8585e-01, -6.8273e-01, -7.9287e-02,\n",
       "          -5.5764e-02,  1.0620e+00, -4.8938e-01, -6.9604e-01,  2.4926e-01,\n",
       "           5.5777e-01, -8.4495e-02, -2.8398e-01, -1.2879e-01,  2.2816e-01,\n",
       "          -3.4639e-01, -6.2371e-01,  4.7821e-01,  4.6521e-02,  3.7367e-01,\n",
       "           1.9539e-01,  3.4601e-01, -1.0118e-01,  3.1458e-01, -3.1020e-01,\n",
       "          -4.6744e-01, -6.5703e-02, -8.0180e-01, -3.2109e-01,  4.7567e-01,\n",
       "           1.3092e-01,  1.1110e-02,  2.7853e-01, -3.4303e-02, -2.3102e-01,\n",
       "           3.3978e-01,  1.2973e+00,  2.5123e-01, -3.4989e-01, -5.7743e-02,\n",
       "           1.7879e-01, -1.3879e-01, -6.7979e-02, -2.4473e-02,  3.8268e-01,\n",
       "           2.4009e-01, -1.3847e-02, -1.9580e-01, -1.1160e-01,  1.9028e-01,\n",
       "           3.0449e-01, -8.0693e-01, -4.8824e-01,  3.5149e-02,  5.1132e-01,\n",
       "          -1.7945e-02,  5.2374e-01, -4.1555e-01,  1.7255e-01,  7.5915e-01,\n",
       "          -8.7814e-01, -2.6196e-02, -3.1632e-01,  3.7241e-01, -1.5728e-01,\n",
       "          -1.0231e-01, -5.8813e-02, -3.0923e-01, -2.6393e-01,  5.2010e-01,\n",
       "           7.0052e-01, -7.2529e-01,  4.9035e-02,  5.2828e-01, -1.7522e-01,\n",
       "           2.5274e-01, -2.8941e-01,  2.9918e-01,  2.1837e-01, -1.6477e-01,\n",
       "          -7.5508e-01,  3.5848e-01, -2.9972e-01, -7.9926e-01,  3.8066e-01,\n",
       "           4.8818e-01, -1.0454e-01,  2.7515e-01,  4.5250e-01, -1.3468e-01,\n",
       "          -1.8195e-01, -4.1190e-01, -1.7831e-01,  3.4320e-01, -2.9934e-01,\n",
       "          -1.2907e-01, -4.8147e-01,  8.9596e-02,  3.9613e-01, -2.7420e-01,\n",
       "           4.1781e-01,  5.4320e-01,  6.7119e-01, -2.0617e-01, -1.1387e-02,\n",
       "          -2.3988e-01,  5.9828e-01, -2.4608e-02,  4.6002e-01, -2.5630e-01,\n",
       "          -6.7048e-02,  1.9995e-01, -6.9685e-02, -6.6078e-01,  5.5042e-01,\n",
       "          -2.7937e-01,  1.1294e-01, -1.8635e-01, -3.7932e-01,  4.8356e-01,\n",
       "          -1.4661e-01,  3.2172e-01, -4.0035e-01, -2.9600e-01,  6.3504e-02,\n",
       "          -5.9920e-03,  4.3308e-01,  2.0072e-01,  6.8190e-01, -9.5146e-02,\n",
       "          -8.3198e-02, -8.4242e-01,  1.0142e-01, -2.1159e-02,  2.7165e-01,\n",
       "          -3.4013e-01, -1.7132e-01,  1.1667e-01, -4.3542e-03,  5.8688e-01,\n",
       "           5.4168e-01,  1.1591e-03, -3.3582e-01, -4.1439e-01, -2.4306e-01,\n",
       "          -1.0676e-01, -1.4936e-02,  4.3686e-03, -1.5014e-01,  2.9382e-01,\n",
       "          -7.2478e-01, -2.6974e-01, -1.7688e-01,  7.9030e-02, -3.5031e-01,\n",
       "           4.1871e-01,  2.8606e-02,  4.4707e-02, -6.2274e-02,  1.2186e-01,\n",
       "           1.7288e-01,  8.6999e-01,  2.5057e-01, -8.6439e-01, -5.5986e-03,\n",
       "           3.8717e-01, -5.3427e-01, -1.0952e-01, -3.7453e-01,  3.6571e-01,\n",
       "           5.7549e-01,  1.1308e+00,  4.9560e-01]], device='mps:0'),\n",
       " tensor([[-2.6577e-01, -6.7066e-01, -6.0269e-01, -5.2367e-01, -7.9395e-01,\n",
       "          -2.4217e-01, -5.8013e-01,  8.3204e-01, -3.2099e-02,  4.9018e-01,\n",
       "          -2.9258e-02,  1.0862e-02,  2.3058e-01, -6.4151e-01, -4.3356e-01,\n",
       "          -5.4208e-01,  3.8364e-01,  4.1898e-01,  7.9644e-01,  5.1741e-01,\n",
       "           2.0445e-01,  2.3494e-01,  1.4122e-01,  1.3840e-01, -6.6261e-01,\n",
       "           3.5448e-01, -1.3099e-01, -1.2571e-01,  1.6891e-01, -6.6922e-01,\n",
       "          -1.1045e-01,  4.6788e-01, -3.2223e-01,  8.9012e-01, -4.7914e-01,\n",
       "           4.6155e-02, -3.1712e-02, -2.8916e-01,  6.2823e-01, -6.7831e-02,\n",
       "           6.5303e-01, -4.0231e-01,  2.4848e-01,  7.2275e-02,  1.6833e-01,\n",
       "           9.1796e-02, -4.3425e-02, -2.8453e-01, -3.0347e-01, -3.5952e-01,\n",
       "           3.0932e-01, -1.3373e-01,  5.6969e-01,  5.4862e-01, -2.2519e-01,\n",
       "          -4.3334e-01,  1.3249e-01,  3.2824e-01, -1.7524e-01,  1.1446e-01,\n",
       "          -1.8427e-01,  8.4005e-01,  4.2860e-01,  1.3079e-01, -5.9423e-02,\n",
       "           1.0415e-01, -4.8162e-01, -8.1482e-01, -5.1310e-01, -6.5730e-02,\n",
       "           3.6047e-02,  2.7320e-01,  3.7815e-02, -6.0810e-02,  3.9335e-01,\n",
       "          -1.2666e-01,  2.0458e-01, -6.1285e-03, -5.1121e-01, -4.2549e-02,\n",
       "          -2.5231e-02,  1.1568e-01, -2.0474e-01, -1.7414e-01, -1.8819e-01,\n",
       "          -6.2250e-02,  2.5652e-01, -1.4409e-02,  2.8027e-01,  1.0972e-01,\n",
       "           4.2771e-01,  3.2410e-02, -5.1681e-01, -3.5949e-01,  5.2593e-01,\n",
       "          -2.7955e-01,  1.1497e-01, -2.9371e-01, -2.6202e-02, -3.6122e-01,\n",
       "           3.3866e-01,  1.5197e-01, -3.0569e-01,  1.3341e-01, -5.4478e-01,\n",
       "          -6.9583e-01,  2.4319e-01,  2.2851e-01, -4.8444e-01,  1.6722e-01,\n",
       "           3.3344e-02, -1.6031e-01, -5.5857e-02, -6.2718e-01, -5.3883e-01,\n",
       "           1.2784e-03,  6.7855e-02, -2.6820e-01, -3.1000e-01,  5.1698e-01,\n",
       "           3.9158e-01, -5.9688e-01,  3.1140e-01, -8.8812e-02,  6.1117e-02,\n",
       "          -2.0423e-01, -3.0250e-05,  7.9284e-02,  1.6912e-01,  5.7687e-02,\n",
       "           1.0811e-01,  4.9054e-01, -1.8300e-01,  6.2718e-02, -4.2013e-02,\n",
       "           1.4345e-02,  2.3149e-01, -3.2055e-01,  7.6143e-02,  2.2232e-01,\n",
       "          -8.7417e-02, -7.1873e-02, -1.0607e-01,  2.0668e-02, -1.6659e-01,\n",
       "           3.7991e-02,  2.0268e-01, -8.9789e-03, -6.6323e-01, -1.8461e-01,\n",
       "          -7.4678e-01, -2.3481e-01,  3.3611e-01, -1.1283e-01, -9.9005e-02,\n",
       "           3.7754e-02,  7.4562e-01,  1.2646e-01,  8.6774e-02,  3.4975e-01,\n",
       "           5.8837e-01,  2.0594e-01,  6.8248e-02, -5.8820e-01, -3.2995e-01,\n",
       "           1.9779e-01,  9.8504e-02, -1.3047e-01,  8.8121e-02,  2.9305e-01,\n",
       "           2.7620e-02, -2.1300e-01,  2.0740e-01, -2.7784e-01, -1.0827e-01,\n",
       "          -5.3509e-01,  1.3256e-01,  2.9485e-01, -3.3479e-01, -3.5993e-02,\n",
       "          -1.3240e-01, -1.3372e+00, -5.3108e-02, -3.4137e-01,  8.3613e-02,\n",
       "          -5.0327e-01,  6.0982e-01,  2.7300e-01,  3.1635e-01, -7.3557e-02,\n",
       "          -1.6290e-01,  2.7820e-01,  2.9176e-01,  1.7498e-01,  5.0969e-02,\n",
       "           2.1994e-01,  4.3676e-01,  4.4166e-01, -1.9450e-01,  1.6846e-02,\n",
       "          -5.4098e-01,  2.3667e-01, -3.1968e-01,  1.0647e+00,  1.0185e-01,\n",
       "           7.9731e-02, -3.8121e-01,  2.4227e-02,  3.1342e-01, -1.2017e-02,\n",
       "          -1.6348e-01, -2.2721e-01, -6.5856e-01, -3.7263e-01, -6.2993e-02,\n",
       "           7.4196e-01, -7.7595e-02, -2.8579e-01,  2.4493e-01,  4.0710e-01,\n",
       "          -1.9265e-01,  4.4603e-02, -1.2753e-01, -1.8100e-01,  1.5554e-01,\n",
       "           2.8806e-01, -1.4137e-02,  1.8205e-01, -2.5511e-01, -8.0770e-03,\n",
       "           3.4269e-01, -4.4705e-01,  5.4238e-01,  4.6314e-01, -3.2183e-01,\n",
       "          -4.3166e-01, -5.2093e-01,  1.9282e-01, -6.8705e-01,  2.6217e-01,\n",
       "          -3.5447e-01, -5.1206e-01, -5.6272e-01, -1.7712e-01,  2.4896e-01,\n",
       "           4.0035e-01,  1.8628e-02,  9.4693e-01, -1.2650e-02,  3.3809e-02,\n",
       "          -3.0681e-01, -4.7693e-01,  3.2248e-01, -1.1295e-01, -2.5620e-01,\n",
       "           3.2035e-01, -9.7831e-02, -4.4641e-01,  3.1640e-01, -6.4410e-01,\n",
       "          -1.1664e-01, -1.8753e-01,  7.8657e-02,  4.8399e-01, -9.9686e-02,\n",
       "           1.7783e-01,  2.1578e-01, -8.0510e-01, -2.5621e-01, -1.9388e-01,\n",
       "          -2.5132e-01,  4.6474e-01,  3.2467e-01, -6.1096e-01, -7.1011e-01,\n",
       "          -4.8283e-01, -9.7773e-02, -1.0755e-01,  2.4790e-01, -2.6734e-01,\n",
       "           4.5662e-02, -2.6123e-01,  1.9452e-01,  2.7527e-01, -1.2212e-01,\n",
       "           1.5260e-03,  2.9099e-01, -8.4405e-02,  2.7017e-01, -5.9956e-01,\n",
       "           1.8518e-01,  1.1449e-01, -3.0167e-02,  2.2430e-01, -8.3686e-02,\n",
       "           1.3236e-01,  1.0978e-01,  4.6134e-01,  2.0513e-01, -2.8933e-01,\n",
       "          -1.5352e-01, -3.2550e-01,  3.4995e-01, -3.6565e-01, -4.3459e-01,\n",
       "          -4.7711e-02,  3.5511e-01,  6.5257e-01,  4.4807e-02, -5.3347e-01,\n",
       "           7.5460e-02,  5.1502e-01, -3.6013e-01,  9.0601e-02, -9.2147e-02,\n",
       "          -1.2709e+00,  3.8833e-02, -1.0731e-01,  2.9633e-01, -3.9028e-01,\n",
       "           2.5565e-01,  1.3043e-01,  1.6466e-01, -3.5638e-02, -2.1276e-01,\n",
       "           6.9708e-01,  1.9039e-01,  2.6710e-02, -1.3727e-01, -1.3940e-01,\n",
       "          -2.8256e-01,  2.5804e-02, -1.0290e-01, -3.5947e-01,  7.0172e-02,\n",
       "           1.0990e-01,  4.6378e-01,  6.4379e-02, -3.6316e-01, -4.8417e-01,\n",
       "           4.4059e-01, -1.8891e-01, -1.4405e-01,  4.8595e-01, -2.8493e-01,\n",
       "          -4.2529e-01,  6.0777e-01, -3.4337e-01, -9.3899e-02, -6.4431e-02,\n",
       "           4.5744e-01,  8.9369e-02, -3.8471e-01,  2.5955e-02,  5.6437e-01,\n",
       "           5.1503e-01,  5.0666e-01, -2.2672e-01,  6.2500e-01,  5.1622e-01,\n",
       "          -8.9276e-02,  5.8679e-01,  2.5398e-01, -7.5816e-02, -4.2325e-02,\n",
       "           1.8260e-01,  1.6374e-01, -7.2543e-01, -1.1433e-01,  4.0246e-04,\n",
       "          -1.3128e-01, -7.0790e-01,  2.3766e-01,  5.6609e-02,  4.0776e-01,\n",
       "          -4.8793e-01, -1.4436e-01,  4.0665e-01, -5.7769e-01, -3.4680e-01,\n",
       "           2.4221e-02, -4.3821e-03,  1.2940e-01, -3.0675e-01, -3.7269e-01,\n",
       "          -4.0293e-01,  6.9675e-02,  7.9701e-01,  2.4466e-01, -6.4683e-02,\n",
       "           2.4828e-02, -7.9401e-02, -1.3972e-01,  6.2973e-01, -2.2574e-01,\n",
       "          -3.1430e-01, -1.3445e-03, -2.4237e-01,  1.6706e-01,  2.7073e-01,\n",
       "          -4.3711e-01,  1.1011e-02,  2.6110e-01, -4.2501e-01, -1.9820e-01,\n",
       "          -2.5344e-01,  8.5044e-02,  4.6935e-01, -7.2148e-02, -1.1352e-01,\n",
       "           6.7548e-01,  2.8602e-01, -6.9660e-02,  1.7663e-01,  2.1237e-01,\n",
       "           4.0445e-01, -2.6559e-01,  1.5624e-01, -5.1111e-01, -8.7377e-01,\n",
       "           5.3150e-01,  7.2005e-01,  5.4706e-01,  2.0591e-01, -1.1175e+01,\n",
       "           3.7322e-01,  4.1664e-01, -4.7791e-01, -1.9567e-01,  9.9676e-02,\n",
       "           3.2591e-01, -3.0122e-01, -3.2804e-02,  1.1934e-01, -4.0327e-01,\n",
       "          -1.5991e-01, -2.4807e-01, -1.2486e-01,  2.0513e-02, -3.7656e-01,\n",
       "          -5.3536e-02, -7.8394e-01,  2.4330e-01, -8.0672e-02,  2.5201e-01,\n",
       "           4.1358e-01, -1.0220e-01,  1.7840e-01, -1.7484e-01, -1.7343e-01,\n",
       "           2.0152e-01,  8.9960e-02,  8.0500e-02,  2.9548e-02,  6.7712e-02,\n",
       "           1.9157e-01, -8.9118e-02,  5.5918e-02, -3.3189e-01, -3.0830e-01,\n",
       "          -5.2562e-03,  1.1318e-01,  3.1934e-01, -6.5457e-02, -3.6119e-01,\n",
       "          -6.2906e-01, -6.7328e-01,  2.6700e-01, -4.5591e-02,  1.0529e+00,\n",
       "           1.0097e+00,  7.8647e-01,  6.0274e-01, -9.3165e-02, -2.0951e-01,\n",
       "          -4.8630e-01, -7.6361e-02, -8.8681e-01, -8.0982e-02,  1.6731e-01,\n",
       "           5.2511e-01, -6.9737e-01,  3.0282e-01, -1.8856e-01, -4.0554e-01,\n",
       "           3.8569e-01,  1.9470e-01,  1.0201e-01, -2.3869e-01, -6.4130e-02,\n",
       "           5.3442e-01,  1.0711e-01,  2.7977e-01, -1.6475e-01,  3.6160e-01,\n",
       "          -3.2150e-01,  5.8289e-01,  3.7234e-01, -6.3904e-01,  1.7818e-01,\n",
       "          -9.0635e-02, -2.1180e-01,  1.3278e+00, -3.0041e-01, -1.2012e-01,\n",
       "           5.4107e-02,  5.7188e-01, -5.3807e-01, -1.0169e+00, -5.8696e-02,\n",
       "          -2.4581e-02,  1.0802e-01,  1.4291e-01,  4.0127e-01, -1.0486e-01,\n",
       "          -5.1071e-01, -8.3558e-02,  4.1296e-02,  3.3692e-01,  5.6283e-01,\n",
       "          -1.2076e-01, -2.1952e-01,  5.0071e-01,  3.8848e-01, -5.0421e-01,\n",
       "          -2.2573e-01, -3.4797e-01, -8.6861e-02,  6.8121e-01,  4.0053e-02,\n",
       "          -8.6708e-02,  1.3374e-01,  1.4264e-02,  3.5200e-01, -6.1100e-01,\n",
       "           4.1658e-01, -1.3831e-01, -1.1147e-01, -4.1585e-01, -4.4550e-01,\n",
       "          -1.0505e-02,  5.0333e-01, -2.5244e-01, -4.9965e-01, -3.0528e-01,\n",
       "           1.6144e-01, -1.5725e-01, -4.0185e-02, -4.8251e-01,  5.5208e-01,\n",
       "          -5.4562e-01,  2.7725e-01,  1.9485e-01,  4.5493e-02,  6.2006e-01,\n",
       "           2.5107e-02,  2.4289e-01,  2.8578e-01, -4.5061e-01, -8.1712e-01,\n",
       "          -5.4328e-01, -3.8697e-02, -2.0394e-01, -5.2693e-01,  2.8419e-02,\n",
       "           1.0722e-01,  9.6094e-02, -1.4164e-02,  2.2962e-01,  1.4459e-01,\n",
       "           1.3390e-01,  4.0953e-01, -5.8991e-02, -2.3596e-01, -2.2641e-01,\n",
       "          -4.7295e-01,  2.2112e-01, -2.6237e-02,  2.9580e-01, -4.6250e-01,\n",
       "          -2.9671e-01, -3.5487e-01, -3.0473e-01,  4.0594e-01,  4.8715e-01,\n",
       "          -1.4027e-01, -9.5728e-02, -2.1488e-02,  6.2293e-02,  4.2024e-01,\n",
       "           5.4895e-02, -1.9096e-01, -5.0192e-01,  6.3323e-02,  1.1889e-01,\n",
       "          -8.0940e-03,  9.0770e-02, -2.6138e-01, -6.9737e-02,  2.3377e-02,\n",
       "           1.2533e-01,  9.6679e-01, -3.3073e-01, -6.9147e-01,  5.2965e-01,\n",
       "           3.7989e-01, -8.7111e-02,  1.1756e-01, -2.7854e-01,  4.9200e-02,\n",
       "          -2.2252e-01, -6.3587e-01,  4.7724e-01,  3.4990e-01,  2.2054e-01,\n",
       "           9.0274e-01,  4.6096e-02, -1.0668e-01,  6.8278e-01, -3.4274e-01,\n",
       "          -3.7929e-01, -1.1005e-01, -2.9342e-01,  6.8626e-02,  9.0954e-03,\n",
       "          -6.3332e-02,  1.2438e-01,  1.9329e-01,  1.8541e-02,  6.3168e-02,\n",
       "          -4.6585e-03,  1.1260e+00, -2.1846e-01, -3.2172e-01,  3.6521e-01,\n",
       "           2.4638e-01, -5.3630e-03,  1.1626e-01,  4.3186e-01,  2.3325e-01,\n",
       "           5.3052e-01, -2.1105e-01, -1.2778e-01, -6.0343e-02,  3.5935e-02,\n",
       "           8.3296e-01, -7.6406e-01, -3.5916e-01, -3.0032e-02,  4.9807e-01,\n",
       "          -2.2894e-01,  3.8956e-01, -6.4853e-01,  1.3729e-01,  6.2192e-01,\n",
       "          -7.2654e-01, -1.5547e-03, -2.2163e-01,  4.7940e-01,  4.4890e-02,\n",
       "          -2.1394e-01, -1.7188e-01, -6.7165e-01, -2.3160e-01,  2.7364e-01,\n",
       "           4.6190e-01, -1.0329e+00, -1.1972e-01,  4.2358e-01, -4.7357e-01,\n",
       "           1.6055e-01, -8.9557e-02,  3.7264e-01, -3.5186e-02, -1.5740e-02,\n",
       "          -2.0304e-01,  2.2719e-01, -1.1154e-01, -8.0461e-01,  3.9726e-01,\n",
       "           6.9191e-01, -5.5472e-02, -4.9815e-01,  5.7366e-02,  1.9402e-01,\n",
       "          -1.0345e-01, -4.0704e-01, -2.8463e-01, -1.1969e-01, -3.5404e-01,\n",
       "          -8.0823e-02, -2.3650e-02,  2.2816e-01,  3.4273e-01, -3.6705e-01,\n",
       "           6.6874e-01,  3.8548e-01,  8.3540e-01, -2.0722e-01, -3.3004e-01,\n",
       "           3.3560e-01,  5.3225e-02, -2.4359e-01,  4.1678e-01, -1.8154e-01,\n",
       "          -4.7553e-01,  4.0074e-01,  7.8362e-02, -1.9746e-01,  3.3919e-02,\n",
       "          -2.3759e-01,  2.6375e-01,  1.1080e-01, -3.8346e-01,  3.7723e-01,\n",
       "          -1.9754e-02, -1.3963e-01, -6.7403e-01,  2.0370e-01,  2.1420e-01,\n",
       "          -4.0002e-02,  1.0644e-01, -9.4851e-02,  6.7336e-01, -2.3388e-01,\n",
       "           2.5111e-01, -1.1545e+00, -3.0301e-01, -1.4300e-01, -1.0336e-01,\n",
       "          -5.3599e-01, -1.3637e-01,  2.9602e-01,  4.0697e-01,  5.9500e-01,\n",
       "           1.9397e-01, -2.5805e-01, -6.5605e-01, -4.9590e-01, -1.7672e-02,\n",
       "          -8.8131e-02, -1.6129e-01,  6.0064e-02, -3.8866e-01,  2.5185e-01,\n",
       "          -4.6065e-01, -4.2120e-01,  1.4048e-01,  1.3256e-01, -1.7563e-01,\n",
       "           7.1610e-01, -1.3044e-01,  3.5248e-01,  4.1638e-01,  5.3227e-02,\n",
       "           2.1182e-01,  3.6420e-01,  9.8884e-02, -7.7726e-01, -3.4003e-01,\n",
       "           1.8109e-01, -2.8231e-03, -2.2150e-01, -8.7047e-02,  4.5254e-01,\n",
       "           4.3978e-02,  5.1896e-01,  6.5071e-01]], device='mps:0'),\n",
       " tensor([[-1.1418e-01, -4.2743e-01, -7.2260e-01, -6.3933e-01, -7.8022e-02,\n",
       "          -5.6085e-01, -9.7444e-01,  7.0255e-01,  4.8843e-01,  7.1184e-01,\n",
       "           2.5635e-01,  2.2949e-01,  1.9804e-01, -2.2429e-01, -1.2982e-01,\n",
       "          -3.8039e-01, -1.8212e-02,  3.6377e-01,  7.4515e-02, -4.0144e-01,\n",
       "           8.3199e-01,  4.0793e-01, -1.1195e-01,  1.8140e-01, -6.1533e-01,\n",
       "           9.5521e-02,  5.5987e-02,  1.5592e-01,  3.2368e-01, -7.7032e-01,\n",
       "          -1.9274e-01,  4.8107e-01,  5.2586e-02,  6.9033e-01, -9.0843e-03,\n",
       "          -1.9389e-01,  2.4976e-01, -1.2960e-01,  1.8592e-01,  6.6149e-02,\n",
       "           3.1164e-01,  1.5772e-01,  3.5226e-01,  6.8565e-02,  4.4235e-01,\n",
       "          -2.0117e-01,  2.6766e-01, -6.4730e-01,  1.7283e-01, -2.2068e-01,\n",
       "          -4.5338e-01, -2.6728e-01,  3.5508e-01,  1.2777e-01, -7.7063e-01,\n",
       "          -6.1651e-01, -7.2258e-02,  1.5742e-01, -4.7190e-01,  3.6978e-01,\n",
       "           1.1932e-01,  4.1916e-01,  3.0417e-02,  1.1317e-01,  3.9937e-01,\n",
       "           4.5465e-01, -5.8629e-01, -6.8149e-01, -4.9264e-01, -8.9501e-02,\n",
       "           1.0030e-01,  3.3130e-01, -2.4906e-01, -5.8570e-01,  2.5348e-01,\n",
       "          -1.2351e-01,  4.7854e-01, -5.3089e-03, -6.0739e-01, -8.3552e-03,\n",
       "           4.2371e-01,  1.4891e-01, -5.4971e-02,  1.9372e-02, -9.7926e-02,\n",
       "          -5.6371e-02,  3.1711e-01,  1.8591e-01, -1.1715e-01,  3.5059e-01,\n",
       "           8.1878e-01,  4.2754e-01,  3.1761e-02, -4.4933e-01,  1.8662e-01,\n",
       "          -1.3193e-01,  3.9146e-01, -1.3390e-01,  7.1522e-02, -1.1976e-01,\n",
       "           4.0129e-02,  3.1847e-01,  2.8331e-01, -3.0023e-01, -2.6693e-01,\n",
       "          -7.9652e-01, -6.3445e-02, -5.8767e-02, -5.2738e-01, -5.1041e-03,\n",
       "          -1.3288e-01, -3.9180e-01,  1.3404e-01, -5.8206e-01, -1.1228e-01,\n",
       "           3.7008e-01, -9.4421e-02, -4.9702e-01, -3.5256e-01,  5.7060e-01,\n",
       "           6.1035e-02, -2.0046e-01, -2.8722e-01, -5.2835e-02,  5.5335e-01,\n",
       "          -4.3384e-01,  5.7172e-01, -5.4222e-01,  3.7980e-01,  8.6242e-01,\n",
       "           1.5968e-01,  6.8082e-02, -3.4286e-01, -1.3683e-01, -2.2591e-01,\n",
       "           1.7988e-01,  1.1767e-01, -5.1410e-01,  6.6868e-02,  7.0048e-02,\n",
       "          -1.0259e-01, -1.3545e-01, -5.1185e-01, -3.2670e-01, -3.6276e-02,\n",
       "          -3.1234e-01, -2.3606e-01,  4.4991e-01,  1.3007e-03, -3.5474e-01,\n",
       "          -5.9774e-01, -4.7919e-01,  4.3388e-01, -4.4451e-02, -1.0801e-03,\n",
       "          -3.0292e-01, -1.0861e-01,  7.7864e-02, -1.4194e-01,  8.6647e-02,\n",
       "           8.0961e-01, -1.1594e-01,  3.2994e-01, -2.8057e-01, -1.3921e-01,\n",
       "           3.3787e-01,  2.4864e-01, -1.3246e-01, -1.8179e-02,  6.0183e-02,\n",
       "           3.1254e-01, -1.8515e-01,  6.9839e-02, -4.7296e-01,  5.1302e-03,\n",
       "           3.9153e-02,  5.4970e-01,  4.1528e-01, -1.2645e-01, -2.3431e-01,\n",
       "           8.8300e-02, -1.9227e+00,  7.8008e-01, -8.2620e-01, -5.4370e-02,\n",
       "          -4.8668e-01,  4.3606e-01,  2.8575e-01,  1.8621e-01,  1.1944e-01,\n",
       "          -1.7396e-01, -1.1376e-01,  2.6503e-02,  3.8158e-01,  9.4875e-03,\n",
       "           6.8358e-01,  5.9196e-01, -1.3851e-01,  1.0958e-01,  1.0254e-01,\n",
       "          -3.0793e-01,  4.5035e-01, -5.5995e-02, -4.8428e-01,  1.3838e-01,\n",
       "           2.4356e-01, -3.0038e-01,  1.2927e-01,  1.7494e-01,  8.9433e-03,\n",
       "          -5.0228e-02, -2.0711e-01, -3.3613e-01,  2.7038e-01, -2.9817e-01,\n",
       "           5.2912e-01, -2.3759e-01,  9.6168e-03,  3.7388e-01,  1.2453e-01,\n",
       "          -2.6596e-01,  6.3262e-02, -7.5493e-02, -2.3458e-01, -8.1702e-02,\n",
       "           5.0799e-01, -5.6874e-01,  1.8403e-01, -2.8042e-01, -2.0181e-01,\n",
       "          -1.0137e-01, -1.0480e+00,  4.7027e-01,  3.3153e-01,  7.0918e-02,\n",
       "          -5.6316e-02, -8.4896e-02, -1.3306e-01, -3.4439e-01, -2.4876e-01,\n",
       "          -5.1092e-01, -3.8222e-01, -4.1384e-01, -6.6754e-04,  2.2540e-01,\n",
       "           2.8167e-02, -3.9951e-02,  1.1024e+00,  4.6301e-01,  7.7434e-01,\n",
       "           1.5994e-01,  2.0158e-01,  5.3376e-01,  1.1136e-01, -2.2741e-01,\n",
       "          -8.8031e-02,  7.5215e-03, -2.0065e-01,  4.1026e-01, -3.0941e-01,\n",
       "          -2.1875e-01,  2.3748e-01,  4.0660e-01,  7.5155e-01,  4.0889e-02,\n",
       "          -9.1460e-02,  3.8390e-03,  1.1481e-01, -1.5783e-01,  6.3135e-01,\n",
       "          -2.7045e-01,  5.5682e-02, -1.9083e-01, -3.1005e-01, -7.2249e-01,\n",
       "          -4.5068e-01,  1.9485e-01, -2.6671e-01,  2.5773e-01, -1.1227e+00,\n",
       "          -3.0850e-01,  3.8312e-01, -1.0940e-02,  4.2779e-01,  2.0810e-01,\n",
       "           3.1011e-01,  2.0690e-01,  2.0710e-01, -2.0779e-02, -1.8910e-01,\n",
       "          -3.5013e-02,  2.8509e-01, -3.9585e-01,  4.3052e-01,  3.0796e-01,\n",
       "          -3.9720e-01,  2.0534e-01,  2.2160e-01,  1.2022e-01, -2.9145e-01,\n",
       "          -4.5417e-01, -3.2472e-01, -1.6361e-01, -1.5181e-01, -1.0703e-01,\n",
       "          -1.9928e-01,  4.3994e-01,  5.6727e-01,  2.7638e-02, -2.7430e-01,\n",
       "          -5.7596e-02,  5.6100e-01, -5.7774e-02,  2.6694e-01, -1.9232e-01,\n",
       "          -1.0586e+00, -4.4672e-01, -2.4890e-01, -1.1880e-01, -3.5309e-01,\n",
       "           2.5164e-01, -2.7543e-02, -1.4043e-01, -1.5893e-01, -3.3290e-01,\n",
       "           2.0233e-01,  6.0997e-02,  2.8627e-01, -1.6957e-01,  1.8614e-01,\n",
       "          -2.1409e-01, -2.6816e-01,  2.2424e-02, -2.3625e-01,  9.3003e-02,\n",
       "           2.4144e-01,  6.6493e-02, -1.5103e-01,  4.1886e-01, -2.3275e-01,\n",
       "           7.8724e-01, -5.5086e-01,  1.1249e-02,  4.7414e-01, -3.5539e-01,\n",
       "          -6.1377e-01,  2.1641e-01, -4.2110e-01,  1.1372e-01,  5.7735e-02,\n",
       "           2.0022e-01, -5.1700e-01, -3.7121e-01, -4.8988e-01,  8.0017e-01,\n",
       "           2.8766e-01,  5.5485e-01, -6.7970e-01,  3.5377e-01,  2.8934e-01,\n",
       "          -4.6455e-01,  1.9734e-01, -3.0868e-01, -1.8893e-01, -3.1992e-01,\n",
       "           3.7759e-01,  3.1319e-01, -5.2707e-01, -1.1038e-01, -1.7950e-01,\n",
       "           2.6453e-01, -2.2520e-01,  1.3450e-01,  3.4013e-01,  7.2350e-01,\n",
       "          -2.4565e-01, -1.8269e-01,  3.1211e-01, -5.1684e-01, -4.4199e-01,\n",
       "           1.2182e-01,  2.6149e-02, -1.6446e-01, -1.2462e-02, -3.3089e-01,\n",
       "          -3.1200e-01,  2.1527e-01,  3.2137e-01,  3.2873e-01,  5.4179e-03,\n",
       "           1.2921e-01, -2.1754e-01, -2.7522e-01,  3.8505e-01, -1.9795e-01,\n",
       "          -1.3386e-01,  4.5056e-01, -1.2003e-01, -4.4023e-02,  2.9940e-01,\n",
       "          -2.5893e-01,  2.7013e-01,  3.8679e-01,  5.9588e-01, -3.3450e-01,\n",
       "          -3.6672e-01,  2.2656e-01,  4.1868e-01, -1.7451e-02,  1.8606e-01,\n",
       "          -4.7954e-01,  8.4423e-01, -5.5263e-01, -3.6410e-01,  3.6787e-01,\n",
       "          -2.3068e-01, -1.5708e-01,  2.1586e-01, -1.9405e-01, -3.4027e-01,\n",
       "           2.5154e-01,  4.8401e-02,  1.7694e-01,  2.1754e-01, -1.1128e+01,\n",
       "           2.1036e-01,  6.3077e-01, -3.8542e-01,  4.4238e-02,  3.5218e-02,\n",
       "          -1.3279e-01, -2.8615e-01,  1.1089e-01, -4.5168e-01, -3.2237e-01,\n",
       "           5.9403e-03,  8.6440e-02,  8.7702e-03, -3.3251e-01, -2.6889e-01,\n",
       "          -1.2792e-01, -8.4641e-01,  2.0372e-01, -2.0529e-01,  2.6673e-01,\n",
       "          -3.6992e-01,  3.0577e-01,  4.0037e-01, -3.7710e-01, -7.9932e-02,\n",
       "          -2.8592e-01,  1.0338e-01,  3.7273e-02,  8.3935e-02, -2.9265e-01,\n",
       "          -2.2893e-01, -8.4321e-02,  1.8480e-01, -2.0552e-01, -6.8101e-01,\n",
       "          -9.2089e-02,  1.3163e-01,  4.3866e-01,  1.9581e-01, -6.1002e-01,\n",
       "          -2.7602e-03, -4.4350e-01,  3.8802e-01,  4.6439e-01,  6.2744e-01,\n",
       "           1.2380e+00,  4.9424e-01, -4.9526e-03, -4.5367e-02,  7.9707e-02,\n",
       "          -7.2045e-01,  3.4287e-01, -6.4450e-01, -7.9565e-02, -2.0868e-01,\n",
       "           5.8680e-01, -5.4928e-01, -3.4396e-02, -2.3011e-01, -1.7896e-01,\n",
       "           1.4167e-02,  2.6237e-02,  3.8453e-01, -1.8165e-01, -4.2770e-01,\n",
       "           8.4162e-01, -7.7598e-02,  1.9936e-01, -2.1547e-01, -2.0975e-01,\n",
       "          -4.6069e-01,  2.6226e-01,  2.6892e-01, -3.8940e-01, -6.5375e-03,\n",
       "          -9.8964e-02, -6.4932e-02,  9.1792e-01, -6.8857e-02,  2.0661e-01,\n",
       "           2.0010e-01,  3.5173e-03, -5.3219e-01, -4.3487e-01, -3.0295e-01,\n",
       "           7.8755e-02,  1.4023e-01, -7.8579e-03,  3.5031e-01, -2.8077e-01,\n",
       "          -6.7702e-01,  3.8307e-02,  7.3619e-02,  1.2486e-01,  5.8775e-01,\n",
       "          -4.8320e-01, -2.7477e-01,  1.2170e-01,  2.0798e-01, -6.3947e-01,\n",
       "          -1.9498e-01, -2.6607e-01, -3.4058e-01,  3.5853e-01, -9.3284e-02,\n",
       "           3.2092e-01,  6.5622e-02, -5.1130e-02,  5.2738e-01, -2.1840e-01,\n",
       "           3.7076e-01, -2.7750e-01,  4.2136e-01, -4.9311e-01,  7.2964e-02,\n",
       "           3.7248e-01,  8.0890e-01, -3.3219e-01, -6.0434e-01, -1.7028e-01,\n",
       "           1.8848e-01,  2.6653e-01,  1.5358e-01, -7.7681e-01,  2.7171e-02,\n",
       "          -4.5704e-01,  9.5779e-03, -2.7549e-02,  3.9952e-01,  6.7410e-01,\n",
       "           1.4607e-01, -3.1008e-02, -2.2989e-01, -3.4152e-01, -1.0948e-01,\n",
       "          -3.3062e-01, -1.8914e-02,  4.7147e-01, -5.0398e-01, -1.6211e-01,\n",
       "          -1.5793e-01,  1.3629e-01, -3.4561e-01,  3.3825e-01, -8.6527e-02,\n",
       "           5.0523e-01,  5.7888e-01,  3.0291e-02,  5.7838e-01, -2.0869e-01,\n",
       "          -3.3156e-01, -4.3820e-01, -8.5286e-02, -5.8063e-02, -4.7024e-01,\n",
       "           4.1879e-02,  1.2707e-01, -4.6827e-01, -1.0929e-01, -2.9483e-02,\n",
       "          -5.3709e-01, -3.2703e-01, -3.7821e-01,  5.4790e-02,  3.7660e-01,\n",
       "           2.6391e-02, -5.5826e-01, -4.7311e-02,  6.9194e-02,  2.0596e-01,\n",
       "           2.6573e-01, -2.6262e-01, -3.5280e-01, -7.9648e-01,  8.9418e-02,\n",
       "           1.4000e-02,  9.4462e-01, -3.6996e-01, -5.4305e-01,  2.8718e-01,\n",
       "           2.6548e-01, -7.1391e-02, -1.5647e-01, -9.2560e-02,  1.5106e-01,\n",
       "          -1.8011e-01,  7.5540e-02,  2.7130e-01, -1.2569e-01,  1.2237e-01,\n",
       "           7.8234e-02,  4.9238e-01, -1.4760e-01,  2.8264e-01,  1.2660e-01,\n",
       "          -5.8340e-01, -9.1271e-02, -1.1782e+00, -5.9068e-02, -5.5899e-02,\n",
       "          -2.2886e-01,  1.5079e-01, -1.6813e-01, -2.7420e-02,  3.1934e-02,\n",
       "          -1.0851e-03,  1.0644e+00,  1.3662e-01, -9.3327e-03,  5.3231e-01,\n",
       "           2.3226e-01, -1.7349e-01,  1.8365e-01, -2.3434e-01,  1.2572e-01,\n",
       "           2.7873e-01,  1.7760e-01, -5.2205e-01,  1.5771e-01,  3.5325e-01,\n",
       "           5.0884e-01, -1.1552e+00, -7.6919e-01, -2.5133e-02,  4.3937e-01,\n",
       "          -1.4434e-01,  9.6027e-03, -1.9234e-01,  2.6255e-01,  4.9446e-01,\n",
       "          -1.2182e+00, -3.1310e-01,  5.8438e-01,  5.0454e-01, -3.6039e-02,\n",
       "          -6.0419e-01, -8.2513e-02,  9.2000e-03, -4.3843e-01,  5.8603e-01,\n",
       "           4.8111e-01, -5.5660e-01,  3.0458e-01,  2.3874e-01, -3.7205e-01,\n",
       "          -7.9385e-02, -1.3196e-01,  2.7204e-01,  2.9828e-01,  1.8684e-01,\n",
       "          -6.0796e-01,  3.0275e-01, -6.7057e-01, -3.9097e-01,  2.3010e-01,\n",
       "           5.0302e-01, -3.9938e-02,  2.8966e-01, -1.1752e-01, -1.9297e-03,\n",
       "          -2.1381e-01, -2.1942e-01, -3.8699e-01, -1.1027e-01, -6.6579e-02,\n",
       "           3.9786e-01, -1.0672e-01,  2.4226e-01,  3.1450e-01, -4.0286e-01,\n",
       "           1.9907e-01,  5.6838e-01,  2.5180e-01, -7.1677e-02, -1.2031e-01,\n",
       "          -1.0406e-01,  7.5263e-02, -2.7059e-01,  6.2964e-01, -2.4138e-01,\n",
       "          -3.3699e-01,  4.3508e-01,  2.5083e-02, -4.4687e-01, -1.9116e-01,\n",
       "          -4.1389e-01,  1.4291e-01,  1.2167e-01, -5.4617e-01,  2.6297e-01,\n",
       "           2.9411e-01,  1.9805e-01, -5.1015e-01,  2.3947e-01,  2.9114e-01,\n",
       "          -1.2864e-01,  1.0273e+00,  5.8633e-02,  4.9283e-01,  7.2642e-03,\n",
       "           1.4251e-01, -6.0643e-01, -1.6881e-01,  3.0661e-01,  3.9877e-01,\n",
       "          -2.5154e-01, -3.0881e-02,  3.0149e-01,  6.5951e-02,  9.4433e-01,\n",
       "           5.6249e-01, -2.6987e-02, -6.3389e-01, -6.4409e-02,  1.3366e-01,\n",
       "          -5.3730e-02, -3.5430e-01,  2.2848e-01, -7.4755e-02, -1.3501e-02,\n",
       "          -1.0133e+00, -4.8762e-01,  4.7826e-02,  1.6941e-01, -1.8491e-01,\n",
       "           7.6992e-01, -1.0579e-02,  1.4110e-01,  5.9030e-01,  1.1791e-01,\n",
       "           1.9761e-01,  5.8025e-01,  2.9968e-01, -4.3029e-01,  2.6831e-01,\n",
       "           2.4331e-01, -2.3397e-01,  7.7202e-02, -6.1282e-01,  5.6888e-02,\n",
       "           1.4553e-02,  1.1538e+00,  4.0106e-01]], device='mps:0'),\n",
       " tensor([[ 0.0506, -0.1939, -0.2496,  ...,  0.0751,  0.8018,  0.7382],\n",
       "         [-0.0159, -0.2946, -0.5222,  ...,  0.3881,  0.5525,  0.3929]],\n",
       "        device='mps:0'),\n",
       " tensor([[-0.3538, -0.2121, -0.6888,  ..., -0.0446,  0.4974,  0.1687],\n",
       "         [-0.0111, -0.5057, -0.4670,  ...,  0.2365, -0.0712,  0.4388],\n",
       "         [-0.3025, -0.5761, -0.1930,  ...,  0.2660,  0.4446,  0.3191],\n",
       "         [ 0.0506, -0.6230, -0.6446,  ..., -0.0875,  0.4018,  0.3101]],\n",
       "        device='mps:0'),\n",
       " tensor([[-0.1959, -0.3163, -0.6089,  ..., -0.2600,  0.1989,  0.5143],\n",
       "         [-0.1682, -0.3560, -0.3949,  ..., -0.0628,  0.8556,  0.6643],\n",
       "         [-0.2632, -0.6292, -0.4726,  ..., -0.2028,  0.3527,  0.3565]],\n",
       "        device='mps:0'),\n",
       " tensor([[-2.1449e-01, -7.0454e-01, -6.7822e-01, -2.9824e-01, -2.3922e-01,\n",
       "          -8.0936e-02, -5.2339e-01,  7.3059e-01,  9.8530e-02,  5.0574e-01,\n",
       "           3.2821e-01,  2.0593e-01, -3.1108e-02,  3.5401e-02, -2.2698e-01,\n",
       "          -3.2096e-01,  2.1416e-01,  9.3440e-02,  8.6833e-02,  1.9313e-01,\n",
       "           6.3786e-01,  2.9887e-01, -2.8171e-01,  1.8532e-01, -6.6412e-01,\n",
       "           1.9387e-01, -3.1794e-01,  4.4021e-01,  3.4666e-01, -2.8586e-01,\n",
       "          -2.2759e-01,  5.0949e-01,  4.6054e-02,  7.5082e-01, -7.3993e-02,\n",
       "          -2.8442e-01,  4.0751e-01, -2.0799e-01,  1.4701e-01, -8.1190e-02,\n",
       "           5.0933e-01, -2.8916e-01,  1.9562e-01,  2.1962e-01,  2.0695e-01,\n",
       "           1.0814e-01,  2.3933e-01, -8.8199e-01,  1.6113e-01, -5.3111e-01,\n",
       "          -1.9534e-01, -1.3989e-01,  1.0564e-01,  4.0681e-01,  4.2414e-02,\n",
       "           7.0881e-02, -2.0309e-02, -2.2723e-03, -4.3205e-02,  1.2105e-01,\n",
       "           8.4445e-02,  6.4572e-01,  8.3931e-02, -1.9807e-01, -4.3443e-02,\n",
       "           4.2557e-01, -4.2430e-01, -4.1475e-01, -3.6487e-01, -9.9862e-02,\n",
       "          -1.3210e-01,  6.3424e-02,  2.7433e-02, -2.7986e-01,  3.0103e-01,\n",
       "          -1.0327e-01,  3.7408e-01,  1.1932e-01, -5.8556e-01,  5.9919e-02,\n",
       "           4.2993e-01,  4.3271e-01,  1.0163e-01, -5.6034e-02, -3.6994e-01,\n",
       "           1.3855e-01,  3.4047e-01,  2.0951e-01,  3.4301e-01,  4.8451e-02,\n",
       "           5.0535e-01,  4.4725e-02, -1.7061e-01, -3.7689e-01,  2.0207e-01,\n",
       "           4.4686e-02,  1.4184e-01, -2.7360e-01, -6.5874e-02, -2.6138e-01,\n",
       "           9.7153e-02,  2.5213e-01,  1.9122e-01,  5.5304e-03, -3.7332e-01,\n",
       "          -2.9791e-01,  2.7761e-02, -1.4923e-01, -5.0171e-01, -1.2071e-01,\n",
       "          -2.3499e-01,  1.3511e-01,  8.4198e-02, -2.3793e-01, -2.5759e-01,\n",
       "           2.7861e-01,  2.3122e-02, -6.1644e-02, -2.4986e-01,  3.5860e-01,\n",
       "          -7.1401e-03, -4.2117e-01,  1.6404e-01,  2.4549e-01,  6.3103e-01,\n",
       "          -1.0210e-01,  4.5199e-01, -2.7899e-01,  5.3807e-01,  2.6549e-01,\n",
       "          -1.3887e-01,  4.5990e-02, -2.7574e-01, -2.2970e-01,  5.9324e-03,\n",
       "          -3.4887e-02,  3.5351e-01, -2.2093e-01, -5.8900e-02, -2.2873e-01,\n",
       "          -4.1851e-02, -1.4381e-01, -3.3947e-01, -2.0282e-01,  2.1735e-02,\n",
       "          -1.7352e-01,  2.8433e-03,  2.4490e-02, -6.1198e-01, -3.4061e-01,\n",
       "          -8.1099e-01, -6.7827e-01,  2.4607e-01,  2.1824e-02, -1.8352e-01,\n",
       "          -2.4343e-02,  3.4273e-01,  3.2692e-01, -3.9045e-03,  1.5816e-01,\n",
       "           6.1919e-01, -5.5924e-02,  2.4366e-01, -6.5111e-01,  1.2130e-02,\n",
       "           1.3989e-01,  1.3852e-01,  1.0334e-02,  3.0207e-01,  2.7192e-01,\n",
       "           2.7881e-01, -4.8082e-01, -3.3521e-03, -5.9475e-01,  7.7579e-02,\n",
       "          -2.2848e-01,  4.0705e-01,  2.7719e-01, -3.6614e-01, -7.6351e-02,\n",
       "          -4.6902e-01, -2.3053e+00,  2.8320e-01, -6.4867e-01,  1.5134e-01,\n",
       "          -3.5205e-01,  3.3376e-01,  4.7742e-02,  2.9126e-01, -7.4618e-02,\n",
       "          -4.4197e-01, -2.0643e-01,  7.1511e-02,  2.5255e-01,  2.1678e-02,\n",
       "           3.5479e-01,  3.9292e-01,  1.1650e-01, -1.3211e-01,  9.5427e-02,\n",
       "          -3.9085e-01,  2.6911e-01, -1.5097e-01,  1.8862e-01,  3.2429e-01,\n",
       "           3.3884e-01, -2.6113e-01,  3.0993e-01,  5.3462e-01, -1.4194e-01,\n",
       "          -2.1112e-01, -2.1748e-01, -3.6181e-01, -1.1906e-01, -9.8641e-02,\n",
       "           6.1779e-01, -4.4257e-01,  1.1045e-01,  3.8594e-01,  2.7874e-01,\n",
       "          -1.8429e-01,  7.0850e-02, -7.9206e-02, -2.9913e-01, -7.1582e-03,\n",
       "           5.3529e-01, -2.5412e-01, -7.5643e-02, -4.0178e-01, -2.5430e-01,\n",
       "           1.0710e-01, -3.6585e-01,  5.4040e-01,  1.3650e-01, -2.7751e-01,\n",
       "          -3.0440e-02, -4.6726e-01,  2.6025e-02, -2.2961e-01,  2.7549e-02,\n",
       "          -3.4250e-01, -4.4381e-01, -6.9098e-01,  1.3356e-01,  5.8389e-02,\n",
       "           1.3915e-02, -2.6186e-01,  7.1536e-01,  6.1612e-01,  4.7437e-01,\n",
       "          -1.8139e-01,  1.6190e-02,  3.5486e-01,  1.8585e-01, -1.1704e-01,\n",
       "          -1.2458e-01, -1.2366e-01, -2.8551e-01,  4.3630e-01, -4.1739e-01,\n",
       "           1.3509e-01, -1.2989e-01, -3.9937e-02,  3.4343e-01, -4.1402e-01,\n",
       "           6.1919e-01,  1.1418e-01, -1.6619e-01, -2.3550e-01,  3.4553e-01,\n",
       "          -4.2821e-01,  3.4114e-01,  1.4150e-01, -4.2692e-01, -6.0903e-01,\n",
       "          -4.9354e-01,  3.9425e-01, -1.6426e-01,  1.1292e-01, -8.1100e-01,\n",
       "          -2.2976e-01,  3.3729e-01,  4.4767e-02,  3.4668e-01, -7.7773e-03,\n",
       "           2.5457e-01,  1.6300e-01,  7.4123e-02,  7.5676e-02, -5.9785e-01,\n",
       "          -2.9940e-02,  4.0458e-01, -3.4230e-01,  3.1484e-01,  3.2909e-01,\n",
       "           1.7037e-02,  1.3885e-01,  2.1392e-01, -3.3108e-02, -2.2602e-01,\n",
       "          -3.2048e-01,  2.2266e-01, -1.2187e-01, -1.5287e-01,  3.3797e-02,\n",
       "           1.1218e-01,  5.4054e-02,  5.2593e-01, -1.3887e-01, -3.5497e-01,\n",
       "          -1.4601e-02,  7.2089e-01, -1.4272e-01,  1.2341e-01, -7.0429e-02,\n",
       "          -1.3876e+00, -2.2261e-01, -2.0181e-01, -9.7845e-02, -4.2230e-01,\n",
       "           3.3472e-01, -2.1790e-03, -7.2537e-02, -5.0113e-02, -1.4296e-01,\n",
       "           8.4535e-02,  5.6148e-01,  3.1051e-01,  1.9252e-01,  1.4375e-01,\n",
       "          -1.8416e-01,  6.2568e-02, -3.0960e-01, -2.0726e-01,  1.7834e-02,\n",
       "           1.8351e-01, -7.7169e-02, -3.7460e-02,  1.1015e-02, -2.1886e-01,\n",
       "           9.6019e-01, -3.0926e-01,  3.8861e-02,  1.6496e-01, -3.5839e-01,\n",
       "          -3.3962e-01,  2.4846e-01, -1.9115e-01,  1.4696e-01,  9.0296e-03,\n",
       "          -1.4068e-02, -5.2225e-01, -3.3624e-01, -5.9971e-03,  6.5100e-01,\n",
       "           1.4027e-01,  4.8045e-01, -3.5575e-01,  2.5440e-01,  5.6965e-01,\n",
       "          -4.3151e-01,  1.4340e-01, -1.6650e-01, -1.5876e-01,  2.4645e-01,\n",
       "           1.4621e-01,  2.0576e-01, -2.3414e-01,  1.3722e-01,  1.8341e-02,\n",
       "           6.0046e-02, -2.3356e-01, -1.4233e-01,  1.5209e-01, -2.8139e-02,\n",
       "           9.4217e-02, -4.5048e-01,  9.6335e-02, -5.1419e-01, -5.1183e-01,\n",
       "          -1.8274e-02,  3.1351e-03, -8.5949e-03, -2.2676e-01, -5.1958e-01,\n",
       "          -4.8065e-01,  6.8147e-02,  6.5632e-01,  1.4340e-01,  1.8086e-01,\n",
       "           3.7948e-01,  6.8706e-02, -4.0554e-01,  3.1719e-01,  9.8486e-02,\n",
       "          -4.3327e-01,  4.1227e-01, -6.7402e-01,  3.9696e-01,  3.5643e-01,\n",
       "          -2.2426e-01, -6.0792e-02, -1.3265e-01,  4.5098e-01,  5.2386e-02,\n",
       "          -1.8944e-01,  3.0117e-01,  3.2255e-01,  1.2599e-01, -5.5095e-02,\n",
       "           5.8891e-03,  6.9732e-01, -4.1903e-01,  1.2872e-01, -2.3904e-01,\n",
       "           2.3916e-01, -1.9664e-01, -7.5440e-02, -1.5370e-01, -3.4435e-01,\n",
       "           2.1290e-01,  2.5904e-01,  1.5371e-01,  4.1539e-02, -1.1574e+01,\n",
       "           1.5778e-01,  6.4611e-01, -5.4654e-01, -2.7999e-02, -8.0666e-02,\n",
       "          -7.6214e-02, -2.5721e-01,  5.6782e-03, -3.6491e-01, -2.2660e-01,\n",
       "          -3.4259e-02,  9.2369e-02,  7.1069e-02, -3.6139e-01, -1.1223e-01,\n",
       "          -3.4103e-01, -8.1940e-01, -4.4931e-02, -1.3049e-01,  5.0144e-01,\n",
       "          -3.9520e-01,  2.1677e-01,  1.2436e-01,  1.1785e-02, -5.1300e-01,\n",
       "          -2.1669e-01, -4.2068e-01,  3.3411e-01, -3.3249e-02, -5.4580e-02,\n",
       "           4.6023e-02, -9.7767e-02,  1.4915e-01, -1.6366e-01, -4.5533e-01,\n",
       "           1.4732e-02,  3.2966e-01,  3.8739e-01, -2.4536e-01, -2.3814e-01,\n",
       "           1.3612e-01, -4.4475e-01,  5.6290e-01,  1.9171e-02,  6.7223e-01,\n",
       "           9.8843e-01,  8.5313e-01, -1.0504e-01,  7.1469e-02,  7.8757e-02,\n",
       "          -9.0512e-01,  3.9596e-01, -9.1779e-01,  1.0809e-01,  8.7403e-02,\n",
       "           6.8621e-01, -4.8684e-01, -2.1797e-01,  1.2465e-01, -3.3176e-01,\n",
       "           1.7581e-01,  4.7997e-03,  2.9667e-01, -1.4982e-01, -8.7584e-02,\n",
       "           4.8404e-01, -1.1318e-01,  2.5377e-01, -1.3684e-01, -4.1049e-01,\n",
       "          -5.2916e-02,  4.5801e-01,  1.4162e-01, -2.5695e-01,  2.2955e-01,\n",
       "          -5.5650e-02,  1.7324e-01,  5.9556e-01, -9.7477e-02, -1.5338e-01,\n",
       "           2.1133e-01,  3.7742e-01, -4.2344e-01, -6.2781e-01, -1.7225e-01,\n",
       "           1.9234e-02,  2.4492e-01,  9.7068e-02,  5.2181e-01, -2.9523e-01,\n",
       "          -3.4028e-01, -3.0628e-01, -3.9049e-03,  7.5488e-02,  7.9833e-01,\n",
       "          -3.5370e-01, -2.3046e-01, -1.6219e-01, -1.5390e-01, -5.4544e-01,\n",
       "          -3.2343e-01, -2.0785e-01, -5.7979e-01,  3.4363e-01, -1.7063e-01,\n",
       "          -2.1419e-01, -9.3807e-02,  1.9702e-01,  2.7729e-01, -3.0160e-01,\n",
       "           2.4914e-01, -7.0639e-02,  1.6772e-01, -3.6538e-01, -9.6589e-02,\n",
       "          -5.8679e-02,  8.8670e-01, -2.8509e-01, -4.5203e-01, -2.8266e-01,\n",
       "          -1.3650e-01,  3.4266e-01, -1.3321e-01, -3.6893e-01, -5.9663e-02,\n",
       "          -5.3281e-01, -1.0096e-01, -3.4001e-01,  1.2398e-01,  4.2333e-01,\n",
       "           3.4099e-01,  1.9882e-01,  4.1674e-01, -2.9492e-01, -5.9256e-01,\n",
       "          -6.1071e-01, -2.3221e-01,  1.7918e-01, -4.1660e-01,  2.3933e-02,\n",
       "          -1.2662e-01,  1.1810e-01, -4.8288e-01,  3.2500e-01, -3.4306e-01,\n",
       "           5.3225e-01,  1.4901e-01, -6.5404e-02,  9.4824e-02, -1.5454e-01,\n",
       "          -3.0315e-01, -3.1135e-01, -2.5577e-02, -2.1383e-04, -2.9308e-01,\n",
       "          -3.5529e-01,  3.4658e-01, -3.2742e-01, -1.9652e-01, -9.1171e-02,\n",
       "           5.8883e-03, -3.1505e-02, -3.3703e-01,  2.2127e-01,  1.5611e-01,\n",
       "           5.1888e-02, -2.8746e-01, -4.9281e-01,  6.7151e-02,  1.2024e-01,\n",
       "           1.7797e-01, -1.4067e-01, -1.7871e-01, -7.1177e-01, -2.1203e-02,\n",
       "           2.2721e-01,  1.2209e+00, -3.0134e-01, -1.5559e-01,  5.0491e-01,\n",
       "           3.8731e-01, -2.5203e-01, -7.9592e-02, -1.7318e-01,  1.6439e-01,\n",
       "           2.6809e-02, -5.3355e-01,  4.3414e-01, -3.6624e-01,  4.3651e-01,\n",
       "           2.0912e-01,  7.4561e-02, -2.5968e-01,  1.7785e-01,  1.3109e-01,\n",
       "          -4.3428e-01,  2.3924e-02, -3.1788e-01, -4.9670e-01,  1.8552e-01,\n",
       "          -1.7094e-01, -4.7952e-02, -7.0504e-02,  1.1502e-01,  4.8674e-01,\n",
       "           1.7448e-01,  1.4508e+00,  9.3435e-02,  1.1494e-01,  9.5490e-02,\n",
       "           1.5215e-01, -3.8269e-01,  2.0135e-02, -1.2861e-01,  4.0301e-01,\n",
       "           2.4194e-01,  1.6791e-01, -4.5340e-01,  1.6435e-01, -1.2027e-01,\n",
       "           3.3342e-01, -7.7834e-01, -6.2246e-01,  1.1594e-02,  6.2443e-01,\n",
       "           7.3555e-02,  1.8331e-01, -1.9643e-01,  6.3311e-01,  6.6714e-01,\n",
       "          -1.0752e+00,  3.6182e-02, -2.6472e-01,  1.9773e-01, -1.2163e-01,\n",
       "          -6.5773e-01, -2.4769e-01,  9.3695e-02, -4.6609e-01,  6.5418e-01,\n",
       "           6.5864e-01, -7.9031e-01,  6.3784e-02,  4.1052e-01, -2.2276e-01,\n",
       "          -4.2281e-02,  8.0290e-05,  1.7308e-01,  3.7529e-01, -9.3648e-02,\n",
       "          -2.7696e-01,  2.7165e-01, -3.1393e-01, -5.1805e-01,  2.9079e-01,\n",
       "           5.9012e-01, -9.5003e-02,  9.2033e-02,  1.9321e-01, -2.9840e-01,\n",
       "          -1.3768e-01, -3.9346e-02, -1.5622e-01,  1.6004e-01,  1.3544e-01,\n",
       "           2.4736e-02, -3.6580e-01,  1.4480e-01,  2.5389e-01, -8.7195e-02,\n",
       "           4.8912e-01,  8.0434e-01,  3.5971e-01, -2.4140e-01,  2.0208e-03,\n",
       "           4.2381e-02,  2.1808e-01, -4.1064e-02,  6.6852e-01, -3.0423e-01,\n",
       "          -3.2984e-01,  3.0238e-01,  1.6299e-01, -3.3704e-01,  1.9782e-01,\n",
       "          -2.3993e-01,  1.9140e-01,  7.6162e-02, -2.2715e-01,  3.9988e-01,\n",
       "           9.3421e-02,  5.2397e-02, -4.4329e-01,  2.4024e-01,  3.1145e-01,\n",
       "          -3.6736e-01,  7.6739e-01,  1.6882e-01,  4.0088e-01, -2.9505e-01,\n",
       "           5.1547e-02, -5.8944e-01, -4.2632e-01,  2.9509e-02,  3.8038e-01,\n",
       "          -3.8722e-01, -1.1970e-01,  4.4778e-01, -2.8456e-01,  6.0988e-01,\n",
       "           5.1692e-01, -4.1234e-01, -5.9516e-01,  1.4129e-01,  6.1326e-03,\n",
       "           4.3634e-02, -3.0439e-01,  1.5096e-01,  6.9082e-02,  3.4227e-02,\n",
       "          -8.0103e-01, -1.8143e-01,  1.2405e-01,  2.3414e-01, -2.6937e-01,\n",
       "           6.4494e-01, -6.1737e-02,  2.1208e-02,  2.6466e-01,  1.8926e-01,\n",
       "           1.3478e-01,  5.7465e-01,  1.9131e-01, -5.9506e-01,  8.7627e-02,\n",
       "           2.3120e-01, -2.3044e-01, -2.2299e-01, -4.1241e-01,  3.3408e-01,\n",
       "           1.9442e-01,  6.8585e-01,  6.6179e-01]], device='mps:0'),\n",
       " tensor([[-6.3660e-02, -7.0359e-01, -8.1569e-02, -8.8158e-01,  3.4994e-02,\n",
       "          -4.5083e-01, -6.3052e-01,  2.3072e-01, -5.7101e-02,  8.1769e-01,\n",
       "           1.2302e-01,  1.8531e-01,  4.5536e-01,  2.3235e-01, -6.4351e-01,\n",
       "          -2.3035e-01, -5.4593e-02,  3.8878e-01,  6.5353e-02,  4.1796e-01,\n",
       "           9.6914e-02,  5.7851e-01,  4.6976e-02, -1.5480e-01, -7.6759e-01,\n",
       "           1.6043e-01,  1.1824e-01, -1.7810e-01, -1.4366e-01, -2.8344e-01,\n",
       "          -3.8239e-02,  3.6106e-01,  4.2168e-02,  1.2776e+00, -1.9196e-01,\n",
       "          -3.7363e-01,  3.3735e-01,  4.3436e-02,  4.1075e-01, -8.2654e-01,\n",
       "           5.3940e-01, -1.5442e-01,  2.5672e-01,  1.4481e-01,  3.3940e-01,\n",
       "          -1.0255e-01, -1.3390e-01, -3.8077e-01,  5.0081e-01, -6.1703e-01,\n",
       "          -1.7319e-01, -2.7674e-01, -1.6836e-01,  3.1066e-01, -5.6382e-01,\n",
       "          -3.9552e-01,  1.2492e-01,  1.4951e-01,  1.6566e-02,  4.6492e-02,\n",
       "          -4.5406e-02,  1.1688e-01, -1.6788e-01,  1.2560e-01,  1.9688e-01,\n",
       "          -2.8222e-01, -6.3916e-01, -7.6971e-01, -7.5577e-01, -3.2710e-01,\n",
       "          -1.1027e-01,  1.7113e-01,  7.8641e-02, -2.7047e-01, -1.2197e-02,\n",
       "           3.2937e-02,  8.9621e-02, -3.8588e-01, -1.9903e-01,  3.8849e-03,\n",
       "          -6.6784e-02,  2.4118e-01,  2.6614e-01,  8.1746e-02, -3.5837e-01,\n",
       "          -1.4970e-01,  2.8098e-01,  1.8501e-01,  5.5596e-03,  2.7966e-01,\n",
       "           5.2068e-02, -2.1917e-01, -1.4163e-01, -2.9491e-01,  4.7070e-01,\n",
       "           5.5243e-02,  1.5085e-01, -4.7138e-01,  1.1693e-02, -2.3916e-01,\n",
       "           9.3257e-03, -1.2050e-01,  1.6442e-02,  3.2374e-01, -4.7997e-01,\n",
       "          -2.0281e-01,  1.1067e-01, -1.0554e-01, -4.8690e-01,  2.1337e-01,\n",
       "          -1.4453e-01, -5.2103e-01, -1.2767e-02, -5.8552e-01, -3.8935e-01,\n",
       "           1.9762e-02,  3.1461e-01,  4.2663e-01, -1.7660e-01,  5.3154e-01,\n",
       "           3.1281e-01,  1.2339e-01,  9.4934e-02,  2.0953e-01,  3.6854e-02,\n",
       "          -5.1513e-02,  7.4764e-01, -1.6507e-01,  8.0369e-02,  3.3128e-01,\n",
       "          -1.3148e-01, -4.6809e-02, -5.8858e-01, -2.1626e-01,  3.8663e-01,\n",
       "           1.4215e-01, -1.1366e-01, -2.7202e-01, -2.7648e-01, -1.0333e-01,\n",
       "           2.7028e-01,  2.8396e-01, -6.1453e-01, -4.0374e-01, -1.1232e-01,\n",
       "          -1.6071e-01, -1.1362e-01, -7.9456e-02, -2.9762e-01, -4.7183e-01,\n",
       "          -7.5342e-01, -5.2162e-01,  8.4199e-01,  5.0192e-02, -2.0557e-01,\n",
       "          -2.0806e-01,  5.4474e-01, -1.3480e-01,  3.3857e-01,  1.4177e-01,\n",
       "           8.8030e-01, -1.9183e-01,  8.1270e-02, -7.1075e-01, -5.8633e-02,\n",
       "          -2.4276e-01,  2.1721e-01, -4.5495e-01,  4.8535e-01,  3.0062e-01,\n",
       "           3.2224e-01, -1.8344e-01,  3.9356e-01, -4.2401e-01,  3.5388e-01,\n",
       "          -2.2512e-02,  2.3472e-01,  2.0299e-01, -3.4573e-01,  3.4537e-02,\n",
       "           9.7004e-02, -1.2335e+00,  2.0063e-01, -1.0631e-01, -5.4442e-01,\n",
       "          -2.8806e-01, -2.6192e-01,  1.7322e-01,  3.3236e-01, -3.4066e-01,\n",
       "          -1.5959e-01,  3.6463e-01,  1.4078e-01,  3.6416e-01, -2.1339e-01,\n",
       "          -2.1711e-01,  5.5270e-01,  7.7588e-01, -1.1373e-01, -9.3698e-02,\n",
       "          -2.1684e-01,  3.1208e-01, -1.7494e-01, -1.2096e-01, -2.1082e-01,\n",
       "           3.0381e-01, -3.2752e-01, -1.3203e-01,  9.1041e-02,  2.4047e-01,\n",
       "          -2.3222e-01, -3.8511e-01, -4.2054e-01, -1.1736e-01, -3.0369e-01,\n",
       "           4.7052e-01, -4.5912e-01, -1.4761e-01,  4.5544e-01,  3.8487e-01,\n",
       "          -1.7952e-01,  9.7157e-02,  4.0572e-01,  3.1537e-02,  2.4354e-02,\n",
       "           5.3066e-01, -1.6041e-01, -2.2142e-01,  1.5688e-02, -1.5091e-01,\n",
       "          -2.2892e-01, -8.4899e-01,  5.3104e-01,  1.2316e-01, -1.7696e-01,\n",
       "          -9.6078e-02, -2.3571e-01,  1.6756e-01, -2.3050e-01,  5.1070e-01,\n",
       "           1.1539e-01, -3.5134e-01, -6.6779e-01, -3.6547e-02,  7.0389e-04,\n",
       "           3.9037e-02, -7.5231e-02,  6.1386e-01,  7.4134e-01,  1.0742e-01,\n",
       "          -4.2193e-01,  1.9701e-01,  7.8800e-01,  3.3339e-01, -4.3084e-01,\n",
       "          -1.5554e-02,  1.4011e-01, -7.3580e-01,  2.3123e-01, -2.0928e-01,\n",
       "          -3.6079e-01, -9.6173e-02,  8.2354e-03,  4.7417e-01, -2.4306e-01,\n",
       "           1.0534e-01,  5.4846e-01, -2.3196e-01, -3.5417e-03,  4.2123e-01,\n",
       "          -4.6901e-01,  2.6778e-01, -3.0972e-02, -5.1125e-01, -5.2076e-01,\n",
       "          -5.3294e-01, -2.5733e-02,  5.3655e-01, -2.3351e-02, -6.0166e-01,\n",
       "          -1.0269e-01, -2.2004e-01,  1.4001e-01, -1.6176e-01, -4.8441e-01,\n",
       "           1.4930e-01, -1.7611e-01, -1.6430e-01, -4.6899e-01, -4.0677e-01,\n",
       "          -4.4721e-01, -8.3116e-02, -3.9755e-01, -4.8332e-02,  7.5163e-01,\n",
       "           3.1077e-01,  3.8663e-01,  3.5406e-01,  3.5435e-01, -2.6022e-01,\n",
       "          -1.7493e-01,  4.9319e-01,  1.3410e-01, -2.2369e-01,  5.9610e-01,\n",
       "           3.6790e-01,  4.2034e-01,  9.6918e-01, -3.8569e-01, -3.0960e-01,\n",
       "           9.2053e-02,  9.6189e-01,  4.5396e-03, -8.8250e-01, -1.0678e-01,\n",
       "          -1.5245e+00,  1.9568e-02,  2.5555e-01,  3.2868e-02, -4.2972e-01,\n",
       "          -5.9892e-02,  4.4172e-01, -4.0122e-02,  1.0152e-01, -1.6307e-02,\n",
       "           1.5665e-01,  6.5560e-01, -2.3379e-02,  4.0088e-03, -1.2796e-01,\n",
       "          -4.2723e-01, -7.0946e-02, -5.2741e-01, -9.8233e-02,  2.4980e-01,\n",
       "           3.0833e-01,  4.0444e-01,  1.6377e-02, -3.7455e-01, -1.3981e-01,\n",
       "           2.1881e-01, -2.4181e-01, -2.8994e-01,  3.3207e-02,  5.9594e-02,\n",
       "           1.8066e-01,  3.8795e-01, -2.9973e-01, -3.7632e-01,  2.7444e-01,\n",
       "           3.3607e-02, -2.8521e-01, -3.4620e-01, -1.3035e-01,  7.3802e-01,\n",
       "           3.2815e-01, -1.8282e-01, -9.0580e-02,  3.4452e-01,  9.7667e-02,\n",
       "          -3.3571e-01,  8.5954e-02, -4.8382e-02, -9.6304e-02,  5.7429e-01,\n",
       "           2.1334e-01,  4.0127e-01, -1.8178e-01,  8.4365e-02,  1.7224e-01,\n",
       "           3.1980e-02, -7.0960e-01, -1.0994e-01,  1.9835e-01,  5.1552e-01,\n",
       "          -2.7715e-01, -4.1003e-01, -6.2859e-02, -9.9340e-01, -3.6220e-01,\n",
       "          -2.6464e-02, -2.7296e-01,  2.8657e-01, -3.4627e-01, -5.9818e-01,\n",
       "           1.1096e-01, -3.6553e-01,  6.8905e-01,  1.3755e-02,  1.4406e-02,\n",
       "           5.4772e-01,  1.5788e-01, -2.6293e-01,  5.3009e-01,  1.2232e-01,\n",
       "          -2.4067e-01,  1.4507e-01, -7.4799e-01,  1.6170e-02,  2.1078e-01,\n",
       "          -3.0292e-01, -9.5226e-02,  4.2249e-01,  6.4682e-01,  2.8987e-01,\n",
       "          -3.3913e-01,  1.3462e-01,  1.8931e-01, -4.9877e-01, -7.5750e-03,\n",
       "           1.8786e-01,  2.6535e-01, -2.5993e-01,  3.8056e-01, -2.3739e-01,\n",
       "           1.7358e-01,  6.4413e-02, -1.5831e-01, -4.2978e-02, -5.3697e-01,\n",
       "           4.2401e-01, -1.1300e-01,  2.7115e-01,  4.0954e-01, -1.1155e+01,\n",
       "           1.5285e-01,  1.0540e+00, -1.2958e-01, -3.0586e-02, -2.5433e-01,\n",
       "           3.7116e-01, -4.4917e-01,  2.1879e-01, -1.7120e-01, -4.2764e-01,\n",
       "          -4.8415e-01,  6.0616e-02,  3.0047e-01, -3.7222e-01,  2.3391e-02,\n",
       "           9.1454e-02, -8.1570e-01,  2.2156e-01,  1.2135e-01,  2.2099e-01,\n",
       "          -6.2769e-02, -1.4374e-01,  1.0675e-01, -1.5945e-01, -1.4285e-01,\n",
       "          -1.7716e-01, -2.6887e-01,  2.5599e-01,  5.0296e-01, -8.9852e-02,\n",
       "           4.6643e-01,  1.4933e-01, -2.4000e-02, -2.5029e-01, -7.8371e-01,\n",
       "          -2.8378e-01, -1.9129e-01,  3.4727e-01,  1.7059e-01, -2.3286e-02,\n",
       "          -1.4311e-01, -4.0817e-01,  1.4556e-01,  3.1240e-01,  6.8233e-01,\n",
       "           7.6034e-01,  8.4683e-01,  4.1967e-01, -4.0086e-01, -1.2573e-01,\n",
       "          -8.5509e-01,  5.0727e-02, -7.5983e-01,  2.5698e-01, -1.8777e-01,\n",
       "           5.7542e-01, -5.5664e-01, -1.7341e-01, -2.6495e-01,  3.7254e-02,\n",
       "           3.9540e-01,  3.9478e-02, -2.3732e-01, -2.1306e-01, -1.8839e-01,\n",
       "           2.7860e-01,  3.6534e-02, -1.0641e-01, -6.4241e-01, -3.7915e-01,\n",
       "           7.2197e-03,  3.2114e-01, -2.5012e-02,  3.7460e-01,  5.1288e-01,\n",
       "          -1.5627e-01, -4.7910e-03,  1.7505e+00, -4.7461e-01, -1.2464e-01,\n",
       "          -2.7146e-01,  4.4714e-01, -1.5916e-01, -3.8716e-01,  7.2920e-02,\n",
       "           5.9804e-02,  5.7889e-01,  3.1373e-01,  7.5964e-01, -2.8989e-01,\n",
       "          -4.7750e-01, -2.6761e-02,  3.5545e-01,  1.0502e-01,  1.0012e+00,\n",
       "          -2.2541e-01, -3.7270e-01,  2.0248e-01,  5.6284e-01, -6.0411e-01,\n",
       "          -3.1328e-02, -3.4438e-01, -3.3282e-01, -2.6075e-01, -2.6158e-02,\n",
       "          -1.7032e-02,  5.1753e-02,  2.7162e-01,  5.7080e-01, -3.9706e-01,\n",
       "           3.8989e-01,  1.9004e-01, -1.8248e-01, -2.7241e-01, -8.7440e-01,\n",
       "           5.5249e-02,  7.9455e-01, -3.2301e-01, -7.5664e-01, -1.5510e-01,\n",
       "          -1.5542e-01,  5.0583e-02,  8.7218e-02, -2.5632e-01,  5.3853e-01,\n",
       "          -8.4337e-01, -1.4451e-01,  1.8345e-01, -4.6297e-01,  4.7605e-01,\n",
       "           4.1288e-01,  6.2445e-02,  3.4487e-02, -6.7027e-01, -2.1880e-01,\n",
       "          -8.5919e-01, -1.0029e-01,  5.7802e-01, -2.3881e-01, -1.5432e-01,\n",
       "           4.5230e-01, -1.2899e-01, -1.2159e-01,  1.8253e-01, -1.0790e-01,\n",
       "           5.7591e-02,  1.0994e-01, -1.3841e-01,  2.2322e-01,  1.6592e-03,\n",
       "           2.3268e-01, -5.9201e-01, -2.6329e-01, -1.2813e-01, -4.8841e-01,\n",
       "          -2.6965e-01,  1.4213e-01, -4.4177e-01,  1.3307e-01,  2.9584e-01,\n",
       "           3.1391e-01,  3.3033e-02,  4.4921e-02,  5.0014e-01,  2.9735e-01,\n",
       "          -1.0271e-01, -3.4976e-01, -5.8183e-02, -3.3582e-01,  1.2506e-01,\n",
       "           1.6146e-01, -2.2645e-01, -2.5220e-01, -4.4534e-01, -4.5335e-01,\n",
       "          -2.7473e-01,  1.2781e+00, -2.6287e-01, -3.3188e-01,  2.8896e-01,\n",
       "           4.4421e-01, -1.0096e-01,  1.9864e-01,  1.8132e-01,  3.1285e-01,\n",
       "          -3.0726e-01, -1.6866e-01,  5.1756e-01, -1.7423e-01,  2.3199e-01,\n",
       "           1.1482e-01,  4.3287e-02, -9.0979e-02,  7.7368e-02,  9.9691e-02,\n",
       "          -5.3722e-01,  1.6507e-01, -1.7745e-01,  2.0629e-01,  2.2885e-01,\n",
       "          -2.1714e-01, -2.4027e-01,  1.4243e-01, -2.7966e-01,  9.2641e-02,\n",
       "           1.4310e-01,  9.6745e-01, -5.6540e-02, -2.7167e-01,  6.4677e-01,\n",
       "           2.6552e-01,  1.4555e-01, -1.3182e-01,  2.0353e-01,  1.7392e-01,\n",
       "           3.9072e-01,  5.5611e-01, -5.9553e-01,  5.5038e-02,  6.4210e-02,\n",
       "           1.7920e-01, -4.9438e-01, -2.1127e-01,  3.5170e-01,  2.1303e-01,\n",
       "          -8.9566e-02,  3.2137e-01,  3.8199e-02,  2.5288e-01,  3.7599e-01,\n",
       "          -8.5903e-01,  8.6963e-02, -9.3896e-02, -4.3068e-01, -1.0781e-01,\n",
       "          -4.4690e-01, -3.1473e-01,  8.5380e-02, -4.2354e-01,  3.1221e-01,\n",
       "           5.4388e-01, -1.1241e+00,  1.4731e-01,  5.9514e-01, -4.0355e-01,\n",
       "           2.7757e-01,  9.2658e-02,  4.0264e-01,  1.9387e-01, -3.7439e-01,\n",
       "           6.6266e-02,  1.1098e-01, -4.2346e-01, -6.7246e-01,  2.0327e-01,\n",
       "           7.5403e-01,  1.1349e-01, -2.0150e-01, -2.1585e-01, -4.2441e-03,\n",
       "          -3.5713e-01, -3.9603e-01, -2.4378e-01, -1.6930e-02,  4.0383e-01,\n",
       "          -3.4128e-01, -1.2380e-01, -2.7945e-01,  2.1798e-01,  3.0289e-01,\n",
       "           3.6812e-01,  5.8350e-01,  7.8301e-01,  5.9435e-02, -2.6345e-01,\n",
       "          -2.1735e-01,  7.5820e-01,  6.5724e-03,  7.3939e-01, -1.1760e-01,\n",
       "          -3.3008e-01, -4.1604e-02,  1.1412e-02,  9.5438e-02, -2.5193e-01,\n",
       "          -1.5892e-01,  5.2617e-01, -3.5438e-02, -1.9908e-02,  4.3814e-01,\n",
       "          -1.5645e-01,  1.2825e-01, -4.9445e-01,  1.1428e-01,  3.3915e-01,\n",
       "           1.5803e-02,  5.6122e-01,  1.6382e-01,  4.6431e-01, -6.9501e-01,\n",
       "           8.6815e-02, -5.1235e-01, -1.1081e-01, -1.6269e-01, -2.9335e-01,\n",
       "          -4.1511e-01, -1.9989e-01,  1.9354e-01,  7.9081e-02,  5.2719e-01,\n",
       "           5.9205e-01, -1.5609e-01, -5.9961e-01, -1.8604e-01,  9.4833e-02,\n",
       "           4.2820e-01, -1.3857e-01,  6.8642e-01, -1.2007e-01,  1.5244e-02,\n",
       "          -7.4171e-01, -7.9910e-02,  5.1878e-02,  1.0624e-01, -6.0799e-02,\n",
       "           4.3445e-01, -1.4526e-01,  5.0166e-01,  9.7796e-02,  3.3521e-01,\n",
       "           3.5235e-01,  6.9826e-01,  1.4090e-01, -7.6857e-01, -1.9484e-01,\n",
       "           2.7389e-01, -1.6172e-01,  2.4140e-01, -8.3217e-01,  1.2475e-01,\n",
       "           2.5083e-01,  7.8436e-01,  4.3375e-01]], device='mps:0'),\n",
       " tensor([[-0.1773, -0.3860, -0.5648,  ..., -0.0519,  0.5404,  0.4747],\n",
       "         [ 0.1128, -0.5118, -0.5146,  ...,  0.3397,  0.9600,  0.5317]],\n",
       "        device='mps:0'),\n",
       " tensor([[-1.2362e-01, -4.4161e-01, -4.2594e-01, -4.8846e-01, -4.2322e-01,\n",
       "          -1.9079e-01, -2.8351e-01,  3.4717e-01,  2.8375e-02,  7.4096e-01,\n",
       "           1.1771e-01,  4.7781e-01, -7.7105e-02, -4.2840e-01, -4.9624e-01,\n",
       "          -4.5683e-01,  2.8039e-02,  7.7707e-01,  4.9250e-01,  3.9978e-01,\n",
       "           5.2396e-01,  1.9871e-01,  1.0537e-01, -1.0390e-01, -5.8287e-01,\n",
       "           1.2654e-01, -2.7832e-02,  2.9248e-01,  2.0339e-01, -6.9820e-01,\n",
       "          -1.3586e-01,  5.2963e-01, -2.1649e-01,  1.0466e+00,  1.1951e-01,\n",
       "          -1.9864e-01,  1.8975e-01, -7.5006e-01,  7.5075e-01, -2.6317e-01,\n",
       "           6.7402e-01,  9.5366e-02,  3.0554e-01, -4.1979e-03,  5.0998e-01,\n",
       "          -1.7571e-01,  3.2695e-02, -4.0198e-01,  3.6471e-01, -4.2541e-01,\n",
       "          -2.4349e-01,  7.7562e-02,  5.2744e-01,  4.0718e-01, -4.5064e-01,\n",
       "          -2.8855e-01,  1.5922e-01,  2.2306e-01, -5.0325e-01,  1.9024e-01,\n",
       "          -1.2244e-01,  4.3914e-01, -1.4508e-02, -2.9360e-01,  1.5641e-02,\n",
       "           3.0376e-01, -6.0714e-01, -5.4573e-01, -2.6109e-01, -1.0070e-01,\n",
       "          -1.2492e-01,  6.0714e-01, -1.5089e-01, -2.2262e-01,  3.0028e-01,\n",
       "           7.7276e-02,  1.1295e-01,  1.3524e-01, -4.7049e-01, -3.3357e-02,\n",
       "          -1.1913e-01,  3.5274e-01, -2.2357e-01, -2.4948e-01, -4.5172e-03,\n",
       "           1.3298e-01,  6.7389e-01,  8.0789e-02, -7.9157e-04, -2.5163e-01,\n",
       "           5.1488e-01, -9.2319e-02, -4.7280e-01, -4.9413e-01,  3.4393e-01,\n",
       "          -1.4415e-01,  4.5310e-01, -5.0005e-01, -1.8054e-02, -2.1360e-01,\n",
       "           1.7351e-01,  1.1631e-01, -8.2787e-02,  2.7429e-01, -3.5265e-01,\n",
       "          -7.8467e-01,  1.8565e-01, -6.8191e-02, -4.7824e-01, -1.2932e-01,\n",
       "          -3.9824e-01, -1.3825e-01,  3.4504e-02, -2.9597e-01, -4.0589e-02,\n",
       "          -1.8848e-01,  2.3414e-01, -1.0517e-01, -3.7891e-01,  1.9105e-01,\n",
       "           1.9004e-01, -6.0919e-01,  1.3650e-01,  2.0042e-01,  1.5433e-01,\n",
       "          -5.1449e-01,  6.9936e-01,  1.0222e-03,  3.0155e-01,  4.5256e-01,\n",
       "           1.0977e-01,  3.9636e-02, -4.6812e-01,  1.7348e-01, -2.5534e-02,\n",
       "           2.5873e-01,  5.4837e-02, -3.7838e-01,  3.1360e-01, -9.5610e-03,\n",
       "          -1.1348e-01, -1.2860e-01, -2.1607e-01, -1.5728e-01,  4.7659e-02,\n",
       "          -3.1920e-01,  3.5523e-01, -5.5618e-01, -4.5795e-02, -3.0301e-01,\n",
       "          -5.3354e-01, -7.3968e-01,  3.9309e-01, -1.6730e-01,  2.5321e-01,\n",
       "          -1.7651e-01,  2.7054e-01,  6.6379e-02, -2.2655e-01,  4.4772e-01,\n",
       "           7.0213e-01,  1.7464e-02,  1.5183e-01, -3.1088e-01, -2.1298e-01,\n",
       "           1.2678e-01,  2.2468e-02, -2.1325e-01, -4.6901e-01,  1.1148e-01,\n",
       "           1.9465e-01, -3.3396e-01,  3.4718e-01, -1.0010e-01,  2.6434e-01,\n",
       "          -5.9763e-01,  1.6206e-01,  1.6368e-01, -1.8200e-01, -1.9151e-01,\n",
       "          -1.4779e-01, -1.3319e+00, -6.1661e-02,  5.2320e-01, -1.7357e-01,\n",
       "          -7.3410e-01,  4.6414e-01,  2.0696e-01,  2.1565e-01, -3.8990e-02,\n",
       "          -4.2878e-02,  5.9639e-02,  2.3787e-02,  1.6792e-02, -1.4923e-02,\n",
       "           5.1795e-01,  7.8231e-01,  1.4885e-01,  1.6949e-01, -8.6254e-03,\n",
       "          -3.9116e-01, -5.4968e-02,  1.0129e-01,  1.0556e-01,  4.1179e-01,\n",
       "           2.2195e-02, -3.8455e-01,  5.0753e-02,  1.6459e-02, -2.2708e-01,\n",
       "          -1.9241e-01, -1.1437e-01, -4.5528e-01,  1.9939e-02, -7.1190e-02,\n",
       "           6.9919e-01, -3.9748e-02, -2.4711e-01,  4.7724e-01,  5.3357e-02,\n",
       "          -2.6376e-01,  2.2808e-01, -7.2615e-02, -3.5367e-01, -2.0370e-01,\n",
       "           6.1832e-01, -3.1667e-01,  7.8864e-02, -1.3505e-01, -2.3257e-01,\n",
       "           3.2501e-01, -4.9854e-02,  5.2459e-01,  3.7106e-01, -5.0439e-01,\n",
       "          -3.1091e-01, -1.5449e-01, -1.7601e-01,  1.7244e-03, -8.1001e-02,\n",
       "          -6.1868e-01, -5.1475e-01, -3.9610e-01, -5.7127e-02,  1.0073e-01,\n",
       "           3.1425e-01, -2.4544e-01,  6.3407e-01,  3.5798e-01,  2.6562e-01,\n",
       "           1.0749e-03,  9.0064e-04,  1.0437e-01,  6.2595e-02, -3.4269e-01,\n",
       "           7.9409e-02,  1.7769e-01, -6.0884e-01,  4.0430e-01, -2.5703e-01,\n",
       "           1.9740e-01,  6.8905e-02,  1.4838e-01,  3.5854e-01, -9.6939e-02,\n",
       "          -1.6210e-01,  2.0241e-01, -1.3825e-01, -3.3695e-01,  1.8850e-01,\n",
       "          -8.0994e-02,  5.2229e-01, -1.1805e-01, -4.2133e-01, -5.0459e-01,\n",
       "          -6.4410e-01, -1.9148e-01, -4.2678e-01, -4.0230e-02, -7.5637e-01,\n",
       "          -3.5088e-01,  3.1493e-02, -1.6721e-01, -8.8813e-02, -3.0654e-01,\n",
       "           2.1923e-01,  2.4463e-01,  2.6440e-01, -1.1052e-01, -3.7692e-01,\n",
       "          -2.9997e-01,  1.6449e-01, -2.0923e-01, -2.4015e-01,  1.5488e-01,\n",
       "           3.1832e-01,  7.0109e-02, -1.9254e-01,  2.8779e-02, -5.0702e-01,\n",
       "           2.2906e-01, -1.8052e-03,  1.1079e-01, -1.1267e-01,  7.4379e-03,\n",
       "           3.4816e-02,  2.4173e-01,  2.3723e-01, -2.9987e-01, -4.9188e-01,\n",
       "          -1.3341e-01,  3.1343e-01, -2.7642e-03, -3.9136e-01, -8.1230e-02,\n",
       "          -8.5074e-01, -7.1403e-02,  1.5230e-01, -1.0019e-01, -3.6463e-01,\n",
       "          -5.0672e-02,  4.0591e-01, -3.8825e-01,  8.2075e-02, -4.2358e-01,\n",
       "           5.2100e-01,  4.1775e-01,  2.3570e-01,  1.3671e-01, -5.8512e-01,\n",
       "          -2.5787e-02, -8.7629e-01,  2.4304e-02, -2.1285e-01, -8.5917e-02,\n",
       "          -3.4874e-02,  8.0745e-02, -1.5320e-02,  2.0129e-01, -2.6216e-01,\n",
       "           3.1495e-01, -3.2318e-01, -2.3917e-01,  2.7938e-01, -3.4672e-01,\n",
       "          -2.9286e-02,  4.2912e-01, -6.7098e-01, -1.1583e-01,  4.2381e-01,\n",
       "           1.4905e-01, -1.4762e-01, -7.7817e-02, -3.7704e-01,  6.1556e-01,\n",
       "           5.3769e-01,  2.7160e-01, -4.0760e-01,  5.5115e-01,  2.0987e-01,\n",
       "          -3.5661e-01,  3.0738e-01, -5.9964e-01, -5.9591e-01, -3.2575e-01,\n",
       "           2.0827e-01,  2.4226e-01, -5.8294e-01,  1.4114e-02, -1.5945e-01,\n",
       "           4.2296e-01, -1.0312e-01, -4.7253e-02,  1.0459e-01, -5.5817e-02,\n",
       "          -7.4256e-02, -3.6519e-01,  1.1194e-01, -5.7310e-03, -2.2627e-02,\n",
       "          -4.2614e-02,  6.3737e-02,  2.9268e-01, -2.8903e-01, -2.0238e-02,\n",
       "          -1.8814e-01, -1.0965e-01,  6.7555e-01, -8.7437e-02, -3.9670e-02,\n",
       "          -8.2721e-02, -8.1432e-02, -3.7938e-01,  5.5910e-01,  5.1646e-02,\n",
       "          -8.1738e-02,  5.4349e-01, -4.3953e-02, -4.7574e-02,  5.1090e-01,\n",
       "          -3.7293e-01,  2.0804e-01, -2.5747e-01,  4.1218e-01,  1.6214e-01,\n",
       "          -4.6497e-01,  4.8936e-02,  2.3551e-01,  5.2292e-02, -4.6707e-02,\n",
       "          -1.4628e-01,  6.4109e-01, -1.7297e-01, -7.8448e-02, -4.3875e-01,\n",
       "           1.7633e-01,  4.6082e-02,  7.9699e-02, -2.4066e-01, -5.9250e-01,\n",
       "           5.6618e-01,  1.9521e-01,  4.3532e-01,  4.8575e-01, -1.1463e+01,\n",
       "           8.8910e-02,  6.2774e-01, -4.7311e-01,  2.1503e-01, -1.0048e-01,\n",
       "           9.9219e-02, -2.9396e-01,  2.2937e-01, -4.0033e-01, -3.5375e-02,\n",
       "           2.2172e-02,  6.6639e-02,  1.3353e-01, -8.5681e-02,  8.9212e-02,\n",
       "          -3.5855e-02, -5.9645e-01,  9.9295e-02, -3.2557e-01,  3.3032e-01,\n",
       "          -5.0971e-01,  8.8723e-02,  2.4288e-01,  1.0307e-01, -6.8089e-02,\n",
       "          -5.2494e-02,  1.5514e-01,  5.1825e-01,  3.0650e-01, -1.1863e-01,\n",
       "           8.6938e-02,  5.1298e-02,  3.3254e-01, -6.4515e-02, -4.8763e-01,\n",
       "           1.7968e-01, -3.7176e-01,  5.5811e-02, -6.8456e-01, -3.1275e-01,\n",
       "          -4.0576e-02, -1.7293e-01,  4.5894e-01,  6.1073e-01,  5.4420e-01,\n",
       "           1.5649e-01,  6.6781e-01,  3.3169e-01, -4.6193e-01,  9.1118e-02,\n",
       "          -8.6136e-01,  1.4325e-01, -4.1524e-01, -8.7507e-01, -1.6877e-01,\n",
       "           5.9061e-01, -4.8274e-01,  4.1082e-01, -6.2086e-01, -2.9981e-01,\n",
       "           2.4618e-01, -1.0038e-02, -3.8618e-01, -5.7498e-01,  6.5998e-02,\n",
       "           1.3594e-01, -1.1328e-01,  1.0291e-01, -1.3564e-01, -1.6546e-01,\n",
       "          -1.3708e-01,  4.7517e-01,  1.9785e-01,  2.6347e-01, -1.7230e-01,\n",
       "          -1.3439e-01, -7.5749e-02,  2.0084e+00, -1.3643e-01,  1.4749e-01,\n",
       "           7.9998e-02,  5.3917e-01, -2.2341e-01, -8.2658e-01, -2.3068e-01,\n",
       "          -9.1236e-02,  3.7423e-01,  1.4851e-01,  3.1412e-01, -1.1075e-01,\n",
       "          -6.7238e-01, -1.6019e-01,  3.1632e-01,  2.4496e-01,  1.0424e+00,\n",
       "          -7.0037e-01,  1.4779e-02,  1.9388e-01, -1.5957e-01, -1.0233e-01,\n",
       "           1.6848e-01, -2.9529e-01, -3.1584e-01,  7.3522e-02,  1.8033e-01,\n",
       "           1.6324e-01,  1.4523e-01,  2.7007e-01,  1.9575e-01, -1.3369e-01,\n",
       "           3.4617e-01,  1.4309e-01,  6.8409e-02, -1.2560e-01, -1.5601e-01,\n",
       "           4.4899e-01,  5.0313e-01, -2.3389e-01, -2.4515e-01, -1.8403e-01,\n",
       "           1.2826e-02,  1.9899e-01,  1.0825e-01, -2.7132e-01,  6.0154e-01,\n",
       "          -5.6972e-01, -1.4141e-01, -2.6681e-02, -2.0985e-01,  3.1444e-01,\n",
       "           3.7791e-01, -3.4831e-01, -5.9645e-02, -6.9473e-01, -4.8519e-01,\n",
       "          -7.2762e-01, -5.7397e-02,  2.0914e-01,  6.1532e-02,  1.7627e-01,\n",
       "          -4.7663e-01,  5.3509e-02, -1.9895e-01,  4.6150e-01,  7.4136e-03,\n",
       "           5.3349e-01,  2.1281e-01, -1.5152e-01,  5.8642e-01, -1.7293e-01,\n",
       "          -4.8366e-01, -1.1333e-01, -4.1451e-01,  6.8887e-03, -4.8906e-01,\n",
       "          -3.6374e-01, -1.9666e-01,  1.7241e-01,  8.4415e-03, -2.3495e-02,\n",
       "          -1.6489e-01,  2.5351e-01, -1.2794e-01,  3.6217e-01,  2.1569e-01,\n",
       "           2.2702e-01, -2.3550e-01, -1.0826e-01, -1.3872e-01,  3.1339e-01,\n",
       "          -1.5533e-01,  3.6394e-02, -1.3942e-01, -2.2399e-01, -1.6412e-01,\n",
       "          -3.9072e-02,  9.1531e-01, -4.5627e-02, -4.1597e-01,  2.5863e-01,\n",
       "           8.8458e-01, -2.5246e-01,  2.9726e-02, -2.9713e-02, -1.9727e-01,\n",
       "          -3.5403e-01, -2.5689e-01,  4.0969e-01, -1.5060e-01,  4.9454e-01,\n",
       "           1.3017e-01, -1.3816e-01, -4.2630e-01,  4.7363e-01,  3.8057e-01,\n",
       "          -3.9106e-01, -1.4303e-01, -4.3541e-01, -1.7435e-01,  6.4928e-01,\n",
       "          -3.7715e-02, -2.0598e-01,  2.9341e-02, -1.5940e-01, -2.5000e-01,\n",
       "           4.3768e-01,  6.0109e-01, -2.0140e-01, -1.5229e-01,  1.0535e+00,\n",
       "           3.4829e-02,  2.6993e-02,  4.9015e-01,  4.2173e-01,  2.5088e-01,\n",
       "           5.0914e-01, -3.9050e-02, -3.8989e-01, -1.3289e-01,  2.1062e-02,\n",
       "           3.2393e-01, -7.9569e-01, -4.9901e-01, -1.0025e-02,  9.9389e-02,\n",
       "           1.2067e-01,  6.5324e-01, -3.0071e-01,  3.5033e-01,  6.5359e-01,\n",
       "          -1.0288e+00, -1.2947e-02, -3.5503e-01,  8.9863e-02,  2.1119e-01,\n",
       "          -1.1802e-01,  4.9682e-02, -4.4418e-01, -1.8470e-01,  3.7290e-01,\n",
       "           3.3845e-01, -7.8286e-01, -1.2387e-01,  5.9063e-01, -2.9397e-01,\n",
       "           1.8309e-01, -1.5119e-01,  2.4356e-01,  1.5649e-01,  1.0039e-02,\n",
       "          -1.7725e-01,  2.2726e-01, -2.2402e-02, -3.4803e-01,  2.1471e-01,\n",
       "           5.2139e-01,  6.6893e-02,  9.9387e-02,  2.3831e-01, -1.9443e-01,\n",
       "          -1.3103e-01, -2.6745e-01, -1.4768e-03, -7.0968e-02, -3.5468e-01,\n",
       "           2.5932e-01, -3.0700e-01, -3.8582e-02,  3.3695e-01, -3.8985e-01,\n",
       "           3.0322e-01,  7.4674e-01,  7.9527e-01,  1.6567e-01, -4.3244e-01,\n",
       "           3.1414e-01,  1.8535e-01, -1.3248e-01,  5.7798e-01,  1.2236e-01,\n",
       "          -4.1227e-01,  1.3922e-01,  5.4724e-02, -3.2476e-01,  7.0525e-02,\n",
       "          -1.8109e-01,  8.1571e-02, -1.2600e-01, -5.6131e-01,  2.0046e-01,\n",
       "           6.7197e-02,  1.5775e-01, -5.7764e-01,  9.5378e-02,  4.7205e-01,\n",
       "          -6.0370e-02,  4.6158e-01, -9.8349e-02,  2.2574e-01, -2.6197e-01,\n",
       "          -8.7918e-02, -6.3920e-01, -2.2465e-01,  3.4741e-01, -3.2252e-01,\n",
       "          -5.4661e-01,  3.2818e-02,  8.0396e-02, -1.7726e-01,  5.0332e-01,\n",
       "           7.0652e-01, -3.3488e-01, -1.0958e+00, -3.3693e-01,  7.8758e-02,\n",
       "           7.4292e-02,  8.4491e-02,  4.3285e-01,  4.6636e-02,  2.4779e-01,\n",
       "          -9.0042e-01, -2.1914e-01, -3.1685e-02,  3.6246e-01, -2.6203e-01,\n",
       "           5.8522e-01,  6.6670e-02,  1.0128e-01,  2.8806e-01,  4.4771e-01,\n",
       "          -9.5737e-02,  6.7974e-01, -2.2059e-01, -2.0277e-01,  3.9234e-02,\n",
       "           2.9768e-01, -1.3995e-01,  9.6168e-02, -9.3203e-01,  4.3953e-01,\n",
       "          -1.0001e-01,  6.5424e-01,  4.8821e-01]], device='mps:0'),\n",
       " tensor([[-2.1745e-01, -5.6520e-01, -7.5059e-01, -5.7110e-01, -7.6522e-02,\n",
       "          -7.4041e-02, -5.5838e-01,  5.1191e-01, -2.4243e-01,  1.6206e-01,\n",
       "           2.6668e-01,  3.6576e-01, -1.3485e-01, -4.1383e-02, -5.3481e-01,\n",
       "          -3.0616e-01,  2.0818e-01,  2.8939e-01,  4.7466e-01,  4.4043e-01,\n",
       "           6.3952e-01,  3.0220e-01, -3.0084e-01,  2.5198e-01, -4.8569e-01,\n",
       "           2.0705e-01, -3.4708e-01,  1.2436e-01,  4.3443e-01, -4.0824e-01,\n",
       "          -1.5099e-01,  7.3836e-01, -9.0918e-02,  8.5803e-01, -2.0163e-01,\n",
       "          -1.6923e-01,  2.3076e-01, -2.1181e-01, -3.8958e-02, -2.5433e-01,\n",
       "           3.4860e-01,  8.0505e-02,  7.7759e-02,  3.3254e-01, -3.8954e-02,\n",
       "           1.5907e-01,  2.5675e-01, -9.6393e-01, -1.0435e-01, -2.8264e-01,\n",
       "          -2.1397e-01, -1.8032e-01,  2.5092e-01, -1.0187e-01, -7.0325e-01,\n",
       "          -5.7739e-01,  2.8475e-01,  1.4103e-01, -3.0590e-02, -1.1605e-01,\n",
       "           1.5257e-01,  5.2724e-01,  6.0889e-02,  6.8423e-02, -1.5478e-02,\n",
       "           1.2244e-01, -5.5165e-01, -5.4819e-01, -4.6793e-01, -6.1637e-02,\n",
       "          -1.0104e-01,  7.3611e-02, -1.0074e-03, -3.1666e-01,  3.0920e-01,\n",
       "           5.5072e-02,  2.8887e-01, -3.0122e-01, -7.4641e-01,  4.0302e-01,\n",
       "           2.8677e-01,  2.3409e-01,  9.9047e-02, -1.3266e-01, -3.8821e-01,\n",
       "          -1.7033e-01,  4.1300e-01,  3.9567e-01,  2.2536e-01,  2.2280e-01,\n",
       "           1.9833e-01,  1.6927e-02, -3.2232e-01, -1.9818e-01,  8.8054e-03,\n",
       "          -2.2575e-01,  1.3488e-01, -1.4178e-02, -1.9602e-02, -9.8765e-02,\n",
       "          -9.6850e-02,  2.4077e-01,  3.2805e-01, -6.0579e-02, -6.9014e-01,\n",
       "          -2.0912e-01,  1.0781e-01, -2.9601e-01, -7.2982e-01, -2.4519e-01,\n",
       "          -2.9261e-01, -1.0070e-02,  1.7472e-01, -6.6424e-01, -2.5405e-01,\n",
       "           1.0792e-01,  5.7594e-01, -2.4237e-01, -3.3824e-01,  2.2690e-01,\n",
       "           5.6440e-02, -3.1550e-01,  1.2658e-01,  1.3179e-01,  4.3400e-01,\n",
       "          -1.2841e-01,  4.4688e-01, -5.0698e-01,  6.7741e-01,  3.6616e-01,\n",
       "          -8.2233e-02,  1.6668e-01, -1.9507e-01, -3.4258e-01,  2.8348e-01,\n",
       "          -1.8564e-01,  2.5678e-02, -1.8334e-01,  1.7393e-03,  8.9615e-03,\n",
       "           1.4683e-01, -2.0637e-01, -3.0617e-02, -2.7332e-01,  2.0545e-02,\n",
       "          -1.8017e-01, -2.9831e-01, -1.5906e-01, -1.9587e-01, -5.7116e-01,\n",
       "          -1.0868e+00, -7.5402e-01,  6.8264e-02,  7.2178e-02, -1.8490e-01,\n",
       "          -6.3222e-02,  3.5698e-01,  6.0692e-02,  2.0739e-01,  2.0563e-01,\n",
       "           8.5973e-01,  2.8933e-02,  1.2508e-01, -3.8017e-01, -1.3813e-01,\n",
       "           2.2650e-02,  2.0334e-01, -8.6772e-02,  4.1550e-01,  1.2895e-01,\n",
       "           3.5262e-01, -4.2346e-01,  3.2866e-01,  6.8316e-02,  2.7526e-01,\n",
       "          -8.7462e-01,  5.4732e-01, -1.4459e-01, -1.7908e-01, -1.5121e-01,\n",
       "          -5.8314e-01, -1.7616e+00,  5.1833e-01, -1.3990e-01,  3.1989e-02,\n",
       "          -4.4485e-01,  1.8602e-01,  3.6194e-01,  1.9066e-01, -4.2955e-01,\n",
       "          -6.1830e-02, -1.7965e-01,  1.3316e-01,  3.2204e-01, -2.0543e-01,\n",
       "           2.4869e-01,  4.1786e-01,  4.4021e-01, -2.4843e-02, -9.1083e-02,\n",
       "          -5.7031e-01,  1.9304e-01, -4.6694e-02,  1.0346e-01,  7.5935e-04,\n",
       "           2.8705e-01,  1.0752e-03,  2.9222e-01,  4.9786e-01,  1.1391e-01,\n",
       "           1.9176e-01, -2.3951e-01, -8.0457e-02,  3.0634e-02, -6.1324e-02,\n",
       "           5.7986e-01, -5.1853e-01,  8.5581e-02,  2.6102e-01,  2.3538e-01,\n",
       "           6.6485e-02, -3.4192e-01, -8.5762e-02, -1.9081e-01,  1.3350e-01,\n",
       "           5.4392e-01, -3.8009e-01, -9.1318e-02, -4.7097e-01,  7.7690e-02,\n",
       "           3.5498e-01, -6.6848e-01,  3.4207e-01,  6.2809e-01,  4.2753e-03,\n",
       "          -1.3269e-02, -3.8189e-01,  3.4672e-01,  3.7599e-02,  5.2111e-01,\n",
       "          -2.0681e-01, -5.0772e-01, -7.6612e-01,  3.3087e-01, -5.3521e-02,\n",
       "           5.4238e-01, -3.0100e-01,  5.5218e-01,  6.1158e-01,  1.9056e-01,\n",
       "           2.6553e-01, -4.5840e-02,  1.7666e-01,  3.0890e-01, -1.2057e-01,\n",
       "          -4.6094e-01, -1.9637e-01, -2.4746e-01,  6.1274e-01, -3.4301e-01,\n",
       "          -3.3029e-01,  1.2237e-01, -8.9119e-02,  2.0925e-01, -2.4478e-01,\n",
       "           6.2987e-01,  5.2543e-02,  8.4803e-02, -6.3611e-02,  5.9161e-01,\n",
       "          -6.4275e-01,  2.2753e-02, -2.0051e-01, -1.6333e-01, -5.1427e-01,\n",
       "          -2.8289e-01,  1.0362e-01,  1.2056e-01,  4.5459e-01, -6.1239e-01,\n",
       "          -6.8984e-01, -1.0315e-03,  2.4709e-01,  3.7588e-01, -1.6922e-02,\n",
       "           4.8252e-01, -1.3569e-02, -3.0843e-01,  1.4705e-01, -4.9152e-01,\n",
       "          -3.7260e-01,  4.1489e-01, -3.2005e-01,  1.2605e-01,  2.4713e-01,\n",
       "          -3.8452e-02, -1.0601e-01,  8.5975e-02, -1.5595e-01, -3.1396e-01,\n",
       "           3.6969e-03, -6.7641e-02,  7.8454e-03,  1.7047e-01,  2.6648e-01,\n",
       "          -3.2843e-02,  3.6948e-01,  6.2059e-01, -4.0449e-01, -4.1059e-01,\n",
       "          -1.6771e-01,  4.6376e-01, -1.7396e-01,  5.6286e-02, -1.3038e-01,\n",
       "          -1.3945e+00, -1.5088e-01, -3.2211e-01, -1.0146e-02, -1.4522e-01,\n",
       "           3.9443e-01,  1.9094e-01,  1.2967e-01,  6.1696e-02, -9.4473e-02,\n",
       "           3.5131e-01,  4.4414e-01, -4.0887e-02,  8.2195e-02, -1.7617e-01,\n",
       "          -1.4101e-02, -2.1743e-01, -2.0207e-01, -1.8773e-01,  8.8400e-02,\n",
       "           3.4036e-01, -8.0080e-02, -5.0949e-02, -1.9487e-01, -1.0630e-01,\n",
       "           1.1070e+00, -2.3085e-01, -2.6895e-01,  2.5413e-01, -5.0215e-01,\n",
       "          -3.0356e-01,  2.2885e-01, -5.6430e-01,  1.8081e-01, -3.3625e-02,\n",
       "           1.6488e-01, -2.8580e-01, -9.2728e-02,  8.4375e-03,  6.7032e-01,\n",
       "           1.6526e-01,  3.8402e-01, -1.8840e-01,  3.2847e-01,  5.1674e-01,\n",
       "           7.6042e-02,  2.5358e-01, -2.9382e-01, -9.9473e-02,  3.1699e-01,\n",
       "          -2.0941e-01,  6.0804e-01, -1.8910e-01,  2.3705e-01,  2.8677e-01,\n",
       "          -2.7170e-04, -5.7905e-01,  2.6774e-02,  1.3586e-04,  2.6888e-01,\n",
       "           4.2141e-02, -3.5957e-01,  2.6734e-01, -5.4077e-01, -4.8106e-01,\n",
       "          -1.4466e-01,  3.6980e-02,  9.5158e-02, -5.2314e-02, -1.0669e-01,\n",
       "          -2.8493e-01,  1.1333e-01,  3.9720e-01,  1.0173e-01, -1.3118e-01,\n",
       "           3.8612e-01, -1.4785e-03, -5.0668e-01,  3.6000e-01,  6.4563e-01,\n",
       "          -1.4902e-01,  3.1416e-01, -3.0213e-01,  3.4628e-01,  2.4229e-01,\n",
       "          -2.2951e-01,  4.5759e-02,  1.3037e-01, -5.8146e-01,  3.9093e-01,\n",
       "          -2.8732e-01,  1.1138e-01,  4.1298e-01,  7.9815e-03, -1.1148e-01,\n",
       "          -4.3015e-01, -2.4300e-02, -2.1134e-01,  2.0500e-02, -6.6952e-02,\n",
       "           2.4702e-01, -3.8551e-02,  5.1663e-02, -3.9512e-02, -6.9939e-02,\n",
       "           3.5306e-01,  3.5562e-01,  2.5781e-01,  3.8191e-01, -1.1442e+01,\n",
       "          -1.7367e-02,  8.1083e-01, -4.0185e-01, -6.2278e-02, -3.2474e-01,\n",
       "           1.8207e-01, -1.3725e-01,  9.2679e-02, -1.1016e-01,  1.5311e-01,\n",
       "          -1.4956e-01, -3.9201e-01,  1.0878e-01, -5.6494e-01,  1.1811e-01,\n",
       "          -4.9382e-01, -5.8867e-01,  1.2464e-01, -1.8112e-01,  2.4435e-01,\n",
       "          -5.6414e-01,  1.5738e-01,  1.4924e-01, -2.5041e-01, -5.7961e-01,\n",
       "          -1.3900e-01, -6.9499e-01,  7.6913e-01,  1.3120e-01, -1.7549e-01,\n",
       "           9.7960e-02,  3.1740e-02,  5.9839e-01, -1.9920e-01, -3.8882e-01,\n",
       "          -8.4616e-02, -1.3154e-03,  4.2860e-01, -5.4872e-01, -2.5155e-01,\n",
       "           3.0173e-02, -2.5315e-01,  5.8404e-01,  1.3454e-02,  6.2449e-01,\n",
       "           1.2541e+00,  9.3979e-01, -5.2139e-02,  1.3577e-01, -1.9090e-02,\n",
       "          -5.6188e-01,  3.9431e-01, -4.8762e-01,  1.9359e-01,  1.7056e-01,\n",
       "           3.3034e-01, -6.3570e-01,  1.7809e-01,  1.0194e-01, -8.4891e-02,\n",
       "           2.8406e-02, -1.5541e-02,  4.3766e-01, -5.2522e-01, -3.2954e-02,\n",
       "           4.6035e-01, -5.5636e-02,  2.4722e-01, -1.4065e-01, -9.1112e-02,\n",
       "           6.0506e-02,  8.2388e-01,  1.2386e-01, -8.2695e-01,  4.0247e-02,\n",
       "          -5.2141e-02, -1.4441e-02,  9.2113e-01,  1.2022e-01,  1.0520e-01,\n",
       "          -5.1424e-02,  3.6690e-01,  1.3465e-03, -8.5918e-01, -5.4331e-01,\n",
       "           1.0384e-01, -2.5178e-01,  2.5247e-01,  7.6190e-01, -1.7964e-01,\n",
       "          -1.4318e-01, -1.0675e-01,  5.1993e-02,  1.5440e-01,  1.7945e-01,\n",
       "          -5.3538e-01, -3.8656e-01,  2.5588e-01, -1.2438e-01, -5.5410e-01,\n",
       "          -3.0551e-01, -4.3011e-02, -1.4863e-01,  5.3680e-01, -9.5684e-02,\n",
       "          -2.5568e-01, -1.0149e-01,  2.0620e-01,  3.6639e-01, -1.8171e-01,\n",
       "          -1.3872e-01,  9.9128e-02,  1.0996e-01, -5.4052e-01,  1.4822e-01,\n",
       "          -4.2939e-02,  6.6097e-01, -5.2277e-01, -4.4000e-01, -3.7552e-01,\n",
       "          -2.4690e-01,  3.3773e-01,  9.1097e-02, -5.7951e-01,  4.8725e-01,\n",
       "          -3.5610e-01, -2.1409e-01, -1.3561e-01, -2.7970e-03,  5.7108e-01,\n",
       "           2.9027e-01,  4.9440e-01,  5.3298e-01,  1.9021e-02, -4.5712e-01,\n",
       "          -5.4355e-01,  5.4772e-02,  2.8797e-01, -5.2719e-01, -1.1893e-01,\n",
       "          -1.2608e-03,  1.5793e-01, -2.6004e-02,  2.2981e-01,  1.2959e-01,\n",
       "           3.6761e-01,  3.7729e-01, -4.3884e-01,  1.5503e-01, -1.2696e-01,\n",
       "          -3.1545e-01,  2.8115e-02, -3.5477e-01,  4.0304e-02, -2.7969e-01,\n",
       "          -2.4055e-01, -1.2121e-01, -1.1699e-01,  2.3867e-01,  4.2693e-02,\n",
       "           3.1094e-01, -6.0063e-02, -1.9914e-01,  6.1371e-01, -8.4879e-02,\n",
       "           2.1389e-01, -8.3616e-01, -4.4699e-01,  6.0406e-02, -2.9833e-02,\n",
       "           2.1319e-01, -4.4895e-01, -6.4797e-01, -1.6092e-01, -1.0342e-01,\n",
       "          -2.1000e-01,  1.0349e+00, -3.4865e-01, -5.7658e-02,  1.0562e-01,\n",
       "           2.8633e-01, -2.1862e-01, -3.9024e-02,  2.3740e-03,  1.4605e-01,\n",
       "          -2.7825e-01, -2.2235e-01,  2.5834e-01, -1.6474e-01,  3.3755e-01,\n",
       "           4.5466e-01,  2.1325e-01,  1.4162e-01, -3.1322e-01, -4.9213e-01,\n",
       "          -6.0200e-01, -3.4060e-02, -2.2792e-02, -2.2066e-01,  4.3212e-01,\n",
       "          -3.8047e-01, -5.4427e-02, -1.4326e-02, -1.8352e-01,  1.4095e-01,\n",
       "           8.6151e-02,  1.3830e+00,  8.7182e-02, -2.7718e-01,  2.9299e-01,\n",
       "           2.1709e-01, -8.1450e-02,  6.8001e-02, -1.9247e-01,  2.3583e-01,\n",
       "           5.6706e-01,  2.4442e-01, -3.4077e-01,  4.2829e-02,  2.2220e-01,\n",
       "           1.7138e-01, -8.1374e-01, -6.8881e-01, -1.9476e-02,  4.5777e-01,\n",
       "          -3.7177e-01,  4.4939e-01, -7.2867e-02,  4.4156e-01,  3.0875e-01,\n",
       "          -8.8166e-01, -2.9708e-01, -1.2028e-01,  2.4328e-01, -1.6288e-01,\n",
       "          -3.5080e-01, -3.2117e-01, -1.1116e-01, -2.4520e-01,  4.9277e-01,\n",
       "           4.2995e-01, -4.0441e-01,  3.5205e-01,  3.7650e-01, -2.0391e-01,\n",
       "           2.2168e-01, -7.2082e-02,  1.6195e-01,  6.4723e-01, -2.8200e-01,\n",
       "          -2.3570e-02,  1.7106e-01, -6.8176e-01, -4.4183e-01,  2.0093e-01,\n",
       "           7.0728e-01, -1.6285e-01,  1.4157e-01,  3.9063e-02, -3.9537e-01,\n",
       "          -1.8466e-01, -2.2071e-01, -3.6575e-01, -2.4633e-01,  3.8859e-02,\n",
       "          -2.5683e-01, -1.9056e-01, -4.4471e-02,  1.7739e-01, -1.5265e-01,\n",
       "           3.2475e-01,  7.9369e-01,  3.6923e-01, -2.6740e-01, -2.9940e-01,\n",
       "           3.5138e-01,  3.9822e-01, -4.6313e-01,  7.3435e-01, -1.9012e-01,\n",
       "          -1.2160e-01,  3.6729e-01,  2.1538e-01, -5.5997e-01, -3.7504e-02,\n",
       "          -2.2991e-01,  1.5890e-01, -2.7431e-01, -1.7118e-01,  5.5354e-01,\n",
       "          -5.3033e-02,  3.7587e-01, -6.8894e-01,  2.2169e-01,  4.1686e-01,\n",
       "          -4.2092e-01,  7.7933e-01,  1.7939e-01,  3.9933e-01, -1.7224e-01,\n",
       "           1.6049e-02, -5.1973e-01, -2.5435e-01, -8.8824e-02,  5.4897e-04,\n",
       "          -3.4219e-01, -1.5497e-01,  2.5116e-01, -2.5459e-01,  6.4233e-01,\n",
       "           6.8774e-01, -3.5222e-01, -6.9994e-01, -1.5328e-02, -7.7404e-02,\n",
       "          -2.5265e-01, -2.0348e-01, -1.9987e-01, -7.0056e-02, -2.8473e-01,\n",
       "          -4.8204e-01, -2.9833e-01, -3.9175e-02,  3.2150e-01, -2.8442e-01,\n",
       "           7.5354e-01,  2.9831e-02,  2.9410e-01,  2.8797e-01,  5.8349e-02,\n",
       "           2.8545e-01,  7.1117e-01,  2.7250e-01, -3.6209e-01, -1.9717e-01,\n",
       "           2.4567e-01, -4.4468e-01, -8.2011e-02, -6.3526e-01,  2.5941e-01,\n",
       "           4.0501e-01,  6.0216e-01,  3.9653e-01]], device='mps:0'),\n",
       " tensor([[-4.7220e-01, -3.8117e-01, -6.2516e-01, -7.3973e-01, -2.7943e-01,\n",
       "          -2.6227e-01, -9.7091e-01,  8.4637e-01,  1.0816e-01,  4.7817e-01,\n",
       "           2.9859e-01, -6.6096e-02,  6.6570e-02, -1.8509e-01, -7.1786e-01,\n",
       "          -3.1789e-01, -1.3235e-01,  3.6509e-01,  3.7310e-01,  3.2984e-01,\n",
       "           8.1565e-01,  2.8582e-01, -1.5921e-01, -1.4603e-02, -5.9030e-01,\n",
       "           3.9568e-01,  1.6204e-01,  1.7514e-01,  3.9385e-01, -8.8542e-01,\n",
       "           1.6616e-01,  3.0638e-01, -9.3394e-02,  7.7724e-01, -4.7268e-02,\n",
       "          -2.7849e-01,  2.4065e-01, -4.8312e-01,  7.2961e-01,  3.8307e-01,\n",
       "           1.6465e-01, -3.3334e-01, -1.7177e-01,  1.6416e-01, -1.6606e-01,\n",
       "          -7.9960e-03, -2.7629e-02, -7.4943e-01,  6.2366e-01, -3.8599e-01,\n",
       "          -3.9590e-01, -3.9669e-02,  5.1597e-01,  2.5064e-01, -7.7905e-01,\n",
       "          -5.6334e-01,  4.5312e-01,  4.7027e-01, -5.8684e-01,  3.8373e-01,\n",
       "          -1.2564e-01,  7.2400e-01,  3.2037e-01, -6.8452e-02,  2.7110e-01,\n",
       "           1.7342e-01, -3.8835e-01, -5.7688e-01, -3.2915e-02, -1.1367e-01,\n",
       "          -1.7784e-01, -5.6692e-02, -4.6691e-01, -3.4181e-01,  5.7574e-01,\n",
       "           2.1254e-01,  2.6113e-01,  5.4404e-01, -3.5402e-01,  5.1981e-02,\n",
       "           3.5865e-01,  7.0396e-02,  8.9668e-02,  3.0427e-02,  2.3425e-01,\n",
       "           2.2470e-01,  5.8048e-01,  2.0344e-01, -5.7798e-02, -1.6609e-01,\n",
       "           5.2360e-01,  9.9354e-02, -1.0054e-01, -7.7684e-01,  2.9883e-01,\n",
       "          -4.4370e-01,  2.6337e-01, -2.2284e-01,  1.7631e-01, -7.0128e-02,\n",
       "           2.0104e-01,  6.4211e-01, -1.6110e-01,  8.5570e-02, -3.1819e-01,\n",
       "          -7.5121e-01,  2.4770e-01, -1.1473e-01, -5.8056e-01,  1.4448e-01,\n",
       "          -7.4708e-02, -2.5991e-01,  2.5036e-01, -1.2521e-01, -2.3587e-01,\n",
       "          -8.1233e-02,  2.4232e-01, -1.0280e-01, -3.7440e-01,  4.8518e-01,\n",
       "           2.5100e-01, -5.0992e-01, -4.8613e-01, -8.1599e-02,  3.4820e-01,\n",
       "          -3.7485e-01,  5.2892e-01,  5.4453e-02,  4.1245e-01,  7.9263e-01,\n",
       "           1.3964e-01,  7.4558e-02, -1.7285e-01, -2.5711e-01, -3.0945e-01,\n",
       "           1.5692e-01, -7.2114e-02, -4.0579e-01,  3.9470e-01,  1.5577e-01,\n",
       "          -5.6334e-01, -2.4016e-01, -5.5589e-01, -2.2570e-01, -1.0711e-01,\n",
       "          -4.7424e-01,  1.4801e-01, -1.3357e-01,  2.1710e-01, -4.2236e-01,\n",
       "          -7.6752e-01, -7.8821e-01,  3.9770e-01,  9.6042e-02,  2.5411e-02,\n",
       "          -2.7616e-01,  2.6405e-01,  2.6542e-01, -1.6485e-02,  2.7675e-01,\n",
       "           6.4904e-01,  1.2940e-01,  2.2829e-01, -5.3446e-01, -3.3295e-02,\n",
       "           8.4328e-02, -2.3243e-01, -2.0410e-01,  3.0862e-02,  2.9375e-01,\n",
       "           2.4108e-01, -4.3052e-01,  6.0318e-02, -4.8793e-01,  5.4443e-04,\n",
       "          -5.8874e-01,  4.5771e-01,  2.3259e-01, -3.2538e-01, -1.5534e-01,\n",
       "          -3.7092e-01, -1.9178e+00,  2.8214e-01, -3.4930e-01, -1.3464e-01,\n",
       "          -7.3188e-01,  2.2316e-01,  2.1091e-01,  1.6665e-01,  2.5349e-01,\n",
       "          -1.7133e-01, -1.4836e-01, -2.5622e-01,  4.1424e-01, -4.5472e-02,\n",
       "           6.0199e-01,  6.9804e-01, -8.9213e-02, -4.8654e-01,  1.0606e-02,\n",
       "          -1.1399e-01,  2.0207e-01,  1.6015e-02,  8.8794e-02,  1.7516e-01,\n",
       "           4.3227e-01, -7.7075e-02,  3.1742e-02,  3.0671e-01, -1.9787e-01,\n",
       "           1.0063e-01, -2.7662e-01, -3.8091e-01, -7.4360e-02, -2.4591e-01,\n",
       "           4.8965e-02, -3.5960e-01,  1.0821e-01,  2.7344e-01,  8.9186e-02,\n",
       "          -2.7666e-01,  1.4054e-01, -4.1336e-01, -4.7701e-01, -2.3517e-01,\n",
       "           5.4084e-01, -2.1399e-01,  2.6586e-01, -4.8143e-01, -2.7768e-02,\n",
       "          -3.9256e-01, -4.7822e-01,  4.8124e-01,  5.0621e-01,  2.9946e-01,\n",
       "          -3.5768e-01, -2.7176e-01,  2.6138e-01, -1.5176e-01, -3.6322e-01,\n",
       "          -2.9211e-01, -4.6154e-01, -4.2638e-01,  2.4159e-02,  4.5309e-01,\n",
       "           5.8796e-03, -6.4210e-02,  9.1871e-01,  8.4846e-01,  1.3622e-01,\n",
       "          -4.4417e-03, -1.1102e-01,  3.5660e-01,  3.9509e-01, -2.6069e-01,\n",
       "          -3.0156e-01, -7.8316e-02, -4.2826e-01,  5.4681e-01, -2.6382e-01,\n",
       "           3.6494e-01, -2.1415e-02,  1.2874e-01,  8.3575e-01,  5.8376e-02,\n",
       "          -4.0432e-01,  1.6270e-01,  1.1758e-01, -1.6087e-01, -4.0037e-03,\n",
       "          -1.1391e-01,  2.7116e-01, -5.2223e-01, -1.9393e-01, -5.8126e-01,\n",
       "          -3.1163e-01, -1.6801e-01,  4.7742e-02,  3.7402e-01, -8.2556e-01,\n",
       "          -1.5150e-01,  4.7127e-01, -2.1854e-01,  3.1632e-01,  2.0948e-01,\n",
       "           5.1000e-01,  2.4296e-01,  1.6579e-01, -3.9279e-01,  6.4216e-02,\n",
       "           1.3500e-02,  2.4321e-01, -3.8971e-01,  1.4010e-02,  7.3395e-02,\n",
       "          -2.7710e-01,  2.7947e-01,  3.8154e-01,  8.7040e-02, -5.5827e-01,\n",
       "          -1.0441e-01, -1.4013e-01, -4.5655e-01, -3.2503e-01,  3.1347e-02,\n",
       "          -2.4748e-01,  4.0940e-01,  8.0560e-01, -7.9063e-03, -1.5308e-01,\n",
       "           1.0434e-01,  7.2943e-01,  4.1394e-02,  2.3299e-01, -2.8244e-02,\n",
       "          -1.0814e+00,  5.3330e-03, -1.4501e-01, -2.5200e-01, -3.2169e-01,\n",
       "           1.9506e-01,  5.6313e-01, -2.9121e-01, -4.2384e-01, -6.1260e-01,\n",
       "           3.9891e-01,  2.0875e-01,  5.7389e-01, -1.1632e-01, -2.9270e-01,\n",
       "           1.1068e-01, -5.6310e-01,  1.3029e-01, -2.9292e-01, -1.7047e-01,\n",
       "           2.6215e-01,  1.9004e-01, -5.6641e-02,  5.7105e-01, -4.2109e-01,\n",
       "           8.4425e-01, -1.8820e-01, -3.8530e-01,  2.2636e-01, -1.5825e-01,\n",
       "          -5.5356e-01,  3.7552e-01, -1.9947e-01,  2.5263e-01,  1.3057e-01,\n",
       "           6.0098e-01, -2.8243e-01, -3.2614e-01, -1.4191e-01,  4.9671e-01,\n",
       "          -4.5638e-02,  4.1505e-01, -3.0214e-01,  2.4624e-01,  3.2734e-01,\n",
       "          -3.5899e-01,  3.9232e-01, -5.2367e-01, -6.7999e-01,  1.2628e-01,\n",
       "           3.9814e-01,  3.0460e-01, -2.2110e-01, -1.3129e-01,  4.3180e-01,\n",
       "           3.3035e-01, -1.2623e-01,  2.4834e-01,  3.8397e-01,  8.5411e-02,\n",
       "          -2.0870e-01, -6.4269e-01,  1.5623e-01, -4.6603e-01, -1.7947e-01,\n",
       "           2.1499e-01, -9.0448e-02,  2.6679e-01, -2.8711e-01, -2.7399e-01,\n",
       "          -3.8118e-01,  3.0847e-01,  5.7773e-01,  1.3660e-01,  9.2345e-02,\n",
       "           1.4875e-01, -1.5667e-01, -3.2117e-01,  4.1956e-01, -2.5800e-01,\n",
       "          -3.5455e-01,  3.2314e-01, -1.7264e-01,  6.3286e-02,  2.3572e-01,\n",
       "          -5.3146e-02,  4.0970e-01,  1.1501e-01,  5.7541e-01, -3.6618e-01,\n",
       "          -3.7807e-01, -1.1026e-01, -5.8215e-03, -2.6365e-01,  1.4187e-01,\n",
       "          -8.0293e-02,  6.7881e-01, -2.0557e-01, -1.3850e-01,  6.6142e-01,\n",
       "          -1.7706e-02, -2.6882e-01,  1.7887e-01, -5.1992e-01, -3.3183e-01,\n",
       "           1.0706e-01,  1.6458e-01, -9.8918e-02,  2.9909e-01, -1.0840e+01,\n",
       "          -1.3564e-01,  7.8048e-01, -2.7977e-01, -3.7410e-02,  1.3757e-01,\n",
       "           9.1683e-02, -1.4739e-01,  2.4333e-01, -2.5401e-01, -5.0719e-01,\n",
       "          -2.5330e-01, -2.4513e-01,  5.2730e-02,  4.7149e-02, -1.9591e-01,\n",
       "           1.1218e-01, -8.0355e-01,  6.7369e-02, -5.0298e-01,  6.2393e-03,\n",
       "          -4.6921e-01,  3.9098e-01,  1.6065e-01, -3.7078e-01, -2.5008e-01,\n",
       "          -6.1873e-02, -2.1199e-01,  1.4435e-01,  1.6794e-02, -2.3664e-01,\n",
       "          -2.1415e-01, -5.4830e-02,  2.3026e-01, -1.9538e-01, -8.1314e-01,\n",
       "          -7.5701e-02,  1.7051e-01,  6.0715e-01, -2.3987e-01, -5.7736e-01,\n",
       "          -4.5001e-01, -6.1594e-01,  5.7053e-01,  3.4953e-01,  6.6533e-01,\n",
       "           8.6691e-01,  6.5168e-01, -9.4224e-03,  8.3258e-02,  2.3417e-01,\n",
       "          -5.9118e-01,  2.2453e-01, -5.9415e-01, -1.7977e-01, -4.1392e-01,\n",
       "           4.3002e-01, -7.6436e-01,  4.8619e-01, -2.1065e-01, -2.4843e-01,\n",
       "           1.4993e-01,  7.1768e-02,  2.2850e-01, -3.5889e-01, -6.0469e-01,\n",
       "           5.7483e-01, -1.2457e-01,  5.8783e-02, -3.9472e-01,  2.0251e-01,\n",
       "          -2.4085e-01,  4.6190e-01,  3.9141e-01, -5.3671e-01,  2.1460e-01,\n",
       "           1.2801e-01, -2.1318e-01,  1.1292e+00, -1.0138e-01,  3.8255e-01,\n",
       "           1.9972e-02,  9.5918e-02, -3.7103e-01, -6.3799e-01, -2.2712e-01,\n",
       "           7.6319e-02,  1.7261e-01,  3.2199e-01,  4.0971e-01, -4.3099e-01,\n",
       "          -6.8832e-01,  2.5939e-02,  2.4932e-01,  4.9559e-01,  1.0120e+00,\n",
       "          -4.5622e-01, -4.4803e-01,  5.9273e-02, -7.6108e-02, -6.9715e-01,\n",
       "          -5.4316e-02, -1.6334e-01, -5.1532e-01,  3.0931e-01, -8.7181e-02,\n",
       "           4.0218e-01,  3.0276e-01, -1.0907e-01,  2.4151e-01, -5.7459e-01,\n",
       "           3.1937e-01, -2.1440e-01,  2.0060e-01, -4.3767e-01,  3.2738e-01,\n",
       "          -8.0875e-02,  4.6318e-02, -3.1557e-01, -4.2702e-01, -1.5122e-02,\n",
       "          -9.8425e-02,  3.6040e-01,  1.9512e-01, -4.9564e-01,  4.4620e-01,\n",
       "          -3.7776e-01,  3.5409e-01,  2.9301e-01, -1.2850e-01,  6.3445e-01,\n",
       "          -1.4839e-01, -2.8298e-01,  9.7392e-02, -5.1506e-01, -4.8205e-01,\n",
       "          -4.6258e-01, -7.2125e-02,  3.8370e-01, -8.7626e-01,  5.4107e-02,\n",
       "           8.8796e-02,  3.4584e-01, -3.1635e-01,  2.7102e-01,  1.9009e-01,\n",
       "           1.4357e-01,  6.4283e-01, -1.5143e-01,  2.4453e-01, -2.8499e-01,\n",
       "          -6.4606e-01, -2.5062e-01,  3.3794e-02, -5.9565e-02, -1.0809e+00,\n",
       "          -1.2324e-01,  1.1712e-01, -2.7580e-01,  3.0844e-01, -1.0209e-01,\n",
       "           1.3986e-02, -1.7507e-01, -4.4887e-01,  4.2872e-01,  5.4993e-01,\n",
       "           2.6521e-01, -1.0998e-01, -3.4647e-01,  1.4453e-01,  3.5086e-01,\n",
       "          -5.2975e-02, -2.6779e-01, -4.2765e-01, -4.2188e-01,  1.7990e-01,\n",
       "           1.5634e-01,  7.0416e-01, -4.3747e-01, -5.3674e-01,  2.5951e-01,\n",
       "           4.0279e-01,  1.0212e-01, -4.2356e-02, -2.5791e-01, -1.2231e-01,\n",
       "          -2.4300e-01,  8.3126e-02,  2.8384e-01, -1.6091e-01,  2.9599e-01,\n",
       "           5.6243e-02,  5.9717e-01,  1.8598e-01,  3.7401e-01, -7.7453e-01,\n",
       "          -1.8476e-01,  1.0401e-01, -7.3981e-01,  3.1782e-02,  2.9136e-01,\n",
       "           7.2293e-02,  2.5885e-01, -3.3709e-01, -1.0727e-01,  2.3148e-01,\n",
       "           4.0491e-01,  8.3035e-01,  1.2699e-01,  2.4541e-01,  8.0606e-01,\n",
       "           3.0515e-01,  9.5258e-02,  1.4183e-01,  2.9865e-02,  1.2988e-01,\n",
       "           2.7003e-01, -7.5461e-02, -4.0604e-01, -1.2277e-01,  4.9380e-02,\n",
       "           4.4502e-01, -1.2779e+00, -6.8522e-01, -3.2293e-01,  6.2735e-01,\n",
       "           8.5968e-03,  2.7742e-01, -8.3087e-02,  4.4077e-01,  7.0379e-01,\n",
       "          -1.2355e+00, -4.4086e-01,  4.0380e-01,  4.3200e-01,  3.2965e-02,\n",
       "          -4.6912e-01,  2.5034e-01, -4.1453e-01, -4.4759e-01,  7.2868e-01,\n",
       "           5.0704e-01, -8.3809e-01,  8.9869e-02,  4.4899e-01, -5.8187e-01,\n",
       "           1.9586e-01,  5.8529e-02,  3.2015e-01, -1.4372e-01,  2.5361e-01,\n",
       "          -1.4912e-01,  3.7845e-01, -6.8510e-01, -4.0133e-01, -2.1575e-01,\n",
       "           3.8189e-01,  1.9472e-01,  3.6929e-01, -4.9674e-02,  1.5710e-01,\n",
       "          -4.6428e-01, -3.3347e-01, -2.3800e-01,  2.1891e-02, -3.9075e-01,\n",
       "           3.7246e-01, -3.6399e-01, -8.1400e-03,  3.9065e-01, -2.4918e-01,\n",
       "           5.1944e-01,  5.4253e-01,  4.6021e-01,  1.0712e-01, -8.4266e-02,\n",
       "           5.9500e-02, -5.3368e-02, -1.4617e-01,  5.8240e-01, -2.6604e-02,\n",
       "          -4.2366e-01,  1.8956e-01,  6.1743e-02,  3.8536e-02,  1.3010e-02,\n",
       "          -3.8344e-01,  4.0981e-01,  1.3398e-01, -4.8049e-01,  2.6606e-01,\n",
       "           4.8626e-01,  3.9398e-01, -4.0571e-01,  2.4753e-01,  9.3982e-02,\n",
       "          -1.8204e-01,  2.1849e-01, -1.1833e-01,  2.5589e-01, -8.4626e-01,\n",
       "           2.1881e-01, -4.6776e-01, -2.4703e-01,  2.9519e-01, -7.5118e-02,\n",
       "          -4.8236e-01, -1.2468e-01, -8.6961e-02, -1.3234e-01,  8.5137e-01,\n",
       "           1.0099e+00, -1.7200e-01, -5.8741e-01, -8.4271e-02,  3.0689e-01,\n",
       "          -1.3354e-01, -1.2457e-01,  2.1468e-01, -6.8691e-02, -2.5569e-01,\n",
       "          -1.2394e+00, -3.3661e-01,  1.1347e-01, -9.0199e-02, -3.0067e-01,\n",
       "           9.9320e-01, -7.0659e-02,  5.8198e-02,  4.9054e-01,  3.2113e-01,\n",
       "           2.4073e-01,  3.0128e-01, -2.5320e-02, -2.6189e-01, -1.4367e-01,\n",
       "           1.6038e-01, -9.2211e-02, -1.1339e-01, -6.1974e-01,  2.6997e-01,\n",
       "           1.9572e-01,  2.6877e-01,  8.9997e-01]], device='mps:0'),\n",
       " tensor([[-3.6355e-01, -5.7639e-01, -4.1457e-01, -5.0283e-01, -6.0284e-02,\n",
       "          -5.9142e-01, -2.9968e-01,  7.0318e-01, -2.3932e-01,  1.0118e+00,\n",
       "           1.6173e-01,  7.3109e-02,  1.0005e-01,  4.7311e-01, -4.8394e-01,\n",
       "          -1.6809e-01, -3.6919e-01,  2.7615e-01,  6.8393e-01,  1.7351e-01,\n",
       "           7.0548e-01,  5.9816e-01, -1.0212e-02, -4.0329e-02, -4.2510e-01,\n",
       "           4.0677e-01,  3.2944e-02, -8.7254e-02,  1.3014e-01, -4.7322e-01,\n",
       "           3.5176e-01,  2.7871e-01,  1.0769e-01,  1.0955e+00,  5.2976e-02,\n",
       "          -4.6911e-01,  5.1179e-01, -4.0847e-01, -1.2514e-01, -1.6568e-01,\n",
       "           5.6654e-01, -2.4748e-01,  3.2461e-01, -4.2481e-02,  5.6546e-02,\n",
       "           1.3156e-01, -1.7127e-01, -9.7535e-02,  4.1906e-01, -9.0828e-01,\n",
       "          -2.6304e-01, -5.1435e-02,  3.9086e-01,  6.1185e-01, -6.9639e-01,\n",
       "          -3.8863e-01,  3.8993e-01, -1.4479e-01, -1.4661e-01,  5.2356e-02,\n",
       "           1.0030e-01,  1.6524e-01, -1.9967e-01,  1.6365e-01,  1.6468e-01,\n",
       "          -1.1189e-01, -5.9249e-01, -4.2536e-01, -1.9803e-01, -1.6094e-01,\n",
       "          -1.3677e-01,  3.1963e-02,  4.1627e-01, -2.9232e-01,  8.0891e-02,\n",
       "           2.4143e-02,  1.5719e-01, -9.4937e-02,  9.4487e-02,  8.4795e-02,\n",
       "           9.8399e-02, -7.1774e-02,  1.7304e-01, -8.8617e-02, -8.5615e-03,\n",
       "          -5.2789e-03,  7.0553e-01, -2.3303e-01, -3.6160e-02,  2.2880e-01,\n",
       "           2.9545e-01,  1.1454e-02, -4.4803e-01, -2.8487e-01,  5.2200e-01,\n",
       "          -4.1697e-01,  5.1858e-01, -2.2271e-01,  1.0827e-01, -1.7546e-01,\n",
       "          -7.2692e-02,  1.9464e-01, -2.9051e-01,  6.4835e-02, -3.1452e-01,\n",
       "          -1.9899e-01, -2.5383e-02, -3.6231e-01, -4.6059e-01,  3.9690e-01,\n",
       "           3.8400e-02,  1.9193e-02, -1.1212e-01, -3.8796e-01, -1.0809e-01,\n",
       "          -1.0892e-02,  2.5472e-01, -9.5143e-02, -4.2694e-01,  3.3498e-01,\n",
       "           5.0107e-01, -5.0206e-02,  3.4072e-02,  1.9349e-02,  2.5892e-01,\n",
       "          -7.8038e-02,  7.1748e-01, -1.3444e-01,  2.4802e-01,  5.8337e-01,\n",
       "           9.7632e-02,  1.7167e-01, -4.0961e-01,  1.7680e-01,  1.5526e-01,\n",
       "          -1.0146e-01, -3.1777e-01, -2.1626e-01, -3.2992e-01, -1.6378e-01,\n",
       "           3.4069e-02, -2.7105e-01, -5.6292e-01, -4.9561e-01, -5.6035e-02,\n",
       "          -1.8110e-01,  1.4349e-01, -2.3207e-01,  3.2760e-01, -3.9283e-01,\n",
       "          -5.3651e-01, -2.9925e-01,  7.3020e-01,  6.7211e-04,  9.5871e-02,\n",
       "          -2.7969e-01,  5.7788e-01,  8.9136e-02,  3.2768e-01, -7.2801e-02,\n",
       "           9.8230e-01,  5.7133e-02, -4.7675e-02, -7.4918e-01, -4.7643e-02,\n",
       "          -5.2528e-02,  1.3923e-01, -4.9983e-01,  2.5507e-01,  7.6985e-02,\n",
       "           1.5098e-01, -2.0293e-01, -9.0335e-03, -1.3282e-01,  2.0459e-01,\n",
       "          -1.7443e-01,  3.1851e-01,  2.4223e-01, -1.4078e-01, -2.0150e-01,\n",
       "           2.4757e-01, -1.2566e+00,  2.6479e-01, -3.6115e-01, -2.3872e-01,\n",
       "          -1.3641e-01,  1.4451e-01, -5.5870e-02,  2.2890e-02,  3.8009e-02,\n",
       "          -2.6097e-01,  3.6324e-02, -4.5817e-01,  2.0586e-01, -2.0680e-01,\n",
       "           1.0416e-01,  4.2564e-01,  6.7318e-01,  3.2049e-01,  1.6586e-01,\n",
       "          -4.8888e-03,  1.4397e-01, -2.4480e-01,  1.3897e-01,  1.0399e-01,\n",
       "           3.8237e-01, -6.5473e-01, -9.8087e-02,  2.6900e-01,  2.2620e-01,\n",
       "          -3.7550e-01, -3.0692e-02, -4.4326e-01, -1.9498e-01, -2.7130e-01,\n",
       "           5.9519e-02, -5.0578e-01, -1.4336e-01,  5.2758e-01,  1.5470e-01,\n",
       "          -2.8267e-02,  7.2433e-02,  2.4700e-01, -2.4854e-01,  1.8642e-01,\n",
       "           5.7958e-01, -3.2799e-01, -2.6224e-01, -1.2151e-01,  4.6325e-02,\n",
       "          -1.5202e-01, -8.6695e-01,  5.4390e-01,  4.5353e-01,  2.9829e-02,\n",
       "          -8.7949e-02, -6.0956e-01,  4.1739e-01,  1.4364e-01,  2.0726e-01,\n",
       "           2.3091e-02, -5.1714e-01, -5.1283e-01,  2.3668e-01, -1.0143e-01,\n",
       "           4.5889e-01, -3.7359e-02,  3.8348e-01,  7.7194e-01,  5.6539e-01,\n",
       "          -2.9230e-01,  1.5241e-01,  5.2744e-01,  3.0603e-01, -5.8258e-01,\n",
       "          -2.6138e-01,  1.5836e-01, -6.8704e-01,  1.0400e-01, -3.0415e-01,\n",
       "          -2.7689e-01, -1.2015e-01,  1.8692e-01,  1.9290e-01,  1.1260e-02,\n",
       "           1.9354e-01,  1.1238e-01, -8.9099e-02, -3.4396e-01,  1.0924e-01,\n",
       "          -4.0331e-01,  3.9422e-01, -2.3491e-01, -8.9328e-02, -7.1923e-01,\n",
       "          -4.6336e-01, -3.4141e-02,  2.8713e-01, -4.3271e-02, -7.4487e-01,\n",
       "           8.1011e-02,  2.4181e-02, -2.1292e-01, -1.2444e-01, -3.8278e-01,\n",
       "           5.7042e-01, -1.6574e-01, -5.3670e-02, -3.1788e-01, -1.8751e-01,\n",
       "          -5.9294e-01,  3.7134e-01, -1.7055e-01, -4.5362e-02,  3.1057e-01,\n",
       "           1.4260e-01,  1.5375e-01,  5.7561e-01, -1.3875e-01, -3.7599e-01,\n",
       "           1.9233e-01,  3.6105e-01, -9.7490e-02,  3.4068e-02,  4.3382e-01,\n",
       "           5.8086e-01,  2.4874e-01,  4.6015e-01, -3.3827e-01, -2.1499e-01,\n",
       "          -4.5326e-02,  5.1555e-01, -1.1089e-01, -5.9192e-01,  1.3044e-01,\n",
       "          -1.0595e+00, -2.7029e-01, -6.3880e-02,  1.0124e-02, -3.8105e-01,\n",
       "          -3.7074e-02,  6.1840e-01, -1.1614e-01,  6.9434e-02, -5.5580e-02,\n",
       "           4.0490e-01,  8.3382e-01, -1.2020e-01,  1.6949e-01, -6.8952e-01,\n",
       "          -5.5152e-02, -1.7167e-01, -5.0060e-01, -3.1424e-01, -1.7019e-02,\n",
       "           2.9608e-01,  2.5314e-01,  4.4125e-01, -2.5558e-01, -4.2526e-01,\n",
       "           4.5961e-01, -3.6450e-01, -5.7582e-01,  3.0228e-01,  2.8991e-02,\n",
       "           9.7237e-02,  3.9841e-01, -6.8601e-02,  3.7154e-02,  6.2877e-01,\n",
       "          -7.4783e-02, -2.1692e-02, -3.9929e-01, -4.4706e-01,  6.4240e-01,\n",
       "           4.6178e-01,  1.5984e-01, -2.2428e-01,  6.7277e-01,  3.8775e-01,\n",
       "          -3.8551e-01,  2.6219e-01, -5.8901e-01, -4.4882e-01,  2.3775e-01,\n",
       "           2.5052e-01,  2.5837e-01, -2.4308e-01, -1.0043e-01, -5.8987e-03,\n",
       "          -7.1583e-02, -4.5932e-01,  1.0813e-01, -6.9623e-02,  1.5392e-01,\n",
       "          -1.4782e-01, -2.0250e-01,  1.8869e-01, -7.4750e-01, -2.7405e-01,\n",
       "          -2.9653e-01, -9.8162e-02,  6.5668e-02, -3.3928e-01, -1.6678e-01,\n",
       "          -3.9604e-02, -5.9445e-03,  7.9989e-01, -9.8573e-03,  5.3335e-02,\n",
       "           3.5260e-01, -7.7638e-02, -2.3873e-01,  8.0848e-01,  6.6808e-02,\n",
       "          -2.3597e-01,  4.8079e-01, -3.1659e-01, -1.9737e-02,  3.4057e-01,\n",
       "          -4.5335e-01,  2.9726e-01, -5.5556e-02,  1.8911e-01,  2.2718e-01,\n",
       "          -3.1041e-01, -1.4550e-01,  4.5487e-01, -6.6704e-01, -3.5466e-02,\n",
       "           2.9962e-01,  6.8267e-01, -6.0315e-01, -1.2318e-03,  1.7526e-02,\n",
       "           1.8750e-01, -8.5368e-02, -1.6232e-01,  4.0410e-02, -8.7599e-01,\n",
       "           3.8119e-01, -2.1660e-01, -8.7728e-02,  3.6628e-01, -1.1149e+01,\n",
       "           2.3726e-01,  8.8183e-01, -3.2766e-01, -3.9771e-02, -3.4749e-01,\n",
       "           1.9291e-01, -7.7958e-02,  2.6107e-01, -2.8036e-01, -9.5322e-02,\n",
       "          -2.5367e-01,  2.2686e-02,  8.0204e-02, -2.5156e-01,  4.3455e-02,\n",
       "          -1.7673e-01, -7.2054e-01,  2.6437e-01,  5.9248e-02, -1.2207e-01,\n",
       "          -1.2074e-01, -2.3925e-01,  1.4175e-01, -6.3142e-01,  5.2380e-02,\n",
       "          -2.2003e-01, -2.4293e-01,  8.3487e-04,  6.9521e-01, -2.1532e-01,\n",
       "          -1.2654e-02,  9.7817e-02,  2.8130e-01, -3.9372e-01, -5.1365e-01,\n",
       "           1.2387e-02, -3.3278e-02, -1.5469e-01,  1.7484e-01, -5.7002e-01,\n",
       "          -5.8714e-01, -2.1127e-02,  7.5472e-02,  1.8655e-01,  5.5475e-01,\n",
       "           2.5084e-01,  9.3274e-01,  4.4707e-01, -1.4224e-01, -3.4981e-01,\n",
       "          -1.0015e+00,  6.3469e-02, -9.5381e-01,  3.5078e-01, -5.4649e-02,\n",
       "           7.9073e-01, -4.0021e-01,  2.5469e-01, -3.8538e-01,  1.6499e-02,\n",
       "           1.8450e-01,  2.7618e-01, -6.1267e-02, -4.4336e-01, -6.7098e-02,\n",
       "           4.2739e-01, -2.0773e-01,  3.6439e-02, -4.7796e-01, -3.1042e-01,\n",
       "          -2.9442e-01,  4.4250e-01,  1.9411e-02,  3.9886e-01,  3.5074e-01,\n",
       "          -2.4336e-01, -2.0888e-01,  1.7993e+00, -4.0366e-01,  1.7553e-01,\n",
       "          -1.8192e-01,  1.8362e-01, -4.9901e-01, -8.3152e-01,  1.8161e-01,\n",
       "          -2.0930e-01,  1.5689e-01,  4.5942e-02,  6.2053e-01, -3.4065e-01,\n",
       "          -2.6528e-01,  1.0245e-02, -6.2038e-02, -2.3191e-01,  1.0345e+00,\n",
       "          -1.2127e-01, -3.6922e-01, -1.4760e-01,  3.0626e-01, -7.1314e-01,\n",
       "          -2.8442e-01, -1.3455e-01, -2.7820e-01,  2.0215e-01, -3.3942e-01,\n",
       "           3.5390e-02, -6.0351e-02, -3.4419e-02,  5.7237e-01, -5.5415e-01,\n",
       "           1.4902e-01,  4.3460e-01,  6.0812e-03, -2.2517e-01, -5.3994e-01,\n",
       "           4.8159e-01,  4.4345e-01, -3.1941e-01, -6.1522e-01, -5.2188e-02,\n",
       "          -1.5662e-01, -1.9574e-02, -1.0654e-01, -5.6184e-01,  3.3130e-01,\n",
       "          -7.0560e-01, -7.4080e-02,  5.3353e-01, -3.8275e-02,  4.1007e-01,\n",
       "           2.8641e-01, -4.8787e-01,  2.8231e-01, -3.1828e-01, -1.1510e-01,\n",
       "          -8.0165e-01, -6.6641e-01,  3.2451e-01, -4.8866e-01,  2.8968e-01,\n",
       "           1.8796e-02, -7.0031e-02, -2.7368e-01,  9.3806e-02, -1.5728e-01,\n",
       "           1.2878e-01,  3.6399e-01, -4.5294e-02, -1.2584e-01, -1.1594e-01,\n",
       "           1.6907e-02, -4.1819e-01, -3.8143e-01,  1.1122e-01, -5.4353e-01,\n",
       "          -1.6795e-01, -1.0167e-01, -1.3214e-01, -1.4681e-01,  5.2087e-01,\n",
       "           6.4475e-01, -9.6996e-02, -3.0489e-01,  3.5089e-01,  3.5642e-01,\n",
       "          -2.2080e-01,  1.1692e-01, -5.9096e-01, -1.0907e-01,  2.1660e-01,\n",
       "           1.0487e-01, -1.5369e-01, -3.0138e-01, -2.5180e-01, -7.0079e-01,\n",
       "          -2.2514e-01,  1.0950e+00, -5.1845e-01, -6.8783e-01,  4.6600e-01,\n",
       "           3.0278e-01, -1.9977e-01,  6.3924e-02, -1.2250e-01,  2.6434e-02,\n",
       "          -3.5658e-01,  1.3026e-01,  7.1617e-01,  1.5163e-01,  1.6427e-01,\n",
       "           3.9115e-01,  3.5391e-01, -3.4353e-01,  1.0958e-01, -1.4495e-01,\n",
       "          -4.7247e-01,  1.4219e-01, -4.0098e-02, -1.6355e-01,  9.1140e-02,\n",
       "          -5.1558e-02, -2.6708e-01,  2.7695e-01, -2.7505e-01,  3.2110e-03,\n",
       "           2.7308e-01,  8.5689e-01, -1.1081e-01, -6.9538e-02,  3.1955e-01,\n",
       "           3.9728e-01,  1.3435e-01,  1.4905e-01,  3.7763e-01,  8.3303e-02,\n",
       "           4.2342e-01,  2.3370e-01, -1.5310e-01, -1.7052e-01,  1.5115e-01,\n",
       "           4.1113e-01, -4.1120e-01, -5.8758e-01,  1.1081e-01,  5.7065e-01,\n",
       "          -1.2180e-01,  6.0302e-01, -1.6362e-01,  1.7591e-01,  4.8836e-01,\n",
       "          -9.6338e-01,  6.1833e-02,  2.0248e-01, -6.0485e-01, -2.1245e-01,\n",
       "          -2.2919e-01, -3.4192e-01, -1.6229e-01, -4.7573e-01,  4.6485e-01,\n",
       "           5.6997e-01, -1.2268e+00,  2.3353e-01,  7.0952e-01, -4.3914e-01,\n",
       "           6.0637e-01, -4.7956e-02,  4.7979e-01,  8.8963e-01, -2.2181e-01,\n",
       "          -9.5748e-02,  1.5582e-01, -3.1432e-01, -4.2579e-01,  5.9131e-02,\n",
       "           6.2730e-01,  4.2211e-02,  1.8738e-02, -2.1620e-01, -5.1581e-01,\n",
       "          -4.3150e-01, -3.2225e-01, -2.5392e-01,  1.6589e-01,  6.2313e-01,\n",
       "          -2.2271e-01,  3.8019e-02, -2.0308e-01,  2.4913e-01, -6.9309e-02,\n",
       "           5.3801e-01,  2.8776e-01,  4.7944e-01,  1.9747e-01, -5.4127e-01,\n",
       "          -8.7731e-02,  5.7235e-01, -3.8376e-01,  8.6869e-01, -1.8586e-01,\n",
       "          -3.9331e-01, -8.4133e-02,  1.1606e-01, -1.8034e-01,  2.3142e-01,\n",
       "          -5.2697e-01,  3.8899e-01, -1.8176e-01,  6.6721e-03,  5.8575e-01,\n",
       "          -6.6039e-02,  2.7508e-01, -7.1980e-01, -8.3032e-03,  1.9168e-01,\n",
       "           8.7176e-02,  5.6715e-01, -1.5991e-01,  8.0517e-02, -4.7620e-01,\n",
       "          -2.1484e-01, -1.7699e-01, -9.6950e-02,  4.5661e-01, -1.2172e-01,\n",
       "          -4.9477e-01, -3.3108e-01, -2.3593e-01,  2.1215e-01,  6.7934e-01,\n",
       "           7.3356e-01, -2.0065e-01, -5.0491e-01, -2.2002e-01,  1.5968e-01,\n",
       "           1.9439e-01,  7.4096e-02,  1.6166e-01, -2.6325e-01, -1.9598e-01,\n",
       "          -4.4662e-01, -4.0758e-01,  7.3051e-02,  4.8675e-02, -2.5397e-01,\n",
       "           5.3172e-01, -8.5962e-02,  8.8738e-02,  5.3409e-01,  3.9084e-01,\n",
       "           4.9408e-02,  6.9127e-01,  1.7864e-01, -4.7942e-01, -1.7557e-01,\n",
       "           1.3188e-01, -9.3891e-02,  8.5702e-02, -3.3488e-01,  6.4587e-01,\n",
       "           1.9185e-01,  2.7023e-01,  4.4925e-01]], device='mps:0'),\n",
       " tensor([[-2.5187e-01, -4.0054e-01, -5.8007e-01, -8.3055e-01, -1.2868e-01,\n",
       "          -2.8804e-02, -7.1917e-01,  6.6701e-01,  1.2709e-01,  5.0855e-01,\n",
       "           1.2570e-01,  5.1916e-01,  5.1676e-02, -1.1521e-01, -1.6481e-01,\n",
       "          -4.9820e-01,  5.6222e-02,  3.7338e-01,  6.7187e-01, -2.4111e-02,\n",
       "           2.2844e-01,  5.7979e-01, -2.6323e-01, -1.2386e-02, -8.3013e-01,\n",
       "           2.9613e-01, -1.1576e-01,  2.2409e-01,  1.7718e-01, -6.3092e-01,\n",
       "           4.0999e-02,  4.7417e-01,  2.3068e-02,  1.3497e+00, -5.6080e-01,\n",
       "          -1.3230e-01,  8.2213e-02, -3.9453e-01, -2.3010e-01,  3.7283e-02,\n",
       "           6.1164e-01, -4.1315e-01,  1.3635e-01,  4.0254e-01, -6.1085e-02,\n",
       "           8.9076e-02,  1.3899e-01, -3.7744e-01,  2.5892e-01, -8.3961e-01,\n",
       "          -4.8318e-02, -2.0669e-01,  4.9144e-01,  6.2615e-01, -6.3176e-01,\n",
       "          -2.2975e-01,  2.7419e-01, -3.7716e-02, -3.3020e-02,  1.5349e-01,\n",
       "           9.7980e-02,  3.8582e-01, -2.9995e-01,  2.3257e-01,  3.3441e-01,\n",
       "           2.4458e-01, -3.0392e-01, -4.8889e-01, -5.4078e-01, -5.0111e-02,\n",
       "           9.1859e-03,  2.2490e-02,  1.4846e-01, -2.3272e-01,  4.1973e-01,\n",
       "          -7.1214e-01, -6.4354e-02, -8.9125e-02, -2.5973e-01,  2.9246e-01,\n",
       "          -4.5549e-02,  1.9719e-01,  3.1010e-01,  1.0268e-01,  1.2457e-01,\n",
       "           4.3419e-02,  2.7215e-01,  5.2932e-04,  4.7373e-02,  3.7533e-01,\n",
       "          -1.3375e-01, -8.1722e-02, -2.8313e-01, -3.5814e-01,  3.1229e-01,\n",
       "           4.0598e-02,  2.9907e-01, -1.9557e-01,  1.8579e-01, -3.2474e-01,\n",
       "           1.5588e-01,  1.7768e-01, -1.8083e-01,  5.7286e-02, -3.5172e-01,\n",
       "          -4.5475e-01,  2.3589e-02, -2.8569e-01, -4.8285e-01,  2.2697e-01,\n",
       "          -1.4211e-01, -3.0205e-01, -4.0922e-02, -3.8935e-01, -3.1524e-02,\n",
       "          -7.5164e-02,  2.6013e-01, -1.3230e-01, -3.4806e-01, -8.3033e-02,\n",
       "           2.0734e-01, -2.2808e-01,  1.6077e-01,  2.6882e-01,  1.2232e-01,\n",
       "          -9.6705e-02,  6.5251e-01,  1.8950e-02,  3.1674e-01,  2.5763e-01,\n",
       "           1.1699e-01, -6.5768e-02, -1.9745e-01, -2.3154e-01,  7.5988e-02,\n",
       "          -7.6941e-02,  1.0687e-01, -1.8182e-01, -2.2790e-01, -1.8508e-01,\n",
       "           3.1771e-01, -3.3097e-01, -6.0893e-01, -4.9260e-01, -1.6098e-01,\n",
       "          -1.6326e-01, -2.4804e-02, -8.2473e-03, -3.3693e-01, -3.1956e-01,\n",
       "          -6.7443e-01, -3.6717e-01,  8.1041e-01, -1.6831e-02, -3.3356e-01,\n",
       "          -1.6499e-01,  2.3208e-01,  3.1666e-01, -4.5959e-02,  3.7161e-01,\n",
       "           7.7250e-01, -4.7670e-02,  1.6786e-01, -5.1827e-01, -3.8195e-01,\n",
       "           4.8335e-02,  2.6658e-01, -2.7054e-01,  6.1879e-01,  4.8352e-02,\n",
       "           7.5627e-02, -6.6170e-02,  2.7430e-01, -3.2704e-01, -1.2324e-02,\n",
       "          -5.7573e-01,  3.7750e-01,  2.9539e-01, -3.9892e-01, -6.6567e-02,\n",
       "          -2.0834e-02, -1.2060e+00,  9.6964e-02, -8.3040e-01,  3.1236e-01,\n",
       "          -2.3110e-01,  1.7963e-01,  2.3389e-01,  2.9545e-01, -6.5320e-02,\n",
       "           2.5404e-01,  2.8725e-01,  2.5687e-01,  3.3609e-01, -3.4924e-02,\n",
       "           2.8019e-01,  2.3924e-01,  5.0790e-01,  4.1073e-01,  1.4769e-01,\n",
       "          -3.5440e-01,  2.9683e-01,  3.5781e-01,  9.0598e-02,  1.6954e-01,\n",
       "           9.9825e-02, -2.1636e-01,  2.1091e-01,  1.7501e-01, -1.6618e-01,\n",
       "          -2.2406e-01, -3.1100e-02, -4.6818e-01, -5.6655e-02, -4.8467e-01,\n",
       "           3.2557e-01, -5.1983e-01, -1.1912e-01,  3.4831e-01,  2.4340e-01,\n",
       "          -1.4896e-02,  1.8319e-01,  1.7196e-01, -2.4338e-01,  2.1735e-01,\n",
       "           3.7449e-01, -3.4618e-01, -3.9804e-02, -3.2197e-01, -2.9367e-01,\n",
       "           2.4357e-01, -5.6244e-01,  6.6635e-01,  4.0574e-01,  6.0165e-02,\n",
       "           1.5119e-01, -2.7920e-01, -5.1638e-02, -5.7408e-01,  3.5471e-01,\n",
       "          -3.2130e-01, -3.4955e-01, -5.4801e-01,  6.4487e-02,  4.2773e-01,\n",
       "           2.7565e-01, -6.4915e-02,  7.7868e-01,  4.2408e-01,  4.2813e-01,\n",
       "          -4.3003e-01, -1.5896e-02,  4.0377e-01, -4.1405e-02, -4.6059e-01,\n",
       "           6.7116e-02, -1.4962e-01, -5.1643e-01,  2.8000e-01, -2.6598e-01,\n",
       "          -1.2047e-01, -1.4913e-01,  3.3682e-01,  1.5365e-01, -2.1964e-01,\n",
       "          -3.8461e-01,  3.9587e-01, -1.4349e-01, -4.1723e-01,  4.7123e-01,\n",
       "          -3.5025e-01,  1.5209e-01,  1.0694e-01, -3.8441e-01, -6.2072e-01,\n",
       "          -3.2235e-01,  2.3090e-01,  1.5585e-01,  3.4799e-02, -6.6306e-01,\n",
       "          -4.0571e-01,  3.1197e-01,  2.4601e-01,  2.9976e-01,  2.9840e-01,\n",
       "           5.7427e-01, -3.0944e-01, -1.6702e-01,  1.4982e-01, -4.1377e-01,\n",
       "          -1.5704e-01, -3.3228e-02, -1.9995e-01,  2.9569e-01,  2.4768e-01,\n",
       "           1.1600e-01,  2.0367e-01,  4.3482e-01, -2.5272e-01, -3.8673e-01,\n",
       "          -7.1543e-02,  2.0161e-01, -9.3629e-02, -5.8306e-02,  2.3951e-01,\n",
       "          -6.3095e-02,  5.9650e-01,  7.2143e-01,  7.7439e-02, -6.4006e-01,\n",
       "           1.7696e-01,  5.2502e-01, -2.7253e-01, -4.0343e-02,  9.6365e-02,\n",
       "          -1.0344e+00, -2.1331e-01, -1.2883e-01, -2.0181e-01, -6.5352e-01,\n",
       "           8.8115e-02,  3.4863e-01,  4.6833e-03, -1.7864e-01, -3.0947e-01,\n",
       "           1.0448e-01,  4.8724e-02, -4.8980e-02,  4.3534e-01, -2.8901e-01,\n",
       "          -3.4797e-01, -2.3438e-01, -2.1532e-01, -5.4201e-01, -3.9968e-01,\n",
       "           3.5914e-01,  2.8129e-01,  1.1986e-01, -2.6486e-01, -5.1832e-01,\n",
       "           4.5577e-01, -6.8169e-01, -1.4632e-01,  2.4840e-01, -4.7702e-01,\n",
       "           1.4650e-02,  4.8990e-01,  1.4999e-01,  1.6336e-01,  6.5258e-02,\n",
       "           7.8557e-02, -3.3903e-01, -5.0030e-01, -1.5000e-01,  6.2323e-01,\n",
       "           3.2283e-01,  3.3506e-01, -1.7236e-01,  3.9096e-01,  4.5167e-01,\n",
       "          -2.9582e-01,  4.1670e-01,  1.0930e-02,  7.7439e-02, -6.0414e-02,\n",
       "           1.9048e-01,  3.4370e-01, -3.6842e-01,  2.0145e-01,  1.7929e-01,\n",
       "          -3.1634e-01, -3.2149e-01,  1.5691e-02,  2.0642e-01, -1.8580e-01,\n",
       "          -3.0875e-01, -2.0361e-01,  9.9698e-02, -3.4856e-01, -4.3104e-01,\n",
       "           2.9161e-01, -3.6543e-02,  7.0495e-02, -4.0357e-01, -4.3462e-01,\n",
       "          -2.0181e-01, -2.6055e-01,  6.8681e-01,  2.1064e-01, -1.9621e-01,\n",
       "           1.8720e-01, -1.0907e-01, -3.6330e-01,  4.7847e-01,  2.0162e-02,\n",
       "          -4.8754e-01,  4.5482e-01, -3.4822e-01,  1.0627e-01,  5.5756e-01,\n",
       "          -1.6769e-01, -1.0840e-01,  1.1008e-01,  1.0820e-01,  2.9026e-01,\n",
       "          -3.9884e-01,  1.5516e-01, -6.3476e-02, -3.1885e-01,  3.1820e-01,\n",
       "           2.6411e-01,  2.2781e-01, -1.7137e-01, -3.3426e-02,  3.5677e-02,\n",
       "           2.8782e-01, -6.0853e-02, -3.9888e-01, -4.7347e-01, -4.6132e-01,\n",
       "           1.2832e-01,  1.8927e-01,  5.6207e-01,  3.4144e-01, -1.1394e+01,\n",
       "           1.2938e-01,  7.8031e-01, -4.0713e-01, -8.2819e-02, -5.0320e-02,\n",
       "           1.0976e-02, -1.3708e-01,  4.2955e-01, -2.7734e-01, -4.4105e-01,\n",
       "          -3.5727e-01, -1.3308e-01,  1.5454e-01, -3.2189e-01, -2.6945e-01,\n",
       "          -1.4686e-02, -7.0935e-01,  4.0963e-01, -3.0286e-01,  5.0298e-01,\n",
       "          -2.3511e-01, -8.3798e-02,  1.5816e-01, -3.5851e-01, -1.3439e-02,\n",
       "           5.9766e-02, -1.6033e-01,  1.4651e-01,  4.2486e-01, -5.2072e-02,\n",
       "           3.3244e-02, -4.1258e-03,  4.8461e-02, -2.5723e-01, -5.9999e-01,\n",
       "           1.9296e-02, -1.4141e-01,  3.3422e-01, -4.4133e-02, -7.1265e-01,\n",
       "          -2.2533e-01, -2.7418e-01,  3.1407e-01,  2.7021e-01,  9.0598e-01,\n",
       "           2.2955e-01,  1.0686e+00,  1.5022e-01,  1.7748e-01,  2.6230e-02,\n",
       "          -7.8764e-01, -3.3269e-02, -4.1907e-01,  2.3280e-02,  1.0114e-01,\n",
       "           3.6576e-01, -7.9367e-01,  2.5251e-01, -3.5521e-01,  3.6487e-02,\n",
       "           4.9094e-01,  3.1768e-02,  2.7700e-01, -3.2298e-02, -7.8123e-02,\n",
       "           1.2089e-01, -2.8544e-02,  1.1098e-01, -2.2583e-01, -2.4084e-01,\n",
       "          -1.0177e-01,  5.9888e-01,  3.9592e-01, -7.2211e-01,  3.6112e-01,\n",
       "           4.9258e-02, -1.8622e-01,  1.4556e+00, -4.8573e-02,  1.8361e-02,\n",
       "           1.9800e-03,  2.9604e-01, -5.0236e-01, -7.3516e-01,  3.6511e-01,\n",
       "          -1.0211e-01,  5.6427e-02,  1.0127e-01,  4.8148e-01, -9.7481e-02,\n",
       "          -4.7940e-01, -1.8593e-03,  1.0556e-01, -1.3380e-01,  5.4710e-01,\n",
       "          -7.2824e-01, -1.6963e-01, -1.1223e-01,  3.1723e-01, -8.4054e-01,\n",
       "          -6.1126e-02, -7.0210e-01, -3.9328e-01,  5.0586e-01, -2.5896e-01,\n",
       "           1.5852e-01, -2.7443e-02, -4.3439e-02,  5.5574e-01, -2.1489e-01,\n",
       "           7.1555e-02,  2.2813e-01,  8.4995e-02, -1.6733e-01,  3.9349e-02,\n",
       "           4.1630e-01,  4.9571e-01, -4.2740e-01, -6.3911e-01, -7.6901e-03,\n",
       "           1.7180e-01,  1.4933e-01,  3.4019e-01, -5.3463e-01,  3.6431e-01,\n",
       "          -7.9162e-01, -1.8769e-01,  7.2949e-02, -4.3273e-02,  3.6175e-01,\n",
       "           2.3925e-01,  7.2038e-02,  2.2256e-01, -7.7756e-01, -6.4498e-01,\n",
       "          -6.5932e-01, -2.1445e-01,  2.6143e-01, -5.4236e-01, -8.3654e-02,\n",
       "          -1.1033e-01, -4.8837e-03, -5.0292e-02, -6.6102e-02,  1.1939e-02,\n",
       "           1.3632e-01,  2.3587e-01, -1.3023e-01, -1.3178e-01, -9.2464e-02,\n",
       "          -5.5566e-01, -1.1753e-01, -9.9591e-02, -1.2505e-01, -5.6783e-01,\n",
       "          -1.3797e-01,  3.3075e-02, -2.5373e-01, -6.8124e-02,  3.5608e-01,\n",
       "          -2.9832e-02, -2.7856e-02, -4.8850e-01,  7.2846e-01,  4.0279e-01,\n",
       "           5.2088e-02, -2.8298e-01, -5.5189e-01, -1.0567e-01,  2.9592e-01,\n",
       "           1.9555e-01, -6.7007e-02, -1.7709e-01, -2.4145e-01, -4.2522e-01,\n",
       "           1.4302e-01,  7.1147e-01, -6.6296e-02, -6.8115e-01,  2.6621e-01,\n",
       "           5.8223e-01, -1.2882e-01, -2.4432e-01,  2.0983e-02, -9.3284e-02,\n",
       "          -3.1851e-01, -6.0317e-01,  7.8821e-01, -1.1246e-01, -5.8638e-02,\n",
       "           2.2427e-01,  2.0359e-01, -9.2588e-03,  4.0209e-01, -1.5770e-01,\n",
       "          -3.5434e-01, -1.6477e-03, -7.3600e-01,  4.3257e-01,  6.1666e-02,\n",
       "          -3.1489e-01,  7.6082e-01,  2.7260e-01, -6.0593e-02, -4.5779e-02,\n",
       "           2.2863e-01,  7.9941e-01, -2.4636e-01,  6.4653e-02,  5.3779e-01,\n",
       "           8.8110e-02, -1.1229e-01,  3.9580e-02,  4.1665e-02,  4.5410e-02,\n",
       "           3.7012e-01, -3.2145e-02, -4.1564e-01, -6.1194e-02,  3.8924e-01,\n",
       "           3.9295e-01, -5.4610e-01, -7.3487e-01,  2.2333e-02,  6.8551e-01,\n",
       "          -2.9054e-01,  2.3654e-01, -3.2708e-02,  3.0880e-01,  1.2001e-01,\n",
       "          -9.2160e-01, -1.5016e-01, -1.0056e-01,  2.8229e-01, -2.5482e-02,\n",
       "          -1.6139e-01, -1.4727e-01, -2.3686e-01, -1.6208e-01,  6.8308e-01,\n",
       "           2.1991e-01, -6.9160e-01,  3.4058e-02,  5.7281e-01, -6.0551e-01,\n",
       "           4.6052e-01, -2.4577e-01, -3.4302e-02,  3.6435e-01,  1.3117e-01,\n",
       "           4.9067e-02,  1.7856e-01, -4.2138e-01, -4.0598e-01,  1.7824e-01,\n",
       "           9.1672e-01, -7.6613e-02,  1.1166e-01, -3.9384e-01, -1.7182e-01,\n",
       "          -7.1855e-02, -3.1913e-01, -3.1286e-01,  3.7185e-01,  2.0529e-01,\n",
       "          -3.2552e-02, -1.3271e-01, -2.2992e-01,  3.0816e-01,  1.1821e-01,\n",
       "           3.2536e-01,  5.8187e-01,  6.4368e-01, -1.0077e-02, -2.4936e-01,\n",
       "          -6.1235e-02,  1.4568e-01, -1.9524e-01,  5.2704e-01, -2.0627e-01,\n",
       "          -3.3453e-01, -3.8069e-03,  3.0134e-01, -6.1681e-01, -5.8725e-02,\n",
       "          -4.5426e-01,  4.2361e-01, -1.6955e-01,  3.7868e-02,  2.8833e-01,\n",
       "           4.0522e-02,  5.7925e-02, -2.2651e-01,  5.5807e-01,  4.4511e-01,\n",
       "           1.6609e-01,  4.5662e-01,  1.6104e-01,  4.5272e-02, -5.1597e-01,\n",
       "           2.1978e-01, -1.0213e+00, -9.4260e-02, -4.6640e-02, -1.5401e-01,\n",
       "          -4.4224e-01, -2.8126e-02,  1.6814e-01,  1.8538e-01,  6.9597e-01,\n",
       "           7.2967e-01, -3.0871e-01, -4.5290e-01, -1.8510e-01,  1.5304e-01,\n",
       "           2.7705e-01, -1.9517e-01,  3.5971e-01,  9.6103e-02,  2.4432e-01,\n",
       "          -5.0967e-01, -4.3165e-01, -4.2252e-02,  3.1411e-01, -2.0171e-01,\n",
       "           5.7118e-01, -2.0164e-01, -1.3863e-01,  9.0832e-02,  1.3770e-01,\n",
       "           1.6639e-01,  7.9664e-01,  4.3416e-01, -6.3425e-01,  8.7469e-02,\n",
       "          -1.3980e-02, -2.9988e-01,  2.5983e-02, -4.3577e-01,  2.8858e-01,\n",
       "           1.2984e-01,  4.9280e-01,  6.7504e-01]], device='mps:0'),\n",
       " tensor([[-2.1631e-01, -5.6879e-01, -5.0655e-01, -6.0363e-01, -4.6338e-02,\n",
       "          -2.8912e-01, -5.9600e-01, -7.8058e-02, -2.7196e-01,  6.7845e-01,\n",
       "           1.5684e-01,  6.2665e-01,  9.7880e-02, -3.2688e-01, -2.6516e-01,\n",
       "           1.6174e-01,  5.9546e-02,  8.1000e-01, -6.7015e-02,  3.9866e-01,\n",
       "           3.7120e-01,  5.1056e-01,  4.2955e-02,  1.5750e-01, -6.2472e-02,\n",
       "          -1.7588e-01, -2.7612e-02,  2.5302e-01,  3.6570e-01, -2.7421e-01,\n",
       "           1.9090e-02,  7.1553e-01,  1.7919e-01,  8.0977e-01, -2.6216e-02,\n",
       "           9.7563e-03,  2.3100e-01, -2.4927e-01,  1.1458e+00, -1.6137e-01,\n",
       "           6.6109e-01, -3.3676e-02,  3.0831e-01,  1.0292e-01,  1.2588e-02,\n",
       "          -1.8503e-01,  7.5427e-02, -8.7086e-01,  4.6124e-01, -2.5988e-01,\n",
       "          -2.1725e-02, -5.8084e-02,  1.8834e-01,  2.9530e-01, -2.8328e-01,\n",
       "           5.5854e-03, -1.9293e-01,  4.8348e-01, -6.3731e-02,  1.9888e-01,\n",
       "           2.0461e-01,  3.0759e-01,  2.3910e-02, -4.8404e-01,  2.7013e-01,\n",
       "          -1.4567e-01, -3.0291e-01, -3.4401e-01, -5.2826e-02, -5.0763e-02,\n",
       "           6.2140e-02,  4.3126e-01,  1.8498e-01, -2.5980e-01,  3.3777e-01,\n",
       "          -3.9040e-01,  2.0172e-01,  9.4802e-02, -1.7198e-01,  3.9648e-01,\n",
       "           3.1835e-01,  2.4132e-01,  2.1040e-01, -7.0260e-02, -5.6549e-02,\n",
       "          -2.8877e-01,  3.3179e-01,  1.3559e-01,  3.6578e-01,  2.1065e-01,\n",
       "           2.2515e-01,  2.1454e-01, -5.4224e-01, -1.1619e-01,  1.9072e-02,\n",
       "           2.5700e-02,  1.2148e-01,  1.5298e-02, -4.9383e-02, -2.8108e-01,\n",
       "           2.0862e-01, -9.0513e-02,  2.7072e-01,  4.1947e-01, -8.0695e-01,\n",
       "          -4.7553e-01,  1.6905e-01, -3.4655e-03, -6.9372e-01, -2.6694e-01,\n",
       "           5.5072e-02, -3.1695e-01,  8.5638e-02, -1.2325e-01, -1.3591e-01,\n",
       "           7.3579e-02,  2.8151e-01,  2.6451e-01, -2.0425e-01,  3.9723e-01,\n",
       "           2.3560e-01, -4.3717e-01,  3.1889e-02, -5.8617e-02,  4.1842e-01,\n",
       "           4.6246e-02,  6.1457e-01, -2.8707e-01,  2.0205e-01,  3.8897e-01,\n",
       "          -1.0744e-01, -4.4705e-02, -2.3384e-01, -3.0292e-01,  8.2466e-02,\n",
       "           1.1305e-01,  2.3918e-01, -3.6555e-01, -2.5647e-02,  2.7365e-01,\n",
       "           5.8876e-02, -4.0393e-01, -2.4972e-01, -2.2026e-01, -2.0415e-01,\n",
       "          -2.0038e-01,  9.6230e-02, -4.1194e-01, -1.4498e-01, -5.4795e-01,\n",
       "          -1.0843e+00, -7.3233e-01,  6.6644e-01, -1.0615e-01,  2.5775e-01,\n",
       "           4.9187e-02,  1.3459e-01,  3.1153e-01,  1.2374e-03,  3.8371e-02,\n",
       "           3.6247e-01, -1.0305e-01,  2.6378e-01, -5.1854e-01,  1.1491e-01,\n",
       "          -1.6559e-01,  7.8061e-02, -1.0258e-01,  3.6614e-01,  6.4727e-02,\n",
       "           3.9231e-01, -4.8038e-01, -4.7041e-02, -1.8927e-01,  3.3835e-01,\n",
       "          -5.8047e-01,  3.3277e-01,  1.4199e-01, -1.5510e-01, -5.1727e-01,\n",
       "           3.8408e-01, -2.0154e+00, -2.2306e-02, -4.3558e-01, -2.5228e-01,\n",
       "          -4.2514e-01,  2.4833e-01,  1.7436e-01, -7.4038e-03, -3.0732e-01,\n",
       "          -3.6107e-01, -2.7998e-01,  4.2298e-02, -4.5966e-02,  1.3639e-02,\n",
       "           2.7881e-01,  6.4413e-01,  1.7534e-01, -1.1637e-01, -6.3286e-02,\n",
       "          -2.1283e-01,  4.2467e-01,  5.9378e-02, -1.8809e-01,  9.2145e-02,\n",
       "           3.1221e-01, -6.2036e-02,  7.9998e-02,  2.4063e-01,  1.0375e-01,\n",
       "          -2.1403e-01,  6.5158e-03, -4.0290e-01,  5.9182e-02, -4.6915e-01,\n",
       "           3.4721e-01, -2.6711e-01, -2.4813e-01,  4.4703e-01,  4.4835e-01,\n",
       "           3.9680e-02, -1.0921e-01,  1.8373e-01, -1.2926e-01,  4.2363e-02,\n",
       "           4.3965e-01,  2.8184e-02,  7.8786e-02,  6.5906e-02, -2.7845e-01,\n",
       "           9.7707e-02, -4.1945e-01,  4.1088e-01,  5.3973e-01, -2.0002e-01,\n",
       "          -2.9136e-01, -1.9957e-01,  2.7150e-01, -2.9497e-01, -7.7260e-02,\n",
       "          -6.7137e-02, -7.1911e-01, -6.6563e-01,  3.7160e-02, -6.7154e-03,\n",
       "           7.5724e-02, -2.3353e-01,  6.4373e-01,  2.9381e-01,  2.6265e-01,\n",
       "          -5.2661e-02, -2.9022e-01,  4.1878e-01,  5.6355e-01, -8.3285e-02,\n",
       "          -1.2914e-01,  2.5191e-01, -1.6888e-01,  5.8967e-02, -4.5613e-01,\n",
       "          -6.1222e-01,  2.6388e-02, -1.0019e-01,  4.3252e-01,  2.0524e-01,\n",
       "           8.4066e-01,  5.1034e-01,  2.4139e-02, -1.2164e-01,  1.4937e-01,\n",
       "          -3.9279e-01, -5.0392e-02, -3.1349e-01, -3.7812e-01, -5.6336e-01,\n",
       "          -8.2920e-01, -1.3594e-01,  9.5789e-02,  1.5284e-01, -7.1554e-01,\n",
       "          -4.1578e-01, -1.4560e-01,  8.0103e-02, -3.0503e-01, -7.9230e-02,\n",
       "          -2.2315e-02, -2.7803e-02,  2.0370e-01, -4.6967e-01, -4.7665e-01,\n",
       "           2.0189e-01,  1.8842e-01, -3.9885e-01,  1.6089e-01,  3.1255e-01,\n",
       "          -2.1373e-01,  7.2679e-02,  5.0024e-01, -4.3841e-02, -3.3644e-01,\n",
       "          -2.6233e-01,  7.4800e-01, -4.6172e-02, -3.3839e-01,  1.3299e-01,\n",
       "           1.8434e-02,  3.3970e-01,  5.1081e-01, -4.6183e-01, -1.4254e-01,\n",
       "           2.9297e-02,  6.1524e-01, -8.2191e-02, -1.8406e-01, -3.7351e-01,\n",
       "          -7.9973e-01,  1.3492e-02, -2.2072e-01, -9.5437e-03, -6.7318e-01,\n",
       "           2.3108e-01,  1.9567e-01, -3.8812e-01,  2.8227e-01, -1.7220e-01,\n",
       "           5.4303e-03,  5.8904e-01,  2.7883e-02, -7.9265e-02, -2.4715e-01,\n",
       "           1.7760e-01, -2.5420e-01, -1.5192e-01, -3.1086e-01,  1.7462e-01,\n",
       "           3.8130e-01,  2.7165e-01,  8.1509e-02,  1.3921e-01, -1.2188e-01,\n",
       "           8.6672e-01, -2.0151e-01,  6.3506e-02,  3.3862e-01, -3.1107e-01,\n",
       "           7.9508e-02,  2.8294e-01, -1.1266e-01, -2.8465e-01,  2.7597e-01,\n",
       "           3.2155e-01, -2.7336e-01, -6.7579e-01, -4.1885e-02,  9.8228e-01,\n",
       "           5.4097e-01,  5.0923e-01, -7.8628e-02, -9.6353e-02,  6.9569e-01,\n",
       "          -2.7151e-01, -6.6157e-02, -1.5785e-01, -2.6203e-01, -5.3945e-02,\n",
       "           4.0444e-01,  4.7233e-01, -1.6351e-01,  3.1030e-01,  4.1901e-01,\n",
       "          -1.3248e-02, -1.3141e-01, -3.6655e-01,  1.5841e-03,  1.0149e-01,\n",
       "          -3.3447e-01, -3.9534e-01,  3.2069e-01, -3.3008e-01, -3.0508e-01,\n",
       "           1.7542e-03, -4.7132e-02, -2.3295e-01, -2.7458e-01, -2.3847e-01,\n",
       "          -1.8564e-01, -2.9270e-01,  3.3282e-01, -1.2979e-01,  1.9156e-02,\n",
       "           2.8455e-01,  1.0631e-01, -3.2868e-01,  5.2642e-01,  2.0817e-01,\n",
       "          -3.2972e-01,  2.0132e-01, -5.4856e-01,  1.1008e-01, -2.1196e-02,\n",
       "          -3.2981e-01,  2.5885e-01,  3.5340e-01,  3.5237e-01,  5.9690e-02,\n",
       "          -3.9848e-01, -3.8610e-02,  1.2683e-01,  1.6795e-01,  1.6531e-01,\n",
       "          -2.1233e-01,  4.5524e-01,  9.4694e-02,  1.0706e-01, -4.6294e-01,\n",
       "           4.6974e-01, -9.3066e-02, -9.9754e-02,  2.3106e-01, -5.4320e-01,\n",
       "           4.4529e-01,  7.1244e-02,  9.2450e-02,  1.7399e-01, -1.1617e+01,\n",
       "           2.6461e-01,  8.0068e-01, -4.4409e-01, -3.5727e-02, -2.8187e-02,\n",
       "           2.0091e-01, -4.6935e-01,  2.0433e-01, -1.9712e-01, -9.5583e-02,\n",
       "          -1.9171e-01, -1.0732e-01,  6.8686e-02, -2.7529e-01, -3.5662e-01,\n",
       "           1.9909e-01, -7.5351e-01,  6.4530e-02, -1.0688e-01,  4.4863e-01,\n",
       "          -4.6211e-01,  6.3557e-02,  2.3006e-01, -4.4749e-01, -2.0124e-01,\n",
       "          -2.7824e-01, -8.2858e-02, -7.4699e-02,  1.0357e-01, -1.4749e-01,\n",
       "           2.4923e-01, -4.8681e-02,  4.3312e-01, -2.1869e-01, -7.3470e-01,\n",
       "          -7.2671e-02, -8.4322e-02,  2.2006e-01, -1.8369e-01, -3.1478e-01,\n",
       "           2.9691e-02, -3.6416e-01,  3.0509e-01,  2.0543e-01,  5.6374e-01,\n",
       "           1.2763e+00,  7.1969e-01, -1.3149e-01, -5.8333e-01,  2.1880e-01,\n",
       "          -4.7028e-01,  2.8213e-01, -1.0296e+00,  3.8948e-01, -1.6873e-01,\n",
       "           1.5871e-01, -3.4144e-01,  8.6015e-02, -1.9687e-01, -2.1140e-01,\n",
       "           1.1060e-02, -8.7471e-02,  3.8358e-02, -3.6275e-01, -2.7527e-01,\n",
       "           4.5205e-01,  2.2488e-01,  2.5899e-04, -1.2873e-01, -1.7915e-01,\n",
       "           1.9349e-01,  1.7946e-01, -4.2383e-02,  2.2156e-02,  4.4065e-01,\n",
       "          -1.7527e-01, -9.5596e-02,  4.4144e-01, -1.2124e-01,  9.6767e-02,\n",
       "           9.2512e-02,  5.2332e-02, -2.4078e-01, -4.3112e-01, -2.9907e-01,\n",
       "          -5.3129e-02,  2.6936e-01, -3.3048e-01,  7.0461e-01, -2.5876e-01,\n",
       "          -1.5315e-01,  7.2480e-03, -1.9619e-01,  4.5679e-01,  5.9708e-01,\n",
       "          -1.4746e-01,  3.6221e-01,  1.3861e-01,  7.0669e-03, -7.6555e-01,\n",
       "          -1.3369e-02,  1.2547e-01, -2.9664e-01,  3.8892e-02, -1.0485e-01,\n",
       "          -3.7114e-02,  4.2065e-03, -5.3498e-02,  4.7855e-01, -3.7215e-01,\n",
       "           2.5256e-01,  3.0782e-02, -1.3581e-02, -3.4819e-01, -3.1701e-01,\n",
       "           4.2182e-01,  4.0804e-01, -6.4771e-01, -4.4345e-01, -1.2335e-01,\n",
       "          -2.6616e-01,  1.3978e-01, -3.7983e-01, -6.5338e-01,  9.6642e-02,\n",
       "          -3.6057e-01, -7.2974e-02, -1.3437e-01, -4.1247e-01,  2.9852e-01,\n",
       "           2.9640e-01, -2.0136e-01,  1.7397e-01, -4.3517e-01, -3.0258e-01,\n",
       "          -3.7193e-01, -2.0569e-01,  5.7586e-01,  2.2628e-01,  8.5490e-02,\n",
       "           1.9385e-01,  1.2832e-01, -2.1589e-01,  4.0943e-01,  1.3430e-01,\n",
       "           1.3838e-01,  4.4206e-01, -3.5333e-01, -1.9270e-01, -2.2392e-01,\n",
       "          -2.8652e-01, -4.0174e-01, -1.0238e-01, -2.3623e-01, -2.6133e-01,\n",
       "          -1.8433e-01, -1.5567e-02, -8.2628e-02,  1.1969e-01,  1.1301e-01,\n",
       "           4.6444e-01,  1.4001e-01, -2.8661e-01,  3.6544e-01,  3.2100e-01,\n",
       "          -2.3475e-02, -4.4633e-01, -2.0507e-01, -3.9061e-02, -5.8902e-02,\n",
       "           3.3314e-01, -1.1795e-01, -6.6516e-01, -3.9176e-01, -5.2639e-01,\n",
       "          -1.9548e-01,  1.3145e+00, -4.0361e-01, -2.0385e-01,  8.3117e-02,\n",
       "          -9.9951e-03, -1.8413e-01, -1.1763e-01, -1.3783e-01,  4.0034e-01,\n",
       "           1.1180e-02, -2.7714e-01,  5.0436e-01, -3.7552e-01,  4.5968e-01,\n",
       "           3.0170e-01,  2.6553e-01, -1.2178e-01,  4.2196e-01, -1.6624e-01,\n",
       "          -2.2762e-01,  3.2612e-02, -3.9988e-01, -1.2578e-01,  6.7230e-01,\n",
       "          -2.0047e-01,  3.5182e-01, -2.4750e-01, -7.9776e-01, -2.1671e-01,\n",
       "           2.1137e-01,  1.4103e+00,  1.0657e-01, -1.8941e-02,  4.1653e-01,\n",
       "           1.6127e-02,  4.7903e-01,  1.7623e-01, -2.2187e-02,  4.4816e-01,\n",
       "           4.4311e-01, -3.4016e-01, -5.4887e-01,  1.3119e-01,  2.7153e-01,\n",
       "           7.1921e-01, -6.5892e-01, -4.5387e-01,  7.2496e-02,  2.0358e-01,\n",
       "          -4.6174e-02,  5.0784e-01, -1.5616e-01,  4.7412e-01,  1.8080e-01,\n",
       "          -1.0113e+00,  1.2370e-01,  1.9483e-01,  1.0745e-04, -4.1514e-01,\n",
       "          -4.8601e-01, -4.7699e-01,  8.2194e-03, -5.0960e-01,  4.1600e-02,\n",
       "           3.4278e-01, -1.1409e+00,  3.0139e-01,  4.5998e-01, -2.5776e-01,\n",
       "           8.8351e-02, -1.8485e-02,  4.1146e-01,  1.7957e-01, -2.3043e-01,\n",
       "           1.9886e-01, -1.9602e-01, -4.8360e-01, -3.9460e-01,  2.9145e-01,\n",
       "           4.4565e-01,  4.4217e-02,  4.4372e-02,  2.1813e-01, -2.0178e-01,\n",
       "          -1.4440e-01,  1.9500e-01, -1.4385e-01, -1.4474e-01,  6.4596e-02,\n",
       "           1.0054e-01, -7.2394e-01, -1.3062e-02,  3.9219e-01,  7.3007e-02,\n",
       "           3.9132e-01,  5.3921e-01,  4.4303e-01, -2.6695e-01,  2.8660e-02,\n",
       "           6.0031e-02,  4.1774e-01,  2.6973e-02,  5.6032e-01, -2.4787e-01,\n",
       "          -3.3472e-01, -1.8002e-01,  3.8496e-02, -6.7302e-01,  1.0457e-01,\n",
       "          -3.2817e-01, -6.3488e-02, -1.8878e-01, -1.2423e-01,  3.6497e-01,\n",
       "           4.9218e-02,  2.3826e-01, -6.9528e-01,  3.4946e-01, -1.3122e-01,\n",
       "           6.1595e-02,  2.8545e-01,  4.7246e-01,  5.0385e-01, -8.5157e-02,\n",
       "          -1.6988e-01, -7.0414e-01, -3.8761e-01,  6.2701e-02,  1.9431e-02,\n",
       "          -9.2516e-02, -1.1489e-01,  1.1755e-01, -6.0092e-02,  7.6993e-01,\n",
       "           6.4433e-01, -2.8590e-01, -5.6653e-01, -5.2583e-01, -1.7768e-01,\n",
       "           1.2363e-01,  8.5953e-02,  3.3321e-01,  3.7668e-01, -1.8442e-01,\n",
       "          -8.2852e-01,  1.4782e-01, -1.5650e-01,  1.4902e-01, -2.0469e-01,\n",
       "           6.1509e-01, -2.9886e-01, -1.2609e-01,  1.9811e-01,  9.6722e-02,\n",
       "          -1.6161e-02,  5.4803e-01, -1.2083e-01, -2.8660e-01, -1.2710e-01,\n",
       "           4.0935e-01, -1.9857e-01, -1.3442e-02, -3.9843e-01,  2.7544e-01,\n",
       "          -1.2134e-02,  4.3220e-01,  3.4299e-01]], device='mps:0'),\n",
       " tensor([[-0.2258, -0.7099, -0.3778,  ...,  0.3292,  0.7059,  0.7494],\n",
       "         [-0.2344, -0.5215, -0.7335,  ...,  0.3040, -0.1277,  0.6510],\n",
       "         [ 0.0350, -0.5838, -0.5010,  ...,  0.3068, -0.0212,  0.5104],\n",
       "         [-0.3292, -0.2590, -0.5016,  ..., -0.0148,  0.3509,  0.1607]],\n",
       "        device='mps:0'),\n",
       " tensor([[-4.1347e-01, -4.3901e-01, -6.9203e-01, -1.2400e+00, -1.2758e-01,\n",
       "          -5.3978e-01, -9.4498e-01,  1.6571e-01,  1.7030e-01,  2.8097e-01,\n",
       "           3.5965e-01,  8.2116e-02, -1.0276e-01, -5.2835e-01, -3.2350e-01,\n",
       "          -1.2156e-01, -2.0967e-01,  2.4758e-01,  5.0734e-01, -1.5269e-02,\n",
       "           5.8476e-01,  4.7184e-01, -1.5202e-01,  3.5045e-02, -5.0647e-01,\n",
       "           4.1017e-01, -5.1430e-01, -1.3742e-02,  4.0260e-01, -1.3499e+00,\n",
       "          -2.0352e-02,  5.3810e-01, -3.7075e-01,  8.6957e-01, -3.2733e-01,\n",
       "          -3.6168e-01, -1.0241e-01, -5.5529e-01,  7.0600e-01,  8.9833e-02,\n",
       "           4.0371e-01,  9.5415e-02,  4.1225e-01,  4.7913e-01,  2.6307e-02,\n",
       "           2.1599e-01,  1.6853e-01, -5.5185e-01,  8.6011e-01, -8.7422e-01,\n",
       "          -1.4943e-01, -4.3396e-01,  4.8912e-01, -1.6872e-01, -9.2601e-01,\n",
       "          -2.4918e-01,  1.8055e-01,  1.8479e-01, -7.6272e-02, -2.9789e-02,\n",
       "           3.9033e-01,  4.7458e-01,  1.4963e-01,  2.2658e-01, -4.4835e-01,\n",
       "           2.2515e-01, -4.1775e-01, -5.7486e-01, -5.7536e-01, -1.6014e-01,\n",
       "           1.2578e-01,  4.2070e-01,  1.1278e-01, -2.0992e-01,  2.8684e-01,\n",
       "          -4.4964e-01,  6.0778e-01, -5.6345e-01, -5.4045e-01,  5.2306e-02,\n",
       "           1.8865e-01,  2.7805e-02, -2.3450e-02,  7.1419e-02, -2.0057e-01,\n",
       "          -2.4646e-02,  3.8310e-01,  1.7152e-01,  1.6721e-02,  1.4835e-01,\n",
       "          -7.2011e-02,  4.8597e-01, -4.1974e-01, -4.2844e-01,  3.0942e-01,\n",
       "          -3.2697e-01,  5.2123e-01,  5.6855e-02,  5.4856e-03, -2.4375e-01,\n",
       "           1.6771e-01,  2.2593e-01,  2.3450e-01,  1.7855e-01, -1.0066e+00,\n",
       "          -2.6541e-01,  2.0748e-01, -1.5632e-01, -6.1310e-01, -2.3142e-01,\n",
       "          -1.4838e-01, -1.3787e-01,  4.5258e-01, -4.0332e-01, -2.5240e-02,\n",
       "          -2.9294e-02,  5.0729e-01,  7.3814e-02,  2.3105e-01,  6.5385e-01,\n",
       "           1.2047e-01,  8.2284e-02, -7.3157e-02, -1.2343e-01,  2.6213e-01,\n",
       "          -4.7535e-02,  4.2474e-01, -1.9139e-01,  1.4772e-01,  1.4938e-01,\n",
       "           1.3051e-01,  4.1845e-01, -4.1131e-01, -2.7193e-01,  1.7892e-01,\n",
       "           2.8608e-01, -4.9318e-01, -3.4831e-01, -3.5170e-01, -2.7926e-01,\n",
       "           4.1860e-01,  1.7969e-01, -1.3084e-01, -1.2310e-01,  1.9259e-01,\n",
       "          -1.1705e-01,  1.7434e-01, -4.4351e-01, -2.8846e-01, -2.0897e-01,\n",
       "          -5.7741e-01, -4.4241e-01,  3.7144e-01,  1.1653e-01, -1.3535e-01,\n",
       "          -3.6025e-02,  3.6025e-01, -2.0432e-01,  1.5873e-02, -6.4234e-02,\n",
       "           1.1118e+00, -3.6500e-02, -2.6187e-02, -1.0846e-01, -1.5465e-01,\n",
       "          -1.4975e-01, -1.2064e-01, -3.0352e-02,  2.2273e-01, -1.7906e-01,\n",
       "           2.7075e-01,  2.2002e-03, -5.4106e-02, -1.1551e-01, -1.4143e-02,\n",
       "          -2.5777e-01,  5.2695e-01,  4.1161e-02, -6.0024e-02, -4.4895e-02,\n",
       "           3.0400e-01, -1.6453e+00,  4.1127e-01, -3.0757e-01,  1.0071e-01,\n",
       "          -4.7495e-01,  1.7669e-01,  1.0879e-01,  3.7074e-01, -5.3186e-02,\n",
       "           2.6222e-01, -9.0766e-02,  3.1848e-01,  4.1244e-01, -1.7939e-01,\n",
       "           2.4022e-01,  7.0568e-01,  3.8660e-01,  2.7344e-01, -1.0624e-01,\n",
       "          -7.2187e-01,  1.1724e-01, -1.2679e-03, -1.7293e-02, -3.4049e-01,\n",
       "           2.1723e-01, -1.8107e-01,  5.2667e-02,  3.1664e-01, -2.2209e-02,\n",
       "          -7.7820e-02, -3.2395e-01, -1.2079e-01, -1.2811e-01, -2.3440e-01,\n",
       "           2.1996e-01, -3.8608e-01, -3.0281e-01,  4.3385e-01,  4.5038e-03,\n",
       "           7.8162e-02, -3.5936e-01, -6.4829e-02, -1.9983e-01, -1.0754e-01,\n",
       "           5.1851e-01, -3.5703e-01, -1.0627e-01, -2.6315e-01, -5.0310e-01,\n",
       "          -5.0344e-01, -1.0195e+00,  4.0681e-01,  5.5246e-01,  3.6681e-01,\n",
       "          -2.7182e-01, -1.0792e-01,  9.8381e-02,  1.7930e-02,  1.5674e-01,\n",
       "          -1.4252e-01, -1.7358e-01, -6.8074e-01,  3.8610e-02,  4.6670e-02,\n",
       "           4.7277e-01, -1.6596e-02,  1.1394e+00,  8.4187e-01,  6.2451e-01,\n",
       "          -7.6208e-02, -8.6604e-02,  3.7087e-01,  3.1354e-01, -4.1017e-01,\n",
       "           2.0422e-01, -1.3297e-01,  2.7548e-01,  2.2591e-01, -2.1288e-01,\n",
       "          -7.8471e-01,  2.9122e-01,  1.3361e-01,  5.8356e-01, -2.0727e-01,\n",
       "           1.2132e-01, -5.1915e-02,  4.6372e-02, -5.7989e-02,  1.7861e-01,\n",
       "          -7.7950e-01,  1.0324e-01, -4.2899e-01, -1.5142e-01, -4.4518e-01,\n",
       "           4.1054e-02,  2.2579e-01, -9.5984e-02,  2.8236e-01, -9.0202e-01,\n",
       "          -7.1843e-01,  1.1620e-01,  9.9497e-02, -6.5258e-02, -5.3326e-01,\n",
       "           3.3250e-01, -2.7684e-01, -3.2983e-01, -1.2639e-01, -2.6242e-01,\n",
       "          -5.0600e-01,  2.4265e-02, -3.2540e-01,  3.0072e-02,  2.1025e-01,\n",
       "          -4.8195e-01,  2.0958e-02,  1.3700e-02, -1.7885e-02,  2.2760e-02,\n",
       "          -7.8223e-02,  7.8753e-02, -4.8553e-01, -3.6977e-02,  4.3134e-01,\n",
       "           2.0526e-01,  4.0724e-01,  7.8622e-01, -4.8421e-01, -5.2483e-01,\n",
       "          -2.3306e-01,  8.7031e-01, -7.7444e-04, -3.1477e-01, -1.1625e-01,\n",
       "          -1.1600e+00,  6.8063e-02, -1.4372e-01, -3.8941e-01, -5.8809e-02,\n",
       "           3.1798e-01, -1.9885e-01, -7.8603e-02, -9.3834e-03, -5.6501e-02,\n",
       "           4.3016e-01,  3.8037e-01, -3.7464e-01,  8.4349e-02, -1.0404e-01,\n",
       "          -3.3917e-02,  6.8788e-02, -2.6729e-01, -6.5354e-01,  1.7252e-01,\n",
       "           3.3549e-01,  2.0758e-01, -1.8075e-01,  4.6360e-02,  4.2442e-02,\n",
       "           5.2294e-01, -2.9622e-01, -3.6275e-01,  3.7861e-01, -1.2216e-01,\n",
       "           6.3670e-02,  2.7327e-01,  8.7765e-02,  2.6647e-01, -5.0212e-01,\n",
       "          -2.8919e-02,  6.7957e-02, -7.1600e-02, -1.7311e-01,  2.5801e-01,\n",
       "           3.4067e-01,  1.5975e-01, -3.4364e-01,  5.0749e-01,  1.9141e-01,\n",
       "          -3.3060e-01,  3.6913e-01, -4.0219e-01, -1.5877e-01,  1.5324e-01,\n",
       "           3.4153e-02,  1.2379e-01, -2.5036e-01,  2.4639e-02,  2.2318e-01,\n",
       "           3.1704e-01, -6.8085e-01,  3.3720e-01,  5.7455e-02,  5.3831e-01,\n",
       "          -2.3447e-01, -5.6427e-01,  3.4953e-01, -4.5018e-01, -2.0726e-01,\n",
       "          -1.2084e-01,  1.0308e-01,  1.7102e-01, -2.2628e-01, -2.0695e-01,\n",
       "          -6.7442e-02,  3.5347e-01,  2.6274e-01,  1.4673e-01,  2.0698e-01,\n",
       "           1.2940e-01,  1.2904e-01, -2.1142e-01,  6.5848e-01,  5.0060e-01,\n",
       "          -2.8802e-01,  3.3236e-01, -8.1326e-02,  4.3830e-01, -5.1184e-01,\n",
       "          -1.9754e-01,  1.3808e-01,  6.2942e-02, -1.5498e-02,  2.0835e-01,\n",
       "          -5.4719e-01,  2.7611e-01,  7.4925e-01, -5.3120e-01, -1.0939e-01,\n",
       "          -3.2811e-01,  4.7193e-01, -4.0461e-01,  3.1759e-01, -1.0236e-01,\n",
       "          -3.4291e-02,  3.0994e-02,  2.2664e-01,  8.2122e-02, -2.7282e-01,\n",
       "           2.0485e-01, -1.5531e-01,  5.1384e-01,  4.4340e-01, -1.1242e+01,\n",
       "           2.3908e-01,  8.4745e-01, -5.1019e-01,  1.6000e-02, -4.6411e-02,\n",
       "           2.0085e-01, -3.8402e-01,  1.9119e-01, -2.5953e-01, -1.7598e-01,\n",
       "          -6.7264e-02,  2.0738e-01,  7.1337e-02, -5.5430e-01, -9.3066e-02,\n",
       "          -3.1437e-02, -7.1368e-01, -2.7665e-02, -5.7804e-02,  6.2039e-01,\n",
       "          -4.4375e-01,  2.6925e-01,  8.2842e-02, -2.4656e-01, -4.0836e-01,\n",
       "          -4.4657e-01, -8.5993e-03,  1.3328e-01,  5.6668e-01, -2.0730e-01,\n",
       "          -1.8187e-01, -6.5063e-02,  3.1947e-01, -1.3541e-01, -2.0534e-01,\n",
       "           1.5417e-01, -2.1345e-01,  7.0728e-01,  6.9934e-02, -7.0345e-02,\n",
       "          -5.6883e-02, -1.9867e-01,  5.5390e-01,  4.9770e-01,  4.3056e-01,\n",
       "           1.5165e+00,  5.8027e-01,  5.8628e-02,  1.5969e-01,  1.8981e-01,\n",
       "          -7.3080e-01,  2.5380e-01, -5.2006e-01,  2.4096e-01, -2.0139e-01,\n",
       "           5.1996e-01, -4.7331e-01,  3.0967e-02, -3.3406e-01, -9.7831e-02,\n",
       "           8.0008e-02,  1.3622e-01,  6.2633e-02, -4.7986e-01, -1.3936e-01,\n",
       "           3.5606e-01, -1.8090e-01,  3.1201e-01, -6.2307e-01,  5.6602e-02,\n",
       "          -3.0741e-01,  5.8207e-01, -2.7221e-01,  3.6218e-01,  3.6927e-01,\n",
       "           2.8972e-01, -2.1961e-01,  1.3026e+00, -1.7115e-01, -3.3722e-02,\n",
       "           2.2786e-01, -1.2350e-01, -2.6429e-01, -6.6320e-01, -2.4745e-01,\n",
       "           1.6069e-02,  1.3563e-01, -1.1188e-01,  3.3520e-01, -4.8001e-01,\n",
       "          -3.5078e-01, -1.4201e-01, -1.4978e-01,  4.0511e-01,  3.0024e-01,\n",
       "          -7.7571e-01, -2.1894e-01, -1.5439e-01,  1.2473e-01, -9.2506e-01,\n",
       "          -5.6284e-01, -2.5704e-01, -4.5411e-01,  3.9263e-01,  2.9206e-01,\n",
       "          -1.4189e-01, -1.0567e-01,  3.8574e-01,  1.0989e-01, -4.5921e-01,\n",
       "          -1.2140e-03,  6.0162e-01,  6.2252e-02, -2.7942e-01, -1.2940e-01,\n",
       "           2.0027e-02,  2.4066e-01, -5.3136e-01, -6.8256e-01,  3.3435e-03,\n",
       "          -3.7094e-02, -1.7993e-02,  3.7147e-01, -4.9197e-01,  3.8788e-01,\n",
       "          -5.4376e-01,  1.2668e-01,  1.1874e-02,  5.7590e-03,  4.1686e-01,\n",
       "           1.2091e-01, -2.5502e-01,  6.6470e-01, -4.1684e-01, -2.1558e-01,\n",
       "          -4.5694e-01, -3.0175e-01,  5.3438e-01, -6.8341e-01, -7.2711e-02,\n",
       "          -1.5036e-01,  1.1447e-01, -2.8205e-01,  3.8856e-02,  2.3223e-01,\n",
       "           4.0479e-01,  6.7164e-01, -2.0496e-01,  2.7816e-02,  5.4446e-02,\n",
       "          -9.2790e-02, -2.7517e-01, -1.4573e-01,  2.6468e-01, -2.0848e-01,\n",
       "          -3.0230e-02, -3.1652e-01, -2.0353e-01, -5.2899e-02,  8.2162e-02,\n",
       "           2.6949e-01,  8.5283e-02, -1.9947e-01,  6.1727e-01,  2.9950e-01,\n",
       "           5.3195e-02, -8.0589e-01, -8.1932e-02,  5.0664e-02,  6.9055e-02,\n",
       "          -1.9313e-02, -2.4064e-01, -2.2741e-01, -5.3817e-01, -3.3156e-01,\n",
       "           1.6466e-01,  7.2254e-01, -3.7743e-01, -3.0078e-01,  1.7487e-01,\n",
       "          -8.9394e-03, -3.8473e-01,  4.5676e-02, -2.8434e-01,  1.4539e-01,\n",
       "           1.7455e-02, -1.7413e-01,  3.2077e-01, -9.7799e-02,  5.2128e-01,\n",
       "           8.0309e-02,  1.0727e-01,  1.3937e-01,  3.8618e-01, -2.5429e-01,\n",
       "          -5.7749e-01, -1.3265e-01, -2.2005e-01,  7.3497e-02, -3.1132e-01,\n",
       "          -2.8688e-02, -1.3505e-02, -3.7635e-02,  2.1870e-02,  2.7179e-01,\n",
       "           1.5133e-01,  1.2578e+00, -4.3966e-02,  2.3948e-01,  2.7142e-01,\n",
       "           4.6736e-01,  3.7462e-01,  3.6234e-01,  1.3319e-01,  4.3524e-01,\n",
       "           3.7599e-01,  1.5735e-01, -3.1410e-01,  2.4755e-01,  2.2484e-01,\n",
       "           1.2268e-01, -7.3828e-01, -5.6706e-01, -1.9613e-01,  6.7149e-01,\n",
       "          -2.0438e-01,  5.6153e-01, -2.3400e-01, -7.4849e-02,  2.1643e-01,\n",
       "          -7.2049e-01, -7.9486e-02, -5.3689e-01,  3.1082e-01, -2.6091e-01,\n",
       "          -7.0177e-01, -3.4580e-01, -4.4542e-02, -3.3996e-01,  3.4828e-01,\n",
       "           4.2412e-01, -4.4169e-01, -1.7591e-02,  3.5534e-01,  9.8777e-02,\n",
       "           3.1740e-01, -2.3717e-01,  9.9333e-02,  4.3405e-01,  4.8472e-01,\n",
       "          -4.0921e-01, -1.4780e-01, -6.5222e-01, -6.2641e-01,  1.4806e-01,\n",
       "           4.2285e-01,  1.9914e-01,  2.3008e-01,  7.4865e-02, -5.3168e-01,\n",
       "          -6.2918e-02, -2.7565e-01, -3.5760e-01, -8.0286e-02,  4.2289e-01,\n",
       "          -2.0397e-01,  3.5604e-02,  2.6468e-01,  3.3542e-01, -2.4597e-01,\n",
       "           4.7931e-01,  6.2135e-01,  9.3573e-01, -2.2083e-02, -3.7265e-01,\n",
       "          -2.7457e-01,  4.8271e-01, -1.3799e-01,  5.4499e-01, -3.9921e-01,\n",
       "           6.6887e-02,  2.2925e-02, -1.0567e-01, -6.6472e-01,  4.9038e-02,\n",
       "          -4.0117e-01,  5.0421e-01, -1.4728e-01, -3.5309e-01,  4.6237e-01,\n",
       "           2.0585e-01,  2.7651e-01, -7.4734e-01, -5.7675e-02,  3.8886e-01,\n",
       "          -1.3416e-01,  3.2545e-01,  1.0133e-01,  2.6927e-01,  8.6497e-02,\n",
       "           2.2631e-01, -4.7220e-01,  9.3591e-02,  3.2818e-01, -3.6883e-02,\n",
       "          -9.2540e-02, -6.4405e-01,  2.4280e-01,  3.1125e-01,  8.7650e-01,\n",
       "           5.1471e-01, -1.2370e-02, -5.5263e-01, -5.5901e-01,  3.9163e-01,\n",
       "           3.4511e-01, -2.7857e-01,  5.7488e-03, -1.6368e-01,  6.0814e-02,\n",
       "          -5.0879e-01, -1.6377e-01,  1.9272e-02, -5.5369e-02, -1.0304e-01,\n",
       "           4.7687e-01, -4.8516e-01,  3.4733e-01,  2.6517e-01,  1.5993e-01,\n",
       "           6.4958e-01,  2.2887e-01,  5.3471e-01, -5.2067e-01, -1.4776e-01,\n",
       "           3.0486e-01, -2.3907e-01, -1.7167e-01, -8.0743e-01,  3.9924e-01,\n",
       "           3.6913e-01,  4.9033e-01,  6.0624e-01]], device='mps:0'),\n",
       " tensor([[-0.2258, -0.7099, -0.3778,  ...,  0.3292,  0.7059,  0.7494],\n",
       "         [-0.0932, -0.6293, -0.4879,  ...,  0.3041, -0.1958,  0.5576],\n",
       "         [-0.3292, -0.2590, -0.5016,  ..., -0.0148,  0.3509,  0.1607]],\n",
       "        device='mps:0'),\n",
       " tensor([[ 1.1277e-01, -5.1182e-01, -5.1456e-01, -9.9081e-02, -3.0116e-01,\n",
       "           3.3007e-03, -4.0947e-01,  4.0427e-01, -5.8056e-02,  7.3590e-01,\n",
       "           2.2527e-01,  8.2791e-02, -3.8724e-02, -7.5126e-02, -1.5840e-01,\n",
       "          -7.3986e-01,  3.8029e-02,  3.1506e-01,  2.4049e-01,  1.5258e-01,\n",
       "           8.4897e-01,  3.7601e-01,  3.7846e-02,  1.5338e-01, -8.9169e-01,\n",
       "           2.4549e-02,  1.7534e-01, -3.8121e-01,  3.0474e-01, -9.4057e-01,\n",
       "          -3.1590e-02,  4.1782e-01, -3.6549e-01,  9.4881e-01, -2.1749e-01,\n",
       "          -2.3738e-01,  2.7768e-01,  1.1423e-01,  1.5727e-03,  8.7533e-02,\n",
       "           6.3608e-01, -3.9554e-01,  3.5800e-01,  1.7469e-01,  5.4999e-01,\n",
       "          -9.7317e-02,  1.1774e-02, -5.5940e-01,  4.2788e-01, -2.4478e-01,\n",
       "          -8.0416e-02, -1.6983e-01,  3.4295e-01,  4.9191e-01, -6.8602e-01,\n",
       "          -5.2333e-01, -6.2734e-02,  1.8164e-01, -2.1619e-01,  2.9696e-01,\n",
       "           1.3140e-01,  5.4967e-01,  1.4880e-01,  8.9218e-02, -9.8997e-02,\n",
       "           2.7962e-01, -6.9163e-01, -1.1292e-01, -4.6817e-01, -1.7222e-01,\n",
       "           7.9175e-02,  3.5231e-01, -2.4967e-01, -7.3913e-02,  7.4744e-01,\n",
       "          -3.1754e-01,  3.1779e-01, -1.0437e-01, -5.2653e-01,  2.5923e-01,\n",
       "           5.3624e-01,  2.2665e-01, -3.8162e-01, -4.3491e-01, -2.9896e-01,\n",
       "           3.1739e-01,  3.6286e-01,  2.1409e-01,  4.1664e-01,  2.3004e-01,\n",
       "           6.9510e-01,  1.1862e-01, -9.0323e-01, -1.9789e-01,  3.0084e-01,\n",
       "          -2.8430e-01,  3.6802e-01, -1.0774e-01,  1.9963e-01,  2.4580e-01,\n",
       "           9.4402e-02,  1.0678e-01, -1.2301e-01,  3.3043e-01, -7.3431e-01,\n",
       "          -3.8387e-01,  2.7538e-01,  7.5278e-02, -7.1095e-01, -5.3743e-02,\n",
       "          -1.3470e-01,  2.0865e-01,  1.6720e-01, -4.9918e-01, -2.6117e-01,\n",
       "           8.9412e-02, -1.5191e-01,  3.4209e-02, -4.2780e-01,  3.6346e-01,\n",
       "           2.9865e-01, -4.8785e-01,  1.3818e-01,  1.4371e-01,  4.4715e-01,\n",
       "          -4.1340e-01,  3.8903e-01, -2.5446e-01,  2.2889e-01,  8.2835e-01,\n",
       "          -1.2485e-01,  2.6252e-02, -4.0830e-02, -1.7487e-01, -2.8440e-01,\n",
       "          -3.4081e-01,  2.2259e-01, -7.0968e-01, -2.4259e-01,  1.8416e-01,\n",
       "           2.6748e-01, -3.7059e-02, -4.8123e-01, -1.1090e-01,  2.2158e-01,\n",
       "          -2.0902e-01, -3.0049e-01,  7.2718e-02, -2.7801e-01, -3.0796e-01,\n",
       "          -1.0773e+00, -6.4338e-01,  3.3934e-01,  1.0304e-01, -8.3425e-02,\n",
       "          -3.2985e-02,  2.6094e-01,  1.2604e-01, -1.1825e-01,  2.1861e-02,\n",
       "           1.4631e-01,  1.5870e-01,  4.0601e-01, -7.2045e-01, -2.6279e-01,\n",
       "           5.1222e-01, -6.9927e-02, -1.1101e-01, -3.8262e-01,  1.1619e-01,\n",
       "           9.3115e-02, -4.6417e-01,  1.4190e-02, -5.3166e-01,  2.5739e-01,\n",
       "          -6.9917e-01,  2.6343e-01,  7.8697e-02,  4.6975e-02, -5.2005e-01,\n",
       "          -5.1269e-01, -1.6613e+00,  4.4969e-02, -1.1536e-01,  4.2832e-01,\n",
       "          -6.4780e-01,  6.8155e-01,  4.9667e-01,  1.3957e-01, -1.0725e-01,\n",
       "          -1.1613e-01, -2.0495e-01, -1.3738e-01,  4.8684e-01, -1.1487e-01,\n",
       "           4.7349e-01,  6.1281e-01,  3.6342e-01,  1.0458e-01,  3.8659e-02,\n",
       "          -5.1090e-01,  3.2178e-01, -6.3780e-02,  6.5283e-02, -4.0803e-03,\n",
       "           5.2045e-01, -1.5508e-01, -5.5974e-02,  4.7542e-01, -9.0455e-02,\n",
       "           2.1110e-02, -5.0237e-01, -2.0562e-01, -1.7373e-01,  3.3842e-02,\n",
       "           6.1372e-01,  9.7513e-02, -1.1752e-01,  6.5636e-01,  6.4063e-01,\n",
       "          -4.3595e-02, -1.1716e-01, -2.8002e-01, -4.5596e-01, -3.9645e-01,\n",
       "           4.4838e-01, -4.1673e-01, -3.8331e-02, -1.3722e-01, -2.2336e-03,\n",
       "           2.9366e-01, -7.7720e-01,  7.8321e-01,  2.7313e-01,  1.2754e-01,\n",
       "          -2.9649e-01, -3.8133e-01, -8.9398e-02,  1.1919e-01,  8.8294e-02,\n",
       "           8.6567e-02, -6.9762e-01, -4.4501e-01,  2.6233e-01, -1.7112e-02,\n",
       "           2.9408e-02, -4.8063e-01,  9.5151e-01,  3.2366e-02,  2.1866e-01,\n",
       "          -8.1848e-02, -2.3095e-01,  2.8605e-01,  5.2168e-02, -4.2972e-01,\n",
       "          -1.1705e-01, -5.8341e-02, -3.5230e-01,  3.8333e-01, -5.6102e-01,\n",
       "          -2.8672e-01,  4.4379e-02,  2.3252e-01,  3.8893e-01, -1.8287e-02,\n",
       "           3.8934e-01, -9.1873e-02, -5.2896e-01, -2.6387e-01,  3.8154e-01,\n",
       "          -2.8519e-01,  3.6897e-01, -4.2110e-01, -2.0175e-01, -6.3157e-01,\n",
       "          -7.1356e-01,  3.0949e-01, -4.4762e-01,  2.3137e-01, -9.6326e-01,\n",
       "          -4.6904e-01,  1.5038e-01, -3.9511e-02,  7.9127e-02, -2.8288e-01,\n",
       "           3.9424e-01,  2.3657e-01,  1.0638e-01, -1.3625e-02, -6.8624e-01,\n",
       "          -1.7361e-01,  5.1713e-01, -3.0739e-01,  3.0249e-01,  3.2178e-01,\n",
       "          -1.5219e-01, -2.7008e-01,  2.0505e-01, -6.8380e-02, -1.3222e-01,\n",
       "          -2.4847e-01, -2.3051e-01,  7.3440e-02, -2.9512e-01, -4.0589e-01,\n",
       "          -1.9683e-01,  3.5653e-01,  3.1357e-01, -1.9608e-01, -4.0936e-01,\n",
       "          -3.1297e-02,  6.9414e-01, -8.3245e-02,  2.4383e-01, -1.2362e-01,\n",
       "          -1.4305e+00, -4.0833e-01, -1.8540e-02, -1.3066e-01, -2.5434e-01,\n",
       "           2.5155e-01,  6.0078e-02, -4.4774e-02, -2.3862e-02, -3.9032e-01,\n",
       "           4.2100e-01,  9.6530e-02, -5.5141e-01,  1.4950e-01, -2.0711e-01,\n",
       "          -1.2827e-01, -1.0196e-01,  5.8219e-02,  1.1231e-02, -1.1398e-01,\n",
       "           1.6417e-01, -2.6502e-02,  2.1900e-01,  8.7820e-02, -3.6880e-01,\n",
       "           5.6536e-01, -4.0455e-01, -9.2397e-02,  2.2993e-01, -6.7463e-01,\n",
       "          -5.8048e-01,  3.5097e-01,  2.4469e-01, -1.4927e-01,  8.5738e-02,\n",
       "           2.5571e-01, -1.1733e-01, -1.8922e-01,  1.1565e-01,  5.0862e-01,\n",
       "           1.6608e-01,  1.5372e-01, -3.3662e-01,  6.7240e-01,  6.1452e-01,\n",
       "          -6.7974e-01,  3.0024e-01, -8.6431e-02, -7.1191e-02, -7.8438e-02,\n",
       "           2.5682e-01,  2.9384e-01, -2.6061e-01,  3.1474e-01,  1.1061e-01,\n",
       "           8.3275e-02, -1.8821e-02, -4.8232e-02,  1.8014e-01,  6.5893e-01,\n",
       "          -6.6205e-02, -4.0549e-01,  3.5343e-01, -4.1343e-01, -2.6268e-01,\n",
       "          -7.5984e-02, -3.7208e-02,  6.5495e-02, -3.1240e-01, -3.4070e-01,\n",
       "          -4.4867e-01, -4.5409e-02,  5.9015e-01, -1.3339e-01, -1.2243e-01,\n",
       "           6.4085e-03,  1.1937e-01, -3.9825e-01,  4.0693e-01, -1.6292e-01,\n",
       "          -6.7273e-03,  1.6511e-01, -4.9061e-01,  5.9405e-01,  2.2158e-01,\n",
       "          -1.9118e-01,  7.7159e-02,  3.7849e-01,  2.9315e-01,  7.3291e-02,\n",
       "          -6.2785e-01, -1.9814e-01,  1.2483e-01,  8.7468e-02,  1.3790e-01,\n",
       "           5.2048e-01,  8.7935e-01, -4.8997e-01,  1.1694e-01, -4.3027e-02,\n",
       "           1.9913e-01, -2.3995e-01,  3.2426e-01,  2.5199e-02, -6.5157e-01,\n",
       "           5.7959e-01, -5.5682e-02,  1.2983e-01, -8.3751e-04, -1.1148e+01,\n",
       "          -2.4697e-02,  6.7893e-01, -5.5838e-01,  2.0324e-01,  1.4167e-01,\n",
       "           2.7728e-01, -3.5224e-01,  2.1719e-04, -1.7645e-01, -4.1684e-01,\n",
       "          -2.9137e-01,  1.3225e-01,  2.1139e-01, -2.9905e-01,  8.3671e-02,\n",
       "          -6.6111e-02, -7.4170e-01,  4.0852e-01,  6.7087e-02,  3.1935e-01,\n",
       "          -8.0690e-02,  3.8597e-01, -1.1090e-01, -1.3520e-02, -3.8995e-01,\n",
       "           1.1861e-01,  2.4902e-02,  5.2355e-02,  3.9441e-02, -2.2822e-01,\n",
       "           2.3118e-01,  4.8091e-02,  4.5015e-01, -2.8330e-01, -7.0757e-01,\n",
       "          -1.9601e-01, -1.5070e-02,  1.7927e-02, -1.7013e-02, -2.8357e-01,\n",
       "           6.6553e-02, -5.8297e-01,  2.2333e-01,  1.1373e-01,  8.5199e-01,\n",
       "           9.8358e-01,  6.7080e-01,  7.6940e-02, -5.4386e-02, -1.6916e-02,\n",
       "          -5.6321e-01,  1.6130e-01, -8.1082e-01, -1.3062e-01, -2.1928e-01,\n",
       "           7.0352e-01, -3.6822e-01,  1.1476e-01,  2.6222e-01, -3.4481e-01,\n",
       "           2.0382e-01,  3.7136e-02,  3.1790e-01, -4.3110e-01, -5.0590e-01,\n",
       "           5.2232e-01, -7.6970e-03,  1.8870e-01, -5.3201e-01,  2.1314e-01,\n",
       "          -3.1748e-01,  2.6052e-01,  1.8880e-01, -5.2874e-01, -2.0760e-01,\n",
       "          -4.9469e-01,  1.3613e-01,  8.4277e-01,  1.4568e-01,  7.7079e-02,\n",
       "           3.6946e-01,  2.0022e-01, -4.6711e-01, -7.1561e-01, -4.9342e-01,\n",
       "           2.4731e-01, -2.3498e-01,  3.4844e-01,  5.7382e-01, -7.8887e-02,\n",
       "          -4.7290e-01,  1.1404e-01,  1.6852e-01,  2.4731e-01,  6.4383e-01,\n",
       "          -3.0459e-01, -4.1848e-02, -2.9485e-02,  9.0120e-02, -3.1542e-01,\n",
       "          -3.7731e-01, -3.2564e-01, -1.7041e-01,  4.5536e-01, -4.5159e-02,\n",
       "           3.8940e-01,  2.2186e-01,  5.9770e-02,  1.9004e-01, -8.8353e-01,\n",
       "           3.4893e-01, -2.2614e-01,  3.7178e-01, -1.8391e-01, -4.1715e-01,\n",
       "           2.0961e-01,  5.8875e-01, -2.8693e-01, -4.0336e-01, -2.5849e-01,\n",
       "           2.3852e-01,  1.3124e-01, -1.3747e-01, -8.3652e-01,  1.2572e-01,\n",
       "          -6.3280e-01, -8.3095e-02, -2.7914e-02,  5.4686e-01,  8.0290e-01,\n",
       "           2.7624e-01, -6.7492e-02,  8.6376e-02, -3.8386e-01, -6.1207e-01,\n",
       "          -6.3800e-01,  1.3163e-02, -1.3512e-01, -7.8098e-01,  2.1135e-02,\n",
       "           2.8876e-02, -2.0300e-02,  5.8427e-02, -2.3546e-01,  1.7729e-01,\n",
       "           4.8705e-01,  3.6141e-01, -1.2408e-01,  5.9419e-02, -3.7922e-01,\n",
       "          -6.2190e-01, -8.6745e-02,  7.7854e-02,  1.6172e-01, -5.1278e-01,\n",
       "          -2.9601e-01,  1.1747e-01, -3.1322e-02,  2.7528e-02, -1.9899e-01,\n",
       "           1.3669e-01, -2.2503e-01, -1.2000e-02,  2.7866e-01,  4.6635e-01,\n",
       "           1.1528e-02, -2.7282e-01, -4.3857e-01, -1.8906e-02,  1.5881e-01,\n",
       "           2.4698e-01, -2.2711e-01, -3.0300e-01, -3.8354e-01,  2.0940e-01,\n",
       "           2.9705e-01,  1.0639e+00, -4.8833e-01, -4.0899e-01,  1.9583e-01,\n",
       "           3.5230e-01, -1.0565e-01, -2.8806e-01, -3.5628e-01,  2.0647e-01,\n",
       "           1.1352e-01, -8.0433e-02, -3.4162e-02, -7.7830e-02,  5.1550e-01,\n",
       "           1.6482e-01,  4.7041e-01,  3.0757e-01,  4.8310e-01,  1.9579e-01,\n",
       "          -1.9577e-01, -8.1087e-02, -6.1276e-01, -5.7265e-01,  1.7477e-01,\n",
       "          -3.4696e-02,  1.7945e-01,  2.9475e-01, -5.4680e-01,  3.5253e-01,\n",
       "           3.9298e-01,  1.0477e+00,  3.8225e-02, -1.9548e-01,  4.0182e-01,\n",
       "           3.8969e-01, -4.5758e-01, -5.5331e-02,  3.1775e-01,  9.0078e-02,\n",
       "           4.2217e-01,  2.6969e-01, -2.5045e-01, -2.6832e-01,  1.3747e-01,\n",
       "           3.6461e-01, -5.7917e-01, -3.7681e-01, -1.5167e-02,  5.5110e-01,\n",
       "          -2.4375e-01,  2.1840e-01,  4.5848e-02,  3.7804e-01,  3.7463e-01,\n",
       "          -1.0699e+00,  1.6807e-01, -3.2041e-01,  2.7435e-01, -2.4271e-02,\n",
       "          -7.8250e-01, -3.4543e-02, -1.9427e-01, -1.1596e-01,  3.3735e-01,\n",
       "           4.1202e-01, -7.9805e-01,  3.3972e-01,  2.3964e-01, -3.8364e-01,\n",
       "           4.7577e-02, -1.7531e-01,  2.4049e-01,  3.3544e-01,  3.2362e-02,\n",
       "          -1.4892e-01,  1.8644e-01, -1.9447e-01, -8.0775e-01,  5.7712e-01,\n",
       "           6.0096e-01, -2.1005e-01,  1.4164e-01,  1.1858e-01, -8.4184e-02,\n",
       "          -3.2385e-01, -3.8343e-01, -1.2416e-01,  1.6063e-01,  9.6145e-02,\n",
       "          -9.8046e-02, -4.7037e-01, -4.0440e-02,  3.8794e-01, -3.9096e-01,\n",
       "           4.3372e-01,  3.4804e-01,  5.3842e-01, -2.8332e-01, -4.2575e-02,\n",
       "           1.7443e-01,  5.7542e-01, -1.1207e-01,  3.8094e-01, -3.8443e-01,\n",
       "          -8.6311e-03,  4.5158e-01, -1.4772e-01,  7.1035e-03,  2.3732e-01,\n",
       "          -5.0991e-01,  2.9103e-01,  1.0851e-01, -5.6297e-01,  5.5767e-01,\n",
       "           5.0509e-01, -7.6896e-02, -4.0047e-01,  1.5744e-01,  2.9919e-01,\n",
       "           3.9453e-02,  7.8684e-01,  2.8627e-01,  3.8282e-01,  2.7073e-01,\n",
       "          -6.7450e-02, -5.4737e-01, -3.2683e-01, -5.6482e-03,  2.6335e-01,\n",
       "           1.1920e-01, -3.6386e-01,  3.2594e-01,  1.8073e-01,  5.2975e-01,\n",
       "           6.1907e-01, -1.8066e-01, -9.0544e-01, -5.6162e-01,  9.6928e-02,\n",
       "          -1.8944e-01, -1.1782e-01, -1.4313e-01,  5.4460e-02,  1.6554e-01,\n",
       "          -7.7022e-01, -5.7397e-01, -1.9516e-01,  2.1820e-01, -3.5735e-01,\n",
       "           6.6382e-01,  4.6434e-02, -3.8504e-02,  3.3171e-01,  3.8933e-01,\n",
       "           2.1859e-01,  3.6118e-01, -5.0593e-02, -5.7660e-01,  1.3136e-02,\n",
       "           9.6468e-02, -4.2191e-01, -1.8386e-01, -4.0805e-01,  5.0626e-01,\n",
       "           3.3973e-01,  9.6001e-01,  5.3169e-01]], device='mps:0'),\n",
       " tensor([[-6.1750e-01, -4.2829e-01, -2.3934e-01, -9.4116e-01, -3.4548e-01,\n",
       "          -6.5496e-02, -6.3874e-01,  9.7828e-01, -9.5892e-02,  4.0096e-01,\n",
       "           1.4919e-01, -5.1603e-03,  2.0980e-01,  1.3680e-01, -6.9506e-01,\n",
       "           1.6740e-01,  4.6094e-01,  2.1159e-01,  3.5951e-01, -4.4329e-02,\n",
       "           7.9832e-01,  3.4245e-01,  1.8019e-01, -2.5205e-01, -4.5876e-01,\n",
       "           5.8994e-01, -2.4628e-01, -9.2014e-02,  2.0509e-01, -8.1949e-01,\n",
       "          -1.3586e-01, -4.9757e-02, -3.6281e-01,  8.7602e-01,  1.1948e-01,\n",
       "          -2.7310e-01, -1.5276e-01, -2.7996e-01,  2.3494e-01, -6.3637e-02,\n",
       "           3.7814e-01,  2.6592e-02, -2.9347e-01,  3.1386e-01, -1.6421e-01,\n",
       "          -1.4057e-01,  1.4047e-02, -7.8429e-01,  1.3666e-02,  1.8955e-01,\n",
       "           5.8851e-02, -4.7205e-02,  1.9231e-01,  3.7030e-01, -1.0617e+00,\n",
       "          -2.6348e-01,  2.7603e-01,  2.9849e-01, -1.7932e-01, -5.7784e-02,\n",
       "          -1.9989e-01,  9.0201e-01,  1.7949e-01, -2.7822e-01,  1.7824e-01,\n",
       "          -4.4899e-01, -5.4601e-01, -6.0781e-01, -1.0682e-01, -1.1967e-01,\n",
       "          -1.0002e-01,  9.7528e-02,  1.4009e-01, -2.9385e-01,  2.3170e-01,\n",
       "           2.1813e-01,  4.7024e-02, -2.7652e-02, -4.3158e-01, -2.6546e-01,\n",
       "           3.6947e-01,  3.4425e-02, -1.4730e-02,  2.0326e-01, -5.8744e-01,\n",
       "          -2.0226e-01,  3.7959e-01,  1.5703e-01,  2.5914e-01,  2.2662e-03,\n",
       "           5.9967e-01,  1.2024e-02,  8.0339e-02, -4.5369e-01,  5.8427e-01,\n",
       "          -5.4955e-01,  1.0329e-01, -1.6007e-01,  4.9532e-02, -5.2088e-01,\n",
       "          -9.8735e-02,  2.4494e-01, -1.8085e-01,  1.6053e-01, -4.6592e-01,\n",
       "          -3.2983e-02,  2.4514e-01,  3.0468e-01, -4.9183e-01,  4.2685e-02,\n",
       "          -6.5134e-03, -2.9791e-01,  7.3694e-02, -6.6407e-01, -8.5795e-01,\n",
       "           5.8354e-01,  1.8428e-02, -4.5240e-01, -2.3930e-01,  7.4443e-01,\n",
       "           2.9738e-01, -2.3201e-01, -5.3072e-02, -3.5548e-01,  2.0259e-01,\n",
       "          -1.8518e-01,  2.5689e-01, -2.7018e-01,  3.8439e-01,  5.5026e-01,\n",
       "           3.1581e-01,  2.8489e-01, -1.8327e-01,  2.7637e-01,  1.4465e-01,\n",
       "           3.6947e-02, -6.6614e-02, -1.5929e-01,  1.5180e-01,  1.2041e-01,\n",
       "          -2.9580e-01, -3.2999e-01,  5.4746e-03,  1.1182e-01, -3.3279e-01,\n",
       "          -1.0202e-01,  1.8680e-01,  3.4808e-01, -1.4092e-01, -2.5403e-01,\n",
       "          -7.8984e-01, -4.2328e-01,  1.0217e-01, -1.8560e-01, -1.7135e-01,\n",
       "          -4.1911e-01,  3.8121e-01, -6.6660e-02,  4.7638e-01,  3.4189e-01,\n",
       "           6.9978e-01, -1.5311e-01,  5.9776e-02, -5.6412e-01,  3.5197e-01,\n",
       "           3.8270e-02, -6.1110e-01, -1.7247e-01,  1.5680e-01, -4.1813e-04,\n",
       "           6.8195e-01,  7.0979e-02,  4.0124e-01,  1.2070e-01, -3.2553e-01,\n",
       "           8.2089e-03,  1.9890e-01, -1.2864e-01, -1.4297e-01,  7.3766e-01,\n",
       "          -5.5188e-01, -2.3029e+00,  1.0721e-01, -3.5844e-01, -3.6529e-01,\n",
       "          -1.8135e-01,  3.7255e-01,  3.3772e-02,  4.9966e-01,  1.5486e-01,\n",
       "           3.7598e-02,  1.6729e-01, -2.8025e-02,  3.6851e-01, -8.4399e-02,\n",
       "           2.6751e-01,  3.3341e-01,  1.2988e-01, -2.3298e-01, -1.0956e-01,\n",
       "          -3.9227e-01,  2.4031e-01, -9.8916e-02,  8.6010e-01,  2.3260e-01,\n",
       "           2.0943e-01, -4.3657e-03, -4.7262e-02,  4.8964e-01,  4.3099e-01,\n",
       "          -6.6541e-02, -1.6585e-01, -4.9522e-01,  4.6269e-03, -4.1279e-02,\n",
       "          -1.2963e-02, -4.5611e-01,  1.3753e-01,  4.2841e-01, -1.2416e-01,\n",
       "          -1.8354e-01,  1.5525e-01, -4.7715e-02, -3.1517e-01, -2.6492e-02,\n",
       "           6.8453e-01,  1.9080e-01, -2.2994e-01,  1.2377e-01, -1.1896e-01,\n",
       "           3.7006e-02, -4.6975e-01,  7.0044e-01,  6.1147e-01,  1.5585e-01,\n",
       "           1.6214e-02, -6.9943e-01,  2.3699e-01, -3.1574e-01, -4.7647e-03,\n",
       "          -1.5027e-01, -1.5731e-01, -4.2986e-01, -5.7285e-02, -5.5854e-02,\n",
       "           4.0115e-01,  7.9943e-02,  4.6653e-01,  1.1837e+00,  2.7973e-01,\n",
       "           1.9184e-02,  5.6097e-01,  2.0801e-01,  4.2492e-01, -4.0180e-01,\n",
       "           3.3082e-01,  8.8825e-02, -6.3896e-01,  2.7188e-01, -4.1593e-01,\n",
       "           4.4963e-02, -4.1432e-02, -3.7926e-01,  1.4208e-01, -2.1276e-02,\n",
       "           3.4237e-01,  3.3517e-01,  1.5186e-01,  1.6838e-01,  2.9411e-01,\n",
       "          -3.1322e-01,  5.9715e-01, -6.0809e-02, -1.1127e-01, -6.3492e-01,\n",
       "          -7.6732e-01,  9.5206e-02,  4.4743e-01, -1.2880e-01, -6.1279e-01,\n",
       "           1.0462e-01, -2.3866e-01, -1.2257e-01, -3.5064e-02,  2.1191e-01,\n",
       "           2.7455e-01,  5.0687e-02, -9.5436e-02,  6.4791e-02, -4.6950e-01,\n",
       "          -4.9277e-01,  4.9140e-02, -1.5758e-01, -1.3063e-01,  5.3918e-01,\n",
       "          -7.9415e-02,  1.1575e-01,  5.7221e-01,  1.9799e-02, -4.5976e-01,\n",
       "          -2.5640e-01,  1.4802e-01,  2.8667e-01, -7.2082e-02,  2.1981e-01,\n",
       "           3.4119e-01,  2.6467e-02,  5.1649e-01,  1.4823e-01, -2.1046e-01,\n",
       "          -2.7239e-01,  5.1173e-01, -1.5755e-01,  6.1403e-02, -1.2527e-01,\n",
       "          -8.7207e-01, -5.6150e-02, -7.8413e-02, -1.6833e-01, -5.4440e-01,\n",
       "           2.6566e-01,  1.4351e-01,  1.0561e-01,  2.2350e-01, -1.5984e-01,\n",
       "           7.0123e-01,  6.5285e-01, -3.3100e-01, -1.6943e-01, -4.0104e-01,\n",
       "           9.3922e-02, -6.8374e-02, -2.9329e-01, -1.4009e-02,  9.5368e-03,\n",
       "          -2.2765e-01, -1.2516e-01, -1.0805e-01,  2.8921e-01, -1.8876e-01,\n",
       "           5.2862e-01,  1.2189e-01,  8.3649e-02,  4.3873e-01,  2.0612e-02,\n",
       "           1.3071e-01,  3.6515e-01, -4.2117e-01,  1.2870e-01,  4.0424e-01,\n",
       "           4.0578e-01, -1.0551e+00, -8.3291e-01, -1.4936e-01,  3.6111e-01,\n",
       "           2.2860e-01,  2.5189e-01,  1.6999e-02, -7.2215e-03,  8.9455e-02,\n",
       "          -2.3402e-01,  3.5135e-01, -6.6415e-02, -5.7412e-01,  3.0561e-01,\n",
       "           2.3567e-01,  5.3263e-01, -2.7377e-01, -5.0579e-01, -6.5233e-02,\n",
       "           2.4616e-01, -3.6381e-01,  1.1392e-01,  2.3480e-01,  4.5466e-01,\n",
       "          -3.6015e-01, -5.6164e-01, -7.0282e-02, -7.3986e-01, -1.3578e-01,\n",
       "           3.2483e-01, -2.1437e-01,  2.7178e-02, -2.4758e-01,  1.5231e-01,\n",
       "          -1.7925e-02, -1.8879e-01,  6.0757e-01,  2.0522e-01, -2.5855e-01,\n",
       "           1.1981e-01, -2.3892e-01, -1.9475e-01,  5.0091e-01, -5.0226e-01,\n",
       "          -3.0348e-01, -5.0960e-02, -1.5349e-01, -3.2418e-01,  3.2181e-01,\n",
       "          -4.1389e-01, -1.4861e-01,  1.6748e-01, -6.4736e-02, -1.3068e-01,\n",
       "          -3.5233e-01,  1.4268e-01, -1.2707e-01, -4.9883e-02, -1.7654e-02,\n",
       "          -1.3353e-01,  6.0900e-02,  1.3308e-01,  6.4538e-02,  1.4388e-01,\n",
       "           5.3696e-01, -1.1464e-01,  2.7582e-03, -2.5900e-01, -2.2469e-01,\n",
       "           5.2498e-01, -9.4973e-02, -3.3117e-01,  5.4928e-01, -1.1156e+01,\n",
       "           1.0386e-01,  7.2631e-01, -5.5808e-01,  5.5128e-02, -2.8539e-01,\n",
       "           2.1085e-01, -3.0009e-01,  7.9101e-02,  2.4967e-01, -2.8460e-01,\n",
       "          -1.2277e-01, -2.7364e-01, -2.3193e-01, -4.0872e-01, -3.9742e-01,\n",
       "          -4.7229e-01, -5.6789e-01,  1.1772e-01, -1.7663e-01, -1.0996e-01,\n",
       "          -5.7036e-01, -1.9379e-01,  2.9696e-01, -4.9559e-03, -5.1069e-01,\n",
       "           5.9541e-02, -2.2948e-01,  2.6399e-01,  4.6126e-02,  2.2045e-01,\n",
       "          -1.9949e-02,  1.4108e-01,  2.0669e-01, -4.4946e-02, -4.5100e-01,\n",
       "          -4.4853e-01,  2.2129e-01, -1.5955e-01, -2.2267e-01, -5.4017e-02,\n",
       "          -3.3838e-01, -2.7689e-01,  1.3447e-01,  3.7594e-01,  6.2596e-01,\n",
       "           1.1595e+00,  5.6208e-01,  2.9245e-01, -1.0625e-01, -8.5438e-02,\n",
       "          -1.0762e+00,  5.5900e-01, -3.7318e-01,  8.1787e-02, -2.5119e-01,\n",
       "           6.6201e-01, -4.9906e-01,  5.4685e-01, -4.0754e-01, -2.5500e-01,\n",
       "           3.7179e-01,  3.0194e-01,  7.1104e-03,  2.8239e-02, -2.7432e-01,\n",
       "           4.0257e-01, -2.6094e-01, -3.6915e-02, -3.7819e-01,  4.0081e-01,\n",
       "          -3.9472e-01,  5.0770e-01, -5.6534e-02,  1.2945e-01,  1.8567e-01,\n",
       "          -3.5001e-01,  1.6610e-01,  1.0337e+00,  1.7414e-01,  4.1071e-01,\n",
       "           1.7420e-01,  5.1160e-01,  5.3064e-02, -8.8958e-01,  1.3691e-01,\n",
       "           2.0502e-01,  2.0789e-01, -6.7096e-02,  5.7131e-01, -2.5517e-02,\n",
       "          -6.5181e-01, -2.3094e-01, -3.2169e-01,  4.3108e-01,  6.5352e-01,\n",
       "          -2.2597e-01, -4.4343e-01,  2.3894e-01, -7.9776e-03, -5.3395e-01,\n",
       "           9.0074e-02, -2.2320e-01, -5.2792e-01,  8.7275e-02, -5.8914e-01,\n",
       "           1.4985e-01, -1.2428e-01,  5.4238e-02,  6.6887e-01, -4.0159e-01,\n",
       "          -1.6113e-01,  3.6602e-02, -4.4532e-02, -4.1270e-01, -4.9520e-01,\n",
       "          -7.7083e-02,  4.1073e-01, -1.9776e-01, -2.7976e-01,  7.0636e-02,\n",
       "          -3.0673e-01,  2.2584e-01, -1.7377e-01, -6.4935e-01,  3.2879e-01,\n",
       "          -6.0415e-01,  2.0578e-01,  2.5360e-01, -2.2507e-01,  4.9017e-01,\n",
       "           1.1337e-01, -8.1817e-02,  1.9397e-01, -9.9369e-02, -7.8554e-01,\n",
       "          -3.2573e-01, -2.7299e-02,  3.2662e-01, -4.2844e-01,  1.5776e-01,\n",
       "           1.5022e-01, -4.5752e-02, -2.2574e-01,  2.7904e-01,  1.0174e-01,\n",
       "           3.7762e-01,  3.9846e-01, -5.5006e-01,  1.1727e-01, -2.9867e-01,\n",
       "          -2.9262e-02, -6.1174e-02,  1.7952e-01,  2.3654e-01, -7.0764e-01,\n",
       "          -1.5224e-01, -2.6434e-01, -3.0965e-01, -2.1650e-01, -2.5120e-01,\n",
       "           2.0680e-01,  1.4038e-01, -1.5897e-02, -1.1308e-01,  5.8077e-01,\n",
       "           4.0814e-02, -1.1167e-01, -4.5997e-01,  3.1443e-01,  4.3178e-01,\n",
       "           2.1794e-01,  7.5875e-02, -1.2830e-01, -5.5532e-01, -1.1922e-02,\n",
       "          -4.3644e-01,  9.1081e-01, -7.3906e-01, -7.1970e-01,  6.6086e-02,\n",
       "           1.4951e-02, -8.8259e-04, -2.3450e-01, -5.0265e-01,  8.1298e-02,\n",
       "          -2.1017e-01, -1.4891e-01,  6.1550e-03, -1.0607e-02,  1.6299e-01,\n",
       "           4.0742e-01,  3.6838e-01,  1.5951e-01, -3.3938e-01, -4.4081e-01,\n",
       "          -6.0244e-01,  1.4567e-02, -1.6795e-01,  2.9330e-01, -2.6752e-01,\n",
       "          -3.2661e-01,  1.6043e-02, -9.1892e-03, -2.2272e-01,  5.2876e-01,\n",
       "           1.2427e-01,  1.6396e+00, -3.7287e-01, -4.4551e-01,  2.9018e-01,\n",
       "           7.2306e-02, -2.7073e-02, -2.8130e-01,  2.5074e-01,  9.5447e-02,\n",
       "           4.0194e-02, -3.6304e-01,  1.5023e-02, -2.0558e-02,  3.2773e-01,\n",
       "           3.2056e-01, -1.0703e+00, -2.0238e-01, -1.7993e-01,  5.5664e-01,\n",
       "           3.2388e-02,  6.3185e-01, -2.2305e-01,  3.0577e-01,  4.6106e-01,\n",
       "          -8.6667e-01, -2.3000e-01, -1.6129e-01,  2.7295e-01, -1.0583e-01,\n",
       "          -9.8735e-02, -1.6178e-01, -3.7165e-02, -1.6936e-01,  3.9538e-01,\n",
       "           7.4751e-01, -7.8426e-01,  7.4081e-01,  7.0467e-01, -9.8234e-02,\n",
       "           5.6815e-01,  9.4187e-02,  3.2641e-01, -1.4659e-01,  2.2669e-02,\n",
       "           2.6200e-01,  1.8542e-01, -6.3165e-01, -2.3962e-01,  2.4473e-01,\n",
       "           4.0158e-01, -1.1943e-01,  1.2108e-01,  7.9292e-02,  3.0960e-02,\n",
       "          -3.7691e-01, -1.0145e-01, -2.4552e-01, -3.9904e-01,  1.1356e-01,\n",
       "           1.4603e-02, -1.5001e-01, -8.6997e-02,  4.7709e-01, -1.7580e-01,\n",
       "           7.5067e-01,  4.2910e-01,  3.7145e-01, -3.5103e-01,  8.9950e-04,\n",
       "          -5.7565e-02,  6.0671e-01, -3.1670e-01,  7.7385e-01,  6.3337e-02,\n",
       "          -2.3140e-01,  2.7251e-01,  6.0547e-03,  1.7829e-01,  2.9892e-01,\n",
       "          -2.3519e-01,  1.2702e-01, -3.0718e-01, -2.3711e-01,  2.8330e-01,\n",
       "          -1.8640e-01,  1.1542e-01, -7.2270e-01, -6.3602e-03,  3.9857e-01,\n",
       "          -3.2051e-01,  4.9873e-01, -2.3074e-01,  6.1026e-01, -4.3509e-01,\n",
       "           3.2669e-02, -4.7470e-01, -4.3275e-01, -2.8508e-01,  5.1854e-02,\n",
       "          -8.6027e-01, -1.9453e-01,  1.7684e-01,  3.7405e-02,  2.1707e-01,\n",
       "           7.5155e-01, -3.3794e-01, -2.4771e-01, -2.0386e-01,  4.0190e-02,\n",
       "           1.8299e-01, -8.2071e-02, -3.5586e-01,  2.1161e-01, -3.5607e-01,\n",
       "          -1.0520e+00, -2.3117e-01,  8.9990e-02,  3.8419e-01, -3.5853e-01,\n",
       "           1.1570e+00,  2.5132e-02, -2.0226e-01,  6.7253e-01,  2.0808e-01,\n",
       "           6.4157e-01,  6.0323e-01,  1.3628e-03, -4.1259e-01, -5.4109e-02,\n",
       "          -1.5877e-01, -3.2881e-01,  1.5757e-01, -1.8346e-01,  5.0556e-01,\n",
       "           5.4802e-01, -2.0806e-02,  5.4756e-01]], device='mps:0'),\n",
       " tensor([[-6.1750e-01, -4.2829e-01, -2.3934e-01, -9.4116e-01, -3.4548e-01,\n",
       "          -6.5496e-02, -6.3874e-01,  9.7828e-01, -9.5892e-02,  4.0096e-01,\n",
       "           1.4919e-01, -5.1603e-03,  2.0980e-01,  1.3680e-01, -6.9506e-01,\n",
       "           1.6740e-01,  4.6094e-01,  2.1159e-01,  3.5951e-01, -4.4329e-02,\n",
       "           7.9832e-01,  3.4245e-01,  1.8019e-01, -2.5205e-01, -4.5876e-01,\n",
       "           5.8994e-01, -2.4628e-01, -9.2014e-02,  2.0509e-01, -8.1949e-01,\n",
       "          -1.3586e-01, -4.9757e-02, -3.6281e-01,  8.7602e-01,  1.1948e-01,\n",
       "          -2.7310e-01, -1.5276e-01, -2.7996e-01,  2.3494e-01, -6.3637e-02,\n",
       "           3.7814e-01,  2.6592e-02, -2.9347e-01,  3.1386e-01, -1.6421e-01,\n",
       "          -1.4057e-01,  1.4047e-02, -7.8429e-01,  1.3666e-02,  1.8955e-01,\n",
       "           5.8851e-02, -4.7205e-02,  1.9231e-01,  3.7030e-01, -1.0617e+00,\n",
       "          -2.6348e-01,  2.7603e-01,  2.9849e-01, -1.7932e-01, -5.7784e-02,\n",
       "          -1.9989e-01,  9.0201e-01,  1.7949e-01, -2.7822e-01,  1.7824e-01,\n",
       "          -4.4899e-01, -5.4601e-01, -6.0781e-01, -1.0682e-01, -1.1967e-01,\n",
       "          -1.0002e-01,  9.7528e-02,  1.4009e-01, -2.9385e-01,  2.3170e-01,\n",
       "           2.1813e-01,  4.7024e-02, -2.7652e-02, -4.3158e-01, -2.6546e-01,\n",
       "           3.6947e-01,  3.4425e-02, -1.4730e-02,  2.0326e-01, -5.8744e-01,\n",
       "          -2.0226e-01,  3.7959e-01,  1.5703e-01,  2.5914e-01,  2.2662e-03,\n",
       "           5.9967e-01,  1.2024e-02,  8.0339e-02, -4.5369e-01,  5.8427e-01,\n",
       "          -5.4955e-01,  1.0329e-01, -1.6007e-01,  4.9532e-02, -5.2088e-01,\n",
       "          -9.8735e-02,  2.4494e-01, -1.8085e-01,  1.6053e-01, -4.6592e-01,\n",
       "          -3.2983e-02,  2.4514e-01,  3.0468e-01, -4.9183e-01,  4.2685e-02,\n",
       "          -6.5134e-03, -2.9791e-01,  7.3694e-02, -6.6407e-01, -8.5795e-01,\n",
       "           5.8354e-01,  1.8428e-02, -4.5240e-01, -2.3930e-01,  7.4443e-01,\n",
       "           2.9738e-01, -2.3201e-01, -5.3072e-02, -3.5548e-01,  2.0259e-01,\n",
       "          -1.8518e-01,  2.5689e-01, -2.7018e-01,  3.8439e-01,  5.5026e-01,\n",
       "           3.1581e-01,  2.8489e-01, -1.8327e-01,  2.7637e-01,  1.4465e-01,\n",
       "           3.6947e-02, -6.6614e-02, -1.5929e-01,  1.5180e-01,  1.2041e-01,\n",
       "          -2.9580e-01, -3.2999e-01,  5.4746e-03,  1.1182e-01, -3.3279e-01,\n",
       "          -1.0202e-01,  1.8680e-01,  3.4808e-01, -1.4092e-01, -2.5403e-01,\n",
       "          -7.8984e-01, -4.2328e-01,  1.0217e-01, -1.8560e-01, -1.7135e-01,\n",
       "          -4.1911e-01,  3.8121e-01, -6.6660e-02,  4.7638e-01,  3.4189e-01,\n",
       "           6.9978e-01, -1.5311e-01,  5.9776e-02, -5.6412e-01,  3.5197e-01,\n",
       "           3.8270e-02, -6.1110e-01, -1.7247e-01,  1.5680e-01, -4.1813e-04,\n",
       "           6.8195e-01,  7.0979e-02,  4.0124e-01,  1.2070e-01, -3.2553e-01,\n",
       "           8.2089e-03,  1.9890e-01, -1.2864e-01, -1.4297e-01,  7.3766e-01,\n",
       "          -5.5188e-01, -2.3029e+00,  1.0721e-01, -3.5844e-01, -3.6529e-01,\n",
       "          -1.8135e-01,  3.7255e-01,  3.3772e-02,  4.9966e-01,  1.5486e-01,\n",
       "           3.7598e-02,  1.6729e-01, -2.8025e-02,  3.6851e-01, -8.4399e-02,\n",
       "           2.6751e-01,  3.3341e-01,  1.2988e-01, -2.3298e-01, -1.0956e-01,\n",
       "          -3.9227e-01,  2.4031e-01, -9.8916e-02,  8.6010e-01,  2.3260e-01,\n",
       "           2.0943e-01, -4.3657e-03, -4.7262e-02,  4.8964e-01,  4.3099e-01,\n",
       "          -6.6541e-02, -1.6585e-01, -4.9522e-01,  4.6269e-03, -4.1279e-02,\n",
       "          -1.2963e-02, -4.5611e-01,  1.3753e-01,  4.2841e-01, -1.2416e-01,\n",
       "          -1.8354e-01,  1.5525e-01, -4.7715e-02, -3.1517e-01, -2.6492e-02,\n",
       "           6.8453e-01,  1.9080e-01, -2.2994e-01,  1.2377e-01, -1.1896e-01,\n",
       "           3.7006e-02, -4.6975e-01,  7.0044e-01,  6.1147e-01,  1.5585e-01,\n",
       "           1.6214e-02, -6.9943e-01,  2.3699e-01, -3.1574e-01, -4.7647e-03,\n",
       "          -1.5027e-01, -1.5731e-01, -4.2986e-01, -5.7285e-02, -5.5854e-02,\n",
       "           4.0115e-01,  7.9943e-02,  4.6653e-01,  1.1837e+00,  2.7973e-01,\n",
       "           1.9184e-02,  5.6097e-01,  2.0801e-01,  4.2492e-01, -4.0180e-01,\n",
       "           3.3082e-01,  8.8825e-02, -6.3896e-01,  2.7188e-01, -4.1593e-01,\n",
       "           4.4963e-02, -4.1432e-02, -3.7926e-01,  1.4208e-01, -2.1276e-02,\n",
       "           3.4237e-01,  3.3517e-01,  1.5186e-01,  1.6838e-01,  2.9411e-01,\n",
       "          -3.1322e-01,  5.9715e-01, -6.0809e-02, -1.1127e-01, -6.3492e-01,\n",
       "          -7.6732e-01,  9.5206e-02,  4.4743e-01, -1.2880e-01, -6.1279e-01,\n",
       "           1.0462e-01, -2.3866e-01, -1.2257e-01, -3.5064e-02,  2.1191e-01,\n",
       "           2.7455e-01,  5.0687e-02, -9.5436e-02,  6.4791e-02, -4.6950e-01,\n",
       "          -4.9277e-01,  4.9140e-02, -1.5758e-01, -1.3063e-01,  5.3918e-01,\n",
       "          -7.9415e-02,  1.1575e-01,  5.7221e-01,  1.9799e-02, -4.5976e-01,\n",
       "          -2.5640e-01,  1.4802e-01,  2.8667e-01, -7.2082e-02,  2.1981e-01,\n",
       "           3.4119e-01,  2.6467e-02,  5.1649e-01,  1.4823e-01, -2.1046e-01,\n",
       "          -2.7239e-01,  5.1173e-01, -1.5755e-01,  6.1403e-02, -1.2527e-01,\n",
       "          -8.7207e-01, -5.6150e-02, -7.8413e-02, -1.6833e-01, -5.4440e-01,\n",
       "           2.6566e-01,  1.4351e-01,  1.0561e-01,  2.2350e-01, -1.5984e-01,\n",
       "           7.0123e-01,  6.5285e-01, -3.3100e-01, -1.6943e-01, -4.0104e-01,\n",
       "           9.3922e-02, -6.8374e-02, -2.9329e-01, -1.4009e-02,  9.5368e-03,\n",
       "          -2.2765e-01, -1.2516e-01, -1.0805e-01,  2.8921e-01, -1.8876e-01,\n",
       "           5.2862e-01,  1.2189e-01,  8.3649e-02,  4.3873e-01,  2.0612e-02,\n",
       "           1.3071e-01,  3.6515e-01, -4.2117e-01,  1.2870e-01,  4.0424e-01,\n",
       "           4.0578e-01, -1.0551e+00, -8.3291e-01, -1.4936e-01,  3.6111e-01,\n",
       "           2.2860e-01,  2.5189e-01,  1.6999e-02, -7.2215e-03,  8.9455e-02,\n",
       "          -2.3402e-01,  3.5135e-01, -6.6415e-02, -5.7412e-01,  3.0561e-01,\n",
       "           2.3567e-01,  5.3263e-01, -2.7377e-01, -5.0579e-01, -6.5233e-02,\n",
       "           2.4616e-01, -3.6381e-01,  1.1392e-01,  2.3480e-01,  4.5466e-01,\n",
       "          -3.6015e-01, -5.6164e-01, -7.0282e-02, -7.3986e-01, -1.3578e-01,\n",
       "           3.2483e-01, -2.1437e-01,  2.7178e-02, -2.4758e-01,  1.5231e-01,\n",
       "          -1.7925e-02, -1.8879e-01,  6.0757e-01,  2.0522e-01, -2.5855e-01,\n",
       "           1.1981e-01, -2.3892e-01, -1.9475e-01,  5.0091e-01, -5.0226e-01,\n",
       "          -3.0348e-01, -5.0960e-02, -1.5349e-01, -3.2418e-01,  3.2181e-01,\n",
       "          -4.1389e-01, -1.4861e-01,  1.6748e-01, -6.4736e-02, -1.3068e-01,\n",
       "          -3.5233e-01,  1.4268e-01, -1.2707e-01, -4.9883e-02, -1.7654e-02,\n",
       "          -1.3353e-01,  6.0900e-02,  1.3308e-01,  6.4538e-02,  1.4388e-01,\n",
       "           5.3696e-01, -1.1464e-01,  2.7582e-03, -2.5900e-01, -2.2469e-01,\n",
       "           5.2498e-01, -9.4973e-02, -3.3117e-01,  5.4928e-01, -1.1156e+01,\n",
       "           1.0386e-01,  7.2631e-01, -5.5808e-01,  5.5128e-02, -2.8539e-01,\n",
       "           2.1085e-01, -3.0009e-01,  7.9101e-02,  2.4967e-01, -2.8460e-01,\n",
       "          -1.2277e-01, -2.7364e-01, -2.3193e-01, -4.0872e-01, -3.9742e-01,\n",
       "          -4.7229e-01, -5.6789e-01,  1.1772e-01, -1.7663e-01, -1.0996e-01,\n",
       "          -5.7036e-01, -1.9379e-01,  2.9696e-01, -4.9559e-03, -5.1069e-01,\n",
       "           5.9541e-02, -2.2948e-01,  2.6399e-01,  4.6126e-02,  2.2045e-01,\n",
       "          -1.9949e-02,  1.4108e-01,  2.0669e-01, -4.4946e-02, -4.5100e-01,\n",
       "          -4.4853e-01,  2.2129e-01, -1.5955e-01, -2.2267e-01, -5.4017e-02,\n",
       "          -3.3838e-01, -2.7689e-01,  1.3447e-01,  3.7594e-01,  6.2596e-01,\n",
       "           1.1595e+00,  5.6208e-01,  2.9245e-01, -1.0625e-01, -8.5438e-02,\n",
       "          -1.0762e+00,  5.5900e-01, -3.7318e-01,  8.1787e-02, -2.5119e-01,\n",
       "           6.6201e-01, -4.9906e-01,  5.4685e-01, -4.0754e-01, -2.5500e-01,\n",
       "           3.7179e-01,  3.0194e-01,  7.1104e-03,  2.8239e-02, -2.7432e-01,\n",
       "           4.0257e-01, -2.6094e-01, -3.6915e-02, -3.7819e-01,  4.0081e-01,\n",
       "          -3.9472e-01,  5.0770e-01, -5.6534e-02,  1.2945e-01,  1.8567e-01,\n",
       "          -3.5001e-01,  1.6610e-01,  1.0337e+00,  1.7414e-01,  4.1071e-01,\n",
       "           1.7420e-01,  5.1160e-01,  5.3064e-02, -8.8958e-01,  1.3691e-01,\n",
       "           2.0502e-01,  2.0789e-01, -6.7096e-02,  5.7131e-01, -2.5517e-02,\n",
       "          -6.5181e-01, -2.3094e-01, -3.2169e-01,  4.3108e-01,  6.5352e-01,\n",
       "          -2.2597e-01, -4.4343e-01,  2.3894e-01, -7.9776e-03, -5.3395e-01,\n",
       "           9.0074e-02, -2.2320e-01, -5.2792e-01,  8.7275e-02, -5.8914e-01,\n",
       "           1.4985e-01, -1.2428e-01,  5.4238e-02,  6.6887e-01, -4.0159e-01,\n",
       "          -1.6113e-01,  3.6602e-02, -4.4532e-02, -4.1270e-01, -4.9520e-01,\n",
       "          -7.7083e-02,  4.1073e-01, -1.9776e-01, -2.7976e-01,  7.0636e-02,\n",
       "          -3.0673e-01,  2.2584e-01, -1.7377e-01, -6.4935e-01,  3.2879e-01,\n",
       "          -6.0415e-01,  2.0578e-01,  2.5360e-01, -2.2507e-01,  4.9017e-01,\n",
       "           1.1337e-01, -8.1817e-02,  1.9397e-01, -9.9369e-02, -7.8554e-01,\n",
       "          -3.2573e-01, -2.7299e-02,  3.2662e-01, -4.2844e-01,  1.5776e-01,\n",
       "           1.5022e-01, -4.5752e-02, -2.2574e-01,  2.7904e-01,  1.0174e-01,\n",
       "           3.7762e-01,  3.9846e-01, -5.5006e-01,  1.1727e-01, -2.9867e-01,\n",
       "          -2.9262e-02, -6.1174e-02,  1.7952e-01,  2.3654e-01, -7.0764e-01,\n",
       "          -1.5224e-01, -2.6434e-01, -3.0965e-01, -2.1650e-01, -2.5120e-01,\n",
       "           2.0680e-01,  1.4038e-01, -1.5897e-02, -1.1308e-01,  5.8077e-01,\n",
       "           4.0814e-02, -1.1167e-01, -4.5997e-01,  3.1443e-01,  4.3178e-01,\n",
       "           2.1794e-01,  7.5875e-02, -1.2830e-01, -5.5532e-01, -1.1922e-02,\n",
       "          -4.3644e-01,  9.1081e-01, -7.3906e-01, -7.1970e-01,  6.6086e-02,\n",
       "           1.4951e-02, -8.8259e-04, -2.3450e-01, -5.0265e-01,  8.1298e-02,\n",
       "          -2.1017e-01, -1.4891e-01,  6.1550e-03, -1.0607e-02,  1.6299e-01,\n",
       "           4.0742e-01,  3.6838e-01,  1.5951e-01, -3.3938e-01, -4.4081e-01,\n",
       "          -6.0244e-01,  1.4567e-02, -1.6795e-01,  2.9330e-01, -2.6752e-01,\n",
       "          -3.2661e-01,  1.6043e-02, -9.1892e-03, -2.2272e-01,  5.2876e-01,\n",
       "           1.2427e-01,  1.6396e+00, -3.7287e-01, -4.4551e-01,  2.9018e-01,\n",
       "           7.2306e-02, -2.7073e-02, -2.8130e-01,  2.5074e-01,  9.5447e-02,\n",
       "           4.0194e-02, -3.6304e-01,  1.5023e-02, -2.0558e-02,  3.2773e-01,\n",
       "           3.2056e-01, -1.0703e+00, -2.0238e-01, -1.7993e-01,  5.5664e-01,\n",
       "           3.2388e-02,  6.3185e-01, -2.2305e-01,  3.0577e-01,  4.6106e-01,\n",
       "          -8.6667e-01, -2.3000e-01, -1.6129e-01,  2.7295e-01, -1.0583e-01,\n",
       "          -9.8735e-02, -1.6178e-01, -3.7165e-02, -1.6936e-01,  3.9538e-01,\n",
       "           7.4751e-01, -7.8426e-01,  7.4081e-01,  7.0467e-01, -9.8234e-02,\n",
       "           5.6815e-01,  9.4187e-02,  3.2641e-01, -1.4659e-01,  2.2669e-02,\n",
       "           2.6200e-01,  1.8542e-01, -6.3165e-01, -2.3962e-01,  2.4473e-01,\n",
       "           4.0158e-01, -1.1943e-01,  1.2108e-01,  7.9292e-02,  3.0960e-02,\n",
       "          -3.7691e-01, -1.0145e-01, -2.4552e-01, -3.9904e-01,  1.1356e-01,\n",
       "           1.4603e-02, -1.5001e-01, -8.6997e-02,  4.7709e-01, -1.7580e-01,\n",
       "           7.5067e-01,  4.2910e-01,  3.7145e-01, -3.5103e-01,  8.9950e-04,\n",
       "          -5.7565e-02,  6.0671e-01, -3.1670e-01,  7.7385e-01,  6.3337e-02,\n",
       "          -2.3140e-01,  2.7251e-01,  6.0547e-03,  1.7829e-01,  2.9892e-01,\n",
       "          -2.3519e-01,  1.2702e-01, -3.0718e-01, -2.3711e-01,  2.8330e-01,\n",
       "          -1.8640e-01,  1.1542e-01, -7.2270e-01, -6.3602e-03,  3.9857e-01,\n",
       "          -3.2051e-01,  4.9873e-01, -2.3074e-01,  6.1026e-01, -4.3509e-01,\n",
       "           3.2669e-02, -4.7470e-01, -4.3275e-01, -2.8508e-01,  5.1854e-02,\n",
       "          -8.6027e-01, -1.9453e-01,  1.7684e-01,  3.7405e-02,  2.1707e-01,\n",
       "           7.5155e-01, -3.3794e-01, -2.4771e-01, -2.0386e-01,  4.0190e-02,\n",
       "           1.8299e-01, -8.2071e-02, -3.5586e-01,  2.1161e-01, -3.5607e-01,\n",
       "          -1.0520e+00, -2.3117e-01,  8.9990e-02,  3.8419e-01, -3.5853e-01,\n",
       "           1.1570e+00,  2.5132e-02, -2.0226e-01,  6.7253e-01,  2.0808e-01,\n",
       "           6.4157e-01,  6.0323e-01,  1.3628e-03, -4.1259e-01, -5.4109e-02,\n",
       "          -1.5877e-01, -3.2881e-01,  1.5757e-01, -1.8346e-01,  5.0556e-01,\n",
       "           5.4802e-01, -2.0806e-02,  5.4756e-01]], device='mps:0'),\n",
       " tensor([[-2.0446e-01, -6.1197e-01, -6.5120e-01, -3.5261e-01, -3.3717e-01,\n",
       "           9.9856e-02, -4.5953e-01,  5.8790e-01, -1.4437e-01,  3.7456e-01,\n",
       "           4.1935e-01,  1.6350e-01,  1.8905e-01, -4.3175e-03, -4.2141e-01,\n",
       "          -4.1585e-01, -2.0535e-01,  1.0751e-01,  1.3956e-01,  1.2101e-01,\n",
       "           9.5831e-01,  3.5226e-01, -1.7911e-01,  2.6967e-01, -7.5653e-01,\n",
       "          -8.5554e-02, -1.5043e-01, -8.9743e-02,  1.3548e-01, -6.7518e-01,\n",
       "           7.2961e-02,  3.1344e-01, -2.3560e-02,  8.6246e-01, -4.4735e-01,\n",
       "          -2.5926e-01,  3.2523e-01, -6.1248e-02,  6.3086e-01, -1.4933e-01,\n",
       "           6.0873e-01, -6.3741e-01,  2.2667e-01,  5.7834e-02,  3.1013e-01,\n",
       "           3.7719e-01,  5.7968e-01, -8.9667e-01,  1.0517e-01, -6.4309e-01,\n",
       "          -1.8241e-01,  1.6567e-01,  1.5905e-01,  2.5134e-01, -1.4850e-01,\n",
       "          -3.2971e-01,  4.6116e-01,  2.8429e-01, -4.2475e-01,  6.2628e-02,\n",
       "          -7.7068e-02,  5.2646e-01,  2.5778e-01,  1.3607e-03, -6.2107e-02,\n",
       "           3.9288e-01, -6.1120e-01, -2.2377e-01, -5.2536e-01, -8.5499e-02,\n",
       "           1.2208e-01,  1.1222e-01,  2.2155e-01,  4.2537e-02,  3.8132e-01,\n",
       "          -2.7553e-01,  2.1399e-01,  1.1997e-01, -6.9298e-01, -1.1323e-01,\n",
       "           1.0265e-01,  4.7453e-01, -4.5418e-02, -3.1688e-01,  1.5859e-02,\n",
       "           1.5226e-02,  5.2812e-01,  4.6394e-01,  3.0875e-01, -1.9721e-01,\n",
       "           3.3609e-01, -9.5169e-03, -2.8336e-01, -3.8854e-01,  2.5322e-01,\n",
       "           5.3726e-03,  3.2695e-01,  1.1161e-01, -2.0247e-01, -3.4893e-01,\n",
       "           8.1865e-02,  1.3777e-01,  1.9856e-01,  5.8127e-02, -3.0235e-01,\n",
       "          -5.3580e-01,  4.6347e-02, -1.3120e-01, -6.4147e-01, -2.9654e-01,\n",
       "          -5.7965e-02, -2.4084e-02,  3.7966e-01, -3.3482e-01, -3.1406e-01,\n",
       "          -7.3994e-03, -8.3844e-02,  2.0243e-01, -3.7230e-01,  5.7430e-01,\n",
       "           2.4810e-01, -4.3176e-01,  6.1906e-02,  2.1824e-01,  4.3013e-01,\n",
       "           1.6769e-02,  3.3426e-01, -2.8384e-01,  3.7158e-01,  4.0219e-02,\n",
       "          -2.4621e-01,  1.8211e-01, -3.3254e-01, -3.3194e-01, -4.5525e-01,\n",
       "          -8.3937e-02,  4.3599e-01, -4.0811e-01, -3.5430e-01, -2.1593e-01,\n",
       "           8.2480e-03, -1.3021e-01, -5.0094e-01, -1.0588e-01,  1.9923e-01,\n",
       "           5.7744e-03,  3.0704e-02, -1.2729e-01, -6.3775e-01, -2.8115e-01,\n",
       "          -5.9582e-01, -5.7904e-01,  3.4301e-01, -1.5643e-01, -1.4632e-01,\n",
       "           1.0421e-01,  2.3775e-01, -1.0556e-01, -1.9955e-01,  2.6347e-01,\n",
       "           7.3691e-01, -2.2450e-01,  1.6618e-01, -5.3461e-01, -3.1379e-02,\n",
       "           2.4071e-01, -1.3981e-01, -5.6826e-02, -1.5626e-01,  1.1269e-01,\n",
       "           3.8810e-02, -2.5280e-01, -7.1981e-02, -6.1064e-01, -3.1599e-02,\n",
       "          -5.6583e-01,  3.8993e-01,  1.8385e-01, -9.1691e-02,  1.7381e-02,\n",
       "           2.5439e-01, -1.9152e+00,  1.2429e-01, -3.0732e-01,  9.5472e-02,\n",
       "          -4.5823e-01,  3.3125e-01,  3.7425e-01,  1.5901e-01, -4.2723e-02,\n",
       "          -2.7771e-01, -9.2223e-02, -1.1283e-01,  4.8513e-01, -8.0233e-02,\n",
       "           3.1868e-01,  3.6242e-01,  1.4201e-01, -3.6301e-01,  1.3980e-01,\n",
       "          -4.8167e-01,  6.7723e-02, -2.5123e-01, -3.3501e-01,  2.4615e-01,\n",
       "           4.1520e-01, -1.0203e-01,  3.0823e-02,  6.0223e-01,  3.1167e-02,\n",
       "          -6.7769e-03, -2.3592e-01, -2.6022e-01, -3.0181e-01,  6.0519e-03,\n",
       "           6.4079e-01, -1.5120e-01, -2.2379e-02,  6.2725e-01,  1.1903e-01,\n",
       "          -1.5426e-01,  7.5605e-02, -1.0385e-01, -3.5719e-01, -8.0490e-02,\n",
       "           3.4120e-01, -2.0422e-01, -6.6409e-02, -2.4670e-01, -1.4158e-01,\n",
       "          -1.1625e-01, -8.1931e-01,  3.6328e-01,  1.8454e-01, -2.5861e-01,\n",
       "          -3.2955e-01, -4.3044e-01,  3.0230e-01, -2.1751e-01,  5.8580e-02,\n",
       "          -1.1971e-01, -4.6770e-01, -8.3534e-01,  3.3918e-02, -2.2350e-01,\n",
       "           1.5454e-01, -1.8955e-01,  8.4345e-01,  3.3650e-01,  4.1863e-01,\n",
       "          -1.3202e-01, -2.0679e-01,  5.9660e-01,  1.1616e-01, -3.3904e-01,\n",
       "           5.9368e-02,  5.1809e-02, -3.3175e-01,  3.3928e-01, -3.9834e-01,\n",
       "          -2.2246e-01, -2.0873e-01,  1.4234e-01,  4.5859e-01, -3.9386e-01,\n",
       "           4.2435e-01,  4.6714e-01, -5.2504e-01, -2.1936e-01,  2.4552e-01,\n",
       "          -4.3375e-01,  1.2567e-01,  6.3162e-01, -4.0988e-01, -6.1270e-01,\n",
       "          -1.9425e-01,  2.7640e-01, -4.2183e-01,  3.8016e-01, -1.0082e+00,\n",
       "           2.4453e-02,  4.5823e-01, -2.4745e-01,  1.0366e-01, -1.7507e-01,\n",
       "           2.0321e-01,  1.6952e-01, -2.4362e-01, -3.1977e-01, -7.7713e-01,\n",
       "           2.0325e-01,  4.9669e-03, -1.3188e-01,  3.5081e-01,  2.8144e-01,\n",
       "           1.1463e-01,  2.9644e-01,  9.0387e-02,  6.7408e-02, -3.7675e-01,\n",
       "          -6.7455e-02, -3.5337e-02, -9.9382e-02, -1.4856e-01,  1.9744e-01,\n",
       "           3.1813e-01,  1.1330e-01,  6.0128e-01, -6.1010e-01, -5.0777e-01,\n",
       "           1.7238e-02,  8.8318e-01, -7.8310e-02,  1.5924e-01, -4.7719e-02,\n",
       "          -1.3880e+00, -3.9435e-01, -4.7267e-01, -2.5403e-01, -3.7448e-01,\n",
       "           3.9156e-01, -3.5281e-01, -4.3930e-02,  2.1778e-02, -2.4405e-01,\n",
       "           5.8036e-02,  1.5822e-01, -8.4619e-02,  2.2582e-01, -2.2292e-01,\n",
       "          -2.1889e-01,  1.0467e-01, -3.4941e-01, -1.7239e-01, -1.8321e-01,\n",
       "           1.6423e-01,  1.3770e-02, -1.5817e-01, -1.4071e-01, -2.8396e-01,\n",
       "           7.5439e-01, -4.0584e-01, -1.3723e-01,  2.2278e-01, -1.8008e-01,\n",
       "          -1.5530e-01,  6.3187e-01, -7.4788e-01, -1.4984e-01,  2.3756e-01,\n",
       "          -1.0832e-01, -3.1036e-01, -3.1138e-01, -1.0912e-01,  5.8585e-01,\n",
       "           2.4527e-01,  6.1429e-01, -5.4751e-01,  3.0181e-01,  5.2733e-01,\n",
       "          -5.6894e-01,  2.2055e-01, -1.5847e-01, -5.3117e-02,  3.4238e-01,\n",
       "           2.1791e-01,  1.6639e-01, -2.4175e-01,  3.5339e-01, -5.8353e-04,\n",
       "          -8.6562e-02, -3.1866e-01, -1.6466e-01,  2.2896e-01, -5.5917e-02,\n",
       "          -1.1628e-01, -6.1302e-01,  4.8145e-01, -2.7111e-01, -6.2022e-01,\n",
       "           1.3283e-01, -6.2701e-02, -6.6285e-03, -5.4448e-02, -3.2490e-01,\n",
       "          -2.8777e-01,  7.0026e-02,  5.3480e-01,  2.5565e-02,  4.1365e-01,\n",
       "           3.6509e-01, -1.1559e-01, -2.6395e-01,  3.8986e-01,  7.8452e-02,\n",
       "          -2.1889e-01,  3.4036e-01, -7.9317e-01,  4.9327e-01,  5.1424e-01,\n",
       "          -2.2179e-01, -4.9643e-01, -3.0877e-02,  2.6544e-01,  3.3100e-01,\n",
       "          -2.8362e-02,  3.1811e-01,  4.4280e-01, -1.3974e-01, -1.0768e-01,\n",
       "           4.7582e-01,  5.3581e-01,  1.2455e-01,  1.8514e-01,  5.7003e-02,\n",
       "           2.5835e-01, -2.2476e-01,  2.2551e-01, -4.7252e-02, -7.0870e-01,\n",
       "           4.1380e-01,  4.7418e-01,  2.0469e-01,  1.7820e-01, -1.1355e+01,\n",
       "           3.5577e-01,  4.3555e-01, -5.7965e-01,  3.8480e-01,  5.0049e-02,\n",
       "           2.1169e-01, -4.7182e-01, -4.2457e-03, -4.9899e-01, -1.7725e-01,\n",
       "          -1.5366e-01,  1.6604e-01,  1.1480e-01, -3.7481e-01,  1.3913e-01,\n",
       "          -2.2740e-01, -9.3869e-01,  3.1457e-02, -9.9158e-02,  3.7596e-01,\n",
       "          -1.9342e-01,  1.2610e-01,  1.7042e-01, -1.5520e-01, -2.2380e-01,\n",
       "          -4.2603e-01, -3.9607e-01,  3.5483e-01,  2.3231e-01,  1.5710e-01,\n",
       "           1.3729e-01, -4.0130e-02,  1.3776e-01, -3.4871e-01, -5.3901e-01,\n",
       "          -1.7272e-01,  9.9460e-02,  2.5214e-01, -6.2425e-02, -3.0212e-02,\n",
       "           1.0265e-01, -2.7183e-01,  4.3654e-01, -9.0303e-03,  5.7613e-01,\n",
       "           6.5250e-01,  7.9796e-01,  8.5160e-02, -7.9172e-02,  2.6781e-01,\n",
       "          -7.4284e-01,  3.2061e-01, -1.0194e+00,  2.8498e-01,  2.8890e-02,\n",
       "           7.2549e-01, -6.8794e-01, -3.9187e-01,  3.1365e-02, -1.8613e-01,\n",
       "           2.0309e-01, -5.2316e-02,  2.8782e-01, -3.6944e-01, -1.6552e-01,\n",
       "           5.7290e-01, -2.6575e-01,  1.7173e-01, -1.4493e-01, -4.7500e-02,\n",
       "          -1.5662e-01,  3.9679e-01,  1.6728e-01, -4.8029e-01,  7.8973e-02,\n",
       "           1.1977e-01, -1.7437e-01,  5.6773e-01,  1.5232e-02, -1.4090e-01,\n",
       "           3.0495e-01,  1.1288e-01, -1.5661e-01, -4.8935e-01, -3.1469e-01,\n",
       "          -2.4221e-01,  3.9133e-01,  3.6759e-01,  6.5765e-01, -5.0155e-01,\n",
       "          -4.6597e-01, -1.6269e-01,  9.1459e-02,  1.4252e-01,  6.4813e-01,\n",
       "          -6.6447e-01,  2.6466e-01, -1.1468e-01, -1.8074e-01, -4.1854e-01,\n",
       "          -2.8198e-01, -1.9023e-01, -6.1233e-01,  1.6281e-01,  2.6104e-01,\n",
       "          -3.9158e-02, -7.2241e-03,  1.1613e-01,  9.5793e-02, -4.6837e-01,\n",
       "           2.6582e-01, -2.4979e-01,  2.7484e-01, -4.2814e-01,  1.0287e-01,\n",
       "          -1.4440e-01,  6.5077e-01, -4.2493e-01, -2.2475e-01, -3.2421e-01,\n",
       "          -2.5474e-01,  4.6548e-01,  3.3122e-02, -2.0910e-01,  2.1302e-02,\n",
       "          -6.6626e-01, -2.5043e-01, -1.7037e-01,  4.3265e-01,  5.8927e-01,\n",
       "           5.2866e-02, -7.4443e-02,  3.2495e-01, -5.2601e-01, -6.4400e-01,\n",
       "          -1.0182e+00, -6.6626e-02,  1.4001e-01, -6.6224e-01, -3.2268e-04,\n",
       "          -5.9929e-02, -6.4441e-02, -2.2564e-01,  9.6926e-02, -1.2329e-01,\n",
       "           4.3934e-01,  3.5485e-01, -3.3960e-01,  3.3246e-01, -2.6391e-01,\n",
       "          -3.2259e-01, -1.2727e-01, -2.7226e-01, -9.5962e-02, -2.3795e-01,\n",
       "          -1.1079e-01,  2.1746e-01, -2.3436e-01,  2.4333e-01,  3.2556e-01,\n",
       "           9.8292e-03,  2.7591e-02, -1.1129e-01,  1.2529e-01, -1.2838e-01,\n",
       "           7.7934e-02, -1.3609e-01, -3.8530e-02,  9.9226e-02,  7.4469e-02,\n",
       "           2.1914e-02, -1.4346e-01, -3.2895e-01, -2.1315e-01, -9.7360e-03,\n",
       "           6.9312e-02,  1.3550e+00, -1.7184e-01, -3.7272e-01,  4.7074e-01,\n",
       "           7.5560e-01, -4.3493e-01, -1.9175e-01,  1.0276e-02,  3.2891e-01,\n",
       "          -1.3878e-01, -1.7642e-01,  2.6787e-01, -1.9511e-01,  3.8617e-01,\n",
       "           9.5725e-02,  2.4829e-01, -1.7678e-01,  2.8686e-01,  1.1832e-01,\n",
       "          -1.6524e-01,  6.9805e-02, -4.4547e-01, -3.9223e-01,  3.0680e-01,\n",
       "          -2.1749e-02,  9.3136e-02, -2.1857e-01,  1.8633e-01,  4.5473e-01,\n",
       "           4.0357e-01,  7.8608e-01,  2.2033e-02,  9.3640e-02,  5.1990e-01,\n",
       "           1.7403e-01, -1.0740e-01,  2.1576e-01,  1.3863e-01,  2.6687e-01,\n",
       "           2.8553e-01,  3.8414e-01, -1.8430e-01,  1.4641e-01,  5.0828e-02,\n",
       "           2.6382e-01, -5.7501e-01, -4.3013e-01, -1.0506e-01,  6.6808e-01,\n",
       "          -3.8844e-01,  3.5382e-01, -2.2238e-02,  5.7493e-01,  5.4092e-01,\n",
       "          -8.3656e-01, -1.1576e-01,  3.1182e-02,  2.5347e-03, -1.2489e-01,\n",
       "          -5.8907e-01, -4.1538e-02,  1.0043e-01, -2.7401e-01,  3.7928e-01,\n",
       "           7.1619e-01, -6.1966e-01,  3.1257e-01,  3.1729e-01, -6.7739e-01,\n",
       "           2.5265e-02,  1.5000e-01,  9.2001e-02,  4.0077e-01, -4.8042e-02,\n",
       "          -2.0537e-01,  1.0943e-01, -7.6782e-02, -7.0749e-01,  4.0043e-01,\n",
       "           7.4198e-01,  2.4401e-01, -1.3833e-02,  1.0418e-01, -2.0272e-01,\n",
       "          -1.4871e-01, -4.7360e-01,  5.5543e-02,  1.6344e-01, -1.8461e-02,\n",
       "          -3.9720e-02, -1.2252e-01,  5.7282e-02,  8.9110e-02, -2.4997e-01,\n",
       "           4.9904e-01,  5.3516e-01,  4.7863e-01,  8.6978e-03, -2.3684e-01,\n",
       "          -8.7472e-02,  1.1482e-01, -1.3876e-01,  6.1790e-01, -1.3934e-01,\n",
       "          -3.3911e-01,  2.2228e-01,  2.8945e-01, -2.4145e-01,  1.6531e-01,\n",
       "          -3.5034e-01,  3.2541e-01,  1.6008e-01, -4.1667e-01,  4.3973e-01,\n",
       "           2.3738e-01,  2.3086e-01, -2.5404e-01,  1.2111e-01,  4.1491e-01,\n",
       "           5.8522e-02,  5.1465e-01, -2.2737e-02,  2.9813e-01, -2.4458e-01,\n",
       "           1.1972e-01, -5.0284e-01, -1.1951e-01,  7.1770e-04,  1.2702e-01,\n",
       "          -1.0748e-01, -1.0557e-01,  5.5317e-01, -5.5652e-02,  1.0575e+00,\n",
       "           6.8609e-01, -3.0630e-01, -6.4550e-01, -2.1561e-01,  7.1945e-02,\n",
       "          -1.4667e-01, -2.5251e-01, -8.8988e-04, -3.9614e-02,  2.0844e-01,\n",
       "          -1.0178e+00, -2.8086e-01,  9.7994e-02,  1.3319e-01, -3.9523e-01,\n",
       "           5.6600e-01,  9.8862e-02,  2.7929e-01,  9.9155e-01,  1.4863e-01,\n",
       "           1.6879e-01,  7.8061e-01,  5.9899e-02, -6.4476e-01,  4.9834e-02,\n",
       "           3.1766e-01, -1.6349e-01, -4.1936e-01, -7.8399e-01,  3.2991e-01,\n",
       "           2.1745e-01,  1.0502e+00,  3.8100e-01]], device='mps:0'),\n",
       " tensor([[-2.3826e-01, -9.8948e-01, -8.5186e-01, -8.4256e-01, -3.1978e-01,\n",
       "          -1.2611e-01, -4.2812e-01,  5.4158e-01,  6.1484e-02,  6.2180e-01,\n",
       "           1.4647e-01,  1.3534e-01,  1.6851e-02, -7.3653e-01, -6.9962e-01,\n",
       "          -6.9098e-02, -2.2383e-01,  3.8579e-01,  2.3332e-01,  3.7459e-01,\n",
       "           8.4867e-01,  2.3615e-01, -4.1344e-02,  1.6753e-01, -6.9576e-01,\n",
       "          -2.9169e-02, -2.4327e-02,  2.0645e-02,  2.4977e-01, -1.1554e+00,\n",
       "           4.9488e-01,  6.8174e-01, -1.0081e-01,  7.2590e-01, -1.3882e-01,\n",
       "          -2.1526e-01,  1.0325e-02, -2.8419e-01,  4.5258e-02, -3.2876e-01,\n",
       "           7.4670e-01,  1.2569e-01,  6.6426e-01, -9.2673e-02,  3.2800e-01,\n",
       "          -2.2908e-01,  2.1222e-01, -5.1727e-01,  3.2323e-01, -5.2252e-01,\n",
       "          -1.9203e-01,  1.3116e-02,  4.2819e-01,  1.9183e-01, -7.1999e-01,\n",
       "          -4.7734e-01,  9.5416e-03,  2.6379e-01, -7.8391e-01,  2.7055e-01,\n",
       "          -1.5596e-01,  3.9128e-01,  1.9872e-01,  2.8867e-02,  3.4965e-01,\n",
       "           2.1008e-02, -4.0546e-01,  2.2245e-02, -1.4713e-01, -3.1518e-03,\n",
       "           2.0481e-01,  3.6333e-01, -1.5350e-01, -2.8678e-01,  7.3597e-01,\n",
       "          -4.4390e-01,  1.8770e-01,  1.7028e-01, -6.7573e-01, -3.7933e-01,\n",
       "          -3.1855e-01,  1.6132e-01, -1.3734e-01, -2.8409e-01,  1.9451e-01,\n",
       "          -8.7122e-02,  4.7255e-01,  1.9420e-01,  1.4378e-01, -3.3788e-01,\n",
       "           3.9461e-01,  1.6375e-01, -5.1908e-01,  9.5400e-02,  3.5618e-01,\n",
       "          -2.0424e-01,  5.5616e-01, -3.2661e-01, -1.4644e-01, -2.7704e-01,\n",
       "           3.3249e-02,  4.1699e-01,  3.7319e-02,  1.2222e-01, -1.0814e+00,\n",
       "          -6.3536e-01,  8.2715e-03, -1.0569e-01, -8.1196e-01, -1.5418e-01,\n",
       "          -1.2928e-01, -4.9370e-02,  1.4969e-01, -8.4357e-01,  2.4682e-02,\n",
       "          -9.3749e-04,  8.1436e-02, -2.0079e-01, -1.3834e-01,  3.7661e-01,\n",
       "           3.1295e-01, -5.6213e-01,  2.5680e-01, -3.6024e-02,  2.4561e-02,\n",
       "          -7.3603e-02,  5.7231e-01, -1.2989e-01,  3.2361e-01,  5.2009e-01,\n",
       "          -2.9822e-01,  4.5720e-02, -3.6381e-01,  2.2739e-01, -3.6909e-03,\n",
       "          -9.3856e-02,  6.8121e-02, -5.9271e-01,  1.2008e-01,  2.1556e-01,\n",
       "          -2.3503e-01, -8.7281e-02, -9.7686e-02,  3.6105e-01,  1.8518e-01,\n",
       "          -1.7413e-01,  3.9814e-01, -1.3092e-01, -3.6792e-02, -2.5970e-01,\n",
       "          -7.1223e-01, -6.4111e-01,  5.2037e-01, -5.9768e-02,  1.7131e-01,\n",
       "          -1.5951e-01,  5.8257e-02, -1.9825e-02, -2.7515e-01,  2.4821e-01,\n",
       "           8.0844e-01, -6.0128e-02,  3.9272e-01, -9.4129e-01, -8.8894e-02,\n",
       "           8.1738e-02, -2.2270e-02, -2.7817e-01, -1.1105e-01, -1.1177e-01,\n",
       "           1.4348e-01, -3.7743e-01,  8.6289e-02, -5.0282e-01,  4.4592e-01,\n",
       "          -9.9624e-01, -3.0776e-02,  2.3947e-01, -1.0020e-01, -5.9626e-02,\n",
       "           3.6320e-01, -1.0052e+00,  1.6463e-01, -1.3295e-01,  2.2973e-01,\n",
       "          -4.0131e-01,  6.4484e-01, -2.9789e-03,  2.4669e-01, -2.0089e-01,\n",
       "           3.6512e-01, -1.3615e-01, -1.4065e-01,  5.4693e-01,  2.3785e-01,\n",
       "           3.0864e-01,  4.8404e-01,  1.8683e-01,  9.6097e-02, -2.6555e-02,\n",
       "          -1.1036e+00,  7.9368e-02, -2.3877e-01,  4.6239e-01,  4.2555e-01,\n",
       "           5.3610e-01, -4.3793e-01,  3.2449e-02, -2.7181e-01,  8.3538e-02,\n",
       "           1.0404e-01, -2.4834e-01, -4.7443e-01, -1.2711e-01, -4.6976e-01,\n",
       "           1.7689e-01,  2.9577e-03, -4.2854e-01,  3.7472e-01,  2.2045e-01,\n",
       "          -8.9011e-02, -3.4417e-01,  7.5840e-02, -3.8285e-01, -7.9154e-02,\n",
       "           3.3689e-01, -4.9356e-01,  1.0667e-01, -1.6332e-01, -5.0871e-02,\n",
       "           1.3296e-01, -9.1217e-01,  6.1597e-01,  3.3236e-01, -2.3338e-01,\n",
       "          -6.7038e-01, -3.3023e-01,  5.8144e-02, -5.7846e-02, -1.3082e-01,\n",
       "          -6.1106e-01, -4.8945e-01, -5.1485e-01,  1.1666e-01, -1.4884e-01,\n",
       "           3.3103e-01, -1.1911e-01,  9.8709e-01,  2.9596e-01,  1.3024e-01,\n",
       "          -2.8196e-02, -3.9310e-01,  7.6656e-02, -8.4519e-02, -2.2819e-01,\n",
       "          -4.6951e-01, -1.5234e-01, -5.2943e-02,  3.5153e-01, -3.1228e-01,\n",
       "          -5.5904e-01, -1.5518e-02, -1.1661e-01,  5.3903e-01, -1.4941e-01,\n",
       "           2.4477e-01, -1.9108e-01, -4.2569e-01, -1.6480e-01,  5.0973e-01,\n",
       "          -2.5837e-01,  3.3291e-01, -7.0805e-02, -3.8768e-01, -5.0787e-01,\n",
       "          -5.6285e-01,  1.9107e-02, -4.3659e-01,  5.7657e-01, -7.5030e-01,\n",
       "          -2.8552e-01, -3.6206e-01,  9.5385e-02, -9.7998e-02, -7.9773e-02,\n",
       "           1.1980e-01,  6.0945e-03,  1.8312e-01, -3.9363e-01, -4.8527e-01,\n",
       "          -9.9952e-02,  2.8662e-01, -2.6916e-01,  9.0172e-03,  2.3202e-01,\n",
       "          -4.6226e-02, -1.4692e-02,  2.6031e-01,  4.7815e-01, -1.4480e-01,\n",
       "          -4.4872e-01, -3.0692e-01, -5.2168e-02, -3.9658e-01,  1.9507e-02,\n",
       "           1.4306e-01,  3.5113e-01,  6.6600e-01, -3.0127e-01, -4.2530e-01,\n",
       "           8.7098e-03,  4.9157e-01, -1.8636e-01, -4.0586e-02,  2.7786e-02,\n",
       "          -1.3841e+00, -2.9968e-02, -2.9593e-01, -2.2757e-02, -1.2177e-01,\n",
       "          -5.9418e-02,  1.5594e-01, -2.4717e-02, -5.9531e-02, -3.4989e-01,\n",
       "           3.4660e-01,  3.3304e-01, -3.2338e-02, -4.2007e-02, -1.6391e-01,\n",
       "           6.3178e-02, -3.1460e-01, -8.6197e-03, -2.0917e-01, -1.2777e-01,\n",
       "           2.9236e-01,  4.5611e-01,  1.7501e-01, -1.4801e-01, -1.1176e-01,\n",
       "           2.6613e-01, -2.9001e-01, -1.0433e+00,  4.9242e-01, -4.9659e-01,\n",
       "          -7.0446e-01,  2.7031e-01, -1.3836e-01, -2.2008e-01,  2.9435e-01,\n",
       "           3.0186e-01, -1.3344e-01, -4.3200e-01, -1.3175e-01,  3.4126e-01,\n",
       "           6.5665e-01,  7.2629e-01, -2.9121e-01,  6.1532e-01,  2.4301e-01,\n",
       "          -2.3900e-01,  1.0319e-01, -2.6226e-01, -1.4586e-01, -1.9464e-01,\n",
       "           1.9260e-01,  4.0164e-01, -6.3168e-01, -2.9008e-02,  2.4701e-01,\n",
       "          -5.2206e-02,  2.1193e-02, -1.3054e-01,  4.4731e-01,  1.2402e-02,\n",
       "          -3.1307e-01, -4.8404e-01,  5.6730e-01, -3.0516e-01, -3.1022e-01,\n",
       "           7.6992e-02,  2.8638e-01,  1.4391e-02, -2.1397e-01, -1.9868e-01,\n",
       "          -3.2809e-01, -1.3661e-02,  6.9513e-01,  7.3853e-02,  3.1340e-01,\n",
       "          -2.6003e-02,  3.5852e-03, -1.7255e-01,  5.6526e-01,  2.5124e-01,\n",
       "          -3.3160e-01,  4.7172e-02, -1.7316e-01, -4.1505e-03,  3.5919e-01,\n",
       "          -3.8587e-01, -6.2402e-02,  1.1354e-01, -3.1321e-01,  1.8025e-01,\n",
       "          -6.0826e-01, -3.2461e-02,  7.2874e-01, -2.1043e-01,  1.2132e-01,\n",
       "           4.9732e-02,  3.4279e-01, -4.9801e-01,  1.9242e-02,  1.4070e-01,\n",
       "           1.9707e-01,  2.3748e-03,  6.8189e-02, -1.3914e-01, -4.4307e-01,\n",
       "           6.6466e-01,  3.9761e-01,  4.8002e-01,  3.9898e-01, -1.1195e+01,\n",
       "           1.6309e-01,  5.5161e-01, -2.5630e-01,  2.6039e-01,  1.8899e-01,\n",
       "           4.6184e-01, -4.5244e-01,  2.8223e-01, -2.3957e-01, -2.1267e-01,\n",
       "          -3.7617e-02,  1.8662e-01,  1.6723e-02, -5.0267e-01, -9.5912e-03,\n",
       "          -1.3974e-01, -7.3337e-01,  4.4726e-01,  1.9045e-01,  4.4207e-01,\n",
       "           4.5868e-02,  4.0577e-01,  1.7504e-01, -1.7238e-01, -5.4371e-01,\n",
       "           3.0592e-01, -3.6281e-01,  3.1361e-01,  5.0040e-01,  1.1778e-01,\n",
       "           2.0892e-01,  2.0327e-01,  6.2734e-02, -3.9494e-01, -6.1904e-01,\n",
       "          -2.8156e-01, -1.1134e-01,  9.2804e-02,  1.6724e-01, -3.3059e-01,\n",
       "          -1.8673e-01, -8.9827e-02,  3.7125e-01,  2.6552e-01,  5.9522e-01,\n",
       "           1.0729e+00,  6.6106e-01,  2.8758e-01, -3.4452e-01, -5.9945e-02,\n",
       "          -7.7988e-01,  1.2191e-01, -8.9657e-01, -1.8021e-02, -1.5854e-01,\n",
       "           5.4542e-01, -7.2852e-01,  8.1421e-02, -5.0230e-01, -6.0405e-02,\n",
       "           3.4659e-01, -1.2501e-02,  2.0471e-01, -4.6069e-01, -3.3272e-01,\n",
       "           4.0733e-01, -1.4545e-01,  2.0554e-01, -2.8070e-01, -1.4497e-01,\n",
       "          -4.9205e-01,  5.4911e-01,  2.7553e-01, -1.1012e-01,  5.8623e-03,\n",
       "          -2.9821e-02, -4.6359e-01,  1.7494e+00,  5.8417e-04, -1.0512e-01,\n",
       "           6.2314e-02,  2.7701e-01, -5.8352e-01, -7.1388e-01, -1.9215e-01,\n",
       "           3.6404e-02, -2.4834e-02,  1.9540e-01,  4.6667e-01, -3.8979e-02,\n",
       "          -3.8257e-01, -6.5428e-03,  3.5312e-01,  1.2769e-01,  7.3415e-01,\n",
       "          -5.3257e-01,  3.0047e-01,  1.6068e-01,  1.8098e-01, -4.4304e-01,\n",
       "           2.4307e-02, -5.0331e-02, -3.0448e-01,  3.7729e-01,  2.0361e-01,\n",
       "           1.8844e-01, -1.6010e-01, -1.8468e-01,  2.5086e-01, -6.0451e-01,\n",
       "           3.2121e-01,  8.4700e-02,  6.0196e-02, -4.0937e-01,  7.6817e-02,\n",
       "           3.8787e-01,  6.9470e-01, -3.9294e-01, -2.3357e-01, -4.4951e-01,\n",
       "           5.0644e-01,  3.1133e-01, -2.1917e-02, -4.9028e-01,  2.0401e-01,\n",
       "          -7.1524e-01,  1.4366e-01, -9.9357e-03,  3.0964e-01,  3.0912e-01,\n",
       "           2.5314e-01,  2.5636e-01,  5.5774e-01, -1.9231e-01, -5.6165e-01,\n",
       "          -5.0988e-01, -2.6352e-01,  1.5310e-01, -3.2051e-01,  1.2061e-01,\n",
       "           1.9167e-01, -3.0438e-02, -2.2860e-02,  4.1186e-01, -8.1402e-02,\n",
       "           5.3516e-01,  8.0382e-02, -1.7649e-01,  3.7712e-02, -3.4276e-01,\n",
       "          -2.6033e-01, -3.7526e-01, -8.5275e-02, -1.0536e-02, -2.1216e-01,\n",
       "          -1.6360e-01, -9.7076e-02,  1.3798e-01,  4.2480e-01,  2.8877e-02,\n",
       "           1.0814e-03, -2.3970e-01,  1.3716e-01,  6.1988e-01,  5.7224e-03,\n",
       "           8.3499e-04, -1.4748e-01, -2.3217e-01, -8.1074e-02, -1.8393e-01,\n",
       "           1.3988e-01,  5.6228e-02, -2.3655e-01, -6.4541e-01, -6.1312e-02,\n",
       "          -2.4069e-01,  1.3640e+00, -7.8438e-02, -6.0735e-01,  1.6491e-01,\n",
       "           3.3034e-01, -7.4894e-02,  1.2742e-01, -7.2691e-02,  2.3515e-01,\n",
       "          -6.0664e-01,  1.2437e-01,  4.2332e-01, -1.5484e-02,  2.8946e-01,\n",
       "           6.1516e-01,  2.2693e-01, -1.5114e-01,  8.0302e-01,  2.4073e-01,\n",
       "          -3.7684e-01, -1.3068e-01, -2.2997e-01,  1.5271e-01,  3.9491e-01,\n",
       "          -7.6537e-02, -1.1395e-01,  2.1830e-02, -7.5642e-02, -3.5195e-03,\n",
       "           2.0677e-01,  1.1749e+00, -2.5485e-01, -1.0071e-02,  5.6229e-01,\n",
       "           1.9148e-01, -1.0814e-01,  1.6731e-01,  4.7872e-01,  3.3632e-01,\n",
       "           3.6498e-01, -1.0140e-01, -1.6956e-01, -6.6799e-02,  3.3419e-02,\n",
       "           8.5460e-02, -5.3348e-01, -4.8368e-01,  1.8911e-02,  3.9912e-01,\n",
       "           3.2651e-01,  3.9622e-01, -1.0054e-01,  4.5582e-01,  6.6135e-01,\n",
       "          -8.4592e-01, -1.7822e-01,  5.2667e-02,  4.1555e-01, -4.7444e-02,\n",
       "          -1.2412e-01,  1.5815e-01, -1.2368e-01, -6.6644e-02,  2.8188e-01,\n",
       "           5.2184e-01, -8.8771e-01, -3.2638e-03,  2.6990e-01, -5.8304e-01,\n",
       "          -1.9644e-02,  1.0946e-01,  4.9663e-01,  3.3455e-01,  9.3217e-02,\n",
       "           2.8244e-02, -3.6868e-03, -1.5591e-01, -6.3941e-01,  1.8785e-01,\n",
       "           5.5164e-01, -1.4382e-01,  2.0123e-02,  3.5426e-02, -4.3531e-01,\n",
       "          -1.3001e-01, -6.0319e-01, -9.4121e-02, -1.2285e-01, -2.6515e-01,\n",
       "          -3.1979e-02, -3.1815e-01,  4.1892e-02,  4.7528e-01, -6.1369e-01,\n",
       "           3.2773e-01,  7.6535e-01,  4.9775e-01, -6.0046e-02, -1.8654e-02,\n",
       "          -2.0821e-03,  4.3341e-02,  7.3781e-02,  4.9567e-01, -8.2810e-02,\n",
       "          -4.3953e-01,  2.8885e-01,  1.6224e-01, -8.9925e-02,  1.6085e-01,\n",
       "          -6.0397e-01,  2.4190e-01,  9.5453e-02, -2.2519e-01,  3.1848e-01,\n",
       "           2.4106e-03,  3.8505e-01, -7.6775e-01,  2.1931e-01,  2.1983e-01,\n",
       "           1.3066e-01,  2.7412e-01,  9.7944e-02,  3.1354e-01, -1.0849e-01,\n",
       "           3.4227e-02, -4.8711e-01, -4.6712e-01,  1.8801e-01, -2.3430e-02,\n",
       "          -2.4801e-01,  5.8986e-03,  1.0220e-01,  5.5721e-01,  7.0030e-01,\n",
       "           6.8487e-01,  1.1286e-01, -8.8836e-01, -5.4721e-01, -9.1254e-02,\n",
       "           5.0443e-02, -3.6743e-01, -2.6688e-02, -3.9771e-01,  1.8954e-01,\n",
       "          -5.9067e-01, -5.5495e-01,  2.4269e-02,  1.1116e-01, -4.0207e-01,\n",
       "           6.5952e-01, -3.0483e-01,  4.5083e-01,  3.2201e-01,  3.0152e-01,\n",
       "           1.9961e-01,  5.1226e-01,  1.7712e-01, -6.9996e-01,  2.1716e-02,\n",
       "           3.2020e-01, -2.1408e-01,  5.0366e-01, -2.8455e-01,  3.3972e-01,\n",
       "           1.7364e-02,  1.1144e+00,  5.1342e-01]], device='mps:0'),\n",
       " tensor([[-1.4377e-01, -3.5339e-01, -5.3634e-01, -4.9456e-01, -5.1886e-02,\n",
       "          -2.5693e-01, -6.9681e-01,  3.1808e-01,  8.6652e-02,  5.1654e-01,\n",
       "           2.5260e-01,  1.3162e-01,  3.3845e-03, -3.4256e-01, -9.0008e-01,\n",
       "          -2.2430e-01,  1.7443e-01,  4.5454e-01,  5.7781e-01,  3.7904e-01,\n",
       "           3.3494e-01,  3.2062e-01, -1.2243e-01,  1.2883e-01, -6.6967e-01,\n",
       "           1.4636e-01, -2.6573e-01, -5.5923e-02,  1.1721e-01, -4.0657e-01,\n",
       "          -1.8805e-02,  3.5102e-01, -3.7222e-01,  8.9251e-01, -2.1152e-01,\n",
       "          -2.8909e-01,  2.3486e-02, -4.2648e-01,  6.4426e-01, -1.0732e-01,\n",
       "           8.0893e-01, -5.5422e-02,  3.3311e-01,  6.2481e-02,  1.5511e-01,\n",
       "          -6.8196e-02,  3.4319e-02, -3.7309e-01, -6.2158e-02, -4.0182e-01,\n",
       "           3.7403e-04,  4.6154e-02,  7.2672e-01,  8.4193e-02, -9.0040e-01,\n",
       "          -3.9226e-01, -2.1874e-01,  5.3091e-01, -4.6501e-01,  1.9677e-02,\n",
       "          -1.3296e-01,  4.7974e-01,  1.9625e-01,  7.3650e-02, -2.5842e-01,\n",
       "           1.1453e-01, -3.0628e-01, -6.0462e-01, -4.9562e-01, -1.4414e-01,\n",
       "           2.1002e-01,  3.8112e-01,  2.6528e-01, -6.1453e-01,  5.4329e-01,\n",
       "          -3.4927e-01,  4.4851e-01, -1.6241e-01, -5.7691e-01, -7.6040e-02,\n",
       "           3.1163e-01,  3.3718e-01, -2.5736e-01,  2.4491e-02,  3.2310e-01,\n",
       "           5.1780e-02,  5.6526e-01,  1.4282e-01,  2.8508e-02, -1.8635e-01,\n",
       "           3.8831e-01,  8.1135e-03, -9.4702e-01, -1.3060e-01,  1.3874e-01,\n",
       "          -4.9431e-01,  3.9847e-02, -3.2263e-01,  5.9491e-02, -1.4452e-01,\n",
       "           9.8331e-03,  3.6659e-01,  1.3483e-01,  3.2132e-01, -9.8990e-01,\n",
       "          -4.4463e-01,  1.7400e-01, -1.2367e-01, -6.9027e-01, -5.1993e-02,\n",
       "           5.5120e-02, -8.6914e-02,  5.3725e-01, -2.7357e-01, -4.8890e-01,\n",
       "           3.4404e-02,  4.9849e-02,  2.8178e-01, -4.9994e-04,  6.5672e-01,\n",
       "           3.2249e-01, -1.4702e-01, -4.9088e-02, -1.3937e-01,  2.6952e-01,\n",
       "          -3.0511e-01,  3.4786e-01, -3.5301e-01,  4.3460e-01,  2.2329e-01,\n",
       "           3.7377e-02,  1.2477e-01, -5.3757e-01, -8.0503e-03,  3.7463e-02,\n",
       "           1.6149e-01,  2.0054e-01, -6.3603e-01,  2.3898e-01,  3.3749e-01,\n",
       "           8.5077e-02, -3.2517e-03, -2.6044e-01, -4.0277e-03, -3.0857e-01,\n",
       "          -1.4844e-01,  2.9062e-01, -2.7679e-01, -8.9774e-02, -2.2734e-01,\n",
       "          -6.2794e-01, -5.0908e-01,  2.3385e-01, -1.2103e-01,  1.7286e-01,\n",
       "          -1.2852e-02,  2.7716e-01, -2.1905e-01, -3.2973e-01, -7.2670e-02,\n",
       "           6.5128e-01,  5.0284e-02,  4.0820e-02, -4.3340e-01, -2.7870e-01,\n",
       "           1.0885e-01,  2.1568e-01, -7.7751e-02, -4.2641e-01,  3.0010e-02,\n",
       "           3.1737e-01, -4.6850e-02, -1.8039e-02, -4.5309e-01,  1.3986e-01,\n",
       "          -6.6008e-01,  4.4838e-01,  2.0563e-01,  2.2296e-01,  1.1239e-01,\n",
       "          -5.2571e-02, -1.5319e+00,  4.8858e-01, -5.3618e-01,  1.1577e-01,\n",
       "          -1.8829e-01,  3.8406e-01,  1.5898e-01,  2.4491e-01, -8.6149e-02,\n",
       "           5.3582e-02,  2.5225e-02, -1.1612e-01,  3.6091e-01, -1.7275e-02,\n",
       "           4.4956e-01,  6.1877e-01,  2.2927e-01,  1.6664e-01, -7.9420e-03,\n",
       "          -1.0215e+00,  6.4986e-02,  2.3450e-01, -5.5388e-02,  2.6168e-01,\n",
       "           4.3630e-01, -1.5857e-01, -1.8584e-01,  1.6024e-01, -7.1109e-03,\n",
       "           2.6564e-02, -4.9057e-01, -4.0424e-01, -2.1819e-01,  7.7592e-02,\n",
       "           7.2278e-03, -1.5676e-01, -2.4066e-01,  7.3743e-01, -5.0529e-03,\n",
       "          -4.8860e-02,  5.5519e-02, -3.6798e-01, -3.8508e-01, -2.9032e-01,\n",
       "           9.4347e-02, -1.9156e-01, -3.0501e-03, -6.4518e-01, -1.8981e-01,\n",
       "           3.2960e-01, -7.5000e-01,  4.8094e-01,  3.8496e-01,  3.7133e-04,\n",
       "          -9.0272e-02, -1.0712e-01,  1.0419e-01, -1.0585e-01,  3.6810e-02,\n",
       "          -7.4216e-03, -3.5432e-01, -6.1034e-01,  2.7165e-01, -1.2673e-01,\n",
       "           3.0242e-01, -1.2405e-01,  5.9864e-01,  5.0067e-01,  4.1290e-01,\n",
       "           2.8772e-01, -1.6564e-01,  1.7661e-01,  3.9415e-01, -3.7117e-01,\n",
       "           2.1260e-01,  3.1802e-02, -2.0875e-01,  2.2951e-01, -4.8719e-01,\n",
       "          -5.3262e-01,  1.6847e-01,  3.5203e-01,  4.9152e-01, -8.6072e-02,\n",
       "           1.1345e-01,  3.2766e-01, -1.3928e-01, -1.0070e-01,  3.5444e-01,\n",
       "          -3.7061e-01,  7.6707e-02, -1.4006e-01, -1.5900e-01, -4.8223e-01,\n",
       "          -6.9099e-01, -1.7761e-01, -1.6863e-01,  1.2109e-01, -9.1467e-01,\n",
       "          -3.1044e-01,  1.7290e-01,  2.3754e-02, -4.8192e-02, -1.1823e-01,\n",
       "           3.6563e-02, -8.2622e-02,  1.2264e-02, -4.8097e-01, -8.4119e-01,\n",
       "          -1.1328e-01,  8.4687e-02, -1.7354e-01,  1.1131e-01,  2.2855e-01,\n",
       "          -1.6713e-01, -2.2917e-01,  3.5749e-01,  7.3729e-02, -1.9815e-01,\n",
       "          -1.5114e-01, -1.9106e-01,  1.2153e-01, -6.0016e-02,  1.2904e-01,\n",
       "           7.2433e-02,  3.4854e-01,  7.1746e-01, -7.7168e-01, -5.9605e-01,\n",
       "          -6.6696e-02,  4.0949e-01, -2.4112e-01, -2.4620e-01, -1.1831e-01,\n",
       "          -1.0040e+00, -1.7854e-01,  2.6551e-02, -3.0759e-01, -3.8163e-01,\n",
       "           3.4242e-01, -1.0039e-01, -8.9868e-03, -5.0261e-02, -4.3412e-01,\n",
       "           5.0820e-01,  9.5370e-01, -1.2584e-01, -2.7551e-01, -4.8335e-01,\n",
       "          -2.0211e-02, -2.2929e-01, -1.5375e-01, -4.4088e-01, -1.9653e-01,\n",
       "           3.1382e-01,  2.4982e-01, -3.7299e-01, -1.5074e-01, -2.7865e-01,\n",
       "           4.7552e-01, -5.9756e-01,  1.9871e-01,  2.9464e-01, -3.7171e-01,\n",
       "          -2.1167e-01,  4.6122e-01, -1.5472e-01,  9.1506e-02,  1.5522e-01,\n",
       "           1.0678e-01, -6.8769e-02, -1.6548e-01, -1.5693e-01,  9.5020e-01,\n",
       "           4.0899e-01,  1.9186e-01, -3.0768e-01,  5.5355e-01,  3.9915e-01,\n",
       "          -2.2649e-01,  2.5943e-01, -1.8648e-01, -3.0715e-01, -2.4450e-01,\n",
       "           9.3625e-02,  9.2667e-02, -4.1848e-01,  2.2849e-01, -4.9524e-02,\n",
       "           2.3394e-01, -2.9335e-01,  2.0832e-01,  9.5334e-02,  2.8972e-01,\n",
       "           3.2617e-02, -2.9147e-01,  5.5028e-01, -2.7886e-01,  1.5665e-01,\n",
       "          -2.6506e-01, -2.0474e-01,  2.0551e-01, -2.6119e-02, -5.1385e-01,\n",
       "          -3.1081e-01, -9.4319e-02,  5.6396e-01, -1.1403e-01, -2.1612e-02,\n",
       "           2.8447e-01, -3.4506e-01, -4.1756e-01,  1.3022e-01, -2.6786e-01,\n",
       "          -7.9530e-02,  2.4247e-01, -1.0200e-01, -6.2871e-02,  2.7486e-01,\n",
       "          -3.3798e-02, -1.9316e-01, -2.0761e-02, -2.3981e-02,  2.7087e-01,\n",
       "          -3.7747e-01,  3.6542e-03,  4.2682e-01, -1.0020e-01,  2.8702e-01,\n",
       "           1.7970e-01,  4.0208e-01, -8.2380e-02,  1.5204e-02, -5.6097e-01,\n",
       "          -1.1402e-01, -2.1131e-01,  8.3132e-02, -2.9606e-02, -5.2771e-01,\n",
       "           5.5282e-01,  4.3985e-01,  8.8702e-02,  4.1910e-01, -1.1534e+01,\n",
       "           9.5258e-02,  4.7440e-01, -3.2415e-01,  3.2352e-01,  2.2484e-01,\n",
       "           1.3936e-01, -3.2025e-01,  1.9101e-01, -2.0227e-01, -1.7819e-01,\n",
       "          -4.5774e-01, -5.0356e-02,  1.3142e-02, -4.8614e-01,  3.9332e-02,\n",
       "          -1.2464e-01, -6.1531e-01,  7.9218e-02, -1.3511e-01,  2.6491e-01,\n",
       "          -2.4662e-01,  9.7381e-02,  1.4166e-01, -1.3334e-01, -5.0099e-02,\n",
       "           2.3738e-02, -2.7515e-01,  5.4071e-01,  1.1970e-01, -9.7257e-02,\n",
       "           1.4224e-01, -2.1359e-01,  2.4150e-01, -2.1950e-01, -4.5378e-01,\n",
       "          -1.0075e-01, -1.1904e-01, -3.7262e-02, -1.3390e-01, -7.7843e-02,\n",
       "          -3.0477e-01, -5.0960e-01,  1.0866e-01,  3.6725e-01,  5.3166e-01,\n",
       "           1.1795e+00,  8.0081e-01,  3.8607e-01,  8.5570e-04,  7.8720e-02,\n",
       "          -4.5270e-01,  2.9220e-01, -7.2562e-02, -2.4382e-01, -2.7895e-01,\n",
       "           5.8078e-01, -6.0418e-01,  4.3898e-02, -2.5668e-01, -1.7958e-01,\n",
       "           3.6628e-01,  1.6101e-01, -1.2604e-01, -2.7369e-01, -8.2986e-03,\n",
       "           4.8775e-01, -2.3151e-01,  1.8496e-01, -4.7167e-01,  1.9530e-01,\n",
       "          -7.2079e-02,  3.4669e-01,  5.1873e-02, -7.1069e-02, -3.7878e-01,\n",
       "          -2.6082e-01, -1.8377e-01,  1.3559e+00,  7.9852e-02,  1.9681e-01,\n",
       "           5.6962e-01,  7.2901e-03, -6.9790e-01, -5.5505e-01, -4.8313e-01,\n",
       "          -7.3330e-02, -8.7465e-02,  9.1930e-02,  5.1256e-01, -3.4002e-01,\n",
       "          -7.4317e-01, -2.2213e-01,  1.8918e-01,  5.5941e-01,  5.5623e-01,\n",
       "          -3.5597e-01, -1.8058e-01, -1.6288e-01, -3.3618e-02, -5.5369e-01,\n",
       "          -7.7926e-02, -5.5450e-02, -2.0654e-01,  2.4948e-01,  2.1920e-02,\n",
       "          -1.0258e-01, -8.6256e-03, -2.1265e-01,  7.6559e-03, -6.0568e-01,\n",
       "           2.8988e-01, -9.7380e-02, -4.6749e-02, -3.3996e-01, -3.1118e-01,\n",
       "           4.4484e-01,  6.2927e-01, -1.6053e-01, -1.2107e-01, -2.0224e-01,\n",
       "           3.5473e-01,  2.8807e-01,  8.7794e-02, -3.6056e-01,  3.7164e-01,\n",
       "          -4.8103e-01,  4.3662e-02,  1.7431e-01,  4.6729e-01,  5.4979e-01,\n",
       "          -2.6758e-01, -5.3820e-01,  2.1175e-01, -2.5987e-01, -3.7751e-01,\n",
       "          -6.0560e-01, -8.4008e-02,  1.3395e-01, -5.6481e-01, -1.2253e-01,\n",
       "          -3.7867e-01,  2.3447e-01, -3.7237e-01,  2.4020e-01,  3.9839e-02,\n",
       "           2.9359e-01,  4.7748e-01, -2.5885e-02,  4.7927e-01, -2.6202e-01,\n",
       "           1.8927e-02,  3.4049e-01, -2.6260e-01,  1.4226e-02, -3.2795e-01,\n",
       "          -5.9153e-01, -4.9066e-02, -1.2252e-01,  2.3597e-01, -1.5389e-01,\n",
       "           2.5321e-01,  8.7510e-02, -1.2482e-01,  4.3954e-01,  4.4465e-01,\n",
       "           1.3627e-01, -5.1491e-01,  1.2865e-01, -1.2985e-03,  3.9092e-02,\n",
       "           3.9072e-01,  1.7983e-01,  8.2741e-02, -4.6287e-01, -1.5133e-01,\n",
       "          -2.3033e-01,  1.0406e+00, -2.2388e-02, -9.5083e-02,  1.1606e-01,\n",
       "           4.5899e-01, -1.6139e-01, -2.1277e-01, -1.8920e-01,  3.6141e-01,\n",
       "          -2.1079e-01, -3.6315e-01,  3.4721e-01,  9.0328e-02,  5.5239e-01,\n",
       "           7.0195e-03,  3.5713e-01, -1.7679e-01,  7.2278e-01,  4.4275e-01,\n",
       "          -3.6996e-01, -9.3348e-02, -3.6926e-01, -2.7661e-01,  4.5076e-01,\n",
       "          -7.7695e-02,  3.1801e-01, -1.2224e-01,  7.4557e-02, -1.7302e-01,\n",
       "           1.5753e-01,  1.2106e+00, -3.4125e-01, -2.0339e-01,  6.6461e-01,\n",
       "           3.4429e-01,  2.5891e-01,  2.8457e-01,  4.5806e-02,  2.3445e-01,\n",
       "           5.8242e-01, -2.1121e-01, -3.7708e-01,  2.0801e-01,  4.0874e-01,\n",
       "           5.9887e-01, -8.7002e-01, -3.9942e-01, -8.8226e-02,  1.5519e-01,\n",
       "          -4.9212e-01,  2.5871e-01,  1.8562e-02,  2.5176e-01,  7.2352e-01,\n",
       "          -1.0426e+00,  1.4111e-01, -2.2935e-01,  4.1728e-01,  2.4571e-02,\n",
       "          -4.4434e-01,  5.6553e-02, -9.3964e-02, -1.0673e-01,  5.3025e-01,\n",
       "           4.8934e-01, -1.2644e+00,  1.3391e-01,  5.3754e-01, -3.0881e-01,\n",
       "           1.0552e-01, -5.5998e-02,  2.4829e-01,  1.0712e-01,  2.5325e-01,\n",
       "          -3.8970e-01,  3.2141e-02, -5.5435e-01, -6.6460e-01,  2.0342e-01,\n",
       "           6.3216e-01,  1.7715e-01, -5.8213e-02,  5.8209e-02, -2.3681e-01,\n",
       "          -2.0749e-01, -5.9957e-01, -3.0250e-01, -6.2136e-02, -1.0400e-01,\n",
       "          -1.7009e-02, -3.1611e-01,  1.3663e-01,  1.3665e-01, -3.0847e-01,\n",
       "           6.5143e-01,  2.4142e-01,  9.2265e-01, -1.7151e-02,  1.1846e-01,\n",
       "           2.1958e-02,  2.8022e-01, -2.6195e-03,  5.0591e-01, -4.5114e-02,\n",
       "          -2.0985e-01,  3.3030e-01, -2.4325e-02, -3.5653e-01,  6.9002e-03,\n",
       "          -4.1190e-01,  4.3693e-01,  8.7995e-02, -3.0946e-01,  3.2244e-01,\n",
       "          -7.2257e-02,  1.8918e-01, -6.6377e-01,  4.6004e-01,  2.0187e-01,\n",
       "           5.8912e-02,  8.1044e-01,  2.2829e-01,  5.6647e-01,  1.6959e-01,\n",
       "           2.2678e-02, -9.6541e-01, -1.6068e-02, -2.1736e-01,  1.3132e-02,\n",
       "           6.9987e-02, -1.8030e-01,  5.4668e-02,  9.4569e-02,  6.8689e-01,\n",
       "           6.0243e-01, -2.8026e-01, -6.1400e-01, -3.6858e-01,  1.3248e-01,\n",
       "          -8.8998e-02, -1.3074e-01, -3.7086e-02, -2.1796e-01,  2.3184e-01,\n",
       "          -5.4442e-01, -3.6110e-01,  9.5467e-02,  3.9623e-01, -2.7848e-01,\n",
       "           5.5323e-01,  7.4920e-02,  3.0466e-01,  1.3305e-01,  2.6188e-01,\n",
       "           1.9911e-01,  4.6279e-01,  1.5046e-01, -5.5550e-01,  2.2801e-02,\n",
       "           1.8402e-01, -2.2380e-01,  2.3907e-01, -8.8521e-01,  3.0932e-01,\n",
       "           2.3739e-01,  4.9221e-01,  4.6405e-01]], device='mps:0'),\n",
       " tensor([[-2.7791e-02, -7.5747e-01, -5.3074e-01, -5.1430e-01, -3.4417e-01,\n",
       "           7.2208e-02, -3.2733e-01,  1.9133e-01, -1.5003e-01,  5.7895e-01,\n",
       "           2.7665e-01,  1.8669e-01, -4.0441e-02, -1.9787e-02, -2.7559e-01,\n",
       "          -3.6770e-01, -5.5075e-01,  4.2714e-01,  4.8599e-01,  1.2291e-01,\n",
       "           6.0920e-01,  4.0016e-01, -5.7060e-01,  2.7172e-01, -1.0960e+00,\n",
       "           2.2592e-01, -9.0650e-02,  1.6102e-01,  2.6783e-01, -4.2947e-01,\n",
       "          -2.6593e-01,  3.3457e-01, -1.0081e-01,  1.1911e+00, -7.2573e-02,\n",
       "          -4.2319e-01,  1.2905e-01, -4.8449e-01,  2.2964e-01,  1.0587e-01,\n",
       "           3.3713e-01, -2.3616e-01,  3.0495e-01,  2.5426e-01,  2.2269e-01,\n",
       "          -1.0156e-01,  3.5441e-01, -6.7506e-01,  5.3763e-02, -5.4964e-01,\n",
       "          -2.7866e-01, -2.3157e-01,  3.8469e-01,  3.0486e-01, -1.7402e-01,\n",
       "          -1.7631e-01,  1.9710e-01,  2.8978e-01, -1.6215e-01,  1.9001e-01,\n",
       "          -3.1728e-01,  3.9028e-01, -2.9116e-02, -2.5337e-02, -4.3117e-01,\n",
       "           5.2338e-02, -1.0542e+00, -2.0097e-01, -6.1678e-01, -2.9287e-01,\n",
       "           6.1601e-02, -4.4671e-02,  2.6994e-02,  8.2943e-02,  6.2566e-01,\n",
       "          -2.9527e-01, -3.5686e-02, -6.8721e-02, -8.9378e-01,  3.5990e-02,\n",
       "           9.3805e-02,  2.8378e-01,  3.4715e-01, -1.8362e-01,  4.0523e-02,\n",
       "          -1.1761e-01,  5.2336e-01,  3.2885e-01,  5.5885e-02, -1.1072e-01,\n",
       "           2.6658e-01,  1.4680e-01,  3.1471e-01, -2.3564e-01,  2.1408e-01,\n",
       "          -1.6953e-01,  4.2040e-01, -7.0445e-02, -1.3569e-01, -1.7957e-01,\n",
       "           1.7683e-01,  2.7367e-01,  1.6921e-01,  1.8827e-01, -7.0935e-02,\n",
       "          -4.6949e-01,  6.4462e-03, -2.7295e-01, -9.2490e-01,  5.3292e-02,\n",
       "           2.5952e-02,  8.0243e-02,  3.5441e-01, -5.5720e-01, -4.9249e-01,\n",
       "           2.3824e-01,  3.1274e-01, -7.8117e-02, -2.4385e-01,  7.2875e-02,\n",
       "           1.2005e-01, -3.7322e-01,  3.0937e-01,  1.4655e-01,  3.8964e-01,\n",
       "          -2.4076e-01,  5.1062e-01, -1.8687e-01,  4.6773e-01,  1.7625e-01,\n",
       "          -2.8715e-02, -1.4186e-02,  2.4065e-02, -2.5766e-01, -7.0436e-02,\n",
       "          -3.2938e-01,  4.0965e-01, -2.1077e-01,  1.5229e-01, -2.7347e-01,\n",
       "          -1.9571e-02, -1.5879e-01, -4.2139e-01, -2.4379e-01,  5.2323e-01,\n",
       "          -2.3310e-01,  3.4454e-01, -2.5864e-01, -5.3971e-01, -5.4621e-01,\n",
       "          -4.6331e-01, -6.6221e-01,  6.5904e-04, -5.8988e-02,  1.5486e-02,\n",
       "          -1.4458e-01, -4.7299e-02,  3.4941e-01, -1.5764e-01, -1.3263e-01,\n",
       "           1.3229e+00, -1.6963e-01,  2.6547e-01, -5.2711e-01, -9.2709e-02,\n",
       "          -2.5951e-01,  3.5857e-01, -2.1281e-01, -5.0373e-02, -1.1031e-02,\n",
       "           1.5012e-01, -3.6325e-01,  3.2426e-01, -2.7495e-01,  4.9574e-02,\n",
       "          -3.8145e-01,  6.7830e-01,  1.0315e-01, -3.7603e-03, -2.1426e-01,\n",
       "           8.8470e-02, -1.8509e+00, -6.0244e-02, -4.9988e-02, -1.1984e-01,\n",
       "          -3.8304e-01,  1.1412e-01, -8.1008e-02,  4.7211e-01,  1.0523e-01,\n",
       "          -3.0391e-01,  1.4457e-02,  3.4290e-01,  1.7766e-01,  4.6617e-02,\n",
       "           3.7586e-01,  5.1268e-01,  5.2866e-01,  8.3422e-02, -1.2969e-01,\n",
       "          -2.2910e-01,  1.8840e-01, -1.4922e-01, -4.9168e-03,  1.3625e-01,\n",
       "           1.9163e-01, -1.7921e-02,  4.5103e-02,  4.5549e-01, -3.9944e-02,\n",
       "          -1.3034e-01, -3.0975e-01, -2.0571e-01, -1.5386e-01, -3.0506e-01,\n",
       "           6.6388e-01, -6.9598e-02,  1.5083e-01,  1.1394e-01, -7.6365e-02,\n",
       "          -1.9021e-01, -3.6627e-01,  7.4035e-02, -4.3048e-01, -1.0458e-01,\n",
       "           5.2612e-01, -2.4620e-01, -9.6266e-02, -1.5057e-01, -2.1628e-01,\n",
       "           4.6617e-01, -6.3064e-01,  6.9359e-01, -6.5753e-02, -1.5876e-01,\n",
       "          -6.3829e-02, -3.6947e-01,  8.8525e-02, -4.7900e-02, -2.1367e-01,\n",
       "          -6.6109e-01, -6.2912e-01, -9.3484e-01,  3.0931e-02,  2.0012e-01,\n",
       "           3.8940e-01, -2.0282e-01,  7.0611e-01,  4.1043e-01,  4.4392e-01,\n",
       "          -1.9033e-01, -2.5634e-01,  1.9540e-01,  2.4175e-02, -1.6470e-01,\n",
       "          -8.3407e-02, -3.6482e-02, -2.3338e-01,  5.4958e-01, -2.6909e-01,\n",
       "           3.4143e-02, -8.4658e-02,  1.4949e-01,  5.6272e-01, -2.6404e-02,\n",
       "           2.7543e-01,  3.6181e-02, -5.5097e-01, -2.6403e-02,  1.4938e-01,\n",
       "          -3.9414e-01,  1.3074e-01,  6.3249e-01, -2.9116e-01, -5.8419e-01,\n",
       "          -5.6450e-01,  1.3798e-02, -1.0222e-01,  2.9551e-01, -1.1019e+00,\n",
       "          -1.6877e-01,  2.8032e-01,  3.0136e-02,  1.8251e-01, -3.0887e-01,\n",
       "           2.9186e-01,  5.8483e-02,  3.7632e-02, -7.8447e-02, -5.1537e-01,\n",
       "          -2.3439e-01,  1.6469e-01, -3.6657e-01, -3.5963e-02,  1.6028e-01,\n",
       "           2.8092e-01, -9.8223e-02,  4.3126e-02,  2.2221e-01, -4.8483e-01,\n",
       "          -8.7845e-02, -3.3919e-02,  1.5657e-01,  6.3212e-02,  6.2429e-02,\n",
       "           7.6042e-02,  1.9293e-01,  7.3101e-01, -3.7007e-02, -6.0240e-01,\n",
       "          -2.4535e-01,  5.4258e-01,  7.7029e-02,  1.5324e-01, -1.7752e-01,\n",
       "          -1.4260e+00, -6.3469e-01,  3.8337e-02,  2.1822e-02, -9.6237e-02,\n",
       "           3.7186e-01,  1.4415e-01,  2.9691e-02,  1.0135e-01, -1.4892e-01,\n",
       "           2.7125e-01,  2.5571e-01,  4.1796e-01,  2.6634e-01,  1.2224e-01,\n",
       "          -9.5239e-02, -2.5680e-01, -3.4749e-01, -2.5056e-01,  5.0008e-02,\n",
       "           3.0551e-01,  2.9752e-01, -2.5921e-01, -9.4277e-02, -9.0702e-02,\n",
       "           7.3119e-01, -4.5527e-01, -5.6552e-01,  3.9878e-01, -5.2456e-01,\n",
       "          -2.7738e-01,  4.5605e-01, -3.7995e-01,  6.1826e-02, -3.0256e-02,\n",
       "           1.6101e-01, -7.3998e-02,  5.4701e-02,  5.9767e-02,  1.4680e-01,\n",
       "           1.8510e-01,  4.4459e-01, -9.6782e-02,  3.8301e-01,  4.8393e-01,\n",
       "          -3.7868e-01,  1.0631e-01, -3.1295e-01,  3.3607e-02,  3.3206e-02,\n",
       "           3.5477e-02,  1.0657e-01, -5.8454e-01, -7.1591e-02,  1.8413e-01,\n",
       "           3.9868e-02, -5.0830e-01, -1.9227e-01,  1.5691e-01, -1.5280e-01,\n",
       "          -1.9063e-02, -6.1864e-01,  7.4105e-02,  2.7357e-02, -1.0623e-01,\n",
       "           2.0196e-01,  3.6702e-02,  1.0747e-01, -2.9594e-01, -3.5062e-01,\n",
       "          -4.3860e-01,  1.3449e-01,  6.4194e-01, -1.6284e-02, -1.8554e-02,\n",
       "          -1.0969e-02, -1.8924e-01, -3.0452e-01,  4.8185e-01,  1.5484e-01,\n",
       "          -2.7278e-01,  1.5357e-01, -1.5461e-01,  4.3362e-01,  4.5208e-01,\n",
       "          -3.5378e-01, -1.2043e-01, -1.1214e-01,  3.6801e-01, -1.9150e-01,\n",
       "          -1.7609e-01,  2.8881e-01,  4.5279e-01,  1.9897e-01,  9.4702e-02,\n",
       "           2.5524e-01,  6.2194e-01, -7.1092e-01,  2.7741e-01, -5.0384e-02,\n",
       "           1.6210e-01, -3.1364e-01,  1.1810e-01, -4.5970e-01, -3.1796e-01,\n",
       "           2.0608e-01,  3.4692e-01,  1.5942e-01, -8.2525e-02, -1.1386e+01,\n",
       "           4.7410e-01,  7.1675e-01, -6.4620e-01, -1.2380e-02,  2.0990e-01,\n",
       "           1.4467e-01, -4.2321e-01,  1.2785e-01, -4.0986e-01, -4.8454e-01,\n",
       "           5.3751e-02, -3.1024e-01,  8.5046e-02, -2.1082e-01, -3.5012e-02,\n",
       "          -3.3293e-01, -8.8604e-01,  5.2616e-01, -2.3722e-01,  1.2282e-01,\n",
       "          -4.0554e-01,  2.2834e-01,  6.7441e-02, -3.7185e-01, -2.9844e-01,\n",
       "          -3.3378e-01, -4.8637e-01,  2.5874e-01,  1.4526e-01, -3.6388e-01,\n",
       "           1.6897e-01,  7.8214e-02,  1.5488e-01, -3.1408e-01, -5.7322e-01,\n",
       "           6.5890e-02, -1.4286e-01,  6.9077e-01, -2.3317e-01,  2.3521e-01,\n",
       "           1.5418e-01, -1.8939e-01,  4.0595e-01, -6.7756e-02,  5.4405e-01,\n",
       "           9.0099e-01,  4.9506e-01, -9.3677e-02, -3.1796e-01,  2.9697e-01,\n",
       "          -1.1296e+00,  2.3765e-01, -7.4326e-01,  3.7037e-02, -4.4715e-02,\n",
       "           5.2303e-01, -5.4682e-01, -1.6110e-02,  1.6453e-01, -1.3098e-01,\n",
       "           3.8056e-01,  1.5443e-02,  3.0641e-01, -3.8540e-01, -1.8694e-01,\n",
       "           5.1715e-01,  1.1540e-01,  3.9804e-01, -2.1344e-01, -2.7691e-02,\n",
       "          -3.5584e-01,  6.3533e-01,  4.1696e-01,  6.3335e-03,  1.0379e-01,\n",
       "           9.4310e-02,  1.5461e-01,  8.2262e-01,  1.5306e-01,  3.7272e-02,\n",
       "           7.5334e-02,  4.9919e-02, -2.0803e-02, -2.5454e-01, -1.9359e-01,\n",
       "          -3.8527e-02,  2.3219e-01,  6.4573e-01,  5.1229e-01, -2.5336e-01,\n",
       "          -1.8886e-01, -2.0242e-01, -5.9704e-02,  1.3338e-01,  5.6888e-01,\n",
       "          -7.2007e-01,  1.4785e-02,  2.0411e-01, -2.3545e-01, -2.6105e-01,\n",
       "          -9.1127e-02, -2.2821e-01, -6.1202e-01,  4.0102e-01, -3.4853e-03,\n",
       "          -3.2201e-01,  8.7693e-02,  5.6282e-01,  1.7925e-01, -2.9609e-01,\n",
       "           4.8297e-01, -1.4248e-01,  5.2177e-01, -7.0864e-02,  3.5673e-02,\n",
       "           3.0329e-01,  6.1367e-01, -2.1669e-01, -3.9304e-01, -6.3589e-02,\n",
       "          -7.9591e-02,  2.2539e-01,  3.5268e-03,  7.8083e-02, -1.6630e-02,\n",
       "          -3.5275e-01, -1.3786e-01,  3.3867e-02,  2.5058e-01,  2.6520e-01,\n",
       "           6.0789e-02,  3.0671e-01,  2.9447e-01, -3.1026e-01, -2.5075e-01,\n",
       "          -7.3840e-01, -2.4288e-01,  2.2993e-01, -1.9391e-01, -9.8460e-02,\n",
       "          -2.1162e-02, -3.4200e-02, -3.9679e-01,  3.1646e-01, -8.8932e-02,\n",
       "           1.4835e-01,  1.1151e-01, -4.4561e-01,  2.3351e-01, -3.3892e-02,\n",
       "          -3.1032e-01, -9.2286e-02, -3.7850e-01,  8.2453e-02,  3.6353e-02,\n",
       "          -4.8302e-01,  2.5437e-01, -2.5304e-01, -6.9363e-01, -3.3891e-01,\n",
       "           1.0299e-02,  1.8091e-02, -3.8864e-01,  1.0920e-01,  5.7217e-02,\n",
       "          -2.6965e-02, -5.7936e-01, -2.5618e-01, -5.4129e-03,  4.7410e-02,\n",
       "           3.7889e-01, -4.1326e-01,  5.3043e-02, -5.8918e-01,  3.3814e-01,\n",
       "           8.6381e-02,  8.8725e-01, -9.5259e-02, -4.3658e-01,  4.1371e-01,\n",
       "           6.8976e-01, -3.6204e-01,  2.3138e-01, -2.0207e-01,  1.7246e-01,\n",
       "          -2.2461e-01, -1.5982e-01,  4.5024e-01, -2.3929e-01,  5.3763e-01,\n",
       "           7.6361e-02,  3.0156e-01, -5.0755e-01, -3.4200e-02, -4.5787e-01,\n",
       "          -2.6191e-01, -1.0136e-01, -4.3868e-01, -2.1873e-01, -9.5628e-02,\n",
       "          -3.6356e-01, -2.6224e-01, -2.7189e-01,  1.6586e-02, -1.9311e-01,\n",
       "           3.5684e-01,  1.1473e+00,  3.5222e-02, -4.8536e-02,  3.1878e-01,\n",
       "           2.3119e-01, -2.1068e-01,  2.0884e-01, -7.6245e-02,  3.3463e-01,\n",
       "           2.9654e-01,  3.1331e-01, -4.5673e-01,  2.5522e-01, -8.5886e-02,\n",
       "           3.3382e-01, -8.9138e-01, -7.5951e-01,  1.8828e-02,  2.3505e-01,\n",
       "          -4.1163e-02,  7.9677e-02, -2.6933e-01,  6.0446e-01,  6.9945e-01,\n",
       "          -1.3167e+00, -6.0678e-02, -2.4393e-01,  1.2029e-01, -4.5133e-04,\n",
       "          -4.4085e-01, -3.7242e-01, -3.3575e-01, -2.8487e-01,  3.2123e-01,\n",
       "           7.5280e-01, -4.5114e-01,  1.6297e-02,  1.8465e-01, -1.1076e-01,\n",
       "           2.3514e-01, -5.0295e-03,  2.5102e-01,  4.3425e-01, -1.0414e-01,\n",
       "          -1.4909e-01,  4.2579e-01, -3.1190e-01, -4.2031e-01,  1.5833e-01,\n",
       "           7.0937e-01, -1.1853e-01,  1.3689e-01, -1.8297e-02, -2.6799e-01,\n",
       "          -7.1901e-02, -1.1732e-01, -1.9540e-01,  2.4573e-01,  3.1104e-01,\n",
       "           1.0016e-01, -3.2782e-01,  2.3126e-01,  2.1841e-01, -5.2982e-02,\n",
       "           5.0868e-02,  9.0378e-01,  7.3841e-01,  1.8846e-02, -7.4087e-02,\n",
       "           2.7607e-01,  2.4221e-01, -5.0505e-02,  5.1696e-01, -1.7273e-01,\n",
       "          -2.8848e-01,  1.9170e-01,  8.5894e-02, -2.9880e-01, -9.4752e-03,\n",
       "          -2.0048e-01,  1.8031e-01,  1.8487e-01, -1.6871e-01,  5.3942e-01,\n",
       "           3.3819e-01,  1.9990e-01, -4.1667e-01,  6.3733e-05,  2.5826e-01,\n",
       "          -2.8037e-01,  8.7311e-01,  1.6816e-01,  4.6258e-01, -3.0563e-01,\n",
       "           2.2482e-01, -6.0811e-01, -4.3120e-01,  1.4365e-01,  7.4790e-02,\n",
       "          -3.6079e-01,  1.8614e-01, -1.2540e-01, -1.5393e-01,  7.3765e-01,\n",
       "           5.8692e-01, -3.4385e-01, -5.9073e-01, -5.4749e-02,  3.2823e-02,\n",
       "          -2.9436e-02, -2.4520e-01,  7.4517e-02, -2.1959e-02,  1.6957e-01,\n",
       "          -7.0204e-01, -2.6400e-01,  1.2722e-01,  2.0311e-01, -4.2548e-01,\n",
       "           8.2082e-01,  7.0018e-02,  1.7358e-01,  7.3647e-01,  1.3118e-01,\n",
       "           1.7952e-01,  6.3050e-01,  6.9376e-02, -5.5670e-01,  1.2268e-01,\n",
       "           2.8105e-01, -1.1468e-01, -5.5720e-02, -6.3967e-01,  3.2919e-01,\n",
       "           8.0147e-02,  9.4187e-01,  7.0432e-01]], device='mps:0'),\n",
       " tensor([[-0.4654, -0.4115, -0.4370,  ...,  0.1625,  0.4338,  0.5908],\n",
       "         [-0.2933, -0.5248, -0.8307,  ...,  0.0862, -0.1052,  0.4124],\n",
       "         [-0.4405, -0.5345, -0.7140,  ...,  0.4177,  0.7024,  0.3268],\n",
       "         [-0.3587, -0.4049, -0.7857,  ..., -0.2691,  0.2327,  0.1502],\n",
       "         [-0.1968, -0.4631, -0.8415,  ...,  0.0922,  1.0152,  0.6667]],\n",
       "        device='mps:0'),\n",
       " tensor([[-5.6844e-01, -2.5393e-01, -5.0750e-01, -8.7244e-01, -3.2844e-01,\n",
       "          -3.6569e-01, -9.0136e-01,  7.7320e-01,  8.1617e-02,  4.9008e-01,\n",
       "           9.4442e-02,  1.7855e-01,  2.8632e-01, -1.8449e-01, -5.2880e-01,\n",
       "          -3.7395e-01,  1.3400e-01,  3.8131e-01,  3.5838e-01,  1.0985e-01,\n",
       "           5.8069e-01,  1.8646e-01, -1.6628e-01,  6.8736e-02, -6.3904e-01,\n",
       "           2.5407e-01,  2.1383e-01,  3.1362e-02,  1.9298e-01, -8.6934e-01,\n",
       "           1.5436e-01,  4.0132e-01,  1.4270e-01,  9.5617e-01, -2.6119e-01,\n",
       "          -1.3691e-01,  7.3087e-02, -4.0375e-01,  5.6589e-01,  4.6931e-01,\n",
       "           2.6016e-01, -4.5020e-01,  6.3035e-03,  1.8017e-01, -1.3233e-01,\n",
       "           7.1244e-02, -6.5158e-02, -6.6997e-01,  4.3002e-01, -6.2951e-01,\n",
       "          -4.0576e-01, -1.7096e-01,  6.4477e-01,  4.3165e-01, -9.5134e-01,\n",
       "          -5.1421e-01,  3.6650e-01,  3.7034e-01, -6.0353e-01,  2.5449e-01,\n",
       "          -5.6246e-02,  7.4130e-01,  3.2109e-01, -3.4573e-03,  4.6800e-01,\n",
       "           2.8329e-01, -2.3211e-01, -6.1568e-01, -1.7770e-01, -1.6986e-01,\n",
       "          -1.4760e-02, -1.5805e-01, -4.9675e-01, -1.8398e-01,  5.6503e-01,\n",
       "           1.4111e-01,  1.8994e-01,  6.2489e-01, -3.7451e-01,  1.5743e-01,\n",
       "           1.1599e-01,  1.8623e-01,  9.5090e-02,  1.9397e-02,  2.6867e-01,\n",
       "           3.0676e-01,  6.6873e-01,  1.7461e-01, -4.1899e-02,  7.6356e-03,\n",
       "           4.0632e-01, -1.3571e-01, -1.9968e-01, -9.0196e-01,  3.4882e-01,\n",
       "          -2.3752e-01,  2.9844e-01, -3.2413e-01,  2.0613e-01, -1.9357e-01,\n",
       "           2.7677e-01,  4.7619e-01, -3.1541e-01, -2.9064e-02, -3.1801e-01,\n",
       "          -7.2487e-01,  2.4777e-01,  1.3619e-01, -4.8521e-01,  2.4988e-01,\n",
       "           6.3011e-02, -2.1547e-01,  4.0180e-02, -1.5635e-01, -2.3985e-01,\n",
       "          -1.3972e-01,  2.8368e-01, -1.5412e-01, -6.8293e-01,  3.4060e-01,\n",
       "           1.7510e-01, -3.9404e-01, -3.4872e-01, -7.5550e-02,  2.1179e-01,\n",
       "          -3.4693e-01,  4.3310e-01,  5.8511e-02,  2.1442e-01,  7.8144e-01,\n",
       "          -2.3282e-03,  1.7204e-01, -1.7537e-01, -2.0757e-01, -3.1992e-01,\n",
       "           1.2913e-01,  1.1688e-01, -9.7004e-02,  7.2642e-02,  8.6313e-02,\n",
       "          -2.5738e-01, -2.4418e-01, -7.5146e-01, -5.0822e-01, -2.4911e-01,\n",
       "          -4.5583e-01,  2.1359e-01, -4.9226e-02, -4.0862e-02, -4.1612e-01,\n",
       "          -7.7786e-01, -5.3849e-01,  7.3452e-01,  2.2917e-02, -5.7738e-02,\n",
       "          -3.4818e-01,  1.8976e-01,  3.5786e-01, -6.8795e-02,  5.2628e-01,\n",
       "           4.0225e-01,  2.8456e-01,  1.1433e-01, -5.6339e-01, -1.9941e-01,\n",
       "           8.1964e-02, -2.7762e-01, -3.5982e-01,  1.1957e-01,  3.0120e-01,\n",
       "           7.2749e-02, -1.9488e-01,  1.1217e-01, -6.4809e-01, -1.6623e-01,\n",
       "          -6.0746e-01,  4.2744e-01,  2.4433e-01, -4.7229e-01, -1.0579e-01,\n",
       "          -3.3341e-01, -1.6541e+00,  2.9410e-01, -7.4445e-01, -2.2061e-01,\n",
       "          -9.1557e-01,  2.5267e-02,  2.7081e-01,  2.3996e-01,  1.7481e-01,\n",
       "           3.0564e-02,  1.9829e-01, -2.6402e-02,  3.0950e-01,  1.3367e-01,\n",
       "           2.1250e-01,  4.2880e-01, -3.5856e-02, -3.1493e-01,  1.7774e-01,\n",
       "          -2.9625e-01,  3.1158e-01, -2.8723e-02,  1.6269e-01,  3.5764e-01,\n",
       "           3.8199e-01, -6.8891e-02,  2.9418e-01,  1.9944e-01, -2.8504e-01,\n",
       "           7.7662e-02, -2.0543e-01, -3.6897e-01,  1.2753e-02, -1.4946e-01,\n",
       "           5.7360e-02, -4.4640e-01,  6.0071e-03,  2.8395e-01,  1.4853e-01,\n",
       "          -1.1382e-01,  2.8209e-01, -2.7212e-01, -4.6728e-01,  5.0678e-02,\n",
       "           5.0747e-01, -2.5028e-01,  4.1382e-01, -4.3106e-01,  1.1839e-01,\n",
       "          -2.8436e-01, -4.3520e-01,  5.3732e-01,  6.0607e-01,  2.4345e-01,\n",
       "          -1.8731e-01, -3.4659e-01,  3.4032e-01, -1.7321e-01, -3.5382e-01,\n",
       "          -2.7126e-01, -3.0751e-01, -6.6050e-02, -1.4941e-01,  6.6002e-01,\n",
       "           1.4670e-01, -3.1667e-02,  1.0202e+00,  6.9238e-01,  2.0488e-01,\n",
       "          -4.7313e-02, -1.3113e-01,  3.3900e-01,  2.8843e-01, -4.0147e-01,\n",
       "          -2.2174e-01, -4.5826e-02, -5.3387e-01,  5.6696e-01, -2.8629e-01,\n",
       "           3.1854e-01,  8.9703e-02,  9.8121e-02,  7.2426e-01,  4.0746e-01,\n",
       "          -5.5773e-01, -1.0385e-02, -3.8020e-02, -2.2596e-01, -8.8121e-02,\n",
       "           5.8970e-04,  2.4156e-01, -4.0000e-01, -2.9996e-01, -7.2638e-01,\n",
       "          -2.3492e-01, -2.1963e-01,  1.6584e-01,  3.0846e-01, -6.2015e-01,\n",
       "          -2.4234e-01,  3.6705e-01, -2.8687e-01,  2.0502e-01,  2.5558e-01,\n",
       "           4.2903e-01, -3.4778e-02, -9.3549e-02, -2.0382e-01, -7.2037e-02,\n",
       "          -7.4568e-02,  1.1676e-01, -2.9530e-01,  2.0986e-02,  5.1684e-03,\n",
       "          -1.1136e-01,  4.3742e-01,  3.4734e-01,  7.8739e-02, -6.1562e-01,\n",
       "           4.4871e-02, -1.5161e-01, -2.5925e-01, -2.2698e-01, -3.9915e-02,\n",
       "          -3.3285e-01,  3.5655e-01,  8.4436e-01, -4.9563e-02, -2.6343e-01,\n",
       "           1.9915e-01,  5.1512e-01, -1.3016e-01,  1.4426e-01, -2.5021e-02,\n",
       "          -9.0834e-01, -2.5524e-01, -1.9994e-01, -1.2356e-01, -4.6987e-01,\n",
       "           1.1757e-01,  6.0037e-01, -2.2855e-01, -3.6448e-01, -6.8986e-01,\n",
       "           4.8317e-01,  1.6688e-01,  3.3535e-01,  1.3286e-01, -2.9904e-01,\n",
       "          -4.7916e-02, -8.2334e-01,  1.2087e-01, -5.1199e-01, -4.3309e-01,\n",
       "           2.3617e-01,  2.8534e-01,  4.3335e-02,  3.0027e-01, -4.8130e-01,\n",
       "           6.8084e-01, -2.2633e-01, -1.7373e-01,  3.8647e-01, -3.8197e-01,\n",
       "          -2.4056e-01,  5.5692e-01, -1.3181e-01,  3.6477e-01,  5.5141e-02,\n",
       "           5.7557e-01, -2.1858e-01, -5.1305e-01, -1.2626e-01,  4.3394e-01,\n",
       "           2.2980e-01,  3.8016e-01, -2.2147e-01,  6.9901e-02,  2.4774e-01,\n",
       "          -3.1906e-01,  5.5489e-01, -3.6133e-01, -4.8930e-01,  1.1730e-01,\n",
       "           3.6468e-01,  1.4321e-01, -3.3744e-01, -1.9847e-01,  5.3661e-01,\n",
       "           6.1608e-02, -2.2447e-02,  3.8687e-01,  4.1707e-01,  7.5413e-02,\n",
       "          -4.1578e-01, -5.8536e-01,  1.2953e-01, -3.9358e-01, -2.0976e-01,\n",
       "           3.5414e-01, -2.8840e-01,  2.8654e-01, -3.9533e-01, -4.0960e-01,\n",
       "          -4.1113e-01,  1.3799e-01,  6.9650e-01,  2.6831e-01,  7.2333e-02,\n",
       "           2.6272e-01, -4.3981e-02, -2.9518e-01,  5.3852e-01, -1.2421e-01,\n",
       "          -3.4740e-01,  3.1586e-01, -2.1424e-01,  5.4590e-02,  2.9407e-01,\n",
       "          -2.2166e-01,  1.6948e-01,  2.2397e-01,  4.9950e-01, -3.4496e-01,\n",
       "          -2.7747e-01, -1.1121e-01, -2.1886e-01, -1.9180e-01,  1.6433e-01,\n",
       "           1.2031e-01,  5.6576e-01, -1.2635e-01, -7.3580e-02,  5.6376e-01,\n",
       "           1.4494e-02, -2.4303e-01,  1.4364e-01, -5.4605e-01, -4.0354e-01,\n",
       "           8.3474e-03,  1.2952e-01,  1.1459e-01,  3.5272e-01, -1.0831e+01,\n",
       "          -1.4639e-01,  7.7977e-01, -4.0763e-01, -9.5451e-03,  6.6180e-02,\n",
       "           1.1330e-01, -1.9354e-02,  2.2326e-01, -2.0652e-02, -6.5077e-01,\n",
       "          -4.5718e-01, -2.8242e-01,  1.1227e-01,  1.4893e-01, -3.7762e-01,\n",
       "           3.6570e-02, -7.7520e-01, -2.6453e-02, -5.8563e-01,  1.1856e-01,\n",
       "          -3.1038e-01,  1.1832e-01,  2.6041e-01, -4.3650e-01, -1.8852e-01,\n",
       "           1.1934e-01, -1.7524e-01,  9.0805e-02,  8.6730e-02, -1.5353e-01,\n",
       "          -1.3314e-01, -8.9663e-02,  1.7702e-01, -1.7022e-01, -7.7946e-01,\n",
       "          -4.5533e-02, -1.0732e-02,  5.5088e-01, -1.8145e-01, -7.5737e-01,\n",
       "          -5.5589e-01, -7.2691e-01,  4.4681e-01,  4.6576e-01,  8.4337e-01,\n",
       "           7.3694e-01,  7.8225e-01,  2.8821e-01,  1.1654e-01, -7.8781e-03,\n",
       "          -4.5251e-01,  7.6623e-02, -5.4234e-01, -2.0190e-01, -1.6733e-01,\n",
       "           3.1524e-01, -7.4778e-01,  5.4153e-01, -3.3615e-01, -2.3183e-01,\n",
       "           2.1240e-01,  3.8045e-02,  2.9980e-01, -1.5400e-01, -5.7548e-01,\n",
       "           6.9551e-01, -8.1846e-02,  8.7726e-02, -3.6636e-01,  1.2998e-01,\n",
       "          -1.1565e-01,  4.2252e-01,  4.9964e-01, -4.2858e-01,  5.0750e-01,\n",
       "          -2.9582e-02, -3.5795e-01,  1.5328e+00, -2.8105e-01,  3.4889e-01,\n",
       "          -3.9615e-02,  1.9863e-01, -5.1251e-01, -7.2759e-01,  5.4114e-02,\n",
       "           7.7376e-03, -5.2419e-02,  5.1108e-01,  3.0878e-01, -3.2255e-01,\n",
       "          -7.7520e-01,  1.9643e-01,  1.9483e-01,  4.1738e-01,  8.3253e-01,\n",
       "          -3.7144e-01, -2.2433e-01,  1.3851e-01,  4.9568e-02, -9.1221e-01,\n",
       "           1.5130e-03, -4.3752e-01, -3.9226e-01,  4.8852e-01,  4.2343e-02,\n",
       "           5.0267e-01,  1.8224e-01, -5.7171e-03,  5.0501e-01, -4.4502e-01,\n",
       "           3.0443e-01, -2.1046e-01,  1.4119e-01, -4.2433e-01,  1.0247e-01,\n",
       "           3.6354e-02,  1.8873e-01, -2.9864e-01, -4.8747e-01,  1.9256e-02,\n",
       "           5.4759e-03,  1.6032e-01,  4.2622e-01, -5.6251e-01,  5.0070e-01,\n",
       "          -4.8306e-01, -6.8047e-03,  1.4265e-01, -2.3568e-01,  7.4557e-01,\n",
       "          -1.8767e-01, -3.2433e-01, -1.5232e-01, -7.4690e-01, -5.0003e-01,\n",
       "          -5.5172e-01, -1.4172e-01,  1.9040e-01, -9.0813e-01, -1.5122e-02,\n",
       "           1.7257e-01,  3.3071e-01, -3.2582e-01,  2.3588e-01,  3.5528e-01,\n",
       "          -2.3571e-02,  5.9669e-01, -4.8350e-03,  2.2136e-01, -1.7667e-01,\n",
       "          -5.2245e-01,  5.5999e-03,  1.7072e-01,  1.7848e-02, -1.0474e+00,\n",
       "          -2.8334e-01, -8.6200e-02, -4.5816e-01,  3.3449e-01,  2.6597e-02,\n",
       "          -7.7005e-02, -1.1522e-01, -4.6150e-01,  4.5910e-01,  5.7866e-01,\n",
       "           2.6352e-01, -1.0573e-02, -2.1847e-01,  6.8300e-02,  4.7001e-01,\n",
       "           3.7492e-02, -2.0095e-01, -3.8934e-01, -3.0347e-01,  2.2958e-01,\n",
       "           3.3929e-01,  5.1629e-01, -4.2074e-01, -7.4347e-01,  2.6759e-01,\n",
       "           5.2226e-01,  1.6757e-01, -1.0236e-01, -2.4258e-01, -2.6495e-01,\n",
       "          -2.6086e-01, -2.0432e-01,  4.2878e-01, -2.3748e-01,  1.3720e-01,\n",
       "           1.2240e-01,  6.0109e-01,  3.4140e-01,  4.1248e-01, -7.8067e-01,\n",
       "          -2.4939e-01, -2.0628e-02, -9.6719e-01,  1.4958e-01,  1.0040e-01,\n",
       "          -9.1728e-02,  4.3231e-01, -1.2503e-01, -1.5137e-01,  2.4529e-01,\n",
       "           5.3670e-01,  6.4405e-01,  1.3705e-01,  2.8725e-01,  8.6653e-01,\n",
       "           4.5630e-01,  7.6329e-02,  5.2303e-02,  3.7178e-01,  5.7313e-02,\n",
       "           2.7836e-01, -8.3007e-02, -3.8735e-01, -1.3449e-01,  1.5554e-01,\n",
       "           5.8080e-01, -1.1659e+00, -4.9298e-01, -2.5056e-01,  3.4759e-01,\n",
       "          -4.3638e-01,  1.2396e-01, -5.5119e-02,  3.2727e-01,  7.0494e-01,\n",
       "          -1.2319e+00, -4.9664e-01,  4.3808e-01,  2.5547e-01,  6.9514e-03,\n",
       "          -3.5297e-01,  1.8096e-01, -5.0733e-01, -2.1436e-01,  6.9160e-01,\n",
       "           3.2753e-01, -8.7155e-01,  3.8160e-02,  5.9753e-01, -5.1593e-01,\n",
       "           1.8231e-01,  1.0520e-02,  1.2073e-01, -3.5920e-02,  2.1339e-01,\n",
       "          -1.0837e-01,  3.6255e-01, -5.6160e-01, -4.2000e-01, -4.6695e-02,\n",
       "           5.9002e-01,  9.8017e-02,  2.4896e-01, -1.9997e-01,  3.5892e-01,\n",
       "          -2.7904e-01, -3.0208e-01, -2.4714e-01,  2.8074e-02, -2.1321e-01,\n",
       "           2.8785e-01, -1.3190e-01, -1.8593e-01,  2.6220e-01, -1.4457e-01,\n",
       "           5.4373e-01,  4.6093e-01,  6.4338e-01,  9.0278e-02, -1.7838e-01,\n",
       "           2.4395e-01,  6.3504e-02, -3.0764e-01,  5.1212e-01,  1.4861e-02,\n",
       "          -4.7605e-01,  1.3018e-01,  6.3130e-02,  5.3125e-02, -1.4310e-01,\n",
       "          -1.8210e-01,  2.8353e-01,  4.7433e-03, -3.4418e-01,  2.0436e-01,\n",
       "           3.2605e-01,  2.2278e-01, -5.5433e-01,  2.7091e-01,  3.4276e-01,\n",
       "          -1.6297e-02,  1.6658e-02, -1.3971e-01,  1.2699e-01, -6.8731e-01,\n",
       "           2.1912e-01, -7.5659e-01, -2.1822e-01,  2.2174e-01, -1.2384e-01,\n",
       "          -4.9953e-01,  3.5551e-02, -1.5131e-01,  3.5066e-03,  8.9862e-01,\n",
       "           8.9985e-01, -1.2377e-01, -4.9015e-01, -1.2054e-01,  2.6347e-01,\n",
       "          -5.6200e-02, -2.3647e-01,  2.6229e-01, -5.1455e-02, -8.3190e-02,\n",
       "          -1.2290e+00, -2.8710e-01,  2.0861e-01,  7.9545e-02, -1.8979e-01,\n",
       "           1.0801e+00, -2.4190e-01,  1.2983e-01,  6.1443e-01,  2.0776e-01,\n",
       "           3.0221e-01,  4.1798e-01,  1.9241e-02, -3.1289e-01, -1.1601e-01,\n",
       "           1.7249e-01, -6.4133e-02, -1.7759e-01, -5.8779e-01,  2.5980e-01,\n",
       "           7.9414e-02,  4.5242e-01,  6.2170e-01]], device='mps:0'),\n",
       " tensor([[-0.4654, -0.4115, -0.4370,  ...,  0.1625,  0.4338,  0.5908],\n",
       "         [-0.1827, -0.4651, -0.5438,  ...,  0.3005,  0.2362,  0.8631],\n",
       "         [ 0.0202, -0.7650, -0.5593,  ...,  0.2759,  0.8834,  0.5347],\n",
       "         [-0.2427, -0.6267, -0.1893,  ...,  0.2335,  0.0749,  0.3544]],\n",
       "        device='mps:0'),\n",
       " tensor([[-0.4603, -0.2720, -0.3283,  ...,  0.0235,  0.5795,  0.3375],\n",
       "         [-0.2619, -0.4089, -0.6686,  ...,  0.2546,  0.4931,  0.4269]],\n",
       "        device='mps:0'),\n",
       " tensor([[-0.2258, -0.7099, -0.3778,  ...,  0.3292,  0.7059,  0.7494],\n",
       "         [-0.2137, -0.4179, -0.6477,  ...,  0.2712,  0.3571,  0.6939]],\n",
       "        device='mps:0'),\n",
       " tensor([[-3.3416e-01, -6.9727e-01, -5.7646e-01, -7.0791e-01, -3.2003e-01,\n",
       "          -2.2332e-01, -6.3648e-01,  2.5279e-01, -8.1671e-02,  6.1022e-01,\n",
       "           2.9752e-01, -2.0794e-01,  5.0398e-02, -4.2130e-01, -4.2355e-01,\n",
       "          -3.4549e-01, -7.1015e-02,  2.7223e-01,  5.8022e-01, -1.3977e-01,\n",
       "           8.0654e-01,  3.8043e-01,  1.5041e-03, -9.2914e-02, -8.3693e-01,\n",
       "           2.3907e-01, -1.0462e-01, -2.2861e-02,  5.6454e-01, -1.1010e+00,\n",
       "          -1.4101e-01,  7.5896e-01, -8.1642e-02,  8.8141e-01,  5.3614e-02,\n",
       "          -1.4256e-01,  2.8339e-01, -6.5535e-02,  6.2216e-01, -3.8454e-02,\n",
       "           2.6776e-01,  2.5552e-01,  3.6413e-01,  2.1785e-01,  4.6563e-01,\n",
       "           8.1336e-02,  3.2178e-01, -5.2900e-01,  4.3513e-01, -7.2181e-01,\n",
       "          -2.8531e-01, -4.3060e-01,  2.9796e-01,  3.5650e-01, -3.0342e-01,\n",
       "          -1.4488e-01,  2.8378e-01,  8.5246e-02, -2.0507e-01,  1.4092e-01,\n",
       "           3.0537e-01,  8.3522e-01, -1.8492e-02, -9.7093e-02,  4.8365e-02,\n",
       "           2.9346e-01, -6.7686e-01, -3.8111e-01, -4.0186e-01, -2.7309e-01,\n",
       "          -1.2424e-01,  6.4183e-01, -2.5160e-01, -4.2749e-01,  2.0041e-01,\n",
       "          -1.5505e-01,  1.6831e-01, -3.9593e-01, -3.6591e-01,  6.3421e-03,\n",
       "           9.7756e-02,  1.4769e-01, -4.6800e-02, -3.4931e-01, -2.5699e-01,\n",
       "          -5.7188e-02,  7.6372e-01, -1.5139e-01,  1.2975e-01,  3.9793e-01,\n",
       "           3.0579e-01,  4.7351e-02,  2.1868e-01, -4.6723e-01,  2.3223e-01,\n",
       "          -3.0701e-01,  5.3967e-01, -6.7540e-02, -6.3484e-02, -7.7312e-02,\n",
       "           4.0471e-01,  1.4186e-01,  5.9490e-03,  6.9628e-02, -2.8747e-01,\n",
       "          -3.3064e-01,  2.7671e-01, -2.9778e-02, -5.8288e-01, -4.9613e-02,\n",
       "          -3.7266e-02,  2.5325e-01,  1.2607e-01, -4.3328e-01,  1.6088e-01,\n",
       "          -3.6972e-03,  4.9533e-02, -1.4586e-01, -7.9959e-02,  6.3989e-01,\n",
       "           4.0008e-01, -2.6903e-01,  2.9788e-01,  2.2517e-01,  5.6478e-01,\n",
       "          -2.1257e-01,  6.6395e-01, -1.4601e-01,  9.8229e-02,  8.9385e-02,\n",
       "          -7.7865e-02, -1.3457e-01, -2.7860e-01,  1.6129e-01, -2.9300e-01,\n",
       "           6.9536e-02, -5.7229e-03, -4.4968e-01,  7.2778e-02, -4.3273e-02,\n",
       "          -4.7929e-01,  4.9804e-02, -7.8925e-01, -3.6298e-01,  1.9542e-01,\n",
       "          -1.6540e-01,  1.6606e-01,  3.6909e-02, -3.4127e-01, -2.4767e-01,\n",
       "          -6.5755e-01, -5.4225e-01,  6.6914e-01,  4.6571e-02, -1.2061e-02,\n",
       "          -6.0084e-02, -8.3698e-02,  2.7714e-02,  2.6671e-02,  2.6154e-01,\n",
       "           5.9385e-01, -3.9082e-01,  3.3943e-02, -3.8508e-01, -6.5618e-02,\n",
       "           4.2267e-02, -1.1221e-01, -3.2925e-01, -2.6622e-01,  2.1045e-02,\n",
       "           3.5531e-01,  8.9032e-02, -1.9064e-01, -5.6346e-01,  1.8388e-02,\n",
       "           7.5832e-02,  1.4607e-02,  4.7215e-01, -7.5680e-02, -3.9468e-01,\n",
       "          -1.1305e-01, -1.8361e+00,  1.8184e-02, -4.0680e-01, -2.1415e-01,\n",
       "          -6.4589e-01,  4.0815e-01,  3.7047e-01,  2.1306e-01,  1.3177e-01,\n",
       "           1.2887e-01, -6.3362e-02,  9.5247e-02,  2.1475e-01, -8.7943e-02,\n",
       "           1.1910e-01,  5.3328e-01,  1.9747e-01,  2.3356e-01, -4.3444e-02,\n",
       "          -2.5538e-01,  4.6095e-01, -1.3050e-01,  2.5399e-02, -9.5282e-02,\n",
       "           9.7905e-02, -8.3855e-01,  2.4692e-01,  3.4070e-01, -4.5538e-01,\n",
       "          -2.6543e-01, -8.0456e-02, -4.1242e-01, -2.6294e-01, -4.1798e-01,\n",
       "           3.7954e-01, -1.2222e-01, -1.1466e-01,  4.4401e-01,  3.4593e-01,\n",
       "          -1.6741e-01,  4.9526e-02, -9.2034e-02, -3.8711e-01, -2.2033e-01,\n",
       "           8.3358e-01, -2.0255e-01, -1.2642e-01,  1.0190e-02, -3.0244e-01,\n",
       "          -5.1496e-02, -6.9100e-01,  7.3005e-01,  5.8259e-01, -1.7861e-01,\n",
       "          -2.6669e-01, -3.2557e-01, -2.3180e-01, -5.4286e-02,  5.6958e-02,\n",
       "          -3.4574e-01, -4.6584e-01, -8.5665e-01,  1.4300e-01,  1.5465e-01,\n",
       "           6.2447e-02, -1.8214e-01,  8.5428e-01,  3.6791e-01,  4.5529e-01,\n",
       "          -9.2644e-02,  6.9782e-02,  2.1649e-01,  2.4662e-01, -2.4678e-01,\n",
       "           3.4445e-02,  2.3621e-01, -2.1180e-01,  2.4591e-01, -2.4951e-01,\n",
       "          -8.0105e-02,  1.7191e-01, -1.3770e-01,  3.7988e-01, -4.6587e-01,\n",
       "           5.0383e-01, -9.4810e-02, -8.7523e-02, -2.0155e-01,  2.6662e-01,\n",
       "          -5.0960e-01,  3.9407e-01, -6.8633e-01, -1.4630e-01, -6.5880e-01,\n",
       "          -5.6583e-01,  3.8285e-02, -1.7355e-01,  1.5545e-01, -9.7159e-01,\n",
       "          -7.0342e-02,  1.6345e-01, -7.7111e-02,  1.5956e-01, -3.3935e-01,\n",
       "           1.3909e-01, -2.0125e-01,  1.0753e-01, -8.7266e-02, -4.3928e-01,\n",
       "          -1.7614e-01,  5.4299e-01, -9.6258e-02,  1.9465e-01,  2.5821e-01,\n",
       "          -2.8345e-02,  1.5095e-01, -1.6725e-01, -2.6420e-01, -2.2211e-02,\n",
       "          -5.3552e-02, -6.5877e-02,  1.1502e-02, -3.8009e-01, -3.8337e-01,\n",
       "           3.2932e-01,  1.7712e-01,  4.1282e-01, -4.5017e-02, -1.6964e-01,\n",
       "          -1.9015e-02,  6.6452e-01, -2.3814e-02, -5.9642e-02, -2.1074e-01,\n",
       "          -1.3892e+00, -2.3279e-01,  1.5257e-01, -4.3709e-01, -1.5645e-02,\n",
       "           2.0286e-01, -6.7407e-03, -1.9971e-01,  2.8245e-01, -2.7601e-01,\n",
       "           3.1363e-01,  1.8291e-01, -4.2832e-01,  7.9204e-02, -1.9739e-01,\n",
       "          -1.6124e-01, -1.6680e-01,  4.5180e-02, -1.1840e-01, -2.2065e-02,\n",
       "           2.0411e-01,  3.1850e-01,  5.3347e-02,  4.7438e-02, -6.0210e-01,\n",
       "           8.3867e-01, -9.9290e-02, -1.0382e-01,  5.1250e-01, -5.8508e-01,\n",
       "           4.0905e-02,  5.3033e-01, -2.3868e-01,  2.4423e-01, -1.8466e-01,\n",
       "           2.7083e-01, -1.8191e-01, -6.0559e-01, -1.8177e-01,  6.2538e-01,\n",
       "           1.6368e-01,  5.5597e-01, -3.6544e-01,  1.3571e-01,  3.5721e-01,\n",
       "          -4.7461e-01,  3.7617e-01, -3.1509e-01, -1.7980e-01, -3.7426e-01,\n",
       "           2.8997e-01,  3.7829e-01, -3.2779e-01,  2.9836e-01,  1.3156e-01,\n",
       "           1.9039e-01, -5.7098e-01, -1.4059e-01,  1.2181e-01,  1.5129e-01,\n",
       "          -2.7706e-02, -4.9354e-01,  1.1438e-01, -5.3401e-01, -6.7150e-02,\n",
       "          -1.1640e-01,  1.3553e-02,  3.7189e-01, -2.7712e-01, -4.2219e-01,\n",
       "          -3.3274e-01,  2.2309e-01,  5.4677e-01,  4.5440e-01,  1.4475e-01,\n",
       "          -9.4219e-02,  6.5433e-02, -1.3120e-01,  5.4316e-01,  1.9376e-01,\n",
       "           1.8011e-02,  7.4663e-01, -4.5142e-01,  2.1670e-01,  1.1089e-01,\n",
       "          -2.6963e-01,  1.6481e-01,  1.4785e-02,  3.3309e-01, -1.0889e-01,\n",
       "          -3.9504e-01,  2.7386e-01,  4.8952e-01, -4.7059e-02,  1.1037e-01,\n",
       "          -2.7669e-02,  9.5783e-01, -2.8922e-01,  1.2826e-01,  3.6485e-01,\n",
       "           2.8460e-01, -3.0898e-02,  2.9775e-01, -8.8897e-02, -7.0588e-01,\n",
       "           4.9931e-01, -9.4940e-02,  5.7490e-02,  1.1930e-01, -1.1329e+01,\n",
       "          -5.1654e-03,  1.0678e+00, -8.0619e-01,  7.1447e-02,  2.0073e-01,\n",
       "          -9.5438e-02, -3.9149e-01,  3.0142e-01, -4.2470e-01, -3.7225e-01,\n",
       "           1.4928e-02,  2.4402e-01,  4.5280e-02, -2.3768e-01, -2.1228e-01,\n",
       "          -7.0876e-02, -8.6302e-01, -1.2778e-01, -1.6736e-01,  5.0317e-01,\n",
       "          -9.6622e-02,  3.4257e-01,  1.6937e-01,  1.2497e-01, -4.6997e-01,\n",
       "          -2.8431e-01, -4.0994e-01, -1.5720e-02,  1.8063e-02, -9.6911e-02,\n",
       "          -2.2088e-01, -1.7044e-01,  3.5959e-01, -5.2345e-02, -5.5205e-01,\n",
       "          -6.2798e-02, -3.7589e-01,  5.0238e-01,  9.5381e-02, -1.7516e-01,\n",
       "           9.7877e-02, -1.1091e-01,  2.7321e-01,  1.7216e-01,  3.8620e-01,\n",
       "           8.1849e-01,  7.1061e-01,  1.9120e-01, -1.9526e-01, -1.1342e-01,\n",
       "          -6.8034e-01,  3.5892e-01, -8.9220e-01, -3.7856e-01, -1.8180e-01,\n",
       "           6.3247e-01, -2.9908e-01,  9.9227e-02,  1.0175e-01, -2.9221e-01,\n",
       "           2.9762e-01, -7.4580e-02,  1.6456e-01, -5.2511e-01, -2.0663e-01,\n",
       "           4.5943e-01,  1.8093e-01,  4.1295e-02, -2.3527e-01,  6.3798e-02,\n",
       "          -2.8845e-01,  8.8068e-01,  1.9685e-01,  4.6274e-02,  1.4481e-01,\n",
       "          -1.0583e-01,  1.6446e-01,  9.2260e-01, -1.7359e-01, -1.3846e-02,\n",
       "           1.2565e-01,  2.9656e-01, -9.6384e-01, -6.7107e-01, -2.2712e-02,\n",
       "           1.4940e-01,  4.5978e-01,  3.1269e-01,  2.6045e-01, -2.6901e-01,\n",
       "          -2.2134e-01,  6.2198e-02, -3.7500e-03,  1.2866e-01,  8.7976e-01,\n",
       "          -4.2986e-01, -2.9442e-01, -2.1377e-01,  4.4853e-02, -4.7484e-01,\n",
       "          -8.4157e-02, -3.0696e-01, -1.1195e-01,  1.2318e-01,  1.7830e-01,\n",
       "           1.6851e-01,  2.4532e-01,  4.3014e-01,  2.7173e-01, -4.0702e-01,\n",
       "           1.3284e-01, -1.5886e-01,  5.4806e-01, -2.7478e-01, -5.7039e-01,\n",
       "          -3.0745e-02,  5.0063e-01, -3.8053e-01, -5.1804e-01, -2.8901e-01,\n",
       "          -1.4631e-02,  1.8121e-01, -9.7369e-02, -4.3523e-01,  3.6594e-01,\n",
       "          -7.7226e-01, -1.6635e-01, -4.0572e-01, -1.5395e-01,  4.0858e-01,\n",
       "           1.2633e-01,  4.4419e-02,  1.7341e-01, -2.6177e-01, -4.9739e-01,\n",
       "          -4.9609e-01,  1.0737e-01,  3.6556e-01, -2.8205e-01, -1.6193e-01,\n",
       "          -3.8768e-01,  6.6087e-02, -2.6272e-01,  2.5992e-01,  5.7026e-03,\n",
       "           4.4397e-01,  2.7045e-01,  1.5937e-02,  1.5734e-01, -1.7181e-01,\n",
       "          -3.8432e-01, -1.5107e-01, -1.3204e-01, -1.6862e-02, -4.2672e-01,\n",
       "           3.7380e-02,  2.6092e-01,  2.9589e-01,  1.4823e-01, -2.6617e-01,\n",
       "          -2.1999e-01,  6.1351e-02, -2.6521e-01,  4.2195e-01,  3.5426e-01,\n",
       "           7.4122e-02, -1.5521e-01, -2.3889e-01,  5.3976e-02,  3.9518e-01,\n",
       "          -4.4450e-02, -8.4947e-02, -1.4378e-01, -5.0739e-01,  1.0021e-01,\n",
       "           1.5478e-01,  9.0121e-01, -1.7055e-01, -4.8081e-01,  6.1324e-01,\n",
       "           3.6535e-01, -4.7710e-01,  1.4137e-01, -2.6730e-01,  1.4002e-01,\n",
       "          -2.6479e-01, -2.4814e-02,  3.1547e-01, -2.1668e-01,  3.3493e-01,\n",
       "           1.8902e-01,  2.5629e-01, -3.2813e-01,  2.3016e-01,  2.1392e-01,\n",
       "          -3.9472e-01,  7.3393e-02, -3.7806e-01, -4.0375e-01,  1.7865e-01,\n",
       "           2.9759e-01, -2.7988e-01, -3.4560e-02, -3.6555e-02,  3.6548e-01,\n",
       "           3.9596e-01,  9.7720e-01, -1.0669e-01, -2.4935e-01,  1.4688e-01,\n",
       "           9.2641e-03, -2.0435e-01,  1.0038e-01,  3.0518e-01,  4.8008e-01,\n",
       "           2.5744e-01,  5.6626e-01, -2.9816e-01,  1.8929e-01, -1.2615e-01,\n",
       "           2.8800e-01, -7.8113e-01, -7.1038e-01, -4.3246e-02,  4.9317e-01,\n",
       "          -6.4955e-02,  4.3512e-01, -3.5439e-01,  3.4361e-01,  4.7235e-01,\n",
       "          -1.0613e+00, -7.2334e-02, -1.5243e-02, -1.3543e-01, -1.6932e-01,\n",
       "          -7.6603e-01, -4.3640e-01, -2.6210e-01, -3.3106e-01,  2.5116e-01,\n",
       "           5.5224e-01, -7.3479e-01,  2.6770e-01,  2.4949e-01, -3.8968e-01,\n",
       "          -1.5880e-03,  5.6334e-02,  5.9973e-02,  4.9321e-01,  1.9318e-02,\n",
       "          -2.2182e-01,  2.4416e-01,  1.6588e-01, -5.7084e-01,  3.5182e-01,\n",
       "           2.9037e-01, -6.3325e-02,  4.0599e-02,  2.1032e-01, -3.2333e-01,\n",
       "          -7.0250e-02, -3.0276e-01, -6.5169e-02, -3.6681e-02,  2.9702e-01,\n",
       "           2.5571e-01, -2.5158e-01, -7.2709e-02,  4.9719e-01, -1.7977e-01,\n",
       "           2.9897e-01,  3.9468e-01,  8.2487e-01, -1.0458e-01, -2.5226e-01,\n",
       "           2.5694e-01,  3.5125e-01, -2.8052e-01,  4.7870e-01, -7.0324e-02,\n",
       "          -2.1443e-01,  3.7254e-01, -1.7366e-01, -4.2377e-01,  2.4793e-02,\n",
       "          -1.3480e-01,  1.9668e-01,  4.2370e-02, -4.7091e-01,  3.0873e-01,\n",
       "           1.4592e-01,  3.9139e-02, -7.8645e-01,  1.6378e-01,  3.3181e-01,\n",
       "          -3.9618e-01,  7.6909e-01,  1.5391e-01,  2.0412e-01,  5.8777e-02,\n",
       "          -2.4512e-04, -5.6668e-01, -3.4248e-01,  3.3739e-01,  2.6819e-02,\n",
       "          -4.2305e-01, -3.5357e-01, -7.9694e-02, -4.8796e-02,  8.4841e-01,\n",
       "           2.9501e-01, -1.5473e-01, -3.9374e-01, -2.7715e-01,  4.7428e-02,\n",
       "           2.0496e-01, -4.1135e-01,  5.5184e-01, -1.0125e-01,  2.2389e-01,\n",
       "          -9.3087e-01, -2.9296e-01, -1.1890e-02,  6.6954e-02, -2.6450e-01,\n",
       "           4.7821e-01, -3.0364e-01,  5.0749e-01,  6.9926e-01,  3.7564e-01,\n",
       "           3.6739e-01,  3.4187e-01,  2.2520e-01, -2.9601e-01,  1.9862e-02,\n",
       "           4.8287e-01, -6.3450e-01, -2.2454e-01, -6.1671e-01,  5.5564e-01,\n",
       "           4.8649e-02,  7.4311e-01,  7.0701e-01]], device='mps:0'),\n",
       " tensor([[-2.6266e-01, -5.4108e-01, -8.1349e-01, -6.3133e-01, -1.3691e-01,\n",
       "          -6.6972e-01, -8.4270e-01,  4.7538e-01, -2.3973e-04,  4.2006e-01,\n",
       "           1.1549e-01,  2.5543e-01,  2.0258e-01, -4.0047e-01, -4.5308e-01,\n",
       "          -3.9355e-01,  2.8448e-01,  3.5454e-01,  4.7794e-01, -6.1003e-02,\n",
       "           8.1927e-01,  2.5216e-01, -7.7887e-02, -2.1555e-01, -4.7194e-01,\n",
       "          -1.1792e-01,  9.8752e-02,  8.0323e-02,  4.8133e-01, -1.4314e+00,\n",
       "          -2.1931e-01,  6.2917e-01, -2.0070e-02,  8.7410e-01,  1.4616e-01,\n",
       "          -1.1058e-01,  3.0338e-01,  2.1086e-02,  2.0576e-01,  1.3709e-01,\n",
       "          -8.0848e-02,  3.1800e-04,  2.9667e-01, -3.0308e-02,  4.7123e-01,\n",
       "           5.7275e-02, -4.4132e-02, -5.2569e-01,  3.7299e-01, -4.0594e-01,\n",
       "          -4.3263e-01, -3.1174e-01,  1.7939e-01,  3.3655e-01, -9.5912e-01,\n",
       "          -5.6035e-01,  2.8891e-01,  1.6364e-01, -5.0055e-01,  4.5378e-01,\n",
       "           1.2062e-02,  5.5155e-01,  4.5668e-02, -3.8529e-02,  4.3315e-01,\n",
       "           3.4735e-01, -7.3735e-01, -3.5862e-02, -4.4048e-01, -4.4998e-01,\n",
       "          -1.5686e-01,  3.2822e-01, -2.0337e-01, -4.4006e-01,  3.7338e-01,\n",
       "          -2.3296e-01,  2.4539e-01, -7.2800e-02, -6.6876e-01,  7.2317e-03,\n",
       "           3.9612e-01,  4.1710e-02,  9.9750e-02, -2.3855e-01, -3.5976e-01,\n",
       "          -1.4741e-01,  2.8523e-01,  2.6427e-01,  1.9441e-01,  1.1246e-01,\n",
       "           4.1545e-01,  6.3592e-02,  1.8527e-01, -2.6916e-01,  1.4060e-01,\n",
       "          -2.3824e-01,  3.1896e-01,  7.8945e-02, -6.7607e-03, -8.9089e-02,\n",
       "           2.7249e-01,  1.9619e-01,  6.6743e-02, -7.3712e-02, -2.3709e-01,\n",
       "          -6.6636e-01,  9.0584e-02, -4.3192e-02, -5.1725e-01,  1.9354e-02,\n",
       "          -1.7433e-01,  5.7784e-02,  6.0883e-02, -7.2445e-01, -6.1994e-01,\n",
       "           2.9755e-01, -2.4140e-01, -3.8502e-01, -2.2379e-01,  3.9178e-01,\n",
       "           2.0167e-01, -4.1831e-02, -1.8917e-01, -4.0440e-02,  6.1577e-01,\n",
       "          -1.6221e-01,  6.5707e-01,  6.7521e-02,  3.6104e-01,  7.5984e-01,\n",
       "          -2.6919e-01,  2.7653e-02, -2.0968e-01, -9.8825e-02, -1.8291e-01,\n",
       "          -2.4164e-03,  9.6738e-02, -6.1700e-01, -1.2445e-01,  8.8345e-02,\n",
       "          -5.0101e-01,  2.0597e-01, -5.3363e-01, -2.0894e-01,  1.7455e-01,\n",
       "          -1.3779e-01, -3.1420e-01,  3.6680e-01, -2.0097e-01, -2.4866e-01,\n",
       "          -6.1398e-01, -4.7640e-01,  7.7195e-01,  9.1083e-02, -1.4410e-01,\n",
       "          -3.8867e-01, -1.7988e-01,  6.9716e-02, -4.6820e-02,  4.2661e-01,\n",
       "           9.2341e-01,  5.5426e-02,  2.7130e-01, -1.2137e-01, -2.0268e-01,\n",
       "           3.0668e-01, -7.1288e-02, -3.5608e-01,  4.1286e-02, -1.3529e-01,\n",
       "           3.3866e-01, -7.5766e-02, -7.4250e-02, -1.5851e-01, -5.8585e-02,\n",
       "          -3.2458e-01,  2.2228e-01,  4.8018e-01,  8.9704e-02, -3.2805e-01,\n",
       "          -1.3924e-01, -1.4480e+00,  3.5369e-01, -8.1850e-01, -4.2228e-03,\n",
       "          -5.8735e-01,  4.0716e-01,  4.3214e-01, -1.6383e-02,  8.4175e-02,\n",
       "           2.6755e-01,  1.1050e-01,  1.6471e-01,  6.4072e-01, -7.9764e-02,\n",
       "           3.7119e-01,  6.1627e-01, -1.5190e-01,  3.2005e-01,  1.0265e-01,\n",
       "          -2.7971e-01,  4.6909e-01, -3.7925e-01,  3.2783e-01,  3.3879e-01,\n",
       "           2.1467e-01, -1.5858e-01,  1.1533e-02,  2.0114e-01, -2.7845e-01,\n",
       "          -8.3067e-02, -2.4663e-01, -6.1683e-01, -4.3876e-01, -3.0918e-01,\n",
       "           3.8139e-01, -3.1745e-01, -2.4879e-02,  1.7365e-01,  3.0088e-01,\n",
       "          -1.7550e-01,  7.2577e-02, -2.4029e-01, -4.0278e-01, -3.3865e-01,\n",
       "           6.4802e-01, -4.2306e-01,  4.1414e-01,  5.3097e-02, -1.7779e-01,\n",
       "           1.9341e-02, -8.4001e-01,  5.1474e-01,  4.6433e-01,  9.5946e-02,\n",
       "          -3.4156e-01, -3.1894e-01,  2.3039e-02, -4.0131e-02, -1.3870e-01,\n",
       "          -1.7899e-01, -4.4067e-01, -4.8943e-01, -4.0787e-02,  2.1680e-01,\n",
       "           9.7272e-02, -3.8971e-03,  1.1201e+00,  4.6043e-01,  4.9510e-01,\n",
       "          -1.9400e-01, -2.3233e-01,  2.3110e-01,  2.4638e-01, -2.3311e-01,\n",
       "          -1.3648e-01, -7.6656e-03, -4.6024e-01,  2.9849e-01, -2.3769e-01,\n",
       "          -2.9935e-01, -5.0190e-02,  1.3620e-01,  4.6303e-01, -3.2260e-01,\n",
       "           6.0590e-02, -2.5849e-01, -7.0711e-02, -1.8277e-01,  3.0643e-01,\n",
       "          -3.8842e-01,  3.7714e-01, -4.9497e-01, -6.1728e-01, -7.4062e-01,\n",
       "          -4.9049e-01, -4.9687e-02,  3.1133e-02,  6.5406e-01, -9.4043e-01,\n",
       "          -8.5121e-02,  2.9875e-02, -1.5860e-05,  3.2640e-01,  2.0598e-02,\n",
       "           1.9344e-01,  9.8396e-03, -3.8719e-02, -2.6977e-01, -3.9000e-01,\n",
       "          -1.7995e-01,  4.0397e-01, -3.8856e-01,  3.5986e-01,  3.8431e-01,\n",
       "          -1.9441e-01,  8.8651e-02,  8.8448e-02,  5.7892e-03,  1.3247e-01,\n",
       "          -3.4669e-01, -4.6997e-01, -1.6902e-01, -5.1910e-01, -3.3561e-01,\n",
       "          -2.3128e-02,  3.3870e-01,  2.9096e-01,  3.2224e-01,  4.1324e-02,\n",
       "          -2.0409e-01,  8.4974e-01, -5.9498e-03,  1.9569e-01, -1.9319e-01,\n",
       "          -1.2605e+00, -1.5192e-01, -4.0283e-02, -2.5726e-01, -3.6661e-02,\n",
       "           7.2585e-02, -2.5324e-02, -2.3821e-01, -1.0040e-01, -2.0412e-01,\n",
       "           2.2830e-01, -6.2941e-02, -5.2502e-02,  6.0973e-02,  2.7919e-01,\n",
       "          -1.7512e-01, -2.7459e-01, -8.6703e-03, -6.8207e-02,  1.1732e-01,\n",
       "           3.1499e-01,  1.5453e-01,  5.4270e-02,  1.6290e-01, -2.8927e-01,\n",
       "           6.3013e-01, -4.9613e-01, -8.5185e-02,  4.6296e-01, -4.6783e-01,\n",
       "          -3.4128e-01,  4.6028e-01, -3.4331e-01,  3.7714e-02, -3.9159e-01,\n",
       "           4.2445e-01, -6.9309e-01, -4.3274e-01, -2.9472e-02,  7.1936e-01,\n",
       "           4.9719e-02,  7.4154e-01, -2.6045e-01,  3.4829e-01,  4.6567e-02,\n",
       "          -5.0740e-01,  5.2275e-01, -2.3929e-01, -2.6422e-01, -2.2041e-01,\n",
       "           3.6547e-01,  2.9924e-01, -3.0585e-01,  1.3073e-01, -9.0399e-02,\n",
       "           2.3945e-01, -2.5490e-01,  1.7569e-01,  3.7818e-01,  6.0855e-01,\n",
       "          -2.8042e-02, -3.7584e-01,  3.2132e-01, -7.6642e-01, -3.5367e-01,\n",
       "           2.0462e-01, -7.8488e-03,  4.0151e-02,  1.3967e-01, -1.7044e-01,\n",
       "          -2.4971e-01,  2.3696e-01,  3.2449e-01,  6.6861e-01,  4.8278e-03,\n",
       "           9.1353e-02, -8.4993e-02, -1.5133e-01,  1.1525e-01, -3.3006e-01,\n",
       "          -6.1404e-02,  4.0034e-01, -3.0652e-01,  5.3568e-02, -6.7189e-02,\n",
       "          -4.5364e-01,  2.4825e-01,  1.7958e-01,  7.2498e-01, -1.8527e-01,\n",
       "          -4.9071e-01, -3.5878e-03,  4.0266e-01, -1.4547e-01, -5.7418e-03,\n",
       "          -3.1384e-01,  9.6976e-01, -4.3947e-01,  2.3603e-01,  9.6493e-01,\n",
       "          -1.8364e-01, -1.5896e-01,  2.5034e-01, -3.4917e-01, -4.2272e-01,\n",
       "           8.0153e-01,  2.8592e-01, -4.1408e-02,  2.2964e-01, -1.1119e+01,\n",
       "          -3.1632e-02,  8.5982e-01, -6.3289e-01,  6.9987e-02,  7.9408e-02,\n",
       "          -7.0727e-02, -4.0079e-01,  3.3154e-01, -6.5682e-01, -3.0413e-01,\n",
       "          -2.5787e-01,  1.9566e-02,  8.3151e-02, -1.4098e-01, -2.8644e-01,\n",
       "          -1.8328e-01, -7.5159e-01,  6.7131e-02, -7.2784e-02,  6.3706e-01,\n",
       "           8.4613e-02,  3.1610e-01,  3.5203e-01, -1.6415e-01, -2.3529e-01,\n",
       "          -2.3920e-01, -2.3091e-02,  6.6573e-03, -2.0843e-02, -5.9309e-02,\n",
       "          -3.2556e-01, -1.4055e-02,  5.8857e-02, -8.3926e-02, -5.7391e-01,\n",
       "          -3.0660e-01, -1.9300e-01,  5.9924e-01,  2.7687e-01, -5.2731e-01,\n",
       "          -3.7101e-02, -2.5517e-01,  6.7512e-02,  3.9749e-01,  7.8229e-01,\n",
       "           1.0401e+00,  3.0492e-01,  3.4704e-01,  1.8271e-01,  4.9310e-02,\n",
       "          -5.9561e-01,  3.0047e-01, -7.8531e-01, -1.9251e-01, -1.8024e-01,\n",
       "           4.0143e-01, -3.8326e-01,  4.2605e-02, -4.2017e-02, -1.6702e-01,\n",
       "           9.5769e-02,  1.8224e-01,  1.4971e-02, -2.4391e-01, -5.1034e-01,\n",
       "           1.0222e+00,  1.4660e-01,  5.4033e-02, -5.9968e-01, -2.0842e-01,\n",
       "          -4.9312e-01,  4.8233e-01,  3.8102e-01,  1.5234e-01,  7.4784e-03,\n",
       "          -3.7727e-02, -3.1177e-02,  1.6945e+00, -3.0040e-02,  2.5148e-02,\n",
       "          -5.2535e-02,  3.9059e-01, -5.8486e-01, -6.1357e-01, -1.9905e-01,\n",
       "           2.3633e-01,  1.2815e-01,  9.8986e-02,  6.8309e-01, -1.5201e-01,\n",
       "          -7.2948e-01,  3.0545e-01,  2.5178e-01,  1.4797e-01,  5.6483e-01,\n",
       "          -3.6281e-01, -1.9165e-01,  5.4949e-02,  5.9189e-02, -3.8826e-01,\n",
       "           8.0543e-02, -2.5124e-01, -2.3085e-01,  1.9600e-01,  2.0101e-01,\n",
       "           3.3471e-01,  3.0737e-02, -5.6907e-02,  4.6887e-01, -3.4037e-01,\n",
       "           2.8030e-01, -4.4923e-01,  2.6829e-01, -3.6593e-01, -1.6619e-01,\n",
       "           1.6334e-01,  9.3059e-01, -2.7350e-01, -8.1986e-01, -5.0347e-01,\n",
       "          -2.5060e-02,  3.0715e-01,  6.2480e-02, -6.9077e-01,  9.0640e-02,\n",
       "          -8.4632e-01,  9.0698e-05, -5.1730e-02,  1.6585e-01,  8.4758e-01,\n",
       "           2.2172e-01,  6.1354e-02,  2.0120e-01, -6.1163e-01, -1.7876e-01,\n",
       "          -1.9748e-01,  8.5680e-02,  3.8414e-01, -4.6688e-01, -5.9069e-02,\n",
       "          -3.8034e-02,  5.7777e-02,  2.4308e-02,  1.1385e-01,  3.0996e-01,\n",
       "           1.0060e-01,  4.6304e-01, -3.3380e-02,  2.2286e-01, -4.6061e-01,\n",
       "          -6.3214e-01,  9.3734e-03,  1.1418e-01,  1.5477e-01, -7.6915e-01,\n",
       "          -1.3800e-01,  2.1285e-01, -1.3191e-01,  2.1905e-01, -1.3937e-01,\n",
       "          -4.3563e-01,  1.0023e-01, -1.2129e-01,  1.3601e-01,  1.2601e-01,\n",
       "           1.2220e-01, -5.6001e-01,  9.3421e-02,  8.0909e-02,  2.6042e-01,\n",
       "           1.8314e-01,  1.5844e-01, -5.2466e-01, -4.5615e-01,  3.4491e-01,\n",
       "          -1.7724e-02,  1.0122e+00, -6.3845e-01, -4.5599e-01,  5.3464e-01,\n",
       "          -7.9927e-02,  7.0500e-02, -1.1043e-01, -4.8671e-02,  3.2686e-01,\n",
       "          -4.0959e-01,  3.2096e-01,  1.7294e-01, -2.6699e-01,  4.0043e-02,\n",
       "           3.8272e-01,  1.6933e-01,  1.9958e-02,  7.4485e-02,  4.8453e-02,\n",
       "          -4.3933e-01,  1.6867e-01, -8.6632e-01, -2.6911e-01,  1.7733e-01,\n",
       "          -3.0427e-01,  2.5305e-01,  3.6216e-02, -1.4142e-01,  2.8383e-01,\n",
       "           2.4257e-01,  9.8175e-01,  1.1746e-01, -6.0173e-01,  3.9082e-01,\n",
       "           2.3449e-01, -5.1544e-01,  1.8939e-02,  3.0417e-01,  3.6703e-01,\n",
       "           3.0896e-01,  6.9871e-02,  9.8028e-02, -9.1676e-02,  1.5331e-01,\n",
       "           3.4382e-01, -8.3916e-01, -5.3233e-01,  1.0523e-01,  3.8168e-01,\n",
       "          -2.9748e-01,  2.0571e-01, -3.0806e-01,  1.9518e-01,  1.5614e-01,\n",
       "          -1.2869e+00, -3.7614e-01, -1.5263e-01,  2.5917e-01,  7.5901e-02,\n",
       "          -5.3863e-01, -8.9483e-02, -1.6612e-02, -1.8652e-01,  4.0135e-01,\n",
       "           5.1967e-01, -5.4207e-01,  3.2619e-01,  4.2189e-01, -5.1376e-01,\n",
       "           1.8822e-01,  1.7014e-01,  4.6579e-01, -1.3816e-03,  2.7426e-02,\n",
       "           1.2628e-01,  2.6945e-01, -4.2500e-01, -5.1276e-01,  2.9745e-01,\n",
       "           3.5692e-01,  8.9223e-02,  2.0428e-01, -8.0270e-03,  7.9689e-01,\n",
       "          -3.4010e-01, -4.6450e-01, -2.4113e-01,  1.6855e-02, -3.1894e-02,\n",
       "           3.1592e-01, -1.2352e-02,  5.1729e-02,  6.0497e-01, -3.6260e-01,\n",
       "           3.9817e-01,  4.9929e-01,  4.5733e-01, -2.6211e-01, -1.2517e-01,\n",
       "          -8.1964e-02,  3.4306e-01, -2.8631e-01,  3.2069e-01, -2.5268e-01,\n",
       "          -2.5276e-01,  5.0323e-01,  6.5257e-02, -2.8358e-01,  8.3093e-02,\n",
       "          -2.2783e-01,  2.0762e-02,  1.5728e-01, -4.2777e-01,  1.6062e-01,\n",
       "           2.0126e-02,  5.6257e-02, -7.6353e-01, -5.2135e-02,  1.4875e-01,\n",
       "          -1.1973e-01,  8.3027e-01,  1.0108e-01,  4.0690e-01, -6.0928e-02,\n",
       "           1.1068e-01, -6.2276e-01, -3.5595e-01,  1.2104e-01, -2.0912e-01,\n",
       "          -2.8147e-01, -2.0001e-01, -3.0687e-02,  5.3147e-01,  9.1463e-01,\n",
       "           4.3715e-01,  3.4076e-02, -7.4412e-01, -4.7798e-01,  1.1029e-01,\n",
       "          -9.9453e-02, -3.0691e-01,  9.6144e-02,  2.0223e-01,  1.2044e-01,\n",
       "          -9.2531e-01, -4.5191e-01, -4.2049e-01, -3.0757e-02, -2.8077e-02,\n",
       "           9.3585e-01, -1.6013e-01,  3.4739e-01,  5.3994e-01,  1.4898e-01,\n",
       "           5.1590e-01,  3.7034e-01,  1.2405e-01, -6.8195e-01, -2.3094e-02,\n",
       "          -1.0124e-02, -3.1715e-01,  6.9124e-02, -5.4183e-01,  3.6074e-01,\n",
       "           9.9143e-02,  7.8874e-01,  4.6557e-01]], device='mps:0'),\n",
       " tensor([[-0.0153, -0.4994, -0.5617,  ...,  0.2217,  0.2081,  0.4534],\n",
       "         [-0.2068, -0.4178, -0.3105,  ...,  0.2512,  0.9903,  0.5041],\n",
       "         [-0.1689, -0.4450, -0.6101,  ...,  0.4169,  0.1256,  0.6819],\n",
       "         [-0.2188, -0.4577, -0.5994,  ...,  0.3145,  0.5079,  0.5582],\n",
       "         [-0.0986, -0.6295, -0.5474,  ...,  0.1507,  0.8416,  0.6531]],\n",
       "        device='mps:0'),\n",
       " tensor([[-0.0986, -0.6295, -0.5474,  ...,  0.1507,  0.8416,  0.6531],\n",
       "         [-0.2176, -0.4739, -0.7415,  ...,  0.0314,  0.6988,  0.3850]],\n",
       "        device='mps:0'),\n",
       " tensor([[ 0.0185, -0.5409, -0.3428,  ...,  0.1157,  0.4021,  0.6327],\n",
       "         [-0.4569, -0.7393, -0.3966,  ...,  0.3974,  0.0024,  0.6986],\n",
       "         [-0.1088, -0.7064, -0.2639,  ...,  0.3692, -0.0780,  0.5435],\n",
       "         [-0.3292, -0.2590, -0.5016,  ..., -0.0148,  0.3509,  0.1607],\n",
       "         [-0.0548, -0.3751, -0.4766,  ...,  0.0464,  0.8423,  0.4111],\n",
       "         [-0.0717, -0.5610, -0.7517,  ...,  0.1111,  0.6377,  0.4663]],\n",
       "        device='mps:0'),\n",
       " tensor([[-6.1750e-01, -4.2829e-01, -2.3934e-01, -9.4116e-01, -3.4548e-01,\n",
       "          -6.5496e-02, -6.3874e-01,  9.7828e-01, -9.5892e-02,  4.0096e-01,\n",
       "           1.4919e-01, -5.1603e-03,  2.0980e-01,  1.3680e-01, -6.9506e-01,\n",
       "           1.6740e-01,  4.6094e-01,  2.1159e-01,  3.5951e-01, -4.4329e-02,\n",
       "           7.9832e-01,  3.4245e-01,  1.8019e-01, -2.5205e-01, -4.5876e-01,\n",
       "           5.8994e-01, -2.4628e-01, -9.2014e-02,  2.0509e-01, -8.1949e-01,\n",
       "          -1.3586e-01, -4.9757e-02, -3.6281e-01,  8.7602e-01,  1.1948e-01,\n",
       "          -2.7310e-01, -1.5276e-01, -2.7996e-01,  2.3494e-01, -6.3637e-02,\n",
       "           3.7814e-01,  2.6592e-02, -2.9347e-01,  3.1386e-01, -1.6421e-01,\n",
       "          -1.4057e-01,  1.4047e-02, -7.8429e-01,  1.3666e-02,  1.8955e-01,\n",
       "           5.8851e-02, -4.7205e-02,  1.9231e-01,  3.7030e-01, -1.0617e+00,\n",
       "          -2.6348e-01,  2.7603e-01,  2.9849e-01, -1.7932e-01, -5.7784e-02,\n",
       "          -1.9989e-01,  9.0201e-01,  1.7949e-01, -2.7822e-01,  1.7824e-01,\n",
       "          -4.4899e-01, -5.4601e-01, -6.0781e-01, -1.0682e-01, -1.1967e-01,\n",
       "          -1.0002e-01,  9.7528e-02,  1.4009e-01, -2.9385e-01,  2.3170e-01,\n",
       "           2.1813e-01,  4.7024e-02, -2.7652e-02, -4.3158e-01, -2.6546e-01,\n",
       "           3.6947e-01,  3.4425e-02, -1.4730e-02,  2.0326e-01, -5.8744e-01,\n",
       "          -2.0226e-01,  3.7959e-01,  1.5703e-01,  2.5914e-01,  2.2662e-03,\n",
       "           5.9967e-01,  1.2024e-02,  8.0339e-02, -4.5369e-01,  5.8427e-01,\n",
       "          -5.4955e-01,  1.0329e-01, -1.6007e-01,  4.9532e-02, -5.2088e-01,\n",
       "          -9.8735e-02,  2.4494e-01, -1.8085e-01,  1.6053e-01, -4.6592e-01,\n",
       "          -3.2983e-02,  2.4514e-01,  3.0468e-01, -4.9183e-01,  4.2685e-02,\n",
       "          -6.5134e-03, -2.9791e-01,  7.3694e-02, -6.6407e-01, -8.5795e-01,\n",
       "           5.8354e-01,  1.8428e-02, -4.5240e-01, -2.3930e-01,  7.4443e-01,\n",
       "           2.9738e-01, -2.3201e-01, -5.3072e-02, -3.5548e-01,  2.0259e-01,\n",
       "          -1.8518e-01,  2.5689e-01, -2.7018e-01,  3.8439e-01,  5.5026e-01,\n",
       "           3.1581e-01,  2.8489e-01, -1.8327e-01,  2.7637e-01,  1.4465e-01,\n",
       "           3.6947e-02, -6.6614e-02, -1.5929e-01,  1.5180e-01,  1.2041e-01,\n",
       "          -2.9580e-01, -3.2999e-01,  5.4746e-03,  1.1182e-01, -3.3279e-01,\n",
       "          -1.0202e-01,  1.8680e-01,  3.4808e-01, -1.4092e-01, -2.5403e-01,\n",
       "          -7.8984e-01, -4.2328e-01,  1.0217e-01, -1.8560e-01, -1.7135e-01,\n",
       "          -4.1911e-01,  3.8121e-01, -6.6660e-02,  4.7638e-01,  3.4189e-01,\n",
       "           6.9978e-01, -1.5311e-01,  5.9776e-02, -5.6412e-01,  3.5197e-01,\n",
       "           3.8270e-02, -6.1110e-01, -1.7247e-01,  1.5680e-01, -4.1813e-04,\n",
       "           6.8195e-01,  7.0979e-02,  4.0124e-01,  1.2070e-01, -3.2553e-01,\n",
       "           8.2089e-03,  1.9890e-01, -1.2864e-01, -1.4297e-01,  7.3766e-01,\n",
       "          -5.5188e-01, -2.3029e+00,  1.0721e-01, -3.5844e-01, -3.6529e-01,\n",
       "          -1.8135e-01,  3.7255e-01,  3.3772e-02,  4.9966e-01,  1.5486e-01,\n",
       "           3.7598e-02,  1.6729e-01, -2.8025e-02,  3.6851e-01, -8.4399e-02,\n",
       "           2.6751e-01,  3.3341e-01,  1.2988e-01, -2.3298e-01, -1.0956e-01,\n",
       "          -3.9227e-01,  2.4031e-01, -9.8916e-02,  8.6010e-01,  2.3260e-01,\n",
       "           2.0943e-01, -4.3657e-03, -4.7262e-02,  4.8964e-01,  4.3099e-01,\n",
       "          -6.6541e-02, -1.6585e-01, -4.9522e-01,  4.6269e-03, -4.1279e-02,\n",
       "          -1.2963e-02, -4.5611e-01,  1.3753e-01,  4.2841e-01, -1.2416e-01,\n",
       "          -1.8354e-01,  1.5525e-01, -4.7715e-02, -3.1517e-01, -2.6492e-02,\n",
       "           6.8453e-01,  1.9080e-01, -2.2994e-01,  1.2377e-01, -1.1896e-01,\n",
       "           3.7006e-02, -4.6975e-01,  7.0044e-01,  6.1147e-01,  1.5585e-01,\n",
       "           1.6214e-02, -6.9943e-01,  2.3699e-01, -3.1574e-01, -4.7647e-03,\n",
       "          -1.5027e-01, -1.5731e-01, -4.2986e-01, -5.7285e-02, -5.5854e-02,\n",
       "           4.0115e-01,  7.9943e-02,  4.6653e-01,  1.1837e+00,  2.7973e-01,\n",
       "           1.9184e-02,  5.6097e-01,  2.0801e-01,  4.2492e-01, -4.0180e-01,\n",
       "           3.3082e-01,  8.8825e-02, -6.3896e-01,  2.7188e-01, -4.1593e-01,\n",
       "           4.4963e-02, -4.1432e-02, -3.7926e-01,  1.4208e-01, -2.1276e-02,\n",
       "           3.4237e-01,  3.3517e-01,  1.5186e-01,  1.6838e-01,  2.9411e-01,\n",
       "          -3.1322e-01,  5.9715e-01, -6.0809e-02, -1.1127e-01, -6.3492e-01,\n",
       "          -7.6732e-01,  9.5206e-02,  4.4743e-01, -1.2880e-01, -6.1279e-01,\n",
       "           1.0462e-01, -2.3866e-01, -1.2257e-01, -3.5064e-02,  2.1191e-01,\n",
       "           2.7455e-01,  5.0687e-02, -9.5436e-02,  6.4791e-02, -4.6950e-01,\n",
       "          -4.9277e-01,  4.9140e-02, -1.5758e-01, -1.3063e-01,  5.3918e-01,\n",
       "          -7.9415e-02,  1.1575e-01,  5.7221e-01,  1.9799e-02, -4.5976e-01,\n",
       "          -2.5640e-01,  1.4802e-01,  2.8667e-01, -7.2082e-02,  2.1981e-01,\n",
       "           3.4119e-01,  2.6467e-02,  5.1649e-01,  1.4823e-01, -2.1046e-01,\n",
       "          -2.7239e-01,  5.1173e-01, -1.5755e-01,  6.1403e-02, -1.2527e-01,\n",
       "          -8.7207e-01, -5.6150e-02, -7.8413e-02, -1.6833e-01, -5.4440e-01,\n",
       "           2.6566e-01,  1.4351e-01,  1.0561e-01,  2.2350e-01, -1.5984e-01,\n",
       "           7.0123e-01,  6.5285e-01, -3.3100e-01, -1.6943e-01, -4.0104e-01,\n",
       "           9.3922e-02, -6.8374e-02, -2.9329e-01, -1.4009e-02,  9.5368e-03,\n",
       "          -2.2765e-01, -1.2516e-01, -1.0805e-01,  2.8921e-01, -1.8876e-01,\n",
       "           5.2862e-01,  1.2189e-01,  8.3649e-02,  4.3873e-01,  2.0612e-02,\n",
       "           1.3071e-01,  3.6515e-01, -4.2117e-01,  1.2870e-01,  4.0424e-01,\n",
       "           4.0578e-01, -1.0551e+00, -8.3291e-01, -1.4936e-01,  3.6111e-01,\n",
       "           2.2860e-01,  2.5189e-01,  1.6999e-02, -7.2215e-03,  8.9455e-02,\n",
       "          -2.3402e-01,  3.5135e-01, -6.6415e-02, -5.7412e-01,  3.0561e-01,\n",
       "           2.3567e-01,  5.3263e-01, -2.7377e-01, -5.0579e-01, -6.5233e-02,\n",
       "           2.4616e-01, -3.6381e-01,  1.1392e-01,  2.3480e-01,  4.5466e-01,\n",
       "          -3.6015e-01, -5.6164e-01, -7.0282e-02, -7.3986e-01, -1.3578e-01,\n",
       "           3.2483e-01, -2.1437e-01,  2.7178e-02, -2.4758e-01,  1.5231e-01,\n",
       "          -1.7925e-02, -1.8879e-01,  6.0757e-01,  2.0522e-01, -2.5855e-01,\n",
       "           1.1981e-01, -2.3892e-01, -1.9475e-01,  5.0091e-01, -5.0226e-01,\n",
       "          -3.0348e-01, -5.0960e-02, -1.5349e-01, -3.2418e-01,  3.2181e-01,\n",
       "          -4.1389e-01, -1.4861e-01,  1.6748e-01, -6.4736e-02, -1.3068e-01,\n",
       "          -3.5233e-01,  1.4268e-01, -1.2707e-01, -4.9883e-02, -1.7654e-02,\n",
       "          -1.3353e-01,  6.0900e-02,  1.3308e-01,  6.4538e-02,  1.4388e-01,\n",
       "           5.3696e-01, -1.1464e-01,  2.7582e-03, -2.5900e-01, -2.2469e-01,\n",
       "           5.2498e-01, -9.4973e-02, -3.3117e-01,  5.4928e-01, -1.1156e+01,\n",
       "           1.0386e-01,  7.2631e-01, -5.5808e-01,  5.5128e-02, -2.8539e-01,\n",
       "           2.1085e-01, -3.0009e-01,  7.9101e-02,  2.4967e-01, -2.8460e-01,\n",
       "          -1.2277e-01, -2.7364e-01, -2.3193e-01, -4.0872e-01, -3.9742e-01,\n",
       "          -4.7229e-01, -5.6789e-01,  1.1772e-01, -1.7663e-01, -1.0996e-01,\n",
       "          -5.7036e-01, -1.9379e-01,  2.9696e-01, -4.9559e-03, -5.1069e-01,\n",
       "           5.9541e-02, -2.2948e-01,  2.6399e-01,  4.6126e-02,  2.2045e-01,\n",
       "          -1.9949e-02,  1.4108e-01,  2.0669e-01, -4.4946e-02, -4.5100e-01,\n",
       "          -4.4853e-01,  2.2129e-01, -1.5955e-01, -2.2267e-01, -5.4017e-02,\n",
       "          -3.3838e-01, -2.7689e-01,  1.3447e-01,  3.7594e-01,  6.2596e-01,\n",
       "           1.1595e+00,  5.6208e-01,  2.9245e-01, -1.0625e-01, -8.5438e-02,\n",
       "          -1.0762e+00,  5.5900e-01, -3.7318e-01,  8.1787e-02, -2.5119e-01,\n",
       "           6.6201e-01, -4.9906e-01,  5.4685e-01, -4.0754e-01, -2.5500e-01,\n",
       "           3.7179e-01,  3.0194e-01,  7.1104e-03,  2.8239e-02, -2.7432e-01,\n",
       "           4.0257e-01, -2.6094e-01, -3.6915e-02, -3.7819e-01,  4.0081e-01,\n",
       "          -3.9472e-01,  5.0770e-01, -5.6534e-02,  1.2945e-01,  1.8567e-01,\n",
       "          -3.5001e-01,  1.6610e-01,  1.0337e+00,  1.7414e-01,  4.1071e-01,\n",
       "           1.7420e-01,  5.1160e-01,  5.3064e-02, -8.8958e-01,  1.3691e-01,\n",
       "           2.0502e-01,  2.0789e-01, -6.7096e-02,  5.7131e-01, -2.5517e-02,\n",
       "          -6.5181e-01, -2.3094e-01, -3.2169e-01,  4.3108e-01,  6.5352e-01,\n",
       "          -2.2597e-01, -4.4343e-01,  2.3894e-01, -7.9776e-03, -5.3395e-01,\n",
       "           9.0074e-02, -2.2320e-01, -5.2792e-01,  8.7275e-02, -5.8914e-01,\n",
       "           1.4985e-01, -1.2428e-01,  5.4238e-02,  6.6887e-01, -4.0159e-01,\n",
       "          -1.6113e-01,  3.6602e-02, -4.4532e-02, -4.1270e-01, -4.9520e-01,\n",
       "          -7.7083e-02,  4.1073e-01, -1.9776e-01, -2.7976e-01,  7.0636e-02,\n",
       "          -3.0673e-01,  2.2584e-01, -1.7377e-01, -6.4935e-01,  3.2879e-01,\n",
       "          -6.0415e-01,  2.0578e-01,  2.5360e-01, -2.2507e-01,  4.9017e-01,\n",
       "           1.1337e-01, -8.1817e-02,  1.9397e-01, -9.9369e-02, -7.8554e-01,\n",
       "          -3.2573e-01, -2.7299e-02,  3.2662e-01, -4.2844e-01,  1.5776e-01,\n",
       "           1.5022e-01, -4.5752e-02, -2.2574e-01,  2.7904e-01,  1.0174e-01,\n",
       "           3.7762e-01,  3.9846e-01, -5.5006e-01,  1.1727e-01, -2.9867e-01,\n",
       "          -2.9262e-02, -6.1174e-02,  1.7952e-01,  2.3654e-01, -7.0764e-01,\n",
       "          -1.5224e-01, -2.6434e-01, -3.0965e-01, -2.1650e-01, -2.5120e-01,\n",
       "           2.0680e-01,  1.4038e-01, -1.5897e-02, -1.1308e-01,  5.8077e-01,\n",
       "           4.0814e-02, -1.1167e-01, -4.5997e-01,  3.1443e-01,  4.3178e-01,\n",
       "           2.1794e-01,  7.5875e-02, -1.2830e-01, -5.5532e-01, -1.1922e-02,\n",
       "          -4.3644e-01,  9.1081e-01, -7.3906e-01, -7.1970e-01,  6.6086e-02,\n",
       "           1.4951e-02, -8.8259e-04, -2.3450e-01, -5.0265e-01,  8.1298e-02,\n",
       "          -2.1017e-01, -1.4891e-01,  6.1550e-03, -1.0607e-02,  1.6299e-01,\n",
       "           4.0742e-01,  3.6838e-01,  1.5951e-01, -3.3938e-01, -4.4081e-01,\n",
       "          -6.0244e-01,  1.4567e-02, -1.6795e-01,  2.9330e-01, -2.6752e-01,\n",
       "          -3.2661e-01,  1.6043e-02, -9.1892e-03, -2.2272e-01,  5.2876e-01,\n",
       "           1.2427e-01,  1.6396e+00, -3.7287e-01, -4.4551e-01,  2.9018e-01,\n",
       "           7.2306e-02, -2.7073e-02, -2.8130e-01,  2.5074e-01,  9.5447e-02,\n",
       "           4.0194e-02, -3.6304e-01,  1.5023e-02, -2.0558e-02,  3.2773e-01,\n",
       "           3.2056e-01, -1.0703e+00, -2.0238e-01, -1.7993e-01,  5.5664e-01,\n",
       "           3.2388e-02,  6.3185e-01, -2.2305e-01,  3.0577e-01,  4.6106e-01,\n",
       "          -8.6667e-01, -2.3000e-01, -1.6129e-01,  2.7295e-01, -1.0583e-01,\n",
       "          -9.8735e-02, -1.6178e-01, -3.7165e-02, -1.6936e-01,  3.9538e-01,\n",
       "           7.4751e-01, -7.8426e-01,  7.4081e-01,  7.0467e-01, -9.8234e-02,\n",
       "           5.6815e-01,  9.4187e-02,  3.2641e-01, -1.4659e-01,  2.2669e-02,\n",
       "           2.6200e-01,  1.8542e-01, -6.3165e-01, -2.3962e-01,  2.4473e-01,\n",
       "           4.0158e-01, -1.1943e-01,  1.2108e-01,  7.9292e-02,  3.0960e-02,\n",
       "          -3.7691e-01, -1.0145e-01, -2.4552e-01, -3.9904e-01,  1.1356e-01,\n",
       "           1.4603e-02, -1.5001e-01, -8.6997e-02,  4.7709e-01, -1.7580e-01,\n",
       "           7.5067e-01,  4.2910e-01,  3.7145e-01, -3.5103e-01,  8.9950e-04,\n",
       "          -5.7565e-02,  6.0671e-01, -3.1670e-01,  7.7385e-01,  6.3337e-02,\n",
       "          -2.3140e-01,  2.7251e-01,  6.0547e-03,  1.7829e-01,  2.9892e-01,\n",
       "          -2.3519e-01,  1.2702e-01, -3.0718e-01, -2.3711e-01,  2.8330e-01,\n",
       "          -1.8640e-01,  1.1542e-01, -7.2270e-01, -6.3602e-03,  3.9857e-01,\n",
       "          -3.2051e-01,  4.9873e-01, -2.3074e-01,  6.1026e-01, -4.3509e-01,\n",
       "           3.2669e-02, -4.7470e-01, -4.3275e-01, -2.8508e-01,  5.1854e-02,\n",
       "          -8.6027e-01, -1.9453e-01,  1.7684e-01,  3.7405e-02,  2.1707e-01,\n",
       "           7.5155e-01, -3.3794e-01, -2.4771e-01, -2.0386e-01,  4.0190e-02,\n",
       "           1.8299e-01, -8.2071e-02, -3.5586e-01,  2.1161e-01, -3.5607e-01,\n",
       "          -1.0520e+00, -2.3117e-01,  8.9990e-02,  3.8419e-01, -3.5853e-01,\n",
       "           1.1570e+00,  2.5132e-02, -2.0226e-01,  6.7253e-01,  2.0808e-01,\n",
       "           6.4157e-01,  6.0323e-01,  1.3628e-03, -4.1259e-01, -5.4109e-02,\n",
       "          -1.5877e-01, -3.2881e-01,  1.5757e-01, -1.8346e-01,  5.0556e-01,\n",
       "           5.4802e-01, -2.0806e-02,  5.4756e-01]], device='mps:0'),\n",
       " tensor([[-1.1097e-02, -5.0567e-01, -4.6699e-01, -6.4640e-01, -3.9556e-01,\n",
       "          -2.5266e-01, -5.5990e-01,  3.0801e-01, -1.2536e-01,  1.4189e-01,\n",
       "           2.9133e-01,  5.3879e-01, -3.6706e-01, -2.8859e-01, -4.8290e-01,\n",
       "          -5.3699e-02, -6.1741e-01, -1.6918e-02,  3.4690e-01,  3.0491e-01,\n",
       "           5.3718e-01,  1.8073e-01, -4.1129e-01,  2.9149e-01, -3.7959e-01,\n",
       "           6.6561e-02, -3.4985e-01, -3.1546e-01,  2.6210e-01,  2.8012e-01,\n",
       "           5.3618e-02,  7.7692e-01, -2.4307e-02,  1.0962e+00, -1.8413e-01,\n",
       "          -2.1778e-01,  1.5359e-01,  2.3320e-01,  8.6439e-01, -6.2204e-01,\n",
       "           5.6395e-01,  3.6489e-01,  2.7586e-02, -4.0063e-02,  2.7845e-01,\n",
       "           3.5451e-01,  1.4567e-01, -5.7913e-01, -2.7161e-01, -3.4037e-01,\n",
       "          -1.2979e-01, -1.5164e-01,  5.0058e-01, -2.3292e-01, -7.9546e-01,\n",
       "          -1.9253e-01,  1.1288e-02,  2.6982e-01,  1.1117e-01,  2.3591e-01,\n",
       "           2.8900e-02,  7.9425e-01, -1.9339e-01, -3.2966e-03,  2.0559e-01,\n",
       "           2.5502e-01, -7.6243e-01, -1.2873e-01, -7.2822e-01,  1.1548e-01,\n",
       "           2.1413e-01,  3.5162e-01,  1.0500e-01, -2.6816e-01, -1.9644e-01,\n",
       "           6.0243e-02,  1.6956e-01,  1.8610e-01, -2.0035e-01,  3.5967e-02,\n",
       "           2.1140e-01,  4.4140e-01,  4.5980e-01,  5.9922e-02,  1.6117e-01,\n",
       "           5.6960e-02,  2.0080e-01, -3.9869e-02, -3.1738e-02,  1.7672e-01,\n",
       "           1.1984e-01, -9.9577e-02, -3.7483e-01, -4.3956e-01, -9.5205e-02,\n",
       "          -3.0408e-01,  3.2250e-02, -2.9469e-01, -2.3954e-01, -3.3261e-01,\n",
       "           2.5982e-01,  9.2762e-02,  3.2750e-01, -9.3082e-02, -6.6514e-01,\n",
       "          -3.5248e-01,  4.0832e-01,  3.4586e-03, -2.8129e-01, -3.8448e-01,\n",
       "          -6.6713e-02, -4.9346e-01,  1.2765e-01, -4.4650e-01,  1.9893e-01,\n",
       "           1.0835e-01,  1.8686e-01, -3.2670e-01,  6.5172e-02,  8.8101e-02,\n",
       "           6.1068e-02, -5.0938e-01,  2.7421e-01,  1.8326e-01,  9.3046e-02,\n",
       "           2.2489e-01,  6.8015e-01, -1.6210e-01,  4.3454e-01,  3.5722e-01,\n",
       "           3.1169e-01,  1.6221e-01, -4.1686e-01, -5.0259e-01, -7.5302e-02,\n",
       "           1.8488e-02, -5.9908e-01, -5.7442e-01,  6.1723e-02, -2.2868e-01,\n",
       "           1.6627e-01, -7.3083e-02, -5.1768e-01, -5.4307e-01, -3.1494e-01,\n",
       "          -3.0772e-01, -2.0517e-02, -3.2393e-01, -4.8159e-01, -2.5398e-01,\n",
       "          -6.1867e-01, -7.2891e-01,  1.9443e-01,  1.5164e-01,  2.7714e-01,\n",
       "          -4.3163e-01,  5.4580e-01, -3.0612e-01, -4.8554e-01,  3.7642e-01,\n",
       "           4.3432e-01, -5.0330e-01,  2.9526e-01, -6.3499e-01,  5.1171e-02,\n",
       "           2.0480e-01,  2.5291e-02, -3.4147e-02,  4.2807e-01,  1.1727e-01,\n",
       "           3.9831e-01,  2.7152e-02,  1.2476e-01, -2.9187e-01,  1.5015e-01,\n",
       "          -4.5427e-02, -3.2147e-01, -1.1176e-01,  1.7279e-02, -5.2967e-02,\n",
       "           5.4936e-01, -1.7306e+00, -1.7852e-01,  7.0060e-02, -2.7460e-01,\n",
       "          -1.3173e-01,  2.1473e-01,  5.0407e-03,  3.2283e-01, -3.5674e-01,\n",
       "           8.3698e-02,  8.7128e-02,  8.4654e-02, -1.0725e-01, -5.0983e-02,\n",
       "           2.9101e-01,  7.5929e-01,  4.0793e-01,  9.7463e-02, -2.6657e-01,\n",
       "          -6.0645e-01,  4.1554e-01,  1.9081e-01,  2.0730e-01,  3.9000e-01,\n",
       "           6.7931e-02, -6.9343e-01,  2.9674e-01,  1.9859e-01,  3.7002e-01,\n",
       "           2.5546e-02, -5.8496e-02, -4.8962e-01, -2.4682e-01, -4.3705e-01,\n",
       "           2.7759e-01, -1.3436e-01, -2.5922e-01,  6.3173e-01,  1.9273e-01,\n",
       "          -5.3531e-02,  3.9653e-01,  1.8733e-01, -4.1041e-01,  4.1633e-01,\n",
       "           5.2814e-01,  7.9114e-02,  1.4573e-01, -2.8055e-01, -1.6673e-01,\n",
       "          -3.1532e-01, -4.4420e-01,  4.4703e-01,  8.3941e-01, -7.2084e-03,\n",
       "           5.5343e-01, -3.5757e-01,  5.1174e-01, -3.6424e-01, -6.7779e-03,\n",
       "          -4.6486e-02,  6.8700e-02, -3.3988e-01,  1.6284e-01,  1.5551e-01,\n",
       "           1.0340e-01,  1.9709e-01,  8.5425e-01,  4.6494e-01,  1.4684e-01,\n",
       "           4.1031e-01,  2.5417e-02,  3.2080e-01,  7.8684e-01, -3.0254e-01,\n",
       "           3.4990e-01, -7.1578e-03, -4.1842e-01,  5.6461e-01, -3.5139e-01,\n",
       "           1.4248e-01, -2.3020e-01,  2.2987e-01,  1.7866e-01, -2.7854e-01,\n",
       "           2.1426e-01,  6.5273e-01,  9.5860e-02, -3.9283e-01,  7.3417e-01,\n",
       "          -4.0214e-01,  7.1262e-01, -3.0138e-02, -2.6992e-02, -1.0120e-01,\n",
       "          -7.4577e-01, -6.8569e-01,  1.4294e-01,  3.0808e-01, -7.6332e-01,\n",
       "          -3.4295e-01,  1.2380e-01,  1.3255e-01,  3.5068e-01,  2.4156e-01,\n",
       "           4.6144e-01, -1.7016e-01, -4.7185e-02, -4.4024e-01, -4.6015e-01,\n",
       "           1.1917e-01,  7.3311e-02,  9.5311e-02, -9.8199e-02,  9.2587e-02,\n",
       "           6.1067e-01,  2.4297e-01,  2.6799e-01,  1.4946e-01, -3.4610e-01,\n",
       "          -2.7723e-02,  1.7135e-01,  3.7874e-01, -6.1900e-01, -2.1807e-01,\n",
       "           5.3543e-02,  8.3164e-02,  6.6826e-01, -2.1832e-01, -1.1546e-01,\n",
       "          -1.2450e-01,  5.9739e-01, -2.2108e-01,  6.8002e-02,  1.0269e-01,\n",
       "          -6.8586e-01,  2.7619e-01, -5.9939e-02, -8.5297e-02, -5.2861e-01,\n",
       "           2.2997e-01, -5.7787e-02, -1.6068e-01, -7.3269e-03, -7.7601e-02,\n",
       "           2.2454e-01,  1.3406e-01,  2.8044e-01, -5.6270e-01, -4.2092e-01,\n",
       "           1.1781e-01, -3.2365e-01, -6.6016e-02, -4.5861e-01,  3.6766e-01,\n",
       "           1.3890e-01,  4.3829e-01, -1.0837e-01, -2.8576e-02, -1.3190e-01,\n",
       "           5.2866e-01, -6.2780e-01,  1.5573e-02,  4.8525e-01, -3.5595e-01,\n",
       "          -1.8139e-01,  4.7644e-01, -2.8011e-01,  4.9664e-02,  2.2982e-01,\n",
       "           3.7738e-01, -1.1530e-01, -4.7105e-01, -5.5247e-02,  9.3761e-01,\n",
       "           4.8980e-01,  2.0874e-02, -6.3820e-01,  1.2560e-01,  4.2066e-01,\n",
       "          -3.4219e-01,  1.0777e-01, -3.7280e-01, -2.3519e-01,  8.6456e-02,\n",
       "           3.3260e-02,  5.0344e-01, -5.8998e-02,  1.9741e-01,  4.6778e-01,\n",
       "           1.7011e-01, -6.9331e-01, -3.0557e-01,  6.3124e-01,  5.4670e-01,\n",
       "          -1.2247e-01, -5.4832e-01,  1.1480e-01, -6.4563e-01,  3.1198e-02,\n",
       "          -3.8623e-03, -3.5755e-01,  2.9653e-01, -1.4962e-01, -2.5683e-01,\n",
       "          -2.0003e-01, -5.3748e-01,  4.4973e-03,  1.0291e-01, -6.9595e-02,\n",
       "           4.1738e-01, -7.9610e-03, -3.1154e-01,  1.0058e+00, -2.7292e-01,\n",
       "          -3.1356e-02,  7.4139e-01, -5.4234e-01,  1.0457e-02,  2.4642e-01,\n",
       "          -1.3433e-01, -1.2416e-02,  2.4323e-02,  1.0660e+00,  1.1515e-05,\n",
       "           3.6402e-02, -2.0346e-01,  4.2203e-01, -4.7659e-01,  1.9394e-01,\n",
       "          -4.9958e-01,  4.8310e-01, -6.9399e-02,  2.4253e-01, -8.2329e-01,\n",
       "           4.3257e-01, -3.0604e-01, -3.2040e-01, -6.3922e-01,  5.0359e-02,\n",
       "          -5.0711e-02,  6.6524e-02, -5.8707e-02,  6.0787e-01, -1.0953e+01,\n",
       "           1.0946e-01,  8.4108e-01, -3.2618e-02,  1.3653e-01,  4.3336e-01,\n",
       "           1.2158e-01, -4.6684e-01,  7.2509e-01, -2.3242e-01, -6.6092e-03,\n",
       "          -3.5861e-01, -1.2444e-01, -1.3819e-01, -6.3519e-01,  4.0758e-01,\n",
       "          -1.3255e-01, -5.9114e-01,  2.8540e-01, -1.6945e-01,  2.4978e-01,\n",
       "          -2.3651e-01,  1.5671e-01,  5.9352e-01, -6.1010e-01, -3.3943e-01,\n",
       "          -1.1958e-01, -2.9991e-01,  5.1723e-01,  4.0217e-01, -2.3672e-01,\n",
       "           2.9096e-01,  3.3529e-02,  2.9455e-01, -5.5694e-01, -4.8133e-01,\n",
       "           4.3369e-01, -3.9950e-01,  2.1957e-01, -9.9943e-01, -3.3265e-01,\n",
       "          -1.8082e-01, -5.0951e-01, -2.7870e-02,  5.3812e-01,  7.5524e-01,\n",
       "           1.2821e+00,  1.1534e+00,  9.8198e-02,  4.6362e-01, -1.0523e-01,\n",
       "          -6.6935e-01,  5.1619e-02, -5.6525e-01, -4.1224e-01,  8.7196e-02,\n",
       "           2.2385e-01, -5.4352e-01,  4.7854e-01, -5.9897e-01,  1.8912e-01,\n",
       "           1.4329e-01,  7.9852e-02,  1.7368e-01, -5.1897e-01, -5.5039e-01,\n",
       "          -1.3282e-01,  8.1201e-02,  8.6324e-02, -6.7881e-01,  1.1881e-01,\n",
       "           1.7006e-01,  6.8197e-01, -2.5397e-01, -4.4268e-01,  4.1137e-01,\n",
       "          -2.2746e-03,  9.3809e-02,  1.3158e+00, -2.6036e-01, -1.7413e-01,\n",
       "          -4.8428e-02, -2.3194e-01, -3.3484e-01, -3.9656e-01, -4.5613e-01,\n",
       "          -2.5020e-02,  3.6573e-01,  5.1948e-01,  2.6062e-01, -2.3329e-02,\n",
       "          -3.0054e-01, -4.1445e-02,  5.1977e-02,  3.3862e-01,  9.6772e-01,\n",
       "          -6.6887e-01,  2.6280e-01, -1.9148e-01,  9.2178e-02,  2.2223e-01,\n",
       "          -5.7106e-01, -1.6871e-02,  2.9434e-03,  1.3975e-02,  5.7281e-01,\n",
       "           1.9705e-01,  2.6453e-01,  3.1190e-02,  2.2859e-01, -4.1682e-01,\n",
       "          -4.4044e-01,  2.9125e-04, -1.3190e-01, -5.8294e-01, -4.1034e-01,\n",
       "          -1.3514e-01,  1.3712e-01, -2.8585e-01, -5.3585e-01, -2.5012e-01,\n",
       "           1.4192e-01,  2.5900e-03,  4.9952e-02, -3.3857e-01,  4.7379e-01,\n",
       "          -7.6918e-01, -2.1429e-01, -2.7578e-01, -3.8495e-03,  1.4633e-01,\n",
       "           2.5508e-01, -5.1518e-01, -2.7838e-01, -1.0123e-01, -9.2314e-01,\n",
       "          -8.4489e-01,  2.2927e-02,  7.4620e-01, -1.1960e-01,  1.1861e-01,\n",
       "           8.5428e-02, -3.1671e-03, -3.7708e-01,  2.5194e-01, -2.4644e-01,\n",
       "           2.8149e-01,  4.3265e-01, -4.9583e-01,  2.9773e-02, -1.2792e-01,\n",
       "          -1.8992e-02, -1.0887e-01, -8.8647e-02, -5.4131e-01, -6.1371e-01,\n",
       "          -4.1197e-01,  1.3736e-01, -2.2747e-01,  4.1516e-01,  6.8072e-02,\n",
       "           1.2582e-01, -2.0826e-01,  9.8224e-02,  8.2540e-01,  6.0197e-01,\n",
       "           2.3014e-01, -3.2920e-01, -7.7666e-01,  3.4140e-02,  4.1642e-02,\n",
       "          -1.6450e-01,  2.1317e-01, -1.6384e-01, -3.0657e-01, -1.0112e-01,\n",
       "           1.2620e-02,  1.0002e+00, -2.0790e-01, -1.9155e-01,  1.0507e-01,\n",
       "           7.4626e-01, -6.0834e-01, -5.7221e-01,  3.7612e-01,  5.0418e-02,\n",
       "          -5.6831e-01, -1.9450e-01,  4.0315e-01, -1.7111e-01,  8.9921e-02,\n",
       "           3.7572e-01,  1.7821e-01, -1.8612e-01,  2.2711e-01, -6.7291e-02,\n",
       "          -1.7001e-01,  3.2731e-01, -1.6533e-01, -7.7254e-02,  4.0384e-01,\n",
       "           9.9517e-02,  1.5584e-01, -2.5262e-01, -4.8444e-01, -2.7025e-01,\n",
       "           1.5439e-01,  1.1110e+00, -1.7989e-01, -3.2527e-01,  6.6881e-01,\n",
       "          -3.0956e-01,  3.8928e-01,  4.8478e-01, -2.2611e-01, -5.0807e-02,\n",
       "           5.8049e-01,  3.0288e-01, -7.3236e-01,  2.2417e-01,  3.1670e-01,\n",
       "          -1.7335e-01, -5.3095e-01, -3.2838e-01, -2.1330e-01, -3.5189e-02,\n",
       "          -3.6356e-01,  5.2132e-01,  1.2255e-01,  3.9253e-01, -1.0843e-01,\n",
       "          -8.7207e-01,  4.7185e-01,  2.6128e-01,  8.0250e-01, -6.2568e-03,\n",
       "          -4.4880e-01, -3.5541e-01, -3.5790e-01, -6.6453e-02,  3.6543e-01,\n",
       "           2.0116e-01, -1.0333e+00,  4.6071e-02,  4.0507e-01, -3.9812e-02,\n",
       "           4.7083e-01, -1.4254e-01,  1.9634e-01,  4.7891e-01, -4.3918e-02,\n",
       "          -2.1856e-01,  1.3161e-01, -3.3027e-02, -2.7190e-01,  4.2861e-01,\n",
       "           4.6755e-01,  1.6062e-02,  2.5454e-02,  6.4881e-02, -4.2427e-01,\n",
       "          -3.4980e-01,  7.7064e-02, -1.7966e-01, -3.5675e-01, -1.4295e-01,\n",
       "          -4.0726e-01, -5.1772e-01, -6.6164e-02, -1.1357e-01,  1.3233e-01,\n",
       "           3.1032e-01,  1.3036e-01,  3.8431e-01, -2.3708e-01, -1.2541e-01,\n",
       "          -2.7977e-01,  4.5653e-01, -2.7595e-01,  8.0808e-01,  2.0566e-01,\n",
       "          -1.4423e-01,  2.0720e-01,  2.1514e-01, -2.9046e-01,  1.3028e-01,\n",
       "          -2.7828e-01,  2.8810e-01, -5.7798e-01, -3.4110e-01, -5.9710e-02,\n",
       "          -7.7948e-02, -1.5049e-01, -2.8505e-01, -1.0179e-01,  1.0608e-01,\n",
       "          -9.9043e-02,  9.7868e-01, -4.1759e-01,  6.3833e-01, -1.0342e-01,\n",
       "          -1.1350e-01, -7.5089e-01,  3.9397e-01,  3.6376e-01, -1.4798e-01,\n",
       "          -5.6434e-01, -2.1653e-01, -8.9158e-02, -4.2284e-01,  8.8621e-01,\n",
       "           5.8157e-01, -7.4673e-01, -3.7246e-01, -1.5919e-01, -2.4113e-01,\n",
       "          -4.7389e-02,  6.9820e-02, -3.2710e-01, -1.3295e-02, -1.6580e-01,\n",
       "          -9.8715e-01,  2.7641e-01, -3.6473e-02,  4.9205e-01, -2.5466e-01,\n",
       "           7.6979e-01, -8.6017e-02, -1.2749e-01,  1.1690e+00,  1.1169e-01,\n",
       "          -6.1844e-02,  5.9531e-01, -1.6837e-01, -6.2973e-01, -7.0295e-02,\n",
       "           4.4239e-02, -3.1098e-01,  8.7670e-02, -4.7880e-01,  2.9353e-01,\n",
       "           2.3648e-01, -7.1226e-02,  4.3877e-01]], device='mps:0'),\n",
       " tensor([[-0.0253, -0.6578, -0.4489,  ...,  0.3289,  0.5162,  0.7200],\n",
       "         [-0.3830, -0.7319, -0.8671,  ...,  0.3136, -0.0203,  0.5986],\n",
       "         [-0.2804, -0.8977, -0.7813,  ...,  0.0441,  0.7521,  0.7393]],\n",
       "        device='mps:0'),\n",
       " tensor([[-3.4035e-01, -5.6578e-01, -5.3090e-01, -7.5276e-01, -2.5144e-01,\n",
       "          -3.4343e-01, -5.2487e-01,  1.6663e-01, -1.1688e-01,  5.0762e-01,\n",
       "           1.9987e-01,  6.1251e-01, -4.9478e-02, -3.3432e-01, -1.0654e+00,\n",
       "          -8.8404e-02,  2.6499e-01,  7.5388e-01,  3.9237e-01,  6.5717e-01,\n",
       "           2.3958e-01,  2.2084e-01,  1.4782e-02,  1.4073e-01,  3.8319e-02,\n",
       "           4.0178e-01, -3.6560e-01,  7.7484e-02, -1.0499e-01, -4.5536e-02,\n",
       "          -1.9359e-04,  5.1959e-01,  1.7765e-01,  8.6059e-01, -1.9302e-01,\n",
       "          -3.0958e-02,  9.0361e-02, -7.3582e-01,  1.0443e+00, -6.9503e-02,\n",
       "           6.3101e-01, -1.3886e-01,  2.5250e-01,  6.0425e-02,  8.6935e-02,\n",
       "           1.9825e-02,  2.2315e-01, -7.8389e-01,  5.1324e-01, -3.7818e-01,\n",
       "          -1.4449e-01, -6.7538e-02,  5.1606e-01,  3.2794e-01, -4.8012e-01,\n",
       "          -3.9583e-01,  8.6292e-03,  1.0557e-01, -3.5177e-01,  5.5570e-02,\n",
       "          -4.3717e-02,  4.2279e-01,  1.0264e-01, -4.4007e-02,  1.5166e-01,\n",
       "          -2.2157e-01, -5.6724e-01, -3.8446e-01, -1.4523e-01, -1.0370e-01,\n",
       "           8.6235e-02,  2.8146e-01,  2.9811e-01, -1.6570e-01,  2.7194e-01,\n",
       "          -2.3957e-01,  2.6117e-02,  8.9432e-02, -2.3959e-01, -9.3266e-02,\n",
       "          -1.2953e-01,  2.7113e-01,  1.1546e-01, -4.0593e-02, -1.3170e-02,\n",
       "          -1.6330e-01,  3.9874e-01,  1.1438e-01,  3.3605e-01, -1.1058e-01,\n",
       "          -6.1618e-02, -6.6280e-02, -4.6252e-01, -1.8351e-01, -5.8992e-02,\n",
       "           1.1537e-01,  3.1760e-01, -3.5405e-01,  2.8188e-02, -3.8261e-01,\n",
       "          -1.0269e-01,  8.6151e-02, -2.6252e-02,  4.5176e-01, -8.6401e-01,\n",
       "          -4.1487e-01,  2.0987e-01, -5.7543e-02, -4.6091e-01, -1.8139e-01,\n",
       "          -3.0471e-02, -1.1624e-01,  5.2729e-02, -3.7123e-01, -1.8220e-01,\n",
       "          -6.7321e-03,  2.0451e-01,  1.8665e-01, -4.5058e-01,  2.4403e-01,\n",
       "           2.0776e-01, -4.7065e-01,  5.8712e-02, -1.7573e-01,  2.3669e-01,\n",
       "           2.7916e-02,  4.6429e-01, -3.2882e-01,  2.3926e-01,  3.8941e-01,\n",
       "          -9.4267e-02, -7.4739e-04, -1.8279e-01, -2.1045e-01,  1.9474e-03,\n",
       "           4.3763e-01,  2.5178e-01, -2.1505e-01,  2.3669e-01,  2.6298e-01,\n",
       "          -1.2409e-02, -3.8348e-01, -3.1957e-01, -3.2298e-01,  9.3375e-03,\n",
       "          -3.8921e-01, -8.3865e-02, -4.7760e-01, -5.1996e-01, -3.8441e-01,\n",
       "          -7.9886e-01, -7.2398e-01,  5.2915e-01, -1.8013e-01,  3.9908e-01,\n",
       "          -1.5493e-01,  2.4422e-01,  1.7491e-01, -3.5473e-03,  3.5381e-01,\n",
       "           6.7555e-01, -2.9846e-01,  4.9166e-02, -4.4497e-01, -1.3679e-01,\n",
       "           7.7887e-02,  1.1471e-02, -3.1779e-01, -2.5515e-01,  2.3228e-01,\n",
       "           1.8063e-01, -3.9630e-01,  7.0047e-02, -4.0706e-01,  9.8065e-02,\n",
       "          -6.0968e-01,  2.0502e-01, -3.2778e-02, -2.7339e-01,  3.3450e-02,\n",
       "           2.7502e-01, -1.4363e+00, -1.2087e-01,  1.1078e-01, -3.5882e-02,\n",
       "          -5.1129e-01,  1.7086e-01,  1.0068e-01,  7.2272e-02, -3.9598e-02,\n",
       "          -1.0016e-01,  5.8159e-02,  1.9529e-01,  4.2517e-02, -2.2985e-02,\n",
       "           1.1300e-01,  6.7517e-01,  2.4066e-01,  1.2758e-01,  2.9138e-01,\n",
       "          -3.2940e-01, -3.2318e-03,  1.1949e-01, -2.5752e-01,  3.6157e-01,\n",
       "           1.9658e-01,  2.4081e-02, -1.4488e-01,  1.8762e-01,  1.4024e-01,\n",
       "           5.8009e-02, -1.1471e-01, -2.3157e-01, -1.2038e-01,  5.9121e-02,\n",
       "           1.5047e-01, -1.5241e-01, -2.0150e-01,  6.3963e-01, -2.7680e-01,\n",
       "           1.4022e-01,  1.5061e-01,  4.9398e-02, -1.7004e-01,  1.8029e-01,\n",
       "           2.5292e-01, -1.6766e-01,  9.1358e-03,  1.1414e-01, -4.3519e-01,\n",
       "           1.5061e-01, -5.0899e-02,  4.7298e-01,  3.6647e-01, -2.8928e-01,\n",
       "          -4.5480e-01, -3.7938e-01,  2.2636e-01, -1.1798e-01,  8.7402e-02,\n",
       "          -4.1434e-01, -4.3884e-01, -7.4628e-01,  1.5776e-01, -1.1115e-01,\n",
       "           4.1324e-01, -1.8392e-01,  6.2225e-01,  2.8351e-01,  1.3393e-01,\n",
       "           3.1059e-02, -1.9508e-01,  2.4722e-01,  3.0855e-01, -3.3637e-01,\n",
       "           3.2501e-01,  2.1148e-02, -7.7435e-01,  1.9922e-01, -2.8672e-01,\n",
       "          -2.8586e-01,  1.6385e-01, -2.0418e-01,  3.5952e-01,  4.0208e-01,\n",
       "           2.5869e-01,  2.5028e-01,  1.4028e-01, -7.5533e-02,  3.9186e-01,\n",
       "          -1.9137e-01,  2.8118e-01, -4.1343e-02, -9.1576e-02, -8.1202e-01,\n",
       "          -6.3466e-01, -2.1614e-01,  1.7120e-01,  6.3634e-02, -8.5756e-01,\n",
       "          -1.1887e-01,  3.6567e-01, -1.6631e-01, -2.9497e-01,  8.6795e-02,\n",
       "           1.4195e-01,  3.6342e-02,  1.6402e-01, -6.4003e-01, -4.7438e-01,\n",
       "           6.5405e-02, -7.7412e-02, -5.7368e-01,  1.0190e-01,  2.7372e-01,\n",
       "          -3.0178e-01,  4.5298e-02,  1.3001e-01, -1.1842e-01, -1.7052e-01,\n",
       "          -1.2753e-01,  4.5434e-01,  1.2510e-01, -3.9700e-01,  1.6360e-01,\n",
       "           3.5658e-01,  3.8144e-01,  7.3561e-01, -5.5972e-01, -2.3297e-01,\n",
       "           1.1152e-01,  8.2247e-01, -1.2187e-01, -5.5062e-01, -2.0507e-01,\n",
       "          -9.7149e-01, -1.8276e-01, -5.5021e-01, -2.1298e-01, -7.7718e-01,\n",
       "          -2.1080e-02,  4.9067e-01, -1.0112e-01,  6.4480e-02, -5.0809e-01,\n",
       "           3.6435e-01,  7.1099e-01, -9.0209e-02, -3.9561e-02, -2.2394e-01,\n",
       "           1.8721e-01, -5.1193e-01, -1.3563e-01, -6.2016e-01, -6.4809e-02,\n",
       "           3.1850e-01,  3.0235e-01, -5.1896e-02, -1.9796e-01, -3.5830e-01,\n",
       "           4.9979e-01, -1.9964e-01,  1.2311e-01,  5.9015e-01, -3.8843e-01,\n",
       "           3.0807e-02,  4.8126e-01, -2.2210e-01, -2.5184e-01,  3.3946e-01,\n",
       "           3.4671e-01, -2.1502e-01, -6.0047e-01,  1.7192e-02,  1.8847e-01,\n",
       "           7.5329e-01,  2.2723e-01, -3.5550e-01,  8.3557e-02,  3.1420e-01,\n",
       "          -3.5169e-01,  2.7870e-01,  1.6475e-01, -3.4172e-01, -4.3494e-01,\n",
       "           2.7170e-01, -8.6119e-04, -2.6942e-01,  2.0623e-01,  2.9215e-01,\n",
       "           3.5546e-01, -9.5321e-02, -1.0837e-01,  1.7652e-01,  4.2539e-01,\n",
       "          -9.3434e-02, -3.3007e-01,  5.8618e-01, -3.4974e-01, -1.3441e-01,\n",
       "           2.3638e-02, -7.2486e-02,  6.3076e-02, -2.6596e-02, -4.0269e-01,\n",
       "          -5.4227e-02, -1.6812e-01,  3.7903e-01, -2.4990e-01, -2.7984e-01,\n",
       "           3.0894e-01,  3.0171e-02, -1.5221e-01,  8.8874e-01,  4.3803e-02,\n",
       "          -3.5868e-01,  1.2511e-01, -2.9237e-01,  2.4500e-02,  7.6374e-02,\n",
       "          -3.3724e-01, -3.7200e-02,  9.3923e-02,  4.0965e-01,  8.1050e-02,\n",
       "          -3.7296e-01, -1.2336e-01,  8.8886e-02,  2.6187e-02,  4.3573e-02,\n",
       "           1.8342e-01,  3.5163e-01, -8.2277e-02,  2.2054e-01, -6.6681e-01,\n",
       "           8.8920e-02,  3.1084e-02, -9.0070e-02, -5.6741e-02, -1.0163e-01,\n",
       "           7.9476e-01,  2.5764e-01, -6.3570e-02,  5.6636e-01, -1.1506e+01,\n",
       "           3.6522e-01,  5.5993e-01, -3.3966e-01,  1.7118e-01,  9.5875e-02,\n",
       "           2.1232e-01, -2.6377e-01,  6.3218e-02, -1.6909e-01,  1.1839e-01,\n",
       "          -2.4837e-01, -9.0619e-02,  2.6592e-01, -3.2673e-01, -8.2203e-02,\n",
       "          -1.2013e-01, -9.7737e-01,  5.0457e-02, -1.3769e-01,  1.7939e-01,\n",
       "          -1.9278e-01,  8.2745e-02,  2.9583e-01, -1.3651e-01,  5.7948e-02,\n",
       "          -2.2092e-01, -2.0124e-02,  4.0679e-01,  5.4219e-01, -1.1953e-01,\n",
       "           2.5137e-01,  6.5059e-02,  1.7985e-01,  6.1856e-02, -4.7585e-01,\n",
       "           1.7269e-02, -1.7177e-01,  3.0077e-01, -7.0484e-01, -4.0175e-01,\n",
       "           1.6548e-01, -9.6894e-01,  2.1708e-01,  3.6382e-01,  5.0226e-01,\n",
       "           9.7341e-01,  6.1395e-01, -3.0295e-02, -2.5736e-01, -5.2451e-02,\n",
       "          -6.6667e-01,  2.0744e-01, -5.9393e-01,  4.1227e-01, -5.2273e-02,\n",
       "           5.2963e-01, -8.5103e-01,  2.6529e-01, -8.3731e-01, -4.0504e-01,\n",
       "           9.9237e-02,  2.6092e-01, -8.3415e-02, -3.7302e-01, -4.0908e-01,\n",
       "           3.6017e-01,  6.3237e-02,  2.6228e-01, -6.3206e-02,  7.6335e-03,\n",
       "           7.2718e-02,  1.8721e-01, -3.5828e-01, -2.0780e-02,  3.5896e-01,\n",
       "           1.0797e-02, -4.5610e-02,  1.6192e+00, -1.3096e-01, -1.0232e-02,\n",
       "          -3.9907e-02, -8.8227e-02, -4.6643e-01, -5.2284e-01,  7.9180e-03,\n",
       "           2.0233e-02, -1.9609e-02,  9.4443e-02,  5.5323e-01, -2.9618e-01,\n",
       "          -5.7405e-01, -1.0710e-02,  8.0596e-02,  4.9648e-01,  7.8468e-01,\n",
       "          -5.3073e-01,  1.6379e-01,  1.8744e-01,  7.8776e-02, -8.2043e-01,\n",
       "           1.4494e-01,  1.6950e-01, -3.6898e-01,  5.9706e-02, -3.8987e-02,\n",
       "          -6.2627e-02, -4.6275e-02, -4.2415e-02,  5.7119e-01, -6.0674e-01,\n",
       "           1.9846e-01,  5.9612e-02,  3.8458e-02, -8.8281e-02, -2.5802e-01,\n",
       "           4.1911e-01,  2.6636e-01, -3.5906e-01, -5.6402e-01, -2.7852e-01,\n",
       "           1.1672e-01,  1.2852e-01,  1.8005e-01, -3.7028e-01,  7.0468e-02,\n",
       "          -3.3543e-01,  4.4153e-02,  3.3103e-01, -3.8054e-01,  9.8031e-02,\n",
       "           2.5320e-01, -5.7682e-01,  4.2709e-01, -5.5540e-01, -5.1891e-01,\n",
       "          -6.5812e-01, -3.1448e-01,  5.4992e-01, -6.9092e-06,  9.3353e-02,\n",
       "           6.6568e-02,  9.3047e-02, -3.0846e-01,  3.2079e-01,  1.4715e-01,\n",
       "           1.4476e-01,  1.2021e-01, -1.9628e-03,  6.5290e-02, -2.5956e-02,\n",
       "          -7.3944e-02, -1.2483e-01, -3.3594e-01, -1.0236e-02, -4.1450e-01,\n",
       "          -4.0937e-01, -2.3794e-01, -2.8703e-01, -1.1012e-01,  2.6990e-01,\n",
       "           1.3753e-01,  4.5247e-02, -3.3213e-01,  3.0884e-01,  2.7883e-01,\n",
       "           2.2900e-02, -3.8634e-01, -7.9049e-02,  1.7271e-02, -1.2580e-02,\n",
       "           3.8376e-01,  1.3976e-01, -4.2743e-01, -2.5520e-01, -4.1502e-01,\n",
       "           6.0236e-02,  8.0850e-01, -6.9961e-02, -4.0933e-01, -5.8701e-02,\n",
       "           1.2925e-01,  1.9290e-01,  8.7157e-02,  1.0220e-01,  3.4755e-01,\n",
       "          -2.3057e-01, -5.9283e-01,  4.9020e-01, -2.9571e-01,  2.0514e-01,\n",
       "           7.2530e-01, -4.2598e-02, -4.7616e-02,  5.1576e-01,  5.4891e-02,\n",
       "          -1.7875e-01,  1.5591e-01, -3.5194e-01, -2.0683e-01,  4.3298e-01,\n",
       "          -6.6201e-02,  3.9363e-01, -2.3388e-01, -4.6556e-01, -1.7104e-01,\n",
       "           3.0307e-01,  1.0255e+00,  3.7737e-02, -5.8712e-02,  7.4443e-01,\n",
       "          -4.2174e-02,  6.9914e-01,  5.0021e-01,  4.0923e-01,  2.0395e-01,\n",
       "           3.4513e-01, -2.1156e-01, -1.1140e-01,  3.4569e-01,  2.3090e-01,\n",
       "           5.4860e-01, -6.9298e-01, -1.4204e-01,  1.3611e-02,  2.8917e-01,\n",
       "          -6.4001e-02,  4.8246e-01,  2.5629e-02,  4.4978e-01,  5.4525e-01,\n",
       "          -1.2797e+00, -2.6560e-02,  2.1129e-02,  2.1978e-01, -2.2149e-01,\n",
       "          -3.9128e-01, -2.2830e-01, -8.2216e-03, -1.6676e-01,  1.9265e-01,\n",
       "           4.8268e-01, -1.0132e+00, -6.8484e-02,  7.8457e-01,  3.5811e-01,\n",
       "           1.5953e-01, -4.1745e-01,  2.9902e-01,  2.4434e-01, -1.4429e-01,\n",
       "           4.9326e-01,  8.3072e-02, -4.8864e-01, -2.2220e-01,  6.8346e-02,\n",
       "           5.2640e-01,  8.4489e-02, -1.4156e-01, -1.8969e-02, -4.0898e-01,\n",
       "           1.1136e-01, -1.8055e-01, -7.3705e-02, -6.8588e-02,  1.8393e-01,\n",
       "           1.0852e-01, -4.5120e-01, -2.8170e-01,  4.2672e-01,  5.9401e-02,\n",
       "           3.1527e-01,  4.4118e-01,  6.0604e-01, -8.9284e-02, -5.0373e-01,\n",
       "          -1.1063e-02,  2.8675e-01,  1.1464e-01,  5.0543e-01, -5.1435e-01,\n",
       "          -3.5180e-01, -3.0241e-01,  1.1618e-01, -4.5416e-01, -1.0579e-01,\n",
       "          -2.0988e-01,  2.4023e-01, -2.0015e-01, -1.1685e-01,  3.7733e-01,\n",
       "           1.7843e-01,  9.7660e-02, -7.2775e-01,  5.3606e-01,  2.0987e-01,\n",
       "           7.3808e-02, -2.8772e-02,  2.4407e-01,  4.6953e-01, -1.2758e-01,\n",
       "           1.3366e-01, -8.7790e-01, -1.6247e-01, -3.0110e-01, -1.0884e-01,\n",
       "          -2.8862e-01, -1.2821e-01, -1.9165e-01, -1.9965e-01,  6.5761e-01,\n",
       "           8.5550e-01, -2.2824e-01, -6.8494e-01, -3.1371e-01,  1.8610e-01,\n",
       "           2.2765e-01,  1.5711e-01,  4.7329e-01,  3.8646e-01,  1.8263e-01,\n",
       "          -6.5470e-01, -2.3980e-02,  1.4965e-01,  3.7158e-01, -1.7319e-01,\n",
       "           8.7425e-01, -3.3272e-01, -2.0150e-01,  1.6799e-01,  5.1634e-03,\n",
       "           3.9245e-01,  4.5606e-01,  5.6969e-02, -2.8047e-01, -1.0086e-01,\n",
       "           4.3921e-02, -1.3820e-01,  1.3669e-01, -7.6370e-01,  1.4727e-01,\n",
       "          -1.7199e-01,  2.8743e-01,  3.7342e-01]], device='mps:0'),\n",
       " tensor([[-0.2188, -0.4577, -0.5994,  ...,  0.3145,  0.5079,  0.5582],\n",
       "         [-0.3342, -0.6973, -0.5765,  ...,  0.0486,  0.7431,  0.7070]],\n",
       "        device='mps:0'),\n",
       " tensor([[-0.1898, -0.4285, -0.4452,  ...,  0.1449,  0.6232,  0.6814],\n",
       "         [-0.1443, -0.6102, -0.5612,  ..., -0.1199,  0.5122,  0.1498]],\n",
       "        device='mps:0'),\n",
       " tensor([[-0.4215, -0.3219, -0.7371,  ...,  0.1094,  0.4889,  0.6566],\n",
       "         [-0.2457, -0.5121, -0.3791,  ...,  0.0229,  0.6675,  0.3177],\n",
       "         [-0.3328, -0.5342, -0.4353,  ...,  0.0206,  0.4514,  0.5724]],\n",
       "        device='mps:0'),\n",
       " tensor([[-6.2616e-02, -6.2647e-01, -7.0276e-01, -7.1558e-01, -2.8144e-01,\n",
       "          -1.6163e-01, -8.7709e-01,  3.6268e-01,  1.2576e-01,  5.9786e-01,\n",
       "           2.8059e-01,  2.7990e-01,  1.2964e-01, -2.9739e-01, -6.7620e-01,\n",
       "           1.2485e-01,  1.2045e-01,  2.6275e-01,  1.6394e-01,  1.0986e-01,\n",
       "           6.6733e-01,  3.4572e-01,  7.2301e-02, -9.9001e-02, -5.2645e-01,\n",
       "           3.5221e-01, -1.0247e-01, -1.2800e-01,  4.8585e-01, -1.2739e-01,\n",
       "           1.0805e-02,  2.6421e-01, -1.5857e-01,  1.1057e+00,  4.6431e-02,\n",
       "          -8.7014e-02,  1.4985e-01, -1.7339e-01,  3.6943e-01,  4.8486e-02,\n",
       "           5.1489e-01,  8.5748e-03,  1.1385e-01,  2.9544e-01,  5.9498e-02,\n",
       "           2.1140e-01,  3.3738e-01, -7.3412e-01,  2.4225e-01, -5.0904e-01,\n",
       "          -1.4421e-01, -1.2879e-01,  4.3693e-01,  3.4752e-01, -6.1938e-01,\n",
       "          -1.5226e-01,  1.7658e-01,  1.7700e-01, -3.8094e-01,  5.7837e-01,\n",
       "           7.7456e-03,  5.2039e-01, -1.1219e-01,  5.6928e-02, -6.0641e-02,\n",
       "           2.2301e-01, -1.8694e-01, -4.7731e-01, -2.0693e-01, -3.3899e-01,\n",
       "          -2.3704e-01,  2.3273e-01,  4.6058e-02, -2.0383e-01,  4.8662e-01,\n",
       "          -4.1023e-01,  2.5189e-01, -3.6104e-01, -7.5358e-02,  7.7211e-04,\n",
       "           2.7676e-01,  3.5111e-01, -1.5597e-01,  1.2574e-02,  4.3904e-02,\n",
       "           5.4012e-02, -3.1553e-03,  1.7637e-01, -6.6285e-02,  3.7809e-01,\n",
       "          -1.7252e-01,  1.1085e-01, -3.4724e-01, -1.4165e-01,  3.0672e-01,\n",
       "           6.2280e-02, -5.4002e-02, -1.8348e-03,  2.5848e-01,  1.8095e-01,\n",
       "           1.3316e-01, -8.1850e-02,  1.2541e-02, -4.9147e-02, -1.3866e-01,\n",
       "          -3.7121e-01,  3.0219e-01, -3.5727e-01, -5.3572e-01, -8.9114e-02,\n",
       "          -2.2678e-01, -2.0834e-01,  2.4095e-01, -1.2899e-01, -5.8584e-02,\n",
       "           5.2857e-02,  1.4616e-02, -6.9177e-02,  7.7893e-02,  3.1706e-01,\n",
       "           3.4116e-01, -2.6142e-01, -3.7110e-02, -4.1103e-02,  4.0389e-01,\n",
       "          -8.3567e-02,  5.0486e-01, -6.3763e-02,  1.9561e-01,  1.9710e-01,\n",
       "           2.1314e-01, -7.2560e-02, -3.5530e-01, -1.6748e-01, -9.2310e-02,\n",
       "           2.4388e-01,  1.5179e-01, -2.9361e-01,  3.6513e-01, -3.1711e-02,\n",
       "          -4.2278e-03, -1.2962e-01, -6.7690e-01, -2.7977e-01, -1.7267e-01,\n",
       "          -2.9240e-01,  3.2325e-02, -1.3392e-01, -6.0883e-01, -1.2325e-01,\n",
       "          -8.8410e-01, -6.5719e-01,  3.4764e-01, -3.7706e-01, -7.2839e-02,\n",
       "          -2.5295e-06,  2.7332e-01, -3.7283e-02, -1.3345e-01,  9.2422e-02,\n",
       "           8.6093e-01, -3.7257e-01, -1.0155e-01, -5.7483e-01,  8.7619e-02,\n",
       "           4.3489e-02, -1.6390e-01, -1.8396e-01, -4.6625e-02,  1.1536e-01,\n",
       "           2.9032e-01, -3.0347e-01,  1.5016e-01, -2.8636e-01,  1.1897e-01,\n",
       "          -5.2865e-01,  3.8697e-01,  2.4944e-01,  2.2011e-01, -1.6624e-01,\n",
       "           1.8996e-01, -2.0789e+00, -1.4910e-02, -7.1562e-01,  1.4990e-01,\n",
       "          -3.4002e-02,  1.6094e-01,  2.5705e-01,  3.1807e-01, -8.5558e-02,\n",
       "           2.4227e-02,  1.2994e-01,  6.0720e-02,  1.7780e-01,  3.0290e-02,\n",
       "           8.5993e-02,  4.2734e-01,  1.0144e-01, -6.2154e-02,  4.6363e-02,\n",
       "          -3.3884e-01,  4.3572e-01,  2.0554e-01,  2.4188e-01, -7.8520e-03,\n",
       "           1.0108e-01, -3.4169e-01,  1.3350e-01,  3.5792e-01, -1.6305e-01,\n",
       "          -1.9152e-01,  1.4795e-01, -6.1009e-01,  6.4222e-02, -1.3318e-01,\n",
       "           2.4521e-01, -1.2892e-01,  3.6794e-03,  6.0353e-01,  2.9885e-01,\n",
       "           2.0611e-01,  2.3138e-01,  2.0087e-01, -3.8862e-01, -2.7582e-02,\n",
       "           2.7418e-01, -3.1822e-01, -2.7486e-02, -5.2360e-01, -2.9816e-01,\n",
       "           2.9255e-01, -6.7586e-01,  4.0643e-01,  3.6271e-01,  1.2730e-02,\n",
       "          -5.1865e-03, -3.9674e-01,  1.8170e-01, -3.2388e-01,  3.8983e-01,\n",
       "          -2.2252e-01, -4.8237e-01, -6.0014e-01,  1.9894e-01,  4.8443e-01,\n",
       "           3.2823e-01,  6.0846e-02,  6.4689e-01,  4.0682e-01,  4.8346e-01,\n",
       "          -1.7029e-01, -9.3965e-03,  4.3252e-01,  2.7174e-01, -3.5879e-01,\n",
       "          -5.5175e-02,  1.9570e-01, -2.8277e-01,  2.7704e-01, -1.2009e-01,\n",
       "          -4.1986e-01, -1.2058e-01,  4.4856e-01,  4.7218e-01, -5.5729e-01,\n",
       "           9.6198e-02,  6.4694e-01, -3.4297e-01, -4.4990e-01, -5.2557e-03,\n",
       "          -5.0208e-01,  2.0353e-01,  8.8648e-02, -1.9668e-01, -5.6403e-01,\n",
       "          -7.0264e-01, -1.1072e-01, -5.0530e-01,  1.2782e-01, -7.0494e-01,\n",
       "          -3.3147e-01, -1.3909e-02, -8.1837e-02,  7.3904e-02,  1.7344e-01,\n",
       "           2.1771e-01, -7.1470e-02, -5.1346e-02, -1.2863e-01, -5.5667e-01,\n",
       "          -3.9013e-02,  1.6707e-01, -3.7411e-01,  3.8257e-01,  1.4180e-01,\n",
       "           6.2359e-02,  1.9231e-01,  2.7552e-01, -3.5967e-01, -4.2561e-02,\n",
       "          -1.0399e-01,  6.3338e-02, -3.7767e-01, -1.8284e-01,  1.6221e-01,\n",
       "           2.5527e-01,  4.3982e-02,  5.5676e-01, -9.8989e-03, -4.9037e-01,\n",
       "           9.7279e-02,  6.7626e-01, -3.0749e-01,  1.1219e-01, -1.6722e-01,\n",
       "          -8.1230e-01, -1.7303e-01, -1.4474e-01, -5.0151e-01, -3.7506e-01,\n",
       "           3.2481e-01,  4.3004e-02, -3.4700e-02, -1.9535e-02, -4.7990e-01,\n",
       "           2.0226e-01,  4.4602e-01,  2.3309e-01,  1.8308e-01, -3.7078e-01,\n",
       "          -1.6060e-01, -1.5167e-01, -4.5093e-01, -2.3121e-01, -2.6221e-02,\n",
       "           3.0377e-01,  1.7798e-01,  6.0163e-02, -1.0477e-01, -1.4368e-01,\n",
       "           6.2727e-01, -2.8161e-01,  1.6625e-02,  2.0776e-01, -3.7056e-01,\n",
       "          -3.4593e-01,  5.0320e-01,  6.7809e-02,  2.2294e-01,  4.6225e-02,\n",
       "           3.6316e-01, -5.6646e-01, -6.2602e-01, -1.3417e-01,  8.9179e-01,\n",
       "           1.5994e-01,  3.4170e-01, -3.8552e-01,  1.1696e-01,  4.9943e-01,\n",
       "          -4.2539e-01,  4.7704e-01, -3.5382e-01, -1.0500e-01, -3.2598e-01,\n",
       "           4.0893e-01,  3.3302e-01, -2.0704e-01,  3.7979e-01,  2.0070e-01,\n",
       "          -1.3786e-01, -5.4466e-02, -2.3674e-01,  3.6240e-01,  4.9872e-01,\n",
       "          -5.3611e-03, -3.7018e-01,  2.2186e-01, -1.8560e-01, -1.2012e-01,\n",
       "          -1.0254e-01, -3.2722e-03,  1.2326e-01, -6.5918e-01, -2.9480e-01,\n",
       "          -1.7014e-01, -5.3133e-03,  7.3499e-01,  2.2859e-01, -3.5801e-01,\n",
       "           1.3965e-01, -3.2208e-01, -3.9333e-01,  3.1746e-01,  1.2561e-01,\n",
       "          -3.7691e-01,  4.0955e-01, -4.8433e-01,  7.0593e-02, -1.1285e-02,\n",
       "          -1.2369e-01,  2.7728e-02,  1.1395e-01,  3.9460e-01,  2.5117e-01,\n",
       "          -2.2488e-01, -4.0668e-02,  1.2723e-01, -1.0980e-01,  2.0994e-01,\n",
       "          -1.1511e-04,  4.9680e-01, -2.1882e-01, -2.9162e-01,  9.7461e-02,\n",
       "           4.0597e-01, -3.5028e-01,  1.8268e-01, -1.7362e-01, -5.4194e-01,\n",
       "           6.3471e-01,  1.0150e-01,  2.5641e-01,  1.3090e-01, -1.1699e+01,\n",
       "           7.1592e-02,  7.3831e-01, -1.9325e-01,  2.8363e-01,  4.0517e-01,\n",
       "           1.1602e-02, -3.7668e-01,  4.4858e-01, -1.3840e-01, -3.4024e-01,\n",
       "          -4.2958e-01, -3.8199e-01,  1.4233e-01, -1.5751e-01, -2.0805e-01,\n",
       "           4.1685e-01, -7.9191e-01,  5.9132e-02, -2.8132e-01,  5.7196e-01,\n",
       "          -3.3759e-01, -1.2832e-01,  3.5796e-01, -1.5764e-01, -2.8902e-01,\n",
       "           2.3761e-01,  1.7273e-02, -1.6691e-01,  3.2209e-02, -1.6032e-02,\n",
       "           1.7587e-01, -3.3271e-01, -1.5027e-01, -1.2007e-01, -4.6988e-01,\n",
       "          -2.7005e-01, -3.0493e-01,  1.2092e-01,  7.8455e-02, -1.7796e-01,\n",
       "           7.7184e-02, -1.7859e-01,  2.2969e-01,  4.8338e-01,  6.2378e-01,\n",
       "           1.3198e+00,  9.0670e-01, -1.8994e-02,  1.9499e-02,  1.7116e-01,\n",
       "          -5.5958e-01,  2.4145e-01, -3.6729e-01,  2.8099e-01, -1.5925e-01,\n",
       "           2.4387e-01, -5.4304e-01,  2.4829e-01, -2.2367e-01,  1.0939e-01,\n",
       "           5.3557e-01, -4.0237e-02,  3.0196e-01, -3.0008e-01, -2.3120e-01,\n",
       "           1.2109e-01,  3.4968e-02,  6.0281e-02, -2.2445e-01, -7.0501e-03,\n",
       "          -7.1388e-03,  4.6860e-01, -3.9338e-02, -1.0576e+00,  2.4495e-01,\n",
       "           3.2278e-02,  1.2890e-01,  2.8982e-01,  1.8912e-01,  2.4288e-02,\n",
       "           1.3797e-03,  3.3766e-01, -5.5161e-01, -4.2301e-01,  2.3988e-01,\n",
       "          -7.5664e-02,  2.2611e-01,  4.2581e-01,  2.5577e-01, -1.9563e-01,\n",
       "          -3.2047e-01, -8.0159e-02,  1.1424e-01,  2.6139e-01,  5.9414e-01,\n",
       "          -7.6424e-01, -7.0030e-03,  1.6680e-01, -1.9423e-01, -1.0671e-01,\n",
       "          -3.2879e-04, -2.6335e-01, -3.5521e-01,  3.4951e-01, -5.5845e-01,\n",
       "           3.8535e-02, -2.3141e-01, -3.2059e-01,  3.6518e-01, -2.9619e-01,\n",
       "           1.8127e-01,  4.9069e-01, -1.3729e-02, -1.0096e-01, -3.7150e-01,\n",
       "           1.7551e-01,  2.6098e-01, -6.2737e-01, -2.1799e-01,  2.9550e-02,\n",
       "          -1.9310e-01,  5.3267e-01,  3.0115e-02, -3.1969e-01,  4.2068e-01,\n",
       "          -7.5378e-01,  2.5273e-01,  2.7860e-01,  9.8125e-02,  5.6108e-01,\n",
       "           1.0236e-01,  3.8592e-01,  1.4133e-01, -2.6726e-01, -5.6057e-01,\n",
       "          -9.4410e-01, -6.6615e-02,  3.3471e-01,  5.1896e-02,  1.2819e-01,\n",
       "          -2.6217e-01, -2.9069e-02, -1.2689e-01,  5.5164e-01, -6.5168e-02,\n",
       "           3.8720e-01,  1.2669e-01, -2.9272e-01, -1.6031e-03, -9.4970e-03,\n",
       "          -4.2841e-02,  1.3770e-01, -9.0373e-02, -2.5524e-01, -6.4782e-01,\n",
       "          -2.4416e-01, -7.7096e-02,  2.8265e-01, -4.0933e-01, -1.4547e-01,\n",
       "           1.0118e-01,  2.1869e-01, -5.5722e-02,  5.9932e-01,  3.6780e-01,\n",
       "          -2.1750e-01, -3.3352e-01, -5.6900e-01, -4.1145e-03, -6.4206e-02,\n",
       "           4.2841e-01,  4.1005e-01, -2.3083e-01, -6.6280e-01, -4.0858e-01,\n",
       "          -2.2012e-01,  8.5329e-01, -1.4409e-01, -3.2051e-01,  9.6469e-02,\n",
       "           4.4245e-01, -4.5867e-02, -4.0926e-01,  2.0025e-02,  2.3056e-02,\n",
       "          -1.9329e-01, -3.1854e-01,  4.2301e-01, -3.8030e-02,  2.5466e-01,\n",
       "           3.3309e-01,  2.3452e-01, -1.8591e-01,  6.3817e-01,  1.5373e-01,\n",
       "          -4.2977e-01, -8.0507e-02, -4.8788e-01,  1.2446e-01,  1.6184e-01,\n",
       "          -8.3537e-02,  6.4485e-01, -4.9381e-02, -4.5344e-01,  1.3276e-01,\n",
       "           3.0587e-01,  1.1167e+00, -1.1363e-01,  3.2400e-01,  4.2685e-01,\n",
       "           5.8359e-02, -2.3759e-01,  1.2959e-01, -2.8071e-01,  1.7872e-01,\n",
       "           4.3174e-01, -5.8123e-01, -6.1045e-01, -8.2382e-02,  4.2457e-01,\n",
       "           6.9720e-01, -4.8775e-01, -6.0306e-01, -4.1329e-01,  6.5184e-01,\n",
       "           3.3971e-02,  5.7363e-01, -4.1938e-02,  1.5482e-01,  3.8577e-01,\n",
       "          -6.4881e-01,  1.6504e-02,  1.1204e-01,  1.9935e-02, -6.3568e-02,\n",
       "          -4.1715e-01, -2.3002e-01, -2.5846e-01, -3.4178e-01,  5.5237e-01,\n",
       "           2.0446e-01, -1.0322e+00,  1.7012e-01,  5.6138e-01, -1.0252e+00,\n",
       "          -8.2842e-02, -3.5042e-01, -6.8051e-02,  1.6212e-01,  5.3855e-02,\n",
       "          -4.1600e-01,  7.6217e-02, -5.0585e-01, -5.2586e-01,  1.2954e-01,\n",
       "           7.4824e-01, -1.6515e-01,  1.5522e-01, -1.1954e-02, -5.8556e-01,\n",
       "          -2.3896e-01, -1.2349e-01, -4.1819e-01,  1.7169e-01, -3.2041e-03,\n",
       "          -1.4074e-01, -2.9339e-01, -1.2374e-01,  3.9420e-01,  6.3078e-02,\n",
       "           2.7390e-01,  2.8186e-01,  4.7710e-01,  8.3079e-02, -3.8005e-01,\n",
       "          -7.2862e-02,  1.3310e-01, -3.3126e-03,  4.3775e-01, -2.6136e-01,\n",
       "          -5.2265e-01,  1.0978e-01, -1.8887e-01, -7.3482e-01, -1.1607e-02,\n",
       "          -5.0759e-01,  1.0867e-01, -2.4182e-01,  7.1465e-02,  3.9294e-01,\n",
       "           3.0331e-01, -5.3342e-02, -1.3184e-01,  2.4586e-01,  1.6124e-01,\n",
       "           7.8310e-02,  8.7854e-01,  4.2669e-01,  3.2882e-01, -1.7049e-01,\n",
       "           3.3070e-01, -6.7632e-01, -1.8839e-01, -6.9532e-02, -2.0193e-02,\n",
       "           3.4640e-02, -3.7705e-01,  2.7600e-01,  2.0283e-01,  5.4257e-01,\n",
       "           4.4060e-01, -2.2287e-01, -6.4283e-01, -2.1943e-01, -7.4912e-03,\n",
       "           1.2109e-01, -1.8831e-01,  4.5566e-02,  1.9784e-01,  2.4691e-01,\n",
       "          -7.0805e-01, -2.0705e-01,  2.4480e-02,  1.1950e-01, -2.6640e-01,\n",
       "           8.0735e-01, -1.2580e-01, -4.8619e-02, -1.0108e-01,  1.9369e-01,\n",
       "          -9.0291e-02,  3.7250e-01,  6.0236e-02, -5.2538e-01, -3.6914e-02,\n",
       "           3.0285e-01, -3.4393e-01,  3.0927e-01, -4.0999e-01, -4.4029e-02,\n",
       "           1.2167e-01,  4.0328e-01,  6.1774e-01]], device='mps:0'),\n",
       " tensor([[-4.5377e-01, -4.6540e-01, -3.4826e-01, -1.0267e+00, -3.6390e-01,\n",
       "          -7.8980e-02, -5.7723e-01,  7.7333e-01,  1.0090e-01,  8.1387e-01,\n",
       "           2.6808e-01,  1.9819e-01,  1.4300e-02, -4.5675e-02, -6.7256e-01,\n",
       "          -6.1621e-02,  3.4699e-01,  4.8986e-01,  2.9110e-01,  3.1961e-01,\n",
       "           8.0945e-01,  5.5465e-01,  2.0671e-01, -2.6120e-01, -1.2193e-01,\n",
       "           4.3934e-01, -1.1218e-01,  2.7679e-01,  2.7720e-01, -3.2604e-01,\n",
       "          -9.3478e-02,  4.8509e-01, -2.8750e-01,  1.1813e+00, -2.5383e-01,\n",
       "          -1.5607e-01, -2.4188e-02, -1.3221e-01, -2.7532e-01, -1.1285e-01,\n",
       "           6.6173e-01, -2.7007e-01, -1.0609e-01,  4.2035e-01,  1.0413e-01,\n",
       "          -9.5271e-02, -2.8993e-02, -7.0744e-01,  4.1511e-01, -4.3433e-01,\n",
       "           4.7825e-02, -2.9410e-01,  1.1814e-02,  3.8485e-01, -7.8292e-01,\n",
       "          -4.8292e-01,  2.8092e-01,  2.6784e-01, -3.1456e-01, -2.5278e-01,\n",
       "           1.3556e-01,  6.6750e-01, -7.6199e-02, -1.0568e-01,  3.8248e-01,\n",
       "          -2.2104e-01, -3.8898e-01, -8.7203e-01, -9.7665e-02,  8.2073e-02,\n",
       "          -1.8151e-01,  4.5989e-01, -1.8072e-01, -5.3471e-02,  3.3507e-01,\n",
       "          -5.7473e-02,  2.9617e-01, -2.9161e-01, -3.4071e-01, -3.0827e-02,\n",
       "           3.1839e-01,  1.5401e-01,  8.0310e-02, -9.9241e-02, -2.0615e-01,\n",
       "          -2.3839e-01,  5.0606e-01,  1.1046e-01,  3.2783e-01,  3.4875e-02,\n",
       "           4.7885e-01, -5.6379e-02, -9.7173e-04, -3.8532e-01,  3.1006e-01,\n",
       "          -8.6314e-02,  3.2731e-01, -9.6577e-02, -3.4611e-01, -2.8645e-01,\n",
       "          -1.1859e-01,  2.1368e-01, -2.2163e-01,  1.5224e-01, -6.7939e-01,\n",
       "          -3.0860e-01,  2.0772e-01, -1.3363e-01, -4.0252e-01,  3.1865e-01,\n",
       "          -2.4144e-01, -3.9750e-01,  1.4466e-01, -6.7065e-01, -1.6097e-01,\n",
       "           8.4845e-02,  6.1557e-01, -2.3819e-01, -2.8606e-01,  1.7892e-01,\n",
       "           2.5878e-01, -3.9271e-01,  1.1690e-01, -8.8343e-02,  1.1060e-01,\n",
       "          -2.4678e-01,  4.1031e-01, -2.7143e-01,  2.0071e-01,  7.4425e-01,\n",
       "           3.1643e-01, -2.1285e-01, -4.1456e-01, -9.9720e-02,  2.3693e-01,\n",
       "          -3.1492e-02, -2.2609e-01, -1.4534e-01, -9.4292e-02,  9.6513e-02,\n",
       "           1.6030e-02, -1.6643e-01, -2.0581e-01, -4.8806e-02, -1.2199e-01,\n",
       "          -2.5974e-01,  2.2924e-02,  1.2429e-01,  1.7284e-01, -1.7112e-01,\n",
       "          -8.3243e-01, -5.6912e-01,  5.9879e-01,  9.7075e-02, -2.3431e-01,\n",
       "           4.1478e-02,  3.8688e-01, -5.8266e-02,  5.7073e-01,  3.4957e-01,\n",
       "           2.4812e-01, -3.3565e-01, -2.4754e-01, -2.2415e-01, -8.5620e-02,\n",
       "          -3.7378e-02,  1.7497e-01, -3.5350e-01,  2.0656e-01,  1.5846e-01,\n",
       "           4.5382e-01, -2.5513e-01,  1.3631e-01, -1.9110e-01, -2.6198e-01,\n",
       "          -1.5784e-01,  5.4807e-02, -1.4121e-01, -2.6802e-01,  4.5267e-01,\n",
       "           1.0146e-02, -1.7812e+00,  5.0499e-01, -5.2383e-01, -3.5665e-01,\n",
       "          -4.5071e-02,  3.5560e-01,  3.6205e-01,  5.2713e-01, -4.3111e-02,\n",
       "          -2.0681e-01, -3.5911e-01, -6.4064e-04,  3.4104e-01, -1.6312e-01,\n",
       "           4.9619e-01,  5.1570e-01,  1.9015e-01, -3.0299e-01,  2.9790e-02,\n",
       "          -2.0583e-01,  4.6037e-01,  4.1197e-02,  2.0714e-01,  8.4674e-02,\n",
       "           1.2783e-01, -4.4912e-02, -1.2043e-01,  2.8106e-01,  1.5702e-01,\n",
       "          -9.3855e-02, -2.1034e-01, -3.7327e-01,  7.4437e-02, -2.4541e-01,\n",
       "           2.1770e-01, -5.7103e-01, -3.4832e-01,  2.8332e-01,  1.4751e-02,\n",
       "          -2.2670e-01,  1.5614e-02,  2.4499e-02, -6.6547e-02, -3.4726e-02,\n",
       "           5.7392e-01, -3.6486e-01, -6.5909e-02, -2.6906e-01, -1.7485e-01,\n",
       "          -6.9790e-02, -2.5088e-01,  6.9706e-01,  5.4226e-01,  1.3862e-01,\n",
       "          -2.2848e-01, -2.4926e-01,  2.3482e-01, -3.3206e-01,  1.7711e-01,\n",
       "          -1.4118e-01, -3.2390e-01, -7.7595e-01,  8.8460e-02, -2.3701e-01,\n",
       "           2.9816e-01,  1.3491e-01,  5.5916e-01,  8.4952e-01,  4.4164e-01,\n",
       "          -1.3104e-01,  1.5511e-01,  5.2849e-01,  1.9447e-01, -2.5820e-01,\n",
       "           4.2209e-01,  2.4780e-02, -4.2856e-01,  3.3893e-01, -4.0078e-01,\n",
       "           2.7864e-02, -3.5936e-02, -2.5213e-01,  1.8535e-01, -2.5041e-01,\n",
       "           5.3590e-01,  5.1747e-01,  2.2939e-02,  1.1109e-01,  5.5617e-01,\n",
       "          -1.1608e-01,  2.7770e-01, -5.2769e-01, -1.5571e-01, -7.5354e-01,\n",
       "          -3.5045e-01, -1.2397e-01,  1.4403e-01, -1.6514e-01, -5.6909e-01,\n",
       "          -2.1000e-01, -1.5057e-01,  6.4924e-02,  1.7325e-01, -8.0698e-03,\n",
       "           3.2543e-01,  1.5241e-01, -2.7110e-02, -9.3120e-02, -2.1121e-01,\n",
       "          -3.8303e-01,  3.6617e-01, -7.3690e-01,  6.6418e-03,  3.7645e-01,\n",
       "          -1.1787e-01,  3.2148e-01,  3.9732e-01, -1.5339e-01, -4.5713e-01,\n",
       "          -2.9904e-01,  2.1679e-01,  1.7216e-01, -1.1815e-01,  3.9725e-01,\n",
       "           1.3225e-01,  2.8751e-01,  3.6737e-01, -3.4175e-01, -4.2027e-01,\n",
       "          -2.7852e-01,  3.4345e-01, -5.7295e-02, -1.0185e-01, -2.5541e-01,\n",
       "          -9.2812e-01, -1.2970e-01, -1.2721e-01, -6.6214e-02, -5.6386e-01,\n",
       "           5.0563e-02,  9.6751e-02,  2.5181e-01,  3.3041e-01, -2.2258e-01,\n",
       "           7.3862e-01,  4.0292e-01,  1.3137e-01, -2.2616e-01, -4.3675e-01,\n",
       "          -2.1503e-01,  1.6315e-01, -2.3397e-01, -5.4453e-01,  9.9130e-02,\n",
       "           1.0886e-01, -1.1016e-01, -1.1349e-01,  2.2911e-01, -2.1360e-01,\n",
       "           5.6404e-01,  1.3995e-03, -1.2226e-01,  4.4004e-01,  7.1901e-02,\n",
       "           7.3040e-02,  2.4619e-01, -9.2326e-02, -2.6328e-01,  2.6945e-01,\n",
       "           4.2679e-01, -1.0988e+00, -3.7692e-01, -3.9414e-01,  4.9880e-01,\n",
       "           2.7004e-01,  3.4991e-01, -1.5881e-01,  4.9467e-01,  3.1682e-01,\n",
       "           2.0066e-02,  3.6394e-01,  7.5882e-02, -1.7806e-01,  3.1894e-01,\n",
       "           2.5467e-01,  4.1024e-01, -3.9450e-01, -7.4368e-02, -5.7149e-01,\n",
       "           1.6475e-01, -5.8065e-01,  1.5932e-01,  1.6111e-01,  5.7808e-01,\n",
       "          -1.5312e-01, -4.5268e-01,  3.4160e-01, -1.0806e+00, -4.3018e-01,\n",
       "           2.6389e-01, -2.1937e-01, -1.1823e-01, -1.5385e-01, -1.2345e-01,\n",
       "          -2.5464e-02, -1.4947e-01,  5.8454e-01,  1.1130e-01, -2.4654e-01,\n",
       "           1.4568e-01, -2.2743e-01, -2.0386e-01,  3.9171e-01, -1.3771e-01,\n",
       "          -2.8872e-01,  5.0498e-01,  1.0539e-01, -3.0649e-01,  3.4502e-01,\n",
       "          -3.6442e-01,  3.8943e-02,  2.4377e-01, -2.9072e-01, -3.2983e-01,\n",
       "          -2.7391e-01,  6.4890e-02,  2.4573e-01, -2.4030e-01, -2.6020e-02,\n",
       "          -6.7974e-01, -2.6061e-01,  1.2073e-01, -2.1568e-01,  2.0777e-01,\n",
       "           4.3722e-01,  4.2449e-02, -1.5840e-01,  1.1993e-01, -2.0983e-01,\n",
       "           3.4071e-01, -1.2555e-01, -7.4874e-02,  4.6267e-01, -1.1339e+01,\n",
       "          -6.4208e-02,  8.3852e-01, -2.9606e-01,  1.4091e-01, -4.1062e-01,\n",
       "          -7.1281e-02, -4.2681e-01,  2.0892e-01, -2.4017e-01, -1.2188e-01,\n",
       "           1.3450e-01, -8.5041e-02,  2.2042e-02, -3.6502e-01, -1.7072e-01,\n",
       "           1.0834e-02, -8.1628e-01,  1.2766e-01, -1.9468e-01,  1.9028e-01,\n",
       "          -4.8772e-01, -5.8744e-03,  1.5114e-01, -1.2475e-01, -9.2739e-02,\n",
       "          -1.2794e-01, -5.0054e-01,  1.9995e-01,  2.6760e-01, -4.8114e-02,\n",
       "          -2.2911e-01,  4.1787e-01, -2.9600e-02, -2.6428e-01, -4.6966e-01,\n",
       "          -8.3021e-02,  2.9721e-01,  1.0342e-01, -1.6667e-01, -2.9426e-01,\n",
       "          -1.6334e-01, -6.0351e-01,  1.7931e-01,  2.9297e-01,  5.4385e-01,\n",
       "           9.4112e-01,  7.9280e-01,  4.1449e-02, -6.1736e-02, -2.4515e-01,\n",
       "          -8.9853e-01,  4.3381e-01, -6.5781e-01,  2.5201e-01, -1.5001e-01,\n",
       "           6.9912e-01, -5.6967e-01,  2.5047e-01, -4.0086e-01, -1.0294e-01,\n",
       "           1.9581e-01,  5.4286e-02,  7.7195e-03, -1.8560e-01, -1.5852e-01,\n",
       "           2.9846e-01, -3.5204e-01,  2.1949e-01, -2.3039e-01,  4.9672e-01,\n",
       "          -3.1145e-01,  4.0398e-01,  2.1793e-01,  1.4279e-01,  4.4354e-01,\n",
       "           8.6958e-02,  9.5083e-02,  7.9764e-01, -2.0888e-01,  2.8125e-01,\n",
       "          -2.3267e-01,  7.6125e-01,  9.7142e-02, -8.7455e-01,  1.9892e-01,\n",
       "           2.6012e-01, -4.0379e-01, -6.1237e-01,  7.3868e-01, -7.0426e-02,\n",
       "          -4.2429e-01, -2.0891e-01, -1.2131e-01,  1.5141e-01,  5.4276e-01,\n",
       "          -4.0725e-01, -1.6554e-01,  2.0328e-01,  1.4087e-01, -8.5313e-01,\n",
       "           3.4912e-03, -9.5819e-02, -2.2888e-01,  4.1717e-01, -5.8332e-01,\n",
       "           2.8089e-01,  6.6559e-02, -1.5099e-01,  4.5433e-01, -1.5078e-01,\n",
       "           1.2942e-02, -1.7108e-01,  1.8248e-01, -3.7326e-01, -4.9405e-01,\n",
       "           8.9837e-02,  4.8305e-01, -2.2259e-01, -5.6235e-01, -1.9381e-01,\n",
       "          -1.4826e-01,  2.9155e-02, -2.0645e-01, -9.2238e-01,  4.4351e-01,\n",
       "          -6.9231e-01, -1.0383e-01,  3.2829e-02, -8.9938e-04,  4.1140e-01,\n",
       "           7.6374e-02, -2.3810e-01,  1.2457e-01, -4.2153e-01, -3.4297e-01,\n",
       "          -3.4963e-01, -1.2627e-01,  3.4832e-01, -5.2762e-01, -6.2287e-03,\n",
       "           1.8584e-01,  2.1021e-02, -2.2510e-01,  3.2217e-01, -4.8431e-02,\n",
       "           3.7038e-01,  3.2694e-01, -3.3359e-01, -4.7647e-02, -2.4343e-01,\n",
       "           1.3811e-01, -5.7848e-01, -3.0840e-02,  1.6657e-01, -6.7666e-01,\n",
       "           5.9311e-02, -1.9061e-01, -5.2691e-01,  2.0322e-01,  2.1473e-01,\n",
       "           3.0293e-01, -1.2551e-03, -8.2857e-02,  2.6384e-01,  2.2716e-01,\n",
       "          -9.0438e-02, -2.2691e-02, -7.1690e-01,  1.8480e-01,  6.6080e-01,\n",
       "           1.5643e-01, -1.4661e-02, -3.8433e-01, -2.2930e-01, -5.6191e-01,\n",
       "          -5.6033e-01,  8.0323e-01, -5.0553e-01, -6.1994e-01,  1.7321e-01,\n",
       "           2.4255e-01, -5.4988e-02,  1.4885e-01,  1.9443e-01,  1.3903e-01,\n",
       "           1.2431e-01, -3.5548e-01,  4.5100e-02,  2.7692e-01,  3.3172e-01,\n",
       "           2.6971e-01,  4.2955e-01,  3.3402e-01, -2.3678e-01, -2.1624e-02,\n",
       "          -5.3817e-01,  9.8771e-02, -3.1655e-01,  2.1937e-01,  2.5662e-01,\n",
       "          -1.2458e-01, -7.4034e-02, -1.1511e-01, -3.5221e-01,  2.3909e-02,\n",
       "           3.2012e-02,  1.4794e+00, -1.6893e-02, -2.2239e-01,  3.9674e-01,\n",
       "           1.1422e-01,  3.6497e-01, -1.3351e-01,  1.5055e-01,  4.4790e-02,\n",
       "           4.6769e-01,  2.0342e-02, -1.2108e-01,  2.0956e-01,  4.9599e-01,\n",
       "           3.0792e-02, -8.6193e-01, -5.4159e-01, -1.6837e-01,  4.3661e-01,\n",
       "           1.4241e-01,  7.4222e-01, -1.4235e-01,  3.6067e-01,  3.3598e-01,\n",
       "          -1.1410e+00, -2.5949e-01, -2.1525e-01,  4.6616e-01, -9.9254e-02,\n",
       "           1.0814e-01, -3.5478e-01,  1.1949e-01, -3.9568e-01,  6.3540e-01,\n",
       "           7.5808e-01, -6.5598e-01,  3.3207e-01,  4.1979e-01, -1.5197e-01,\n",
       "           5.5854e-01, -2.5619e-01,  5.8639e-01,  1.9523e-01, -1.4421e-01,\n",
       "          -1.5540e-01,  1.2865e-01, -3.9943e-01, -9.3076e-02,  2.1188e-01,\n",
       "           5.6957e-01, -1.1218e-01,  1.0976e-01, -2.4707e-01,  8.9793e-02,\n",
       "          -5.0616e-01, -1.0909e-01, -2.3858e-01, -2.4258e-01,  3.7940e-01,\n",
       "          -1.5953e-01, -1.2217e-01, -7.7210e-02,  5.0987e-01, -1.2564e-01,\n",
       "           4.8427e-01,  5.3966e-01,  4.0581e-01, -2.3818e-01, -2.1711e-01,\n",
       "           4.0151e-02,  4.1900e-01, -5.5514e-02,  7.9403e-01, -1.8900e-01,\n",
       "          -4.8858e-01,  2.1342e-01,  8.6464e-02, -2.1183e-01,  7.8846e-02,\n",
       "          -2.6943e-01,  2.5475e-01, -5.7346e-02, -1.8844e-01,  1.5142e-01,\n",
       "          -1.3979e-01,  3.3697e-01, -8.2360e-01,  1.1760e-01,  9.4606e-02,\n",
       "          -7.4495e-02,  5.2425e-01, -2.7541e-01,  4.5440e-01, -1.4852e-01,\n",
       "          -1.3582e-01, -2.4192e-01, -2.9916e-01,  3.6948e-02,  8.6087e-02,\n",
       "          -6.1182e-01, -1.3388e-01,  2.0912e-01,  8.5063e-02,  4.1030e-01,\n",
       "           9.5362e-01, -2.1452e-01, -3.8599e-01, -4.0412e-01, -2.1011e-01,\n",
       "          -2.2722e-02,  7.5514e-02, -1.0996e-01, -9.8546e-02,  1.2538e-02,\n",
       "          -7.1806e-01, -4.6335e-01,  1.1233e-01,  4.4731e-01, -3.4165e-01,\n",
       "           7.1108e-01, -1.6188e-01, -1.6318e-01,  4.5594e-01,  6.1110e-02,\n",
       "           5.0819e-01,  7.3969e-01,  8.8296e-02, -5.3562e-01, -6.1764e-02,\n",
       "           1.4071e-02, -4.9295e-01,  4.1528e-01, -1.9480e-01,  2.4153e-01,\n",
       "           2.3398e-01,  2.2262e-01,  4.3282e-01]], device='mps:0'),\n",
       " tensor([[-0.1860, -0.5475, -0.2602,  ...,  0.0794,  0.4866,  0.3014],\n",
       "         [-0.0844, -0.9954, -0.7668,  ...,  0.0671,  0.4228,  0.6851]],\n",
       "        device='mps:0'),\n",
       " tensor([[-0.1835, -0.6425, -0.3709,  ...,  0.2866,  0.3817,  0.7981],\n",
       "         [-0.2567, -0.3027, -0.5871,  ...,  0.0705,  0.7106,  0.7832],\n",
       "         [-0.2330, -0.4411, -0.4565,  ...,  0.0257,  0.5397,  0.4576],\n",
       "         [-0.3359, -0.7143, -0.8535,  ..., -0.0995,  0.7248,  0.5324]],\n",
       "        device='mps:0'),\n",
       " tensor([[-1.1097e-02, -5.0567e-01, -4.6699e-01, -6.4640e-01, -3.9556e-01,\n",
       "          -2.5266e-01, -5.5990e-01,  3.0801e-01, -1.2536e-01,  1.4189e-01,\n",
       "           2.9133e-01,  5.3879e-01, -3.6706e-01, -2.8859e-01, -4.8290e-01,\n",
       "          -5.3699e-02, -6.1741e-01, -1.6918e-02,  3.4690e-01,  3.0491e-01,\n",
       "           5.3718e-01,  1.8073e-01, -4.1129e-01,  2.9149e-01, -3.7959e-01,\n",
       "           6.6561e-02, -3.4985e-01, -3.1546e-01,  2.6210e-01,  2.8012e-01,\n",
       "           5.3618e-02,  7.7692e-01, -2.4307e-02,  1.0962e+00, -1.8413e-01,\n",
       "          -2.1778e-01,  1.5359e-01,  2.3320e-01,  8.6439e-01, -6.2204e-01,\n",
       "           5.6395e-01,  3.6489e-01,  2.7586e-02, -4.0063e-02,  2.7845e-01,\n",
       "           3.5451e-01,  1.4567e-01, -5.7913e-01, -2.7161e-01, -3.4037e-01,\n",
       "          -1.2979e-01, -1.5164e-01,  5.0058e-01, -2.3292e-01, -7.9546e-01,\n",
       "          -1.9253e-01,  1.1288e-02,  2.6982e-01,  1.1117e-01,  2.3591e-01,\n",
       "           2.8900e-02,  7.9425e-01, -1.9339e-01, -3.2966e-03,  2.0559e-01,\n",
       "           2.5502e-01, -7.6243e-01, -1.2873e-01, -7.2822e-01,  1.1548e-01,\n",
       "           2.1413e-01,  3.5162e-01,  1.0500e-01, -2.6816e-01, -1.9644e-01,\n",
       "           6.0243e-02,  1.6956e-01,  1.8610e-01, -2.0035e-01,  3.5967e-02,\n",
       "           2.1140e-01,  4.4140e-01,  4.5980e-01,  5.9922e-02,  1.6117e-01,\n",
       "           5.6960e-02,  2.0080e-01, -3.9869e-02, -3.1738e-02,  1.7672e-01,\n",
       "           1.1984e-01, -9.9577e-02, -3.7483e-01, -4.3956e-01, -9.5205e-02,\n",
       "          -3.0408e-01,  3.2250e-02, -2.9469e-01, -2.3954e-01, -3.3261e-01,\n",
       "           2.5982e-01,  9.2762e-02,  3.2750e-01, -9.3082e-02, -6.6514e-01,\n",
       "          -3.5248e-01,  4.0832e-01,  3.4586e-03, -2.8129e-01, -3.8448e-01,\n",
       "          -6.6713e-02, -4.9346e-01,  1.2765e-01, -4.4650e-01,  1.9893e-01,\n",
       "           1.0835e-01,  1.8686e-01, -3.2670e-01,  6.5172e-02,  8.8101e-02,\n",
       "           6.1068e-02, -5.0938e-01,  2.7421e-01,  1.8326e-01,  9.3046e-02,\n",
       "           2.2489e-01,  6.8015e-01, -1.6210e-01,  4.3454e-01,  3.5722e-01,\n",
       "           3.1169e-01,  1.6221e-01, -4.1686e-01, -5.0259e-01, -7.5302e-02,\n",
       "           1.8488e-02, -5.9908e-01, -5.7442e-01,  6.1723e-02, -2.2868e-01,\n",
       "           1.6627e-01, -7.3083e-02, -5.1768e-01, -5.4307e-01, -3.1494e-01,\n",
       "          -3.0772e-01, -2.0517e-02, -3.2393e-01, -4.8159e-01, -2.5398e-01,\n",
       "          -6.1867e-01, -7.2891e-01,  1.9443e-01,  1.5164e-01,  2.7714e-01,\n",
       "          -4.3163e-01,  5.4580e-01, -3.0612e-01, -4.8554e-01,  3.7642e-01,\n",
       "           4.3432e-01, -5.0330e-01,  2.9526e-01, -6.3499e-01,  5.1171e-02,\n",
       "           2.0480e-01,  2.5291e-02, -3.4147e-02,  4.2807e-01,  1.1727e-01,\n",
       "           3.9831e-01,  2.7152e-02,  1.2476e-01, -2.9187e-01,  1.5015e-01,\n",
       "          -4.5427e-02, -3.2147e-01, -1.1176e-01,  1.7279e-02, -5.2967e-02,\n",
       "           5.4936e-01, -1.7306e+00, -1.7852e-01,  7.0060e-02, -2.7460e-01,\n",
       "          -1.3173e-01,  2.1473e-01,  5.0407e-03,  3.2283e-01, -3.5674e-01,\n",
       "           8.3698e-02,  8.7128e-02,  8.4654e-02, -1.0725e-01, -5.0983e-02,\n",
       "           2.9101e-01,  7.5929e-01,  4.0793e-01,  9.7463e-02, -2.6657e-01,\n",
       "          -6.0645e-01,  4.1554e-01,  1.9081e-01,  2.0730e-01,  3.9000e-01,\n",
       "           6.7931e-02, -6.9343e-01,  2.9674e-01,  1.9859e-01,  3.7002e-01,\n",
       "           2.5546e-02, -5.8496e-02, -4.8962e-01, -2.4682e-01, -4.3705e-01,\n",
       "           2.7759e-01, -1.3436e-01, -2.5922e-01,  6.3173e-01,  1.9273e-01,\n",
       "          -5.3531e-02,  3.9653e-01,  1.8733e-01, -4.1041e-01,  4.1633e-01,\n",
       "           5.2814e-01,  7.9114e-02,  1.4573e-01, -2.8055e-01, -1.6673e-01,\n",
       "          -3.1532e-01, -4.4420e-01,  4.4703e-01,  8.3941e-01, -7.2084e-03,\n",
       "           5.5343e-01, -3.5757e-01,  5.1174e-01, -3.6424e-01, -6.7779e-03,\n",
       "          -4.6486e-02,  6.8700e-02, -3.3988e-01,  1.6284e-01,  1.5551e-01,\n",
       "           1.0340e-01,  1.9709e-01,  8.5425e-01,  4.6494e-01,  1.4684e-01,\n",
       "           4.1031e-01,  2.5417e-02,  3.2080e-01,  7.8684e-01, -3.0254e-01,\n",
       "           3.4990e-01, -7.1578e-03, -4.1842e-01,  5.6461e-01, -3.5139e-01,\n",
       "           1.4248e-01, -2.3020e-01,  2.2987e-01,  1.7866e-01, -2.7854e-01,\n",
       "           2.1426e-01,  6.5273e-01,  9.5860e-02, -3.9283e-01,  7.3417e-01,\n",
       "          -4.0214e-01,  7.1262e-01, -3.0138e-02, -2.6992e-02, -1.0120e-01,\n",
       "          -7.4577e-01, -6.8569e-01,  1.4294e-01,  3.0808e-01, -7.6332e-01,\n",
       "          -3.4295e-01,  1.2380e-01,  1.3255e-01,  3.5068e-01,  2.4156e-01,\n",
       "           4.6144e-01, -1.7016e-01, -4.7185e-02, -4.4024e-01, -4.6015e-01,\n",
       "           1.1917e-01,  7.3311e-02,  9.5311e-02, -9.8199e-02,  9.2587e-02,\n",
       "           6.1067e-01,  2.4297e-01,  2.6799e-01,  1.4946e-01, -3.4610e-01,\n",
       "          -2.7723e-02,  1.7135e-01,  3.7874e-01, -6.1900e-01, -2.1807e-01,\n",
       "           5.3543e-02,  8.3164e-02,  6.6826e-01, -2.1832e-01, -1.1546e-01,\n",
       "          -1.2450e-01,  5.9739e-01, -2.2108e-01,  6.8002e-02,  1.0269e-01,\n",
       "          -6.8586e-01,  2.7619e-01, -5.9939e-02, -8.5297e-02, -5.2861e-01,\n",
       "           2.2997e-01, -5.7787e-02, -1.6068e-01, -7.3269e-03, -7.7601e-02,\n",
       "           2.2454e-01,  1.3406e-01,  2.8044e-01, -5.6270e-01, -4.2092e-01,\n",
       "           1.1781e-01, -3.2365e-01, -6.6016e-02, -4.5861e-01,  3.6766e-01,\n",
       "           1.3890e-01,  4.3829e-01, -1.0837e-01, -2.8576e-02, -1.3190e-01,\n",
       "           5.2866e-01, -6.2780e-01,  1.5573e-02,  4.8525e-01, -3.5595e-01,\n",
       "          -1.8139e-01,  4.7644e-01, -2.8011e-01,  4.9664e-02,  2.2982e-01,\n",
       "           3.7738e-01, -1.1530e-01, -4.7105e-01, -5.5247e-02,  9.3761e-01,\n",
       "           4.8980e-01,  2.0874e-02, -6.3820e-01,  1.2560e-01,  4.2066e-01,\n",
       "          -3.4219e-01,  1.0777e-01, -3.7280e-01, -2.3519e-01,  8.6456e-02,\n",
       "           3.3260e-02,  5.0344e-01, -5.8998e-02,  1.9741e-01,  4.6778e-01,\n",
       "           1.7011e-01, -6.9331e-01, -3.0557e-01,  6.3124e-01,  5.4670e-01,\n",
       "          -1.2247e-01, -5.4832e-01,  1.1480e-01, -6.4563e-01,  3.1198e-02,\n",
       "          -3.8623e-03, -3.5755e-01,  2.9653e-01, -1.4962e-01, -2.5683e-01,\n",
       "          -2.0003e-01, -5.3748e-01,  4.4973e-03,  1.0291e-01, -6.9595e-02,\n",
       "           4.1738e-01, -7.9610e-03, -3.1154e-01,  1.0058e+00, -2.7292e-01,\n",
       "          -3.1356e-02,  7.4139e-01, -5.4234e-01,  1.0457e-02,  2.4642e-01,\n",
       "          -1.3433e-01, -1.2416e-02,  2.4323e-02,  1.0660e+00,  1.1515e-05,\n",
       "           3.6402e-02, -2.0346e-01,  4.2203e-01, -4.7659e-01,  1.9394e-01,\n",
       "          -4.9958e-01,  4.8310e-01, -6.9399e-02,  2.4253e-01, -8.2329e-01,\n",
       "           4.3257e-01, -3.0604e-01, -3.2040e-01, -6.3922e-01,  5.0359e-02,\n",
       "          -5.0711e-02,  6.6524e-02, -5.8707e-02,  6.0787e-01, -1.0953e+01,\n",
       "           1.0946e-01,  8.4108e-01, -3.2618e-02,  1.3653e-01,  4.3336e-01,\n",
       "           1.2158e-01, -4.6684e-01,  7.2509e-01, -2.3242e-01, -6.6092e-03,\n",
       "          -3.5861e-01, -1.2444e-01, -1.3819e-01, -6.3519e-01,  4.0758e-01,\n",
       "          -1.3255e-01, -5.9114e-01,  2.8540e-01, -1.6945e-01,  2.4978e-01,\n",
       "          -2.3651e-01,  1.5671e-01,  5.9352e-01, -6.1010e-01, -3.3943e-01,\n",
       "          -1.1958e-01, -2.9991e-01,  5.1723e-01,  4.0217e-01, -2.3672e-01,\n",
       "           2.9096e-01,  3.3529e-02,  2.9455e-01, -5.5694e-01, -4.8133e-01,\n",
       "           4.3369e-01, -3.9950e-01,  2.1957e-01, -9.9943e-01, -3.3265e-01,\n",
       "          -1.8082e-01, -5.0951e-01, -2.7870e-02,  5.3812e-01,  7.5524e-01,\n",
       "           1.2821e+00,  1.1534e+00,  9.8198e-02,  4.6362e-01, -1.0523e-01,\n",
       "          -6.6935e-01,  5.1619e-02, -5.6525e-01, -4.1224e-01,  8.7196e-02,\n",
       "           2.2385e-01, -5.4352e-01,  4.7854e-01, -5.9897e-01,  1.8912e-01,\n",
       "           1.4329e-01,  7.9852e-02,  1.7368e-01, -5.1897e-01, -5.5039e-01,\n",
       "          -1.3282e-01,  8.1201e-02,  8.6324e-02, -6.7881e-01,  1.1881e-01,\n",
       "           1.7006e-01,  6.8197e-01, -2.5397e-01, -4.4268e-01,  4.1137e-01,\n",
       "          -2.2746e-03,  9.3809e-02,  1.3158e+00, -2.6036e-01, -1.7413e-01,\n",
       "          -4.8428e-02, -2.3194e-01, -3.3484e-01, -3.9656e-01, -4.5613e-01,\n",
       "          -2.5020e-02,  3.6573e-01,  5.1948e-01,  2.6062e-01, -2.3329e-02,\n",
       "          -3.0054e-01, -4.1445e-02,  5.1977e-02,  3.3862e-01,  9.6772e-01,\n",
       "          -6.6887e-01,  2.6280e-01, -1.9148e-01,  9.2178e-02,  2.2223e-01,\n",
       "          -5.7106e-01, -1.6871e-02,  2.9434e-03,  1.3975e-02,  5.7281e-01,\n",
       "           1.9705e-01,  2.6453e-01,  3.1190e-02,  2.2859e-01, -4.1682e-01,\n",
       "          -4.4044e-01,  2.9125e-04, -1.3190e-01, -5.8294e-01, -4.1034e-01,\n",
       "          -1.3514e-01,  1.3712e-01, -2.8585e-01, -5.3585e-01, -2.5012e-01,\n",
       "           1.4192e-01,  2.5900e-03,  4.9952e-02, -3.3857e-01,  4.7379e-01,\n",
       "          -7.6918e-01, -2.1429e-01, -2.7578e-01, -3.8495e-03,  1.4633e-01,\n",
       "           2.5508e-01, -5.1518e-01, -2.7838e-01, -1.0123e-01, -9.2314e-01,\n",
       "          -8.4489e-01,  2.2927e-02,  7.4620e-01, -1.1960e-01,  1.1861e-01,\n",
       "           8.5428e-02, -3.1671e-03, -3.7708e-01,  2.5194e-01, -2.4644e-01,\n",
       "           2.8149e-01,  4.3265e-01, -4.9583e-01,  2.9773e-02, -1.2792e-01,\n",
       "          -1.8992e-02, -1.0887e-01, -8.8647e-02, -5.4131e-01, -6.1371e-01,\n",
       "          -4.1197e-01,  1.3736e-01, -2.2747e-01,  4.1516e-01,  6.8072e-02,\n",
       "           1.2582e-01, -2.0826e-01,  9.8224e-02,  8.2540e-01,  6.0197e-01,\n",
       "           2.3014e-01, -3.2920e-01, -7.7666e-01,  3.4140e-02,  4.1642e-02,\n",
       "          -1.6450e-01,  2.1317e-01, -1.6384e-01, -3.0657e-01, -1.0112e-01,\n",
       "           1.2620e-02,  1.0002e+00, -2.0790e-01, -1.9155e-01,  1.0507e-01,\n",
       "           7.4626e-01, -6.0834e-01, -5.7221e-01,  3.7612e-01,  5.0418e-02,\n",
       "          -5.6831e-01, -1.9450e-01,  4.0315e-01, -1.7111e-01,  8.9921e-02,\n",
       "           3.7572e-01,  1.7821e-01, -1.8612e-01,  2.2711e-01, -6.7291e-02,\n",
       "          -1.7001e-01,  3.2731e-01, -1.6533e-01, -7.7254e-02,  4.0384e-01,\n",
       "           9.9517e-02,  1.5584e-01, -2.5262e-01, -4.8444e-01, -2.7025e-01,\n",
       "           1.5439e-01,  1.1110e+00, -1.7989e-01, -3.2527e-01,  6.6881e-01,\n",
       "          -3.0956e-01,  3.8928e-01,  4.8478e-01, -2.2611e-01, -5.0807e-02,\n",
       "           5.8049e-01,  3.0288e-01, -7.3236e-01,  2.2417e-01,  3.1670e-01,\n",
       "          -1.7335e-01, -5.3095e-01, -3.2838e-01, -2.1330e-01, -3.5189e-02,\n",
       "          -3.6356e-01,  5.2132e-01,  1.2255e-01,  3.9253e-01, -1.0843e-01,\n",
       "          -8.7207e-01,  4.7185e-01,  2.6128e-01,  8.0250e-01, -6.2568e-03,\n",
       "          -4.4880e-01, -3.5541e-01, -3.5790e-01, -6.6453e-02,  3.6543e-01,\n",
       "           2.0116e-01, -1.0333e+00,  4.6071e-02,  4.0507e-01, -3.9812e-02,\n",
       "           4.7083e-01, -1.4254e-01,  1.9634e-01,  4.7891e-01, -4.3918e-02,\n",
       "          -2.1856e-01,  1.3161e-01, -3.3027e-02, -2.7190e-01,  4.2861e-01,\n",
       "           4.6755e-01,  1.6062e-02,  2.5454e-02,  6.4881e-02, -4.2427e-01,\n",
       "          -3.4980e-01,  7.7064e-02, -1.7966e-01, -3.5675e-01, -1.4295e-01,\n",
       "          -4.0726e-01, -5.1772e-01, -6.6164e-02, -1.1357e-01,  1.3233e-01,\n",
       "           3.1032e-01,  1.3036e-01,  3.8431e-01, -2.3708e-01, -1.2541e-01,\n",
       "          -2.7977e-01,  4.5653e-01, -2.7595e-01,  8.0808e-01,  2.0566e-01,\n",
       "          -1.4423e-01,  2.0720e-01,  2.1514e-01, -2.9046e-01,  1.3028e-01,\n",
       "          -2.7828e-01,  2.8810e-01, -5.7798e-01, -3.4110e-01, -5.9710e-02,\n",
       "          -7.7948e-02, -1.5049e-01, -2.8505e-01, -1.0179e-01,  1.0608e-01,\n",
       "          -9.9043e-02,  9.7868e-01, -4.1759e-01,  6.3833e-01, -1.0342e-01,\n",
       "          -1.1350e-01, -7.5089e-01,  3.9397e-01,  3.6376e-01, -1.4798e-01,\n",
       "          -5.6434e-01, -2.1653e-01, -8.9158e-02, -4.2284e-01,  8.8621e-01,\n",
       "           5.8157e-01, -7.4673e-01, -3.7246e-01, -1.5919e-01, -2.4113e-01,\n",
       "          -4.7389e-02,  6.9820e-02, -3.2710e-01, -1.3295e-02, -1.6580e-01,\n",
       "          -9.8715e-01,  2.7641e-01, -3.6473e-02,  4.9205e-01, -2.5466e-01,\n",
       "           7.6979e-01, -8.6017e-02, -1.2749e-01,  1.1690e+00,  1.1169e-01,\n",
       "          -6.1844e-02,  5.9531e-01, -1.6837e-01, -6.2973e-01, -7.0295e-02,\n",
       "           4.4239e-02, -3.1098e-01,  8.7670e-02, -4.7880e-01,  2.9353e-01,\n",
       "           2.3648e-01, -7.1226e-02,  4.3877e-01]], device='mps:0'),\n",
       " tensor([[-0.1672, -0.5249, -0.4540,  ...,  0.3771,  0.2016,  0.5340],\n",
       "         [-0.3292, -0.2590, -0.5016,  ..., -0.0148,  0.3509,  0.1607]],\n",
       "        device='mps:0'),\n",
       " tensor([[-2.7791e-02, -7.5747e-01, -5.3074e-01, -5.1430e-01, -3.4417e-01,\n",
       "           7.2208e-02, -3.2733e-01,  1.9133e-01, -1.5003e-01,  5.7895e-01,\n",
       "           2.7665e-01,  1.8669e-01, -4.0441e-02, -1.9787e-02, -2.7559e-01,\n",
       "          -3.6770e-01, -5.5075e-01,  4.2714e-01,  4.8599e-01,  1.2291e-01,\n",
       "           6.0920e-01,  4.0016e-01, -5.7060e-01,  2.7172e-01, -1.0960e+00,\n",
       "           2.2592e-01, -9.0650e-02,  1.6102e-01,  2.6783e-01, -4.2947e-01,\n",
       "          -2.6593e-01,  3.3457e-01, -1.0081e-01,  1.1911e+00, -7.2573e-02,\n",
       "          -4.2319e-01,  1.2905e-01, -4.8449e-01,  2.2964e-01,  1.0587e-01,\n",
       "           3.3713e-01, -2.3616e-01,  3.0495e-01,  2.5426e-01,  2.2269e-01,\n",
       "          -1.0156e-01,  3.5441e-01, -6.7506e-01,  5.3763e-02, -5.4964e-01,\n",
       "          -2.7866e-01, -2.3157e-01,  3.8469e-01,  3.0486e-01, -1.7402e-01,\n",
       "          -1.7631e-01,  1.9710e-01,  2.8978e-01, -1.6215e-01,  1.9001e-01,\n",
       "          -3.1728e-01,  3.9028e-01, -2.9116e-02, -2.5337e-02, -4.3117e-01,\n",
       "           5.2338e-02, -1.0542e+00, -2.0097e-01, -6.1678e-01, -2.9287e-01,\n",
       "           6.1601e-02, -4.4671e-02,  2.6994e-02,  8.2943e-02,  6.2566e-01,\n",
       "          -2.9527e-01, -3.5686e-02, -6.8721e-02, -8.9378e-01,  3.5990e-02,\n",
       "           9.3805e-02,  2.8378e-01,  3.4715e-01, -1.8362e-01,  4.0523e-02,\n",
       "          -1.1761e-01,  5.2336e-01,  3.2885e-01,  5.5885e-02, -1.1072e-01,\n",
       "           2.6658e-01,  1.4680e-01,  3.1471e-01, -2.3564e-01,  2.1408e-01,\n",
       "          -1.6953e-01,  4.2040e-01, -7.0445e-02, -1.3569e-01, -1.7957e-01,\n",
       "           1.7683e-01,  2.7367e-01,  1.6921e-01,  1.8827e-01, -7.0935e-02,\n",
       "          -4.6949e-01,  6.4462e-03, -2.7295e-01, -9.2490e-01,  5.3292e-02,\n",
       "           2.5952e-02,  8.0243e-02,  3.5441e-01, -5.5720e-01, -4.9249e-01,\n",
       "           2.3824e-01,  3.1274e-01, -7.8117e-02, -2.4385e-01,  7.2875e-02,\n",
       "           1.2005e-01, -3.7322e-01,  3.0937e-01,  1.4655e-01,  3.8964e-01,\n",
       "          -2.4076e-01,  5.1062e-01, -1.8687e-01,  4.6773e-01,  1.7625e-01,\n",
       "          -2.8715e-02, -1.4186e-02,  2.4065e-02, -2.5766e-01, -7.0436e-02,\n",
       "          -3.2938e-01,  4.0965e-01, -2.1077e-01,  1.5229e-01, -2.7347e-01,\n",
       "          -1.9571e-02, -1.5879e-01, -4.2139e-01, -2.4379e-01,  5.2323e-01,\n",
       "          -2.3310e-01,  3.4454e-01, -2.5864e-01, -5.3971e-01, -5.4621e-01,\n",
       "          -4.6331e-01, -6.6221e-01,  6.5904e-04, -5.8988e-02,  1.5486e-02,\n",
       "          -1.4458e-01, -4.7299e-02,  3.4941e-01, -1.5764e-01, -1.3263e-01,\n",
       "           1.3229e+00, -1.6963e-01,  2.6547e-01, -5.2711e-01, -9.2709e-02,\n",
       "          -2.5951e-01,  3.5857e-01, -2.1281e-01, -5.0373e-02, -1.1031e-02,\n",
       "           1.5012e-01, -3.6325e-01,  3.2426e-01, -2.7495e-01,  4.9574e-02,\n",
       "          -3.8145e-01,  6.7830e-01,  1.0315e-01, -3.7603e-03, -2.1426e-01,\n",
       "           8.8470e-02, -1.8509e+00, -6.0244e-02, -4.9988e-02, -1.1984e-01,\n",
       "          -3.8304e-01,  1.1412e-01, -8.1008e-02,  4.7211e-01,  1.0523e-01,\n",
       "          -3.0391e-01,  1.4457e-02,  3.4290e-01,  1.7766e-01,  4.6617e-02,\n",
       "           3.7586e-01,  5.1268e-01,  5.2866e-01,  8.3422e-02, -1.2969e-01,\n",
       "          -2.2910e-01,  1.8840e-01, -1.4922e-01, -4.9168e-03,  1.3625e-01,\n",
       "           1.9163e-01, -1.7921e-02,  4.5103e-02,  4.5549e-01, -3.9944e-02,\n",
       "          -1.3034e-01, -3.0975e-01, -2.0571e-01, -1.5386e-01, -3.0506e-01,\n",
       "           6.6388e-01, -6.9598e-02,  1.5083e-01,  1.1394e-01, -7.6365e-02,\n",
       "          -1.9021e-01, -3.6627e-01,  7.4035e-02, -4.3048e-01, -1.0458e-01,\n",
       "           5.2612e-01, -2.4620e-01, -9.6266e-02, -1.5057e-01, -2.1628e-01,\n",
       "           4.6617e-01, -6.3064e-01,  6.9359e-01, -6.5753e-02, -1.5876e-01,\n",
       "          -6.3829e-02, -3.6947e-01,  8.8525e-02, -4.7900e-02, -2.1367e-01,\n",
       "          -6.6109e-01, -6.2912e-01, -9.3484e-01,  3.0931e-02,  2.0012e-01,\n",
       "           3.8940e-01, -2.0282e-01,  7.0611e-01,  4.1043e-01,  4.4392e-01,\n",
       "          -1.9033e-01, -2.5634e-01,  1.9540e-01,  2.4175e-02, -1.6470e-01,\n",
       "          -8.3407e-02, -3.6482e-02, -2.3338e-01,  5.4958e-01, -2.6909e-01,\n",
       "           3.4143e-02, -8.4658e-02,  1.4949e-01,  5.6272e-01, -2.6404e-02,\n",
       "           2.7543e-01,  3.6181e-02, -5.5097e-01, -2.6403e-02,  1.4938e-01,\n",
       "          -3.9414e-01,  1.3074e-01,  6.3249e-01, -2.9116e-01, -5.8419e-01,\n",
       "          -5.6450e-01,  1.3798e-02, -1.0222e-01,  2.9551e-01, -1.1019e+00,\n",
       "          -1.6877e-01,  2.8032e-01,  3.0136e-02,  1.8251e-01, -3.0887e-01,\n",
       "           2.9186e-01,  5.8483e-02,  3.7632e-02, -7.8447e-02, -5.1537e-01,\n",
       "          -2.3439e-01,  1.6469e-01, -3.6657e-01, -3.5963e-02,  1.6028e-01,\n",
       "           2.8092e-01, -9.8223e-02,  4.3126e-02,  2.2221e-01, -4.8483e-01,\n",
       "          -8.7845e-02, -3.3919e-02,  1.5657e-01,  6.3212e-02,  6.2429e-02,\n",
       "           7.6042e-02,  1.9293e-01,  7.3101e-01, -3.7007e-02, -6.0240e-01,\n",
       "          -2.4535e-01,  5.4258e-01,  7.7029e-02,  1.5324e-01, -1.7752e-01,\n",
       "          -1.4260e+00, -6.3469e-01,  3.8337e-02,  2.1822e-02, -9.6237e-02,\n",
       "           3.7186e-01,  1.4415e-01,  2.9691e-02,  1.0135e-01, -1.4892e-01,\n",
       "           2.7125e-01,  2.5571e-01,  4.1796e-01,  2.6634e-01,  1.2224e-01,\n",
       "          -9.5239e-02, -2.5680e-01, -3.4749e-01, -2.5056e-01,  5.0008e-02,\n",
       "           3.0551e-01,  2.9752e-01, -2.5921e-01, -9.4277e-02, -9.0702e-02,\n",
       "           7.3119e-01, -4.5527e-01, -5.6552e-01,  3.9878e-01, -5.2456e-01,\n",
       "          -2.7738e-01,  4.5605e-01, -3.7995e-01,  6.1826e-02, -3.0256e-02,\n",
       "           1.6101e-01, -7.3998e-02,  5.4701e-02,  5.9767e-02,  1.4680e-01,\n",
       "           1.8510e-01,  4.4459e-01, -9.6782e-02,  3.8301e-01,  4.8393e-01,\n",
       "          -3.7868e-01,  1.0631e-01, -3.1295e-01,  3.3607e-02,  3.3206e-02,\n",
       "           3.5477e-02,  1.0657e-01, -5.8454e-01, -7.1591e-02,  1.8413e-01,\n",
       "           3.9868e-02, -5.0830e-01, -1.9227e-01,  1.5691e-01, -1.5280e-01,\n",
       "          -1.9063e-02, -6.1864e-01,  7.4105e-02,  2.7357e-02, -1.0623e-01,\n",
       "           2.0196e-01,  3.6702e-02,  1.0747e-01, -2.9594e-01, -3.5062e-01,\n",
       "          -4.3860e-01,  1.3449e-01,  6.4194e-01, -1.6284e-02, -1.8554e-02,\n",
       "          -1.0969e-02, -1.8924e-01, -3.0452e-01,  4.8185e-01,  1.5484e-01,\n",
       "          -2.7278e-01,  1.5357e-01, -1.5461e-01,  4.3362e-01,  4.5208e-01,\n",
       "          -3.5378e-01, -1.2043e-01, -1.1214e-01,  3.6801e-01, -1.9150e-01,\n",
       "          -1.7609e-01,  2.8881e-01,  4.5279e-01,  1.9897e-01,  9.4702e-02,\n",
       "           2.5524e-01,  6.2194e-01, -7.1092e-01,  2.7741e-01, -5.0384e-02,\n",
       "           1.6210e-01, -3.1364e-01,  1.1810e-01, -4.5970e-01, -3.1796e-01,\n",
       "           2.0608e-01,  3.4692e-01,  1.5942e-01, -8.2525e-02, -1.1386e+01,\n",
       "           4.7410e-01,  7.1675e-01, -6.4620e-01, -1.2380e-02,  2.0990e-01,\n",
       "           1.4467e-01, -4.2321e-01,  1.2785e-01, -4.0986e-01, -4.8454e-01,\n",
       "           5.3751e-02, -3.1024e-01,  8.5046e-02, -2.1082e-01, -3.5012e-02,\n",
       "          -3.3293e-01, -8.8604e-01,  5.2616e-01, -2.3722e-01,  1.2282e-01,\n",
       "          -4.0554e-01,  2.2834e-01,  6.7441e-02, -3.7185e-01, -2.9844e-01,\n",
       "          -3.3378e-01, -4.8637e-01,  2.5874e-01,  1.4526e-01, -3.6388e-01,\n",
       "           1.6897e-01,  7.8214e-02,  1.5488e-01, -3.1408e-01, -5.7322e-01,\n",
       "           6.5890e-02, -1.4286e-01,  6.9077e-01, -2.3317e-01,  2.3521e-01,\n",
       "           1.5418e-01, -1.8939e-01,  4.0595e-01, -6.7756e-02,  5.4405e-01,\n",
       "           9.0099e-01,  4.9506e-01, -9.3677e-02, -3.1796e-01,  2.9697e-01,\n",
       "          -1.1296e+00,  2.3765e-01, -7.4326e-01,  3.7037e-02, -4.4715e-02,\n",
       "           5.2303e-01, -5.4682e-01, -1.6110e-02,  1.6453e-01, -1.3098e-01,\n",
       "           3.8056e-01,  1.5443e-02,  3.0641e-01, -3.8540e-01, -1.8694e-01,\n",
       "           5.1715e-01,  1.1540e-01,  3.9804e-01, -2.1344e-01, -2.7691e-02,\n",
       "          -3.5584e-01,  6.3533e-01,  4.1696e-01,  6.3335e-03,  1.0379e-01,\n",
       "           9.4310e-02,  1.5461e-01,  8.2262e-01,  1.5306e-01,  3.7272e-02,\n",
       "           7.5334e-02,  4.9919e-02, -2.0803e-02, -2.5454e-01, -1.9359e-01,\n",
       "          -3.8527e-02,  2.3219e-01,  6.4573e-01,  5.1229e-01, -2.5336e-01,\n",
       "          -1.8886e-01, -2.0242e-01, -5.9704e-02,  1.3338e-01,  5.6888e-01,\n",
       "          -7.2007e-01,  1.4785e-02,  2.0411e-01, -2.3545e-01, -2.6105e-01,\n",
       "          -9.1127e-02, -2.2821e-01, -6.1202e-01,  4.0102e-01, -3.4853e-03,\n",
       "          -3.2201e-01,  8.7693e-02,  5.6282e-01,  1.7925e-01, -2.9609e-01,\n",
       "           4.8297e-01, -1.4248e-01,  5.2177e-01, -7.0864e-02,  3.5673e-02,\n",
       "           3.0329e-01,  6.1367e-01, -2.1669e-01, -3.9304e-01, -6.3589e-02,\n",
       "          -7.9591e-02,  2.2539e-01,  3.5268e-03,  7.8083e-02, -1.6630e-02,\n",
       "          -3.5275e-01, -1.3786e-01,  3.3867e-02,  2.5058e-01,  2.6520e-01,\n",
       "           6.0789e-02,  3.0671e-01,  2.9447e-01, -3.1026e-01, -2.5075e-01,\n",
       "          -7.3840e-01, -2.4288e-01,  2.2993e-01, -1.9391e-01, -9.8460e-02,\n",
       "          -2.1162e-02, -3.4200e-02, -3.9679e-01,  3.1646e-01, -8.8932e-02,\n",
       "           1.4835e-01,  1.1151e-01, -4.4561e-01,  2.3351e-01, -3.3892e-02,\n",
       "          -3.1032e-01, -9.2286e-02, -3.7850e-01,  8.2453e-02,  3.6353e-02,\n",
       "          -4.8302e-01,  2.5437e-01, -2.5304e-01, -6.9363e-01, -3.3891e-01,\n",
       "           1.0299e-02,  1.8091e-02, -3.8864e-01,  1.0920e-01,  5.7217e-02,\n",
       "          -2.6965e-02, -5.7936e-01, -2.5618e-01, -5.4129e-03,  4.7410e-02,\n",
       "           3.7889e-01, -4.1326e-01,  5.3043e-02, -5.8918e-01,  3.3814e-01,\n",
       "           8.6381e-02,  8.8725e-01, -9.5259e-02, -4.3658e-01,  4.1371e-01,\n",
       "           6.8976e-01, -3.6204e-01,  2.3138e-01, -2.0207e-01,  1.7246e-01,\n",
       "          -2.2461e-01, -1.5982e-01,  4.5024e-01, -2.3929e-01,  5.3763e-01,\n",
       "           7.6361e-02,  3.0156e-01, -5.0755e-01, -3.4200e-02, -4.5787e-01,\n",
       "          -2.6191e-01, -1.0136e-01, -4.3868e-01, -2.1873e-01, -9.5628e-02,\n",
       "          -3.6356e-01, -2.6224e-01, -2.7189e-01,  1.6586e-02, -1.9311e-01,\n",
       "           3.5684e-01,  1.1473e+00,  3.5222e-02, -4.8536e-02,  3.1878e-01,\n",
       "           2.3119e-01, -2.1068e-01,  2.0884e-01, -7.6245e-02,  3.3463e-01,\n",
       "           2.9654e-01,  3.1331e-01, -4.5673e-01,  2.5522e-01, -8.5886e-02,\n",
       "           3.3382e-01, -8.9138e-01, -7.5951e-01,  1.8828e-02,  2.3505e-01,\n",
       "          -4.1163e-02,  7.9677e-02, -2.6933e-01,  6.0446e-01,  6.9945e-01,\n",
       "          -1.3167e+00, -6.0678e-02, -2.4393e-01,  1.2029e-01, -4.5133e-04,\n",
       "          -4.4085e-01, -3.7242e-01, -3.3575e-01, -2.8487e-01,  3.2123e-01,\n",
       "           7.5280e-01, -4.5114e-01,  1.6297e-02,  1.8465e-01, -1.1076e-01,\n",
       "           2.3514e-01, -5.0295e-03,  2.5102e-01,  4.3425e-01, -1.0414e-01,\n",
       "          -1.4909e-01,  4.2579e-01, -3.1190e-01, -4.2031e-01,  1.5833e-01,\n",
       "           7.0937e-01, -1.1853e-01,  1.3689e-01, -1.8297e-02, -2.6799e-01,\n",
       "          -7.1901e-02, -1.1732e-01, -1.9540e-01,  2.4573e-01,  3.1104e-01,\n",
       "           1.0016e-01, -3.2782e-01,  2.3126e-01,  2.1841e-01, -5.2982e-02,\n",
       "           5.0868e-02,  9.0378e-01,  7.3841e-01,  1.8846e-02, -7.4087e-02,\n",
       "           2.7607e-01,  2.4221e-01, -5.0505e-02,  5.1696e-01, -1.7273e-01,\n",
       "          -2.8848e-01,  1.9170e-01,  8.5894e-02, -2.9880e-01, -9.4752e-03,\n",
       "          -2.0048e-01,  1.8031e-01,  1.8487e-01, -1.6871e-01,  5.3942e-01,\n",
       "           3.3819e-01,  1.9990e-01, -4.1667e-01,  6.3733e-05,  2.5826e-01,\n",
       "          -2.8037e-01,  8.7311e-01,  1.6816e-01,  4.6258e-01, -3.0563e-01,\n",
       "           2.2482e-01, -6.0811e-01, -4.3120e-01,  1.4365e-01,  7.4790e-02,\n",
       "          -3.6079e-01,  1.8614e-01, -1.2540e-01, -1.5393e-01,  7.3765e-01,\n",
       "           5.8692e-01, -3.4385e-01, -5.9073e-01, -5.4749e-02,  3.2823e-02,\n",
       "          -2.9436e-02, -2.4520e-01,  7.4517e-02, -2.1959e-02,  1.6957e-01,\n",
       "          -7.0204e-01, -2.6400e-01,  1.2722e-01,  2.0311e-01, -4.2548e-01,\n",
       "           8.2082e-01,  7.0018e-02,  1.7358e-01,  7.3647e-01,  1.3118e-01,\n",
       "           1.7952e-01,  6.3050e-01,  6.9376e-02, -5.5670e-01,  1.2268e-01,\n",
       "           2.8105e-01, -1.1468e-01, -5.5720e-02, -6.3967e-01,  3.2919e-01,\n",
       "           8.0147e-02,  9.4187e-01,  7.0432e-01]], device='mps:0'),\n",
       " tensor([[-0.4654, -0.4115, -0.4370,  ...,  0.1625,  0.4338,  0.5908],\n",
       "         [-0.1242, -0.5179, -0.4093,  ...,  0.3691, -0.0529,  0.5134],\n",
       "         [-0.0916, -0.6086, -0.3076,  ...,  0.3904,  0.6534,  0.4003],\n",
       "         [-0.7184, -0.4251, -0.6743,  ...,  0.1987,  0.1606,  0.2312],\n",
       "         [-0.4491, -0.3888, -0.8171,  ...,  0.2574,  0.1279,  0.4124]],\n",
       "        device='mps:0'),\n",
       " tensor([[-0.0248, -0.4215, -0.6790,  ...,  0.1175,  0.3931,  0.9420],\n",
       "         [-0.3413, -0.8224, -0.4278,  ...,  0.2859,  0.5113,  0.5540]],\n",
       "        device='mps:0'),\n",
       " tensor([[ 9.2527e-02, -6.7690e-01, -6.5961e-01, -6.7476e-01, -2.4173e-01,\n",
       "          -2.1967e-01, -5.9736e-01,  1.5405e-01,  2.4009e-01,  4.4930e-01,\n",
       "           2.5711e-01, -5.9154e-02, -7.5814e-02, -2.2846e-01, -4.2886e-01,\n",
       "          -1.8564e-01, -5.0430e-01,  3.2359e-01,  7.2550e-01,  1.1514e-02,\n",
       "           7.6498e-01,  6.7626e-01, -3.6783e-01,  1.7408e-01, -8.8976e-01,\n",
       "           1.0173e-01, -8.7582e-02,  2.5819e-01,  1.1992e-01, -4.9268e-01,\n",
       "           4.1939e-02,  3.3562e-01, -6.0114e-02,  1.2479e+00,  2.7327e-02,\n",
       "          -3.1041e-01,  2.4144e-01, -4.5027e-01,  7.8802e-02,  2.3770e-01,\n",
       "           3.9584e-01, -7.6745e-02,  1.7257e-01,  5.8488e-01,  2.6486e-01,\n",
       "          -1.2689e-01,  3.7247e-01, -4.3754e-01,  5.2169e-01, -4.2791e-01,\n",
       "          -3.0750e-01, -3.4547e-01,  1.5350e-01,  1.0028e-01, -1.7589e-01,\n",
       "          -4.0415e-01,  3.1373e-01,  3.9113e-01, -9.6441e-02,  5.1185e-02,\n",
       "          -3.6627e-01,  3.1297e-01, -1.8621e-01,  1.9179e-01, -2.6939e-01,\n",
       "           4.9894e-02, -9.4640e-01, -5.9543e-01, -6.2259e-01, -1.0610e-01,\n",
       "           8.5857e-02,  1.9498e-01,  1.7502e-01,  4.9815e-02,  7.0415e-01,\n",
       "          -5.2611e-01,  8.6877e-02, -2.1241e-01, -6.0591e-01,  1.2487e-01,\n",
       "          -1.3546e-02,  6.0045e-02,  2.8357e-01,  2.0969e-02,  1.8058e-01,\n",
       "          -1.2566e-01,  4.9549e-01,  2.2228e-01,  3.6799e-03,  1.6412e-01,\n",
       "           2.3338e-01,  1.8316e-01,  3.1028e-01, -2.6951e-02,  5.4812e-02,\n",
       "          -5.1217e-02,  5.0375e-01, -2.1930e-01, -2.2121e-01, -3.0912e-01,\n",
       "           1.3948e-01,  2.4957e-01,  2.3887e-01,  3.0172e-01, -3.1474e-01,\n",
       "          -6.5304e-01, -7.3087e-02, -3.1741e-01, -7.7487e-01,  6.1738e-02,\n",
       "          -3.2210e-01,  1.8978e-02,  3.0999e-01, -6.3994e-01, -1.2155e-01,\n",
       "          -3.1466e-02,  6.6034e-02, -1.5386e-01, -1.3673e-01,  1.5042e-01,\n",
       "           5.9989e-02, -4.9810e-01,  2.9317e-01, -9.1656e-03,  1.4631e-01,\n",
       "          -2.3530e-02,  4.8667e-01, -2.4477e-01,  1.7211e-01,  2.8143e-01,\n",
       "           6.0149e-02, -2.2871e-01,  1.2881e-03, -1.7444e-01, -1.7909e-01,\n",
       "           1.7552e-02,  4.9230e-01, -5.2099e-02, -1.7399e-01, -4.0571e-01,\n",
       "           2.3500e-01, -2.5356e-03, -3.4836e-01, -2.4052e-01,  2.3948e-01,\n",
       "          -2.9420e-01, -1.9975e-01, -2.8037e-01, -3.1515e-01, -6.2244e-01,\n",
       "          -3.7004e-01, -4.3175e-01,  1.3035e-01, -2.3071e-01,  1.6086e-02,\n",
       "          -2.7307e-01, -3.5304e-01,  8.8337e-03,  2.6647e-02,  1.2896e-01,\n",
       "           1.0893e+00, -4.6513e-01,  1.8682e-01, -3.3143e-01, -2.0413e-01,\n",
       "          -7.9460e-02,  3.2027e-01, -1.5268e-01, -1.6311e-01,  4.4760e-02,\n",
       "           1.6796e-01, -2.1038e-01,  4.1518e-01, -2.0239e-01,  1.8004e-02,\n",
       "          -4.6573e-01,  6.9239e-01,  1.8499e-01, -3.0110e-02, -5.5018e-02,\n",
       "           4.5887e-01, -1.4771e+00,  2.3287e-02, -4.5786e-01,  1.7348e-01,\n",
       "          -3.0874e-02,  1.5836e-01,  3.9731e-02,  3.8329e-01,  2.4096e-03,\n",
       "          -6.3158e-02, -7.8805e-02,  4.4973e-01,  1.9331e-01,  9.0756e-02,\n",
       "           3.1072e-01,  4.9602e-01,  4.3236e-01,  2.0558e-01,  1.1082e-01,\n",
       "          -3.7917e-01,  3.2583e-01,  1.3182e-01, -4.7920e-01,  1.3741e-01,\n",
       "           5.2713e-03, -4.7174e-01,  9.8007e-02,  1.2276e-01,  2.3342e-01,\n",
       "          -1.5627e-01, -3.8818e-01, -2.9775e-01, -1.8633e-01, -6.6318e-01,\n",
       "           5.8744e-01, -2.2404e-01, -1.7328e-01,  1.8100e-01, -6.9102e-02,\n",
       "          -1.7886e-01, -9.8428e-02,  1.5290e-01, -3.1445e-01, -1.2976e-01,\n",
       "           6.6082e-01, -5.9319e-01,  8.5485e-03, -7.0989e-02, -2.1786e-01,\n",
       "           2.7004e-01, -9.1717e-01,  7.1317e-01, -5.2586e-02,  1.9477e-01,\n",
       "          -5.4997e-02, -3.5977e-01, -3.1522e-02, -2.7631e-01, -1.7972e-01,\n",
       "          -8.2649e-01, -3.1887e-01, -1.0254e+00,  3.0757e-02, -4.5129e-02,\n",
       "           2.3145e-01, -6.8055e-02,  7.4625e-01,  5.1677e-01,  5.9123e-01,\n",
       "          -2.4705e-01, -1.7290e-01,  4.6177e-01, -1.4106e-01, -2.7065e-01,\n",
       "           2.4353e-01, -1.4893e-03, -2.9689e-01,  4.5966e-01, -2.9652e-01,\n",
       "           2.1600e-02, -2.9714e-01,  3.4073e-01,  6.3371e-01, -1.5879e-01,\n",
       "           5.1999e-02, -1.8982e-01, -1.2904e-01, -1.4732e-01,  5.6983e-01,\n",
       "          -3.6249e-01, -7.1523e-02,  3.6672e-01, -1.8202e-01, -5.9991e-01,\n",
       "          -5.1046e-01,  3.5203e-01,  1.1487e-01,  1.3619e-01, -1.2934e+00,\n",
       "          -3.4805e-01,  4.7691e-01,  2.7427e-01,  1.1170e-01, -1.9196e-01,\n",
       "           2.4938e-01, -3.3040e-01,  1.1417e-01, -1.1465e-01, -3.3377e-01,\n",
       "          -1.1081e-01,  1.2423e-01, -6.3092e-01,  1.9188e-01,  4.3529e-01,\n",
       "           1.3085e-02,  2.2521e-01,  2.2843e-01, -1.7933e-02, -3.5655e-01,\n",
       "          -3.1768e-01,  4.3756e-02, -1.2728e-01, -4.8219e-02,  1.0451e-01,\n",
       "           1.0425e-01,  3.9726e-01,  5.2073e-01, -1.3780e-01, -6.2039e-01,\n",
       "          -1.1878e-02,  7.0862e-01,  9.0874e-02, -2.7442e-01, -7.7977e-02,\n",
       "          -1.1928e+00, -3.4952e-01, -9.6744e-02, -2.3630e-01, -3.9115e-01,\n",
       "           9.8506e-02,  6.3862e-02, -1.1067e-01,  3.8250e-02, -2.2251e-01,\n",
       "           5.8545e-02,  3.7507e-01,  3.5871e-01,  1.2167e-01,  9.5850e-02,\n",
       "          -3.0516e-01, -1.5048e-01, -4.0721e-01, -3.4759e-01, -2.2326e-01,\n",
       "           1.1130e-01,  3.2721e-01, -3.6835e-01,  2.5918e-02,  6.4476e-02,\n",
       "           5.8058e-01, -5.2502e-01, -6.8590e-01,  4.6516e-01, -3.8738e-01,\n",
       "          -1.3878e-01,  2.4595e-01, -5.2946e-02, -1.0838e-02, -1.8987e-02,\n",
       "          -7.4822e-02, -4.8342e-01, -1.2670e-01, -2.4635e-01,  2.7628e-01,\n",
       "           3.8550e-01,  3.3274e-01, -4.5019e-01,  4.4314e-01,  2.6783e-01,\n",
       "          -3.1777e-01,  8.5874e-02, -1.0175e-01,  7.1443e-02, -1.8663e-01,\n",
       "           2.3270e-01,  2.8426e-01, -6.0129e-01, -1.4251e-01,  1.3899e-02,\n",
       "          -7.5893e-02, -4.3838e-01, -9.4557e-02,  2.2544e-01, -2.7461e-01,\n",
       "          -2.1456e-01, -4.9662e-01,  9.7228e-02,  8.8612e-02, -8.3540e-02,\n",
       "           2.8050e-01,  2.1304e-01, -1.0072e-01, -2.7576e-02, -3.8252e-01,\n",
       "          -1.8189e-01, -3.3397e-02,  5.0621e-01,  1.5698e-01,  1.5807e-03,\n",
       "           2.5419e-01, -2.2324e-01, -1.2680e-01,  4.2776e-01,  1.3549e-01,\n",
       "          -4.3668e-01,  4.9449e-01,  1.6796e-02,  2.1033e-01,  4.4183e-01,\n",
       "          -2.1709e-01,  1.3926e-01,  3.1911e-02,  7.6300e-01,  8.8504e-02,\n",
       "          -1.0757e-01,  1.4861e-01,  4.6372e-01, -2.9404e-01,  3.3587e-01,\n",
       "           3.0005e-01,  3.0241e-01, -7.4521e-01,  3.6952e-01,  8.0656e-02,\n",
       "           6.3549e-02, -6.1875e-02, -1.4495e-01, -4.5076e-01, -2.3025e-01,\n",
       "           3.7939e-01,  2.5044e-01,  2.9632e-01, -2.9228e-02, -1.1310e+01,\n",
       "           5.0152e-01,  7.5210e-01, -5.5978e-01, -3.8583e-01,  1.4938e-02,\n",
       "          -1.5754e-02, -3.2916e-01,  2.3378e-01, -2.5149e-01, -4.9787e-01,\n",
       "          -5.2326e-02,  3.3808e-02,  3.1030e-02, -4.0579e-01, -8.6203e-02,\n",
       "          -4.4928e-02, -6.1177e-01,  5.9641e-01,  1.5383e-02,  4.5525e-01,\n",
       "          -4.2322e-01,  2.5693e-01,  2.2368e-01, -5.2613e-01,  1.9883e-02,\n",
       "          -3.0066e-01, -2.6651e-01,  2.6799e-01,  4.0738e-01, -4.5722e-01,\n",
       "           1.2347e-01, -1.1578e-01,  4.6030e-02, -4.8203e-01, -4.8744e-01,\n",
       "           2.1304e-01, -2.2144e-01,  1.1329e+00, -5.2081e-02, -1.2981e-01,\n",
       "           6.6406e-03, -1.2326e-01,  3.6260e-01,  2.2316e-01,  4.8313e-01,\n",
       "           8.6575e-01,  8.3523e-01, -6.6254e-03, -1.6270e-01, -7.8468e-03,\n",
       "          -1.0438e+00,  3.1016e-01, -4.4246e-01, -9.5141e-02, -1.0734e-01,\n",
       "           3.7645e-01, -6.3508e-01,  9.1822e-03, -3.4988e-01,  1.1762e-02,\n",
       "           6.5213e-01,  6.4293e-02,  2.0466e-01, -4.3457e-01, -3.0054e-01,\n",
       "           4.8670e-01, -1.2167e-01,  4.6133e-01, -2.8246e-01, -1.7943e-01,\n",
       "          -4.0117e-01,  5.8213e-01,  4.4489e-01,  3.0163e-01,  1.7450e-01,\n",
       "           2.2161e-01, -1.1357e-01,  1.3177e+00,  2.2645e-01,  6.6530e-02,\n",
       "           1.5884e-01, -3.4728e-02, -6.8295e-02, -4.8803e-01, -2.5607e-01,\n",
       "          -1.6371e-01,  3.1301e-01,  3.7072e-01,  4.4555e-01, -6.3468e-02,\n",
       "          -5.4665e-01, -3.2873e-01,  8.0593e-02,  7.0071e-02,  5.5756e-01,\n",
       "          -7.5871e-01, -4.3110e-02, -1.2527e-01, -6.3291e-02, -7.4014e-01,\n",
       "          -1.7195e-02, -4.5558e-01, -7.7874e-01,  4.4952e-01,  5.5543e-02,\n",
       "          -1.8693e-01, -1.9082e-02,  4.3301e-01,  3.9075e-01, -1.6431e-01,\n",
       "           3.6937e-01, -4.2573e-01,  5.2139e-01, -3.8830e-01, -1.9002e-02,\n",
       "           3.8134e-01,  7.3444e-01, -2.8205e-01, -4.5904e-01, -1.1028e-01,\n",
       "           3.3531e-01,  4.8421e-02,  2.5143e-01, -1.4043e-01, -2.0192e-01,\n",
       "          -5.6873e-01, -2.2038e-02,  2.5104e-01,  2.5854e-01,  2.8629e-01,\n",
       "          -7.1849e-02, -2.5193e-03,  3.9088e-01, -5.7502e-01, -1.6908e-01,\n",
       "          -7.5777e-01, -3.0235e-01,  3.6264e-01, -3.6656e-01, -1.1324e-01,\n",
       "          -2.5294e-01, -7.8630e-02, -2.6578e-01,  2.0977e-01, -9.1546e-02,\n",
       "           2.8362e-01, -2.6563e-02, -2.5663e-01,  1.9265e-01,  3.7247e-02,\n",
       "          -3.9608e-01, -8.5364e-02, -2.0948e-01,  1.2267e-01,  4.5973e-02,\n",
       "          -2.3224e-01,  7.4326e-02, -5.5124e-01, -6.2444e-01, -8.9522e-02,\n",
       "          -9.1214e-02, -1.2000e-01, -6.1688e-01,  6.3422e-01,  2.7517e-01,\n",
       "          -6.3834e-02, -2.1295e-01, -1.5100e-01, -2.1952e-01,  6.3204e-02,\n",
       "           3.7311e-01, -3.3614e-01,  5.5177e-01, -6.5943e-01, -2.4407e-01,\n",
       "          -1.2107e-01,  1.0879e+00, -9.3831e-02, -7.2886e-01,  3.2577e-01,\n",
       "           4.9528e-01, -2.8316e-01,  1.2212e-01,  1.5994e-02,  1.2259e-01,\n",
       "          -2.6059e-01, -8.5484e-02,  3.6812e-01, -1.9476e-01,  3.7021e-01,\n",
       "           1.2692e-01,  1.1333e-01, -4.5577e-01,  3.3983e-02,  2.7847e-01,\n",
       "          -3.2319e-01, -9.8257e-02, -3.7547e-01,  1.8372e-02,  2.3215e-02,\n",
       "          -2.4764e-01,  1.8034e-01, -1.3381e-01, -1.2543e-01, -6.2393e-02,\n",
       "           2.8299e-01,  9.1515e-01,  5.0674e-02, -3.7280e-02,  3.5725e-01,\n",
       "          -2.6677e-02,  3.0744e-01,  2.3491e-01,  1.3114e-01,  2.2877e-01,\n",
       "           7.7546e-02,  1.0396e-02, -4.5192e-01,  2.0325e-01,  1.9798e-02,\n",
       "           3.7101e-01, -9.3175e-01, -8.3407e-01,  2.0493e-01,  2.9776e-01,\n",
       "           1.6041e-01,  2.6664e-01, -4.3334e-01,  6.1390e-01,  8.1072e-01,\n",
       "          -1.0811e+00, -1.1748e-01, -9.2776e-02, -1.5083e-02, -1.0464e-01,\n",
       "          -5.2966e-01, -8.0201e-02, -1.0080e-01, -3.8947e-01,  3.8898e-01,\n",
       "           9.1721e-01, -6.5005e-01, -5.7787e-02,  4.5661e-01, -1.9656e-01,\n",
       "           2.7879e-01, -4.0231e-01,  2.8169e-01,  4.9076e-01,  2.5801e-01,\n",
       "          -1.9543e-01,  3.3268e-01, -5.0585e-01, -3.3431e-01,  3.7815e-02,\n",
       "           8.1172e-01, -2.0222e-01,  1.2741e-01, -2.7863e-01, -2.3113e-01,\n",
       "          -1.5312e-01, -3.4103e-01, -2.1885e-01,  1.1047e-01,  1.5903e-01,\n",
       "           1.3588e-01, -4.7185e-02,  2.1258e-01,  3.8301e-01, -1.6682e-01,\n",
       "           5.3450e-03,  8.9391e-01,  5.8832e-01,  9.5066e-03, -9.5438e-02,\n",
       "          -1.6280e-01,  1.2837e-01, -6.5964e-02,  4.0262e-01, -1.9610e-01,\n",
       "          -2.6272e-01,  7.4795e-02,  1.4707e-01, -4.7910e-01,  1.1720e-01,\n",
       "          -4.3351e-01,  1.6852e-01,  4.1548e-02, -2.3318e-01,  2.4775e-01,\n",
       "           2.4639e-01,  3.3907e-01, -4.3047e-01,  2.5493e-01,  2.0443e-01,\n",
       "           1.4497e-01,  6.8241e-01,  9.7092e-02,  3.4217e-01, -1.6484e-01,\n",
       "           6.6163e-02, -7.3952e-01, -2.9111e-01, -5.2399e-02,  2.4940e-01,\n",
       "          -2.6337e-01,  2.8880e-02,  4.0319e-01,  2.4177e-01,  9.4725e-01,\n",
       "           4.0713e-01,  1.2077e-02, -5.6126e-01, -1.0598e-01,  1.0644e-01,\n",
       "           1.2572e-01, -3.7734e-01,  3.2798e-01, -1.7763e-01,  1.4219e-01,\n",
       "          -5.6209e-01, -4.6403e-01,  3.7269e-02,  1.4412e-01, -6.5614e-02,\n",
       "           9.1478e-01, -1.7699e-02,  1.2145e-01,  4.7267e-01,  1.6013e-01,\n",
       "           9.9121e-02,  7.5079e-01,  8.7889e-02, -5.5907e-01,  1.6583e-01,\n",
       "           1.9119e-01,  3.3675e-03,  2.9908e-01, -4.5660e-01,  2.4141e-01,\n",
       "           2.4457e-02,  8.7964e-01,  7.5721e-01]], device='mps:0'),\n",
       " tensor([[-0.1672, -0.5249, -0.4540,  ...,  0.3771,  0.2016,  0.5340],\n",
       "         [-0.3292, -0.2590, -0.5016,  ..., -0.0148,  0.3509,  0.1607]],\n",
       "        device='mps:0'),\n",
       " tensor([[-0.0253, -0.6578, -0.4489,  ...,  0.3289,  0.5162,  0.7200],\n",
       "         [-0.3702, -0.6568, -0.6846,  ...,  0.1169, -0.0985,  0.2539],\n",
       "         [-0.2941, -0.2550, -0.2842,  ..., -0.0493,  0.7798,  0.0119]],\n",
       "        device='mps:0'),\n",
       " tensor([[-0.4654, -0.4115, -0.4370,  ...,  0.1625,  0.4338,  0.5908],\n",
       "         [-0.0932, -0.6293, -0.4879,  ...,  0.3041, -0.1958,  0.5576],\n",
       "         [-0.3292, -0.2590, -0.5016,  ..., -0.0148,  0.3509,  0.1607]],\n",
       "        device='mps:0'),\n",
       " tensor([[-1.7731e-01, -3.8604e-01, -5.6476e-01, -2.2650e-01, -1.3208e-01,\n",
       "          -3.1625e-01, -5.9292e-01,  4.5825e-01,  2.2151e-01,  5.2940e-01,\n",
       "           3.8502e-01,  1.4343e-01,  6.5615e-02, -2.5563e-01, -4.4222e-01,\n",
       "          -1.6916e-01,  1.6884e-01,  5.0107e-01,  3.6126e-01, -7.6046e-03,\n",
       "           5.9085e-01,  3.7327e-01, -2.0317e-01,  2.5562e-01, -7.5220e-01,\n",
       "           2.6747e-02, -1.6815e-01, -2.5680e-02,  9.6832e-02, -3.0143e-01,\n",
       "          -1.1712e-01,  2.7595e-01, -3.7936e-01,  7.2144e-01, -1.4269e-01,\n",
       "          -2.3945e-01,  1.6259e-01, -1.0450e-01,  5.3649e-01, -1.3024e-01,\n",
       "           5.7769e-01, -2.9111e-01,  4.9321e-01,  3.4162e-02,  2.2711e-01,\n",
       "           1.3206e-01,  2.9819e-01, -4.0790e-01,  1.3721e-01, -5.3671e-01,\n",
       "          -1.6362e-02, -7.3465e-02,  4.1546e-01,  1.0724e-01, -6.3685e-01,\n",
       "          -2.7220e-01, -7.3533e-02,  4.2370e-01, -3.4661e-01,  1.4839e-01,\n",
       "           1.4452e-01,  4.6875e-01,  1.2890e-01,  1.1326e-01, -3.3348e-01,\n",
       "           3.0093e-01, -7.0323e-01, -8.0455e-01, -5.7621e-01, -1.8468e-01,\n",
       "           2.6210e-01,  2.4685e-01,  3.2711e-01, -6.3178e-01,  2.8882e-01,\n",
       "          -5.5500e-01,  4.7778e-01, -3.7641e-02, -3.8107e-01,  2.4976e-02,\n",
       "           4.6662e-01,  1.6673e-01, -2.3893e-01,  2.6411e-01,  2.9965e-01,\n",
       "          -6.2089e-02,  4.4818e-01, -2.9706e-02,  5.6407e-02,  3.8863e-01,\n",
       "           5.0392e-01,  2.6142e-01, -5.4248e-01, -2.1306e-01,  2.3546e-01,\n",
       "          -3.7940e-01,  2.9621e-01, -1.0776e-01,  2.0922e-02, -8.6342e-02,\n",
       "          -3.6881e-02,  3.4334e-01,  3.1658e-01, -6.2528e-02, -1.0724e+00,\n",
       "          -4.3983e-01, -2.4311e-02, -2.7814e-01, -3.4665e-01, -1.0267e-02,\n",
       "           7.0862e-03, -2.1454e-01,  4.5842e-01, -4.4114e-01, -3.9926e-01,\n",
       "           1.3807e-01, -3.8646e-02, -1.5082e-02, -1.4568e-01,  6.0048e-01,\n",
       "           1.6661e-01, -5.3057e-02,  1.5341e-01,  8.8933e-02,  4.9861e-01,\n",
       "          -3.9381e-01,  3.7161e-01, -4.2227e-01,  3.0903e-01,  6.8681e-02,\n",
       "          -2.0183e-02, -7.2510e-02, -2.5207e-01, -2.0895e-01, -2.4617e-02,\n",
       "           2.1165e-02,  2.8627e-01, -4.8458e-01,  5.5727e-02,  5.6803e-02,\n",
       "           2.5004e-01, -1.2498e-01, -4.2667e-01, -1.9318e-01, -6.4563e-02,\n",
       "          -4.0302e-02,  1.9246e-01, -1.1367e-01, -2.7591e-01, -2.6032e-01,\n",
       "          -7.7253e-01, -4.4007e-01,  3.9722e-01,  1.0952e-01,  9.2693e-02,\n",
       "           1.1479e-01,  2.1754e-01, -1.9775e-01, -1.8512e-01, -1.5673e-01,\n",
       "           5.8319e-01, -9.5195e-02,  6.6991e-02, -3.6372e-01, -3.0891e-01,\n",
       "           7.3167e-02,  1.7092e-01, -1.7339e-01,  4.3863e-02,  1.4590e-02,\n",
       "           2.2225e-01, -8.8855e-02, -1.0134e-03, -7.5909e-01, -3.0093e-02,\n",
       "          -2.8356e-01,  3.3814e-01,  2.7840e-01,  1.8768e-01, -2.4537e-01,\n",
       "           5.0899e-02, -1.7294e+00,  7.8671e-01, -5.9166e-01,  2.4877e-01,\n",
       "          -1.8150e-01,  3.1718e-01,  9.1189e-02,  1.8509e-01, -2.9267e-01,\n",
       "          -1.2026e-01, -3.9309e-02,  4.1623e-02,  2.9227e-01, -8.4757e-02,\n",
       "           4.8559e-01,  5.5627e-01,  3.3170e-01,  2.0436e-01, -9.9033e-02,\n",
       "          -6.0977e-01,  3.3235e-01,  1.6388e-01, -2.7455e-01,  2.9857e-02,\n",
       "           2.4361e-01, -3.2791e-01, -5.7703e-02,  3.4800e-01,  3.3556e-02,\n",
       "          -8.5587e-02, -3.0852e-01, -4.0912e-01, -1.8006e-01, -1.6990e-01,\n",
       "           3.0904e-01, -2.3500e-01, -2.0646e-01,  7.4699e-01,  9.5515e-02,\n",
       "          -2.2466e-02, -7.9463e-02, -2.8277e-01, -4.4039e-01, -2.0940e-01,\n",
       "           2.9576e-01, -5.3435e-01, -1.7453e-01, -2.8854e-01, -4.6020e-01,\n",
       "           4.6218e-01, -9.4786e-01,  6.3073e-01,  2.5225e-01,  1.6028e-01,\n",
       "           6.5808e-02, -8.2977e-02, -1.3264e-01, -3.6344e-01, -1.9702e-02,\n",
       "          -2.1377e-02, -5.7642e-01, -9.0486e-01,  1.5472e-01, -7.9254e-02,\n",
       "           1.2900e-01, -4.1901e-02,  6.2316e-01,  4.2972e-01,  5.9118e-01,\n",
       "           2.3140e-01,  5.5355e-03,  4.6999e-01,  4.3441e-01, -4.1017e-01,\n",
       "           4.6498e-01,  1.8871e-01, -4.7838e-02,  6.9646e-02, -3.2186e-01,\n",
       "          -9.1404e-01,  2.0856e-02,  3.2947e-01,  4.9930e-01, -1.7114e-01,\n",
       "           4.2813e-01,  2.6662e-01, -1.3998e-01, -1.7031e-01,  3.1347e-01,\n",
       "          -4.8944e-01, -1.5127e-01,  9.5415e-02, -1.4029e-01, -6.5634e-01,\n",
       "          -4.8841e-01,  2.7357e-02, -2.9199e-01,  2.0484e-01, -9.9702e-01,\n",
       "          -2.6273e-01,  5.6883e-01, -2.8623e-02,  9.3732e-02, -2.3345e-01,\n",
       "           7.0240e-02, -1.0750e-01,  1.0928e-01, -1.5161e-01, -8.9822e-01,\n",
       "          -1.4258e-01,  9.6331e-02, -2.2318e-01,  3.2530e-01,  3.7809e-01,\n",
       "          -2.0083e-01, -2.9617e-01,  2.8364e-01, -9.0550e-02, -2.5039e-01,\n",
       "          -1.3545e-01, -3.7063e-01, -2.4607e-02, -1.3576e-01, -1.1638e-01,\n",
       "           6.9800e-02,  4.9085e-01,  4.8805e-01, -7.7597e-01, -5.5824e-01,\n",
       "          -2.2992e-02,  5.0756e-01, -1.6302e-01,  3.7266e-02, -2.2967e-01,\n",
       "          -1.0825e+00, -2.0374e-01, -1.1835e-01, -3.9640e-01, -3.8528e-01,\n",
       "           5.0161e-01, -1.9830e-01, -2.0671e-01,  9.6594e-02, -3.4539e-01,\n",
       "           3.4160e-01,  8.4200e-01, -4.9032e-02, -4.7416e-02, -2.9725e-01,\n",
       "           1.7477e-02, -1.8692e-01,  4.8213e-02, -1.9462e-01, -2.5485e-02,\n",
       "           3.1570e-01,  1.2741e-01, -1.8351e-01, -8.2635e-02, -1.7372e-01,\n",
       "           7.9347e-01, -5.6947e-01,  1.3088e-01,  3.8738e-01, -2.4876e-01,\n",
       "          -1.3259e-01,  4.8066e-01,  1.4439e-01,  1.9491e-01, -1.0024e-01,\n",
       "          -2.6493e-02, -2.8608e-01, -1.8423e-01, -3.1248e-01,  9.6616e-01,\n",
       "           3.5980e-01,  2.2368e-01, -3.0496e-01,  5.7904e-01,  5.6810e-01,\n",
       "          -3.8525e-01,  7.6086e-02,  8.8516e-02,  3.4151e-03, -3.1432e-01,\n",
       "           3.2810e-01,  2.4729e-01, -4.2766e-01, -2.5123e-02,  4.5374e-03,\n",
       "           1.4485e-01, -5.5797e-01,  1.6191e-01,  1.0782e-01,  8.5195e-02,\n",
       "          -1.4724e-03, -2.6111e-01,  5.2888e-01, -2.4531e-01, -2.8912e-01,\n",
       "          -1.0092e-01, -1.2182e-01,  4.1012e-02, -2.4548e-01, -3.6887e-01,\n",
       "          -2.7428e-01,  2.4193e-02,  3.5074e-01,  1.4285e-02,  7.4821e-02,\n",
       "           2.5544e-01, -2.6796e-01, -3.4638e-01,  4.5500e-01, -1.7660e-02,\n",
       "          -1.6810e-01,  4.0768e-01, -2.4890e-01, -1.1639e-01,  1.3881e-01,\n",
       "           6.2926e-02,  1.0677e-01, -5.8094e-02,  3.7998e-01,  1.7175e-01,\n",
       "          -2.8992e-01,  3.6344e-02,  2.1408e-01, -8.0245e-02,  3.1665e-01,\n",
       "           5.1861e-02,  8.2278e-01, -3.7109e-01, -2.7164e-02, -3.8441e-01,\n",
       "          -1.2821e-01, -1.6499e-01,  7.5490e-03,  1.5024e-01, -7.5773e-01,\n",
       "           3.0131e-01,  2.8842e-01,  3.4034e-01,  1.1331e-01, -1.1458e+01,\n",
       "           1.6305e-01,  6.4218e-01, -5.1374e-01,  1.8438e-01,  1.7594e-01,\n",
       "          -1.4040e-01, -3.9794e-01,  4.0875e-02, -3.4785e-01, -2.6394e-01,\n",
       "          -6.5590e-02,  1.9075e-01, -1.2756e-02, -5.5478e-01,  3.3680e-02,\n",
       "          -2.5398e-01, -6.3192e-01,  2.6160e-01, -9.0323e-02,  4.1335e-01,\n",
       "          -3.0876e-01,  1.8731e-01,  2.3832e-01, -3.6034e-01, -2.7614e-02,\n",
       "          -5.1547e-01, -5.5807e-02,  3.7042e-01,  3.2770e-01, -3.5254e-01,\n",
       "           2.2243e-01, -1.8729e-01,  3.8908e-01, -2.6449e-01, -5.2363e-01,\n",
       "           1.3092e-01, -9.6993e-02,  3.7952e-01,  1.1884e-02, -2.4878e-01,\n",
       "          -2.5432e-02, -5.2584e-01,  5.9952e-02,  2.0270e-01,  3.4279e-01,\n",
       "           1.2541e+00,  9.4924e-01,  1.3760e-01, -1.3198e-01,  1.6614e-01,\n",
       "          -7.3513e-01,  2.0677e-01, -6.2811e-01, -8.1585e-02, -6.4376e-02,\n",
       "           8.3128e-01, -4.3261e-01, -3.3743e-01,  7.3012e-02, -8.0493e-02,\n",
       "           3.1300e-01, -7.9471e-02,  2.6202e-01, -3.1186e-01,  2.9341e-03,\n",
       "           6.1375e-01, -1.8722e-02,  3.8063e-01, -4.3721e-01, -6.2672e-02,\n",
       "          -4.2457e-02,  2.9525e-01,  3.7279e-01,  6.2687e-02, -2.5474e-01,\n",
       "          -4.1450e-01,  4.2406e-02,  5.2243e-01,  1.3990e-01,  1.4541e-01,\n",
       "           5.8211e-01, -2.3165e-01, -4.1432e-01, -6.0104e-01, -3.9034e-01,\n",
       "          -7.6601e-03,  1.3684e-01, -3.9867e-01,  6.4750e-01, -4.2253e-01,\n",
       "          -2.7567e-01, -1.6427e-01, -8.8628e-02,  3.8882e-01,  5.0639e-01,\n",
       "          -3.3627e-01,  8.0837e-02, -1.8585e-01, -4.4707e-03, -6.7852e-01,\n",
       "          -3.5458e-01, -8.4538e-02, -2.9987e-01,  2.3231e-01,  1.7305e-01,\n",
       "          -1.0020e-01,  9.0697e-03,  1.2051e-01,  7.3101e-02, -5.4356e-01,\n",
       "           3.0479e-01, -1.0749e-01,  6.8380e-02, -3.9586e-01, -5.1756e-01,\n",
       "           4.0833e-01,  7.0232e-01, -4.2415e-01, -3.3873e-01,  9.8320e-02,\n",
       "           1.3073e-01,  1.8252e-01,  1.2615e-01, -6.4414e-01,  1.1900e-01,\n",
       "          -4.1931e-01, -1.3812e-01,  1.3512e-01,  7.2335e-01,  6.6950e-01,\n",
       "          -5.3703e-02, -3.2049e-01,  2.4632e-01, -5.0640e-02, -2.5765e-01,\n",
       "          -7.0535e-01, -2.9464e-02,  2.1372e-01, -5.3812e-01,  9.7345e-02,\n",
       "          -3.3445e-01,  1.9755e-01, -1.7248e-01,  1.0825e-01, -7.4644e-02,\n",
       "           4.1662e-01,  4.1987e-01,  7.6810e-02,  4.3515e-01, -1.7908e-01,\n",
       "          -2.1990e-01,  1.9709e-01, -1.7846e-01, -1.1703e-01, -1.7569e-01,\n",
       "          -1.7366e-01,  9.7459e-02, -4.0722e-01, -3.6435e-02,  1.7903e-01,\n",
       "          -1.2723e-01,  6.7948e-02, -3.4229e-01,  2.2739e-01,  3.3538e-01,\n",
       "          -1.0467e-01, -3.7966e-01, -2.3884e-01,  4.7521e-02,  7.1481e-02,\n",
       "           3.8731e-01, -2.6195e-01, -1.0102e-01, -7.3811e-01, -2.0636e-01,\n",
       "           3.6787e-02,  1.0410e+00, -3.7965e-02, -3.0778e-01,  3.5156e-01,\n",
       "           3.9051e-01, -2.4726e-01, -3.4583e-01, -1.8699e-01,  7.4840e-01,\n",
       "          -2.3598e-01, -1.5465e-01,  2.2099e-01,  2.5510e-02,  6.2240e-01,\n",
       "           1.3154e-01,  2.2300e-01, -1.8512e-01,  4.7019e-01,  2.1848e-01,\n",
       "          -4.4102e-01, -1.5073e-01, -3.9303e-01, -3.2472e-01,  4.3754e-01,\n",
       "          -6.9776e-02,  2.7454e-01, -1.1790e-01,  8.1553e-02,  1.4253e-02,\n",
       "           5.4206e-02,  1.1645e+00, -2.0185e-01, -2.3960e-01,  4.7684e-01,\n",
       "           2.7079e-01,  3.4262e-01,  3.7917e-02, -1.3477e-01,  9.5921e-02,\n",
       "           4.9705e-01, -1.6706e-01, -4.9624e-01,  1.5664e-01,  3.6272e-01,\n",
       "           5.9108e-01, -8.7613e-01, -3.6622e-01,  2.0509e-01,  2.2296e-01,\n",
       "          -3.2121e-01,  1.9818e-01, -7.6228e-02,  3.2510e-01,  3.6266e-01,\n",
       "          -1.2771e+00,  1.7642e-01,  7.1262e-02,  1.5943e-01, -1.5269e-01,\n",
       "          -6.7715e-01, -2.5385e-01,  1.0595e-01, -6.2916e-02,  4.2076e-01,\n",
       "           4.4789e-01, -7.1509e-01,  1.2452e-01,  3.1511e-01, -3.7691e-01,\n",
       "           1.2320e-01, -2.7571e-01,  1.6643e-01,  5.2170e-01,  1.6129e-01,\n",
       "          -4.2505e-01,  2.5049e-01, -5.0826e-01, -6.1173e-01,  1.4561e-01,\n",
       "           6.9575e-01,  2.1324e-02, -1.1322e-01, -1.1804e-01, -2.9289e-01,\n",
       "          -1.9107e-01, -3.1790e-01, -3.4967e-01, -2.7542e-02,  1.4802e-01,\n",
       "           2.4431e-02, -3.6606e-02,  2.4884e-01,  1.1143e-01, -2.2300e-01,\n",
       "           5.8287e-01,  4.8747e-01,  7.4334e-01, -1.5062e-02,  1.1963e-02,\n",
       "          -2.2207e-01,  2.9970e-01, -4.7196e-02,  4.7552e-01, -3.4433e-01,\n",
       "          -2.0911e-01,  2.2006e-01,  3.6212e-02, -6.2294e-01, -1.0381e-01,\n",
       "          -3.5492e-01,  3.0044e-01,  1.1075e-02, -2.7609e-01,  4.8880e-01,\n",
       "           2.0553e-02,  1.8280e-01, -5.5884e-01,  4.6756e-01,  1.6596e-01,\n",
       "           2.2073e-02,  9.8054e-01,  1.1685e-01,  5.3145e-01,  5.8221e-01,\n",
       "           1.1828e-01, -8.1138e-01, -1.8763e-01, -7.2256e-02,  1.5035e-01,\n",
       "           1.1544e-02,  1.5872e-03,  1.5741e-01,  2.8459e-02,  9.3799e-01,\n",
       "           4.5776e-01, -2.2136e-01, -3.9686e-01, -3.7782e-01,  1.8491e-01,\n",
       "          -8.4548e-02, -3.3508e-01,  2.0442e-01, -2.2187e-01,  1.7388e-02,\n",
       "          -5.8935e-01, -5.5885e-01, -1.2302e-01,  2.7758e-01, -2.6880e-01,\n",
       "           5.4517e-01,  4.6664e-02,  5.2447e-03,  3.4141e-01, -2.9763e-02,\n",
       "           2.1521e-01,  6.7382e-01,  2.2993e-01, -2.9428e-01,  1.8031e-01,\n",
       "           2.1968e-01, -1.3524e-01,  2.0995e-01, -8.6762e-01,  3.6296e-01,\n",
       "          -5.1872e-02,  5.4038e-01,  4.7469e-01]], device='mps:0'),\n",
       " tensor([[-0.1979, -0.6462, -0.8826,  ...,  0.1613,  1.0377,  0.5252],\n",
       "         [-0.2841, -0.9073, -0.8214,  ...,  0.1888,  0.6933,  0.5187]],\n",
       "        device='mps:0'),\n",
       " tensor([[-1.1843e-01, -4.9125e-01, -4.4954e-01, -4.7549e-01, -4.0872e-01,\n",
       "          -1.7667e-01, -2.5509e-01,  3.5023e-01,  5.6414e-02,  7.8048e-01,\n",
       "           1.3201e-01,  4.4899e-01, -9.8497e-02, -4.8542e-01, -5.3704e-01,\n",
       "          -4.4376e-01,  5.8553e-02,  7.7363e-01,  5.1058e-01,  4.1094e-01,\n",
       "           5.3900e-01,  2.0034e-01,  5.4456e-02, -6.5173e-02, -6.1024e-01,\n",
       "           1.0546e-01, -1.7607e-02,  2.4498e-01,  2.0119e-01, -6.7148e-01,\n",
       "          -1.4522e-01,  5.4882e-01, -1.7624e-01,  1.0560e+00,  1.1553e-01,\n",
       "          -1.9609e-01,  2.1077e-01, -7.2339e-01,  8.6133e-01, -2.8613e-01,\n",
       "           6.8612e-01,  1.2632e-01,  3.0012e-01,  2.6015e-02,  5.9929e-01,\n",
       "          -1.6909e-01,  4.0121e-02, -4.2217e-01,  2.8623e-01, -4.1505e-01,\n",
       "          -2.5753e-01,  9.3101e-02,  5.2881e-01,  3.5953e-01, -4.6355e-01,\n",
       "          -2.6639e-01,  9.9449e-02,  1.9461e-01, -4.9407e-01,  1.9014e-01,\n",
       "          -1.2198e-01,  4.2450e-01,  4.1447e-02, -3.2885e-01,  4.2058e-02,\n",
       "           3.1123e-01, -5.9306e-01, -6.2452e-01, -2.2265e-01, -9.1382e-02,\n",
       "          -1.2320e-01,  6.2860e-01, -1.5038e-01, -2.8279e-01,  2.8585e-01,\n",
       "           6.3625e-02,  8.9663e-02,  1.1703e-01, -4.9317e-01, -8.0637e-02,\n",
       "          -5.7839e-02,  3.2798e-01, -2.1525e-01, -2.8452e-01,  3.7490e-02,\n",
       "           1.0552e-01,  6.7334e-01,  8.1529e-02,  1.0869e-04, -3.0614e-01,\n",
       "           5.3542e-01, -1.1140e-01, -4.7955e-01, -4.8801e-01,  3.2239e-01,\n",
       "          -1.8767e-01,  4.2698e-01, -4.9915e-01, -2.2783e-03, -1.9399e-01,\n",
       "           1.5669e-01,  8.6628e-02, -7.5250e-02,  2.8543e-01, -3.5514e-01,\n",
       "          -8.1363e-01,  1.7685e-01, -7.9770e-02, -5.4005e-01, -9.0156e-02,\n",
       "          -4.5764e-01, -1.2212e-01,  6.4277e-02, -2.7974e-01, -1.2378e-02,\n",
       "          -2.1342e-01,  2.5515e-01, -1.0848e-01, -3.2897e-01,  2.3762e-01,\n",
       "           2.0994e-01, -6.0787e-01,  1.2496e-01,  2.0787e-01,  1.2410e-01,\n",
       "          -5.1917e-01,  6.8795e-01, -3.2654e-02,  3.1108e-01,  4.6305e-01,\n",
       "           1.2660e-01,  7.1300e-02, -4.7467e-01,  1.4219e-01, -4.1049e-02,\n",
       "           2.8503e-01,  6.6237e-02, -4.3768e-01,  3.2051e-01,  2.3550e-02,\n",
       "          -1.1658e-01, -9.9193e-02, -2.3060e-01, -1.1763e-01,  5.5177e-02,\n",
       "          -2.8678e-01,  3.5494e-01, -5.5512e-01, -1.5799e-02, -3.3146e-01,\n",
       "          -5.8973e-01, -7.4327e-01,  3.8825e-01, -1.7411e-01,  2.5130e-01,\n",
       "          -2.0379e-01,  2.9492e-01,  3.7698e-02, -2.1000e-01,  4.2616e-01,\n",
       "           7.2926e-01,  1.4223e-03,  1.6985e-01, -3.2607e-01, -2.1011e-01,\n",
       "           1.2847e-01,  2.3687e-04, -2.1526e-01, -5.4019e-01,  1.1055e-01,\n",
       "           2.0863e-01, -3.3273e-01,  3.5968e-01, -5.1097e-02,  2.8188e-01,\n",
       "          -6.1826e-01,  2.1862e-01,  1.6538e-01, -1.3687e-01, -1.4783e-01,\n",
       "          -1.4564e-01, -1.4414e+00, -4.4067e-02,  6.2913e-01, -1.6822e-01,\n",
       "          -6.9555e-01,  4.5542e-01,  2.0639e-01,  1.9344e-01, -5.2174e-02,\n",
       "          -4.6837e-02,  2.2271e-02,  3.5276e-03,  3.6602e-02, -1.4320e-02,\n",
       "           5.4736e-01,  7.9141e-01,  1.5330e-01,  1.2494e-01, -5.1446e-02,\n",
       "          -3.6515e-01, -8.1040e-02,  8.5083e-02,  7.6117e-02,  4.1627e-01,\n",
       "           4.4468e-02, -4.7015e-01,  1.0771e-02, -5.5460e-03, -2.0119e-01,\n",
       "          -1.4629e-01, -1.3392e-01, -5.0249e-01, -6.1996e-03, -6.3792e-02,\n",
       "           7.3909e-01, -7.2855e-02, -2.3897e-01,  4.8215e-01,  6.2015e-02,\n",
       "          -2.8928e-01,  2.7693e-01, -6.2137e-02, -3.7981e-01, -2.1381e-01,\n",
       "           6.4128e-01, -3.2355e-01,  6.6944e-02, -1.3272e-01, -2.5872e-01,\n",
       "           3.1507e-01, -6.9532e-02,  5.1788e-01,  4.0309e-01, -5.0759e-01,\n",
       "          -3.2531e-01, -1.3455e-01, -1.8765e-01,  9.0643e-03, -8.0010e-02,\n",
       "          -6.0228e-01, -5.1392e-01, -4.0992e-01, -2.2038e-02,  1.0987e-01,\n",
       "           2.9754e-01, -2.9714e-01,  6.7753e-01,  3.6299e-01,  1.9725e-01,\n",
       "           2.4780e-02,  4.3675e-02,  1.1994e-01,  5.3441e-02, -3.4287e-01,\n",
       "           6.9167e-02,  1.9886e-01, -6.1421e-01,  4.3722e-01, -2.6436e-01,\n",
       "           1.8432e-01,  7.1850e-02,  1.3532e-01,  3.7301e-01, -1.4101e-01,\n",
       "          -2.0560e-01,  1.8450e-01, -1.2828e-01, -2.7618e-01,  1.9370e-01,\n",
       "          -4.6582e-02,  5.2971e-01, -1.3754e-01, -4.3314e-01, -4.9645e-01,\n",
       "          -7.3085e-01, -2.0760e-01, -4.2804e-01, -7.5591e-02, -7.8481e-01,\n",
       "          -2.9246e-01,  3.2435e-02, -1.9836e-01, -6.3685e-02, -2.9866e-01,\n",
       "           2.3263e-01,  2.9278e-01,  2.9077e-01, -1.7379e-01, -4.1811e-01,\n",
       "          -3.3560e-01,  1.6295e-01, -1.9432e-01, -2.6607e-01,  1.3411e-01,\n",
       "           3.2452e-01,  4.7071e-02, -2.0613e-01,  3.6466e-02, -4.9056e-01,\n",
       "           1.7483e-01, -3.5511e-02,  7.0117e-02, -1.3143e-01, -1.3696e-03,\n",
       "           4.4811e-02,  2.4325e-01,  2.4922e-01, -3.3876e-01, -4.7399e-01,\n",
       "          -1.6807e-01,  3.1060e-01,  8.5682e-03, -3.6262e-01, -7.7696e-02,\n",
       "          -8.9466e-01, -6.5726e-02,  1.9089e-01, -8.8163e-02, -3.4969e-01,\n",
       "          -6.2157e-02,  4.2100e-01, -4.4429e-01,  5.9495e-02, -3.9767e-01,\n",
       "           4.9000e-01,  4.5727e-01,  2.4337e-01,  1.1970e-01, -6.0170e-01,\n",
       "          -4.5312e-02, -9.0227e-01,  5.8616e-03, -1.8296e-01, -8.1327e-02,\n",
       "          -6.4708e-03,  1.1128e-01, -1.9528e-02,  1.9193e-01, -2.5705e-01,\n",
       "           3.1224e-01, -3.3130e-01, -2.7784e-01,  3.0660e-01, -3.4597e-01,\n",
       "          -1.1566e-01,  4.3952e-01, -7.8265e-01, -1.3381e-01,  4.4059e-01,\n",
       "           1.6090e-01, -1.0686e-01, -4.6664e-02, -4.2064e-01,  6.3477e-01,\n",
       "           5.2201e-01,  2.3062e-01, -4.6050e-01,  5.4718e-01,  2.4439e-01,\n",
       "          -3.7726e-01,  2.4953e-01, -6.2328e-01, -6.0188e-01, -2.9783e-01,\n",
       "           2.1497e-01,  2.2826e-01, -6.0660e-01,  3.5390e-03, -2.0624e-01,\n",
       "           3.9608e-01, -1.5528e-01, -7.6873e-02,  1.3661e-01, -5.0682e-02,\n",
       "          -1.0200e-01, -3.5822e-01,  1.6772e-01, -3.9396e-02, -1.1573e-02,\n",
       "          -6.0214e-02,  4.6782e-02,  2.9405e-01, -2.9950e-01, -6.4050e-02,\n",
       "          -1.7604e-01, -1.3440e-01,  6.9773e-01, -6.6083e-02, -5.7879e-02,\n",
       "          -7.1709e-02, -1.1930e-01, -3.9032e-01,  5.5134e-01,  4.0389e-02,\n",
       "          -5.8277e-02,  5.6901e-01, -4.7148e-02, -6.0063e-02,  4.8363e-01,\n",
       "          -3.9280e-01,  1.7847e-01, -2.4921e-01,  4.5169e-01,  1.4454e-01,\n",
       "          -4.8072e-01,  3.2744e-02,  2.6179e-01,  4.0403e-02, -4.8223e-02,\n",
       "          -1.4297e-01,  6.5462e-01, -1.5934e-01, -6.4371e-02, -4.1898e-01,\n",
       "           1.6821e-01, -1.1769e-02,  6.6671e-02, -2.2665e-01, -5.5834e-01,\n",
       "           5.5948e-01,  2.1423e-01,  4.3357e-01,  4.9375e-01, -1.1396e+01,\n",
       "           1.4777e-01,  6.4551e-01, -4.9262e-01,  2.2747e-01, -1.0482e-01,\n",
       "           5.8263e-02, -2.7878e-01,  2.5245e-01, -4.1181e-01, -2.4950e-02,\n",
       "          -2.1924e-02,  7.2583e-02,  7.7419e-02, -6.3761e-02,  8.9130e-02,\n",
       "          -5.0520e-02, -5.9096e-01,  1.0741e-01, -2.9670e-01,  3.1465e-01,\n",
       "          -5.0609e-01,  8.8639e-02,  2.6618e-01,  7.3192e-02, -9.1892e-02,\n",
       "          -5.3813e-03,  1.1847e-01,  5.2263e-01,  3.2941e-01, -9.8777e-02,\n",
       "           6.6251e-02,  6.2720e-02,  3.1605e-01, -6.8059e-02, -4.4587e-01,\n",
       "           1.7839e-01, -4.5551e-01,  3.6690e-02, -7.3401e-01, -2.7023e-01,\n",
       "          -3.6243e-02, -1.6648e-01,  4.3838e-01,  5.6466e-01,  5.6267e-01,\n",
       "           2.6017e-01,  7.1506e-01,  3.1789e-01, -4.2845e-01,  1.0097e-01,\n",
       "          -8.5582e-01,  1.5457e-01, -3.1680e-01, -8.9881e-01, -2.1993e-01,\n",
       "           6.0184e-01, -5.6351e-01,  4.3406e-01, -5.9727e-01, -3.0626e-01,\n",
       "           2.7208e-01,  3.4472e-03, -3.9201e-01, -5.7653e-01,  9.4556e-02,\n",
       "           1.2115e-01, -8.3143e-02,  6.8269e-02, -1.8233e-01, -1.5131e-01,\n",
       "          -1.7911e-01,  5.1450e-01,  1.8307e-01,  2.0498e-01, -2.1435e-01,\n",
       "          -1.3951e-01, -1.0002e-01,  1.8767e+00, -1.1902e-01,  1.1669e-01,\n",
       "           6.9184e-02,  5.5544e-01, -1.4281e-01, -7.7821e-01, -2.3498e-01,\n",
       "          -9.5871e-02,  4.1552e-01,  1.1551e-01,  3.4504e-01, -1.3550e-01,\n",
       "          -7.1638e-01, -1.5671e-01,  3.0726e-01,  2.2647e-01,  1.0958e+00,\n",
       "          -7.0764e-01,  5.2158e-02,  2.1162e-01, -1.4750e-01, -3.1446e-02,\n",
       "           1.8388e-01, -2.4775e-01, -3.3862e-01,  4.7719e-02,  1.8229e-01,\n",
       "           1.6181e-01,  1.3264e-01,  2.4809e-01,  1.9802e-01, -1.4573e-01,\n",
       "           3.4378e-01,  1.4549e-01,  6.9914e-02, -1.5082e-01, -2.1689e-01,\n",
       "           4.5651e-01,  4.6021e-01, -2.1681e-01, -2.2095e-01, -1.5841e-01,\n",
       "           2.1320e-02,  1.9005e-01,  6.2956e-02, -3.1216e-01,  6.0847e-01,\n",
       "          -5.2055e-01, -1.1587e-01, -2.5972e-02, -1.6415e-01,  2.9892e-01,\n",
       "           4.1080e-01, -2.5861e-01, -6.7394e-02, -6.7898e-01, -5.4052e-01,\n",
       "          -7.8129e-01, -5.8475e-02,  2.3322e-01,  6.8942e-02,  2.0690e-01,\n",
       "          -4.9857e-01,  4.7265e-02, -1.9734e-01,  4.4969e-01,  1.6178e-02,\n",
       "           4.1182e-01,  2.3602e-01, -1.6684e-01,  5.4020e-01, -1.9142e-01,\n",
       "          -4.2697e-01, -1.0291e-01, -4.4923e-01, -7.8897e-04, -5.4838e-01,\n",
       "          -3.7745e-01, -1.8653e-01,  1.3104e-01,  2.8906e-02, -3.7324e-02,\n",
       "          -1.8423e-01,  2.4917e-01, -9.6391e-02,  3.4800e-01,  2.1397e-01,\n",
       "           2.5284e-01, -2.6514e-01, -1.5230e-01, -1.3215e-01,  3.0796e-01,\n",
       "          -1.2394e-01,  4.1160e-02, -1.3251e-01, -2.1940e-01, -1.5720e-01,\n",
       "          -4.6176e-02,  9.8240e-01, -3.2307e-02, -4.2742e-01,  2.8003e-01,\n",
       "           9.1249e-01, -2.3737e-01,  6.7229e-02, -6.5526e-02, -1.4996e-01,\n",
       "          -3.5342e-01, -2.1655e-01,  3.9432e-01, -9.9262e-02,  4.8768e-01,\n",
       "           1.4889e-01, -1.2049e-01, -4.2603e-01,  4.9076e-01,  3.8655e-01,\n",
       "          -3.7306e-01, -1.2400e-01, -4.2752e-01, -1.7923e-01,  6.7717e-01,\n",
       "          -3.8744e-02, -1.8415e-01,  6.9159e-02, -1.1170e-01, -2.6273e-01,\n",
       "           4.6625e-01,  6.8166e-01, -2.2159e-01, -1.6276e-01,  1.1109e+00,\n",
       "           9.5206e-03,  8.9293e-02,  5.2950e-01,  4.1857e-01,  2.4586e-01,\n",
       "           4.6988e-01,  5.9000e-02, -3.9153e-01, -1.1854e-01,  3.6253e-02,\n",
       "           3.1178e-01, -8.2427e-01, -4.5042e-01,  1.1416e-02,  1.1277e-01,\n",
       "           1.5427e-01,  6.5869e-01, -2.8953e-01,  3.5052e-01,  6.7410e-01,\n",
       "          -1.0488e+00, -2.1073e-02, -3.7624e-01,  1.2600e-01,  2.6588e-01,\n",
       "          -1.6181e-01,  4.5596e-02, -4.6192e-01, -1.8573e-01,  3.7531e-01,\n",
       "           3.1766e-01, -7.7497e-01, -1.0643e-01,  5.6161e-01, -3.4698e-01,\n",
       "           1.8723e-01, -1.2057e-01,  2.8547e-01,  1.5277e-01,  3.3463e-02,\n",
       "          -1.1250e-01,  2.5535e-01, -4.4077e-03, -3.6677e-01,  2.1675e-01,\n",
       "           5.1481e-01,  1.0485e-01,  9.8100e-02,  2.7660e-01, -1.7148e-01,\n",
       "          -1.4190e-01, -2.8038e-01, -1.7408e-02, -6.3570e-02, -3.9107e-01,\n",
       "           2.6544e-01, -3.1543e-01, -6.2714e-02,  3.3658e-01, -3.9430e-01,\n",
       "           3.1753e-01,  6.8831e-01,  7.9935e-01,  1.6777e-01, -4.1383e-01,\n",
       "           2.9205e-01,  1.8568e-01, -1.1669e-01,  5.8992e-01,  1.4283e-01,\n",
       "          -3.9695e-01,  1.5034e-01,  2.5312e-02, -2.6518e-01,  4.5055e-02,\n",
       "          -1.8812e-01,  9.2188e-02, -1.1799e-01, -5.7053e-01,  2.1846e-01,\n",
       "           8.6715e-02,  1.6197e-01, -5.4280e-01,  1.0276e-01,  4.4631e-01,\n",
       "          -8.3580e-02,  5.0737e-01, -9.5251e-02,  2.9891e-01, -2.6866e-01,\n",
       "          -1.0410e-01, -6.3715e-01, -2.2859e-01,  3.3310e-01, -3.6184e-01,\n",
       "          -5.1146e-01,  3.1482e-03,  1.0657e-01, -1.5551e-01,  4.9831e-01,\n",
       "           7.1390e-01, -3.2386e-01, -1.0535e+00, -3.5296e-01,  7.8013e-02,\n",
       "           7.8945e-02,  1.1154e-01,  4.4472e-01,  5.9865e-02,  2.6502e-01,\n",
       "          -9.0734e-01, -2.4123e-01, -2.5153e-02,  3.0498e-01, -2.7307e-01,\n",
       "           6.1096e-01,  8.9648e-02,  1.0572e-01,  3.1457e-01,  4.7469e-01,\n",
       "          -9.2520e-02,  7.4312e-01, -2.1416e-01, -1.7656e-01,  8.0637e-02,\n",
       "           3.0046e-01, -1.3728e-01,  1.1061e-01, -9.2221e-01,  4.6079e-01,\n",
       "          -8.2631e-02,  5.9883e-01,  4.8924e-01]], device='mps:0'),\n",
       " tensor([[-1.4909e-01, -3.8725e-01, -5.4046e-01, -7.4076e-01, -2.8953e-01,\n",
       "          -2.9537e-01, -8.7437e-01,  3.5888e-01, -8.9489e-02,  2.1202e-01,\n",
       "           1.4615e-01,  1.4133e-01,  3.2720e-01, -2.2380e-01, -3.7354e-01,\n",
       "          -5.3304e-01,  7.5485e-02,  4.4279e-01,  4.3411e-01,  2.9058e-01,\n",
       "           3.3511e-01,  6.5856e-01,  6.5031e-02,  1.7751e-01, -4.6303e-01,\n",
       "           3.9375e-01, -7.2822e-03, -9.7942e-02,  4.8101e-01, -4.9192e-01,\n",
       "          -1.7860e-01,  3.5128e-01, -1.9682e-01,  1.4640e+00, -1.5939e-02,\n",
       "          -2.8745e-01,  3.1636e-01, -3.7043e-01,  1.0141e-01,  7.9319e-02,\n",
       "           6.1295e-01, -3.1130e-01, -2.4562e-03,  1.7343e-01,  2.6626e-01,\n",
       "           4.8815e-01,  1.8825e-01, -3.8243e-01,  6.5666e-01, -6.4806e-01,\n",
       "          -3.2249e-01, -3.7004e-02,  1.7426e-01,  5.7865e-01, -4.9964e-01,\n",
       "          -6.5385e-01, -4.3685e-02, -1.4191e-01, -2.7100e-01,  4.6448e-01,\n",
       "          -1.8960e-01,  5.4213e-01, -1.7718e-01,  1.5874e-01,  5.0124e-01,\n",
       "           5.3343e-01, -4.0419e-01, -7.7499e-01, -3.7645e-01, -1.9102e-01,\n",
       "           2.0253e-01,  3.2754e-01, -3.1351e-01, -3.2908e-02,  3.5884e-01,\n",
       "          -6.6156e-01, -8.8474e-04, -1.2204e-03, -3.6783e-01,  2.5080e-01,\n",
       "          -9.1024e-02,  1.0321e-01,  2.7410e-01,  5.0311e-02, -5.8253e-02,\n",
       "          -1.1946e-01,  5.1225e-01,  7.3092e-02, -1.2405e-02,  2.6166e-01,\n",
       "           2.5456e-01, -1.5874e-01, -3.6116e-01, -4.9533e-01,  3.2551e-02,\n",
       "           2.9895e-01,  2.8447e-01, -2.9973e-01,  1.8339e-01,  2.4451e-02,\n",
       "           3.0515e-01, -1.2171e-01, -1.0464e-01,  1.1635e-01, -4.9076e-01,\n",
       "          -6.8535e-01,  2.8414e-01, -2.4639e-01, -3.9576e-01,  1.6824e-02,\n",
       "          -4.2892e-01, -2.6622e-02,  1.9724e-01, -7.7839e-01, -2.0528e-01,\n",
       "          -7.9354e-02, -9.5013e-02,  1.0455e-01, -1.2049e-01, -2.0063e-01,\n",
       "           3.4183e-03, -1.4271e-01,  3.6565e-01,  2.4701e-01, -4.2730e-02,\n",
       "          -3.2864e-01,  5.5652e-01, -4.5324e-03,  4.3798e-01,  3.8386e-01,\n",
       "           7.0150e-02, -7.7691e-02, -2.1514e-01, -1.9226e-01,  9.2681e-02,\n",
       "           1.4055e-01,  2.2742e-01, -3.0564e-01, -8.7572e-02,  3.6725e-01,\n",
       "           6.1534e-03,  1.1842e-01, -4.1067e-01, -2.8244e-01, -1.5084e-01,\n",
       "          -2.1028e-01, -1.0266e-01, -1.6986e-01, -3.2699e-01, -2.7786e-01,\n",
       "          -6.3450e-01, -6.5014e-01,  6.1042e-01, -1.3163e-02, -3.8226e-02,\n",
       "           4.0997e-02,  5.5215e-01, -3.3914e-01,  2.7744e-02,  2.1633e-01,\n",
       "           1.0016e+00, -2.7012e-01,  1.9423e-01, -4.3225e-01, -3.1009e-01,\n",
       "           2.2082e-01,  7.9273e-02, -1.2290e-01, -1.0478e-01, -5.8679e-02,\n",
       "           1.5834e-01, -2.0948e-01,  3.0316e-01, -4.9165e-02,  6.4976e-02,\n",
       "          -7.0985e-01,  2.7000e-01,  1.2702e-01,  1.3906e-01, -1.5477e-01,\n",
       "          -3.4254e-01, -1.1725e+00,  2.0437e-01, -1.1065e-01,  5.7918e-02,\n",
       "          -3.8552e-01,  2.2103e-01,  4.8722e-01,  1.3417e-01,  2.0848e-01,\n",
       "          -5.0608e-02, -5.8360e-02,  7.9120e-02,  2.2990e-01, -1.4703e-03,\n",
       "           4.1048e-01,  5.9631e-01,  1.9631e-01,  1.7343e-01, -3.5540e-03,\n",
       "          -3.8188e-01,  1.8000e-01,  1.9698e-01, -7.8339e-02, -3.1041e-02,\n",
       "           3.8581e-01,  1.1909e-01, -9.3310e-02,  6.4419e-02, -4.2006e-02,\n",
       "          -9.6829e-02,  4.8152e-02, -6.7869e-01,  1.1805e-01, -1.6008e-01,\n",
       "           4.4952e-01, -3.7438e-01,  5.2035e-02,  3.8302e-01,  9.8268e-02,\n",
       "          -1.6795e-02,  1.6211e-01,  1.2124e-01, -3.1848e-01,  4.6799e-02,\n",
       "           3.7315e-01, -3.4584e-01, -2.0020e-01, -2.2757e-01, -2.7336e-01,\n",
       "           2.6207e-01, -5.2248e-01,  5.0706e-01,  6.0758e-01, -6.8721e-02,\n",
       "          -9.8871e-02, -6.4064e-01,  1.0211e-03, -3.5396e-01,  1.6297e-01,\n",
       "          -4.7742e-01, -5.0154e-01, -4.4583e-01,  5.0678e-02,  3.7215e-01,\n",
       "           3.2945e-01,  5.8393e-03,  7.4296e-01,  3.3262e-01,  1.9303e-01,\n",
       "          -3.4882e-02, -6.5045e-02,  2.3491e-01,  1.1948e-01, -2.4240e-01,\n",
       "           8.5116e-03,  3.7532e-02, -5.0801e-01,  4.3154e-01, -8.6298e-02,\n",
       "           9.2211e-02,  1.6097e-01,  1.9828e-01,  2.8500e-01, -6.2433e-02,\n",
       "          -2.7862e-01,  6.5267e-01,  1.5256e-02, -5.1497e-01,  7.7540e-01,\n",
       "          -4.1193e-01,  3.6159e-01, -1.1035e-01, -3.5939e-01, -4.1219e-01,\n",
       "          -7.7714e-01,  3.1096e-02, -1.0138e-01,  1.9776e-01, -5.9144e-01,\n",
       "           4.7616e-03,  1.1725e-01, -1.1841e-01,  5.5841e-02,  2.2366e-01,\n",
       "           2.1942e-01, -2.1013e-02,  2.7489e-02, -1.0384e-01, -4.5082e-01,\n",
       "           7.8251e-04,  1.5638e-01, -2.5553e-01,  1.7833e-01,  2.8164e-01,\n",
       "          -2.2433e-01, -6.8428e-02,  1.7891e-01, -8.7833e-02, -5.8200e-01,\n",
       "          -8.0460e-03, -2.0327e-01, -1.8531e-01, -2.4313e-01,  1.3408e-02,\n",
       "          -1.6777e-01,  5.2505e-01,  3.2453e-01, -2.4766e-01, -4.2548e-01,\n",
       "          -7.9352e-02,  9.5312e-01, -1.2967e-01, -2.5882e-01, -1.2258e-01,\n",
       "          -9.3864e-01, -1.7607e-01, -1.7201e-02, -2.1731e-01, -4.4016e-01,\n",
       "           9.8431e-02,  3.8238e-01, -1.1019e-01,  7.3192e-02, -2.2230e-01,\n",
       "           1.2468e-01,  3.8016e-01, -8.0226e-02,  2.3874e-01, -3.4631e-01,\n",
       "          -1.3636e-01, -5.5957e-01, -1.4487e-01, -2.2106e-01, -1.6730e-01,\n",
       "           2.4770e-01,  2.5126e-01,  1.2808e-01, -2.1246e-01, -3.4243e-01,\n",
       "           4.8252e-01, -4.3320e-01,  3.9375e-01,  5.4937e-01, -3.7415e-01,\n",
       "          -1.6276e-01,  1.8255e-01, -3.6099e-01, -2.6990e-01,  2.6736e-01,\n",
       "           2.4007e-01, -3.7573e-01, -4.9322e-01, -1.1599e-01,  7.5727e-01,\n",
       "           2.3919e-01,  8.5792e-02, -3.6000e-01,  1.5448e-01,  2.9126e-01,\n",
       "          -6.0147e-01,  3.3063e-01, -2.9838e-01, -1.2681e-01, -3.1545e-01,\n",
       "           1.6815e-01,  1.6393e-01, -1.8340e-01,  3.2142e-01, -1.3031e-01,\n",
       "          -4.1305e-02, -7.9510e-02, -2.8694e-01,  1.5881e-01,  7.3161e-01,\n",
       "           3.6484e-02, -1.5481e-01,  1.9075e-01, -1.3188e-01, -2.3456e-01,\n",
       "          -2.6925e-02, -1.1053e-01,  1.0465e-01, -5.2553e-01, -3.3404e-01,\n",
       "          -1.2270e-01, -2.9141e-01,  6.6590e-01,  2.8797e-01, -2.7719e-01,\n",
       "           4.0167e-01, -1.4251e-01, -3.4218e-01,  6.8706e-01,  1.0277e-02,\n",
       "          -3.6331e-01,  4.0851e-01, -3.8801e-01,  1.0200e-02,  4.8971e-01,\n",
       "          -1.8031e-01,  2.2134e-01,  2.5076e-01,  4.4796e-01,  4.4143e-01,\n",
       "          -4.9559e-01,  1.1775e-01,  5.4268e-02, -3.4106e-01,  3.4993e-01,\n",
       "          -1.8275e-01,  4.9301e-01, -1.7704e-01,  2.2202e-02, -2.2105e-01,\n",
       "           1.5581e-01, -1.1453e-01, -3.2515e-01, -1.7093e-01, -4.5477e-01,\n",
       "           1.1332e+00,  2.2463e-01,  2.6448e-01,  1.8485e-01, -1.1483e+01,\n",
       "           2.3567e-01,  7.7685e-01, -3.0786e-01, -1.1959e-01,  9.7272e-03,\n",
       "          -1.3645e-02, -4.8425e-01,  4.1859e-01, -2.7868e-01, -2.0772e-02,\n",
       "          -3.9494e-01, -1.4911e-01,  2.5746e-01, -1.3164e-01, -2.3898e-02,\n",
       "          -1.1501e-01, -5.9739e-01,  1.3219e-01, -1.4433e-01,  3.5491e-01,\n",
       "          -1.0003e-01,  1.3950e-02,  2.7266e-01, -2.4277e-01, -1.2164e-01,\n",
       "           9.5344e-02, -6.5466e-02,  3.0282e-01,  3.3887e-01, -1.6645e-01,\n",
       "          -1.1948e-01,  1.0823e-01,  8.2848e-02, -1.6634e-01, -7.5323e-01,\n",
       "          -1.6723e-01, -2.0174e-01,  2.8315e-02, -2.3903e-01, -5.7666e-01,\n",
       "           5.2639e-02, -3.4017e-01,  2.2836e-01,  6.8048e-01,  7.6581e-01,\n",
       "           9.0153e-01,  7.1068e-01,  2.8622e-01,  1.4237e-01, -9.1913e-03,\n",
       "          -7.2790e-01, -6.9989e-02, -3.5365e-01,  2.7544e-02, -1.4470e-01,\n",
       "           5.5411e-01, -6.7613e-01,  2.6860e-01, -4.0837e-01, -1.2664e-01,\n",
       "           5.8274e-01,  1.4427e-01,  1.9111e-02, -4.8926e-01, -2.1310e-01,\n",
       "           5.6714e-02, -9.4522e-02,  2.3437e-01, -3.1856e-01, -2.2234e-01,\n",
       "          -8.3443e-02,  5.3229e-01,  7.2447e-02, -6.2213e-01,  1.3398e-01,\n",
       "           1.1471e-01, -1.6080e-01,  1.9294e+00, -1.2947e-01,  2.5232e-01,\n",
       "           1.0253e-01,  2.4563e-02, -8.4341e-01, -3.9301e-01, -2.3967e-02,\n",
       "           1.1547e-01, -1.9127e-01,  4.0477e-01,  5.9815e-01, -1.3853e-01,\n",
       "          -4.3424e-01,  2.5962e-01,  5.5903e-01,  1.0420e-01,  1.1292e+00,\n",
       "          -5.0568e-01, -3.4077e-01,  2.2346e-01,  2.8548e-01, -4.5079e-01,\n",
       "           1.3986e-01, -3.2531e-01, -2.9601e-01,  2.6286e-01, -6.1492e-02,\n",
       "           2.9159e-01, -2.8469e-01, -7.1316e-02,  3.5920e-01, -1.9069e-01,\n",
       "          -7.5760e-02, -3.2200e-01, -7.7770e-02, -3.0488e-01, -1.2544e-01,\n",
       "           9.3310e-02,  8.7209e-01, -3.7571e-01, -4.6089e-01, -5.0828e-01,\n",
       "           1.3303e-01,  2.9897e-01,  2.1843e-01, -4.9436e-01,  2.3564e-01,\n",
       "          -4.5049e-01,  1.4762e-01,  1.5183e-01,  4.2383e-02,  4.3218e-01,\n",
       "           3.3461e-01,  2.2883e-02,  2.1647e-01, -5.2578e-01, -8.1970e-01,\n",
       "          -5.6133e-01, -4.6198e-02,  3.6846e-01, -4.1957e-02, -1.2493e-01,\n",
       "          -2.9224e-03, -2.0093e-01, -1.7242e-01, -1.1362e-01,  1.9834e-01,\n",
       "          -2.6842e-02,  1.5991e-02, -3.8558e-01,  2.5496e-01, -9.9457e-02,\n",
       "          -4.2658e-01,  7.2584e-02, -1.1363e-01, -1.8188e-01, -6.0151e-01,\n",
       "          -6.0703e-01, -1.1065e-01,  3.8596e-01, -4.0194e-01,  7.3806e-02,\n",
       "           1.2945e-01,  1.8810e-01, -7.9130e-02,  9.0609e-01,  4.2808e-01,\n",
       "           2.0657e-02, -4.1389e-01, -1.0571e-01, -1.8442e-01,  1.1878e-01,\n",
       "           3.1107e-01,  3.3152e-01, -5.4552e-01, -1.2939e-01, -4.8812e-01,\n",
       "          -2.8438e-01,  9.1400e-01, -2.3524e-01, -3.9421e-01,  1.1694e-01,\n",
       "           1.1598e-01,  2.6767e-01,  8.3855e-02,  2.2641e-01,  2.8631e-02,\n",
       "          -5.2286e-01, -7.0115e-01, -4.1319e-02, -1.9789e-01,  1.8766e-01,\n",
       "           5.5710e-01, -9.3840e-02, -5.7516e-02,  4.1168e-01,  7.0973e-02,\n",
       "          -1.0427e-01,  8.9174e-02, -4.6646e-01, -5.2083e-02,  5.3159e-01,\n",
       "          -4.3940e-02,  2.5694e-01,  1.9772e-01, -8.0988e-02, -2.3896e-01,\n",
       "           5.1085e-01,  1.0069e+00, -1.6487e-01, -1.2642e-01,  7.8150e-01,\n",
       "          -2.2215e-02, -2.2734e-01,  2.5457e-01,  2.4629e-02,  1.7495e-01,\n",
       "           3.9677e-01, -3.8618e-01, -2.4342e-01, -2.0153e-01,  3.4650e-01,\n",
       "           2.5531e-01, -6.4989e-01, -5.5135e-01, -1.0183e-01,  5.6892e-02,\n",
       "          -8.4229e-02,  3.8447e-01,  4.3420e-03,  2.0017e-01,  1.2854e-01,\n",
       "          -9.9581e-01, -1.0776e-01, -4.6044e-01,  2.9698e-04, -1.5785e-02,\n",
       "          -2.8082e-01, -2.3964e-01, -4.4183e-01,  8.0700e-03,  4.2609e-01,\n",
       "           6.7926e-02, -1.0181e+00, -2.9130e-02,  6.2360e-01, -8.2859e-01,\n",
       "           2.5361e-02, -4.3784e-01,  3.0603e-01, -6.5595e-02,  8.4007e-02,\n",
       "           2.5118e-02,  1.9176e-01, -6.1476e-01, -4.5625e-01,  2.6410e-01,\n",
       "           7.3775e-01, -1.5005e-02,  2.6759e-01, -2.4350e-01, -2.9402e-01,\n",
       "          -4.1112e-01, -4.7959e-01, -1.4782e-03,  1.0971e-01, -1.2660e-01,\n",
       "          -8.1293e-03, -2.2537e-01, -4.1665e-01,  3.9565e-01,  1.2271e-01,\n",
       "           3.6658e-01,  5.5232e-01,  3.4947e-01,  1.2986e-01, -4.3342e-01,\n",
       "           5.4614e-02,  2.5755e-01, -4.8771e-02,  3.8852e-01, -3.3476e-01,\n",
       "          -3.9273e-01,  6.9014e-02, -1.7055e-01, -3.8186e-01,  5.6135e-02,\n",
       "          -3.8845e-01,  2.9961e-01, -1.9409e-01,  1.1884e-01,  4.6161e-01,\n",
       "           2.3983e-01,  1.2382e-01, -4.9320e-01,  4.5861e-01,  5.7099e-01,\n",
       "          -2.0605e-01,  5.3892e-01,  2.0078e-01,  5.6958e-01, -5.1132e-01,\n",
       "           1.7044e-02, -8.3500e-01, -2.2189e-02, -6.2679e-03, -4.2054e-01,\n",
       "          -1.9253e-01, -3.3916e-01,  3.0638e-01,  5.2977e-02,  5.5913e-01,\n",
       "           4.0959e-01, -3.7785e-02, -8.2725e-01, -8.0325e-02,  2.4043e-01,\n",
       "           1.0740e-01, -1.8259e-01,  1.4515e-01,  9.2354e-02,  3.4760e-01,\n",
       "          -6.5628e-01, -5.7824e-01, -1.8649e-01,  1.5759e-01, -2.9795e-02,\n",
       "           8.9344e-01, -1.0358e-01, -2.0966e-01, -7.7817e-02,  4.3588e-01,\n",
       "          -8.8625e-02,  8.6523e-01,  4.4504e-02, -2.4255e-01,  7.5842e-02,\n",
       "           1.1282e-01, -2.2094e-02,  6.1821e-01, -4.8180e-01,  2.7568e-01,\n",
       "           2.0659e-01,  5.0871e-01,  4.7371e-01]], device='mps:0'),\n",
       " tensor([[-6.1750e-01, -4.2829e-01, -2.3934e-01, -9.4116e-01, -3.4548e-01,\n",
       "          -6.5496e-02, -6.3874e-01,  9.7828e-01, -9.5892e-02,  4.0096e-01,\n",
       "           1.4919e-01, -5.1603e-03,  2.0980e-01,  1.3680e-01, -6.9506e-01,\n",
       "           1.6740e-01,  4.6094e-01,  2.1159e-01,  3.5951e-01, -4.4329e-02,\n",
       "           7.9832e-01,  3.4245e-01,  1.8019e-01, -2.5205e-01, -4.5876e-01,\n",
       "           5.8994e-01, -2.4628e-01, -9.2014e-02,  2.0509e-01, -8.1949e-01,\n",
       "          -1.3586e-01, -4.9757e-02, -3.6281e-01,  8.7602e-01,  1.1948e-01,\n",
       "          -2.7310e-01, -1.5276e-01, -2.7996e-01,  2.3494e-01, -6.3637e-02,\n",
       "           3.7814e-01,  2.6592e-02, -2.9347e-01,  3.1386e-01, -1.6421e-01,\n",
       "          -1.4057e-01,  1.4047e-02, -7.8429e-01,  1.3666e-02,  1.8955e-01,\n",
       "           5.8851e-02, -4.7205e-02,  1.9231e-01,  3.7030e-01, -1.0617e+00,\n",
       "          -2.6348e-01,  2.7603e-01,  2.9849e-01, -1.7932e-01, -5.7784e-02,\n",
       "          -1.9989e-01,  9.0201e-01,  1.7949e-01, -2.7822e-01,  1.7824e-01,\n",
       "          -4.4899e-01, -5.4601e-01, -6.0781e-01, -1.0682e-01, -1.1967e-01,\n",
       "          -1.0002e-01,  9.7528e-02,  1.4009e-01, -2.9385e-01,  2.3170e-01,\n",
       "           2.1813e-01,  4.7024e-02, -2.7652e-02, -4.3158e-01, -2.6546e-01,\n",
       "           3.6947e-01,  3.4425e-02, -1.4730e-02,  2.0326e-01, -5.8744e-01,\n",
       "          -2.0226e-01,  3.7959e-01,  1.5703e-01,  2.5914e-01,  2.2662e-03,\n",
       "           5.9967e-01,  1.2024e-02,  8.0339e-02, -4.5369e-01,  5.8427e-01,\n",
       "          -5.4955e-01,  1.0329e-01, -1.6007e-01,  4.9532e-02, -5.2088e-01,\n",
       "          -9.8735e-02,  2.4494e-01, -1.8085e-01,  1.6053e-01, -4.6592e-01,\n",
       "          -3.2983e-02,  2.4514e-01,  3.0468e-01, -4.9183e-01,  4.2685e-02,\n",
       "          -6.5134e-03, -2.9791e-01,  7.3694e-02, -6.6407e-01, -8.5795e-01,\n",
       "           5.8354e-01,  1.8428e-02, -4.5240e-01, -2.3930e-01,  7.4443e-01,\n",
       "           2.9738e-01, -2.3201e-01, -5.3072e-02, -3.5548e-01,  2.0259e-01,\n",
       "          -1.8518e-01,  2.5689e-01, -2.7018e-01,  3.8439e-01,  5.5026e-01,\n",
       "           3.1581e-01,  2.8489e-01, -1.8327e-01,  2.7637e-01,  1.4465e-01,\n",
       "           3.6947e-02, -6.6614e-02, -1.5929e-01,  1.5180e-01,  1.2041e-01,\n",
       "          -2.9580e-01, -3.2999e-01,  5.4746e-03,  1.1182e-01, -3.3279e-01,\n",
       "          -1.0202e-01,  1.8680e-01,  3.4808e-01, -1.4092e-01, -2.5403e-01,\n",
       "          -7.8984e-01, -4.2328e-01,  1.0217e-01, -1.8560e-01, -1.7135e-01,\n",
       "          -4.1911e-01,  3.8121e-01, -6.6660e-02,  4.7638e-01,  3.4189e-01,\n",
       "           6.9978e-01, -1.5311e-01,  5.9776e-02, -5.6412e-01,  3.5197e-01,\n",
       "           3.8270e-02, -6.1110e-01, -1.7247e-01,  1.5680e-01, -4.1813e-04,\n",
       "           6.8195e-01,  7.0979e-02,  4.0124e-01,  1.2070e-01, -3.2553e-01,\n",
       "           8.2089e-03,  1.9890e-01, -1.2864e-01, -1.4297e-01,  7.3766e-01,\n",
       "          -5.5188e-01, -2.3029e+00,  1.0721e-01, -3.5844e-01, -3.6529e-01,\n",
       "          -1.8135e-01,  3.7255e-01,  3.3772e-02,  4.9966e-01,  1.5486e-01,\n",
       "           3.7598e-02,  1.6729e-01, -2.8025e-02,  3.6851e-01, -8.4399e-02,\n",
       "           2.6751e-01,  3.3341e-01,  1.2988e-01, -2.3298e-01, -1.0956e-01,\n",
       "          -3.9227e-01,  2.4031e-01, -9.8916e-02,  8.6010e-01,  2.3260e-01,\n",
       "           2.0943e-01, -4.3657e-03, -4.7262e-02,  4.8964e-01,  4.3099e-01,\n",
       "          -6.6541e-02, -1.6585e-01, -4.9522e-01,  4.6269e-03, -4.1279e-02,\n",
       "          -1.2963e-02, -4.5611e-01,  1.3753e-01,  4.2841e-01, -1.2416e-01,\n",
       "          -1.8354e-01,  1.5525e-01, -4.7715e-02, -3.1517e-01, -2.6492e-02,\n",
       "           6.8453e-01,  1.9080e-01, -2.2994e-01,  1.2377e-01, -1.1896e-01,\n",
       "           3.7006e-02, -4.6975e-01,  7.0044e-01,  6.1147e-01,  1.5585e-01,\n",
       "           1.6214e-02, -6.9943e-01,  2.3699e-01, -3.1574e-01, -4.7647e-03,\n",
       "          -1.5027e-01, -1.5731e-01, -4.2986e-01, -5.7285e-02, -5.5854e-02,\n",
       "           4.0115e-01,  7.9943e-02,  4.6653e-01,  1.1837e+00,  2.7973e-01,\n",
       "           1.9184e-02,  5.6097e-01,  2.0801e-01,  4.2492e-01, -4.0180e-01,\n",
       "           3.3082e-01,  8.8825e-02, -6.3896e-01,  2.7188e-01, -4.1593e-01,\n",
       "           4.4963e-02, -4.1432e-02, -3.7926e-01,  1.4208e-01, -2.1276e-02,\n",
       "           3.4237e-01,  3.3517e-01,  1.5186e-01,  1.6838e-01,  2.9411e-01,\n",
       "          -3.1322e-01,  5.9715e-01, -6.0809e-02, -1.1127e-01, -6.3492e-01,\n",
       "          -7.6732e-01,  9.5206e-02,  4.4743e-01, -1.2880e-01, -6.1279e-01,\n",
       "           1.0462e-01, -2.3866e-01, -1.2257e-01, -3.5064e-02,  2.1191e-01,\n",
       "           2.7455e-01,  5.0687e-02, -9.5436e-02,  6.4791e-02, -4.6950e-01,\n",
       "          -4.9277e-01,  4.9140e-02, -1.5758e-01, -1.3063e-01,  5.3918e-01,\n",
       "          -7.9415e-02,  1.1575e-01,  5.7221e-01,  1.9799e-02, -4.5976e-01,\n",
       "          -2.5640e-01,  1.4802e-01,  2.8667e-01, -7.2082e-02,  2.1981e-01,\n",
       "           3.4119e-01,  2.6467e-02,  5.1649e-01,  1.4823e-01, -2.1046e-01,\n",
       "          -2.7239e-01,  5.1173e-01, -1.5755e-01,  6.1403e-02, -1.2527e-01,\n",
       "          -8.7207e-01, -5.6150e-02, -7.8413e-02, -1.6833e-01, -5.4440e-01,\n",
       "           2.6566e-01,  1.4351e-01,  1.0561e-01,  2.2350e-01, -1.5984e-01,\n",
       "           7.0123e-01,  6.5285e-01, -3.3100e-01, -1.6943e-01, -4.0104e-01,\n",
       "           9.3922e-02, -6.8374e-02, -2.9329e-01, -1.4009e-02,  9.5368e-03,\n",
       "          -2.2765e-01, -1.2516e-01, -1.0805e-01,  2.8921e-01, -1.8876e-01,\n",
       "           5.2862e-01,  1.2189e-01,  8.3649e-02,  4.3873e-01,  2.0612e-02,\n",
       "           1.3071e-01,  3.6515e-01, -4.2117e-01,  1.2870e-01,  4.0424e-01,\n",
       "           4.0578e-01, -1.0551e+00, -8.3291e-01, -1.4936e-01,  3.6111e-01,\n",
       "           2.2860e-01,  2.5189e-01,  1.6999e-02, -7.2215e-03,  8.9455e-02,\n",
       "          -2.3402e-01,  3.5135e-01, -6.6415e-02, -5.7412e-01,  3.0561e-01,\n",
       "           2.3567e-01,  5.3263e-01, -2.7377e-01, -5.0579e-01, -6.5233e-02,\n",
       "           2.4616e-01, -3.6381e-01,  1.1392e-01,  2.3480e-01,  4.5466e-01,\n",
       "          -3.6015e-01, -5.6164e-01, -7.0282e-02, -7.3986e-01, -1.3578e-01,\n",
       "           3.2483e-01, -2.1437e-01,  2.7178e-02, -2.4758e-01,  1.5231e-01,\n",
       "          -1.7925e-02, -1.8879e-01,  6.0757e-01,  2.0522e-01, -2.5855e-01,\n",
       "           1.1981e-01, -2.3892e-01, -1.9475e-01,  5.0091e-01, -5.0226e-01,\n",
       "          -3.0348e-01, -5.0960e-02, -1.5349e-01, -3.2418e-01,  3.2181e-01,\n",
       "          -4.1389e-01, -1.4861e-01,  1.6748e-01, -6.4736e-02, -1.3068e-01,\n",
       "          -3.5233e-01,  1.4268e-01, -1.2707e-01, -4.9883e-02, -1.7654e-02,\n",
       "          -1.3353e-01,  6.0900e-02,  1.3308e-01,  6.4538e-02,  1.4388e-01,\n",
       "           5.3696e-01, -1.1464e-01,  2.7582e-03, -2.5900e-01, -2.2469e-01,\n",
       "           5.2498e-01, -9.4973e-02, -3.3117e-01,  5.4928e-01, -1.1156e+01,\n",
       "           1.0386e-01,  7.2631e-01, -5.5808e-01,  5.5128e-02, -2.8539e-01,\n",
       "           2.1085e-01, -3.0009e-01,  7.9101e-02,  2.4967e-01, -2.8460e-01,\n",
       "          -1.2277e-01, -2.7364e-01, -2.3193e-01, -4.0872e-01, -3.9742e-01,\n",
       "          -4.7229e-01, -5.6789e-01,  1.1772e-01, -1.7663e-01, -1.0996e-01,\n",
       "          -5.7036e-01, -1.9379e-01,  2.9696e-01, -4.9559e-03, -5.1069e-01,\n",
       "           5.9541e-02, -2.2948e-01,  2.6399e-01,  4.6126e-02,  2.2045e-01,\n",
       "          -1.9949e-02,  1.4108e-01,  2.0669e-01, -4.4946e-02, -4.5100e-01,\n",
       "          -4.4853e-01,  2.2129e-01, -1.5955e-01, -2.2267e-01, -5.4017e-02,\n",
       "          -3.3838e-01, -2.7689e-01,  1.3447e-01,  3.7594e-01,  6.2596e-01,\n",
       "           1.1595e+00,  5.6208e-01,  2.9245e-01, -1.0625e-01, -8.5438e-02,\n",
       "          -1.0762e+00,  5.5900e-01, -3.7318e-01,  8.1787e-02, -2.5119e-01,\n",
       "           6.6201e-01, -4.9906e-01,  5.4685e-01, -4.0754e-01, -2.5500e-01,\n",
       "           3.7179e-01,  3.0194e-01,  7.1104e-03,  2.8239e-02, -2.7432e-01,\n",
       "           4.0257e-01, -2.6094e-01, -3.6915e-02, -3.7819e-01,  4.0081e-01,\n",
       "          -3.9472e-01,  5.0770e-01, -5.6534e-02,  1.2945e-01,  1.8567e-01,\n",
       "          -3.5001e-01,  1.6610e-01,  1.0337e+00,  1.7414e-01,  4.1071e-01,\n",
       "           1.7420e-01,  5.1160e-01,  5.3064e-02, -8.8958e-01,  1.3691e-01,\n",
       "           2.0502e-01,  2.0789e-01, -6.7096e-02,  5.7131e-01, -2.5517e-02,\n",
       "          -6.5181e-01, -2.3094e-01, -3.2169e-01,  4.3108e-01,  6.5352e-01,\n",
       "          -2.2597e-01, -4.4343e-01,  2.3894e-01, -7.9776e-03, -5.3395e-01,\n",
       "           9.0074e-02, -2.2320e-01, -5.2792e-01,  8.7275e-02, -5.8914e-01,\n",
       "           1.4985e-01, -1.2428e-01,  5.4238e-02,  6.6887e-01, -4.0159e-01,\n",
       "          -1.6113e-01,  3.6602e-02, -4.4532e-02, -4.1270e-01, -4.9520e-01,\n",
       "          -7.7083e-02,  4.1073e-01, -1.9776e-01, -2.7976e-01,  7.0636e-02,\n",
       "          -3.0673e-01,  2.2584e-01, -1.7377e-01, -6.4935e-01,  3.2879e-01,\n",
       "          -6.0415e-01,  2.0578e-01,  2.5360e-01, -2.2507e-01,  4.9017e-01,\n",
       "           1.1337e-01, -8.1817e-02,  1.9397e-01, -9.9369e-02, -7.8554e-01,\n",
       "          -3.2573e-01, -2.7299e-02,  3.2662e-01, -4.2844e-01,  1.5776e-01,\n",
       "           1.5022e-01, -4.5752e-02, -2.2574e-01,  2.7904e-01,  1.0174e-01,\n",
       "           3.7762e-01,  3.9846e-01, -5.5006e-01,  1.1727e-01, -2.9867e-01,\n",
       "          -2.9262e-02, -6.1174e-02,  1.7952e-01,  2.3654e-01, -7.0764e-01,\n",
       "          -1.5224e-01, -2.6434e-01, -3.0965e-01, -2.1650e-01, -2.5120e-01,\n",
       "           2.0680e-01,  1.4038e-01, -1.5897e-02, -1.1308e-01,  5.8077e-01,\n",
       "           4.0814e-02, -1.1167e-01, -4.5997e-01,  3.1443e-01,  4.3178e-01,\n",
       "           2.1794e-01,  7.5875e-02, -1.2830e-01, -5.5532e-01, -1.1922e-02,\n",
       "          -4.3644e-01,  9.1081e-01, -7.3906e-01, -7.1970e-01,  6.6086e-02,\n",
       "           1.4951e-02, -8.8259e-04, -2.3450e-01, -5.0265e-01,  8.1298e-02,\n",
       "          -2.1017e-01, -1.4891e-01,  6.1550e-03, -1.0607e-02,  1.6299e-01,\n",
       "           4.0742e-01,  3.6838e-01,  1.5951e-01, -3.3938e-01, -4.4081e-01,\n",
       "          -6.0244e-01,  1.4567e-02, -1.6795e-01,  2.9330e-01, -2.6752e-01,\n",
       "          -3.2661e-01,  1.6043e-02, -9.1892e-03, -2.2272e-01,  5.2876e-01,\n",
       "           1.2427e-01,  1.6396e+00, -3.7287e-01, -4.4551e-01,  2.9018e-01,\n",
       "           7.2306e-02, -2.7073e-02, -2.8130e-01,  2.5074e-01,  9.5447e-02,\n",
       "           4.0194e-02, -3.6304e-01,  1.5023e-02, -2.0558e-02,  3.2773e-01,\n",
       "           3.2056e-01, -1.0703e+00, -2.0238e-01, -1.7993e-01,  5.5664e-01,\n",
       "           3.2388e-02,  6.3185e-01, -2.2305e-01,  3.0577e-01,  4.6106e-01,\n",
       "          -8.6667e-01, -2.3000e-01, -1.6129e-01,  2.7295e-01, -1.0583e-01,\n",
       "          -9.8735e-02, -1.6178e-01, -3.7165e-02, -1.6936e-01,  3.9538e-01,\n",
       "           7.4751e-01, -7.8426e-01,  7.4081e-01,  7.0467e-01, -9.8234e-02,\n",
       "           5.6815e-01,  9.4187e-02,  3.2641e-01, -1.4659e-01,  2.2669e-02,\n",
       "           2.6200e-01,  1.8542e-01, -6.3165e-01, -2.3962e-01,  2.4473e-01,\n",
       "           4.0158e-01, -1.1943e-01,  1.2108e-01,  7.9292e-02,  3.0960e-02,\n",
       "          -3.7691e-01, -1.0145e-01, -2.4552e-01, -3.9904e-01,  1.1356e-01,\n",
       "           1.4603e-02, -1.5001e-01, -8.6997e-02,  4.7709e-01, -1.7580e-01,\n",
       "           7.5067e-01,  4.2910e-01,  3.7145e-01, -3.5103e-01,  8.9950e-04,\n",
       "          -5.7565e-02,  6.0671e-01, -3.1670e-01,  7.7385e-01,  6.3337e-02,\n",
       "          -2.3140e-01,  2.7251e-01,  6.0547e-03,  1.7829e-01,  2.9892e-01,\n",
       "          -2.3519e-01,  1.2702e-01, -3.0718e-01, -2.3711e-01,  2.8330e-01,\n",
       "          -1.8640e-01,  1.1542e-01, -7.2270e-01, -6.3602e-03,  3.9857e-01,\n",
       "          -3.2051e-01,  4.9873e-01, -2.3074e-01,  6.1026e-01, -4.3509e-01,\n",
       "           3.2669e-02, -4.7470e-01, -4.3275e-01, -2.8508e-01,  5.1854e-02,\n",
       "          -8.6027e-01, -1.9453e-01,  1.7684e-01,  3.7405e-02,  2.1707e-01,\n",
       "           7.5155e-01, -3.3794e-01, -2.4771e-01, -2.0386e-01,  4.0190e-02,\n",
       "           1.8299e-01, -8.2071e-02, -3.5586e-01,  2.1161e-01, -3.5607e-01,\n",
       "          -1.0520e+00, -2.3117e-01,  8.9990e-02,  3.8419e-01, -3.5853e-01,\n",
       "           1.1570e+00,  2.5132e-02, -2.0226e-01,  6.7253e-01,  2.0808e-01,\n",
       "           6.4157e-01,  6.0323e-01,  1.3628e-03, -4.1259e-01, -5.4109e-02,\n",
       "          -1.5877e-01, -3.2881e-01,  1.5757e-01, -1.8346e-01,  5.0556e-01,\n",
       "           5.4802e-01, -2.0806e-02,  5.4756e-01]], device='mps:0'),\n",
       " tensor([[-0.4654, -0.4115, -0.4370,  ...,  0.1625,  0.4338,  0.5908],\n",
       "         [-0.0029, -0.4260, -0.6820,  ...,  0.1889,  0.1934,  0.5362],\n",
       "         [-0.2434, -0.3725, -0.3338,  ...,  0.2009,  0.3829,  0.4365],\n",
       "         [-0.1134, -0.5452, -0.4614,  ...,  0.3774, -0.1768,  0.8147]],\n",
       "        device='mps:0'),\n",
       " tensor([[ 3.7923e-02, -4.3611e-01, -7.6214e-01, -7.5284e-01, -1.9860e-01,\n",
       "          -2.6072e-01, -7.5546e-01,  3.7519e-01,  3.1617e-01,  4.6573e-01,\n",
       "           2.6943e-01,  1.4835e-03,  1.8521e-02, -1.6920e-01, -9.0926e-01,\n",
       "          -1.1284e-01,  6.9604e-02,  4.0457e-01,  2.0448e-01,  9.1098e-02,\n",
       "           7.1824e-01,  6.3230e-01,  4.6756e-02,  1.2618e-01, -3.4998e-01,\n",
       "           1.0459e-01, -1.3590e-02,  1.7764e-01, -2.0423e-02, -4.4196e-01,\n",
       "           5.7088e-02,  5.1977e-01, -3.4201e-01,  1.1198e+00, -2.4976e-01,\n",
       "          -9.6382e-02,  1.8783e-01, -4.7008e-01,  3.1466e-01, -2.4829e-01,\n",
       "           5.7376e-01,  5.7908e-02, -3.1357e-02,  3.1378e-01,  1.9063e-01,\n",
       "           1.2608e-01,  1.5026e-01, -5.3171e-01,  6.0478e-01, -5.1145e-01,\n",
       "          -4.1852e-01, -2.3001e-01,  2.5871e-01,  2.2115e-01, -5.7945e-01,\n",
       "          -1.9485e-01,  2.9961e-01,  3.9516e-01, -4.1993e-01,  1.6155e-01,\n",
       "          -2.0110e-01,  4.4012e-01, -3.4335e-02,  2.1163e-01,  1.4907e-01,\n",
       "           2.3977e-01, -2.7685e-01, -5.8018e-01, -3.6940e-01, -4.2172e-02,\n",
       "           9.3771e-02,  3.2786e-01,  1.0584e-01, -3.1260e-01,  4.1237e-01,\n",
       "          -3.7674e-01,  3.5155e-01, -2.0485e-01, -5.1273e-01,  2.1922e-01,\n",
       "          -2.1033e-02,  3.8033e-01, -5.5320e-02,  5.7552e-03,  3.1522e-01,\n",
       "          -1.5474e-01,  8.8206e-02,  1.6952e-01,  2.1390e-01,  1.0209e-01,\n",
       "           3.2871e-01,  1.5471e-01,  8.8847e-02, -4.6815e-02,  2.5371e-01,\n",
       "           7.7384e-02,  3.7505e-01, -2.7506e-01, -6.3675e-02, -1.9222e-01,\n",
       "          -9.4676e-02,  2.1755e-01,  1.6623e-02,  2.3029e-01, -4.6834e-01,\n",
       "          -5.4439e-01,  1.2503e-01, -2.4460e-01, -7.7261e-01, -1.7396e-01,\n",
       "          -2.1604e-01, -5.3068e-02,  5.1585e-01, -5.4854e-01, -7.1604e-02,\n",
       "           4.4683e-02, -3.3665e-01,  3.8205e-02, -1.7211e-01,  2.6705e-01,\n",
       "           1.9923e-01, -3.1413e-01,  1.7681e-01, -8.7227e-02,  2.1182e-01,\n",
       "          -7.7407e-02,  4.1608e-01, -3.1950e-01,  1.5175e-01,  4.3941e-01,\n",
       "           3.1131e-01, -2.8902e-01, -4.7346e-01,  8.7926e-04,  1.2004e-01,\n",
       "           1.6012e-01,  1.0513e-01, -2.4886e-01, -2.9099e-01,  6.1067e-04,\n",
       "           2.0605e-01, -7.3251e-02, -6.0063e-02, -1.7461e-01, -1.3321e-01,\n",
       "          -3.9104e-01, -5.0333e-03,  1.0543e-01, -2.8710e-01, -3.7997e-01,\n",
       "          -5.4052e-01, -5.5755e-01,  3.7261e-01, -2.6150e-01,  7.0218e-03,\n",
       "          -9.1179e-02, -4.7261e-03, -8.8000e-02, -1.0506e-01,  3.3782e-01,\n",
       "           6.7254e-01, -2.7502e-01,  1.1765e-01, -6.4263e-01, -6.6616e-02,\n",
       "           7.9276e-02,  2.0668e-01, -6.3324e-02, -3.7254e-01,  5.2503e-02,\n",
       "           2.9702e-01, -1.3327e-01,  1.8631e-02, -5.5228e-01, -1.4568e-01,\n",
       "          -7.4331e-01,  4.3149e-01,  2.5933e-01, -5.6096e-02,  2.6362e-01,\n",
       "           4.9566e-01, -1.2484e+00,  2.5880e-01, -4.8250e-01,  1.8582e-01,\n",
       "          -6.2512e-02,  5.6903e-01,  8.8991e-02,  2.3655e-01, -7.0763e-02,\n",
       "           2.5123e-01, -1.1826e-01,  2.2874e-01,  1.8117e-01,  1.0290e-01,\n",
       "           3.2403e-01,  4.0686e-01,  1.2542e-01, -1.1127e-01,  2.5669e-01,\n",
       "          -6.9330e-01,  3.1509e-01,  1.3744e-01, -2.5934e-01,  1.3856e-01,\n",
       "           1.8288e-01, -2.8067e-01,  1.4280e-01, -6.4271e-02,  1.1337e-01,\n",
       "          -2.3597e-01, -2.9806e-01, -4.5596e-01, -1.4385e-01, -4.9193e-01,\n",
       "           5.3967e-01, -1.4097e-01, -2.1508e-01,  6.0695e-01, -1.2381e-02,\n",
       "          -1.2098e-01,  1.3548e-01, -4.4421e-02, -2.7689e-01, -1.7927e-01,\n",
       "           5.0617e-01, -5.8945e-01,  9.7126e-02, -3.3835e-01, -2.9514e-01,\n",
       "           1.8175e-03, -1.1832e+00,  4.1795e-01,  1.6000e-01,  9.9065e-02,\n",
       "          -2.3198e-01, -3.4235e-01,  1.0242e-01, -5.0152e-01, -3.1484e-02,\n",
       "          -5.2649e-01, -3.8505e-01, -8.9331e-01,  2.0804e-01,  8.8172e-02,\n",
       "           3.7361e-02, -6.1479e-02,  7.8332e-01,  4.1371e-01,  5.4204e-01,\n",
       "          -6.2437e-02, -1.1366e-01,  6.7124e-01, -1.6603e-01, -3.7851e-01,\n",
       "           2.8354e-01,  1.4447e-02, -1.7763e-01,  2.4114e-01, -3.2092e-01,\n",
       "          -4.4610e-01, -1.4971e-01,  5.6130e-01,  5.1369e-01, -4.5312e-01,\n",
       "           3.6260e-01, -5.0946e-03, -2.9777e-02, -3.1114e-01,  8.3768e-01,\n",
       "          -1.6106e-01, -1.5852e-01, -8.4657e-02, -5.1505e-01, -5.8612e-01,\n",
       "          -3.8940e-01,  1.5463e-01, -3.1320e-01,  8.7850e-02, -8.8093e-01,\n",
       "          -4.7226e-01,  3.0292e-01,  4.7989e-02,  1.6623e-01, -1.3558e-01,\n",
       "           1.9639e-01, -1.5975e-01,  7.3908e-02, -2.5253e-01, -5.5078e-01,\n",
       "           3.9459e-02,  2.6510e-01, -4.3011e-01,  3.5644e-01,  4.5077e-01,\n",
       "          -3.5331e-01,  4.2655e-01,  1.0385e-01, -7.3887e-02, -3.2297e-01,\n",
       "          -3.1377e-02, -1.0765e-01, -3.3226e-01, -1.2769e-01,  3.7787e-01,\n",
       "           1.2615e-01,  2.8596e-01,  6.5538e-01, -2.7259e-01, -5.8840e-01,\n",
       "           1.0994e-01,  9.4669e-01, -3.3915e-01, -1.3949e-01, -1.9725e-01,\n",
       "          -8.3588e-01, -1.9787e-01, -2.6507e-01, -4.8360e-01, -6.3116e-01,\n",
       "           6.2507e-03, -2.1442e-01, -6.7811e-02, -2.1877e-01, -2.9781e-01,\n",
       "           1.8872e-02,  4.4500e-01,  2.8930e-01,  1.5799e-03, -2.5672e-01,\n",
       "          -2.1855e-01, -2.5672e-02, -4.9185e-01, -3.1099e-01, -1.1286e-01,\n",
       "           3.7843e-03,  1.3719e-01, -1.2531e-01, -3.0467e-01, -1.3630e-01,\n",
       "           3.9004e-01, -4.6271e-01, -5.0591e-01,  1.2080e-01, -2.0992e-01,\n",
       "          -2.3036e-01,  4.2234e-02, -1.0551e-01, -1.4753e-01,  3.8281e-01,\n",
       "           1.4741e-01, -1.4379e-01, -3.4034e-01, -3.2559e-01,  6.1525e-01,\n",
       "           4.2776e-01,  1.9837e-01, -5.4554e-01,  7.0769e-01,  3.5635e-01,\n",
       "          -6.1493e-01,  1.0271e-01,  4.7722e-02, -1.0939e-01, -1.1030e-01,\n",
       "           4.1740e-01,  2.7791e-01, -3.1982e-01,  1.6623e-01,  8.3533e-02,\n",
       "           1.0421e-01, -6.8040e-02, -1.5252e-01,  1.6652e-01, -2.7165e-01,\n",
       "           8.4182e-02, -4.8727e-01,  3.1618e-01,  1.2979e-01, -2.2273e-01,\n",
       "           1.4174e-01,  9.3531e-02, -1.3514e-01, -2.1364e-01, -2.9439e-01,\n",
       "          -1.1088e-01, -4.7049e-02,  5.4489e-01, -1.3587e-01,  1.1494e-02,\n",
       "           1.0938e-01, -1.8909e-01, -2.2728e-01, -1.9277e-02,  1.3397e-01,\n",
       "          -2.5197e-01,  4.9121e-01, -7.4253e-02,  2.3427e-01,  5.2237e-01,\n",
       "          -1.8630e-01, -4.8421e-02,  8.4122e-02,  1.2732e-01, -1.0255e-01,\n",
       "          -3.5482e-01,  6.3412e-02,  3.6660e-01, -3.7085e-01,  1.1304e-01,\n",
       "          -2.7920e-01,  4.4173e-01, -2.2256e-01,  8.5264e-02, -1.0308e-01,\n",
       "          -9.4109e-02,  2.3452e-01, -2.3605e-01, -5.8077e-01, -1.5416e-01,\n",
       "           6.5603e-01,  1.3763e-01,  3.9735e-01,  7.9892e-02, -1.1505e+01,\n",
       "           2.3766e-01,  8.0025e-01, -3.2007e-01, -1.3948e-02, -1.4711e-03,\n",
       "           9.9766e-02, -2.0260e-01,  2.8864e-01, -1.5507e-01, -2.4306e-01,\n",
       "          -3.0913e-01,  5.3102e-02,  1.5295e-01, -3.9772e-01, -2.0144e-02,\n",
       "           1.6491e-01, -6.8032e-01,  3.5710e-01,  2.3157e-03,  7.4859e-01,\n",
       "          -4.2649e-01,  3.6167e-01,  2.2014e-01, -1.9688e-01, -2.7278e-01,\n",
       "          -1.8193e-01, -7.9110e-02,  1.1410e-01,  4.0850e-01, -1.0784e-01,\n",
       "           1.2502e-01, -1.5907e-02,  1.6910e-01, -3.6100e-01, -5.5169e-01,\n",
       "           1.1631e-01, -2.1668e-01,  5.4746e-01,  2.3867e-01, -3.7512e-01,\n",
       "           3.2512e-01, -4.9439e-01,  5.6429e-01,  4.6260e-01,  4.7506e-01,\n",
       "           8.8443e-01,  1.1463e+00,  1.1188e-01, -9.8363e-02, -7.7025e-02,\n",
       "          -1.0508e+00,  2.5426e-01, -2.9830e-01,  2.3546e-02, -1.3959e-01,\n",
       "           4.6025e-01, -6.2009e-01,  7.9597e-02, -5.0293e-01,  5.8823e-02,\n",
       "           6.2386e-01, -2.6315e-02,  1.4423e-01, -4.5739e-01, -1.2755e-01,\n",
       "           1.3755e-01, -5.8906e-01,  2.0036e-01, -1.9578e-01, -1.3766e-01,\n",
       "          -2.3435e-01,  2.8762e-01,  9.7934e-02, -3.4963e-02,  3.0716e-01,\n",
       "          -1.3365e-02, -2.9354e-01,  1.0893e+00,  8.2567e-02, -5.0245e-02,\n",
       "           2.2870e-01,  2.3276e-01, -3.7830e-01, -5.6625e-01, -2.3138e-01,\n",
       "          -1.5890e-01, -7.3161e-02,  2.1202e-01,  6.5364e-01, -6.6347e-02,\n",
       "          -6.6615e-01, -2.1763e-01,  8.6870e-02,  1.6609e-01,  3.8130e-01,\n",
       "          -7.1125e-01,  1.2311e-01, -9.6181e-02, -8.4748e-02, -6.7142e-01,\n",
       "          -1.6870e-01, -4.6122e-01, -8.0278e-01,  4.6539e-01, -1.1341e-01,\n",
       "           9.7715e-02, -6.3966e-03,  1.7979e-01,  3.7020e-01, -2.5863e-01,\n",
       "           3.0143e-01,  2.8906e-01,  1.9851e-01, -3.7039e-01, -9.1157e-02,\n",
       "           5.0372e-01,  8.5142e-01, -1.5343e-01, -5.6622e-01, -3.5078e-01,\n",
       "           2.7882e-01,  4.6284e-01,  3.8384e-01, -6.5135e-01, -4.7615e-02,\n",
       "          -8.4967e-01, -1.8255e-03,  2.3246e-02,  2.2182e-01,  6.5271e-01,\n",
       "          -1.3606e-01, -1.7489e-01,  1.4973e-01, -6.8776e-01, -2.3890e-01,\n",
       "          -8.0775e-01,  9.0802e-03,  1.3328e-01, -5.3568e-01,  2.0766e-02,\n",
       "          -2.2084e-01,  3.5107e-02, -2.1555e-01,  3.1775e-01, -4.3180e-02,\n",
       "           5.8231e-01,  1.3394e-01, -4.3183e-02,  1.5177e-01, -7.6733e-02,\n",
       "          -3.2146e-01, -2.6066e-01, -1.5882e-01, -1.4399e-01, -2.9683e-01,\n",
       "          -4.2971e-02,  1.1188e-01,  8.0679e-03, -3.2664e-01, -1.3912e-02,\n",
       "           1.4071e-01, -3.0977e-01, -1.7385e-01,  5.7069e-01,  2.5914e-01,\n",
       "          -1.5212e-01, -7.1293e-02, -4.2115e-02, -2.0610e-01,  1.4565e-01,\n",
       "           2.4961e-01, -9.2676e-02,  2.4912e-01, -6.3042e-01, -3.1732e-01,\n",
       "          -2.5141e-01,  1.2206e+00, -2.9540e-01, -5.4570e-01,  6.3615e-02,\n",
       "           1.9853e-01, -6.7274e-02, -5.3313e-03, -9.9330e-02,  1.0801e-01,\n",
       "          -2.7465e-01, -9.6143e-02,  2.9869e-01, -1.3709e-02,  2.5484e-01,\n",
       "           2.0811e-01,  6.8637e-02,  1.7620e-01,  4.6117e-01,  6.1676e-01,\n",
       "          -1.8913e-01, -1.4796e-01, -5.1854e-01,  4.5446e-02, -3.8615e-02,\n",
       "          -3.3813e-01,  3.3949e-01, -7.5272e-02, -1.4831e-01,  1.5124e-01,\n",
       "           4.6757e-01,  1.0609e+00, -1.2446e-01,  1.7527e-01,  7.2580e-01,\n",
       "          -1.9442e-01,  4.0487e-01,  2.5667e-01,  4.2968e-01,  1.9985e-01,\n",
       "           4.1268e-01, -1.9190e-01, -3.1504e-01,  1.0962e-01,  3.0382e-01,\n",
       "           1.5910e-01, -6.4786e-01, -6.1037e-01, -1.3968e-01,  5.9264e-01,\n",
       "          -4.0616e-02,  3.6312e-01, -1.8194e-01,  4.7372e-01,  6.7052e-01,\n",
       "          -9.0600e-01,  5.1754e-03, -1.3353e-01,  2.7721e-01,  5.7902e-02,\n",
       "          -5.7289e-01,  2.4729e-01,  1.8793e-01, -4.3069e-01,  3.7450e-01,\n",
       "           6.6293e-01, -7.0765e-01,  9.4907e-02,  5.1865e-01, -6.6586e-01,\n",
       "          -2.4915e-02, -2.7319e-01,  1.5443e-01,  4.2560e-01,  4.4728e-02,\n",
       "          -3.4883e-01,  2.2955e-01, -2.8184e-01, -4.4025e-01,  5.7841e-02,\n",
       "           7.0912e-01, -5.5962e-02,  2.5797e-01, -5.2233e-02, -2.7062e-01,\n",
       "          -1.5723e-01, -4.0819e-01, -2.9042e-01, -1.2105e-01, -3.2689e-02,\n",
       "           1.0744e-01, -5.9078e-02,  1.0331e-01,  3.7123e-01, -4.3106e-01,\n",
       "           4.0383e-01,  5.0816e-01,  4.5129e-01,  1.4192e-01, -1.0497e-01,\n",
       "          -1.7260e-01,  1.7224e-01, -2.8145e-02,  4.4469e-01, -2.1522e-01,\n",
       "          -1.9127e-01,  1.5410e-01,  1.9695e-01, -3.0994e-01,  2.3659e-01,\n",
       "          -6.0500e-01,  3.8165e-02, -1.6969e-01, -2.8534e-01,  3.5610e-01,\n",
       "           2.6130e-01,  3.4343e-01, -6.9950e-01,  2.5013e-01,  2.7160e-01,\n",
       "           3.0245e-01,  4.4579e-01,  1.0390e-01,  2.9093e-01,  1.9281e-01,\n",
       "           4.0131e-02, -5.6405e-01, -8.3026e-02, -2.4544e-01,  2.2122e-01,\n",
       "          -2.4965e-01, -2.0955e-01,  6.9195e-01,  2.4255e-01,  6.6116e-01,\n",
       "           4.0400e-01,  2.6358e-01, -5.7043e-01, -1.5445e-01,  8.3865e-02,\n",
       "          -4.2393e-02, -3.5150e-01,  2.1524e-01, -3.5600e-02,  2.7553e-01,\n",
       "          -8.0785e-01, -4.2374e-01,  5.0325e-02,  2.4625e-01,  6.5630e-02,\n",
       "           5.9476e-01, -1.6523e-01,  3.3607e-01, -1.1727e-01,  1.0698e-01,\n",
       "           2.2024e-01,  8.1495e-01,  7.7855e-02, -6.3969e-01, -1.5519e-02,\n",
       "           1.6102e-01, -1.4560e-01,  1.3427e-01, -6.2520e-01,  2.1842e-01,\n",
       "           6.2388e-02,  9.6463e-01,  5.9641e-01]], device='mps:0'),\n",
       " tensor([[-0.2744, -0.4852, -0.4880,  ...,  0.2064,  0.2977,  0.6606],\n",
       "         [ 0.0506, -0.6230, -0.6446,  ..., -0.0875,  0.4018,  0.3101],\n",
       "         [-0.4643, -0.3750, -0.2078,  ...,  0.2835,  0.5093,  0.4110]],\n",
       "        device='mps:0'),\n",
       " tensor([[-1.1097e-02, -5.0567e-01, -4.6699e-01, -6.4640e-01, -3.9556e-01,\n",
       "          -2.5266e-01, -5.5990e-01,  3.0801e-01, -1.2536e-01,  1.4189e-01,\n",
       "           2.9133e-01,  5.3879e-01, -3.6706e-01, -2.8859e-01, -4.8290e-01,\n",
       "          -5.3699e-02, -6.1741e-01, -1.6918e-02,  3.4690e-01,  3.0491e-01,\n",
       "           5.3718e-01,  1.8073e-01, -4.1129e-01,  2.9149e-01, -3.7959e-01,\n",
       "           6.6561e-02, -3.4985e-01, -3.1546e-01,  2.6210e-01,  2.8012e-01,\n",
       "           5.3618e-02,  7.7692e-01, -2.4307e-02,  1.0962e+00, -1.8413e-01,\n",
       "          -2.1778e-01,  1.5359e-01,  2.3320e-01,  8.6439e-01, -6.2204e-01,\n",
       "           5.6395e-01,  3.6489e-01,  2.7586e-02, -4.0063e-02,  2.7845e-01,\n",
       "           3.5451e-01,  1.4567e-01, -5.7913e-01, -2.7161e-01, -3.4037e-01,\n",
       "          -1.2979e-01, -1.5164e-01,  5.0058e-01, -2.3292e-01, -7.9546e-01,\n",
       "          -1.9253e-01,  1.1288e-02,  2.6982e-01,  1.1117e-01,  2.3591e-01,\n",
       "           2.8900e-02,  7.9425e-01, -1.9339e-01, -3.2966e-03,  2.0559e-01,\n",
       "           2.5502e-01, -7.6243e-01, -1.2873e-01, -7.2822e-01,  1.1548e-01,\n",
       "           2.1413e-01,  3.5162e-01,  1.0500e-01, -2.6816e-01, -1.9644e-01,\n",
       "           6.0243e-02,  1.6956e-01,  1.8610e-01, -2.0035e-01,  3.5967e-02,\n",
       "           2.1140e-01,  4.4140e-01,  4.5980e-01,  5.9922e-02,  1.6117e-01,\n",
       "           5.6960e-02,  2.0080e-01, -3.9869e-02, -3.1738e-02,  1.7672e-01,\n",
       "           1.1984e-01, -9.9577e-02, -3.7483e-01, -4.3956e-01, -9.5205e-02,\n",
       "          -3.0408e-01,  3.2250e-02, -2.9469e-01, -2.3954e-01, -3.3261e-01,\n",
       "           2.5982e-01,  9.2762e-02,  3.2750e-01, -9.3082e-02, -6.6514e-01,\n",
       "          -3.5248e-01,  4.0832e-01,  3.4586e-03, -2.8129e-01, -3.8448e-01,\n",
       "          -6.6713e-02, -4.9346e-01,  1.2765e-01, -4.4650e-01,  1.9893e-01,\n",
       "           1.0835e-01,  1.8686e-01, -3.2670e-01,  6.5172e-02,  8.8101e-02,\n",
       "           6.1068e-02, -5.0938e-01,  2.7421e-01,  1.8326e-01,  9.3046e-02,\n",
       "           2.2489e-01,  6.8015e-01, -1.6210e-01,  4.3454e-01,  3.5722e-01,\n",
       "           3.1169e-01,  1.6221e-01, -4.1686e-01, -5.0259e-01, -7.5302e-02,\n",
       "           1.8488e-02, -5.9908e-01, -5.7442e-01,  6.1723e-02, -2.2868e-01,\n",
       "           1.6627e-01, -7.3083e-02, -5.1768e-01, -5.4307e-01, -3.1494e-01,\n",
       "          -3.0772e-01, -2.0517e-02, -3.2393e-01, -4.8159e-01, -2.5398e-01,\n",
       "          -6.1867e-01, -7.2891e-01,  1.9443e-01,  1.5164e-01,  2.7714e-01,\n",
       "          -4.3163e-01,  5.4580e-01, -3.0612e-01, -4.8554e-01,  3.7642e-01,\n",
       "           4.3432e-01, -5.0330e-01,  2.9526e-01, -6.3499e-01,  5.1171e-02,\n",
       "           2.0480e-01,  2.5291e-02, -3.4147e-02,  4.2807e-01,  1.1727e-01,\n",
       "           3.9831e-01,  2.7152e-02,  1.2476e-01, -2.9187e-01,  1.5015e-01,\n",
       "          -4.5427e-02, -3.2147e-01, -1.1176e-01,  1.7279e-02, -5.2967e-02,\n",
       "           5.4936e-01, -1.7306e+00, -1.7852e-01,  7.0060e-02, -2.7460e-01,\n",
       "          -1.3173e-01,  2.1473e-01,  5.0407e-03,  3.2283e-01, -3.5674e-01,\n",
       "           8.3698e-02,  8.7128e-02,  8.4654e-02, -1.0725e-01, -5.0983e-02,\n",
       "           2.9101e-01,  7.5929e-01,  4.0793e-01,  9.7463e-02, -2.6657e-01,\n",
       "          -6.0645e-01,  4.1554e-01,  1.9081e-01,  2.0730e-01,  3.9000e-01,\n",
       "           6.7931e-02, -6.9343e-01,  2.9674e-01,  1.9859e-01,  3.7002e-01,\n",
       "           2.5546e-02, -5.8496e-02, -4.8962e-01, -2.4682e-01, -4.3705e-01,\n",
       "           2.7759e-01, -1.3436e-01, -2.5922e-01,  6.3173e-01,  1.9273e-01,\n",
       "          -5.3531e-02,  3.9653e-01,  1.8733e-01, -4.1041e-01,  4.1633e-01,\n",
       "           5.2814e-01,  7.9114e-02,  1.4573e-01, -2.8055e-01, -1.6673e-01,\n",
       "          -3.1532e-01, -4.4420e-01,  4.4703e-01,  8.3941e-01, -7.2084e-03,\n",
       "           5.5343e-01, -3.5757e-01,  5.1174e-01, -3.6424e-01, -6.7779e-03,\n",
       "          -4.6486e-02,  6.8700e-02, -3.3988e-01,  1.6284e-01,  1.5551e-01,\n",
       "           1.0340e-01,  1.9709e-01,  8.5425e-01,  4.6494e-01,  1.4684e-01,\n",
       "           4.1031e-01,  2.5417e-02,  3.2080e-01,  7.8684e-01, -3.0254e-01,\n",
       "           3.4990e-01, -7.1578e-03, -4.1842e-01,  5.6461e-01, -3.5139e-01,\n",
       "           1.4248e-01, -2.3020e-01,  2.2987e-01,  1.7866e-01, -2.7854e-01,\n",
       "           2.1426e-01,  6.5273e-01,  9.5860e-02, -3.9283e-01,  7.3417e-01,\n",
       "          -4.0214e-01,  7.1262e-01, -3.0138e-02, -2.6992e-02, -1.0120e-01,\n",
       "          -7.4577e-01, -6.8569e-01,  1.4294e-01,  3.0808e-01, -7.6332e-01,\n",
       "          -3.4295e-01,  1.2380e-01,  1.3255e-01,  3.5068e-01,  2.4156e-01,\n",
       "           4.6144e-01, -1.7016e-01, -4.7185e-02, -4.4024e-01, -4.6015e-01,\n",
       "           1.1917e-01,  7.3311e-02,  9.5311e-02, -9.8199e-02,  9.2587e-02,\n",
       "           6.1067e-01,  2.4297e-01,  2.6799e-01,  1.4946e-01, -3.4610e-01,\n",
       "          -2.7723e-02,  1.7135e-01,  3.7874e-01, -6.1900e-01, -2.1807e-01,\n",
       "           5.3543e-02,  8.3164e-02,  6.6826e-01, -2.1832e-01, -1.1546e-01,\n",
       "          -1.2450e-01,  5.9739e-01, -2.2108e-01,  6.8002e-02,  1.0269e-01,\n",
       "          -6.8586e-01,  2.7619e-01, -5.9939e-02, -8.5297e-02, -5.2861e-01,\n",
       "           2.2997e-01, -5.7787e-02, -1.6068e-01, -7.3269e-03, -7.7601e-02,\n",
       "           2.2454e-01,  1.3406e-01,  2.8044e-01, -5.6270e-01, -4.2092e-01,\n",
       "           1.1781e-01, -3.2365e-01, -6.6016e-02, -4.5861e-01,  3.6766e-01,\n",
       "           1.3890e-01,  4.3829e-01, -1.0837e-01, -2.8576e-02, -1.3190e-01,\n",
       "           5.2866e-01, -6.2780e-01,  1.5573e-02,  4.8525e-01, -3.5595e-01,\n",
       "          -1.8139e-01,  4.7644e-01, -2.8011e-01,  4.9664e-02,  2.2982e-01,\n",
       "           3.7738e-01, -1.1530e-01, -4.7105e-01, -5.5247e-02,  9.3761e-01,\n",
       "           4.8980e-01,  2.0874e-02, -6.3820e-01,  1.2560e-01,  4.2066e-01,\n",
       "          -3.4219e-01,  1.0777e-01, -3.7280e-01, -2.3519e-01,  8.6456e-02,\n",
       "           3.3260e-02,  5.0344e-01, -5.8998e-02,  1.9741e-01,  4.6778e-01,\n",
       "           1.7011e-01, -6.9331e-01, -3.0557e-01,  6.3124e-01,  5.4670e-01,\n",
       "          -1.2247e-01, -5.4832e-01,  1.1480e-01, -6.4563e-01,  3.1198e-02,\n",
       "          -3.8623e-03, -3.5755e-01,  2.9653e-01, -1.4962e-01, -2.5683e-01,\n",
       "          -2.0003e-01, -5.3748e-01,  4.4973e-03,  1.0291e-01, -6.9595e-02,\n",
       "           4.1738e-01, -7.9610e-03, -3.1154e-01,  1.0058e+00, -2.7292e-01,\n",
       "          -3.1356e-02,  7.4139e-01, -5.4234e-01,  1.0457e-02,  2.4642e-01,\n",
       "          -1.3433e-01, -1.2416e-02,  2.4323e-02,  1.0660e+00,  1.1515e-05,\n",
       "           3.6402e-02, -2.0346e-01,  4.2203e-01, -4.7659e-01,  1.9394e-01,\n",
       "          -4.9958e-01,  4.8310e-01, -6.9399e-02,  2.4253e-01, -8.2329e-01,\n",
       "           4.3257e-01, -3.0604e-01, -3.2040e-01, -6.3922e-01,  5.0359e-02,\n",
       "          -5.0711e-02,  6.6524e-02, -5.8707e-02,  6.0787e-01, -1.0953e+01,\n",
       "           1.0946e-01,  8.4108e-01, -3.2618e-02,  1.3653e-01,  4.3336e-01,\n",
       "           1.2158e-01, -4.6684e-01,  7.2509e-01, -2.3242e-01, -6.6092e-03,\n",
       "          -3.5861e-01, -1.2444e-01, -1.3819e-01, -6.3519e-01,  4.0758e-01,\n",
       "          -1.3255e-01, -5.9114e-01,  2.8540e-01, -1.6945e-01,  2.4978e-01,\n",
       "          -2.3651e-01,  1.5671e-01,  5.9352e-01, -6.1010e-01, -3.3943e-01,\n",
       "          -1.1958e-01, -2.9991e-01,  5.1723e-01,  4.0217e-01, -2.3672e-01,\n",
       "           2.9096e-01,  3.3529e-02,  2.9455e-01, -5.5694e-01, -4.8133e-01,\n",
       "           4.3369e-01, -3.9950e-01,  2.1957e-01, -9.9943e-01, -3.3265e-01,\n",
       "          -1.8082e-01, -5.0951e-01, -2.7870e-02,  5.3812e-01,  7.5524e-01,\n",
       "           1.2821e+00,  1.1534e+00,  9.8198e-02,  4.6362e-01, -1.0523e-01,\n",
       "          -6.6935e-01,  5.1619e-02, -5.6525e-01, -4.1224e-01,  8.7196e-02,\n",
       "           2.2385e-01, -5.4352e-01,  4.7854e-01, -5.9897e-01,  1.8912e-01,\n",
       "           1.4329e-01,  7.9852e-02,  1.7368e-01, -5.1897e-01, -5.5039e-01,\n",
       "          -1.3282e-01,  8.1201e-02,  8.6324e-02, -6.7881e-01,  1.1881e-01,\n",
       "           1.7006e-01,  6.8197e-01, -2.5397e-01, -4.4268e-01,  4.1137e-01,\n",
       "          -2.2746e-03,  9.3809e-02,  1.3158e+00, -2.6036e-01, -1.7413e-01,\n",
       "          -4.8428e-02, -2.3194e-01, -3.3484e-01, -3.9656e-01, -4.5613e-01,\n",
       "          -2.5020e-02,  3.6573e-01,  5.1948e-01,  2.6062e-01, -2.3329e-02,\n",
       "          -3.0054e-01, -4.1445e-02,  5.1977e-02,  3.3862e-01,  9.6772e-01,\n",
       "          -6.6887e-01,  2.6280e-01, -1.9148e-01,  9.2178e-02,  2.2223e-01,\n",
       "          -5.7106e-01, -1.6871e-02,  2.9434e-03,  1.3975e-02,  5.7281e-01,\n",
       "           1.9705e-01,  2.6453e-01,  3.1190e-02,  2.2859e-01, -4.1682e-01,\n",
       "          -4.4044e-01,  2.9125e-04, -1.3190e-01, -5.8294e-01, -4.1034e-01,\n",
       "          -1.3514e-01,  1.3712e-01, -2.8585e-01, -5.3585e-01, -2.5012e-01,\n",
       "           1.4192e-01,  2.5900e-03,  4.9952e-02, -3.3857e-01,  4.7379e-01,\n",
       "          -7.6918e-01, -2.1429e-01, -2.7578e-01, -3.8495e-03,  1.4633e-01,\n",
       "           2.5508e-01, -5.1518e-01, -2.7838e-01, -1.0123e-01, -9.2314e-01,\n",
       "          -8.4489e-01,  2.2927e-02,  7.4620e-01, -1.1960e-01,  1.1861e-01,\n",
       "           8.5428e-02, -3.1671e-03, -3.7708e-01,  2.5194e-01, -2.4644e-01,\n",
       "           2.8149e-01,  4.3265e-01, -4.9583e-01,  2.9773e-02, -1.2792e-01,\n",
       "          -1.8992e-02, -1.0887e-01, -8.8647e-02, -5.4131e-01, -6.1371e-01,\n",
       "          -4.1197e-01,  1.3736e-01, -2.2747e-01,  4.1516e-01,  6.8072e-02,\n",
       "           1.2582e-01, -2.0826e-01,  9.8224e-02,  8.2540e-01,  6.0197e-01,\n",
       "           2.3014e-01, -3.2920e-01, -7.7666e-01,  3.4140e-02,  4.1642e-02,\n",
       "          -1.6450e-01,  2.1317e-01, -1.6384e-01, -3.0657e-01, -1.0112e-01,\n",
       "           1.2620e-02,  1.0002e+00, -2.0790e-01, -1.9155e-01,  1.0507e-01,\n",
       "           7.4626e-01, -6.0834e-01, -5.7221e-01,  3.7612e-01,  5.0418e-02,\n",
       "          -5.6831e-01, -1.9450e-01,  4.0315e-01, -1.7111e-01,  8.9921e-02,\n",
       "           3.7572e-01,  1.7821e-01, -1.8612e-01,  2.2711e-01, -6.7291e-02,\n",
       "          -1.7001e-01,  3.2731e-01, -1.6533e-01, -7.7254e-02,  4.0384e-01,\n",
       "           9.9517e-02,  1.5584e-01, -2.5262e-01, -4.8444e-01, -2.7025e-01,\n",
       "           1.5439e-01,  1.1110e+00, -1.7989e-01, -3.2527e-01,  6.6881e-01,\n",
       "          -3.0956e-01,  3.8928e-01,  4.8478e-01, -2.2611e-01, -5.0807e-02,\n",
       "           5.8049e-01,  3.0288e-01, -7.3236e-01,  2.2417e-01,  3.1670e-01,\n",
       "          -1.7335e-01, -5.3095e-01, -3.2838e-01, -2.1330e-01, -3.5189e-02,\n",
       "          -3.6356e-01,  5.2132e-01,  1.2255e-01,  3.9253e-01, -1.0843e-01,\n",
       "          -8.7207e-01,  4.7185e-01,  2.6128e-01,  8.0250e-01, -6.2568e-03,\n",
       "          -4.4880e-01, -3.5541e-01, -3.5790e-01, -6.6453e-02,  3.6543e-01,\n",
       "           2.0116e-01, -1.0333e+00,  4.6071e-02,  4.0507e-01, -3.9812e-02,\n",
       "           4.7083e-01, -1.4254e-01,  1.9634e-01,  4.7891e-01, -4.3918e-02,\n",
       "          -2.1856e-01,  1.3161e-01, -3.3027e-02, -2.7190e-01,  4.2861e-01,\n",
       "           4.6755e-01,  1.6062e-02,  2.5454e-02,  6.4881e-02, -4.2427e-01,\n",
       "          -3.4980e-01,  7.7064e-02, -1.7966e-01, -3.5675e-01, -1.4295e-01,\n",
       "          -4.0726e-01, -5.1772e-01, -6.6164e-02, -1.1357e-01,  1.3233e-01,\n",
       "           3.1032e-01,  1.3036e-01,  3.8431e-01, -2.3708e-01, -1.2541e-01,\n",
       "          -2.7977e-01,  4.5653e-01, -2.7595e-01,  8.0808e-01,  2.0566e-01,\n",
       "          -1.4423e-01,  2.0720e-01,  2.1514e-01, -2.9046e-01,  1.3028e-01,\n",
       "          -2.7828e-01,  2.8810e-01, -5.7798e-01, -3.4110e-01, -5.9710e-02,\n",
       "          -7.7948e-02, -1.5049e-01, -2.8505e-01, -1.0179e-01,  1.0608e-01,\n",
       "          -9.9043e-02,  9.7868e-01, -4.1759e-01,  6.3833e-01, -1.0342e-01,\n",
       "          -1.1350e-01, -7.5089e-01,  3.9397e-01,  3.6376e-01, -1.4798e-01,\n",
       "          -5.6434e-01, -2.1653e-01, -8.9158e-02, -4.2284e-01,  8.8621e-01,\n",
       "           5.8157e-01, -7.4673e-01, -3.7246e-01, -1.5919e-01, -2.4113e-01,\n",
       "          -4.7389e-02,  6.9820e-02, -3.2710e-01, -1.3295e-02, -1.6580e-01,\n",
       "          -9.8715e-01,  2.7641e-01, -3.6473e-02,  4.9205e-01, -2.5466e-01,\n",
       "           7.6979e-01, -8.6017e-02, -1.2749e-01,  1.1690e+00,  1.1169e-01,\n",
       "          -6.1844e-02,  5.9531e-01, -1.6837e-01, -6.2973e-01, -7.0295e-02,\n",
       "           4.4239e-02, -3.1098e-01,  8.7670e-02, -4.7880e-01,  2.9353e-01,\n",
       "           2.3648e-01, -7.1226e-02,  4.3877e-01]], device='mps:0'),\n",
       " tensor([[-1.6242e-01, -7.0038e-01, -6.4050e-01, -5.7401e-01, -5.2307e-01,\n",
       "          -2.7146e-01, -2.7335e-01,  3.7291e-01, -9.6958e-02,  6.4964e-01,\n",
       "           1.5170e-01,  2.6801e-01,  8.5330e-02, -1.8806e-01, -2.2300e-02,\n",
       "          -4.2005e-01,  2.4015e-01,  7.1058e-01,  5.2697e-01,  1.6134e-01,\n",
       "           7.6495e-01,  1.6615e-01,  7.2323e-02,  2.3478e-02, -8.6874e-01,\n",
       "           1.4401e-01, -2.6941e-01, -4.0714e-02,  3.0428e-01, -7.5863e-01,\n",
       "          -3.8205e-01,  1.9576e-01, -4.2277e-02,  1.1939e+00,  2.7977e-01,\n",
       "          -1.5983e-03,  1.5187e-01, -5.0016e-01,  5.7923e-01, -2.1356e-02,\n",
       "           4.3514e-01,  3.2626e-02,  1.2402e-01,  2.3256e-01,  5.7568e-01,\n",
       "          -2.4870e-02,  4.6483e-02, -4.6613e-01,  2.7395e-01, -2.1692e-01,\n",
       "           1.6020e-01,  8.0300e-02,  6.9412e-01,  4.0318e-01, -1.0100e+00,\n",
       "          -7.5921e-01,  3.8768e-01,  1.3175e-01, -5.6965e-01,  2.7855e-01,\n",
       "          -1.3067e-02,  4.9120e-01,  1.5309e-01,  3.7727e-02, -2.4845e-01,\n",
       "           1.9583e-01, -6.5221e-01, -9.1750e-01, -2.9501e-01, -2.7006e-01,\n",
       "          -1.0704e-01,  7.0736e-01, -1.5237e-01, -1.7698e-01, -1.8834e-02,\n",
       "          -7.7560e-02,  1.4350e-01, -6.8143e-02, -3.4723e-01, -1.6803e-01,\n",
       "           4.8939e-02,  2.1572e-02, -1.5116e-01, -2.8349e-01, -3.1992e-01,\n",
       "           1.3467e-01,  4.9109e-01,  2.7328e-01,  3.1822e-01, -1.7827e-01,\n",
       "           5.0664e-01,  4.6780e-02, -1.6823e-01, -7.1850e-01,  3.5289e-01,\n",
       "          -3.4872e-01,  4.2163e-01, -3.9566e-01, -8.4710e-02, -3.3188e-01,\n",
       "           2.1301e-01,  2.1352e-01, -1.8870e-01,  2.0598e-01, -2.0243e-01,\n",
       "          -6.5668e-01,  8.4434e-02, -3.1302e-01, -7.8138e-01, -5.8787e-02,\n",
       "          -2.8282e-01, -4.6164e-01, -5.7959e-02, -4.4877e-01, -1.5116e-01,\n",
       "          -1.9516e-01,  9.5520e-03, -8.2824e-02, -4.1386e-01,  3.4294e-01,\n",
       "           1.5828e-02, -3.5935e-01,  2.8065e-01,  2.7812e-01,  2.4794e-01,\n",
       "          -4.8112e-01,  4.3224e-01,  1.6921e-01,  3.1934e-01,  5.4573e-01,\n",
       "           9.4214e-04,  2.2959e-01, -1.2800e-01,  1.9653e-02, -2.2768e-01,\n",
       "          -1.3802e-01,  1.1540e-01, -3.3846e-01,  6.1959e-02, -1.9162e-01,\n",
       "           5.3290e-03,  7.6019e-02, -3.5955e-01, -5.2119e-02,  1.0485e-01,\n",
       "          -1.0216e-01,  1.0262e-01, -2.2118e-01, -3.5332e-01, -4.5933e-01,\n",
       "          -5.7635e-01, -5.2464e-01,  5.5836e-01, -1.1661e-01, -4.1934e-02,\n",
       "          -1.5776e-01,  5.7446e-01,  1.6270e-01, -3.5791e-01,  2.4243e-01,\n",
       "           7.8397e-01, -1.4168e-01,  2.0700e-01, -1.7219e-01, -1.5483e-01,\n",
       "           3.1463e-01, -4.1436e-01, -1.4263e-01, -3.2898e-01, -2.8233e-02,\n",
       "           3.7240e-01, -3.1474e-01,  2.1667e-01,  3.6003e-01,  2.0594e-01,\n",
       "          -5.7042e-01,  4.9565e-01,  4.6104e-01, -2.9591e-01, -9.3946e-02,\n",
       "          -4.2259e-01, -1.6229e+00,  3.0726e-02,  2.9152e-01, -1.3931e-01,\n",
       "          -6.5735e-01,  5.0211e-01,  4.8923e-01,  7.6640e-02, -2.0383e-01,\n",
       "           1.1236e-01, -1.5082e-01, -9.5284e-03,  3.8578e-01, -2.8985e-01,\n",
       "           4.7139e-01,  4.8316e-01,  2.0403e-01,  3.4324e-01, -9.7658e-03,\n",
       "          -3.7935e-01,  6.0834e-03,  2.7081e-02,  2.6397e-01,  1.8619e-01,\n",
       "           1.5956e-01, -3.9995e-02,  5.8105e-02,  3.9408e-01, -2.4172e-02,\n",
       "          -4.0624e-02, -1.7089e-01, -5.7926e-01, -5.0470e-02,  1.9124e-01,\n",
       "           6.4155e-01, -2.2456e-01, -2.4493e-01,  4.4529e-01,  1.6237e-01,\n",
       "          -3.8950e-01,  2.2050e-01, -2.2974e-01, -3.9150e-01, -2.4641e-01,\n",
       "           7.6712e-01, -1.9489e-01,  8.0535e-02,  1.6024e-01, -2.6362e-01,\n",
       "           3.9931e-01, -1.7430e-01,  5.0831e-01,  2.7717e-01, -2.3696e-01,\n",
       "          -1.7325e-01, -6.1387e-01, -1.1909e-01,  2.6462e-02,  6.0976e-02,\n",
       "          -4.2756e-01, -4.7945e-01, -5.0013e-01, -2.7739e-01,  2.7821e-01,\n",
       "           4.0482e-01, -1.5696e-01,  7.6627e-01,  6.0994e-01,  2.9539e-01,\n",
       "           3.6976e-03, -8.8765e-02,  2.5638e-01,  8.4581e-02, -4.2303e-01,\n",
       "           3.8420e-01,  1.8100e-01, -6.1089e-01,  3.5106e-01, -3.9741e-01,\n",
       "          -1.9527e-01, -1.7833e-02,  2.0362e-01,  2.6719e-01, -4.0318e-01,\n",
       "           3.6024e-01,  4.6547e-02, -3.5422e-01, -2.5936e-01,  1.3301e-02,\n",
       "          -2.9795e-01,  5.7843e-01,  3.7830e-01, -4.9141e-01, -5.9186e-01,\n",
       "          -7.7743e-01, -7.2760e-02, -3.9089e-01,  7.3042e-02, -8.1030e-01,\n",
       "          -4.0239e-01,  1.9795e-01, -1.7676e-01,  1.9436e-02, -1.8883e-01,\n",
       "           1.1450e-01,  1.4922e-01,  2.4642e-02, -1.7016e-02, -3.6694e-01,\n",
       "          -3.8806e-01,  4.8751e-02, -2.8404e-01,  3.9855e-02, -3.8188e-02,\n",
       "           2.9088e-01, -4.3608e-02, -2.7753e-01,  5.2703e-02, -3.7915e-01,\n",
       "           1.7092e-01, -5.3674e-02,  2.6932e-02, -1.7148e-01, -2.7199e-01,\n",
       "           4.6577e-03,  9.1325e-02,  3.2511e-01, -9.1780e-02, -7.5407e-01,\n",
       "          -1.5033e-01,  6.8233e-01,  1.6228e-01, -9.5992e-02,  9.1234e-03,\n",
       "          -1.0360e+00, -9.0141e-02,  1.6485e-01, -6.5452e-02, -2.5652e-01,\n",
       "           2.0138e-01, -4.8738e-02, -2.0096e-01,  2.1763e-01, -5.8373e-01,\n",
       "           6.5732e-01, -1.0787e-02,  8.0051e-02,  1.4502e-01, -4.0099e-01,\n",
       "           2.5355e-02, -7.3821e-01, -1.1567e-01, -1.4171e-01, -1.1011e-01,\n",
       "           2.0496e-02,  1.6521e-01,  5.0393e-03,  4.7869e-01, -2.4951e-01,\n",
       "           4.0432e-01, -2.8500e-01, -3.7698e-02,  3.2068e-01, -4.1196e-01,\n",
       "          -3.1461e-02,  6.1526e-01, -8.4449e-01,  3.7234e-02,  4.6183e-02,\n",
       "           7.5600e-02, -1.2801e-01, -1.6950e-01, -2.5520e-02,  6.2436e-01,\n",
       "           4.9453e-01,  2.3703e-01, -4.0964e-01,  4.9814e-01,  1.2824e-01,\n",
       "          -3.3118e-01,  4.5327e-01, -2.4032e-01, -3.7679e-01, -3.1047e-01,\n",
       "           1.4162e-01,  2.4629e-01, -4.6608e-01, -2.2841e-01, -1.6908e-01,\n",
       "           2.5153e-01, -3.1989e-01,  7.7959e-02,  2.1063e-01,  4.1502e-01,\n",
       "          -3.4311e-01, -3.1888e-01,  1.1408e-01, -2.1551e-01, -3.3674e-01,\n",
       "           1.6527e-01, -1.6066e-02,  9.0713e-02, -1.2400e-01,  2.9879e-02,\n",
       "          -2.5245e-01, -6.0626e-02,  7.1335e-01,  2.1966e-01, -1.1618e-01,\n",
       "          -1.0543e-01,  1.2762e-01, -5.6494e-01,  4.7863e-01,  2.3028e-01,\n",
       "          -6.2099e-02,  3.9678e-01, -1.3814e-01,  6.0775e-02,  2.2432e-01,\n",
       "          -2.3228e-01,  7.4942e-02, -1.6672e-01,  3.7631e-01,  2.3143e-01,\n",
       "          -4.2253e-01,  4.1375e-02,  2.8276e-01, -1.7768e-02, -1.1975e-01,\n",
       "          -1.4280e-01,  7.9805e-01, -4.5214e-01,  3.7852e-01,  1.5841e-01,\n",
       "           1.3545e-01, -1.1286e-01,  1.6391e-01, -2.8432e-01, -6.0964e-01,\n",
       "           6.4294e-01,  3.0948e-01,  4.6692e-01,  5.9437e-01, -1.1288e+01,\n",
       "          -2.6069e-01,  6.2847e-01, -6.5879e-01, -1.2117e-01,  5.2818e-03,\n",
       "          -1.0865e-01, -3.0473e-01,  7.1517e-02, -3.8351e-01, -1.2699e-01,\n",
       "           3.0108e-02, -5.9732e-02,  1.6769e-01, -3.8105e-02, -1.8157e-01,\n",
       "          -2.2825e-01, -6.0100e-01,  1.9052e-01, -5.1396e-02,  5.4467e-01,\n",
       "          -3.4883e-01,  3.1829e-01,  3.3910e-01, -2.8179e-02, -2.6985e-01,\n",
       "          -7.2717e-02,  3.2758e-02,  4.8420e-01,  3.5909e-01, -1.4535e-01,\n",
       "           1.7118e-01,  8.2732e-02,  4.4020e-01, -8.7911e-03, -4.0903e-01,\n",
       "          -9.5736e-02, -5.7798e-01,  1.5798e-01, -5.0040e-01, -2.7930e-01,\n",
       "          -1.9627e-01, -6.6823e-02,  3.6748e-01,  2.4904e-01,  7.2846e-01,\n",
       "           7.0996e-01,  5.1166e-01,  2.4686e-01, -3.7895e-01,  1.5767e-01,\n",
       "          -9.6770e-01,  3.2034e-01, -6.1387e-01, -4.0750e-01,  6.2399e-02,\n",
       "           5.8684e-01, -5.6322e-01,  2.3628e-01, -1.9624e-01, -4.3279e-01,\n",
       "           4.0106e-01,  8.6140e-03, -1.6987e-01, -4.3943e-01, -2.0975e-01,\n",
       "           5.5474e-01,  3.7674e-02,  1.2673e-01, -3.7348e-01, -1.8957e-01,\n",
       "          -3.1772e-01,  3.8277e-01,  3.7006e-01,  2.3599e-02, -1.6485e-01,\n",
       "           3.0001e-02, -6.3529e-02,  1.5314e+00,  1.6926e-02,  7.2207e-02,\n",
       "           2.6649e-01,  4.2900e-01, -1.1463e-01, -1.0533e+00,  3.8076e-02,\n",
       "           8.4828e-02,  5.9370e-01,  1.4787e-01,  4.2166e-01, -1.1578e-01,\n",
       "          -6.4764e-01,  1.4822e-01,  2.4530e-01,  1.5993e-01,  4.1286e-01,\n",
       "          -8.2456e-01,  5.4753e-02,  1.9023e-01, -1.3646e-01,  6.5687e-03,\n",
       "           3.9871e-02, -3.0036e-01, -2.1068e-01,  3.5384e-01,  2.4806e-01,\n",
       "           1.0819e-01,  2.3031e-02,  2.9180e-01,  2.7162e-01, -2.0694e-01,\n",
       "           1.3416e-01,  3.3998e-02, -9.3074e-02, -2.1848e-01, -1.4588e-01,\n",
       "           7.2682e-02,  4.9926e-01, -3.0886e-01, -2.3153e-01, -8.1042e-02,\n",
       "          -1.9381e-01, -1.4201e-01,  2.2566e-01, -5.3494e-01,  6.7086e-01,\n",
       "          -5.6732e-01,  2.9206e-04,  1.1787e-01, -9.7842e-02,  2.4557e-01,\n",
       "           2.6749e-01,  4.3392e-01,  2.2338e-01, -4.8916e-01, -7.3530e-01,\n",
       "          -7.2245e-01,  2.1286e-01,  1.5666e-01,  2.9249e-02,  2.8156e-01,\n",
       "          -1.2112e-01,  1.4252e-01, -1.1546e-01,  2.1387e-01,  3.7005e-01,\n",
       "           4.7761e-01,  5.0623e-01, -1.5326e-01,  2.0507e-01, -2.6152e-01,\n",
       "          -4.9464e-01,  6.1408e-02, -2.9523e-01,  3.0700e-01, -4.8403e-01,\n",
       "          -5.5654e-02, -1.1325e-01, -3.4495e-01, -1.6532e-01, -1.3025e-01,\n",
       "          -1.2483e-01,  1.0831e-01,  1.4223e-01,  2.3120e-01,  1.6189e-01,\n",
       "           3.5433e-01, -2.7462e-01, -7.3182e-02, -2.8253e-01,  3.2615e-01,\n",
       "          -1.0701e-01, -4.4685e-02, -2.4827e-01, -3.1653e-01, -7.0411e-02,\n",
       "           2.1311e-01,  6.7827e-01, -1.1858e-01, -6.2601e-01,  3.7081e-01,\n",
       "           5.8397e-01, -1.1082e-01, -3.0060e-01, -2.0124e-01, -1.8456e-01,\n",
       "          -2.4621e-01, -5.0744e-01,  3.4336e-01,  8.4708e-02,  5.8445e-01,\n",
       "           2.3479e-01, -2.0927e-01, -6.8227e-01,  5.9995e-01, -8.2623e-02,\n",
       "          -6.7978e-01, -1.9709e-01, -5.6960e-01, -2.4725e-01,  2.9192e-01,\n",
       "          -8.6309e-03, -8.8094e-02,  7.8582e-02, -1.4702e-01,  1.8489e-01,\n",
       "           3.4682e-01,  9.0617e-01, -6.2635e-03, -6.7841e-02,  7.9005e-01,\n",
       "           4.2095e-01, -1.4924e-01,  2.9555e-01,  2.9191e-01,  2.2767e-01,\n",
       "           3.7441e-01, -3.9379e-02, -3.3540e-01,  6.8580e-02,  2.1775e-02,\n",
       "           5.7403e-01, -7.9143e-01, -2.4789e-01,  3.9712e-02,  3.2272e-01,\n",
       "           1.2775e-01,  4.0751e-01, -4.6374e-01,  5.7863e-02,  3.7124e-01,\n",
       "          -1.2007e+00, -2.9415e-02, -3.1140e-01, -1.0755e-03,  1.2526e-01,\n",
       "          -4.4278e-01, -7.4082e-02, -6.0228e-01, -1.0281e-01,  4.3412e-01,\n",
       "           2.3210e-01, -6.5461e-01,  1.3414e-01,  7.3085e-01, -4.0653e-01,\n",
       "           1.1225e-01, -3.1652e-01,  1.4969e-01,  2.8878e-01,  4.0029e-01,\n",
       "          -3.9644e-02,  2.3110e-01, -3.9129e-03, -5.1225e-01,  3.1435e-01,\n",
       "           5.7257e-01, -1.9029e-02,  2.8252e-03,  2.2493e-01,  3.2604e-01,\n",
       "          -1.9133e-01, -2.9503e-01, -6.9515e-02, -6.7775e-02, -2.5456e-01,\n",
       "           2.0068e-01, -1.9466e-01,  5.2102e-02,  1.2202e-01, -2.0261e-01,\n",
       "           3.3141e-01,  6.6473e-01,  8.7264e-01, -1.1238e-01, -5.5727e-01,\n",
       "           2.5160e-01,  1.7126e-01, -7.0603e-02,  3.3696e-01, -5.8602e-02,\n",
       "          -3.7362e-01,  1.5869e-01,  1.2515e-01, -1.9165e-01,  1.1684e-01,\n",
       "          -1.3141e-01, -7.2971e-02,  8.9393e-03, -5.0550e-01,  3.5926e-01,\n",
       "          -9.0743e-02,  6.8522e-02, -4.0276e-01, -9.6612e-03,  4.9805e-01,\n",
       "          -4.7082e-01,  7.6358e-01, -2.1559e-01,  3.7516e-01, -1.3090e-01,\n",
       "           1.2462e-01, -9.6600e-01, -3.0389e-01,  4.3235e-01, -3.4929e-01,\n",
       "          -3.9428e-01, -7.0174e-03,  6.9262e-03,  2.0757e-01,  5.0747e-01,\n",
       "           3.0740e-01, -1.7350e-01, -8.6439e-01, -4.5825e-01,  1.5008e-01,\n",
       "           9.5087e-03, -9.9254e-02,  3.6242e-01,  2.1016e-01,  1.9399e-01,\n",
       "          -7.8467e-01, -5.1859e-01, -1.8190e-01,  5.7571e-02, -2.6757e-01,\n",
       "           9.1948e-01,  3.2004e-02, -2.2867e-01,  3.0810e-01,  2.2264e-01,\n",
       "           2.8780e-01,  7.6753e-01, -8.6152e-02, -2.2249e-01, -1.3838e-01,\n",
       "           1.7958e-01, -6.7637e-02, -2.3260e-01, -9.2015e-01,  7.0337e-01,\n",
       "           2.4602e-01,  2.2410e-01,  5.0990e-01]], device='mps:0'),\n",
       " tensor([[-9.8608e-02, -6.2949e-01, -5.4737e-01, -4.6335e-01, -3.7454e-01,\n",
       "           1.0248e-01, -2.8097e-01,  4.4707e-01, -1.3869e-01,  5.9022e-01,\n",
       "           3.3979e-01,  3.6518e-02, -7.9829e-02, -1.9712e-01, -4.0961e-01,\n",
       "          -5.4994e-01,  1.7904e-01,  2.6107e-01,  2.6766e-01, -1.3636e-01,\n",
       "           9.8026e-01,  3.6199e-01, -8.5372e-02,  1.8939e-01, -6.9872e-01,\n",
       "           3.9986e-01,  4.5490e-02,  8.0399e-02,  4.6963e-01, -6.9729e-01,\n",
       "          -2.0828e-01,  5.4662e-01, -1.2219e-01,  1.1649e+00, -1.8525e-01,\n",
       "          -2.5129e-01,  4.1644e-01, -2.6874e-01,  8.0973e-01,  4.0156e-02,\n",
       "           6.2909e-01, -1.2839e-01,  2.5974e-01,  2.1204e-01,  2.5654e-01,\n",
       "          -1.6577e-02,  1.0736e-01, -6.2638e-01,  5.4556e-01, -5.1867e-01,\n",
       "          -2.1843e-01, -1.5953e-01,  3.6604e-01,  4.8748e-01, -6.7281e-02,\n",
       "           3.5607e-02,  3.0541e-01,  1.6232e-01, -2.5763e-01,  2.0386e-01,\n",
       "           1.3393e-01,  6.9371e-01,  5.7586e-02, -1.9525e-01, -3.0287e-01,\n",
       "           3.0687e-01, -5.2685e-01, -7.0972e-01, -3.4711e-01, -1.5503e-01,\n",
       "          -6.3145e-02,  3.6453e-01, -2.3968e-01, -8.3683e-02,  7.8001e-02,\n",
       "          -8.8895e-02,  3.9835e-02, -1.7901e-01, -2.7637e-01, -9.8806e-03,\n",
       "           6.4246e-02,  3.2661e-01, -1.9744e-01, -4.1654e-01, -2.1876e-01,\n",
       "           3.0475e-01,  7.8369e-01,  6.3690e-02,  2.5669e-01,  8.5968e-02,\n",
       "           3.1750e-01,  9.4549e-02, -3.4083e-01, -4.9471e-01,  3.1401e-01,\n",
       "          -1.3255e-01,  4.9667e-01, -1.4940e-01, -6.5687e-02, -4.7700e-02,\n",
       "           3.7095e-01,  7.2304e-02,  1.7323e-03,  1.1325e-01, -6.6663e-02,\n",
       "          -4.8542e-01,  1.4204e-01, -2.5931e-01, -6.6520e-01, -1.1522e-01,\n",
       "          -7.9637e-02,  1.3277e-01,  1.5561e-02, -3.1430e-01,  2.6189e-01,\n",
       "          -7.3846e-02,  1.9145e-01, -1.3642e-02, -3.3838e-01,  3.7293e-01,\n",
       "           2.5609e-01, -4.0848e-01,  2.2958e-02,  1.6951e-01,  4.6388e-01,\n",
       "          -2.2394e-01,  6.0049e-01, -2.8694e-01,  2.5392e-01,  4.8866e-01,\n",
       "           8.4207e-02, -6.0898e-02, -9.6990e-02, -1.7751e-01, -3.9547e-01,\n",
       "          -4.2915e-02,  7.3082e-02, -3.2798e-01,  1.0024e-01,  2.1027e-02,\n",
       "          -1.5484e-01, -1.5575e-01, -7.1363e-01, -3.2433e-01,  1.6862e-01,\n",
       "          -2.9423e-01,  2.9391e-01, -7.2839e-02, -5.6183e-01, -3.6233e-01,\n",
       "          -8.1793e-01, -4.5420e-01,  5.0598e-01,  3.1144e-02, -1.4616e-01,\n",
       "          -4.4906e-02,  1.9702e-01,  1.8656e-01, -1.9001e-02,  3.6581e-01,\n",
       "           4.5356e-01, -1.7970e-01,  5.5773e-02, -4.7383e-01,  5.1199e-02,\n",
       "           3.6512e-02, -1.1536e-01, -1.4678e-01, -1.8904e-01,  8.2812e-02,\n",
       "           1.8413e-01, -2.0916e-01, -2.0209e-02, -4.3147e-01,  1.4387e-01,\n",
       "          -9.5740e-02,  8.7516e-02,  2.8605e-01, -3.1460e-01, -3.4357e-01,\n",
       "          -4.4457e-01, -2.0415e+00,  1.3037e-01, -1.6708e-01, -1.9640e-01,\n",
       "          -7.8661e-01,  5.6989e-01,  4.0038e-01,  2.3491e-01,  1.2055e-02,\n",
       "          -4.1441e-01, -1.3256e-01, -2.2065e-01,  2.0346e-01, -1.5937e-01,\n",
       "           4.1104e-01,  3.6856e-01,  2.1222e-01, -8.2505e-02, -5.3417e-02,\n",
       "          -3.3993e-01,  1.9027e-01,  6.6698e-02,  4.3789e-02,  1.3752e-02,\n",
       "           8.6470e-02, -3.4516e-01,  1.5865e-01,  6.3413e-01, -3.2035e-01,\n",
       "          -2.6413e-01, -1.5432e-01, -3.1116e-01, -1.1825e-01, -2.1011e-01,\n",
       "           7.5833e-01, -1.0075e-01, -1.9090e-01,  4.1462e-01,  2.9186e-01,\n",
       "          -2.0576e-01,  2.6479e-02, -1.7243e-01, -4.6025e-01, -8.4004e-02,\n",
       "           8.3166e-01, -2.6809e-01, -1.8435e-01, -1.3576e-01, -2.8072e-01,\n",
       "           9.9627e-02, -3.9657e-01,  7.0865e-01,  1.3131e-01, -2.4437e-01,\n",
       "          -1.3250e-01, -4.9085e-01, -3.6081e-02, -6.5221e-02,  2.7934e-01,\n",
       "          -3.2480e-01, -4.4413e-01, -6.2940e-01,  9.8728e-02,  4.2580e-01,\n",
       "           1.0494e-01, -2.2581e-01,  5.8588e-01,  3.7420e-01,  5.5104e-01,\n",
       "          -7.6877e-02,  1.8477e-03,  4.0139e-01,  4.0454e-02, -3.1153e-01,\n",
       "           4.0137e-02,  1.1839e-01, -6.1671e-02,  2.8916e-01, -5.0007e-01,\n",
       "           2.1460e-01,  1.3048e-01,  4.2285e-02,  4.0869e-01, -3.3156e-01,\n",
       "           3.3759e-01, -6.4578e-04, -4.2352e-01, -2.2792e-01,  3.1587e-01,\n",
       "          -3.4318e-01,  4.7882e-01, -1.1497e-01, -1.3533e-01, -6.5793e-01,\n",
       "          -3.0023e-01,  9.9027e-02, -5.2265e-01,  4.0820e-02, -8.5922e-01,\n",
       "          -1.4457e-01,  4.3467e-01, -1.1843e-01,  2.8682e-01, -4.3268e-01,\n",
       "           2.7079e-01,  1.1150e-01,  1.0583e-01, -1.8427e-01, -4.8035e-01,\n",
       "           6.4368e-03,  3.5471e-01, -1.9635e-01,  6.7501e-02,  1.8728e-02,\n",
       "           1.4821e-01,  7.4914e-02, -1.4411e-01,  1.8116e-02, -3.9952e-01,\n",
       "           1.2334e-01,  2.7083e-01, -6.4203e-02, -1.1696e-01, -2.5127e-01,\n",
       "           8.1457e-02,  5.7341e-02,  4.5826e-01, -1.5574e-01, -4.7792e-01,\n",
       "           1.3602e-02,  5.6159e-01,  4.2917e-02,  9.5691e-02, -4.1476e-02,\n",
       "          -1.3511e+00, -3.8201e-01,  1.3646e-01, -2.3684e-01, -1.6250e-01,\n",
       "           1.4728e-01, -4.5766e-02, -1.0990e-01,  1.3434e-01, -3.3727e-01,\n",
       "           4.9913e-01, -7.4794e-03, -9.4830e-02,  2.6318e-01, -2.9595e-01,\n",
       "          -2.6245e-01, -5.5022e-02, -5.5963e-02, -1.9048e-01, -2.1108e-01,\n",
       "           1.0760e-01,  2.4895e-01, -1.6610e-02,  2.9324e-01, -4.2315e-01,\n",
       "           6.9357e-01,  2.0403e-02, -3.6256e-01,  1.8028e-01, -4.6817e-01,\n",
       "          -2.2470e-01,  6.2020e-01, -2.1111e-01,  1.7508e-01,  7.0164e-02,\n",
       "           2.2583e-01, -1.6263e-01, -2.5922e-01, -1.5555e-01,  5.6515e-01,\n",
       "           9.7645e-02,  3.8631e-01, -3.1465e-01,  4.6445e-01,  6.0830e-01,\n",
       "          -4.3876e-01,  2.7575e-01, -3.0324e-01, -3.5509e-01, -1.4900e-01,\n",
       "           3.7937e-01,  2.4503e-01, -2.4568e-01,  1.2754e-01,  2.1491e-01,\n",
       "           2.4189e-01, -2.8123e-01, -1.4375e-01,  1.5623e-01, -1.6800e-02,\n",
       "          -1.6159e-01, -5.9860e-01,  4.8387e-02, -4.0540e-01, -3.7598e-01,\n",
       "          -5.5542e-02, -9.6940e-02,  2.3018e-01, -3.4557e-01, -2.8109e-01,\n",
       "          -4.1082e-01,  1.5328e-01,  7.9392e-01, -8.2566e-02,  1.3355e-02,\n",
       "          -2.4290e-01,  1.3134e-01, -3.9568e-01,  5.6959e-01,  2.4837e-01,\n",
       "          -5.6746e-02,  5.0092e-01, -5.9799e-01,  3.2502e-01,  1.6325e-01,\n",
       "          -2.5296e-01,  7.9720e-02, -1.1265e-01,  4.2524e-01, -1.9241e-01,\n",
       "          -2.6666e-01,  1.8088e-01,  2.9568e-01,  8.0022e-02,  3.3680e-02,\n",
       "           8.9946e-02,  7.7279e-01, -5.6155e-01, -1.1355e-01, -3.2412e-02,\n",
       "           4.6990e-01, -1.4868e-01,  1.7842e-01, -2.5184e-01, -4.5494e-01,\n",
       "           2.8096e-01,  7.6220e-03,  3.0259e-02,  1.5848e-01, -1.1470e+01,\n",
       "           1.7761e-02,  7.9956e-01, -5.9407e-01,  3.9142e-02, -7.9952e-04,\n",
       "           1.4375e-01, -2.8145e-01,  3.6361e-03, -2.5851e-01, -3.7282e-01,\n",
       "           1.6253e-01,  5.8747e-02,  6.8827e-02, -1.9252e-01, -1.2168e-01,\n",
       "          -4.4456e-02, -9.0133e-01,  1.4013e-01, -2.9238e-01,  3.1633e-01,\n",
       "          -4.6944e-01,  3.5652e-01,  1.0775e-01,  4.8818e-02, -3.3732e-01,\n",
       "          -6.3534e-02, -4.1684e-01,  2.3683e-01,  1.0364e-01, -1.6207e-01,\n",
       "          -1.3703e-01, -1.5828e-01,  4.6034e-01, -1.8993e-01, -4.8947e-01,\n",
       "           1.5862e-01, -7.0142e-02,  4.5043e-01, -1.6798e-02, -2.3410e-01,\n",
       "           5.0011e-02, -4.1456e-01,  3.0195e-01,  6.4668e-02,  5.0457e-01,\n",
       "           4.7572e-01,  6.3490e-01,  1.0739e-01, -2.1848e-01,  4.9891e-02,\n",
       "          -8.9429e-01,  3.3171e-01, -9.8824e-01, -2.6344e-01,  3.6779e-02,\n",
       "           7.4622e-01, -4.6007e-01,  9.9323e-03,  8.2971e-02, -3.9312e-01,\n",
       "           2.4107e-01, -5.5413e-02,  1.0686e-01, -6.2270e-01, -1.1555e-01,\n",
       "           3.6365e-01,  1.0164e-01,  1.3567e-01, -3.5856e-01, -1.3223e-02,\n",
       "          -4.3049e-02,  5.9870e-01,  1.5647e-01, -2.5151e-01,  1.0429e-01,\n",
       "          -8.0721e-03,  2.3224e-01,  6.2748e-01,  4.7218e-02,  1.3164e-01,\n",
       "           1.7459e-01,  5.1672e-01, -5.4105e-01, -4.7941e-01, -1.2530e-01,\n",
       "           6.3722e-02,  2.8461e-01,  8.3486e-02,  3.6163e-01, -3.9872e-01,\n",
       "          -3.1281e-01, -7.8561e-02,  3.1281e-02,  5.0728e-02,  1.2098e+00,\n",
       "          -5.8519e-01, -2.2122e-01, -1.0256e-01, -9.1470e-02, -1.3571e-01,\n",
       "          -3.9828e-02, -3.3651e-01, -4.2370e-01,  3.3182e-01,  1.1019e-01,\n",
       "           1.7328e-01,  1.5282e-02,  3.9299e-01,  1.9321e-01, -5.2099e-01,\n",
       "           2.8514e-01,  1.4470e-01,  4.3958e-01, -1.8846e-01, -4.1365e-01,\n",
       "          -1.0991e-01,  3.3157e-01, -3.8281e-01, -3.4973e-01, -1.0983e-01,\n",
       "           3.0412e-02,  5.9583e-02, -5.8809e-02, -3.7567e-01,  2.5850e-01,\n",
       "          -6.2094e-01, -9.3514e-03, -4.2726e-01, -1.1183e-01,  4.7884e-01,\n",
       "           1.2318e-01, -3.6857e-02,  1.1137e-01, -4.2935e-01, -6.7978e-01,\n",
       "          -6.4684e-01, -7.4975e-02,  5.2269e-02, -3.7956e-01, -1.6842e-02,\n",
       "          -1.8671e-01,  6.6050e-02, -3.6548e-01,  1.8633e-01, -8.2184e-02,\n",
       "           7.5332e-01,  2.3533e-01, -1.8486e-01,  1.2410e-01, -1.4424e-01,\n",
       "          -3.6447e-01, -2.1959e-01, -2.4676e-01,  2.2621e-01, -5.0366e-01,\n",
       "          -1.3080e-01,  1.8011e-01,  1.1438e-01,  6.4878e-03, -3.4448e-01,\n",
       "          -1.2932e-01,  2.0570e-01, -1.3707e-01,  3.0997e-01,  1.7373e-01,\n",
       "           2.0329e-01, -6.4335e-02, -4.6287e-01,  8.7786e-02,  4.5179e-01,\n",
       "          -1.1348e-01, -2.7396e-01,  3.8658e-02, -7.0830e-01,  2.1125e-01,\n",
       "           3.8231e-01,  8.2746e-01, -2.5998e-01, -5.3996e-01,  4.6996e-01,\n",
       "           6.2219e-01, -4.6785e-01, -5.7024e-02, -2.1133e-01,  1.2442e-01,\n",
       "          -1.0488e-01, -3.4183e-01,  2.6033e-01, -2.5177e-01,  4.2375e-01,\n",
       "          -9.9348e-02,  2.0811e-01, -1.9413e-01,  4.1807e-01,  9.7647e-02,\n",
       "          -2.0238e-01, -3.5246e-02, -6.2741e-01, -4.3863e-01,  2.3873e-01,\n",
       "           2.6324e-01, -1.5522e-01, -2.3160e-02, -1.2275e-01,  1.1149e-01,\n",
       "           4.8271e-01,  8.5393e-01,  8.1221e-02,  1.4584e-01,  3.8317e-01,\n",
       "           2.3356e-01, -2.7630e-01,  5.9953e-02,  2.9270e-01,  4.0766e-01,\n",
       "           3.5259e-01,  5.3145e-01, -3.4176e-01,  7.2203e-02, -1.6344e-01,\n",
       "           2.0465e-01, -5.7619e-01, -7.6698e-01, -1.3126e-01,  4.3082e-01,\n",
       "           1.5541e-02,  2.9468e-01, -2.7851e-01,  3.4408e-01,  6.3093e-01,\n",
       "          -1.1306e+00, -5.9181e-02, -3.9846e-02, -9.3994e-02, -6.2591e-02,\n",
       "          -5.4897e-01, -2.2413e-01, -3.5234e-01, -2.4418e-01,  5.2852e-01,\n",
       "           5.8811e-01, -8.5286e-01,  1.6085e-01,  2.2953e-01, -1.5005e-01,\n",
       "           9.4328e-02, -9.1644e-02,  3.2863e-02,  5.0134e-01,  1.5175e-01,\n",
       "          -5.7932e-01,  4.5778e-01,  1.5595e-01, -4.3637e-01,  2.6165e-01,\n",
       "           4.1776e-01, -1.8687e-02,  1.5952e-01,  2.2085e-01, -4.9751e-01,\n",
       "          -1.0141e-02, -4.6200e-01, -1.0629e-01,  1.0751e-01, -6.7537e-02,\n",
       "           2.4815e-02, -4.9794e-01, -6.6236e-02,  3.4484e-01,  2.9505e-03,\n",
       "           5.0621e-01,  6.0668e-01,  7.6526e-01, -5.6484e-02, -2.1306e-01,\n",
       "           3.3852e-01,  2.5943e-01, -9.1834e-02,  5.2227e-01, -2.1885e-02,\n",
       "          -1.6835e-01,  3.1658e-01,  2.3124e-02, -2.2045e-01,  1.7294e-01,\n",
       "          -2.5761e-01,  2.6423e-01,  1.0870e-01, -4.9724e-01,  3.6477e-01,\n",
       "           3.7087e-01,  1.5082e-01, -3.0923e-01,  1.8171e-01,  3.9828e-01,\n",
       "          -3.1100e-01,  6.8356e-01,  1.6607e-04,  2.2973e-01,  6.3476e-02,\n",
       "          -1.1506e-01, -5.5647e-01, -3.1374e-01,  2.1509e-01,  9.2105e-02,\n",
       "          -5.0547e-01, -9.5287e-02,  1.5106e-01, -4.9544e-01,  7.8060e-01,\n",
       "           6.2385e-01, -3.7763e-01, -3.5664e-01, -7.1127e-02,  1.1295e-01,\n",
       "           1.2269e-01, -3.3879e-01,  2.9572e-01,  1.1312e-01,  1.0076e-01,\n",
       "          -1.0219e+00, -2.9326e-01,  2.4530e-01,  4.9147e-02, -4.2291e-01,\n",
       "           5.9939e-01,  7.9206e-02,  5.0359e-02,  3.8428e-01,  3.0534e-01,\n",
       "           1.4161e-01,  4.3910e-01,  7.0775e-02, -2.1375e-01,  6.5311e-02,\n",
       "           2.5600e-01, -3.0970e-01, -3.2766e-01, -6.3757e-01,  5.3684e-01,\n",
       "           1.5068e-01,  8.4160e-01,  6.5313e-01]], device='mps:0'),\n",
       " tensor([[-1.2744e-01, -3.6886e-01, -5.8966e-01, -3.8612e-01, -3.9839e-01,\n",
       "          -2.6093e-02, -6.9988e-01,  4.7690e-01,  1.4660e-01,  3.5934e-01,\n",
       "           2.2341e-01,  2.1011e-01, -1.9782e-02, -1.3628e-01, -3.2271e-01,\n",
       "          -2.9192e-01,  1.9477e-01,  1.1105e-02,  4.7480e-01,  3.7473e-01,\n",
       "           2.3759e-01,  1.2863e-01, -2.7636e-01, -8.4524e-03, -5.7920e-01,\n",
       "           5.2122e-01, -1.9992e-01,  2.4295e-01,  4.0208e-01, -8.7319e-01,\n",
       "          -1.5724e-01,  3.4887e-01, -1.1519e-01,  9.5447e-01,  1.0815e-01,\n",
       "          -2.1800e-01,  1.9717e-01, -4.1613e-01,  4.4315e-01, -7.8625e-02,\n",
       "           4.9483e-01, -1.1715e-01, -6.8274e-02,  5.8175e-02,  3.3054e-01,\n",
       "           1.0223e-01,  4.2262e-01, -8.6963e-01,  5.7195e-02, -8.5951e-01,\n",
       "           1.4489e-01,  8.5214e-02,  5.6111e-02,  3.9715e-01, -8.3769e-02,\n",
       "           1.1703e-03,  2.0640e-01,  3.2689e-01, -1.7768e-01,  2.4573e-01,\n",
       "          -1.4667e-01,  5.9904e-01,  2.4168e-01, -2.2416e-01, -9.1909e-02,\n",
       "           2.3362e-01, -4.1027e-01, -8.0082e-02, -5.9253e-01, -3.8848e-02,\n",
       "          -3.0027e-01,  5.0558e-02,  7.0746e-02, -4.2309e-01,  6.6069e-01,\n",
       "          -2.8641e-02,  3.6976e-01,  3.1896e-01, -5.6682e-01, -1.4716e-01,\n",
       "           2.0574e-02,  5.4220e-01, -7.3909e-02, -9.7348e-02, -6.0546e-02,\n",
       "           1.6835e-01,  4.3324e-01,  1.8741e-01,  4.9216e-01, -2.7530e-01,\n",
       "           3.4666e-01,  2.0740e-01, -4.8388e-01, -3.4634e-01,  3.2555e-01,\n",
       "          -2.2010e-01,  2.0870e-02,  3.5313e-02,  7.2057e-02, -4.8509e-01,\n",
       "           3.2062e-02,  3.0252e-01,  3.4019e-02,  2.2994e-01, -6.9579e-01,\n",
       "          -3.2647e-01,  1.2075e-01, -1.2859e-01, -6.9336e-01, -3.0060e-01,\n",
       "          -3.1885e-02, -1.9062e-01,  2.9386e-01, -1.7405e-01,  1.9779e-01,\n",
       "          -1.3300e-02,  1.4326e-01,  9.9442e-02, -1.3655e-02,  4.7251e-02,\n",
       "          -2.4289e-01, -2.2602e-01,  2.7060e-01,  9.9331e-02,  5.9831e-01,\n",
       "          -1.0524e-01,  1.9171e-01,  1.0947e-01,  6.0067e-01,  1.3412e-01,\n",
       "           5.4718e-02,  1.1605e-01, -2.3171e-01, -2.8301e-02,  1.7790e-01,\n",
       "           1.2551e-01,  9.3539e-03, -2.0242e-01, -6.2051e-02, -5.5221e-02,\n",
       "          -1.9351e-01, -7.9696e-02,  1.2172e-02,  6.4713e-02,  2.3389e-02,\n",
       "          -2.9732e-01,  6.4722e-01, -2.3968e-01, -4.6100e-01, -2.8527e-01,\n",
       "          -5.9287e-01, -3.3712e-01, -1.6464e-01, -1.8269e-01, -2.5387e-01,\n",
       "          -4.1172e-02,  2.9951e-01,  7.9301e-03,  2.9821e-02,  2.7362e-01,\n",
       "           7.7804e-01,  3.3612e-02,  2.9417e-01, -3.8371e-01, -1.7314e-01,\n",
       "           3.5786e-01, -5.2665e-02,  1.7453e-02,  3.4050e-01,  4.9258e-02,\n",
       "           2.5971e-01, -4.6870e-01,  2.9129e-01, -1.9721e-02,  3.3761e-01,\n",
       "          -4.5630e-01,  6.3396e-01,  1.8206e-01, -2.2251e-01,  1.3218e-01,\n",
       "          -2.7067e-01, -1.9681e+00,  2.7409e-01, -5.3402e-01, -3.3687e-01,\n",
       "          -5.4313e-01,  5.0271e-01,  1.1955e-02,  3.0021e-01, -3.5581e-02,\n",
       "          -1.5728e-02, -2.0107e-01, -1.7325e-01,  2.4484e-01, -1.5260e-01,\n",
       "           3.4010e-01,  4.3245e-01,  2.2335e-01,  2.0987e-02,  7.1640e-02,\n",
       "          -3.5186e-01,  2.1606e-01,  3.1683e-01,  1.2299e-01,  4.6625e-01,\n",
       "           4.3921e-01, -1.9788e-01,  1.4637e-01,  3.0379e-01,  1.7299e-01,\n",
       "           4.2774e-02, -4.8766e-02, -6.0864e-01, -4.3481e-01,  8.4589e-02,\n",
       "           6.1821e-01, -3.7447e-02, -1.8553e-01,  3.0917e-01,  2.7359e-01,\n",
       "          -1.2981e-01, -7.1170e-02, -2.2230e-01, -2.3223e-01,  2.4117e-01,\n",
       "           2.4292e-01, -1.9781e-01,  6.7176e-02, -4.5396e-01, -1.8121e-01,\n",
       "           1.2644e-01, -4.4649e-01,  4.4478e-01,  5.0948e-01, -9.6100e-02,\n",
       "          -2.2433e-01, -7.5417e-01,  1.1847e-01, -4.5513e-01, -5.0902e-02,\n",
       "          -2.9625e-01, -4.4381e-01, -6.4586e-01,  8.3141e-02,  2.4983e-02,\n",
       "           4.1063e-01, -1.4773e-01,  5.3868e-01,  7.7313e-01,  5.1901e-01,\n",
       "          -3.1456e-02, -1.4849e-01,  4.0876e-01,  3.4347e-01, -1.0435e-01,\n",
       "          -2.8603e-01, -9.3219e-02, -1.8155e-01,  4.9330e-01, -1.1836e-01,\n",
       "           1.2650e-02, -3.6974e-01,  2.9917e-01,  6.0181e-01, -6.0250e-01,\n",
       "           4.5421e-01,  4.0299e-01, -3.2465e-01, -2.8221e-01,  2.0783e-01,\n",
       "          -5.4009e-01,  3.5731e-01,  4.4142e-01, -2.1193e-01, -5.1534e-01,\n",
       "          -5.3220e-01,  1.3967e-01, -4.5681e-01,  1.6254e-01, -9.4993e-01,\n",
       "          -7.2121e-01,  1.9764e-01,  2.7853e-02,  2.2033e-01,  2.3177e-01,\n",
       "           2.9691e-01,  2.6055e-01, -3.0842e-01, -4.7068e-03, -5.5665e-01,\n",
       "          -1.3382e-01,  1.9658e-01, -5.9442e-01,  1.9406e-01,  3.1458e-01,\n",
       "          -9.5471e-02, -2.3677e-01,  1.1607e-01,  2.6327e-01, -3.6988e-01,\n",
       "          -3.4352e-01,  1.1162e-01, -1.2854e-01,  6.2280e-02,  1.5006e-01,\n",
       "           1.9091e-01,  2.6202e-01,  6.1249e-01, -1.9113e-01, -3.4774e-01,\n",
       "          -2.3449e-01,  5.6052e-01, -9.7700e-02,  2.6829e-01, -2.4674e-03,\n",
       "          -1.0923e+00, -1.6681e-01, -1.3233e-01, -1.6892e-01, -2.3808e-01,\n",
       "           3.2043e-01,  3.8827e-02, -1.5429e-03, -5.3960e-03, -1.8113e-01,\n",
       "          -1.2497e-01,  6.3763e-01,  4.8824e-01,  1.2392e-01, -1.6461e-01,\n",
       "          -3.1110e-01,  2.6912e-01, -4.7266e-01, -3.9657e-02, -3.9847e-01,\n",
       "           2.2630e-01,  5.4890e-02, -2.2261e-01, -1.6871e-01, -9.3712e-02,\n",
       "           8.4236e-01, -2.8847e-01,  3.3317e-02,  3.9876e-01,  1.7333e-02,\n",
       "          -1.2513e-01,  2.1090e-01, -2.1270e-01, -2.7171e-02, -6.2933e-02,\n",
       "          -1.5418e-01, -3.6306e-01, -5.9896e-01, -1.6418e-01,  7.6700e-01,\n",
       "           3.5505e-01,  6.3605e-01, -4.3754e-01,  1.1850e-01,  4.0720e-01,\n",
       "          -3.0629e-01,  3.4225e-01, -4.0022e-01, -2.6250e-01,  1.8061e-01,\n",
       "           2.2927e-01,  2.0317e-01, -2.0997e-01,  2.1466e-01, -4.5316e-02,\n",
       "          -2.4904e-01, -2.3962e-01, -1.5913e-01,  1.2842e-01,  1.9884e-01,\n",
       "          -1.6377e-01, -4.9167e-01,  4.6913e-01, -4.6686e-01, -4.4366e-01,\n",
       "          -1.3138e-01, -3.9469e-02,  2.6926e-02, -2.3624e-01, -6.2990e-01,\n",
       "          -5.3244e-01,  1.0719e-01,  7.8798e-01,  2.2912e-01,  2.7124e-01,\n",
       "           3.5317e-01, -1.5187e-01, -4.7546e-01,  5.5343e-01,  1.1960e-01,\n",
       "          -7.3996e-01,  5.0864e-01, -8.7861e-02,  4.5199e-01,  1.5250e-01,\n",
       "          -1.0490e-01, -3.9812e-01, -1.6321e-01,  5.7516e-02,  1.1066e-01,\n",
       "          -1.0170e-01,  2.6733e-01,  2.7944e-01, -9.1626e-02, -1.2806e-01,\n",
       "          -3.6408e-01,  6.5188e-01, -1.9893e-01,  3.3909e-01, -7.6062e-02,\n",
       "           4.4720e-01, -9.0910e-02,  2.9596e-01, -4.1676e-02, -3.2361e-01,\n",
       "           3.2309e-01,  4.3930e-01,  6.6604e-02,  2.8283e-01, -1.1342e+01,\n",
       "           2.5073e-01,  3.3695e-01, -2.7721e-01, -3.3814e-01, -2.0797e-01,\n",
       "           1.4309e-02, -1.1250e-01,  5.9957e-02, -5.2643e-02,  8.4555e-02,\n",
       "           5.7257e-02, -1.5938e-01, -1.2961e-01, -3.1405e-01, -2.5103e-01,\n",
       "          -3.1555e-01, -7.8943e-01, -3.5223e-02, -1.1858e-01,  3.7357e-01,\n",
       "          -1.4416e-01, -1.0051e-01,  1.0652e-01,  1.8536e-02, -2.1217e-01,\n",
       "          -1.2537e-03, -3.9634e-01,  2.5957e-01,  1.0211e-01,  1.2887e-01,\n",
       "           6.0551e-02, -2.1079e-01, -3.0225e-01, -4.0741e-01, -1.6218e-01,\n",
       "          -5.7135e-02,  2.3947e-01,  6.7048e-01, -4.9515e-02, -2.8464e-01,\n",
       "           8.6184e-02, -3.8013e-01,  5.4413e-01,  1.3969e-01,  6.4884e-01,\n",
       "           1.6804e+00,  8.6468e-01,  1.2536e-03,  9.2417e-02,  9.4459e-02,\n",
       "          -1.1914e+00,  5.3354e-01, -6.5743e-01,  2.3642e-01, -8.0591e-02,\n",
       "           6.6101e-01, -5.4793e-01,  9.8762e-02,  2.3464e-02, -4.5766e-02,\n",
       "           2.5879e-01, -1.4390e-01,  2.1979e-01, -3.4649e-01,  1.4607e-02,\n",
       "           1.7742e-01, -4.1623e-01,  3.9326e-01, -2.2080e-01, -2.9110e-01,\n",
       "           1.9209e-01,  4.3571e-01,  1.8156e-01, -3.9204e-01,  4.6669e-01,\n",
       "           1.4567e-01, -1.2916e-01,  2.4630e-01, -1.1670e-01, -3.6927e-01,\n",
       "           2.4790e-01,  2.9172e-01, -3.9819e-01, -7.9393e-01, -2.4674e-01,\n",
       "           1.4939e-01,  8.4515e-02,  9.8346e-02,  3.2097e-01, -3.4363e-01,\n",
       "          -1.1931e-01, -3.0171e-01,  8.9557e-02,  3.1143e-02,  6.0974e-01,\n",
       "          -3.9739e-01, -3.1709e-01,  1.8110e-01, -3.4176e-02, -5.6142e-02,\n",
       "          -3.1976e-01, -3.9476e-02, -3.5016e-01,  2.9538e-01,  1.3379e-01,\n",
       "          -3.1464e-01, -2.4862e-01,  2.8423e-03, -2.0509e-02, -2.8011e-01,\n",
       "           1.6176e-01,  1.3696e-01, -3.9499e-02, -5.6251e-01, -2.6655e-01,\n",
       "           1.4900e-01,  6.5725e-01, -3.7705e-01, -4.7637e-01, -1.6537e-01,\n",
       "          -3.7458e-03,  2.7367e-01, -4.9678e-02, -3.8691e-01,  2.4280e-01,\n",
       "          -7.4630e-01,  3.7875e-02, -1.2885e-01,  2.5627e-01,  3.9825e-01,\n",
       "           2.1553e-01,  2.2306e-01,  6.3958e-01, -2.4253e-01, -7.5381e-01,\n",
       "          -7.2268e-01, -3.7877e-01,  1.3895e-01, -2.3283e-01,  1.5624e-01,\n",
       "          -7.5009e-02, -2.0488e-02, -2.2027e-01,  5.4608e-01, -1.5348e-01,\n",
       "           6.5817e-01,  8.1926e-02, -2.2133e-01,  2.8428e-01, -2.6495e-01,\n",
       "          -2.0247e-01, -1.2891e-01, -3.3237e-02,  1.2341e-02, -4.8486e-01,\n",
       "          -6.8411e-02,  1.2113e-02, -3.2645e-01, -1.6233e-01,  3.1744e-01,\n",
       "           1.5508e-01,  2.2397e-01,  8.1840e-02,  1.3771e-01,  1.5091e-01,\n",
       "          -5.8332e-02, -3.4181e-01, -4.6389e-01, -1.4840e-01, -2.1854e-02,\n",
       "           2.9120e-01,  1.5057e-01, -1.6943e-01, -3.4826e-01, -1.5630e-01,\n",
       "           3.3120e-01,  1.0920e+00, -4.2546e-01, -5.0576e-02,  4.3019e-01,\n",
       "           6.1310e-01, -2.8793e-01, -5.9560e-02, -2.5380e-01, -1.8582e-01,\n",
       "           1.2802e-01, -5.5977e-01,  3.4554e-01, -3.5107e-01,  4.4924e-01,\n",
       "           1.4727e-01,  1.2839e-01, -3.6234e-01,  2.8734e-01,  8.8045e-02,\n",
       "          -6.2931e-01, -1.6030e-01, -1.6852e-01, -4.3430e-01,  5.2212e-01,\n",
       "          -3.1425e-01, -1.1537e-01, -1.8644e-01, -2.3570e-01,  6.2245e-01,\n",
       "           3.0525e-01,  1.3974e+00,  1.4168e-01, -6.3228e-02,  6.4992e-01,\n",
       "           1.7198e-01, -5.9372e-01,  1.0333e-01, -3.2490e-01,  3.3897e-01,\n",
       "           2.3329e-01,  1.8528e-02, -3.0002e-01,  7.2502e-03,  1.6217e-01,\n",
       "           3.4970e-01, -6.9435e-01, -9.1765e-01, -2.3259e-01,  5.1759e-01,\n",
       "          -3.9585e-01,  3.3931e-01, -2.6317e-01,  2.7665e-01,  5.2981e-01,\n",
       "          -9.7005e-01, -5.9644e-02, -5.6020e-01,  3.9886e-01, -5.1046e-02,\n",
       "          -5.1848e-01, -2.9517e-01,  2.6993e-01, -6.0905e-01,  6.8855e-01,\n",
       "           5.6781e-01, -7.0979e-01, -9.0777e-02,  6.0233e-01, -3.4466e-01,\n",
       "           2.7055e-01, -1.1937e-01,  9.5849e-02,  1.3332e-01, -5.8360e-02,\n",
       "          -4.7854e-01,  4.4713e-02, -5.7373e-01, -8.0471e-01,  2.4943e-01,\n",
       "           4.7264e-01, -1.9956e-01,  2.9839e-01,  8.8237e-02, -5.7208e-01,\n",
       "          -1.7800e-01, -3.2724e-01, -1.0209e-01,  2.6329e-01, -2.3215e-01,\n",
       "          -1.8293e-01, -2.3268e-01,  2.2084e-01,  2.0970e-01, -6.6922e-02,\n",
       "           6.1877e-01,  4.4335e-01,  3.7257e-01, -1.0860e-01,  2.8005e-02,\n",
       "           7.9098e-02,  3.2362e-01, -2.6026e-02,  6.8501e-01, -2.4993e-01,\n",
       "          -4.3910e-01,  2.5426e-01,  1.2443e-01, -2.5166e-01,  4.4084e-02,\n",
       "          -4.7607e-01,  2.5556e-01, -8.7111e-04, -2.6517e-01,  5.1456e-01,\n",
       "           1.6735e-01,  1.6049e-01, -5.1357e-01,  1.8915e-02,  3.8806e-01,\n",
       "          -1.2695e-01,  5.1835e-01,  4.0893e-02,  6.1101e-01, -4.2597e-01,\n",
       "           5.2654e-02, -6.3957e-01, -1.6493e-01, -1.2452e-01,  2.8040e-01,\n",
       "          -5.9614e-01,  1.4522e-02,  6.0262e-01,  6.3799e-02,  6.5544e-01,\n",
       "           5.3148e-01, -4.2138e-01, -7.2301e-01,  9.2347e-02,  1.0394e-01,\n",
       "          -1.1149e-01, -2.5791e-01, -5.1979e-01,  8.5114e-03,  3.5523e-01,\n",
       "          -7.4568e-01, -2.0172e-01, -4.5950e-02, -3.4070e-02, -2.5010e-01,\n",
       "           5.7131e-01,  1.4834e-01, -6.8530e-02,  2.6508e-01,  1.1563e-01,\n",
       "           7.7642e-03,  4.2474e-01,  8.0413e-02, -6.4725e-01, -2.6003e-01,\n",
       "           2.0198e-01, -5.0822e-01, -1.7224e-01, -4.8987e-01,  2.1055e-01,\n",
       "           1.2388e-01,  9.0132e-01,  6.6990e-01]], device='mps:0'),\n",
       " tensor([[ 2.9008e-03, -4.3318e-01, -5.7510e-01, -2.9224e-01, -3.0295e-01,\n",
       "          -3.0902e-01, -7.0551e-01,  1.4946e-01,  2.0271e-01,  4.7205e-01,\n",
       "           2.5574e-01,  7.7520e-02, -6.7837e-02, -1.7097e-01, -5.3232e-01,\n",
       "          -5.6924e-01, -5.1277e-01,  3.1787e-01,  4.3404e-01,  1.6802e-01,\n",
       "           6.5901e-01,  1.4081e-01, -9.3628e-02,  2.6067e-01, -7.1014e-01,\n",
       "          -2.1855e-03,  6.6080e-02, -4.4363e-01,  1.7925e-01, -6.4305e-01,\n",
       "           3.2709e-02,  6.0367e-01, -3.9635e-01,  8.9624e-01, -3.3298e-01,\n",
       "          -1.5641e-01,  1.7262e-01,  4.6312e-03,  8.1842e-01,  2.5720e-02,\n",
       "           7.3634e-01, -1.5888e-01,  5.2904e-01,  1.1068e-01,  2.4160e-01,\n",
       "          -2.8973e-02, -1.6938e-01, -3.2165e-01,  3.0245e-01, -3.9500e-01,\n",
       "          -1.2622e-01, -3.3266e-02,  6.8284e-01,  4.1331e-01, -8.1996e-01,\n",
       "          -5.5008e-01, -3.9824e-01,  3.1298e-01, -2.6294e-01,  1.0984e-01,\n",
       "          -1.9660e-01,  7.2583e-01,  1.4372e-01,  1.6310e-01, -4.2463e-01,\n",
       "           2.0633e-01, -3.4804e-01, -3.3247e-01, -3.7174e-01, -9.6018e-03,\n",
       "           1.6379e-01,  2.0134e-01,  1.7143e-01, -2.8161e-01,  4.1567e-01,\n",
       "          -2.3257e-01,  5.0418e-01,  1.0886e-01, -6.3116e-01,  3.2847e-02,\n",
       "           2.8541e-01,  1.0365e-01, -2.7131e-01, -2.5437e-01,  1.1793e-01,\n",
       "           2.2065e-01,  6.8505e-01,  1.8298e-01,  2.8383e-01, -1.2345e-01,\n",
       "           5.6294e-01,  3.3569e-01, -9.9418e-01, -1.8295e-01,  2.9424e-01,\n",
       "          -4.9301e-01,  2.6005e-01, -3.2882e-01,  2.0763e-01,  2.2080e-01,\n",
       "           1.1194e-01,  4.1759e-01,  1.8716e-03,  3.8913e-01, -7.8902e-01,\n",
       "          -4.3756e-01,  1.8094e-01, -8.4630e-02, -8.3873e-01,  1.4949e-01,\n",
       "          -1.5590e-01, -9.5115e-02,  3.2331e-01, -2.6191e-01, -1.0773e-01,\n",
       "           1.1039e-01, -1.0347e-01, -8.2128e-02, -2.6510e-01,  5.2203e-01,\n",
       "           4.4625e-01, -5.6405e-01, -8.0613e-02,  2.3796e-02,  4.0746e-01,\n",
       "          -2.5385e-01,  1.2245e-01, -5.1367e-02,  3.9581e-01,  5.9330e-01,\n",
       "          -6.6486e-02,  3.3022e-01, -1.3930e-01,  2.6400e-01, -2.7601e-01,\n",
       "          -3.3297e-02,  4.4827e-01, -5.8564e-01,  2.9972e-01,  1.1670e-01,\n",
       "          -1.1354e-02, -2.0174e-01, -2.2868e-01,  4.2478e-02, -8.8563e-02,\n",
       "          -2.2647e-01, -6.1850e-02,  1.2947e-01,  8.7484e-02, -2.0371e-01,\n",
       "          -8.4197e-01, -3.9703e-01,  2.1804e-01, -2.5167e-02,  5.2559e-02,\n",
       "          -1.9434e-01,  1.7957e-01,  9.4118e-02, -5.7709e-01, -1.7865e-02,\n",
       "           1.4812e-01,  7.9035e-02,  1.1075e-01, -8.5492e-01, -2.8576e-01,\n",
       "           2.2605e-01,  2.3150e-01, -2.6078e-02, -4.1231e-01,  8.3800e-02,\n",
       "           2.5207e-01, -6.3335e-01,  7.6737e-02, -5.5214e-01,  2.1183e-01,\n",
       "          -5.7531e-01,  4.4172e-01,  5.4371e-01, -6.2511e-02, -3.5810e-02,\n",
       "          -2.3147e-01, -1.6969e+00,  5.4792e-01, -4.6250e-01,  1.9679e-01,\n",
       "          -4.1286e-01,  4.4326e-01,  2.7262e-01,  4.1395e-01, -2.7798e-02,\n",
       "          -9.2991e-02,  2.0318e-02, -1.0426e-01,  7.5314e-01, -1.3832e-01,\n",
       "           2.5403e-01,  6.2674e-01,  2.5665e-01,  2.4562e-01, -4.8169e-02,\n",
       "          -9.8139e-01,  2.0837e-01, -7.1379e-02,  1.3371e-01,  1.1735e-01,\n",
       "           6.3985e-01, -1.9032e-01, -1.4820e-01,  3.7905e-01, -7.6476e-02,\n",
       "          -4.6011e-02, -4.5591e-01, -3.6211e-01, -3.8628e-01,  9.5566e-02,\n",
       "           1.0861e-01,  2.5016e-01, -1.0774e-01,  7.7999e-01,  3.7650e-01,\n",
       "           9.8801e-02, -6.8960e-02, -3.7857e-01, -3.9673e-01, -4.7297e-01,\n",
       "           9.9967e-02, -3.3185e-01, -1.1991e-01, -4.7646e-01,  4.2284e-02,\n",
       "           3.8592e-01, -4.3989e-01,  6.9669e-01,  2.6012e-01, -1.1800e-02,\n",
       "          -1.9348e-01, -3.3543e-01,  2.9913e-01,  2.4258e-01, -2.8852e-01,\n",
       "           7.2609e-02, -3.6820e-01, -5.8002e-01,  2.0072e-01,  2.2913e-02,\n",
       "           7.2096e-02, -2.1823e-01,  6.7334e-01,  2.9726e-01,  4.2970e-01,\n",
       "           2.5276e-02, -2.2866e-01,  2.2387e-01,  6.6673e-04, -3.9608e-01,\n",
       "          -1.0320e-01,  1.7578e-01, -5.6051e-01,  4.1299e-01, -5.2683e-01,\n",
       "          -3.8485e-01,  1.0183e-01,  2.4261e-01,  5.0383e-01, -2.1974e-01,\n",
       "           1.3439e-01, -7.3837e-02, -5.5185e-01, -1.3036e-01,  4.5205e-01,\n",
       "          -2.5934e-01,  3.2608e-01, -3.0867e-01, -9.0311e-02, -6.3887e-01,\n",
       "          -9.2495e-01,  4.1301e-02, -3.9276e-01,  1.4450e-01, -9.5231e-01,\n",
       "          -1.2686e-01,  4.0218e-01,  1.2400e-01,  4.2741e-02, -3.9088e-01,\n",
       "           2.2293e-01,  1.4388e-01,  1.5778e-01, -2.9596e-01, -6.1219e-01,\n",
       "           8.7324e-02,  3.3742e-01, -2.9066e-01,  1.1917e-01,  1.1107e-01,\n",
       "          -9.6766e-02, -1.6120e-01,  3.3282e-01, -9.1660e-02, -1.1767e-01,\n",
       "          -1.6654e-01, -2.8272e-01,  2.9661e-01, -2.9406e-01, -1.9643e-01,\n",
       "          -2.5631e-02,  2.9923e-01,  7.5711e-01, -4.7277e-01, -4.9892e-01,\n",
       "          -9.6487e-02,  4.5680e-01, -3.8258e-01, -1.0932e-01,  4.1455e-02,\n",
       "          -1.2470e+00, -3.8990e-01, -2.6826e-01, -1.1916e-01, -4.7917e-01,\n",
       "           2.0424e-01, -5.9518e-02,  9.7846e-02, -1.2375e-01, -3.3414e-01,\n",
       "           4.7337e-01,  4.7367e-01, -2.1490e-01, -1.3274e-01, -4.4255e-01,\n",
       "          -9.9170e-02, -1.7272e-02, -1.9134e-01, -4.5052e-01, -8.9570e-02,\n",
       "           2.5038e-01,  3.4361e-01,  1.7014e-01, -2.1798e-02, -3.8831e-01,\n",
       "           5.8279e-01, -4.7840e-01, -4.4396e-03,  3.3559e-01, -5.3767e-01,\n",
       "          -4.5989e-01,  3.6484e-01,  5.4087e-01,  1.0478e-01,  9.5992e-02,\n",
       "           3.1130e-01, -3.8600e-02, -1.3857e-01,  2.2337e-02,  6.4674e-01,\n",
       "           2.1257e-01,  6.4012e-02, -4.4862e-01,  7.2104e-01,  6.1371e-01,\n",
       "          -3.0666e-01,  3.8584e-01, -2.0005e-01, -1.3643e-01, -2.1894e-01,\n",
       "           3.3748e-01,  1.8107e-01, -3.4043e-01,  4.2699e-01,  1.2281e-01,\n",
       "           2.3199e-01, -3.4493e-01,  1.4740e-01,  2.5875e-01,  5.0456e-01,\n",
       "           5.5007e-02, -3.6868e-01,  5.3678e-01, -6.7980e-01, -1.0480e-01,\n",
       "          -1.7328e-01,  5.6144e-02, -1.7223e-02, -4.4960e-01, -5.4589e-01,\n",
       "          -4.4261e-01, -2.4712e-03,  8.8584e-01, -2.0165e-01, -9.3400e-02,\n",
       "           1.3818e-01, -1.7137e-01, -5.5071e-01,  2.2931e-01, -2.2211e-01,\n",
       "           6.8223e-02,  1.5310e-01, -3.1930e-01,  3.6119e-01,  4.0810e-01,\n",
       "          -2.1061e-01,  6.1050e-02,  2.1038e-01, -1.9187e-01,  7.7587e-02,\n",
       "          -4.8391e-01, -1.1384e-01,  1.9657e-01, -3.4664e-02,  2.7519e-01,\n",
       "           6.6931e-01,  6.9479e-01, -3.0869e-01,  1.5120e-01, -5.6590e-02,\n",
       "          -7.3166e-02, -3.0530e-01,  9.5942e-02, -2.4872e-01, -3.5566e-01,\n",
       "           5.6952e-01,  3.3583e-01, -1.2317e-01,  1.5769e-01, -1.1014e+01,\n",
       "           5.7043e-02,  6.0067e-01, -5.3532e-01,  1.3932e-01,  2.9124e-01,\n",
       "           2.9364e-01, -3.4247e-01,  7.1448e-02, -3.6925e-01, -5.6547e-01,\n",
       "          -3.8205e-01,  2.3381e-01,  1.2487e-01, -4.2990e-01, -1.3665e-01,\n",
       "           3.0906e-02, -8.3550e-01,  3.3535e-01, -4.6629e-02,  3.3457e-01,\n",
       "           8.6116e-02,  3.3810e-01,  3.9840e-02,  8.3288e-02, -1.6583e-01,\n",
       "           3.7278e-01, -2.8027e-01,  6.0193e-01, -3.8722e-02, -1.6512e-01,\n",
       "          -4.2897e-02, -2.0674e-02, -6.4845e-02, -4.5879e-01, -6.8197e-01,\n",
       "           2.3473e-02,  1.8587e-01,  9.2198e-02,  7.3224e-02, -2.6841e-01,\n",
       "          -4.0987e-01, -3.0278e-01, -9.7811e-02,  2.8679e-01,  8.2634e-01,\n",
       "           1.1918e+00,  7.1487e-01,  3.3222e-01, -9.4547e-02, -1.8631e-01,\n",
       "          -6.1559e-01,  2.1331e-02, -5.0033e-01,  1.8328e-01, -2.3031e-02,\n",
       "           6.8930e-01, -3.7036e-01,  7.5741e-02, -1.2043e-01, -2.9035e-01,\n",
       "           1.8387e-01,  1.8176e-01,  8.5827e-02, -2.6625e-01, -4.1798e-01,\n",
       "           5.4259e-01, -2.6546e-01,  2.9890e-01, -6.2619e-01,  1.5658e-01,\n",
       "          -2.8134e-01,  1.6062e-01,  2.6367e-01, -1.0212e-02, -1.8326e-01,\n",
       "          -4.5013e-01, -2.5052e-02,  8.4265e-01,  2.1917e-01,  1.5873e-01,\n",
       "           2.9027e-01,  3.1146e-02, -5.1296e-01, -7.4625e-01, -4.8962e-01,\n",
       "           2.5239e-01, -1.1520e-01,  3.8487e-01,  6.9332e-01, -2.8750e-01,\n",
       "          -5.8566e-01,  1.4496e-02,  3.5707e-01,  4.4037e-01,  7.4944e-01,\n",
       "          -2.8622e-01, -4.3490e-01, -1.5871e-01, -6.3709e-02, -4.9993e-01,\n",
       "          -4.2481e-01, -3.1368e-01, -8.4482e-02,  5.2145e-01, -1.5447e-01,\n",
       "           6.1562e-02, -7.4770e-02, -1.7973e-01,  3.4009e-02, -9.5669e-01,\n",
       "           3.8895e-01, -5.1840e-02,  2.7311e-01, -3.3994e-01, -2.0684e-01,\n",
       "           5.7380e-01,  6.1054e-01, -7.0168e-02, -3.0208e-01, -2.6372e-01,\n",
       "           4.3557e-01,  2.5120e-01, -1.4795e-02, -6.6843e-01,  2.1524e-01,\n",
       "          -6.7585e-01,  2.0516e-01,  2.8714e-02,  6.3078e-01,  6.7216e-01,\n",
       "          -2.8674e-01, -4.8212e-01,  3.9118e-01, -4.2283e-01, -6.1316e-01,\n",
       "          -6.9039e-01,  1.2102e-02, -1.0841e-01, -9.1811e-01, -2.0477e-01,\n",
       "          -2.2875e-01,  1.2882e-01, -3.9815e-01,  5.9858e-03,  4.0124e-02,\n",
       "           4.5691e-01,  2.5480e-01,  2.1126e-01,  3.7373e-01, -3.6897e-01,\n",
       "          -4.3890e-01, -5.8223e-02, -1.0887e-01,  6.2030e-02, -2.7808e-01,\n",
       "          -2.8758e-01, -8.0520e-02, -3.9928e-01,  3.8137e-01, -5.7638e-01,\n",
       "           1.6346e-01, -2.9173e-01, -1.7904e-01,  7.0671e-01,  2.9046e-01,\n",
       "           9.9292e-03, -1.9432e-01, -4.4975e-02, -9.4879e-02,  4.6101e-02,\n",
       "           1.6246e-01,  2.7730e-01, -2.1981e-01, -4.1932e-01,  2.0276e-01,\n",
       "           3.5311e-01,  8.5217e-01, -3.1084e-01, -3.3221e-01,  3.2910e-01,\n",
       "          -1.8631e-01, -4.8205e-02, -2.6293e-01, -2.5644e-01, -2.9902e-02,\n",
       "           1.9434e-02, -2.3918e-01,  1.7585e-01, -1.7603e-02,  4.4624e-01,\n",
       "          -1.4217e-01,  2.9756e-01,  2.5069e-01,  8.3888e-01,  3.7305e-01,\n",
       "          -4.7839e-01, -1.4253e-01, -5.5989e-01, -4.0481e-01,  4.7892e-02,\n",
       "           5.4161e-02,  1.6470e-01,  3.0551e-02, -2.8619e-01,  5.8100e-02,\n",
       "           1.5400e-01,  1.1251e+00, -9.5506e-02,  1.9398e-02,  3.9107e-01,\n",
       "           4.2571e-01, -1.0435e-01, -4.8866e-02, -3.5783e-02,  2.4645e-01,\n",
       "           4.4425e-01, -5.8087e-02, -5.8253e-01,  1.9214e-01,  2.2808e-01,\n",
       "           6.2300e-01, -6.4917e-01, -5.1769e-01,  3.2581e-02,  2.6700e-01,\n",
       "          -5.1171e-01, -2.2532e-02,  1.3236e-01,  3.6826e-01,  7.5981e-01,\n",
       "          -1.0444e+00,  8.2931e-02, -1.5796e-01,  7.2639e-01,  1.0896e-02,\n",
       "          -7.1458e-01,  2.6614e-01, -4.1565e-01, -9.4976e-03,  2.7889e-01,\n",
       "           4.9473e-01, -1.2221e+00,  2.7148e-01,  5.5618e-01, -2.7365e-01,\n",
       "          -7.1851e-02, -1.6617e-01,  2.6189e-01,  2.6486e-01,  2.2950e-01,\n",
       "          -6.2857e-01,  1.9076e-01, -4.6970e-01, -8.5066e-01,  3.8991e-01,\n",
       "           5.6044e-01,  3.7853e-02,  7.0010e-02,  1.7572e-01, -2.9422e-01,\n",
       "          -3.5490e-01, -4.1980e-01, -3.8370e-02,  1.0768e-01, -4.7205e-02,\n",
       "          -8.3533e-03, -3.2014e-01,  6.4911e-02,  3.4863e-01, -3.6372e-01,\n",
       "           6.2221e-01,  3.5652e-01,  8.1211e-01, -1.9887e-01, -1.3219e-01,\n",
       "           3.2905e-01,  1.0725e-01, -3.0084e-02,  4.6104e-01, -3.4345e-01,\n",
       "          -1.6983e-01,  3.9980e-01, -6.4964e-02, -5.4108e-02,  9.0883e-02,\n",
       "          -3.7358e-01,  4.4988e-01,  3.1859e-01, -4.3310e-01,  4.1442e-01,\n",
       "           4.1450e-01, -1.4567e-01, -5.8048e-01,  1.8934e-01,  1.8414e-01,\n",
       "          -2.4746e-02,  8.1809e-01,  1.8213e-01,  5.1762e-01,  3.0404e-01,\n",
       "          -4.0902e-02, -9.8929e-01, -1.6498e-01, -1.7807e-01,  1.2507e-01,\n",
       "          -2.6171e-02, -3.4749e-01,  1.8356e-01,  3.1663e-01,  6.5158e-01,\n",
       "           6.7385e-01, -2.8107e-01, -5.7312e-01, -3.5271e-01,  2.2942e-01,\n",
       "          -1.1414e-01, -1.2183e-01, -4.1816e-02, -1.3172e-01,  3.4459e-01,\n",
       "          -6.0439e-01, -6.0133e-01, -2.7838e-02,  1.3782e-01, -3.0948e-01,\n",
       "           9.4016e-01,  6.5917e-02,  3.9938e-01,  1.8772e-01,  3.9906e-01,\n",
       "           1.6729e-01,  2.6603e-01,  5.2915e-03, -8.3636e-01, -9.2227e-03,\n",
       "           3.6457e-01, -2.4896e-01,  4.1452e-02, -3.6876e-01,  3.7147e-01,\n",
       "           4.0842e-01,  6.3983e-01,  5.4919e-01]], device='mps:0'),\n",
       " tensor([[-6.1750e-01, -4.2829e-01, -2.3934e-01, -9.4116e-01, -3.4548e-01,\n",
       "          -6.5496e-02, -6.3874e-01,  9.7828e-01, -9.5892e-02,  4.0096e-01,\n",
       "           1.4919e-01, -5.1603e-03,  2.0980e-01,  1.3680e-01, -6.9506e-01,\n",
       "           1.6740e-01,  4.6094e-01,  2.1159e-01,  3.5951e-01, -4.4329e-02,\n",
       "           7.9832e-01,  3.4245e-01,  1.8019e-01, -2.5205e-01, -4.5876e-01,\n",
       "           5.8994e-01, -2.4628e-01, -9.2014e-02,  2.0509e-01, -8.1949e-01,\n",
       "          -1.3586e-01, -4.9757e-02, -3.6281e-01,  8.7602e-01,  1.1948e-01,\n",
       "          -2.7310e-01, -1.5276e-01, -2.7996e-01,  2.3494e-01, -6.3637e-02,\n",
       "           3.7814e-01,  2.6592e-02, -2.9347e-01,  3.1386e-01, -1.6421e-01,\n",
       "          -1.4057e-01,  1.4047e-02, -7.8429e-01,  1.3666e-02,  1.8955e-01,\n",
       "           5.8851e-02, -4.7205e-02,  1.9231e-01,  3.7030e-01, -1.0617e+00,\n",
       "          -2.6348e-01,  2.7603e-01,  2.9849e-01, -1.7932e-01, -5.7784e-02,\n",
       "          -1.9989e-01,  9.0201e-01,  1.7949e-01, -2.7822e-01,  1.7824e-01,\n",
       "          -4.4899e-01, -5.4601e-01, -6.0781e-01, -1.0682e-01, -1.1967e-01,\n",
       "          -1.0002e-01,  9.7528e-02,  1.4009e-01, -2.9385e-01,  2.3170e-01,\n",
       "           2.1813e-01,  4.7024e-02, -2.7652e-02, -4.3158e-01, -2.6546e-01,\n",
       "           3.6947e-01,  3.4425e-02, -1.4730e-02,  2.0326e-01, -5.8744e-01,\n",
       "          -2.0226e-01,  3.7959e-01,  1.5703e-01,  2.5914e-01,  2.2662e-03,\n",
       "           5.9967e-01,  1.2024e-02,  8.0339e-02, -4.5369e-01,  5.8427e-01,\n",
       "          -5.4955e-01,  1.0329e-01, -1.6007e-01,  4.9532e-02, -5.2088e-01,\n",
       "          -9.8735e-02,  2.4494e-01, -1.8085e-01,  1.6053e-01, -4.6592e-01,\n",
       "          -3.2983e-02,  2.4514e-01,  3.0468e-01, -4.9183e-01,  4.2685e-02,\n",
       "          -6.5134e-03, -2.9791e-01,  7.3694e-02, -6.6407e-01, -8.5795e-01,\n",
       "           5.8354e-01,  1.8428e-02, -4.5240e-01, -2.3930e-01,  7.4443e-01,\n",
       "           2.9738e-01, -2.3201e-01, -5.3072e-02, -3.5548e-01,  2.0259e-01,\n",
       "          -1.8518e-01,  2.5689e-01, -2.7018e-01,  3.8439e-01,  5.5026e-01,\n",
       "           3.1581e-01,  2.8489e-01, -1.8327e-01,  2.7637e-01,  1.4465e-01,\n",
       "           3.6947e-02, -6.6614e-02, -1.5929e-01,  1.5180e-01,  1.2041e-01,\n",
       "          -2.9580e-01, -3.2999e-01,  5.4746e-03,  1.1182e-01, -3.3279e-01,\n",
       "          -1.0202e-01,  1.8680e-01,  3.4808e-01, -1.4092e-01, -2.5403e-01,\n",
       "          -7.8984e-01, -4.2328e-01,  1.0217e-01, -1.8560e-01, -1.7135e-01,\n",
       "          -4.1911e-01,  3.8121e-01, -6.6660e-02,  4.7638e-01,  3.4189e-01,\n",
       "           6.9978e-01, -1.5311e-01,  5.9776e-02, -5.6412e-01,  3.5197e-01,\n",
       "           3.8270e-02, -6.1110e-01, -1.7247e-01,  1.5680e-01, -4.1813e-04,\n",
       "           6.8195e-01,  7.0979e-02,  4.0124e-01,  1.2070e-01, -3.2553e-01,\n",
       "           8.2089e-03,  1.9890e-01, -1.2864e-01, -1.4297e-01,  7.3766e-01,\n",
       "          -5.5188e-01, -2.3029e+00,  1.0721e-01, -3.5844e-01, -3.6529e-01,\n",
       "          -1.8135e-01,  3.7255e-01,  3.3772e-02,  4.9966e-01,  1.5486e-01,\n",
       "           3.7598e-02,  1.6729e-01, -2.8025e-02,  3.6851e-01, -8.4399e-02,\n",
       "           2.6751e-01,  3.3341e-01,  1.2988e-01, -2.3298e-01, -1.0956e-01,\n",
       "          -3.9227e-01,  2.4031e-01, -9.8916e-02,  8.6010e-01,  2.3260e-01,\n",
       "           2.0943e-01, -4.3657e-03, -4.7262e-02,  4.8964e-01,  4.3099e-01,\n",
       "          -6.6541e-02, -1.6585e-01, -4.9522e-01,  4.6269e-03, -4.1279e-02,\n",
       "          -1.2963e-02, -4.5611e-01,  1.3753e-01,  4.2841e-01, -1.2416e-01,\n",
       "          -1.8354e-01,  1.5525e-01, -4.7715e-02, -3.1517e-01, -2.6492e-02,\n",
       "           6.8453e-01,  1.9080e-01, -2.2994e-01,  1.2377e-01, -1.1896e-01,\n",
       "           3.7006e-02, -4.6975e-01,  7.0044e-01,  6.1147e-01,  1.5585e-01,\n",
       "           1.6214e-02, -6.9943e-01,  2.3699e-01, -3.1574e-01, -4.7647e-03,\n",
       "          -1.5027e-01, -1.5731e-01, -4.2986e-01, -5.7285e-02, -5.5854e-02,\n",
       "           4.0115e-01,  7.9943e-02,  4.6653e-01,  1.1837e+00,  2.7973e-01,\n",
       "           1.9184e-02,  5.6097e-01,  2.0801e-01,  4.2492e-01, -4.0180e-01,\n",
       "           3.3082e-01,  8.8825e-02, -6.3896e-01,  2.7188e-01, -4.1593e-01,\n",
       "           4.4963e-02, -4.1432e-02, -3.7926e-01,  1.4208e-01, -2.1276e-02,\n",
       "           3.4237e-01,  3.3517e-01,  1.5186e-01,  1.6838e-01,  2.9411e-01,\n",
       "          -3.1322e-01,  5.9715e-01, -6.0809e-02, -1.1127e-01, -6.3492e-01,\n",
       "          -7.6732e-01,  9.5206e-02,  4.4743e-01, -1.2880e-01, -6.1279e-01,\n",
       "           1.0462e-01, -2.3866e-01, -1.2257e-01, -3.5064e-02,  2.1191e-01,\n",
       "           2.7455e-01,  5.0687e-02, -9.5436e-02,  6.4791e-02, -4.6950e-01,\n",
       "          -4.9277e-01,  4.9140e-02, -1.5758e-01, -1.3063e-01,  5.3918e-01,\n",
       "          -7.9415e-02,  1.1575e-01,  5.7221e-01,  1.9799e-02, -4.5976e-01,\n",
       "          -2.5640e-01,  1.4802e-01,  2.8667e-01, -7.2082e-02,  2.1981e-01,\n",
       "           3.4119e-01,  2.6467e-02,  5.1649e-01,  1.4823e-01, -2.1046e-01,\n",
       "          -2.7239e-01,  5.1173e-01, -1.5755e-01,  6.1403e-02, -1.2527e-01,\n",
       "          -8.7207e-01, -5.6150e-02, -7.8413e-02, -1.6833e-01, -5.4440e-01,\n",
       "           2.6566e-01,  1.4351e-01,  1.0561e-01,  2.2350e-01, -1.5984e-01,\n",
       "           7.0123e-01,  6.5285e-01, -3.3100e-01, -1.6943e-01, -4.0104e-01,\n",
       "           9.3922e-02, -6.8374e-02, -2.9329e-01, -1.4009e-02,  9.5368e-03,\n",
       "          -2.2765e-01, -1.2516e-01, -1.0805e-01,  2.8921e-01, -1.8876e-01,\n",
       "           5.2862e-01,  1.2189e-01,  8.3649e-02,  4.3873e-01,  2.0612e-02,\n",
       "           1.3071e-01,  3.6515e-01, -4.2117e-01,  1.2870e-01,  4.0424e-01,\n",
       "           4.0578e-01, -1.0551e+00, -8.3291e-01, -1.4936e-01,  3.6111e-01,\n",
       "           2.2860e-01,  2.5189e-01,  1.6999e-02, -7.2215e-03,  8.9455e-02,\n",
       "          -2.3402e-01,  3.5135e-01, -6.6415e-02, -5.7412e-01,  3.0561e-01,\n",
       "           2.3567e-01,  5.3263e-01, -2.7377e-01, -5.0579e-01, -6.5233e-02,\n",
       "           2.4616e-01, -3.6381e-01,  1.1392e-01,  2.3480e-01,  4.5466e-01,\n",
       "          -3.6015e-01, -5.6164e-01, -7.0282e-02, -7.3986e-01, -1.3578e-01,\n",
       "           3.2483e-01, -2.1437e-01,  2.7178e-02, -2.4758e-01,  1.5231e-01,\n",
       "          -1.7925e-02, -1.8879e-01,  6.0757e-01,  2.0522e-01, -2.5855e-01,\n",
       "           1.1981e-01, -2.3892e-01, -1.9475e-01,  5.0091e-01, -5.0226e-01,\n",
       "          -3.0348e-01, -5.0960e-02, -1.5349e-01, -3.2418e-01,  3.2181e-01,\n",
       "          -4.1389e-01, -1.4861e-01,  1.6748e-01, -6.4736e-02, -1.3068e-01,\n",
       "          -3.5233e-01,  1.4268e-01, -1.2707e-01, -4.9883e-02, -1.7654e-02,\n",
       "          -1.3353e-01,  6.0900e-02,  1.3308e-01,  6.4538e-02,  1.4388e-01,\n",
       "           5.3696e-01, -1.1464e-01,  2.7582e-03, -2.5900e-01, -2.2469e-01,\n",
       "           5.2498e-01, -9.4973e-02, -3.3117e-01,  5.4928e-01, -1.1156e+01,\n",
       "           1.0386e-01,  7.2631e-01, -5.5808e-01,  5.5128e-02, -2.8539e-01,\n",
       "           2.1085e-01, -3.0009e-01,  7.9101e-02,  2.4967e-01, -2.8460e-01,\n",
       "          -1.2277e-01, -2.7364e-01, -2.3193e-01, -4.0872e-01, -3.9742e-01,\n",
       "          -4.7229e-01, -5.6789e-01,  1.1772e-01, -1.7663e-01, -1.0996e-01,\n",
       "          -5.7036e-01, -1.9379e-01,  2.9696e-01, -4.9559e-03, -5.1069e-01,\n",
       "           5.9541e-02, -2.2948e-01,  2.6399e-01,  4.6126e-02,  2.2045e-01,\n",
       "          -1.9949e-02,  1.4108e-01,  2.0669e-01, -4.4946e-02, -4.5100e-01,\n",
       "          -4.4853e-01,  2.2129e-01, -1.5955e-01, -2.2267e-01, -5.4017e-02,\n",
       "          -3.3838e-01, -2.7689e-01,  1.3447e-01,  3.7594e-01,  6.2596e-01,\n",
       "           1.1595e+00,  5.6208e-01,  2.9245e-01, -1.0625e-01, -8.5438e-02,\n",
       "          -1.0762e+00,  5.5900e-01, -3.7318e-01,  8.1787e-02, -2.5119e-01,\n",
       "           6.6201e-01, -4.9906e-01,  5.4685e-01, -4.0754e-01, -2.5500e-01,\n",
       "           3.7179e-01,  3.0194e-01,  7.1104e-03,  2.8239e-02, -2.7432e-01,\n",
       "           4.0257e-01, -2.6094e-01, -3.6915e-02, -3.7819e-01,  4.0081e-01,\n",
       "          -3.9472e-01,  5.0770e-01, -5.6534e-02,  1.2945e-01,  1.8567e-01,\n",
       "          -3.5001e-01,  1.6610e-01,  1.0337e+00,  1.7414e-01,  4.1071e-01,\n",
       "           1.7420e-01,  5.1160e-01,  5.3064e-02, -8.8958e-01,  1.3691e-01,\n",
       "           2.0502e-01,  2.0789e-01, -6.7096e-02,  5.7131e-01, -2.5517e-02,\n",
       "          -6.5181e-01, -2.3094e-01, -3.2169e-01,  4.3108e-01,  6.5352e-01,\n",
       "          -2.2597e-01, -4.4343e-01,  2.3894e-01, -7.9776e-03, -5.3395e-01,\n",
       "           9.0074e-02, -2.2320e-01, -5.2792e-01,  8.7275e-02, -5.8914e-01,\n",
       "           1.4985e-01, -1.2428e-01,  5.4238e-02,  6.6887e-01, -4.0159e-01,\n",
       "          -1.6113e-01,  3.6602e-02, -4.4532e-02, -4.1270e-01, -4.9520e-01,\n",
       "          -7.7083e-02,  4.1073e-01, -1.9776e-01, -2.7976e-01,  7.0636e-02,\n",
       "          -3.0673e-01,  2.2584e-01, -1.7377e-01, -6.4935e-01,  3.2879e-01,\n",
       "          -6.0415e-01,  2.0578e-01,  2.5360e-01, -2.2507e-01,  4.9017e-01,\n",
       "           1.1337e-01, -8.1817e-02,  1.9397e-01, -9.9369e-02, -7.8554e-01,\n",
       "          -3.2573e-01, -2.7299e-02,  3.2662e-01, -4.2844e-01,  1.5776e-01,\n",
       "           1.5022e-01, -4.5752e-02, -2.2574e-01,  2.7904e-01,  1.0174e-01,\n",
       "           3.7762e-01,  3.9846e-01, -5.5006e-01,  1.1727e-01, -2.9867e-01,\n",
       "          -2.9262e-02, -6.1174e-02,  1.7952e-01,  2.3654e-01, -7.0764e-01,\n",
       "          -1.5224e-01, -2.6434e-01, -3.0965e-01, -2.1650e-01, -2.5120e-01,\n",
       "           2.0680e-01,  1.4038e-01, -1.5897e-02, -1.1308e-01,  5.8077e-01,\n",
       "           4.0814e-02, -1.1167e-01, -4.5997e-01,  3.1443e-01,  4.3178e-01,\n",
       "           2.1794e-01,  7.5875e-02, -1.2830e-01, -5.5532e-01, -1.1922e-02,\n",
       "          -4.3644e-01,  9.1081e-01, -7.3906e-01, -7.1970e-01,  6.6086e-02,\n",
       "           1.4951e-02, -8.8259e-04, -2.3450e-01, -5.0265e-01,  8.1298e-02,\n",
       "          -2.1017e-01, -1.4891e-01,  6.1550e-03, -1.0607e-02,  1.6299e-01,\n",
       "           4.0742e-01,  3.6838e-01,  1.5951e-01, -3.3938e-01, -4.4081e-01,\n",
       "          -6.0244e-01,  1.4567e-02, -1.6795e-01,  2.9330e-01, -2.6752e-01,\n",
       "          -3.2661e-01,  1.6043e-02, -9.1892e-03, -2.2272e-01,  5.2876e-01,\n",
       "           1.2427e-01,  1.6396e+00, -3.7287e-01, -4.4551e-01,  2.9018e-01,\n",
       "           7.2306e-02, -2.7073e-02, -2.8130e-01,  2.5074e-01,  9.5447e-02,\n",
       "           4.0194e-02, -3.6304e-01,  1.5023e-02, -2.0558e-02,  3.2773e-01,\n",
       "           3.2056e-01, -1.0703e+00, -2.0238e-01, -1.7993e-01,  5.5664e-01,\n",
       "           3.2388e-02,  6.3185e-01, -2.2305e-01,  3.0577e-01,  4.6106e-01,\n",
       "          -8.6667e-01, -2.3000e-01, -1.6129e-01,  2.7295e-01, -1.0583e-01,\n",
       "          -9.8735e-02, -1.6178e-01, -3.7165e-02, -1.6936e-01,  3.9538e-01,\n",
       "           7.4751e-01, -7.8426e-01,  7.4081e-01,  7.0467e-01, -9.8234e-02,\n",
       "           5.6815e-01,  9.4187e-02,  3.2641e-01, -1.4659e-01,  2.2669e-02,\n",
       "           2.6200e-01,  1.8542e-01, -6.3165e-01, -2.3962e-01,  2.4473e-01,\n",
       "           4.0158e-01, -1.1943e-01,  1.2108e-01,  7.9292e-02,  3.0960e-02,\n",
       "          -3.7691e-01, -1.0145e-01, -2.4552e-01, -3.9904e-01,  1.1356e-01,\n",
       "           1.4603e-02, -1.5001e-01, -8.6997e-02,  4.7709e-01, -1.7580e-01,\n",
       "           7.5067e-01,  4.2910e-01,  3.7145e-01, -3.5103e-01,  8.9950e-04,\n",
       "          -5.7565e-02,  6.0671e-01, -3.1670e-01,  7.7385e-01,  6.3337e-02,\n",
       "          -2.3140e-01,  2.7251e-01,  6.0547e-03,  1.7829e-01,  2.9892e-01,\n",
       "          -2.3519e-01,  1.2702e-01, -3.0718e-01, -2.3711e-01,  2.8330e-01,\n",
       "          -1.8640e-01,  1.1542e-01, -7.2270e-01, -6.3602e-03,  3.9857e-01,\n",
       "          -3.2051e-01,  4.9873e-01, -2.3074e-01,  6.1026e-01, -4.3509e-01,\n",
       "           3.2669e-02, -4.7470e-01, -4.3275e-01, -2.8508e-01,  5.1854e-02,\n",
       "          -8.6027e-01, -1.9453e-01,  1.7684e-01,  3.7405e-02,  2.1707e-01,\n",
       "           7.5155e-01, -3.3794e-01, -2.4771e-01, -2.0386e-01,  4.0190e-02,\n",
       "           1.8299e-01, -8.2071e-02, -3.5586e-01,  2.1161e-01, -3.5607e-01,\n",
       "          -1.0520e+00, -2.3117e-01,  8.9990e-02,  3.8419e-01, -3.5853e-01,\n",
       "           1.1570e+00,  2.5132e-02, -2.0226e-01,  6.7253e-01,  2.0808e-01,\n",
       "           6.4157e-01,  6.0323e-01,  1.3628e-03, -4.1259e-01, -5.4109e-02,\n",
       "          -1.5877e-01, -3.2881e-01,  1.5757e-01, -1.8346e-01,  5.0556e-01,\n",
       "           5.4802e-01, -2.0806e-02,  5.4756e-01]], device='mps:0'),\n",
       " tensor([[-0.2258, -0.7099, -0.3778,  ...,  0.3292,  0.7059,  0.7494],\n",
       "         [-0.0932, -0.6293, -0.4879,  ...,  0.3041, -0.1958,  0.5576],\n",
       "         [-0.3292, -0.2590, -0.5016,  ..., -0.0148,  0.3509,  0.1607]],\n",
       "        device='mps:0'),\n",
       " tensor([[-0.0184, -0.5590, -0.5915,  ...,  0.2161,  0.3434,  0.9342],\n",
       "         [-0.5102, -0.5245, -0.6019,  ..., -0.0406,  0.7921,  0.5311]],\n",
       "        device='mps:0'),\n",
       " tensor([[-0.1521, -0.6347, -0.7744,  ...,  0.2264, -0.4010,  0.7173],\n",
       "         [-0.2953, -0.8775, -0.4259,  ...,  0.1367,  0.0478,  0.5037]],\n",
       "        device='mps:0'),\n",
       " tensor([[-0.3402, -0.6569, -0.3442,  ...,  0.2599,  0.7381,  0.5866],\n",
       "         [-0.0238, -0.5721, -0.8340,  ...,  0.0483,  0.8204,  0.5286]],\n",
       "        device='mps:0'),\n",
       " tensor([[-0.4986, -0.1214, -0.4676,  ...,  0.1459,  0.8064,  0.3478],\n",
       "         [-0.0548, -0.4843, -0.5199,  ...,  0.3330,  0.0168,  0.3173],\n",
       "         [-0.3530, -0.3965, -0.3571,  ...,  0.0269,  0.8932,  0.6202],\n",
       "         [-0.1403, -0.4789, -0.6126,  ...,  0.4490,  0.9203,  0.2500],\n",
       "         [-0.1813, -0.5438, -0.6217,  ...,  0.1651,  0.9105,  0.6850],\n",
       "         [-0.3777, -0.8370, -0.5973,  ...,  0.3308,  0.3320,  0.2288]],\n",
       "        device='mps:0'),\n",
       " tensor([[-0.1973, -0.4202, -0.7866,  ...,  0.1423,  0.6287,  0.5201],\n",
       "         [-0.3292, -0.2590, -0.5016,  ..., -0.0148,  0.3509,  0.1607],\n",
       "         [-0.3742, -1.0694, -1.0601,  ..., -0.0281,  0.5891,  0.7007]],\n",
       "        device='mps:0'),\n",
       " tensor([[-7.7611e-01, -6.0926e-01, -1.1787e-01, -1.0640e+00, -3.1583e-01,\n",
       "          -8.6267e-02, -7.6483e-01,  9.2128e-01,  2.7002e-01,  3.7649e-01,\n",
       "           8.7499e-02,  4.1714e-01,  2.4632e-01, -2.7338e-01, -6.7152e-01,\n",
       "          -7.9374e-02,  1.5870e-01,  4.1646e-01, -7.9433e-02,  1.6746e-01,\n",
       "           3.5979e-01,  4.9060e-01,  7.5620e-02, -1.9239e-01, -6.7870e-01,\n",
       "           3.6117e-01, -1.3505e-02,  2.5924e-01,  2.9026e-01, -6.4409e-01,\n",
       "          -1.8089e-01,  1.3904e-01, -4.0658e-01,  1.1366e+00,  2.6674e-02,\n",
       "          -3.0685e-01,  6.7086e-02, -1.0912e-02,  3.3724e-01, -1.6781e-01,\n",
       "           5.8725e-01,  1.5386e-01, -1.8457e-01,  3.9306e-01,  2.4695e-02,\n",
       "           1.4839e-01,  1.0083e-01, -5.2092e-01,  5.1091e-02, -2.4924e-01,\n",
       "          -9.1981e-03, -1.4916e-01,  4.1145e-01,  4.2003e-01, -8.7722e-01,\n",
       "          -4.4840e-01,  1.7193e-01,  3.0803e-01, -3.1370e-01, -3.4767e-01,\n",
       "          -8.8392e-02,  6.7159e-01,  1.8736e-01, -4.0113e-01,  3.2594e-01,\n",
       "          -2.7862e-01, -3.9193e-01, -7.5738e-01, -2.0888e-01, -3.0523e-01,\n",
       "          -2.0261e-01,  2.9258e-01,  2.4439e-02, -2.6322e-01,  1.4535e-01,\n",
       "           1.4496e-01,  1.0139e-01,  1.4443e-01, -5.4568e-01, -3.0232e-01,\n",
       "           3.0880e-01,  5.8307e-02,  1.8512e-01, -1.6273e-01, -2.1340e-01,\n",
       "          -1.8400e-01,  3.2121e-01, -1.0120e-01,  8.3677e-02, -2.8053e-01,\n",
       "           4.9796e-01, -1.2428e-01, -5.8351e-02, -2.0952e-01,  3.7108e-01,\n",
       "          -3.7241e-01,  3.8112e-01, -4.1952e-01,  2.5338e-02, -3.5476e-01,\n",
       "          -1.1026e-01,  3.4463e-01, -4.5569e-01,  4.3380e-02, -8.6978e-01,\n",
       "          -1.9347e-01,  3.7376e-01,  4.8510e-02, -7.6099e-01,  3.8419e-01,\n",
       "           6.4721e-02, -2.6483e-01,  1.6827e-01, -6.8798e-01, -4.0337e-01,\n",
       "           3.5363e-01,  1.9094e-01, -2.2372e-01, -2.2978e-01,  3.0602e-01,\n",
       "           5.7816e-01, -1.8993e-01,  3.8308e-01, -3.1500e-01, -9.0651e-02,\n",
       "          -3.6299e-01,  5.3724e-01, -2.9667e-01,  3.3187e-01,  4.9091e-01,\n",
       "           3.4459e-01,  3.5344e-01, -3.6064e-01,  1.7609e-01,  2.2007e-01,\n",
       "          -6.4815e-02, -1.4574e-01, -2.3581e-01,  3.7500e-01,  3.3386e-01,\n",
       "          -4.4378e-01, -1.6613e-01,  7.9672e-02,  4.9999e-02, -1.6292e-01,\n",
       "          -1.3209e-01,  1.1388e-01,  3.3711e-01, -3.7937e-01, -5.0006e-01,\n",
       "          -1.1100e+00, -3.7886e-01,  2.6264e-01, -2.9083e-01, -5.6477e-02,\n",
       "          -3.8939e-01,  4.6649e-01, -6.8099e-02,  4.2340e-01,  2.6147e-01,\n",
       "           4.5898e-01, -2.2711e-01,  1.8306e-01, -6.7828e-01,  1.4424e-01,\n",
       "           4.6581e-02, -5.7980e-01, -1.1083e-01,  1.0714e-01,  4.7456e-02,\n",
       "           5.8144e-01, -6.0817e-02,  3.4058e-01,  2.2957e-01, -6.7439e-02,\n",
       "          -1.8591e-02,  3.8512e-01, -2.8090e-02, -1.9682e-01,  4.9172e-01,\n",
       "          -4.4480e-01, -1.7906e+00,  6.7731e-02, -3.7295e-01, -2.4978e-01,\n",
       "          -1.2911e-01,  2.1964e-01, -1.0053e-01,  7.2979e-01,  1.3512e-01,\n",
       "           1.6537e-01,  2.3284e-02,  6.4205e-02,  6.1590e-01,  6.6559e-02,\n",
       "           7.0437e-01,  2.2367e-01,  2.5559e-01, -4.0492e-01, -1.4754e-01,\n",
       "          -1.0921e-01,  3.4398e-01,  1.9815e-01,  7.0458e-01,  1.2290e-01,\n",
       "           4.4800e-01, -6.8702e-02,  1.1456e-01,  1.6965e-01,  2.9036e-01,\n",
       "          -1.0234e-01, -9.3230e-02, -5.3960e-01, -2.0857e-01, -2.4399e-01,\n",
       "           7.0145e-02, -4.7111e-01,  1.0131e-01,  6.6774e-01,  1.4782e-01,\n",
       "          -2.3682e-01,  2.0315e-01, -1.0984e-01, -3.3356e-01,  8.3820e-02,\n",
       "           4.2603e-01, -1.5716e-02, -9.8476e-02,  1.8908e-02, -1.8369e-01,\n",
       "          -4.2660e-01, -1.2279e-01,  6.5492e-01,  5.0196e-01, -4.9594e-03,\n",
       "           1.8190e-01, -1.8560e-01, -3.4036e-02, -2.4997e-01,  2.1814e-01,\n",
       "          -3.3029e-01, -2.0772e-01, -2.4000e-01, -8.8967e-04, -2.7357e-01,\n",
       "           4.6117e-01,  1.9515e-01,  3.4596e-01,  9.7995e-01,  3.5252e-01,\n",
       "          -1.3282e-01,  4.9795e-01,  1.0548e-01,  8.9236e-02, -4.0801e-01,\n",
       "           1.3282e-01,  4.1431e-02, -5.6534e-01,  5.9087e-01, -3.4730e-01,\n",
       "          -6.0888e-02, -1.7708e-01,  1.1331e-01,  1.8898e-01, -1.6088e-01,\n",
       "           2.7780e-01,  4.4364e-01, -1.0948e-01,  7.8641e-04,  5.7977e-01,\n",
       "          -6.1212e-01,  5.3111e-01, -2.5898e-01, -1.4000e-01, -6.5929e-01,\n",
       "          -9.2288e-01, -3.0148e-01,  2.6868e-01,  1.9056e-02, -4.4041e-01,\n",
       "           9.5264e-03, -2.0278e-01,  1.3338e-01,  4.0489e-02, -1.8322e-01,\n",
       "           2.0011e-01,  6.6336e-03,  2.6320e-02, -2.7059e-01, -3.5435e-01,\n",
       "          -3.3183e-01,  2.1377e-02, -3.1489e-01, -1.4312e-01,  1.3885e-01,\n",
       "           3.6202e-01,  4.6631e-02,  5.9175e-01, -1.5776e-01, -5.0567e-01,\n",
       "          -1.1412e-01,  9.0993e-02,  2.8621e-01, -2.8619e-01,  3.0996e-01,\n",
       "           2.9626e-01,  1.2143e-01,  6.8184e-01, -2.9263e-01, -3.6936e-01,\n",
       "          -1.6363e-01,  5.6569e-01, -1.9600e-01,  7.7361e-02, -1.7146e-01,\n",
       "          -1.2156e+00, -2.7134e-01, -1.0566e-01, -7.8763e-02, -6.0420e-01,\n",
       "           2.7985e-01, -8.9898e-02, -4.2795e-02, -2.0294e-02, -2.9210e-01,\n",
       "           8.0329e-01,  5.3894e-01, -1.0754e-01,  2.3309e-02, -3.7469e-01,\n",
       "          -1.5335e-01, -2.5923e-01, -1.5644e-01, -1.4319e-01,  2.1083e-01,\n",
       "           3.3959e-02,  1.7998e-02, -2.3199e-01,  2.5818e-01, -5.1250e-01,\n",
       "           5.6120e-01,  3.5343e-02,  2.6174e-01,  4.1334e-01, -1.7440e-01,\n",
       "          -7.0880e-03,  2.9414e-01, -2.4418e-01, -7.7057e-02,  2.7634e-01,\n",
       "           3.1004e-01, -9.5212e-01, -5.1009e-01, -2.8998e-01,  5.0158e-01,\n",
       "           1.0484e-01, -1.5769e-02, -3.3580e-02,  1.3302e-02,  3.5838e-01,\n",
       "          -1.3682e-01,  4.2705e-01, -2.7304e-01, -3.9463e-01,  4.1937e-01,\n",
       "           1.9348e-01,  4.4094e-01, -2.7794e-01, -7.6944e-02, -1.9636e-02,\n",
       "           8.6838e-02, -2.0867e-01,  1.7694e-01,  1.4487e-01,  5.9896e-01,\n",
       "          -2.7905e-01, -5.6952e-01, -9.8497e-02, -8.1404e-01, -3.2022e-01,\n",
       "           3.9128e-01,  2.5789e-02,  3.2119e-01, -2.8320e-01, -2.4963e-01,\n",
       "          -4.5410e-02, -2.6859e-01,  7.4586e-01,  5.0966e-01, -1.5913e-01,\n",
       "           1.5504e-01, -1.2622e-01, -4.4267e-01,  3.7352e-01, -4.5244e-01,\n",
       "          -3.7546e-01,  2.2603e-01, -2.3292e-01, -2.2891e-01,  1.2024e-01,\n",
       "          -2.9179e-01, -2.6092e-01,  2.4583e-01, -2.9012e-02, -2.0807e-02,\n",
       "          -4.9239e-01,  2.4357e-01,  1.3106e-01, -1.7474e-01, -1.4601e-01,\n",
       "           7.6459e-02,  1.6942e-01, -1.3340e-01, -1.7454e-01,  3.1735e-01,\n",
       "           3.3934e-01,  4.6853e-02, -1.1605e-01, -2.5210e-01, -1.7305e-02,\n",
       "           2.5167e-01, -2.3211e-02, -4.1639e-01,  3.4658e-01, -1.1106e+01,\n",
       "           1.0504e-01,  7.1573e-01, -3.8253e-01,  2.0115e-01, -2.4505e-02,\n",
       "           7.8860e-02, -2.5375e-01, -9.2045e-03, -8.9896e-02, -2.9909e-01,\n",
       "          -4.3760e-01,  6.6427e-02, -1.7524e-01, -4.8494e-01, -2.6129e-01,\n",
       "          -2.2463e-01, -6.7504e-01,  7.2328e-02, -4.0304e-01,  4.6106e-02,\n",
       "          -4.0322e-01, -6.5675e-02,  2.1397e-01,  1.8516e-01, -2.6273e-01,\n",
       "          -1.2515e-01, -3.7561e-01,  3.9453e-01, -4.3939e-02,  2.1094e-01,\n",
       "          -7.2534e-02,  6.5628e-02, -8.4189e-02, -3.3603e-01, -7.0643e-01,\n",
       "          -3.0427e-01,  4.5997e-02,  5.1683e-02, -4.8462e-02, -9.8761e-02,\n",
       "          -5.4025e-01, -4.0086e-01,  2.1965e-01,  2.4717e-01,  7.0190e-01,\n",
       "           7.4396e-01,  5.9364e-01,  1.0945e-01,  1.3975e-02, -2.8945e-01,\n",
       "          -1.1530e+00,  5.0231e-01, -3.7230e-01, -9.1031e-02, -5.1419e-01,\n",
       "           6.0343e-01, -4.8404e-01,  2.5345e-01, -3.3815e-01, -2.6641e-01,\n",
       "           2.7081e-01,  3.5636e-01, -1.3866e-01, -5.2928e-02, -1.7237e-01,\n",
       "           3.5771e-01, -2.5290e-01,  1.5395e-01, -9.3758e-02,  2.6043e-01,\n",
       "          -2.5511e-01,  3.9954e-01, -1.3738e-01, -1.4642e-01,  3.2815e-01,\n",
       "          -2.6072e-01,  3.9174e-02,  1.2599e+00, -2.4615e-01,  2.7132e-01,\n",
       "          -1.1791e-01,  4.2004e-01, -2.7168e-02, -8.3786e-01,  1.1718e-01,\n",
       "           2.2799e-01,  3.3693e-01, -6.5000e-02,  6.8635e-01, -7.3416e-02,\n",
       "          -5.5963e-01, -1.4080e-01,  1.8438e-01,  5.5227e-01,  8.0256e-01,\n",
       "          -1.9028e-01, -4.7103e-01,  2.9278e-01,  2.2523e-02, -3.5325e-01,\n",
       "           4.1952e-01, -3.1284e-01, -1.6562e-01,  1.4905e-01, -9.4511e-01,\n",
       "           8.8251e-02,  5.9215e-02,  2.2444e-02,  6.3211e-01, -2.6673e-01,\n",
       "           1.6983e-03,  1.8687e-01, -7.2675e-02, -5.4493e-01, -1.4846e-01,\n",
       "          -1.8520e-01,  6.6005e-01, -1.1613e-01, -3.4219e-01, -5.6555e-02,\n",
       "          -8.1339e-02,  2.2354e-01, -9.3754e-02, -5.8464e-01,  2.3635e-01,\n",
       "          -5.1705e-01,  1.1337e-01, -3.3886e-02, -3.3860e-01,  5.1906e-01,\n",
       "           4.6517e-01, -1.5233e-01,  1.4550e-01, -4.6621e-01, -8.6591e-01,\n",
       "          -2.8904e-01, -4.0662e-01,  1.9112e-01, -1.9203e-01,  2.0722e-01,\n",
       "          -3.8124e-02,  3.4875e-02, -2.0340e-01,  3.1555e-01,  1.9955e-01,\n",
       "           1.8074e-01,  4.1602e-01, -4.3114e-01,  2.9108e-01, -7.6504e-02,\n",
       "          -7.1603e-02,  5.7420e-02,  1.6474e-01,  5.2320e-02, -8.3155e-01,\n",
       "          -2.9030e-01, -2.4031e-02, -2.3431e-01, -1.7045e-01, -3.8480e-01,\n",
       "           4.1302e-01, -1.1371e-01, -4.7603e-02, -1.7801e-01,  5.2133e-01,\n",
       "          -3.7934e-02, -3.0868e-01, -4.0018e-01,  1.1899e-01,  1.9168e-01,\n",
       "           4.5668e-01,  1.0456e-01, -2.7063e-01, -7.7268e-01, -1.3778e-01,\n",
       "          -2.4037e-01,  1.0391e+00, -5.2415e-01, -6.3644e-01,  1.9308e-01,\n",
       "           5.2181e-01,  1.5072e-01, -1.3919e-01, -5.7820e-01,  1.9132e-01,\n",
       "          -1.9453e-01, -7.0282e-01,  1.4964e-01, -1.4359e-01,  2.8282e-01,\n",
       "           5.0103e-01,  6.3936e-01,  1.6586e-01, -3.8710e-01, -1.6847e-02,\n",
       "          -5.0519e-01,  7.1554e-02, -1.8306e-01,  1.7325e-01, -2.6914e-01,\n",
       "          -2.8724e-01, -3.0614e-01,  1.6367e-01,  2.4286e-01,  3.0013e-01,\n",
       "           3.1120e-01,  1.6530e+00, -2.3911e-01, -2.7961e-01,  5.9665e-01,\n",
       "           9.0213e-03, -2.4908e-02, -2.2733e-01,  2.8403e-01,  1.1474e-01,\n",
       "           2.6548e-01, -3.0710e-01, -2.0369e-01, -7.1450e-02,  8.5490e-02,\n",
       "           3.8125e-01, -1.0456e+00, -2.4686e-01, -1.6465e-01,  5.7844e-01,\n",
       "           1.3285e-01,  7.7519e-01, -3.9708e-01,  3.8031e-01,  4.0349e-01,\n",
       "          -9.1632e-01, -1.8390e-01,  7.4529e-02,  2.8054e-01, -2.5684e-01,\n",
       "          -2.3060e-01, -2.3385e-01, -2.5440e-02, -9.9006e-02,  6.5606e-01,\n",
       "           3.7995e-01, -7.3328e-01,  5.7676e-01,  6.1132e-01, -1.2815e-02,\n",
       "           4.1695e-01,  1.6172e-01,  1.9861e-01, -1.5513e-01, -2.0143e-01,\n",
       "           2.6425e-01,  2.2313e-01, -4.5190e-01, -2.4440e-01,  2.8130e-01,\n",
       "           4.7389e-01, -1.5121e-01,  1.6101e-01, -1.9505e-01, -9.6667e-02,\n",
       "          -3.6360e-01, -1.3938e-01, -2.3518e-01, -2.6297e-01, -1.1675e-01,\n",
       "           1.0009e-01, -1.0131e-01,  1.1767e-01,  3.3478e-01, -5.7070e-02,\n",
       "           7.6098e-01,  3.5911e-01,  5.1213e-01, -2.7551e-01, -1.6322e-01,\n",
       "           1.1151e-01,  3.6967e-01, -9.4679e-02,  8.7862e-01,  7.3330e-02,\n",
       "          -4.0101e-01, -3.6010e-02, -1.0515e-01, -2.4222e-01,  1.2275e-01,\n",
       "          -1.1474e-01,  4.5187e-01,  5.7993e-02, -1.2680e-01,  3.1383e-01,\n",
       "          -2.3638e-01,  3.5392e-01, -7.9113e-01,  1.8509e-01,  3.8241e-01,\n",
       "          -3.0961e-01,  4.8416e-01, -2.3816e-02,  7.2519e-01, -2.4295e-01,\n",
       "           1.5822e-01, -7.6637e-01, -3.1679e-01, -1.7306e-01,  8.3996e-02,\n",
       "          -8.9049e-01, -2.6289e-01,  3.1644e-01,  1.2364e-01,  1.2562e-01,\n",
       "           8.4939e-01, -1.2116e-01, -3.0483e-01, -2.3912e-01,  1.8344e-01,\n",
       "           9.1992e-02, -6.1688e-02,  2.7403e-01,  3.0162e-02,  1.1545e-01,\n",
       "          -9.3505e-01, -2.9236e-01,  1.7188e-01,  2.2567e-01, -4.1036e-01,\n",
       "           9.1631e-01,  1.3134e-01,  1.1352e-01,  7.5199e-01,  2.7749e-01,\n",
       "           7.8049e-01,  6.6786e-01,  2.0860e-01, -6.3340e-01, -2.1855e-01,\n",
       "           7.2659e-02, -4.7093e-01,  1.4314e-01, -1.6040e-01,  1.7869e-01,\n",
       "           4.9239e-01,  4.9623e-01,  4.3228e-01]], device='mps:0'),\n",
       " tensor([[-0.0743, -0.4193, -0.5377,  ...,  0.2489,  0.3612,  0.6175],\n",
       "         [-0.2961, -0.6861, -0.6414,  ...,  0.3717,  0.2839,  0.5318]],\n",
       "        device='mps:0'),\n",
       " tensor([[-0.1230, -0.2743, -0.7632,  ...,  0.0595,  0.8762,  0.7366],\n",
       "         [-0.2143, -0.4630, -0.6023,  ...,  0.1281,  0.4524,  0.6160]],\n",
       "        device='mps:0'),\n",
       " tensor([[-3.6574e-01, -6.4201e-01, -3.8116e-01, -6.8043e-01,  2.5284e-02,\n",
       "          -2.1048e-01, -5.0085e-01,  5.6467e-01, -2.5217e-02, -1.8448e-02,\n",
       "           1.0611e-01, -9.8275e-02,  2.4870e-02,  2.8472e-01, -7.7405e-01,\n",
       "           9.2675e-02,  3.2293e-02, -1.0692e-01,  1.3482e-01, -4.3608e-02,\n",
       "           8.2122e-01,  5.8324e-01, -1.6158e-02, -5.4814e-02, -4.3541e-01,\n",
       "          -5.1552e-02, -1.8004e-01,  2.4182e-01,  1.5429e-01, -3.9311e-01,\n",
       "           3.1121e-01,  5.4872e-01, -5.3917e-01,  1.1615e+00, -4.2926e-02,\n",
       "          -2.4630e-01, -9.6363e-03, -5.2133e-01,  2.0321e-01, -2.3021e-01,\n",
       "          -1.6121e-01,  1.6962e-01,  5.0674e-01,  4.0999e-01,  4.8922e-01,\n",
       "           3.2770e-01, -6.7295e-03, -9.5665e-02,  9.2150e-03, -4.3067e-01,\n",
       "          -4.4111e-01, -8.1222e-03,  1.8788e-01,  3.7541e-01, -3.7775e-01,\n",
       "          -3.3299e-02,  7.1501e-01,  4.0057e-01, -3.9838e-01,  1.0956e-01,\n",
       "          -2.8334e-01,  6.6823e-01, -2.2259e-01, -1.5115e-01, -3.5625e-02,\n",
       "          -1.3742e-01, -3.0596e-01, -5.6061e-01,  1.2652e-01, -1.1049e-01,\n",
       "           3.5563e-02, -3.9302e-01,  1.4360e-01, -7.2882e-02, -2.5910e-02,\n",
       "           1.9753e-02,  1.5737e-01,  2.6958e-01, -4.7910e-01, -1.7403e-02,\n",
       "          -9.3610e-02, -1.1140e-01,  4.9058e-01, -2.3494e-01, -8.7400e-02,\n",
       "          -3.3561e-02,  3.7487e-01,  4.3673e-02,  2.4173e-01,  1.4955e-01,\n",
       "           6.3722e-01, -2.6374e-01, -9.9457e-02, -6.8665e-01,  4.3909e-01,\n",
       "           2.6164e-02,  3.6944e-01, -1.1727e-01,  1.8091e-01, -5.8328e-01,\n",
       "          -1.2364e-01,  1.0820e-02, -1.9752e-01, -3.0425e-01, -2.7480e-01,\n",
       "          -3.8762e-01,  4.8497e-01,  3.6907e-01, -1.0712e+00,  6.2793e-03,\n",
       "           1.8516e-02, -2.2658e-01, -9.6829e-03, -7.1886e-01, -2.1227e-01,\n",
       "          -4.0918e-02,  2.9682e-01, -7.4513e-02, -3.6314e-01,  1.8315e-01,\n",
       "           5.0821e-01, -7.2372e-02, -2.3744e-01,  1.5829e-01,  3.2064e-01,\n",
       "          -4.3105e-01,  3.8972e-01, -2.8791e-02, -1.1305e-03,  5.6955e-01,\n",
       "          -1.2091e-01, -2.8363e-01, -4.0995e-01, -2.1416e-01,  1.1970e-01,\n",
       "          -2.8661e-01,  1.3291e-01, -1.3773e-01, -7.4371e-02, -1.4726e-01,\n",
       "          -3.5953e-01,  7.6476e-02,  9.0664e-02, -2.9639e-01,  1.8283e-02,\n",
       "          -4.4380e-01,  3.0480e-01, -6.3378e-02, -6.9256e-02, -1.5947e-01,\n",
       "          -6.1174e-01, -3.7595e-01,  5.8579e-01,  4.3330e-02,  1.5932e-01,\n",
       "          -6.6284e-01,  7.4245e-01,  8.6503e-02,  4.8153e-01,  4.1220e-02,\n",
       "           4.9433e-01,  1.5757e-01,  3.3976e-01, -6.0038e-01, -1.6289e-01,\n",
       "           2.0693e-01, -3.3331e-01, -2.3320e-01, -1.1482e-01,  2.8127e-01,\n",
       "           7.7620e-02, -8.8573e-02,  4.7506e-01, -2.9636e-01, -2.1081e-01,\n",
       "          -2.5993e-01,  1.6462e-02,  2.2230e-01, -1.9887e-01,  3.1568e-01,\n",
       "           3.0709e-01, -1.5645e+00,  4.2686e-01, -2.9999e-01, -6.6220e-01,\n",
       "          -2.9658e-02,  1.3383e-02, -1.5157e-01,  4.7298e-01,  4.8421e-02,\n",
       "          -3.5287e-01, -7.4812e-02, -7.0360e-02,  1.4798e-01, -1.1635e-01,\n",
       "           2.2344e-01,  4.4230e-01,  1.4273e-01, -3.1860e-01, -4.3578e-02,\n",
       "          -1.8810e-01,  4.5379e-01,  4.7945e-02,  6.8987e-01,  2.9829e-01,\n",
       "           2.6048e-01, -3.4794e-02,  1.7792e-01,  2.6431e-01,  2.6708e-01,\n",
       "           3.6485e-02, -7.2677e-02, -6.7856e-01,  1.2723e-01, -5.4304e-01,\n",
       "           3.8493e-01, -4.0737e-01, -5.7342e-01,  1.3785e-01,  2.2530e-01,\n",
       "          -2.0329e-01, -9.1859e-02,  1.5237e-02, -1.8741e-01, -5.9478e-02,\n",
       "           1.0135e+00, -3.1305e-01,  1.8470e-02,  1.0010e-01,  8.3199e-02,\n",
       "          -8.4897e-02, -8.3064e-01,  4.7166e-01,  4.0642e-01, -5.0533e-02,\n",
       "          -1.4929e-01, -2.6659e-01,  5.9748e-01, -4.1607e-02, -2.8942e-01,\n",
       "          -3.1914e-01, -3.2422e-01, -4.5474e-01, -2.5027e-01,  2.2183e-01,\n",
       "           1.8011e-01, -3.8705e-02,  6.3950e-01,  4.2596e-01,  3.1737e-01,\n",
       "          -2.3475e-01, -3.0463e-02, -1.8129e-01,  9.1710e-02, -3.4017e-01,\n",
       "           1.2370e-02,  1.5032e-02, -5.7648e-01,  5.2378e-01, -1.4187e-01,\n",
       "          -5.3881e-01, -5.6217e-02,  3.4904e-01,  4.9617e-01, -2.0771e-01,\n",
       "           3.3031e-01,  2.2450e-01, -1.8884e-01,  3.7978e-01,  3.6569e-01,\n",
       "           2.4707e-02,  6.1415e-01,  2.1891e-01, -8.9961e-01, -4.6033e-01,\n",
       "          -3.4677e-01, -5.3938e-01,  1.7966e-01, -3.3926e-01, -1.6111e-01,\n",
       "          -6.5670e-02, -4.2738e-01, -7.6962e-02, -2.6643e-01, -2.2777e-01,\n",
       "          -9.3339e-02,  1.7803e-01, -5.0965e-02, -4.5848e-01, -2.3458e-01,\n",
       "          -7.1839e-01,  2.8023e-01,  1.0635e-01, -1.1142e-01,  1.7551e-01,\n",
       "           4.0993e-01,  3.2721e-01,  4.2593e-01,  4.8364e-01, -2.8290e-01,\n",
       "          -3.8511e-01, -4.6505e-01,  6.3453e-01, -2.3411e-01,  4.0744e-01,\n",
       "           2.0306e-01,  7.4918e-02,  7.4018e-01, -3.8872e-01, -4.4197e-01,\n",
       "          -1.6579e-01,  6.1072e-01, -3.5303e-02, -3.2708e-01,  1.8014e-01,\n",
       "          -1.0248e+00, -3.3646e-01, -2.1877e-02, -4.2301e-03, -5.0281e-01,\n",
       "           6.4857e-02,  1.2375e-01, -3.2968e-01,  3.2285e-01,  6.6198e-02,\n",
       "           4.7160e-01,  8.0589e-01, -8.1440e-02, -1.0962e-01, -1.0198e-01,\n",
       "          -2.2814e-01, -8.1145e-02, -9.0077e-02, -4.0996e-01, -4.0325e-01,\n",
       "          -4.0862e-02,  4.1039e-01,  3.0178e-01,  2.1718e-01, -1.5888e-01,\n",
       "           7.5233e-01,  8.3399e-02, -9.1312e-02,  3.5424e-01, -4.6293e-01,\n",
       "           1.4574e-01,  2.0732e-01, -9.8637e-01, -4.7974e-02,  2.3366e-01,\n",
       "           1.7717e-01, -4.6145e-01,  1.2221e-01, -7.2887e-02,  2.6095e-01,\n",
       "           4.9720e-01,  1.5027e-01,  5.7054e-01,  5.2830e-01,  7.6548e-01,\n",
       "          -3.2936e-01,  5.1614e-01,  1.9591e-02, -3.8961e-01, -2.9262e-01,\n",
       "           5.9958e-01,  2.4961e-01, -9.2411e-01, -2.1386e-01, -3.7308e-01,\n",
       "           8.2386e-03, -6.4126e-01,  7.1344e-02, -2.6252e-02, -1.2227e-01,\n",
       "          -4.7220e-01, -7.7308e-01, -1.7482e-02, -1.0302e+00, -5.7997e-01,\n",
       "           7.2808e-02, -1.5842e-02,  1.4564e-01, -6.2857e-01, -4.2383e-01,\n",
       "          -3.6325e-01, -1.2916e-01,  3.3714e-01,  3.6478e-01, -5.5752e-02,\n",
       "           3.2105e-01,  1.3223e-01, -6.5226e-02,  2.0412e-02, -1.0480e-01,\n",
       "          -7.7323e-02,  1.7837e-01, -2.1854e-01, -5.6643e-02,  5.3415e-01,\n",
       "          -3.1276e-01,  2.2316e-01,  1.2804e-01, -6.7550e-02,  3.0109e-01,\n",
       "          -6.1516e-01,  5.3496e-02, -2.3873e-01, -5.3411e-01,  3.7850e-01,\n",
       "          -5.0815e-01,  8.6531e-02, -2.8231e-01,  1.9126e-01,  2.7163e-01,\n",
       "           2.4636e-01, -2.7678e-01, -9.8383e-02, -2.4445e-02, -2.6937e-01,\n",
       "           4.1077e-01, -7.8466e-02, -6.6302e-03,  3.5900e-01, -1.1049e+01,\n",
       "           7.3103e-02,  1.0576e+00, -1.1710e-01, -3.0611e-01, -9.4873e-02,\n",
       "           2.4621e-01, -4.6211e-01,  4.5943e-01, -3.1712e-01, -4.8154e-01,\n",
       "          -2.5218e-01, -2.4711e-02,  1.4232e-01, -7.0663e-01, -5.6128e-01,\n",
       "          -1.6794e-01, -9.0206e-01,  1.9769e-01, -2.7094e-01,  1.1694e-01,\n",
       "          -4.3772e-02,  2.5932e-01,  4.6993e-01, -4.1853e-01, -4.6460e-01,\n",
       "          -3.2908e-01, -1.9637e-01,  1.5998e-01,  3.7072e-01, -6.1366e-02,\n",
       "          -8.2209e-02,  5.1987e-01, -9.8187e-02, -3.4302e-01, -6.0965e-01,\n",
       "          -3.0679e-01,  6.3869e-02,  5.4158e-01, -1.5408e-01, -5.8522e-01,\n",
       "          -2.7093e-01, -1.3590e-02,  5.7188e-02,  2.5482e-01,  4.4475e-01,\n",
       "           1.1606e+00,  6.6664e-01,  4.9826e-01, -8.2384e-02, -5.3981e-01,\n",
       "          -1.0710e+00,  2.1865e-01, -2.8295e-01,  1.8301e-01, -1.4045e-01,\n",
       "           5.0854e-01, -6.6093e-01, -2.5442e-01, -4.8855e-01, -1.4245e-01,\n",
       "           3.6252e-01,  1.7075e-02,  1.1792e-02, -1.0619e-01,  4.2064e-02,\n",
       "           4.6740e-01,  6.8424e-02,  4.9468e-02, -2.1668e-01, -5.1386e-02,\n",
       "          -3.8331e-02,  8.0797e-01,  4.1160e-01,  5.8123e-01,  5.3451e-01,\n",
       "          -5.4584e-01, -3.4606e-01,  2.0300e+00, -1.0196e-01,  1.1954e-01,\n",
       "          -2.6989e-02,  7.9871e-02, -3.2352e-01, -6.1768e-01,  1.6043e-01,\n",
       "           1.8545e-01,  4.9216e-02,  4.1505e-01,  7.6536e-01, -6.8807e-02,\n",
       "          -6.8840e-01,  3.9271e-02,  4.0389e-01,  5.0786e-01,  6.4969e-01,\n",
       "          -2.0627e-01,  3.5640e-01,  1.4063e-03,  6.1361e-01, -4.0076e-01,\n",
       "          -1.0032e-01, -5.6035e-01, -2.8546e-01, -3.1990e-02, -3.1639e-01,\n",
       "           8.2178e-02,  3.9284e-02,  2.0867e-01,  4.5236e-01, -5.1592e-01,\n",
       "           7.8131e-02, -1.3980e-01,  7.8139e-02, -2.3248e-01, -2.2408e-01,\n",
       "           3.1980e-01,  5.1606e-01, -1.9527e-01, -4.6912e-01, -1.1871e-01,\n",
       "           4.2201e-02,  2.4786e-01,  1.8960e-01, -4.7242e-01, -2.8964e-01,\n",
       "          -8.5285e-01, -4.3929e-02,  7.8306e-02, -1.3160e-01,  6.6855e-01,\n",
       "           1.5886e-01, -1.8007e-01,  3.0327e-01, -4.4337e-01, -5.7473e-01,\n",
       "          -6.1631e-01,  1.0183e-02,  1.6682e-01, -5.5135e-01,  8.2404e-02,\n",
       "           2.1627e-01, -7.7723e-02, -6.5687e-02,  6.5576e-01,  7.8596e-02,\n",
       "           3.4096e-02,  2.1850e-01, -4.7326e-02,  1.8583e-01, -3.4507e-01,\n",
       "           3.9943e-03, -3.0384e-01,  4.6522e-02, -1.3332e-01, -4.0151e-01,\n",
       "           1.1787e-01, -2.6825e-04,  1.6361e-01, -2.6926e-01,  2.3719e-01,\n",
       "           2.4040e-01, -5.8158e-02,  3.1301e-01,  2.5379e-01,  5.7178e-01,\n",
       "           4.7746e-02, -4.1499e-01, -1.8952e-01, -1.2826e-01,  3.5808e-01,\n",
       "           1.4912e-01, -8.3443e-02, -2.4185e-01, -4.6559e-01, -2.7121e-01,\n",
       "          -4.8933e-01,  5.1874e-01, -6.5424e-01, -5.5202e-01,  3.1295e-01,\n",
       "          -1.4707e-01,  2.3372e-01, -2.2586e-01,  2.2144e-01,  2.8480e-02,\n",
       "          -4.4666e-01, -2.9188e-02,  2.7150e-01,  3.3440e-01,  1.2613e-01,\n",
       "           3.6097e-01,  3.7395e-01, -1.4790e-01,  5.6968e-01, -3.5821e-01,\n",
       "          -1.9202e-01,  9.3395e-02, -2.7615e-01,  3.2126e-01,  1.0308e-01,\n",
       "          -6.6965e-01, -1.8235e-01, -6.1488e-02, -5.8360e-02,  1.6962e-01,\n",
       "           5.5257e-01,  1.4479e+00, -6.1746e-01, -5.1843e-02,  1.0310e+00,\n",
       "           1.4269e-01, -1.4005e-01, -1.1874e-01,  1.8052e-01,  3.5426e-02,\n",
       "           1.1753e-01,  1.1638e-02, -3.5391e-01, -1.3672e-01,  3.9956e-01,\n",
       "           2.5333e-01, -9.6841e-01, -2.7614e-01, -2.5443e-02,  2.4421e-01,\n",
       "           2.6599e-01,  3.2689e-01, -4.9218e-01,  2.1322e-01,  5.8895e-01,\n",
       "          -9.0662e-01, -1.4266e-01, -2.6182e-01,  1.2505e-01,  6.7665e-02,\n",
       "          -2.1504e-01, -6.8589e-02, -1.4953e-01, -2.8161e-01,  3.9984e-01,\n",
       "           7.5241e-01, -7.7193e-01,  1.4645e-01,  3.4417e-01, -2.7428e-01,\n",
       "           3.3789e-01,  2.3154e-01,  1.3543e-01,  1.9894e-01,  5.7394e-03,\n",
       "          -3.1730e-01,  6.4379e-01, -5.1588e-01, -3.1033e-01, -1.8344e-01,\n",
       "           5.1586e-01,  1.1527e-01,  1.5900e-02, -2.8562e-01,  3.1469e-01,\n",
       "           7.9061e-02, -3.8173e-01, -2.8283e-01,  7.8457e-02, -1.8556e-01,\n",
       "           3.4683e-02,  2.3743e-01,  6.3914e-02,  2.0986e-01, -4.1984e-01,\n",
       "           5.0312e-01,  6.1022e-01,  9.1397e-01, -1.0573e-01, -4.5204e-01,\n",
       "           5.2393e-02,  3.1346e-01, -3.1122e-01,  5.7742e-01,  5.2647e-01,\n",
       "          -5.0725e-01,  3.4245e-01,  1.6799e-01, -1.8465e-01,  3.5628e-01,\n",
       "          -1.2262e-01,  3.3023e-01, -2.5320e-01,  5.2861e-02,  4.2071e-01,\n",
       "          -1.2848e-01,  3.1701e-01, -7.8017e-01, -1.3220e-01,  1.3710e-01,\n",
       "          -2.0701e-01, -8.9271e-03,  1.6801e-01,  1.7894e-01, -2.8145e-01,\n",
       "           1.1830e-01, -3.3216e-01, -2.3181e-01,  2.3820e-01, -4.3129e-01,\n",
       "          -7.0299e-01, -1.8829e-01,  4.2950e-02,  3.7519e-01,  5.2732e-01,\n",
       "           4.7279e-01,  2.9072e-01, -1.3236e-01, -3.2629e-01,  1.3847e-01,\n",
       "           5.3355e-02, -2.2347e-01,  4.1832e-01, -4.4473e-01, -4.7679e-02,\n",
       "          -6.2764e-01, -2.7887e-01, -2.8782e-01, -1.6181e-01, -1.7931e-01,\n",
       "           8.8088e-01, -3.9125e-01,  6.1203e-02,  5.1385e-01,  2.9722e-01,\n",
       "           3.5548e-01,  8.0119e-01,  4.3415e-01, -4.9589e-01,  6.0161e-02,\n",
       "           8.0465e-02, -2.0692e-01, -2.8229e-01, -4.4126e-01,  2.5052e-01,\n",
       "           2.1208e-01,  8.4368e-01,  4.9462e-01]], device='mps:0'),\n",
       " tensor([[-0.1626, -0.3850, -0.4584,  ...,  0.0859,  0.8194,  0.3343],\n",
       "         [-0.3397, -0.9407, -0.5348,  ...,  0.4867,  0.9440,  0.6982],\n",
       "         [-0.2394, -0.4080, -0.6520,  ...,  0.1409,  0.6880,  0.2775]],\n",
       "        device='mps:0'),\n",
       " tensor([[-0.1230, -0.2743, -0.7632,  ...,  0.0595,  0.8762,  0.7366],\n",
       "         [-0.2143, -0.4630, -0.6023,  ...,  0.1281,  0.4524,  0.6160]],\n",
       "        device='mps:0'),\n",
       " tensor([[-2.5187e-01, -4.0054e-01, -5.8007e-01, -8.3055e-01, -1.2868e-01,\n",
       "          -2.8804e-02, -7.1917e-01,  6.6701e-01,  1.2709e-01,  5.0855e-01,\n",
       "           1.2570e-01,  5.1916e-01,  5.1676e-02, -1.1521e-01, -1.6481e-01,\n",
       "          -4.9820e-01,  5.6222e-02,  3.7338e-01,  6.7187e-01, -2.4111e-02,\n",
       "           2.2844e-01,  5.7979e-01, -2.6323e-01, -1.2386e-02, -8.3013e-01,\n",
       "           2.9613e-01, -1.1576e-01,  2.2409e-01,  1.7718e-01, -6.3092e-01,\n",
       "           4.0999e-02,  4.7417e-01,  2.3068e-02,  1.3497e+00, -5.6080e-01,\n",
       "          -1.3230e-01,  8.2213e-02, -3.9453e-01, -2.3010e-01,  3.7283e-02,\n",
       "           6.1164e-01, -4.1315e-01,  1.3635e-01,  4.0254e-01, -6.1085e-02,\n",
       "           8.9076e-02,  1.3899e-01, -3.7744e-01,  2.5892e-01, -8.3961e-01,\n",
       "          -4.8318e-02, -2.0669e-01,  4.9144e-01,  6.2615e-01, -6.3176e-01,\n",
       "          -2.2975e-01,  2.7419e-01, -3.7716e-02, -3.3020e-02,  1.5349e-01,\n",
       "           9.7980e-02,  3.8582e-01, -2.9995e-01,  2.3257e-01,  3.3441e-01,\n",
       "           2.4458e-01, -3.0392e-01, -4.8889e-01, -5.4078e-01, -5.0111e-02,\n",
       "           9.1859e-03,  2.2490e-02,  1.4846e-01, -2.3272e-01,  4.1973e-01,\n",
       "          -7.1214e-01, -6.4354e-02, -8.9125e-02, -2.5973e-01,  2.9246e-01,\n",
       "          -4.5549e-02,  1.9719e-01,  3.1010e-01,  1.0268e-01,  1.2457e-01,\n",
       "           4.3419e-02,  2.7215e-01,  5.2932e-04,  4.7373e-02,  3.7533e-01,\n",
       "          -1.3375e-01, -8.1722e-02, -2.8313e-01, -3.5814e-01,  3.1229e-01,\n",
       "           4.0598e-02,  2.9907e-01, -1.9557e-01,  1.8579e-01, -3.2474e-01,\n",
       "           1.5588e-01,  1.7768e-01, -1.8083e-01,  5.7286e-02, -3.5172e-01,\n",
       "          -4.5475e-01,  2.3589e-02, -2.8569e-01, -4.8285e-01,  2.2697e-01,\n",
       "          -1.4211e-01, -3.0205e-01, -4.0922e-02, -3.8935e-01, -3.1524e-02,\n",
       "          -7.5164e-02,  2.6013e-01, -1.3230e-01, -3.4806e-01, -8.3033e-02,\n",
       "           2.0734e-01, -2.2808e-01,  1.6077e-01,  2.6882e-01,  1.2232e-01,\n",
       "          -9.6705e-02,  6.5251e-01,  1.8950e-02,  3.1674e-01,  2.5763e-01,\n",
       "           1.1699e-01, -6.5768e-02, -1.9745e-01, -2.3154e-01,  7.5988e-02,\n",
       "          -7.6941e-02,  1.0687e-01, -1.8182e-01, -2.2790e-01, -1.8508e-01,\n",
       "           3.1771e-01, -3.3097e-01, -6.0893e-01, -4.9260e-01, -1.6098e-01,\n",
       "          -1.6326e-01, -2.4804e-02, -8.2473e-03, -3.3693e-01, -3.1956e-01,\n",
       "          -6.7443e-01, -3.6717e-01,  8.1041e-01, -1.6831e-02, -3.3356e-01,\n",
       "          -1.6499e-01,  2.3208e-01,  3.1666e-01, -4.5959e-02,  3.7161e-01,\n",
       "           7.7250e-01, -4.7670e-02,  1.6786e-01, -5.1827e-01, -3.8195e-01,\n",
       "           4.8335e-02,  2.6658e-01, -2.7054e-01,  6.1879e-01,  4.8352e-02,\n",
       "           7.5627e-02, -6.6170e-02,  2.7430e-01, -3.2704e-01, -1.2324e-02,\n",
       "          -5.7573e-01,  3.7750e-01,  2.9539e-01, -3.9892e-01, -6.6567e-02,\n",
       "          -2.0834e-02, -1.2060e+00,  9.6964e-02, -8.3040e-01,  3.1236e-01,\n",
       "          -2.3110e-01,  1.7963e-01,  2.3389e-01,  2.9545e-01, -6.5320e-02,\n",
       "           2.5404e-01,  2.8725e-01,  2.5687e-01,  3.3609e-01, -3.4924e-02,\n",
       "           2.8019e-01,  2.3924e-01,  5.0790e-01,  4.1073e-01,  1.4769e-01,\n",
       "          -3.5440e-01,  2.9683e-01,  3.5781e-01,  9.0598e-02,  1.6954e-01,\n",
       "           9.9825e-02, -2.1636e-01,  2.1091e-01,  1.7501e-01, -1.6618e-01,\n",
       "          -2.2406e-01, -3.1100e-02, -4.6818e-01, -5.6655e-02, -4.8467e-01,\n",
       "           3.2557e-01, -5.1983e-01, -1.1912e-01,  3.4831e-01,  2.4340e-01,\n",
       "          -1.4896e-02,  1.8319e-01,  1.7196e-01, -2.4338e-01,  2.1735e-01,\n",
       "           3.7449e-01, -3.4618e-01, -3.9804e-02, -3.2197e-01, -2.9367e-01,\n",
       "           2.4357e-01, -5.6244e-01,  6.6635e-01,  4.0574e-01,  6.0165e-02,\n",
       "           1.5119e-01, -2.7920e-01, -5.1638e-02, -5.7408e-01,  3.5471e-01,\n",
       "          -3.2130e-01, -3.4955e-01, -5.4801e-01,  6.4487e-02,  4.2773e-01,\n",
       "           2.7565e-01, -6.4915e-02,  7.7868e-01,  4.2408e-01,  4.2813e-01,\n",
       "          -4.3003e-01, -1.5896e-02,  4.0377e-01, -4.1405e-02, -4.6059e-01,\n",
       "           6.7116e-02, -1.4962e-01, -5.1643e-01,  2.8000e-01, -2.6598e-01,\n",
       "          -1.2047e-01, -1.4913e-01,  3.3682e-01,  1.5365e-01, -2.1964e-01,\n",
       "          -3.8461e-01,  3.9587e-01, -1.4349e-01, -4.1723e-01,  4.7123e-01,\n",
       "          -3.5025e-01,  1.5209e-01,  1.0694e-01, -3.8441e-01, -6.2072e-01,\n",
       "          -3.2235e-01,  2.3090e-01,  1.5585e-01,  3.4799e-02, -6.6306e-01,\n",
       "          -4.0571e-01,  3.1197e-01,  2.4601e-01,  2.9976e-01,  2.9840e-01,\n",
       "           5.7427e-01, -3.0944e-01, -1.6702e-01,  1.4982e-01, -4.1377e-01,\n",
       "          -1.5704e-01, -3.3228e-02, -1.9995e-01,  2.9569e-01,  2.4768e-01,\n",
       "           1.1600e-01,  2.0367e-01,  4.3482e-01, -2.5272e-01, -3.8673e-01,\n",
       "          -7.1543e-02,  2.0161e-01, -9.3629e-02, -5.8306e-02,  2.3951e-01,\n",
       "          -6.3095e-02,  5.9650e-01,  7.2143e-01,  7.7439e-02, -6.4006e-01,\n",
       "           1.7696e-01,  5.2502e-01, -2.7253e-01, -4.0343e-02,  9.6365e-02,\n",
       "          -1.0344e+00, -2.1331e-01, -1.2883e-01, -2.0181e-01, -6.5352e-01,\n",
       "           8.8115e-02,  3.4863e-01,  4.6833e-03, -1.7864e-01, -3.0947e-01,\n",
       "           1.0448e-01,  4.8724e-02, -4.8980e-02,  4.3534e-01, -2.8901e-01,\n",
       "          -3.4797e-01, -2.3438e-01, -2.1532e-01, -5.4201e-01, -3.9968e-01,\n",
       "           3.5914e-01,  2.8129e-01,  1.1986e-01, -2.6486e-01, -5.1832e-01,\n",
       "           4.5577e-01, -6.8169e-01, -1.4632e-01,  2.4840e-01, -4.7702e-01,\n",
       "           1.4650e-02,  4.8990e-01,  1.4999e-01,  1.6336e-01,  6.5258e-02,\n",
       "           7.8557e-02, -3.3903e-01, -5.0030e-01, -1.5000e-01,  6.2323e-01,\n",
       "           3.2283e-01,  3.3506e-01, -1.7236e-01,  3.9096e-01,  4.5167e-01,\n",
       "          -2.9582e-01,  4.1670e-01,  1.0930e-02,  7.7439e-02, -6.0414e-02,\n",
       "           1.9048e-01,  3.4370e-01, -3.6842e-01,  2.0145e-01,  1.7929e-01,\n",
       "          -3.1634e-01, -3.2149e-01,  1.5691e-02,  2.0642e-01, -1.8580e-01,\n",
       "          -3.0875e-01, -2.0361e-01,  9.9698e-02, -3.4856e-01, -4.3104e-01,\n",
       "           2.9161e-01, -3.6543e-02,  7.0495e-02, -4.0357e-01, -4.3462e-01,\n",
       "          -2.0181e-01, -2.6055e-01,  6.8681e-01,  2.1064e-01, -1.9621e-01,\n",
       "           1.8720e-01, -1.0907e-01, -3.6330e-01,  4.7847e-01,  2.0162e-02,\n",
       "          -4.8754e-01,  4.5482e-01, -3.4822e-01,  1.0627e-01,  5.5756e-01,\n",
       "          -1.6769e-01, -1.0840e-01,  1.1008e-01,  1.0820e-01,  2.9026e-01,\n",
       "          -3.9884e-01,  1.5516e-01, -6.3476e-02, -3.1885e-01,  3.1820e-01,\n",
       "           2.6411e-01,  2.2781e-01, -1.7137e-01, -3.3426e-02,  3.5677e-02,\n",
       "           2.8782e-01, -6.0853e-02, -3.9888e-01, -4.7347e-01, -4.6132e-01,\n",
       "           1.2832e-01,  1.8927e-01,  5.6207e-01,  3.4144e-01, -1.1394e+01,\n",
       "           1.2938e-01,  7.8031e-01, -4.0713e-01, -8.2819e-02, -5.0320e-02,\n",
       "           1.0976e-02, -1.3708e-01,  4.2955e-01, -2.7734e-01, -4.4105e-01,\n",
       "          -3.5727e-01, -1.3308e-01,  1.5454e-01, -3.2189e-01, -2.6945e-01,\n",
       "          -1.4686e-02, -7.0935e-01,  4.0963e-01, -3.0286e-01,  5.0298e-01,\n",
       "          -2.3511e-01, -8.3798e-02,  1.5816e-01, -3.5851e-01, -1.3439e-02,\n",
       "           5.9766e-02, -1.6033e-01,  1.4651e-01,  4.2486e-01, -5.2072e-02,\n",
       "           3.3244e-02, -4.1258e-03,  4.8461e-02, -2.5723e-01, -5.9999e-01,\n",
       "           1.9296e-02, -1.4141e-01,  3.3422e-01, -4.4133e-02, -7.1265e-01,\n",
       "          -2.2533e-01, -2.7418e-01,  3.1407e-01,  2.7021e-01,  9.0598e-01,\n",
       "           2.2955e-01,  1.0686e+00,  1.5022e-01,  1.7748e-01,  2.6230e-02,\n",
       "          -7.8764e-01, -3.3269e-02, -4.1907e-01,  2.3280e-02,  1.0114e-01,\n",
       "           3.6576e-01, -7.9367e-01,  2.5251e-01, -3.5521e-01,  3.6487e-02,\n",
       "           4.9094e-01,  3.1768e-02,  2.7700e-01, -3.2298e-02, -7.8123e-02,\n",
       "           1.2089e-01, -2.8544e-02,  1.1098e-01, -2.2583e-01, -2.4084e-01,\n",
       "          -1.0177e-01,  5.9888e-01,  3.9592e-01, -7.2211e-01,  3.6112e-01,\n",
       "           4.9258e-02, -1.8622e-01,  1.4556e+00, -4.8573e-02,  1.8361e-02,\n",
       "           1.9800e-03,  2.9604e-01, -5.0236e-01, -7.3516e-01,  3.6511e-01,\n",
       "          -1.0211e-01,  5.6427e-02,  1.0127e-01,  4.8148e-01, -9.7481e-02,\n",
       "          -4.7940e-01, -1.8593e-03,  1.0556e-01, -1.3380e-01,  5.4710e-01,\n",
       "          -7.2824e-01, -1.6963e-01, -1.1223e-01,  3.1723e-01, -8.4054e-01,\n",
       "          -6.1126e-02, -7.0210e-01, -3.9328e-01,  5.0586e-01, -2.5896e-01,\n",
       "           1.5852e-01, -2.7443e-02, -4.3439e-02,  5.5574e-01, -2.1489e-01,\n",
       "           7.1555e-02,  2.2813e-01,  8.4995e-02, -1.6733e-01,  3.9349e-02,\n",
       "           4.1630e-01,  4.9571e-01, -4.2740e-01, -6.3911e-01, -7.6901e-03,\n",
       "           1.7180e-01,  1.4933e-01,  3.4019e-01, -5.3463e-01,  3.6431e-01,\n",
       "          -7.9162e-01, -1.8769e-01,  7.2949e-02, -4.3273e-02,  3.6175e-01,\n",
       "           2.3925e-01,  7.2038e-02,  2.2256e-01, -7.7756e-01, -6.4498e-01,\n",
       "          -6.5932e-01, -2.1445e-01,  2.6143e-01, -5.4236e-01, -8.3654e-02,\n",
       "          -1.1033e-01, -4.8837e-03, -5.0292e-02, -6.6102e-02,  1.1939e-02,\n",
       "           1.3632e-01,  2.3587e-01, -1.3023e-01, -1.3178e-01, -9.2464e-02,\n",
       "          -5.5566e-01, -1.1753e-01, -9.9591e-02, -1.2505e-01, -5.6783e-01,\n",
       "          -1.3797e-01,  3.3075e-02, -2.5373e-01, -6.8124e-02,  3.5608e-01,\n",
       "          -2.9832e-02, -2.7856e-02, -4.8850e-01,  7.2846e-01,  4.0279e-01,\n",
       "           5.2088e-02, -2.8298e-01, -5.5189e-01, -1.0567e-01,  2.9592e-01,\n",
       "           1.9555e-01, -6.7007e-02, -1.7709e-01, -2.4145e-01, -4.2522e-01,\n",
       "           1.4302e-01,  7.1147e-01, -6.6296e-02, -6.8115e-01,  2.6621e-01,\n",
       "           5.8223e-01, -1.2882e-01, -2.4432e-01,  2.0983e-02, -9.3284e-02,\n",
       "          -3.1851e-01, -6.0317e-01,  7.8821e-01, -1.1246e-01, -5.8638e-02,\n",
       "           2.2427e-01,  2.0359e-01, -9.2588e-03,  4.0209e-01, -1.5770e-01,\n",
       "          -3.5434e-01, -1.6477e-03, -7.3600e-01,  4.3257e-01,  6.1666e-02,\n",
       "          -3.1489e-01,  7.6082e-01,  2.7260e-01, -6.0593e-02, -4.5779e-02,\n",
       "           2.2863e-01,  7.9941e-01, -2.4636e-01,  6.4653e-02,  5.3779e-01,\n",
       "           8.8110e-02, -1.1229e-01,  3.9580e-02,  4.1665e-02,  4.5410e-02,\n",
       "           3.7012e-01, -3.2145e-02, -4.1564e-01, -6.1194e-02,  3.8924e-01,\n",
       "           3.9295e-01, -5.4610e-01, -7.3487e-01,  2.2333e-02,  6.8551e-01,\n",
       "          -2.9054e-01,  2.3654e-01, -3.2708e-02,  3.0880e-01,  1.2001e-01,\n",
       "          -9.2160e-01, -1.5016e-01, -1.0056e-01,  2.8229e-01, -2.5482e-02,\n",
       "          -1.6139e-01, -1.4727e-01, -2.3686e-01, -1.6208e-01,  6.8308e-01,\n",
       "           2.1991e-01, -6.9160e-01,  3.4058e-02,  5.7281e-01, -6.0551e-01,\n",
       "           4.6052e-01, -2.4577e-01, -3.4302e-02,  3.6435e-01,  1.3117e-01,\n",
       "           4.9067e-02,  1.7856e-01, -4.2138e-01, -4.0598e-01,  1.7824e-01,\n",
       "           9.1672e-01, -7.6613e-02,  1.1166e-01, -3.9384e-01, -1.7182e-01,\n",
       "          -7.1855e-02, -3.1913e-01, -3.1286e-01,  3.7185e-01,  2.0529e-01,\n",
       "          -3.2552e-02, -1.3271e-01, -2.2992e-01,  3.0816e-01,  1.1821e-01,\n",
       "           3.2536e-01,  5.8187e-01,  6.4368e-01, -1.0077e-02, -2.4936e-01,\n",
       "          -6.1235e-02,  1.4568e-01, -1.9524e-01,  5.2704e-01, -2.0627e-01,\n",
       "          -3.3453e-01, -3.8069e-03,  3.0134e-01, -6.1681e-01, -5.8725e-02,\n",
       "          -4.5426e-01,  4.2361e-01, -1.6955e-01,  3.7868e-02,  2.8833e-01,\n",
       "           4.0522e-02,  5.7925e-02, -2.2651e-01,  5.5807e-01,  4.4511e-01,\n",
       "           1.6609e-01,  4.5662e-01,  1.6104e-01,  4.5272e-02, -5.1597e-01,\n",
       "           2.1978e-01, -1.0213e+00, -9.4260e-02, -4.6640e-02, -1.5401e-01,\n",
       "          -4.4224e-01, -2.8126e-02,  1.6814e-01,  1.8538e-01,  6.9597e-01,\n",
       "           7.2967e-01, -3.0871e-01, -4.5290e-01, -1.8510e-01,  1.5304e-01,\n",
       "           2.7705e-01, -1.9517e-01,  3.5971e-01,  9.6103e-02,  2.4432e-01,\n",
       "          -5.0967e-01, -4.3165e-01, -4.2252e-02,  3.1411e-01, -2.0171e-01,\n",
       "           5.7118e-01, -2.0164e-01, -1.3863e-01,  9.0832e-02,  1.3770e-01,\n",
       "           1.6639e-01,  7.9664e-01,  4.3416e-01, -6.3425e-01,  8.7469e-02,\n",
       "          -1.3980e-02, -2.9988e-01,  2.5983e-02, -4.3577e-01,  2.8858e-01,\n",
       "           1.2984e-01,  4.9280e-01,  6.7504e-01]], device='mps:0'),\n",
       " tensor([[-0.2986, -0.7212, -0.5251,  ...,  0.3061,  0.8097,  0.1645],\n",
       "         [ 0.0748, -0.2566, -0.5776,  ...,  0.2010,  1.2229,  0.3245]],\n",
       "        device='mps:0'),\n",
       " tensor([[-0.1505, -0.7134, -0.3094,  ...,  0.3297,  0.0925,  0.4436],\n",
       "         [-0.3292, -0.2590, -0.5016,  ..., -0.0148,  0.3509,  0.1607]],\n",
       "        device='mps:0'),\n",
       " tensor([[-0.3825, -0.5211, -0.7059,  ...,  0.1836,  1.1140,  0.4890],\n",
       "         [-0.2685, -0.4092, -0.8361,  ..., -0.0684,  0.0418,  0.6510],\n",
       "         [-0.2475, -0.7169, -0.6231,  ...,  0.2672,  0.5322,  0.5050],\n",
       "         [-0.0583, -0.6757, -0.8037,  ...,  0.1694,  0.7852,  0.5016]],\n",
       "        device='mps:0'),\n",
       " tensor([[-3.7330e-01, -4.3563e-01, -3.9906e-01, -8.7912e-01, -2.9993e-01,\n",
       "          -4.2529e-01, -4.9387e-01,  8.3807e-01, -1.4804e-01,  8.3468e-01,\n",
       "           2.5768e-01,  2.4697e-01,  2.1780e-01,  2.2983e-01, -6.0866e-01,\n",
       "          -1.4258e-01, -5.8298e-01,  4.4077e-01,  6.8131e-01,  5.6858e-01,\n",
       "           8.5224e-01,  5.6634e-01,  4.1325e-02, -2.0508e-01, -4.3196e-01,\n",
       "           5.5375e-01,  1.1628e-01, -1.2326e-01, -6.8571e-03, -3.2234e-01,\n",
       "          -9.9030e-02,  1.9631e-01,  4.2344e-02,  9.9706e-01,  6.0629e-02,\n",
       "          -1.6206e-01,  3.0908e-01, -3.2658e-01, -3.8757e-01, -4.2795e-01,\n",
       "           5.8308e-01, -2.1369e-01,  1.8079e-01,  1.1607e-01,  1.1280e-01,\n",
       "          -1.5397e-01,  2.9237e-03, -3.5541e-01,  3.3282e-01, -8.5186e-01,\n",
       "          -2.3034e-01, -2.4235e-01,  2.0091e-01,  5.2927e-01, -8.6555e-01,\n",
       "          -3.3380e-01,  1.9529e-01,  1.9573e-02, -2.7749e-01,  4.5230e-02,\n",
       "           1.0038e-01,  2.2931e-01,  8.7179e-02,  3.8366e-02,  4.7978e-01,\n",
       "          -3.3107e-03, -3.7004e-01, -7.8375e-01, -6.3425e-01, -2.3411e-01,\n",
       "          -9.2147e-02,  2.9986e-01, -1.2443e-01, -3.9335e-01,  2.7354e-02,\n",
       "           9.7849e-02,  4.2066e-01, -1.8259e-01, -3.0797e-01,  9.7147e-02,\n",
       "           1.4592e-01, -8.3907e-03, -8.8849e-02,  1.5923e-02, -3.2036e-02,\n",
       "          -1.3219e-01,  7.3125e-01, -1.6721e-01,  2.0639e-01, -1.3859e-02,\n",
       "           4.3311e-01, -9.8102e-02, -2.7453e-01, -4.4259e-01,  4.1971e-01,\n",
       "          -2.2450e-01,  5.4927e-01, -1.3504e-01,  3.4382e-01, -2.7664e-01,\n",
       "           2.1260e-01,  3.0183e-01, -2.4012e-01,  2.7830e-01, -5.0913e-01,\n",
       "          -4.7134e-01, -1.5401e-02, -1.2060e-01, -2.6015e-01,  4.2575e-01,\n",
       "           2.0281e-02, -2.7076e-01,  7.9879e-02, -4.0117e-01, -2.9419e-01,\n",
       "           2.3439e-01,  3.4107e-01, -1.2963e-01, -4.2595e-01,  2.9983e-01,\n",
       "           3.3500e-01, -2.2323e-01,  8.5301e-02,  6.8300e-02,  2.2206e-01,\n",
       "          -3.5102e-01,  6.3847e-01,  3.9178e-02,  4.3307e-02,  2.2765e-02,\n",
       "          -8.2804e-03,  1.9012e-01, -2.9458e-01,  2.4041e-01,  2.3394e-01,\n",
       "           1.1344e-01, -5.1910e-01, -2.9641e-01, -2.5141e-01,  6.6487e-02,\n",
       "          -6.8810e-03, -4.2198e-01, -6.6668e-01,  2.3266e-02,  1.4582e-01,\n",
       "          -9.6412e-02,  2.7199e-01, -1.4651e-01,  3.2146e-01, -4.9560e-01,\n",
       "          -5.9382e-01, -6.1629e-01,  6.2127e-01,  1.8620e-01, -1.0481e-01,\n",
       "          -2.3555e-01,  2.6801e-01,  2.4787e-02,  2.2299e-01,  9.1248e-02,\n",
       "           8.7076e-01, -2.1596e-01, -1.9572e-01, -4.1764e-01,  1.1730e-01,\n",
       "          -1.6114e-01, -6.8335e-02, -5.4669e-01, -6.1882e-03,  2.5476e-01,\n",
       "           4.3498e-01,  2.3985e-02,  1.0637e-01, -3.9237e-01, -1.2032e-01,\n",
       "          -1.7988e-01,  1.8378e-01,  7.0826e-02, -1.2409e-01, -3.7524e-02,\n",
       "           1.8154e-01, -1.4628e+00, -1.7218e-03, -2.3231e-01, -3.0675e-01,\n",
       "          -2.1397e-01,  3.4101e-01,  3.0077e-01,  1.4889e-01, -2.1606e-01,\n",
       "           1.4610e-01, -4.3024e-02, -1.3326e-01,  4.4508e-01, -2.8805e-01,\n",
       "           4.8085e-01,  8.0443e-01,  2.6888e-01,  3.5346e-01,  1.8127e-01,\n",
       "          -6.7653e-02,  5.1563e-02,  2.0304e-01,  2.4841e-01,  3.6636e-01,\n",
       "           2.4121e-01, -4.7403e-01, -2.9736e-01,  1.4963e-01,  1.2181e-01,\n",
       "          -3.9559e-01, -5.4511e-02, -5.2180e-01, -1.6851e-01, -5.0793e-01,\n",
       "           1.2186e-01, -3.0072e-01,  9.2125e-02,  4.8993e-01, -7.8319e-02,\n",
       "          -5.0982e-02,  5.9609e-02,  8.3177e-03, -3.1299e-01,  1.4715e-01,\n",
       "           4.7415e-01, -9.1142e-02,  3.3471e-02, -2.0477e-01,  2.2621e-02,\n",
       "           1.3266e-01, -6.9271e-01,  6.7040e-01,  6.5506e-01,  1.6479e-02,\n",
       "           1.0775e-01, -1.7044e-01,  7.0169e-02,  8.3362e-02,  2.8949e-01,\n",
       "           2.5252e-01, -6.3166e-01, -8.3849e-01, -2.8517e-02, -2.1365e-01,\n",
       "           1.3882e-01, -2.1557e-01,  2.4303e-01,  8.5828e-01,  1.7138e-01,\n",
       "          -1.6206e-01,  9.0529e-02,  3.6506e-01,  1.4735e-01, -2.8851e-01,\n",
       "          -1.1732e-01,  6.0505e-02, -7.8541e-01,  1.4946e-01, -5.6698e-01,\n",
       "           1.3943e-01, -1.4194e-01,  6.8244e-02,  2.0636e-01, -1.5684e-01,\n",
       "           2.3205e-01,  3.9322e-01,  9.1301e-02, -1.8953e-01,  3.6506e-01,\n",
       "          -2.2512e-01,  2.4045e-01, -3.0861e-01, -9.5253e-02, -7.0906e-01,\n",
       "          -6.1271e-01,  7.3121e-03,  4.3042e-01, -2.5703e-01, -5.0679e-01,\n",
       "          -1.2645e-02, -1.8568e-01, -2.6004e-01, -2.8564e-01, -1.7272e-01,\n",
       "           4.7499e-01, -8.3918e-02,  1.1529e-01, -2.4101e-01, -3.1171e-01,\n",
       "          -6.0102e-01,  3.1251e-01, -6.5396e-01, -1.4228e-01,  3.2872e-01,\n",
       "           3.4773e-01,  6.5156e-02,  2.5190e-01, -9.9570e-02, -1.8821e-01,\n",
       "           1.5580e-01,  3.5111e-01, -3.9531e-03, -1.0806e-01,  2.8256e-01,\n",
       "           3.9927e-01,  4.1358e-01,  4.1941e-01, -1.8233e-01, -3.4320e-01,\n",
       "          -7.4377e-02,  4.7964e-01, -9.3541e-02, -5.1118e-01, -2.1616e-01,\n",
       "          -9.9465e-01, -1.0130e-01, -1.3354e-01, -4.2022e-01, -4.4955e-01,\n",
       "           1.8271e-01,  2.9857e-01,  7.1365e-03,  7.0406e-03, -7.0425e-02,\n",
       "           2.5522e-01,  5.2772e-01, -3.4056e-01,  3.1461e-01, -5.0884e-01,\n",
       "           1.0316e-01, -3.0779e-01, -1.0539e-01, -1.4639e-01, -2.2702e-01,\n",
       "           3.1927e-01,  1.9870e-01,  2.8745e-01, -3.3925e-01, -2.6409e-01,\n",
       "           3.0638e-01, -7.0312e-02, -5.3066e-01,  4.3216e-01, -5.4784e-02,\n",
       "           3.9462e-01,  2.1559e-01, -1.3285e-01,  7.0420e-03,  4.7229e-01,\n",
       "           2.8269e-02, -3.3278e-01, -5.3906e-01, -1.9110e-01,  6.8328e-01,\n",
       "           3.7131e-01,  3.5385e-02, -2.0897e-03,  4.4664e-01,  2.7291e-01,\n",
       "          -5.3640e-01,  2.5678e-01, -3.1737e-01, -3.1406e-01,  2.1918e-01,\n",
       "           8.2434e-02,  3.4183e-01, -3.0140e-01,  1.7066e-01, -3.7548e-01,\n",
       "           8.2295e-02, -6.1386e-01,  1.0628e-01,  1.2381e-01,  3.5131e-01,\n",
       "          -7.9348e-02, -3.0230e-01,  2.2523e-01, -8.1400e-01,  9.8728e-03,\n",
       "          -1.0537e-01, -1.5910e-01,  1.3273e-01, -5.4382e-01, -2.3930e-01,\n",
       "          -1.2235e-01, -2.7630e-01,  6.1257e-01,  4.3151e-01, -5.9885e-03,\n",
       "           2.1047e-01, -1.5674e-01, -1.9861e-01,  3.3940e-01, -2.3198e-01,\n",
       "          -2.2358e-01,  4.2311e-01, -1.7395e-01, -1.7185e-01,  4.5124e-01,\n",
       "          -2.7529e-01, -1.4388e-02,  1.4193e-01,  8.7711e-02,  1.6626e-01,\n",
       "          -1.4436e-01, -1.5605e-01,  5.1627e-01, -5.7370e-01,  3.8941e-02,\n",
       "          -1.1348e-01,  7.6225e-01, -2.8692e-01, -1.5591e-01, -3.2933e-01,\n",
       "           2.2673e-01,  7.2862e-02, -9.8360e-04,  1.7539e-01, -6.0452e-01,\n",
       "           4.7995e-01,  1.6004e-01, -8.8582e-02,  4.5386e-01, -1.1277e+01,\n",
       "          -2.6203e-02,  8.6372e-01, -3.7298e-01,  1.6081e-01, -4.6382e-01,\n",
       "           1.3315e-01,  4.5830e-02,  1.8168e-01, -1.7168e-01, -2.2666e-01,\n",
       "          -4.0394e-01, -1.0159e-01, -7.3791e-02, -8.7253e-02,  9.0844e-02,\n",
       "           6.1434e-02, -7.3849e-01,  1.5514e-01, -1.4089e-01, -7.3357e-02,\n",
       "          -3.4354e-01, -2.4592e-01,  2.7957e-01, -2.3701e-01,  7.9528e-02,\n",
       "          -1.9835e-01, -3.5392e-01,  2.9319e-01,  1.6187e-01, -2.2736e-01,\n",
       "           1.1644e-01, -9.0771e-02,  3.6619e-01, -4.7792e-02, -6.4464e-01,\n",
       "          -1.2448e-01, -4.0683e-02,  1.0660e-01, -3.7017e-01, -1.4678e-01,\n",
       "          -2.0294e-01, -2.4074e-02,  3.4039e-01,  4.2897e-01,  6.5152e-01,\n",
       "           5.0415e-01,  8.6785e-01,  2.0021e-01, -2.1650e-01, -3.0444e-01,\n",
       "          -7.0698e-01,  2.4402e-01, -9.9359e-01,  1.2839e-01, -2.3527e-01,\n",
       "           8.6665e-01, -3.3434e-01,  4.5119e-01, -1.9701e-01, -1.4415e-01,\n",
       "           2.7744e-02,  1.4815e-01, -1.5451e-01, -3.1304e-01, -2.3470e-01,\n",
       "           8.3050e-02, -3.1313e-02, -4.7684e-02, -5.3036e-01, -6.4897e-02,\n",
       "          -2.0977e-01,  6.4223e-01,  1.5430e-01,  1.1421e-01,  1.4542e-01,\n",
       "          -7.5388e-02, -7.6686e-02,  1.4633e+00, -4.2114e-01,  4.2814e-01,\n",
       "          -1.2509e-01,  4.9242e-01, -2.5619e-01, -9.5161e-01, -1.7766e-02,\n",
       "           9.1607e-02,  1.1849e-01,  1.4258e-01,  4.5519e-01, -2.6459e-01,\n",
       "          -4.1147e-01, -1.2100e-01, -1.3494e-01, -7.3936e-02,  1.0959e+00,\n",
       "          -3.1138e-01, -1.4651e-01, -1.4868e-01,  1.8283e-01, -5.4871e-01,\n",
       "          -2.4579e-01, -2.2493e-01, -1.3331e-01,  1.4627e-02,  4.8148e-02,\n",
       "           6.4020e-02,  2.6959e-01,  1.2688e-01,  4.9571e-01, -2.2255e-01,\n",
       "           1.9608e-01,  1.6265e-01,  2.2577e-01, -8.1023e-02, -9.1621e-01,\n",
       "           4.9153e-01,  8.0658e-02, -1.2145e-01, -7.1865e-01, -9.4122e-02,\n",
       "          -1.3295e-01, -5.0573e-02,  3.5937e-02, -3.1941e-01,  6.3315e-01,\n",
       "          -6.5618e-01, -2.9457e-01,  2.7986e-01, -5.1041e-02,  4.6976e-01,\n",
       "           6.9139e-02, -6.0615e-01,  4.4035e-01, -2.7947e-01, -3.4780e-01,\n",
       "          -6.2296e-01, -2.0503e-01,  4.9483e-01, -4.3503e-01,  1.6505e-01,\n",
       "           6.8709e-02, -6.1557e-02, -1.7799e-01,  3.9885e-01,  4.9335e-02,\n",
       "           4.0229e-02,  3.6274e-01, -1.7719e-01,  1.1028e-01, -2.2713e-01,\n",
       "          -6.4851e-02, -1.1554e-01, -3.2002e-01, -1.3948e-01, -6.3937e-01,\n",
       "          -4.7404e-02, -9.4816e-02, -1.3970e-01, -3.5756e-02,  1.5560e-02,\n",
       "           3.9325e-01, -5.5706e-02, -5.8825e-01,  6.7098e-01,  2.7360e-01,\n",
       "          -5.8202e-02, -4.3316e-02, -4.6645e-01, -2.2195e-01,  6.1688e-01,\n",
       "          -4.2440e-02, -7.1388e-02, -3.7537e-01, -4.0942e-02, -4.5655e-01,\n",
       "           2.7158e-02,  1.2123e+00, -2.9441e-01, -6.1812e-01,  3.2690e-01,\n",
       "           5.8111e-01, -4.3046e-01,  6.4334e-02, -1.5329e-01, -7.5591e-03,\n",
       "          -4.5112e-01,  1.6648e-01,  2.6122e-01, -1.3462e-01,  3.0719e-01,\n",
       "           4.1818e-01,  1.2577e-01,  1.9447e-01,  9.0073e-02, -1.6781e-01,\n",
       "          -5.3150e-01,  8.7577e-02, -4.6837e-02, -1.2657e-01,  1.1970e-01,\n",
       "           6.3164e-02, -5.0271e-01, -1.4370e-02, -1.8972e-01,  4.4942e-01,\n",
       "           3.1736e-01,  9.2225e-01, -2.9638e-01, -2.1511e-01,  4.7866e-01,\n",
       "           5.0648e-01, -5.6369e-02,  3.0271e-01,  3.7541e-01, -2.0625e-02,\n",
       "           4.5233e-01,  1.3122e-01, -4.5061e-01, -8.0224e-02,  3.1025e-01,\n",
       "           3.0019e-01, -7.5909e-01, -4.7562e-01,  2.1785e-03,  4.7039e-01,\n",
       "           1.2921e-01,  6.9922e-01, -7.2862e-02,  3.9100e-01,  5.4489e-01,\n",
       "          -1.1108e+00,  1.4330e-01,  1.9419e-01, -3.5238e-01, -8.0417e-03,\n",
       "          -9.0851e-02, -1.9183e-01,  2.8050e-02, -4.5726e-01,  4.0929e-01,\n",
       "           5.2671e-01, -1.0289e+00,  2.2136e-01,  6.9771e-01, -3.3456e-01,\n",
       "           4.2134e-01, -1.4520e-01,  4.9815e-01,  5.9385e-01,  3.3454e-02,\n",
       "           2.2149e-01, -1.8779e-01, -1.1599e-01, -1.6498e-01,  2.1616e-01,\n",
       "           4.5339e-01, -4.6873e-02, -1.3379e-01, -9.0155e-02, -1.5788e-01,\n",
       "          -2.4250e-01,  2.7183e-03, -2.2120e-01, -1.9077e-01,  5.0745e-01,\n",
       "          -1.8578e-01, -1.7018e-01, -1.0750e-01,  1.6829e-01, -2.6362e-01,\n",
       "           3.8931e-01,  4.2900e-01,  9.1312e-01, -3.3820e-02, -2.8248e-01,\n",
       "          -1.7877e-01,  3.7717e-01, -2.9673e-01,  9.3972e-01,  2.7302e-02,\n",
       "          -2.5792e-01, -1.4068e-01,  9.5767e-02, -7.5826e-02,  6.0059e-02,\n",
       "          -3.5799e-01,  9.7694e-02, -2.1135e-01, -1.9997e-01,  2.4918e-01,\n",
       "          -1.7253e-01,  4.8232e-03, -7.9165e-01,  3.1122e-01,  1.2318e-01,\n",
       "          -1.2590e-01,  5.8434e-01, -3.8359e-01,  3.7729e-01, -2.1765e-01,\n",
       "          -1.2799e-01, -3.0587e-01, -2.7245e-01,  5.9273e-01, -2.7162e-01,\n",
       "          -5.9155e-01, -3.5065e-01, -1.0695e-01, -2.5998e-01,  7.2006e-01,\n",
       "           1.0686e+00, -3.2422e-01, -4.3352e-01, -2.2734e-01, -3.2932e-02,\n",
       "          -2.1783e-03,  2.5669e-01, -1.7045e-02, -3.0181e-01, -1.1049e-01,\n",
       "          -5.4632e-01, -2.4343e-01, -2.5080e-02,  3.8137e-01, -2.1579e-01,\n",
       "           5.0319e-01, -7.1049e-02, -1.5799e-01,  8.9018e-01,  4.4370e-01,\n",
       "           7.2323e-02,  7.9881e-01, -1.0325e-01, -4.7279e-01, -3.3936e-01,\n",
       "           1.2045e-01, -3.8146e-01,  1.5275e-01, -5.8603e-01,  4.6049e-01,\n",
       "           2.7097e-02,  2.2748e-01,  5.9120e-01]], device='mps:0'),\n",
       " tensor([[-2.5187e-01, -4.0054e-01, -5.8007e-01, -8.3055e-01, -1.2868e-01,\n",
       "          -2.8804e-02, -7.1917e-01,  6.6701e-01,  1.2709e-01,  5.0855e-01,\n",
       "           1.2570e-01,  5.1916e-01,  5.1676e-02, -1.1521e-01, -1.6481e-01,\n",
       "          -4.9820e-01,  5.6222e-02,  3.7338e-01,  6.7187e-01, -2.4111e-02,\n",
       "           2.2844e-01,  5.7979e-01, -2.6323e-01, -1.2386e-02, -8.3013e-01,\n",
       "           2.9613e-01, -1.1576e-01,  2.2409e-01,  1.7718e-01, -6.3092e-01,\n",
       "           4.0999e-02,  4.7417e-01,  2.3068e-02,  1.3497e+00, -5.6080e-01,\n",
       "          -1.3230e-01,  8.2213e-02, -3.9453e-01, -2.3010e-01,  3.7283e-02,\n",
       "           6.1164e-01, -4.1315e-01,  1.3635e-01,  4.0254e-01, -6.1085e-02,\n",
       "           8.9076e-02,  1.3899e-01, -3.7744e-01,  2.5892e-01, -8.3961e-01,\n",
       "          -4.8318e-02, -2.0669e-01,  4.9144e-01,  6.2615e-01, -6.3176e-01,\n",
       "          -2.2975e-01,  2.7419e-01, -3.7716e-02, -3.3020e-02,  1.5349e-01,\n",
       "           9.7980e-02,  3.8582e-01, -2.9995e-01,  2.3257e-01,  3.3441e-01,\n",
       "           2.4458e-01, -3.0392e-01, -4.8889e-01, -5.4078e-01, -5.0111e-02,\n",
       "           9.1859e-03,  2.2490e-02,  1.4846e-01, -2.3272e-01,  4.1973e-01,\n",
       "          -7.1214e-01, -6.4354e-02, -8.9125e-02, -2.5973e-01,  2.9246e-01,\n",
       "          -4.5549e-02,  1.9719e-01,  3.1010e-01,  1.0268e-01,  1.2457e-01,\n",
       "           4.3419e-02,  2.7215e-01,  5.2932e-04,  4.7373e-02,  3.7533e-01,\n",
       "          -1.3375e-01, -8.1722e-02, -2.8313e-01, -3.5814e-01,  3.1229e-01,\n",
       "           4.0598e-02,  2.9907e-01, -1.9557e-01,  1.8579e-01, -3.2474e-01,\n",
       "           1.5588e-01,  1.7768e-01, -1.8083e-01,  5.7286e-02, -3.5172e-01,\n",
       "          -4.5475e-01,  2.3589e-02, -2.8569e-01, -4.8285e-01,  2.2697e-01,\n",
       "          -1.4211e-01, -3.0205e-01, -4.0922e-02, -3.8935e-01, -3.1524e-02,\n",
       "          -7.5164e-02,  2.6013e-01, -1.3230e-01, -3.4806e-01, -8.3033e-02,\n",
       "           2.0734e-01, -2.2808e-01,  1.6077e-01,  2.6882e-01,  1.2232e-01,\n",
       "          -9.6705e-02,  6.5251e-01,  1.8950e-02,  3.1674e-01,  2.5763e-01,\n",
       "           1.1699e-01, -6.5768e-02, -1.9745e-01, -2.3154e-01,  7.5988e-02,\n",
       "          -7.6941e-02,  1.0687e-01, -1.8182e-01, -2.2790e-01, -1.8508e-01,\n",
       "           3.1771e-01, -3.3097e-01, -6.0893e-01, -4.9260e-01, -1.6098e-01,\n",
       "          -1.6326e-01, -2.4804e-02, -8.2473e-03, -3.3693e-01, -3.1956e-01,\n",
       "          -6.7443e-01, -3.6717e-01,  8.1041e-01, -1.6831e-02, -3.3356e-01,\n",
       "          -1.6499e-01,  2.3208e-01,  3.1666e-01, -4.5959e-02,  3.7161e-01,\n",
       "           7.7250e-01, -4.7670e-02,  1.6786e-01, -5.1827e-01, -3.8195e-01,\n",
       "           4.8335e-02,  2.6658e-01, -2.7054e-01,  6.1879e-01,  4.8352e-02,\n",
       "           7.5627e-02, -6.6170e-02,  2.7430e-01, -3.2704e-01, -1.2324e-02,\n",
       "          -5.7573e-01,  3.7750e-01,  2.9539e-01, -3.9892e-01, -6.6567e-02,\n",
       "          -2.0834e-02, -1.2060e+00,  9.6964e-02, -8.3040e-01,  3.1236e-01,\n",
       "          -2.3110e-01,  1.7963e-01,  2.3389e-01,  2.9545e-01, -6.5320e-02,\n",
       "           2.5404e-01,  2.8725e-01,  2.5687e-01,  3.3609e-01, -3.4924e-02,\n",
       "           2.8019e-01,  2.3924e-01,  5.0790e-01,  4.1073e-01,  1.4769e-01,\n",
       "          -3.5440e-01,  2.9683e-01,  3.5781e-01,  9.0598e-02,  1.6954e-01,\n",
       "           9.9825e-02, -2.1636e-01,  2.1091e-01,  1.7501e-01, -1.6618e-01,\n",
       "          -2.2406e-01, -3.1100e-02, -4.6818e-01, -5.6655e-02, -4.8467e-01,\n",
       "           3.2557e-01, -5.1983e-01, -1.1912e-01,  3.4831e-01,  2.4340e-01,\n",
       "          -1.4896e-02,  1.8319e-01,  1.7196e-01, -2.4338e-01,  2.1735e-01,\n",
       "           3.7449e-01, -3.4618e-01, -3.9804e-02, -3.2197e-01, -2.9367e-01,\n",
       "           2.4357e-01, -5.6244e-01,  6.6635e-01,  4.0574e-01,  6.0165e-02,\n",
       "           1.5119e-01, -2.7920e-01, -5.1638e-02, -5.7408e-01,  3.5471e-01,\n",
       "          -3.2130e-01, -3.4955e-01, -5.4801e-01,  6.4487e-02,  4.2773e-01,\n",
       "           2.7565e-01, -6.4915e-02,  7.7868e-01,  4.2408e-01,  4.2813e-01,\n",
       "          -4.3003e-01, -1.5896e-02,  4.0377e-01, -4.1405e-02, -4.6059e-01,\n",
       "           6.7116e-02, -1.4962e-01, -5.1643e-01,  2.8000e-01, -2.6598e-01,\n",
       "          -1.2047e-01, -1.4913e-01,  3.3682e-01,  1.5365e-01, -2.1964e-01,\n",
       "          -3.8461e-01,  3.9587e-01, -1.4349e-01, -4.1723e-01,  4.7123e-01,\n",
       "          -3.5025e-01,  1.5209e-01,  1.0694e-01, -3.8441e-01, -6.2072e-01,\n",
       "          -3.2235e-01,  2.3090e-01,  1.5585e-01,  3.4799e-02, -6.6306e-01,\n",
       "          -4.0571e-01,  3.1197e-01,  2.4601e-01,  2.9976e-01,  2.9840e-01,\n",
       "           5.7427e-01, -3.0944e-01, -1.6702e-01,  1.4982e-01, -4.1377e-01,\n",
       "          -1.5704e-01, -3.3228e-02, -1.9995e-01,  2.9569e-01,  2.4768e-01,\n",
       "           1.1600e-01,  2.0367e-01,  4.3482e-01, -2.5272e-01, -3.8673e-01,\n",
       "          -7.1543e-02,  2.0161e-01, -9.3629e-02, -5.8306e-02,  2.3951e-01,\n",
       "          -6.3095e-02,  5.9650e-01,  7.2143e-01,  7.7439e-02, -6.4006e-01,\n",
       "           1.7696e-01,  5.2502e-01, -2.7253e-01, -4.0343e-02,  9.6365e-02,\n",
       "          -1.0344e+00, -2.1331e-01, -1.2883e-01, -2.0181e-01, -6.5352e-01,\n",
       "           8.8115e-02,  3.4863e-01,  4.6833e-03, -1.7864e-01, -3.0947e-01,\n",
       "           1.0448e-01,  4.8724e-02, -4.8980e-02,  4.3534e-01, -2.8901e-01,\n",
       "          -3.4797e-01, -2.3438e-01, -2.1532e-01, -5.4201e-01, -3.9968e-01,\n",
       "           3.5914e-01,  2.8129e-01,  1.1986e-01, -2.6486e-01, -5.1832e-01,\n",
       "           4.5577e-01, -6.8169e-01, -1.4632e-01,  2.4840e-01, -4.7702e-01,\n",
       "           1.4650e-02,  4.8990e-01,  1.4999e-01,  1.6336e-01,  6.5258e-02,\n",
       "           7.8557e-02, -3.3903e-01, -5.0030e-01, -1.5000e-01,  6.2323e-01,\n",
       "           3.2283e-01,  3.3506e-01, -1.7236e-01,  3.9096e-01,  4.5167e-01,\n",
       "          -2.9582e-01,  4.1670e-01,  1.0930e-02,  7.7439e-02, -6.0414e-02,\n",
       "           1.9048e-01,  3.4370e-01, -3.6842e-01,  2.0145e-01,  1.7929e-01,\n",
       "          -3.1634e-01, -3.2149e-01,  1.5691e-02,  2.0642e-01, -1.8580e-01,\n",
       "          -3.0875e-01, -2.0361e-01,  9.9698e-02, -3.4856e-01, -4.3104e-01,\n",
       "           2.9161e-01, -3.6543e-02,  7.0495e-02, -4.0357e-01, -4.3462e-01,\n",
       "          -2.0181e-01, -2.6055e-01,  6.8681e-01,  2.1064e-01, -1.9621e-01,\n",
       "           1.8720e-01, -1.0907e-01, -3.6330e-01,  4.7847e-01,  2.0162e-02,\n",
       "          -4.8754e-01,  4.5482e-01, -3.4822e-01,  1.0627e-01,  5.5756e-01,\n",
       "          -1.6769e-01, -1.0840e-01,  1.1008e-01,  1.0820e-01,  2.9026e-01,\n",
       "          -3.9884e-01,  1.5516e-01, -6.3476e-02, -3.1885e-01,  3.1820e-01,\n",
       "           2.6411e-01,  2.2781e-01, -1.7137e-01, -3.3426e-02,  3.5677e-02,\n",
       "           2.8782e-01, -6.0853e-02, -3.9888e-01, -4.7347e-01, -4.6132e-01,\n",
       "           1.2832e-01,  1.8927e-01,  5.6207e-01,  3.4144e-01, -1.1394e+01,\n",
       "           1.2938e-01,  7.8031e-01, -4.0713e-01, -8.2819e-02, -5.0320e-02,\n",
       "           1.0976e-02, -1.3708e-01,  4.2955e-01, -2.7734e-01, -4.4105e-01,\n",
       "          -3.5727e-01, -1.3308e-01,  1.5454e-01, -3.2189e-01, -2.6945e-01,\n",
       "          -1.4686e-02, -7.0935e-01,  4.0963e-01, -3.0286e-01,  5.0298e-01,\n",
       "          -2.3511e-01, -8.3798e-02,  1.5816e-01, -3.5851e-01, -1.3439e-02,\n",
       "           5.9766e-02, -1.6033e-01,  1.4651e-01,  4.2486e-01, -5.2072e-02,\n",
       "           3.3244e-02, -4.1258e-03,  4.8461e-02, -2.5723e-01, -5.9999e-01,\n",
       "           1.9296e-02, -1.4141e-01,  3.3422e-01, -4.4133e-02, -7.1265e-01,\n",
       "          -2.2533e-01, -2.7418e-01,  3.1407e-01,  2.7021e-01,  9.0598e-01,\n",
       "           2.2955e-01,  1.0686e+00,  1.5022e-01,  1.7748e-01,  2.6230e-02,\n",
       "          -7.8764e-01, -3.3269e-02, -4.1907e-01,  2.3280e-02,  1.0114e-01,\n",
       "           3.6576e-01, -7.9367e-01,  2.5251e-01, -3.5521e-01,  3.6487e-02,\n",
       "           4.9094e-01,  3.1768e-02,  2.7700e-01, -3.2298e-02, -7.8123e-02,\n",
       "           1.2089e-01, -2.8544e-02,  1.1098e-01, -2.2583e-01, -2.4084e-01,\n",
       "          -1.0177e-01,  5.9888e-01,  3.9592e-01, -7.2211e-01,  3.6112e-01,\n",
       "           4.9258e-02, -1.8622e-01,  1.4556e+00, -4.8573e-02,  1.8361e-02,\n",
       "           1.9800e-03,  2.9604e-01, -5.0236e-01, -7.3516e-01,  3.6511e-01,\n",
       "          -1.0211e-01,  5.6427e-02,  1.0127e-01,  4.8148e-01, -9.7481e-02,\n",
       "          -4.7940e-01, -1.8593e-03,  1.0556e-01, -1.3380e-01,  5.4710e-01,\n",
       "          -7.2824e-01, -1.6963e-01, -1.1223e-01,  3.1723e-01, -8.4054e-01,\n",
       "          -6.1126e-02, -7.0210e-01, -3.9328e-01,  5.0586e-01, -2.5896e-01,\n",
       "           1.5852e-01, -2.7443e-02, -4.3439e-02,  5.5574e-01, -2.1489e-01,\n",
       "           7.1555e-02,  2.2813e-01,  8.4995e-02, -1.6733e-01,  3.9349e-02,\n",
       "           4.1630e-01,  4.9571e-01, -4.2740e-01, -6.3911e-01, -7.6901e-03,\n",
       "           1.7180e-01,  1.4933e-01,  3.4019e-01, -5.3463e-01,  3.6431e-01,\n",
       "          -7.9162e-01, -1.8769e-01,  7.2949e-02, -4.3273e-02,  3.6175e-01,\n",
       "           2.3925e-01,  7.2038e-02,  2.2256e-01, -7.7756e-01, -6.4498e-01,\n",
       "          -6.5932e-01, -2.1445e-01,  2.6143e-01, -5.4236e-01, -8.3654e-02,\n",
       "          -1.1033e-01, -4.8837e-03, -5.0292e-02, -6.6102e-02,  1.1939e-02,\n",
       "           1.3632e-01,  2.3587e-01, -1.3023e-01, -1.3178e-01, -9.2464e-02,\n",
       "          -5.5566e-01, -1.1753e-01, -9.9591e-02, -1.2505e-01, -5.6783e-01,\n",
       "          -1.3797e-01,  3.3075e-02, -2.5373e-01, -6.8124e-02,  3.5608e-01,\n",
       "          -2.9832e-02, -2.7856e-02, -4.8850e-01,  7.2846e-01,  4.0279e-01,\n",
       "           5.2088e-02, -2.8298e-01, -5.5189e-01, -1.0567e-01,  2.9592e-01,\n",
       "           1.9555e-01, -6.7007e-02, -1.7709e-01, -2.4145e-01, -4.2522e-01,\n",
       "           1.4302e-01,  7.1147e-01, -6.6296e-02, -6.8115e-01,  2.6621e-01,\n",
       "           5.8223e-01, -1.2882e-01, -2.4432e-01,  2.0983e-02, -9.3284e-02,\n",
       "          -3.1851e-01, -6.0317e-01,  7.8821e-01, -1.1246e-01, -5.8638e-02,\n",
       "           2.2427e-01,  2.0359e-01, -9.2588e-03,  4.0209e-01, -1.5770e-01,\n",
       "          -3.5434e-01, -1.6477e-03, -7.3600e-01,  4.3257e-01,  6.1666e-02,\n",
       "          -3.1489e-01,  7.6082e-01,  2.7260e-01, -6.0593e-02, -4.5779e-02,\n",
       "           2.2863e-01,  7.9941e-01, -2.4636e-01,  6.4653e-02,  5.3779e-01,\n",
       "           8.8110e-02, -1.1229e-01,  3.9580e-02,  4.1665e-02,  4.5410e-02,\n",
       "           3.7012e-01, -3.2145e-02, -4.1564e-01, -6.1194e-02,  3.8924e-01,\n",
       "           3.9295e-01, -5.4610e-01, -7.3487e-01,  2.2333e-02,  6.8551e-01,\n",
       "          -2.9054e-01,  2.3654e-01, -3.2708e-02,  3.0880e-01,  1.2001e-01,\n",
       "          -9.2160e-01, -1.5016e-01, -1.0056e-01,  2.8229e-01, -2.5482e-02,\n",
       "          -1.6139e-01, -1.4727e-01, -2.3686e-01, -1.6208e-01,  6.8308e-01,\n",
       "           2.1991e-01, -6.9160e-01,  3.4058e-02,  5.7281e-01, -6.0551e-01,\n",
       "           4.6052e-01, -2.4577e-01, -3.4302e-02,  3.6435e-01,  1.3117e-01,\n",
       "           4.9067e-02,  1.7856e-01, -4.2138e-01, -4.0598e-01,  1.7824e-01,\n",
       "           9.1672e-01, -7.6613e-02,  1.1166e-01, -3.9384e-01, -1.7182e-01,\n",
       "          -7.1855e-02, -3.1913e-01, -3.1286e-01,  3.7185e-01,  2.0529e-01,\n",
       "          -3.2552e-02, -1.3271e-01, -2.2992e-01,  3.0816e-01,  1.1821e-01,\n",
       "           3.2536e-01,  5.8187e-01,  6.4368e-01, -1.0077e-02, -2.4936e-01,\n",
       "          -6.1235e-02,  1.4568e-01, -1.9524e-01,  5.2704e-01, -2.0627e-01,\n",
       "          -3.3453e-01, -3.8069e-03,  3.0134e-01, -6.1681e-01, -5.8725e-02,\n",
       "          -4.5426e-01,  4.2361e-01, -1.6955e-01,  3.7868e-02,  2.8833e-01,\n",
       "           4.0522e-02,  5.7925e-02, -2.2651e-01,  5.5807e-01,  4.4511e-01,\n",
       "           1.6609e-01,  4.5662e-01,  1.6104e-01,  4.5272e-02, -5.1597e-01,\n",
       "           2.1978e-01, -1.0213e+00, -9.4260e-02, -4.6640e-02, -1.5401e-01,\n",
       "          -4.4224e-01, -2.8126e-02,  1.6814e-01,  1.8538e-01,  6.9597e-01,\n",
       "           7.2967e-01, -3.0871e-01, -4.5290e-01, -1.8510e-01,  1.5304e-01,\n",
       "           2.7705e-01, -1.9517e-01,  3.5971e-01,  9.6103e-02,  2.4432e-01,\n",
       "          -5.0967e-01, -4.3165e-01, -4.2252e-02,  3.1411e-01, -2.0171e-01,\n",
       "           5.7118e-01, -2.0164e-01, -1.3863e-01,  9.0832e-02,  1.3770e-01,\n",
       "           1.6639e-01,  7.9664e-01,  4.3416e-01, -6.3425e-01,  8.7469e-02,\n",
       "          -1.3980e-02, -2.9988e-01,  2.5983e-02, -4.3577e-01,  2.8858e-01,\n",
       "           1.2984e-01,  4.9280e-01,  6.7504e-01]], device='mps:0'),\n",
       " tensor([[-6.1750e-01, -4.2829e-01, -2.3934e-01, -9.4116e-01, -3.4548e-01,\n",
       "          -6.5496e-02, -6.3874e-01,  9.7828e-01, -9.5892e-02,  4.0096e-01,\n",
       "           1.4919e-01, -5.1603e-03,  2.0980e-01,  1.3680e-01, -6.9506e-01,\n",
       "           1.6740e-01,  4.6094e-01,  2.1159e-01,  3.5951e-01, -4.4329e-02,\n",
       "           7.9832e-01,  3.4245e-01,  1.8019e-01, -2.5205e-01, -4.5876e-01,\n",
       "           5.8994e-01, -2.4628e-01, -9.2014e-02,  2.0509e-01, -8.1949e-01,\n",
       "          -1.3586e-01, -4.9757e-02, -3.6281e-01,  8.7602e-01,  1.1948e-01,\n",
       "          -2.7310e-01, -1.5276e-01, -2.7996e-01,  2.3494e-01, -6.3637e-02,\n",
       "           3.7814e-01,  2.6592e-02, -2.9347e-01,  3.1386e-01, -1.6421e-01,\n",
       "          -1.4057e-01,  1.4047e-02, -7.8429e-01,  1.3666e-02,  1.8955e-01,\n",
       "           5.8851e-02, -4.7205e-02,  1.9231e-01,  3.7030e-01, -1.0617e+00,\n",
       "          -2.6348e-01,  2.7603e-01,  2.9849e-01, -1.7932e-01, -5.7784e-02,\n",
       "          -1.9989e-01,  9.0201e-01,  1.7949e-01, -2.7822e-01,  1.7824e-01,\n",
       "          -4.4899e-01, -5.4601e-01, -6.0781e-01, -1.0682e-01, -1.1967e-01,\n",
       "          -1.0002e-01,  9.7528e-02,  1.4009e-01, -2.9385e-01,  2.3170e-01,\n",
       "           2.1813e-01,  4.7024e-02, -2.7652e-02, -4.3158e-01, -2.6546e-01,\n",
       "           3.6947e-01,  3.4425e-02, -1.4730e-02,  2.0326e-01, -5.8744e-01,\n",
       "          -2.0226e-01,  3.7959e-01,  1.5703e-01,  2.5914e-01,  2.2662e-03,\n",
       "           5.9967e-01,  1.2024e-02,  8.0339e-02, -4.5369e-01,  5.8427e-01,\n",
       "          -5.4955e-01,  1.0329e-01, -1.6007e-01,  4.9532e-02, -5.2088e-01,\n",
       "          -9.8735e-02,  2.4494e-01, -1.8085e-01,  1.6053e-01, -4.6592e-01,\n",
       "          -3.2983e-02,  2.4514e-01,  3.0468e-01, -4.9183e-01,  4.2685e-02,\n",
       "          -6.5134e-03, -2.9791e-01,  7.3694e-02, -6.6407e-01, -8.5795e-01,\n",
       "           5.8354e-01,  1.8428e-02, -4.5240e-01, -2.3930e-01,  7.4443e-01,\n",
       "           2.9738e-01, -2.3201e-01, -5.3072e-02, -3.5548e-01,  2.0259e-01,\n",
       "          -1.8518e-01,  2.5689e-01, -2.7018e-01,  3.8439e-01,  5.5026e-01,\n",
       "           3.1581e-01,  2.8489e-01, -1.8327e-01,  2.7637e-01,  1.4465e-01,\n",
       "           3.6947e-02, -6.6614e-02, -1.5929e-01,  1.5180e-01,  1.2041e-01,\n",
       "          -2.9580e-01, -3.2999e-01,  5.4746e-03,  1.1182e-01, -3.3279e-01,\n",
       "          -1.0202e-01,  1.8680e-01,  3.4808e-01, -1.4092e-01, -2.5403e-01,\n",
       "          -7.8984e-01, -4.2328e-01,  1.0217e-01, -1.8560e-01, -1.7135e-01,\n",
       "          -4.1911e-01,  3.8121e-01, -6.6660e-02,  4.7638e-01,  3.4189e-01,\n",
       "           6.9978e-01, -1.5311e-01,  5.9776e-02, -5.6412e-01,  3.5197e-01,\n",
       "           3.8270e-02, -6.1110e-01, -1.7247e-01,  1.5680e-01, -4.1813e-04,\n",
       "           6.8195e-01,  7.0979e-02,  4.0124e-01,  1.2070e-01, -3.2553e-01,\n",
       "           8.2089e-03,  1.9890e-01, -1.2864e-01, -1.4297e-01,  7.3766e-01,\n",
       "          -5.5188e-01, -2.3029e+00,  1.0721e-01, -3.5844e-01, -3.6529e-01,\n",
       "          -1.8135e-01,  3.7255e-01,  3.3772e-02,  4.9966e-01,  1.5486e-01,\n",
       "           3.7598e-02,  1.6729e-01, -2.8025e-02,  3.6851e-01, -8.4399e-02,\n",
       "           2.6751e-01,  3.3341e-01,  1.2988e-01, -2.3298e-01, -1.0956e-01,\n",
       "          -3.9227e-01,  2.4031e-01, -9.8916e-02,  8.6010e-01,  2.3260e-01,\n",
       "           2.0943e-01, -4.3657e-03, -4.7262e-02,  4.8964e-01,  4.3099e-01,\n",
       "          -6.6541e-02, -1.6585e-01, -4.9522e-01,  4.6269e-03, -4.1279e-02,\n",
       "          -1.2963e-02, -4.5611e-01,  1.3753e-01,  4.2841e-01, -1.2416e-01,\n",
       "          -1.8354e-01,  1.5525e-01, -4.7715e-02, -3.1517e-01, -2.6492e-02,\n",
       "           6.8453e-01,  1.9080e-01, -2.2994e-01,  1.2377e-01, -1.1896e-01,\n",
       "           3.7006e-02, -4.6975e-01,  7.0044e-01,  6.1147e-01,  1.5585e-01,\n",
       "           1.6214e-02, -6.9943e-01,  2.3699e-01, -3.1574e-01, -4.7647e-03,\n",
       "          -1.5027e-01, -1.5731e-01, -4.2986e-01, -5.7285e-02, -5.5854e-02,\n",
       "           4.0115e-01,  7.9943e-02,  4.6653e-01,  1.1837e+00,  2.7973e-01,\n",
       "           1.9184e-02,  5.6097e-01,  2.0801e-01,  4.2492e-01, -4.0180e-01,\n",
       "           3.3082e-01,  8.8825e-02, -6.3896e-01,  2.7188e-01, -4.1593e-01,\n",
       "           4.4963e-02, -4.1432e-02, -3.7926e-01,  1.4208e-01, -2.1276e-02,\n",
       "           3.4237e-01,  3.3517e-01,  1.5186e-01,  1.6838e-01,  2.9411e-01,\n",
       "          -3.1322e-01,  5.9715e-01, -6.0809e-02, -1.1127e-01, -6.3492e-01,\n",
       "          -7.6732e-01,  9.5206e-02,  4.4743e-01, -1.2880e-01, -6.1279e-01,\n",
       "           1.0462e-01, -2.3866e-01, -1.2257e-01, -3.5064e-02,  2.1191e-01,\n",
       "           2.7455e-01,  5.0687e-02, -9.5436e-02,  6.4791e-02, -4.6950e-01,\n",
       "          -4.9277e-01,  4.9140e-02, -1.5758e-01, -1.3063e-01,  5.3918e-01,\n",
       "          -7.9415e-02,  1.1575e-01,  5.7221e-01,  1.9799e-02, -4.5976e-01,\n",
       "          -2.5640e-01,  1.4802e-01,  2.8667e-01, -7.2082e-02,  2.1981e-01,\n",
       "           3.4119e-01,  2.6467e-02,  5.1649e-01,  1.4823e-01, -2.1046e-01,\n",
       "          -2.7239e-01,  5.1173e-01, -1.5755e-01,  6.1403e-02, -1.2527e-01,\n",
       "          -8.7207e-01, -5.6150e-02, -7.8413e-02, -1.6833e-01, -5.4440e-01,\n",
       "           2.6566e-01,  1.4351e-01,  1.0561e-01,  2.2350e-01, -1.5984e-01,\n",
       "           7.0123e-01,  6.5285e-01, -3.3100e-01, -1.6943e-01, -4.0104e-01,\n",
       "           9.3922e-02, -6.8374e-02, -2.9329e-01, -1.4009e-02,  9.5368e-03,\n",
       "          -2.2765e-01, -1.2516e-01, -1.0805e-01,  2.8921e-01, -1.8876e-01,\n",
       "           5.2862e-01,  1.2189e-01,  8.3649e-02,  4.3873e-01,  2.0612e-02,\n",
       "           1.3071e-01,  3.6515e-01, -4.2117e-01,  1.2870e-01,  4.0424e-01,\n",
       "           4.0578e-01, -1.0551e+00, -8.3291e-01, -1.4936e-01,  3.6111e-01,\n",
       "           2.2860e-01,  2.5189e-01,  1.6999e-02, -7.2215e-03,  8.9455e-02,\n",
       "          -2.3402e-01,  3.5135e-01, -6.6415e-02, -5.7412e-01,  3.0561e-01,\n",
       "           2.3567e-01,  5.3263e-01, -2.7377e-01, -5.0579e-01, -6.5233e-02,\n",
       "           2.4616e-01, -3.6381e-01,  1.1392e-01,  2.3480e-01,  4.5466e-01,\n",
       "          -3.6015e-01, -5.6164e-01, -7.0282e-02, -7.3986e-01, -1.3578e-01,\n",
       "           3.2483e-01, -2.1437e-01,  2.7178e-02, -2.4758e-01,  1.5231e-01,\n",
       "          -1.7925e-02, -1.8879e-01,  6.0757e-01,  2.0522e-01, -2.5855e-01,\n",
       "           1.1981e-01, -2.3892e-01, -1.9475e-01,  5.0091e-01, -5.0226e-01,\n",
       "          -3.0348e-01, -5.0960e-02, -1.5349e-01, -3.2418e-01,  3.2181e-01,\n",
       "          -4.1389e-01, -1.4861e-01,  1.6748e-01, -6.4736e-02, -1.3068e-01,\n",
       "          -3.5233e-01,  1.4268e-01, -1.2707e-01, -4.9883e-02, -1.7654e-02,\n",
       "          -1.3353e-01,  6.0900e-02,  1.3308e-01,  6.4538e-02,  1.4388e-01,\n",
       "           5.3696e-01, -1.1464e-01,  2.7582e-03, -2.5900e-01, -2.2469e-01,\n",
       "           5.2498e-01, -9.4973e-02, -3.3117e-01,  5.4928e-01, -1.1156e+01,\n",
       "           1.0386e-01,  7.2631e-01, -5.5808e-01,  5.5128e-02, -2.8539e-01,\n",
       "           2.1085e-01, -3.0009e-01,  7.9101e-02,  2.4967e-01, -2.8460e-01,\n",
       "          -1.2277e-01, -2.7364e-01, -2.3193e-01, -4.0872e-01, -3.9742e-01,\n",
       "          -4.7229e-01, -5.6789e-01,  1.1772e-01, -1.7663e-01, -1.0996e-01,\n",
       "          -5.7036e-01, -1.9379e-01,  2.9696e-01, -4.9559e-03, -5.1069e-01,\n",
       "           5.9541e-02, -2.2948e-01,  2.6399e-01,  4.6126e-02,  2.2045e-01,\n",
       "          -1.9949e-02,  1.4108e-01,  2.0669e-01, -4.4946e-02, -4.5100e-01,\n",
       "          -4.4853e-01,  2.2129e-01, -1.5955e-01, -2.2267e-01, -5.4017e-02,\n",
       "          -3.3838e-01, -2.7689e-01,  1.3447e-01,  3.7594e-01,  6.2596e-01,\n",
       "           1.1595e+00,  5.6208e-01,  2.9245e-01, -1.0625e-01, -8.5438e-02,\n",
       "          -1.0762e+00,  5.5900e-01, -3.7318e-01,  8.1787e-02, -2.5119e-01,\n",
       "           6.6201e-01, -4.9906e-01,  5.4685e-01, -4.0754e-01, -2.5500e-01,\n",
       "           3.7179e-01,  3.0194e-01,  7.1104e-03,  2.8239e-02, -2.7432e-01,\n",
       "           4.0257e-01, -2.6094e-01, -3.6915e-02, -3.7819e-01,  4.0081e-01,\n",
       "          -3.9472e-01,  5.0770e-01, -5.6534e-02,  1.2945e-01,  1.8567e-01,\n",
       "          -3.5001e-01,  1.6610e-01,  1.0337e+00,  1.7414e-01,  4.1071e-01,\n",
       "           1.7420e-01,  5.1160e-01,  5.3064e-02, -8.8958e-01,  1.3691e-01,\n",
       "           2.0502e-01,  2.0789e-01, -6.7096e-02,  5.7131e-01, -2.5517e-02,\n",
       "          -6.5181e-01, -2.3094e-01, -3.2169e-01,  4.3108e-01,  6.5352e-01,\n",
       "          -2.2597e-01, -4.4343e-01,  2.3894e-01, -7.9776e-03, -5.3395e-01,\n",
       "           9.0074e-02, -2.2320e-01, -5.2792e-01,  8.7275e-02, -5.8914e-01,\n",
       "           1.4985e-01, -1.2428e-01,  5.4238e-02,  6.6887e-01, -4.0159e-01,\n",
       "          -1.6113e-01,  3.6602e-02, -4.4532e-02, -4.1270e-01, -4.9520e-01,\n",
       "          -7.7083e-02,  4.1073e-01, -1.9776e-01, -2.7976e-01,  7.0636e-02,\n",
       "          -3.0673e-01,  2.2584e-01, -1.7377e-01, -6.4935e-01,  3.2879e-01,\n",
       "          -6.0415e-01,  2.0578e-01,  2.5360e-01, -2.2507e-01,  4.9017e-01,\n",
       "           1.1337e-01, -8.1817e-02,  1.9397e-01, -9.9369e-02, -7.8554e-01,\n",
       "          -3.2573e-01, -2.7299e-02,  3.2662e-01, -4.2844e-01,  1.5776e-01,\n",
       "           1.5022e-01, -4.5752e-02, -2.2574e-01,  2.7904e-01,  1.0174e-01,\n",
       "           3.7762e-01,  3.9846e-01, -5.5006e-01,  1.1727e-01, -2.9867e-01,\n",
       "          -2.9262e-02, -6.1174e-02,  1.7952e-01,  2.3654e-01, -7.0764e-01,\n",
       "          -1.5224e-01, -2.6434e-01, -3.0965e-01, -2.1650e-01, -2.5120e-01,\n",
       "           2.0680e-01,  1.4038e-01, -1.5897e-02, -1.1308e-01,  5.8077e-01,\n",
       "           4.0814e-02, -1.1167e-01, -4.5997e-01,  3.1443e-01,  4.3178e-01,\n",
       "           2.1794e-01,  7.5875e-02, -1.2830e-01, -5.5532e-01, -1.1922e-02,\n",
       "          -4.3644e-01,  9.1081e-01, -7.3906e-01, -7.1970e-01,  6.6086e-02,\n",
       "           1.4951e-02, -8.8259e-04, -2.3450e-01, -5.0265e-01,  8.1298e-02,\n",
       "          -2.1017e-01, -1.4891e-01,  6.1550e-03, -1.0607e-02,  1.6299e-01,\n",
       "           4.0742e-01,  3.6838e-01,  1.5951e-01, -3.3938e-01, -4.4081e-01,\n",
       "          -6.0244e-01,  1.4567e-02, -1.6795e-01,  2.9330e-01, -2.6752e-01,\n",
       "          -3.2661e-01,  1.6043e-02, -9.1892e-03, -2.2272e-01,  5.2876e-01,\n",
       "           1.2427e-01,  1.6396e+00, -3.7287e-01, -4.4551e-01,  2.9018e-01,\n",
       "           7.2306e-02, -2.7073e-02, -2.8130e-01,  2.5074e-01,  9.5447e-02,\n",
       "           4.0194e-02, -3.6304e-01,  1.5023e-02, -2.0558e-02,  3.2773e-01,\n",
       "           3.2056e-01, -1.0703e+00, -2.0238e-01, -1.7993e-01,  5.5664e-01,\n",
       "           3.2388e-02,  6.3185e-01, -2.2305e-01,  3.0577e-01,  4.6106e-01,\n",
       "          -8.6667e-01, -2.3000e-01, -1.6129e-01,  2.7295e-01, -1.0583e-01,\n",
       "          -9.8735e-02, -1.6178e-01, -3.7165e-02, -1.6936e-01,  3.9538e-01,\n",
       "           7.4751e-01, -7.8426e-01,  7.4081e-01,  7.0467e-01, -9.8234e-02,\n",
       "           5.6815e-01,  9.4187e-02,  3.2641e-01, -1.4659e-01,  2.2669e-02,\n",
       "           2.6200e-01,  1.8542e-01, -6.3165e-01, -2.3962e-01,  2.4473e-01,\n",
       "           4.0158e-01, -1.1943e-01,  1.2108e-01,  7.9292e-02,  3.0960e-02,\n",
       "          -3.7691e-01, -1.0145e-01, -2.4552e-01, -3.9904e-01,  1.1356e-01,\n",
       "           1.4603e-02, -1.5001e-01, -8.6997e-02,  4.7709e-01, -1.7580e-01,\n",
       "           7.5067e-01,  4.2910e-01,  3.7145e-01, -3.5103e-01,  8.9950e-04,\n",
       "          -5.7565e-02,  6.0671e-01, -3.1670e-01,  7.7385e-01,  6.3337e-02,\n",
       "          -2.3140e-01,  2.7251e-01,  6.0547e-03,  1.7829e-01,  2.9892e-01,\n",
       "          -2.3519e-01,  1.2702e-01, -3.0718e-01, -2.3711e-01,  2.8330e-01,\n",
       "          -1.8640e-01,  1.1542e-01, -7.2270e-01, -6.3602e-03,  3.9857e-01,\n",
       "          -3.2051e-01,  4.9873e-01, -2.3074e-01,  6.1026e-01, -4.3509e-01,\n",
       "           3.2669e-02, -4.7470e-01, -4.3275e-01, -2.8508e-01,  5.1854e-02,\n",
       "          -8.6027e-01, -1.9453e-01,  1.7684e-01,  3.7405e-02,  2.1707e-01,\n",
       "           7.5155e-01, -3.3794e-01, -2.4771e-01, -2.0386e-01,  4.0190e-02,\n",
       "           1.8299e-01, -8.2071e-02, -3.5586e-01,  2.1161e-01, -3.5607e-01,\n",
       "          -1.0520e+00, -2.3117e-01,  8.9990e-02,  3.8419e-01, -3.5853e-01,\n",
       "           1.1570e+00,  2.5132e-02, -2.0226e-01,  6.7253e-01,  2.0808e-01,\n",
       "           6.4157e-01,  6.0323e-01,  1.3628e-03, -4.1259e-01, -5.4109e-02,\n",
       "          -1.5877e-01, -3.2881e-01,  1.5757e-01, -1.8346e-01,  5.0556e-01,\n",
       "           5.4802e-01, -2.0806e-02,  5.4756e-01]], device='mps:0'),\n",
       " tensor([[-0.2744, -0.4852, -0.4880,  ...,  0.2064,  0.2977,  0.6606],\n",
       "         [-0.2434, -0.3725, -0.3338,  ...,  0.2009,  0.3829,  0.4365],\n",
       "         [ 0.0506, -0.6230, -0.6446,  ..., -0.0875,  0.4018,  0.3101],\n",
       "         ...,\n",
       "         [-0.2434, -0.3725, -0.3338,  ...,  0.2009,  0.3829,  0.4365],\n",
       "         [ 0.0506, -0.6230, -0.6446,  ..., -0.0875,  0.4018,  0.3101],\n",
       "         [-0.3784, -0.2626, -0.4447,  ...,  0.0806,  1.1661,  0.3650]],\n",
       "        device='mps:0'),\n",
       " tensor([[-0.0548, -0.3751, -0.4766,  ...,  0.0464,  0.8423,  0.4111],\n",
       "         [-0.0583, -0.6757, -0.8037,  ...,  0.1694,  0.7852,  0.5016]],\n",
       "        device='mps:0'),\n",
       " tensor([[-0.0111, -0.5057, -0.4670,  ...,  0.2365, -0.0712,  0.4388],\n",
       "         [-0.3292, -0.2590, -0.5016,  ..., -0.0148,  0.3509,  0.1607]],\n",
       "        device='mps:0'),\n",
       " tensor([[-0.0111, -0.5057, -0.4670,  ...,  0.2365, -0.0712,  0.4388],\n",
       "         [-0.3292, -0.2590, -0.5016,  ..., -0.0148,  0.3509,  0.1607]],\n",
       "        device='mps:0'),\n",
       " tensor([[-0.3011, -0.2392, -0.3565,  ...,  0.1903,  0.6847,  0.4711],\n",
       "         [-0.4278, -0.5156, -0.8224,  ..., -0.1083,  0.4120,  0.3513]],\n",
       "        device='mps:0'),\n",
       " tensor([[-2.6669e-01, -4.3706e-01, -4.7596e-01, -6.7304e-01, -1.3138e-01,\n",
       "           6.4374e-04, -6.0597e-01,  5.9090e-01,  6.7669e-03,  6.4348e-01,\n",
       "           1.3153e-01,  2.7624e-01,  1.3447e-02, -2.7649e-01, -5.7975e-01,\n",
       "          -5.0599e-01,  2.4398e-01,  5.8716e-01,  7.5801e-01,  3.8037e-01,\n",
       "           2.3905e-01,  6.1888e-01, -2.5811e-01, -1.7227e-01, -9.2400e-01,\n",
       "           3.5704e-01, -1.1709e-01,  3.4438e-01,  2.4135e-01, -5.9088e-01,\n",
       "           4.9509e-04,  3.3321e-01, -2.1608e-02,  1.2997e+00, -5.0299e-01,\n",
       "          -2.6962e-01,  1.0103e-01, -7.9520e-01, -1.3835e-01, -7.2011e-02,\n",
       "           7.8072e-01, -2.2671e-01, -5.0898e-02,  2.4541e-01,  9.0136e-02,\n",
       "          -6.5820e-02,  1.4531e-01, -3.5947e-01, -1.6407e-01, -6.3451e-01,\n",
       "           8.1026e-02,  2.8061e-02,  5.0322e-01,  6.3953e-01, -6.0556e-01,\n",
       "          -3.1544e-01,  4.4217e-02,  5.1450e-03, -1.3031e-01,  1.6744e-01,\n",
       "          -1.5517e-01,  2.2217e-01, -3.8968e-02,  1.1978e-01,  2.4615e-01,\n",
       "           2.1713e-01, -2.0339e-01, -4.9474e-01, -4.5195e-01, -3.0046e-03,\n",
       "           2.9934e-02,  2.4908e-01,  1.0440e-01, -3.2297e-01,  4.5519e-01,\n",
       "          -5.1012e-01, -7.7386e-02, -8.9061e-02, -4.8942e-01,  6.7425e-02,\n",
       "          -4.9286e-02,  3.0154e-01,  1.3302e-01, -1.6004e-01,  2.2767e-01,\n",
       "           1.7675e-01,  4.7197e-01,  3.8563e-02,  1.1828e-02,  5.7524e-02,\n",
       "           9.6185e-02, -2.4340e-01, -5.5609e-01, -1.9307e-01,  2.3140e-01,\n",
       "          -1.4339e-01,  1.6008e-01, -3.7226e-01, -1.8733e-02, -2.6769e-01,\n",
       "           1.1829e-02,  1.4753e-01, -1.9106e-01,  9.5430e-02, -4.1773e-01,\n",
       "          -5.4564e-01,  1.4099e-01, -1.5372e-01, -7.1785e-01,  2.0021e-01,\n",
       "          -1.0204e-01, -1.5055e-02,  1.0202e-01, -2.1675e-01, -3.1958e-01,\n",
       "          -2.0219e-01,  2.5920e-01, -1.1339e-02, -2.2266e-01,  3.2632e-01,\n",
       "           3.5906e-01, -3.0213e-01,  1.1375e-01,  9.2287e-02,  4.5701e-02,\n",
       "          -2.1335e-01,  6.0362e-01,  2.4028e-02,  4.3650e-01,  1.7077e-01,\n",
       "           1.0153e-01,  7.7920e-02, -4.5629e-01, -2.6933e-01,  1.2104e-01,\n",
       "           1.3717e-01,  2.6540e-01, -4.9232e-01,  2.4692e-02,  1.4262e-01,\n",
       "           1.2387e-01, -2.1885e-01, -5.6537e-01, -2.1133e-01, -1.1971e-01,\n",
       "          -2.5224e-01,  1.1765e-02, -2.2651e-01, -1.1036e-01, -2.6001e-01,\n",
       "          -6.5849e-01, -6.2506e-01,  5.7964e-01, -1.9432e-01, -1.3871e-01,\n",
       "          -2.2300e-01,  3.5561e-01,  4.9680e-02, -7.4630e-02,  3.6573e-01,\n",
       "           7.6619e-01,  5.6569e-02,  4.3075e-02, -4.2795e-01, -3.5467e-01,\n",
       "           2.2589e-01,  2.4553e-01, -3.5334e-01,  1.3043e-01,  2.0456e-01,\n",
       "           1.5307e-01, -1.0832e-02,  2.7463e-01, -2.2198e-01,  2.4185e-01,\n",
       "          -6.6922e-01,  6.6483e-01,  2.0678e-01, -2.7953e-01,  5.1875e-02,\n",
       "          -1.0417e-01, -1.2402e+00,  1.6376e-01, -5.8857e-01,  1.7824e-01,\n",
       "          -2.0540e-01,  3.5234e-01,  3.8504e-01,  2.1492e-01, -3.8152e-02,\n",
       "           2.7009e-01,  2.5973e-01,  1.5486e-01,  2.7364e-01, -5.6590e-02,\n",
       "           5.4587e-01,  4.4938e-01,  4.5008e-01,  2.7313e-01,  1.2965e-01,\n",
       "          -5.3941e-01,  3.9300e-02,  4.9174e-01, -6.9616e-02,  3.6034e-01,\n",
       "           1.5707e-01, -3.9769e-01,  6.1221e-02,  3.8432e-02, -4.7404e-02,\n",
       "          -2.0827e-01, -1.9611e-01, -5.9737e-01, -2.0864e-01, -3.5310e-01,\n",
       "           2.7306e-01, -3.6915e-01, -2.3400e-01,  4.6606e-01,  1.4711e-01,\n",
       "          -2.4735e-02,  2.2147e-01,  3.1945e-02, -3.3354e-01,  1.0792e-01,\n",
       "           2.8013e-01, -2.4420e-01, -1.5576e-02, -4.3904e-01, -1.5632e-01,\n",
       "           3.3399e-01, -4.3383e-01,  6.1239e-01,  2.8382e-01, -3.3190e-02,\n",
       "           2.0303e-02, -1.2578e-01,  2.8065e-02, -3.3142e-01,  3.7946e-01,\n",
       "          -3.2269e-01, -4.5179e-01, -4.3334e-01,  2.1128e-01,  1.2049e-01,\n",
       "           2.3977e-01, -2.4026e-01,  6.3917e-01,  5.0884e-01,  3.2966e-01,\n",
       "          -2.8186e-01, -6.5457e-02,  1.7808e-01,  9.9536e-03, -4.1537e-01,\n",
       "           8.7899e-02, -1.8099e-01, -6.1234e-01,  1.6878e-01, -4.1831e-01,\n",
       "           1.2814e-01, -5.9025e-02,  1.2498e-01,  1.6226e-01, -8.4770e-02,\n",
       "          -5.2078e-01,  4.5037e-01,  4.3352e-02, -3.8129e-01,  3.8771e-01,\n",
       "          -3.4068e-01,  1.8967e-01, -3.2728e-02, -4.5297e-01, -4.3795e-01,\n",
       "          -6.7658e-01,  1.5078e-01,  1.5339e-01, -1.0142e-03, -6.8293e-01,\n",
       "          -2.7571e-01,  4.1142e-01,  1.0819e-01,  8.2559e-02,  4.5421e-01,\n",
       "           5.0181e-01, -5.2925e-02, -5.9691e-02,  9.9037e-02, -4.2089e-01,\n",
       "          -3.2972e-01,  3.9187e-02, -2.7262e-01,  1.9374e-01,  3.2319e-01,\n",
       "           2.4645e-01,  1.0160e-01,  2.5225e-01, -1.3414e-01, -4.3414e-01,\n",
       "          -1.8845e-01,  1.9938e-01, -1.6081e-01, -8.2010e-02,  1.9281e-01,\n",
       "          -5.1205e-02,  6.0268e-01,  5.4264e-01, -1.6438e-01, -5.8651e-01,\n",
       "           2.0490e-01,  5.6551e-01, -1.8465e-01, -1.8872e-01,  1.0269e-01,\n",
       "          -9.6697e-01, -1.8496e-01,  6.4934e-02, -8.9887e-02, -5.7390e-01,\n",
       "           1.0109e-01,  3.4863e-01, -1.5122e-01, -1.4832e-01, -4.1673e-01,\n",
       "           3.4936e-01,  5.3502e-01, -2.7166e-02,  2.5459e-01, -3.1492e-01,\n",
       "          -3.9519e-01, -5.3522e-01, -8.9294e-02, -5.2208e-01, -5.0580e-01,\n",
       "           3.5540e-01,  2.7834e-01, -8.6759e-02, -1.2290e-01, -4.0563e-01,\n",
       "           3.5520e-01, -6.0802e-01, -8.6681e-02,  3.4036e-01, -4.8614e-01,\n",
       "          -8.3471e-02,  4.8270e-01, -2.6538e-01,  4.0032e-02,  2.1176e-01,\n",
       "           1.5432e-02, -2.6222e-01, -4.6079e-01, -8.0798e-02,  7.4098e-01,\n",
       "           3.6566e-01,  2.7895e-01, -2.5122e-01,  3.9871e-01,  4.0345e-01,\n",
       "          -3.0865e-01,  4.2468e-01, -8.6272e-02, -1.9507e-01, -4.0550e-01,\n",
       "           1.0274e-01,  2.9723e-01, -4.4318e-01,  1.3295e-01,  1.4606e-01,\n",
       "          -1.2874e-01, -1.7816e-01, -4.2518e-02,  1.0418e-01, -2.1360e-02,\n",
       "          -3.2653e-01, -1.2787e-01,  2.7417e-01, -2.3525e-01, -6.4287e-02,\n",
       "           7.0331e-02, -7.3733e-02,  2.2464e-01, -3.2135e-01, -4.2933e-01,\n",
       "          -3.0237e-01, -2.0227e-01,  8.8525e-01,  2.9675e-01, -2.5456e-01,\n",
       "           2.9469e-01, -2.3396e-01, -3.9108e-01,  2.4452e-01, -3.5424e-01,\n",
       "          -3.0656e-01,  4.2482e-01, -2.9462e-01, -1.0329e-01,  5.6171e-01,\n",
       "          -1.7964e-01, -1.4489e-01, -4.5861e-04,  1.5629e-01,  3.1616e-01,\n",
       "          -2.5958e-01,  1.3797e-01,  7.5066e-02, -2.9145e-01,  2.5187e-01,\n",
       "           3.7264e-01,  4.4885e-02, -2.0926e-01, -1.5341e-01, -3.0785e-01,\n",
       "           2.7671e-01, -1.4718e-01, -3.6931e-01, -5.0086e-01, -6.5042e-01,\n",
       "           3.4967e-01,  2.6955e-01,  3.8810e-01,  3.5468e-01, -1.1377e+01,\n",
       "           2.0013e-02,  5.0456e-01, -4.2194e-01,  2.0937e-03, -1.5111e-01,\n",
       "           1.2082e-01, -1.6535e-01,  4.3084e-01, -2.7223e-01, -3.2614e-01,\n",
       "          -6.9926e-01, -1.1167e-01,  1.2102e-01, -3.1484e-01, -1.9683e-01,\n",
       "          -1.1599e-01, -7.5337e-01,  3.1500e-01, -1.7982e-01,  1.7717e-01,\n",
       "          -2.7249e-01, -1.7803e-01,  1.5075e-01, -2.9849e-01, -2.8032e-02,\n",
       "           3.1434e-01, -1.1683e-01,  2.6480e-01,  3.9000e-01,  1.2891e-01,\n",
       "           6.1290e-02,  3.2118e-02,  5.6437e-02, -8.4044e-02, -5.0284e-01,\n",
       "          -2.5283e-01, -1.7709e-01, -2.0110e-02, -2.5312e-01, -5.8619e-01,\n",
       "          -3.7296e-01, -2.3237e-01,  3.9207e-01,  2.7948e-01,  9.5278e-01,\n",
       "           3.9367e-01,  8.1366e-01,  4.0400e-01,  1.3884e-01,  5.1258e-02,\n",
       "          -6.0169e-01,  2.3417e-02, -5.7934e-02, -4.7860e-01, -1.5314e-01,\n",
       "           3.9025e-01, -8.4697e-01,  4.0645e-01, -4.9992e-01, -1.6174e-01,\n",
       "           5.0095e-01,  1.5167e-01,  2.1177e-02, -1.5478e-01, -1.1835e-01,\n",
       "           2.9361e-02,  2.8784e-02,  9.4043e-02, -1.6661e-01, -2.7722e-01,\n",
       "          -1.2523e-01,  8.5651e-01, -1.5642e-03, -6.9428e-01, -8.4744e-02,\n",
       "           3.6425e-02, -2.6455e-01,  1.8203e+00,  4.2966e-02,  5.3375e-02,\n",
       "           1.0681e-01,  4.9972e-01, -8.7441e-01, -7.4027e-01,  1.6256e-01,\n",
       "          -2.2990e-01, -2.1354e-02,  2.7314e-01,  4.1352e-01, -1.5877e-01,\n",
       "          -7.9939e-01, -1.7606e-01,  1.3340e-01,  1.2208e-01,  8.1355e-01,\n",
       "          -6.7446e-01, -1.7792e-01,  1.4477e-01,  2.9987e-01, -7.1651e-01,\n",
       "           4.9153e-02, -4.7479e-01, -1.7856e-01,  4.4306e-01, -3.2465e-01,\n",
       "           4.9508e-02,  2.0475e-02, -2.0761e-01,  4.4174e-01, -3.3644e-01,\n",
       "           9.1400e-02, -6.2004e-02,  1.0473e-01, -1.6457e-01,  1.0357e-01,\n",
       "           5.4552e-01,  4.2707e-01, -3.5946e-01, -3.3354e-01, -1.5038e-01,\n",
       "           3.0716e-01,  1.8745e-01,  1.7077e-01, -2.5328e-01,  5.3390e-01,\n",
       "          -5.5036e-01, -1.4622e-01,  3.2583e-01,  4.8665e-02,  2.0818e-01,\n",
       "           1.5231e-01, -9.6376e-02,  1.7477e-01, -6.7863e-01, -7.5909e-01,\n",
       "          -8.0018e-01, -2.4358e-01,  1.5439e-01, -4.4855e-01, -1.4862e-01,\n",
       "          -2.3419e-01,  7.3974e-02, -5.1005e-02,  1.4870e-02, -2.5686e-02,\n",
       "           1.6097e-02,  1.6317e-01, -2.4016e-01,  1.0677e-01, -2.4245e-01,\n",
       "          -3.3148e-01,  1.2349e-02, -2.0928e-01, -9.6694e-02, -5.2250e-01,\n",
       "          -6.1963e-01, -3.7462e-02, -1.0307e-01,  1.8196e-01,  1.0656e-01,\n",
       "           1.0292e-01, -1.7712e-02, -4.3886e-01,  6.1135e-01,  5.1758e-01,\n",
       "           1.6445e-01, -3.3695e-01, -3.2353e-01, -1.0853e-01,  1.7682e-01,\n",
       "           2.0129e-01,  2.2317e-01, -1.2276e-02, -1.6320e-01, -5.0478e-01,\n",
       "          -9.8563e-02,  9.8739e-01,  1.3774e-01, -4.7945e-01,  2.5363e-01,\n",
       "           9.6138e-01, -1.3354e-01,  1.6805e-01, -2.9215e-02,  1.1194e-01,\n",
       "          -4.4008e-01, -7.0675e-01,  8.0018e-01, -3.6636e-02,  1.2877e-01,\n",
       "           2.9275e-01,  1.2451e-01, -7.7632e-03,  5.2487e-01,  1.2346e-02,\n",
       "          -3.8720e-01,  3.6415e-02, -4.3676e-01,  2.8684e-01,  4.6460e-01,\n",
       "          -2.2969e-01,  5.9022e-01,  1.9322e-01,  3.9133e-02, -1.9496e-01,\n",
       "           2.8252e-01,  9.0577e-01, -3.9320e-01, -2.3260e-01,  9.4466e-01,\n",
       "           1.8525e-01,  1.9796e-02,  1.4980e-01,  1.0566e-01,  9.9099e-02,\n",
       "           4.4159e-01, -1.1097e-01, -1.8993e-01, -4.2436e-02,  2.9372e-01,\n",
       "           5.2662e-01, -7.4004e-01, -7.0469e-01,  5.2254e-02,  6.6033e-01,\n",
       "          -2.6628e-01,  3.4612e-01, -1.2615e-02,  4.7841e-01,  7.4639e-01,\n",
       "          -9.3128e-01, -1.1187e-01, -3.5959e-01,  4.2181e-01,  9.3330e-02,\n",
       "          -3.0429e-02,  2.0069e-02, -2.4715e-01, -2.1373e-01,  5.5320e-01,\n",
       "           2.8602e-01, -1.0652e+00, -1.0496e-01,  6.4288e-01, -5.5859e-01,\n",
       "           3.2355e-01, -1.0464e-01,  9.4316e-02,  9.5133e-02,  1.8910e-01,\n",
       "           1.5151e-01,  9.9232e-02, -5.0697e-01, -3.8816e-01,  1.8828e-01,\n",
       "           9.9329e-01, -7.8510e-02,  3.1882e-02, -2.5602e-01, -1.4773e-01,\n",
       "          -2.4202e-02, -4.7863e-01, -2.7923e-01,  3.7689e-01,  2.1389e-02,\n",
       "          -7.8835e-02, -3.5621e-01, -2.7949e-01,  1.0702e-01,  6.3132e-02,\n",
       "           2.4612e-01,  5.1465e-01,  7.9325e-01,  2.3308e-02, -7.1828e-02,\n",
       "           1.0114e-01,  9.7149e-02, -1.3890e-01,  4.7611e-01,  5.3982e-02,\n",
       "          -4.0961e-01,  1.5278e-01,  1.0081e-01, -4.1020e-01, -1.5330e-03,\n",
       "          -4.2404e-01,  4.4031e-01,  2.5315e-03, -1.7649e-01,  2.8251e-01,\n",
       "          -1.4685e-01,  5.0563e-02, -1.5809e-01,  6.3118e-01,  4.4867e-01,\n",
       "           7.4203e-02,  6.3175e-01,  1.7669e-01,  2.0984e-01, -4.9467e-01,\n",
       "           1.4856e-01, -7.5069e-01, -2.0604e-01, -1.3657e-01, -1.3802e-01,\n",
       "          -3.1078e-01, -4.1866e-02,  1.9206e-02,  2.2115e-01,  5.3296e-01,\n",
       "           8.6237e-01, -2.9525e-01, -6.2137e-01, -3.1109e-01,  9.9329e-02,\n",
       "           1.2273e-01, -8.8691e-02,  4.1858e-01, -8.8168e-02,  4.0425e-01,\n",
       "          -4.9122e-01, -2.8049e-01, -3.0361e-02,  4.3579e-01, -2.2783e-01,\n",
       "           7.0317e-01, -8.0943e-04, -3.5070e-02,  8.6417e-02,  3.1666e-01,\n",
       "           6.4023e-02,  8.4298e-01,  2.8706e-01, -4.1373e-01,  1.4737e-01,\n",
       "           1.7830e-02, -3.3744e-01,  1.9099e-01, -5.7530e-01,  2.2969e-01,\n",
       "           1.4163e-01,  5.5768e-01,  7.2432e-01]], device='mps:0'),\n",
       " tensor([[-9.8608e-02, -6.2949e-01, -5.4737e-01, -4.6335e-01, -3.7454e-01,\n",
       "           1.0248e-01, -2.8097e-01,  4.4707e-01, -1.3869e-01,  5.9022e-01,\n",
       "           3.3979e-01,  3.6518e-02, -7.9829e-02, -1.9712e-01, -4.0961e-01,\n",
       "          -5.4994e-01,  1.7904e-01,  2.6107e-01,  2.6766e-01, -1.3636e-01,\n",
       "           9.8026e-01,  3.6199e-01, -8.5372e-02,  1.8939e-01, -6.9872e-01,\n",
       "           3.9986e-01,  4.5490e-02,  8.0399e-02,  4.6963e-01, -6.9729e-01,\n",
       "          -2.0828e-01,  5.4662e-01, -1.2219e-01,  1.1649e+00, -1.8525e-01,\n",
       "          -2.5129e-01,  4.1644e-01, -2.6874e-01,  8.0973e-01,  4.0156e-02,\n",
       "           6.2909e-01, -1.2839e-01,  2.5974e-01,  2.1204e-01,  2.5654e-01,\n",
       "          -1.6577e-02,  1.0736e-01, -6.2638e-01,  5.4556e-01, -5.1867e-01,\n",
       "          -2.1843e-01, -1.5953e-01,  3.6604e-01,  4.8748e-01, -6.7281e-02,\n",
       "           3.5607e-02,  3.0541e-01,  1.6232e-01, -2.5763e-01,  2.0386e-01,\n",
       "           1.3393e-01,  6.9371e-01,  5.7586e-02, -1.9525e-01, -3.0287e-01,\n",
       "           3.0687e-01, -5.2685e-01, -7.0972e-01, -3.4711e-01, -1.5503e-01,\n",
       "          -6.3145e-02,  3.6453e-01, -2.3968e-01, -8.3683e-02,  7.8001e-02,\n",
       "          -8.8895e-02,  3.9835e-02, -1.7901e-01, -2.7637e-01, -9.8806e-03,\n",
       "           6.4246e-02,  3.2661e-01, -1.9744e-01, -4.1654e-01, -2.1876e-01,\n",
       "           3.0475e-01,  7.8369e-01,  6.3690e-02,  2.5669e-01,  8.5968e-02,\n",
       "           3.1750e-01,  9.4549e-02, -3.4083e-01, -4.9471e-01,  3.1401e-01,\n",
       "          -1.3255e-01,  4.9667e-01, -1.4940e-01, -6.5687e-02, -4.7700e-02,\n",
       "           3.7095e-01,  7.2304e-02,  1.7323e-03,  1.1325e-01, -6.6663e-02,\n",
       "          -4.8542e-01,  1.4204e-01, -2.5931e-01, -6.6520e-01, -1.1522e-01,\n",
       "          -7.9637e-02,  1.3277e-01,  1.5561e-02, -3.1430e-01,  2.6189e-01,\n",
       "          -7.3846e-02,  1.9145e-01, -1.3642e-02, -3.3838e-01,  3.7293e-01,\n",
       "           2.5609e-01, -4.0848e-01,  2.2958e-02,  1.6951e-01,  4.6388e-01,\n",
       "          -2.2394e-01,  6.0049e-01, -2.8694e-01,  2.5392e-01,  4.8866e-01,\n",
       "           8.4207e-02, -6.0898e-02, -9.6990e-02, -1.7751e-01, -3.9547e-01,\n",
       "          -4.2915e-02,  7.3082e-02, -3.2798e-01,  1.0024e-01,  2.1027e-02,\n",
       "          -1.5484e-01, -1.5575e-01, -7.1363e-01, -3.2433e-01,  1.6862e-01,\n",
       "          -2.9423e-01,  2.9391e-01, -7.2839e-02, -5.6183e-01, -3.6233e-01,\n",
       "          -8.1793e-01, -4.5420e-01,  5.0598e-01,  3.1144e-02, -1.4616e-01,\n",
       "          -4.4906e-02,  1.9702e-01,  1.8656e-01, -1.9001e-02,  3.6581e-01,\n",
       "           4.5356e-01, -1.7970e-01,  5.5773e-02, -4.7383e-01,  5.1199e-02,\n",
       "           3.6512e-02, -1.1536e-01, -1.4678e-01, -1.8904e-01,  8.2812e-02,\n",
       "           1.8413e-01, -2.0916e-01, -2.0209e-02, -4.3147e-01,  1.4387e-01,\n",
       "          -9.5740e-02,  8.7516e-02,  2.8605e-01, -3.1460e-01, -3.4357e-01,\n",
       "          -4.4457e-01, -2.0415e+00,  1.3037e-01, -1.6708e-01, -1.9640e-01,\n",
       "          -7.8661e-01,  5.6989e-01,  4.0038e-01,  2.3491e-01,  1.2055e-02,\n",
       "          -4.1441e-01, -1.3256e-01, -2.2065e-01,  2.0346e-01, -1.5937e-01,\n",
       "           4.1104e-01,  3.6856e-01,  2.1222e-01, -8.2505e-02, -5.3417e-02,\n",
       "          -3.3993e-01,  1.9027e-01,  6.6698e-02,  4.3789e-02,  1.3752e-02,\n",
       "           8.6470e-02, -3.4516e-01,  1.5865e-01,  6.3413e-01, -3.2035e-01,\n",
       "          -2.6413e-01, -1.5432e-01, -3.1116e-01, -1.1825e-01, -2.1011e-01,\n",
       "           7.5833e-01, -1.0075e-01, -1.9090e-01,  4.1462e-01,  2.9186e-01,\n",
       "          -2.0576e-01,  2.6479e-02, -1.7243e-01, -4.6025e-01, -8.4004e-02,\n",
       "           8.3166e-01, -2.6809e-01, -1.8435e-01, -1.3576e-01, -2.8072e-01,\n",
       "           9.9627e-02, -3.9657e-01,  7.0865e-01,  1.3131e-01, -2.4437e-01,\n",
       "          -1.3250e-01, -4.9085e-01, -3.6081e-02, -6.5221e-02,  2.7934e-01,\n",
       "          -3.2480e-01, -4.4413e-01, -6.2940e-01,  9.8728e-02,  4.2580e-01,\n",
       "           1.0494e-01, -2.2581e-01,  5.8588e-01,  3.7420e-01,  5.5104e-01,\n",
       "          -7.6877e-02,  1.8477e-03,  4.0139e-01,  4.0454e-02, -3.1153e-01,\n",
       "           4.0137e-02,  1.1839e-01, -6.1671e-02,  2.8916e-01, -5.0007e-01,\n",
       "           2.1460e-01,  1.3048e-01,  4.2285e-02,  4.0869e-01, -3.3156e-01,\n",
       "           3.3759e-01, -6.4578e-04, -4.2352e-01, -2.2792e-01,  3.1587e-01,\n",
       "          -3.4318e-01,  4.7882e-01, -1.1497e-01, -1.3533e-01, -6.5793e-01,\n",
       "          -3.0023e-01,  9.9027e-02, -5.2265e-01,  4.0820e-02, -8.5922e-01,\n",
       "          -1.4457e-01,  4.3467e-01, -1.1843e-01,  2.8682e-01, -4.3268e-01,\n",
       "           2.7079e-01,  1.1150e-01,  1.0583e-01, -1.8427e-01, -4.8035e-01,\n",
       "           6.4368e-03,  3.5471e-01, -1.9635e-01,  6.7501e-02,  1.8728e-02,\n",
       "           1.4821e-01,  7.4914e-02, -1.4411e-01,  1.8116e-02, -3.9952e-01,\n",
       "           1.2334e-01,  2.7083e-01, -6.4203e-02, -1.1696e-01, -2.5127e-01,\n",
       "           8.1457e-02,  5.7341e-02,  4.5826e-01, -1.5574e-01, -4.7792e-01,\n",
       "           1.3602e-02,  5.6159e-01,  4.2917e-02,  9.5691e-02, -4.1476e-02,\n",
       "          -1.3511e+00, -3.8201e-01,  1.3646e-01, -2.3684e-01, -1.6250e-01,\n",
       "           1.4728e-01, -4.5766e-02, -1.0990e-01,  1.3434e-01, -3.3727e-01,\n",
       "           4.9913e-01, -7.4794e-03, -9.4830e-02,  2.6318e-01, -2.9595e-01,\n",
       "          -2.6245e-01, -5.5022e-02, -5.5963e-02, -1.9048e-01, -2.1108e-01,\n",
       "           1.0760e-01,  2.4895e-01, -1.6610e-02,  2.9324e-01, -4.2315e-01,\n",
       "           6.9357e-01,  2.0403e-02, -3.6256e-01,  1.8028e-01, -4.6817e-01,\n",
       "          -2.2470e-01,  6.2020e-01, -2.1111e-01,  1.7508e-01,  7.0164e-02,\n",
       "           2.2583e-01, -1.6263e-01, -2.5922e-01, -1.5555e-01,  5.6515e-01,\n",
       "           9.7645e-02,  3.8631e-01, -3.1465e-01,  4.6445e-01,  6.0830e-01,\n",
       "          -4.3876e-01,  2.7575e-01, -3.0324e-01, -3.5509e-01, -1.4900e-01,\n",
       "           3.7937e-01,  2.4503e-01, -2.4568e-01,  1.2754e-01,  2.1491e-01,\n",
       "           2.4189e-01, -2.8123e-01, -1.4375e-01,  1.5623e-01, -1.6800e-02,\n",
       "          -1.6159e-01, -5.9860e-01,  4.8387e-02, -4.0540e-01, -3.7598e-01,\n",
       "          -5.5542e-02, -9.6940e-02,  2.3018e-01, -3.4557e-01, -2.8109e-01,\n",
       "          -4.1082e-01,  1.5328e-01,  7.9392e-01, -8.2566e-02,  1.3355e-02,\n",
       "          -2.4290e-01,  1.3134e-01, -3.9568e-01,  5.6959e-01,  2.4837e-01,\n",
       "          -5.6746e-02,  5.0092e-01, -5.9799e-01,  3.2502e-01,  1.6325e-01,\n",
       "          -2.5296e-01,  7.9720e-02, -1.1265e-01,  4.2524e-01, -1.9241e-01,\n",
       "          -2.6666e-01,  1.8088e-01,  2.9568e-01,  8.0022e-02,  3.3680e-02,\n",
       "           8.9946e-02,  7.7279e-01, -5.6155e-01, -1.1355e-01, -3.2412e-02,\n",
       "           4.6990e-01, -1.4868e-01,  1.7842e-01, -2.5184e-01, -4.5494e-01,\n",
       "           2.8096e-01,  7.6220e-03,  3.0259e-02,  1.5848e-01, -1.1470e+01,\n",
       "           1.7761e-02,  7.9956e-01, -5.9407e-01,  3.9142e-02, -7.9952e-04,\n",
       "           1.4375e-01, -2.8145e-01,  3.6361e-03, -2.5851e-01, -3.7282e-01,\n",
       "           1.6253e-01,  5.8747e-02,  6.8827e-02, -1.9252e-01, -1.2168e-01,\n",
       "          -4.4456e-02, -9.0133e-01,  1.4013e-01, -2.9238e-01,  3.1633e-01,\n",
       "          -4.6944e-01,  3.5652e-01,  1.0775e-01,  4.8818e-02, -3.3732e-01,\n",
       "          -6.3534e-02, -4.1684e-01,  2.3683e-01,  1.0364e-01, -1.6207e-01,\n",
       "          -1.3703e-01, -1.5828e-01,  4.6034e-01, -1.8993e-01, -4.8947e-01,\n",
       "           1.5862e-01, -7.0142e-02,  4.5043e-01, -1.6798e-02, -2.3410e-01,\n",
       "           5.0011e-02, -4.1456e-01,  3.0195e-01,  6.4668e-02,  5.0457e-01,\n",
       "           4.7572e-01,  6.3490e-01,  1.0739e-01, -2.1848e-01,  4.9891e-02,\n",
       "          -8.9429e-01,  3.3171e-01, -9.8824e-01, -2.6344e-01,  3.6779e-02,\n",
       "           7.4622e-01, -4.6007e-01,  9.9323e-03,  8.2971e-02, -3.9312e-01,\n",
       "           2.4107e-01, -5.5413e-02,  1.0686e-01, -6.2270e-01, -1.1555e-01,\n",
       "           3.6365e-01,  1.0164e-01,  1.3567e-01, -3.5856e-01, -1.3223e-02,\n",
       "          -4.3049e-02,  5.9870e-01,  1.5647e-01, -2.5151e-01,  1.0429e-01,\n",
       "          -8.0721e-03,  2.3224e-01,  6.2748e-01,  4.7218e-02,  1.3164e-01,\n",
       "           1.7459e-01,  5.1672e-01, -5.4105e-01, -4.7941e-01, -1.2530e-01,\n",
       "           6.3722e-02,  2.8461e-01,  8.3486e-02,  3.6163e-01, -3.9872e-01,\n",
       "          -3.1281e-01, -7.8561e-02,  3.1281e-02,  5.0728e-02,  1.2098e+00,\n",
       "          -5.8519e-01, -2.2122e-01, -1.0256e-01, -9.1470e-02, -1.3571e-01,\n",
       "          -3.9828e-02, -3.3651e-01, -4.2370e-01,  3.3182e-01,  1.1019e-01,\n",
       "           1.7328e-01,  1.5282e-02,  3.9299e-01,  1.9321e-01, -5.2099e-01,\n",
       "           2.8514e-01,  1.4470e-01,  4.3958e-01, -1.8846e-01, -4.1365e-01,\n",
       "          -1.0991e-01,  3.3157e-01, -3.8281e-01, -3.4973e-01, -1.0983e-01,\n",
       "           3.0412e-02,  5.9583e-02, -5.8809e-02, -3.7567e-01,  2.5850e-01,\n",
       "          -6.2094e-01, -9.3514e-03, -4.2726e-01, -1.1183e-01,  4.7884e-01,\n",
       "           1.2318e-01, -3.6857e-02,  1.1137e-01, -4.2935e-01, -6.7978e-01,\n",
       "          -6.4684e-01, -7.4975e-02,  5.2269e-02, -3.7956e-01, -1.6842e-02,\n",
       "          -1.8671e-01,  6.6050e-02, -3.6548e-01,  1.8633e-01, -8.2184e-02,\n",
       "           7.5332e-01,  2.3533e-01, -1.8486e-01,  1.2410e-01, -1.4424e-01,\n",
       "          -3.6447e-01, -2.1959e-01, -2.4676e-01,  2.2621e-01, -5.0366e-01,\n",
       "          -1.3080e-01,  1.8011e-01,  1.1438e-01,  6.4878e-03, -3.4448e-01,\n",
       "          -1.2932e-01,  2.0570e-01, -1.3707e-01,  3.0997e-01,  1.7373e-01,\n",
       "           2.0329e-01, -6.4335e-02, -4.6287e-01,  8.7786e-02,  4.5179e-01,\n",
       "          -1.1348e-01, -2.7396e-01,  3.8658e-02, -7.0830e-01,  2.1125e-01,\n",
       "           3.8231e-01,  8.2746e-01, -2.5998e-01, -5.3996e-01,  4.6996e-01,\n",
       "           6.2219e-01, -4.6785e-01, -5.7024e-02, -2.1133e-01,  1.2442e-01,\n",
       "          -1.0488e-01, -3.4183e-01,  2.6033e-01, -2.5177e-01,  4.2375e-01,\n",
       "          -9.9348e-02,  2.0811e-01, -1.9413e-01,  4.1807e-01,  9.7647e-02,\n",
       "          -2.0238e-01, -3.5246e-02, -6.2741e-01, -4.3863e-01,  2.3873e-01,\n",
       "           2.6324e-01, -1.5522e-01, -2.3160e-02, -1.2275e-01,  1.1149e-01,\n",
       "           4.8271e-01,  8.5393e-01,  8.1221e-02,  1.4584e-01,  3.8317e-01,\n",
       "           2.3356e-01, -2.7630e-01,  5.9953e-02,  2.9270e-01,  4.0766e-01,\n",
       "           3.5259e-01,  5.3145e-01, -3.4176e-01,  7.2203e-02, -1.6344e-01,\n",
       "           2.0465e-01, -5.7619e-01, -7.6698e-01, -1.3126e-01,  4.3082e-01,\n",
       "           1.5541e-02,  2.9468e-01, -2.7851e-01,  3.4408e-01,  6.3093e-01,\n",
       "          -1.1306e+00, -5.9181e-02, -3.9846e-02, -9.3994e-02, -6.2591e-02,\n",
       "          -5.4897e-01, -2.2413e-01, -3.5234e-01, -2.4418e-01,  5.2852e-01,\n",
       "           5.8811e-01, -8.5286e-01,  1.6085e-01,  2.2953e-01, -1.5005e-01,\n",
       "           9.4328e-02, -9.1644e-02,  3.2863e-02,  5.0134e-01,  1.5175e-01,\n",
       "          -5.7932e-01,  4.5778e-01,  1.5595e-01, -4.3637e-01,  2.6165e-01,\n",
       "           4.1776e-01, -1.8687e-02,  1.5952e-01,  2.2085e-01, -4.9751e-01,\n",
       "          -1.0141e-02, -4.6200e-01, -1.0629e-01,  1.0751e-01, -6.7537e-02,\n",
       "           2.4815e-02, -4.9794e-01, -6.6236e-02,  3.4484e-01,  2.9505e-03,\n",
       "           5.0621e-01,  6.0668e-01,  7.6526e-01, -5.6484e-02, -2.1306e-01,\n",
       "           3.3852e-01,  2.5943e-01, -9.1834e-02,  5.2227e-01, -2.1885e-02,\n",
       "          -1.6835e-01,  3.1658e-01,  2.3124e-02, -2.2045e-01,  1.7294e-01,\n",
       "          -2.5761e-01,  2.6423e-01,  1.0870e-01, -4.9724e-01,  3.6477e-01,\n",
       "           3.7087e-01,  1.5082e-01, -3.0923e-01,  1.8171e-01,  3.9828e-01,\n",
       "          -3.1100e-01,  6.8356e-01,  1.6607e-04,  2.2973e-01,  6.3476e-02,\n",
       "          -1.1506e-01, -5.5647e-01, -3.1374e-01,  2.1509e-01,  9.2105e-02,\n",
       "          -5.0547e-01, -9.5287e-02,  1.5106e-01, -4.9544e-01,  7.8060e-01,\n",
       "           6.2385e-01, -3.7763e-01, -3.5664e-01, -7.1127e-02,  1.1295e-01,\n",
       "           1.2269e-01, -3.3879e-01,  2.9572e-01,  1.1312e-01,  1.0076e-01,\n",
       "          -1.0219e+00, -2.9326e-01,  2.4530e-01,  4.9147e-02, -4.2291e-01,\n",
       "           5.9939e-01,  7.9206e-02,  5.0359e-02,  3.8428e-01,  3.0534e-01,\n",
       "           1.4161e-01,  4.3910e-01,  7.0775e-02, -2.1375e-01,  6.5311e-02,\n",
       "           2.5600e-01, -3.0970e-01, -3.2766e-01, -6.3757e-01,  5.3684e-01,\n",
       "           1.5068e-01,  8.4160e-01,  6.5313e-01]], device='mps:0'),\n",
       " tensor([[ 4.1996e-02, -7.2195e-01, -5.2851e-01, -5.1034e-01, -2.5314e-01,\n",
       "          -3.1635e-01, -2.8946e-01,  5.9984e-01, -1.1747e-01,  7.0401e-01,\n",
       "           9.6475e-02,  3.6403e-01,  1.8809e-01, -2.7645e-01, -7.9964e-01,\n",
       "          -4.8550e-01,  8.7368e-03,  2.1978e-01, -3.4828e-02,  3.5115e-01,\n",
       "           5.7684e-01,  4.6877e-01,  4.0164e-02,  9.0115e-02, -1.8701e-01,\n",
       "           3.1114e-01, -2.4456e-02, -1.4454e-01,  1.0148e-01, -9.7157e-01,\n",
       "          -3.9865e-01,  7.8972e-01,  1.8364e-01,  1.0266e+00, -6.1282e-01,\n",
       "          -4.0360e-01, -3.7091e-02, -1.8689e-01,  5.2737e-01, -2.9897e-01,\n",
       "           5.3119e-01, -5.9364e-01,  2.5571e-01,  4.0321e-01,  9.8940e-01,\n",
       "           1.1851e-01, -4.1626e-02, -7.5587e-01,  1.3515e-01, -2.9840e-01,\n",
       "          -4.0105e-01, -1.1812e-01,  3.7929e-01,  2.1614e-01, -2.6738e-01,\n",
       "          -6.3268e-01, -8.5300e-02,  1.1410e-01, -4.8719e-01, -4.7487e-02,\n",
       "          -7.1504e-02,  3.9185e-01, -9.7085e-02,  2.6362e-01,  1.3422e-01,\n",
       "           2.3544e-01, -3.3408e-01, -2.3819e-01, -6.2966e-01, -7.2678e-03,\n",
       "           1.6136e-01, -2.3102e-01, -9.4671e-02, -4.0223e-01,  2.8533e-01,\n",
       "          -3.3589e-01,  2.8497e-01, -2.3217e-01, -7.2594e-01,  1.8640e-02,\n",
       "          -1.1440e-02,  5.9732e-01, -4.0847e-01, -2.7074e-01, -2.9465e-01,\n",
       "          -2.1619e-01,  7.3327e-01,  2.7036e-01,  3.9177e-01, -7.2944e-01,\n",
       "           1.5096e-02,  1.7405e-01, -3.3069e-01, -5.5721e-01,  2.3398e-01,\n",
       "           1.0582e-02,  4.4319e-01, -1.8598e-01, -1.7094e-01, -6.2842e-01,\n",
       "          -3.9479e-02,  1.7603e-01,  4.9749e-02,  4.5001e-01, -7.1238e-01,\n",
       "          -6.2676e-01,  9.0222e-02, -1.3916e-01, -8.1397e-01, -2.5790e-01,\n",
       "          -7.6894e-02, -2.0937e-01,  3.5581e-01, -5.7569e-01, -2.1230e-01,\n",
       "          -7.9744e-02,  1.6771e-01,  2.5780e-01, -3.4001e-01,  5.4150e-01,\n",
       "           2.7939e-01, -2.8914e-01,  5.1230e-02,  1.5497e-01, -3.6194e-02,\n",
       "           1.0528e-01,  4.9095e-01, -2.7060e-01,  3.2038e-01,  2.4526e-01,\n",
       "          -1.4441e-01,  4.7543e-02, -5.7406e-01, -1.9501e-01,  1.4322e-01,\n",
       "          -1.8845e-01,  1.6540e-01, -5.2188e-01, -6.4139e-01,  2.4246e-02,\n",
       "           4.3305e-02,  1.6044e-01, -2.3713e-01, -3.4509e-01,  2.0898e-01,\n",
       "          -3.1106e-01, -1.2688e-01, -1.1428e-01, -7.2933e-01, -4.1833e-01,\n",
       "          -3.3588e-01, -5.1946e-01,  5.4545e-01, -2.1554e-01, -1.6898e-01,\n",
       "          -2.3405e-01,  2.9966e-01, -1.4947e-01,  1.9485e-01,  3.4919e-01,\n",
       "           6.9925e-01, -6.0058e-02,  3.2265e-01, -5.9747e-01, -8.8737e-02,\n",
       "           3.9937e-01, -1.7733e-01, -1.6061e-01,  5.1315e-01,  3.7885e-01,\n",
       "           2.3960e-01, -3.0973e-01,  1.3062e-01, -4.6515e-01,  3.5853e-01,\n",
       "          -2.3794e-01,  6.0039e-01,  2.4167e-01, -3.1163e-01,  6.3453e-02,\n",
       "           1.2409e-01, -1.6597e+00, -8.6468e-02, -2.5353e-01, -1.8906e-01,\n",
       "          -5.3263e-01,  6.1854e-01,  3.9403e-01,  3.5619e-01, -1.9650e-01,\n",
       "           2.4900e-01, -1.4032e-01,  1.5130e-01,  1.8249e-01,  1.0154e-01,\n",
       "           2.1015e-01,  4.7949e-01,  1.5780e-01,  1.1452e-01,  2.3763e-02,\n",
       "          -6.9437e-01,  2.7443e-01,  8.3749e-02,  2.7291e-01,  1.5183e-01,\n",
       "           1.7930e-01,  4.1651e-02, -7.9899e-03,  9.6045e-02,  2.2135e-01,\n",
       "          -5.2910e-02, -2.7919e-01, -3.1515e-01, -4.8726e-01, -6.0650e-01,\n",
       "           6.4296e-01, -2.9767e-01, -3.5011e-01,  3.8537e-01,  4.1485e-01,\n",
       "          -2.2945e-01, -1.6290e-01,  3.4648e-01, -2.3777e-01, -1.7233e-01,\n",
       "           4.5943e-01, -8.2158e-02, -2.7460e-02,  5.2322e-04, -2.7323e-01,\n",
       "           1.4089e-01, -4.1645e-01,  1.7926e-01,  2.4725e-01, -8.5927e-03,\n",
       "          -4.9412e-01, -2.0743e-01,  2.6369e-02, -5.0950e-01,  1.6685e-01,\n",
       "           4.9666e-02, -2.2874e-01, -5.6948e-01,  5.9468e-02,  1.3645e-01,\n",
       "           6.2197e-02, -1.9554e-01,  1.0693e+00,  5.3146e-01,  5.0364e-01,\n",
       "          -2.7178e-01, -2.2124e-01,  5.0998e-01, -1.2210e-01, -3.9656e-01,\n",
       "           2.2412e-01,  1.5916e-01,  6.2459e-02,  1.7010e-01, -1.5814e-01,\n",
       "          -5.2594e-03,  1.7405e-01,  2.6248e-01,  4.9563e-01, -6.3678e-01,\n",
       "          -2.1386e-01,  3.9688e-01, -6.4470e-01, -1.4409e-01,  2.6378e-01,\n",
       "          -1.5812e-01,  3.6637e-01,  1.1914e-01, -8.6514e-01, -5.8750e-01,\n",
       "          -2.5693e-01,  3.4140e-03, -5.8516e-02,  3.0717e-01, -9.8148e-01,\n",
       "           1.5329e-01,  2.1383e-01, -1.7904e-01,  2.7423e-01, -3.5658e-01,\n",
       "           2.5285e-01,  7.5734e-02, -3.7416e-01, -4.4276e-01, -6.6812e-01,\n",
       "           6.6736e-02, -2.8842e-01, -4.1756e-01,  3.8268e-01,  1.9258e-01,\n",
       "           3.9879e-01,  5.7436e-01,  2.0146e-01,  3.8490e-01, -5.4265e-01,\n",
       "           2.2104e-01, -2.2435e-02, -5.3984e-01, -2.4545e-01, -4.9867e-02,\n",
       "           3.8680e-01,  1.6945e-01,  4.6477e-01, -4.8512e-01, -6.4099e-01,\n",
       "           1.4133e-01,  6.0082e-01, -1.7073e-01, -1.7975e-01, -1.0469e-01,\n",
       "          -1.5410e+00, -1.7418e-01, -3.7242e-01, -2.4866e-02, -6.7151e-01,\n",
       "           5.9601e-02, -1.2764e-01, -8.5242e-03, -2.3436e-01, -7.2911e-02,\n",
       "           1.3006e-01,  3.1430e-01,  5.2335e-02, -1.9662e-02, -3.3080e-01,\n",
       "          -5.5543e-01,  1.8459e-01, -2.3934e-01, -2.8761e-01, -2.4175e-01,\n",
       "           2.3740e-02,  5.4195e-01, -2.6213e-01, -1.3934e-01, -2.2273e-01,\n",
       "           3.9284e-01, -2.5105e-01, -4.6565e-01,  1.9214e-01, -1.2475e-01,\n",
       "          -2.2535e-04,  5.2562e-01, -2.2709e-01, -3.8190e-01, -8.7214e-02,\n",
       "          -1.1182e-01, -2.6078e-01, -4.8285e-01, -5.7076e-02,  5.8159e-01,\n",
       "           5.0698e-01,  1.1813e-02, -5.4336e-01,  5.1669e-01,  1.8551e-01,\n",
       "          -5.0285e-01,  3.0972e-01,  2.1863e-01,  9.2948e-02,  9.0802e-02,\n",
       "           2.4190e-01,  3.4499e-01, -4.2993e-01,  6.3092e-02,  1.1101e-01,\n",
       "           1.6735e-01, -5.1919e-01, -2.4122e-01,  3.1374e-01,  3.5258e-01,\n",
       "          -1.2897e-01, -4.5679e-01,  3.9646e-01, -4.4078e-01, -5.5130e-01,\n",
       "          -6.4852e-02,  1.2110e-02,  9.2860e-02, -2.0410e-02, -6.5676e-01,\n",
       "          -3.3070e-01,  2.3536e-01,  4.5375e-01,  4.0680e-01, -1.9968e-01,\n",
       "          -7.0772e-02, -8.6968e-02, -9.0614e-02,  3.4848e-01, -3.6649e-02,\n",
       "           1.4808e-01,  1.9017e-01,  9.2420e-02,  1.1110e-01,  4.1193e-01,\n",
       "          -2.9344e-01, -7.3396e-01,  1.9732e-01,  2.7212e-01, -1.8035e-01,\n",
       "          -1.6151e-01,  5.1943e-02,  3.2984e-01, -3.7521e-01, -1.8457e-01,\n",
       "           2.4051e-01,  3.2081e-01, -2.0330e-01,  3.5863e-01, -4.6395e-01,\n",
       "           2.1871e-01,  2.5685e-01,  1.4271e-01, -9.5561e-02, -2.6830e-01,\n",
       "           4.8567e-01,  4.3755e-01,  2.9889e-01,  2.6597e-01, -1.1117e+01,\n",
       "           2.1139e-01,  5.4733e-01, -3.3483e-01,  1.6206e-01, -2.3924e-02,\n",
       "           1.4773e-01, -2.3816e-01, -2.2781e-01, -6.2281e-01,  6.4850e-03,\n",
       "          -3.0360e-01,  1.5595e-01,  1.4133e-01, -3.2908e-01, -2.4791e-01,\n",
       "          -2.3053e-01, -3.8529e-01,  3.9393e-01, -4.1747e-02,  4.8230e-01,\n",
       "           9.2491e-02,  1.0064e-01,  1.2882e-01, -3.4561e-02,  3.1640e-02,\n",
       "           1.8670e-01, -4.1074e-01,  3.9596e-01,  5.3937e-01,  1.4807e-01,\n",
       "           7.5288e-02,  1.1948e-01,  2.0517e-01, -1.6542e-01, -4.8035e-01,\n",
       "           5.1690e-01, -1.6675e-01, -8.9695e-03,  3.8606e-01, -1.1628e-01,\n",
       "           4.4005e-02, -7.4485e-01,  4.0533e-01,  3.6593e-02,  7.5355e-01,\n",
       "           9.3719e-01,  9.0069e-01,  3.4168e-01, -3.3790e-01, -2.6996e-01,\n",
       "          -5.1228e-01,  2.0853e-01, -6.7584e-01, -2.0738e-01, -4.4492e-01,\n",
       "           8.3418e-01, -5.1663e-01,  9.2893e-02, -3.5175e-01, -4.8357e-01,\n",
       "           6.1363e-01,  1.7950e-01, -1.5570e-02, -2.8800e-01, -2.6691e-01,\n",
       "           3.4300e-01, -7.9984e-03,  3.3488e-01, -3.7053e-01,  5.0961e-01,\n",
       "          -1.7098e-01,  2.4939e-01,  2.9130e-01,  1.4060e-01,  4.8199e-01,\n",
       "           2.1571e-01, -9.7343e-02,  1.4173e+00, -7.3546e-02, -1.5495e-01,\n",
       "          -3.7366e-02,  2.0012e-01, -6.1400e-01, -3.7815e-01, -4.2842e-01,\n",
       "           7.1866e-02,  1.1937e-01,  1.5509e-01,  3.6230e-01, -1.2074e-01,\n",
       "          -7.9255e-01, -1.8148e-02, -4.9097e-02,  3.1719e-01,  1.1955e+00,\n",
       "          -5.9662e-01,  2.2923e-01,  1.2434e-01,  1.7857e-02, -3.8315e-01,\n",
       "           8.1580e-02, -2.3233e-01, -4.0775e-01,  1.4719e-01,  5.1995e-02,\n",
       "           7.7415e-02, -1.8900e-01,  1.4003e-01,  4.9706e-01, -2.2346e-01,\n",
       "           3.0413e-01,  6.1930e-02,  2.9248e-01, -2.7351e-01, -1.7634e-01,\n",
       "           3.4615e-01,  3.9395e-01, -1.6013e-01, -7.2885e-01, -3.4862e-01,\n",
       "          -2.2157e-01,  8.3370e-02,  1.8916e-01, -3.2585e-01,  2.5831e-01,\n",
       "          -6.4408e-01, -2.4095e-01, -9.9382e-02, -1.0446e-01,  3.9807e-01,\n",
       "           1.7949e-01,  1.9204e-01, -7.2679e-03, -9.3095e-01, -5.6223e-01,\n",
       "          -7.2915e-01, -1.6302e-01, -1.8879e-01, -2.0741e-01, -2.0515e-01,\n",
       "           3.8035e-01, -1.6728e-01, -1.5467e-01,  1.5847e-01,  1.8968e-01,\n",
       "           6.5057e-01,  4.8220e-01, -5.6797e-02,  1.4816e-02, -8.1851e-02,\n",
       "          -3.6895e-01, -1.8817e-01, -3.2527e-01,  2.8309e-02, -2.9161e-01,\n",
       "           8.9658e-02,  1.8359e-01, -6.3697e-02,  2.7432e-01,  1.6539e-01,\n",
       "          -1.3928e-01, -2.5385e-01,  8.6481e-02,  1.2484e-01,  2.2150e-02,\n",
       "           2.0463e-01, -4.2719e-02, -1.8665e-01, -4.5450e-02,  4.0442e-01,\n",
       "           1.9607e-01, -5.8075e-02, -1.8064e-01, -1.0689e-01, -7.3734e-02,\n",
       "           4.4962e-01,  1.1987e+00, -2.6552e-01, -4.6645e-01,  1.3478e-01,\n",
       "           6.3244e-01,  8.6234e-03, -5.0724e-02,  7.6674e-02,  1.8918e-01,\n",
       "          -1.3619e-03, -6.7547e-01,  3.1259e-01, -2.8536e-02,  3.6014e-01,\n",
       "           2.9450e-01, -3.7487e-01,  1.9751e-01,  6.3056e-01,  1.8950e-01,\n",
       "          -3.1452e-02,  1.4233e-01, -5.1266e-01, -1.8071e-01,  2.3579e-01,\n",
       "          -1.3654e-01,  2.0432e-01, -3.9479e-01,  1.2811e-01, -5.2366e-02,\n",
       "           1.2022e-01,  1.0123e+00,  4.2723e-01, -2.8362e-01,  5.0505e-01,\n",
       "           3.2144e-01, -1.5640e-01,  3.1493e-01,  8.8109e-02,  7.3548e-02,\n",
       "           2.9078e-01,  2.0404e-01, -3.2153e-01,  1.0831e-01,  4.3536e-01,\n",
       "           5.3144e-01, -2.0153e-01, -6.0102e-01, -1.4913e-01,  8.8675e-02,\n",
       "           1.3767e-01,  4.3405e-01, -4.2225e-02,  4.9409e-01,  8.7937e-01,\n",
       "          -1.0165e+00, -1.7164e-01, -2.8076e-01,  3.9212e-01, -7.6194e-02,\n",
       "          -5.8635e-01, -4.8312e-02, -2.3311e-01, -4.0197e-01, -1.1029e-01,\n",
       "           5.4016e-01, -1.2031e+00,  3.4558e-01,  2.1284e-01, -1.5809e-01,\n",
       "           1.9751e-01, -8.7633e-02,  1.2516e-01,  3.4900e-01,  1.2786e-01,\n",
       "           2.5912e-01,  3.6734e-01, -7.2968e-02, -9.7313e-01,  4.4956e-01,\n",
       "           5.4442e-01,  2.7565e-01,  7.2402e-02, -1.2922e-02, -1.3951e-01,\n",
       "          -2.1718e-02, -6.1536e-01, -3.9435e-01,  1.1335e-01, -7.2724e-01,\n",
       "          -4.7472e-02, -4.0034e-01, -1.9115e-02,  2.3535e-01,  1.0366e-01,\n",
       "           5.5476e-01,  5.0155e-01,  7.7175e-01, -3.5134e-01, -2.6605e-01,\n",
       "          -1.5238e-01,  4.1912e-01, -1.7531e-01,  4.9802e-01,  8.4179e-02,\n",
       "          -2.8729e-01,  2.5887e-01,  2.5466e-01,  1.6790e-01,  1.2018e-01,\n",
       "          -2.5856e-01,  5.0194e-01, -2.9929e-01, -2.5178e-01,  2.4354e-01,\n",
       "           2.5844e-01,  3.7921e-02, -5.3718e-01, -2.6064e-02,  1.1435e-01,\n",
       "          -7.3542e-02,  8.1194e-01, -4.2120e-01,  3.8590e-01, -4.3711e-01,\n",
       "           2.8001e-01, -6.6945e-01, -3.2370e-01, -4.3960e-02, -3.2312e-01,\n",
       "          -2.5008e-01, -8.6087e-02,  4.3126e-01,  4.9529e-01,  6.1654e-01,\n",
       "           4.8578e-01, -1.8644e-01, -5.3280e-01, -3.6557e-01, -1.3126e-01,\n",
       "           4.2881e-02,  1.7392e-01,  9.9740e-02,  5.4126e-02,  2.2842e-01,\n",
       "          -1.0236e+00, -2.6397e-01,  2.4196e-01,  4.3392e-01, -9.7281e-02,\n",
       "           3.8846e-01, -5.0532e-02, -9.1840e-02,  4.8263e-01,  3.7944e-01,\n",
       "           2.7831e-01,  6.7085e-01,  6.5123e-02, -9.0602e-01,  2.3161e-01,\n",
       "           1.5675e-01, -3.0430e-01,  6.3728e-02, -8.6250e-01,  4.3700e-01,\n",
       "           1.1221e-01,  5.2606e-01,  4.0842e-01]], device='mps:0'),\n",
       " tensor([[-1.7731e-01, -3.8604e-01, -5.6476e-01, -2.2650e-01, -1.3208e-01,\n",
       "          -3.1625e-01, -5.9292e-01,  4.5825e-01,  2.2151e-01,  5.2940e-01,\n",
       "           3.8502e-01,  1.4343e-01,  6.5615e-02, -2.5563e-01, -4.4222e-01,\n",
       "          -1.6916e-01,  1.6884e-01,  5.0107e-01,  3.6126e-01, -7.6046e-03,\n",
       "           5.9085e-01,  3.7327e-01, -2.0317e-01,  2.5562e-01, -7.5220e-01,\n",
       "           2.6747e-02, -1.6815e-01, -2.5680e-02,  9.6832e-02, -3.0143e-01,\n",
       "          -1.1712e-01,  2.7595e-01, -3.7936e-01,  7.2144e-01, -1.4269e-01,\n",
       "          -2.3945e-01,  1.6259e-01, -1.0450e-01,  5.3649e-01, -1.3024e-01,\n",
       "           5.7769e-01, -2.9111e-01,  4.9321e-01,  3.4162e-02,  2.2711e-01,\n",
       "           1.3206e-01,  2.9819e-01, -4.0790e-01,  1.3721e-01, -5.3671e-01,\n",
       "          -1.6362e-02, -7.3465e-02,  4.1546e-01,  1.0724e-01, -6.3685e-01,\n",
       "          -2.7220e-01, -7.3533e-02,  4.2370e-01, -3.4661e-01,  1.4839e-01,\n",
       "           1.4452e-01,  4.6875e-01,  1.2890e-01,  1.1326e-01, -3.3348e-01,\n",
       "           3.0093e-01, -7.0323e-01, -8.0455e-01, -5.7621e-01, -1.8468e-01,\n",
       "           2.6210e-01,  2.4685e-01,  3.2711e-01, -6.3178e-01,  2.8882e-01,\n",
       "          -5.5500e-01,  4.7778e-01, -3.7641e-02, -3.8107e-01,  2.4976e-02,\n",
       "           4.6662e-01,  1.6673e-01, -2.3893e-01,  2.6411e-01,  2.9965e-01,\n",
       "          -6.2089e-02,  4.4818e-01, -2.9706e-02,  5.6407e-02,  3.8863e-01,\n",
       "           5.0392e-01,  2.6142e-01, -5.4248e-01, -2.1306e-01,  2.3546e-01,\n",
       "          -3.7940e-01,  2.9621e-01, -1.0776e-01,  2.0922e-02, -8.6342e-02,\n",
       "          -3.6881e-02,  3.4334e-01,  3.1658e-01, -6.2528e-02, -1.0724e+00,\n",
       "          -4.3983e-01, -2.4311e-02, -2.7814e-01, -3.4665e-01, -1.0267e-02,\n",
       "           7.0862e-03, -2.1454e-01,  4.5842e-01, -4.4114e-01, -3.9926e-01,\n",
       "           1.3807e-01, -3.8646e-02, -1.5082e-02, -1.4568e-01,  6.0048e-01,\n",
       "           1.6661e-01, -5.3057e-02,  1.5341e-01,  8.8933e-02,  4.9861e-01,\n",
       "          -3.9381e-01,  3.7161e-01, -4.2227e-01,  3.0903e-01,  6.8681e-02,\n",
       "          -2.0183e-02, -7.2510e-02, -2.5207e-01, -2.0895e-01, -2.4617e-02,\n",
       "           2.1165e-02,  2.8627e-01, -4.8458e-01,  5.5727e-02,  5.6803e-02,\n",
       "           2.5004e-01, -1.2498e-01, -4.2667e-01, -1.9318e-01, -6.4563e-02,\n",
       "          -4.0302e-02,  1.9246e-01, -1.1367e-01, -2.7591e-01, -2.6032e-01,\n",
       "          -7.7253e-01, -4.4007e-01,  3.9722e-01,  1.0952e-01,  9.2693e-02,\n",
       "           1.1479e-01,  2.1754e-01, -1.9775e-01, -1.8512e-01, -1.5673e-01,\n",
       "           5.8319e-01, -9.5195e-02,  6.6991e-02, -3.6372e-01, -3.0891e-01,\n",
       "           7.3167e-02,  1.7092e-01, -1.7339e-01,  4.3863e-02,  1.4590e-02,\n",
       "           2.2225e-01, -8.8855e-02, -1.0134e-03, -7.5909e-01, -3.0093e-02,\n",
       "          -2.8356e-01,  3.3814e-01,  2.7840e-01,  1.8768e-01, -2.4537e-01,\n",
       "           5.0899e-02, -1.7294e+00,  7.8671e-01, -5.9166e-01,  2.4877e-01,\n",
       "          -1.8150e-01,  3.1718e-01,  9.1189e-02,  1.8509e-01, -2.9267e-01,\n",
       "          -1.2026e-01, -3.9309e-02,  4.1623e-02,  2.9227e-01, -8.4757e-02,\n",
       "           4.8559e-01,  5.5627e-01,  3.3170e-01,  2.0436e-01, -9.9033e-02,\n",
       "          -6.0977e-01,  3.3235e-01,  1.6388e-01, -2.7455e-01,  2.9857e-02,\n",
       "           2.4361e-01, -3.2791e-01, -5.7703e-02,  3.4800e-01,  3.3556e-02,\n",
       "          -8.5587e-02, -3.0852e-01, -4.0912e-01, -1.8006e-01, -1.6990e-01,\n",
       "           3.0904e-01, -2.3500e-01, -2.0646e-01,  7.4699e-01,  9.5515e-02,\n",
       "          -2.2466e-02, -7.9463e-02, -2.8277e-01, -4.4039e-01, -2.0940e-01,\n",
       "           2.9576e-01, -5.3435e-01, -1.7453e-01, -2.8854e-01, -4.6020e-01,\n",
       "           4.6218e-01, -9.4786e-01,  6.3073e-01,  2.5225e-01,  1.6028e-01,\n",
       "           6.5808e-02, -8.2977e-02, -1.3264e-01, -3.6344e-01, -1.9702e-02,\n",
       "          -2.1377e-02, -5.7642e-01, -9.0486e-01,  1.5472e-01, -7.9254e-02,\n",
       "           1.2900e-01, -4.1901e-02,  6.2316e-01,  4.2972e-01,  5.9118e-01,\n",
       "           2.3140e-01,  5.5355e-03,  4.6999e-01,  4.3441e-01, -4.1017e-01,\n",
       "           4.6498e-01,  1.8871e-01, -4.7838e-02,  6.9646e-02, -3.2186e-01,\n",
       "          -9.1404e-01,  2.0856e-02,  3.2947e-01,  4.9930e-01, -1.7114e-01,\n",
       "           4.2813e-01,  2.6662e-01, -1.3998e-01, -1.7031e-01,  3.1347e-01,\n",
       "          -4.8944e-01, -1.5127e-01,  9.5415e-02, -1.4029e-01, -6.5634e-01,\n",
       "          -4.8841e-01,  2.7357e-02, -2.9199e-01,  2.0484e-01, -9.9702e-01,\n",
       "          -2.6273e-01,  5.6883e-01, -2.8623e-02,  9.3732e-02, -2.3345e-01,\n",
       "           7.0240e-02, -1.0750e-01,  1.0928e-01, -1.5161e-01, -8.9822e-01,\n",
       "          -1.4258e-01,  9.6331e-02, -2.2318e-01,  3.2530e-01,  3.7809e-01,\n",
       "          -2.0083e-01, -2.9617e-01,  2.8364e-01, -9.0550e-02, -2.5039e-01,\n",
       "          -1.3545e-01, -3.7063e-01, -2.4607e-02, -1.3576e-01, -1.1638e-01,\n",
       "           6.9800e-02,  4.9085e-01,  4.8805e-01, -7.7597e-01, -5.5824e-01,\n",
       "          -2.2992e-02,  5.0756e-01, -1.6302e-01,  3.7266e-02, -2.2967e-01,\n",
       "          -1.0825e+00, -2.0374e-01, -1.1835e-01, -3.9640e-01, -3.8528e-01,\n",
       "           5.0161e-01, -1.9830e-01, -2.0671e-01,  9.6594e-02, -3.4539e-01,\n",
       "           3.4160e-01,  8.4200e-01, -4.9032e-02, -4.7416e-02, -2.9725e-01,\n",
       "           1.7477e-02, -1.8692e-01,  4.8213e-02, -1.9462e-01, -2.5485e-02,\n",
       "           3.1570e-01,  1.2741e-01, -1.8351e-01, -8.2635e-02, -1.7372e-01,\n",
       "           7.9347e-01, -5.6947e-01,  1.3088e-01,  3.8738e-01, -2.4876e-01,\n",
       "          -1.3259e-01,  4.8066e-01,  1.4439e-01,  1.9491e-01, -1.0024e-01,\n",
       "          -2.6493e-02, -2.8608e-01, -1.8423e-01, -3.1248e-01,  9.6616e-01,\n",
       "           3.5980e-01,  2.2368e-01, -3.0496e-01,  5.7904e-01,  5.6810e-01,\n",
       "          -3.8525e-01,  7.6086e-02,  8.8516e-02,  3.4151e-03, -3.1432e-01,\n",
       "           3.2810e-01,  2.4729e-01, -4.2766e-01, -2.5123e-02,  4.5374e-03,\n",
       "           1.4485e-01, -5.5797e-01,  1.6191e-01,  1.0782e-01,  8.5195e-02,\n",
       "          -1.4724e-03, -2.6111e-01,  5.2888e-01, -2.4531e-01, -2.8912e-01,\n",
       "          -1.0092e-01, -1.2182e-01,  4.1012e-02, -2.4548e-01, -3.6887e-01,\n",
       "          -2.7428e-01,  2.4193e-02,  3.5074e-01,  1.4285e-02,  7.4821e-02,\n",
       "           2.5544e-01, -2.6796e-01, -3.4638e-01,  4.5500e-01, -1.7660e-02,\n",
       "          -1.6810e-01,  4.0768e-01, -2.4890e-01, -1.1639e-01,  1.3881e-01,\n",
       "           6.2926e-02,  1.0677e-01, -5.8094e-02,  3.7998e-01,  1.7175e-01,\n",
       "          -2.8992e-01,  3.6344e-02,  2.1408e-01, -8.0245e-02,  3.1665e-01,\n",
       "           5.1861e-02,  8.2278e-01, -3.7109e-01, -2.7164e-02, -3.8441e-01,\n",
       "          -1.2821e-01, -1.6499e-01,  7.5490e-03,  1.5024e-01, -7.5773e-01,\n",
       "           3.0131e-01,  2.8842e-01,  3.4034e-01,  1.1331e-01, -1.1458e+01,\n",
       "           1.6305e-01,  6.4218e-01, -5.1374e-01,  1.8438e-01,  1.7594e-01,\n",
       "          -1.4040e-01, -3.9794e-01,  4.0875e-02, -3.4785e-01, -2.6394e-01,\n",
       "          -6.5590e-02,  1.9075e-01, -1.2756e-02, -5.5478e-01,  3.3680e-02,\n",
       "          -2.5398e-01, -6.3192e-01,  2.6160e-01, -9.0323e-02,  4.1335e-01,\n",
       "          -3.0876e-01,  1.8731e-01,  2.3832e-01, -3.6034e-01, -2.7614e-02,\n",
       "          -5.1547e-01, -5.5807e-02,  3.7042e-01,  3.2770e-01, -3.5254e-01,\n",
       "           2.2243e-01, -1.8729e-01,  3.8908e-01, -2.6449e-01, -5.2363e-01,\n",
       "           1.3092e-01, -9.6993e-02,  3.7952e-01,  1.1884e-02, -2.4878e-01,\n",
       "          -2.5432e-02, -5.2584e-01,  5.9952e-02,  2.0270e-01,  3.4279e-01,\n",
       "           1.2541e+00,  9.4924e-01,  1.3760e-01, -1.3198e-01,  1.6614e-01,\n",
       "          -7.3513e-01,  2.0677e-01, -6.2811e-01, -8.1585e-02, -6.4376e-02,\n",
       "           8.3128e-01, -4.3261e-01, -3.3743e-01,  7.3012e-02, -8.0493e-02,\n",
       "           3.1300e-01, -7.9471e-02,  2.6202e-01, -3.1186e-01,  2.9341e-03,\n",
       "           6.1375e-01, -1.8722e-02,  3.8063e-01, -4.3721e-01, -6.2672e-02,\n",
       "          -4.2457e-02,  2.9525e-01,  3.7279e-01,  6.2687e-02, -2.5474e-01,\n",
       "          -4.1450e-01,  4.2406e-02,  5.2243e-01,  1.3990e-01,  1.4541e-01,\n",
       "           5.8211e-01, -2.3165e-01, -4.1432e-01, -6.0104e-01, -3.9034e-01,\n",
       "          -7.6601e-03,  1.3684e-01, -3.9867e-01,  6.4750e-01, -4.2253e-01,\n",
       "          -2.7567e-01, -1.6427e-01, -8.8628e-02,  3.8882e-01,  5.0639e-01,\n",
       "          -3.3627e-01,  8.0837e-02, -1.8585e-01, -4.4707e-03, -6.7852e-01,\n",
       "          -3.5458e-01, -8.4538e-02, -2.9987e-01,  2.3231e-01,  1.7305e-01,\n",
       "          -1.0020e-01,  9.0697e-03,  1.2051e-01,  7.3101e-02, -5.4356e-01,\n",
       "           3.0479e-01, -1.0749e-01,  6.8380e-02, -3.9586e-01, -5.1756e-01,\n",
       "           4.0833e-01,  7.0232e-01, -4.2415e-01, -3.3873e-01,  9.8320e-02,\n",
       "           1.3073e-01,  1.8252e-01,  1.2615e-01, -6.4414e-01,  1.1900e-01,\n",
       "          -4.1931e-01, -1.3812e-01,  1.3512e-01,  7.2335e-01,  6.6950e-01,\n",
       "          -5.3703e-02, -3.2049e-01,  2.4632e-01, -5.0640e-02, -2.5765e-01,\n",
       "          -7.0535e-01, -2.9464e-02,  2.1372e-01, -5.3812e-01,  9.7345e-02,\n",
       "          -3.3445e-01,  1.9755e-01, -1.7248e-01,  1.0825e-01, -7.4644e-02,\n",
       "           4.1662e-01,  4.1987e-01,  7.6810e-02,  4.3515e-01, -1.7908e-01,\n",
       "          -2.1990e-01,  1.9709e-01, -1.7846e-01, -1.1703e-01, -1.7569e-01,\n",
       "          -1.7366e-01,  9.7459e-02, -4.0722e-01, -3.6435e-02,  1.7903e-01,\n",
       "          -1.2723e-01,  6.7948e-02, -3.4229e-01,  2.2739e-01,  3.3538e-01,\n",
       "          -1.0467e-01, -3.7966e-01, -2.3884e-01,  4.7521e-02,  7.1481e-02,\n",
       "           3.8731e-01, -2.6195e-01, -1.0102e-01, -7.3811e-01, -2.0636e-01,\n",
       "           3.6787e-02,  1.0410e+00, -3.7965e-02, -3.0778e-01,  3.5156e-01,\n",
       "           3.9051e-01, -2.4726e-01, -3.4583e-01, -1.8699e-01,  7.4840e-01,\n",
       "          -2.3598e-01, -1.5465e-01,  2.2099e-01,  2.5510e-02,  6.2240e-01,\n",
       "           1.3154e-01,  2.2300e-01, -1.8512e-01,  4.7019e-01,  2.1848e-01,\n",
       "          -4.4102e-01, -1.5073e-01, -3.9303e-01, -3.2472e-01,  4.3754e-01,\n",
       "          -6.9776e-02,  2.7454e-01, -1.1790e-01,  8.1553e-02,  1.4253e-02,\n",
       "           5.4206e-02,  1.1645e+00, -2.0185e-01, -2.3960e-01,  4.7684e-01,\n",
       "           2.7079e-01,  3.4262e-01,  3.7917e-02, -1.3477e-01,  9.5921e-02,\n",
       "           4.9705e-01, -1.6706e-01, -4.9624e-01,  1.5664e-01,  3.6272e-01,\n",
       "           5.9108e-01, -8.7613e-01, -3.6622e-01,  2.0509e-01,  2.2296e-01,\n",
       "          -3.2121e-01,  1.9818e-01, -7.6228e-02,  3.2510e-01,  3.6266e-01,\n",
       "          -1.2771e+00,  1.7642e-01,  7.1262e-02,  1.5943e-01, -1.5269e-01,\n",
       "          -6.7715e-01, -2.5385e-01,  1.0595e-01, -6.2916e-02,  4.2076e-01,\n",
       "           4.4789e-01, -7.1509e-01,  1.2452e-01,  3.1511e-01, -3.7691e-01,\n",
       "           1.2320e-01, -2.7571e-01,  1.6643e-01,  5.2170e-01,  1.6129e-01,\n",
       "          -4.2505e-01,  2.5049e-01, -5.0826e-01, -6.1173e-01,  1.4561e-01,\n",
       "           6.9575e-01,  2.1324e-02, -1.1322e-01, -1.1804e-01, -2.9289e-01,\n",
       "          -1.9107e-01, -3.1790e-01, -3.4967e-01, -2.7542e-02,  1.4802e-01,\n",
       "           2.4431e-02, -3.6606e-02,  2.4884e-01,  1.1143e-01, -2.2300e-01,\n",
       "           5.8287e-01,  4.8747e-01,  7.4334e-01, -1.5062e-02,  1.1963e-02,\n",
       "          -2.2207e-01,  2.9970e-01, -4.7196e-02,  4.7552e-01, -3.4433e-01,\n",
       "          -2.0911e-01,  2.2006e-01,  3.6212e-02, -6.2294e-01, -1.0381e-01,\n",
       "          -3.5492e-01,  3.0044e-01,  1.1075e-02, -2.7609e-01,  4.8880e-01,\n",
       "           2.0553e-02,  1.8280e-01, -5.5884e-01,  4.6756e-01,  1.6596e-01,\n",
       "           2.2073e-02,  9.8054e-01,  1.1685e-01,  5.3145e-01,  5.8221e-01,\n",
       "           1.1828e-01, -8.1138e-01, -1.8763e-01, -7.2256e-02,  1.5035e-01,\n",
       "           1.1544e-02,  1.5872e-03,  1.5741e-01,  2.8459e-02,  9.3799e-01,\n",
       "           4.5776e-01, -2.2136e-01, -3.9686e-01, -3.7782e-01,  1.8491e-01,\n",
       "          -8.4548e-02, -3.3508e-01,  2.0442e-01, -2.2187e-01,  1.7388e-02,\n",
       "          -5.8935e-01, -5.5885e-01, -1.2302e-01,  2.7758e-01, -2.6880e-01,\n",
       "           5.4517e-01,  4.6664e-02,  5.2447e-03,  3.4141e-01, -2.9763e-02,\n",
       "           2.1521e-01,  6.7382e-01,  2.2993e-01, -2.9428e-01,  1.8031e-01,\n",
       "           2.1968e-01, -1.3524e-01,  2.0995e-01, -8.6762e-01,  3.6296e-01,\n",
       "          -5.1872e-02,  5.4038e-01,  4.7469e-01]], device='mps:0'),\n",
       " tensor([[-2.7791e-02, -7.5747e-01, -5.3074e-01, -5.1430e-01, -3.4417e-01,\n",
       "           7.2208e-02, -3.2733e-01,  1.9133e-01, -1.5003e-01,  5.7895e-01,\n",
       "           2.7665e-01,  1.8669e-01, -4.0441e-02, -1.9787e-02, -2.7559e-01,\n",
       "          -3.6770e-01, -5.5075e-01,  4.2714e-01,  4.8599e-01,  1.2291e-01,\n",
       "           6.0920e-01,  4.0016e-01, -5.7060e-01,  2.7172e-01, -1.0960e+00,\n",
       "           2.2592e-01, -9.0650e-02,  1.6102e-01,  2.6783e-01, -4.2947e-01,\n",
       "          -2.6593e-01,  3.3457e-01, -1.0081e-01,  1.1911e+00, -7.2573e-02,\n",
       "          -4.2319e-01,  1.2905e-01, -4.8449e-01,  2.2964e-01,  1.0587e-01,\n",
       "           3.3713e-01, -2.3616e-01,  3.0495e-01,  2.5426e-01,  2.2269e-01,\n",
       "          -1.0156e-01,  3.5441e-01, -6.7506e-01,  5.3763e-02, -5.4964e-01,\n",
       "          -2.7866e-01, -2.3157e-01,  3.8469e-01,  3.0486e-01, -1.7402e-01,\n",
       "          -1.7631e-01,  1.9710e-01,  2.8978e-01, -1.6215e-01,  1.9001e-01,\n",
       "          -3.1728e-01,  3.9028e-01, -2.9116e-02, -2.5337e-02, -4.3117e-01,\n",
       "           5.2338e-02, -1.0542e+00, -2.0097e-01, -6.1678e-01, -2.9287e-01,\n",
       "           6.1601e-02, -4.4671e-02,  2.6994e-02,  8.2943e-02,  6.2566e-01,\n",
       "          -2.9527e-01, -3.5686e-02, -6.8721e-02, -8.9378e-01,  3.5990e-02,\n",
       "           9.3805e-02,  2.8378e-01,  3.4715e-01, -1.8362e-01,  4.0523e-02,\n",
       "          -1.1761e-01,  5.2336e-01,  3.2885e-01,  5.5885e-02, -1.1072e-01,\n",
       "           2.6658e-01,  1.4680e-01,  3.1471e-01, -2.3564e-01,  2.1408e-01,\n",
       "          -1.6953e-01,  4.2040e-01, -7.0445e-02, -1.3569e-01, -1.7957e-01,\n",
       "           1.7683e-01,  2.7367e-01,  1.6921e-01,  1.8827e-01, -7.0935e-02,\n",
       "          -4.6949e-01,  6.4462e-03, -2.7295e-01, -9.2490e-01,  5.3292e-02,\n",
       "           2.5952e-02,  8.0243e-02,  3.5441e-01, -5.5720e-01, -4.9249e-01,\n",
       "           2.3824e-01,  3.1274e-01, -7.8117e-02, -2.4385e-01,  7.2875e-02,\n",
       "           1.2005e-01, -3.7322e-01,  3.0937e-01,  1.4655e-01,  3.8964e-01,\n",
       "          -2.4076e-01,  5.1062e-01, -1.8687e-01,  4.6773e-01,  1.7625e-01,\n",
       "          -2.8715e-02, -1.4186e-02,  2.4065e-02, -2.5766e-01, -7.0436e-02,\n",
       "          -3.2938e-01,  4.0965e-01, -2.1077e-01,  1.5229e-01, -2.7347e-01,\n",
       "          -1.9571e-02, -1.5879e-01, -4.2139e-01, -2.4379e-01,  5.2323e-01,\n",
       "          -2.3310e-01,  3.4454e-01, -2.5864e-01, -5.3971e-01, -5.4621e-01,\n",
       "          -4.6331e-01, -6.6221e-01,  6.5904e-04, -5.8988e-02,  1.5486e-02,\n",
       "          -1.4458e-01, -4.7299e-02,  3.4941e-01, -1.5764e-01, -1.3263e-01,\n",
       "           1.3229e+00, -1.6963e-01,  2.6547e-01, -5.2711e-01, -9.2709e-02,\n",
       "          -2.5951e-01,  3.5857e-01, -2.1281e-01, -5.0373e-02, -1.1031e-02,\n",
       "           1.5012e-01, -3.6325e-01,  3.2426e-01, -2.7495e-01,  4.9574e-02,\n",
       "          -3.8145e-01,  6.7830e-01,  1.0315e-01, -3.7603e-03, -2.1426e-01,\n",
       "           8.8470e-02, -1.8509e+00, -6.0244e-02, -4.9988e-02, -1.1984e-01,\n",
       "          -3.8304e-01,  1.1412e-01, -8.1008e-02,  4.7211e-01,  1.0523e-01,\n",
       "          -3.0391e-01,  1.4457e-02,  3.4290e-01,  1.7766e-01,  4.6617e-02,\n",
       "           3.7586e-01,  5.1268e-01,  5.2866e-01,  8.3422e-02, -1.2969e-01,\n",
       "          -2.2910e-01,  1.8840e-01, -1.4922e-01, -4.9168e-03,  1.3625e-01,\n",
       "           1.9163e-01, -1.7921e-02,  4.5103e-02,  4.5549e-01, -3.9944e-02,\n",
       "          -1.3034e-01, -3.0975e-01, -2.0571e-01, -1.5386e-01, -3.0506e-01,\n",
       "           6.6388e-01, -6.9598e-02,  1.5083e-01,  1.1394e-01, -7.6365e-02,\n",
       "          -1.9021e-01, -3.6627e-01,  7.4035e-02, -4.3048e-01, -1.0458e-01,\n",
       "           5.2612e-01, -2.4620e-01, -9.6266e-02, -1.5057e-01, -2.1628e-01,\n",
       "           4.6617e-01, -6.3064e-01,  6.9359e-01, -6.5753e-02, -1.5876e-01,\n",
       "          -6.3829e-02, -3.6947e-01,  8.8525e-02, -4.7900e-02, -2.1367e-01,\n",
       "          -6.6109e-01, -6.2912e-01, -9.3484e-01,  3.0931e-02,  2.0012e-01,\n",
       "           3.8940e-01, -2.0282e-01,  7.0611e-01,  4.1043e-01,  4.4392e-01,\n",
       "          -1.9033e-01, -2.5634e-01,  1.9540e-01,  2.4175e-02, -1.6470e-01,\n",
       "          -8.3407e-02, -3.6482e-02, -2.3338e-01,  5.4958e-01, -2.6909e-01,\n",
       "           3.4143e-02, -8.4658e-02,  1.4949e-01,  5.6272e-01, -2.6404e-02,\n",
       "           2.7543e-01,  3.6181e-02, -5.5097e-01, -2.6403e-02,  1.4938e-01,\n",
       "          -3.9414e-01,  1.3074e-01,  6.3249e-01, -2.9116e-01, -5.8419e-01,\n",
       "          -5.6450e-01,  1.3798e-02, -1.0222e-01,  2.9551e-01, -1.1019e+00,\n",
       "          -1.6877e-01,  2.8032e-01,  3.0136e-02,  1.8251e-01, -3.0887e-01,\n",
       "           2.9186e-01,  5.8483e-02,  3.7632e-02, -7.8447e-02, -5.1537e-01,\n",
       "          -2.3439e-01,  1.6469e-01, -3.6657e-01, -3.5963e-02,  1.6028e-01,\n",
       "           2.8092e-01, -9.8223e-02,  4.3126e-02,  2.2221e-01, -4.8483e-01,\n",
       "          -8.7845e-02, -3.3919e-02,  1.5657e-01,  6.3212e-02,  6.2429e-02,\n",
       "           7.6042e-02,  1.9293e-01,  7.3101e-01, -3.7007e-02, -6.0240e-01,\n",
       "          -2.4535e-01,  5.4258e-01,  7.7029e-02,  1.5324e-01, -1.7752e-01,\n",
       "          -1.4260e+00, -6.3469e-01,  3.8337e-02,  2.1822e-02, -9.6237e-02,\n",
       "           3.7186e-01,  1.4415e-01,  2.9691e-02,  1.0135e-01, -1.4892e-01,\n",
       "           2.7125e-01,  2.5571e-01,  4.1796e-01,  2.6634e-01,  1.2224e-01,\n",
       "          -9.5239e-02, -2.5680e-01, -3.4749e-01, -2.5056e-01,  5.0008e-02,\n",
       "           3.0551e-01,  2.9752e-01, -2.5921e-01, -9.4277e-02, -9.0702e-02,\n",
       "           7.3119e-01, -4.5527e-01, -5.6552e-01,  3.9878e-01, -5.2456e-01,\n",
       "          -2.7738e-01,  4.5605e-01, -3.7995e-01,  6.1826e-02, -3.0256e-02,\n",
       "           1.6101e-01, -7.3998e-02,  5.4701e-02,  5.9767e-02,  1.4680e-01,\n",
       "           1.8510e-01,  4.4459e-01, -9.6782e-02,  3.8301e-01,  4.8393e-01,\n",
       "          -3.7868e-01,  1.0631e-01, -3.1295e-01,  3.3607e-02,  3.3206e-02,\n",
       "           3.5477e-02,  1.0657e-01, -5.8454e-01, -7.1591e-02,  1.8413e-01,\n",
       "           3.9868e-02, -5.0830e-01, -1.9227e-01,  1.5691e-01, -1.5280e-01,\n",
       "          -1.9063e-02, -6.1864e-01,  7.4105e-02,  2.7357e-02, -1.0623e-01,\n",
       "           2.0196e-01,  3.6702e-02,  1.0747e-01, -2.9594e-01, -3.5062e-01,\n",
       "          -4.3860e-01,  1.3449e-01,  6.4194e-01, -1.6284e-02, -1.8554e-02,\n",
       "          -1.0969e-02, -1.8924e-01, -3.0452e-01,  4.8185e-01,  1.5484e-01,\n",
       "          -2.7278e-01,  1.5357e-01, -1.5461e-01,  4.3362e-01,  4.5208e-01,\n",
       "          -3.5378e-01, -1.2043e-01, -1.1214e-01,  3.6801e-01, -1.9150e-01,\n",
       "          -1.7609e-01,  2.8881e-01,  4.5279e-01,  1.9897e-01,  9.4702e-02,\n",
       "           2.5524e-01,  6.2194e-01, -7.1092e-01,  2.7741e-01, -5.0384e-02,\n",
       "           1.6210e-01, -3.1364e-01,  1.1810e-01, -4.5970e-01, -3.1796e-01,\n",
       "           2.0608e-01,  3.4692e-01,  1.5942e-01, -8.2525e-02, -1.1386e+01,\n",
       "           4.7410e-01,  7.1675e-01, -6.4620e-01, -1.2380e-02,  2.0990e-01,\n",
       "           1.4467e-01, -4.2321e-01,  1.2785e-01, -4.0986e-01, -4.8454e-01,\n",
       "           5.3751e-02, -3.1024e-01,  8.5046e-02, -2.1082e-01, -3.5012e-02,\n",
       "          -3.3293e-01, -8.8604e-01,  5.2616e-01, -2.3722e-01,  1.2282e-01,\n",
       "          -4.0554e-01,  2.2834e-01,  6.7441e-02, -3.7185e-01, -2.9844e-01,\n",
       "          -3.3378e-01, -4.8637e-01,  2.5874e-01,  1.4526e-01, -3.6388e-01,\n",
       "           1.6897e-01,  7.8214e-02,  1.5488e-01, -3.1408e-01, -5.7322e-01,\n",
       "           6.5890e-02, -1.4286e-01,  6.9077e-01, -2.3317e-01,  2.3521e-01,\n",
       "           1.5418e-01, -1.8939e-01,  4.0595e-01, -6.7756e-02,  5.4405e-01,\n",
       "           9.0099e-01,  4.9506e-01, -9.3677e-02, -3.1796e-01,  2.9697e-01,\n",
       "          -1.1296e+00,  2.3765e-01, -7.4326e-01,  3.7037e-02, -4.4715e-02,\n",
       "           5.2303e-01, -5.4682e-01, -1.6110e-02,  1.6453e-01, -1.3098e-01,\n",
       "           3.8056e-01,  1.5443e-02,  3.0641e-01, -3.8540e-01, -1.8694e-01,\n",
       "           5.1715e-01,  1.1540e-01,  3.9804e-01, -2.1344e-01, -2.7691e-02,\n",
       "          -3.5584e-01,  6.3533e-01,  4.1696e-01,  6.3335e-03,  1.0379e-01,\n",
       "           9.4310e-02,  1.5461e-01,  8.2262e-01,  1.5306e-01,  3.7272e-02,\n",
       "           7.5334e-02,  4.9919e-02, -2.0803e-02, -2.5454e-01, -1.9359e-01,\n",
       "          -3.8527e-02,  2.3219e-01,  6.4573e-01,  5.1229e-01, -2.5336e-01,\n",
       "          -1.8886e-01, -2.0242e-01, -5.9704e-02,  1.3338e-01,  5.6888e-01,\n",
       "          -7.2007e-01,  1.4785e-02,  2.0411e-01, -2.3545e-01, -2.6105e-01,\n",
       "          -9.1127e-02, -2.2821e-01, -6.1202e-01,  4.0102e-01, -3.4853e-03,\n",
       "          -3.2201e-01,  8.7693e-02,  5.6282e-01,  1.7925e-01, -2.9609e-01,\n",
       "           4.8297e-01, -1.4248e-01,  5.2177e-01, -7.0864e-02,  3.5673e-02,\n",
       "           3.0329e-01,  6.1367e-01, -2.1669e-01, -3.9304e-01, -6.3589e-02,\n",
       "          -7.9591e-02,  2.2539e-01,  3.5268e-03,  7.8083e-02, -1.6630e-02,\n",
       "          -3.5275e-01, -1.3786e-01,  3.3867e-02,  2.5058e-01,  2.6520e-01,\n",
       "           6.0789e-02,  3.0671e-01,  2.9447e-01, -3.1026e-01, -2.5075e-01,\n",
       "          -7.3840e-01, -2.4288e-01,  2.2993e-01, -1.9391e-01, -9.8460e-02,\n",
       "          -2.1162e-02, -3.4200e-02, -3.9679e-01,  3.1646e-01, -8.8932e-02,\n",
       "           1.4835e-01,  1.1151e-01, -4.4561e-01,  2.3351e-01, -3.3892e-02,\n",
       "          -3.1032e-01, -9.2286e-02, -3.7850e-01,  8.2453e-02,  3.6353e-02,\n",
       "          -4.8302e-01,  2.5437e-01, -2.5304e-01, -6.9363e-01, -3.3891e-01,\n",
       "           1.0299e-02,  1.8091e-02, -3.8864e-01,  1.0920e-01,  5.7217e-02,\n",
       "          -2.6965e-02, -5.7936e-01, -2.5618e-01, -5.4129e-03,  4.7410e-02,\n",
       "           3.7889e-01, -4.1326e-01,  5.3043e-02, -5.8918e-01,  3.3814e-01,\n",
       "           8.6381e-02,  8.8725e-01, -9.5259e-02, -4.3658e-01,  4.1371e-01,\n",
       "           6.8976e-01, -3.6204e-01,  2.3138e-01, -2.0207e-01,  1.7246e-01,\n",
       "          -2.2461e-01, -1.5982e-01,  4.5024e-01, -2.3929e-01,  5.3763e-01,\n",
       "           7.6361e-02,  3.0156e-01, -5.0755e-01, -3.4200e-02, -4.5787e-01,\n",
       "          -2.6191e-01, -1.0136e-01, -4.3868e-01, -2.1873e-01, -9.5628e-02,\n",
       "          -3.6356e-01, -2.6224e-01, -2.7189e-01,  1.6586e-02, -1.9311e-01,\n",
       "           3.5684e-01,  1.1473e+00,  3.5222e-02, -4.8536e-02,  3.1878e-01,\n",
       "           2.3119e-01, -2.1068e-01,  2.0884e-01, -7.6245e-02,  3.3463e-01,\n",
       "           2.9654e-01,  3.1331e-01, -4.5673e-01,  2.5522e-01, -8.5886e-02,\n",
       "           3.3382e-01, -8.9138e-01, -7.5951e-01,  1.8828e-02,  2.3505e-01,\n",
       "          -4.1163e-02,  7.9677e-02, -2.6933e-01,  6.0446e-01,  6.9945e-01,\n",
       "          -1.3167e+00, -6.0678e-02, -2.4393e-01,  1.2029e-01, -4.5133e-04,\n",
       "          -4.4085e-01, -3.7242e-01, -3.3575e-01, -2.8487e-01,  3.2123e-01,\n",
       "           7.5280e-01, -4.5114e-01,  1.6297e-02,  1.8465e-01, -1.1076e-01,\n",
       "           2.3514e-01, -5.0295e-03,  2.5102e-01,  4.3425e-01, -1.0414e-01,\n",
       "          -1.4909e-01,  4.2579e-01, -3.1190e-01, -4.2031e-01,  1.5833e-01,\n",
       "           7.0937e-01, -1.1853e-01,  1.3689e-01, -1.8297e-02, -2.6799e-01,\n",
       "          -7.1901e-02, -1.1732e-01, -1.9540e-01,  2.4573e-01,  3.1104e-01,\n",
       "           1.0016e-01, -3.2782e-01,  2.3126e-01,  2.1841e-01, -5.2982e-02,\n",
       "           5.0868e-02,  9.0378e-01,  7.3841e-01,  1.8846e-02, -7.4087e-02,\n",
       "           2.7607e-01,  2.4221e-01, -5.0505e-02,  5.1696e-01, -1.7273e-01,\n",
       "          -2.8848e-01,  1.9170e-01,  8.5894e-02, -2.9880e-01, -9.4752e-03,\n",
       "          -2.0048e-01,  1.8031e-01,  1.8487e-01, -1.6871e-01,  5.3942e-01,\n",
       "           3.3819e-01,  1.9990e-01, -4.1667e-01,  6.3733e-05,  2.5826e-01,\n",
       "          -2.8037e-01,  8.7311e-01,  1.6816e-01,  4.6258e-01, -3.0563e-01,\n",
       "           2.2482e-01, -6.0811e-01, -4.3120e-01,  1.4365e-01,  7.4790e-02,\n",
       "          -3.6079e-01,  1.8614e-01, -1.2540e-01, -1.5393e-01,  7.3765e-01,\n",
       "           5.8692e-01, -3.4385e-01, -5.9073e-01, -5.4749e-02,  3.2823e-02,\n",
       "          -2.9436e-02, -2.4520e-01,  7.4517e-02, -2.1959e-02,  1.6957e-01,\n",
       "          -7.0204e-01, -2.6400e-01,  1.2722e-01,  2.0311e-01, -4.2548e-01,\n",
       "           8.2082e-01,  7.0018e-02,  1.7358e-01,  7.3647e-01,  1.3118e-01,\n",
       "           1.7952e-01,  6.3050e-01,  6.9376e-02, -5.5670e-01,  1.2268e-01,\n",
       "           2.8105e-01, -1.1468e-01, -5.5720e-02, -6.3967e-01,  3.2919e-01,\n",
       "           8.0147e-02,  9.4187e-01,  7.0432e-01]], device='mps:0'),\n",
       " tensor([[-6.2616e-02, -6.2647e-01, -7.0276e-01, -7.1558e-01, -2.8144e-01,\n",
       "          -1.6163e-01, -8.7709e-01,  3.6268e-01,  1.2576e-01,  5.9786e-01,\n",
       "           2.8059e-01,  2.7990e-01,  1.2964e-01, -2.9739e-01, -6.7620e-01,\n",
       "           1.2485e-01,  1.2045e-01,  2.6275e-01,  1.6394e-01,  1.0986e-01,\n",
       "           6.6733e-01,  3.4572e-01,  7.2301e-02, -9.9001e-02, -5.2645e-01,\n",
       "           3.5221e-01, -1.0247e-01, -1.2800e-01,  4.8585e-01, -1.2739e-01,\n",
       "           1.0805e-02,  2.6421e-01, -1.5857e-01,  1.1057e+00,  4.6431e-02,\n",
       "          -8.7014e-02,  1.4985e-01, -1.7339e-01,  3.6943e-01,  4.8486e-02,\n",
       "           5.1489e-01,  8.5748e-03,  1.1385e-01,  2.9544e-01,  5.9498e-02,\n",
       "           2.1140e-01,  3.3738e-01, -7.3412e-01,  2.4225e-01, -5.0904e-01,\n",
       "          -1.4421e-01, -1.2879e-01,  4.3693e-01,  3.4752e-01, -6.1938e-01,\n",
       "          -1.5226e-01,  1.7658e-01,  1.7700e-01, -3.8094e-01,  5.7837e-01,\n",
       "           7.7456e-03,  5.2039e-01, -1.1219e-01,  5.6928e-02, -6.0641e-02,\n",
       "           2.2301e-01, -1.8694e-01, -4.7731e-01, -2.0693e-01, -3.3899e-01,\n",
       "          -2.3704e-01,  2.3273e-01,  4.6058e-02, -2.0383e-01,  4.8662e-01,\n",
       "          -4.1023e-01,  2.5189e-01, -3.6104e-01, -7.5358e-02,  7.7211e-04,\n",
       "           2.7676e-01,  3.5111e-01, -1.5597e-01,  1.2574e-02,  4.3904e-02,\n",
       "           5.4012e-02, -3.1553e-03,  1.7637e-01, -6.6285e-02,  3.7809e-01,\n",
       "          -1.7252e-01,  1.1085e-01, -3.4724e-01, -1.4165e-01,  3.0672e-01,\n",
       "           6.2280e-02, -5.4002e-02, -1.8348e-03,  2.5848e-01,  1.8095e-01,\n",
       "           1.3316e-01, -8.1850e-02,  1.2541e-02, -4.9147e-02, -1.3866e-01,\n",
       "          -3.7121e-01,  3.0219e-01, -3.5727e-01, -5.3572e-01, -8.9114e-02,\n",
       "          -2.2678e-01, -2.0834e-01,  2.4095e-01, -1.2899e-01, -5.8584e-02,\n",
       "           5.2857e-02,  1.4616e-02, -6.9177e-02,  7.7893e-02,  3.1706e-01,\n",
       "           3.4116e-01, -2.6142e-01, -3.7110e-02, -4.1103e-02,  4.0389e-01,\n",
       "          -8.3567e-02,  5.0486e-01, -6.3763e-02,  1.9561e-01,  1.9710e-01,\n",
       "           2.1314e-01, -7.2560e-02, -3.5530e-01, -1.6748e-01, -9.2310e-02,\n",
       "           2.4388e-01,  1.5179e-01, -2.9361e-01,  3.6513e-01, -3.1711e-02,\n",
       "          -4.2278e-03, -1.2962e-01, -6.7690e-01, -2.7977e-01, -1.7267e-01,\n",
       "          -2.9240e-01,  3.2325e-02, -1.3392e-01, -6.0883e-01, -1.2325e-01,\n",
       "          -8.8410e-01, -6.5719e-01,  3.4764e-01, -3.7706e-01, -7.2839e-02,\n",
       "          -2.5295e-06,  2.7332e-01, -3.7283e-02, -1.3345e-01,  9.2422e-02,\n",
       "           8.6093e-01, -3.7257e-01, -1.0155e-01, -5.7483e-01,  8.7619e-02,\n",
       "           4.3489e-02, -1.6390e-01, -1.8396e-01, -4.6625e-02,  1.1536e-01,\n",
       "           2.9032e-01, -3.0347e-01,  1.5016e-01, -2.8636e-01,  1.1897e-01,\n",
       "          -5.2865e-01,  3.8697e-01,  2.4944e-01,  2.2011e-01, -1.6624e-01,\n",
       "           1.8996e-01, -2.0789e+00, -1.4910e-02, -7.1562e-01,  1.4990e-01,\n",
       "          -3.4002e-02,  1.6094e-01,  2.5705e-01,  3.1807e-01, -8.5558e-02,\n",
       "           2.4227e-02,  1.2994e-01,  6.0720e-02,  1.7780e-01,  3.0290e-02,\n",
       "           8.5993e-02,  4.2734e-01,  1.0144e-01, -6.2154e-02,  4.6363e-02,\n",
       "          -3.3884e-01,  4.3572e-01,  2.0554e-01,  2.4188e-01, -7.8520e-03,\n",
       "           1.0108e-01, -3.4169e-01,  1.3350e-01,  3.5792e-01, -1.6305e-01,\n",
       "          -1.9152e-01,  1.4795e-01, -6.1009e-01,  6.4222e-02, -1.3318e-01,\n",
       "           2.4521e-01, -1.2892e-01,  3.6794e-03,  6.0353e-01,  2.9885e-01,\n",
       "           2.0611e-01,  2.3138e-01,  2.0087e-01, -3.8862e-01, -2.7582e-02,\n",
       "           2.7418e-01, -3.1822e-01, -2.7486e-02, -5.2360e-01, -2.9816e-01,\n",
       "           2.9255e-01, -6.7586e-01,  4.0643e-01,  3.6271e-01,  1.2730e-02,\n",
       "          -5.1865e-03, -3.9674e-01,  1.8170e-01, -3.2388e-01,  3.8983e-01,\n",
       "          -2.2252e-01, -4.8237e-01, -6.0014e-01,  1.9894e-01,  4.8443e-01,\n",
       "           3.2823e-01,  6.0846e-02,  6.4689e-01,  4.0682e-01,  4.8346e-01,\n",
       "          -1.7029e-01, -9.3965e-03,  4.3252e-01,  2.7174e-01, -3.5879e-01,\n",
       "          -5.5175e-02,  1.9570e-01, -2.8277e-01,  2.7704e-01, -1.2009e-01,\n",
       "          -4.1986e-01, -1.2058e-01,  4.4856e-01,  4.7218e-01, -5.5729e-01,\n",
       "           9.6198e-02,  6.4694e-01, -3.4297e-01, -4.4990e-01, -5.2557e-03,\n",
       "          -5.0208e-01,  2.0353e-01,  8.8648e-02, -1.9668e-01, -5.6403e-01,\n",
       "          -7.0264e-01, -1.1072e-01, -5.0530e-01,  1.2782e-01, -7.0494e-01,\n",
       "          -3.3147e-01, -1.3909e-02, -8.1837e-02,  7.3904e-02,  1.7344e-01,\n",
       "           2.1771e-01, -7.1470e-02, -5.1346e-02, -1.2863e-01, -5.5667e-01,\n",
       "          -3.9013e-02,  1.6707e-01, -3.7411e-01,  3.8257e-01,  1.4180e-01,\n",
       "           6.2359e-02,  1.9231e-01,  2.7552e-01, -3.5967e-01, -4.2561e-02,\n",
       "          -1.0399e-01,  6.3338e-02, -3.7767e-01, -1.8284e-01,  1.6221e-01,\n",
       "           2.5527e-01,  4.3982e-02,  5.5676e-01, -9.8989e-03, -4.9037e-01,\n",
       "           9.7279e-02,  6.7626e-01, -3.0749e-01,  1.1219e-01, -1.6722e-01,\n",
       "          -8.1230e-01, -1.7303e-01, -1.4474e-01, -5.0151e-01, -3.7506e-01,\n",
       "           3.2481e-01,  4.3004e-02, -3.4700e-02, -1.9535e-02, -4.7990e-01,\n",
       "           2.0226e-01,  4.4602e-01,  2.3309e-01,  1.8308e-01, -3.7078e-01,\n",
       "          -1.6060e-01, -1.5167e-01, -4.5093e-01, -2.3121e-01, -2.6221e-02,\n",
       "           3.0377e-01,  1.7798e-01,  6.0163e-02, -1.0477e-01, -1.4368e-01,\n",
       "           6.2727e-01, -2.8161e-01,  1.6625e-02,  2.0776e-01, -3.7056e-01,\n",
       "          -3.4593e-01,  5.0320e-01,  6.7809e-02,  2.2294e-01,  4.6225e-02,\n",
       "           3.6316e-01, -5.6646e-01, -6.2602e-01, -1.3417e-01,  8.9179e-01,\n",
       "           1.5994e-01,  3.4170e-01, -3.8552e-01,  1.1696e-01,  4.9943e-01,\n",
       "          -4.2539e-01,  4.7704e-01, -3.5382e-01, -1.0500e-01, -3.2598e-01,\n",
       "           4.0893e-01,  3.3302e-01, -2.0704e-01,  3.7979e-01,  2.0070e-01,\n",
       "          -1.3786e-01, -5.4466e-02, -2.3674e-01,  3.6240e-01,  4.9872e-01,\n",
       "          -5.3611e-03, -3.7018e-01,  2.2186e-01, -1.8560e-01, -1.2012e-01,\n",
       "          -1.0254e-01, -3.2722e-03,  1.2326e-01, -6.5918e-01, -2.9480e-01,\n",
       "          -1.7014e-01, -5.3133e-03,  7.3499e-01,  2.2859e-01, -3.5801e-01,\n",
       "           1.3965e-01, -3.2208e-01, -3.9333e-01,  3.1746e-01,  1.2561e-01,\n",
       "          -3.7691e-01,  4.0955e-01, -4.8433e-01,  7.0593e-02, -1.1285e-02,\n",
       "          -1.2369e-01,  2.7728e-02,  1.1395e-01,  3.9460e-01,  2.5117e-01,\n",
       "          -2.2488e-01, -4.0668e-02,  1.2723e-01, -1.0980e-01,  2.0994e-01,\n",
       "          -1.1511e-04,  4.9680e-01, -2.1882e-01, -2.9162e-01,  9.7461e-02,\n",
       "           4.0597e-01, -3.5028e-01,  1.8268e-01, -1.7362e-01, -5.4194e-01,\n",
       "           6.3471e-01,  1.0150e-01,  2.5641e-01,  1.3090e-01, -1.1699e+01,\n",
       "           7.1592e-02,  7.3831e-01, -1.9325e-01,  2.8363e-01,  4.0517e-01,\n",
       "           1.1602e-02, -3.7668e-01,  4.4858e-01, -1.3840e-01, -3.4024e-01,\n",
       "          -4.2958e-01, -3.8199e-01,  1.4233e-01, -1.5751e-01, -2.0805e-01,\n",
       "           4.1685e-01, -7.9191e-01,  5.9132e-02, -2.8132e-01,  5.7196e-01,\n",
       "          -3.3759e-01, -1.2832e-01,  3.5796e-01, -1.5764e-01, -2.8902e-01,\n",
       "           2.3761e-01,  1.7273e-02, -1.6691e-01,  3.2209e-02, -1.6032e-02,\n",
       "           1.7587e-01, -3.3271e-01, -1.5027e-01, -1.2007e-01, -4.6988e-01,\n",
       "          -2.7005e-01, -3.0493e-01,  1.2092e-01,  7.8455e-02, -1.7796e-01,\n",
       "           7.7184e-02, -1.7859e-01,  2.2969e-01,  4.8338e-01,  6.2378e-01,\n",
       "           1.3198e+00,  9.0670e-01, -1.8994e-02,  1.9499e-02,  1.7116e-01,\n",
       "          -5.5958e-01,  2.4145e-01, -3.6729e-01,  2.8099e-01, -1.5925e-01,\n",
       "           2.4387e-01, -5.4304e-01,  2.4829e-01, -2.2367e-01,  1.0939e-01,\n",
       "           5.3557e-01, -4.0237e-02,  3.0196e-01, -3.0008e-01, -2.3120e-01,\n",
       "           1.2109e-01,  3.4968e-02,  6.0281e-02, -2.2445e-01, -7.0501e-03,\n",
       "          -7.1388e-03,  4.6860e-01, -3.9338e-02, -1.0576e+00,  2.4495e-01,\n",
       "           3.2278e-02,  1.2890e-01,  2.8982e-01,  1.8912e-01,  2.4288e-02,\n",
       "           1.3797e-03,  3.3766e-01, -5.5161e-01, -4.2301e-01,  2.3988e-01,\n",
       "          -7.5664e-02,  2.2611e-01,  4.2581e-01,  2.5577e-01, -1.9563e-01,\n",
       "          -3.2047e-01, -8.0159e-02,  1.1424e-01,  2.6139e-01,  5.9414e-01,\n",
       "          -7.6424e-01, -7.0030e-03,  1.6680e-01, -1.9423e-01, -1.0671e-01,\n",
       "          -3.2879e-04, -2.6335e-01, -3.5521e-01,  3.4951e-01, -5.5845e-01,\n",
       "           3.8535e-02, -2.3141e-01, -3.2059e-01,  3.6518e-01, -2.9619e-01,\n",
       "           1.8127e-01,  4.9069e-01, -1.3729e-02, -1.0096e-01, -3.7150e-01,\n",
       "           1.7551e-01,  2.6098e-01, -6.2737e-01, -2.1799e-01,  2.9550e-02,\n",
       "          -1.9310e-01,  5.3267e-01,  3.0115e-02, -3.1969e-01,  4.2068e-01,\n",
       "          -7.5378e-01,  2.5273e-01,  2.7860e-01,  9.8125e-02,  5.6108e-01,\n",
       "           1.0236e-01,  3.8592e-01,  1.4133e-01, -2.6726e-01, -5.6057e-01,\n",
       "          -9.4410e-01, -6.6615e-02,  3.3471e-01,  5.1896e-02,  1.2819e-01,\n",
       "          -2.6217e-01, -2.9069e-02, -1.2689e-01,  5.5164e-01, -6.5168e-02,\n",
       "           3.8720e-01,  1.2669e-01, -2.9272e-01, -1.6031e-03, -9.4970e-03,\n",
       "          -4.2841e-02,  1.3770e-01, -9.0373e-02, -2.5524e-01, -6.4782e-01,\n",
       "          -2.4416e-01, -7.7096e-02,  2.8265e-01, -4.0933e-01, -1.4547e-01,\n",
       "           1.0118e-01,  2.1869e-01, -5.5722e-02,  5.9932e-01,  3.6780e-01,\n",
       "          -2.1750e-01, -3.3352e-01, -5.6900e-01, -4.1145e-03, -6.4206e-02,\n",
       "           4.2841e-01,  4.1005e-01, -2.3083e-01, -6.6280e-01, -4.0858e-01,\n",
       "          -2.2012e-01,  8.5329e-01, -1.4409e-01, -3.2051e-01,  9.6469e-02,\n",
       "           4.4245e-01, -4.5867e-02, -4.0926e-01,  2.0025e-02,  2.3056e-02,\n",
       "          -1.9329e-01, -3.1854e-01,  4.2301e-01, -3.8030e-02,  2.5466e-01,\n",
       "           3.3309e-01,  2.3452e-01, -1.8591e-01,  6.3817e-01,  1.5373e-01,\n",
       "          -4.2977e-01, -8.0507e-02, -4.8788e-01,  1.2446e-01,  1.6184e-01,\n",
       "          -8.3537e-02,  6.4485e-01, -4.9381e-02, -4.5344e-01,  1.3276e-01,\n",
       "           3.0587e-01,  1.1167e+00, -1.1363e-01,  3.2400e-01,  4.2685e-01,\n",
       "           5.8359e-02, -2.3759e-01,  1.2959e-01, -2.8071e-01,  1.7872e-01,\n",
       "           4.3174e-01, -5.8123e-01, -6.1045e-01, -8.2382e-02,  4.2457e-01,\n",
       "           6.9720e-01, -4.8775e-01, -6.0306e-01, -4.1329e-01,  6.5184e-01,\n",
       "           3.3971e-02,  5.7363e-01, -4.1938e-02,  1.5482e-01,  3.8577e-01,\n",
       "          -6.4881e-01,  1.6504e-02,  1.1204e-01,  1.9935e-02, -6.3568e-02,\n",
       "          -4.1715e-01, -2.3002e-01, -2.5846e-01, -3.4178e-01,  5.5237e-01,\n",
       "           2.0446e-01, -1.0322e+00,  1.7012e-01,  5.6138e-01, -1.0252e+00,\n",
       "          -8.2842e-02, -3.5042e-01, -6.8051e-02,  1.6212e-01,  5.3855e-02,\n",
       "          -4.1600e-01,  7.6217e-02, -5.0585e-01, -5.2586e-01,  1.2954e-01,\n",
       "           7.4824e-01, -1.6515e-01,  1.5522e-01, -1.1954e-02, -5.8556e-01,\n",
       "          -2.3896e-01, -1.2349e-01, -4.1819e-01,  1.7169e-01, -3.2041e-03,\n",
       "          -1.4074e-01, -2.9339e-01, -1.2374e-01,  3.9420e-01,  6.3078e-02,\n",
       "           2.7390e-01,  2.8186e-01,  4.7710e-01,  8.3079e-02, -3.8005e-01,\n",
       "          -7.2862e-02,  1.3310e-01, -3.3126e-03,  4.3775e-01, -2.6136e-01,\n",
       "          -5.2265e-01,  1.0978e-01, -1.8887e-01, -7.3482e-01, -1.1607e-02,\n",
       "          -5.0759e-01,  1.0867e-01, -2.4182e-01,  7.1465e-02,  3.9294e-01,\n",
       "           3.0331e-01, -5.3342e-02, -1.3184e-01,  2.4586e-01,  1.6124e-01,\n",
       "           7.8310e-02,  8.7854e-01,  4.2669e-01,  3.2882e-01, -1.7049e-01,\n",
       "           3.3070e-01, -6.7632e-01, -1.8839e-01, -6.9532e-02, -2.0193e-02,\n",
       "           3.4640e-02, -3.7705e-01,  2.7600e-01,  2.0283e-01,  5.4257e-01,\n",
       "           4.4060e-01, -2.2287e-01, -6.4283e-01, -2.1943e-01, -7.4912e-03,\n",
       "           1.2109e-01, -1.8831e-01,  4.5566e-02,  1.9784e-01,  2.4691e-01,\n",
       "          -7.0805e-01, -2.0705e-01,  2.4480e-02,  1.1950e-01, -2.6640e-01,\n",
       "           8.0735e-01, -1.2580e-01, -4.8619e-02, -1.0108e-01,  1.9369e-01,\n",
       "          -9.0291e-02,  3.7250e-01,  6.0236e-02, -5.2538e-01, -3.6914e-02,\n",
       "           3.0285e-01, -3.4393e-01,  3.0927e-01, -4.0999e-01, -4.4029e-02,\n",
       "           1.2167e-01,  4.0328e-01,  6.1774e-01]], device='mps:0'),\n",
       " tensor([[-3.1277e-01, -8.1765e-01, -7.8675e-01, -9.2829e-01, -3.0912e-01,\n",
       "          -2.6843e-02, -5.4915e-01,  5.9046e-01,  6.2958e-02,  4.7953e-01,\n",
       "           2.3160e-01, -1.4474e-01,  1.9234e-01, -6.2158e-01, -8.3536e-01,\n",
       "          -4.5126e-02, -7.7122e-02,  4.1625e-01,  6.5700e-01,  2.5457e-02,\n",
       "           6.5821e-01,  1.0012e-01, -6.5034e-02,  1.1765e-01, -3.3757e-01,\n",
       "           2.8379e-01, -7.6012e-02, -9.5163e-02,  2.4548e-01, -1.1102e+00,\n",
       "           1.9077e-01,  6.4230e-01, -1.3973e-01,  8.4030e-01, -1.1508e-01,\n",
       "          -5.7295e-02,  1.6146e-02, -5.3312e-01,  1.8913e-01, -2.8841e-01,\n",
       "           8.2416e-01,  3.9986e-02,  5.5866e-01, -2.4573e-01,  6.1504e-01,\n",
       "          -7.6680e-02,  2.3098e-01, -6.9784e-01,  7.6574e-02, -2.8193e-01,\n",
       "          -1.9769e-01, -5.6537e-02,  4.8610e-01,  6.0348e-01, -7.5519e-01,\n",
       "          -4.3490e-01,  1.2197e-01,  2.9996e-01, -5.7325e-01,  3.1535e-01,\n",
       "          -3.4342e-01,  1.8888e-01, -1.2490e-01, -4.6445e-02,  6.6445e-02,\n",
       "          -4.5886e-02, -3.2581e-01, -2.2738e-01, -3.1213e-01,  3.9674e-02,\n",
       "           1.5422e-01,  4.1724e-01, -2.5153e-01, -2.6315e-01,  5.4710e-01,\n",
       "          -1.4698e-01,  1.2158e-01,  2.5022e-02, -3.8817e-01,  1.8600e-02,\n",
       "          -2.8119e-01,  1.9108e-01, -3.9669e-01, -3.6559e-01, -7.6709e-03,\n",
       "          -8.6979e-02,  1.5231e-01,  1.6200e-01,  1.5674e-01, -3.3576e-01,\n",
       "           3.0137e-01,  2.0569e-03, -1.6732e-01,  2.2323e-01,  3.3745e-01,\n",
       "          -1.3297e-01,  3.6939e-01, -3.8469e-01,  1.3681e-01, -4.3229e-01,\n",
       "           4.1799e-02,  1.9198e-01, -4.7950e-02,  1.2809e-02, -7.7214e-01,\n",
       "          -7.3107e-01,  1.2865e-01, -2.9567e-02, -8.7981e-01, -2.0349e-01,\n",
       "          -2.1636e-01, -2.5038e-02,  1.7336e-01, -6.4409e-01, -1.9227e-01,\n",
       "           5.6568e-02, -2.2121e-01, -2.6665e-01, -1.6423e-01,  7.0397e-01,\n",
       "           1.1948e-01, -1.9474e-01,  7.7885e-02, -5.1522e-02,  1.6444e-01,\n",
       "          -2.9241e-01,  3.8458e-01, -2.6771e-01,  5.1034e-01,  1.1680e-01,\n",
       "          -2.1783e-01, -7.4575e-02, -5.1241e-01, -4.0803e-02,  5.8365e-02,\n",
       "           9.9553e-02,  3.8884e-01, -3.8259e-01,  8.9446e-02,  1.9868e-01,\n",
       "          -1.3459e-01, -2.6634e-01,  1.4133e-03,  1.9259e-01,  3.6179e-02,\n",
       "          -3.0910e-02,  9.9909e-02, -2.9255e-01, -2.3225e-01, -2.0005e-01,\n",
       "          -6.9523e-01, -8.8669e-01,  4.8729e-01, -4.5391e-01,  2.8924e-01,\n",
       "          -1.5282e-01,  7.0597e-02,  1.1364e-01,  7.6246e-02,  2.7272e-01,\n",
       "           8.9641e-01, -3.6326e-02,  2.6167e-01, -8.7124e-01, -1.4706e-01,\n",
       "          -3.1462e-02, -1.8133e-01, -1.7235e-01, -2.1349e-02, -1.6596e-01,\n",
       "           2.2603e-01, -2.2091e-01,  3.9463e-01, -7.1600e-01,  3.6045e-01,\n",
       "          -6.5252e-01,  3.8158e-01,  1.9494e-01, -1.4397e-01, -1.8038e-01,\n",
       "           3.1940e-01, -1.3834e+00,  2.3688e-01, -4.0394e-01,  9.2862e-02,\n",
       "          -6.1670e-01,  8.9086e-01,  1.0508e-01,  2.4744e-01,  1.6204e-01,\n",
       "           5.1022e-01, -2.8122e-01,  4.3315e-02,  5.5484e-01,  5.4156e-02,\n",
       "           4.6068e-01,  3.8942e-01,  4.9742e-02,  1.5685e-02,  7.9776e-02,\n",
       "          -7.1038e-01,  7.0864e-02, -1.8230e-02,  2.7971e-01,  3.0958e-01,\n",
       "           3.5920e-01, -2.2516e-01,  1.9470e-01, -1.2412e-01,  2.2354e-01,\n",
       "          -8.4929e-02, -6.5889e-02, -5.9038e-01, -2.1220e-01, -4.9047e-01,\n",
       "           2.9879e-01, -2.0580e-01, -4.0232e-01,  4.5105e-01,  1.6083e-01,\n",
       "          -9.8325e-02, -1.4739e-01,  1.0595e-01, -2.1218e-01,  7.3435e-02,\n",
       "           2.2152e-01, -3.9285e-01,  4.8395e-02, -1.4512e-01, -2.4840e-01,\n",
       "           1.0210e-01, -8.3278e-01,  5.7747e-01,  2.9455e-01, -2.9652e-01,\n",
       "          -5.3871e-01, -2.5473e-01,  9.6485e-02, -3.2083e-02, -2.5881e-01,\n",
       "          -3.7275e-01, -5.1092e-01, -4.6301e-01,  1.3298e-01, -1.3543e-01,\n",
       "           2.4994e-01, -3.5434e-02,  8.3131e-01,  4.6293e-01,  3.1407e-01,\n",
       "          -3.3373e-02, -2.6627e-01,  2.3030e-01, -2.4585e-01, -4.6488e-01,\n",
       "          -2.5860e-01, -1.2864e-01, -5.0870e-01,  2.2397e-01, -2.9672e-01,\n",
       "          -5.0623e-01,  6.4482e-02,  1.9096e-01,  2.5666e-01, -4.2751e-02,\n",
       "          -2.4388e-01,  9.2594e-02, -2.4917e-01, -3.5385e-01,  3.5596e-01,\n",
       "           1.3964e-01,  1.8974e-01,  2.5169e-02, -5.5958e-01, -5.9531e-01,\n",
       "          -5.7389e-01,  1.1464e-01, -5.3638e-01,  5.4403e-01, -4.6552e-01,\n",
       "          -2.0782e-01,  1.0246e-01, -1.4251e-01, -2.5505e-01, -1.5300e-01,\n",
       "           2.1213e-01, -1.5010e-02,  1.4784e-01, -3.4668e-01, -5.6048e-01,\n",
       "          -1.6959e-01,  2.8583e-01, -4.4844e-01,  1.0797e-01,  1.8824e-01,\n",
       "           3.6102e-01,  1.0574e-01,  2.1844e-01,  1.9897e-01, -3.3613e-01,\n",
       "          -2.9029e-01, -2.9696e-01, -4.3131e-02,  1.5862e-02,  2.0683e-01,\n",
       "          -7.5377e-02,  3.5212e-01,  4.7848e-01, -4.1914e-01, -6.0363e-01,\n",
       "           1.3238e-01,  4.9823e-01, -1.7710e-01, -1.7809e-01, -2.4784e-01,\n",
       "          -8.7520e-01, -2.5475e-01, -5.1827e-01,  8.1741e-02, -3.1857e-01,\n",
       "           2.0508e-01,  2.4792e-01,  1.1184e-01,  9.1486e-02, -4.0808e-01,\n",
       "          -3.4878e-02,  2.3834e-01,  2.6841e-01,  2.5444e-01, -2.5573e-01,\n",
       "          -7.5600e-02, -3.6710e-01, -1.4635e-01, -2.1597e-01, -9.3711e-02,\n",
       "           2.4354e-01,  3.2237e-01, -4.4551e-02, -1.8386e-01, -2.0617e-01,\n",
       "           4.6593e-01, -7.1115e-02, -3.9653e-01,  4.9744e-01, -3.3136e-01,\n",
       "          -4.1853e-01,  2.4576e-01, -3.4430e-01, -8.4198e-02,  4.2637e-01,\n",
       "           2.2248e-01, -2.3168e-01, -3.7317e-01, -2.1564e-02,  5.2056e-01,\n",
       "           6.6802e-01,  6.1537e-01, -4.7101e-01,  4.5705e-01,  1.8181e-01,\n",
       "          -2.1965e-01,  2.0810e-01, -3.5318e-01, -5.8077e-03, -2.7844e-01,\n",
       "           6.2632e-02,  3.5643e-01, -7.1869e-01,  1.6721e-01,  2.1461e-01,\n",
       "          -1.3838e-01,  1.9430e-01, -3.8620e-02,  3.0284e-01,  1.1464e-01,\n",
       "          -2.7239e-01, -4.0078e-01,  4.8636e-01, -9.0497e-02, -5.4654e-02,\n",
       "           2.7325e-01,  4.6809e-01, -1.6113e-01, -6.1697e-01, -2.7188e-01,\n",
       "          -2.5341e-01, -1.7912e-01,  1.0263e+00,  1.9470e-01,  1.1228e-01,\n",
       "           2.4115e-01,  3.8039e-02, -5.4463e-02,  3.3885e-01,  3.9573e-01,\n",
       "          -2.6284e-01, -4.8418e-02, -1.1261e-01,  1.9518e-01,  4.6590e-01,\n",
       "          -4.5606e-01, -3.8270e-01,  9.2707e-02, -4.4268e-01,  1.8036e-02,\n",
       "          -5.0276e-01,  8.6227e-02,  5.8472e-01, -2.0673e-01,  1.4841e-01,\n",
       "           9.2537e-02,  9.3781e-01, -4.7944e-01,  1.2609e-01, -4.1453e-02,\n",
       "           1.6249e-01, -1.1395e-03, -1.8509e-02, -2.0273e-02, -6.3946e-01,\n",
       "           7.5169e-01,  5.2601e-01, -1.0795e-02,  3.9353e-01, -1.1231e+01,\n",
       "          -8.2799e-02,  5.6685e-01, -3.3629e-01,  5.6062e-01, -7.9312e-02,\n",
       "           3.4310e-01, -1.3846e-01,  2.3830e-01, -5.0592e-01,  1.3440e-01,\n",
       "          -3.6496e-01,  1.3906e-01, -2.4858e-01, -3.4307e-01, -3.1929e-01,\n",
       "          -1.7711e-02, -7.1733e-01,  4.6480e-01, -9.0171e-02,  6.3683e-01,\n",
       "          -1.9295e-01,  4.4139e-02,  2.8335e-01, -8.0026e-02, -2.5491e-01,\n",
       "          -2.1581e-01, -3.7548e-01, -5.2375e-02,  2.8817e-01,  7.7898e-02,\n",
       "           5.0711e-02, -7.8969e-03,  1.7758e-01, -1.4380e-01, -6.2872e-01,\n",
       "          -2.1153e-01, -3.6581e-01, -2.9643e-02, -8.9068e-02, -2.2596e-01,\n",
       "          -1.7708e-01,  2.1914e-02,  3.8478e-01,  3.5703e-01,  7.2916e-01,\n",
       "           1.0035e+00,  2.6757e-01,  1.0147e-01, -2.9319e-01, -8.2124e-03,\n",
       "          -7.0106e-01,  1.5951e-01, -9.5963e-01,  2.0677e-01, -1.2414e-01,\n",
       "           5.3636e-01, -5.2416e-01,  2.5285e-01, -3.6590e-01, -1.7093e-01,\n",
       "           5.0717e-01, -1.0823e-01,  6.3880e-02, -2.6194e-01, -3.1761e-01,\n",
       "           4.2127e-01, -6.0306e-01,  4.8909e-02, -2.7592e-01, -1.1524e-01,\n",
       "          -4.1540e-01,  4.3822e-01,  4.1389e-01, -4.6767e-01, -1.3333e-01,\n",
       "          -1.3578e-01, -3.5736e-01,  1.3059e+00, -1.5503e-02,  2.1202e-01,\n",
       "           1.7973e-01,  3.1309e-01, -6.0312e-01, -8.2752e-01, -2.2010e-01,\n",
       "           4.5707e-02,  1.4803e-01,  1.6564e-01,  6.2880e-01, -4.0211e-02,\n",
       "          -3.9680e-01,  4.0423e-02,  3.9486e-01,  3.0246e-01,  5.3777e-01,\n",
       "          -8.1350e-01,  3.0008e-01,  1.4610e-02, -1.4601e-01, -3.6772e-01,\n",
       "           1.6169e-01, -2.3086e-01, -3.8956e-01,  2.7331e-01,  2.2747e-01,\n",
       "           2.4331e-01,  2.8317e-01, -1.9094e-02,  4.3357e-01, -3.9433e-01,\n",
       "           3.3308e-01,  5.2255e-02,  2.3098e-01, -6.3662e-01,  8.1904e-02,\n",
       "           5.8523e-01,  5.4617e-01, -6.0024e-02, -2.2600e-01, -3.4586e-01,\n",
       "           4.0098e-01,  2.5896e-01, -2.0244e-01, -5.1861e-01,  1.5651e-01,\n",
       "          -6.3354e-01, -1.2262e-01,  2.2055e-01,  3.3569e-01,  5.7886e-01,\n",
       "           1.4744e-01,  1.7942e-01, -3.8780e-02, -1.9293e-01, -6.0106e-01,\n",
       "          -5.8665e-01,  5.7788e-02,  3.4150e-01, -1.7771e-01,  2.6929e-01,\n",
       "           1.2960e-02, -9.0936e-02, -3.0172e-01,  4.8054e-01,  2.1461e-01,\n",
       "           5.0475e-01,  1.3950e-01, -7.8181e-02,  1.0385e-01, -4.3411e-01,\n",
       "          -1.4861e-01,  1.1281e-01,  6.6470e-02, -2.5911e-01, -3.2800e-01,\n",
       "          -1.9160e-01, -1.3327e-01, -1.7859e-01, -1.5692e-01, -3.4679e-02,\n",
       "          -8.9137e-02, -9.8149e-02, -7.1678e-02,  3.6193e-01,  1.5084e-01,\n",
       "           2.3321e-02, -2.6274e-01, -1.0461e-01, -2.1590e-01,  8.3339e-02,\n",
       "           3.8580e-01, -4.8376e-02, -4.5682e-01, -2.4061e-01, -7.8442e-02,\n",
       "           2.6830e-02,  1.2271e+00, -2.2949e-01, -7.0127e-01,  2.4651e-01,\n",
       "           2.0158e-01,  1.5512e-01,  2.3941e-01, -1.6654e-01,  2.3156e-01,\n",
       "          -5.3010e-01,  3.3307e-02,  3.6495e-01,  8.5679e-02,  3.7131e-01,\n",
       "           3.8363e-01,  3.8752e-02, -3.2286e-01,  1.1580e+00,  1.1134e-02,\n",
       "          -4.3539e-01, -3.0508e-01, -5.0743e-01, -1.9106e-01,  4.0813e-01,\n",
       "          -1.0745e-01,  2.0454e-01, -2.5837e-02,  1.9611e-01,  4.1180e-01,\n",
       "           1.2875e-01,  9.8244e-01, -1.5758e-01,  1.2218e-01,  5.0160e-01,\n",
       "           2.3177e-01, -1.5453e-01,  2.5084e-01,  2.7724e-01,  1.5313e-01,\n",
       "           2.3875e-01, -2.5518e-01, -3.3225e-01, -6.7178e-02,  3.4269e-01,\n",
       "           4.0544e-01, -6.9295e-01, -2.9738e-01, -1.5370e-01,  5.1131e-01,\n",
       "           1.4733e-01,  5.1412e-01, -2.0618e-01,  4.5406e-01,  9.6169e-01,\n",
       "          -1.1868e+00, -1.1669e-01, -2.2752e-02,  5.2636e-01, -1.1658e-01,\n",
       "          -1.7329e-01,  1.4383e-01,  1.0116e-02, -2.9441e-01,  8.1929e-02,\n",
       "           4.5200e-01, -9.3458e-01,  2.1961e-01,  5.6323e-01, -8.3371e-01,\n",
       "          -1.0950e-01, -1.1786e-01,  4.6318e-01,  1.5961e-01,  1.2146e-01,\n",
       "           1.3420e-01, -1.1980e-01,  1.3127e-01, -6.9132e-01,  2.9462e-01,\n",
       "           5.7352e-01,  5.4046e-02,  1.9922e-01,  1.4790e-01, -1.4738e-01,\n",
       "          -1.2821e-01, -3.5142e-01, -2.5477e-01,  9.8025e-02, -4.8513e-02,\n",
       "           8.1793e-02, -4.4941e-01,  1.7753e-01,  3.9250e-01, -4.7586e-01,\n",
       "           2.4361e-01,  6.3974e-01,  6.5597e-01, -3.0338e-02, -9.6304e-02,\n",
       "           9.9444e-02,  1.9920e-01,  6.5217e-02,  5.0328e-01, -8.6927e-02,\n",
       "          -2.7155e-01,  6.9034e-02,  2.1187e-01, -2.5872e-01,  2.1388e-01,\n",
       "          -4.7262e-01,  1.9244e-01, -2.1823e-01, -1.7387e-01,  4.4845e-01,\n",
       "           1.3136e-02,  4.1882e-01, -7.5546e-01,  1.5900e-01,  1.1572e-01,\n",
       "           1.7012e-01,  6.4528e-01,  2.8922e-01,  3.0974e-01,  4.8786e-02,\n",
       "           4.1070e-02, -4.2988e-01, -4.0735e-01,  3.2082e-01,  3.4582e-02,\n",
       "          -4.5969e-01, -1.4713e-01, -3.5809e-02,  6.4528e-01,  7.3049e-01,\n",
       "           6.7527e-01,  2.2773e-02, -6.5807e-01, -4.7754e-01, -1.3736e-01,\n",
       "           2.7910e-03, -4.4993e-01,  1.4194e-01, -1.3685e-01,  1.9805e-01,\n",
       "          -6.0786e-01, -6.6149e-01, -8.2228e-02,  4.3641e-02, -4.6873e-01,\n",
       "           5.6773e-01, -1.7996e-01,  5.0517e-01,  1.4561e-01,  4.1485e-01,\n",
       "           5.3772e-02,  6.9230e-01,  1.9095e-01, -5.0441e-01, -1.7062e-02,\n",
       "           4.7248e-01, -7.4181e-02,  2.0249e-01, -3.1354e-01,  8.6693e-02,\n",
       "           5.9158e-02,  7.7016e-01,  6.7568e-01]], device='mps:0'),\n",
       " tensor([[-6.1750e-01, -4.2829e-01, -2.3934e-01, -9.4116e-01, -3.4548e-01,\n",
       "          -6.5496e-02, -6.3874e-01,  9.7828e-01, -9.5892e-02,  4.0096e-01,\n",
       "           1.4919e-01, -5.1603e-03,  2.0980e-01,  1.3680e-01, -6.9506e-01,\n",
       "           1.6740e-01,  4.6094e-01,  2.1159e-01,  3.5951e-01, -4.4329e-02,\n",
       "           7.9832e-01,  3.4245e-01,  1.8019e-01, -2.5205e-01, -4.5876e-01,\n",
       "           5.8994e-01, -2.4628e-01, -9.2014e-02,  2.0509e-01, -8.1949e-01,\n",
       "          -1.3586e-01, -4.9757e-02, -3.6281e-01,  8.7602e-01,  1.1948e-01,\n",
       "          -2.7310e-01, -1.5276e-01, -2.7996e-01,  2.3494e-01, -6.3637e-02,\n",
       "           3.7814e-01,  2.6592e-02, -2.9347e-01,  3.1386e-01, -1.6421e-01,\n",
       "          -1.4057e-01,  1.4047e-02, -7.8429e-01,  1.3666e-02,  1.8955e-01,\n",
       "           5.8851e-02, -4.7205e-02,  1.9231e-01,  3.7030e-01, -1.0617e+00,\n",
       "          -2.6348e-01,  2.7603e-01,  2.9849e-01, -1.7932e-01, -5.7784e-02,\n",
       "          -1.9989e-01,  9.0201e-01,  1.7949e-01, -2.7822e-01,  1.7824e-01,\n",
       "          -4.4899e-01, -5.4601e-01, -6.0781e-01, -1.0682e-01, -1.1967e-01,\n",
       "          -1.0002e-01,  9.7528e-02,  1.4009e-01, -2.9385e-01,  2.3170e-01,\n",
       "           2.1813e-01,  4.7024e-02, -2.7652e-02, -4.3158e-01, -2.6546e-01,\n",
       "           3.6947e-01,  3.4425e-02, -1.4730e-02,  2.0326e-01, -5.8744e-01,\n",
       "          -2.0226e-01,  3.7959e-01,  1.5703e-01,  2.5914e-01,  2.2662e-03,\n",
       "           5.9967e-01,  1.2024e-02,  8.0339e-02, -4.5369e-01,  5.8427e-01,\n",
       "          -5.4955e-01,  1.0329e-01, -1.6007e-01,  4.9532e-02, -5.2088e-01,\n",
       "          -9.8735e-02,  2.4494e-01, -1.8085e-01,  1.6053e-01, -4.6592e-01,\n",
       "          -3.2983e-02,  2.4514e-01,  3.0468e-01, -4.9183e-01,  4.2685e-02,\n",
       "          -6.5134e-03, -2.9791e-01,  7.3694e-02, -6.6407e-01, -8.5795e-01,\n",
       "           5.8354e-01,  1.8428e-02, -4.5240e-01, -2.3930e-01,  7.4443e-01,\n",
       "           2.9738e-01, -2.3201e-01, -5.3072e-02, -3.5548e-01,  2.0259e-01,\n",
       "          -1.8518e-01,  2.5689e-01, -2.7018e-01,  3.8439e-01,  5.5026e-01,\n",
       "           3.1581e-01,  2.8489e-01, -1.8327e-01,  2.7637e-01,  1.4465e-01,\n",
       "           3.6947e-02, -6.6614e-02, -1.5929e-01,  1.5180e-01,  1.2041e-01,\n",
       "          -2.9580e-01, -3.2999e-01,  5.4746e-03,  1.1182e-01, -3.3279e-01,\n",
       "          -1.0202e-01,  1.8680e-01,  3.4808e-01, -1.4092e-01, -2.5403e-01,\n",
       "          -7.8984e-01, -4.2328e-01,  1.0217e-01, -1.8560e-01, -1.7135e-01,\n",
       "          -4.1911e-01,  3.8121e-01, -6.6660e-02,  4.7638e-01,  3.4189e-01,\n",
       "           6.9978e-01, -1.5311e-01,  5.9776e-02, -5.6412e-01,  3.5197e-01,\n",
       "           3.8270e-02, -6.1110e-01, -1.7247e-01,  1.5680e-01, -4.1813e-04,\n",
       "           6.8195e-01,  7.0979e-02,  4.0124e-01,  1.2070e-01, -3.2553e-01,\n",
       "           8.2089e-03,  1.9890e-01, -1.2864e-01, -1.4297e-01,  7.3766e-01,\n",
       "          -5.5188e-01, -2.3029e+00,  1.0721e-01, -3.5844e-01, -3.6529e-01,\n",
       "          -1.8135e-01,  3.7255e-01,  3.3772e-02,  4.9966e-01,  1.5486e-01,\n",
       "           3.7598e-02,  1.6729e-01, -2.8025e-02,  3.6851e-01, -8.4399e-02,\n",
       "           2.6751e-01,  3.3341e-01,  1.2988e-01, -2.3298e-01, -1.0956e-01,\n",
       "          -3.9227e-01,  2.4031e-01, -9.8916e-02,  8.6010e-01,  2.3260e-01,\n",
       "           2.0943e-01, -4.3657e-03, -4.7262e-02,  4.8964e-01,  4.3099e-01,\n",
       "          -6.6541e-02, -1.6585e-01, -4.9522e-01,  4.6269e-03, -4.1279e-02,\n",
       "          -1.2963e-02, -4.5611e-01,  1.3753e-01,  4.2841e-01, -1.2416e-01,\n",
       "          -1.8354e-01,  1.5525e-01, -4.7715e-02, -3.1517e-01, -2.6492e-02,\n",
       "           6.8453e-01,  1.9080e-01, -2.2994e-01,  1.2377e-01, -1.1896e-01,\n",
       "           3.7006e-02, -4.6975e-01,  7.0044e-01,  6.1147e-01,  1.5585e-01,\n",
       "           1.6214e-02, -6.9943e-01,  2.3699e-01, -3.1574e-01, -4.7647e-03,\n",
       "          -1.5027e-01, -1.5731e-01, -4.2986e-01, -5.7285e-02, -5.5854e-02,\n",
       "           4.0115e-01,  7.9943e-02,  4.6653e-01,  1.1837e+00,  2.7973e-01,\n",
       "           1.9184e-02,  5.6097e-01,  2.0801e-01,  4.2492e-01, -4.0180e-01,\n",
       "           3.3082e-01,  8.8825e-02, -6.3896e-01,  2.7188e-01, -4.1593e-01,\n",
       "           4.4963e-02, -4.1432e-02, -3.7926e-01,  1.4208e-01, -2.1276e-02,\n",
       "           3.4237e-01,  3.3517e-01,  1.5186e-01,  1.6838e-01,  2.9411e-01,\n",
       "          -3.1322e-01,  5.9715e-01, -6.0809e-02, -1.1127e-01, -6.3492e-01,\n",
       "          -7.6732e-01,  9.5206e-02,  4.4743e-01, -1.2880e-01, -6.1279e-01,\n",
       "           1.0462e-01, -2.3866e-01, -1.2257e-01, -3.5064e-02,  2.1191e-01,\n",
       "           2.7455e-01,  5.0687e-02, -9.5436e-02,  6.4791e-02, -4.6950e-01,\n",
       "          -4.9277e-01,  4.9140e-02, -1.5758e-01, -1.3063e-01,  5.3918e-01,\n",
       "          -7.9415e-02,  1.1575e-01,  5.7221e-01,  1.9799e-02, -4.5976e-01,\n",
       "          -2.5640e-01,  1.4802e-01,  2.8667e-01, -7.2082e-02,  2.1981e-01,\n",
       "           3.4119e-01,  2.6467e-02,  5.1649e-01,  1.4823e-01, -2.1046e-01,\n",
       "          -2.7239e-01,  5.1173e-01, -1.5755e-01,  6.1403e-02, -1.2527e-01,\n",
       "          -8.7207e-01, -5.6150e-02, -7.8413e-02, -1.6833e-01, -5.4440e-01,\n",
       "           2.6566e-01,  1.4351e-01,  1.0561e-01,  2.2350e-01, -1.5984e-01,\n",
       "           7.0123e-01,  6.5285e-01, -3.3100e-01, -1.6943e-01, -4.0104e-01,\n",
       "           9.3922e-02, -6.8374e-02, -2.9329e-01, -1.4009e-02,  9.5368e-03,\n",
       "          -2.2765e-01, -1.2516e-01, -1.0805e-01,  2.8921e-01, -1.8876e-01,\n",
       "           5.2862e-01,  1.2189e-01,  8.3649e-02,  4.3873e-01,  2.0612e-02,\n",
       "           1.3071e-01,  3.6515e-01, -4.2117e-01,  1.2870e-01,  4.0424e-01,\n",
       "           4.0578e-01, -1.0551e+00, -8.3291e-01, -1.4936e-01,  3.6111e-01,\n",
       "           2.2860e-01,  2.5189e-01,  1.6999e-02, -7.2215e-03,  8.9455e-02,\n",
       "          -2.3402e-01,  3.5135e-01, -6.6415e-02, -5.7412e-01,  3.0561e-01,\n",
       "           2.3567e-01,  5.3263e-01, -2.7377e-01, -5.0579e-01, -6.5233e-02,\n",
       "           2.4616e-01, -3.6381e-01,  1.1392e-01,  2.3480e-01,  4.5466e-01,\n",
       "          -3.6015e-01, -5.6164e-01, -7.0282e-02, -7.3986e-01, -1.3578e-01,\n",
       "           3.2483e-01, -2.1437e-01,  2.7178e-02, -2.4758e-01,  1.5231e-01,\n",
       "          -1.7925e-02, -1.8879e-01,  6.0757e-01,  2.0522e-01, -2.5855e-01,\n",
       "           1.1981e-01, -2.3892e-01, -1.9475e-01,  5.0091e-01, -5.0226e-01,\n",
       "          -3.0348e-01, -5.0960e-02, -1.5349e-01, -3.2418e-01,  3.2181e-01,\n",
       "          -4.1389e-01, -1.4861e-01,  1.6748e-01, -6.4736e-02, -1.3068e-01,\n",
       "          -3.5233e-01,  1.4268e-01, -1.2707e-01, -4.9883e-02, -1.7654e-02,\n",
       "          -1.3353e-01,  6.0900e-02,  1.3308e-01,  6.4538e-02,  1.4388e-01,\n",
       "           5.3696e-01, -1.1464e-01,  2.7582e-03, -2.5900e-01, -2.2469e-01,\n",
       "           5.2498e-01, -9.4973e-02, -3.3117e-01,  5.4928e-01, -1.1156e+01,\n",
       "           1.0386e-01,  7.2631e-01, -5.5808e-01,  5.5128e-02, -2.8539e-01,\n",
       "           2.1085e-01, -3.0009e-01,  7.9101e-02,  2.4967e-01, -2.8460e-01,\n",
       "          -1.2277e-01, -2.7364e-01, -2.3193e-01, -4.0872e-01, -3.9742e-01,\n",
       "          -4.7229e-01, -5.6789e-01,  1.1772e-01, -1.7663e-01, -1.0996e-01,\n",
       "          -5.7036e-01, -1.9379e-01,  2.9696e-01, -4.9559e-03, -5.1069e-01,\n",
       "           5.9541e-02, -2.2948e-01,  2.6399e-01,  4.6126e-02,  2.2045e-01,\n",
       "          -1.9949e-02,  1.4108e-01,  2.0669e-01, -4.4946e-02, -4.5100e-01,\n",
       "          -4.4853e-01,  2.2129e-01, -1.5955e-01, -2.2267e-01, -5.4017e-02,\n",
       "          -3.3838e-01, -2.7689e-01,  1.3447e-01,  3.7594e-01,  6.2596e-01,\n",
       "           1.1595e+00,  5.6208e-01,  2.9245e-01, -1.0625e-01, -8.5438e-02,\n",
       "          -1.0762e+00,  5.5900e-01, -3.7318e-01,  8.1787e-02, -2.5119e-01,\n",
       "           6.6201e-01, -4.9906e-01,  5.4685e-01, -4.0754e-01, -2.5500e-01,\n",
       "           3.7179e-01,  3.0194e-01,  7.1104e-03,  2.8239e-02, -2.7432e-01,\n",
       "           4.0257e-01, -2.6094e-01, -3.6915e-02, -3.7819e-01,  4.0081e-01,\n",
       "          -3.9472e-01,  5.0770e-01, -5.6534e-02,  1.2945e-01,  1.8567e-01,\n",
       "          -3.5001e-01,  1.6610e-01,  1.0337e+00,  1.7414e-01,  4.1071e-01,\n",
       "           1.7420e-01,  5.1160e-01,  5.3064e-02, -8.8958e-01,  1.3691e-01,\n",
       "           2.0502e-01,  2.0789e-01, -6.7096e-02,  5.7131e-01, -2.5517e-02,\n",
       "          -6.5181e-01, -2.3094e-01, -3.2169e-01,  4.3108e-01,  6.5352e-01,\n",
       "          -2.2597e-01, -4.4343e-01,  2.3894e-01, -7.9776e-03, -5.3395e-01,\n",
       "           9.0074e-02, -2.2320e-01, -5.2792e-01,  8.7275e-02, -5.8914e-01,\n",
       "           1.4985e-01, -1.2428e-01,  5.4238e-02,  6.6887e-01, -4.0159e-01,\n",
       "          -1.6113e-01,  3.6602e-02, -4.4532e-02, -4.1270e-01, -4.9520e-01,\n",
       "          -7.7083e-02,  4.1073e-01, -1.9776e-01, -2.7976e-01,  7.0636e-02,\n",
       "          -3.0673e-01,  2.2584e-01, -1.7377e-01, -6.4935e-01,  3.2879e-01,\n",
       "          -6.0415e-01,  2.0578e-01,  2.5360e-01, -2.2507e-01,  4.9017e-01,\n",
       "           1.1337e-01, -8.1817e-02,  1.9397e-01, -9.9369e-02, -7.8554e-01,\n",
       "          -3.2573e-01, -2.7299e-02,  3.2662e-01, -4.2844e-01,  1.5776e-01,\n",
       "           1.5022e-01, -4.5752e-02, -2.2574e-01,  2.7904e-01,  1.0174e-01,\n",
       "           3.7762e-01,  3.9846e-01, -5.5006e-01,  1.1727e-01, -2.9867e-01,\n",
       "          -2.9262e-02, -6.1174e-02,  1.7952e-01,  2.3654e-01, -7.0764e-01,\n",
       "          -1.5224e-01, -2.6434e-01, -3.0965e-01, -2.1650e-01, -2.5120e-01,\n",
       "           2.0680e-01,  1.4038e-01, -1.5897e-02, -1.1308e-01,  5.8077e-01,\n",
       "           4.0814e-02, -1.1167e-01, -4.5997e-01,  3.1443e-01,  4.3178e-01,\n",
       "           2.1794e-01,  7.5875e-02, -1.2830e-01, -5.5532e-01, -1.1922e-02,\n",
       "          -4.3644e-01,  9.1081e-01, -7.3906e-01, -7.1970e-01,  6.6086e-02,\n",
       "           1.4951e-02, -8.8259e-04, -2.3450e-01, -5.0265e-01,  8.1298e-02,\n",
       "          -2.1017e-01, -1.4891e-01,  6.1550e-03, -1.0607e-02,  1.6299e-01,\n",
       "           4.0742e-01,  3.6838e-01,  1.5951e-01, -3.3938e-01, -4.4081e-01,\n",
       "          -6.0244e-01,  1.4567e-02, -1.6795e-01,  2.9330e-01, -2.6752e-01,\n",
       "          -3.2661e-01,  1.6043e-02, -9.1892e-03, -2.2272e-01,  5.2876e-01,\n",
       "           1.2427e-01,  1.6396e+00, -3.7287e-01, -4.4551e-01,  2.9018e-01,\n",
       "           7.2306e-02, -2.7073e-02, -2.8130e-01,  2.5074e-01,  9.5447e-02,\n",
       "           4.0194e-02, -3.6304e-01,  1.5023e-02, -2.0558e-02,  3.2773e-01,\n",
       "           3.2056e-01, -1.0703e+00, -2.0238e-01, -1.7993e-01,  5.5664e-01,\n",
       "           3.2388e-02,  6.3185e-01, -2.2305e-01,  3.0577e-01,  4.6106e-01,\n",
       "          -8.6667e-01, -2.3000e-01, -1.6129e-01,  2.7295e-01, -1.0583e-01,\n",
       "          -9.8735e-02, -1.6178e-01, -3.7165e-02, -1.6936e-01,  3.9538e-01,\n",
       "           7.4751e-01, -7.8426e-01,  7.4081e-01,  7.0467e-01, -9.8234e-02,\n",
       "           5.6815e-01,  9.4187e-02,  3.2641e-01, -1.4659e-01,  2.2669e-02,\n",
       "           2.6200e-01,  1.8542e-01, -6.3165e-01, -2.3962e-01,  2.4473e-01,\n",
       "           4.0158e-01, -1.1943e-01,  1.2108e-01,  7.9292e-02,  3.0960e-02,\n",
       "          -3.7691e-01, -1.0145e-01, -2.4552e-01, -3.9904e-01,  1.1356e-01,\n",
       "           1.4603e-02, -1.5001e-01, -8.6997e-02,  4.7709e-01, -1.7580e-01,\n",
       "           7.5067e-01,  4.2910e-01,  3.7145e-01, -3.5103e-01,  8.9950e-04,\n",
       "          -5.7565e-02,  6.0671e-01, -3.1670e-01,  7.7385e-01,  6.3337e-02,\n",
       "          -2.3140e-01,  2.7251e-01,  6.0547e-03,  1.7829e-01,  2.9892e-01,\n",
       "          -2.3519e-01,  1.2702e-01, -3.0718e-01, -2.3711e-01,  2.8330e-01,\n",
       "          -1.8640e-01,  1.1542e-01, -7.2270e-01, -6.3602e-03,  3.9857e-01,\n",
       "          -3.2051e-01,  4.9873e-01, -2.3074e-01,  6.1026e-01, -4.3509e-01,\n",
       "           3.2669e-02, -4.7470e-01, -4.3275e-01, -2.8508e-01,  5.1854e-02,\n",
       "          -8.6027e-01, -1.9453e-01,  1.7684e-01,  3.7405e-02,  2.1707e-01,\n",
       "           7.5155e-01, -3.3794e-01, -2.4771e-01, -2.0386e-01,  4.0190e-02,\n",
       "           1.8299e-01, -8.2071e-02, -3.5586e-01,  2.1161e-01, -3.5607e-01,\n",
       "          -1.0520e+00, -2.3117e-01,  8.9990e-02,  3.8419e-01, -3.5853e-01,\n",
       "           1.1570e+00,  2.5132e-02, -2.0226e-01,  6.7253e-01,  2.0808e-01,\n",
       "           6.4157e-01,  6.0323e-01,  1.3628e-03, -4.1259e-01, -5.4109e-02,\n",
       "          -1.5877e-01, -3.2881e-01,  1.5757e-01, -1.8346e-01,  5.0556e-01,\n",
       "           5.4802e-01, -2.0806e-02,  5.4756e-01]], device='mps:0'),\n",
       " tensor([[-5.8125e-01, -5.6701e-01, -6.0709e-01, -7.6086e-01, -5.2899e-02,\n",
       "          -1.8295e-01, -4.5541e-01,  3.2835e-01,  1.5329e-01,  2.2138e-01,\n",
       "           1.2290e-01,  1.6363e-01, -2.6092e-02, -3.1458e-01, -4.7278e-01,\n",
       "          -2.4505e-01, -4.1356e-01,  2.9014e-01,  4.0426e-01,  5.5223e-01,\n",
       "           5.5459e-01,  2.6780e-01,  3.2187e-02,  1.2552e-02, -6.5403e-01,\n",
       "           7.2130e-02, -2.8635e-01,  4.4151e-03,  5.3700e-01, -1.0424e+00,\n",
       "           1.9329e-01,  6.5369e-01, -1.7744e-01,  9.3082e-01, -1.4861e-01,\n",
       "          -1.8363e-02,  5.5486e-02, -5.3270e-01,  1.2330e-01, -2.5389e-01,\n",
       "           8.0928e-01,  2.8371e-02, -2.0711e-02,  2.4278e-01,  7.9321e-01,\n",
       "           3.8230e-02,  4.2287e-01, -4.3086e-01, -2.5549e-01, -5.1724e-02,\n",
       "          -1.5972e-01,  1.5419e-01,  5.8196e-01,  9.4509e-02, -7.5436e-01,\n",
       "          -6.7960e-01, -3.0432e-01,  2.0134e-01, -2.7380e-01,  8.4216e-02,\n",
       "          -1.4197e-01,  2.9230e-01,  3.7791e-01, -1.5631e-01,  3.3354e-01,\n",
       "          -3.0520e-02, -4.0307e-01, -2.7911e-01, -5.8823e-01,  1.2289e-02,\n",
       "           1.9674e-01,  5.3417e-01,  3.2210e-01, -2.1841e-01,  8.7342e-01,\n",
       "           8.1705e-02,  3.8221e-01, -1.9522e-01, -6.4455e-01, -4.9331e-02,\n",
       "           2.5762e-01,  2.8977e-01, -2.2887e-01, -4.8592e-01, -1.2639e-01,\n",
       "          -2.6671e-01,  6.4543e-01,  2.1379e-01,  4.0531e-01, -2.1432e-01,\n",
       "           6.4120e-01, -1.8565e-01, -3.0470e-01, -2.7015e-01,  1.7894e-01,\n",
       "          -1.2939e-03,  3.4285e-01, -1.3186e-01,  9.3058e-02, -2.2083e-01,\n",
       "          -3.5229e-01,  1.1447e-01, -1.3783e-01,  2.3106e-01, -4.0019e-01,\n",
       "          -7.0122e-01,  3.0559e-01,  1.1233e-01, -9.4508e-01, -4.3849e-01,\n",
       "           5.1950e-02,  2.8005e-01,  2.2401e-01, -5.4590e-01,  1.8308e-01,\n",
       "           7.1659e-02, -2.1678e-01,  6.3296e-02, -4.7154e-02,  7.3005e-01,\n",
       "           2.2205e-01, -6.8802e-01,  2.7955e-01,  1.2424e-01,  1.2306e-01,\n",
       "          -2.6606e-01,  5.2057e-01, -1.4451e-02,  3.1896e-01,  6.0951e-01,\n",
       "           6.2058e-02,  3.0808e-01, -3.9731e-02, -1.2732e-01, -1.0905e-01,\n",
       "           2.3562e-01,  2.9939e-01, -3.8607e-01, -1.6463e-01,  4.4244e-01,\n",
       "           1.3412e-01, -3.3997e-02, -2.3001e-01, -1.8661e-01, -7.5352e-02,\n",
       "          -4.9532e-02, -7.0645e-01, -2.5596e-01,  4.8857e-02, -1.8622e-01,\n",
       "          -6.5160e-01, -9.6669e-01,  4.2793e-01, -1.8521e-01,  3.7495e-02,\n",
       "           7.6854e-02,  2.4003e-01,  7.2770e-02,  1.0242e-01, -1.3991e-01,\n",
       "           1.0200e+00, -3.7677e-02,  1.8749e-01, -5.5982e-01, -5.6480e-02,\n",
       "           4.7211e-01, -3.9471e-01, -2.6060e-01, -5.0113e-01,  9.5540e-02,\n",
       "           2.3924e-01, -1.8523e-01,  2.5962e-01, -6.1207e-01,  1.6349e-01,\n",
       "          -4.8388e-01,  9.4443e-01, -5.7841e-03,  9.8833e-02,  1.3364e-02,\n",
       "          -1.1472e-01, -1.5091e+00,  2.7947e-01, -3.4562e-01,  2.2459e-01,\n",
       "          -5.8765e-01,  6.7493e-01,  4.6599e-01, -7.1238e-02,  2.3993e-01,\n",
       "          -9.1328e-02, -2.0976e-01,  1.7988e-02,  2.2033e-01,  2.3376e-02,\n",
       "           2.1780e-01,  6.3392e-01,  2.5714e-02, -4.7532e-02,  3.6520e-01,\n",
       "          -6.6709e-01, -1.1634e-02,  1.3687e-01, -9.8252e-02,  1.7700e-01,\n",
       "           3.8480e-01,  2.5224e-02,  9.7063e-02, -6.7001e-02,  4.1798e-01,\n",
       "           1.5302e-02, -1.6144e-01, -3.8140e-01, -1.2106e-01, -3.2668e-02,\n",
       "           3.6539e-01, -3.1093e-01, -1.0837e-01,  4.5004e-01,  7.9832e-02,\n",
       "          -3.8652e-01,  2.5172e-01, -3.4062e-01, -1.4847e-01,  3.1302e-01,\n",
       "           3.1691e-01, -2.9343e-01,  2.5422e-01, -1.4025e-01, -1.5783e-01,\n",
       "           2.2754e-01, -6.4696e-01,  6.3509e-01,  6.9181e-01, -2.0811e-01,\n",
       "          -2.8726e-01, -2.4787e-01,  2.2060e-01, -8.2704e-02, -3.1642e-01,\n",
       "          -1.5010e-01, -4.3498e-01, -5.1878e-01,  1.5102e-01, -1.9666e-01,\n",
       "          -4.5642e-02, -1.5838e-01,  9.3813e-01, -2.6837e-02,  4.2402e-01,\n",
       "           2.6015e-03, -6.1778e-02,  1.8552e-01,  2.2671e-02, -3.3609e-01,\n",
       "           3.4484e-01, -1.6487e-01, -5.9427e-01,  3.0683e-01, -3.9763e-01,\n",
       "          -3.6214e-01,  6.4549e-02,  9.1911e-03,  3.2640e-01, -2.3144e-01,\n",
       "          -4.2379e-01,  3.1276e-01, -7.7254e-03, -3.9106e-01,  3.1301e-01,\n",
       "          -5.4134e-02,  2.2738e-01, -3.0651e-01, -5.2413e-01, -4.9942e-01,\n",
       "          -9.6753e-01,  1.3783e-01, -1.9638e-01, -4.8626e-02, -1.0751e+00,\n",
       "          -3.9520e-01,  2.3029e-01, -1.3873e-01,  3.5862e-02,  4.0926e-01,\n",
       "           3.5843e-01,  6.4132e-02,  7.3936e-02,  8.8761e-04, -5.2274e-01,\n",
       "          -2.9655e-01,  1.8896e-01, -2.9958e-01,  4.6521e-02,  2.1681e-01,\n",
       "          -3.3301e-01,  3.4246e-02,  3.9500e-01, -1.8062e-01, -2.4787e-01,\n",
       "          -4.5549e-01, -3.2597e-01,  1.6095e-01, -6.8677e-02, -6.8769e-03,\n",
       "          -1.6611e-01,  2.8945e-01,  7.6739e-01, -5.6441e-01, -7.7109e-01,\n",
       "           7.5579e-02,  2.4140e-01, -1.4286e-01, -2.4809e-01, -2.5021e-01,\n",
       "          -9.2259e-01, -2.6915e-02, -1.7244e-01, -1.6725e-01, -5.0933e-01,\n",
       "           2.5351e-01, -4.3928e-02, -1.3423e-01, -1.1113e-01, -2.9242e-01,\n",
       "           1.4208e-01,  3.3421e-01, -2.7283e-01,  1.1322e-01, -2.9268e-01,\n",
       "           1.8873e-01, -3.0141e-01,  7.2321e-03, -3.4182e-01,  6.0634e-02,\n",
       "           1.8711e-01, -2.0632e-01,  5.5838e-02, -7.3778e-02, -3.1496e-01,\n",
       "           3.8900e-01, -4.0732e-01, -5.8766e-02,  3.5700e-01, -4.5896e-01,\n",
       "          -4.3196e-01,  4.3506e-01, -8.7744e-01, -5.6263e-01,  1.6586e-01,\n",
       "           3.3870e-01, -2.2666e-01, -4.1438e-01, -5.3312e-02,  2.7649e-01,\n",
       "           5.7517e-01,  2.7700e-01, -5.1972e-01,  2.1680e-01,  1.7434e-01,\n",
       "          -3.5159e-01,  3.8993e-01, -1.9005e-01, -1.9936e-01, -5.1625e-01,\n",
       "          -5.1670e-02,  2.8095e-01, -6.7578e-01,  1.5098e-01, -3.1328e-01,\n",
       "          -3.4116e-01,  3.8056e-02, -1.5104e-01,  2.0485e-01,  7.2477e-01,\n",
       "          -1.9960e-01, -2.9160e-01,  4.7028e-01, -3.7544e-01,  1.1620e-01,\n",
       "           8.4313e-02,  5.7862e-02, -6.7045e-02, -3.5411e-01, -2.1431e-01,\n",
       "          -2.8269e-01, -3.3205e-01,  2.3794e-01,  3.6801e-01,  1.4242e-02,\n",
       "           7.0119e-02, -2.0462e-01, -1.5543e-01,  9.7277e-01, -3.1691e-01,\n",
       "          -3.9943e-01,  2.2692e-01,  2.1494e-01,  2.2726e-01, -2.8748e-01,\n",
       "          -1.7302e-01, -2.8388e-01,  6.3040e-01, -2.4195e-01,  2.5752e-01,\n",
       "          -4.7725e-01,  1.5182e-02,  2.0099e-01, -2.8956e-01, -7.0461e-02,\n",
       "           6.8967e-01,  6.1830e-01,  2.2874e-01,  4.1937e-01, -3.1751e-01,\n",
       "           7.8736e-02, -6.9358e-02,  1.0949e-01,  4.1084e-01, -8.0356e-01,\n",
       "           5.3574e-01,  4.6789e-01,  4.6241e-01,  2.1917e-02, -1.1062e+01,\n",
       "           1.4933e-01,  4.5052e-01, -3.8443e-01,  4.0179e-01, -2.8539e-01,\n",
       "           7.1007e-01, -2.3680e-01, -3.6101e-02, -1.3829e-01,  1.1163e-01,\n",
       "          -5.3106e-01,  3.1114e-02, -1.3019e-02, -4.1297e-01, -6.2060e-02,\n",
       "          -2.4520e-01, -6.9370e-01,  8.9877e-03, -2.1956e-01,  1.2912e-01,\n",
       "          -4.3137e-02,  1.2544e-01,  1.8596e-01,  9.4200e-02,  1.0136e-01,\n",
       "          -1.7058e-01, -2.4823e-01,  4.6804e-01,  4.1543e-01,  2.5877e-01,\n",
       "           1.7289e-02,  7.4320e-02,  2.7404e-01, -2.6912e-01, -4.7287e-01,\n",
       "          -7.5024e-02, -1.6265e-01, -1.5382e-01, -3.8110e-01, -2.4374e-01,\n",
       "          -3.8759e-01, -2.3122e-02,  5.8981e-01,  1.8492e-01,  9.9867e-01,\n",
       "           1.4936e+00,  5.7914e-01,  2.7833e-01, -9.0346e-02,  7.6317e-02,\n",
       "           5.1604e-02,  5.2978e-01, -6.8027e-01, -6.1479e-02, -6.9028e-01,\n",
       "           2.4417e-01, -1.1004e+00,  2.1557e-01, -4.1523e-01, -5.1783e-01,\n",
       "           2.9694e-01, -1.6271e-01,  3.2349e-01, -3.6633e-01, -1.9989e-01,\n",
       "           4.9332e-01, -5.0237e-01,  1.1235e-01, -1.3648e-02,  1.1596e-01,\n",
       "          -5.9574e-01,  5.8643e-01,  6.7600e-02, -4.0591e-01, -3.0374e-01,\n",
       "          -2.2595e-01, -1.6827e-01,  1.6499e+00,  1.0238e-01,  6.3182e-03,\n",
       "           3.3148e-01,  1.2002e-01, -2.4678e-01, -5.1067e-01,  3.4279e-01,\n",
       "          -3.2295e-02, -2.3322e-01,  1.8517e-01,  5.5575e-01, -2.3738e-01,\n",
       "          -1.1152e+00,  2.9230e-02,  2.3309e-01,  3.9285e-01,  6.9187e-01,\n",
       "          -8.9288e-01, -1.6080e-01,  5.5083e-02, -1.6676e-01, -8.8859e-01,\n",
       "           2.5817e-02, -5.7065e-02, -4.3525e-01,  3.2931e-01,  4.2958e-02,\n",
       "           2.2200e-02,  2.3154e-01, -3.5754e-01,  4.9859e-01, -3.1470e-01,\n",
       "           6.1742e-02, -8.5203e-02,  3.8069e-01, -4.4371e-01, -1.3333e-01,\n",
       "           4.0263e-01,  6.8200e-01, -1.2682e-01, -1.7887e-01, -4.6578e-01,\n",
       "           1.8188e-01,  1.9430e-01, -2.3578e-02, -3.6714e-01,  3.7050e-01,\n",
       "          -3.8336e-01, -2.6788e-01,  6.2797e-02,  1.1603e-01,  2.9290e-01,\n",
       "           2.6263e-02, -1.3425e-01,  3.2624e-01, -4.8082e-01, -1.0270e+00,\n",
       "          -6.0785e-01, -1.9437e-01, -2.7057e-02, -2.4267e-01,  8.9363e-02,\n",
       "          -9.0969e-02, -2.4874e-01,  5.5497e-02,  1.4090e-01,  2.4560e-01,\n",
       "           3.5831e-01,  4.1688e-01, -5.2434e-01,  1.1731e-01, -2.0947e-01,\n",
       "          -3.5393e-01,  1.4944e-02, -1.1839e-01,  2.2336e-02, -3.0764e-01,\n",
       "          -1.4743e-01, -1.5521e-01, -2.3476e-01, -6.2826e-03,  1.3356e-01,\n",
       "           1.7946e-01, -2.2486e-01, -1.7963e-01,  3.8874e-01,  4.6101e-01,\n",
       "           1.0100e-01, -3.7585e-01, -6.6035e-02, -7.2042e-03,  1.5482e-01,\n",
       "           1.4839e-01, -8.6226e-02, -6.4187e-01,  2.0878e-01, -1.5298e-01,\n",
       "          -1.1746e-02,  1.3891e+00, -1.9750e-01, -5.1458e-01,  8.2327e-02,\n",
       "           2.2427e-01, -5.5395e-02,  8.5149e-02, -5.8916e-01,  1.9627e-01,\n",
       "          -3.6696e-01, -4.6783e-01,  2.3598e-01, -8.4911e-03,  1.0345e-01,\n",
       "           4.2273e-01, -1.4191e-02, -6.5073e-02,  8.4024e-01,  4.4751e-01,\n",
       "          -3.0334e-01, -6.0323e-02, -4.2283e-01, -4.9448e-01,  1.6937e-01,\n",
       "           6.4510e-02,  2.7785e-01, -1.0816e-01,  4.1375e-01,  2.7764e-01,\n",
       "           3.4132e-01,  9.0311e-01, -4.6131e-02, -8.1850e-02,  9.5740e-01,\n",
       "          -6.3082e-02, -3.1440e-01,  5.0672e-01,  1.4596e-01, -9.0632e-03,\n",
       "           7.0893e-02,  1.1201e-01, -3.3358e-02,  2.2370e-01,  2.3093e-01,\n",
       "           4.2855e-01, -7.9807e-01, -1.6753e-01, -5.5622e-02,  6.4398e-01,\n",
       "          -3.5335e-01,  6.6968e-01, -1.0068e-01,  6.2934e-01,  4.8384e-01,\n",
       "          -7.4275e-01, -2.4621e-01, -2.8155e-01,  4.3065e-01,  2.6382e-02,\n",
       "          -5.0845e-01,  1.5545e-01,  1.0541e-01, -2.6312e-01,  5.1493e-02,\n",
       "           3.1225e-01, -9.7981e-01,  3.4935e-01,  7.0233e-01, -1.5454e-01,\n",
       "           9.1163e-02, -2.0100e-01,  2.8027e-01,  1.8079e-01, -6.7407e-02,\n",
       "           8.2740e-02, -4.5222e-02, -4.2797e-02, -5.6106e-01,  1.5273e-01,\n",
       "           5.8410e-01,  1.3786e-01,  8.8295e-05, -6.0881e-02,  1.7480e-01,\n",
       "          -2.0773e-01, -1.6881e-01, -1.6801e-01,  1.7257e-01, -2.6533e-01,\n",
       "           3.4370e-02, -4.1383e-01, -1.8526e-01,  2.0536e-01, -6.1263e-01,\n",
       "           2.6945e-01,  2.9664e-01,  6.3086e-01, -1.6590e-01, -2.8048e-01,\n",
       "           2.2841e-01,  4.6448e-01, -4.9136e-01,  5.6423e-01,  3.9014e-02,\n",
       "          -4.1381e-01,  1.7606e-02, -8.7256e-02, -1.7381e-01,  2.2214e-01,\n",
       "          -2.1966e-01,  3.6535e-01, -9.0760e-02, -1.2684e-01,  6.7261e-01,\n",
       "           1.4555e-01,  2.9486e-01, -7.6388e-01,  3.1234e-01,  1.9128e-01,\n",
       "           7.7186e-02,  8.2148e-01, -2.4473e-03,  3.1802e-01, -5.1749e-01,\n",
       "           5.3432e-02, -7.5045e-01, -2.6638e-02, -3.0928e-02, -8.3624e-03,\n",
       "          -2.7236e-01, -1.6344e-01,  5.0439e-01,  6.1688e-01,  8.2246e-01,\n",
       "           6.6253e-01,  7.2680e-02, -6.9248e-01, -5.9265e-01,  1.6514e-01,\n",
       "           5.9948e-02, -9.5385e-02,  1.5750e-01, -1.5135e-02,  4.6853e-01,\n",
       "          -5.6746e-01, -5.3956e-01,  2.4191e-01,  9.2813e-02, -2.7803e-01,\n",
       "           5.5420e-01, -3.9384e-02, -9.3511e-02,  5.0030e-01,  1.0280e-01,\n",
       "           4.6612e-01,  7.1132e-01,  2.6620e-02, -8.0462e-01, -1.7798e-01,\n",
       "           6.0975e-01, -3.7398e-01, -9.6133e-03, -4.7242e-01,  5.7219e-01,\n",
       "           2.9203e-01,  5.4013e-01,  6.8356e-01]], device='mps:0'),\n",
       " tensor([[-1.1097e-02, -5.0567e-01, -4.6699e-01, -6.4640e-01, -3.9556e-01,\n",
       "          -2.5266e-01, -5.5990e-01,  3.0801e-01, -1.2536e-01,  1.4189e-01,\n",
       "           2.9133e-01,  5.3879e-01, -3.6706e-01, -2.8859e-01, -4.8290e-01,\n",
       "          -5.3699e-02, -6.1741e-01, -1.6918e-02,  3.4690e-01,  3.0491e-01,\n",
       "           5.3718e-01,  1.8073e-01, -4.1129e-01,  2.9149e-01, -3.7959e-01,\n",
       "           6.6561e-02, -3.4985e-01, -3.1546e-01,  2.6210e-01,  2.8012e-01,\n",
       "           5.3618e-02,  7.7692e-01, -2.4307e-02,  1.0962e+00, -1.8413e-01,\n",
       "          -2.1778e-01,  1.5359e-01,  2.3320e-01,  8.6439e-01, -6.2204e-01,\n",
       "           5.6395e-01,  3.6489e-01,  2.7586e-02, -4.0063e-02,  2.7845e-01,\n",
       "           3.5451e-01,  1.4567e-01, -5.7913e-01, -2.7161e-01, -3.4037e-01,\n",
       "          -1.2979e-01, -1.5164e-01,  5.0058e-01, -2.3292e-01, -7.9546e-01,\n",
       "          -1.9253e-01,  1.1288e-02,  2.6982e-01,  1.1117e-01,  2.3591e-01,\n",
       "           2.8900e-02,  7.9425e-01, -1.9339e-01, -3.2966e-03,  2.0559e-01,\n",
       "           2.5502e-01, -7.6243e-01, -1.2873e-01, -7.2822e-01,  1.1548e-01,\n",
       "           2.1413e-01,  3.5162e-01,  1.0500e-01, -2.6816e-01, -1.9644e-01,\n",
       "           6.0243e-02,  1.6956e-01,  1.8610e-01, -2.0035e-01,  3.5967e-02,\n",
       "           2.1140e-01,  4.4140e-01,  4.5980e-01,  5.9922e-02,  1.6117e-01,\n",
       "           5.6960e-02,  2.0080e-01, -3.9869e-02, -3.1738e-02,  1.7672e-01,\n",
       "           1.1984e-01, -9.9577e-02, -3.7483e-01, -4.3956e-01, -9.5205e-02,\n",
       "          -3.0408e-01,  3.2250e-02, -2.9469e-01, -2.3954e-01, -3.3261e-01,\n",
       "           2.5982e-01,  9.2762e-02,  3.2750e-01, -9.3082e-02, -6.6514e-01,\n",
       "          -3.5248e-01,  4.0832e-01,  3.4586e-03, -2.8129e-01, -3.8448e-01,\n",
       "          -6.6713e-02, -4.9346e-01,  1.2765e-01, -4.4650e-01,  1.9893e-01,\n",
       "           1.0835e-01,  1.8686e-01, -3.2670e-01,  6.5172e-02,  8.8101e-02,\n",
       "           6.1068e-02, -5.0938e-01,  2.7421e-01,  1.8326e-01,  9.3046e-02,\n",
       "           2.2489e-01,  6.8015e-01, -1.6210e-01,  4.3454e-01,  3.5722e-01,\n",
       "           3.1169e-01,  1.6221e-01, -4.1686e-01, -5.0259e-01, -7.5302e-02,\n",
       "           1.8488e-02, -5.9908e-01, -5.7442e-01,  6.1723e-02, -2.2868e-01,\n",
       "           1.6627e-01, -7.3083e-02, -5.1768e-01, -5.4307e-01, -3.1494e-01,\n",
       "          -3.0772e-01, -2.0517e-02, -3.2393e-01, -4.8159e-01, -2.5398e-01,\n",
       "          -6.1867e-01, -7.2891e-01,  1.9443e-01,  1.5164e-01,  2.7714e-01,\n",
       "          -4.3163e-01,  5.4580e-01, -3.0612e-01, -4.8554e-01,  3.7642e-01,\n",
       "           4.3432e-01, -5.0330e-01,  2.9526e-01, -6.3499e-01,  5.1171e-02,\n",
       "           2.0480e-01,  2.5291e-02, -3.4147e-02,  4.2807e-01,  1.1727e-01,\n",
       "           3.9831e-01,  2.7152e-02,  1.2476e-01, -2.9187e-01,  1.5015e-01,\n",
       "          -4.5427e-02, -3.2147e-01, -1.1176e-01,  1.7279e-02, -5.2967e-02,\n",
       "           5.4936e-01, -1.7306e+00, -1.7852e-01,  7.0060e-02, -2.7460e-01,\n",
       "          -1.3173e-01,  2.1473e-01,  5.0407e-03,  3.2283e-01, -3.5674e-01,\n",
       "           8.3698e-02,  8.7128e-02,  8.4654e-02, -1.0725e-01, -5.0983e-02,\n",
       "           2.9101e-01,  7.5929e-01,  4.0793e-01,  9.7463e-02, -2.6657e-01,\n",
       "          -6.0645e-01,  4.1554e-01,  1.9081e-01,  2.0730e-01,  3.9000e-01,\n",
       "           6.7931e-02, -6.9343e-01,  2.9674e-01,  1.9859e-01,  3.7002e-01,\n",
       "           2.5546e-02, -5.8496e-02, -4.8962e-01, -2.4682e-01, -4.3705e-01,\n",
       "           2.7759e-01, -1.3436e-01, -2.5922e-01,  6.3173e-01,  1.9273e-01,\n",
       "          -5.3531e-02,  3.9653e-01,  1.8733e-01, -4.1041e-01,  4.1633e-01,\n",
       "           5.2814e-01,  7.9114e-02,  1.4573e-01, -2.8055e-01, -1.6673e-01,\n",
       "          -3.1532e-01, -4.4420e-01,  4.4703e-01,  8.3941e-01, -7.2084e-03,\n",
       "           5.5343e-01, -3.5757e-01,  5.1174e-01, -3.6424e-01, -6.7779e-03,\n",
       "          -4.6486e-02,  6.8700e-02, -3.3988e-01,  1.6284e-01,  1.5551e-01,\n",
       "           1.0340e-01,  1.9709e-01,  8.5425e-01,  4.6494e-01,  1.4684e-01,\n",
       "           4.1031e-01,  2.5417e-02,  3.2080e-01,  7.8684e-01, -3.0254e-01,\n",
       "           3.4990e-01, -7.1578e-03, -4.1842e-01,  5.6461e-01, -3.5139e-01,\n",
       "           1.4248e-01, -2.3020e-01,  2.2987e-01,  1.7866e-01, -2.7854e-01,\n",
       "           2.1426e-01,  6.5273e-01,  9.5860e-02, -3.9283e-01,  7.3417e-01,\n",
       "          -4.0214e-01,  7.1262e-01, -3.0138e-02, -2.6992e-02, -1.0120e-01,\n",
       "          -7.4577e-01, -6.8569e-01,  1.4294e-01,  3.0808e-01, -7.6332e-01,\n",
       "          -3.4295e-01,  1.2380e-01,  1.3255e-01,  3.5068e-01,  2.4156e-01,\n",
       "           4.6144e-01, -1.7016e-01, -4.7185e-02, -4.4024e-01, -4.6015e-01,\n",
       "           1.1917e-01,  7.3311e-02,  9.5311e-02, -9.8199e-02,  9.2587e-02,\n",
       "           6.1067e-01,  2.4297e-01,  2.6799e-01,  1.4946e-01, -3.4610e-01,\n",
       "          -2.7723e-02,  1.7135e-01,  3.7874e-01, -6.1900e-01, -2.1807e-01,\n",
       "           5.3543e-02,  8.3164e-02,  6.6826e-01, -2.1832e-01, -1.1546e-01,\n",
       "          -1.2450e-01,  5.9739e-01, -2.2108e-01,  6.8002e-02,  1.0269e-01,\n",
       "          -6.8586e-01,  2.7619e-01, -5.9939e-02, -8.5297e-02, -5.2861e-01,\n",
       "           2.2997e-01, -5.7787e-02, -1.6068e-01, -7.3269e-03, -7.7601e-02,\n",
       "           2.2454e-01,  1.3406e-01,  2.8044e-01, -5.6270e-01, -4.2092e-01,\n",
       "           1.1781e-01, -3.2365e-01, -6.6016e-02, -4.5861e-01,  3.6766e-01,\n",
       "           1.3890e-01,  4.3829e-01, -1.0837e-01, -2.8576e-02, -1.3190e-01,\n",
       "           5.2866e-01, -6.2780e-01,  1.5573e-02,  4.8525e-01, -3.5595e-01,\n",
       "          -1.8139e-01,  4.7644e-01, -2.8011e-01,  4.9664e-02,  2.2982e-01,\n",
       "           3.7738e-01, -1.1530e-01, -4.7105e-01, -5.5247e-02,  9.3761e-01,\n",
       "           4.8980e-01,  2.0874e-02, -6.3820e-01,  1.2560e-01,  4.2066e-01,\n",
       "          -3.4219e-01,  1.0777e-01, -3.7280e-01, -2.3519e-01,  8.6456e-02,\n",
       "           3.3260e-02,  5.0344e-01, -5.8998e-02,  1.9741e-01,  4.6778e-01,\n",
       "           1.7011e-01, -6.9331e-01, -3.0557e-01,  6.3124e-01,  5.4670e-01,\n",
       "          -1.2247e-01, -5.4832e-01,  1.1480e-01, -6.4563e-01,  3.1198e-02,\n",
       "          -3.8623e-03, -3.5755e-01,  2.9653e-01, -1.4962e-01, -2.5683e-01,\n",
       "          -2.0003e-01, -5.3748e-01,  4.4973e-03,  1.0291e-01, -6.9595e-02,\n",
       "           4.1738e-01, -7.9610e-03, -3.1154e-01,  1.0058e+00, -2.7292e-01,\n",
       "          -3.1356e-02,  7.4139e-01, -5.4234e-01,  1.0457e-02,  2.4642e-01,\n",
       "          -1.3433e-01, -1.2416e-02,  2.4323e-02,  1.0660e+00,  1.1515e-05,\n",
       "           3.6402e-02, -2.0346e-01,  4.2203e-01, -4.7659e-01,  1.9394e-01,\n",
       "          -4.9958e-01,  4.8310e-01, -6.9399e-02,  2.4253e-01, -8.2329e-01,\n",
       "           4.3257e-01, -3.0604e-01, -3.2040e-01, -6.3922e-01,  5.0359e-02,\n",
       "          -5.0711e-02,  6.6524e-02, -5.8707e-02,  6.0787e-01, -1.0953e+01,\n",
       "           1.0946e-01,  8.4108e-01, -3.2618e-02,  1.3653e-01,  4.3336e-01,\n",
       "           1.2158e-01, -4.6684e-01,  7.2509e-01, -2.3242e-01, -6.6092e-03,\n",
       "          -3.5861e-01, -1.2444e-01, -1.3819e-01, -6.3519e-01,  4.0758e-01,\n",
       "          -1.3255e-01, -5.9114e-01,  2.8540e-01, -1.6945e-01,  2.4978e-01,\n",
       "          -2.3651e-01,  1.5671e-01,  5.9352e-01, -6.1010e-01, -3.3943e-01,\n",
       "          -1.1958e-01, -2.9991e-01,  5.1723e-01,  4.0217e-01, -2.3672e-01,\n",
       "           2.9096e-01,  3.3529e-02,  2.9455e-01, -5.5694e-01, -4.8133e-01,\n",
       "           4.3369e-01, -3.9950e-01,  2.1957e-01, -9.9943e-01, -3.3265e-01,\n",
       "          -1.8082e-01, -5.0951e-01, -2.7870e-02,  5.3812e-01,  7.5524e-01,\n",
       "           1.2821e+00,  1.1534e+00,  9.8198e-02,  4.6362e-01, -1.0523e-01,\n",
       "          -6.6935e-01,  5.1619e-02, -5.6525e-01, -4.1224e-01,  8.7196e-02,\n",
       "           2.2385e-01, -5.4352e-01,  4.7854e-01, -5.9897e-01,  1.8912e-01,\n",
       "           1.4329e-01,  7.9852e-02,  1.7368e-01, -5.1897e-01, -5.5039e-01,\n",
       "          -1.3282e-01,  8.1201e-02,  8.6324e-02, -6.7881e-01,  1.1881e-01,\n",
       "           1.7006e-01,  6.8197e-01, -2.5397e-01, -4.4268e-01,  4.1137e-01,\n",
       "          -2.2746e-03,  9.3809e-02,  1.3158e+00, -2.6036e-01, -1.7413e-01,\n",
       "          -4.8428e-02, -2.3194e-01, -3.3484e-01, -3.9656e-01, -4.5613e-01,\n",
       "          -2.5020e-02,  3.6573e-01,  5.1948e-01,  2.6062e-01, -2.3329e-02,\n",
       "          -3.0054e-01, -4.1445e-02,  5.1977e-02,  3.3862e-01,  9.6772e-01,\n",
       "          -6.6887e-01,  2.6280e-01, -1.9148e-01,  9.2178e-02,  2.2223e-01,\n",
       "          -5.7106e-01, -1.6871e-02,  2.9434e-03,  1.3975e-02,  5.7281e-01,\n",
       "           1.9705e-01,  2.6453e-01,  3.1190e-02,  2.2859e-01, -4.1682e-01,\n",
       "          -4.4044e-01,  2.9125e-04, -1.3190e-01, -5.8294e-01, -4.1034e-01,\n",
       "          -1.3514e-01,  1.3712e-01, -2.8585e-01, -5.3585e-01, -2.5012e-01,\n",
       "           1.4192e-01,  2.5900e-03,  4.9952e-02, -3.3857e-01,  4.7379e-01,\n",
       "          -7.6918e-01, -2.1429e-01, -2.7578e-01, -3.8495e-03,  1.4633e-01,\n",
       "           2.5508e-01, -5.1518e-01, -2.7838e-01, -1.0123e-01, -9.2314e-01,\n",
       "          -8.4489e-01,  2.2927e-02,  7.4620e-01, -1.1960e-01,  1.1861e-01,\n",
       "           8.5428e-02, -3.1671e-03, -3.7708e-01,  2.5194e-01, -2.4644e-01,\n",
       "           2.8149e-01,  4.3265e-01, -4.9583e-01,  2.9773e-02, -1.2792e-01,\n",
       "          -1.8992e-02, -1.0887e-01, -8.8647e-02, -5.4131e-01, -6.1371e-01,\n",
       "          -4.1197e-01,  1.3736e-01, -2.2747e-01,  4.1516e-01,  6.8072e-02,\n",
       "           1.2582e-01, -2.0826e-01,  9.8224e-02,  8.2540e-01,  6.0197e-01,\n",
       "           2.3014e-01, -3.2920e-01, -7.7666e-01,  3.4140e-02,  4.1642e-02,\n",
       "          -1.6450e-01,  2.1317e-01, -1.6384e-01, -3.0657e-01, -1.0112e-01,\n",
       "           1.2620e-02,  1.0002e+00, -2.0790e-01, -1.9155e-01,  1.0507e-01,\n",
       "           7.4626e-01, -6.0834e-01, -5.7221e-01,  3.7612e-01,  5.0418e-02,\n",
       "          -5.6831e-01, -1.9450e-01,  4.0315e-01, -1.7111e-01,  8.9921e-02,\n",
       "           3.7572e-01,  1.7821e-01, -1.8612e-01,  2.2711e-01, -6.7291e-02,\n",
       "          -1.7001e-01,  3.2731e-01, -1.6533e-01, -7.7254e-02,  4.0384e-01,\n",
       "           9.9517e-02,  1.5584e-01, -2.5262e-01, -4.8444e-01, -2.7025e-01,\n",
       "           1.5439e-01,  1.1110e+00, -1.7989e-01, -3.2527e-01,  6.6881e-01,\n",
       "          -3.0956e-01,  3.8928e-01,  4.8478e-01, -2.2611e-01, -5.0807e-02,\n",
       "           5.8049e-01,  3.0288e-01, -7.3236e-01,  2.2417e-01,  3.1670e-01,\n",
       "          -1.7335e-01, -5.3095e-01, -3.2838e-01, -2.1330e-01, -3.5189e-02,\n",
       "          -3.6356e-01,  5.2132e-01,  1.2255e-01,  3.9253e-01, -1.0843e-01,\n",
       "          -8.7207e-01,  4.7185e-01,  2.6128e-01,  8.0250e-01, -6.2568e-03,\n",
       "          -4.4880e-01, -3.5541e-01, -3.5790e-01, -6.6453e-02,  3.6543e-01,\n",
       "           2.0116e-01, -1.0333e+00,  4.6071e-02,  4.0507e-01, -3.9812e-02,\n",
       "           4.7083e-01, -1.4254e-01,  1.9634e-01,  4.7891e-01, -4.3918e-02,\n",
       "          -2.1856e-01,  1.3161e-01, -3.3027e-02, -2.7190e-01,  4.2861e-01,\n",
       "           4.6755e-01,  1.6062e-02,  2.5454e-02,  6.4881e-02, -4.2427e-01,\n",
       "          -3.4980e-01,  7.7064e-02, -1.7966e-01, -3.5675e-01, -1.4295e-01,\n",
       "          -4.0726e-01, -5.1772e-01, -6.6164e-02, -1.1357e-01,  1.3233e-01,\n",
       "           3.1032e-01,  1.3036e-01,  3.8431e-01, -2.3708e-01, -1.2541e-01,\n",
       "          -2.7977e-01,  4.5653e-01, -2.7595e-01,  8.0808e-01,  2.0566e-01,\n",
       "          -1.4423e-01,  2.0720e-01,  2.1514e-01, -2.9046e-01,  1.3028e-01,\n",
       "          -2.7828e-01,  2.8810e-01, -5.7798e-01, -3.4110e-01, -5.9710e-02,\n",
       "          -7.7948e-02, -1.5049e-01, -2.8505e-01, -1.0179e-01,  1.0608e-01,\n",
       "          -9.9043e-02,  9.7868e-01, -4.1759e-01,  6.3833e-01, -1.0342e-01,\n",
       "          -1.1350e-01, -7.5089e-01,  3.9397e-01,  3.6376e-01, -1.4798e-01,\n",
       "          -5.6434e-01, -2.1653e-01, -8.9158e-02, -4.2284e-01,  8.8621e-01,\n",
       "           5.8157e-01, -7.4673e-01, -3.7246e-01, -1.5919e-01, -2.4113e-01,\n",
       "          -4.7389e-02,  6.9820e-02, -3.2710e-01, -1.3295e-02, -1.6580e-01,\n",
       "          -9.8715e-01,  2.7641e-01, -3.6473e-02,  4.9205e-01, -2.5466e-01,\n",
       "           7.6979e-01, -8.6017e-02, -1.2749e-01,  1.1690e+00,  1.1169e-01,\n",
       "          -6.1844e-02,  5.9531e-01, -1.6837e-01, -6.2973e-01, -7.0295e-02,\n",
       "           4.4239e-02, -3.1098e-01,  8.7670e-02, -4.7880e-01,  2.9353e-01,\n",
       "           2.3648e-01, -7.1226e-02,  4.3877e-01]], device='mps:0'),\n",
       " tensor([[ 0.0269, -0.7027, -0.5242,  ...,  0.2728,  0.6374,  0.8627],\n",
       "         [-0.2330, -0.4411, -0.4565,  ...,  0.0257,  0.5397,  0.4576],\n",
       "         [-0.0120, -0.5748, -0.5633,  ...,  0.3189,  0.5923,  0.7314],\n",
       "         [-0.0294, -0.4941, -0.6745,  ...,  0.2266,  0.4689,  0.6529]],\n",
       "        device='mps:0'),\n",
       " tensor([[-0.0548, -0.3751, -0.4766,  ...,  0.0464,  0.8423,  0.4111],\n",
       "         [-0.0543, -0.6656, -0.7601,  ...,  0.1333,  0.5108,  0.4749]],\n",
       "        device='mps:0'),\n",
       " tensor([[-0.0111, -0.5057, -0.4670,  ...,  0.2365, -0.0712,  0.4388],\n",
       "         [-0.3292, -0.2590, -0.5016,  ..., -0.0148,  0.3509,  0.1607]],\n",
       "        device='mps:0'),\n",
       " tensor([[-6.1750e-01, -4.2829e-01, -2.3934e-01, -9.4116e-01, -3.4548e-01,\n",
       "          -6.5496e-02, -6.3874e-01,  9.7828e-01, -9.5892e-02,  4.0096e-01,\n",
       "           1.4919e-01, -5.1603e-03,  2.0980e-01,  1.3680e-01, -6.9506e-01,\n",
       "           1.6740e-01,  4.6094e-01,  2.1159e-01,  3.5951e-01, -4.4329e-02,\n",
       "           7.9832e-01,  3.4245e-01,  1.8019e-01, -2.5205e-01, -4.5876e-01,\n",
       "           5.8994e-01, -2.4628e-01, -9.2014e-02,  2.0509e-01, -8.1949e-01,\n",
       "          -1.3586e-01, -4.9757e-02, -3.6281e-01,  8.7602e-01,  1.1948e-01,\n",
       "          -2.7310e-01, -1.5276e-01, -2.7996e-01,  2.3494e-01, -6.3637e-02,\n",
       "           3.7814e-01,  2.6592e-02, -2.9347e-01,  3.1386e-01, -1.6421e-01,\n",
       "          -1.4057e-01,  1.4047e-02, -7.8429e-01,  1.3666e-02,  1.8955e-01,\n",
       "           5.8851e-02, -4.7205e-02,  1.9231e-01,  3.7030e-01, -1.0617e+00,\n",
       "          -2.6348e-01,  2.7603e-01,  2.9849e-01, -1.7932e-01, -5.7784e-02,\n",
       "          -1.9989e-01,  9.0201e-01,  1.7949e-01, -2.7822e-01,  1.7824e-01,\n",
       "          -4.4899e-01, -5.4601e-01, -6.0781e-01, -1.0682e-01, -1.1967e-01,\n",
       "          -1.0002e-01,  9.7528e-02,  1.4009e-01, -2.9385e-01,  2.3170e-01,\n",
       "           2.1813e-01,  4.7024e-02, -2.7652e-02, -4.3158e-01, -2.6546e-01,\n",
       "           3.6947e-01,  3.4425e-02, -1.4730e-02,  2.0326e-01, -5.8744e-01,\n",
       "          -2.0226e-01,  3.7959e-01,  1.5703e-01,  2.5914e-01,  2.2662e-03,\n",
       "           5.9967e-01,  1.2024e-02,  8.0339e-02, -4.5369e-01,  5.8427e-01,\n",
       "          -5.4955e-01,  1.0329e-01, -1.6007e-01,  4.9532e-02, -5.2088e-01,\n",
       "          -9.8735e-02,  2.4494e-01, -1.8085e-01,  1.6053e-01, -4.6592e-01,\n",
       "          -3.2983e-02,  2.4514e-01,  3.0468e-01, -4.9183e-01,  4.2685e-02,\n",
       "          -6.5134e-03, -2.9791e-01,  7.3694e-02, -6.6407e-01, -8.5795e-01,\n",
       "           5.8354e-01,  1.8428e-02, -4.5240e-01, -2.3930e-01,  7.4443e-01,\n",
       "           2.9738e-01, -2.3201e-01, -5.3072e-02, -3.5548e-01,  2.0259e-01,\n",
       "          -1.8518e-01,  2.5689e-01, -2.7018e-01,  3.8439e-01,  5.5026e-01,\n",
       "           3.1581e-01,  2.8489e-01, -1.8327e-01,  2.7637e-01,  1.4465e-01,\n",
       "           3.6947e-02, -6.6614e-02, -1.5929e-01,  1.5180e-01,  1.2041e-01,\n",
       "          -2.9580e-01, -3.2999e-01,  5.4746e-03,  1.1182e-01, -3.3279e-01,\n",
       "          -1.0202e-01,  1.8680e-01,  3.4808e-01, -1.4092e-01, -2.5403e-01,\n",
       "          -7.8984e-01, -4.2328e-01,  1.0217e-01, -1.8560e-01, -1.7135e-01,\n",
       "          -4.1911e-01,  3.8121e-01, -6.6660e-02,  4.7638e-01,  3.4189e-01,\n",
       "           6.9978e-01, -1.5311e-01,  5.9776e-02, -5.6412e-01,  3.5197e-01,\n",
       "           3.8270e-02, -6.1110e-01, -1.7247e-01,  1.5680e-01, -4.1813e-04,\n",
       "           6.8195e-01,  7.0979e-02,  4.0124e-01,  1.2070e-01, -3.2553e-01,\n",
       "           8.2089e-03,  1.9890e-01, -1.2864e-01, -1.4297e-01,  7.3766e-01,\n",
       "          -5.5188e-01, -2.3029e+00,  1.0721e-01, -3.5844e-01, -3.6529e-01,\n",
       "          -1.8135e-01,  3.7255e-01,  3.3772e-02,  4.9966e-01,  1.5486e-01,\n",
       "           3.7598e-02,  1.6729e-01, -2.8025e-02,  3.6851e-01, -8.4399e-02,\n",
       "           2.6751e-01,  3.3341e-01,  1.2988e-01, -2.3298e-01, -1.0956e-01,\n",
       "          -3.9227e-01,  2.4031e-01, -9.8916e-02,  8.6010e-01,  2.3260e-01,\n",
       "           2.0943e-01, -4.3657e-03, -4.7262e-02,  4.8964e-01,  4.3099e-01,\n",
       "          -6.6541e-02, -1.6585e-01, -4.9522e-01,  4.6269e-03, -4.1279e-02,\n",
       "          -1.2963e-02, -4.5611e-01,  1.3753e-01,  4.2841e-01, -1.2416e-01,\n",
       "          -1.8354e-01,  1.5525e-01, -4.7715e-02, -3.1517e-01, -2.6492e-02,\n",
       "           6.8453e-01,  1.9080e-01, -2.2994e-01,  1.2377e-01, -1.1896e-01,\n",
       "           3.7006e-02, -4.6975e-01,  7.0044e-01,  6.1147e-01,  1.5585e-01,\n",
       "           1.6214e-02, -6.9943e-01,  2.3699e-01, -3.1574e-01, -4.7647e-03,\n",
       "          -1.5027e-01, -1.5731e-01, -4.2986e-01, -5.7285e-02, -5.5854e-02,\n",
       "           4.0115e-01,  7.9943e-02,  4.6653e-01,  1.1837e+00,  2.7973e-01,\n",
       "           1.9184e-02,  5.6097e-01,  2.0801e-01,  4.2492e-01, -4.0180e-01,\n",
       "           3.3082e-01,  8.8825e-02, -6.3896e-01,  2.7188e-01, -4.1593e-01,\n",
       "           4.4963e-02, -4.1432e-02, -3.7926e-01,  1.4208e-01, -2.1276e-02,\n",
       "           3.4237e-01,  3.3517e-01,  1.5186e-01,  1.6838e-01,  2.9411e-01,\n",
       "          -3.1322e-01,  5.9715e-01, -6.0809e-02, -1.1127e-01, -6.3492e-01,\n",
       "          -7.6732e-01,  9.5206e-02,  4.4743e-01, -1.2880e-01, -6.1279e-01,\n",
       "           1.0462e-01, -2.3866e-01, -1.2257e-01, -3.5064e-02,  2.1191e-01,\n",
       "           2.7455e-01,  5.0687e-02, -9.5436e-02,  6.4791e-02, -4.6950e-01,\n",
       "          -4.9277e-01,  4.9140e-02, -1.5758e-01, -1.3063e-01,  5.3918e-01,\n",
       "          -7.9415e-02,  1.1575e-01,  5.7221e-01,  1.9799e-02, -4.5976e-01,\n",
       "          -2.5640e-01,  1.4802e-01,  2.8667e-01, -7.2082e-02,  2.1981e-01,\n",
       "           3.4119e-01,  2.6467e-02,  5.1649e-01,  1.4823e-01, -2.1046e-01,\n",
       "          -2.7239e-01,  5.1173e-01, -1.5755e-01,  6.1403e-02, -1.2527e-01,\n",
       "          -8.7207e-01, -5.6150e-02, -7.8413e-02, -1.6833e-01, -5.4440e-01,\n",
       "           2.6566e-01,  1.4351e-01,  1.0561e-01,  2.2350e-01, -1.5984e-01,\n",
       "           7.0123e-01,  6.5285e-01, -3.3100e-01, -1.6943e-01, -4.0104e-01,\n",
       "           9.3922e-02, -6.8374e-02, -2.9329e-01, -1.4009e-02,  9.5368e-03,\n",
       "          -2.2765e-01, -1.2516e-01, -1.0805e-01,  2.8921e-01, -1.8876e-01,\n",
       "           5.2862e-01,  1.2189e-01,  8.3649e-02,  4.3873e-01,  2.0612e-02,\n",
       "           1.3071e-01,  3.6515e-01, -4.2117e-01,  1.2870e-01,  4.0424e-01,\n",
       "           4.0578e-01, -1.0551e+00, -8.3291e-01, -1.4936e-01,  3.6111e-01,\n",
       "           2.2860e-01,  2.5189e-01,  1.6999e-02, -7.2215e-03,  8.9455e-02,\n",
       "          -2.3402e-01,  3.5135e-01, -6.6415e-02, -5.7412e-01,  3.0561e-01,\n",
       "           2.3567e-01,  5.3263e-01, -2.7377e-01, -5.0579e-01, -6.5233e-02,\n",
       "           2.4616e-01, -3.6381e-01,  1.1392e-01,  2.3480e-01,  4.5466e-01,\n",
       "          -3.6015e-01, -5.6164e-01, -7.0282e-02, -7.3986e-01, -1.3578e-01,\n",
       "           3.2483e-01, -2.1437e-01,  2.7178e-02, -2.4758e-01,  1.5231e-01,\n",
       "          -1.7925e-02, -1.8879e-01,  6.0757e-01,  2.0522e-01, -2.5855e-01,\n",
       "           1.1981e-01, -2.3892e-01, -1.9475e-01,  5.0091e-01, -5.0226e-01,\n",
       "          -3.0348e-01, -5.0960e-02, -1.5349e-01, -3.2418e-01,  3.2181e-01,\n",
       "          -4.1389e-01, -1.4861e-01,  1.6748e-01, -6.4736e-02, -1.3068e-01,\n",
       "          -3.5233e-01,  1.4268e-01, -1.2707e-01, -4.9883e-02, -1.7654e-02,\n",
       "          -1.3353e-01,  6.0900e-02,  1.3308e-01,  6.4538e-02,  1.4388e-01,\n",
       "           5.3696e-01, -1.1464e-01,  2.7582e-03, -2.5900e-01, -2.2469e-01,\n",
       "           5.2498e-01, -9.4973e-02, -3.3117e-01,  5.4928e-01, -1.1156e+01,\n",
       "           1.0386e-01,  7.2631e-01, -5.5808e-01,  5.5128e-02, -2.8539e-01,\n",
       "           2.1085e-01, -3.0009e-01,  7.9101e-02,  2.4967e-01, -2.8460e-01,\n",
       "          -1.2277e-01, -2.7364e-01, -2.3193e-01, -4.0872e-01, -3.9742e-01,\n",
       "          -4.7229e-01, -5.6789e-01,  1.1772e-01, -1.7663e-01, -1.0996e-01,\n",
       "          -5.7036e-01, -1.9379e-01,  2.9696e-01, -4.9559e-03, -5.1069e-01,\n",
       "           5.9541e-02, -2.2948e-01,  2.6399e-01,  4.6126e-02,  2.2045e-01,\n",
       "          -1.9949e-02,  1.4108e-01,  2.0669e-01, -4.4946e-02, -4.5100e-01,\n",
       "          -4.4853e-01,  2.2129e-01, -1.5955e-01, -2.2267e-01, -5.4017e-02,\n",
       "          -3.3838e-01, -2.7689e-01,  1.3447e-01,  3.7594e-01,  6.2596e-01,\n",
       "           1.1595e+00,  5.6208e-01,  2.9245e-01, -1.0625e-01, -8.5438e-02,\n",
       "          -1.0762e+00,  5.5900e-01, -3.7318e-01,  8.1787e-02, -2.5119e-01,\n",
       "           6.6201e-01, -4.9906e-01,  5.4685e-01, -4.0754e-01, -2.5500e-01,\n",
       "           3.7179e-01,  3.0194e-01,  7.1104e-03,  2.8239e-02, -2.7432e-01,\n",
       "           4.0257e-01, -2.6094e-01, -3.6915e-02, -3.7819e-01,  4.0081e-01,\n",
       "          -3.9472e-01,  5.0770e-01, -5.6534e-02,  1.2945e-01,  1.8567e-01,\n",
       "          -3.5001e-01,  1.6610e-01,  1.0337e+00,  1.7414e-01,  4.1071e-01,\n",
       "           1.7420e-01,  5.1160e-01,  5.3064e-02, -8.8958e-01,  1.3691e-01,\n",
       "           2.0502e-01,  2.0789e-01, -6.7096e-02,  5.7131e-01, -2.5517e-02,\n",
       "          -6.5181e-01, -2.3094e-01, -3.2169e-01,  4.3108e-01,  6.5352e-01,\n",
       "          -2.2597e-01, -4.4343e-01,  2.3894e-01, -7.9776e-03, -5.3395e-01,\n",
       "           9.0074e-02, -2.2320e-01, -5.2792e-01,  8.7275e-02, -5.8914e-01,\n",
       "           1.4985e-01, -1.2428e-01,  5.4238e-02,  6.6887e-01, -4.0159e-01,\n",
       "          -1.6113e-01,  3.6602e-02, -4.4532e-02, -4.1270e-01, -4.9520e-01,\n",
       "          -7.7083e-02,  4.1073e-01, -1.9776e-01, -2.7976e-01,  7.0636e-02,\n",
       "          -3.0673e-01,  2.2584e-01, -1.7377e-01, -6.4935e-01,  3.2879e-01,\n",
       "          -6.0415e-01,  2.0578e-01,  2.5360e-01, -2.2507e-01,  4.9017e-01,\n",
       "           1.1337e-01, -8.1817e-02,  1.9397e-01, -9.9369e-02, -7.8554e-01,\n",
       "          -3.2573e-01, -2.7299e-02,  3.2662e-01, -4.2844e-01,  1.5776e-01,\n",
       "           1.5022e-01, -4.5752e-02, -2.2574e-01,  2.7904e-01,  1.0174e-01,\n",
       "           3.7762e-01,  3.9846e-01, -5.5006e-01,  1.1727e-01, -2.9867e-01,\n",
       "          -2.9262e-02, -6.1174e-02,  1.7952e-01,  2.3654e-01, -7.0764e-01,\n",
       "          -1.5224e-01, -2.6434e-01, -3.0965e-01, -2.1650e-01, -2.5120e-01,\n",
       "           2.0680e-01,  1.4038e-01, -1.5897e-02, -1.1308e-01,  5.8077e-01,\n",
       "           4.0814e-02, -1.1167e-01, -4.5997e-01,  3.1443e-01,  4.3178e-01,\n",
       "           2.1794e-01,  7.5875e-02, -1.2830e-01, -5.5532e-01, -1.1922e-02,\n",
       "          -4.3644e-01,  9.1081e-01, -7.3906e-01, -7.1970e-01,  6.6086e-02,\n",
       "           1.4951e-02, -8.8259e-04, -2.3450e-01, -5.0265e-01,  8.1298e-02,\n",
       "          -2.1017e-01, -1.4891e-01,  6.1550e-03, -1.0607e-02,  1.6299e-01,\n",
       "           4.0742e-01,  3.6838e-01,  1.5951e-01, -3.3938e-01, -4.4081e-01,\n",
       "          -6.0244e-01,  1.4567e-02, -1.6795e-01,  2.9330e-01, -2.6752e-01,\n",
       "          -3.2661e-01,  1.6043e-02, -9.1892e-03, -2.2272e-01,  5.2876e-01,\n",
       "           1.2427e-01,  1.6396e+00, -3.7287e-01, -4.4551e-01,  2.9018e-01,\n",
       "           7.2306e-02, -2.7073e-02, -2.8130e-01,  2.5074e-01,  9.5447e-02,\n",
       "           4.0194e-02, -3.6304e-01,  1.5023e-02, -2.0558e-02,  3.2773e-01,\n",
       "           3.2056e-01, -1.0703e+00, -2.0238e-01, -1.7993e-01,  5.5664e-01,\n",
       "           3.2388e-02,  6.3185e-01, -2.2305e-01,  3.0577e-01,  4.6106e-01,\n",
       "          -8.6667e-01, -2.3000e-01, -1.6129e-01,  2.7295e-01, -1.0583e-01,\n",
       "          -9.8735e-02, -1.6178e-01, -3.7165e-02, -1.6936e-01,  3.9538e-01,\n",
       "           7.4751e-01, -7.8426e-01,  7.4081e-01,  7.0467e-01, -9.8234e-02,\n",
       "           5.6815e-01,  9.4187e-02,  3.2641e-01, -1.4659e-01,  2.2669e-02,\n",
       "           2.6200e-01,  1.8542e-01, -6.3165e-01, -2.3962e-01,  2.4473e-01,\n",
       "           4.0158e-01, -1.1943e-01,  1.2108e-01,  7.9292e-02,  3.0960e-02,\n",
       "          -3.7691e-01, -1.0145e-01, -2.4552e-01, -3.9904e-01,  1.1356e-01,\n",
       "           1.4603e-02, -1.5001e-01, -8.6997e-02,  4.7709e-01, -1.7580e-01,\n",
       "           7.5067e-01,  4.2910e-01,  3.7145e-01, -3.5103e-01,  8.9950e-04,\n",
       "          -5.7565e-02,  6.0671e-01, -3.1670e-01,  7.7385e-01,  6.3337e-02,\n",
       "          -2.3140e-01,  2.7251e-01,  6.0547e-03,  1.7829e-01,  2.9892e-01,\n",
       "          -2.3519e-01,  1.2702e-01, -3.0718e-01, -2.3711e-01,  2.8330e-01,\n",
       "          -1.8640e-01,  1.1542e-01, -7.2270e-01, -6.3602e-03,  3.9857e-01,\n",
       "          -3.2051e-01,  4.9873e-01, -2.3074e-01,  6.1026e-01, -4.3509e-01,\n",
       "           3.2669e-02, -4.7470e-01, -4.3275e-01, -2.8508e-01,  5.1854e-02,\n",
       "          -8.6027e-01, -1.9453e-01,  1.7684e-01,  3.7405e-02,  2.1707e-01,\n",
       "           7.5155e-01, -3.3794e-01, -2.4771e-01, -2.0386e-01,  4.0190e-02,\n",
       "           1.8299e-01, -8.2071e-02, -3.5586e-01,  2.1161e-01, -3.5607e-01,\n",
       "          -1.0520e+00, -2.3117e-01,  8.9990e-02,  3.8419e-01, -3.5853e-01,\n",
       "           1.1570e+00,  2.5132e-02, -2.0226e-01,  6.7253e-01,  2.0808e-01,\n",
       "           6.4157e-01,  6.0323e-01,  1.3628e-03, -4.1259e-01, -5.4109e-02,\n",
       "          -1.5877e-01, -3.2881e-01,  1.5757e-01, -1.8346e-01,  5.0556e-01,\n",
       "           5.4802e-01, -2.0806e-02,  5.4756e-01]], device='mps:0'),\n",
       " tensor([[-0.0738, -0.7564, -0.3511,  ..., -0.0289,  0.4287,  0.3059],\n",
       "         [-0.2268, -0.9329, -0.6384,  ...,  0.2005,  0.5393,  0.6164]],\n",
       "        device='mps:0'),\n",
       " tensor([[-0.0907, -0.3410, -0.6425,  ...,  0.0212,  0.4597,  0.5145],\n",
       "         [-0.1476, -0.7085, -0.5662,  ..., -0.0943,  0.2921,  0.4532],\n",
       "         [-0.3292, -0.2590, -0.5016,  ..., -0.0148,  0.3509,  0.1607]],\n",
       "        device='mps:0'),\n",
       " tensor([[-0.1672, -0.5249, -0.4540,  ...,  0.3771,  0.2016,  0.5340],\n",
       "         [-0.3292, -0.2590, -0.5016,  ..., -0.0148,  0.3509,  0.1607]],\n",
       "        device='mps:0'),\n",
       " tensor([[-3.0175e-01, -6.6980e-01, -5.8058e-01, -1.0169e+00, -4.7769e-01,\n",
       "          -1.2967e-03, -6.6155e-01,  2.1546e-01,  1.0829e-01,  2.8188e-01,\n",
       "           2.6802e-01, -7.4553e-03,  8.8102e-02,  1.2687e-02, -2.8827e-01,\n",
       "           2.5617e-01, -2.6220e-01,  2.1438e-01,  7.7033e-02, -3.7727e-02,\n",
       "           8.1075e-01,  3.6768e-01, -1.6212e-01, -2.0573e-01, -3.3086e-01,\n",
       "           4.5605e-01, -1.3148e-01,  4.8222e-02,  3.5668e-01, -7.9078e-01,\n",
       "          -2.5202e-01,  1.1725e-01, -2.9749e-01,  1.0378e+00, -1.1594e-01,\n",
       "          -3.3333e-01, -1.7748e-01, -4.3787e-01,  2.1301e-01, -5.0568e-02,\n",
       "           4.5755e-01, -1.4778e-01, -2.4105e-01,  2.0005e-01, -1.2606e-02,\n",
       "           9.5042e-02,  3.7443e-01, -8.8176e-01,  3.3878e-01, -1.1266e-01,\n",
       "           1.8889e-01, -3.4865e-01,  5.0318e-01,  2.1380e-01, -8.5959e-01,\n",
       "           5.2285e-02,  2.7071e-01, -4.0013e-02, -8.2586e-01,  1.2128e-01,\n",
       "          -4.6670e-01,  5.9963e-01, -1.1233e-01,  1.0830e-01,  1.9111e-01,\n",
       "          -1.8788e-01, -5.8713e-01,  1.3197e-01, -2.4455e-01, -5.4474e-02,\n",
       "          -5.4218e-03,  6.5278e-03,  1.1025e-02,  5.1945e-02, -1.4253e-01,\n",
       "          -4.2182e-01, -1.2333e-01, -1.5679e-01, -4.8684e-01, -2.8943e-01,\n",
       "           1.1703e-01,  2.2252e-01, -9.0303e-02, -2.5004e-01,  1.6458e-01,\n",
       "          -2.2152e-02, -1.1224e-01,  2.6949e-01,  1.3030e-01,  8.0149e-02,\n",
       "           3.0522e-01,  2.0928e-01, -1.5859e-02, -2.0004e-01,  5.0817e-01,\n",
       "          -3.4280e-01,  5.8038e-01, -1.4811e-01, -1.0314e-01, -1.4254e-01,\n",
       "          -8.5900e-02,  1.6763e-01,  2.0530e-01,  3.5774e-01, -1.2954e-01,\n",
       "          -2.2287e-01, -1.0609e-01, -1.5258e-01, -6.7798e-01, -3.9055e-02,\n",
       "          -2.1100e-01, -3.5712e-01, -7.7076e-02, -6.5853e-01, -4.8448e-01,\n",
       "           3.3116e-01,  8.6550e-02, -3.0879e-01, -6.8888e-02,  6.9461e-01,\n",
       "           4.5933e-01, -4.6570e-01,  9.2314e-02, -2.0174e-01,  3.6437e-01,\n",
       "          -2.6446e-01,  3.1500e-01, -2.6301e-01,  1.6895e-01,  3.9658e-01,\n",
       "           2.4446e-01,  1.7780e-01, -2.5048e-01, -9.5926e-02, -4.5958e-03,\n",
       "          -2.0377e-01,  3.0765e-02, -1.7716e-01,  2.5017e-01, -1.6240e-01,\n",
       "          -1.1815e-01, -1.9138e-01, -3.8586e-01, -2.9267e-01,  2.0262e-02,\n",
       "          -3.8745e-01,  1.7052e-01,  2.0249e-01,  2.3481e-01,  2.0374e-01,\n",
       "          -5.9123e-01, -3.7090e-01,  2.6279e-01, -1.5877e-01, -1.7998e-01,\n",
       "          -2.7334e-01,  4.8190e-01,  1.1280e-01,  3.4257e-02,  4.9299e-01,\n",
       "           4.8370e-01, -3.4823e-01,  3.8388e-01, -5.6908e-01,  4.7970e-01,\n",
       "          -8.0979e-02,  2.2367e-02, -3.4129e-01,  6.6072e-01, -2.2073e-01,\n",
       "           4.2082e-01,  4.4552e-02, -5.8999e-02, -4.2000e-01,  2.1493e-01,\n",
       "           3.2119e-01,  5.5215e-01,  2.8715e-01, -2.9960e-01, -5.3479e-02,\n",
       "           5.6045e-02, -1.5361e+00,  1.7164e-01, -9.1554e-01, -2.4568e-01,\n",
       "          -2.9564e-01,  4.7877e-01, -1.0528e-01,  5.3352e-01,  2.0833e-01,\n",
       "           1.5799e-01,  2.2647e-01,  4.1759e-01,  6.8457e-01,  1.3707e-01,\n",
       "           3.0360e-01,  6.7200e-01,  5.1151e-01,  1.5972e-01, -9.9524e-02,\n",
       "          -8.4006e-01,  4.5848e-01, -2.4269e-01,  4.0564e-01, -4.0343e-01,\n",
       "           2.8765e-01, -4.5223e-01,  2.4525e-01,  5.3122e-01,  6.8031e-02,\n",
       "          -1.8708e-01, -3.3998e-01, -5.5491e-01, -1.4993e-01, -5.1624e-01,\n",
       "          -6.4144e-02, -3.5381e-01, -1.2625e-01,  2.2048e-01,  2.4657e-01,\n",
       "           2.5758e-01, -1.0877e-01,  8.6218e-02, -4.4103e-01, -2.5500e-01,\n",
       "           4.7314e-01, -4.2033e-02, -2.8710e-01, -2.4144e-01, -2.3728e-01,\n",
       "           2.8834e-02, -6.8555e-01,  5.9320e-01,  5.4095e-01,  2.8945e-01,\n",
       "          -3.7894e-01, -2.6274e-01,  3.3479e-01, -8.1051e-02, -1.9290e-01,\n",
       "          -3.2112e-01, -2.8164e-01, -6.3114e-01,  3.0389e-02,  3.2483e-01,\n",
       "           4.5472e-01,  2.2502e-02,  5.9776e-01,  9.2431e-01,  6.2616e-01,\n",
       "          -6.3152e-02,  4.1419e-01,  3.5893e-01,  4.0719e-01, -3.6893e-01,\n",
       "           2.3675e-01,  2.5578e-02, -3.0394e-01,  1.8930e-01, -3.4051e-01,\n",
       "          -5.4999e-01, -3.0810e-01,  1.8355e-01,  3.9312e-01, -4.0218e-01,\n",
       "           1.3738e-01,  3.6324e-01, -9.4898e-02, -2.0045e-01,  2.3112e-01,\n",
       "          -4.5769e-01,  2.7052e-01, -3.8193e-02, -2.4676e-01, -7.8228e-01,\n",
       "          -8.0561e-01,  1.7671e-01,  4.0383e-01,  2.7239e-02, -6.2392e-01,\n",
       "           4.0250e-01, -7.7847e-02, -1.6572e-01,  5.7036e-02, -2.0384e-01,\n",
       "           5.1000e-01, -2.4829e-01, -4.3042e-02,  2.3431e-01, -5.3414e-02,\n",
       "          -2.6091e-01,  2.8407e-01, -2.1272e-01, -1.4648e-01,  2.0817e-01,\n",
       "           1.1829e-01,  2.4851e-01,  2.9531e-01, -1.8469e-01, -1.0712e-01,\n",
       "          -1.4378e-01,  1.4497e-01, -3.1745e-01, -1.1662e-01,  2.7385e-01,\n",
       "           4.4367e-01, -8.9016e-02,  5.4896e-01, -2.2052e-01, -4.3292e-01,\n",
       "          -5.1639e-02,  7.8045e-01, -2.5523e-01,  4.0706e-01, -4.3595e-01,\n",
       "          -1.0067e+00, -2.3247e-01, -2.4278e-01, -1.6148e-01, -4.9052e-01,\n",
       "           3.3004e-01, -2.3283e-01,  9.8914e-02, -2.6420e-02, -2.8987e-01,\n",
       "           3.0162e-01,  6.6029e-01,  4.9695e-01, -3.8492e-02, -2.6121e-01,\n",
       "          -9.2459e-02, -1.8654e-01, -6.9434e-01, -2.8268e-01, -3.3254e-01,\n",
       "           2.9086e-01,  5.4987e-01, -9.4563e-03, -3.4511e-02, -2.2907e-01,\n",
       "           6.7731e-01,  4.2296e-02, -2.2174e-01,  4.3139e-01, -2.6503e-01,\n",
       "          -3.1949e-02,  4.2142e-01,  2.4983e-01,  1.4853e-01,  2.7700e-02,\n",
       "           3.8850e-01, -2.4014e-01, -5.2190e-01,  2.2986e-02,  7.3507e-01,\n",
       "           2.4180e-01,  4.1704e-01, -1.2385e-01,  6.9499e-01,  1.1516e-01,\n",
       "          -4.0155e-01,  1.9317e-01, -3.4905e-01, -9.7611e-02, -4.1392e-02,\n",
       "           2.1507e-01,  1.7865e-01, -4.6424e-01, -1.4887e-01, -3.2423e-01,\n",
       "           5.4446e-01, -4.4284e-01,  1.0868e-01,  6.0624e-01,  9.5384e-02,\n",
       "          -8.9992e-02, -5.1526e-01,  4.1846e-02, -9.2054e-01, -4.2575e-01,\n",
       "           1.3709e-03, -6.5500e-02,  3.1680e-01, -1.4904e-01, -6.4729e-02,\n",
       "          -9.6847e-03, -2.5828e-01,  6.6431e-01,  2.4458e-01,  2.2317e-02,\n",
       "           2.4482e-02, -4.2979e-01, -1.6413e-01,  7.3132e-01, -2.8516e-01,\n",
       "          -2.0146e-01,  2.4408e-01, -1.7036e-01,  7.5170e-02,  9.1916e-01,\n",
       "          -5.4459e-01, -1.2919e-01,  1.6822e-01, -9.2317e-02, -4.5496e-01,\n",
       "          -2.6936e-01,  3.7161e-01,  3.1036e-01, -2.3544e-01,  9.3051e-03,\n",
       "          -1.9547e-01, -4.8565e-02, -1.3911e-01,  1.4568e-01,  4.0195e-01,\n",
       "           3.8214e-01, -3.3720e-01, -9.5011e-02, -2.3665e-01, -1.5221e-01,\n",
       "           5.8983e-01,  4.1325e-02,  5.0294e-02,  4.4798e-01, -1.1088e+01,\n",
       "           3.1967e-01,  8.8400e-01, -4.8213e-01, -3.6244e-01,  1.7139e-01,\n",
       "           2.1689e-01, -3.2129e-01,  5.4136e-01, -3.4162e-01, -4.5191e-01,\n",
       "          -3.6412e-01, -2.6337e-01, -5.2896e-02, -5.3673e-01, -2.6602e-01,\n",
       "          -4.5073e-02, -6.6599e-01,  4.9827e-01, -5.6950e-02,  4.1197e-01,\n",
       "          -7.3141e-01,  2.9915e-01,  2.4368e-01, -2.5628e-01, -5.1343e-01,\n",
       "          -4.1890e-01,  5.6851e-02, -5.3908e-02,  1.5332e-01, -1.4587e-02,\n",
       "          -7.8227e-02, -1.4075e-01,  3.3179e-01, -1.0998e-01, -6.0080e-01,\n",
       "          -2.6099e-01, -1.5141e-01,  3.0497e-01,  3.4405e-01, -1.6240e-02,\n",
       "           1.4509e-01, -6.8089e-02,  4.0110e-01,  3.9231e-01,  8.2129e-01,\n",
       "           1.4654e+00,  3.3271e-01,  1.3907e-01,  1.6580e-01, -3.7775e-01,\n",
       "          -7.0736e-01,  3.5929e-01, -8.6010e-01,  1.6285e-02,  1.0245e-01,\n",
       "           4.0998e-01, -5.2030e-01,  1.8141e-01, -1.0184e-01, -2.9502e-01,\n",
       "           4.7893e-01, -4.8131e-02,  3.6590e-01,  6.5942e-02, -2.0825e-01,\n",
       "           3.6483e-01,  9.8140e-03,  1.0428e-01, -2.6566e-01,  7.0489e-02,\n",
       "          -1.0515e-01,  9.0629e-01,  5.6924e-02,  2.2025e-01,  3.5505e-01,\n",
       "           1.9906e-01,  5.6627e-02,  9.7982e-01,  1.4887e-02,  2.3439e-02,\n",
       "           2.4081e-01,  1.2057e-01, -8.2163e-01, -6.6525e-01, -3.0672e-01,\n",
       "          -1.9990e-01,  2.9884e-01,  2.1077e-01,  4.5599e-01,  1.0191e-01,\n",
       "          -2.4030e-01, -4.7529e-01, -8.4384e-02,  3.4405e-01,  2.7867e-01,\n",
       "          -8.1985e-01, -2.7656e-01,  3.8855e-01,  5.9943e-03, -3.3150e-01,\n",
       "          -1.0185e-01,  3.9454e-02, -6.9953e-01,  2.8364e-01, -8.8777e-01,\n",
       "           4.4082e-01, -1.1780e-01,  1.2180e-01,  3.3622e-01, -2.0828e-01,\n",
       "           1.4357e-01,  2.4272e-01,  1.5712e-01, -1.2949e-01, -3.2449e-01,\n",
       "          -6.2829e-02,  5.6284e-01, -3.0667e-01, -6.1434e-01,  7.8437e-02,\n",
       "          -4.1754e-01, -3.2532e-02,  2.0604e-01, -4.0847e-01,  2.9599e-01,\n",
       "          -9.2509e-01,  5.7991e-01,  5.3225e-01,  1.3132e-01,  4.2593e-01,\n",
       "           1.0221e-01,  2.7063e-01,  3.5230e-01, -3.5712e-01, -6.5350e-01,\n",
       "          -4.5414e-01,  1.1263e-01,  4.4292e-01, -4.3713e-01, -6.4580e-02,\n",
       "          -2.5887e-01, -1.8874e-02, -3.6240e-01,  4.6295e-01, -4.5580e-01,\n",
       "           5.0356e-01,  2.7012e-01, -3.2513e-01,  8.5783e-02, -9.8713e-02,\n",
       "          -3.8531e-02, -3.5799e-02, -9.5866e-02, -2.5031e-01, -6.3330e-01,\n",
       "          -2.7211e-01,  6.7283e-02,  1.7325e-01, -4.6052e-01, -4.7682e-01,\n",
       "          -2.3481e-01, -3.3311e-01, -3.1033e-01,  5.6504e-01,  4.5932e-01,\n",
       "           3.9209e-03, -2.8593e-01, -8.9424e-02,  6.8840e-02,  1.4673e-01,\n",
       "           5.1449e-01,  1.8069e-02, -1.3468e-01, -1.1048e+00, -2.6467e-01,\n",
       "          -2.6220e-01,  1.1618e+00, -5.0380e-01, -2.3770e-01,  1.6040e-01,\n",
       "          -3.5048e-01, -1.3186e-01,  3.4729e-01, -4.3105e-01, -2.8086e-01,\n",
       "          -3.8445e-02,  3.7695e-02,  4.9161e-01, -1.4846e-01,  4.3220e-01,\n",
       "           3.6677e-01,  1.2157e-01, -2.9789e-01,  1.2610e-01, -5.6793e-01,\n",
       "          -5.4133e-01,  1.5364e-01, -3.2761e-01,  2.5915e-01, -1.4858e-01,\n",
       "          -4.2085e-01,  4.1585e-02, -2.8757e-01,  3.6606e-02,  6.1003e-01,\n",
       "           1.4671e-01,  1.3806e+00, -3.4816e-01, -7.2524e-02,  2.0323e-02,\n",
       "           1.1091e-02,  2.8904e-01, -4.7353e-02, -2.8690e-01,  4.2239e-01,\n",
       "           3.8130e-01, -5.1135e-01, -2.1610e-02,  2.4837e-01,  8.4575e-02,\n",
       "           1.5806e-01, -7.0628e-01, -5.6971e-01, -4.5014e-01,  8.3924e-01,\n",
       "           4.2194e-01,  6.0600e-02, -1.3581e-01,  3.6786e-01,  6.1553e-01,\n",
       "          -7.2307e-01,  1.2407e-01, -5.6063e-01,  3.4624e-01, -2.8673e-02,\n",
       "          -3.8840e-02, -1.4214e-01, -4.1968e-01, -2.4922e-01,  5.1180e-01,\n",
       "           8.8041e-01, -8.0026e-01,  2.7926e-01,  2.8429e-01, -6.0185e-01,\n",
       "           2.7400e-01, -3.0303e-01,  4.9336e-01,  2.6875e-01, -1.5132e-01,\n",
       "           7.2616e-02,  3.0180e-01, -8.4330e-02, -4.9152e-01,  4.2695e-02,\n",
       "           6.6296e-01, -1.1219e-01,  1.9635e-01, -7.4417e-02, -4.0446e-01,\n",
       "           1.0495e-01,  3.4093e-02, -4.6593e-01, -1.8069e-01,  2.7051e-01,\n",
       "          -1.6951e-01, -5.1150e-01,  3.0621e-02,  3.2784e-01, -9.6532e-02,\n",
       "           3.0489e-01,  6.1836e-01,  1.8625e-01,  2.8380e-01, -2.6720e-01,\n",
       "          -1.7008e-01,  2.8282e-01, -3.1094e-01,  4.9413e-01, -2.5900e-01,\n",
       "          -2.8052e-01,  5.1395e-01, -5.1076e-02, -2.3065e-01,  1.3898e-01,\n",
       "          -1.9188e-01,  1.2284e-01, -2.0852e-01, -8.5311e-02,  1.3929e-01,\n",
       "           1.5890e-01, -2.0449e-01, -3.2401e-01, -1.5393e-01,  2.0152e-01,\n",
       "          -3.0676e-01,  6.9643e-01, -5.2115e-02,  2.7670e-01, -3.2876e-01,\n",
       "           3.2702e-01, -6.1801e-01, -3.5319e-01,  3.7352e-01,  4.6685e-03,\n",
       "          -4.5457e-01, -7.3337e-01,  3.5639e-01,  1.9663e-01,  6.8855e-01,\n",
       "           4.1244e-01, -4.6492e-02, -4.1667e-01, -3.7578e-01,  4.9655e-02,\n",
       "          -5.1082e-02, -1.9918e-01,  4.2887e-02,  2.6552e-01, -3.0464e-03,\n",
       "          -7.6106e-01, -5.7014e-01,  5.2966e-02,  3.2732e-01, -1.7900e-01,\n",
       "           7.9234e-01, -4.4085e-01,  6.0370e-02,  6.0840e-01,  4.8676e-01,\n",
       "           1.1522e-01,  7.3154e-01,  1.1035e-01, -7.9930e-01,  5.9814e-04,\n",
       "          -1.4466e-01, -2.9386e-01,  2.4061e-01, -9.9708e-02,  2.6837e-01,\n",
       "           3.2979e-01,  1.6883e-01,  7.6565e-01]], device='mps:0'),\n",
       " tensor([[-5.9860e-01, -4.5544e-01, -2.0495e-01, -9.9041e-01, -8.1314e-02,\n",
       "          -1.9984e-01, -7.7969e-01,  7.5656e-01,  3.6911e-01,  3.4839e-01,\n",
       "           1.8849e-01,  2.9132e-01,  1.1570e-01, -5.5162e-02, -4.2499e-01,\n",
       "          -2.6595e-02,  3.1957e-01,  1.8309e-01, -1.1967e-01,  3.5016e-01,\n",
       "           4.3796e-01,  7.1288e-01,  6.4353e-02, -6.4002e-02, -7.3509e-01,\n",
       "           3.4310e-01, -5.5715e-03,  1.1177e-01,  3.9750e-01, -7.5368e-01,\n",
       "          -2.1617e-01,  5.2113e-01, -4.5483e-01,  1.3612e+00,  7.8467e-02,\n",
       "          -1.1303e-01,  1.5515e-01, -2.2220e-01,  3.4746e-01, -1.4186e-01,\n",
       "           5.1737e-01, -2.4259e-01, -9.2146e-02,  4.7829e-01,  7.8971e-02,\n",
       "           2.7414e-02,  2.0566e-01, -7.1450e-01,  2.9711e-02, -3.9848e-01,\n",
       "          -2.6967e-01, -2.5453e-01,  3.8509e-01,  4.6869e-01, -1.0277e+00,\n",
       "          -3.5598e-01,  1.0431e-01,  1.2733e-01, -3.7737e-01, -4.0268e-01,\n",
       "          -6.9772e-02,  6.1548e-01,  1.8538e-01, -3.4681e-01,  6.0965e-02,\n",
       "          -2.2985e-01, -2.1532e-01, -9.3869e-01, -2.4700e-01, -3.0693e-01,\n",
       "          -3.1605e-01,  8.8882e-02,  3.3414e-01, -7.0922e-02,  1.4585e-01,\n",
       "          -5.6231e-03,  2.0138e-01, -1.5274e-01, -6.1270e-01, -1.9408e-01,\n",
       "           2.5172e-01,  3.4447e-02,  2.6500e-01, -2.0325e-01, -6.3303e-02,\n",
       "           3.7827e-02,  1.6505e-01,  9.7989e-02,  1.2768e-01, -7.6256e-02,\n",
       "           3.8571e-01, -2.8593e-02, -3.6835e-01, -3.2013e-01,  3.0865e-01,\n",
       "          -7.4527e-02,  3.4472e-01, -1.4833e-01, -3.8327e-02, -2.8776e-01,\n",
       "          -2.3267e-02,  1.4860e-01, -3.9167e-01,  1.1074e-01, -8.5394e-01,\n",
       "          -4.3707e-01,  2.9595e-01, -5.3644e-03, -5.6887e-01,  1.4143e-01,\n",
       "           3.2778e-02, -3.8304e-01, -1.8402e-02, -5.3814e-01, -1.7859e-01,\n",
       "           5.0414e-01,  2.2701e-01,  6.4475e-02, -1.6710e-01,  2.6412e-01,\n",
       "           7.0536e-01, -2.2707e-01,  3.2908e-01, -3.5696e-01,  2.0945e-02,\n",
       "          -4.4040e-01,  5.7535e-01, -1.6467e-01,  2.8593e-01,  3.1008e-01,\n",
       "           2.0934e-01,  4.3873e-01, -4.8224e-01,  1.5692e-01, -1.0650e-01,\n",
       "           1.3460e-01, -1.0051e-01, -3.2462e-01, -6.5179e-02,  2.2859e-01,\n",
       "          -2.5197e-01,  5.4853e-02, -1.1710e-01, -9.2754e-02,  9.9975e-02,\n",
       "          -1.0423e-01, -1.1969e-01,  2.4246e-01, -2.5604e-01, -6.0118e-01,\n",
       "          -7.7695e-01, -5.8951e-01,  3.9355e-01, -2.9351e-01, -1.0902e-02,\n",
       "          -4.2263e-01,  4.6809e-01,  2.7796e-02,  3.8687e-01,  4.2275e-01,\n",
       "           4.2025e-01, -2.0702e-01,  3.2037e-01, -6.4038e-01, -1.7148e-02,\n",
       "           6.1098e-02, -6.5288e-01, -6.6619e-02,  1.6748e-01, -1.2524e-01,\n",
       "           4.0588e-01, -3.0200e-01,  5.3869e-01,  4.9501e-01,  1.4246e-01,\n",
       "          -1.9903e-01,  1.1684e-01, -6.4403e-02, -4.4396e-02,  5.2325e-01,\n",
       "          -6.1588e-01, -1.5197e+00,  9.2848e-02, -5.8008e-01, -1.8335e-01,\n",
       "          -3.1158e-01, -1.5475e-02,  4.8052e-02,  5.9592e-01, -1.8403e-02,\n",
       "          -1.1464e-01,  1.6298e-01,  2.0782e-01,  5.7566e-01, -8.8849e-02,\n",
       "           4.5005e-01,  2.6423e-01,  3.6112e-01, -2.6686e-01, -2.0611e-01,\n",
       "          -4.6937e-02,  2.3812e-01,  2.9197e-01,  6.4920e-01,  2.0618e-01,\n",
       "           5.8734e-01, -1.2297e-02, -1.5561e-02,  6.0426e-02,  1.4034e-01,\n",
       "          -1.7315e-02, -1.7193e-01, -4.0371e-01, -8.6022e-02, -1.2144e-01,\n",
       "           3.0286e-01, -4.9510e-01,  1.9104e-01,  6.7924e-01,  2.4236e-01,\n",
       "          -1.6548e-01,  2.0552e-01,  3.3463e-02, -3.8871e-01,  7.9140e-03,\n",
       "           4.1203e-01, -2.3942e-01, -2.4862e-02,  5.4032e-02, -8.9140e-02,\n",
       "          -3.3784e-01, -1.9724e-01,  7.3610e-01,  1.5040e-01, -1.2010e-01,\n",
       "           1.7676e-01, -7.0618e-01,  1.7500e-01, -9.7911e-02,  8.2944e-02,\n",
       "          -4.3299e-01, -1.8721e-01, -4.1070e-01, -9.9211e-02,  1.8718e-02,\n",
       "           5.6319e-01,  2.1590e-01,  3.0955e-01,  5.6402e-01,  2.4497e-01,\n",
       "           6.6851e-02,  3.1579e-01,  1.8748e-01,  2.8609e-01, -1.8642e-01,\n",
       "           1.1467e-01,  2.7135e-02, -5.3980e-01,  5.6686e-01, -3.1413e-01,\n",
       "          -7.7845e-02, -1.3693e-02, -5.9498e-02,  1.1296e-01, -3.7215e-01,\n",
       "           2.2865e-01,  3.9566e-01,  8.1840e-02,  6.9792e-02,  5.5599e-01,\n",
       "          -2.5513e-01,  6.7303e-01, -1.2557e-01, -1.2991e-01, -5.2590e-01,\n",
       "          -9.3563e-01, -3.1319e-01,  1.9173e-01, -2.3722e-01, -5.0349e-01,\n",
       "           4.3298e-03, -2.7820e-01,  3.9957e-01,  9.0593e-02, -1.7225e-01,\n",
       "           2.6476e-01,  2.2508e-01, -1.9207e-01, -1.1053e-01, -3.7384e-01,\n",
       "          -2.3748e-01,  3.9715e-02, -2.5050e-01, -1.0968e-01,  3.2802e-01,\n",
       "           2.7639e-01, -3.5376e-01,  4.7770e-01, -9.9282e-02, -5.2693e-01,\n",
       "           1.0833e-01, -6.9226e-02,  3.3360e-01,  1.3319e-02,  2.1802e-01,\n",
       "          -5.0342e-04,  2.1066e-01,  4.4246e-01, -6.2338e-01, -2.6982e-02,\n",
       "           5.3141e-02,  5.6786e-01, -1.1649e-01, -2.8985e-01, -5.9144e-02,\n",
       "          -1.0719e+00, -3.2174e-01, -2.9860e-01, -1.2594e-01, -4.4880e-01,\n",
       "           2.6632e-01, -1.4700e-01,  4.2714e-02, -2.0183e-01, -3.7469e-01,\n",
       "           7.2825e-01,  6.5353e-01, -1.4404e-01,  9.2025e-02, -4.0752e-01,\n",
       "          -1.7940e-01, -3.1292e-01, -2.1093e-01, -1.5596e-01,  2.2651e-01,\n",
       "           5.6586e-02, -1.3885e-01, -4.6458e-01,  3.3275e-01, -3.6563e-01,\n",
       "           4.6088e-01, -2.3076e-01,  5.7267e-03,  4.9511e-01, -1.5569e-01,\n",
       "          -1.5665e-01,  3.1059e-01, -4.1362e-01,  1.6624e-01,  3.3708e-01,\n",
       "           2.0823e-01, -6.1248e-01, -7.4437e-01, -1.0138e-01,  3.7190e-01,\n",
       "           8.6987e-02, -1.1934e-01, -3.5410e-02,  2.9074e-01,  3.5005e-01,\n",
       "          -2.3241e-01,  4.3671e-01, -3.1977e-01, -3.3055e-01,  3.6318e-01,\n",
       "          -3.6896e-02,  2.2032e-01, -5.1169e-01,  1.7128e-01, -3.4785e-01,\n",
       "           2.4218e-01, -2.0741e-01,  6.2636e-02,  6.1095e-02,  8.1634e-01,\n",
       "          -1.1401e-01, -5.1077e-01, -6.8725e-02, -9.9887e-01, -4.2537e-01,\n",
       "           2.5691e-01, -7.7476e-02,  3.4377e-01, -3.2775e-01, -1.3090e-01,\n",
       "           2.2971e-02, -4.6052e-01,  5.7917e-01,  6.1515e-01, -4.1154e-02,\n",
       "           1.3773e-01, -2.6947e-02, -5.4190e-01,  5.8382e-01, -5.2608e-01,\n",
       "          -5.3940e-01, -1.1212e-01, -3.2412e-01, -1.7556e-01,  9.4227e-02,\n",
       "          -2.9402e-01, -2.2282e-01, -5.3876e-03,  9.2314e-02,  1.3085e-01,\n",
       "          -4.3219e-01,  8.3971e-02,  1.1902e-01, -1.3461e-01, -1.2422e-01,\n",
       "           1.2647e-01,  1.5106e-01, -1.3944e-01, -1.1859e-01,  2.6290e-01,\n",
       "           7.5986e-02,  3.1687e-02, -1.8678e-02, -2.4102e-02,  1.5125e-01,\n",
       "           2.5244e-01,  1.3433e-01, -4.9068e-01,  4.0921e-01, -1.1162e+01,\n",
       "           2.4978e-01,  6.4035e-01, -4.0446e-01,  2.7672e-01, -1.2858e-01,\n",
       "           1.8583e-02, -1.3081e-01, -1.3592e-01, -5.9788e-02, -2.2731e-01,\n",
       "          -4.2943e-01,  1.9695e-01, -1.1257e-01, -3.9354e-01, -3.4564e-01,\n",
       "          -1.8842e-01, -8.1544e-01,  2.0130e-01, -3.0130e-01,  2.9682e-01,\n",
       "          -3.4074e-01,  3.4454e-02,  2.5098e-01,  9.8259e-02, -3.6712e-01,\n",
       "          -2.8498e-01, -4.8045e-01,  3.8256e-01, -1.0146e-01,  1.9143e-01,\n",
       "           3.6459e-02, -3.0929e-01, -2.7625e-02,  6.3163e-02, -5.6803e-01,\n",
       "          -9.8806e-02,  1.6030e-01,  8.1632e-02,  1.6237e-01, -1.9051e-01,\n",
       "          -3.4941e-01, -1.9401e-01,  3.1898e-01,  3.7182e-01,  7.6721e-01,\n",
       "           9.1825e-01,  8.2098e-01,  1.7627e-01,  8.0398e-02, -2.6066e-01,\n",
       "          -9.5902e-01,  5.7870e-01, -5.3772e-01,  2.0727e-01, -4.4787e-01,\n",
       "           5.6822e-01, -6.0854e-01,  1.1967e-01, -1.3842e-01, -1.4657e-01,\n",
       "           2.6821e-01,  5.1104e-01, -1.5270e-01, -6.0409e-02, -2.9396e-01,\n",
       "           4.2650e-01, -1.0483e-01,  1.2202e-01, -4.3981e-01,  2.4418e-01,\n",
       "          -7.1877e-03,  1.7582e-01,  4.4853e-02, -1.1193e-01,  2.7812e-01,\n",
       "          -3.3125e-01, -1.4825e-01,  1.4142e+00, -1.6269e-01,  4.3625e-01,\n",
       "          -6.7999e-02,  3.3162e-01, -1.6429e-02, -9.3521e-01,  2.3904e-01,\n",
       "           3.2875e-01,  7.5479e-02, -5.5498e-02,  7.7080e-01, -4.0066e-02,\n",
       "          -8.7516e-01,  1.4240e-01,  3.3390e-01,  4.6987e-01,  8.1681e-01,\n",
       "          -2.7822e-01, -3.5752e-01,  1.6938e-01,  7.9915e-02, -2.0738e-01,\n",
       "           5.8706e-01, -2.2869e-01, -3.4771e-01,  1.7734e-01, -6.5578e-01,\n",
       "          -1.8643e-01, -4.5959e-02, -4.3168e-02,  5.4314e-01, -1.9200e-01,\n",
       "          -5.8171e-02, -1.3693e-01,  1.2850e-01, -3.9601e-01, -9.7987e-02,\n",
       "           9.4800e-02,  6.0267e-01, -2.1064e-01, -5.9371e-01, -2.1621e-01,\n",
       "          -2.2556e-01,  3.4079e-01, -1.6342e-02, -6.3582e-01,  6.1230e-03,\n",
       "          -4.9840e-01, -4.9863e-03,  1.7850e-01, -1.9080e-01,  6.9512e-01,\n",
       "           4.7588e-01, -9.7122e-02,  1.9378e-01, -6.9210e-01, -8.8465e-01,\n",
       "          -2.1439e-01, -3.1066e-01,  2.6278e-01, -4.9480e-01,  8.6109e-02,\n",
       "           2.9209e-01, -6.2350e-02,  3.0006e-03,  3.7832e-01,  3.1683e-01,\n",
       "          -4.5994e-02,  3.0766e-01, -5.6030e-01,  8.7494e-02, -6.5637e-02,\n",
       "          -2.8148e-01,  2.4568e-01,  1.8772e-01,  2.1387e-01, -6.3365e-01,\n",
       "           3.1140e-04,  6.0849e-02, -3.7324e-01, -5.5840e-01, -1.0580e-01,\n",
       "           3.6406e-01, -3.0931e-02,  1.0848e-01,  8.2628e-02,  1.2935e-01,\n",
       "           8.6632e-02, -4.8749e-01, -3.1145e-01, -1.8180e-02,  3.4713e-01,\n",
       "           4.8715e-01,  1.4755e-01, -3.5381e-01, -4.0795e-01, -8.2105e-02,\n",
       "          -1.7391e-01,  1.0365e+00, -5.4816e-01, -5.7670e-01,  6.2325e-02,\n",
       "           2.3082e-01, -2.9305e-02, -1.1714e-01, -6.4695e-01,  5.3211e-01,\n",
       "           9.3319e-02, -7.5362e-01,  2.4068e-02, -5.0026e-02,  3.5098e-01,\n",
       "           4.7221e-01,  5.7608e-01,  2.2900e-01, -2.3776e-01,  1.8777e-02,\n",
       "          -7.1182e-01,  3.3353e-01, -4.2238e-01,  5.9667e-02, -5.5355e-02,\n",
       "          -3.0194e-01, -1.9765e-01,  1.7957e-01,  9.0448e-02,  1.7757e-01,\n",
       "           4.4674e-01,  1.7031e+00, -3.3093e-01, -3.1251e-01,  4.7404e-01,\n",
       "          -1.6550e-01, -2.6870e-01, -2.1059e-01,  1.6694e-01,  1.6250e-01,\n",
       "           1.8284e-01, -4.2130e-01, -1.1938e-01, -2.4274e-01,  3.5567e-02,\n",
       "           3.0561e-01, -9.7901e-01, -2.3762e-01, -1.9769e-01,  6.1780e-01,\n",
       "          -4.4716e-02,  8.3731e-01, -4.5113e-02,  3.3406e-01,  3.4324e-01,\n",
       "          -8.4671e-01, -2.3422e-01, -2.7151e-01,  3.2004e-01, -2.4707e-01,\n",
       "          -3.4248e-02, -4.0783e-01, -3.6820e-01, -1.2496e-01,  5.4228e-01,\n",
       "           2.7681e-01, -9.0385e-01,  5.2613e-01,  5.6738e-01, -1.4871e-01,\n",
       "           2.6722e-01,  1.8826e-01,  3.7068e-01, -1.9473e-02, -3.5163e-01,\n",
       "           2.3784e-01,  1.9902e-01, -5.7395e-01, -5.6530e-01,  2.2680e-01,\n",
       "           5.6264e-01,  2.6752e-02,  3.4954e-03, -6.0508e-02,  1.1857e-01,\n",
       "          -1.3066e-01, -1.6192e-01, -9.0313e-02, -1.4879e-01, -7.0790e-02,\n",
       "           9.4811e-02, -5.3253e-02,  1.1092e-01,  3.9042e-01,  1.4010e-01,\n",
       "           7.5863e-01,  2.6301e-01,  4.5390e-01, -3.8667e-01, -2.8015e-01,\n",
       "          -2.8722e-01,  4.9595e-01, -3.6257e-02,  8.7691e-01,  5.0523e-02,\n",
       "          -1.5057e-02, -2.6750e-02, -2.4822e-01, -3.8220e-01,  2.2564e-01,\n",
       "          -8.6326e-02,  4.9772e-01,  1.8077e-01, -1.8360e-01,  4.8215e-01,\n",
       "          -2.6579e-01,  4.4952e-01, -5.6281e-01,  1.5924e-01,  4.9080e-01,\n",
       "          -3.2271e-01,  5.5379e-01, -3.9824e-02,  6.8040e-01, -4.1699e-01,\n",
       "          -1.3062e-01, -9.7449e-01, -1.7237e-01, -2.2420e-01,  2.5141e-01,\n",
       "          -7.6481e-01,  1.4507e-01,  2.2615e-01,  2.3917e-01, -1.3702e-03,\n",
       "           6.4555e-01, -1.9265e-01, -2.0890e-01, -2.5134e-01,  1.7888e-01,\n",
       "          -3.1306e-02, -6.6624e-02,  7.5423e-02, -2.7028e-03,  7.3544e-02,\n",
       "          -6.9967e-01, -2.6810e-01,  5.3730e-02, -3.6626e-02, -4.6412e-01,\n",
       "           9.2597e-01, -7.0027e-02,  2.9567e-01,  5.7658e-01,  2.6951e-01,\n",
       "           4.6745e-01,  5.4313e-01,  3.2018e-01, -7.4021e-01, -1.2987e-01,\n",
       "           6.4983e-02, -5.7094e-01,  9.6192e-02, -1.1553e-02,  2.7364e-01,\n",
       "           9.1178e-01,  5.5582e-01,  2.8817e-01]], device='mps:0'),\n",
       " tensor([[-1.7731e-01, -3.8604e-01, -5.6476e-01, -2.2650e-01, -1.3208e-01,\n",
       "          -3.1625e-01, -5.9292e-01,  4.5825e-01,  2.2151e-01,  5.2940e-01,\n",
       "           3.8502e-01,  1.4343e-01,  6.5615e-02, -2.5563e-01, -4.4222e-01,\n",
       "          -1.6916e-01,  1.6884e-01,  5.0107e-01,  3.6126e-01, -7.6046e-03,\n",
       "           5.9085e-01,  3.7327e-01, -2.0317e-01,  2.5562e-01, -7.5220e-01,\n",
       "           2.6747e-02, -1.6815e-01, -2.5680e-02,  9.6832e-02, -3.0143e-01,\n",
       "          -1.1712e-01,  2.7595e-01, -3.7936e-01,  7.2144e-01, -1.4269e-01,\n",
       "          -2.3945e-01,  1.6259e-01, -1.0450e-01,  5.3649e-01, -1.3024e-01,\n",
       "           5.7769e-01, -2.9111e-01,  4.9321e-01,  3.4162e-02,  2.2711e-01,\n",
       "           1.3206e-01,  2.9819e-01, -4.0790e-01,  1.3721e-01, -5.3671e-01,\n",
       "          -1.6362e-02, -7.3465e-02,  4.1546e-01,  1.0724e-01, -6.3685e-01,\n",
       "          -2.7220e-01, -7.3533e-02,  4.2370e-01, -3.4661e-01,  1.4839e-01,\n",
       "           1.4452e-01,  4.6875e-01,  1.2890e-01,  1.1326e-01, -3.3348e-01,\n",
       "           3.0093e-01, -7.0323e-01, -8.0455e-01, -5.7621e-01, -1.8468e-01,\n",
       "           2.6210e-01,  2.4685e-01,  3.2711e-01, -6.3178e-01,  2.8882e-01,\n",
       "          -5.5500e-01,  4.7778e-01, -3.7641e-02, -3.8107e-01,  2.4976e-02,\n",
       "           4.6662e-01,  1.6673e-01, -2.3893e-01,  2.6411e-01,  2.9965e-01,\n",
       "          -6.2089e-02,  4.4818e-01, -2.9706e-02,  5.6407e-02,  3.8863e-01,\n",
       "           5.0392e-01,  2.6142e-01, -5.4248e-01, -2.1306e-01,  2.3546e-01,\n",
       "          -3.7940e-01,  2.9621e-01, -1.0776e-01,  2.0922e-02, -8.6342e-02,\n",
       "          -3.6881e-02,  3.4334e-01,  3.1658e-01, -6.2528e-02, -1.0724e+00,\n",
       "          -4.3983e-01, -2.4311e-02, -2.7814e-01, -3.4665e-01, -1.0267e-02,\n",
       "           7.0862e-03, -2.1454e-01,  4.5842e-01, -4.4114e-01, -3.9926e-01,\n",
       "           1.3807e-01, -3.8646e-02, -1.5082e-02, -1.4568e-01,  6.0048e-01,\n",
       "           1.6661e-01, -5.3057e-02,  1.5341e-01,  8.8933e-02,  4.9861e-01,\n",
       "          -3.9381e-01,  3.7161e-01, -4.2227e-01,  3.0903e-01,  6.8681e-02,\n",
       "          -2.0183e-02, -7.2510e-02, -2.5207e-01, -2.0895e-01, -2.4617e-02,\n",
       "           2.1165e-02,  2.8627e-01, -4.8458e-01,  5.5727e-02,  5.6803e-02,\n",
       "           2.5004e-01, -1.2498e-01, -4.2667e-01, -1.9318e-01, -6.4563e-02,\n",
       "          -4.0302e-02,  1.9246e-01, -1.1367e-01, -2.7591e-01, -2.6032e-01,\n",
       "          -7.7253e-01, -4.4007e-01,  3.9722e-01,  1.0952e-01,  9.2693e-02,\n",
       "           1.1479e-01,  2.1754e-01, -1.9775e-01, -1.8512e-01, -1.5673e-01,\n",
       "           5.8319e-01, -9.5195e-02,  6.6991e-02, -3.6372e-01, -3.0891e-01,\n",
       "           7.3167e-02,  1.7092e-01, -1.7339e-01,  4.3863e-02,  1.4590e-02,\n",
       "           2.2225e-01, -8.8855e-02, -1.0134e-03, -7.5909e-01, -3.0093e-02,\n",
       "          -2.8356e-01,  3.3814e-01,  2.7840e-01,  1.8768e-01, -2.4537e-01,\n",
       "           5.0899e-02, -1.7294e+00,  7.8671e-01, -5.9166e-01,  2.4877e-01,\n",
       "          -1.8150e-01,  3.1718e-01,  9.1189e-02,  1.8509e-01, -2.9267e-01,\n",
       "          -1.2026e-01, -3.9309e-02,  4.1623e-02,  2.9227e-01, -8.4757e-02,\n",
       "           4.8559e-01,  5.5627e-01,  3.3170e-01,  2.0436e-01, -9.9033e-02,\n",
       "          -6.0977e-01,  3.3235e-01,  1.6388e-01, -2.7455e-01,  2.9857e-02,\n",
       "           2.4361e-01, -3.2791e-01, -5.7703e-02,  3.4800e-01,  3.3556e-02,\n",
       "          -8.5587e-02, -3.0852e-01, -4.0912e-01, -1.8006e-01, -1.6990e-01,\n",
       "           3.0904e-01, -2.3500e-01, -2.0646e-01,  7.4699e-01,  9.5515e-02,\n",
       "          -2.2466e-02, -7.9463e-02, -2.8277e-01, -4.4039e-01, -2.0940e-01,\n",
       "           2.9576e-01, -5.3435e-01, -1.7453e-01, -2.8854e-01, -4.6020e-01,\n",
       "           4.6218e-01, -9.4786e-01,  6.3073e-01,  2.5225e-01,  1.6028e-01,\n",
       "           6.5808e-02, -8.2977e-02, -1.3264e-01, -3.6344e-01, -1.9702e-02,\n",
       "          -2.1377e-02, -5.7642e-01, -9.0486e-01,  1.5472e-01, -7.9254e-02,\n",
       "           1.2900e-01, -4.1901e-02,  6.2316e-01,  4.2972e-01,  5.9118e-01,\n",
       "           2.3140e-01,  5.5355e-03,  4.6999e-01,  4.3441e-01, -4.1017e-01,\n",
       "           4.6498e-01,  1.8871e-01, -4.7838e-02,  6.9646e-02, -3.2186e-01,\n",
       "          -9.1404e-01,  2.0856e-02,  3.2947e-01,  4.9930e-01, -1.7114e-01,\n",
       "           4.2813e-01,  2.6662e-01, -1.3998e-01, -1.7031e-01,  3.1347e-01,\n",
       "          -4.8944e-01, -1.5127e-01,  9.5415e-02, -1.4029e-01, -6.5634e-01,\n",
       "          -4.8841e-01,  2.7357e-02, -2.9199e-01,  2.0484e-01, -9.9702e-01,\n",
       "          -2.6273e-01,  5.6883e-01, -2.8623e-02,  9.3732e-02, -2.3345e-01,\n",
       "           7.0240e-02, -1.0750e-01,  1.0928e-01, -1.5161e-01, -8.9822e-01,\n",
       "          -1.4258e-01,  9.6331e-02, -2.2318e-01,  3.2530e-01,  3.7809e-01,\n",
       "          -2.0083e-01, -2.9617e-01,  2.8364e-01, -9.0550e-02, -2.5039e-01,\n",
       "          -1.3545e-01, -3.7063e-01, -2.4607e-02, -1.3576e-01, -1.1638e-01,\n",
       "           6.9800e-02,  4.9085e-01,  4.8805e-01, -7.7597e-01, -5.5824e-01,\n",
       "          -2.2992e-02,  5.0756e-01, -1.6302e-01,  3.7266e-02, -2.2967e-01,\n",
       "          -1.0825e+00, -2.0374e-01, -1.1835e-01, -3.9640e-01, -3.8528e-01,\n",
       "           5.0161e-01, -1.9830e-01, -2.0671e-01,  9.6594e-02, -3.4539e-01,\n",
       "           3.4160e-01,  8.4200e-01, -4.9032e-02, -4.7416e-02, -2.9725e-01,\n",
       "           1.7477e-02, -1.8692e-01,  4.8213e-02, -1.9462e-01, -2.5485e-02,\n",
       "           3.1570e-01,  1.2741e-01, -1.8351e-01, -8.2635e-02, -1.7372e-01,\n",
       "           7.9347e-01, -5.6947e-01,  1.3088e-01,  3.8738e-01, -2.4876e-01,\n",
       "          -1.3259e-01,  4.8066e-01,  1.4439e-01,  1.9491e-01, -1.0024e-01,\n",
       "          -2.6493e-02, -2.8608e-01, -1.8423e-01, -3.1248e-01,  9.6616e-01,\n",
       "           3.5980e-01,  2.2368e-01, -3.0496e-01,  5.7904e-01,  5.6810e-01,\n",
       "          -3.8525e-01,  7.6086e-02,  8.8516e-02,  3.4151e-03, -3.1432e-01,\n",
       "           3.2810e-01,  2.4729e-01, -4.2766e-01, -2.5123e-02,  4.5374e-03,\n",
       "           1.4485e-01, -5.5797e-01,  1.6191e-01,  1.0782e-01,  8.5195e-02,\n",
       "          -1.4724e-03, -2.6111e-01,  5.2888e-01, -2.4531e-01, -2.8912e-01,\n",
       "          -1.0092e-01, -1.2182e-01,  4.1012e-02, -2.4548e-01, -3.6887e-01,\n",
       "          -2.7428e-01,  2.4193e-02,  3.5074e-01,  1.4285e-02,  7.4821e-02,\n",
       "           2.5544e-01, -2.6796e-01, -3.4638e-01,  4.5500e-01, -1.7660e-02,\n",
       "          -1.6810e-01,  4.0768e-01, -2.4890e-01, -1.1639e-01,  1.3881e-01,\n",
       "           6.2926e-02,  1.0677e-01, -5.8094e-02,  3.7998e-01,  1.7175e-01,\n",
       "          -2.8992e-01,  3.6344e-02,  2.1408e-01, -8.0245e-02,  3.1665e-01,\n",
       "           5.1861e-02,  8.2278e-01, -3.7109e-01, -2.7164e-02, -3.8441e-01,\n",
       "          -1.2821e-01, -1.6499e-01,  7.5490e-03,  1.5024e-01, -7.5773e-01,\n",
       "           3.0131e-01,  2.8842e-01,  3.4034e-01,  1.1331e-01, -1.1458e+01,\n",
       "           1.6305e-01,  6.4218e-01, -5.1374e-01,  1.8438e-01,  1.7594e-01,\n",
       "          -1.4040e-01, -3.9794e-01,  4.0875e-02, -3.4785e-01, -2.6394e-01,\n",
       "          -6.5590e-02,  1.9075e-01, -1.2756e-02, -5.5478e-01,  3.3680e-02,\n",
       "          -2.5398e-01, -6.3192e-01,  2.6160e-01, -9.0323e-02,  4.1335e-01,\n",
       "          -3.0876e-01,  1.8731e-01,  2.3832e-01, -3.6034e-01, -2.7614e-02,\n",
       "          -5.1547e-01, -5.5807e-02,  3.7042e-01,  3.2770e-01, -3.5254e-01,\n",
       "           2.2243e-01, -1.8729e-01,  3.8908e-01, -2.6449e-01, -5.2363e-01,\n",
       "           1.3092e-01, -9.6993e-02,  3.7952e-01,  1.1884e-02, -2.4878e-01,\n",
       "          -2.5432e-02, -5.2584e-01,  5.9952e-02,  2.0270e-01,  3.4279e-01,\n",
       "           1.2541e+00,  9.4924e-01,  1.3760e-01, -1.3198e-01,  1.6614e-01,\n",
       "          -7.3513e-01,  2.0677e-01, -6.2811e-01, -8.1585e-02, -6.4376e-02,\n",
       "           8.3128e-01, -4.3261e-01, -3.3743e-01,  7.3012e-02, -8.0493e-02,\n",
       "           3.1300e-01, -7.9471e-02,  2.6202e-01, -3.1186e-01,  2.9341e-03,\n",
       "           6.1375e-01, -1.8722e-02,  3.8063e-01, -4.3721e-01, -6.2672e-02,\n",
       "          -4.2457e-02,  2.9525e-01,  3.7279e-01,  6.2687e-02, -2.5474e-01,\n",
       "          -4.1450e-01,  4.2406e-02,  5.2243e-01,  1.3990e-01,  1.4541e-01,\n",
       "           5.8211e-01, -2.3165e-01, -4.1432e-01, -6.0104e-01, -3.9034e-01,\n",
       "          -7.6601e-03,  1.3684e-01, -3.9867e-01,  6.4750e-01, -4.2253e-01,\n",
       "          -2.7567e-01, -1.6427e-01, -8.8628e-02,  3.8882e-01,  5.0639e-01,\n",
       "          -3.3627e-01,  8.0837e-02, -1.8585e-01, -4.4707e-03, -6.7852e-01,\n",
       "          -3.5458e-01, -8.4538e-02, -2.9987e-01,  2.3231e-01,  1.7305e-01,\n",
       "          -1.0020e-01,  9.0697e-03,  1.2051e-01,  7.3101e-02, -5.4356e-01,\n",
       "           3.0479e-01, -1.0749e-01,  6.8380e-02, -3.9586e-01, -5.1756e-01,\n",
       "           4.0833e-01,  7.0232e-01, -4.2415e-01, -3.3873e-01,  9.8320e-02,\n",
       "           1.3073e-01,  1.8252e-01,  1.2615e-01, -6.4414e-01,  1.1900e-01,\n",
       "          -4.1931e-01, -1.3812e-01,  1.3512e-01,  7.2335e-01,  6.6950e-01,\n",
       "          -5.3703e-02, -3.2049e-01,  2.4632e-01, -5.0640e-02, -2.5765e-01,\n",
       "          -7.0535e-01, -2.9464e-02,  2.1372e-01, -5.3812e-01,  9.7345e-02,\n",
       "          -3.3445e-01,  1.9755e-01, -1.7248e-01,  1.0825e-01, -7.4644e-02,\n",
       "           4.1662e-01,  4.1987e-01,  7.6810e-02,  4.3515e-01, -1.7908e-01,\n",
       "          -2.1990e-01,  1.9709e-01, -1.7846e-01, -1.1703e-01, -1.7569e-01,\n",
       "          -1.7366e-01,  9.7459e-02, -4.0722e-01, -3.6435e-02,  1.7903e-01,\n",
       "          -1.2723e-01,  6.7948e-02, -3.4229e-01,  2.2739e-01,  3.3538e-01,\n",
       "          -1.0467e-01, -3.7966e-01, -2.3884e-01,  4.7521e-02,  7.1481e-02,\n",
       "           3.8731e-01, -2.6195e-01, -1.0102e-01, -7.3811e-01, -2.0636e-01,\n",
       "           3.6787e-02,  1.0410e+00, -3.7965e-02, -3.0778e-01,  3.5156e-01,\n",
       "           3.9051e-01, -2.4726e-01, -3.4583e-01, -1.8699e-01,  7.4840e-01,\n",
       "          -2.3598e-01, -1.5465e-01,  2.2099e-01,  2.5510e-02,  6.2240e-01,\n",
       "           1.3154e-01,  2.2300e-01, -1.8512e-01,  4.7019e-01,  2.1848e-01,\n",
       "          -4.4102e-01, -1.5073e-01, -3.9303e-01, -3.2472e-01,  4.3754e-01,\n",
       "          -6.9776e-02,  2.7454e-01, -1.1790e-01,  8.1553e-02,  1.4253e-02,\n",
       "           5.4206e-02,  1.1645e+00, -2.0185e-01, -2.3960e-01,  4.7684e-01,\n",
       "           2.7079e-01,  3.4262e-01,  3.7917e-02, -1.3477e-01,  9.5921e-02,\n",
       "           4.9705e-01, -1.6706e-01, -4.9624e-01,  1.5664e-01,  3.6272e-01,\n",
       "           5.9108e-01, -8.7613e-01, -3.6622e-01,  2.0509e-01,  2.2296e-01,\n",
       "          -3.2121e-01,  1.9818e-01, -7.6228e-02,  3.2510e-01,  3.6266e-01,\n",
       "          -1.2771e+00,  1.7642e-01,  7.1262e-02,  1.5943e-01, -1.5269e-01,\n",
       "          -6.7715e-01, -2.5385e-01,  1.0595e-01, -6.2916e-02,  4.2076e-01,\n",
       "           4.4789e-01, -7.1509e-01,  1.2452e-01,  3.1511e-01, -3.7691e-01,\n",
       "           1.2320e-01, -2.7571e-01,  1.6643e-01,  5.2170e-01,  1.6129e-01,\n",
       "          -4.2505e-01,  2.5049e-01, -5.0826e-01, -6.1173e-01,  1.4561e-01,\n",
       "           6.9575e-01,  2.1324e-02, -1.1322e-01, -1.1804e-01, -2.9289e-01,\n",
       "          -1.9107e-01, -3.1790e-01, -3.4967e-01, -2.7542e-02,  1.4802e-01,\n",
       "           2.4431e-02, -3.6606e-02,  2.4884e-01,  1.1143e-01, -2.2300e-01,\n",
       "           5.8287e-01,  4.8747e-01,  7.4334e-01, -1.5062e-02,  1.1963e-02,\n",
       "          -2.2207e-01,  2.9970e-01, -4.7196e-02,  4.7552e-01, -3.4433e-01,\n",
       "          -2.0911e-01,  2.2006e-01,  3.6212e-02, -6.2294e-01, -1.0381e-01,\n",
       "          -3.5492e-01,  3.0044e-01,  1.1075e-02, -2.7609e-01,  4.8880e-01,\n",
       "           2.0553e-02,  1.8280e-01, -5.5884e-01,  4.6756e-01,  1.6596e-01,\n",
       "           2.2073e-02,  9.8054e-01,  1.1685e-01,  5.3145e-01,  5.8221e-01,\n",
       "           1.1828e-01, -8.1138e-01, -1.8763e-01, -7.2256e-02,  1.5035e-01,\n",
       "           1.1544e-02,  1.5872e-03,  1.5741e-01,  2.8459e-02,  9.3799e-01,\n",
       "           4.5776e-01, -2.2136e-01, -3.9686e-01, -3.7782e-01,  1.8491e-01,\n",
       "          -8.4548e-02, -3.3508e-01,  2.0442e-01, -2.2187e-01,  1.7388e-02,\n",
       "          -5.8935e-01, -5.5885e-01, -1.2302e-01,  2.7758e-01, -2.6880e-01,\n",
       "           5.4517e-01,  4.6664e-02,  5.2447e-03,  3.4141e-01, -2.9763e-02,\n",
       "           2.1521e-01,  6.7382e-01,  2.2993e-01, -2.9428e-01,  1.8031e-01,\n",
       "           2.1968e-01, -1.3524e-01,  2.0995e-01, -8.6762e-01,  3.6296e-01,\n",
       "          -5.1872e-02,  5.4038e-01,  4.7469e-01]], device='mps:0'),\n",
       " tensor([[-0.2841, -0.9073, -0.8214,  ...,  0.1888,  0.6933,  0.5187],\n",
       "         [-0.2643, -0.8360, -0.8564,  ...,  0.1610,  0.5643,  0.6620]],\n",
       "        device='mps:0'),\n",
       " tensor([[-2.9081e-01, -3.4246e-01, -6.1361e-01, -8.6322e-01, -1.4702e-01,\n",
       "           2.9555e-02, -7.9563e-01,  7.2588e-01,  1.8982e-01,  5.0297e-01,\n",
       "           1.4437e-01,  4.5770e-01,  1.1316e-02, -2.0062e-01, -1.5305e-01,\n",
       "          -4.9358e-01,  4.4290e-02,  4.0279e-01,  6.7501e-01,  1.3703e-02,\n",
       "           2.5296e-01,  6.8128e-01, -2.7320e-01, -1.2174e-02, -9.1870e-01,\n",
       "           3.1070e-01, -1.4332e-01,  1.1436e-01,  2.1067e-01, -6.5824e-01,\n",
       "           3.7449e-02,  4.9470e-01,  1.0611e-01,  1.4655e+00, -5.9531e-01,\n",
       "          -1.6066e-01,  1.0024e-01, -3.4678e-01, -9.8152e-02,  2.6610e-02,\n",
       "           5.3161e-01, -3.9453e-01,  1.8502e-01,  4.9419e-01, -6.1832e-02,\n",
       "           9.4301e-02,  1.1348e-01, -3.9382e-01,  2.6957e-01, -8.2592e-01,\n",
       "          -1.0821e-01, -1.6103e-01,  4.1228e-01,  5.2381e-01, -7.1708e-01,\n",
       "          -2.3165e-01,  2.5715e-01, -7.7403e-02, -5.5917e-02,  1.6285e-01,\n",
       "           1.4140e-01,  4.6158e-01, -2.9742e-01,  3.2124e-01,  3.6779e-01,\n",
       "           2.6087e-01, -2.5505e-01, -5.7454e-01, -5.5802e-01,  3.3219e-02,\n",
       "           2.1946e-03,  5.1744e-02,  2.1430e-01, -2.1326e-01,  3.5179e-01,\n",
       "          -7.3041e-01, -9.1061e-02, -8.0725e-02, -2.6825e-01,  3.7433e-01,\n",
       "          -7.7350e-02,  2.0975e-01,  3.6683e-01,  1.1851e-01,  1.5445e-01,\n",
       "           1.0644e-02,  2.3474e-01, -3.0152e-02,  6.9133e-02,  3.9366e-01,\n",
       "          -2.2777e-01, -4.6005e-02, -2.3522e-01, -3.5811e-01,  3.3577e-01,\n",
       "          -4.4921e-02,  2.8690e-01, -1.7314e-01,  2.0282e-01, -3.7063e-01,\n",
       "           1.2787e-01,  1.9046e-01, -2.5722e-01,  7.3058e-02, -4.4702e-01,\n",
       "          -4.2942e-01,  1.3233e-02, -2.3976e-01, -4.4031e-01,  2.5484e-01,\n",
       "          -1.4660e-01, -3.2896e-01, -5.3408e-02, -3.7252e-01,  6.9930e-02,\n",
       "          -2.6884e-02,  3.0832e-01, -1.0944e-01, -2.6649e-01, -1.0607e-01,\n",
       "           2.5796e-01, -2.7805e-01,  1.8230e-01,  2.9883e-01,  1.0908e-01,\n",
       "          -1.2859e-01,  6.7095e-01,  5.4591e-02,  3.5249e-01,  1.8430e-01,\n",
       "           1.3247e-01, -7.2004e-02, -1.8499e-01, -2.2456e-01,  9.0717e-02,\n",
       "          -8.4815e-02,  1.0671e-01, -1.7169e-01, -2.4376e-01, -2.1581e-01,\n",
       "           3.3059e-01, -3.1506e-01, -6.6383e-01, -5.2244e-01, -1.8615e-01,\n",
       "          -1.7100e-01, -7.6204e-02, -2.2739e-02, -3.1201e-01, -4.4289e-01,\n",
       "          -7.4636e-01, -2.8709e-01,  7.9972e-01, -5.7533e-02, -3.6645e-01,\n",
       "          -1.8557e-01,  3.0437e-01,  3.7106e-01, -1.0042e-01,  4.1648e-01,\n",
       "           7.7684e-01, -4.0635e-02,  1.2543e-01, -5.5030e-01, -4.4700e-01,\n",
       "           3.2396e-02,  2.9258e-01, -3.1212e-01,  6.9698e-01,  6.8279e-02,\n",
       "           5.4975e-02, -9.4131e-03,  3.6377e-01, -2.8036e-01, -2.2490e-02,\n",
       "          -5.7097e-01,  3.6908e-01,  3.1411e-01, -3.8492e-01,  6.5335e-03,\n",
       "          -3.0718e-03, -1.3254e+00,  7.9837e-02, -8.7727e-01,  2.8398e-01,\n",
       "          -1.9191e-01,  1.9271e-01,  2.0876e-01,  3.2324e-01, -6.1898e-02,\n",
       "           3.2284e-01,  3.2249e-01,  1.6788e-01,  2.6400e-01, -7.3091e-02,\n",
       "           2.3737e-01,  1.2526e-01,  4.8843e-01,  3.6065e-01,  1.2071e-01,\n",
       "          -2.9076e-01,  3.0741e-01,  3.7339e-01,  3.0259e-02,  1.3957e-01,\n",
       "           9.9413e-02, -2.5778e-01,  1.7601e-01,  1.3098e-01, -1.1175e-01,\n",
       "          -1.5187e-01, -3.0435e-02, -4.8504e-01, -1.1352e-01, -5.2726e-01,\n",
       "           4.0771e-01, -5.5564e-01, -1.1305e-01,  4.1230e-01,  2.7225e-01,\n",
       "          -3.3799e-02,  1.8883e-01,  8.2140e-02, -2.3442e-01,  1.7880e-01,\n",
       "           4.5898e-01, -3.9206e-01, -4.6603e-02, -3.7320e-01, -3.0302e-01,\n",
       "           2.3759e-01, -5.5579e-01,  7.1978e-01,  4.6788e-01,  7.6883e-02,\n",
       "           1.9113e-01, -2.5794e-01, -1.4381e-01, -6.5846e-01,  3.3800e-01,\n",
       "          -3.3778e-01, -3.1129e-01, -5.6699e-01,  7.7491e-02,  4.9894e-01,\n",
       "           2.7724e-01, -6.6076e-02,  7.8137e-01,  3.8867e-01,  3.5353e-01,\n",
       "          -4.7208e-01,  5.9533e-02,  4.1616e-01, -1.5283e-02, -4.5259e-01,\n",
       "           1.5769e-02, -8.9152e-02, -4.5754e-01,  3.1081e-01, -2.6948e-01,\n",
       "          -4.0523e-02, -1.0804e-01,  2.7430e-01,  1.3430e-01, -2.3669e-01,\n",
       "          -4.2311e-01,  4.3373e-01, -1.4084e-01, -4.3205e-01,  4.4138e-01,\n",
       "          -3.7218e-01,  1.8896e-01,  3.1535e-02, -3.7750e-01, -6.2929e-01,\n",
       "          -3.1492e-01,  2.2351e-01,  1.7066e-01,  3.0821e-02, -7.3396e-01,\n",
       "          -3.4176e-01,  3.3295e-01,  3.1491e-01,  3.1869e-01,  2.9044e-01,\n",
       "           6.2194e-01, -3.3788e-01, -1.9488e-01,  1.6709e-01, -4.2424e-01,\n",
       "          -9.9766e-02, -6.7796e-02, -2.1129e-01,  2.8430e-01,  2.0728e-01,\n",
       "           1.3746e-01,  2.1763e-01,  4.5221e-01, -3.1802e-01, -3.7308e-01,\n",
       "          -7.0759e-02,  1.3849e-01, -2.5263e-02, -5.3610e-02,  2.2123e-01,\n",
       "          -6.4396e-02,  6.2242e-01,  7.5648e-01,  1.3055e-01, -6.5010e-01,\n",
       "           1.4659e-01,  5.1771e-01, -2.9651e-01, -1.0113e-02,  1.6700e-01,\n",
       "          -1.1184e+00, -2.4027e-01, -8.6775e-02, -2.1056e-01, -5.9125e-01,\n",
       "           7.2027e-02,  4.3137e-01,  1.8076e-02, -2.0543e-01, -2.7052e-01,\n",
       "           9.5510e-02,  1.1981e-01, -1.4153e-01,  5.1287e-01, -3.3815e-01,\n",
       "          -3.0536e-01, -2.0003e-01, -2.3677e-01, -5.7483e-01, -4.8991e-01,\n",
       "           3.5148e-01,  2.2996e-01,  1.0467e-01, -3.5152e-01, -4.9799e-01,\n",
       "           4.5495e-01, -7.0870e-01, -1.7862e-01,  2.5698e-01, -4.8046e-01,\n",
       "          -4.3825e-02,  5.1606e-01,  1.0322e-01,  2.1133e-01,  7.5907e-02,\n",
       "           8.5243e-02, -2.9797e-01, -4.6131e-01, -1.4776e-01,  6.7127e-01,\n",
       "           2.6160e-01,  3.5839e-01, -1.7753e-01,  3.5534e-01,  4.9747e-01,\n",
       "          -2.8540e-01,  4.8430e-01,  2.5910e-02,  6.3767e-02, -6.4572e-02,\n",
       "           1.7880e-01,  3.2566e-01, -3.8045e-01,  2.9748e-01,  1.2423e-01,\n",
       "          -4.2083e-01, -3.9629e-01,  8.7954e-03,  2.0341e-01, -1.6362e-01,\n",
       "          -3.0513e-01, -2.1567e-01,  1.3845e-01, -3.6132e-01, -4.7115e-01,\n",
       "           3.3023e-01, -9.8021e-02,  5.7053e-02, -4.3758e-01, -4.1566e-01,\n",
       "          -2.7373e-01, -2.7277e-01,  6.8827e-01,  2.9845e-01, -1.7444e-01,\n",
       "           2.0833e-01, -1.0279e-01, -3.2641e-01,  5.4818e-01, -3.7558e-02,\n",
       "          -4.9551e-01,  4.8053e-01, -3.4858e-01,  7.3427e-02,  5.9145e-01,\n",
       "          -1.8349e-01, -1.1045e-01,  5.9823e-02,  5.4361e-02,  3.0851e-01,\n",
       "          -3.9709e-01,  1.4324e-01, -3.1543e-02, -2.7796e-01,  3.3399e-01,\n",
       "           2.5296e-01,  2.4322e-01, -1.6547e-01, -1.0487e-01,  1.0004e-01,\n",
       "           3.0643e-01, -1.2382e-01, -4.0334e-01, -5.2873e-01, -4.0032e-01,\n",
       "           1.2463e-01,  1.8654e-01,  5.3235e-01,  2.9334e-01, -1.1226e+01,\n",
       "           1.8872e-01,  7.7968e-01, -4.6649e-01, -9.0135e-03, -5.1439e-02,\n",
       "          -4.2698e-02, -1.1157e-01,  4.0243e-01, -2.2953e-01, -4.5989e-01,\n",
       "          -3.7077e-01, -1.6350e-01,  1.1649e-01, -3.7756e-01, -2.7495e-01,\n",
       "          -1.4786e-02, -7.0837e-01,  4.1731e-01, -2.7579e-01,  5.1562e-01,\n",
       "          -1.6539e-01, -1.6084e-01,  1.8167e-01, -3.5050e-01,  5.1494e-02,\n",
       "           1.5740e-01, -1.9956e-01,  9.2161e-02,  4.2488e-01, -6.5572e-02,\n",
       "           4.0860e-03, -1.4096e-02,  5.6879e-02, -2.3899e-01, -5.6600e-01,\n",
       "           1.3975e-02, -1.9311e-01,  3.6576e-01, -5.2175e-02, -7.7538e-01,\n",
       "          -2.8429e-01, -3.4296e-01,  2.8488e-01,  3.4896e-01,  8.9941e-01,\n",
       "           4.1432e-01,  1.1548e+00,  2.0407e-01,  2.1170e-01,  2.2573e-02,\n",
       "          -8.2125e-01,  4.0045e-03, -3.7539e-01,  1.4738e-02,  8.9116e-02,\n",
       "           3.8608e-01, -8.2107e-01,  3.3754e-01, -3.4994e-01,  4.4553e-02,\n",
       "           5.1407e-01,  6.7545e-02,  2.8395e-01, -1.8858e-02, -5.9149e-02,\n",
       "           8.4166e-02, -2.9320e-02,  1.0342e-01, -2.1000e-01, -1.8696e-01,\n",
       "          -1.1408e-01,  6.4027e-01,  4.0303e-01, -7.6878e-01,  3.9564e-01,\n",
       "          -9.0177e-03, -1.8766e-01,  1.2889e+00,  1.9062e-03,  5.3525e-02,\n",
       "          -5.2373e-02,  1.9753e-01, -4.3513e-01, -7.5634e-01,  3.9382e-01,\n",
       "          -9.2386e-02,  3.1987e-02,  4.8932e-02,  5.1488e-01, -4.4782e-02,\n",
       "          -5.6284e-01,  1.8424e-02,  1.1922e-01, -1.6813e-01,  5.4050e-01,\n",
       "          -7.5339e-01, -2.2061e-01, -1.1110e-01,  3.0232e-01, -8.1961e-01,\n",
       "          -7.6965e-02, -7.2943e-01, -3.6169e-01,  4.4102e-01, -2.9278e-01,\n",
       "           1.6572e-01,  1.8525e-02, -7.8544e-02,  5.8231e-01, -2.2101e-01,\n",
       "           2.9185e-02,  2.1157e-01,  1.3322e-01, -2.3389e-01, -9.2427e-03,\n",
       "           3.6302e-01,  5.0553e-01, -4.1878e-01, -6.1246e-01,  4.3244e-02,\n",
       "           2.2746e-01,  2.0461e-01,  3.1006e-01, -5.6253e-01,  3.3156e-01,\n",
       "          -7.6507e-01, -2.6103e-01, -2.6415e-02, -6.5982e-02,  3.2392e-01,\n",
       "           2.2773e-01,  5.8764e-02,  2.7414e-01, -7.6456e-01, -6.7247e-01,\n",
       "          -6.6072e-01, -2.5461e-01,  2.9526e-01, -5.6322e-01, -1.3924e-03,\n",
       "          -1.6405e-01,  6.9548e-03,  1.1208e-02, -9.3658e-02, -1.4767e-02,\n",
       "           3.5045e-02,  2.6346e-01, -1.2895e-01, -1.4748e-01, -9.9612e-02,\n",
       "          -5.6071e-01, -1.4993e-02, -7.0952e-02, -2.0604e-01, -6.4565e-01,\n",
       "          -1.2499e-01,  1.7052e-02, -2.8379e-01, -7.4496e-02,  3.7789e-01,\n",
       "          -1.0053e-01, -3.4002e-02, -4.9457e-01,  8.2264e-01,  4.8846e-01,\n",
       "           5.8259e-02, -2.9293e-01, -6.5764e-01, -8.5947e-02,  2.5510e-01,\n",
       "           2.5986e-01, -4.7233e-02, -1.4916e-01, -2.6185e-01, -3.7392e-01,\n",
       "           1.7138e-01,  7.1575e-01, -7.1715e-02, -7.5529e-01,  2.6483e-01,\n",
       "           6.4045e-01, -1.2545e-01, -2.9774e-01, -3.2146e-03, -1.8685e-02,\n",
       "          -2.9010e-01, -5.7084e-01,  7.5440e-01, -7.5690e-02, -7.7504e-02,\n",
       "           2.0478e-01,  2.6735e-01,  2.4623e-02,  4.3622e-01, -1.8759e-01,\n",
       "          -3.1093e-01, -3.7286e-02, -6.9741e-01,  4.7492e-01,  9.9028e-02,\n",
       "          -3.5227e-01,  7.5104e-01,  2.9523e-01, -1.2125e-01,  3.3642e-02,\n",
       "           2.6925e-01,  8.9505e-01, -3.1073e-01,  4.8013e-02,  5.9103e-01,\n",
       "           4.2431e-02, -1.4203e-01,  9.3838e-02,  1.0584e-02, -4.3494e-02,\n",
       "           3.3119e-01,  5.7846e-02, -4.6939e-01, -1.0917e-01,  3.5860e-01,\n",
       "           4.0143e-01, -6.0677e-01, -7.6633e-01,  1.5120e-02,  7.5098e-01,\n",
       "          -3.0635e-01,  2.1695e-01, -3.8207e-03,  2.7657e-01,  2.6928e-02,\n",
       "          -9.6982e-01, -1.5315e-01, -1.1560e-01,  3.8632e-01,  7.9077e-03,\n",
       "          -1.1736e-01, -1.4485e-01, -1.7340e-01, -1.6275e-01,  7.1849e-01,\n",
       "           2.2133e-01, -6.5045e-01,  5.9786e-02,  5.2007e-01, -6.8016e-01,\n",
       "           5.0422e-01, -1.7132e-01, -9.2284e-02,  3.2389e-01,  1.8685e-01,\n",
       "           4.7780e-02,  1.8465e-01, -4.4811e-01, -4.3728e-01,  1.8665e-01,\n",
       "           9.0192e-01, -4.7576e-02,  8.7023e-02, -3.7908e-01, -1.4860e-01,\n",
       "          -1.0458e-01, -3.0311e-01, -3.3159e-01,  3.9705e-01,  2.1882e-01,\n",
       "           1.9242e-02, -7.2473e-02, -2.4511e-01,  2.9653e-01,  1.0007e-01,\n",
       "           3.3009e-01,  6.0030e-01,  6.4647e-01, -5.1901e-02, -2.5065e-01,\n",
       "          -1.4112e-02,  1.6173e-01, -2.3909e-01,  5.0775e-01, -1.6140e-01,\n",
       "          -3.0625e-01,  1.7359e-02,  2.7768e-01, -6.3036e-01, -1.2381e-01,\n",
       "          -4.1311e-01,  4.7585e-01, -1.0176e-01,  2.9503e-02,  3.0193e-01,\n",
       "           3.6938e-02,  2.8693e-02, -2.2905e-01,  5.4198e-01,  4.8472e-01,\n",
       "           1.9507e-01,  4.4784e-01,  1.6693e-01,  7.2157e-02, -4.7641e-01,\n",
       "           2.7368e-01, -1.0420e+00, -7.5432e-02, -8.4596e-02, -1.2210e-01,\n",
       "          -4.7481e-01, -4.4736e-02,  2.0176e-01,  1.6941e-01,  6.3582e-01,\n",
       "           7.3628e-01, -3.9026e-01, -4.4445e-01, -2.2545e-01,  1.9528e-01,\n",
       "           2.6898e-01, -1.7829e-01,  3.4329e-01,  6.0807e-02,  3.1945e-01,\n",
       "          -5.3761e-01, -3.9001e-01,  1.7498e-02,  2.9927e-01, -2.2351e-01,\n",
       "           6.0448e-01, -2.1332e-01, -2.0640e-01,  8.0174e-02,  1.4445e-01,\n",
       "           1.9218e-01,  7.7926e-01,  4.9605e-01, -5.9700e-01,  1.2727e-01,\n",
       "          -7.9618e-02, -2.9950e-01, -7.4013e-02, -3.8503e-01,  3.0926e-01,\n",
       "           1.6641e-01,  3.8763e-01,  6.3903e-01]], device='mps:0'),\n",
       " tensor([[-0.1555, -0.8327, -0.2983,  ...,  0.1656,  0.5149,  0.8016],\n",
       "         [-0.1689, -0.4450, -0.6101,  ...,  0.4169,  0.1256,  0.6819]],\n",
       "        device='mps:0'),\n",
       " tensor([[-1.1097e-02, -5.0567e-01, -4.6699e-01, -6.4640e-01, -3.9556e-01,\n",
       "          -2.5266e-01, -5.5990e-01,  3.0801e-01, -1.2536e-01,  1.4189e-01,\n",
       "           2.9133e-01,  5.3879e-01, -3.6706e-01, -2.8859e-01, -4.8290e-01,\n",
       "          -5.3699e-02, -6.1741e-01, -1.6918e-02,  3.4690e-01,  3.0491e-01,\n",
       "           5.3718e-01,  1.8073e-01, -4.1129e-01,  2.9149e-01, -3.7959e-01,\n",
       "           6.6561e-02, -3.4985e-01, -3.1546e-01,  2.6210e-01,  2.8012e-01,\n",
       "           5.3618e-02,  7.7692e-01, -2.4307e-02,  1.0962e+00, -1.8413e-01,\n",
       "          -2.1778e-01,  1.5359e-01,  2.3320e-01,  8.6439e-01, -6.2204e-01,\n",
       "           5.6395e-01,  3.6489e-01,  2.7586e-02, -4.0063e-02,  2.7845e-01,\n",
       "           3.5451e-01,  1.4567e-01, -5.7913e-01, -2.7161e-01, -3.4037e-01,\n",
       "          -1.2979e-01, -1.5164e-01,  5.0058e-01, -2.3292e-01, -7.9546e-01,\n",
       "          -1.9253e-01,  1.1288e-02,  2.6982e-01,  1.1117e-01,  2.3591e-01,\n",
       "           2.8900e-02,  7.9425e-01, -1.9339e-01, -3.2966e-03,  2.0559e-01,\n",
       "           2.5502e-01, -7.6243e-01, -1.2873e-01, -7.2822e-01,  1.1548e-01,\n",
       "           2.1413e-01,  3.5162e-01,  1.0500e-01, -2.6816e-01, -1.9644e-01,\n",
       "           6.0243e-02,  1.6956e-01,  1.8610e-01, -2.0035e-01,  3.5967e-02,\n",
       "           2.1140e-01,  4.4140e-01,  4.5980e-01,  5.9922e-02,  1.6117e-01,\n",
       "           5.6960e-02,  2.0080e-01, -3.9869e-02, -3.1738e-02,  1.7672e-01,\n",
       "           1.1984e-01, -9.9577e-02, -3.7483e-01, -4.3956e-01, -9.5205e-02,\n",
       "          -3.0408e-01,  3.2250e-02, -2.9469e-01, -2.3954e-01, -3.3261e-01,\n",
       "           2.5982e-01,  9.2762e-02,  3.2750e-01, -9.3082e-02, -6.6514e-01,\n",
       "          -3.5248e-01,  4.0832e-01,  3.4586e-03, -2.8129e-01, -3.8448e-01,\n",
       "          -6.6713e-02, -4.9346e-01,  1.2765e-01, -4.4650e-01,  1.9893e-01,\n",
       "           1.0835e-01,  1.8686e-01, -3.2670e-01,  6.5172e-02,  8.8101e-02,\n",
       "           6.1068e-02, -5.0938e-01,  2.7421e-01,  1.8326e-01,  9.3046e-02,\n",
       "           2.2489e-01,  6.8015e-01, -1.6210e-01,  4.3454e-01,  3.5722e-01,\n",
       "           3.1169e-01,  1.6221e-01, -4.1686e-01, -5.0259e-01, -7.5302e-02,\n",
       "           1.8488e-02, -5.9908e-01, -5.7442e-01,  6.1723e-02, -2.2868e-01,\n",
       "           1.6627e-01, -7.3083e-02, -5.1768e-01, -5.4307e-01, -3.1494e-01,\n",
       "          -3.0772e-01, -2.0517e-02, -3.2393e-01, -4.8159e-01, -2.5398e-01,\n",
       "          -6.1867e-01, -7.2891e-01,  1.9443e-01,  1.5164e-01,  2.7714e-01,\n",
       "          -4.3163e-01,  5.4580e-01, -3.0612e-01, -4.8554e-01,  3.7642e-01,\n",
       "           4.3432e-01, -5.0330e-01,  2.9526e-01, -6.3499e-01,  5.1171e-02,\n",
       "           2.0480e-01,  2.5291e-02, -3.4147e-02,  4.2807e-01,  1.1727e-01,\n",
       "           3.9831e-01,  2.7152e-02,  1.2476e-01, -2.9187e-01,  1.5015e-01,\n",
       "          -4.5427e-02, -3.2147e-01, -1.1176e-01,  1.7279e-02, -5.2967e-02,\n",
       "           5.4936e-01, -1.7306e+00, -1.7852e-01,  7.0060e-02, -2.7460e-01,\n",
       "          -1.3173e-01,  2.1473e-01,  5.0407e-03,  3.2283e-01, -3.5674e-01,\n",
       "           8.3698e-02,  8.7128e-02,  8.4654e-02, -1.0725e-01, -5.0983e-02,\n",
       "           2.9101e-01,  7.5929e-01,  4.0793e-01,  9.7463e-02, -2.6657e-01,\n",
       "          -6.0645e-01,  4.1554e-01,  1.9081e-01,  2.0730e-01,  3.9000e-01,\n",
       "           6.7931e-02, -6.9343e-01,  2.9674e-01,  1.9859e-01,  3.7002e-01,\n",
       "           2.5546e-02, -5.8496e-02, -4.8962e-01, -2.4682e-01, -4.3705e-01,\n",
       "           2.7759e-01, -1.3436e-01, -2.5922e-01,  6.3173e-01,  1.9273e-01,\n",
       "          -5.3531e-02,  3.9653e-01,  1.8733e-01, -4.1041e-01,  4.1633e-01,\n",
       "           5.2814e-01,  7.9114e-02,  1.4573e-01, -2.8055e-01, -1.6673e-01,\n",
       "          -3.1532e-01, -4.4420e-01,  4.4703e-01,  8.3941e-01, -7.2084e-03,\n",
       "           5.5343e-01, -3.5757e-01,  5.1174e-01, -3.6424e-01, -6.7779e-03,\n",
       "          -4.6486e-02,  6.8700e-02, -3.3988e-01,  1.6284e-01,  1.5551e-01,\n",
       "           1.0340e-01,  1.9709e-01,  8.5425e-01,  4.6494e-01,  1.4684e-01,\n",
       "           4.1031e-01,  2.5417e-02,  3.2080e-01,  7.8684e-01, -3.0254e-01,\n",
       "           3.4990e-01, -7.1578e-03, -4.1842e-01,  5.6461e-01, -3.5139e-01,\n",
       "           1.4248e-01, -2.3020e-01,  2.2987e-01,  1.7866e-01, -2.7854e-01,\n",
       "           2.1426e-01,  6.5273e-01,  9.5860e-02, -3.9283e-01,  7.3417e-01,\n",
       "          -4.0214e-01,  7.1262e-01, -3.0138e-02, -2.6992e-02, -1.0120e-01,\n",
       "          -7.4577e-01, -6.8569e-01,  1.4294e-01,  3.0808e-01, -7.6332e-01,\n",
       "          -3.4295e-01,  1.2380e-01,  1.3255e-01,  3.5068e-01,  2.4156e-01,\n",
       "           4.6144e-01, -1.7016e-01, -4.7185e-02, -4.4024e-01, -4.6015e-01,\n",
       "           1.1917e-01,  7.3311e-02,  9.5311e-02, -9.8199e-02,  9.2587e-02,\n",
       "           6.1067e-01,  2.4297e-01,  2.6799e-01,  1.4946e-01, -3.4610e-01,\n",
       "          -2.7723e-02,  1.7135e-01,  3.7874e-01, -6.1900e-01, -2.1807e-01,\n",
       "           5.3543e-02,  8.3164e-02,  6.6826e-01, -2.1832e-01, -1.1546e-01,\n",
       "          -1.2450e-01,  5.9739e-01, -2.2108e-01,  6.8002e-02,  1.0269e-01,\n",
       "          -6.8586e-01,  2.7619e-01, -5.9939e-02, -8.5297e-02, -5.2861e-01,\n",
       "           2.2997e-01, -5.7787e-02, -1.6068e-01, -7.3269e-03, -7.7601e-02,\n",
       "           2.2454e-01,  1.3406e-01,  2.8044e-01, -5.6270e-01, -4.2092e-01,\n",
       "           1.1781e-01, -3.2365e-01, -6.6016e-02, -4.5861e-01,  3.6766e-01,\n",
       "           1.3890e-01,  4.3829e-01, -1.0837e-01, -2.8576e-02, -1.3190e-01,\n",
       "           5.2866e-01, -6.2780e-01,  1.5573e-02,  4.8525e-01, -3.5595e-01,\n",
       "          -1.8139e-01,  4.7644e-01, -2.8011e-01,  4.9664e-02,  2.2982e-01,\n",
       "           3.7738e-01, -1.1530e-01, -4.7105e-01, -5.5247e-02,  9.3761e-01,\n",
       "           4.8980e-01,  2.0874e-02, -6.3820e-01,  1.2560e-01,  4.2066e-01,\n",
       "          -3.4219e-01,  1.0777e-01, -3.7280e-01, -2.3519e-01,  8.6456e-02,\n",
       "           3.3260e-02,  5.0344e-01, -5.8998e-02,  1.9741e-01,  4.6778e-01,\n",
       "           1.7011e-01, -6.9331e-01, -3.0557e-01,  6.3124e-01,  5.4670e-01,\n",
       "          -1.2247e-01, -5.4832e-01,  1.1480e-01, -6.4563e-01,  3.1198e-02,\n",
       "          -3.8623e-03, -3.5755e-01,  2.9653e-01, -1.4962e-01, -2.5683e-01,\n",
       "          -2.0003e-01, -5.3748e-01,  4.4973e-03,  1.0291e-01, -6.9595e-02,\n",
       "           4.1738e-01, -7.9610e-03, -3.1154e-01,  1.0058e+00, -2.7292e-01,\n",
       "          -3.1356e-02,  7.4139e-01, -5.4234e-01,  1.0457e-02,  2.4642e-01,\n",
       "          -1.3433e-01, -1.2416e-02,  2.4323e-02,  1.0660e+00,  1.1515e-05,\n",
       "           3.6402e-02, -2.0346e-01,  4.2203e-01, -4.7659e-01,  1.9394e-01,\n",
       "          -4.9958e-01,  4.8310e-01, -6.9399e-02,  2.4253e-01, -8.2329e-01,\n",
       "           4.3257e-01, -3.0604e-01, -3.2040e-01, -6.3922e-01,  5.0359e-02,\n",
       "          -5.0711e-02,  6.6524e-02, -5.8707e-02,  6.0787e-01, -1.0953e+01,\n",
       "           1.0946e-01,  8.4108e-01, -3.2618e-02,  1.3653e-01,  4.3336e-01,\n",
       "           1.2158e-01, -4.6684e-01,  7.2509e-01, -2.3242e-01, -6.6092e-03,\n",
       "          -3.5861e-01, -1.2444e-01, -1.3819e-01, -6.3519e-01,  4.0758e-01,\n",
       "          -1.3255e-01, -5.9114e-01,  2.8540e-01, -1.6945e-01,  2.4978e-01,\n",
       "          -2.3651e-01,  1.5671e-01,  5.9352e-01, -6.1010e-01, -3.3943e-01,\n",
       "          -1.1958e-01, -2.9991e-01,  5.1723e-01,  4.0217e-01, -2.3672e-01,\n",
       "           2.9096e-01,  3.3529e-02,  2.9455e-01, -5.5694e-01, -4.8133e-01,\n",
       "           4.3369e-01, -3.9950e-01,  2.1957e-01, -9.9943e-01, -3.3265e-01,\n",
       "          -1.8082e-01, -5.0951e-01, -2.7870e-02,  5.3812e-01,  7.5524e-01,\n",
       "           1.2821e+00,  1.1534e+00,  9.8198e-02,  4.6362e-01, -1.0523e-01,\n",
       "          -6.6935e-01,  5.1619e-02, -5.6525e-01, -4.1224e-01,  8.7196e-02,\n",
       "           2.2385e-01, -5.4352e-01,  4.7854e-01, -5.9897e-01,  1.8912e-01,\n",
       "           1.4329e-01,  7.9852e-02,  1.7368e-01, -5.1897e-01, -5.5039e-01,\n",
       "          -1.3282e-01,  8.1201e-02,  8.6324e-02, -6.7881e-01,  1.1881e-01,\n",
       "           1.7006e-01,  6.8197e-01, -2.5397e-01, -4.4268e-01,  4.1137e-01,\n",
       "          -2.2746e-03,  9.3809e-02,  1.3158e+00, -2.6036e-01, -1.7413e-01,\n",
       "          -4.8428e-02, -2.3194e-01, -3.3484e-01, -3.9656e-01, -4.5613e-01,\n",
       "          -2.5020e-02,  3.6573e-01,  5.1948e-01,  2.6062e-01, -2.3329e-02,\n",
       "          -3.0054e-01, -4.1445e-02,  5.1977e-02,  3.3862e-01,  9.6772e-01,\n",
       "          -6.6887e-01,  2.6280e-01, -1.9148e-01,  9.2178e-02,  2.2223e-01,\n",
       "          -5.7106e-01, -1.6871e-02,  2.9434e-03,  1.3975e-02,  5.7281e-01,\n",
       "           1.9705e-01,  2.6453e-01,  3.1190e-02,  2.2859e-01, -4.1682e-01,\n",
       "          -4.4044e-01,  2.9125e-04, -1.3190e-01, -5.8294e-01, -4.1034e-01,\n",
       "          -1.3514e-01,  1.3712e-01, -2.8585e-01, -5.3585e-01, -2.5012e-01,\n",
       "           1.4192e-01,  2.5900e-03,  4.9952e-02, -3.3857e-01,  4.7379e-01,\n",
       "          -7.6918e-01, -2.1429e-01, -2.7578e-01, -3.8495e-03,  1.4633e-01,\n",
       "           2.5508e-01, -5.1518e-01, -2.7838e-01, -1.0123e-01, -9.2314e-01,\n",
       "          -8.4489e-01,  2.2927e-02,  7.4620e-01, -1.1960e-01,  1.1861e-01,\n",
       "           8.5428e-02, -3.1671e-03, -3.7708e-01,  2.5194e-01, -2.4644e-01,\n",
       "           2.8149e-01,  4.3265e-01, -4.9583e-01,  2.9773e-02, -1.2792e-01,\n",
       "          -1.8992e-02, -1.0887e-01, -8.8647e-02, -5.4131e-01, -6.1371e-01,\n",
       "          -4.1197e-01,  1.3736e-01, -2.2747e-01,  4.1516e-01,  6.8072e-02,\n",
       "           1.2582e-01, -2.0826e-01,  9.8224e-02,  8.2540e-01,  6.0197e-01,\n",
       "           2.3014e-01, -3.2920e-01, -7.7666e-01,  3.4140e-02,  4.1642e-02,\n",
       "          -1.6450e-01,  2.1317e-01, -1.6384e-01, -3.0657e-01, -1.0112e-01,\n",
       "           1.2620e-02,  1.0002e+00, -2.0790e-01, -1.9155e-01,  1.0507e-01,\n",
       "           7.4626e-01, -6.0834e-01, -5.7221e-01,  3.7612e-01,  5.0418e-02,\n",
       "          -5.6831e-01, -1.9450e-01,  4.0315e-01, -1.7111e-01,  8.9921e-02,\n",
       "           3.7572e-01,  1.7821e-01, -1.8612e-01,  2.2711e-01, -6.7291e-02,\n",
       "          -1.7001e-01,  3.2731e-01, -1.6533e-01, -7.7254e-02,  4.0384e-01,\n",
       "           9.9517e-02,  1.5584e-01, -2.5262e-01, -4.8444e-01, -2.7025e-01,\n",
       "           1.5439e-01,  1.1110e+00, -1.7989e-01, -3.2527e-01,  6.6881e-01,\n",
       "          -3.0956e-01,  3.8928e-01,  4.8478e-01, -2.2611e-01, -5.0807e-02,\n",
       "           5.8049e-01,  3.0288e-01, -7.3236e-01,  2.2417e-01,  3.1670e-01,\n",
       "          -1.7335e-01, -5.3095e-01, -3.2838e-01, -2.1330e-01, -3.5189e-02,\n",
       "          -3.6356e-01,  5.2132e-01,  1.2255e-01,  3.9253e-01, -1.0843e-01,\n",
       "          -8.7207e-01,  4.7185e-01,  2.6128e-01,  8.0250e-01, -6.2568e-03,\n",
       "          -4.4880e-01, -3.5541e-01, -3.5790e-01, -6.6453e-02,  3.6543e-01,\n",
       "           2.0116e-01, -1.0333e+00,  4.6071e-02,  4.0507e-01, -3.9812e-02,\n",
       "           4.7083e-01, -1.4254e-01,  1.9634e-01,  4.7891e-01, -4.3918e-02,\n",
       "          -2.1856e-01,  1.3161e-01, -3.3027e-02, -2.7190e-01,  4.2861e-01,\n",
       "           4.6755e-01,  1.6062e-02,  2.5454e-02,  6.4881e-02, -4.2427e-01,\n",
       "          -3.4980e-01,  7.7064e-02, -1.7966e-01, -3.5675e-01, -1.4295e-01,\n",
       "          -4.0726e-01, -5.1772e-01, -6.6164e-02, -1.1357e-01,  1.3233e-01,\n",
       "           3.1032e-01,  1.3036e-01,  3.8431e-01, -2.3708e-01, -1.2541e-01,\n",
       "          -2.7977e-01,  4.5653e-01, -2.7595e-01,  8.0808e-01,  2.0566e-01,\n",
       "          -1.4423e-01,  2.0720e-01,  2.1514e-01, -2.9046e-01,  1.3028e-01,\n",
       "          -2.7828e-01,  2.8810e-01, -5.7798e-01, -3.4110e-01, -5.9710e-02,\n",
       "          -7.7948e-02, -1.5049e-01, -2.8505e-01, -1.0179e-01,  1.0608e-01,\n",
       "          -9.9043e-02,  9.7868e-01, -4.1759e-01,  6.3833e-01, -1.0342e-01,\n",
       "          -1.1350e-01, -7.5089e-01,  3.9397e-01,  3.6376e-01, -1.4798e-01,\n",
       "          -5.6434e-01, -2.1653e-01, -8.9158e-02, -4.2284e-01,  8.8621e-01,\n",
       "           5.8157e-01, -7.4673e-01, -3.7246e-01, -1.5919e-01, -2.4113e-01,\n",
       "          -4.7389e-02,  6.9820e-02, -3.2710e-01, -1.3295e-02, -1.6580e-01,\n",
       "          -9.8715e-01,  2.7641e-01, -3.6473e-02,  4.9205e-01, -2.5466e-01,\n",
       "           7.6979e-01, -8.6017e-02, -1.2749e-01,  1.1690e+00,  1.1169e-01,\n",
       "          -6.1844e-02,  5.9531e-01, -1.6837e-01, -6.2973e-01, -7.0295e-02,\n",
       "           4.4239e-02, -3.1098e-01,  8.7670e-02, -4.7880e-01,  2.9353e-01,\n",
       "           2.3648e-01, -7.1226e-02,  4.3877e-01]], device='mps:0'),\n",
       " tensor([[-0.1879, -0.7771, -0.6931,  ...,  0.0884,  0.0610,  0.3990],\n",
       "         [-0.1127, -0.5503, -0.8677,  ...,  0.1306,  0.4647,  0.4332]],\n",
       "        device='mps:0'),\n",
       " tensor([[-0.2455, -0.3062, -0.3376,  ...,  0.0991,  0.2277,  0.4125],\n",
       "         [-0.3292, -0.2590, -0.5016,  ..., -0.0148,  0.3509,  0.1607]],\n",
       "        device='mps:0'),\n",
       " tensor([[-9.2943e-02, -5.1867e-01, -4.2151e-01, -7.7446e-01, -1.9057e-01,\n",
       "          -2.2396e-01, -5.1457e-01,  4.4190e-01,  5.2480e-02,  4.2398e-01,\n",
       "           2.2176e-01,  6.2463e-01,  3.1867e-02, -7.8361e-02, -2.3694e-01,\n",
       "          -3.1725e-01, -1.3960e-02,  5.1445e-01,  3.6228e-01,  2.5588e-01,\n",
       "           2.9757e-01,  3.5936e-01,  1.2859e-01, -5.8801e-02, -5.1358e-01,\n",
       "           6.1143e-01,  7.7832e-02,  2.2829e-01,  1.5635e-01, -5.8690e-01,\n",
       "           3.6965e-02,  6.1030e-01, -4.3539e-02,  1.0644e+00, -2.6128e-01,\n",
       "          -4.6399e-02,  2.4385e-02, -4.0618e-01, -1.6344e-01, -2.7204e-01,\n",
       "           5.6419e-01, -3.2149e-01,  3.0268e-02,  9.8450e-02, -6.9517e-02,\n",
       "           1.3431e-01,  2.5402e-01, -5.0583e-01,  4.9717e-01, -5.9803e-01,\n",
       "          -1.2879e-01,  3.4321e-02,  4.2965e-01,  3.5747e-01, -5.7634e-01,\n",
       "          -2.6854e-02,  8.2437e-02,  8.5138e-02, -1.5777e-01,  2.3861e-01,\n",
       "          -2.2360e-01,  5.2223e-01, -2.3275e-01,  8.0734e-02,  1.0427e-01,\n",
       "           4.5304e-01, -1.1783e-01, -3.9161e-01, -2.7221e-01, -2.3335e-01,\n",
       "          -1.6212e-02,  1.2914e-01,  2.0951e-01, -1.7257e-01,  2.6381e-01,\n",
       "          -6.1187e-01,  2.5094e-02, -4.5973e-02, -2.3167e-01,  1.4397e-01,\n",
       "           8.8465e-02,  3.7858e-01,  3.6692e-01, -1.3157e-01,  2.3953e-02,\n",
       "          -8.9270e-02,  1.1213e-01,  9.1852e-02,  9.0633e-02,  2.1274e-01,\n",
       "           2.0553e-01, -2.7641e-03, -4.0554e-01, -2.2123e-01,  4.2108e-01,\n",
       "           1.2868e-01,  4.3819e-02, -6.5257e-02, -2.8937e-02, -1.7914e-01,\n",
       "           1.8685e-01,  2.5377e-01, -8.3141e-02,  7.4648e-02, -5.6890e-01,\n",
       "          -5.4631e-01,  1.4134e-01, -3.2404e-01, -5.0280e-01, -1.5450e-02,\n",
       "          -6.8557e-02, -2.5369e-01,  3.5821e-02, -5.4305e-01, -1.8615e-01,\n",
       "           6.8135e-02,  7.9087e-02, -1.3831e-01, -2.9615e-01,  2.7097e-01,\n",
       "           3.3578e-01, -1.8470e-01,  7.4140e-02,  8.3791e-02,  1.1347e-01,\n",
       "          -4.4156e-02,  5.6254e-01,  2.0241e-01,  2.9952e-01,  2.5343e-01,\n",
       "           1.3599e-01, -1.1554e-01, -2.6321e-01, -8.3334e-02, -1.3655e-01,\n",
       "           3.6232e-02,  1.8988e-01, -2.1220e-01,  6.5294e-02, -4.0189e-02,\n",
       "           1.3838e-01, -3.3112e-01, -2.6565e-01, -3.4677e-01, -8.2066e-02,\n",
       "          -3.3907e-01,  8.2016e-02, -4.2652e-01, -1.8696e-01, -2.2009e-01,\n",
       "          -7.3253e-01, -7.2751e-01,  2.3950e-01, -1.4049e-01, -4.1473e-02,\n",
       "          -1.9473e-01,  4.8899e-01,  1.9361e-01, -1.7118e-01,  3.1806e-01,\n",
       "           1.1162e+00, -6.9239e-02,  1.5861e-01, -4.6988e-01, -2.5737e-01,\n",
       "           8.2930e-02,  1.2362e-02, -2.7040e-04,  2.4316e-01,  9.3801e-02,\n",
       "           1.7260e-01, -1.1642e-01,  1.0619e-01, -2.5987e-01, -1.9190e-01,\n",
       "          -5.3665e-01,  9.6788e-02, -2.4487e-02, -2.0821e-01, -1.3820e-01,\n",
       "          -2.2937e-01, -9.5521e-01,  2.4208e-01, -1.6910e-01,  2.5545e-01,\n",
       "          -3.0291e-01,  3.4762e-01,  2.6882e-01,  1.5252e-01,  1.7013e-02,\n",
       "          -4.6511e-02,  1.6795e-01,  1.6740e-01,  1.7838e-01,  2.0641e-01,\n",
       "           4.0897e-01,  2.3726e-01, -1.6604e-01,  3.8324e-01,  8.0601e-02,\n",
       "          -5.1555e-01,  2.5941e-01,  1.6103e-01,  2.8787e-01,  2.2890e-01,\n",
       "          -1.4215e-01, -4.5573e-02,  1.7840e-01,  7.0792e-02, -8.6163e-02,\n",
       "          -1.9885e-01, -2.5697e-02, -3.4244e-01,  3.8163e-01, -2.1675e-01,\n",
       "           4.6934e-01, -4.4044e-01, -1.9166e-02,  4.4032e-01, -2.2395e-03,\n",
       "           9.2646e-03,  3.2197e-01,  2.4743e-01, -2.9510e-01,  1.0552e-01,\n",
       "           2.6434e-01, -2.2862e-01,  3.0595e-02, -3.6483e-01, -3.7937e-01,\n",
       "           1.6697e-02, -5.6619e-01,  4.2332e-01,  1.8158e-01, -2.0175e-01,\n",
       "          -1.4647e-01, -2.7312e-01, -3.8644e-03, -2.0085e-01,  2.0732e-01,\n",
       "          -4.5552e-01, -3.0136e-01, -4.3634e-01,  2.9926e-01,  3.9459e-01,\n",
       "           3.1536e-01, -5.3616e-02,  5.3628e-01,  3.4586e-01,  2.9709e-01,\n",
       "          -2.7457e-01, -2.4985e-01,  4.4867e-01, -3.4734e-02, -5.8293e-01,\n",
       "           2.6863e-01, -2.0079e-01, -7.5456e-01,  2.5072e-01, -2.0495e-01,\n",
       "          -3.0446e-03,  6.5649e-02,  1.5935e-01,  4.6208e-02, -6.4714e-02,\n",
       "          -2.5240e-01,  3.1395e-01, -6.2903e-02, -5.2867e-01,  4.9016e-01,\n",
       "          -1.3822e-01,  3.7566e-01, -1.0417e-02, -4.8046e-01, -5.6135e-01,\n",
       "          -2.6933e-01,  1.4230e-01,  1.9252e-01,  1.6272e-01, -7.9902e-01,\n",
       "          -1.5301e-01,  1.3952e-01,  5.9592e-02,  1.6891e-01,  3.0869e-01,\n",
       "           4.4769e-01, -1.1690e-01,  1.7469e-01,  1.1995e-01, -5.4637e-01,\n",
       "           1.2999e-02, -3.5063e-02, -1.6753e-01,  1.7567e-01,  3.7785e-01,\n",
       "           5.2270e-02,  2.1085e-01,  1.3569e-01, -2.7558e-01, -3.3805e-01,\n",
       "           1.0131e-01, -5.3771e-02, -1.9799e-01, -1.5888e-01,  2.4383e-01,\n",
       "           1.7817e-02,  5.3152e-01,  3.5332e-01, -3.7493e-01, -4.2040e-01,\n",
       "           8.2757e-02,  5.4965e-01, -1.9105e-01, -2.4015e-01, -1.1130e-01,\n",
       "          -6.2011e-01, -1.5233e-01, -2.0917e-01, -2.4935e-01, -5.8298e-01,\n",
       "          -2.0453e-02,  2.9904e-01, -1.6137e-01, -1.5872e-01, -5.7924e-01,\n",
       "           3.8645e-01,  2.0040e-02,  2.0055e-01,  2.0652e-02, -4.0932e-01,\n",
       "          -2.8034e-01, -4.4787e-01, -3.0653e-01, -2.5959e-01, -3.2899e-01,\n",
       "           1.2684e-01,  8.4073e-02,  3.9753e-01, -3.5173e-01, -3.4186e-01,\n",
       "           7.0060e-01, -4.7391e-01,  1.6551e-01,  9.1017e-02, -6.8031e-01,\n",
       "           2.8930e-02,  4.5017e-01, -2.0389e-02, -9.7838e-02,  1.0003e-02,\n",
       "           5.3608e-01, -7.0143e-02, -3.4341e-01, -1.2460e-01,  5.4390e-01,\n",
       "           3.4922e-01,  1.3967e-01, -3.7441e-01,  3.6886e-01,  3.8473e-01,\n",
       "          -5.2262e-01,  5.5366e-01, -4.2080e-02, -4.3795e-02, -3.3945e-01,\n",
       "           1.3391e-01,  3.3648e-01, -4.6580e-01,  9.1308e-02,  1.7037e-02,\n",
       "          -1.6696e-02, -4.1966e-02, -4.3099e-02,  1.8304e-01,  4.9913e-03,\n",
       "          -1.5079e-01, -3.0479e-01,  1.2694e-01, -9.3054e-02, -3.0030e-01,\n",
       "           4.1686e-01,  1.2879e-01, -3.0593e-02, -7.1777e-01, -2.6585e-03,\n",
       "          -1.0752e-01, -1.7732e-01,  6.9021e-01,  1.9072e-01, -4.6136e-01,\n",
       "           1.4423e-01, -2.8225e-01, -3.2058e-01,  7.5830e-01,  1.1928e-01,\n",
       "          -4.5461e-01,  4.3538e-01, -4.1035e-01, -6.3804e-02,  4.3978e-01,\n",
       "          -3.1080e-01, -1.0902e-02,  3.6776e-02,  2.6098e-01,  1.4709e-01,\n",
       "          -4.3816e-01,  2.2331e-02, -7.8046e-02, -1.5892e-01,  3.8092e-01,\n",
       "           2.3167e-01,  3.5339e-01,  4.8395e-02, -2.7662e-01, -1.6943e-01,\n",
       "           3.0428e-01, -1.1517e-02, -2.4713e-01, -5.0641e-01, -6.1377e-01,\n",
       "           6.4683e-01,  3.6649e-01,  6.1451e-01,  2.9460e-01, -1.1580e+01,\n",
       "           2.8545e-01,  8.6653e-01, -6.1033e-01, -2.7406e-02,  1.1163e-01,\n",
       "           8.7825e-02, -2.7959e-01,  3.8720e-01, -3.5337e-01, -1.1492e-01,\n",
       "          -4.0180e-01, -3.3200e-01,  2.2824e-01, -3.4710e-01, -2.9598e-01,\n",
       "           9.6841e-02, -7.8569e-01,  3.9733e-01, -4.3685e-01,  5.0306e-01,\n",
       "          -1.3499e-01, -2.3712e-02,  2.8183e-01, -3.6615e-01, -8.2184e-02,\n",
       "          -1.2977e-01,  3.0279e-02,  3.9260e-01,  4.3940e-01,  1.4226e-01,\n",
       "           5.7532e-02, -1.0734e-01,  1.8506e-01, -1.3101e-01, -4.3430e-01,\n",
       "          -1.6962e-01, -3.1628e-01,  1.5577e-01, -9.5731e-02, -5.0758e-01,\n",
       "          -1.8937e-01, -3.1567e-01,  2.5060e-01,  5.7899e-01,  7.1320e-01,\n",
       "           4.0370e-01,  8.9640e-01,  1.4961e-01, -3.3297e-01, -5.4384e-02,\n",
       "          -7.7437e-01,  2.3757e-02, -7.0105e-01,  1.8239e-02, -3.0725e-02,\n",
       "           5.7433e-01, -8.0394e-01,  4.0038e-01, -5.0386e-01,  1.1153e-02,\n",
       "           4.0719e-01,  1.7703e-01,  6.8571e-02, -4.1527e-01,  9.1731e-02,\n",
       "           4.6076e-02, -1.5773e-01, -6.5972e-02, -1.1630e-01, -1.6304e-01,\n",
       "          -2.5915e-01,  4.1510e-01,  3.1806e-01, -7.9638e-01,  1.9130e-01,\n",
       "          -2.0579e-01, -3.9832e-02,  1.6578e+00,  2.5034e-01,  1.8514e-01,\n",
       "           1.1363e-01,  6.0324e-01, -4.5429e-01, -5.8243e-01,  3.1820e-01,\n",
       "           6.2383e-03,  1.7879e-01,  3.2710e-01,  5.2999e-01, -1.6325e-01,\n",
       "          -8.2248e-01,  1.6216e-01,  2.3546e-01,  8.5903e-02,  7.6315e-01,\n",
       "          -8.3252e-01,  5.6413e-03,  1.7048e-01,  1.5428e-01, -5.4929e-01,\n",
       "          -8.7870e-02, -5.2210e-01, -3.2305e-01,  2.1525e-01, -3.1334e-01,\n",
       "           1.9125e-01, -8.6808e-02, -2.4996e-01,  5.5689e-01,  4.0336e-02,\n",
       "           4.4744e-02,  3.6161e-02,  5.9362e-02, -1.5473e-01,  1.5542e-01,\n",
       "           3.3492e-01,  4.1340e-01, -3.6598e-01, -5.8182e-01, -2.6772e-01,\n",
       "          -1.0444e-01,  4.9459e-01,  4.4946e-01, -3.2994e-01,  3.5291e-01,\n",
       "          -3.9041e-01,  1.7044e-02,  2.1843e-01, -1.9702e-01,  3.4356e-01,\n",
       "           5.1333e-02,  1.2143e-01,  1.4112e-01, -7.5418e-01, -6.3319e-01,\n",
       "          -4.0534e-01, -2.2166e-01,  4.3467e-01, -2.9345e-01, -1.2274e-01,\n",
       "          -1.4829e-01,  6.9441e-02, -1.5084e-01,  4.4336e-01,  6.0617e-02,\n",
       "           3.2013e-01,  1.9074e-01, -6.9949e-02, -4.0842e-02, -9.0872e-02,\n",
       "          -6.4207e-01,  1.2742e-01, -2.7253e-02, -1.8564e-01, -4.7011e-01,\n",
       "          -2.5898e-01, -1.0735e-01,  7.0683e-02, -1.4060e-01,  3.1043e-01,\n",
       "          -1.0760e-01,  1.2879e-01, -3.8891e-01,  6.4276e-01,  2.9661e-01,\n",
       "          -1.5508e-02, -2.8712e-01, -2.6636e-01,  1.7780e-01,  2.9855e-01,\n",
       "           2.1271e-01, -1.1391e-02, -3.2005e-01, -2.2492e-01, -3.3290e-01,\n",
       "           1.5671e-02,  9.6115e-01, -1.6654e-01, -3.2137e-01,  7.1564e-02,\n",
       "           2.5003e-01,  1.2502e-01, -2.8889e-01,  2.1762e-01,  3.0157e-02,\n",
       "          -4.1561e-01, -5.5127e-01,  4.3455e-01, -3.9854e-03, -4.7582e-02,\n",
       "           3.4120e-01, -1.6655e-01, -2.3784e-01,  6.6539e-01, -2.1076e-01,\n",
       "          -3.5708e-01, -1.1260e-01, -7.2213e-01,  2.9644e-01,  1.6338e-01,\n",
       "          -3.9167e-01,  6.2299e-01,  5.8569e-02, -5.0089e-02,  2.6817e-01,\n",
       "           2.8892e-01,  7.7713e-01, -5.0974e-01,  1.0283e-01,  6.2111e-01,\n",
       "          -1.5532e-01, -1.3598e-01,  2.6910e-01, -5.0818e-02,  1.2021e-01,\n",
       "           3.5380e-01, -4.7294e-01, -2.9973e-01, -9.8611e-02,  2.1723e-01,\n",
       "           4.0556e-01, -6.4296e-01, -5.7028e-01, -2.5036e-01,  8.2152e-01,\n",
       "           8.0753e-02,  4.8663e-01,  2.0072e-01,  3.9480e-01,  2.5922e-01,\n",
       "          -1.1007e+00, -1.4357e-01, -3.0050e-01,  1.7209e-01,  4.0352e-02,\n",
       "          -3.1519e-01,  5.9256e-02, -1.5222e-01,  5.8426e-02,  4.3758e-01,\n",
       "           1.9288e-01, -8.9514e-01,  1.6417e-01,  8.8059e-01, -9.5545e-01,\n",
       "           5.5152e-01, -2.1840e-01,  1.3746e-01,  1.8114e-01, -5.8500e-02,\n",
       "           1.7376e-01,  7.0432e-02, -3.8167e-01, -2.8007e-01,  1.6290e-01,\n",
       "           8.5254e-01, -6.2212e-02,  2.2114e-01, -5.0933e-02, -2.1437e-01,\n",
       "          -7.5335e-02, -1.6921e-01, -2.7157e-01,  4.8908e-02, -4.6553e-02,\n",
       "          -6.1227e-02, -2.6623e-01, -2.2663e-01,  3.8289e-01,  8.6154e-02,\n",
       "           2.3193e-01,  6.1992e-01,  3.4196e-01, -1.7068e-01, -3.3235e-01,\n",
       "           1.0794e-01,  8.4547e-02, -1.9243e-01,  3.7130e-01, -3.6933e-01,\n",
       "          -3.8892e-01, -8.6191e-02,  9.2702e-02, -7.3873e-01,  1.7486e-02,\n",
       "          -3.7905e-01,  5.1263e-01, -4.5067e-01, -8.9179e-02,  6.0367e-01,\n",
       "           1.0916e-01,  1.7747e-01, -4.4371e-01,  5.4294e-01,  4.5230e-01,\n",
       "           1.3459e-01,  5.3151e-01,  2.3746e-01,  6.8383e-05, -5.9834e-01,\n",
       "           1.9777e-01, -9.9329e-01, -4.3168e-01, -1.3043e-01, -2.5010e-01,\n",
       "          -2.6453e-01, -7.8286e-02, -4.8092e-02,  3.8613e-01,  5.2744e-01,\n",
       "           6.7262e-01, -4.1592e-01, -5.0653e-01, -2.4761e-01,  4.9086e-02,\n",
       "           2.2783e-02, -1.2054e-01,  4.0841e-01,  4.4693e-02,  1.3992e-01,\n",
       "          -4.5281e-01, -4.3623e-01, -1.3375e-01,  3.9473e-01, -2.4382e-01,\n",
       "           7.5593e-01, -1.8028e-01, -1.6491e-01,  2.1178e-01,  7.4194e-02,\n",
       "           3.2002e-02,  7.1648e-01, -4.7916e-02, -6.7933e-01,  1.2514e-01,\n",
       "           5.6558e-02, -2.8674e-02,  2.6129e-01, -4.0813e-01,  1.0849e-01,\n",
       "          -8.1714e-02,  4.3304e-01,  3.9476e-01]], device='mps:0'),\n",
       " tensor([[-1.7614e-01, -5.8749e-01, -3.5802e-01, -1.0230e+00, -2.7551e-01,\n",
       "          -4.0262e-02, -8.5206e-01,  2.0346e-01,  3.3503e-01,  3.4728e-01,\n",
       "           2.1668e-01,  4.0528e-01, -6.9469e-02,  1.4801e-02, -5.1288e-01,\n",
       "          -2.2614e-01,  1.6378e-01,  2.5336e-01,  6.8510e-01,  1.1683e-01,\n",
       "           1.8234e-01,  4.4697e-01, -1.5832e-01, -4.9634e-02, -3.6537e-01,\n",
       "           1.9070e-01, -1.9841e-01,  2.3795e-01, -7.8445e-03, -4.6238e-01,\n",
       "           2.5665e-01,  3.4648e-01, -4.2807e-01,  1.4162e+00, -2.9775e-01,\n",
       "          -2.1308e-01,  8.2004e-02, -2.9639e-01, -3.0269e-01,  1.1109e-01,\n",
       "           5.9814e-01, -1.9909e-01, -2.2982e-01,  3.1273e-01, -3.4196e-02,\n",
       "           1.8301e-01,  1.0062e-01, -2.9022e-01,  5.0568e-01, -5.9002e-01,\n",
       "          -1.3629e-01, -3.7746e-01,  2.1880e-01,  3.8247e-01, -4.6266e-01,\n",
       "          -2.9720e-01,  2.1812e-01,  1.4988e-01, -2.1488e-01,  1.1886e-01,\n",
       "          -7.9123e-02,  3.9341e-01, -1.2118e-01,  1.8456e-01,  1.6926e-01,\n",
       "           3.4797e-01, -5.0807e-01, -6.5750e-01, -6.4595e-01, -1.6275e-02,\n",
       "           1.5502e-01,  3.0507e-01,  9.2226e-02, -2.7995e-02,  2.9219e-01,\n",
       "          -4.9067e-01,  2.0200e-01, -1.4132e-02, -3.6383e-01,  2.4186e-01,\n",
       "          -2.2957e-01,  4.2885e-01,  3.1956e-01,  2.0169e-01,  2.0612e-01,\n",
       "          -5.6283e-02,  2.9078e-01, -3.0172e-02,  1.8520e-02,  3.8230e-01,\n",
       "          -1.1129e-01,  8.7467e-02, -8.0353e-02, -1.4495e-01,  1.0699e-01,\n",
       "           1.8117e-01,  2.1215e-01, -2.6830e-01, -9.8620e-02, -2.2677e-01,\n",
       "           1.4849e-01, -3.3044e-02, -8.9273e-02,  2.7185e-01, -4.5808e-01,\n",
       "          -5.8438e-01,  9.1718e-02, -3.0031e-01, -4.5676e-01,  3.0003e-02,\n",
       "          -3.0395e-01, -1.9434e-01,  1.5160e-01, -2.2299e-01,  2.1092e-01,\n",
       "          -1.1786e-01, -5.5463e-02,  5.4748e-02, -1.5349e-01,  1.4516e-02,\n",
       "          -1.0023e-01, -2.7571e-01,  1.9385e-01,  4.4648e-02,  1.2932e-01,\n",
       "          -4.6716e-02,  6.2585e-01,  4.2309e-02,  1.2413e-01,  4.9745e-01,\n",
       "           1.4654e-01, -8.2920e-02, -2.6324e-01, -2.1829e-01, -4.2277e-02,\n",
       "           2.5880e-01,  7.7860e-02, -1.9708e-02, -1.8201e-01, -2.1863e-01,\n",
       "           2.8411e-01,  3.9164e-02, -3.1539e-01, -2.0375e-01, -2.3885e-01,\n",
       "          -3.8593e-01, -2.5494e-01, -2.0587e-01, -5.0981e-01, -4.4611e-01,\n",
       "          -8.2060e-01, -5.5616e-01,  6.4001e-01, -1.9643e-01, -2.3516e-01,\n",
       "          -1.3706e-01,  2.5442e-01,  1.4085e-02,  4.1483e-02,  3.3790e-01,\n",
       "           6.3541e-01, -2.7567e-01, -4.8947e-02, -3.7977e-01, -7.3702e-02,\n",
       "           6.9975e-02,  1.8970e-01, -7.8576e-02,  1.3551e-01,  5.4505e-02,\n",
       "           1.8973e-01, -1.9960e-01,  5.8888e-01, -1.7729e-01,  1.2618e-02,\n",
       "          -8.4935e-01,  4.9333e-01,  1.9954e-01, -1.6157e-01,  1.1710e-01,\n",
       "           3.3931e-01, -1.1931e+00,  1.2620e-01, -1.0227e+00,  3.2824e-01,\n",
       "          -1.7098e-01,  1.8095e-01,  3.2597e-01,  2.7577e-01,  5.5632e-02,\n",
       "           2.1186e-01, -9.9121e-03,  3.2520e-01,  3.9732e-01,  2.0288e-02,\n",
       "           2.5561e-01,  3.7830e-01,  3.9603e-01,  2.3333e-01,  8.7525e-02,\n",
       "          -4.6283e-01,  3.5844e-01,  5.1188e-01,  9.6246e-02,  1.9932e-01,\n",
       "           1.3285e-01, -1.7003e-01,  1.8194e-01, -4.7599e-03, -2.2653e-02,\n",
       "          -2.4232e-01, -2.7171e-01, -5.3104e-01, -5.0171e-02, -4.5602e-01,\n",
       "           1.8974e-01, -4.4759e-01, -2.3165e-01,  4.9004e-01,  2.0560e-01,\n",
       "          -5.8674e-03,  2.9285e-01,  1.7700e-01, -1.9742e-01,  1.7903e-01,\n",
       "           3.6679e-01, -3.7862e-01,  2.1561e-03, -1.9138e-01, -2.0779e-01,\n",
       "           4.1036e-01, -8.3581e-01,  5.2290e-01,  5.0503e-01,  1.8949e-01,\n",
       "           1.5468e-01, -1.6453e-01,  1.0726e-01, -6.0864e-01,  3.0500e-01,\n",
       "          -3.3508e-01, -2.8021e-01, -6.3127e-01, -3.2163e-02,  2.6321e-01,\n",
       "           2.0939e-01,  3.2734e-03,  7.8382e-01,  5.2561e-01,  4.9839e-01,\n",
       "          -2.4440e-01, -7.1029e-02,  3.1526e-01,  2.4145e-01, -5.7815e-01,\n",
       "           3.1173e-01, -1.9860e-01, -4.2880e-01,  2.8787e-01, -2.2835e-01,\n",
       "          -5.3193e-02, -8.2633e-02,  2.8505e-01,  3.9954e-01, -1.8936e-01,\n",
       "          -2.1729e-01,  6.3627e-02,  6.5801e-02, -3.3103e-01,  5.3294e-01,\n",
       "          -3.3218e-01,  3.3568e-01, -1.2785e-01, -4.5420e-01, -6.0830e-01,\n",
       "          -3.2012e-01,  7.5990e-03, -1.8086e-01,  1.3518e-01, -6.0369e-01,\n",
       "          -5.3184e-01,  2.9145e-01,  7.0159e-02,  4.4066e-02,  9.3039e-02,\n",
       "           2.9553e-01, -3.2811e-01, -1.8485e-01, -2.0442e-01, -4.3013e-01,\n",
       "          -1.0626e-01,  5.5425e-02, -6.8973e-01,  2.1155e-01,  3.6262e-01,\n",
       "           2.3407e-02,  4.3709e-01,  4.8475e-01, -1.9884e-01, -3.9754e-01,\n",
       "          -1.5716e-01,  2.4385e-01, -2.5395e-01, -1.9611e-01,  2.6151e-01,\n",
       "           3.6015e-02,  4.6225e-01,  5.5650e-01, -3.5037e-02, -4.3253e-01,\n",
       "           1.8470e-01,  7.1793e-01, -1.9044e-01, -2.0873e-01,  6.0083e-03,\n",
       "          -9.7548e-01, -7.8421e-02, -9.7703e-02, -4.0406e-01, -6.5467e-01,\n",
       "          -1.0242e-01,  1.6247e-01,  4.1146e-03, -8.4313e-02, -3.5131e-01,\n",
       "           1.1079e-01,  4.8121e-01,  1.0656e-01,  3.6841e-01, -1.3932e-01,\n",
       "          -4.4117e-01, -1.3803e-01, -4.1224e-01, -5.8410e-01, -2.4412e-01,\n",
       "           1.9565e-01,  4.2279e-01,  2.8224e-02, -1.8967e-01, -2.9446e-01,\n",
       "           3.8764e-01, -3.5967e-01, -1.8551e-01,  2.1402e-01, -3.6530e-01,\n",
       "          -3.4331e-02,  3.9263e-01,  4.4703e-01,  9.3187e-02,  1.3090e-02,\n",
       "          -2.0790e-02, -2.7593e-01, -4.0255e-01, -2.9048e-01,  6.3862e-01,\n",
       "           5.5537e-01,  2.4306e-01, -4.3878e-01,  3.5191e-01,  4.3891e-01,\n",
       "          -3.2233e-01,  3.5813e-01, -1.9689e-01, -6.6888e-02, -1.4280e-01,\n",
       "           2.6177e-01,  5.1533e-01, -1.7291e-01,  2.9446e-01,  2.7587e-01,\n",
       "          -2.5831e-01, -1.6174e-01, -1.3777e-01,  3.2027e-01, -2.0562e-02,\n",
       "          -1.6339e-01, -2.1425e-01,  1.6837e-01, -1.3232e-03, -3.5641e-01,\n",
       "           2.8122e-01,  8.1080e-02, -1.5214e-01, -2.0127e-01, -6.3402e-01,\n",
       "          -1.0510e-03, -4.2574e-01,  5.4687e-01,  8.5314e-02, -3.6663e-01,\n",
       "           3.7945e-01, -1.1153e-01, -4.6167e-01,  6.0215e-01,  1.0126e-02,\n",
       "          -4.6889e-01,  5.1139e-01, -1.7366e-01,  1.7471e-01,  2.6800e-01,\n",
       "          -9.6391e-02,  1.4733e-01,  3.4559e-02,  6.0019e-01,  4.3388e-01,\n",
       "          -1.5452e-01, -4.1834e-02,  2.4696e-01, -4.0157e-01,  4.8827e-01,\n",
       "          -1.4159e-01,  4.4179e-01,  3.5865e-02,  5.5619e-02, -1.3735e-01,\n",
       "           2.4115e-01, -4.1934e-02, -3.8763e-01, -4.3656e-01, -3.2445e-01,\n",
       "           4.1797e-01,  1.4815e-01,  3.0259e-01,  2.0858e-01, -1.1440e+01,\n",
       "           2.8408e-01,  8.6672e-01, -4.1510e-01, -1.0259e-01, -1.7257e-01,\n",
       "           7.8554e-03, -3.5723e-01,  5.1084e-01, -5.2429e-02, -4.5512e-01,\n",
       "          -4.7200e-01, -2.1718e-01,  2.3182e-01, -5.0717e-01, -1.0071e-01,\n",
       "           1.5686e-01, -6.4411e-01,  2.6426e-01, -8.6464e-02,  7.7427e-01,\n",
       "          -4.3489e-01,  6.6019e-02,  2.2426e-01, -2.9222e-01,  3.6921e-02,\n",
       "           1.8945e-01,  5.6739e-02,  3.2735e-01,  4.3975e-01, -5.4637e-02,\n",
       "           1.0907e-01, -1.2817e-01, -7.2709e-02, -4.7155e-01, -4.5018e-01,\n",
       "           1.0718e-01, -4.5356e-01,  6.1574e-01, -1.7738e-01, -6.1719e-01,\n",
       "          -7.3560e-02, -3.4541e-01,  2.8051e-01,  2.9518e-01,  6.7493e-01,\n",
       "           9.7734e-01,  9.3128e-01,  2.2472e-01,  3.6179e-01,  3.3649e-02,\n",
       "          -7.5636e-01,  1.8178e-01, -2.7020e-01, -1.2551e-01,  1.2280e-01,\n",
       "           3.8567e-01, -8.0513e-01,  1.6771e-01, -4.3596e-01,  3.7745e-02,\n",
       "           7.0978e-01,  3.5794e-02,  1.4096e-01, -3.6255e-01, -1.7046e-01,\n",
       "           7.9639e-02, -1.9999e-01,  2.1605e-01, -4.5426e-01, -3.5187e-01,\n",
       "          -4.5643e-02,  2.7080e-01,  2.5588e-01, -9.5273e-02,  4.3303e-01,\n",
       "           3.6737e-01, -1.4664e-01,  1.2865e+00,  2.8703e-02, -7.5999e-02,\n",
       "          -2.2399e-02,  1.3058e-01, -4.7545e-01, -5.8969e-01, -1.6053e-01,\n",
       "          -6.7661e-02, -5.0352e-02,  5.8265e-02,  3.7852e-01, -5.6180e-02,\n",
       "          -5.5075e-01, -3.6072e-02,  1.1656e-01,  1.0534e-01,  7.4453e-01,\n",
       "          -8.2455e-01, -2.5613e-01, -1.8590e-01,  7.3763e-02, -8.7501e-01,\n",
       "          -4.2014e-02, -5.1090e-01, -7.6148e-01,  4.9889e-01, -1.2170e-01,\n",
       "           2.5119e-01, -2.0634e-01,  3.4217e-02,  5.0820e-01, -1.2499e-01,\n",
       "           7.4107e-02,  6.7300e-02,  2.7937e-01, -2.7528e-01, -3.6588e-01,\n",
       "           3.2383e-01,  4.1790e-01, -2.1081e-01, -6.4888e-01, -1.1401e-01,\n",
       "           3.2435e-01, -1.3542e-01,  4.9550e-01, -4.6173e-01,  4.4143e-02,\n",
       "          -9.5193e-01, -4.3019e-03, -3.9190e-03, -8.4407e-02,  4.1453e-01,\n",
       "           9.1924e-02, -2.7209e-01,  1.5418e-01, -6.9187e-01, -4.1589e-01,\n",
       "          -8.1779e-01, -2.4179e-01,  1.4407e-01, -5.2683e-01, -1.7039e-01,\n",
       "          -1.8392e-01, -1.1571e-01, -2.0320e-02,  6.5289e-02, -3.6723e-02,\n",
       "           3.3165e-01,  9.3338e-02, -2.7144e-01, -1.6026e-01,  1.8651e-02,\n",
       "          -4.1166e-01, -1.4496e-01, -1.9238e-01,  5.4924e-03, -7.0790e-01,\n",
       "          -2.3814e-01,  1.1461e-01,  6.8359e-02, -2.1760e-01, -2.8679e-02,\n",
       "           1.7137e-01,  1.2980e-01, -3.9035e-01,  9.6516e-01,  4.2760e-01,\n",
       "           1.0279e-01, -3.7809e-01, -3.2426e-01, -1.2709e-01,  2.3585e-01,\n",
       "           1.6038e-01,  6.5167e-02,  8.4681e-02, -5.6620e-01, -6.4375e-01,\n",
       "           1.9720e-01,  7.3751e-01, -1.9981e-01, -5.0881e-01, -1.4986e-02,\n",
       "           4.2746e-01, -1.5042e-02, -6.3733e-02,  2.1279e-01,  1.4412e-01,\n",
       "          -1.8094e-01, -4.2080e-01,  4.8719e-01, -2.7092e-01,  5.3509e-02,\n",
       "           4.3953e-01, -5.3768e-02, -1.6833e-02,  2.5855e-01,  3.1938e-01,\n",
       "          -3.1925e-01, -7.1982e-02, -4.9652e-01,  2.2127e-01,  4.7249e-02,\n",
       "          -5.1562e-01,  8.0227e-01,  7.0630e-02, -1.5259e-01, -3.8539e-02,\n",
       "           3.4457e-01,  9.7020e-01, -1.3363e-03,  2.0419e-01,  8.2567e-01,\n",
       "           4.2065e-02,  3.4566e-01,  3.6262e-01,  3.6471e-01,  2.9030e-01,\n",
       "           3.2097e-01, -1.0485e-01, -5.8949e-01,  4.7583e-03,  3.5014e-01,\n",
       "           1.7573e-01, -5.5299e-01, -6.0410e-01, -1.6108e-01,  7.3744e-01,\n",
       "           1.0074e-01,  4.1783e-01, -3.7119e-01,  5.1360e-01,  4.9028e-01,\n",
       "          -8.1544e-01, -1.1107e-01, -3.6174e-01,  8.6522e-02,  9.0050e-02,\n",
       "          -4.1330e-01,  2.5312e-02,  9.7974e-03, -8.6930e-02,  7.2329e-01,\n",
       "           4.8497e-01, -1.0391e+00, -1.3207e-01,  6.9199e-01, -5.0798e-01,\n",
       "           2.5290e-01, -4.6972e-01, -4.7001e-02,  3.6721e-01,  2.4417e-01,\n",
       "          -2.5694e-01,  2.0378e-03, -3.0780e-01, -3.6706e-01,  1.8169e-01,\n",
       "           7.3122e-01, -2.1473e-01,  9.9233e-02, -2.9653e-01, -3.5736e-01,\n",
       "          -3.1443e-01, -3.2372e-01, -2.1382e-01, -4.8743e-02,  1.6010e-01,\n",
       "          -1.1684e-01,  7.3328e-03, -2.2013e-01,  3.4689e-01, -1.1010e-01,\n",
       "           3.2356e-01,  4.2742e-01,  4.9158e-01, -1.5864e-02, -2.9975e-01,\n",
       "          -3.3077e-01,  2.5213e-01, -1.0164e-01,  3.3758e-01, -2.6408e-01,\n",
       "          -3.6077e-01, -2.7779e-02,  1.2287e-01, -4.7394e-01,  2.2007e-01,\n",
       "          -6.1151e-01,  2.7277e-01, -1.7342e-01, -2.0179e-02,  2.4015e-01,\n",
       "           2.3243e-01,  2.1663e-01, -4.6588e-01,  4.6449e-01,  4.1972e-01,\n",
       "           1.2592e-01,  2.8265e-01,  3.0306e-01,  6.0499e-02, -2.3870e-01,\n",
       "           8.3599e-02, -9.6397e-01, -4.1337e-02, -1.3807e-01, -1.5032e-01,\n",
       "          -2.9964e-01, -2.6773e-01,  5.6849e-01,  2.3993e-01,  6.9110e-01,\n",
       "           5.0521e-01, -1.7087e-01, -5.7037e-01, -2.9213e-01,  1.7344e-01,\n",
       "           2.1405e-01, -1.4886e-01,  3.0544e-01,  1.1362e-01,  2.1320e-01,\n",
       "          -6.0712e-01, -3.0321e-01, -3.7033e-02,  2.1753e-01,  6.9741e-02,\n",
       "           7.3239e-01, -3.9063e-01, -2.9549e-01,  6.8877e-02,  2.3525e-01,\n",
       "           1.6941e-01,  7.3786e-01,  1.4470e-01, -6.8182e-01, -9.3803e-02,\n",
       "          -4.7710e-03, -1.2981e-01,  3.2907e-01, -5.9411e-01,  1.8814e-01,\n",
       "           3.7874e-02,  5.2011e-01,  7.7260e-01]], device='mps:0'),\n",
       " tensor([[-0.0111, -0.5057, -0.4670,  ...,  0.2365, -0.0712,  0.4388],\n",
       "         [-0.5452, -0.4140, -0.4096,  ...,  0.6097,  0.5846,  0.2192],\n",
       "         [-0.3002, -0.5320, -0.5890,  ...,  0.1243,  0.5166, -0.0538],\n",
       "         [-0.4742, -0.7223, -0.5108,  ...,  0.1736,  0.4514,  0.2745]],\n",
       "        device='mps:0'),\n",
       " tensor([[-8.1238e-02, -6.8254e-01, -7.8284e-01, -8.5821e-01, -1.4702e-01,\n",
       "          -1.1712e-02, -9.5023e-01,  5.0302e-01,  6.8472e-02,  3.9371e-01,\n",
       "           3.6339e-01,  4.0499e-01,  2.5400e-01, -4.5633e-01, -6.2405e-01,\n",
       "          -4.1382e-01, -2.0438e-01,  3.8264e-01,  2.5253e-01,  2.7438e-01,\n",
       "           3.6820e-01,  2.8209e-01,  4.8321e-03,  2.4986e-01, -8.4405e-01,\n",
       "           2.6384e-01, -8.6443e-04, -4.2804e-01,  4.5918e-02, -5.9396e-03,\n",
       "          -1.4156e-01,  3.7263e-01, -5.1785e-02,  1.2053e+00, -1.4171e-01,\n",
       "          -4.9855e-01,  5.4689e-01, -1.6273e-01,  5.3073e-01, -2.4957e-01,\n",
       "           2.3933e-01, -5.4422e-02,  3.2922e-01,  3.2067e-01,  2.7292e-01,\n",
       "          -5.4024e-03,  1.5518e-01, -9.0450e-01,  2.5999e-01, -3.4979e-01,\n",
       "          -1.2642e-01, -2.2277e-01,  3.0166e-01,  3.0843e-01, -7.1406e-02,\n",
       "          -3.5654e-01,  5.5184e-01,  1.3448e-01,  3.2684e-01,  4.5051e-01,\n",
       "          -2.9193e-01,  9.8112e-01, -1.6529e-02, -2.1960e-01,  2.2380e-01,\n",
       "           9.2382e-02, -5.8582e-01, -6.3849e-01, -6.7319e-01,  6.1551e-02,\n",
       "          -2.1092e-02,  5.2076e-01, -4.0816e-02,  5.3454e-02,  2.3750e-01,\n",
       "          -3.2271e-02,  2.5882e-01, -1.0931e-01, -4.3471e-01,  3.6278e-01,\n",
       "           2.1838e-01,  5.9192e-02,  8.5644e-02,  2.2519e-01, -2.2728e-01,\n",
       "          -5.9513e-02,  7.7360e-01,  1.2858e-01,  3.3878e-01,  1.2645e-01,\n",
       "           2.8060e-01, -2.5596e-01, -1.7729e-01, -2.5357e-01,  2.1868e-01,\n",
       "           3.3957e-01,  6.6363e-01, -6.2779e-01,  2.1053e-01, -2.1258e-01,\n",
       "           1.3515e-01,  3.8618e-02,  5.7150e-02, -6.7864e-02, -3.5492e-01,\n",
       "          -4.8870e-01,  2.7722e-02, -1.0980e-01, -7.5234e-02, -1.8211e-01,\n",
       "          -5.3608e-01,  9.3207e-02,  2.0463e-01, -4.4529e-01, -3.1925e-01,\n",
       "           3.8218e-01, -1.9719e-03, -2.1398e-01,  1.7146e-03,  4.9962e-01,\n",
       "           1.9837e-02, -6.7673e-01,  3.8057e-01,  1.2427e-01,  1.0347e-01,\n",
       "          -1.4441e-01,  2.8478e-01, -1.7931e-01,  3.7924e-01, -5.8045e-02,\n",
       "           1.9189e-01,  1.2775e-01, -2.2156e-01, -4.3640e-02,  9.7017e-02,\n",
       "          -7.9093e-02,  4.0645e-02, -2.0214e-01,  2.9767e-01, -8.9884e-02,\n",
       "           5.0828e-01,  1.3818e-01, -3.0394e-01, -3.3318e-01, -3.7055e-01,\n",
       "           4.3000e-02, -1.3240e-01,  1.8436e-01, -5.0589e-01, -3.4849e-01,\n",
       "          -7.7698e-01, -6.0197e-01,  3.1898e-01, -1.1916e-01, -4.0232e-02,\n",
       "           1.4413e-01,  2.6597e-01, -1.5506e-01, -1.0968e-01,  4.3875e-01,\n",
       "           6.2543e-01, -2.7412e-01, -2.7428e-02, -5.3568e-01, -7.6230e-02,\n",
       "           1.1683e-01,  3.5015e-01, -8.0128e-02, -3.2778e-01,  6.4389e-02,\n",
       "           4.4149e-01, -2.5909e-02,  6.4342e-01, -2.9061e-01,  6.6526e-02,\n",
       "          -2.1203e-01, -2.9732e-01,  1.5562e-01, -2.3922e-01,  3.0975e-01,\n",
       "           1.7143e-01, -1.3686e+00,  1.5296e-01, -2.2669e-02, -7.4285e-01,\n",
       "          -1.7495e-01,  8.4183e-02,  1.9034e-01,  3.8054e-01, -3.8956e-02,\n",
       "          -2.9945e-01,  5.5781e-01,  2.5557e-01,  2.0563e-01,  1.2950e-01,\n",
       "           5.5655e-02,  8.1536e-01, -1.0029e-01, -3.7536e-01,  1.1529e-01,\n",
       "          -8.1276e-01,  1.7802e-01,  2.4672e-01, -7.8931e-02,  1.0486e-01,\n",
       "           3.0885e-01, -4.0654e-01,  3.1431e-01,  2.9576e-01, -6.1424e-02,\n",
       "          -1.3911e-01, -3.2807e-01, -6.0996e-01, -1.9123e-01, -2.2203e-01,\n",
       "           8.8550e-01, -2.5676e-01,  2.9969e-01,  5.3695e-01,  1.0887e-01,\n",
       "          -1.3726e-01,  1.2411e-01,  1.9546e-01,  3.4862e-02, -2.8987e-01,\n",
       "           5.8259e-01, -3.4953e-01, -2.6934e-01, -1.3982e-01, -4.2211e-01,\n",
       "          -2.1544e-02, -4.6506e-01,  1.1890e+00,  6.5792e-01, -2.3418e-01,\n",
       "           6.5243e-02, -4.1404e-01,  1.2846e-01, -6.0039e-01,  3.3491e-01,\n",
       "          -4.9458e-01, -1.9310e-01, -7.7586e-01,  2.5857e-02, -2.3181e-02,\n",
       "           3.2761e-01,  7.9570e-02,  9.1122e-01,  3.5086e-01,  1.5360e-01,\n",
       "           1.0998e-01, -3.0849e-02,  3.6776e-01, -7.9283e-02, -2.2413e-01,\n",
       "           1.6608e-01,  2.1193e-01, -8.7314e-01,  8.7803e-01, -2.1380e-01,\n",
       "           1.5303e-01, -3.1496e-01, -6.7507e-02,  2.8876e-01, -2.0396e-01,\n",
       "           1.5430e-01,  6.6233e-01, -2.7582e-01, -4.5209e-01,  7.0182e-01,\n",
       "          -5.4589e-01,  8.9963e-02,  3.6847e-02, -2.5926e-01, -3.7217e-01,\n",
       "          -2.9975e-01,  1.6823e-01, -2.4892e-02, -6.7023e-02, -6.7591e-01,\n",
       "          -3.1791e-01,  1.3189e-01,  1.6776e-01,  5.4133e-01,  1.0252e-01,\n",
       "           3.0641e-01,  6.3392e-02,  3.8239e-02, -1.8789e-01, -6.4458e-01,\n",
       "           1.1235e-01,  1.3303e-01, -2.0690e-01,  1.6862e-01,  2.7924e-02,\n",
       "          -1.5227e-01,  7.1696e-02,  1.1401e-01, -1.1322e-01, -4.6170e-01,\n",
       "          -2.5786e-02, -2.5574e-01,  4.0922e-01, -4.5430e-01, -3.3688e-01,\n",
       "           1.1340e-02,  1.7706e-01,  6.5277e-01,  1.4288e-01, -5.1425e-01,\n",
       "          -2.1825e-01,  4.3709e-01, -1.4949e-01, -5.7625e-01, -7.6874e-02,\n",
       "          -1.2380e+00,  2.0505e-01, -1.5143e-01,  2.3743e-02, -8.0043e-01,\n",
       "           1.9442e-01,  3.7399e-01, -3.6557e-01,  3.1547e-01, -7.9599e-02,\n",
       "           2.1996e-01,  1.4795e-01, -4.5422e-02, -2.0548e-01, -2.1570e-01,\n",
       "           3.9903e-01, -4.6550e-01, -3.4716e-01, -2.7658e-01,  3.2347e-01,\n",
       "          -1.9840e-01,  1.9939e-01, -2.0414e-01,  1.6384e-02, -2.2867e-01,\n",
       "           6.7056e-01, -5.8238e-01,  2.3700e-01,  3.4312e-01, -2.9597e-01,\n",
       "           5.3873e-02,  1.4202e-01, -4.3869e-01, -1.9601e-01,  5.3683e-01,\n",
       "           3.9776e-01, -5.0874e-01, -9.3945e-02,  5.1853e-03,  7.1439e-01,\n",
       "           1.0863e-01,  1.2122e-01, -5.2585e-01,  7.8086e-02,  2.3400e-01,\n",
       "          -3.7308e-01, -2.9492e-02, -1.8813e-01, -3.4618e-01,  9.1004e-02,\n",
       "          -3.5262e-02,  3.6941e-01, -3.1596e-01, -1.4007e-01, -2.2512e-03,\n",
       "          -1.7488e-01, -9.1045e-01, -1.7435e-01,  6.3888e-02,  6.1108e-01,\n",
       "          -4.4530e-02, -4.8100e-01,  1.3464e-01, -7.0299e-01, -2.0784e-01,\n",
       "          -1.8930e-01,  1.5148e-02, -8.2581e-02, -4.0589e-01, -2.3052e-01,\n",
       "          -1.0565e-01, -2.0554e-01,  5.4511e-01,  3.6979e-01, -7.7291e-02,\n",
       "           5.9600e-01,  1.4242e-01, -1.8991e-01,  6.7437e-01, -3.3533e-02,\n",
       "          -3.7891e-01,  4.9413e-01, -5.1745e-01,  2.3551e-01,  7.1042e-01,\n",
       "          -2.6997e-01,  3.4196e-01,  7.2963e-01,  2.8887e-01, -3.6308e-02,\n",
       "          -4.0591e-01,  2.0402e-01, -9.1175e-02, -3.4577e-01,  4.0640e-01,\n",
       "          -2.0259e-01,  2.4894e-01, -5.4054e-01,  1.2489e-01, -2.7970e-01,\n",
       "           1.5428e-01, -1.5434e-01, -3.7552e-01, -7.5617e-02, -5.5450e-02,\n",
       "           3.9612e-01,  2.1058e-02,  3.7063e-01,  8.9591e-02, -1.0941e+01,\n",
       "           4.0896e-01,  7.9667e-01, -5.4739e-01,  1.6996e-01, -5.4529e-02,\n",
       "           2.3010e-01, -4.8697e-01,  4.1433e-01, -5.7121e-01, -1.3745e-01,\n",
       "          -2.5062e-01, -8.3349e-02,  9.2729e-02, -1.7676e-01, -2.6181e-01,\n",
       "          -3.6784e-01, -5.1660e-01,  5.9018e-02, -1.4793e-01,  6.8787e-01,\n",
       "          -3.2056e-01,  1.0668e-01,  2.3819e-01, -4.9510e-02, -2.1127e-01,\n",
       "          -1.0429e-01, -1.0781e-01,  8.3553e-01,  2.7993e-01, -1.8303e-02,\n",
       "          -5.0317e-02,  3.4271e-01,  4.0071e-01,  1.0540e-01, -8.3132e-01,\n",
       "           3.6390e-01,  1.5531e-01,  1.7715e-01, -3.7652e-01, -2.9092e-01,\n",
       "          -4.9873e-01, -8.4516e-01,  1.0931e-01,  5.5940e-01,  7.5442e-01,\n",
       "           8.0117e-01,  1.3531e+00,  3.6676e-01,  1.6930e-01, -1.6899e-01,\n",
       "          -6.9579e-01,  1.6943e-01, -7.4212e-01, -1.6019e-01, -2.2052e-02,\n",
       "           4.0782e-01, -5.2573e-01,  1.2986e-01, -4.1854e-01, -2.0464e-01,\n",
       "           3.9354e-01,  7.6379e-03, -1.7021e-02, -4.2552e-01, -4.0402e-01,\n",
       "           1.8625e-01, -2.1958e-01, -7.1547e-02, -1.9639e-01,  2.7173e-01,\n",
       "           6.3061e-02,  5.4905e-01,  3.9612e-01, -2.5567e-01,  4.1481e-01,\n",
       "          -2.2412e-01, -5.6609e-02,  1.9697e+00, -4.2303e-01,  1.2094e-01,\n",
       "          -1.5118e-01,  2.3388e-01, -8.6610e-02, -4.9216e-01,  1.0898e-01,\n",
       "           1.9747e-01,  1.1656e-01,  5.9281e-01,  3.6717e-01,  1.6258e-01,\n",
       "          -7.0861e-01,  1.4844e-01,  2.1485e-01,  4.2210e-01,  1.0781e+00,\n",
       "          -3.4685e-01, -2.4098e-01,  1.1593e-02,  5.2349e-02, -5.7561e-01,\n",
       "          -4.2986e-02, -2.1241e-01, -5.3335e-01,  2.1967e-01,  1.5693e-01,\n",
       "          -1.7127e-02,  3.9480e-01, -2.7921e-01,  6.9187e-01, -1.7981e-01,\n",
       "           1.5646e-01,  1.4935e-01,  6.5668e-02, -6.0470e-01, -2.9457e-01,\n",
       "           7.8761e-02,  8.9981e-01, -1.5100e-01, -7.6249e-01, -6.3626e-01,\n",
       "           2.3216e-01,  4.0410e-01, -4.8511e-02, -3.0990e-01,  4.6758e-01,\n",
       "          -7.2670e-01, -2.5676e-01, -1.8298e-01, -2.7436e-01,  2.6281e-01,\n",
       "           9.0979e-02, -2.9870e-01, -2.0645e-01, -4.5930e-01, -7.3292e-01,\n",
       "          -5.1917e-01,  1.1120e-01,  5.3153e-01, -2.1742e-01, -6.9155e-02,\n",
       "          -3.1639e-01, -9.7186e-02, -1.5750e-01,  7.2136e-02, -1.9287e-01,\n",
       "          -2.8333e-01,  1.6192e-01,  2.5155e-01,  3.7399e-01, -4.7742e-02,\n",
       "          -1.2147e-01, -5.6657e-01, -3.7437e-02,  3.4372e-01, -6.8354e-01,\n",
       "          -4.0905e-01, -1.8339e-02, -1.8585e-01,  3.2216e-01,  2.5456e-01,\n",
       "           1.2476e-01, -6.5141e-02, -2.6504e-01,  4.4735e-01,  3.8062e-01,\n",
       "           6.3390e-03, -5.9769e-01, -3.4258e-01, -7.6699e-02,  1.2386e-01,\n",
       "           1.7695e-01, -4.7578e-02, -5.4417e-01, -1.0946e-01, -2.7955e-01,\n",
       "          -1.7415e-01,  1.0316e+00, -3.4684e-01, -5.4607e-01,  4.7267e-01,\n",
       "           4.2459e-01, -4.7064e-02, -3.9601e-02, -1.5120e-02, -2.9819e-01,\n",
       "          -4.8109e-01, -5.4989e-01,  1.9818e-01, -2.6071e-01,  3.8255e-01,\n",
       "           1.0360e+00, -2.0003e-01, -2.9013e-02,  2.2510e-01,  3.7220e-01,\n",
       "          -1.8653e-01, -1.7511e-01, -4.6492e-01,  7.8535e-02,  1.6397e-01,\n",
       "           3.1879e-01,  1.3882e-01,  1.7706e-01, -2.0491e-01, -6.9445e-03,\n",
       "           5.3202e-01,  1.1155e+00, -1.2616e-01, -5.4864e-01,  6.0156e-01,\n",
       "          -1.2138e-01,  1.6840e-01, -2.6034e-01, -2.0639e-01, -2.0764e-02,\n",
       "           8.2657e-02, -3.5581e-01, -3.1389e-01,  2.1075e-01,  2.8290e-01,\n",
       "           1.0073e-01, -4.9041e-01, -3.4727e-01, -2.2288e-01,  6.5281e-01,\n",
       "          -1.3400e-01,  4.4078e-01, -1.9632e-01,  6.3128e-01,  2.5844e-01,\n",
       "          -1.1164e+00, -5.1755e-02, -3.3659e-01,  4.1359e-01,  5.1423e-02,\n",
       "          -4.9909e-01, -3.2263e-01, -2.9965e-01, -2.3192e-01,  3.2790e-01,\n",
       "           2.2128e-01, -9.9607e-01,  2.6138e-01,  6.7820e-01, -9.3552e-02,\n",
       "          -2.5074e-01, -3.6878e-01,  3.2626e-01,  2.4218e-01, -1.9209e-01,\n",
       "          -2.6693e-01,  4.1681e-01, -2.9625e-01, -2.7526e-01, -1.1704e-01,\n",
       "           8.3041e-01,  5.1388e-02, -6.2027e-02, -2.9563e-01,  7.3826e-03,\n",
       "          -4.3112e-01, -3.6391e-01, -3.0411e-01, -6.6916e-02, -1.9304e-02,\n",
       "           1.3655e-01, -3.1333e-01, -2.0925e-01,  5.2036e-01, -2.8063e-01,\n",
       "           3.0513e-01,  4.7127e-01,  3.1973e-01, -6.6478e-03, -3.3109e-01,\n",
       "          -1.9623e-01,  2.5008e-01, -2.7990e-02,  6.3841e-01, -1.1352e-01,\n",
       "          -3.9979e-01,  3.2312e-01,  1.3751e-01, -2.0148e-01, -2.2390e-01,\n",
       "          -3.2695e-01,  1.1764e-01, -4.0071e-01, -2.5397e-01,  3.8604e-01,\n",
       "           6.7360e-02,  1.0904e-01, -8.9894e-01,  7.1455e-02,  3.9193e-01,\n",
       "          -1.8117e-01,  5.8739e-01,  4.8386e-02,  6.2215e-01, -3.5673e-01,\n",
       "           2.1833e-01, -7.7363e-01,  9.7510e-02, -5.1534e-01, -5.6897e-02,\n",
       "          -2.3072e-01, -3.4397e-01,  6.4170e-01, -2.1036e-01,  5.1389e-01,\n",
       "           5.8055e-01, -1.6410e-01, -6.6763e-01,  1.5672e-01, -1.9564e-01,\n",
       "           3.6032e-01, -5.2849e-01,  4.1713e-01,  1.5973e-01,  3.9070e-01,\n",
       "          -8.6437e-01, -3.6562e-01, -6.9349e-03,  2.4324e-01, -1.3539e-02,\n",
       "           1.1308e+00,  3.3768e-02, -1.7917e-01,  4.2120e-02,  9.9611e-02,\n",
       "           3.8454e-02,  9.8922e-01, -1.2609e-01, -3.7534e-01, -3.9446e-01,\n",
       "           3.1308e-01,  1.3074e-01,  1.9680e-01, -2.4187e-01,  4.6688e-01,\n",
       "           1.5210e-01,  2.6363e-01,  6.5947e-01]], device='mps:0'),\n",
       " tensor([[-2.5187e-01, -4.0054e-01, -5.8007e-01, -8.3055e-01, -1.2868e-01,\n",
       "          -2.8804e-02, -7.1917e-01,  6.6701e-01,  1.2709e-01,  5.0855e-01,\n",
       "           1.2570e-01,  5.1916e-01,  5.1676e-02, -1.1521e-01, -1.6481e-01,\n",
       "          -4.9820e-01,  5.6222e-02,  3.7338e-01,  6.7187e-01, -2.4111e-02,\n",
       "           2.2844e-01,  5.7979e-01, -2.6323e-01, -1.2386e-02, -8.3013e-01,\n",
       "           2.9613e-01, -1.1576e-01,  2.2409e-01,  1.7718e-01, -6.3092e-01,\n",
       "           4.0999e-02,  4.7417e-01,  2.3068e-02,  1.3497e+00, -5.6080e-01,\n",
       "          -1.3230e-01,  8.2213e-02, -3.9453e-01, -2.3010e-01,  3.7283e-02,\n",
       "           6.1164e-01, -4.1315e-01,  1.3635e-01,  4.0254e-01, -6.1085e-02,\n",
       "           8.9076e-02,  1.3899e-01, -3.7744e-01,  2.5892e-01, -8.3961e-01,\n",
       "          -4.8318e-02, -2.0669e-01,  4.9144e-01,  6.2615e-01, -6.3176e-01,\n",
       "          -2.2975e-01,  2.7419e-01, -3.7716e-02, -3.3020e-02,  1.5349e-01,\n",
       "           9.7980e-02,  3.8582e-01, -2.9995e-01,  2.3257e-01,  3.3441e-01,\n",
       "           2.4458e-01, -3.0392e-01, -4.8889e-01, -5.4078e-01, -5.0111e-02,\n",
       "           9.1859e-03,  2.2490e-02,  1.4846e-01, -2.3272e-01,  4.1973e-01,\n",
       "          -7.1214e-01, -6.4354e-02, -8.9125e-02, -2.5973e-01,  2.9246e-01,\n",
       "          -4.5549e-02,  1.9719e-01,  3.1010e-01,  1.0268e-01,  1.2457e-01,\n",
       "           4.3419e-02,  2.7215e-01,  5.2932e-04,  4.7373e-02,  3.7533e-01,\n",
       "          -1.3375e-01, -8.1722e-02, -2.8313e-01, -3.5814e-01,  3.1229e-01,\n",
       "           4.0598e-02,  2.9907e-01, -1.9557e-01,  1.8579e-01, -3.2474e-01,\n",
       "           1.5588e-01,  1.7768e-01, -1.8083e-01,  5.7286e-02, -3.5172e-01,\n",
       "          -4.5475e-01,  2.3589e-02, -2.8569e-01, -4.8285e-01,  2.2697e-01,\n",
       "          -1.4211e-01, -3.0205e-01, -4.0922e-02, -3.8935e-01, -3.1524e-02,\n",
       "          -7.5164e-02,  2.6013e-01, -1.3230e-01, -3.4806e-01, -8.3033e-02,\n",
       "           2.0734e-01, -2.2808e-01,  1.6077e-01,  2.6882e-01,  1.2232e-01,\n",
       "          -9.6705e-02,  6.5251e-01,  1.8950e-02,  3.1674e-01,  2.5763e-01,\n",
       "           1.1699e-01, -6.5768e-02, -1.9745e-01, -2.3154e-01,  7.5988e-02,\n",
       "          -7.6941e-02,  1.0687e-01, -1.8182e-01, -2.2790e-01, -1.8508e-01,\n",
       "           3.1771e-01, -3.3097e-01, -6.0893e-01, -4.9260e-01, -1.6098e-01,\n",
       "          -1.6326e-01, -2.4804e-02, -8.2473e-03, -3.3693e-01, -3.1956e-01,\n",
       "          -6.7443e-01, -3.6717e-01,  8.1041e-01, -1.6831e-02, -3.3356e-01,\n",
       "          -1.6499e-01,  2.3208e-01,  3.1666e-01, -4.5959e-02,  3.7161e-01,\n",
       "           7.7250e-01, -4.7670e-02,  1.6786e-01, -5.1827e-01, -3.8195e-01,\n",
       "           4.8335e-02,  2.6658e-01, -2.7054e-01,  6.1879e-01,  4.8352e-02,\n",
       "           7.5627e-02, -6.6170e-02,  2.7430e-01, -3.2704e-01, -1.2324e-02,\n",
       "          -5.7573e-01,  3.7750e-01,  2.9539e-01, -3.9892e-01, -6.6567e-02,\n",
       "          -2.0834e-02, -1.2060e+00,  9.6964e-02, -8.3040e-01,  3.1236e-01,\n",
       "          -2.3110e-01,  1.7963e-01,  2.3389e-01,  2.9545e-01, -6.5320e-02,\n",
       "           2.5404e-01,  2.8725e-01,  2.5687e-01,  3.3609e-01, -3.4924e-02,\n",
       "           2.8019e-01,  2.3924e-01,  5.0790e-01,  4.1073e-01,  1.4769e-01,\n",
       "          -3.5440e-01,  2.9683e-01,  3.5781e-01,  9.0598e-02,  1.6954e-01,\n",
       "           9.9825e-02, -2.1636e-01,  2.1091e-01,  1.7501e-01, -1.6618e-01,\n",
       "          -2.2406e-01, -3.1100e-02, -4.6818e-01, -5.6655e-02, -4.8467e-01,\n",
       "           3.2557e-01, -5.1983e-01, -1.1912e-01,  3.4831e-01,  2.4340e-01,\n",
       "          -1.4896e-02,  1.8319e-01,  1.7196e-01, -2.4338e-01,  2.1735e-01,\n",
       "           3.7449e-01, -3.4618e-01, -3.9804e-02, -3.2197e-01, -2.9367e-01,\n",
       "           2.4357e-01, -5.6244e-01,  6.6635e-01,  4.0574e-01,  6.0165e-02,\n",
       "           1.5119e-01, -2.7920e-01, -5.1638e-02, -5.7408e-01,  3.5471e-01,\n",
       "          -3.2130e-01, -3.4955e-01, -5.4801e-01,  6.4487e-02,  4.2773e-01,\n",
       "           2.7565e-01, -6.4915e-02,  7.7868e-01,  4.2408e-01,  4.2813e-01,\n",
       "          -4.3003e-01, -1.5896e-02,  4.0377e-01, -4.1405e-02, -4.6059e-01,\n",
       "           6.7116e-02, -1.4962e-01, -5.1643e-01,  2.8000e-01, -2.6598e-01,\n",
       "          -1.2047e-01, -1.4913e-01,  3.3682e-01,  1.5365e-01, -2.1964e-01,\n",
       "          -3.8461e-01,  3.9587e-01, -1.4349e-01, -4.1723e-01,  4.7123e-01,\n",
       "          -3.5025e-01,  1.5209e-01,  1.0694e-01, -3.8441e-01, -6.2072e-01,\n",
       "          -3.2235e-01,  2.3090e-01,  1.5585e-01,  3.4799e-02, -6.6306e-01,\n",
       "          -4.0571e-01,  3.1197e-01,  2.4601e-01,  2.9976e-01,  2.9840e-01,\n",
       "           5.7427e-01, -3.0944e-01, -1.6702e-01,  1.4982e-01, -4.1377e-01,\n",
       "          -1.5704e-01, -3.3228e-02, -1.9995e-01,  2.9569e-01,  2.4768e-01,\n",
       "           1.1600e-01,  2.0367e-01,  4.3482e-01, -2.5272e-01, -3.8673e-01,\n",
       "          -7.1543e-02,  2.0161e-01, -9.3629e-02, -5.8306e-02,  2.3951e-01,\n",
       "          -6.3095e-02,  5.9650e-01,  7.2143e-01,  7.7439e-02, -6.4006e-01,\n",
       "           1.7696e-01,  5.2502e-01, -2.7253e-01, -4.0343e-02,  9.6365e-02,\n",
       "          -1.0344e+00, -2.1331e-01, -1.2883e-01, -2.0181e-01, -6.5352e-01,\n",
       "           8.8115e-02,  3.4863e-01,  4.6833e-03, -1.7864e-01, -3.0947e-01,\n",
       "           1.0448e-01,  4.8724e-02, -4.8980e-02,  4.3534e-01, -2.8901e-01,\n",
       "          -3.4797e-01, -2.3438e-01, -2.1532e-01, -5.4201e-01, -3.9968e-01,\n",
       "           3.5914e-01,  2.8129e-01,  1.1986e-01, -2.6486e-01, -5.1832e-01,\n",
       "           4.5577e-01, -6.8169e-01, -1.4632e-01,  2.4840e-01, -4.7702e-01,\n",
       "           1.4650e-02,  4.8990e-01,  1.4999e-01,  1.6336e-01,  6.5258e-02,\n",
       "           7.8557e-02, -3.3903e-01, -5.0030e-01, -1.5000e-01,  6.2323e-01,\n",
       "           3.2283e-01,  3.3506e-01, -1.7236e-01,  3.9096e-01,  4.5167e-01,\n",
       "          -2.9582e-01,  4.1670e-01,  1.0930e-02,  7.7439e-02, -6.0414e-02,\n",
       "           1.9048e-01,  3.4370e-01, -3.6842e-01,  2.0145e-01,  1.7929e-01,\n",
       "          -3.1634e-01, -3.2149e-01,  1.5691e-02,  2.0642e-01, -1.8580e-01,\n",
       "          -3.0875e-01, -2.0361e-01,  9.9698e-02, -3.4856e-01, -4.3104e-01,\n",
       "           2.9161e-01, -3.6543e-02,  7.0495e-02, -4.0357e-01, -4.3462e-01,\n",
       "          -2.0181e-01, -2.6055e-01,  6.8681e-01,  2.1064e-01, -1.9621e-01,\n",
       "           1.8720e-01, -1.0907e-01, -3.6330e-01,  4.7847e-01,  2.0162e-02,\n",
       "          -4.8754e-01,  4.5482e-01, -3.4822e-01,  1.0627e-01,  5.5756e-01,\n",
       "          -1.6769e-01, -1.0840e-01,  1.1008e-01,  1.0820e-01,  2.9026e-01,\n",
       "          -3.9884e-01,  1.5516e-01, -6.3476e-02, -3.1885e-01,  3.1820e-01,\n",
       "           2.6411e-01,  2.2781e-01, -1.7137e-01, -3.3426e-02,  3.5677e-02,\n",
       "           2.8782e-01, -6.0853e-02, -3.9888e-01, -4.7347e-01, -4.6132e-01,\n",
       "           1.2832e-01,  1.8927e-01,  5.6207e-01,  3.4144e-01, -1.1394e+01,\n",
       "           1.2938e-01,  7.8031e-01, -4.0713e-01, -8.2819e-02, -5.0320e-02,\n",
       "           1.0976e-02, -1.3708e-01,  4.2955e-01, -2.7734e-01, -4.4105e-01,\n",
       "          -3.5727e-01, -1.3308e-01,  1.5454e-01, -3.2189e-01, -2.6945e-01,\n",
       "          -1.4686e-02, -7.0935e-01,  4.0963e-01, -3.0286e-01,  5.0298e-01,\n",
       "          -2.3511e-01, -8.3798e-02,  1.5816e-01, -3.5851e-01, -1.3439e-02,\n",
       "           5.9766e-02, -1.6033e-01,  1.4651e-01,  4.2486e-01, -5.2072e-02,\n",
       "           3.3244e-02, -4.1258e-03,  4.8461e-02, -2.5723e-01, -5.9999e-01,\n",
       "           1.9296e-02, -1.4141e-01,  3.3422e-01, -4.4133e-02, -7.1265e-01,\n",
       "          -2.2533e-01, -2.7418e-01,  3.1407e-01,  2.7021e-01,  9.0598e-01,\n",
       "           2.2955e-01,  1.0686e+00,  1.5022e-01,  1.7748e-01,  2.6230e-02,\n",
       "          -7.8764e-01, -3.3269e-02, -4.1907e-01,  2.3280e-02,  1.0114e-01,\n",
       "           3.6576e-01, -7.9367e-01,  2.5251e-01, -3.5521e-01,  3.6487e-02,\n",
       "           4.9094e-01,  3.1768e-02,  2.7700e-01, -3.2298e-02, -7.8123e-02,\n",
       "           1.2089e-01, -2.8544e-02,  1.1098e-01, -2.2583e-01, -2.4084e-01,\n",
       "          -1.0177e-01,  5.9888e-01,  3.9592e-01, -7.2211e-01,  3.6112e-01,\n",
       "           4.9258e-02, -1.8622e-01,  1.4556e+00, -4.8573e-02,  1.8361e-02,\n",
       "           1.9800e-03,  2.9604e-01, -5.0236e-01, -7.3516e-01,  3.6511e-01,\n",
       "          -1.0211e-01,  5.6427e-02,  1.0127e-01,  4.8148e-01, -9.7481e-02,\n",
       "          -4.7940e-01, -1.8593e-03,  1.0556e-01, -1.3380e-01,  5.4710e-01,\n",
       "          -7.2824e-01, -1.6963e-01, -1.1223e-01,  3.1723e-01, -8.4054e-01,\n",
       "          -6.1126e-02, -7.0210e-01, -3.9328e-01,  5.0586e-01, -2.5896e-01,\n",
       "           1.5852e-01, -2.7443e-02, -4.3439e-02,  5.5574e-01, -2.1489e-01,\n",
       "           7.1555e-02,  2.2813e-01,  8.4995e-02, -1.6733e-01,  3.9349e-02,\n",
       "           4.1630e-01,  4.9571e-01, -4.2740e-01, -6.3911e-01, -7.6901e-03,\n",
       "           1.7180e-01,  1.4933e-01,  3.4019e-01, -5.3463e-01,  3.6431e-01,\n",
       "          -7.9162e-01, -1.8769e-01,  7.2949e-02, -4.3273e-02,  3.6175e-01,\n",
       "           2.3925e-01,  7.2038e-02,  2.2256e-01, -7.7756e-01, -6.4498e-01,\n",
       "          -6.5932e-01, -2.1445e-01,  2.6143e-01, -5.4236e-01, -8.3654e-02,\n",
       "          -1.1033e-01, -4.8837e-03, -5.0292e-02, -6.6102e-02,  1.1939e-02,\n",
       "           1.3632e-01,  2.3587e-01, -1.3023e-01, -1.3178e-01, -9.2464e-02,\n",
       "          -5.5566e-01, -1.1753e-01, -9.9591e-02, -1.2505e-01, -5.6783e-01,\n",
       "          -1.3797e-01,  3.3075e-02, -2.5373e-01, -6.8124e-02,  3.5608e-01,\n",
       "          -2.9832e-02, -2.7856e-02, -4.8850e-01,  7.2846e-01,  4.0279e-01,\n",
       "           5.2088e-02, -2.8298e-01, -5.5189e-01, -1.0567e-01,  2.9592e-01,\n",
       "           1.9555e-01, -6.7007e-02, -1.7709e-01, -2.4145e-01, -4.2522e-01,\n",
       "           1.4302e-01,  7.1147e-01, -6.6296e-02, -6.8115e-01,  2.6621e-01,\n",
       "           5.8223e-01, -1.2882e-01, -2.4432e-01,  2.0983e-02, -9.3284e-02,\n",
       "          -3.1851e-01, -6.0317e-01,  7.8821e-01, -1.1246e-01, -5.8638e-02,\n",
       "           2.2427e-01,  2.0359e-01, -9.2588e-03,  4.0209e-01, -1.5770e-01,\n",
       "          -3.5434e-01, -1.6477e-03, -7.3600e-01,  4.3257e-01,  6.1666e-02,\n",
       "          -3.1489e-01,  7.6082e-01,  2.7260e-01, -6.0593e-02, -4.5779e-02,\n",
       "           2.2863e-01,  7.9941e-01, -2.4636e-01,  6.4653e-02,  5.3779e-01,\n",
       "           8.8110e-02, -1.1229e-01,  3.9580e-02,  4.1665e-02,  4.5410e-02,\n",
       "           3.7012e-01, -3.2145e-02, -4.1564e-01, -6.1194e-02,  3.8924e-01,\n",
       "           3.9295e-01, -5.4610e-01, -7.3487e-01,  2.2333e-02,  6.8551e-01,\n",
       "          -2.9054e-01,  2.3654e-01, -3.2708e-02,  3.0880e-01,  1.2001e-01,\n",
       "          -9.2160e-01, -1.5016e-01, -1.0056e-01,  2.8229e-01, -2.5482e-02,\n",
       "          -1.6139e-01, -1.4727e-01, -2.3686e-01, -1.6208e-01,  6.8308e-01,\n",
       "           2.1991e-01, -6.9160e-01,  3.4058e-02,  5.7281e-01, -6.0551e-01,\n",
       "           4.6052e-01, -2.4577e-01, -3.4302e-02,  3.6435e-01,  1.3117e-01,\n",
       "           4.9067e-02,  1.7856e-01, -4.2138e-01, -4.0598e-01,  1.7824e-01,\n",
       "           9.1672e-01, -7.6613e-02,  1.1166e-01, -3.9384e-01, -1.7182e-01,\n",
       "          -7.1855e-02, -3.1913e-01, -3.1286e-01,  3.7185e-01,  2.0529e-01,\n",
       "          -3.2552e-02, -1.3271e-01, -2.2992e-01,  3.0816e-01,  1.1821e-01,\n",
       "           3.2536e-01,  5.8187e-01,  6.4368e-01, -1.0077e-02, -2.4936e-01,\n",
       "          -6.1235e-02,  1.4568e-01, -1.9524e-01,  5.2704e-01, -2.0627e-01,\n",
       "          -3.3453e-01, -3.8069e-03,  3.0134e-01, -6.1681e-01, -5.8725e-02,\n",
       "          -4.5426e-01,  4.2361e-01, -1.6955e-01,  3.7868e-02,  2.8833e-01,\n",
       "           4.0522e-02,  5.7925e-02, -2.2651e-01,  5.5807e-01,  4.4511e-01,\n",
       "           1.6609e-01,  4.5662e-01,  1.6104e-01,  4.5272e-02, -5.1597e-01,\n",
       "           2.1978e-01, -1.0213e+00, -9.4260e-02, -4.6640e-02, -1.5401e-01,\n",
       "          -4.4224e-01, -2.8126e-02,  1.6814e-01,  1.8538e-01,  6.9597e-01,\n",
       "           7.2967e-01, -3.0871e-01, -4.5290e-01, -1.8510e-01,  1.5304e-01,\n",
       "           2.7705e-01, -1.9517e-01,  3.5971e-01,  9.6103e-02,  2.4432e-01,\n",
       "          -5.0967e-01, -4.3165e-01, -4.2252e-02,  3.1411e-01, -2.0171e-01,\n",
       "           5.7118e-01, -2.0164e-01, -1.3863e-01,  9.0832e-02,  1.3770e-01,\n",
       "           1.6639e-01,  7.9664e-01,  4.3416e-01, -6.3425e-01,  8.7469e-02,\n",
       "          -1.3980e-02, -2.9988e-01,  2.5983e-02, -4.3577e-01,  2.8858e-01,\n",
       "           1.2984e-01,  4.9280e-01,  6.7504e-01]], device='mps:0'),\n",
       " tensor([[-4.7220e-01, -3.8117e-01, -6.2516e-01, -7.3973e-01, -2.7943e-01,\n",
       "          -2.6227e-01, -9.7091e-01,  8.4637e-01,  1.0816e-01,  4.7817e-01,\n",
       "           2.9859e-01, -6.6096e-02,  6.6570e-02, -1.8509e-01, -7.1786e-01,\n",
       "          -3.1789e-01, -1.3235e-01,  3.6509e-01,  3.7310e-01,  3.2984e-01,\n",
       "           8.1565e-01,  2.8582e-01, -1.5921e-01, -1.4603e-02, -5.9030e-01,\n",
       "           3.9568e-01,  1.6204e-01,  1.7514e-01,  3.9385e-01, -8.8542e-01,\n",
       "           1.6616e-01,  3.0638e-01, -9.3394e-02,  7.7724e-01, -4.7268e-02,\n",
       "          -2.7849e-01,  2.4065e-01, -4.8312e-01,  7.2961e-01,  3.8307e-01,\n",
       "           1.6465e-01, -3.3334e-01, -1.7177e-01,  1.6416e-01, -1.6606e-01,\n",
       "          -7.9960e-03, -2.7629e-02, -7.4943e-01,  6.2366e-01, -3.8599e-01,\n",
       "          -3.9590e-01, -3.9669e-02,  5.1597e-01,  2.5064e-01, -7.7905e-01,\n",
       "          -5.6334e-01,  4.5312e-01,  4.7027e-01, -5.8684e-01,  3.8373e-01,\n",
       "          -1.2564e-01,  7.2400e-01,  3.2037e-01, -6.8452e-02,  2.7110e-01,\n",
       "           1.7342e-01, -3.8835e-01, -5.7688e-01, -3.2915e-02, -1.1367e-01,\n",
       "          -1.7784e-01, -5.6692e-02, -4.6691e-01, -3.4181e-01,  5.7574e-01,\n",
       "           2.1254e-01,  2.6113e-01,  5.4404e-01, -3.5402e-01,  5.1981e-02,\n",
       "           3.5865e-01,  7.0396e-02,  8.9668e-02,  3.0427e-02,  2.3425e-01,\n",
       "           2.2470e-01,  5.8048e-01,  2.0344e-01, -5.7798e-02, -1.6609e-01,\n",
       "           5.2360e-01,  9.9354e-02, -1.0054e-01, -7.7684e-01,  2.9883e-01,\n",
       "          -4.4370e-01,  2.6337e-01, -2.2284e-01,  1.7631e-01, -7.0128e-02,\n",
       "           2.0104e-01,  6.4211e-01, -1.6110e-01,  8.5570e-02, -3.1819e-01,\n",
       "          -7.5121e-01,  2.4770e-01, -1.1473e-01, -5.8056e-01,  1.4448e-01,\n",
       "          -7.4708e-02, -2.5991e-01,  2.5036e-01, -1.2521e-01, -2.3587e-01,\n",
       "          -8.1233e-02,  2.4232e-01, -1.0280e-01, -3.7440e-01,  4.8518e-01,\n",
       "           2.5100e-01, -5.0992e-01, -4.8613e-01, -8.1599e-02,  3.4820e-01,\n",
       "          -3.7485e-01,  5.2892e-01,  5.4453e-02,  4.1245e-01,  7.9263e-01,\n",
       "           1.3964e-01,  7.4558e-02, -1.7285e-01, -2.5711e-01, -3.0945e-01,\n",
       "           1.5692e-01, -7.2114e-02, -4.0579e-01,  3.9470e-01,  1.5577e-01,\n",
       "          -5.6334e-01, -2.4016e-01, -5.5589e-01, -2.2570e-01, -1.0711e-01,\n",
       "          -4.7424e-01,  1.4801e-01, -1.3357e-01,  2.1710e-01, -4.2236e-01,\n",
       "          -7.6752e-01, -7.8821e-01,  3.9770e-01,  9.6042e-02,  2.5411e-02,\n",
       "          -2.7616e-01,  2.6405e-01,  2.6542e-01, -1.6485e-02,  2.7675e-01,\n",
       "           6.4904e-01,  1.2940e-01,  2.2829e-01, -5.3446e-01, -3.3295e-02,\n",
       "           8.4328e-02, -2.3243e-01, -2.0410e-01,  3.0862e-02,  2.9375e-01,\n",
       "           2.4108e-01, -4.3052e-01,  6.0318e-02, -4.8793e-01,  5.4443e-04,\n",
       "          -5.8874e-01,  4.5771e-01,  2.3259e-01, -3.2538e-01, -1.5534e-01,\n",
       "          -3.7092e-01, -1.9178e+00,  2.8214e-01, -3.4930e-01, -1.3464e-01,\n",
       "          -7.3188e-01,  2.2316e-01,  2.1091e-01,  1.6665e-01,  2.5349e-01,\n",
       "          -1.7133e-01, -1.4836e-01, -2.5622e-01,  4.1424e-01, -4.5472e-02,\n",
       "           6.0199e-01,  6.9804e-01, -8.9213e-02, -4.8654e-01,  1.0606e-02,\n",
       "          -1.1399e-01,  2.0207e-01,  1.6015e-02,  8.8794e-02,  1.7516e-01,\n",
       "           4.3227e-01, -7.7075e-02,  3.1742e-02,  3.0671e-01, -1.9787e-01,\n",
       "           1.0063e-01, -2.7662e-01, -3.8091e-01, -7.4360e-02, -2.4591e-01,\n",
       "           4.8965e-02, -3.5960e-01,  1.0821e-01,  2.7344e-01,  8.9186e-02,\n",
       "          -2.7666e-01,  1.4054e-01, -4.1336e-01, -4.7701e-01, -2.3517e-01,\n",
       "           5.4084e-01, -2.1399e-01,  2.6586e-01, -4.8143e-01, -2.7768e-02,\n",
       "          -3.9256e-01, -4.7822e-01,  4.8124e-01,  5.0621e-01,  2.9946e-01,\n",
       "          -3.5768e-01, -2.7176e-01,  2.6138e-01, -1.5176e-01, -3.6322e-01,\n",
       "          -2.9211e-01, -4.6154e-01, -4.2638e-01,  2.4159e-02,  4.5309e-01,\n",
       "           5.8796e-03, -6.4210e-02,  9.1871e-01,  8.4846e-01,  1.3622e-01,\n",
       "          -4.4417e-03, -1.1102e-01,  3.5660e-01,  3.9509e-01, -2.6069e-01,\n",
       "          -3.0156e-01, -7.8316e-02, -4.2826e-01,  5.4681e-01, -2.6382e-01,\n",
       "           3.6494e-01, -2.1415e-02,  1.2874e-01,  8.3575e-01,  5.8376e-02,\n",
       "          -4.0432e-01,  1.6270e-01,  1.1758e-01, -1.6087e-01, -4.0037e-03,\n",
       "          -1.1391e-01,  2.7116e-01, -5.2223e-01, -1.9393e-01, -5.8126e-01,\n",
       "          -3.1163e-01, -1.6801e-01,  4.7742e-02,  3.7402e-01, -8.2556e-01,\n",
       "          -1.5150e-01,  4.7127e-01, -2.1854e-01,  3.1632e-01,  2.0948e-01,\n",
       "           5.1000e-01,  2.4296e-01,  1.6579e-01, -3.9279e-01,  6.4216e-02,\n",
       "           1.3500e-02,  2.4321e-01, -3.8971e-01,  1.4010e-02,  7.3395e-02,\n",
       "          -2.7710e-01,  2.7947e-01,  3.8154e-01,  8.7040e-02, -5.5827e-01,\n",
       "          -1.0441e-01, -1.4013e-01, -4.5655e-01, -3.2503e-01,  3.1347e-02,\n",
       "          -2.4748e-01,  4.0940e-01,  8.0560e-01, -7.9063e-03, -1.5308e-01,\n",
       "           1.0434e-01,  7.2943e-01,  4.1394e-02,  2.3299e-01, -2.8244e-02,\n",
       "          -1.0814e+00,  5.3330e-03, -1.4501e-01, -2.5200e-01, -3.2169e-01,\n",
       "           1.9506e-01,  5.6313e-01, -2.9121e-01, -4.2384e-01, -6.1260e-01,\n",
       "           3.9891e-01,  2.0875e-01,  5.7389e-01, -1.1632e-01, -2.9270e-01,\n",
       "           1.1068e-01, -5.6310e-01,  1.3029e-01, -2.9292e-01, -1.7047e-01,\n",
       "           2.6215e-01,  1.9004e-01, -5.6641e-02,  5.7105e-01, -4.2109e-01,\n",
       "           8.4425e-01, -1.8820e-01, -3.8530e-01,  2.2636e-01, -1.5825e-01,\n",
       "          -5.5356e-01,  3.7552e-01, -1.9947e-01,  2.5263e-01,  1.3057e-01,\n",
       "           6.0098e-01, -2.8243e-01, -3.2614e-01, -1.4191e-01,  4.9671e-01,\n",
       "          -4.5638e-02,  4.1505e-01, -3.0214e-01,  2.4624e-01,  3.2734e-01,\n",
       "          -3.5899e-01,  3.9232e-01, -5.2367e-01, -6.7999e-01,  1.2628e-01,\n",
       "           3.9814e-01,  3.0460e-01, -2.2110e-01, -1.3129e-01,  4.3180e-01,\n",
       "           3.3035e-01, -1.2623e-01,  2.4834e-01,  3.8397e-01,  8.5411e-02,\n",
       "          -2.0870e-01, -6.4269e-01,  1.5623e-01, -4.6603e-01, -1.7947e-01,\n",
       "           2.1499e-01, -9.0448e-02,  2.6679e-01, -2.8711e-01, -2.7399e-01,\n",
       "          -3.8118e-01,  3.0847e-01,  5.7773e-01,  1.3660e-01,  9.2345e-02,\n",
       "           1.4875e-01, -1.5667e-01, -3.2117e-01,  4.1956e-01, -2.5800e-01,\n",
       "          -3.5455e-01,  3.2314e-01, -1.7264e-01,  6.3286e-02,  2.3572e-01,\n",
       "          -5.3146e-02,  4.0970e-01,  1.1501e-01,  5.7541e-01, -3.6618e-01,\n",
       "          -3.7807e-01, -1.1026e-01, -5.8215e-03, -2.6365e-01,  1.4187e-01,\n",
       "          -8.0293e-02,  6.7881e-01, -2.0557e-01, -1.3850e-01,  6.6142e-01,\n",
       "          -1.7706e-02, -2.6882e-01,  1.7887e-01, -5.1992e-01, -3.3183e-01,\n",
       "           1.0706e-01,  1.6458e-01, -9.8918e-02,  2.9909e-01, -1.0840e+01,\n",
       "          -1.3564e-01,  7.8048e-01, -2.7977e-01, -3.7410e-02,  1.3757e-01,\n",
       "           9.1683e-02, -1.4739e-01,  2.4333e-01, -2.5401e-01, -5.0719e-01,\n",
       "          -2.5330e-01, -2.4513e-01,  5.2730e-02,  4.7149e-02, -1.9591e-01,\n",
       "           1.1218e-01, -8.0355e-01,  6.7369e-02, -5.0298e-01,  6.2393e-03,\n",
       "          -4.6921e-01,  3.9098e-01,  1.6065e-01, -3.7078e-01, -2.5008e-01,\n",
       "          -6.1873e-02, -2.1199e-01,  1.4435e-01,  1.6794e-02, -2.3664e-01,\n",
       "          -2.1415e-01, -5.4830e-02,  2.3026e-01, -1.9538e-01, -8.1314e-01,\n",
       "          -7.5701e-02,  1.7051e-01,  6.0715e-01, -2.3987e-01, -5.7736e-01,\n",
       "          -4.5001e-01, -6.1594e-01,  5.7053e-01,  3.4953e-01,  6.6533e-01,\n",
       "           8.6691e-01,  6.5168e-01, -9.4224e-03,  8.3258e-02,  2.3417e-01,\n",
       "          -5.9118e-01,  2.2453e-01, -5.9415e-01, -1.7977e-01, -4.1392e-01,\n",
       "           4.3002e-01, -7.6436e-01,  4.8619e-01, -2.1065e-01, -2.4843e-01,\n",
       "           1.4993e-01,  7.1768e-02,  2.2850e-01, -3.5889e-01, -6.0469e-01,\n",
       "           5.7483e-01, -1.2457e-01,  5.8783e-02, -3.9472e-01,  2.0251e-01,\n",
       "          -2.4085e-01,  4.6190e-01,  3.9141e-01, -5.3671e-01,  2.1460e-01,\n",
       "           1.2801e-01, -2.1318e-01,  1.1292e+00, -1.0138e-01,  3.8255e-01,\n",
       "           1.9972e-02,  9.5918e-02, -3.7103e-01, -6.3799e-01, -2.2712e-01,\n",
       "           7.6319e-02,  1.7261e-01,  3.2199e-01,  4.0971e-01, -4.3099e-01,\n",
       "          -6.8832e-01,  2.5939e-02,  2.4932e-01,  4.9559e-01,  1.0120e+00,\n",
       "          -4.5622e-01, -4.4803e-01,  5.9273e-02, -7.6108e-02, -6.9715e-01,\n",
       "          -5.4316e-02, -1.6334e-01, -5.1532e-01,  3.0931e-01, -8.7181e-02,\n",
       "           4.0218e-01,  3.0276e-01, -1.0907e-01,  2.4151e-01, -5.7459e-01,\n",
       "           3.1937e-01, -2.1440e-01,  2.0060e-01, -4.3767e-01,  3.2738e-01,\n",
       "          -8.0875e-02,  4.6318e-02, -3.1557e-01, -4.2702e-01, -1.5122e-02,\n",
       "          -9.8425e-02,  3.6040e-01,  1.9512e-01, -4.9564e-01,  4.4620e-01,\n",
       "          -3.7776e-01,  3.5409e-01,  2.9301e-01, -1.2850e-01,  6.3445e-01,\n",
       "          -1.4839e-01, -2.8298e-01,  9.7392e-02, -5.1506e-01, -4.8205e-01,\n",
       "          -4.6258e-01, -7.2125e-02,  3.8370e-01, -8.7626e-01,  5.4107e-02,\n",
       "           8.8796e-02,  3.4584e-01, -3.1635e-01,  2.7102e-01,  1.9009e-01,\n",
       "           1.4357e-01,  6.4283e-01, -1.5143e-01,  2.4453e-01, -2.8499e-01,\n",
       "          -6.4606e-01, -2.5062e-01,  3.3794e-02, -5.9565e-02, -1.0809e+00,\n",
       "          -1.2324e-01,  1.1712e-01, -2.7580e-01,  3.0844e-01, -1.0209e-01,\n",
       "           1.3986e-02, -1.7507e-01, -4.4887e-01,  4.2872e-01,  5.4993e-01,\n",
       "           2.6521e-01, -1.0998e-01, -3.4647e-01,  1.4453e-01,  3.5086e-01,\n",
       "          -5.2975e-02, -2.6779e-01, -4.2765e-01, -4.2188e-01,  1.7990e-01,\n",
       "           1.5634e-01,  7.0416e-01, -4.3747e-01, -5.3674e-01,  2.5951e-01,\n",
       "           4.0279e-01,  1.0212e-01, -4.2356e-02, -2.5791e-01, -1.2231e-01,\n",
       "          -2.4300e-01,  8.3126e-02,  2.8384e-01, -1.6091e-01,  2.9599e-01,\n",
       "           5.6243e-02,  5.9717e-01,  1.8598e-01,  3.7401e-01, -7.7453e-01,\n",
       "          -1.8476e-01,  1.0401e-01, -7.3981e-01,  3.1782e-02,  2.9136e-01,\n",
       "           7.2293e-02,  2.5885e-01, -3.3709e-01, -1.0727e-01,  2.3148e-01,\n",
       "           4.0491e-01,  8.3035e-01,  1.2699e-01,  2.4541e-01,  8.0606e-01,\n",
       "           3.0515e-01,  9.5258e-02,  1.4183e-01,  2.9865e-02,  1.2988e-01,\n",
       "           2.7003e-01, -7.5461e-02, -4.0604e-01, -1.2277e-01,  4.9380e-02,\n",
       "           4.4502e-01, -1.2779e+00, -6.8522e-01, -3.2293e-01,  6.2735e-01,\n",
       "           8.5968e-03,  2.7742e-01, -8.3087e-02,  4.4077e-01,  7.0379e-01,\n",
       "          -1.2355e+00, -4.4086e-01,  4.0380e-01,  4.3200e-01,  3.2965e-02,\n",
       "          -4.6912e-01,  2.5034e-01, -4.1453e-01, -4.4759e-01,  7.2868e-01,\n",
       "           5.0704e-01, -8.3809e-01,  8.9869e-02,  4.4899e-01, -5.8187e-01,\n",
       "           1.9586e-01,  5.8529e-02,  3.2015e-01, -1.4372e-01,  2.5361e-01,\n",
       "          -1.4912e-01,  3.7845e-01, -6.8510e-01, -4.0133e-01, -2.1575e-01,\n",
       "           3.8189e-01,  1.9472e-01,  3.6929e-01, -4.9674e-02,  1.5710e-01,\n",
       "          -4.6428e-01, -3.3347e-01, -2.3800e-01,  2.1891e-02, -3.9075e-01,\n",
       "           3.7246e-01, -3.6399e-01, -8.1400e-03,  3.9065e-01, -2.4918e-01,\n",
       "           5.1944e-01,  5.4253e-01,  4.6021e-01,  1.0712e-01, -8.4266e-02,\n",
       "           5.9500e-02, -5.3368e-02, -1.4617e-01,  5.8240e-01, -2.6604e-02,\n",
       "          -4.2366e-01,  1.8956e-01,  6.1743e-02,  3.8536e-02,  1.3010e-02,\n",
       "          -3.8344e-01,  4.0981e-01,  1.3398e-01, -4.8049e-01,  2.6606e-01,\n",
       "           4.8626e-01,  3.9398e-01, -4.0571e-01,  2.4753e-01,  9.3982e-02,\n",
       "          -1.8204e-01,  2.1849e-01, -1.1833e-01,  2.5589e-01, -8.4626e-01,\n",
       "           2.1881e-01, -4.6776e-01, -2.4703e-01,  2.9519e-01, -7.5118e-02,\n",
       "          -4.8236e-01, -1.2468e-01, -8.6961e-02, -1.3234e-01,  8.5137e-01,\n",
       "           1.0099e+00, -1.7200e-01, -5.8741e-01, -8.4271e-02,  3.0689e-01,\n",
       "          -1.3354e-01, -1.2457e-01,  2.1468e-01, -6.8691e-02, -2.5569e-01,\n",
       "          -1.2394e+00, -3.3661e-01,  1.1347e-01, -9.0199e-02, -3.0067e-01,\n",
       "           9.9320e-01, -7.0659e-02,  5.8198e-02,  4.9054e-01,  3.2113e-01,\n",
       "           2.4073e-01,  3.0128e-01, -2.5320e-02, -2.6189e-01, -1.4367e-01,\n",
       "           1.6038e-01, -9.2211e-02, -1.1339e-01, -6.1974e-01,  2.6997e-01,\n",
       "           1.9572e-01,  2.6877e-01,  8.9997e-01]], device='mps:0'),\n",
       " tensor([[-0.2276, -1.0336, -0.7305,  ...,  0.3808,  0.8645,  0.8293],\n",
       "         [-0.0238, -0.5721, -0.8340,  ...,  0.0483,  0.8204,  0.5286]],\n",
       "        device='mps:0'),\n",
       " tensor([[-0.2118, -0.8953, -0.5708,  ...,  0.3504,  0.6032,  0.6196],\n",
       "         [-0.1609, -0.3542, -0.6481,  ...,  0.2981,  0.3474,  0.7390]],\n",
       "        device='mps:0'),\n",
       " tensor([[-0.3670, -0.8917, -0.7430,  ...,  0.1222,  0.8454,  0.5251],\n",
       "         [-0.2804, -0.8977, -0.7813,  ...,  0.0441,  0.7521,  0.7393],\n",
       "         [-0.2170, -0.9206, -0.6892,  ...,  0.0033,  0.6907,  0.6555]],\n",
       "        device='mps:0'),\n",
       " tensor([[-6.1750e-01, -4.2829e-01, -2.3934e-01, -9.4116e-01, -3.4548e-01,\n",
       "          -6.5496e-02, -6.3874e-01,  9.7828e-01, -9.5892e-02,  4.0096e-01,\n",
       "           1.4919e-01, -5.1603e-03,  2.0980e-01,  1.3680e-01, -6.9506e-01,\n",
       "           1.6740e-01,  4.6094e-01,  2.1159e-01,  3.5951e-01, -4.4329e-02,\n",
       "           7.9832e-01,  3.4245e-01,  1.8019e-01, -2.5205e-01, -4.5876e-01,\n",
       "           5.8994e-01, -2.4628e-01, -9.2014e-02,  2.0509e-01, -8.1949e-01,\n",
       "          -1.3586e-01, -4.9757e-02, -3.6281e-01,  8.7602e-01,  1.1948e-01,\n",
       "          -2.7310e-01, -1.5276e-01, -2.7996e-01,  2.3494e-01, -6.3637e-02,\n",
       "           3.7814e-01,  2.6592e-02, -2.9347e-01,  3.1386e-01, -1.6421e-01,\n",
       "          -1.4057e-01,  1.4047e-02, -7.8429e-01,  1.3666e-02,  1.8955e-01,\n",
       "           5.8851e-02, -4.7205e-02,  1.9231e-01,  3.7030e-01, -1.0617e+00,\n",
       "          -2.6348e-01,  2.7603e-01,  2.9849e-01, -1.7932e-01, -5.7784e-02,\n",
       "          -1.9989e-01,  9.0201e-01,  1.7949e-01, -2.7822e-01,  1.7824e-01,\n",
       "          -4.4899e-01, -5.4601e-01, -6.0781e-01, -1.0682e-01, -1.1967e-01,\n",
       "          -1.0002e-01,  9.7528e-02,  1.4009e-01, -2.9385e-01,  2.3170e-01,\n",
       "           2.1813e-01,  4.7024e-02, -2.7652e-02, -4.3158e-01, -2.6546e-01,\n",
       "           3.6947e-01,  3.4425e-02, -1.4730e-02,  2.0326e-01, -5.8744e-01,\n",
       "          -2.0226e-01,  3.7959e-01,  1.5703e-01,  2.5914e-01,  2.2662e-03,\n",
       "           5.9967e-01,  1.2024e-02,  8.0339e-02, -4.5369e-01,  5.8427e-01,\n",
       "          -5.4955e-01,  1.0329e-01, -1.6007e-01,  4.9532e-02, -5.2088e-01,\n",
       "          -9.8735e-02,  2.4494e-01, -1.8085e-01,  1.6053e-01, -4.6592e-01,\n",
       "          -3.2983e-02,  2.4514e-01,  3.0468e-01, -4.9183e-01,  4.2685e-02,\n",
       "          -6.5134e-03, -2.9791e-01,  7.3694e-02, -6.6407e-01, -8.5795e-01,\n",
       "           5.8354e-01,  1.8428e-02, -4.5240e-01, -2.3930e-01,  7.4443e-01,\n",
       "           2.9738e-01, -2.3201e-01, -5.3072e-02, -3.5548e-01,  2.0259e-01,\n",
       "          -1.8518e-01,  2.5689e-01, -2.7018e-01,  3.8439e-01,  5.5026e-01,\n",
       "           3.1581e-01,  2.8489e-01, -1.8327e-01,  2.7637e-01,  1.4465e-01,\n",
       "           3.6947e-02, -6.6614e-02, -1.5929e-01,  1.5180e-01,  1.2041e-01,\n",
       "          -2.9580e-01, -3.2999e-01,  5.4746e-03,  1.1182e-01, -3.3279e-01,\n",
       "          -1.0202e-01,  1.8680e-01,  3.4808e-01, -1.4092e-01, -2.5403e-01,\n",
       "          -7.8984e-01, -4.2328e-01,  1.0217e-01, -1.8560e-01, -1.7135e-01,\n",
       "          -4.1911e-01,  3.8121e-01, -6.6660e-02,  4.7638e-01,  3.4189e-01,\n",
       "           6.9978e-01, -1.5311e-01,  5.9776e-02, -5.6412e-01,  3.5197e-01,\n",
       "           3.8270e-02, -6.1110e-01, -1.7247e-01,  1.5680e-01, -4.1813e-04,\n",
       "           6.8195e-01,  7.0979e-02,  4.0124e-01,  1.2070e-01, -3.2553e-01,\n",
       "           8.2089e-03,  1.9890e-01, -1.2864e-01, -1.4297e-01,  7.3766e-01,\n",
       "          -5.5188e-01, -2.3029e+00,  1.0721e-01, -3.5844e-01, -3.6529e-01,\n",
       "          -1.8135e-01,  3.7255e-01,  3.3772e-02,  4.9966e-01,  1.5486e-01,\n",
       "           3.7598e-02,  1.6729e-01, -2.8025e-02,  3.6851e-01, -8.4399e-02,\n",
       "           2.6751e-01,  3.3341e-01,  1.2988e-01, -2.3298e-01, -1.0956e-01,\n",
       "          -3.9227e-01,  2.4031e-01, -9.8916e-02,  8.6010e-01,  2.3260e-01,\n",
       "           2.0943e-01, -4.3657e-03, -4.7262e-02,  4.8964e-01,  4.3099e-01,\n",
       "          -6.6541e-02, -1.6585e-01, -4.9522e-01,  4.6269e-03, -4.1279e-02,\n",
       "          -1.2963e-02, -4.5611e-01,  1.3753e-01,  4.2841e-01, -1.2416e-01,\n",
       "          -1.8354e-01,  1.5525e-01, -4.7715e-02, -3.1517e-01, -2.6492e-02,\n",
       "           6.8453e-01,  1.9080e-01, -2.2994e-01,  1.2377e-01, -1.1896e-01,\n",
       "           3.7006e-02, -4.6975e-01,  7.0044e-01,  6.1147e-01,  1.5585e-01,\n",
       "           1.6214e-02, -6.9943e-01,  2.3699e-01, -3.1574e-01, -4.7647e-03,\n",
       "          -1.5027e-01, -1.5731e-01, -4.2986e-01, -5.7285e-02, -5.5854e-02,\n",
       "           4.0115e-01,  7.9943e-02,  4.6653e-01,  1.1837e+00,  2.7973e-01,\n",
       "           1.9184e-02,  5.6097e-01,  2.0801e-01,  4.2492e-01, -4.0180e-01,\n",
       "           3.3082e-01,  8.8825e-02, -6.3896e-01,  2.7188e-01, -4.1593e-01,\n",
       "           4.4963e-02, -4.1432e-02, -3.7926e-01,  1.4208e-01, -2.1276e-02,\n",
       "           3.4237e-01,  3.3517e-01,  1.5186e-01,  1.6838e-01,  2.9411e-01,\n",
       "          -3.1322e-01,  5.9715e-01, -6.0809e-02, -1.1127e-01, -6.3492e-01,\n",
       "          -7.6732e-01,  9.5206e-02,  4.4743e-01, -1.2880e-01, -6.1279e-01,\n",
       "           1.0462e-01, -2.3866e-01, -1.2257e-01, -3.5064e-02,  2.1191e-01,\n",
       "           2.7455e-01,  5.0687e-02, -9.5436e-02,  6.4791e-02, -4.6950e-01,\n",
       "          -4.9277e-01,  4.9140e-02, -1.5758e-01, -1.3063e-01,  5.3918e-01,\n",
       "          -7.9415e-02,  1.1575e-01,  5.7221e-01,  1.9799e-02, -4.5976e-01,\n",
       "          -2.5640e-01,  1.4802e-01,  2.8667e-01, -7.2082e-02,  2.1981e-01,\n",
       "           3.4119e-01,  2.6467e-02,  5.1649e-01,  1.4823e-01, -2.1046e-01,\n",
       "          -2.7239e-01,  5.1173e-01, -1.5755e-01,  6.1403e-02, -1.2527e-01,\n",
       "          -8.7207e-01, -5.6150e-02, -7.8413e-02, -1.6833e-01, -5.4440e-01,\n",
       "           2.6566e-01,  1.4351e-01,  1.0561e-01,  2.2350e-01, -1.5984e-01,\n",
       "           7.0123e-01,  6.5285e-01, -3.3100e-01, -1.6943e-01, -4.0104e-01,\n",
       "           9.3922e-02, -6.8374e-02, -2.9329e-01, -1.4009e-02,  9.5368e-03,\n",
       "          -2.2765e-01, -1.2516e-01, -1.0805e-01,  2.8921e-01, -1.8876e-01,\n",
       "           5.2862e-01,  1.2189e-01,  8.3649e-02,  4.3873e-01,  2.0612e-02,\n",
       "           1.3071e-01,  3.6515e-01, -4.2117e-01,  1.2870e-01,  4.0424e-01,\n",
       "           4.0578e-01, -1.0551e+00, -8.3291e-01, -1.4936e-01,  3.6111e-01,\n",
       "           2.2860e-01,  2.5189e-01,  1.6999e-02, -7.2215e-03,  8.9455e-02,\n",
       "          -2.3402e-01,  3.5135e-01, -6.6415e-02, -5.7412e-01,  3.0561e-01,\n",
       "           2.3567e-01,  5.3263e-01, -2.7377e-01, -5.0579e-01, -6.5233e-02,\n",
       "           2.4616e-01, -3.6381e-01,  1.1392e-01,  2.3480e-01,  4.5466e-01,\n",
       "          -3.6015e-01, -5.6164e-01, -7.0282e-02, -7.3986e-01, -1.3578e-01,\n",
       "           3.2483e-01, -2.1437e-01,  2.7178e-02, -2.4758e-01,  1.5231e-01,\n",
       "          -1.7925e-02, -1.8879e-01,  6.0757e-01,  2.0522e-01, -2.5855e-01,\n",
       "           1.1981e-01, -2.3892e-01, -1.9475e-01,  5.0091e-01, -5.0226e-01,\n",
       "          -3.0348e-01, -5.0960e-02, -1.5349e-01, -3.2418e-01,  3.2181e-01,\n",
       "          -4.1389e-01, -1.4861e-01,  1.6748e-01, -6.4736e-02, -1.3068e-01,\n",
       "          -3.5233e-01,  1.4268e-01, -1.2707e-01, -4.9883e-02, -1.7654e-02,\n",
       "          -1.3353e-01,  6.0900e-02,  1.3308e-01,  6.4538e-02,  1.4388e-01,\n",
       "           5.3696e-01, -1.1464e-01,  2.7582e-03, -2.5900e-01, -2.2469e-01,\n",
       "           5.2498e-01, -9.4973e-02, -3.3117e-01,  5.4928e-01, -1.1156e+01,\n",
       "           1.0386e-01,  7.2631e-01, -5.5808e-01,  5.5128e-02, -2.8539e-01,\n",
       "           2.1085e-01, -3.0009e-01,  7.9101e-02,  2.4967e-01, -2.8460e-01,\n",
       "          -1.2277e-01, -2.7364e-01, -2.3193e-01, -4.0872e-01, -3.9742e-01,\n",
       "          -4.7229e-01, -5.6789e-01,  1.1772e-01, -1.7663e-01, -1.0996e-01,\n",
       "          -5.7036e-01, -1.9379e-01,  2.9696e-01, -4.9559e-03, -5.1069e-01,\n",
       "           5.9541e-02, -2.2948e-01,  2.6399e-01,  4.6126e-02,  2.2045e-01,\n",
       "          -1.9949e-02,  1.4108e-01,  2.0669e-01, -4.4946e-02, -4.5100e-01,\n",
       "          -4.4853e-01,  2.2129e-01, -1.5955e-01, -2.2267e-01, -5.4017e-02,\n",
       "          -3.3838e-01, -2.7689e-01,  1.3447e-01,  3.7594e-01,  6.2596e-01,\n",
       "           1.1595e+00,  5.6208e-01,  2.9245e-01, -1.0625e-01, -8.5438e-02,\n",
       "          -1.0762e+00,  5.5900e-01, -3.7318e-01,  8.1787e-02, -2.5119e-01,\n",
       "           6.6201e-01, -4.9906e-01,  5.4685e-01, -4.0754e-01, -2.5500e-01,\n",
       "           3.7179e-01,  3.0194e-01,  7.1104e-03,  2.8239e-02, -2.7432e-01,\n",
       "           4.0257e-01, -2.6094e-01, -3.6915e-02, -3.7819e-01,  4.0081e-01,\n",
       "          -3.9472e-01,  5.0770e-01, -5.6534e-02,  1.2945e-01,  1.8567e-01,\n",
       "          -3.5001e-01,  1.6610e-01,  1.0337e+00,  1.7414e-01,  4.1071e-01,\n",
       "           1.7420e-01,  5.1160e-01,  5.3064e-02, -8.8958e-01,  1.3691e-01,\n",
       "           2.0502e-01,  2.0789e-01, -6.7096e-02,  5.7131e-01, -2.5517e-02,\n",
       "          -6.5181e-01, -2.3094e-01, -3.2169e-01,  4.3108e-01,  6.5352e-01,\n",
       "          -2.2597e-01, -4.4343e-01,  2.3894e-01, -7.9776e-03, -5.3395e-01,\n",
       "           9.0074e-02, -2.2320e-01, -5.2792e-01,  8.7275e-02, -5.8914e-01,\n",
       "           1.4985e-01, -1.2428e-01,  5.4238e-02,  6.6887e-01, -4.0159e-01,\n",
       "          -1.6113e-01,  3.6602e-02, -4.4532e-02, -4.1270e-01, -4.9520e-01,\n",
       "          -7.7083e-02,  4.1073e-01, -1.9776e-01, -2.7976e-01,  7.0636e-02,\n",
       "          -3.0673e-01,  2.2584e-01, -1.7377e-01, -6.4935e-01,  3.2879e-01,\n",
       "          -6.0415e-01,  2.0578e-01,  2.5360e-01, -2.2507e-01,  4.9017e-01,\n",
       "           1.1337e-01, -8.1817e-02,  1.9397e-01, -9.9369e-02, -7.8554e-01,\n",
       "          -3.2573e-01, -2.7299e-02,  3.2662e-01, -4.2844e-01,  1.5776e-01,\n",
       "           1.5022e-01, -4.5752e-02, -2.2574e-01,  2.7904e-01,  1.0174e-01,\n",
       "           3.7762e-01,  3.9846e-01, -5.5006e-01,  1.1727e-01, -2.9867e-01,\n",
       "          -2.9262e-02, -6.1174e-02,  1.7952e-01,  2.3654e-01, -7.0764e-01,\n",
       "          -1.5224e-01, -2.6434e-01, -3.0965e-01, -2.1650e-01, -2.5120e-01,\n",
       "           2.0680e-01,  1.4038e-01, -1.5897e-02, -1.1308e-01,  5.8077e-01,\n",
       "           4.0814e-02, -1.1167e-01, -4.5997e-01,  3.1443e-01,  4.3178e-01,\n",
       "           2.1794e-01,  7.5875e-02, -1.2830e-01, -5.5532e-01, -1.1922e-02,\n",
       "          -4.3644e-01,  9.1081e-01, -7.3906e-01, -7.1970e-01,  6.6086e-02,\n",
       "           1.4951e-02, -8.8259e-04, -2.3450e-01, -5.0265e-01,  8.1298e-02,\n",
       "          -2.1017e-01, -1.4891e-01,  6.1550e-03, -1.0607e-02,  1.6299e-01,\n",
       "           4.0742e-01,  3.6838e-01,  1.5951e-01, -3.3938e-01, -4.4081e-01,\n",
       "          -6.0244e-01,  1.4567e-02, -1.6795e-01,  2.9330e-01, -2.6752e-01,\n",
       "          -3.2661e-01,  1.6043e-02, -9.1892e-03, -2.2272e-01,  5.2876e-01,\n",
       "           1.2427e-01,  1.6396e+00, -3.7287e-01, -4.4551e-01,  2.9018e-01,\n",
       "           7.2306e-02, -2.7073e-02, -2.8130e-01,  2.5074e-01,  9.5447e-02,\n",
       "           4.0194e-02, -3.6304e-01,  1.5023e-02, -2.0558e-02,  3.2773e-01,\n",
       "           3.2056e-01, -1.0703e+00, -2.0238e-01, -1.7993e-01,  5.5664e-01,\n",
       "           3.2388e-02,  6.3185e-01, -2.2305e-01,  3.0577e-01,  4.6106e-01,\n",
       "          -8.6667e-01, -2.3000e-01, -1.6129e-01,  2.7295e-01, -1.0583e-01,\n",
       "          -9.8735e-02, -1.6178e-01, -3.7165e-02, -1.6936e-01,  3.9538e-01,\n",
       "           7.4751e-01, -7.8426e-01,  7.4081e-01,  7.0467e-01, -9.8234e-02,\n",
       "           5.6815e-01,  9.4187e-02,  3.2641e-01, -1.4659e-01,  2.2669e-02,\n",
       "           2.6200e-01,  1.8542e-01, -6.3165e-01, -2.3962e-01,  2.4473e-01,\n",
       "           4.0158e-01, -1.1943e-01,  1.2108e-01,  7.9292e-02,  3.0960e-02,\n",
       "          -3.7691e-01, -1.0145e-01, -2.4552e-01, -3.9904e-01,  1.1356e-01,\n",
       "           1.4603e-02, -1.5001e-01, -8.6997e-02,  4.7709e-01, -1.7580e-01,\n",
       "           7.5067e-01,  4.2910e-01,  3.7145e-01, -3.5103e-01,  8.9950e-04,\n",
       "          -5.7565e-02,  6.0671e-01, -3.1670e-01,  7.7385e-01,  6.3337e-02,\n",
       "          -2.3140e-01,  2.7251e-01,  6.0547e-03,  1.7829e-01,  2.9892e-01,\n",
       "          -2.3519e-01,  1.2702e-01, -3.0718e-01, -2.3711e-01,  2.8330e-01,\n",
       "          -1.8640e-01,  1.1542e-01, -7.2270e-01, -6.3602e-03,  3.9857e-01,\n",
       "          -3.2051e-01,  4.9873e-01, -2.3074e-01,  6.1026e-01, -4.3509e-01,\n",
       "           3.2669e-02, -4.7470e-01, -4.3275e-01, -2.8508e-01,  5.1854e-02,\n",
       "          -8.6027e-01, -1.9453e-01,  1.7684e-01,  3.7405e-02,  2.1707e-01,\n",
       "           7.5155e-01, -3.3794e-01, -2.4771e-01, -2.0386e-01,  4.0190e-02,\n",
       "           1.8299e-01, -8.2071e-02, -3.5586e-01,  2.1161e-01, -3.5607e-01,\n",
       "          -1.0520e+00, -2.3117e-01,  8.9990e-02,  3.8419e-01, -3.5853e-01,\n",
       "           1.1570e+00,  2.5132e-02, -2.0226e-01,  6.7253e-01,  2.0808e-01,\n",
       "           6.4157e-01,  6.0323e-01,  1.3628e-03, -4.1259e-01, -5.4109e-02,\n",
       "          -1.5877e-01, -3.2881e-01,  1.5757e-01, -1.8346e-01,  5.0556e-01,\n",
       "           5.4802e-01, -2.0806e-02,  5.4756e-01]], device='mps:0'),\n",
       " tensor([[-0.1989, -0.7488, -0.5274,  ...,  0.1034,  0.2036,  0.4835],\n",
       "         [-0.5840, -0.3428, -0.3724,  ...,  0.1756,  0.7563,  0.3118],\n",
       "         [-0.2519, -0.4005, -0.5801,  ...,  0.1298,  0.4928,  0.6750]],\n",
       "        device='mps:0'),\n",
       " tensor([[-0.2744, -0.4852, -0.4880,  ...,  0.2064,  0.2977,  0.6606],\n",
       "         [ 0.0506, -0.6230, -0.6446,  ..., -0.0875,  0.4018,  0.3101],\n",
       "         [-0.3115, -0.6359, -0.3479,  ..., -0.0294,  0.8437,  0.2953]],\n",
       "        device='mps:0'),\n",
       " tensor([[-0.4654, -0.4115, -0.4370,  ...,  0.1625,  0.4338,  0.5908],\n",
       "         [-0.0367, -0.8495, -0.5948,  ...,  0.3991,  0.4067,  0.6434]],\n",
       "        device='mps:0'),\n",
       " tensor([[-0.1932, -0.4948, -0.6750,  ...,  0.2888,  0.2386,  0.7024],\n",
       "         [-0.0583, -0.6757, -0.8037,  ...,  0.1694,  0.7852,  0.5016]],\n",
       "        device='mps:0'),\n",
       " tensor([[-0.3104, -0.5652, -0.2977,  ..., -0.1645,  0.5554,  0.4119],\n",
       "         [-0.4580, -0.0772, -0.3209,  ...,  0.2665,  0.7906,  0.1587],\n",
       "         [-0.2158, -0.6403, -0.4623,  ...,  0.2045,  0.8277,  0.5596]],\n",
       "        device='mps:0'),\n",
       " tensor([[-0.0986, -0.6295, -0.5474,  ...,  0.1507,  0.8416,  0.6531],\n",
       "         [ 0.1498, -0.3364, -0.5901,  ...,  0.2182,  1.0799,  0.6295],\n",
       "         [-0.0278, -0.7575, -0.5307,  ...,  0.0801,  0.9419,  0.7043],\n",
       "         [-0.1266, -0.7452, -0.7321,  ...,  0.0133,  0.7699,  0.1507]],\n",
       "        device='mps:0'),\n",
       " tensor([[-2.1449e-01, -7.0454e-01, -6.7822e-01, -2.9824e-01, -2.3922e-01,\n",
       "          -8.0936e-02, -5.2339e-01,  7.3059e-01,  9.8530e-02,  5.0574e-01,\n",
       "           3.2821e-01,  2.0593e-01, -3.1108e-02,  3.5401e-02, -2.2698e-01,\n",
       "          -3.2096e-01,  2.1416e-01,  9.3440e-02,  8.6833e-02,  1.9313e-01,\n",
       "           6.3786e-01,  2.9887e-01, -2.8171e-01,  1.8532e-01, -6.6412e-01,\n",
       "           1.9387e-01, -3.1794e-01,  4.4021e-01,  3.4666e-01, -2.8586e-01,\n",
       "          -2.2759e-01,  5.0949e-01,  4.6054e-02,  7.5082e-01, -7.3993e-02,\n",
       "          -2.8442e-01,  4.0751e-01, -2.0799e-01,  1.4701e-01, -8.1190e-02,\n",
       "           5.0933e-01, -2.8916e-01,  1.9562e-01,  2.1962e-01,  2.0695e-01,\n",
       "           1.0814e-01,  2.3933e-01, -8.8199e-01,  1.6113e-01, -5.3111e-01,\n",
       "          -1.9534e-01, -1.3989e-01,  1.0564e-01,  4.0681e-01,  4.2414e-02,\n",
       "           7.0881e-02, -2.0309e-02, -2.2723e-03, -4.3205e-02,  1.2105e-01,\n",
       "           8.4445e-02,  6.4572e-01,  8.3931e-02, -1.9807e-01, -4.3443e-02,\n",
       "           4.2557e-01, -4.2430e-01, -4.1475e-01, -3.6487e-01, -9.9862e-02,\n",
       "          -1.3210e-01,  6.3424e-02,  2.7433e-02, -2.7986e-01,  3.0103e-01,\n",
       "          -1.0327e-01,  3.7408e-01,  1.1932e-01, -5.8556e-01,  5.9919e-02,\n",
       "           4.2993e-01,  4.3271e-01,  1.0163e-01, -5.6034e-02, -3.6994e-01,\n",
       "           1.3855e-01,  3.4047e-01,  2.0951e-01,  3.4301e-01,  4.8451e-02,\n",
       "           5.0535e-01,  4.4725e-02, -1.7061e-01, -3.7689e-01,  2.0207e-01,\n",
       "           4.4686e-02,  1.4184e-01, -2.7360e-01, -6.5874e-02, -2.6138e-01,\n",
       "           9.7153e-02,  2.5213e-01,  1.9122e-01,  5.5304e-03, -3.7332e-01,\n",
       "          -2.9791e-01,  2.7761e-02, -1.4923e-01, -5.0171e-01, -1.2071e-01,\n",
       "          -2.3499e-01,  1.3511e-01,  8.4198e-02, -2.3793e-01, -2.5759e-01,\n",
       "           2.7861e-01,  2.3122e-02, -6.1644e-02, -2.4986e-01,  3.5860e-01,\n",
       "          -7.1401e-03, -4.2117e-01,  1.6404e-01,  2.4549e-01,  6.3103e-01,\n",
       "          -1.0210e-01,  4.5199e-01, -2.7899e-01,  5.3807e-01,  2.6549e-01,\n",
       "          -1.3887e-01,  4.5990e-02, -2.7574e-01, -2.2970e-01,  5.9324e-03,\n",
       "          -3.4887e-02,  3.5351e-01, -2.2093e-01, -5.8900e-02, -2.2873e-01,\n",
       "          -4.1851e-02, -1.4381e-01, -3.3947e-01, -2.0282e-01,  2.1735e-02,\n",
       "          -1.7352e-01,  2.8433e-03,  2.4490e-02, -6.1198e-01, -3.4061e-01,\n",
       "          -8.1099e-01, -6.7827e-01,  2.4607e-01,  2.1824e-02, -1.8352e-01,\n",
       "          -2.4343e-02,  3.4273e-01,  3.2692e-01, -3.9045e-03,  1.5816e-01,\n",
       "           6.1919e-01, -5.5924e-02,  2.4366e-01, -6.5111e-01,  1.2130e-02,\n",
       "           1.3989e-01,  1.3852e-01,  1.0334e-02,  3.0207e-01,  2.7192e-01,\n",
       "           2.7881e-01, -4.8082e-01, -3.3521e-03, -5.9475e-01,  7.7579e-02,\n",
       "          -2.2848e-01,  4.0705e-01,  2.7719e-01, -3.6614e-01, -7.6351e-02,\n",
       "          -4.6902e-01, -2.3053e+00,  2.8320e-01, -6.4867e-01,  1.5134e-01,\n",
       "          -3.5205e-01,  3.3376e-01,  4.7742e-02,  2.9126e-01, -7.4618e-02,\n",
       "          -4.4197e-01, -2.0643e-01,  7.1511e-02,  2.5255e-01,  2.1678e-02,\n",
       "           3.5479e-01,  3.9292e-01,  1.1650e-01, -1.3211e-01,  9.5427e-02,\n",
       "          -3.9085e-01,  2.6911e-01, -1.5097e-01,  1.8862e-01,  3.2429e-01,\n",
       "           3.3884e-01, -2.6113e-01,  3.0993e-01,  5.3462e-01, -1.4194e-01,\n",
       "          -2.1112e-01, -2.1748e-01, -3.6181e-01, -1.1906e-01, -9.8641e-02,\n",
       "           6.1779e-01, -4.4257e-01,  1.1045e-01,  3.8594e-01,  2.7874e-01,\n",
       "          -1.8429e-01,  7.0850e-02, -7.9206e-02, -2.9913e-01, -7.1582e-03,\n",
       "           5.3529e-01, -2.5412e-01, -7.5643e-02, -4.0178e-01, -2.5430e-01,\n",
       "           1.0710e-01, -3.6585e-01,  5.4040e-01,  1.3650e-01, -2.7751e-01,\n",
       "          -3.0440e-02, -4.6726e-01,  2.6025e-02, -2.2961e-01,  2.7549e-02,\n",
       "          -3.4250e-01, -4.4381e-01, -6.9098e-01,  1.3356e-01,  5.8389e-02,\n",
       "           1.3915e-02, -2.6186e-01,  7.1536e-01,  6.1612e-01,  4.7437e-01,\n",
       "          -1.8139e-01,  1.6190e-02,  3.5486e-01,  1.8585e-01, -1.1704e-01,\n",
       "          -1.2458e-01, -1.2366e-01, -2.8551e-01,  4.3630e-01, -4.1739e-01,\n",
       "           1.3509e-01, -1.2989e-01, -3.9937e-02,  3.4343e-01, -4.1402e-01,\n",
       "           6.1919e-01,  1.1418e-01, -1.6619e-01, -2.3550e-01,  3.4553e-01,\n",
       "          -4.2821e-01,  3.4114e-01,  1.4150e-01, -4.2692e-01, -6.0903e-01,\n",
       "          -4.9354e-01,  3.9425e-01, -1.6426e-01,  1.1292e-01, -8.1100e-01,\n",
       "          -2.2976e-01,  3.3729e-01,  4.4767e-02,  3.4668e-01, -7.7773e-03,\n",
       "           2.5457e-01,  1.6300e-01,  7.4123e-02,  7.5676e-02, -5.9785e-01,\n",
       "          -2.9940e-02,  4.0458e-01, -3.4230e-01,  3.1484e-01,  3.2909e-01,\n",
       "           1.7037e-02,  1.3885e-01,  2.1392e-01, -3.3108e-02, -2.2602e-01,\n",
       "          -3.2048e-01,  2.2266e-01, -1.2187e-01, -1.5287e-01,  3.3797e-02,\n",
       "           1.1218e-01,  5.4054e-02,  5.2593e-01, -1.3887e-01, -3.5497e-01,\n",
       "          -1.4601e-02,  7.2089e-01, -1.4272e-01,  1.2341e-01, -7.0429e-02,\n",
       "          -1.3876e+00, -2.2261e-01, -2.0181e-01, -9.7845e-02, -4.2230e-01,\n",
       "           3.3472e-01, -2.1790e-03, -7.2537e-02, -5.0113e-02, -1.4296e-01,\n",
       "           8.4535e-02,  5.6148e-01,  3.1051e-01,  1.9252e-01,  1.4375e-01,\n",
       "          -1.8416e-01,  6.2568e-02, -3.0960e-01, -2.0726e-01,  1.7834e-02,\n",
       "           1.8351e-01, -7.7169e-02, -3.7460e-02,  1.1015e-02, -2.1886e-01,\n",
       "           9.6019e-01, -3.0926e-01,  3.8861e-02,  1.6496e-01, -3.5839e-01,\n",
       "          -3.3962e-01,  2.4846e-01, -1.9115e-01,  1.4696e-01,  9.0296e-03,\n",
       "          -1.4068e-02, -5.2225e-01, -3.3624e-01, -5.9971e-03,  6.5100e-01,\n",
       "           1.4027e-01,  4.8045e-01, -3.5575e-01,  2.5440e-01,  5.6965e-01,\n",
       "          -4.3151e-01,  1.4340e-01, -1.6650e-01, -1.5876e-01,  2.4645e-01,\n",
       "           1.4621e-01,  2.0576e-01, -2.3414e-01,  1.3722e-01,  1.8341e-02,\n",
       "           6.0046e-02, -2.3356e-01, -1.4233e-01,  1.5209e-01, -2.8139e-02,\n",
       "           9.4217e-02, -4.5048e-01,  9.6335e-02, -5.1419e-01, -5.1183e-01,\n",
       "          -1.8274e-02,  3.1351e-03, -8.5949e-03, -2.2676e-01, -5.1958e-01,\n",
       "          -4.8065e-01,  6.8147e-02,  6.5632e-01,  1.4340e-01,  1.8086e-01,\n",
       "           3.7948e-01,  6.8706e-02, -4.0554e-01,  3.1719e-01,  9.8486e-02,\n",
       "          -4.3327e-01,  4.1227e-01, -6.7402e-01,  3.9696e-01,  3.5643e-01,\n",
       "          -2.2426e-01, -6.0792e-02, -1.3265e-01,  4.5098e-01,  5.2386e-02,\n",
       "          -1.8944e-01,  3.0117e-01,  3.2255e-01,  1.2599e-01, -5.5095e-02,\n",
       "           5.8891e-03,  6.9732e-01, -4.1903e-01,  1.2872e-01, -2.3904e-01,\n",
       "           2.3916e-01, -1.9664e-01, -7.5440e-02, -1.5370e-01, -3.4435e-01,\n",
       "           2.1290e-01,  2.5904e-01,  1.5371e-01,  4.1539e-02, -1.1574e+01,\n",
       "           1.5778e-01,  6.4611e-01, -5.4654e-01, -2.7999e-02, -8.0666e-02,\n",
       "          -7.6214e-02, -2.5721e-01,  5.6782e-03, -3.6491e-01, -2.2660e-01,\n",
       "          -3.4259e-02,  9.2369e-02,  7.1069e-02, -3.6139e-01, -1.1223e-01,\n",
       "          -3.4103e-01, -8.1940e-01, -4.4931e-02, -1.3049e-01,  5.0144e-01,\n",
       "          -3.9520e-01,  2.1677e-01,  1.2436e-01,  1.1785e-02, -5.1300e-01,\n",
       "          -2.1669e-01, -4.2068e-01,  3.3411e-01, -3.3249e-02, -5.4580e-02,\n",
       "           4.6023e-02, -9.7767e-02,  1.4915e-01, -1.6366e-01, -4.5533e-01,\n",
       "           1.4732e-02,  3.2966e-01,  3.8739e-01, -2.4536e-01, -2.3814e-01,\n",
       "           1.3612e-01, -4.4475e-01,  5.6290e-01,  1.9171e-02,  6.7223e-01,\n",
       "           9.8843e-01,  8.5313e-01, -1.0504e-01,  7.1469e-02,  7.8757e-02,\n",
       "          -9.0512e-01,  3.9596e-01, -9.1779e-01,  1.0809e-01,  8.7403e-02,\n",
       "           6.8621e-01, -4.8684e-01, -2.1797e-01,  1.2465e-01, -3.3176e-01,\n",
       "           1.7581e-01,  4.7997e-03,  2.9667e-01, -1.4982e-01, -8.7584e-02,\n",
       "           4.8404e-01, -1.1318e-01,  2.5377e-01, -1.3684e-01, -4.1049e-01,\n",
       "          -5.2916e-02,  4.5801e-01,  1.4162e-01, -2.5695e-01,  2.2955e-01,\n",
       "          -5.5650e-02,  1.7324e-01,  5.9556e-01, -9.7477e-02, -1.5338e-01,\n",
       "           2.1133e-01,  3.7742e-01, -4.2344e-01, -6.2781e-01, -1.7225e-01,\n",
       "           1.9234e-02,  2.4492e-01,  9.7068e-02,  5.2181e-01, -2.9523e-01,\n",
       "          -3.4028e-01, -3.0628e-01, -3.9049e-03,  7.5488e-02,  7.9833e-01,\n",
       "          -3.5370e-01, -2.3046e-01, -1.6219e-01, -1.5390e-01, -5.4544e-01,\n",
       "          -3.2343e-01, -2.0785e-01, -5.7979e-01,  3.4363e-01, -1.7063e-01,\n",
       "          -2.1419e-01, -9.3807e-02,  1.9702e-01,  2.7729e-01, -3.0160e-01,\n",
       "           2.4914e-01, -7.0639e-02,  1.6772e-01, -3.6538e-01, -9.6589e-02,\n",
       "          -5.8679e-02,  8.8670e-01, -2.8509e-01, -4.5203e-01, -2.8266e-01,\n",
       "          -1.3650e-01,  3.4266e-01, -1.3321e-01, -3.6893e-01, -5.9663e-02,\n",
       "          -5.3281e-01, -1.0096e-01, -3.4001e-01,  1.2398e-01,  4.2333e-01,\n",
       "           3.4099e-01,  1.9882e-01,  4.1674e-01, -2.9492e-01, -5.9256e-01,\n",
       "          -6.1071e-01, -2.3221e-01,  1.7918e-01, -4.1660e-01,  2.3933e-02,\n",
       "          -1.2662e-01,  1.1810e-01, -4.8288e-01,  3.2500e-01, -3.4306e-01,\n",
       "           5.3225e-01,  1.4901e-01, -6.5404e-02,  9.4824e-02, -1.5454e-01,\n",
       "          -3.0315e-01, -3.1135e-01, -2.5577e-02, -2.1383e-04, -2.9308e-01,\n",
       "          -3.5529e-01,  3.4658e-01, -3.2742e-01, -1.9652e-01, -9.1171e-02,\n",
       "           5.8883e-03, -3.1505e-02, -3.3703e-01,  2.2127e-01,  1.5611e-01,\n",
       "           5.1888e-02, -2.8746e-01, -4.9281e-01,  6.7151e-02,  1.2024e-01,\n",
       "           1.7797e-01, -1.4067e-01, -1.7871e-01, -7.1177e-01, -2.1203e-02,\n",
       "           2.2721e-01,  1.2209e+00, -3.0134e-01, -1.5559e-01,  5.0491e-01,\n",
       "           3.8731e-01, -2.5203e-01, -7.9592e-02, -1.7318e-01,  1.6439e-01,\n",
       "           2.6809e-02, -5.3355e-01,  4.3414e-01, -3.6624e-01,  4.3651e-01,\n",
       "           2.0912e-01,  7.4561e-02, -2.5968e-01,  1.7785e-01,  1.3109e-01,\n",
       "          -4.3428e-01,  2.3924e-02, -3.1788e-01, -4.9670e-01,  1.8552e-01,\n",
       "          -1.7094e-01, -4.7952e-02, -7.0504e-02,  1.1502e-01,  4.8674e-01,\n",
       "           1.7448e-01,  1.4508e+00,  9.3435e-02,  1.1494e-01,  9.5490e-02,\n",
       "           1.5215e-01, -3.8269e-01,  2.0135e-02, -1.2861e-01,  4.0301e-01,\n",
       "           2.4194e-01,  1.6791e-01, -4.5340e-01,  1.6435e-01, -1.2027e-01,\n",
       "           3.3342e-01, -7.7834e-01, -6.2246e-01,  1.1594e-02,  6.2443e-01,\n",
       "           7.3555e-02,  1.8331e-01, -1.9643e-01,  6.3311e-01,  6.6714e-01,\n",
       "          -1.0752e+00,  3.6182e-02, -2.6472e-01,  1.9773e-01, -1.2163e-01,\n",
       "          -6.5773e-01, -2.4769e-01,  9.3695e-02, -4.6609e-01,  6.5418e-01,\n",
       "           6.5864e-01, -7.9031e-01,  6.3784e-02,  4.1052e-01, -2.2276e-01,\n",
       "          -4.2281e-02,  8.0290e-05,  1.7308e-01,  3.7529e-01, -9.3648e-02,\n",
       "          -2.7696e-01,  2.7165e-01, -3.1393e-01, -5.1805e-01,  2.9079e-01,\n",
       "           5.9012e-01, -9.5003e-02,  9.2033e-02,  1.9321e-01, -2.9840e-01,\n",
       "          -1.3768e-01, -3.9346e-02, -1.5622e-01,  1.6004e-01,  1.3544e-01,\n",
       "           2.4736e-02, -3.6580e-01,  1.4480e-01,  2.5389e-01, -8.7195e-02,\n",
       "           4.8912e-01,  8.0434e-01,  3.5971e-01, -2.4140e-01,  2.0208e-03,\n",
       "           4.2381e-02,  2.1808e-01, -4.1064e-02,  6.6852e-01, -3.0423e-01,\n",
       "          -3.2984e-01,  3.0238e-01,  1.6299e-01, -3.3704e-01,  1.9782e-01,\n",
       "          -2.3993e-01,  1.9140e-01,  7.6162e-02, -2.2715e-01,  3.9988e-01,\n",
       "           9.3421e-02,  5.2397e-02, -4.4329e-01,  2.4024e-01,  3.1145e-01,\n",
       "          -3.6736e-01,  7.6739e-01,  1.6882e-01,  4.0088e-01, -2.9505e-01,\n",
       "           5.1547e-02, -5.8944e-01, -4.2632e-01,  2.9509e-02,  3.8038e-01,\n",
       "          -3.8722e-01, -1.1970e-01,  4.4778e-01, -2.8456e-01,  6.0988e-01,\n",
       "           5.1692e-01, -4.1234e-01, -5.9516e-01,  1.4129e-01,  6.1326e-03,\n",
       "           4.3634e-02, -3.0439e-01,  1.5096e-01,  6.9082e-02,  3.4227e-02,\n",
       "          -8.0103e-01, -1.8143e-01,  1.2405e-01,  2.3414e-01, -2.6937e-01,\n",
       "           6.4494e-01, -6.1737e-02,  2.1208e-02,  2.6466e-01,  1.8926e-01,\n",
       "           1.3478e-01,  5.7465e-01,  1.9131e-01, -5.9506e-01,  8.7627e-02,\n",
       "           2.3120e-01, -2.3044e-01, -2.2299e-01, -4.1241e-01,  3.3408e-01,\n",
       "           1.9442e-01,  6.8585e-01,  6.6179e-01]], device='mps:0'),\n",
       " tensor([[-0.4654, -0.4115, -0.4370,  ...,  0.1625,  0.4338,  0.5908],\n",
       "         [-0.0932, -0.6293, -0.4879,  ...,  0.3041, -0.1958,  0.5576],\n",
       "         [-0.3292, -0.2590, -0.5016,  ..., -0.0148,  0.3509,  0.1607]],\n",
       "        device='mps:0'),\n",
       " tensor([[ 2.0208e-02, -7.6503e-01, -5.5929e-01, -2.5692e-01, -5.6472e-01,\n",
       "           5.7194e-02, -2.9899e-01,  6.1910e-01,  5.2920e-03,  4.5897e-01,\n",
       "           4.0233e-01,  2.0171e-01,  1.7813e-02, -1.9294e-01, -5.8388e-01,\n",
       "          -3.3605e-01,  3.3281e-01,  4.2457e-01,  3.0132e-01, -6.8317e-02,\n",
       "           7.2122e-01,  4.3789e-01, -3.0936e-01, -1.1078e-02, -6.4128e-01,\n",
       "           3.0362e-01,  8.5775e-02,  1.2002e-01,  3.6188e-01, -4.3962e-01,\n",
       "          -4.3361e-01,  5.5624e-01, -4.7431e-02,  6.8908e-01, -3.6465e-01,\n",
       "          -1.6057e-01,  1.2906e-01, -2.5347e-01,  3.7705e-01,  1.3642e-01,\n",
       "           7.8618e-01, -1.3841e-01,  1.9413e-01,  3.8705e-01,  5.8566e-01,\n",
       "           5.5818e-02,  5.9252e-02, -9.8198e-01,  4.3724e-01, -4.2433e-01,\n",
       "          -8.7132e-02, -1.9352e-01,  2.9384e-01,  4.3466e-01,  4.1407e-02,\n",
       "          -3.8051e-01,  1.3777e-01,  2.5721e-01, -3.6372e-01,  1.6510e-01,\n",
       "          -7.3787e-02,  8.1903e-01, -1.2874e-01, -6.0702e-02, -5.3394e-02,\n",
       "           4.8926e-02, -1.0550e-01, -7.6464e-01, -3.0272e-01, -1.4867e-01,\n",
       "          -2.1655e-01,  1.6044e-01, -3.2349e-01, -1.7165e-01,  4.2270e-01,\n",
       "          -3.6935e-01,  1.4055e-01, -4.9128e-02, -4.9496e-01, -1.3233e-02,\n",
       "           4.0388e-01,  4.8051e-01,  3.8214e-02, -2.0929e-02, -4.3727e-01,\n",
       "          -6.0656e-02,  6.7806e-01,  4.1573e-01,  3.6109e-01, -6.7068e-03,\n",
       "           4.0578e-01,  1.5997e-01, -1.4478e-02, -3.7510e-01,  3.7012e-01,\n",
       "          -2.0713e-01,  3.9778e-01, -1.8583e-01,  6.5082e-02, -9.5916e-02,\n",
       "           7.1551e-02,  1.8106e-01,  1.3193e-01,  1.7613e-01, -5.0597e-01,\n",
       "          -6.4350e-01,  2.0210e-01, -4.2119e-02, -8.4742e-01, -8.1491e-02,\n",
       "          -2.3964e-01,  2.2045e-01,  3.4319e-01, -4.2882e-01, -1.0789e-01,\n",
       "           1.2355e-01,  3.2049e-01, -3.3026e-01, -5.3377e-01,  3.8314e-01,\n",
       "           2.0406e-01, -5.1428e-01,  1.4590e-01,  2.3334e-01,  3.6936e-01,\n",
       "          -5.1731e-02,  2.7681e-01, -3.6767e-01,  4.4541e-01,  4.9211e-01,\n",
       "          -2.3287e-03,  1.9927e-01, -2.3671e-01, -2.1461e-01, -2.8364e-02,\n",
       "          -1.9802e-02,  1.1374e-01, -1.8721e-01,  3.1578e-01,  7.9393e-02,\n",
       "          -1.7306e-01, -1.4675e-01, -4.4787e-01,  6.8730e-02, -8.9550e-02,\n",
       "          -7.4923e-03, -6.0417e-02,  1.3191e-01, -2.0985e-01, -3.4197e-01,\n",
       "          -8.3526e-01, -6.6114e-01,  2.9746e-01,  1.6620e-02, -2.6681e-02,\n",
       "          -8.1852e-03,  4.1030e-01,  1.2823e-01,  2.0095e-01,  4.6233e-01,\n",
       "           3.9786e-01, -1.6751e-01,  2.4818e-01, -5.1207e-01, -6.5876e-02,\n",
       "           3.7497e-01,  2.1087e-01, -2.8988e-01, -8.3998e-02,  2.4633e-01,\n",
       "           2.3304e-01, -4.3286e-01, -1.2071e-01, -7.9654e-01, -3.2950e-02,\n",
       "          -6.0529e-01,  2.8070e-01,  2.2814e-01, -4.3748e-01,  9.8900e-03,\n",
       "          -4.4120e-01, -2.0305e+00,  2.0706e-01, -2.1848e-01,  1.2624e-01,\n",
       "          -6.4365e-01,  3.0940e-01,  3.7443e-01,  3.4252e-01, -1.5505e-01,\n",
       "          -3.8439e-01, -5.8677e-03,  1.1463e-01,  6.6717e-01, -1.3443e-01,\n",
       "           3.3261e-01,  6.4918e-01,  1.0890e-02, -2.5317e-01, -7.8681e-03,\n",
       "          -3.3116e-01,  1.3954e-01, -1.1193e-01, -4.6732e-01,  1.5745e-02,\n",
       "           2.0312e-01,  1.5326e-01,  2.1860e-02,  4.0112e-01,  2.8193e-03,\n",
       "          -9.9428e-02, -3.1422e-01, -5.0077e-03, -4.3295e-01, -8.7508e-02,\n",
       "           3.7966e-01, -1.5067e-01, -1.0608e-01,  3.9688e-01,  1.2532e-01,\n",
       "          -1.3816e-01,  1.1189e-01, -1.4272e-01, -5.7000e-02, -1.6625e-01,\n",
       "           2.8259e-01, -1.1423e-01, -1.0816e-01, -2.9287e-01, -4.0375e-01,\n",
       "          -3.0225e-03, -2.9997e-01,  7.1459e-01, -5.2832e-02, -2.1241e-01,\n",
       "          -4.9738e-01, -2.7527e-01, -9.5959e-02, -4.1868e-02,  1.8233e-01,\n",
       "          -1.9273e-01, -4.2171e-01, -8.5361e-01, -8.9961e-02,  8.8740e-02,\n",
       "          -4.7861e-02, -2.6897e-01,  9.0623e-01,  1.7193e-01,  2.4860e-01,\n",
       "          -3.4133e-02,  3.6571e-02,  4.1085e-01,  2.5951e-02, -1.6019e-01,\n",
       "           1.0218e-01, -2.8330e-02, -8.7010e-02,  4.1328e-01, -2.0672e-01,\n",
       "           2.8802e-02, -1.2390e-01, -4.0113e-02,  5.2599e-01, -1.0086e-01,\n",
       "           2.7946e-01,  4.4659e-01, -4.4787e-01,  1.7640e-01,  3.1108e-01,\n",
       "          -2.8588e-01,  3.4071e-01,  2.1494e-02, -1.5333e-01, -7.8185e-01,\n",
       "          -2.5145e-01,  4.3187e-01, -1.5070e-01,  1.5915e-01, -7.8193e-01,\n",
       "          -2.8628e-02,  2.9829e-01, -9.5702e-02,  2.2307e-01, -3.3996e-01,\n",
       "           1.0336e-01,  2.7664e-01,  5.4592e-02, -1.8774e-01, -6.2797e-01,\n",
       "           2.4925e-02,  2.6298e-01, -3.6244e-01,  3.6625e-01, -2.3536e-02,\n",
       "          -2.7597e-01,  6.5678e-02,  3.4875e-01, -5.6993e-03, -3.0233e-01,\n",
       "          -1.8564e-01, -1.5399e-01, -4.3299e-01, -8.5505e-02, -1.9394e-01,\n",
       "           1.6030e-01, -1.8583e-02,  4.4168e-01, -7.7571e-01, -6.2004e-01,\n",
       "          -8.2860e-02,  6.8730e-01,  1.1407e-01, -2.6381e-01, -3.6458e-01,\n",
       "          -1.4420e+00, -2.3726e-01, -4.1471e-01, -1.6035e-01, -5.0801e-01,\n",
       "           2.4845e-01,  5.6050e-03, -7.0122e-02,  5.1329e-02, -1.9641e-01,\n",
       "           2.2804e-01,  4.2249e-01, -1.4574e-01, -5.9133e-02, -4.7953e-02,\n",
       "          -2.9291e-01,  5.4336e-02, -1.0803e-01, -2.3218e-01, -7.5940e-02,\n",
       "           8.5886e-02,  1.0303e-01, -2.5835e-01, -5.4286e-02, -3.2694e-01,\n",
       "           6.3005e-01, -9.8360e-02,  2.2665e-01,  3.2024e-02, -3.4603e-01,\n",
       "          -1.4111e-01,  5.2638e-01, -2.6100e-01, -9.5505e-02, -3.1647e-02,\n",
       "          -8.1588e-04, -3.7655e-01, -6.9738e-02, -5.0863e-02,  5.1015e-01,\n",
       "           3.5030e-01,  2.1533e-01, -3.5168e-01,  4.2025e-01,  6.7044e-01,\n",
       "          -1.8195e-01,  2.5602e-01,  2.4438e-02, -1.7030e-01,  8.4277e-03,\n",
       "           2.9253e-01,  1.8083e-01, -5.4752e-01,  9.4749e-02,  3.7275e-01,\n",
       "           2.9185e-01, -3.6818e-01, -6.6197e-02,  2.1315e-01,  2.7459e-01,\n",
       "          -3.4370e-01, -3.8059e-01,  3.9311e-01, -3.9890e-01, -3.4777e-01,\n",
       "           4.3958e-02,  9.1178e-02,  1.5579e-01,  7.0696e-02, -5.7060e-01,\n",
       "          -4.3794e-01,  2.4488e-01,  8.7743e-01,  1.2242e-01,  1.5704e-01,\n",
       "           3.1480e-01, -1.1315e-01, -2.4633e-01,  3.3256e-01, -1.6159e-01,\n",
       "          -7.7163e-02,  3.1051e-01, -4.6952e-01,  4.0089e-01,  3.4175e-01,\n",
       "          -1.6735e-01, -2.5179e-01, -1.0563e-01, -3.9040e-01, -1.1559e-01,\n",
       "          -2.1741e-01,  3.9607e-01,  4.9480e-02,  1.2715e-01, -8.5624e-02,\n",
       "           2.8878e-01,  1.4682e-01, -2.8650e-01,  8.1864e-02, -7.2852e-02,\n",
       "           1.0194e-01,  1.1781e-01, -8.8097e-02, -4.7460e-02, -5.2011e-02,\n",
       "           3.8932e-01,  1.0028e-01,  2.4817e-01,  2.0441e-01, -1.1597e+01,\n",
       "           6.8626e-02,  7.5693e-01, -4.7819e-01, -1.7697e-02,  4.0950e-02,\n",
       "           1.6838e-01, -2.1576e-01, -6.7861e-02, -2.9902e-01, -2.2121e-01,\n",
       "           1.6634e-02,  2.9037e-02,  1.5752e-01, -2.8156e-02, -9.0802e-02,\n",
       "          -1.4198e-01, -6.5549e-01,  5.0503e-01, -4.5816e-02,  1.5790e-01,\n",
       "          -2.2957e-01,  3.6382e-01, -5.2319e-02, -1.6315e-01, -1.3850e-01,\n",
       "          -3.3683e-01, -4.1339e-01,  2.7810e-01,  1.4482e-01, -1.1726e-01,\n",
       "          -3.9508e-02,  3.4742e-01,  1.4369e-01, -4.0219e-01, -5.6183e-01,\n",
       "           2.0400e-01, -1.8716e-02,  4.9096e-02,  4.0898e-02, -1.9458e-01,\n",
       "           1.6506e-01, -6.7758e-01,  3.8554e-01,  1.8458e-01,  7.8416e-01,\n",
       "           7.3742e-01,  7.0307e-01,  2.3267e-01, -4.3792e-02,  2.8482e-02,\n",
       "          -8.0305e-01,  1.6911e-01, -8.0406e-01,  1.9478e-01, -8.5476e-02,\n",
       "           4.9424e-01, -6.4510e-01, -6.0906e-02,  8.3130e-02, -3.2829e-01,\n",
       "           2.5847e-01, -7.8931e-02,  2.7519e-01, -2.7523e-01, -1.2579e-01,\n",
       "           3.5699e-01,  1.1202e-01,  2.9189e-01, -1.4989e-02,  1.9941e-01,\n",
       "          -2.1878e-01,  5.6106e-01,  2.6796e-01, -2.9849e-01,  1.1820e-01,\n",
       "           2.0249e-02,  7.7441e-02,  5.5225e-01, -8.4660e-02, -4.6368e-02,\n",
       "           7.2658e-02,  3.4458e-01, -2.6370e-01, -7.9845e-01, -8.8288e-02,\n",
       "           2.4457e-01, -7.1626e-02,  6.4396e-02,  5.0609e-01, -3.0357e-01,\n",
       "          -2.6946e-01, -4.2302e-01,  1.9476e-02,  1.8042e-01,  7.1453e-01,\n",
       "          -5.8034e-01, -1.0122e-01, -1.6808e-02, -8.1182e-02, -7.3097e-01,\n",
       "          -8.3426e-02, -1.9399e-01, -2.7188e-01,  2.1695e-01, -2.7433e-01,\n",
       "          -7.3079e-03,  4.6410e-02,  2.2299e-01,  3.8007e-01, -5.8152e-01,\n",
       "           5.3584e-01,  3.9769e-02,  1.8878e-01, -1.4119e-01,  1.2564e-01,\n",
       "           1.4194e-01,  5.4991e-01, -2.6273e-01, -5.8466e-01, -2.5381e-01,\n",
       "          -1.5310e-01,  3.0415e-01, -1.5312e-01, -3.1279e-01,  1.1880e-01,\n",
       "          -2.5092e-01, -3.4404e-03,  8.2193e-02,  7.3047e-02,  4.6305e-01,\n",
       "           8.4067e-02,  6.7405e-02,  6.4410e-02, -5.4696e-01, -8.2122e-01,\n",
       "          -5.6732e-01,  1.5895e-01,  1.4352e-01, -2.9913e-01, -1.5565e-01,\n",
       "          -8.1726e-02,  7.9281e-02, -4.6112e-01,  2.3830e-01,  3.3623e-02,\n",
       "           7.5235e-01,  4.5941e-01, -1.1610e-01, -1.2501e-01, -2.4152e-01,\n",
       "          -4.7697e-01, -2.4586e-01,  8.6539e-03,  3.4751e-02, -3.2841e-01,\n",
       "          -3.3981e-01,  2.1232e-01, -2.0264e-01,  9.3159e-02,  2.3010e-01,\n",
       "          -2.8576e-01, -3.4762e-01, -2.4335e-01, -1.8679e-02,  3.3081e-01,\n",
       "           1.3124e-01, -5.1549e-02, -3.5166e-01,  9.7943e-02,  3.3827e-01,\n",
       "           9.2638e-02, -2.7596e-01, -3.1185e-01, -1.7647e-01, -3.0932e-02,\n",
       "           2.4132e-01,  8.9103e-01, -2.2223e-01, -2.5462e-01,  2.4756e-01,\n",
       "           1.2559e-01, -1.6163e-01,  2.4328e-01, -2.1901e-01, -8.2403e-02,\n",
       "          -3.1274e-02, -6.0327e-01,  2.3566e-01,  1.2174e-02,  3.7033e-01,\n",
       "           1.0918e-02,  1.0173e-01,  4.5553e-02,  3.5172e-01, -6.4380e-02,\n",
       "          -1.3083e-01, -7.2966e-02, -5.9665e-01, -1.2796e-01, -1.2882e-02,\n",
       "           1.6433e-02,  1.0146e-02, -2.8245e-01, -2.4282e-01,  2.5145e-01,\n",
       "          -4.8975e-02,  1.0930e+00,  3.7531e-01,  4.5117e-02,  1.7778e-01,\n",
       "           1.8941e-01, -4.0156e-01,  2.1312e-01, -3.3692e-02,  2.1484e-01,\n",
       "           3.7423e-01,  2.5027e-01, -2.1613e-01,  8.1767e-02,  1.1973e-01,\n",
       "           3.6036e-01, -7.3182e-01, -6.4988e-01, -6.2273e-02,  4.1248e-01,\n",
       "          -8.2303e-02,  3.2196e-01,  2.8974e-02,  5.7252e-01,  6.7178e-01,\n",
       "          -1.1460e+00, -2.3901e-01, -2.7814e-01,  7.7240e-01, -2.6478e-02,\n",
       "          -4.1822e-01, -1.6920e-01, -5.3889e-01, -4.6541e-01,  3.1793e-01,\n",
       "           6.9987e-01, -1.0613e+00,  1.6571e-01,  1.2003e-01, -1.9229e-01,\n",
       "           5.4059e-02, -5.7781e-01,  3.3046e-01,  1.5885e-02,  2.1187e-01,\n",
       "          -9.0080e-02,  5.4108e-01, -4.1372e-01, -5.0692e-01,  1.5216e-01,\n",
       "           3.9384e-01,  4.3704e-02,  4.0912e-02,  2.8431e-02,  3.3223e-01,\n",
       "          -1.3847e-01, -1.8566e-01, -2.6675e-01,  3.5006e-01, -4.9763e-02,\n",
       "           2.1789e-01, -5.1372e-01, -7.2430e-02,  2.6108e-01, -3.8436e-02,\n",
       "           3.6857e-01,  6.1013e-01,  6.8394e-01, -9.4953e-02, -2.8380e-01,\n",
       "           4.8577e-01,  1.5567e-01, -7.3465e-02,  6.4399e-01, -1.2834e-01,\n",
       "          -4.3102e-01,  6.4595e-01,  2.9300e-02, -2.5687e-01,  4.9953e-02,\n",
       "          -1.7909e-01,  3.3106e-01,  2.3613e-01, -1.9502e-01,  9.3905e-02,\n",
       "           1.3858e-01,  1.2522e-01, -3.8871e-01,  8.9062e-02,  1.8340e-01,\n",
       "          -3.6279e-01,  7.6027e-01, -1.6448e-01,  5.0500e-01, -2.9734e-01,\n",
       "           1.8890e-01, -3.0684e-01, -4.4585e-01, -3.3397e-01, -2.8147e-01,\n",
       "          -2.8851e-01, -1.6761e-01,  1.7526e-01, -9.1573e-02,  7.4284e-01,\n",
       "           3.8549e-01, -4.0562e-01, -3.0646e-01, -2.1099e-01,  1.9009e-01,\n",
       "           1.6331e-01, -1.7484e-01,  7.9365e-02,  2.1902e-01,  1.8608e-01,\n",
       "          -9.0702e-01, -4.3824e-01,  3.0154e-01,  3.2570e-01, -3.4959e-01,\n",
       "           5.0241e-01, -9.3615e-02,  1.9549e-01,  1.3200e-01,  1.4179e-01,\n",
       "           1.4406e-01,  5.4799e-01,  2.3142e-01, -3.4907e-01,  9.9233e-02,\n",
       "           1.8399e-01, -4.6496e-01,  7.9736e-02, -5.1373e-01,  2.3767e-01,\n",
       "           2.7593e-01,  8.8341e-01,  5.3475e-01]], device='mps:0'),\n",
       " tensor([[-2.7514e-01, -6.1363e-01, -9.8980e-01,  ...,  1.6808e-01,\n",
       "           8.7505e-04,  4.9040e-01],\n",
       "         [-4.7962e-01, -5.7906e-01, -4.3584e-01,  ...,  1.9415e-01,\n",
       "           3.7597e-01,  1.5180e-01],\n",
       "         [-2.4524e-01, -7.1187e-01, -3.5790e-01,  ...,  1.2537e-01,\n",
       "           2.6219e-01,  1.0056e-01]], device='mps:0'),\n",
       " tensor([[-1.1097e-02, -5.0567e-01, -4.6699e-01, -6.4640e-01, -3.9556e-01,\n",
       "          -2.5266e-01, -5.5990e-01,  3.0801e-01, -1.2536e-01,  1.4189e-01,\n",
       "           2.9133e-01,  5.3879e-01, -3.6706e-01, -2.8859e-01, -4.8290e-01,\n",
       "          -5.3699e-02, -6.1741e-01, -1.6918e-02,  3.4690e-01,  3.0491e-01,\n",
       "           5.3718e-01,  1.8073e-01, -4.1129e-01,  2.9149e-01, -3.7959e-01,\n",
       "           6.6561e-02, -3.4985e-01, -3.1546e-01,  2.6210e-01,  2.8012e-01,\n",
       "           5.3618e-02,  7.7692e-01, -2.4307e-02,  1.0962e+00, -1.8413e-01,\n",
       "          -2.1778e-01,  1.5359e-01,  2.3320e-01,  8.6439e-01, -6.2204e-01,\n",
       "           5.6395e-01,  3.6489e-01,  2.7586e-02, -4.0063e-02,  2.7845e-01,\n",
       "           3.5451e-01,  1.4567e-01, -5.7913e-01, -2.7161e-01, -3.4037e-01,\n",
       "          -1.2979e-01, -1.5164e-01,  5.0058e-01, -2.3292e-01, -7.9546e-01,\n",
       "          -1.9253e-01,  1.1288e-02,  2.6982e-01,  1.1117e-01,  2.3591e-01,\n",
       "           2.8900e-02,  7.9425e-01, -1.9339e-01, -3.2966e-03,  2.0559e-01,\n",
       "           2.5502e-01, -7.6243e-01, -1.2873e-01, -7.2822e-01,  1.1548e-01,\n",
       "           2.1413e-01,  3.5162e-01,  1.0500e-01, -2.6816e-01, -1.9644e-01,\n",
       "           6.0243e-02,  1.6956e-01,  1.8610e-01, -2.0035e-01,  3.5967e-02,\n",
       "           2.1140e-01,  4.4140e-01,  4.5980e-01,  5.9922e-02,  1.6117e-01,\n",
       "           5.6960e-02,  2.0080e-01, -3.9869e-02, -3.1738e-02,  1.7672e-01,\n",
       "           1.1984e-01, -9.9577e-02, -3.7483e-01, -4.3956e-01, -9.5205e-02,\n",
       "          -3.0408e-01,  3.2250e-02, -2.9469e-01, -2.3954e-01, -3.3261e-01,\n",
       "           2.5982e-01,  9.2762e-02,  3.2750e-01, -9.3082e-02, -6.6514e-01,\n",
       "          -3.5248e-01,  4.0832e-01,  3.4586e-03, -2.8129e-01, -3.8448e-01,\n",
       "          -6.6713e-02, -4.9346e-01,  1.2765e-01, -4.4650e-01,  1.9893e-01,\n",
       "           1.0835e-01,  1.8686e-01, -3.2670e-01,  6.5172e-02,  8.8101e-02,\n",
       "           6.1068e-02, -5.0938e-01,  2.7421e-01,  1.8326e-01,  9.3046e-02,\n",
       "           2.2489e-01,  6.8015e-01, -1.6210e-01,  4.3454e-01,  3.5722e-01,\n",
       "           3.1169e-01,  1.6221e-01, -4.1686e-01, -5.0259e-01, -7.5302e-02,\n",
       "           1.8488e-02, -5.9908e-01, -5.7442e-01,  6.1723e-02, -2.2868e-01,\n",
       "           1.6627e-01, -7.3083e-02, -5.1768e-01, -5.4307e-01, -3.1494e-01,\n",
       "          -3.0772e-01, -2.0517e-02, -3.2393e-01, -4.8159e-01, -2.5398e-01,\n",
       "          -6.1867e-01, -7.2891e-01,  1.9443e-01,  1.5164e-01,  2.7714e-01,\n",
       "          -4.3163e-01,  5.4580e-01, -3.0612e-01, -4.8554e-01,  3.7642e-01,\n",
       "           4.3432e-01, -5.0330e-01,  2.9526e-01, -6.3499e-01,  5.1171e-02,\n",
       "           2.0480e-01,  2.5291e-02, -3.4147e-02,  4.2807e-01,  1.1727e-01,\n",
       "           3.9831e-01,  2.7152e-02,  1.2476e-01, -2.9187e-01,  1.5015e-01,\n",
       "          -4.5427e-02, -3.2147e-01, -1.1176e-01,  1.7279e-02, -5.2967e-02,\n",
       "           5.4936e-01, -1.7306e+00, -1.7852e-01,  7.0060e-02, -2.7460e-01,\n",
       "          -1.3173e-01,  2.1473e-01,  5.0407e-03,  3.2283e-01, -3.5674e-01,\n",
       "           8.3698e-02,  8.7128e-02,  8.4654e-02, -1.0725e-01, -5.0983e-02,\n",
       "           2.9101e-01,  7.5929e-01,  4.0793e-01,  9.7463e-02, -2.6657e-01,\n",
       "          -6.0645e-01,  4.1554e-01,  1.9081e-01,  2.0730e-01,  3.9000e-01,\n",
       "           6.7931e-02, -6.9343e-01,  2.9674e-01,  1.9859e-01,  3.7002e-01,\n",
       "           2.5546e-02, -5.8496e-02, -4.8962e-01, -2.4682e-01, -4.3705e-01,\n",
       "           2.7759e-01, -1.3436e-01, -2.5922e-01,  6.3173e-01,  1.9273e-01,\n",
       "          -5.3531e-02,  3.9653e-01,  1.8733e-01, -4.1041e-01,  4.1633e-01,\n",
       "           5.2814e-01,  7.9114e-02,  1.4573e-01, -2.8055e-01, -1.6673e-01,\n",
       "          -3.1532e-01, -4.4420e-01,  4.4703e-01,  8.3941e-01, -7.2084e-03,\n",
       "           5.5343e-01, -3.5757e-01,  5.1174e-01, -3.6424e-01, -6.7779e-03,\n",
       "          -4.6486e-02,  6.8700e-02, -3.3988e-01,  1.6284e-01,  1.5551e-01,\n",
       "           1.0340e-01,  1.9709e-01,  8.5425e-01,  4.6494e-01,  1.4684e-01,\n",
       "           4.1031e-01,  2.5417e-02,  3.2080e-01,  7.8684e-01, -3.0254e-01,\n",
       "           3.4990e-01, -7.1578e-03, -4.1842e-01,  5.6461e-01, -3.5139e-01,\n",
       "           1.4248e-01, -2.3020e-01,  2.2987e-01,  1.7866e-01, -2.7854e-01,\n",
       "           2.1426e-01,  6.5273e-01,  9.5860e-02, -3.9283e-01,  7.3417e-01,\n",
       "          -4.0214e-01,  7.1262e-01, -3.0138e-02, -2.6992e-02, -1.0120e-01,\n",
       "          -7.4577e-01, -6.8569e-01,  1.4294e-01,  3.0808e-01, -7.6332e-01,\n",
       "          -3.4295e-01,  1.2380e-01,  1.3255e-01,  3.5068e-01,  2.4156e-01,\n",
       "           4.6144e-01, -1.7016e-01, -4.7185e-02, -4.4024e-01, -4.6015e-01,\n",
       "           1.1917e-01,  7.3311e-02,  9.5311e-02, -9.8199e-02,  9.2587e-02,\n",
       "           6.1067e-01,  2.4297e-01,  2.6799e-01,  1.4946e-01, -3.4610e-01,\n",
       "          -2.7723e-02,  1.7135e-01,  3.7874e-01, -6.1900e-01, -2.1807e-01,\n",
       "           5.3543e-02,  8.3164e-02,  6.6826e-01, -2.1832e-01, -1.1546e-01,\n",
       "          -1.2450e-01,  5.9739e-01, -2.2108e-01,  6.8002e-02,  1.0269e-01,\n",
       "          -6.8586e-01,  2.7619e-01, -5.9939e-02, -8.5297e-02, -5.2861e-01,\n",
       "           2.2997e-01, -5.7787e-02, -1.6068e-01, -7.3269e-03, -7.7601e-02,\n",
       "           2.2454e-01,  1.3406e-01,  2.8044e-01, -5.6270e-01, -4.2092e-01,\n",
       "           1.1781e-01, -3.2365e-01, -6.6016e-02, -4.5861e-01,  3.6766e-01,\n",
       "           1.3890e-01,  4.3829e-01, -1.0837e-01, -2.8576e-02, -1.3190e-01,\n",
       "           5.2866e-01, -6.2780e-01,  1.5573e-02,  4.8525e-01, -3.5595e-01,\n",
       "          -1.8139e-01,  4.7644e-01, -2.8011e-01,  4.9664e-02,  2.2982e-01,\n",
       "           3.7738e-01, -1.1530e-01, -4.7105e-01, -5.5247e-02,  9.3761e-01,\n",
       "           4.8980e-01,  2.0874e-02, -6.3820e-01,  1.2560e-01,  4.2066e-01,\n",
       "          -3.4219e-01,  1.0777e-01, -3.7280e-01, -2.3519e-01,  8.6456e-02,\n",
       "           3.3260e-02,  5.0344e-01, -5.8998e-02,  1.9741e-01,  4.6778e-01,\n",
       "           1.7011e-01, -6.9331e-01, -3.0557e-01,  6.3124e-01,  5.4670e-01,\n",
       "          -1.2247e-01, -5.4832e-01,  1.1480e-01, -6.4563e-01,  3.1198e-02,\n",
       "          -3.8623e-03, -3.5755e-01,  2.9653e-01, -1.4962e-01, -2.5683e-01,\n",
       "          -2.0003e-01, -5.3748e-01,  4.4973e-03,  1.0291e-01, -6.9595e-02,\n",
       "           4.1738e-01, -7.9610e-03, -3.1154e-01,  1.0058e+00, -2.7292e-01,\n",
       "          -3.1356e-02,  7.4139e-01, -5.4234e-01,  1.0457e-02,  2.4642e-01,\n",
       "          -1.3433e-01, -1.2416e-02,  2.4323e-02,  1.0660e+00,  1.1515e-05,\n",
       "           3.6402e-02, -2.0346e-01,  4.2203e-01, -4.7659e-01,  1.9394e-01,\n",
       "          -4.9958e-01,  4.8310e-01, -6.9399e-02,  2.4253e-01, -8.2329e-01,\n",
       "           4.3257e-01, -3.0604e-01, -3.2040e-01, -6.3922e-01,  5.0359e-02,\n",
       "          -5.0711e-02,  6.6524e-02, -5.8707e-02,  6.0787e-01, -1.0953e+01,\n",
       "           1.0946e-01,  8.4108e-01, -3.2618e-02,  1.3653e-01,  4.3336e-01,\n",
       "           1.2158e-01, -4.6684e-01,  7.2509e-01, -2.3242e-01, -6.6092e-03,\n",
       "          -3.5861e-01, -1.2444e-01, -1.3819e-01, -6.3519e-01,  4.0758e-01,\n",
       "          -1.3255e-01, -5.9114e-01,  2.8540e-01, -1.6945e-01,  2.4978e-01,\n",
       "          -2.3651e-01,  1.5671e-01,  5.9352e-01, -6.1010e-01, -3.3943e-01,\n",
       "          -1.1958e-01, -2.9991e-01,  5.1723e-01,  4.0217e-01, -2.3672e-01,\n",
       "           2.9096e-01,  3.3529e-02,  2.9455e-01, -5.5694e-01, -4.8133e-01,\n",
       "           4.3369e-01, -3.9950e-01,  2.1957e-01, -9.9943e-01, -3.3265e-01,\n",
       "          -1.8082e-01, -5.0951e-01, -2.7870e-02,  5.3812e-01,  7.5524e-01,\n",
       "           1.2821e+00,  1.1534e+00,  9.8198e-02,  4.6362e-01, -1.0523e-01,\n",
       "          -6.6935e-01,  5.1619e-02, -5.6525e-01, -4.1224e-01,  8.7196e-02,\n",
       "           2.2385e-01, -5.4352e-01,  4.7854e-01, -5.9897e-01,  1.8912e-01,\n",
       "           1.4329e-01,  7.9852e-02,  1.7368e-01, -5.1897e-01, -5.5039e-01,\n",
       "          -1.3282e-01,  8.1201e-02,  8.6324e-02, -6.7881e-01,  1.1881e-01,\n",
       "           1.7006e-01,  6.8197e-01, -2.5397e-01, -4.4268e-01,  4.1137e-01,\n",
       "          -2.2746e-03,  9.3809e-02,  1.3158e+00, -2.6036e-01, -1.7413e-01,\n",
       "          -4.8428e-02, -2.3194e-01, -3.3484e-01, -3.9656e-01, -4.5613e-01,\n",
       "          -2.5020e-02,  3.6573e-01,  5.1948e-01,  2.6062e-01, -2.3329e-02,\n",
       "          -3.0054e-01, -4.1445e-02,  5.1977e-02,  3.3862e-01,  9.6772e-01,\n",
       "          -6.6887e-01,  2.6280e-01, -1.9148e-01,  9.2178e-02,  2.2223e-01,\n",
       "          -5.7106e-01, -1.6871e-02,  2.9434e-03,  1.3975e-02,  5.7281e-01,\n",
       "           1.9705e-01,  2.6453e-01,  3.1190e-02,  2.2859e-01, -4.1682e-01,\n",
       "          -4.4044e-01,  2.9125e-04, -1.3190e-01, -5.8294e-01, -4.1034e-01,\n",
       "          -1.3514e-01,  1.3712e-01, -2.8585e-01, -5.3585e-01, -2.5012e-01,\n",
       "           1.4192e-01,  2.5900e-03,  4.9952e-02, -3.3857e-01,  4.7379e-01,\n",
       "          -7.6918e-01, -2.1429e-01, -2.7578e-01, -3.8495e-03,  1.4633e-01,\n",
       "           2.5508e-01, -5.1518e-01, -2.7838e-01, -1.0123e-01, -9.2314e-01,\n",
       "          -8.4489e-01,  2.2927e-02,  7.4620e-01, -1.1960e-01,  1.1861e-01,\n",
       "           8.5428e-02, -3.1671e-03, -3.7708e-01,  2.5194e-01, -2.4644e-01,\n",
       "           2.8149e-01,  4.3265e-01, -4.9583e-01,  2.9773e-02, -1.2792e-01,\n",
       "          -1.8992e-02, -1.0887e-01, -8.8647e-02, -5.4131e-01, -6.1371e-01,\n",
       "          -4.1197e-01,  1.3736e-01, -2.2747e-01,  4.1516e-01,  6.8072e-02,\n",
       "           1.2582e-01, -2.0826e-01,  9.8224e-02,  8.2540e-01,  6.0197e-01,\n",
       "           2.3014e-01, -3.2920e-01, -7.7666e-01,  3.4140e-02,  4.1642e-02,\n",
       "          -1.6450e-01,  2.1317e-01, -1.6384e-01, -3.0657e-01, -1.0112e-01,\n",
       "           1.2620e-02,  1.0002e+00, -2.0790e-01, -1.9155e-01,  1.0507e-01,\n",
       "           7.4626e-01, -6.0834e-01, -5.7221e-01,  3.7612e-01,  5.0418e-02,\n",
       "          -5.6831e-01, -1.9450e-01,  4.0315e-01, -1.7111e-01,  8.9921e-02,\n",
       "           3.7572e-01,  1.7821e-01, -1.8612e-01,  2.2711e-01, -6.7291e-02,\n",
       "          -1.7001e-01,  3.2731e-01, -1.6533e-01, -7.7254e-02,  4.0384e-01,\n",
       "           9.9517e-02,  1.5584e-01, -2.5262e-01, -4.8444e-01, -2.7025e-01,\n",
       "           1.5439e-01,  1.1110e+00, -1.7989e-01, -3.2527e-01,  6.6881e-01,\n",
       "          -3.0956e-01,  3.8928e-01,  4.8478e-01, -2.2611e-01, -5.0807e-02,\n",
       "           5.8049e-01,  3.0288e-01, -7.3236e-01,  2.2417e-01,  3.1670e-01,\n",
       "          -1.7335e-01, -5.3095e-01, -3.2838e-01, -2.1330e-01, -3.5189e-02,\n",
       "          -3.6356e-01,  5.2132e-01,  1.2255e-01,  3.9253e-01, -1.0843e-01,\n",
       "          -8.7207e-01,  4.7185e-01,  2.6128e-01,  8.0250e-01, -6.2568e-03,\n",
       "          -4.4880e-01, -3.5541e-01, -3.5790e-01, -6.6453e-02,  3.6543e-01,\n",
       "           2.0116e-01, -1.0333e+00,  4.6071e-02,  4.0507e-01, -3.9812e-02,\n",
       "           4.7083e-01, -1.4254e-01,  1.9634e-01,  4.7891e-01, -4.3918e-02,\n",
       "          -2.1856e-01,  1.3161e-01, -3.3027e-02, -2.7190e-01,  4.2861e-01,\n",
       "           4.6755e-01,  1.6062e-02,  2.5454e-02,  6.4881e-02, -4.2427e-01,\n",
       "          -3.4980e-01,  7.7064e-02, -1.7966e-01, -3.5675e-01, -1.4295e-01,\n",
       "          -4.0726e-01, -5.1772e-01, -6.6164e-02, -1.1357e-01,  1.3233e-01,\n",
       "           3.1032e-01,  1.3036e-01,  3.8431e-01, -2.3708e-01, -1.2541e-01,\n",
       "          -2.7977e-01,  4.5653e-01, -2.7595e-01,  8.0808e-01,  2.0566e-01,\n",
       "          -1.4423e-01,  2.0720e-01,  2.1514e-01, -2.9046e-01,  1.3028e-01,\n",
       "          -2.7828e-01,  2.8810e-01, -5.7798e-01, -3.4110e-01, -5.9710e-02,\n",
       "          -7.7948e-02, -1.5049e-01, -2.8505e-01, -1.0179e-01,  1.0608e-01,\n",
       "          -9.9043e-02,  9.7868e-01, -4.1759e-01,  6.3833e-01, -1.0342e-01,\n",
       "          -1.1350e-01, -7.5089e-01,  3.9397e-01,  3.6376e-01, -1.4798e-01,\n",
       "          -5.6434e-01, -2.1653e-01, -8.9158e-02, -4.2284e-01,  8.8621e-01,\n",
       "           5.8157e-01, -7.4673e-01, -3.7246e-01, -1.5919e-01, -2.4113e-01,\n",
       "          -4.7389e-02,  6.9820e-02, -3.2710e-01, -1.3295e-02, -1.6580e-01,\n",
       "          -9.8715e-01,  2.7641e-01, -3.6473e-02,  4.9205e-01, -2.5466e-01,\n",
       "           7.6979e-01, -8.6017e-02, -1.2749e-01,  1.1690e+00,  1.1169e-01,\n",
       "          -6.1844e-02,  5.9531e-01, -1.6837e-01, -6.2973e-01, -7.0295e-02,\n",
       "           4.4239e-02, -3.1098e-01,  8.7670e-02, -4.7880e-01,  2.9353e-01,\n",
       "           2.3648e-01, -7.1226e-02,  4.3877e-01]], device='mps:0'),\n",
       " tensor([[-0.0626, -0.6265, -0.7028,  ...,  0.1217,  0.4033,  0.6177],\n",
       "         [-0.2170, -0.9206, -0.6892,  ...,  0.0033,  0.6907,  0.6555],\n",
       "         [-0.2446, -0.5514, -0.2367,  ...,  0.2636,  0.1883,  0.2926],\n",
       "         [-0.4032, -0.3304, -0.5761,  ...,  0.2794,  0.6603, -0.1920]],\n",
       "        device='mps:0'),\n",
       " tensor([[-3.0175e-01, -6.6980e-01, -5.8058e-01, -1.0169e+00, -4.7769e-01,\n",
       "          -1.2967e-03, -6.6155e-01,  2.1546e-01,  1.0829e-01,  2.8188e-01,\n",
       "           2.6802e-01, -7.4553e-03,  8.8102e-02,  1.2687e-02, -2.8827e-01,\n",
       "           2.5617e-01, -2.6220e-01,  2.1438e-01,  7.7033e-02, -3.7727e-02,\n",
       "           8.1075e-01,  3.6768e-01, -1.6212e-01, -2.0573e-01, -3.3086e-01,\n",
       "           4.5605e-01, -1.3148e-01,  4.8222e-02,  3.5668e-01, -7.9078e-01,\n",
       "          -2.5202e-01,  1.1725e-01, -2.9749e-01,  1.0378e+00, -1.1594e-01,\n",
       "          -3.3333e-01, -1.7748e-01, -4.3787e-01,  2.1301e-01, -5.0568e-02,\n",
       "           4.5755e-01, -1.4778e-01, -2.4105e-01,  2.0005e-01, -1.2606e-02,\n",
       "           9.5042e-02,  3.7443e-01, -8.8176e-01,  3.3878e-01, -1.1266e-01,\n",
       "           1.8889e-01, -3.4865e-01,  5.0318e-01,  2.1380e-01, -8.5959e-01,\n",
       "           5.2285e-02,  2.7071e-01, -4.0013e-02, -8.2586e-01,  1.2128e-01,\n",
       "          -4.6670e-01,  5.9963e-01, -1.1233e-01,  1.0830e-01,  1.9111e-01,\n",
       "          -1.8788e-01, -5.8713e-01,  1.3197e-01, -2.4455e-01, -5.4474e-02,\n",
       "          -5.4218e-03,  6.5278e-03,  1.1025e-02,  5.1945e-02, -1.4253e-01,\n",
       "          -4.2182e-01, -1.2333e-01, -1.5679e-01, -4.8684e-01, -2.8943e-01,\n",
       "           1.1703e-01,  2.2252e-01, -9.0303e-02, -2.5004e-01,  1.6458e-01,\n",
       "          -2.2152e-02, -1.1224e-01,  2.6949e-01,  1.3030e-01,  8.0149e-02,\n",
       "           3.0522e-01,  2.0928e-01, -1.5859e-02, -2.0004e-01,  5.0817e-01,\n",
       "          -3.4280e-01,  5.8038e-01, -1.4811e-01, -1.0314e-01, -1.4254e-01,\n",
       "          -8.5900e-02,  1.6763e-01,  2.0530e-01,  3.5774e-01, -1.2954e-01,\n",
       "          -2.2287e-01, -1.0609e-01, -1.5258e-01, -6.7798e-01, -3.9055e-02,\n",
       "          -2.1100e-01, -3.5712e-01, -7.7076e-02, -6.5853e-01, -4.8448e-01,\n",
       "           3.3116e-01,  8.6550e-02, -3.0879e-01, -6.8888e-02,  6.9461e-01,\n",
       "           4.5933e-01, -4.6570e-01,  9.2314e-02, -2.0174e-01,  3.6437e-01,\n",
       "          -2.6446e-01,  3.1500e-01, -2.6301e-01,  1.6895e-01,  3.9658e-01,\n",
       "           2.4446e-01,  1.7780e-01, -2.5048e-01, -9.5926e-02, -4.5958e-03,\n",
       "          -2.0377e-01,  3.0765e-02, -1.7716e-01,  2.5017e-01, -1.6240e-01,\n",
       "          -1.1815e-01, -1.9138e-01, -3.8586e-01, -2.9267e-01,  2.0262e-02,\n",
       "          -3.8745e-01,  1.7052e-01,  2.0249e-01,  2.3481e-01,  2.0374e-01,\n",
       "          -5.9123e-01, -3.7090e-01,  2.6279e-01, -1.5877e-01, -1.7998e-01,\n",
       "          -2.7334e-01,  4.8190e-01,  1.1280e-01,  3.4257e-02,  4.9299e-01,\n",
       "           4.8370e-01, -3.4823e-01,  3.8388e-01, -5.6908e-01,  4.7970e-01,\n",
       "          -8.0979e-02,  2.2367e-02, -3.4129e-01,  6.6072e-01, -2.2073e-01,\n",
       "           4.2082e-01,  4.4552e-02, -5.8999e-02, -4.2000e-01,  2.1493e-01,\n",
       "           3.2119e-01,  5.5215e-01,  2.8715e-01, -2.9960e-01, -5.3479e-02,\n",
       "           5.6045e-02, -1.5361e+00,  1.7164e-01, -9.1554e-01, -2.4568e-01,\n",
       "          -2.9564e-01,  4.7877e-01, -1.0528e-01,  5.3352e-01,  2.0833e-01,\n",
       "           1.5799e-01,  2.2647e-01,  4.1759e-01,  6.8457e-01,  1.3707e-01,\n",
       "           3.0360e-01,  6.7200e-01,  5.1151e-01,  1.5972e-01, -9.9524e-02,\n",
       "          -8.4006e-01,  4.5848e-01, -2.4269e-01,  4.0564e-01, -4.0343e-01,\n",
       "           2.8765e-01, -4.5223e-01,  2.4525e-01,  5.3122e-01,  6.8031e-02,\n",
       "          -1.8708e-01, -3.3998e-01, -5.5491e-01, -1.4993e-01, -5.1624e-01,\n",
       "          -6.4144e-02, -3.5381e-01, -1.2625e-01,  2.2048e-01,  2.4657e-01,\n",
       "           2.5758e-01, -1.0877e-01,  8.6218e-02, -4.4103e-01, -2.5500e-01,\n",
       "           4.7314e-01, -4.2033e-02, -2.8710e-01, -2.4144e-01, -2.3728e-01,\n",
       "           2.8834e-02, -6.8555e-01,  5.9320e-01,  5.4095e-01,  2.8945e-01,\n",
       "          -3.7894e-01, -2.6274e-01,  3.3479e-01, -8.1051e-02, -1.9290e-01,\n",
       "          -3.2112e-01, -2.8164e-01, -6.3114e-01,  3.0389e-02,  3.2483e-01,\n",
       "           4.5472e-01,  2.2502e-02,  5.9776e-01,  9.2431e-01,  6.2616e-01,\n",
       "          -6.3152e-02,  4.1419e-01,  3.5893e-01,  4.0719e-01, -3.6893e-01,\n",
       "           2.3675e-01,  2.5578e-02, -3.0394e-01,  1.8930e-01, -3.4051e-01,\n",
       "          -5.4999e-01, -3.0810e-01,  1.8355e-01,  3.9312e-01, -4.0218e-01,\n",
       "           1.3738e-01,  3.6324e-01, -9.4898e-02, -2.0045e-01,  2.3112e-01,\n",
       "          -4.5769e-01,  2.7052e-01, -3.8193e-02, -2.4676e-01, -7.8228e-01,\n",
       "          -8.0561e-01,  1.7671e-01,  4.0383e-01,  2.7239e-02, -6.2392e-01,\n",
       "           4.0250e-01, -7.7847e-02, -1.6572e-01,  5.7036e-02, -2.0384e-01,\n",
       "           5.1000e-01, -2.4829e-01, -4.3042e-02,  2.3431e-01, -5.3414e-02,\n",
       "          -2.6091e-01,  2.8407e-01, -2.1272e-01, -1.4648e-01,  2.0817e-01,\n",
       "           1.1829e-01,  2.4851e-01,  2.9531e-01, -1.8469e-01, -1.0712e-01,\n",
       "          -1.4378e-01,  1.4497e-01, -3.1745e-01, -1.1662e-01,  2.7385e-01,\n",
       "           4.4367e-01, -8.9016e-02,  5.4896e-01, -2.2052e-01, -4.3292e-01,\n",
       "          -5.1639e-02,  7.8045e-01, -2.5523e-01,  4.0706e-01, -4.3595e-01,\n",
       "          -1.0067e+00, -2.3247e-01, -2.4278e-01, -1.6148e-01, -4.9052e-01,\n",
       "           3.3004e-01, -2.3283e-01,  9.8914e-02, -2.6420e-02, -2.8987e-01,\n",
       "           3.0162e-01,  6.6029e-01,  4.9695e-01, -3.8492e-02, -2.6121e-01,\n",
       "          -9.2459e-02, -1.8654e-01, -6.9434e-01, -2.8268e-01, -3.3254e-01,\n",
       "           2.9086e-01,  5.4987e-01, -9.4563e-03, -3.4511e-02, -2.2907e-01,\n",
       "           6.7731e-01,  4.2296e-02, -2.2174e-01,  4.3139e-01, -2.6503e-01,\n",
       "          -3.1949e-02,  4.2142e-01,  2.4983e-01,  1.4853e-01,  2.7700e-02,\n",
       "           3.8850e-01, -2.4014e-01, -5.2190e-01,  2.2986e-02,  7.3507e-01,\n",
       "           2.4180e-01,  4.1704e-01, -1.2385e-01,  6.9499e-01,  1.1516e-01,\n",
       "          -4.0155e-01,  1.9317e-01, -3.4905e-01, -9.7611e-02, -4.1392e-02,\n",
       "           2.1507e-01,  1.7865e-01, -4.6424e-01, -1.4887e-01, -3.2423e-01,\n",
       "           5.4446e-01, -4.4284e-01,  1.0868e-01,  6.0624e-01,  9.5384e-02,\n",
       "          -8.9992e-02, -5.1526e-01,  4.1846e-02, -9.2054e-01, -4.2575e-01,\n",
       "           1.3709e-03, -6.5500e-02,  3.1680e-01, -1.4904e-01, -6.4729e-02,\n",
       "          -9.6847e-03, -2.5828e-01,  6.6431e-01,  2.4458e-01,  2.2317e-02,\n",
       "           2.4482e-02, -4.2979e-01, -1.6413e-01,  7.3132e-01, -2.8516e-01,\n",
       "          -2.0146e-01,  2.4408e-01, -1.7036e-01,  7.5170e-02,  9.1916e-01,\n",
       "          -5.4459e-01, -1.2919e-01,  1.6822e-01, -9.2317e-02, -4.5496e-01,\n",
       "          -2.6936e-01,  3.7161e-01,  3.1036e-01, -2.3544e-01,  9.3051e-03,\n",
       "          -1.9547e-01, -4.8565e-02, -1.3911e-01,  1.4568e-01,  4.0195e-01,\n",
       "           3.8214e-01, -3.3720e-01, -9.5011e-02, -2.3665e-01, -1.5221e-01,\n",
       "           5.8983e-01,  4.1325e-02,  5.0294e-02,  4.4798e-01, -1.1088e+01,\n",
       "           3.1967e-01,  8.8400e-01, -4.8213e-01, -3.6244e-01,  1.7139e-01,\n",
       "           2.1689e-01, -3.2129e-01,  5.4136e-01, -3.4162e-01, -4.5191e-01,\n",
       "          -3.6412e-01, -2.6337e-01, -5.2896e-02, -5.3673e-01, -2.6602e-01,\n",
       "          -4.5073e-02, -6.6599e-01,  4.9827e-01, -5.6950e-02,  4.1197e-01,\n",
       "          -7.3141e-01,  2.9915e-01,  2.4368e-01, -2.5628e-01, -5.1343e-01,\n",
       "          -4.1890e-01,  5.6851e-02, -5.3908e-02,  1.5332e-01, -1.4587e-02,\n",
       "          -7.8227e-02, -1.4075e-01,  3.3179e-01, -1.0998e-01, -6.0080e-01,\n",
       "          -2.6099e-01, -1.5141e-01,  3.0497e-01,  3.4405e-01, -1.6240e-02,\n",
       "           1.4509e-01, -6.8089e-02,  4.0110e-01,  3.9231e-01,  8.2129e-01,\n",
       "           1.4654e+00,  3.3271e-01,  1.3907e-01,  1.6580e-01, -3.7775e-01,\n",
       "          -7.0736e-01,  3.5929e-01, -8.6010e-01,  1.6285e-02,  1.0245e-01,\n",
       "           4.0998e-01, -5.2030e-01,  1.8141e-01, -1.0184e-01, -2.9502e-01,\n",
       "           4.7893e-01, -4.8131e-02,  3.6590e-01,  6.5942e-02, -2.0825e-01,\n",
       "           3.6483e-01,  9.8140e-03,  1.0428e-01, -2.6566e-01,  7.0489e-02,\n",
       "          -1.0515e-01,  9.0629e-01,  5.6924e-02,  2.2025e-01,  3.5505e-01,\n",
       "           1.9906e-01,  5.6627e-02,  9.7982e-01,  1.4887e-02,  2.3439e-02,\n",
       "           2.4081e-01,  1.2057e-01, -8.2163e-01, -6.6525e-01, -3.0672e-01,\n",
       "          -1.9990e-01,  2.9884e-01,  2.1077e-01,  4.5599e-01,  1.0191e-01,\n",
       "          -2.4030e-01, -4.7529e-01, -8.4384e-02,  3.4405e-01,  2.7867e-01,\n",
       "          -8.1985e-01, -2.7656e-01,  3.8855e-01,  5.9943e-03, -3.3150e-01,\n",
       "          -1.0185e-01,  3.9454e-02, -6.9953e-01,  2.8364e-01, -8.8777e-01,\n",
       "           4.4082e-01, -1.1780e-01,  1.2180e-01,  3.3622e-01, -2.0828e-01,\n",
       "           1.4357e-01,  2.4272e-01,  1.5712e-01, -1.2949e-01, -3.2449e-01,\n",
       "          -6.2829e-02,  5.6284e-01, -3.0667e-01, -6.1434e-01,  7.8437e-02,\n",
       "          -4.1754e-01, -3.2532e-02,  2.0604e-01, -4.0847e-01,  2.9599e-01,\n",
       "          -9.2509e-01,  5.7991e-01,  5.3225e-01,  1.3132e-01,  4.2593e-01,\n",
       "           1.0221e-01,  2.7063e-01,  3.5230e-01, -3.5712e-01, -6.5350e-01,\n",
       "          -4.5414e-01,  1.1263e-01,  4.4292e-01, -4.3713e-01, -6.4580e-02,\n",
       "          -2.5887e-01, -1.8874e-02, -3.6240e-01,  4.6295e-01, -4.5580e-01,\n",
       "           5.0356e-01,  2.7012e-01, -3.2513e-01,  8.5783e-02, -9.8713e-02,\n",
       "          -3.8531e-02, -3.5799e-02, -9.5866e-02, -2.5031e-01, -6.3330e-01,\n",
       "          -2.7211e-01,  6.7283e-02,  1.7325e-01, -4.6052e-01, -4.7682e-01,\n",
       "          -2.3481e-01, -3.3311e-01, -3.1033e-01,  5.6504e-01,  4.5932e-01,\n",
       "           3.9209e-03, -2.8593e-01, -8.9424e-02,  6.8840e-02,  1.4673e-01,\n",
       "           5.1449e-01,  1.8069e-02, -1.3468e-01, -1.1048e+00, -2.6467e-01,\n",
       "          -2.6220e-01,  1.1618e+00, -5.0380e-01, -2.3770e-01,  1.6040e-01,\n",
       "          -3.5048e-01, -1.3186e-01,  3.4729e-01, -4.3105e-01, -2.8086e-01,\n",
       "          -3.8445e-02,  3.7695e-02,  4.9161e-01, -1.4846e-01,  4.3220e-01,\n",
       "           3.6677e-01,  1.2157e-01, -2.9789e-01,  1.2610e-01, -5.6793e-01,\n",
       "          -5.4133e-01,  1.5364e-01, -3.2761e-01,  2.5915e-01, -1.4858e-01,\n",
       "          -4.2085e-01,  4.1585e-02, -2.8757e-01,  3.6606e-02,  6.1003e-01,\n",
       "           1.4671e-01,  1.3806e+00, -3.4816e-01, -7.2524e-02,  2.0323e-02,\n",
       "           1.1091e-02,  2.8904e-01, -4.7353e-02, -2.8690e-01,  4.2239e-01,\n",
       "           3.8130e-01, -5.1135e-01, -2.1610e-02,  2.4837e-01,  8.4575e-02,\n",
       "           1.5806e-01, -7.0628e-01, -5.6971e-01, -4.5014e-01,  8.3924e-01,\n",
       "           4.2194e-01,  6.0600e-02, -1.3581e-01,  3.6786e-01,  6.1553e-01,\n",
       "          -7.2307e-01,  1.2407e-01, -5.6063e-01,  3.4624e-01, -2.8673e-02,\n",
       "          -3.8840e-02, -1.4214e-01, -4.1968e-01, -2.4922e-01,  5.1180e-01,\n",
       "           8.8041e-01, -8.0026e-01,  2.7926e-01,  2.8429e-01, -6.0185e-01,\n",
       "           2.7400e-01, -3.0303e-01,  4.9336e-01,  2.6875e-01, -1.5132e-01,\n",
       "           7.2616e-02,  3.0180e-01, -8.4330e-02, -4.9152e-01,  4.2695e-02,\n",
       "           6.6296e-01, -1.1219e-01,  1.9635e-01, -7.4417e-02, -4.0446e-01,\n",
       "           1.0495e-01,  3.4093e-02, -4.6593e-01, -1.8069e-01,  2.7051e-01,\n",
       "          -1.6951e-01, -5.1150e-01,  3.0621e-02,  3.2784e-01, -9.6532e-02,\n",
       "           3.0489e-01,  6.1836e-01,  1.8625e-01,  2.8380e-01, -2.6720e-01,\n",
       "          -1.7008e-01,  2.8282e-01, -3.1094e-01,  4.9413e-01, -2.5900e-01,\n",
       "          -2.8052e-01,  5.1395e-01, -5.1076e-02, -2.3065e-01,  1.3898e-01,\n",
       "          -1.9188e-01,  1.2284e-01, -2.0852e-01, -8.5311e-02,  1.3929e-01,\n",
       "           1.5890e-01, -2.0449e-01, -3.2401e-01, -1.5393e-01,  2.0152e-01,\n",
       "          -3.0676e-01,  6.9643e-01, -5.2115e-02,  2.7670e-01, -3.2876e-01,\n",
       "           3.2702e-01, -6.1801e-01, -3.5319e-01,  3.7352e-01,  4.6685e-03,\n",
       "          -4.5457e-01, -7.3337e-01,  3.5639e-01,  1.9663e-01,  6.8855e-01,\n",
       "           4.1244e-01, -4.6492e-02, -4.1667e-01, -3.7578e-01,  4.9655e-02,\n",
       "          -5.1082e-02, -1.9918e-01,  4.2887e-02,  2.6552e-01, -3.0464e-03,\n",
       "          -7.6106e-01, -5.7014e-01,  5.2966e-02,  3.2732e-01, -1.7900e-01,\n",
       "           7.9234e-01, -4.4085e-01,  6.0370e-02,  6.0840e-01,  4.8676e-01,\n",
       "           1.1522e-01,  7.3154e-01,  1.1035e-01, -7.9930e-01,  5.9814e-04,\n",
       "          -1.4466e-01, -2.9386e-01,  2.4061e-01, -9.9708e-02,  2.6837e-01,\n",
       "           3.2979e-01,  1.6883e-01,  7.6565e-01]], device='mps:0'),\n",
       " tensor([[-5.4428e-01, -3.2717e-01, -5.7726e-01, -4.9891e-01, -1.0664e-01,\n",
       "          -4.5807e-01, -5.9307e-01,  5.3891e-01,  3.2190e-01,  6.6029e-01,\n",
       "          -6.7260e-02,  1.2625e-02,  3.9646e-02, -2.8407e-01, -6.9524e-01,\n",
       "          -9.8629e-02,  1.5969e-01,  3.8506e-01,  6.5757e-01,  4.2945e-01,\n",
       "           6.3696e-01,  4.2341e-01, -9.6068e-02, -1.3250e-01, -6.7899e-01,\n",
       "          -1.7653e-01,  6.2424e-02, -1.3791e-01,  8.3816e-02, -4.4329e-01,\n",
       "          -8.1223e-03,  3.3628e-01, -1.7883e-01,  7.3589e-01, -2.0096e-01,\n",
       "          -3.2469e-01, -1.0917e-02, -2.7310e-01,  2.9146e-01,  4.6916e-01,\n",
       "           7.8053e-01,  1.0665e-03,  3.6082e-01,  7.4596e-02,  5.1178e-01,\n",
       "           9.8219e-02,  1.8713e-01, -5.6160e-01,  8.5150e-02, -5.2922e-01,\n",
       "          -3.3751e-01,  2.9670e-02,  5.0643e-01,  1.3561e-01, -8.9152e-01,\n",
       "          -4.0803e-01, -1.4918e-01,  3.7061e-01, -5.7854e-01,  4.7422e-02,\n",
       "          -9.7774e-02,  4.3019e-01,  2.9469e-01, -9.2631e-03,  1.4417e-01,\n",
       "           7.8668e-03, -7.0828e-01, -5.3904e-01, -6.0345e-01, -2.2172e-01,\n",
       "           8.4454e-02,  4.2448e-01, -4.1213e-02, -4.2031e-01,  3.2432e-01,\n",
       "          -5.5053e-01,  5.6453e-01, -1.3849e-01, -4.5773e-01,  1.1507e-01,\n",
       "           3.4806e-01,  2.3905e-01, -3.7845e-01,  1.1068e-01,  1.8739e-01,\n",
       "          -2.7077e-01,  8.7134e-01,  1.6425e-01,  2.0084e-02, -2.9445e-03,\n",
       "           4.5293e-01, -3.0010e-01, -7.0934e-01, -3.5740e-01,  5.3200e-02,\n",
       "          -6.1357e-01,  1.5297e-01, -3.3605e-01,  3.4711e-01, -3.9751e-01,\n",
       "           6.5646e-02,  4.5588e-01,  2.1691e-01,  3.0585e-01, -8.1083e-01,\n",
       "          -4.4396e-01,  1.6058e-01, -1.4151e-01, -3.7897e-01, -6.4861e-02,\n",
       "           7.9145e-02,  8.0743e-02,  6.4411e-01, -6.0580e-01, -4.5235e-01,\n",
       "           2.9922e-01,  3.8846e-01, -9.9376e-03, -8.0919e-02,  4.3617e-01,\n",
       "           1.3693e-01, -1.8769e-01, -1.6875e-01, -1.6879e-01,  3.0022e-01,\n",
       "          -6.4210e-01,  7.0135e-01, -2.7653e-01,  3.3047e-01,  4.4094e-01,\n",
       "          -4.3290e-02,  3.4944e-01, -3.2746e-01, -3.3177e-01, -1.2287e-01,\n",
       "           1.6429e-01,  5.1301e-01, -5.2105e-01, -1.2216e-01,  4.9715e-01,\n",
       "           2.1987e-02,  1.6903e-01, -4.2189e-01, -1.1248e-01,  2.0009e-01,\n",
       "          -1.1021e-01, -2.9823e-01, -5.1535e-01, -4.1705e-01, -1.1118e-01,\n",
       "          -7.2230e-01, -7.7967e-01,  3.1218e-01,  8.8243e-02,  2.5730e-01,\n",
       "          -2.6322e-01,  3.5778e-01, -1.5190e-02, -1.3252e-01, -2.5147e-01,\n",
       "           6.9639e-01,  1.6371e-01,  1.5235e-01, -2.7439e-02, -4.5799e-01,\n",
       "           1.0976e-01, -8.5769e-02, -1.9110e-01, -3.6696e-01,  1.2378e-01,\n",
       "           2.8516e-01,  1.5070e-01,  3.8110e-01, -3.2299e-01,  3.5535e-01,\n",
       "          -4.3543e-01,  3.5087e-01, -2.0051e-02,  2.0010e-01,  1.5727e-01,\n",
       "          -7.1503e-02, -1.5951e+00,  4.9998e-01, -3.3197e-01,  2.8129e-01,\n",
       "          -3.2578e-01,  2.3822e-01,  3.5278e-01,  9.0046e-02, -1.9108e-01,\n",
       "           1.4713e-02,  3.4355e-01, -4.3695e-02,  4.2145e-01, -2.3397e-01,\n",
       "           2.3280e-01,  1.0273e+00,  4.3854e-01,  1.7300e-01,  2.2108e-01,\n",
       "          -7.6975e-01,  2.4435e-01,  1.5634e-01, -5.9166e-01,  1.0001e-01,\n",
       "           2.4530e-01, -4.1543e-01, -1.2111e-01, -2.6260e-02, -1.1042e-01,\n",
       "           3.2991e-02, -3.8546e-01, -4.1912e-01,  2.4175e-01, -7.8122e-02,\n",
       "           2.9581e-01, -1.8854e-01, -5.1581e-02,  7.0446e-01, -4.2881e-02,\n",
       "           2.3683e-02,  1.9313e-01, -4.7818e-01, -3.6183e-01, -1.0881e-01,\n",
       "           1.8376e-01, -1.4579e-01,  2.4311e-01, -1.7515e-01, -2.1393e-01,\n",
       "           1.5473e-01, -7.5816e-01,  7.9576e-01,  2.8851e-01,  9.7707e-02,\n",
       "          -1.5717e-01, -1.0274e-01, -5.8232e-02, -4.3775e-01, -1.5412e-02,\n",
       "           1.2577e-01, -7.1667e-01, -6.9169e-01, -3.7409e-02, -2.7630e-02,\n",
       "           1.9767e-01, -3.9468e-01,  7.7355e-01,  3.5991e-01,  3.9809e-01,\n",
       "           4.3941e-02, -1.4390e-01, -1.4647e-02,  4.1656e-01, -3.4096e-01,\n",
       "           3.2532e-01, -1.0385e-01, -5.0159e-01,  3.6221e-01, -3.8364e-02,\n",
       "          -6.8838e-01,  8.6532e-02,  3.5297e-01,  4.2882e-01, -8.3792e-02,\n",
       "          -1.0957e-01,  2.3681e-01,  1.9867e-01, -4.7369e-02,  4.4329e-01,\n",
       "          -2.7097e-01, -5.9019e-02, -4.6750e-01, -1.9977e-01, -5.7659e-01,\n",
       "          -7.5615e-01, -2.2313e-01, -2.4699e-03, -1.1100e-01, -1.0118e+00,\n",
       "          -1.5108e-01,  2.7289e-01, -1.2459e-01, -1.2133e-01,  7.8632e-03,\n",
       "          -4.6556e-02, -1.1618e-02,  1.0033e-01,  2.2048e-02, -4.7030e-01,\n",
       "          -2.5640e-01,  2.6393e-01, -1.6108e-01,  1.4215e-01,  2.6051e-01,\n",
       "          -2.4287e-01,  1.7129e-02,  2.4995e-01, -1.1245e-01,  9.5832e-02,\n",
       "          -5.0777e-01, -4.2767e-01, -4.2570e-01, -6.6579e-02, -2.8942e-02,\n",
       "           6.5188e-02,  6.6931e-01,  3.3682e-01, -5.4189e-01, -4.3667e-01,\n",
       "          -5.3383e-02,  4.8552e-01, -5.4144e-02, -3.1136e-01,  8.9119e-02,\n",
       "          -1.0608e+00, -4.3515e-01,  7.0883e-02, -4.0694e-01, -4.0642e-01,\n",
       "           1.9051e-01,  1.2719e-02,  1.1150e-01,  2.4656e-01, -4.7487e-01,\n",
       "           5.9798e-01,  7.2190e-01,  2.4084e-01, -2.2205e-01, -2.6893e-01,\n",
       "           6.0828e-02, -6.5972e-01,  7.3061e-02, -3.2830e-01, -9.8520e-02,\n",
       "           3.3933e-01,  1.2090e-01, -5.9241e-01, -3.1336e-01,  1.2006e-01,\n",
       "           6.5468e-01, -4.6509e-01,  1.8092e-01,  2.8025e-01, -3.4949e-01,\n",
       "          -1.9302e-01,  2.0049e-01, -2.6003e-01, -3.1021e-01, -4.7415e-02,\n",
       "          -1.5099e-02, -5.0759e-01,  1.1051e-01, -1.4264e-01,  8.3220e-01,\n",
       "           5.8947e-01,  4.5788e-01, -4.9109e-01,  3.5546e-01,  8.7834e-02,\n",
       "          -4.0971e-01,  1.0552e-01,  9.3760e-02, -3.0757e-02, -4.7961e-01,\n",
       "          -2.1726e-01,  1.2256e-01, -4.7391e-01, -1.0824e-01,  1.7294e-02,\n",
       "           3.7803e-01, -1.7516e-01,  1.4617e-01,  9.3564e-02,  5.1762e-02,\n",
       "          -1.0258e-01, -4.3980e-01,  2.5771e-01, -2.7210e-01, -6.9631e-02,\n",
       "           6.5477e-03, -1.9615e-02,  1.9079e-01, -1.7933e-01, -2.3837e-01,\n",
       "          -1.9930e-01, -2.1222e-01,  6.1852e-01, -5.4897e-02,  1.2334e-01,\n",
       "           4.4780e-01, -1.9773e-01, -1.8060e-01,  4.0610e-01, -5.2460e-01,\n",
       "          -3.0208e-01,  3.2564e-01, -1.6133e-01, -3.9139e-02,  4.6447e-01,\n",
       "          -2.2123e-01, -2.0948e-01,  2.0962e-01,  7.5093e-01,  3.3972e-01,\n",
       "          -5.8285e-01, -9.3099e-02,  6.2323e-01,  2.4291e-01,  1.7725e-01,\n",
       "           1.6241e-01,  5.6225e-01,  4.7793e-02,  1.1080e-01, -2.3500e-01,\n",
       "          -1.3642e-02, -2.5257e-01, -9.9774e-03,  7.2926e-02, -5.7473e-01,\n",
       "           4.4167e-01,  4.2296e-01,  6.9265e-02,  1.9538e-01, -1.1135e+01,\n",
       "           2.0286e-01,  7.4167e-01, -5.5786e-01,  2.8318e-01,  3.5289e-02,\n",
       "           1.7607e-01, -4.3367e-01,  2.2193e-01, -2.8061e-01, -2.5162e-01,\n",
       "          -4.8441e-01, -8.6282e-02,  6.4972e-02, -4.8148e-01, -4.9852e-02,\n",
       "          -3.3706e-01, -8.4769e-01,  1.0731e-01, -3.6594e-01, -3.2308e-02,\n",
       "          -1.8092e-01,  2.7681e-02,  2.1419e-01, -2.3731e-01, -2.0185e-01,\n",
       "          -6.5991e-02,  2.4430e-01,  2.2678e-01,  2.4254e-01,  5.8909e-03,\n",
       "           4.4140e-01,  5.7786e-02,  2.8933e-01, -1.8013e-01, -6.6111e-01,\n",
       "          -3.4821e-02, -1.9609e-01, -1.2656e-02, -3.7478e-01, -4.3611e-01,\n",
       "          -3.0722e-01, -7.9103e-01,  1.6418e-01,  2.1416e-01,  3.6717e-01,\n",
       "           9.9375e-01,  3.7825e-01,  2.6112e-01,  1.0547e-01, -1.1805e-01,\n",
       "          -5.3074e-01,  1.4100e-01, -3.0387e-01, -4.3404e-01, -4.0790e-01,\n",
       "           8.4178e-01, -7.2825e-01, -1.7224e-01, -3.4755e-01, -1.0659e-01,\n",
       "           2.0123e-04,  5.3772e-03,  3.9160e-01, -1.8366e-01, -3.6271e-01,\n",
       "           1.9319e-01,  1.9980e-01,  2.8366e-01, -3.1343e-01,  3.5703e-02,\n",
       "           3.4901e-02,  6.7048e-01,  2.7765e-01,  2.3783e-01, -4.2750e-01,\n",
       "          -1.3814e-01, -1.2845e-01,  1.6125e+00, -4.5635e-02,  1.5336e-01,\n",
       "           5.1424e-01, -9.0454e-02, -4.1793e-01, -5.4065e-01, -1.1638e-01,\n",
       "           9.7362e-02, -8.1557e-02,  1.9645e-01,  2.7584e-01, -1.9937e-01,\n",
       "          -7.5613e-01, -1.7336e-01,  3.1572e-01,  6.1703e-01,  9.3973e-01,\n",
       "          -3.2887e-01,  2.9192e-01, -7.3956e-02,  7.2574e-03, -8.0817e-01,\n",
       "          -5.8342e-02, -1.7065e-01, -4.2599e-01,  1.5810e-01,  3.0947e-01,\n",
       "           6.8128e-02,  1.6617e-01,  2.9773e-02,  2.3819e-01, -4.8885e-01,\n",
       "           6.3676e-01, -2.3213e-01, -1.2480e-01, -1.6853e-01, -5.9609e-01,\n",
       "           3.3569e-01,  2.3408e-01, -3.0715e-01, -4.9486e-01,  4.0774e-03,\n",
       "           1.3571e-01,  6.1237e-02,  2.7703e-01, -5.6222e-01, -2.2088e-02,\n",
       "          -3.6937e-01, -2.7228e-02,  2.0643e-01,  4.2180e-01,  4.3910e-01,\n",
       "           1.0062e-01, -3.3001e-01,  8.0506e-03, -7.6621e-02, -6.0333e-01,\n",
       "          -7.6842e-01,  5.2483e-02,  3.5788e-01, -5.1324e-01,  1.6022e-01,\n",
       "          -2.8137e-01,  9.0422e-02, -3.0854e-01, -4.8027e-04,  3.4970e-01,\n",
       "           1.2397e-01,  3.4243e-01, -4.5443e-02,  5.2898e-01, -2.6764e-01,\n",
       "          -3.8265e-01,  4.3731e-01, -2.6012e-01, -3.3909e-02, -3.4807e-01,\n",
       "          -3.5774e-01,  2.4087e-02, -1.0873e-01,  3.1173e-02, -1.2983e-01,\n",
       "           5.6445e-02, -1.7827e-01, -3.4386e-01,  2.1349e-01,  5.5260e-01,\n",
       "           1.5133e-01, -7.2217e-01, -5.7344e-02, -2.4651e-01, -6.4310e-02,\n",
       "           6.6578e-01,  5.8575e-02, -3.9637e-02, -2.7925e-01, -1.3636e-01,\n",
       "          -4.4550e-02,  6.0860e-01,  1.0592e-01, -2.8721e-01,  2.4798e-01,\n",
       "           6.6968e-01,  3.0196e-02, -2.3565e-01, -5.1461e-01,  5.8945e-01,\n",
       "          -3.7866e-01, -1.8670e-01,  2.3920e-01, -2.3182e-01,  5.6058e-01,\n",
       "           1.4324e-01,  2.5307e-01,  1.5757e-01,  7.4106e-01,  3.7763e-01,\n",
       "          -4.7501e-02, -1.5690e-01, -5.1541e-01, -2.8358e-01,  4.8652e-01,\n",
       "           1.2928e-01,  5.5013e-01,  1.7140e-02,  1.9741e-02, -9.0398e-02,\n",
       "           1.7304e-01,  1.0681e+00, -4.1443e-01, -1.4173e-01,  8.2366e-01,\n",
       "           5.1697e-01,  2.6849e-01,  3.8529e-01,  3.7897e-01, -7.2954e-02,\n",
       "           3.2763e-01,  8.7677e-02, -1.5105e-01, -1.4170e-02,  2.8399e-01,\n",
       "           8.0565e-01, -9.3313e-01, -4.2770e-01,  2.8678e-01,  3.3921e-01,\n",
       "          -4.5222e-01,  3.2071e-01, -1.0564e-01,  5.3785e-01,  6.0226e-01,\n",
       "          -9.2450e-01,  7.6856e-02, -4.8478e-02, -8.6367e-02, -3.2746e-02,\n",
       "          -6.1377e-01,  1.5504e-01, -2.4203e-01, -2.3513e-01,  5.1324e-01,\n",
       "           4.3365e-01, -7.6108e-01,  2.4304e-01,  7.4847e-01, -2.1061e-01,\n",
       "           4.8360e-02, -3.4934e-02, -4.9522e-02,  9.1551e-02,  4.5118e-01,\n",
       "          -1.7003e-01,  2.0422e-01, -7.0104e-01, -6.6338e-01, -1.2292e-02,\n",
       "           5.1639e-01,  1.0963e-01, -1.5116e-01, -5.0165e-02,  4.1024e-01,\n",
       "           9.9256e-03, -6.3158e-01, -2.5607e-01,  1.2468e-01,  9.7188e-02,\n",
       "           2.0708e-02, -1.6027e-01, -3.4336e-02,  2.0147e-01, -5.6954e-01,\n",
       "           4.3899e-01,  3.8414e-01,  9.0014e-01,  3.3429e-02,  3.2523e-02,\n",
       "          -1.1951e-01,  2.7607e-01,  2.3741e-02,  3.9122e-01, -1.2040e-01,\n",
       "          -2.6997e-01,  1.8353e-01,  1.1499e-01, -1.8602e-01, -2.3663e-01,\n",
       "          -2.9987e-01,  5.9091e-01, -6.6580e-02, -2.3584e-01,  1.0610e-01,\n",
       "          -1.6710e-01,  2.2672e-02, -8.1102e-01,  6.6123e-01,  1.9448e-01,\n",
       "          -2.4137e-01,  6.3501e-01,  3.9104e-01,  5.4130e-01,  1.1730e-01,\n",
       "           6.2250e-02, -9.2180e-01, -2.8666e-01, -2.4363e-01, -7.8691e-02,\n",
       "           1.4656e-01, -2.6406e-02, -1.9603e-01, -2.1064e-03,  9.4790e-01,\n",
       "           5.9903e-01,  1.2704e-01, -7.3067e-01, -4.0456e-01,  3.7387e-01,\n",
       "          -1.1514e-01, -1.0457e-01,  1.8444e-01,  1.1284e-01,  5.3369e-01,\n",
       "          -7.9783e-01, -2.4131e-01, -1.6872e-01,  1.9377e-01, -2.8912e-01,\n",
       "           6.1067e-01, -2.6688e-01,  1.5861e-01,  2.7041e-01,  1.1484e-01,\n",
       "           3.1460e-01,  4.5037e-01,  5.2031e-02, -2.4999e-01,  7.7659e-02,\n",
       "           1.0046e-01, -3.9149e-01,  3.0587e-01, -1.0591e+00,  2.5856e-01,\n",
       "          -1.1493e-02,  2.4918e-01,  3.2559e-01]], device='mps:0'),\n",
       " tensor([[-0.2258, -0.7099, -0.3778,  ...,  0.3292,  0.7059,  0.7494],\n",
       "         [-0.0932, -0.6293, -0.4879,  ...,  0.3041, -0.1958,  0.5576],\n",
       "         [-0.3292, -0.2590, -0.5016,  ..., -0.0148,  0.3509,  0.1607]],\n",
       "        device='mps:0'),\n",
       " tensor([[-4.0519e-01, -6.3044e-01, -4.6265e-01, -9.2112e-01, -3.1882e-01,\n",
       "          -2.2368e-01, -5.4733e-01,  3.8875e-02,  1.0271e-01,  4.8600e-01,\n",
       "           3.2953e-01,  4.6674e-01, -1.6412e-01, -4.2071e-01, -1.0761e+00,\n",
       "           9.7057e-02, -4.2540e-01,  9.3580e-02, -1.5000e-02,  3.4832e-01,\n",
       "           3.4745e-01,  2.8814e-01, -5.7310e-02,  2.3828e-01,  1.0276e-01,\n",
       "           1.7826e-01, -1.1325e-01, -2.7233e-01,  6.1170e-01, -3.7864e-01,\n",
       "          -1.8277e-01,  7.6179e-01, -1.2412e-01,  8.7277e-01, -6.0611e-02,\n",
       "           1.0580e-01, -3.0531e-01, -2.1279e-01, -1.0336e-01, -1.1636e-01,\n",
       "           4.8100e-01,  1.5693e-01, -2.3699e-01,  3.3551e-01,  2.4230e-01,\n",
       "           2.5686e-01,  2.2507e-01, -9.0692e-01,  6.9052e-01, -1.7300e-01,\n",
       "          -4.9322e-01,  5.1016e-02,  2.2913e-01,  2.6046e-01, -8.3776e-01,\n",
       "          -4.1498e-01, -1.1455e-01, -3.8839e-02, -3.0751e-01,  4.4343e-01,\n",
       "           3.3365e-01,  5.1028e-01,  2.9545e-02, -9.7963e-02,  2.8331e-01,\n",
       "           1.3938e-01, -2.2146e-01,  5.8028e-03, -2.7691e-01,  8.7652e-02,\n",
       "           1.7582e-01,  1.3314e-01, -6.1468e-02, -1.6493e-02,  4.1835e-01,\n",
       "          -6.4583e-02,  4.7788e-01, -2.3830e-01, -1.4635e-01, -2.1019e-01,\n",
       "           3.2396e-01,  4.3350e-01,  6.9233e-02, -8.2866e-02, -4.8881e-01,\n",
       "           1.4828e-03,  3.8539e-01,  2.8070e-01,  4.7296e-02, -1.0841e-04,\n",
       "           4.8536e-01,  2.0798e-01, -4.5844e-01, -2.0316e-02,  2.4024e-01,\n",
       "          -5.8213e-02,  2.0751e-01, -3.2783e-02, -1.2371e-01, -1.5039e-01,\n",
       "           1.0813e-01, -1.6912e-01,  2.9832e-01,  2.7427e-01, -3.4202e-01,\n",
       "          -4.9598e-01,  3.2974e-01, -2.3432e-02, -3.6303e-01, -3.3959e-01,\n",
       "           1.0144e-01, -1.1390e-01,  6.3676e-02, -4.7652e-01,  2.1496e-01,\n",
       "          -1.4658e-01, -1.3876e-01, -2.6988e-01,  6.6310e-03,  4.8937e-01,\n",
       "          -1.0744e-01, -2.9503e-01,  3.5105e-02,  3.6529e-02,  3.1180e-01,\n",
       "          -3.9693e-02,  6.8859e-01,  8.4773e-02,  1.0380e-01,  8.3254e-01,\n",
       "           7.9135e-02,  3.1605e-01, -5.0363e-02, -3.4829e-01,  7.9164e-02,\n",
       "           1.5564e-01, -2.6330e-01, -2.7233e-01, -1.4743e-01, -2.2156e-01,\n",
       "           1.0501e-01,  8.8833e-02, -5.4985e-02, -5.7953e-02, -1.6896e-01,\n",
       "          -1.7709e-01, -4.8510e-01, -2.9331e-01, -5.5636e-03, -4.4200e-01,\n",
       "          -8.6136e-01, -1.0712e+00,  7.4881e-01, -3.2443e-01,  1.0677e-01,\n",
       "          -1.3538e-01,  2.5806e-01,  5.6206e-02, -1.4961e-01,  1.1207e-02,\n",
       "           5.3439e-01, -2.4506e-01, -7.9671e-02, -4.2483e-01,  4.5117e-01,\n",
       "           3.7957e-01,  7.0318e-02, -2.8659e-02,  2.0846e-01,  4.1933e-02,\n",
       "           1.8168e-01, -1.9826e-01,  1.3482e-01, -2.2869e-01,  7.8072e-02,\n",
       "          -5.9743e-01,  5.1902e-01, -8.9218e-02,  1.1670e-01,  9.8726e-02,\n",
       "           5.1811e-01, -1.9754e+00, -2.8972e-02, -9.4929e-01,  1.7522e-01,\n",
       "          -6.4876e-01,  3.5160e-01,  1.2334e-01,  8.5743e-02,  2.0981e-02,\n",
       "          -2.8968e-01, -6.1813e-02, -4.8072e-02, -2.0393e-01,  1.1793e-01,\n",
       "           5.7695e-02,  5.5744e-01,  2.3281e-01, -9.6790e-02, -2.3336e-02,\n",
       "          -1.7320e-01,  3.1704e-01, -1.7289e-01,  4.3560e-01, -2.0455e-01,\n",
       "           2.7290e-01, -1.3742e-01,  3.1316e-01,  4.2805e-01, -3.6076e-02,\n",
       "           9.1425e-02, -1.6348e-01, -4.6478e-01, -2.4903e-01, -5.2770e-01,\n",
       "           3.2072e-01, -5.2051e-01, -2.2083e-01,  5.8058e-01,  4.1399e-01,\n",
       "          -2.7845e-01, -5.2997e-02,  1.0278e-01, -1.5918e-01,  1.1199e-01,\n",
       "           2.6277e-01, -1.0496e-01,  2.2681e-01, -2.0791e-01, -1.7681e-01,\n",
       "           3.9295e-03, -5.3911e-01,  5.4070e-01,  5.4823e-01,  1.4526e-01,\n",
       "          -2.2678e-01, -6.2293e-02,  2.9475e-01, -4.1056e-01, -3.2711e-01,\n",
       "           9.1270e-02, -3.7081e-01, -3.2061e-01, -9.4538e-02,  3.0280e-01,\n",
       "           3.5661e-01,  1.7740e-02,  1.1455e+00,  5.2520e-01,  5.7958e-01,\n",
       "          -3.2865e-02,  3.2974e-01,  5.6707e-01,  7.8516e-01, -4.9135e-01,\n",
       "           3.9926e-02, -8.6878e-04, -3.0882e-01,  2.0337e-01, -1.9899e-01,\n",
       "          -5.7013e-01,  2.7501e-01,  1.9550e-01,  3.9647e-01, -2.7849e-01,\n",
       "           5.5644e-01,  2.3385e-01, -1.2793e-01, -5.0902e-01,  1.1858e-01,\n",
       "          -2.0661e-01,  3.7407e-01, -4.9262e-01, -2.9859e-01, -7.7641e-01,\n",
       "          -6.3227e-01, -1.5551e-01, -1.5270e-01,  2.0309e-01, -4.6448e-01,\n",
       "          -8.2705e-01,  2.5213e-01, -3.9121e-01, -1.7718e-01, -1.4108e-01,\n",
       "           5.8843e-01,  1.6141e-01, -1.3853e-01, -1.9191e-01, -1.0288e-01,\n",
       "          -3.0307e-01,  8.8467e-02, -3.1752e-01,  5.6740e-02,  1.7095e-01,\n",
       "          -1.2733e-01,  2.2603e-01, -2.1488e-02, -3.1060e-03, -9.6538e-02,\n",
       "           1.1419e-01, -8.9718e-02,  7.5205e-03, -2.8019e-01, -2.0131e-01,\n",
       "          -2.1670e-02, -1.1021e-01,  2.3826e-01, -5.9554e-01, -2.9540e-01,\n",
       "          -1.3198e-01,  3.9024e-01, -4.7994e-02,  3.6842e-01, -3.2242e-01,\n",
       "          -3.0008e-01, -6.2838e-03, -5.4544e-01, -1.5268e-01, -2.4920e-01,\n",
       "           4.5840e-01, -3.1655e-02, -2.6059e-01,  4.0350e-03, -1.9539e-01,\n",
       "          -2.4828e-02,  2.1122e-01,  3.6059e-01,  2.9422e-01, -1.8928e-01,\n",
       "           9.2629e-02,  3.2521e-02, -2.8566e-01, -4.4422e-01,  6.9690e-02,\n",
       "           2.6297e-01,  2.6688e-01,  2.0073e-01, -6.4824e-02,  5.2401e-02,\n",
       "           5.2015e-01, -2.0428e-01, -1.4935e-01,  4.2226e-01, -3.4453e-01,\n",
       "          -1.2787e-01,  2.5283e-01, -2.7592e-02, -1.2916e-01,  8.0973e-02,\n",
       "           2.3298e-01,  1.3012e-01, -7.1959e-02, -6.6004e-02,  3.7704e-01,\n",
       "           2.3838e-01,  1.3186e-01, -4.1292e-01,  4.2844e-02,  5.0355e-01,\n",
       "          -4.5161e-01,  3.5346e-01, -4.0555e-01, -6.3862e-03, -1.8457e-01,\n",
       "           1.0285e-01,  3.5138e-01, -3.3568e-01,  9.5652e-02,  8.6645e-02,\n",
       "          -1.1318e-02, -8.4827e-02, -5.4669e-03,  4.0058e-01,  7.3107e-01,\n",
       "          -4.7431e-02, -1.2711e-01,  3.0843e-01, -6.5938e-02, -6.7675e-02,\n",
       "           1.0048e-01, -1.0804e-01, -1.3532e-02, -4.2834e-01, -2.0116e-01,\n",
       "          -1.7523e-01, -1.7655e-02, -9.7357e-03,  5.4736e-01,  7.6879e-02,\n",
       "           1.9909e-02, -7.4178e-02, -3.2699e-01,  8.8006e-01,  2.3775e-01,\n",
       "          -3.6362e-01,  1.7657e-01,  1.6882e-01,  4.4029e-01,  1.6759e-01,\n",
       "          -2.2554e-01, -3.4309e-01,  4.6120e-01, -2.0796e-01, -2.7290e-01,\n",
       "          -1.2138e-01,  2.4502e-02,  7.6028e-01, -1.9420e-01, -1.4337e-01,\n",
       "          -2.6925e-01,  5.9280e-01,  2.3677e-01,  2.8445e-01, -4.3607e-01,\n",
       "           3.6906e-01, -1.0004e-01,  2.0862e-01, -9.1337e-02, -1.7841e-01,\n",
       "           6.0674e-01,  1.5497e-01,  5.6019e-02,  4.6447e-01, -1.1670e+01,\n",
       "           2.1316e-01,  1.0044e+00, -4.2441e-01,  2.3723e-01, -3.5084e-01,\n",
       "           1.6266e-01, -3.9088e-01,  3.9057e-01, -2.7041e-01,  2.6029e-01,\n",
       "          -4.7681e-01, -3.0693e-02, -7.8948e-02, -3.7879e-01,  6.5810e-02,\n",
       "           1.4263e-01, -5.4908e-01,  4.8931e-02, -7.5297e-02,  5.5030e-01,\n",
       "          -1.6145e-01, -1.0550e-01,  1.3920e-01, -2.8869e-01, -6.0806e-02,\n",
       "          -7.5112e-02, -3.5925e-01, -3.5856e-02,  4.6406e-01,  1.7421e-01,\n",
       "          -8.1633e-02, -8.3174e-02,  2.4124e-01, -1.6907e-01, -5.1737e-01,\n",
       "           1.8314e-01, -4.1743e-01,  7.7319e-03, -2.3772e-01, -6.7522e-01,\n",
       "           1.4862e-01, -1.1346e-01,  2.6745e-01,  4.8366e-01,  6.2217e-01,\n",
       "           1.4887e+00,  5.7373e-01, -1.2124e-01,  6.6728e-02,  1.8626e-01,\n",
       "          -3.9722e-01,  1.5422e-01, -6.4141e-01,  3.5303e-01, -4.0775e-01,\n",
       "           1.6880e-01, -6.4516e-01,  5.8113e-01, -3.3298e-01, -1.4347e-01,\n",
       "          -2.4872e-01,  2.0975e-02,  2.0838e-01, -6.3175e-01, -2.4327e-01,\n",
       "           6.6979e-01, -4.7053e-02,  1.0726e-01, -4.4218e-01,  3.4515e-01,\n",
       "          -1.3407e-01,  9.7497e-02, -1.4254e-02, -2.2688e-01,  2.1465e-01,\n",
       "          -1.8210e-01,  1.1210e-01,  3.3839e-01, -5.1140e-01,  1.7073e-02,\n",
       "           4.4521e-02, -8.7041e-03, -1.4844e-01, -6.5688e-01,  2.2381e-01,\n",
       "           4.3185e-01, -3.6685e-01,  1.9371e-01,  2.1993e-01, -1.2193e-01,\n",
       "          -1.8560e-01,  1.7939e-01, -1.7014e-01,  2.2387e-01,  7.3626e-01,\n",
       "          -6.6589e-01,  3.1301e-01,  1.1072e-01, -9.4234e-02, -2.5234e-01,\n",
       "          -2.5555e-02,  3.2307e-02, -4.1414e-01,  1.0649e-01, -2.0766e-01,\n",
       "          -2.3015e-02, -4.9307e-02, -4.7319e-02,  3.5312e-01, -1.7873e-01,\n",
       "          -1.1606e-01,  1.9784e-01,  1.9468e-01, -5.3724e-01, -3.8092e-01,\n",
       "           3.3847e-01,  4.5813e-01, -1.5359e-01, -5.4885e-01, -3.8023e-01,\n",
       "          -1.1509e-01,  2.0008e-01, -1.1556e-01, -6.5272e-01,  3.3497e-01,\n",
       "          -7.7484e-01,  5.6431e-02, -1.8212e-01, -1.2103e-01,  4.6564e-01,\n",
       "           4.2762e-02, -1.1359e-01,  2.5397e-01, -1.3684e-01, -3.4225e-01,\n",
       "          -8.9286e-01, -7.9506e-02,  2.7247e-01,  1.1756e-01,  3.0920e-01,\n",
       "          -9.7669e-03, -1.1863e-01, -6.6349e-02,  3.4986e-01,  3.7660e-01,\n",
       "           4.0759e-01,  4.2583e-01, -4.8843e-01, -2.4300e-01, -3.3644e-02,\n",
       "          -2.1451e-01,  2.4837e-02,  3.3594e-02, -1.6138e-01, -3.5715e-01,\n",
       "           1.5855e-01, -7.7677e-02, -3.2648e-02, -4.4425e-01,  2.4986e-01,\n",
       "           4.0657e-01, -2.0526e-01,  1.7007e-01,  4.3908e-01,  5.9884e-01,\n",
       "          -2.4765e-03, -2.6506e-01, -6.2403e-01,  1.6140e-02,  6.3243e-02,\n",
       "           8.6495e-02, -9.0306e-02, -7.9234e-01, -2.2603e-01, -2.0982e-01,\n",
       "           3.4784e-02,  9.2012e-01, -3.3546e-01, -5.3215e-01, -1.6513e-01,\n",
       "           4.1247e-01,  2.1558e-01, -6.2373e-02,  1.6351e-01, -1.0045e-01,\n",
       "           1.4053e-01, -5.1129e-01,  3.5793e-01, -8.9282e-02,  1.9397e-01,\n",
       "           2.3136e-02, -7.5718e-02,  3.9328e-01,  2.9840e-01, -3.2549e-01,\n",
       "          -1.2685e-01,  1.2989e-01, -3.4913e-01, -1.5521e-01,  4.1199e-01,\n",
       "          -1.6520e-01,  4.8981e-01, -3.3991e-01, -1.5773e-01,  6.1706e-02,\n",
       "           2.2006e-01,  1.1130e+00,  2.8482e-02,  1.2191e-01,  8.7698e-01,\n",
       "           3.4444e-01, -3.9573e-01,  4.1453e-01,  8.5057e-02,  7.6325e-02,\n",
       "           3.9524e-01, -4.7851e-02, -4.7779e-01,  3.3503e-01,  7.2429e-01,\n",
       "           4.2506e-01, -3.5616e-01, -2.2155e-01, -3.4641e-01,  5.2915e-01,\n",
       "           4.0373e-01,  6.8163e-01, -2.6724e-01,  6.7264e-02,  7.4255e-02,\n",
       "          -7.2849e-01, -3.3019e-02, -2.0992e-01,  6.4544e-01, -1.7928e-01,\n",
       "          -5.6282e-01, -3.5586e-01, -4.5909e-02, -4.6876e-01,  1.6193e-01,\n",
       "           5.2828e-02, -1.1132e+00,  8.3679e-02,  3.9410e-01, -7.7972e-01,\n",
       "           1.9544e-01, -2.8802e-01, -2.0574e-01,  4.1083e-02, -1.4512e-01,\n",
       "          -2.2486e-01, -1.2160e-01,  1.5579e-01, -6.5767e-01,  3.2587e-01,\n",
       "           2.6531e-01, -3.3304e-01,  9.7220e-02,  3.3708e-01, -6.2614e-01,\n",
       "          -2.1650e-01,  2.5058e-01, -3.5112e-01,  2.7250e-02,  9.3970e-02,\n",
       "          -1.5947e-01, -5.1569e-01, -2.1392e-01,  5.4007e-01, -2.6709e-01,\n",
       "           8.6575e-02,  1.9618e-01,  4.0731e-01, -2.0245e-01, -5.2603e-01,\n",
       "           7.9394e-02,  4.7248e-01, -3.8250e-01,  4.8677e-01, -1.0742e-01,\n",
       "          -2.7399e-01, -1.2996e-02, -8.2594e-02, -5.0434e-01,  7.8137e-02,\n",
       "          -1.0177e-01,  1.9676e-01, -5.0390e-01, -4.7724e-02,  6.4325e-02,\n",
       "           1.2647e-01,  2.1433e-01, -5.6631e-01, -1.9235e-01, -3.2375e-02,\n",
       "           1.2962e-01,  4.2762e-01,  2.2932e-01,  4.5193e-01, -2.8995e-01,\n",
       "          -9.4393e-02, -3.9791e-01, -5.0841e-02,  3.9182e-01, -5.3317e-02,\n",
       "          -4.7638e-02, -3.9213e-01,  3.4377e-01,  3.6661e-01,  6.8653e-01,\n",
       "           4.1427e-01, -4.6671e-02, -4.9900e-01, -6.9831e-01, -1.3360e-01,\n",
       "          -1.8370e-01,  3.4862e-02, -3.7386e-03,  6.0978e-03, -1.5733e-02,\n",
       "          -6.3622e-01,  2.3358e-02, -1.8573e-03,  3.0058e-01, -7.6805e-02,\n",
       "           3.3387e-01, -2.1266e-01, -5.4469e-02,  4.9698e-01,  6.1546e-02,\n",
       "           1.7462e-01,  3.4190e-01, -7.9827e-02, -3.7012e-01, -2.5301e-01,\n",
       "           2.9549e-01, -3.9718e-01,  2.8048e-02, -6.0159e-01,  4.4688e-01,\n",
       "           1.7684e-02,  2.0551e-01,  1.8319e-01]], device='mps:0'),\n",
       " tensor([[-0.2344, -0.5215, -0.7335,  ...,  0.3040, -0.1277,  0.6510],\n",
       "         [-0.2974, -0.5650, -0.8205,  ...,  0.3798,  0.1095,  0.6311],\n",
       "         [-0.2344, -0.5215, -0.7335,  ...,  0.3040, -0.1277,  0.6510],\n",
       "         [-0.1133, -0.3282, -0.6784,  ...,  0.1702,  0.2561,  0.4909],\n",
       "         [-0.2344, -0.5215, -0.7335,  ...,  0.3040, -0.1277,  0.6510]],\n",
       "        device='mps:0'),\n",
       " tensor([[-0.2519, -0.4005, -0.5801,  ...,  0.1298,  0.4928,  0.6750],\n",
       "         [-0.1672, -0.5249, -0.4540,  ...,  0.3771,  0.2016,  0.5340],\n",
       "         [-0.3292, -0.2590, -0.5016,  ..., -0.0148,  0.3509,  0.1607],\n",
       "         [-0.2185, -0.9200, -0.8595,  ...,  0.1888,  0.7993,  0.7992],\n",
       "         [-0.0548, -0.3751, -0.4766,  ...,  0.0464,  0.8423,  0.4111],\n",
       "         [-0.0583, -0.6757, -0.8037,  ...,  0.1694,  0.7852,  0.5016]],\n",
       "        device='mps:0'),\n",
       " tensor([[-1.7614e-01, -5.8749e-01, -3.5802e-01, -1.0230e+00, -2.7551e-01,\n",
       "          -4.0262e-02, -8.5206e-01,  2.0346e-01,  3.3503e-01,  3.4728e-01,\n",
       "           2.1668e-01,  4.0528e-01, -6.9469e-02,  1.4801e-02, -5.1288e-01,\n",
       "          -2.2614e-01,  1.6378e-01,  2.5336e-01,  6.8510e-01,  1.1683e-01,\n",
       "           1.8234e-01,  4.4697e-01, -1.5832e-01, -4.9634e-02, -3.6537e-01,\n",
       "           1.9070e-01, -1.9841e-01,  2.3795e-01, -7.8445e-03, -4.6238e-01,\n",
       "           2.5665e-01,  3.4648e-01, -4.2807e-01,  1.4162e+00, -2.9775e-01,\n",
       "          -2.1308e-01,  8.2004e-02, -2.9639e-01, -3.0269e-01,  1.1109e-01,\n",
       "           5.9814e-01, -1.9909e-01, -2.2982e-01,  3.1273e-01, -3.4196e-02,\n",
       "           1.8301e-01,  1.0062e-01, -2.9022e-01,  5.0568e-01, -5.9002e-01,\n",
       "          -1.3629e-01, -3.7746e-01,  2.1880e-01,  3.8247e-01, -4.6266e-01,\n",
       "          -2.9720e-01,  2.1812e-01,  1.4988e-01, -2.1488e-01,  1.1886e-01,\n",
       "          -7.9123e-02,  3.9341e-01, -1.2118e-01,  1.8456e-01,  1.6926e-01,\n",
       "           3.4797e-01, -5.0807e-01, -6.5750e-01, -6.4595e-01, -1.6275e-02,\n",
       "           1.5502e-01,  3.0507e-01,  9.2226e-02, -2.7995e-02,  2.9219e-01,\n",
       "          -4.9067e-01,  2.0200e-01, -1.4132e-02, -3.6383e-01,  2.4186e-01,\n",
       "          -2.2957e-01,  4.2885e-01,  3.1956e-01,  2.0169e-01,  2.0612e-01,\n",
       "          -5.6283e-02,  2.9078e-01, -3.0172e-02,  1.8520e-02,  3.8230e-01,\n",
       "          -1.1129e-01,  8.7467e-02, -8.0353e-02, -1.4495e-01,  1.0699e-01,\n",
       "           1.8117e-01,  2.1215e-01, -2.6830e-01, -9.8620e-02, -2.2677e-01,\n",
       "           1.4849e-01, -3.3044e-02, -8.9273e-02,  2.7185e-01, -4.5808e-01,\n",
       "          -5.8438e-01,  9.1718e-02, -3.0031e-01, -4.5676e-01,  3.0003e-02,\n",
       "          -3.0395e-01, -1.9434e-01,  1.5160e-01, -2.2299e-01,  2.1092e-01,\n",
       "          -1.1786e-01, -5.5463e-02,  5.4748e-02, -1.5349e-01,  1.4516e-02,\n",
       "          -1.0023e-01, -2.7571e-01,  1.9385e-01,  4.4648e-02,  1.2932e-01,\n",
       "          -4.6716e-02,  6.2585e-01,  4.2309e-02,  1.2413e-01,  4.9745e-01,\n",
       "           1.4654e-01, -8.2920e-02, -2.6324e-01, -2.1829e-01, -4.2277e-02,\n",
       "           2.5880e-01,  7.7860e-02, -1.9708e-02, -1.8201e-01, -2.1863e-01,\n",
       "           2.8411e-01,  3.9164e-02, -3.1539e-01, -2.0375e-01, -2.3885e-01,\n",
       "          -3.8593e-01, -2.5494e-01, -2.0587e-01, -5.0981e-01, -4.4611e-01,\n",
       "          -8.2060e-01, -5.5616e-01,  6.4001e-01, -1.9643e-01, -2.3516e-01,\n",
       "          -1.3706e-01,  2.5442e-01,  1.4085e-02,  4.1483e-02,  3.3790e-01,\n",
       "           6.3541e-01, -2.7567e-01, -4.8947e-02, -3.7977e-01, -7.3702e-02,\n",
       "           6.9975e-02,  1.8970e-01, -7.8576e-02,  1.3551e-01,  5.4505e-02,\n",
       "           1.8973e-01, -1.9960e-01,  5.8888e-01, -1.7729e-01,  1.2618e-02,\n",
       "          -8.4935e-01,  4.9333e-01,  1.9954e-01, -1.6157e-01,  1.1710e-01,\n",
       "           3.3931e-01, -1.1931e+00,  1.2620e-01, -1.0227e+00,  3.2824e-01,\n",
       "          -1.7098e-01,  1.8095e-01,  3.2597e-01,  2.7577e-01,  5.5632e-02,\n",
       "           2.1186e-01, -9.9121e-03,  3.2520e-01,  3.9732e-01,  2.0288e-02,\n",
       "           2.5561e-01,  3.7830e-01,  3.9603e-01,  2.3333e-01,  8.7525e-02,\n",
       "          -4.6283e-01,  3.5844e-01,  5.1188e-01,  9.6246e-02,  1.9932e-01,\n",
       "           1.3285e-01, -1.7003e-01,  1.8194e-01, -4.7599e-03, -2.2653e-02,\n",
       "          -2.4232e-01, -2.7171e-01, -5.3104e-01, -5.0171e-02, -4.5602e-01,\n",
       "           1.8974e-01, -4.4759e-01, -2.3165e-01,  4.9004e-01,  2.0560e-01,\n",
       "          -5.8674e-03,  2.9285e-01,  1.7700e-01, -1.9742e-01,  1.7903e-01,\n",
       "           3.6679e-01, -3.7862e-01,  2.1561e-03, -1.9138e-01, -2.0779e-01,\n",
       "           4.1036e-01, -8.3581e-01,  5.2290e-01,  5.0503e-01,  1.8949e-01,\n",
       "           1.5468e-01, -1.6453e-01,  1.0726e-01, -6.0864e-01,  3.0500e-01,\n",
       "          -3.3508e-01, -2.8021e-01, -6.3127e-01, -3.2163e-02,  2.6321e-01,\n",
       "           2.0939e-01,  3.2734e-03,  7.8382e-01,  5.2561e-01,  4.9839e-01,\n",
       "          -2.4440e-01, -7.1029e-02,  3.1526e-01,  2.4145e-01, -5.7815e-01,\n",
       "           3.1173e-01, -1.9860e-01, -4.2880e-01,  2.8787e-01, -2.2835e-01,\n",
       "          -5.3193e-02, -8.2633e-02,  2.8505e-01,  3.9954e-01, -1.8936e-01,\n",
       "          -2.1729e-01,  6.3627e-02,  6.5801e-02, -3.3103e-01,  5.3294e-01,\n",
       "          -3.3218e-01,  3.3568e-01, -1.2785e-01, -4.5420e-01, -6.0830e-01,\n",
       "          -3.2012e-01,  7.5990e-03, -1.8086e-01,  1.3518e-01, -6.0369e-01,\n",
       "          -5.3184e-01,  2.9145e-01,  7.0159e-02,  4.4066e-02,  9.3039e-02,\n",
       "           2.9553e-01, -3.2811e-01, -1.8485e-01, -2.0442e-01, -4.3013e-01,\n",
       "          -1.0626e-01,  5.5425e-02, -6.8973e-01,  2.1155e-01,  3.6262e-01,\n",
       "           2.3407e-02,  4.3709e-01,  4.8475e-01, -1.9884e-01, -3.9754e-01,\n",
       "          -1.5716e-01,  2.4385e-01, -2.5395e-01, -1.9611e-01,  2.6151e-01,\n",
       "           3.6015e-02,  4.6225e-01,  5.5650e-01, -3.5037e-02, -4.3253e-01,\n",
       "           1.8470e-01,  7.1793e-01, -1.9044e-01, -2.0873e-01,  6.0083e-03,\n",
       "          -9.7548e-01, -7.8421e-02, -9.7703e-02, -4.0406e-01, -6.5467e-01,\n",
       "          -1.0242e-01,  1.6247e-01,  4.1146e-03, -8.4313e-02, -3.5131e-01,\n",
       "           1.1079e-01,  4.8121e-01,  1.0656e-01,  3.6841e-01, -1.3932e-01,\n",
       "          -4.4117e-01, -1.3803e-01, -4.1224e-01, -5.8410e-01, -2.4412e-01,\n",
       "           1.9565e-01,  4.2279e-01,  2.8224e-02, -1.8967e-01, -2.9446e-01,\n",
       "           3.8764e-01, -3.5967e-01, -1.8551e-01,  2.1402e-01, -3.6530e-01,\n",
       "          -3.4331e-02,  3.9263e-01,  4.4703e-01,  9.3187e-02,  1.3090e-02,\n",
       "          -2.0790e-02, -2.7593e-01, -4.0255e-01, -2.9048e-01,  6.3862e-01,\n",
       "           5.5537e-01,  2.4306e-01, -4.3878e-01,  3.5191e-01,  4.3891e-01,\n",
       "          -3.2233e-01,  3.5813e-01, -1.9689e-01, -6.6888e-02, -1.4280e-01,\n",
       "           2.6177e-01,  5.1533e-01, -1.7291e-01,  2.9446e-01,  2.7587e-01,\n",
       "          -2.5831e-01, -1.6174e-01, -1.3777e-01,  3.2027e-01, -2.0562e-02,\n",
       "          -1.6339e-01, -2.1425e-01,  1.6837e-01, -1.3232e-03, -3.5641e-01,\n",
       "           2.8122e-01,  8.1080e-02, -1.5214e-01, -2.0127e-01, -6.3402e-01,\n",
       "          -1.0510e-03, -4.2574e-01,  5.4687e-01,  8.5314e-02, -3.6663e-01,\n",
       "           3.7945e-01, -1.1153e-01, -4.6167e-01,  6.0215e-01,  1.0126e-02,\n",
       "          -4.6889e-01,  5.1139e-01, -1.7366e-01,  1.7471e-01,  2.6800e-01,\n",
       "          -9.6391e-02,  1.4733e-01,  3.4559e-02,  6.0019e-01,  4.3388e-01,\n",
       "          -1.5452e-01, -4.1834e-02,  2.4696e-01, -4.0157e-01,  4.8827e-01,\n",
       "          -1.4159e-01,  4.4179e-01,  3.5865e-02,  5.5619e-02, -1.3735e-01,\n",
       "           2.4115e-01, -4.1934e-02, -3.8763e-01, -4.3656e-01, -3.2445e-01,\n",
       "           4.1797e-01,  1.4815e-01,  3.0259e-01,  2.0858e-01, -1.1440e+01,\n",
       "           2.8408e-01,  8.6672e-01, -4.1510e-01, -1.0259e-01, -1.7257e-01,\n",
       "           7.8554e-03, -3.5723e-01,  5.1084e-01, -5.2429e-02, -4.5512e-01,\n",
       "          -4.7200e-01, -2.1718e-01,  2.3182e-01, -5.0717e-01, -1.0071e-01,\n",
       "           1.5686e-01, -6.4411e-01,  2.6426e-01, -8.6464e-02,  7.7427e-01,\n",
       "          -4.3489e-01,  6.6019e-02,  2.2426e-01, -2.9222e-01,  3.6921e-02,\n",
       "           1.8945e-01,  5.6739e-02,  3.2735e-01,  4.3975e-01, -5.4637e-02,\n",
       "           1.0907e-01, -1.2817e-01, -7.2709e-02, -4.7155e-01, -4.5018e-01,\n",
       "           1.0718e-01, -4.5356e-01,  6.1574e-01, -1.7738e-01, -6.1719e-01,\n",
       "          -7.3560e-02, -3.4541e-01,  2.8051e-01,  2.9518e-01,  6.7493e-01,\n",
       "           9.7734e-01,  9.3128e-01,  2.2472e-01,  3.6179e-01,  3.3649e-02,\n",
       "          -7.5636e-01,  1.8178e-01, -2.7020e-01, -1.2551e-01,  1.2280e-01,\n",
       "           3.8567e-01, -8.0513e-01,  1.6771e-01, -4.3596e-01,  3.7745e-02,\n",
       "           7.0978e-01,  3.5794e-02,  1.4096e-01, -3.6255e-01, -1.7046e-01,\n",
       "           7.9639e-02, -1.9999e-01,  2.1605e-01, -4.5426e-01, -3.5187e-01,\n",
       "          -4.5643e-02,  2.7080e-01,  2.5588e-01, -9.5273e-02,  4.3303e-01,\n",
       "           3.6737e-01, -1.4664e-01,  1.2865e+00,  2.8703e-02, -7.5999e-02,\n",
       "          -2.2399e-02,  1.3058e-01, -4.7545e-01, -5.8969e-01, -1.6053e-01,\n",
       "          -6.7661e-02, -5.0352e-02,  5.8265e-02,  3.7852e-01, -5.6180e-02,\n",
       "          -5.5075e-01, -3.6072e-02,  1.1656e-01,  1.0534e-01,  7.4453e-01,\n",
       "          -8.2455e-01, -2.5613e-01, -1.8590e-01,  7.3763e-02, -8.7501e-01,\n",
       "          -4.2014e-02, -5.1090e-01, -7.6148e-01,  4.9889e-01, -1.2170e-01,\n",
       "           2.5119e-01, -2.0634e-01,  3.4217e-02,  5.0820e-01, -1.2499e-01,\n",
       "           7.4107e-02,  6.7300e-02,  2.7937e-01, -2.7528e-01, -3.6588e-01,\n",
       "           3.2383e-01,  4.1790e-01, -2.1081e-01, -6.4888e-01, -1.1401e-01,\n",
       "           3.2435e-01, -1.3542e-01,  4.9550e-01, -4.6173e-01,  4.4143e-02,\n",
       "          -9.5193e-01, -4.3019e-03, -3.9190e-03, -8.4407e-02,  4.1453e-01,\n",
       "           9.1924e-02, -2.7209e-01,  1.5418e-01, -6.9187e-01, -4.1589e-01,\n",
       "          -8.1779e-01, -2.4179e-01,  1.4407e-01, -5.2683e-01, -1.7039e-01,\n",
       "          -1.8392e-01, -1.1571e-01, -2.0320e-02,  6.5289e-02, -3.6723e-02,\n",
       "           3.3165e-01,  9.3338e-02, -2.7144e-01, -1.6026e-01,  1.8651e-02,\n",
       "          -4.1166e-01, -1.4496e-01, -1.9238e-01,  5.4924e-03, -7.0790e-01,\n",
       "          -2.3814e-01,  1.1461e-01,  6.8359e-02, -2.1760e-01, -2.8679e-02,\n",
       "           1.7137e-01,  1.2980e-01, -3.9035e-01,  9.6516e-01,  4.2760e-01,\n",
       "           1.0279e-01, -3.7809e-01, -3.2426e-01, -1.2709e-01,  2.3585e-01,\n",
       "           1.6038e-01,  6.5167e-02,  8.4681e-02, -5.6620e-01, -6.4375e-01,\n",
       "           1.9720e-01,  7.3751e-01, -1.9981e-01, -5.0881e-01, -1.4986e-02,\n",
       "           4.2746e-01, -1.5042e-02, -6.3733e-02,  2.1279e-01,  1.4412e-01,\n",
       "          -1.8094e-01, -4.2080e-01,  4.8719e-01, -2.7092e-01,  5.3509e-02,\n",
       "           4.3953e-01, -5.3768e-02, -1.6833e-02,  2.5855e-01,  3.1938e-01,\n",
       "          -3.1925e-01, -7.1982e-02, -4.9652e-01,  2.2127e-01,  4.7249e-02,\n",
       "          -5.1562e-01,  8.0227e-01,  7.0630e-02, -1.5259e-01, -3.8539e-02,\n",
       "           3.4457e-01,  9.7020e-01, -1.3363e-03,  2.0419e-01,  8.2567e-01,\n",
       "           4.2065e-02,  3.4566e-01,  3.6262e-01,  3.6471e-01,  2.9030e-01,\n",
       "           3.2097e-01, -1.0485e-01, -5.8949e-01,  4.7583e-03,  3.5014e-01,\n",
       "           1.7573e-01, -5.5299e-01, -6.0410e-01, -1.6108e-01,  7.3744e-01,\n",
       "           1.0074e-01,  4.1783e-01, -3.7119e-01,  5.1360e-01,  4.9028e-01,\n",
       "          -8.1544e-01, -1.1107e-01, -3.6174e-01,  8.6522e-02,  9.0050e-02,\n",
       "          -4.1330e-01,  2.5312e-02,  9.7974e-03, -8.6930e-02,  7.2329e-01,\n",
       "           4.8497e-01, -1.0391e+00, -1.3207e-01,  6.9199e-01, -5.0798e-01,\n",
       "           2.5290e-01, -4.6972e-01, -4.7001e-02,  3.6721e-01,  2.4417e-01,\n",
       "          -2.5694e-01,  2.0378e-03, -3.0780e-01, -3.6706e-01,  1.8169e-01,\n",
       "           7.3122e-01, -2.1473e-01,  9.9233e-02, -2.9653e-01, -3.5736e-01,\n",
       "          -3.1443e-01, -3.2372e-01, -2.1382e-01, -4.8743e-02,  1.6010e-01,\n",
       "          -1.1684e-01,  7.3328e-03, -2.2013e-01,  3.4689e-01, -1.1010e-01,\n",
       "           3.2356e-01,  4.2742e-01,  4.9158e-01, -1.5864e-02, -2.9975e-01,\n",
       "          -3.3077e-01,  2.5213e-01, -1.0164e-01,  3.3758e-01, -2.6408e-01,\n",
       "          -3.6077e-01, -2.7779e-02,  1.2287e-01, -4.7394e-01,  2.2007e-01,\n",
       "          -6.1151e-01,  2.7277e-01, -1.7342e-01, -2.0179e-02,  2.4015e-01,\n",
       "           2.3243e-01,  2.1663e-01, -4.6588e-01,  4.6449e-01,  4.1972e-01,\n",
       "           1.2592e-01,  2.8265e-01,  3.0306e-01,  6.0499e-02, -2.3870e-01,\n",
       "           8.3599e-02, -9.6397e-01, -4.1337e-02, -1.3807e-01, -1.5032e-01,\n",
       "          -2.9964e-01, -2.6773e-01,  5.6849e-01,  2.3993e-01,  6.9110e-01,\n",
       "           5.0521e-01, -1.7087e-01, -5.7037e-01, -2.9213e-01,  1.7344e-01,\n",
       "           2.1405e-01, -1.4886e-01,  3.0544e-01,  1.1362e-01,  2.1320e-01,\n",
       "          -6.0712e-01, -3.0321e-01, -3.7033e-02,  2.1753e-01,  6.9741e-02,\n",
       "           7.3239e-01, -3.9063e-01, -2.9549e-01,  6.8877e-02,  2.3525e-01,\n",
       "           1.6941e-01,  7.3786e-01,  1.4470e-01, -6.8182e-01, -9.3803e-02,\n",
       "          -4.7710e-03, -1.2981e-01,  3.2907e-01, -5.9411e-01,  1.8814e-01,\n",
       "           3.7874e-02,  5.2011e-01,  7.7260e-01]], device='mps:0'),\n",
       " tensor([[-0.2623, -0.5417, -0.4505,  ...,  0.1173,  0.5185,  0.6248],\n",
       "         [-0.1711, -0.7371, -0.5961,  ...,  0.1080,  0.3116,  0.3395],\n",
       "         [-0.3292, -0.2590, -0.5016,  ..., -0.0148,  0.3509,  0.1607]],\n",
       "        device='mps:0'),\n",
       " tensor([[-0.4654, -0.4115, -0.4370,  ...,  0.1625,  0.4338,  0.5908],\n",
       "         [-0.1044, -0.7255, -0.6196,  ...,  0.3032,  0.3411,  0.6797]],\n",
       "        device='mps:0'),\n",
       " tensor([[-2.2115e-01, -4.0785e-01, -5.7008e-01, -8.4110e-01, -1.8063e-01,\n",
       "          -6.4874e-02, -8.1797e-01,  4.8747e-01,  1.3876e-01,  5.1025e-01,\n",
       "           1.3308e-01,  4.5201e-01,  1.2124e-01, -2.2697e-02, -3.0274e-01,\n",
       "          -2.7463e-01,  1.9110e-01,  3.0301e-01,  5.1059e-01, -4.2661e-02,\n",
       "           2.0994e-01,  5.2801e-01, -8.4736e-02, -2.8641e-02, -6.8592e-01,\n",
       "           2.9997e-01, -7.5231e-02,  8.7288e-02,  2.4767e-01, -3.5634e-01,\n",
       "           1.3398e-02,  4.1260e-01, -1.3038e-01,  1.5028e+00, -3.4582e-01,\n",
       "           8.9252e-03,  1.2309e-01, -3.5245e-01,  2.9210e-02,  8.3731e-02,\n",
       "           5.9744e-01, -3.4789e-01,  9.0473e-02,  4.4851e-01, -8.2971e-02,\n",
       "           2.0044e-01,  2.3719e-01, -3.5455e-01,  2.5407e-01, -9.2699e-01,\n",
       "          -2.0832e-01, -2.7803e-01,  5.0140e-01,  5.6522e-01, -5.5595e-01,\n",
       "          -2.6412e-01,  2.5578e-01,  7.9181e-02, -4.0664e-01,  1.7214e-01,\n",
       "           5.3818e-02,  4.7333e-01, -1.7820e-01,  2.1154e-01,  2.0118e-01,\n",
       "           3.0639e-01, -2.2901e-01, -4.6961e-01, -5.2300e-01, -2.1040e-01,\n",
       "          -6.5956e-02,  2.9151e-02,  5.5275e-02, -2.6878e-01,  5.1983e-01,\n",
       "          -4.5297e-01,  6.7164e-02, -2.5771e-01, -1.3175e-01,  2.1727e-01,\n",
       "          -7.8943e-02,  2.4914e-01,  1.9719e-01,  1.9665e-01,  1.5592e-01,\n",
       "           2.6325e-02,  1.3043e-01,  6.5136e-02,  3.4640e-02,  2.9002e-01,\n",
       "          -2.1763e-01, -7.0018e-02, -2.9948e-01, -2.1331e-01,  2.5006e-01,\n",
       "           1.1040e-01,  2.2652e-01, -1.7122e-01,  2.2869e-01, -1.1475e-01,\n",
       "           1.3421e-01, -2.5993e-02, -1.1727e-01, -6.2681e-03, -2.7773e-01,\n",
       "          -4.5485e-01,  1.3560e-01, -2.8137e-01, -5.5872e-01,  1.5983e-01,\n",
       "          -2.2923e-01, -3.3457e-01,  4.2271e-02, -4.2907e-01,  1.4274e-02,\n",
       "          -1.2454e-01,  1.0327e-01, -3.2173e-02, -1.8872e-01,  5.5090e-03,\n",
       "           2.3560e-01, -1.5391e-01,  1.3128e-01,  1.7807e-01,  2.4997e-01,\n",
       "          -2.5083e-02,  5.3017e-01,  2.9607e-02,  1.8693e-01,  3.3842e-01,\n",
       "           1.3163e-01, -1.7996e-01, -2.2352e-01, -6.3588e-02,  6.6907e-03,\n",
       "           3.6669e-02,  4.0967e-02, -1.5772e-01,  8.7341e-02, -5.9198e-02,\n",
       "           1.8809e-01, -2.1040e-01, -6.5251e-01, -4.9339e-01, -1.3446e-01,\n",
       "          -2.3077e-01, -6.3704e-02, -5.1876e-02, -4.8677e-01, -3.4766e-01,\n",
       "          -7.5440e-01, -3.9442e-01,  6.9877e-01, -1.6336e-01, -2.0169e-01,\n",
       "          -3.6948e-04,  2.8772e-01,  1.5802e-01, -1.0174e-01,  2.7112e-01,\n",
       "           7.4345e-01, -1.1905e-01,  4.1394e-02, -4.8582e-01, -2.5506e-01,\n",
       "           6.4343e-02,  7.3559e-02, -3.7165e-01,  1.4194e-01,  1.1692e-01,\n",
       "           8.7149e-02, -1.9535e-01,  1.8104e-01, -2.8488e-01,  1.1084e-01,\n",
       "          -6.4854e-01,  3.3547e-01,  2.6510e-01, -2.0923e-01, -1.2559e-01,\n",
       "           3.1449e-02, -1.3046e+00,  1.1282e-01, -6.7069e-01,  1.9442e-01,\n",
       "          -1.7050e-01,  1.4410e-01,  3.1129e-01,  2.6171e-01, -2.4990e-02,\n",
       "           2.6322e-01,  1.2807e-01,  1.7444e-01,  3.1132e-01, -6.1163e-02,\n",
       "           1.5101e-01,  2.9770e-01,  3.8045e-01,  2.5646e-01,  1.2933e-01,\n",
       "          -3.9422e-01,  3.2834e-01,  3.6373e-01,  1.8673e-01,  1.8082e-01,\n",
       "           1.4619e-01, -1.4171e-01,  1.3173e-01,  5.8809e-02, -1.9359e-01,\n",
       "          -2.2467e-01,  5.8982e-02, -4.6934e-01, -1.6353e-02, -2.7697e-01,\n",
       "           1.9581e-01, -3.3842e-01,  1.0616e-03,  5.1638e-01,  2.7234e-01,\n",
       "           6.1655e-02,  2.4680e-01,  9.9117e-02, -3.2177e-01,  2.4660e-01,\n",
       "           2.5621e-01, -3.2565e-01,  6.3398e-02, -4.0649e-01, -2.5452e-01,\n",
       "           2.8386e-01, -6.1659e-01,  5.2713e-01,  5.2735e-01,  1.1534e-02,\n",
       "           4.4759e-02, -3.7464e-01,  3.4684e-02, -4.2542e-01,  3.8320e-01,\n",
       "          -2.7552e-01, -4.4780e-01, -5.7812e-01,  7.0197e-02,  5.0748e-01,\n",
       "           2.8194e-01,  1.5510e-02,  7.2880e-01,  3.2546e-01,  3.6210e-01,\n",
       "          -3.0822e-01, -4.0243e-02,  3.8499e-01,  1.0456e-01, -3.8188e-01,\n",
       "           1.9314e-02, -3.2356e-02, -4.2720e-01,  2.7195e-01, -2.7270e-01,\n",
       "          -1.4005e-01, -1.0777e-01,  3.6784e-01,  3.4209e-01, -3.7914e-01,\n",
       "          -1.3150e-01,  3.0634e-01, -1.0394e-01, -4.3525e-01,  2.9672e-01,\n",
       "          -4.4497e-01,  1.8905e-01,  1.2563e-01, -3.6693e-01, -5.7399e-01,\n",
       "          -3.2053e-01, -6.9191e-02, -1.7400e-01,  1.1770e-01, -7.2510e-01,\n",
       "          -4.5610e-01,  4.9362e-02,  1.0284e-01,  1.0720e-01,  2.9626e-01,\n",
       "           3.9273e-01, -2.3658e-01, -1.7200e-01, -1.3320e-01, -5.5932e-01,\n",
       "          -8.2797e-02,  5.6099e-02, -3.1822e-01,  3.1942e-01,  2.0091e-01,\n",
       "           1.4024e-02,  1.0330e-01,  4.1367e-01, -2.9776e-01, -3.0643e-01,\n",
       "          -4.9441e-02,  5.0167e-02, -2.3929e-01, -1.0467e-01,  2.5486e-01,\n",
       "           5.0751e-02,  4.1962e-01,  9.0461e-01,  5.9333e-02, -5.0320e-01,\n",
       "           1.3741e-01,  6.1045e-01, -3.5701e-01,  6.5412e-03, -1.2161e-01,\n",
       "          -1.0206e+00, -2.4188e-01, -1.5891e-01, -2.9731e-01, -5.3073e-01,\n",
       "           1.0215e-01,  2.2052e-01,  1.2428e-02, -1.5334e-01, -3.7618e-01,\n",
       "           2.0821e-01,  2.2134e-01, -1.1328e-01,  3.9144e-01, -3.0264e-01,\n",
       "          -2.6101e-01, -2.3275e-01, -3.4418e-01, -4.6901e-01, -4.2015e-01,\n",
       "           3.5280e-01,  2.2192e-01,  1.2231e-01, -1.6684e-01, -4.1457e-01,\n",
       "           4.8600e-01, -5.5057e-01, -1.4724e-02,  2.6035e-01, -4.3419e-01,\n",
       "          -1.6682e-01,  5.0294e-01, -4.1691e-02,  2.1870e-01,  1.6419e-01,\n",
       "           1.6186e-01, -3.7523e-01, -5.2219e-01, -2.5891e-01,  6.5956e-01,\n",
       "           2.7140e-01,  3.6756e-01, -2.2686e-01,  2.0260e-01,  5.0353e-01,\n",
       "          -2.8142e-01,  4.8572e-01, -1.2124e-01, -4.3083e-02, -1.7088e-01,\n",
       "           3.3554e-01,  3.0027e-01, -2.2149e-01,  3.4676e-01,  2.4224e-01,\n",
       "          -2.9572e-01, -1.1043e-01, -3.0894e-02,  3.4511e-01,  2.3173e-02,\n",
       "          -9.9270e-02, -2.3328e-01,  1.7864e-01, -2.5785e-01, -3.1786e-01,\n",
       "           2.5711e-01, -1.5047e-01,  1.3902e-01, -4.1619e-01, -2.7322e-01,\n",
       "          -1.9672e-01, -2.2865e-01,  7.9684e-01,  1.7149e-01, -2.7634e-01,\n",
       "           2.1428e-01, -2.1926e-01, -3.9203e-01,  5.0543e-01,  1.1788e-01,\n",
       "          -4.1701e-01,  4.2716e-01, -3.4446e-01,  1.0070e-01,  2.3280e-01,\n",
       "          -7.9724e-02, -1.1698e-01,  4.7742e-02,  2.7417e-01,  3.7730e-01,\n",
       "          -3.7167e-01,  3.7586e-03, -6.2826e-02, -2.0672e-01,  2.4468e-01,\n",
       "           1.7035e-01,  3.5660e-01, -1.9704e-01, -1.7483e-01,  8.3473e-02,\n",
       "           3.6290e-01, -1.1784e-01,  4.9871e-03, -4.0691e-01, -4.5798e-01,\n",
       "           3.6629e-01,  1.7803e-01,  4.8482e-01,  2.1583e-01, -1.1684e+01,\n",
       "           3.3027e-02,  7.5940e-01, -3.1420e-01,  7.9339e-02,  3.6269e-02,\n",
       "           1.4228e-03, -3.1587e-01,  3.3568e-01, -7.5273e-02, -4.2220e-01,\n",
       "          -3.2106e-01, -3.3083e-01,  2.4085e-01, -2.4356e-01, -1.1312e-01,\n",
       "           1.8976e-01, -7.2268e-01,  2.2771e-01, -2.3753e-01,  5.5671e-01,\n",
       "          -2.7443e-01, -9.3054e-02,  2.5924e-01, -2.0426e-01, -4.4191e-02,\n",
       "           2.1536e-01, -4.5893e-02,  2.2346e-01,  2.4036e-01, -4.1898e-02,\n",
       "           1.4362e-01, -1.2398e-01,  7.8931e-03, -1.7247e-01, -6.1442e-01,\n",
       "          -1.2422e-01, -3.0776e-01,  2.1553e-01, -8.6393e-02, -5.8819e-01,\n",
       "           2.4543e-02, -2.8082e-01,  2.6703e-01,  4.0834e-01,  7.3936e-01,\n",
       "           5.1571e-01,  1.1077e+00,  6.6908e-02,  1.9403e-01,  1.6779e-01,\n",
       "          -7.3383e-01,  7.2103e-02, -3.0086e-01,  5.4596e-02, -4.9264e-04,\n",
       "           2.9902e-01, -6.3275e-01,  3.2030e-01, -3.3935e-01,  1.0018e-01,\n",
       "           5.2465e-01, -3.2176e-02,  2.2974e-01, -1.6183e-01, -1.1412e-01,\n",
       "           1.6192e-01, -5.0214e-02,  1.2932e-01, -1.5990e-01, -2.4795e-01,\n",
       "           1.1974e-02,  5.1514e-01,  1.9103e-01, -1.0085e+00,  3.3074e-01,\n",
       "           1.1479e-01, -1.1113e-01,  1.3344e+00,  6.0265e-02,  2.0453e-02,\n",
       "          -7.2403e-02,  2.7630e-01, -4.8136e-01, -5.5133e-01,  3.3743e-01,\n",
       "          -1.7901e-01, -7.6146e-02,  2.4472e-01,  3.9879e-01, -2.0777e-01,\n",
       "          -4.8103e-01,  4.6200e-02,  1.6913e-01,  2.0664e-02,  5.9595e-01,\n",
       "          -8.0440e-01, -1.7446e-01, -2.9740e-02,  1.2390e-01, -6.9377e-01,\n",
       "          -2.8243e-03, -4.3000e-01, -3.3049e-01,  5.1123e-01, -2.6437e-01,\n",
       "           1.1945e-01, -1.1062e-01, -2.2750e-01,  4.4296e-01, -2.7944e-01,\n",
       "           1.6202e-02,  3.6161e-01,  5.1586e-02, -1.8236e-01, -8.4289e-02,\n",
       "           2.7268e-01,  5.5625e-01, -5.0114e-01, -4.0332e-01, -7.6952e-02,\n",
       "           1.3843e-01,  2.5794e-01,  3.1399e-01, -3.8615e-01,  3.8565e-01,\n",
       "          -8.2569e-01, -1.3949e-01,  8.5260e-02,  9.9086e-02,  4.9717e-01,\n",
       "           1.0666e-01,  1.1684e-02,  1.1151e-01, -5.6289e-01, -4.8569e-01,\n",
       "          -7.2403e-01, -1.9239e-01,  1.2706e-01, -2.3196e-01, -2.0672e-02,\n",
       "          -1.6149e-01, -4.2470e-02,  3.3131e-02,  2.0050e-01, -7.2901e-02,\n",
       "           2.0855e-01,  1.6585e-01, -4.7449e-02, -1.1259e-01, -5.3955e-02,\n",
       "          -4.2408e-01,  5.7576e-02, -8.2128e-02, -1.1104e-01, -6.4200e-01,\n",
       "          -2.0823e-01,  2.4172e-02,  8.9488e-02, -2.8029e-01,  1.9129e-01,\n",
       "           1.5103e-01,  1.4656e-01, -2.6516e-01,  6.7193e-01,  3.0966e-01,\n",
       "          -1.5960e-02, -2.6234e-01, -5.4246e-01, -4.5675e-02,  1.9935e-01,\n",
       "           2.9931e-01,  2.5905e-01, -3.2107e-01, -3.3201e-01, -3.0591e-01,\n",
       "           7.5813e-03,  6.9124e-01, -1.5810e-01, -5.0761e-01,  1.0113e-01,\n",
       "           5.7340e-01, -1.7424e-01, -3.3546e-01,  1.3036e-01,  7.7503e-02,\n",
       "          -2.6064e-01, -5.5278e-01,  6.4369e-01, -1.1135e-01,  4.1509e-02,\n",
       "           3.1584e-01,  2.0766e-01, -1.1206e-01,  5.9170e-01, -5.2692e-03,\n",
       "          -4.6380e-01, -5.2345e-02, -8.1214e-01,  3.1456e-01,  1.6943e-01,\n",
       "          -1.6479e-01,  7.6410e-01,  2.3705e-01, -3.1335e-01, -4.2930e-02,\n",
       "           4.3377e-01,  9.6831e-01, -1.0729e-01,  1.9369e-01,  6.5521e-01,\n",
       "           3.6739e-02, -2.1334e-01,  1.8032e-01,  3.7207e-02,  1.0767e-01,\n",
       "           4.5342e-01, -2.6050e-01, -4.4642e-01, -1.0399e-01,  3.6934e-01,\n",
       "           4.3893e-01, -4.5900e-01, -6.9414e-01, -1.5317e-01,  5.3691e-01,\n",
       "          -3.3230e-01,  2.9391e-01, -7.8549e-02,  3.0547e-01,  1.3460e-01,\n",
       "          -9.1134e-01, -1.1964e-01,  8.9327e-02,  4.7779e-02,  2.3483e-02,\n",
       "          -3.5310e-01, -2.0503e-01, -2.0414e-01, -1.2068e-01,  6.7507e-01,\n",
       "           2.2571e-01, -7.5395e-01,  8.1098e-02,  6.0859e-01, -7.0330e-01,\n",
       "           2.6161e-01, -3.0057e-01, -1.0814e-01,  3.7312e-01,  7.1977e-02,\n",
       "          -1.8065e-01,  1.2742e-01, -4.2657e-01, -4.1132e-01,  1.5504e-01,\n",
       "           8.8180e-01, -9.8331e-02,  1.9444e-01, -3.1352e-01, -2.6970e-01,\n",
       "          -2.9879e-01, -4.2308e-01, -3.1602e-01,  2.4105e-01,  1.0052e-01,\n",
       "          -1.3343e-02, -8.0468e-02, -3.3635e-01,  3.7359e-01,  8.4713e-02,\n",
       "           3.8641e-01,  3.9878e-01,  5.7910e-01,  2.4004e-02, -2.8970e-01,\n",
       "          -1.5561e-02,  1.0451e-01, -1.4209e-01,  4.8604e-01, -2.7948e-01,\n",
       "          -4.8096e-01,  1.9265e-02,  7.9697e-02, -5.4979e-01, -1.1590e-01,\n",
       "          -4.8090e-01,  2.4789e-01, -1.4125e-01,  1.2204e-02,  3.8391e-01,\n",
       "           1.8131e-01,  9.0521e-02, -2.9486e-01,  5.2355e-01,  5.0011e-01,\n",
       "           9.4086e-02,  5.7952e-01,  2.4944e-01,  9.6160e-02, -3.1777e-01,\n",
       "           2.8294e-01, -9.6606e-01, -1.5936e-01, -1.0321e-01, -2.2222e-01,\n",
       "          -3.4218e-01, -1.4505e-01,  1.7743e-01,  1.7824e-01,  6.2117e-01,\n",
       "           5.8925e-01, -2.2997e-01, -6.0209e-01, -1.7097e-01,  1.1398e-01,\n",
       "           1.4756e-01, -1.2023e-01,  2.0127e-01,  1.4290e-01,  3.1449e-01,\n",
       "          -5.6819e-01, -3.6697e-01,  4.5629e-02,  1.8809e-01, -1.1195e-01,\n",
       "           7.5055e-01, -2.0544e-01, -1.4336e-01, -4.5771e-02,  8.9154e-02,\n",
       "           2.1665e-01,  6.8106e-01,  2.5200e-01, -6.4751e-01, -1.0427e-02,\n",
       "           6.3059e-02, -3.0872e-01,  1.5917e-01, -6.3080e-01,  1.8315e-01,\n",
       "           2.0125e-01,  4.6589e-01,  5.2440e-01]], device='mps:0'),\n",
       " tensor([[-9.8608e-02, -6.2949e-01, -5.4737e-01, -4.6335e-01, -3.7454e-01,\n",
       "           1.0248e-01, -2.8097e-01,  4.4707e-01, -1.3869e-01,  5.9022e-01,\n",
       "           3.3979e-01,  3.6518e-02, -7.9829e-02, -1.9712e-01, -4.0961e-01,\n",
       "          -5.4994e-01,  1.7904e-01,  2.6107e-01,  2.6766e-01, -1.3636e-01,\n",
       "           9.8026e-01,  3.6199e-01, -8.5372e-02,  1.8939e-01, -6.9872e-01,\n",
       "           3.9986e-01,  4.5490e-02,  8.0399e-02,  4.6963e-01, -6.9729e-01,\n",
       "          -2.0828e-01,  5.4662e-01, -1.2219e-01,  1.1649e+00, -1.8525e-01,\n",
       "          -2.5129e-01,  4.1644e-01, -2.6874e-01,  8.0973e-01,  4.0156e-02,\n",
       "           6.2909e-01, -1.2839e-01,  2.5974e-01,  2.1204e-01,  2.5654e-01,\n",
       "          -1.6577e-02,  1.0736e-01, -6.2638e-01,  5.4556e-01, -5.1867e-01,\n",
       "          -2.1843e-01, -1.5953e-01,  3.6604e-01,  4.8748e-01, -6.7281e-02,\n",
       "           3.5607e-02,  3.0541e-01,  1.6232e-01, -2.5763e-01,  2.0386e-01,\n",
       "           1.3393e-01,  6.9371e-01,  5.7586e-02, -1.9525e-01, -3.0287e-01,\n",
       "           3.0687e-01, -5.2685e-01, -7.0972e-01, -3.4711e-01, -1.5503e-01,\n",
       "          -6.3145e-02,  3.6453e-01, -2.3968e-01, -8.3683e-02,  7.8001e-02,\n",
       "          -8.8895e-02,  3.9835e-02, -1.7901e-01, -2.7637e-01, -9.8806e-03,\n",
       "           6.4246e-02,  3.2661e-01, -1.9744e-01, -4.1654e-01, -2.1876e-01,\n",
       "           3.0475e-01,  7.8369e-01,  6.3690e-02,  2.5669e-01,  8.5968e-02,\n",
       "           3.1750e-01,  9.4549e-02, -3.4083e-01, -4.9471e-01,  3.1401e-01,\n",
       "          -1.3255e-01,  4.9667e-01, -1.4940e-01, -6.5687e-02, -4.7700e-02,\n",
       "           3.7095e-01,  7.2304e-02,  1.7323e-03,  1.1325e-01, -6.6663e-02,\n",
       "          -4.8542e-01,  1.4204e-01, -2.5931e-01, -6.6520e-01, -1.1522e-01,\n",
       "          -7.9637e-02,  1.3277e-01,  1.5561e-02, -3.1430e-01,  2.6189e-01,\n",
       "          -7.3846e-02,  1.9145e-01, -1.3642e-02, -3.3838e-01,  3.7293e-01,\n",
       "           2.5609e-01, -4.0848e-01,  2.2958e-02,  1.6951e-01,  4.6388e-01,\n",
       "          -2.2394e-01,  6.0049e-01, -2.8694e-01,  2.5392e-01,  4.8866e-01,\n",
       "           8.4207e-02, -6.0898e-02, -9.6990e-02, -1.7751e-01, -3.9547e-01,\n",
       "          -4.2915e-02,  7.3082e-02, -3.2798e-01,  1.0024e-01,  2.1027e-02,\n",
       "          -1.5484e-01, -1.5575e-01, -7.1363e-01, -3.2433e-01,  1.6862e-01,\n",
       "          -2.9423e-01,  2.9391e-01, -7.2839e-02, -5.6183e-01, -3.6233e-01,\n",
       "          -8.1793e-01, -4.5420e-01,  5.0598e-01,  3.1144e-02, -1.4616e-01,\n",
       "          -4.4906e-02,  1.9702e-01,  1.8656e-01, -1.9001e-02,  3.6581e-01,\n",
       "           4.5356e-01, -1.7970e-01,  5.5773e-02, -4.7383e-01,  5.1199e-02,\n",
       "           3.6512e-02, -1.1536e-01, -1.4678e-01, -1.8904e-01,  8.2812e-02,\n",
       "           1.8413e-01, -2.0916e-01, -2.0209e-02, -4.3147e-01,  1.4387e-01,\n",
       "          -9.5740e-02,  8.7516e-02,  2.8605e-01, -3.1460e-01, -3.4357e-01,\n",
       "          -4.4457e-01, -2.0415e+00,  1.3037e-01, -1.6708e-01, -1.9640e-01,\n",
       "          -7.8661e-01,  5.6989e-01,  4.0038e-01,  2.3491e-01,  1.2055e-02,\n",
       "          -4.1441e-01, -1.3256e-01, -2.2065e-01,  2.0346e-01, -1.5937e-01,\n",
       "           4.1104e-01,  3.6856e-01,  2.1222e-01, -8.2505e-02, -5.3417e-02,\n",
       "          -3.3993e-01,  1.9027e-01,  6.6698e-02,  4.3789e-02,  1.3752e-02,\n",
       "           8.6470e-02, -3.4516e-01,  1.5865e-01,  6.3413e-01, -3.2035e-01,\n",
       "          -2.6413e-01, -1.5432e-01, -3.1116e-01, -1.1825e-01, -2.1011e-01,\n",
       "           7.5833e-01, -1.0075e-01, -1.9090e-01,  4.1462e-01,  2.9186e-01,\n",
       "          -2.0576e-01,  2.6479e-02, -1.7243e-01, -4.6025e-01, -8.4004e-02,\n",
       "           8.3166e-01, -2.6809e-01, -1.8435e-01, -1.3576e-01, -2.8072e-01,\n",
       "           9.9627e-02, -3.9657e-01,  7.0865e-01,  1.3131e-01, -2.4437e-01,\n",
       "          -1.3250e-01, -4.9085e-01, -3.6081e-02, -6.5221e-02,  2.7934e-01,\n",
       "          -3.2480e-01, -4.4413e-01, -6.2940e-01,  9.8728e-02,  4.2580e-01,\n",
       "           1.0494e-01, -2.2581e-01,  5.8588e-01,  3.7420e-01,  5.5104e-01,\n",
       "          -7.6877e-02,  1.8477e-03,  4.0139e-01,  4.0454e-02, -3.1153e-01,\n",
       "           4.0137e-02,  1.1839e-01, -6.1671e-02,  2.8916e-01, -5.0007e-01,\n",
       "           2.1460e-01,  1.3048e-01,  4.2285e-02,  4.0869e-01, -3.3156e-01,\n",
       "           3.3759e-01, -6.4578e-04, -4.2352e-01, -2.2792e-01,  3.1587e-01,\n",
       "          -3.4318e-01,  4.7882e-01, -1.1497e-01, -1.3533e-01, -6.5793e-01,\n",
       "          -3.0023e-01,  9.9027e-02, -5.2265e-01,  4.0820e-02, -8.5922e-01,\n",
       "          -1.4457e-01,  4.3467e-01, -1.1843e-01,  2.8682e-01, -4.3268e-01,\n",
       "           2.7079e-01,  1.1150e-01,  1.0583e-01, -1.8427e-01, -4.8035e-01,\n",
       "           6.4368e-03,  3.5471e-01, -1.9635e-01,  6.7501e-02,  1.8728e-02,\n",
       "           1.4821e-01,  7.4914e-02, -1.4411e-01,  1.8116e-02, -3.9952e-01,\n",
       "           1.2334e-01,  2.7083e-01, -6.4203e-02, -1.1696e-01, -2.5127e-01,\n",
       "           8.1457e-02,  5.7341e-02,  4.5826e-01, -1.5574e-01, -4.7792e-01,\n",
       "           1.3602e-02,  5.6159e-01,  4.2917e-02,  9.5691e-02, -4.1476e-02,\n",
       "          -1.3511e+00, -3.8201e-01,  1.3646e-01, -2.3684e-01, -1.6250e-01,\n",
       "           1.4728e-01, -4.5766e-02, -1.0990e-01,  1.3434e-01, -3.3727e-01,\n",
       "           4.9913e-01, -7.4794e-03, -9.4830e-02,  2.6318e-01, -2.9595e-01,\n",
       "          -2.6245e-01, -5.5022e-02, -5.5963e-02, -1.9048e-01, -2.1108e-01,\n",
       "           1.0760e-01,  2.4895e-01, -1.6610e-02,  2.9324e-01, -4.2315e-01,\n",
       "           6.9357e-01,  2.0403e-02, -3.6256e-01,  1.8028e-01, -4.6817e-01,\n",
       "          -2.2470e-01,  6.2020e-01, -2.1111e-01,  1.7508e-01,  7.0164e-02,\n",
       "           2.2583e-01, -1.6263e-01, -2.5922e-01, -1.5555e-01,  5.6515e-01,\n",
       "           9.7645e-02,  3.8631e-01, -3.1465e-01,  4.6445e-01,  6.0830e-01,\n",
       "          -4.3876e-01,  2.7575e-01, -3.0324e-01, -3.5509e-01, -1.4900e-01,\n",
       "           3.7937e-01,  2.4503e-01, -2.4568e-01,  1.2754e-01,  2.1491e-01,\n",
       "           2.4189e-01, -2.8123e-01, -1.4375e-01,  1.5623e-01, -1.6800e-02,\n",
       "          -1.6159e-01, -5.9860e-01,  4.8387e-02, -4.0540e-01, -3.7598e-01,\n",
       "          -5.5542e-02, -9.6940e-02,  2.3018e-01, -3.4557e-01, -2.8109e-01,\n",
       "          -4.1082e-01,  1.5328e-01,  7.9392e-01, -8.2566e-02,  1.3355e-02,\n",
       "          -2.4290e-01,  1.3134e-01, -3.9568e-01,  5.6959e-01,  2.4837e-01,\n",
       "          -5.6746e-02,  5.0092e-01, -5.9799e-01,  3.2502e-01,  1.6325e-01,\n",
       "          -2.5296e-01,  7.9720e-02, -1.1265e-01,  4.2524e-01, -1.9241e-01,\n",
       "          -2.6666e-01,  1.8088e-01,  2.9568e-01,  8.0022e-02,  3.3680e-02,\n",
       "           8.9946e-02,  7.7279e-01, -5.6155e-01, -1.1355e-01, -3.2412e-02,\n",
       "           4.6990e-01, -1.4868e-01,  1.7842e-01, -2.5184e-01, -4.5494e-01,\n",
       "           2.8096e-01,  7.6220e-03,  3.0259e-02,  1.5848e-01, -1.1470e+01,\n",
       "           1.7761e-02,  7.9956e-01, -5.9407e-01,  3.9142e-02, -7.9952e-04,\n",
       "           1.4375e-01, -2.8145e-01,  3.6361e-03, -2.5851e-01, -3.7282e-01,\n",
       "           1.6253e-01,  5.8747e-02,  6.8827e-02, -1.9252e-01, -1.2168e-01,\n",
       "          -4.4456e-02, -9.0133e-01,  1.4013e-01, -2.9238e-01,  3.1633e-01,\n",
       "          -4.6944e-01,  3.5652e-01,  1.0775e-01,  4.8818e-02, -3.3732e-01,\n",
       "          -6.3534e-02, -4.1684e-01,  2.3683e-01,  1.0364e-01, -1.6207e-01,\n",
       "          -1.3703e-01, -1.5828e-01,  4.6034e-01, -1.8993e-01, -4.8947e-01,\n",
       "           1.5862e-01, -7.0142e-02,  4.5043e-01, -1.6798e-02, -2.3410e-01,\n",
       "           5.0011e-02, -4.1456e-01,  3.0195e-01,  6.4668e-02,  5.0457e-01,\n",
       "           4.7572e-01,  6.3490e-01,  1.0739e-01, -2.1848e-01,  4.9891e-02,\n",
       "          -8.9429e-01,  3.3171e-01, -9.8824e-01, -2.6344e-01,  3.6779e-02,\n",
       "           7.4622e-01, -4.6007e-01,  9.9323e-03,  8.2971e-02, -3.9312e-01,\n",
       "           2.4107e-01, -5.5413e-02,  1.0686e-01, -6.2270e-01, -1.1555e-01,\n",
       "           3.6365e-01,  1.0164e-01,  1.3567e-01, -3.5856e-01, -1.3223e-02,\n",
       "          -4.3049e-02,  5.9870e-01,  1.5647e-01, -2.5151e-01,  1.0429e-01,\n",
       "          -8.0721e-03,  2.3224e-01,  6.2748e-01,  4.7218e-02,  1.3164e-01,\n",
       "           1.7459e-01,  5.1672e-01, -5.4105e-01, -4.7941e-01, -1.2530e-01,\n",
       "           6.3722e-02,  2.8461e-01,  8.3486e-02,  3.6163e-01, -3.9872e-01,\n",
       "          -3.1281e-01, -7.8561e-02,  3.1281e-02,  5.0728e-02,  1.2098e+00,\n",
       "          -5.8519e-01, -2.2122e-01, -1.0256e-01, -9.1470e-02, -1.3571e-01,\n",
       "          -3.9828e-02, -3.3651e-01, -4.2370e-01,  3.3182e-01,  1.1019e-01,\n",
       "           1.7328e-01,  1.5282e-02,  3.9299e-01,  1.9321e-01, -5.2099e-01,\n",
       "           2.8514e-01,  1.4470e-01,  4.3958e-01, -1.8846e-01, -4.1365e-01,\n",
       "          -1.0991e-01,  3.3157e-01, -3.8281e-01, -3.4973e-01, -1.0983e-01,\n",
       "           3.0412e-02,  5.9583e-02, -5.8809e-02, -3.7567e-01,  2.5850e-01,\n",
       "          -6.2094e-01, -9.3514e-03, -4.2726e-01, -1.1183e-01,  4.7884e-01,\n",
       "           1.2318e-01, -3.6857e-02,  1.1137e-01, -4.2935e-01, -6.7978e-01,\n",
       "          -6.4684e-01, -7.4975e-02,  5.2269e-02, -3.7956e-01, -1.6842e-02,\n",
       "          -1.8671e-01,  6.6050e-02, -3.6548e-01,  1.8633e-01, -8.2184e-02,\n",
       "           7.5332e-01,  2.3533e-01, -1.8486e-01,  1.2410e-01, -1.4424e-01,\n",
       "          -3.6447e-01, -2.1959e-01, -2.4676e-01,  2.2621e-01, -5.0366e-01,\n",
       "          -1.3080e-01,  1.8011e-01,  1.1438e-01,  6.4878e-03, -3.4448e-01,\n",
       "          -1.2932e-01,  2.0570e-01, -1.3707e-01,  3.0997e-01,  1.7373e-01,\n",
       "           2.0329e-01, -6.4335e-02, -4.6287e-01,  8.7786e-02,  4.5179e-01,\n",
       "          -1.1348e-01, -2.7396e-01,  3.8658e-02, -7.0830e-01,  2.1125e-01,\n",
       "           3.8231e-01,  8.2746e-01, -2.5998e-01, -5.3996e-01,  4.6996e-01,\n",
       "           6.2219e-01, -4.6785e-01, -5.7024e-02, -2.1133e-01,  1.2442e-01,\n",
       "          -1.0488e-01, -3.4183e-01,  2.6033e-01, -2.5177e-01,  4.2375e-01,\n",
       "          -9.9348e-02,  2.0811e-01, -1.9413e-01,  4.1807e-01,  9.7647e-02,\n",
       "          -2.0238e-01, -3.5246e-02, -6.2741e-01, -4.3863e-01,  2.3873e-01,\n",
       "           2.6324e-01, -1.5522e-01, -2.3160e-02, -1.2275e-01,  1.1149e-01,\n",
       "           4.8271e-01,  8.5393e-01,  8.1221e-02,  1.4584e-01,  3.8317e-01,\n",
       "           2.3356e-01, -2.7630e-01,  5.9953e-02,  2.9270e-01,  4.0766e-01,\n",
       "           3.5259e-01,  5.3145e-01, -3.4176e-01,  7.2203e-02, -1.6344e-01,\n",
       "           2.0465e-01, -5.7619e-01, -7.6698e-01, -1.3126e-01,  4.3082e-01,\n",
       "           1.5541e-02,  2.9468e-01, -2.7851e-01,  3.4408e-01,  6.3093e-01,\n",
       "          -1.1306e+00, -5.9181e-02, -3.9846e-02, -9.3994e-02, -6.2591e-02,\n",
       "          -5.4897e-01, -2.2413e-01, -3.5234e-01, -2.4418e-01,  5.2852e-01,\n",
       "           5.8811e-01, -8.5286e-01,  1.6085e-01,  2.2953e-01, -1.5005e-01,\n",
       "           9.4328e-02, -9.1644e-02,  3.2863e-02,  5.0134e-01,  1.5175e-01,\n",
       "          -5.7932e-01,  4.5778e-01,  1.5595e-01, -4.3637e-01,  2.6165e-01,\n",
       "           4.1776e-01, -1.8687e-02,  1.5952e-01,  2.2085e-01, -4.9751e-01,\n",
       "          -1.0141e-02, -4.6200e-01, -1.0629e-01,  1.0751e-01, -6.7537e-02,\n",
       "           2.4815e-02, -4.9794e-01, -6.6236e-02,  3.4484e-01,  2.9505e-03,\n",
       "           5.0621e-01,  6.0668e-01,  7.6526e-01, -5.6484e-02, -2.1306e-01,\n",
       "           3.3852e-01,  2.5943e-01, -9.1834e-02,  5.2227e-01, -2.1885e-02,\n",
       "          -1.6835e-01,  3.1658e-01,  2.3124e-02, -2.2045e-01,  1.7294e-01,\n",
       "          -2.5761e-01,  2.6423e-01,  1.0870e-01, -4.9724e-01,  3.6477e-01,\n",
       "           3.7087e-01,  1.5082e-01, -3.0923e-01,  1.8171e-01,  3.9828e-01,\n",
       "          -3.1100e-01,  6.8356e-01,  1.6607e-04,  2.2973e-01,  6.3476e-02,\n",
       "          -1.1506e-01, -5.5647e-01, -3.1374e-01,  2.1509e-01,  9.2105e-02,\n",
       "          -5.0547e-01, -9.5287e-02,  1.5106e-01, -4.9544e-01,  7.8060e-01,\n",
       "           6.2385e-01, -3.7763e-01, -3.5664e-01, -7.1127e-02,  1.1295e-01,\n",
       "           1.2269e-01, -3.3879e-01,  2.9572e-01,  1.1312e-01,  1.0076e-01,\n",
       "          -1.0219e+00, -2.9326e-01,  2.4530e-01,  4.9147e-02, -4.2291e-01,\n",
       "           5.9939e-01,  7.9206e-02,  5.0359e-02,  3.8428e-01,  3.0534e-01,\n",
       "           1.4161e-01,  4.3910e-01,  7.0775e-02, -2.1375e-01,  6.5311e-02,\n",
       "           2.5600e-01, -3.0970e-01, -3.2766e-01, -6.3757e-01,  5.3684e-01,\n",
       "           1.5068e-01,  8.4160e-01,  6.5313e-01]], device='mps:0'),\n",
       " tensor([[-0.3056, -0.7273, -0.4395,  ...,  0.4157,  0.6836,  0.8789],\n",
       "         [-0.4133, -0.7676, -0.3773,  ...,  0.1480,  0.3258,  0.1579],\n",
       "         [-0.3292, -0.2590, -0.5016,  ..., -0.0148,  0.3509,  0.1607],\n",
       "         [-0.3029, -0.6449, -0.6165,  ...,  0.0458,  0.0831,  0.1345],\n",
       "         [-0.2375, -0.1241, -0.3509,  ..., -0.2516,  0.3497,  0.2568]],\n",
       "        device='mps:0'),\n",
       " tensor([[-0.2021, -1.0367, -0.7009,  ...,  0.0666,  0.4522,  0.2952],\n",
       "         [-0.2804, -0.8977, -0.7813,  ...,  0.0441,  0.7521,  0.7393]],\n",
       "        device='mps:0'),\n",
       " tensor([[-1.3933e-01, -5.2264e-01, -3.6336e-01, -7.4571e-01,  2.9388e-02,\n",
       "          -2.4191e-01, -4.2087e-01, -2.3909e-02, -3.0058e-01,  7.4198e-01,\n",
       "           1.4660e-01,  4.5423e-01, -3.8266e-02,  3.0181e-02, -3.8863e-01,\n",
       "           1.2727e-01,  7.9298e-01,  5.1457e-01,  2.8817e-01,  3.1548e-01,\n",
       "           2.2554e-01,  3.9253e-01,  2.1343e-01,  1.8514e-01, -3.8485e-01,\n",
       "          -4.2508e-01, -2.1970e-01, -3.8046e-01,  3.8814e-02, -2.6990e-01,\n",
       "          -2.8576e-02,  4.8749e-01,  1.9990e-01,  8.3447e-01, -1.7562e-01,\n",
       "          -7.3334e-02,  1.8834e-01, -2.7052e-01,  5.8200e-01, -1.5011e-01,\n",
       "           6.2756e-01, -1.8544e-01,  2.2635e-01,  1.7700e-01,  3.8184e-01,\n",
       "          -1.6745e-01,  1.8327e-02, -6.7638e-01,  2.9794e-01, -1.1273e-01,\n",
       "          -2.8189e-01, -8.1443e-02,  6.0670e-01, -3.1280e-03, -6.2087e-01,\n",
       "          -3.2929e-01, -6.2259e-02,  2.9880e-01, -1.9456e-01, -1.5492e-01,\n",
       "          -1.6745e-01,  2.5656e-01, -1.6718e-01, -2.8485e-01,  1.6184e-01,\n",
       "          -4.7529e-01, -2.9647e-01, -6.9564e-01, -1.8800e-01, -8.9128e-02,\n",
       "           2.1451e-01,  4.2546e-01,  2.1691e-02, -3.9388e-01,  1.5965e-01,\n",
       "          -4.5473e-01,  1.8070e-01, -5.4699e-02, -7.1267e-01,  2.5134e-01,\n",
       "           2.8863e-01,  6.2410e-02,  1.2044e-01, -5.6229e-01, -7.7670e-02,\n",
       "          -2.3292e-01,  4.6879e-01,  1.1227e-01,  2.1748e-01, -3.0857e-03,\n",
       "           2.1742e-01,  7.3158e-04, -5.0184e-01, -2.4559e-01,  9.3496e-02,\n",
       "           6.0028e-04,  1.8155e-01, -1.1384e-01,  8.3375e-04, -4.5379e-01,\n",
       "           1.7747e-02,  5.9367e-02,  2.4594e-01,  4.3640e-01, -5.6070e-01,\n",
       "          -5.3067e-01,  2.0357e-01, -5.3519e-02, -5.6847e-01, -2.1183e-01,\n",
       "           9.2360e-02, -6.7028e-02,  1.7329e-01, -2.2628e-01, -5.8184e-01,\n",
       "           6.2133e-02,  1.6328e-01,  3.1432e-01, -2.4532e-01,  4.4845e-01,\n",
       "           5.3427e-01, -1.7790e-01, -2.6353e-01, -1.8183e-01, -6.8791e-02,\n",
       "          -2.4959e-01,  7.4004e-01, -4.1128e-01,  1.8567e-01,  2.5068e-01,\n",
       "          -4.8550e-02,  3.2672e-01, -5.3629e-01,  8.8152e-03,  2.7561e-01,\n",
       "           2.5860e-01,  3.0257e-01, -5.0521e-01,  6.1676e-03,  2.0717e-01,\n",
       "           1.8931e-01, -1.7507e-01, -3.2335e-01, -4.6836e-01, -3.8741e-02,\n",
       "          -1.4006e-01, -8.5986e-02, -3.8146e-01, -2.1721e-01, -2.7724e-01,\n",
       "          -1.0208e+00, -8.0983e-01,  5.1063e-01, -1.7955e-01,  3.4303e-01,\n",
       "          -3.3650e-01,  3.7763e-01,  3.3489e-01, -7.0052e-02, -6.3932e-02,\n",
       "           4.6093e-01, -9.0858e-02,  2.9510e-01, -5.1305e-01,  1.0730e-01,\n",
       "          -1.6521e-01, -3.5034e-01, -2.4500e-01, -8.6472e-02,  1.7806e-02,\n",
       "           4.2684e-02, -1.0941e-01,  2.3489e-01, -1.5786e-01,  3.9429e-01,\n",
       "          -1.7393e-01,  1.3185e-01,  1.2030e-01, -4.2804e-01, -2.9486e-01,\n",
       "          -1.4254e-01, -1.4368e+00,  2.9971e-01, -2.0336e-01, -8.6846e-01,\n",
       "          -5.8579e-01,  1.3216e-01,  2.5853e-01,  4.0864e-02, -1.6827e-01,\n",
       "          -4.4813e-01, -9.0622e-02,  8.7098e-02,  6.8532e-02,  4.6310e-02,\n",
       "           1.0306e-01,  8.6077e-01,  2.3567e-01, -3.9002e-02,  2.8499e-01,\n",
       "          -6.7695e-01,  4.3153e-02,  2.0851e-01, -1.5885e-01,  1.5811e-01,\n",
       "           3.0066e-01,  7.2246e-02, -4.7059e-02,  3.5581e-01,  1.2753e-01,\n",
       "           3.3682e-02, -2.0437e-01, -2.1456e-01,  3.0714e-01, -1.5024e-01,\n",
       "           5.8856e-01, -4.3271e-01,  6.8798e-02,  8.7093e-01,  2.1124e-01,\n",
       "          -6.9191e-02, -3.9965e-03,  1.8051e-01, -3.5037e-01, -4.4747e-02,\n",
       "           2.4640e-01, -6.3875e-02,  1.1993e-01,  1.5837e-01, -2.5259e-01,\n",
       "          -1.4894e-01, -7.6375e-01,  4.0287e-01,  5.2248e-01, -3.5506e-01,\n",
       "          -4.1345e-01,  5.7227e-02,  5.3562e-01, -2.4304e-01, -1.4680e-01,\n",
       "          -1.6282e-01, -6.4948e-01, -6.1477e-01, -5.0470e-02, -5.3886e-02,\n",
       "           1.3133e-01, -4.3501e-01,  5.0577e-01,  5.3968e-01,  1.9248e-01,\n",
       "           1.4489e-01, -1.2459e-02,  2.5170e-01,  5.2871e-01, -4.7676e-01,\n",
       "           6.4807e-02,  3.6417e-01, -2.2409e-01,  1.6106e-01, -3.4339e-01,\n",
       "          -5.3745e-01,  8.9028e-02, -1.0856e-01,  2.5626e-01,  2.7744e-01,\n",
       "           3.7364e-01,  3.6583e-01, -1.8760e-02, -2.0988e-01,  3.9445e-02,\n",
       "          -4.4352e-01,  6.7391e-02, -9.3527e-02, -8.1330e-01, -4.8004e-01,\n",
       "          -1.0621e+00, -3.2868e-01,  3.3382e-01,  2.5483e-01, -9.7929e-01,\n",
       "          -1.2229e-01,  1.9098e-01, -3.0939e-01, -2.6484e-01,  4.2220e-02,\n",
       "          -1.0712e-01,  1.3534e-01,  6.5773e-02, -4.0161e-01, -4.4003e-01,\n",
       "           8.7974e-02,  1.5655e-01, -1.6442e-01,  3.1387e-01,  3.5774e-01,\n",
       "          -3.4515e-01,  3.3456e-01,  2.9267e-01, -4.3051e-02, -3.0173e-01,\n",
       "          -2.2406e-01,  3.9425e-01,  1.7629e-01, -2.0246e-01,  4.3214e-01,\n",
       "          -5.8601e-02,  4.3362e-01,  3.6400e-01, -5.5387e-01, -4.2636e-01,\n",
       "          -3.2024e-03,  8.8936e-01, -4.6996e-02, -3.1838e-01, -2.4169e-01,\n",
       "          -8.0222e-01, -1.1456e-01, -3.1772e-01, -1.5116e-01, -6.5426e-01,\n",
       "           3.1801e-01,  3.6388e-01, -1.4231e-01,  1.5599e-01, -2.0752e-01,\n",
       "           4.2292e-01,  5.8761e-01,  1.6107e-01,  8.8206e-02, -2.5728e-01,\n",
       "          -2.1201e-01, -8.1527e-01, -2.8331e-01, -2.2382e-01, -1.9847e-01,\n",
       "           5.4644e-01,  3.5344e-01,  4.9856e-03, -1.3218e-01, -5.7787e-01,\n",
       "           5.6449e-01, -2.8029e-01, -3.4292e-02,  5.5559e-01, -5.2118e-01,\n",
       "           2.4822e-01,  3.8816e-01, -2.4767e-01, -5.8174e-01,  3.8819e-01,\n",
       "           3.4226e-01, -2.1924e-01, -6.0760e-01, -1.5605e-01,  8.1583e-01,\n",
       "           7.4704e-01,  2.1641e-01, -2.8413e-01,  1.6836e-01,  4.1084e-01,\n",
       "          -5.4776e-01, -5.8217e-03, -3.8318e-01, -3.8142e-01,  2.7298e-02,\n",
       "           2.2778e-01,  3.6565e-01, -7.6888e-02,  8.2797e-02,  2.1583e-01,\n",
       "           9.4972e-02, -2.5290e-01, -2.4647e-01, -4.4155e-02,  4.1593e-01,\n",
       "          -2.6373e-01, -5.6246e-01,  3.9127e-01, -6.5850e-01, -1.3179e-01,\n",
       "          -1.2510e-01, -7.0780e-02, -1.0739e-01,  1.7258e-02, -3.0807e-01,\n",
       "          -9.2946e-02, -5.3918e-02,  6.2759e-01,  1.6127e-01, -2.1469e-01,\n",
       "           2.9675e-01,  1.4404e-01, -5.2568e-02,  6.5011e-01,  1.8180e-01,\n",
       "          -1.4166e-01,  1.9872e-01, -8.1174e-01,  1.3462e-01,  3.9702e-01,\n",
       "          -4.0440e-01, -2.5367e-01,  1.7926e-01,  6.3953e-01,  1.3922e-01,\n",
       "          -4.7511e-01,  5.1666e-02,  1.9152e-01,  2.5495e-01,  1.6456e-01,\n",
       "          -3.7630e-01,  3.4246e-01,  1.4655e-01,  3.1121e-01, -4.8961e-01,\n",
       "           1.1988e-01, -9.1983e-02, -6.5733e-02,  1.6510e-01, -4.7233e-01,\n",
       "           5.0445e-01,  1.4275e-01,  2.6854e-01,  4.4775e-01, -1.1052e+01,\n",
       "           2.6608e-01,  7.2773e-01, -4.3318e-01,  1.8903e-02,  2.4031e-02,\n",
       "           1.7421e-01, -2.3417e-01,  4.0635e-01, -3.3881e-01,  8.8490e-02,\n",
       "          -4.8197e-01,  1.3937e-02,  3.3761e-01, -2.6768e-01, -4.9688e-01,\n",
       "          -6.7492e-02, -7.1474e-01,  1.3321e-01,  3.7926e-02,  3.8882e-01,\n",
       "          -3.6965e-01,  1.3047e-01,  1.5706e-01, -2.2014e-01, -2.6751e-01,\n",
       "           3.0768e-02, -4.4950e-02,  2.6756e-01,  2.7868e-01,  2.0679e-01,\n",
       "           2.6307e-01,  4.0088e-02,  7.0271e-01,  1.5506e-02, -7.1500e-01,\n",
       "           1.2696e-01, -8.7776e-02, -1.5741e-01,  2.1645e-02, -1.3100e-01,\n",
       "          -2.9714e-01, -4.5420e-01,  3.1297e-01,  1.7327e-01,  6.3431e-01,\n",
       "           4.5766e-01,  5.0700e-01,  7.3783e-02, -4.2187e-01,  2.7987e-01,\n",
       "          -4.8199e-01,  3.0483e-01, -5.8551e-01, -4.2767e-02, -1.1369e-01,\n",
       "           2.4137e-01, -6.8979e-01,  1.1981e-01, -4.0837e-01, -4.9371e-01,\n",
       "           1.0339e-01,  2.5263e-01, -4.7958e-02, -4.8002e-01, -2.6382e-01,\n",
       "           2.7238e-01, -1.1096e-02, -8.0711e-02, -2.7166e-01,  8.5864e-02,\n",
       "           2.5990e-01, -2.1789e-01,  1.1639e-01, -1.7145e-01,  5.3762e-01,\n",
       "          -2.4231e-01, -1.5278e-01,  1.5882e+00, -2.1549e-01,  1.8216e-01,\n",
       "           1.0131e-01, -6.1305e-02, -5.8755e-01, -9.4219e-02, -5.9852e-01,\n",
       "          -2.2028e-01, -1.6019e-02,  1.8032e-01,  6.8590e-01, -3.6024e-01,\n",
       "          -6.1643e-01,  6.7971e-02, -1.6929e-01,  5.1900e-01,  6.8247e-01,\n",
       "          -3.5528e-01,  2.3071e-01,  4.5806e-01,  1.3427e-03, -9.7602e-01,\n",
       "           6.5740e-02, -1.5817e-02, -2.8561e-01,  5.7302e-02, -2.3564e-02,\n",
       "           2.8248e-01, -2.0198e-01, -1.4027e-01,  5.9058e-01, -2.0903e-01,\n",
       "           3.7666e-01,  1.0753e-01, -4.6412e-02, -2.7627e-01, -1.0103e+00,\n",
       "           3.5089e-01,  5.4100e-01, -3.8527e-01, -4.7432e-01,  1.4438e-01,\n",
       "          -3.2909e-01,  5.6669e-02, -4.2903e-02, -4.1840e-01,  1.2377e-02,\n",
       "           8.4912e-03,  1.3703e-01,  1.5635e-01, -2.9319e-01,  2.5524e-01,\n",
       "           4.6176e-01, -3.6045e-01,  1.0871e-01, -7.5278e-01, -5.8340e-01,\n",
       "          -6.6798e-01, -2.3791e-01,  5.3597e-01, -9.1541e-03, -1.1783e-02,\n",
       "           5.0111e-01, -1.0529e-01, -3.8790e-01,  4.3568e-03,  2.2031e-01,\n",
       "           3.8152e-02,  5.9409e-01, -9.1800e-02,  2.7011e-01, -1.0014e-01,\n",
       "           1.8189e-02, -4.1231e-01, -4.4161e-01, -1.1575e-01, -2.8353e-01,\n",
       "          -2.9683e-01, -9.1144e-02, -2.8434e-01, -2.2102e-01, -1.0366e-01,\n",
       "           3.5757e-01,  2.4648e-02, -2.9537e-01,  5.0217e-01,  3.9952e-01,\n",
       "           2.1030e-01, -6.7497e-01,  1.8609e-01, -1.2081e-01, -9.3575e-03,\n",
       "           7.5404e-01, -1.9816e-01, -6.7074e-01, -6.0709e-01, -1.2867e-01,\n",
       "          -2.4325e-01,  1.3191e+00, -2.0078e-01, -2.5788e-01,  8.9017e-02,\n",
       "           2.4825e-01,  2.6195e-01,  8.9268e-02, -2.0553e-01,  4.2232e-01,\n",
       "          -1.5465e-01, -3.9382e-01,  2.2085e-01, -1.8031e-01,  3.1756e-01,\n",
       "           3.5950e-01,  1.8495e-01, -2.8083e-01,  4.1039e-01,  1.6438e-01,\n",
       "          -1.0637e-01, -1.3631e-01, -5.1082e-02, -2.2393e-01,  7.6950e-01,\n",
       "          -3.2863e-01,  4.9487e-01,  4.5303e-03,  2.3918e-02, -3.8988e-01,\n",
       "           5.4256e-01,  1.2741e+00, -9.6497e-02, -3.6449e-01,  6.7230e-01,\n",
       "           2.8178e-01,  6.1129e-01,  2.0836e-01,  1.5706e-01,  2.4233e-01,\n",
       "           3.8769e-01,  1.6838e-02, -2.0980e-01,  3.7760e-01,  2.1249e-01,\n",
       "           6.3470e-01, -5.8037e-01, -2.0565e-01,  1.6860e-01,  2.0599e-01,\n",
       "          -4.2280e-01,  5.7974e-01, -2.7815e-02,  4.5913e-01,  3.3985e-01,\n",
       "          -1.1179e+00, -1.3277e-01, -3.4829e-02, -3.8464e-01, -5.6561e-02,\n",
       "          -6.2926e-01, -3.2293e-01, -1.0025e-01, -4.3785e-01,  2.7089e-01,\n",
       "           6.1548e-01, -1.2074e+00,  3.9680e-01,  6.3812e-01, -1.3517e-02,\n",
       "           2.7001e-01,  1.7385e-01,  4.5595e-01,  1.2414e-01,  1.5220e-01,\n",
       "           2.7638e-01,  2.5052e-01, -6.0822e-01, -5.6230e-01,  1.1552e-01,\n",
       "           6.2844e-01,  1.7665e-01, -2.8351e-01,  2.9362e-01, -2.5733e-01,\n",
       "           1.9061e-01,  1.8865e-02, -2.9778e-01, -1.7398e-01,  1.6744e-01,\n",
       "           1.0653e-01, -7.2150e-01, -4.6318e-02,  3.3032e-01, -4.5811e-02,\n",
       "           3.6862e-01,  3.7510e-01,  2.9058e-01, -1.5111e-01,  8.8917e-03,\n",
       "          -9.0775e-02,  6.7447e-01, -1.0619e-01,  5.7329e-01, -1.5024e-01,\n",
       "          -4.2108e-01, -2.4152e-01,  4.3797e-02, -5.8392e-01,  4.0257e-02,\n",
       "          -2.2459e-01,  2.3211e-01, -1.7451e-01, -1.4293e-01,  5.2262e-01,\n",
       "          -1.3146e-01,  7.2502e-02, -7.4386e-01,  4.1497e-01,  2.0514e-01,\n",
       "           6.8001e-02,  8.4092e-01,  8.5104e-02,  1.8626e-01, -6.1130e-01,\n",
       "          -6.5182e-03, -8.5021e-01, -2.5211e-01,  3.0191e-01, -2.1448e-01,\n",
       "          -3.0428e-03, -1.4796e-01, -3.7356e-01,  1.0424e-01,  6.9559e-01,\n",
       "           8.1512e-01, -8.7100e-02, -6.0489e-01, -4.6493e-01,  6.7442e-02,\n",
       "           3.6615e-01, -4.1318e-02,  9.2536e-01,  4.5510e-01, -1.4513e-01,\n",
       "          -1.0385e+00, -4.3094e-01,  7.8504e-02,  4.7902e-01, -2.0514e-01,\n",
       "           9.3278e-01, -1.2942e-01, -1.1889e-02,  1.5334e-01,  1.5119e-01,\n",
       "           4.4616e-01,  5.5380e-01, -8.1820e-02, -7.0676e-01,  1.5334e-01,\n",
       "           5.2390e-01, -1.4322e-01, -1.2726e-01, -8.6649e-01,  3.2411e-01,\n",
       "           1.9242e-01,  1.0713e-01,  3.8929e-01]], device='mps:0'),\n",
       " tensor([[-0.0111, -0.5057, -0.4670,  ...,  0.2365, -0.0712,  0.4388],\n",
       "         [-0.3292, -0.2590, -0.5016,  ..., -0.0148,  0.3509,  0.1607]],\n",
       "        device='mps:0'),\n",
       " tensor([[-0.0400, -0.7229, -0.4817,  ...,  0.1800,  0.8100,  0.4046],\n",
       "         [-0.1636, -0.4891, -0.6446,  ...,  0.1583,  0.6701,  0.5511]],\n",
       "        device='mps:0'),\n",
       " tensor([[-1.0665e-01, -6.9445e-01, -6.4997e-01, -5.2548e-01, -4.4278e-01,\n",
       "          -3.5859e-01, -5.4435e-01,  4.2628e-01,  4.1308e-02,  6.5288e-01,\n",
       "           7.8914e-02,  1.6971e-01,  2.9829e-01, -3.6411e-01, -3.0660e-01,\n",
       "          -6.6551e-02, -3.1187e-01,  2.3371e-01,  4.8499e-01,  1.4651e-01,\n",
       "           4.1529e-01,  2.6838e-01,  1.1044e-01, -1.1498e-01, -7.1794e-01,\n",
       "           3.2349e-02, -1.5023e-02, -2.5431e-02,  4.4441e-01, -8.0531e-01,\n",
       "           5.5127e-02,  5.5975e-01, -1.9038e-01,  9.2704e-01,  1.0480e-01,\n",
       "          -3.5947e-01,  1.8613e-01,  9.6070e-02,  1.6055e-01, -4.2312e-02,\n",
       "           5.2652e-01,  3.8830e-02,  2.5546e-01,  2.7388e-01,  1.8116e-01,\n",
       "          -8.6480e-02,  4.1524e-01, -5.6409e-01,  3.1971e-01, -7.4267e-01,\n",
       "          -4.0433e-01, -1.1830e-01,  4.1100e-01,  2.1386e-01, -5.8188e-01,\n",
       "          -2.6177e-01,  7.8702e-03,  1.8190e-01, -8.0385e-01,  3.6534e-01,\n",
       "          -1.7487e-01,  7.8058e-01,  2.9615e-01,  1.2489e-01,  8.1541e-02,\n",
       "           7.0742e-02, -4.7815e-01,  3.8809e-01, -4.3715e-01, -2.3010e-01,\n",
       "           7.8524e-02,  2.9064e-01,  2.3254e-02, -2.9357e-01,  5.1809e-01,\n",
       "          -3.8674e-01,  1.3434e-01, -5.3327e-01, -2.4359e-01, -3.0220e-01,\n",
       "           2.8505e-02,  3.2440e-01, -1.3390e-01, -1.8060e-01, -1.9198e-01,\n",
       "          -1.7373e-02,  1.2433e-01,  1.7821e-01,  9.0212e-02,  1.1798e-01,\n",
       "           2.1798e-01,  2.8624e-01, -5.3418e-01, -3.4835e-02,  2.6402e-01,\n",
       "          -3.7156e-01,  1.2733e-01,  3.0393e-01,  6.9169e-02, -5.9082e-02,\n",
       "           7.8029e-02, -5.5067e-02,  1.3400e-01,  1.5921e-01, -5.4415e-01,\n",
       "          -5.6557e-01,  3.0321e-01, -2.0676e-02, -6.8780e-01, -3.0978e-01,\n",
       "          -1.5930e-01, -1.5410e-01,  3.8364e-01, -7.0438e-01, -3.3754e-01,\n",
       "          -3.9570e-02,  2.6760e-02, -9.4876e-02, -8.8278e-02,  3.6210e-01,\n",
       "           1.4247e-01, -3.2138e-01,  4.0399e-01, -4.8559e-02,  2.8754e-01,\n",
       "          -3.3948e-01,  5.2511e-01, -4.0676e-02,  1.2206e-01,  4.5884e-01,\n",
       "          -2.7222e-01,  2.2809e-02, -4.0252e-01, -5.3121e-02, -1.5854e-01,\n",
       "           1.3503e-01, -4.7962e-02, -4.5411e-01,  2.0755e-01, -1.2961e-01,\n",
       "          -2.1070e-01,  9.4044e-02, -5.2534e-01, -8.8992e-02,  1.1474e-01,\n",
       "          -8.7369e-02,  6.3174e-02, -3.3044e-01, -4.3266e-01, -9.0922e-03,\n",
       "          -3.3971e-01, -7.7308e-01,  4.2227e-01, -1.9395e-01,  2.0966e-03,\n",
       "          -4.8478e-02, -8.7079e-02, -1.1355e-01,  8.7067e-03,  1.7030e-01,\n",
       "           8.2548e-01,  9.4309e-02,  2.2952e-01, -5.1323e-01, -2.7382e-02,\n",
       "          -7.4625e-02, -6.4664e-01, -3.7294e-01, -7.5044e-02, -1.3075e-01,\n",
       "           1.1090e-01, -2.2257e-01, -1.1683e-01, -7.1724e-01,  1.2141e-02,\n",
       "          -7.1616e-01,  1.0957e-01,  4.9414e-02,  4.6452e-01, -4.7771e-01,\n",
       "          -6.2116e-02, -1.0230e+00, -4.3717e-01, -1.3854e-01,  1.1959e-01,\n",
       "          -6.3614e-01,  2.8485e-01,  5.3316e-02,  1.4425e-01, -1.2701e-01,\n",
       "           1.1468e-01,  1.0378e-01,  2.2657e-01,  4.5452e-01,  6.0309e-02,\n",
       "           2.8160e-01,  7.2084e-01,  3.4962e-01,  1.9043e-01,  1.7958e-01,\n",
       "          -6.3122e-01,  1.6594e-01, -1.8927e-01,  4.0981e-01,  1.1560e-01,\n",
       "           1.5491e-01, -3.3843e-01,  1.2218e-01,  4.1821e-02, -2.8251e-01,\n",
       "           4.5925e-02, -1.2293e-01, -3.3222e-01, -2.4137e-01, -2.4180e-01,\n",
       "           4.2658e-01, -1.3335e-01,  3.6570e-02,  5.4598e-01,  4.4224e-01,\n",
       "          -1.6394e-01, -1.0208e-01,  3.4157e-01, -5.0501e-01, -3.2383e-01,\n",
       "           2.5948e-01, -1.4515e-01,  1.2982e-01, -2.1329e-01, -1.3986e-01,\n",
       "           3.6247e-01, -1.1847e+00,  6.2256e-01,  4.0287e-01,  1.4696e-01,\n",
       "          -3.2548e-01, -1.1554e-01,  7.6921e-02, -1.2451e-01,  2.7053e-02,\n",
       "          -3.7552e-01, -4.8231e-01, -7.8876e-01,  1.9418e-01,  4.4233e-03,\n",
       "           1.3424e-01, -2.0404e-01,  8.8641e-01,  4.8270e-01,  2.6514e-01,\n",
       "          -1.5688e-01, -3.7954e-01,  2.7514e-01,  1.4746e-01, -1.9147e-01,\n",
       "          -1.7092e-01, -4.0202e-02, -6.5738e-02,  2.1582e-01, -5.7746e-02,\n",
       "          -3.0501e-01, -1.5416e-01,  4.6626e-01,  5.6230e-01, -3.6779e-01,\n",
       "           4.2320e-01, -2.1380e-01, -1.5448e-01, -1.6401e-01, -6.6331e-02,\n",
       "          -6.2280e-01,  4.5048e-01, -2.9947e-01, -5.5477e-01, -6.9476e-01,\n",
       "          -6.1714e-01, -1.1463e-01, -1.5801e-01,  3.4001e-01, -1.0923e+00,\n",
       "          -4.7798e-01, -2.6740e-01, -7.5305e-02, -5.6876e-02, -1.9841e-01,\n",
       "           1.2798e-01, -1.8713e-03, -2.5724e-01, -2.9486e-01, -5.1909e-01,\n",
       "          -1.8326e-01,  2.8424e-01, -3.5395e-02,  5.4425e-02,  4.4087e-01,\n",
       "          -1.6876e-01,  1.5525e-03,  4.6364e-02,  1.6716e-01, -2.6884e-02,\n",
       "          -3.4934e-01, -3.4393e-01, -2.0159e-01, -2.6575e-01,  7.8262e-02,\n",
       "           3.3187e-01,  2.3228e-01,  6.9913e-01, -1.8739e-01, -6.2384e-01,\n",
       "          -1.5203e-02,  4.6857e-01, -3.5032e-01,  7.9153e-02, -2.6876e-01,\n",
       "          -1.0763e+00, -3.3028e-01, -2.9107e-01, -4.8692e-01,  2.6798e-01,\n",
       "           2.6904e-01, -1.8307e-02,  1.3118e-01, -7.3555e-02, -4.8607e-01,\n",
       "           2.2648e-01,  6.2656e-01, -4.4016e-01,  2.4406e-01, -5.1103e-02,\n",
       "          -1.5351e-01, -4.2508e-01, -2.7789e-01, -1.1023e-01, -1.4566e-02,\n",
       "           1.9785e-01, -1.0571e-02,  2.1225e-01, -2.2896e-01, -1.5606e-01,\n",
       "           6.5487e-01, -3.4924e-01, -4.7779e-01,  3.5263e-01, -5.0230e-01,\n",
       "          -3.2940e-01,  3.8929e-01, -1.4882e-01,  1.0160e-01,  1.3304e-01,\n",
       "           2.1899e-01, -3.9687e-01, -6.1205e-01, -1.3897e-02,  4.4603e-01,\n",
       "           5.3010e-01,  6.8281e-01, -3.0921e-01,  3.9946e-01,  2.5134e-01,\n",
       "          -5.4473e-01,  3.8569e-01, -2.2241e-01, -1.3654e-02, -6.9072e-02,\n",
       "           1.9577e-01,  4.0704e-01, -5.0823e-01,  3.5418e-02,  1.6317e-01,\n",
       "           1.1472e-01, -2.2650e-01, -1.4876e-01,  5.5815e-01,  3.4787e-01,\n",
       "          -1.2812e-01, -4.4903e-01,  4.5639e-01,  4.2230e-02, -2.1032e-01,\n",
       "           1.3650e-01, -5.6099e-02,  7.8627e-02, -4.4949e-01,  7.3304e-02,\n",
       "          -1.3549e-01,  3.2564e-01,  7.2024e-01,  2.4774e-01, -1.2706e-01,\n",
       "          -6.2105e-02, -3.6559e-01, -2.8637e-01,  5.1323e-01, -1.2305e-01,\n",
       "          -4.4164e-01,  3.5580e-01, -1.6701e-02,  1.2088e-01,  3.7206e-01,\n",
       "          -2.4596e-01, -2.1695e-01,  2.6575e-01,  3.9147e-01,  2.4953e-02,\n",
       "          -3.3297e-01,  3.0263e-01,  2.4747e-01, -3.6885e-02,  2.2992e-01,\n",
       "           5.6460e-02,  7.7857e-01, -9.1012e-02, -5.0115e-02,  1.6216e-01,\n",
       "           1.2583e-01, -3.5270e-01,  5.6855e-01,  2.4005e-02, -9.8946e-01,\n",
       "           8.7014e-01,  3.1844e-01,  6.0177e-01,  3.5398e-01, -1.1426e+01,\n",
       "           4.9936e-01,  6.9161e-01, -5.1674e-01,  1.2811e-01,  3.7344e-01,\n",
       "           5.0341e-01, -4.5127e-01,  3.7097e-01, -1.6880e-01, -3.2606e-01,\n",
       "          -4.0349e-02,  1.2820e-02,  1.7268e-01, -3.2271e-01,  1.0126e-03,\n",
       "           1.8052e-01, -5.8249e-01,  2.6097e-01, -1.0350e-01,  4.6722e-01,\n",
       "           3.1980e-02,  9.1450e-02,  2.5100e-01, -2.0090e-01, -6.2740e-01,\n",
       "           7.7779e-02, -2.1384e-01,  1.1281e-01,  2.1237e-01,  1.2701e-01,\n",
       "           5.0231e-02, -1.0118e-01, -8.7316e-03, -1.5773e-01, -5.4077e-01,\n",
       "          -3.1373e-01, -3.7578e-01,  3.0592e-01, -3.0799e-03,  9.1115e-02,\n",
       "           1.7804e-01, -1.5209e-03,  3.5748e-01,  2.0586e-01,  4.2981e-01,\n",
       "           1.1112e+00,  6.0362e-01, -9.1425e-02, -2.7498e-01, -1.5020e-01,\n",
       "          -7.2183e-01,  1.9577e-01, -7.0263e-01, -9.9048e-02, -2.3871e-02,\n",
       "           7.6208e-01, -5.5795e-01,  1.7443e-01, -1.3724e-02, -1.4452e-01,\n",
       "           3.8289e-01, -2.0825e-01,  1.1136e-01, -2.6944e-01, -1.8138e-01,\n",
       "           3.6497e-01, -2.6563e-01,  7.7600e-02, -3.7321e-01,  3.0179e-01,\n",
       "          -2.0930e-01,  5.4447e-01,  1.8742e-01, -6.1380e-01, -8.6347e-02,\n",
       "           2.9106e-02, -1.1558e-01,  1.8456e+00,  7.8612e-03, -1.8617e-01,\n",
       "           3.5130e-01,  1.4630e-02, -5.8265e-01, -6.1126e-01, -1.1133e-01,\n",
       "          -3.3301e-02,  3.3427e-01,  5.5392e-01,  4.1812e-01, -1.8483e-01,\n",
       "          -3.3355e-01,  3.6994e-02,  1.4980e-01,  2.3689e-01,  5.1889e-01,\n",
       "          -5.8064e-01,  1.8907e-01,  1.8275e-01,  7.9104e-02, -6.0101e-01,\n",
       "          -1.7290e-02, -3.0109e-01, -1.5688e-01,  1.2422e-01, -6.2941e-02,\n",
       "           1.5333e-01,  7.5043e-02, -6.7126e-02, -1.1621e-01, -4.1156e-01,\n",
       "           4.2066e-01,  3.1805e-01,  9.7686e-02, -3.1265e-02, -1.8783e-01,\n",
       "           1.9609e-01,  6.2797e-01, -4.7234e-01, -1.5012e-01, -1.6414e-01,\n",
       "           7.4878e-02,  5.2318e-01,  4.5732e-02, -6.8070e-02,  4.0506e-01,\n",
       "          -7.0231e-01,  1.8167e-02, -1.8888e-01,  2.6671e-01,  5.7947e-01,\n",
       "           2.5370e-01,  4.3893e-01,  3.5784e-01, -1.4140e-01, -3.4918e-01,\n",
       "          -6.9173e-01, -2.2794e-01,  2.8249e-01, -2.7613e-01, -2.5971e-02,\n",
       "           4.0422e-03, -7.8611e-02, -5.4126e-02,  5.1139e-01,  1.6869e-01,\n",
       "           6.9469e-01,  3.2669e-01, -1.7580e-01,  6.5794e-02, -1.4254e-02,\n",
       "          -2.8400e-01,  2.3203e-01,  7.3419e-02, -1.0833e-01, -4.6831e-01,\n",
       "          -3.8749e-01,  1.6927e-01,  3.4941e-01, -4.5530e-01, -1.9854e-01,\n",
       "           1.0091e-01,  7.5958e-02, -2.7452e-02,  1.2700e-02,  2.9530e-01,\n",
       "           2.0690e-02, -4.2450e-01, -2.5327e-01,  4.7477e-04,  2.3976e-02,\n",
       "           4.1348e-01, -1.2106e-01, -1.2069e-01, -3.7981e-01,  2.9268e-02,\n",
       "          -4.4070e-01,  1.0990e+00, -1.3018e-01, -4.5414e-01,  1.4605e-01,\n",
       "           6.0308e-01, -1.1264e-01,  1.1018e-01, -9.8228e-02,  5.0482e-01,\n",
       "          -2.8651e-01,  9.3654e-02,  2.7972e-01, -1.7402e-01,  3.8553e-01,\n",
       "           5.8397e-02,  2.9230e-02, -2.2794e-01,  3.5888e-02,  1.4718e-01,\n",
       "          -5.6763e-01,  8.1753e-03, -2.7674e-01,  5.0954e-02,  5.0624e-01,\n",
       "          -3.1931e-01, -7.4513e-03,  2.0958e-01,  3.0460e-01,  1.5467e-01,\n",
       "           3.3133e-01,  1.0455e+00, -9.3735e-02, -1.8127e-01,  4.5611e-01,\n",
       "           1.4621e-01, -3.2100e-01,  3.6532e-01,  7.8640e-02,  4.1232e-01,\n",
       "           4.3991e-01, -3.8798e-02, -3.3213e-01, -3.9349e-01,  1.3334e-01,\n",
       "           4.4440e-01, -5.2256e-01, -4.5308e-01, -2.5861e-01,  4.4983e-01,\n",
       "          -2.6714e-03,  3.1331e-01,  4.9067e-03,  4.8779e-01,  5.4587e-01,\n",
       "          -9.3822e-01,  3.1480e-02,  8.9428e-02, -1.7189e-02, -2.8958e-02,\n",
       "          -5.1944e-01, -1.3400e-01, -2.4563e-01, -3.3220e-01,  1.7955e-01,\n",
       "           4.7089e-01, -7.4036e-01,  1.7270e-01,  4.4420e-01, -6.6975e-01,\n",
       "           1.8994e-01, -1.5834e-01,  4.2625e-03,  5.1366e-01, -6.0538e-02,\n",
       "          -7.8716e-02,  2.6791e-02, -6.3351e-02, -7.5044e-01,  3.1202e-01,\n",
       "           5.6793e-01, -1.6540e-01,  1.4279e-01,  2.0289e-02, -3.3418e-01,\n",
       "          -3.3213e-03, -4.0291e-01, -2.1839e-01, -8.8785e-02, -1.1882e-01,\n",
       "          -9.9165e-02, -7.4227e-02,  2.2713e-01,  4.0295e-01, -2.4612e-01,\n",
       "           8.1354e-02,  4.9667e-01,  4.9615e-01,  8.6974e-03,  2.4444e-01,\n",
       "           7.0335e-02,  2.0545e-01, -1.7978e-01,  2.0159e-01, -5.0663e-02,\n",
       "          -4.2534e-01,  3.3628e-01, -1.3971e-01, -3.5809e-01, -1.2016e-01,\n",
       "          -2.9041e-01,  1.0708e-01, -2.2698e-02, -2.5389e-01,  2.2835e-01,\n",
       "           2.2191e-02,  1.2626e-01, -6.2222e-01,  1.3444e-01,  1.7726e-01,\n",
       "           5.1205e-02,  8.1369e-01,  1.5905e-01,  3.2786e-01, -8.0516e-02,\n",
       "           1.7447e-01, -7.0375e-01, -4.1953e-01,  2.2980e-01, -1.4134e-01,\n",
       "          -3.5166e-02, -3.2984e-01,  2.4090e-01,  2.0840e-01,  5.5069e-01,\n",
       "           4.0457e-01, -9.9010e-02, -1.1570e+00, -2.7392e-01, -3.2235e-02,\n",
       "          -1.5406e-02, -1.2735e-01,  2.0365e-01, -4.8449e-01,  2.6017e-01,\n",
       "          -7.4287e-01, -3.4545e-01, -9.3460e-02,  4.1150e-01, -3.6667e-01,\n",
       "           5.4990e-01, -6.5561e-02,  2.8974e-01,  4.4765e-01,  7.2593e-02,\n",
       "           2.5255e-01,  6.1210e-01,  4.6853e-02, -8.1408e-01, -4.0423e-02,\n",
       "           2.4193e-01, -3.6803e-01,  9.6091e-02, -8.0278e-01,  3.4686e-01,\n",
       "           2.8446e-01,  7.5386e-01,  4.0509e-01]], device='mps:0'),\n",
       " tensor([[-0.2446, -0.3533, -0.5318,  ...,  0.0083,  0.2499,  0.4224],\n",
       "         [-0.1923, -0.6475, -0.6643,  ...,  0.0531,  1.0075,  0.1900],\n",
       "         [-0.2881, -0.3311, -0.4859,  ..., -0.1017,  0.3507,  0.1910],\n",
       "         [-0.3288, -0.2892, -0.7484,  ...,  0.0909,  0.9524,  0.5098],\n",
       "         [-0.3273, -0.4546, -0.7480,  ...,  0.2674,  1.1347,  0.4993]],\n",
       "        device='mps:0'),\n",
       " tensor([[-0.2422, -0.2850, -0.1854,  ...,  0.0219,  0.7967,  0.2399],\n",
       "         [-0.1975, -0.7585, -0.6995,  ...,  0.0778, -0.0070,  0.3890]],\n",
       "        device='mps:0'),\n",
       " tensor([[ 0.0297, -0.5113, -0.7001,  ...,  0.0205,  0.2145,  0.5025],\n",
       "         [-0.2251, -0.4048, -0.6157,  ...,  0.3186,  0.2124,  0.6282],\n",
       "         [-0.2427, -0.6267, -0.1893,  ...,  0.2335,  0.0749,  0.3544],\n",
       "         [-0.3292, -0.2590, -0.5016,  ..., -0.0148,  0.3509,  0.1607]],\n",
       "        device='mps:0'),\n",
       " tensor([[-1.9929e-01, -4.7277e-01, -9.1353e-01, -7.9424e-01, -4.1503e-01,\n",
       "          -4.4127e-01, -7.5255e-01,  3.1785e-01, -7.8880e-03,  4.7854e-01,\n",
       "           1.6369e-01, -1.5301e-01,  9.8473e-02, -2.5835e-01, -5.9441e-01,\n",
       "          -1.4721e-01,  2.3256e-02,  4.8634e-01,  5.1214e-01,  1.9188e-01,\n",
       "           7.9537e-01,  4.9176e-01,  1.3074e-01, -2.5996e-02, -7.5284e-01,\n",
       "           1.4525e-01, -1.4251e-01,  1.7984e-01,  2.1046e-01, -6.3649e-01,\n",
       "          -7.0903e-02,  2.0048e-01, -3.2711e-01,  1.0834e+00, -1.7641e-01,\n",
       "          -2.5468e-02,  1.6916e-01, -7.1098e-01, -1.3390e-01, -2.1303e-01,\n",
       "           4.6865e-01, -1.8250e-02, -1.9620e-03,  4.0179e-01,  2.8849e-01,\n",
       "           3.8086e-02,  1.4767e-02, -5.5226e-01,  2.1947e-01, -2.1303e-01,\n",
       "          -1.2382e-01, -4.6332e-02,  3.1580e-01,  2.1652e-01, -1.1413e+00,\n",
       "          -6.1922e-01,  3.2170e-01,  3.1767e-01, -1.9214e-01,  2.0664e-01,\n",
       "          -4.1786e-01,  3.0379e-01,  2.1673e-01,  3.6833e-01, -7.8662e-02,\n",
       "           1.5138e-01, -3.3366e-01, -9.5802e-01, -4.3563e-01, -8.4040e-02,\n",
       "           2.6562e-02,  3.8592e-01,  1.8270e-01, -3.4838e-01, -4.3242e-03,\n",
       "          -4.8208e-01,  3.9951e-01, -1.8842e-01, -5.2437e-01,  3.0999e-03,\n",
       "           1.8471e-01, -5.3980e-02,  6.9821e-02, -1.6609e-01,  1.2849e-01,\n",
       "           5.3368e-02,  1.9409e-01,  2.0953e-01,  5.1481e-01,  2.9702e-02,\n",
       "           4.7190e-01,  5.6391e-02, -1.0391e-02, -2.7907e-01,  3.6604e-01,\n",
       "          -1.5805e-01,  4.0231e-01, -3.7242e-01, -2.6040e-01, -4.5380e-01,\n",
       "          -5.2470e-02,  4.2374e-01,  5.3816e-02,  2.4742e-01, -5.3879e-01,\n",
       "          -6.7099e-01,  1.5662e-02, -2.8416e-01, -8.8003e-01, -1.0000e-01,\n",
       "          -2.3763e-01, -2.7911e-01,  1.9439e-01, -6.7699e-01, -3.8291e-01,\n",
       "           1.2942e-02, -4.0506e-01,  1.0587e-01, -1.3842e-01,  6.2952e-01,\n",
       "           3.3072e-01, -2.2649e-01,  1.5778e-01, -5.5456e-02,  1.1554e-01,\n",
       "          -2.3528e-01,  2.9234e-01, -1.7063e-01,  2.7626e-01,  3.3039e-01,\n",
       "           3.0572e-01,  1.4516e-01, -2.4069e-01, -9.5441e-02,  2.6812e-01,\n",
       "           1.1071e-01,  2.4822e-01, -3.4702e-01, -6.2351e-01, -9.7490e-03,\n",
       "           2.0659e-01, -7.0106e-02, -8.0330e-02,  1.3001e-02, -2.0375e-02,\n",
       "          -2.5428e-01, -1.6927e-01,  8.2839e-02, -2.2712e-01, -4.4586e-01,\n",
       "          -5.5310e-01, -2.3193e-01,  2.7520e-01, -3.1649e-01, -1.9272e-01,\n",
       "          -1.9154e-01,  2.9868e-01, -1.5100e-01, -1.1629e-01,  1.7536e-01,\n",
       "           8.0683e-01, -1.9597e-01,  2.5133e-01, -5.5101e-01, -2.2148e-01,\n",
       "           2.8449e-01,  2.4446e-02,  4.1068e-02, -2.6081e-01,  5.1321e-02,\n",
       "           3.4527e-01, -3.0352e-02,  1.9908e-01,  5.9377e-02, -5.6948e-02,\n",
       "          -8.2108e-01,  8.2629e-01,  3.9105e-01, -8.9403e-02,  3.1635e-01,\n",
       "           1.6112e-01, -1.3111e+00,  1.5663e-01, -3.8635e-01,  1.4435e-01,\n",
       "           6.0170e-04,  6.8859e-01,  3.1852e-01,  1.4703e-01, -1.9906e-01,\n",
       "           2.8942e-01,  5.8447e-02,  3.1431e-01,  3.3951e-01, -8.8561e-02,\n",
       "           2.8938e-01,  4.6240e-01,  1.7784e-01,  7.3560e-02,  2.1796e-01,\n",
       "          -7.8350e-01,  2.4292e-01,  2.4527e-01,  8.8320e-02,  1.5260e-01,\n",
       "           1.6644e-01, -2.3246e-01,  1.6347e-01,  6.1614e-02,  3.2612e-01,\n",
       "          -2.0774e-01, -3.3100e-01, -6.3338e-01, -2.8563e-01, -4.5019e-01,\n",
       "           6.9077e-01, -2.1932e-01, -2.5110e-01,  4.3339e-01,  1.5662e-01,\n",
       "          -2.3963e-01,  1.3464e-01, -1.7279e-01, -2.3546e-01, -4.0353e-01,\n",
       "           5.7954e-01, -4.9676e-01,  1.4865e-01, -6.6307e-02, -1.4713e-01,\n",
       "           1.7220e-01, -1.0079e+00,  4.7667e-01,  2.4966e-01,  2.0969e-01,\n",
       "          -2.2365e-01, -4.3317e-01,  8.2690e-02, -4.3509e-01, -3.8481e-02,\n",
       "          -2.9266e-01, -5.3687e-01, -7.3677e-01,  1.2553e-01,  8.1412e-02,\n",
       "           1.8074e-01, -1.0125e-01,  6.8775e-01,  6.7474e-01,  4.2922e-01,\n",
       "          -1.8120e-01, -3.6464e-01,  6.5699e-01, -2.0577e-01, -3.5300e-01,\n",
       "           3.9457e-01,  3.1983e-02, -3.6182e-01,  1.9538e-01, -5.8708e-01,\n",
       "          -4.0378e-01, -2.2225e-01,  4.5734e-01,  2.9422e-01, -3.9198e-01,\n",
       "           2.5561e-01, -2.3261e-02,  2.2396e-02, -2.7771e-01,  4.4748e-01,\n",
       "          -2.5728e-01,  1.8226e-02,  1.9174e-01, -5.2420e-01, -5.6998e-01,\n",
       "          -5.2015e-01,  3.4634e-01,  1.7257e-01,  1.5803e-01, -8.1192e-01,\n",
       "          -4.6806e-01,  6.1364e-01,  1.1708e-01,  9.5115e-02, -4.7374e-02,\n",
       "           2.3783e-01, -3.1228e-01,  1.7400e-01,  4.1419e-02, -4.8036e-01,\n",
       "          -5.0660e-02,  1.2909e-01, -3.0757e-01,  5.0616e-01,  3.6874e-01,\n",
       "          -1.7583e-01,  2.9881e-01,  7.7525e-03, -3.6199e-02, -2.0241e-01,\n",
       "          -1.1310e-01, -2.2071e-01, -4.5178e-01, -1.9585e-01,  6.3210e-02,\n",
       "           5.3845e-02,  2.2778e-01,  4.4002e-01, -2.8384e-01, -7.2818e-01,\n",
       "           2.0010e-01,  1.1893e+00, -8.4648e-02, -1.9909e-01, -1.9114e-01,\n",
       "          -8.9863e-01,  6.6110e-04, -5.6452e-02, -3.1647e-01, -6.3272e-01,\n",
       "           1.1279e-01, -4.3006e-01, -6.3246e-02, -1.7352e-01, -3.8680e-01,\n",
       "           1.1601e-01,  6.3529e-01,  3.8830e-01, -1.1351e-02, -1.1757e-01,\n",
       "          -2.8311e-01, -5.4473e-02, -3.5855e-01, -1.6883e-01, -3.7518e-01,\n",
       "          -1.2064e-02,  3.6152e-01,  3.7116e-02, -1.8946e-01, -5.4858e-02,\n",
       "           3.8808e-01, -4.8643e-01, -3.8395e-01,  1.1649e-01, -2.3856e-01,\n",
       "          -2.3052e-01,  1.5684e-01, -2.2064e-01, -2.5571e-01, -3.0521e-02,\n",
       "           9.9207e-02, -2.0966e-01, -3.6809e-01, -7.3776e-03,  6.2537e-01,\n",
       "           3.0363e-01,  2.3653e-02, -4.9898e-01,  8.2610e-01,  2.5370e-01,\n",
       "          -5.8888e-01,  2.3160e-01,  2.2636e-01, -8.1912e-02, -3.0692e-01,\n",
       "           8.5201e-02,  2.3673e-01, -3.9246e-01, -9.5247e-02, -1.0420e-01,\n",
       "           1.5530e-01, -4.0912e-01,  3.7023e-02,  1.5174e-01,  3.2801e-03,\n",
       "          -1.7646e-01, -4.6323e-01,  2.3108e-01, -2.4803e-01, -2.6237e-01,\n",
       "           2.3728e-01,  1.0933e-01, -1.7135e-01, -3.2972e-01, -1.5641e-01,\n",
       "          -2.3806e-01,  6.2307e-02,  4.9929e-01,  4.4998e-01, -1.2748e-01,\n",
       "           8.3781e-02, -9.1548e-02, -2.3370e-01,  1.1473e-02,  1.5062e-01,\n",
       "          -3.1937e-01,  2.6047e-01,  1.7520e-01,  3.5572e-02,  3.0967e-01,\n",
       "          -3.1343e-01,  9.6872e-02,  1.5276e-01,  1.8953e-01,  1.4137e-02,\n",
       "          -3.9719e-01,  1.0800e-01,  3.7868e-01, -4.2595e-01,  1.2659e-01,\n",
       "           8.4775e-02,  2.3621e-01, -5.5080e-01,  4.4941e-01,  1.8609e-02,\n",
       "          -1.7347e-01,  1.9538e-01, -1.1891e-01, -5.9889e-01, -2.7890e-01,\n",
       "           8.1396e-01,  3.0541e-01,  4.8767e-01,  1.8457e-01, -1.1240e+01,\n",
       "          -6.9319e-02,  8.7039e-01, -5.4841e-01, -4.9819e-01, -6.8730e-02,\n",
       "          -4.5975e-02, -1.0554e-01,  2.3776e-02, -1.6488e-01, -2.2850e-01,\n",
       "          -3.4583e-01, -2.5386e-02,  2.6381e-02, -3.3832e-01, -2.1593e-01,\n",
       "           4.1920e-02, -3.7123e-01,  3.6252e-01, -4.5952e-02,  8.4503e-01,\n",
       "          -5.3507e-01,  2.4689e-01,  2.4920e-01, -2.4199e-01, -2.1420e-01,\n",
       "          -1.0168e-01, -2.2476e-02,  2.4813e-01,  5.4666e-01, -7.3535e-02,\n",
       "           2.8333e-01, -8.0798e-02,  1.3560e-01, -3.3816e-01, -4.6449e-01,\n",
       "           1.0062e-01, -2.1484e-01,  4.2466e-01,  2.1106e-01, -5.9531e-01,\n",
       "          -2.9875e-01, -3.7313e-01,  5.4666e-01,  3.8626e-01,  7.7846e-01,\n",
       "           1.2129e+00,  8.4693e-01,  3.2582e-01, -9.7669e-02, -1.8289e-01,\n",
       "          -8.5451e-01,  3.7598e-01, -3.9299e-01, -1.2747e-01, -9.5632e-02,\n",
       "           4.6351e-01, -6.9287e-01,  4.3880e-02, -3.7283e-01, -1.5561e-01,\n",
       "           7.7716e-01,  1.9910e-01,  2.7873e-01, -3.5407e-01, -2.8023e-01,\n",
       "           4.0040e-01, -3.0489e-01,  2.4003e-01, -2.3937e-01, -2.2662e-01,\n",
       "          -4.6678e-01,  4.6932e-01,  2.9532e-01,  8.2868e-02,  1.1568e-01,\n",
       "          -4.1061e-02, -3.5932e-01,  1.4326e+00,  2.4279e-01, -6.1429e-02,\n",
       "           4.5854e-01,  2.7843e-01, -2.9915e-01, -9.1077e-01, -5.7336e-02,\n",
       "          -3.9430e-03,  9.4398e-02,  1.3102e-01,  7.0788e-01,  3.5885e-02,\n",
       "          -8.1004e-01, -1.7802e-01,  2.0003e-01,  3.9512e-01,  1.6930e-01,\n",
       "          -6.0915e-01,  4.8089e-02,  1.0716e-01, -1.9453e-02, -7.1556e-01,\n",
       "          -4.4702e-02, -3.5905e-01, -4.6949e-01,  5.5092e-01, -5.3614e-02,\n",
       "           3.9010e-02, -2.0928e-01,  1.1284e-01,  5.2534e-01, -2.2549e-01,\n",
       "           3.5372e-01,  1.5566e-01,  3.4678e-02, -3.7479e-01,  5.4208e-02,\n",
       "           4.7621e-01,  8.7817e-01, -2.2108e-01, -4.7994e-01, -2.3174e-01,\n",
       "           2.0383e-01,  1.4607e-01,  2.7097e-01, -6.2258e-01,  1.3273e-01,\n",
       "          -7.1104e-01,  2.4314e-01,  3.2820e-01,  9.2572e-02,  4.4619e-01,\n",
       "          -1.6396e-01,  3.3219e-01,  5.5077e-01, -8.2981e-01, -5.8629e-01,\n",
       "          -7.2361e-01,  2.7797e-02,  6.9036e-02, -5.0863e-01,  2.1168e-01,\n",
       "           1.1308e-01,  2.5612e-01,  3.7071e-02, -3.1052e-02,  1.1601e-01,\n",
       "           3.9106e-01,  3.2890e-01,  9.2080e-02,  1.6217e-02, -1.9252e-01,\n",
       "          -5.4040e-01, -9.0594e-03,  2.7607e-02,  1.3623e-01, -2.1968e-01,\n",
       "           9.7152e-02,  2.3546e-02, -1.7766e-01, -1.0872e-01,  4.4995e-02,\n",
       "          -5.7101e-02, -3.3122e-01, -6.8157e-02,  2.3682e-01,  3.5417e-01,\n",
       "           1.6633e-01, -2.5789e-01,  1.0620e-01, -4.0420e-01,  2.4758e-02,\n",
       "           1.9297e-01,  2.9987e-02,  2.1140e-01, -5.3458e-01, -3.6661e-01,\n",
       "          -1.8766e-01,  1.3024e+00, -2.2188e-01, -7.1848e-01,  2.4056e-01,\n",
       "           1.3753e-01,  1.5178e-02, -2.6584e-01, -1.5431e-01,  1.9085e-02,\n",
       "          -3.0554e-01, -3.9982e-01,  3.5080e-01, -2.6661e-02,  3.8423e-01,\n",
       "           4.3581e-01, -1.3838e-01, -1.8384e-01,  4.9272e-01,  1.0169e-01,\n",
       "          -3.6620e-01, -1.7033e-01, -3.8691e-01,  6.5345e-02, -7.6319e-02,\n",
       "          -4.5108e-01,  2.8021e-01,  3.6154e-02, -1.6075e-01,  2.2997e-01,\n",
       "           2.7373e-01,  1.1877e+00, -2.5678e-01,  1.8457e-02,  7.9242e-01,\n",
       "           2.3615e-01,  4.0053e-01,  7.9188e-02,  4.5226e-01,  1.1147e-01,\n",
       "           2.2174e-01, -3.3957e-01, -1.9149e-01,  1.2282e-01,  2.3292e-01,\n",
       "           5.9342e-01, -7.4440e-01, -4.8953e-01,  1.0369e-01,  6.3395e-01,\n",
       "          -2.5831e-02,  3.2336e-01, -3.5510e-01,  3.9149e-01,  5.9328e-01,\n",
       "          -9.2754e-01, -1.2099e-01, -1.5118e-01,  3.7112e-01,  2.6110e-02,\n",
       "          -3.6830e-01,  3.9884e-01, -2.2138e-01, -3.4428e-01,  2.4502e-01,\n",
       "           4.6183e-01, -8.0152e-01,  2.3669e-01,  8.4727e-01, -6.7025e-01,\n",
       "           1.8110e-01, -3.1925e-01,  3.1130e-01,  2.7687e-01,  4.0369e-01,\n",
       "           6.9927e-02,  3.0576e-01, -4.5065e-01, -4.9560e-01,  1.8690e-02,\n",
       "           8.3464e-01, -8.8543e-02,  1.7367e-02, -3.8948e-02,  2.9304e-01,\n",
       "          -1.1445e-01, -5.6694e-01, -3.6638e-01, -1.4762e-01, -2.3536e-01,\n",
       "           1.3585e-01, -2.1753e-04,  1.8472e-01,  2.1493e-01, -3.8257e-01,\n",
       "           1.7794e-01,  7.3040e-01,  6.3326e-01, -1.7591e-01, -2.8302e-01,\n",
       "          -2.3712e-01,  2.0428e-01, -1.2611e-01,  3.3745e-01, -8.0261e-02,\n",
       "          -2.1521e-01,  2.4862e-01,  2.6322e-01, -2.0006e-01,  3.2894e-01,\n",
       "          -4.1238e-01,  5.1251e-02, -5.9052e-02, -2.9048e-01,  2.9674e-01,\n",
       "          -8.3240e-03,  2.9523e-01, -5.4059e-01,  3.0519e-01,  2.7401e-01,\n",
       "          -1.9523e-01,  5.5406e-01, -1.0010e-01,  5.4378e-01, -2.1883e-01,\n",
       "           1.3050e-01, -7.5742e-01, -3.6809e-01, -1.0226e-01,  5.3774e-02,\n",
       "          -2.1019e-01, -2.4524e-01,  4.7402e-01,  7.0504e-01,  5.2690e-01,\n",
       "           2.3729e-01,  1.9236e-01, -7.0661e-01, -2.3716e-01,  2.6671e-01,\n",
       "           4.9506e-02, -3.9419e-01,  1.6661e-01,  2.4672e-02,  9.2694e-02,\n",
       "          -6.1972e-01, -7.1242e-01, -1.2949e-01,  2.4299e-01,  1.6418e-01,\n",
       "           9.0174e-01, -1.1228e-01,  4.5695e-02, -7.8232e-02,  1.4934e-01,\n",
       "           2.7662e-01,  7.9402e-01,  1.0119e-02, -7.1910e-01, -5.2436e-02,\n",
       "           2.3916e-02, -1.0689e-01,  4.6704e-03, -5.9216e-01,  5.8139e-01,\n",
       "           3.0180e-01,  4.9594e-01,  6.7783e-01]], device='mps:0'),\n",
       " tensor([[-6.1750e-01, -4.2829e-01, -2.3934e-01, -9.4116e-01, -3.4548e-01,\n",
       "          -6.5496e-02, -6.3874e-01,  9.7828e-01, -9.5892e-02,  4.0096e-01,\n",
       "           1.4919e-01, -5.1603e-03,  2.0980e-01,  1.3680e-01, -6.9506e-01,\n",
       "           1.6740e-01,  4.6094e-01,  2.1159e-01,  3.5951e-01, -4.4329e-02,\n",
       "           7.9832e-01,  3.4245e-01,  1.8019e-01, -2.5205e-01, -4.5876e-01,\n",
       "           5.8994e-01, -2.4628e-01, -9.2014e-02,  2.0509e-01, -8.1949e-01,\n",
       "          -1.3586e-01, -4.9757e-02, -3.6281e-01,  8.7602e-01,  1.1948e-01,\n",
       "          -2.7310e-01, -1.5276e-01, -2.7996e-01,  2.3494e-01, -6.3637e-02,\n",
       "           3.7814e-01,  2.6592e-02, -2.9347e-01,  3.1386e-01, -1.6421e-01,\n",
       "          -1.4057e-01,  1.4047e-02, -7.8429e-01,  1.3666e-02,  1.8955e-01,\n",
       "           5.8851e-02, -4.7205e-02,  1.9231e-01,  3.7030e-01, -1.0617e+00,\n",
       "          -2.6348e-01,  2.7603e-01,  2.9849e-01, -1.7932e-01, -5.7784e-02,\n",
       "          -1.9989e-01,  9.0201e-01,  1.7949e-01, -2.7822e-01,  1.7824e-01,\n",
       "          -4.4899e-01, -5.4601e-01, -6.0781e-01, -1.0682e-01, -1.1967e-01,\n",
       "          -1.0002e-01,  9.7528e-02,  1.4009e-01, -2.9385e-01,  2.3170e-01,\n",
       "           2.1813e-01,  4.7024e-02, -2.7652e-02, -4.3158e-01, -2.6546e-01,\n",
       "           3.6947e-01,  3.4425e-02, -1.4730e-02,  2.0326e-01, -5.8744e-01,\n",
       "          -2.0226e-01,  3.7959e-01,  1.5703e-01,  2.5914e-01,  2.2662e-03,\n",
       "           5.9967e-01,  1.2024e-02,  8.0339e-02, -4.5369e-01,  5.8427e-01,\n",
       "          -5.4955e-01,  1.0329e-01, -1.6007e-01,  4.9532e-02, -5.2088e-01,\n",
       "          -9.8735e-02,  2.4494e-01, -1.8085e-01,  1.6053e-01, -4.6592e-01,\n",
       "          -3.2983e-02,  2.4514e-01,  3.0468e-01, -4.9183e-01,  4.2685e-02,\n",
       "          -6.5134e-03, -2.9791e-01,  7.3694e-02, -6.6407e-01, -8.5795e-01,\n",
       "           5.8354e-01,  1.8428e-02, -4.5240e-01, -2.3930e-01,  7.4443e-01,\n",
       "           2.9738e-01, -2.3201e-01, -5.3072e-02, -3.5548e-01,  2.0259e-01,\n",
       "          -1.8518e-01,  2.5689e-01, -2.7018e-01,  3.8439e-01,  5.5026e-01,\n",
       "           3.1581e-01,  2.8489e-01, -1.8327e-01,  2.7637e-01,  1.4465e-01,\n",
       "           3.6947e-02, -6.6614e-02, -1.5929e-01,  1.5180e-01,  1.2041e-01,\n",
       "          -2.9580e-01, -3.2999e-01,  5.4746e-03,  1.1182e-01, -3.3279e-01,\n",
       "          -1.0202e-01,  1.8680e-01,  3.4808e-01, -1.4092e-01, -2.5403e-01,\n",
       "          -7.8984e-01, -4.2328e-01,  1.0217e-01, -1.8560e-01, -1.7135e-01,\n",
       "          -4.1911e-01,  3.8121e-01, -6.6660e-02,  4.7638e-01,  3.4189e-01,\n",
       "           6.9978e-01, -1.5311e-01,  5.9776e-02, -5.6412e-01,  3.5197e-01,\n",
       "           3.8270e-02, -6.1110e-01, -1.7247e-01,  1.5680e-01, -4.1813e-04,\n",
       "           6.8195e-01,  7.0979e-02,  4.0124e-01,  1.2070e-01, -3.2553e-01,\n",
       "           8.2089e-03,  1.9890e-01, -1.2864e-01, -1.4297e-01,  7.3766e-01,\n",
       "          -5.5188e-01, -2.3029e+00,  1.0721e-01, -3.5844e-01, -3.6529e-01,\n",
       "          -1.8135e-01,  3.7255e-01,  3.3772e-02,  4.9966e-01,  1.5486e-01,\n",
       "           3.7598e-02,  1.6729e-01, -2.8025e-02,  3.6851e-01, -8.4399e-02,\n",
       "           2.6751e-01,  3.3341e-01,  1.2988e-01, -2.3298e-01, -1.0956e-01,\n",
       "          -3.9227e-01,  2.4031e-01, -9.8916e-02,  8.6010e-01,  2.3260e-01,\n",
       "           2.0943e-01, -4.3657e-03, -4.7262e-02,  4.8964e-01,  4.3099e-01,\n",
       "          -6.6541e-02, -1.6585e-01, -4.9522e-01,  4.6269e-03, -4.1279e-02,\n",
       "          -1.2963e-02, -4.5611e-01,  1.3753e-01,  4.2841e-01, -1.2416e-01,\n",
       "          -1.8354e-01,  1.5525e-01, -4.7715e-02, -3.1517e-01, -2.6492e-02,\n",
       "           6.8453e-01,  1.9080e-01, -2.2994e-01,  1.2377e-01, -1.1896e-01,\n",
       "           3.7006e-02, -4.6975e-01,  7.0044e-01,  6.1147e-01,  1.5585e-01,\n",
       "           1.6214e-02, -6.9943e-01,  2.3699e-01, -3.1574e-01, -4.7647e-03,\n",
       "          -1.5027e-01, -1.5731e-01, -4.2986e-01, -5.7285e-02, -5.5854e-02,\n",
       "           4.0115e-01,  7.9943e-02,  4.6653e-01,  1.1837e+00,  2.7973e-01,\n",
       "           1.9184e-02,  5.6097e-01,  2.0801e-01,  4.2492e-01, -4.0180e-01,\n",
       "           3.3082e-01,  8.8825e-02, -6.3896e-01,  2.7188e-01, -4.1593e-01,\n",
       "           4.4963e-02, -4.1432e-02, -3.7926e-01,  1.4208e-01, -2.1276e-02,\n",
       "           3.4237e-01,  3.3517e-01,  1.5186e-01,  1.6838e-01,  2.9411e-01,\n",
       "          -3.1322e-01,  5.9715e-01, -6.0809e-02, -1.1127e-01, -6.3492e-01,\n",
       "          -7.6732e-01,  9.5206e-02,  4.4743e-01, -1.2880e-01, -6.1279e-01,\n",
       "           1.0462e-01, -2.3866e-01, -1.2257e-01, -3.5064e-02,  2.1191e-01,\n",
       "           2.7455e-01,  5.0687e-02, -9.5436e-02,  6.4791e-02, -4.6950e-01,\n",
       "          -4.9277e-01,  4.9140e-02, -1.5758e-01, -1.3063e-01,  5.3918e-01,\n",
       "          -7.9415e-02,  1.1575e-01,  5.7221e-01,  1.9799e-02, -4.5976e-01,\n",
       "          -2.5640e-01,  1.4802e-01,  2.8667e-01, -7.2082e-02,  2.1981e-01,\n",
       "           3.4119e-01,  2.6467e-02,  5.1649e-01,  1.4823e-01, -2.1046e-01,\n",
       "          -2.7239e-01,  5.1173e-01, -1.5755e-01,  6.1403e-02, -1.2527e-01,\n",
       "          -8.7207e-01, -5.6150e-02, -7.8413e-02, -1.6833e-01, -5.4440e-01,\n",
       "           2.6566e-01,  1.4351e-01,  1.0561e-01,  2.2350e-01, -1.5984e-01,\n",
       "           7.0123e-01,  6.5285e-01, -3.3100e-01, -1.6943e-01, -4.0104e-01,\n",
       "           9.3922e-02, -6.8374e-02, -2.9329e-01, -1.4009e-02,  9.5368e-03,\n",
       "          -2.2765e-01, -1.2516e-01, -1.0805e-01,  2.8921e-01, -1.8876e-01,\n",
       "           5.2862e-01,  1.2189e-01,  8.3649e-02,  4.3873e-01,  2.0612e-02,\n",
       "           1.3071e-01,  3.6515e-01, -4.2117e-01,  1.2870e-01,  4.0424e-01,\n",
       "           4.0578e-01, -1.0551e+00, -8.3291e-01, -1.4936e-01,  3.6111e-01,\n",
       "           2.2860e-01,  2.5189e-01,  1.6999e-02, -7.2215e-03,  8.9455e-02,\n",
       "          -2.3402e-01,  3.5135e-01, -6.6415e-02, -5.7412e-01,  3.0561e-01,\n",
       "           2.3567e-01,  5.3263e-01, -2.7377e-01, -5.0579e-01, -6.5233e-02,\n",
       "           2.4616e-01, -3.6381e-01,  1.1392e-01,  2.3480e-01,  4.5466e-01,\n",
       "          -3.6015e-01, -5.6164e-01, -7.0282e-02, -7.3986e-01, -1.3578e-01,\n",
       "           3.2483e-01, -2.1437e-01,  2.7178e-02, -2.4758e-01,  1.5231e-01,\n",
       "          -1.7925e-02, -1.8879e-01,  6.0757e-01,  2.0522e-01, -2.5855e-01,\n",
       "           1.1981e-01, -2.3892e-01, -1.9475e-01,  5.0091e-01, -5.0226e-01,\n",
       "          -3.0348e-01, -5.0960e-02, -1.5349e-01, -3.2418e-01,  3.2181e-01,\n",
       "          -4.1389e-01, -1.4861e-01,  1.6748e-01, -6.4736e-02, -1.3068e-01,\n",
       "          -3.5233e-01,  1.4268e-01, -1.2707e-01, -4.9883e-02, -1.7654e-02,\n",
       "          -1.3353e-01,  6.0900e-02,  1.3308e-01,  6.4538e-02,  1.4388e-01,\n",
       "           5.3696e-01, -1.1464e-01,  2.7582e-03, -2.5900e-01, -2.2469e-01,\n",
       "           5.2498e-01, -9.4973e-02, -3.3117e-01,  5.4928e-01, -1.1156e+01,\n",
       "           1.0386e-01,  7.2631e-01, -5.5808e-01,  5.5128e-02, -2.8539e-01,\n",
       "           2.1085e-01, -3.0009e-01,  7.9101e-02,  2.4967e-01, -2.8460e-01,\n",
       "          -1.2277e-01, -2.7364e-01, -2.3193e-01, -4.0872e-01, -3.9742e-01,\n",
       "          -4.7229e-01, -5.6789e-01,  1.1772e-01, -1.7663e-01, -1.0996e-01,\n",
       "          -5.7036e-01, -1.9379e-01,  2.9696e-01, -4.9559e-03, -5.1069e-01,\n",
       "           5.9541e-02, -2.2948e-01,  2.6399e-01,  4.6126e-02,  2.2045e-01,\n",
       "          -1.9949e-02,  1.4108e-01,  2.0669e-01, -4.4946e-02, -4.5100e-01,\n",
       "          -4.4853e-01,  2.2129e-01, -1.5955e-01, -2.2267e-01, -5.4017e-02,\n",
       "          -3.3838e-01, -2.7689e-01,  1.3447e-01,  3.7594e-01,  6.2596e-01,\n",
       "           1.1595e+00,  5.6208e-01,  2.9245e-01, -1.0625e-01, -8.5438e-02,\n",
       "          -1.0762e+00,  5.5900e-01, -3.7318e-01,  8.1787e-02, -2.5119e-01,\n",
       "           6.6201e-01, -4.9906e-01,  5.4685e-01, -4.0754e-01, -2.5500e-01,\n",
       "           3.7179e-01,  3.0194e-01,  7.1104e-03,  2.8239e-02, -2.7432e-01,\n",
       "           4.0257e-01, -2.6094e-01, -3.6915e-02, -3.7819e-01,  4.0081e-01,\n",
       "          -3.9472e-01,  5.0770e-01, -5.6534e-02,  1.2945e-01,  1.8567e-01,\n",
       "          -3.5001e-01,  1.6610e-01,  1.0337e+00,  1.7414e-01,  4.1071e-01,\n",
       "           1.7420e-01,  5.1160e-01,  5.3064e-02, -8.8958e-01,  1.3691e-01,\n",
       "           2.0502e-01,  2.0789e-01, -6.7096e-02,  5.7131e-01, -2.5517e-02,\n",
       "          -6.5181e-01, -2.3094e-01, -3.2169e-01,  4.3108e-01,  6.5352e-01,\n",
       "          -2.2597e-01, -4.4343e-01,  2.3894e-01, -7.9776e-03, -5.3395e-01,\n",
       "           9.0074e-02, -2.2320e-01, -5.2792e-01,  8.7275e-02, -5.8914e-01,\n",
       "           1.4985e-01, -1.2428e-01,  5.4238e-02,  6.6887e-01, -4.0159e-01,\n",
       "          -1.6113e-01,  3.6602e-02, -4.4532e-02, -4.1270e-01, -4.9520e-01,\n",
       "          -7.7083e-02,  4.1073e-01, -1.9776e-01, -2.7976e-01,  7.0636e-02,\n",
       "          -3.0673e-01,  2.2584e-01, -1.7377e-01, -6.4935e-01,  3.2879e-01,\n",
       "          -6.0415e-01,  2.0578e-01,  2.5360e-01, -2.2507e-01,  4.9017e-01,\n",
       "           1.1337e-01, -8.1817e-02,  1.9397e-01, -9.9369e-02, -7.8554e-01,\n",
       "          -3.2573e-01, -2.7299e-02,  3.2662e-01, -4.2844e-01,  1.5776e-01,\n",
       "           1.5022e-01, -4.5752e-02, -2.2574e-01,  2.7904e-01,  1.0174e-01,\n",
       "           3.7762e-01,  3.9846e-01, -5.5006e-01,  1.1727e-01, -2.9867e-01,\n",
       "          -2.9262e-02, -6.1174e-02,  1.7952e-01,  2.3654e-01, -7.0764e-01,\n",
       "          -1.5224e-01, -2.6434e-01, -3.0965e-01, -2.1650e-01, -2.5120e-01,\n",
       "           2.0680e-01,  1.4038e-01, -1.5897e-02, -1.1308e-01,  5.8077e-01,\n",
       "           4.0814e-02, -1.1167e-01, -4.5997e-01,  3.1443e-01,  4.3178e-01,\n",
       "           2.1794e-01,  7.5875e-02, -1.2830e-01, -5.5532e-01, -1.1922e-02,\n",
       "          -4.3644e-01,  9.1081e-01, -7.3906e-01, -7.1970e-01,  6.6086e-02,\n",
       "           1.4951e-02, -8.8259e-04, -2.3450e-01, -5.0265e-01,  8.1298e-02,\n",
       "          -2.1017e-01, -1.4891e-01,  6.1550e-03, -1.0607e-02,  1.6299e-01,\n",
       "           4.0742e-01,  3.6838e-01,  1.5951e-01, -3.3938e-01, -4.4081e-01,\n",
       "          -6.0244e-01,  1.4567e-02, -1.6795e-01,  2.9330e-01, -2.6752e-01,\n",
       "          -3.2661e-01,  1.6043e-02, -9.1892e-03, -2.2272e-01,  5.2876e-01,\n",
       "           1.2427e-01,  1.6396e+00, -3.7287e-01, -4.4551e-01,  2.9018e-01,\n",
       "           7.2306e-02, -2.7073e-02, -2.8130e-01,  2.5074e-01,  9.5447e-02,\n",
       "           4.0194e-02, -3.6304e-01,  1.5023e-02, -2.0558e-02,  3.2773e-01,\n",
       "           3.2056e-01, -1.0703e+00, -2.0238e-01, -1.7993e-01,  5.5664e-01,\n",
       "           3.2388e-02,  6.3185e-01, -2.2305e-01,  3.0577e-01,  4.6106e-01,\n",
       "          -8.6667e-01, -2.3000e-01, -1.6129e-01,  2.7295e-01, -1.0583e-01,\n",
       "          -9.8735e-02, -1.6178e-01, -3.7165e-02, -1.6936e-01,  3.9538e-01,\n",
       "           7.4751e-01, -7.8426e-01,  7.4081e-01,  7.0467e-01, -9.8234e-02,\n",
       "           5.6815e-01,  9.4187e-02,  3.2641e-01, -1.4659e-01,  2.2669e-02,\n",
       "           2.6200e-01,  1.8542e-01, -6.3165e-01, -2.3962e-01,  2.4473e-01,\n",
       "           4.0158e-01, -1.1943e-01,  1.2108e-01,  7.9292e-02,  3.0960e-02,\n",
       "          -3.7691e-01, -1.0145e-01, -2.4552e-01, -3.9904e-01,  1.1356e-01,\n",
       "           1.4603e-02, -1.5001e-01, -8.6997e-02,  4.7709e-01, -1.7580e-01,\n",
       "           7.5067e-01,  4.2910e-01,  3.7145e-01, -3.5103e-01,  8.9950e-04,\n",
       "          -5.7565e-02,  6.0671e-01, -3.1670e-01,  7.7385e-01,  6.3337e-02,\n",
       "          -2.3140e-01,  2.7251e-01,  6.0547e-03,  1.7829e-01,  2.9892e-01,\n",
       "          -2.3519e-01,  1.2702e-01, -3.0718e-01, -2.3711e-01,  2.8330e-01,\n",
       "          -1.8640e-01,  1.1542e-01, -7.2270e-01, -6.3602e-03,  3.9857e-01,\n",
       "          -3.2051e-01,  4.9873e-01, -2.3074e-01,  6.1026e-01, -4.3509e-01,\n",
       "           3.2669e-02, -4.7470e-01, -4.3275e-01, -2.8508e-01,  5.1854e-02,\n",
       "          -8.6027e-01, -1.9453e-01,  1.7684e-01,  3.7405e-02,  2.1707e-01,\n",
       "           7.5155e-01, -3.3794e-01, -2.4771e-01, -2.0386e-01,  4.0190e-02,\n",
       "           1.8299e-01, -8.2071e-02, -3.5586e-01,  2.1161e-01, -3.5607e-01,\n",
       "          -1.0520e+00, -2.3117e-01,  8.9990e-02,  3.8419e-01, -3.5853e-01,\n",
       "           1.1570e+00,  2.5132e-02, -2.0226e-01,  6.7253e-01,  2.0808e-01,\n",
       "           6.4157e-01,  6.0323e-01,  1.3628e-03, -4.1259e-01, -5.4109e-02,\n",
       "          -1.5877e-01, -3.2881e-01,  1.5757e-01, -1.8346e-01,  5.0556e-01,\n",
       "           5.4802e-01, -2.0806e-02,  5.4756e-01]], device='mps:0'),\n",
       " tensor([[-0.4654, -0.4115, -0.4370,  ...,  0.1625,  0.4338,  0.5908],\n",
       "         [-0.0932, -0.6293, -0.4879,  ...,  0.3041, -0.1958,  0.5576]],\n",
       "        device='mps:0'),\n",
       " tensor([[-2.7791e-02, -7.5747e-01, -5.3074e-01, -5.1430e-01, -3.4417e-01,\n",
       "           7.2208e-02, -3.2733e-01,  1.9133e-01, -1.5003e-01,  5.7895e-01,\n",
       "           2.7665e-01,  1.8669e-01, -4.0441e-02, -1.9787e-02, -2.7559e-01,\n",
       "          -3.6770e-01, -5.5075e-01,  4.2714e-01,  4.8599e-01,  1.2291e-01,\n",
       "           6.0920e-01,  4.0016e-01, -5.7060e-01,  2.7172e-01, -1.0960e+00,\n",
       "           2.2592e-01, -9.0650e-02,  1.6102e-01,  2.6783e-01, -4.2947e-01,\n",
       "          -2.6593e-01,  3.3457e-01, -1.0081e-01,  1.1911e+00, -7.2573e-02,\n",
       "          -4.2319e-01,  1.2905e-01, -4.8449e-01,  2.2964e-01,  1.0587e-01,\n",
       "           3.3713e-01, -2.3616e-01,  3.0495e-01,  2.5426e-01,  2.2269e-01,\n",
       "          -1.0156e-01,  3.5441e-01, -6.7506e-01,  5.3763e-02, -5.4964e-01,\n",
       "          -2.7866e-01, -2.3157e-01,  3.8469e-01,  3.0486e-01, -1.7402e-01,\n",
       "          -1.7631e-01,  1.9710e-01,  2.8978e-01, -1.6215e-01,  1.9001e-01,\n",
       "          -3.1728e-01,  3.9028e-01, -2.9116e-02, -2.5337e-02, -4.3117e-01,\n",
       "           5.2338e-02, -1.0542e+00, -2.0097e-01, -6.1678e-01, -2.9287e-01,\n",
       "           6.1601e-02, -4.4671e-02,  2.6994e-02,  8.2943e-02,  6.2566e-01,\n",
       "          -2.9527e-01, -3.5686e-02, -6.8721e-02, -8.9378e-01,  3.5990e-02,\n",
       "           9.3805e-02,  2.8378e-01,  3.4715e-01, -1.8362e-01,  4.0523e-02,\n",
       "          -1.1761e-01,  5.2336e-01,  3.2885e-01,  5.5885e-02, -1.1072e-01,\n",
       "           2.6658e-01,  1.4680e-01,  3.1471e-01, -2.3564e-01,  2.1408e-01,\n",
       "          -1.6953e-01,  4.2040e-01, -7.0445e-02, -1.3569e-01, -1.7957e-01,\n",
       "           1.7683e-01,  2.7367e-01,  1.6921e-01,  1.8827e-01, -7.0935e-02,\n",
       "          -4.6949e-01,  6.4462e-03, -2.7295e-01, -9.2490e-01,  5.3292e-02,\n",
       "           2.5952e-02,  8.0243e-02,  3.5441e-01, -5.5720e-01, -4.9249e-01,\n",
       "           2.3824e-01,  3.1274e-01, -7.8117e-02, -2.4385e-01,  7.2875e-02,\n",
       "           1.2005e-01, -3.7322e-01,  3.0937e-01,  1.4655e-01,  3.8964e-01,\n",
       "          -2.4076e-01,  5.1062e-01, -1.8687e-01,  4.6773e-01,  1.7625e-01,\n",
       "          -2.8715e-02, -1.4186e-02,  2.4065e-02, -2.5766e-01, -7.0436e-02,\n",
       "          -3.2938e-01,  4.0965e-01, -2.1077e-01,  1.5229e-01, -2.7347e-01,\n",
       "          -1.9571e-02, -1.5879e-01, -4.2139e-01, -2.4379e-01,  5.2323e-01,\n",
       "          -2.3310e-01,  3.4454e-01, -2.5864e-01, -5.3971e-01, -5.4621e-01,\n",
       "          -4.6331e-01, -6.6221e-01,  6.5904e-04, -5.8988e-02,  1.5486e-02,\n",
       "          -1.4458e-01, -4.7299e-02,  3.4941e-01, -1.5764e-01, -1.3263e-01,\n",
       "           1.3229e+00, -1.6963e-01,  2.6547e-01, -5.2711e-01, -9.2709e-02,\n",
       "          -2.5951e-01,  3.5857e-01, -2.1281e-01, -5.0373e-02, -1.1031e-02,\n",
       "           1.5012e-01, -3.6325e-01,  3.2426e-01, -2.7495e-01,  4.9574e-02,\n",
       "          -3.8145e-01,  6.7830e-01,  1.0315e-01, -3.7603e-03, -2.1426e-01,\n",
       "           8.8470e-02, -1.8509e+00, -6.0244e-02, -4.9988e-02, -1.1984e-01,\n",
       "          -3.8304e-01,  1.1412e-01, -8.1008e-02,  4.7211e-01,  1.0523e-01,\n",
       "          -3.0391e-01,  1.4457e-02,  3.4290e-01,  1.7766e-01,  4.6617e-02,\n",
       "           3.7586e-01,  5.1268e-01,  5.2866e-01,  8.3422e-02, -1.2969e-01,\n",
       "          -2.2910e-01,  1.8840e-01, -1.4922e-01, -4.9168e-03,  1.3625e-01,\n",
       "           1.9163e-01, -1.7921e-02,  4.5103e-02,  4.5549e-01, -3.9944e-02,\n",
       "          -1.3034e-01, -3.0975e-01, -2.0571e-01, -1.5386e-01, -3.0506e-01,\n",
       "           6.6388e-01, -6.9598e-02,  1.5083e-01,  1.1394e-01, -7.6365e-02,\n",
       "          -1.9021e-01, -3.6627e-01,  7.4035e-02, -4.3048e-01, -1.0458e-01,\n",
       "           5.2612e-01, -2.4620e-01, -9.6266e-02, -1.5057e-01, -2.1628e-01,\n",
       "           4.6617e-01, -6.3064e-01,  6.9359e-01, -6.5753e-02, -1.5876e-01,\n",
       "          -6.3829e-02, -3.6947e-01,  8.8525e-02, -4.7900e-02, -2.1367e-01,\n",
       "          -6.6109e-01, -6.2912e-01, -9.3484e-01,  3.0931e-02,  2.0012e-01,\n",
       "           3.8940e-01, -2.0282e-01,  7.0611e-01,  4.1043e-01,  4.4392e-01,\n",
       "          -1.9033e-01, -2.5634e-01,  1.9540e-01,  2.4175e-02, -1.6470e-01,\n",
       "          -8.3407e-02, -3.6482e-02, -2.3338e-01,  5.4958e-01, -2.6909e-01,\n",
       "           3.4143e-02, -8.4658e-02,  1.4949e-01,  5.6272e-01, -2.6404e-02,\n",
       "           2.7543e-01,  3.6181e-02, -5.5097e-01, -2.6403e-02,  1.4938e-01,\n",
       "          -3.9414e-01,  1.3074e-01,  6.3249e-01, -2.9116e-01, -5.8419e-01,\n",
       "          -5.6450e-01,  1.3798e-02, -1.0222e-01,  2.9551e-01, -1.1019e+00,\n",
       "          -1.6877e-01,  2.8032e-01,  3.0136e-02,  1.8251e-01, -3.0887e-01,\n",
       "           2.9186e-01,  5.8483e-02,  3.7632e-02, -7.8447e-02, -5.1537e-01,\n",
       "          -2.3439e-01,  1.6469e-01, -3.6657e-01, -3.5963e-02,  1.6028e-01,\n",
       "           2.8092e-01, -9.8223e-02,  4.3126e-02,  2.2221e-01, -4.8483e-01,\n",
       "          -8.7845e-02, -3.3919e-02,  1.5657e-01,  6.3212e-02,  6.2429e-02,\n",
       "           7.6042e-02,  1.9293e-01,  7.3101e-01, -3.7007e-02, -6.0240e-01,\n",
       "          -2.4535e-01,  5.4258e-01,  7.7029e-02,  1.5324e-01, -1.7752e-01,\n",
       "          -1.4260e+00, -6.3469e-01,  3.8337e-02,  2.1822e-02, -9.6237e-02,\n",
       "           3.7186e-01,  1.4415e-01,  2.9691e-02,  1.0135e-01, -1.4892e-01,\n",
       "           2.7125e-01,  2.5571e-01,  4.1796e-01,  2.6634e-01,  1.2224e-01,\n",
       "          -9.5239e-02, -2.5680e-01, -3.4749e-01, -2.5056e-01,  5.0008e-02,\n",
       "           3.0551e-01,  2.9752e-01, -2.5921e-01, -9.4277e-02, -9.0702e-02,\n",
       "           7.3119e-01, -4.5527e-01, -5.6552e-01,  3.9878e-01, -5.2456e-01,\n",
       "          -2.7738e-01,  4.5605e-01, -3.7995e-01,  6.1826e-02, -3.0256e-02,\n",
       "           1.6101e-01, -7.3998e-02,  5.4701e-02,  5.9767e-02,  1.4680e-01,\n",
       "           1.8510e-01,  4.4459e-01, -9.6782e-02,  3.8301e-01,  4.8393e-01,\n",
       "          -3.7868e-01,  1.0631e-01, -3.1295e-01,  3.3607e-02,  3.3206e-02,\n",
       "           3.5477e-02,  1.0657e-01, -5.8454e-01, -7.1591e-02,  1.8413e-01,\n",
       "           3.9868e-02, -5.0830e-01, -1.9227e-01,  1.5691e-01, -1.5280e-01,\n",
       "          -1.9063e-02, -6.1864e-01,  7.4105e-02,  2.7357e-02, -1.0623e-01,\n",
       "           2.0196e-01,  3.6702e-02,  1.0747e-01, -2.9594e-01, -3.5062e-01,\n",
       "          -4.3860e-01,  1.3449e-01,  6.4194e-01, -1.6284e-02, -1.8554e-02,\n",
       "          -1.0969e-02, -1.8924e-01, -3.0452e-01,  4.8185e-01,  1.5484e-01,\n",
       "          -2.7278e-01,  1.5357e-01, -1.5461e-01,  4.3362e-01,  4.5208e-01,\n",
       "          -3.5378e-01, -1.2043e-01, -1.1214e-01,  3.6801e-01, -1.9150e-01,\n",
       "          -1.7609e-01,  2.8881e-01,  4.5279e-01,  1.9897e-01,  9.4702e-02,\n",
       "           2.5524e-01,  6.2194e-01, -7.1092e-01,  2.7741e-01, -5.0384e-02,\n",
       "           1.6210e-01, -3.1364e-01,  1.1810e-01, -4.5970e-01, -3.1796e-01,\n",
       "           2.0608e-01,  3.4692e-01,  1.5942e-01, -8.2525e-02, -1.1386e+01,\n",
       "           4.7410e-01,  7.1675e-01, -6.4620e-01, -1.2380e-02,  2.0990e-01,\n",
       "           1.4467e-01, -4.2321e-01,  1.2785e-01, -4.0986e-01, -4.8454e-01,\n",
       "           5.3751e-02, -3.1024e-01,  8.5046e-02, -2.1082e-01, -3.5012e-02,\n",
       "          -3.3293e-01, -8.8604e-01,  5.2616e-01, -2.3722e-01,  1.2282e-01,\n",
       "          -4.0554e-01,  2.2834e-01,  6.7441e-02, -3.7185e-01, -2.9844e-01,\n",
       "          -3.3378e-01, -4.8637e-01,  2.5874e-01,  1.4526e-01, -3.6388e-01,\n",
       "           1.6897e-01,  7.8214e-02,  1.5488e-01, -3.1408e-01, -5.7322e-01,\n",
       "           6.5890e-02, -1.4286e-01,  6.9077e-01, -2.3317e-01,  2.3521e-01,\n",
       "           1.5418e-01, -1.8939e-01,  4.0595e-01, -6.7756e-02,  5.4405e-01,\n",
       "           9.0099e-01,  4.9506e-01, -9.3677e-02, -3.1796e-01,  2.9697e-01,\n",
       "          -1.1296e+00,  2.3765e-01, -7.4326e-01,  3.7037e-02, -4.4715e-02,\n",
       "           5.2303e-01, -5.4682e-01, -1.6110e-02,  1.6453e-01, -1.3098e-01,\n",
       "           3.8056e-01,  1.5443e-02,  3.0641e-01, -3.8540e-01, -1.8694e-01,\n",
       "           5.1715e-01,  1.1540e-01,  3.9804e-01, -2.1344e-01, -2.7691e-02,\n",
       "          -3.5584e-01,  6.3533e-01,  4.1696e-01,  6.3335e-03,  1.0379e-01,\n",
       "           9.4310e-02,  1.5461e-01,  8.2262e-01,  1.5306e-01,  3.7272e-02,\n",
       "           7.5334e-02,  4.9919e-02, -2.0803e-02, -2.5454e-01, -1.9359e-01,\n",
       "          -3.8527e-02,  2.3219e-01,  6.4573e-01,  5.1229e-01, -2.5336e-01,\n",
       "          -1.8886e-01, -2.0242e-01, -5.9704e-02,  1.3338e-01,  5.6888e-01,\n",
       "          -7.2007e-01,  1.4785e-02,  2.0411e-01, -2.3545e-01, -2.6105e-01,\n",
       "          -9.1127e-02, -2.2821e-01, -6.1202e-01,  4.0102e-01, -3.4853e-03,\n",
       "          -3.2201e-01,  8.7693e-02,  5.6282e-01,  1.7925e-01, -2.9609e-01,\n",
       "           4.8297e-01, -1.4248e-01,  5.2177e-01, -7.0864e-02,  3.5673e-02,\n",
       "           3.0329e-01,  6.1367e-01, -2.1669e-01, -3.9304e-01, -6.3589e-02,\n",
       "          -7.9591e-02,  2.2539e-01,  3.5268e-03,  7.8083e-02, -1.6630e-02,\n",
       "          -3.5275e-01, -1.3786e-01,  3.3867e-02,  2.5058e-01,  2.6520e-01,\n",
       "           6.0789e-02,  3.0671e-01,  2.9447e-01, -3.1026e-01, -2.5075e-01,\n",
       "          -7.3840e-01, -2.4288e-01,  2.2993e-01, -1.9391e-01, -9.8460e-02,\n",
       "          -2.1162e-02, -3.4200e-02, -3.9679e-01,  3.1646e-01, -8.8932e-02,\n",
       "           1.4835e-01,  1.1151e-01, -4.4561e-01,  2.3351e-01, -3.3892e-02,\n",
       "          -3.1032e-01, -9.2286e-02, -3.7850e-01,  8.2453e-02,  3.6353e-02,\n",
       "          -4.8302e-01,  2.5437e-01, -2.5304e-01, -6.9363e-01, -3.3891e-01,\n",
       "           1.0299e-02,  1.8091e-02, -3.8864e-01,  1.0920e-01,  5.7217e-02,\n",
       "          -2.6965e-02, -5.7936e-01, -2.5618e-01, -5.4129e-03,  4.7410e-02,\n",
       "           3.7889e-01, -4.1326e-01,  5.3043e-02, -5.8918e-01,  3.3814e-01,\n",
       "           8.6381e-02,  8.8725e-01, -9.5259e-02, -4.3658e-01,  4.1371e-01,\n",
       "           6.8976e-01, -3.6204e-01,  2.3138e-01, -2.0207e-01,  1.7246e-01,\n",
       "          -2.2461e-01, -1.5982e-01,  4.5024e-01, -2.3929e-01,  5.3763e-01,\n",
       "           7.6361e-02,  3.0156e-01, -5.0755e-01, -3.4200e-02, -4.5787e-01,\n",
       "          -2.6191e-01, -1.0136e-01, -4.3868e-01, -2.1873e-01, -9.5628e-02,\n",
       "          -3.6356e-01, -2.6224e-01, -2.7189e-01,  1.6586e-02, -1.9311e-01,\n",
       "           3.5684e-01,  1.1473e+00,  3.5222e-02, -4.8536e-02,  3.1878e-01,\n",
       "           2.3119e-01, -2.1068e-01,  2.0884e-01, -7.6245e-02,  3.3463e-01,\n",
       "           2.9654e-01,  3.1331e-01, -4.5673e-01,  2.5522e-01, -8.5886e-02,\n",
       "           3.3382e-01, -8.9138e-01, -7.5951e-01,  1.8828e-02,  2.3505e-01,\n",
       "          -4.1163e-02,  7.9677e-02, -2.6933e-01,  6.0446e-01,  6.9945e-01,\n",
       "          -1.3167e+00, -6.0678e-02, -2.4393e-01,  1.2029e-01, -4.5133e-04,\n",
       "          -4.4085e-01, -3.7242e-01, -3.3575e-01, -2.8487e-01,  3.2123e-01,\n",
       "           7.5280e-01, -4.5114e-01,  1.6297e-02,  1.8465e-01, -1.1076e-01,\n",
       "           2.3514e-01, -5.0295e-03,  2.5102e-01,  4.3425e-01, -1.0414e-01,\n",
       "          -1.4909e-01,  4.2579e-01, -3.1190e-01, -4.2031e-01,  1.5833e-01,\n",
       "           7.0937e-01, -1.1853e-01,  1.3689e-01, -1.8297e-02, -2.6799e-01,\n",
       "          -7.1901e-02, -1.1732e-01, -1.9540e-01,  2.4573e-01,  3.1104e-01,\n",
       "           1.0016e-01, -3.2782e-01,  2.3126e-01,  2.1841e-01, -5.2982e-02,\n",
       "           5.0868e-02,  9.0378e-01,  7.3841e-01,  1.8846e-02, -7.4087e-02,\n",
       "           2.7607e-01,  2.4221e-01, -5.0505e-02,  5.1696e-01, -1.7273e-01,\n",
       "          -2.8848e-01,  1.9170e-01,  8.5894e-02, -2.9880e-01, -9.4752e-03,\n",
       "          -2.0048e-01,  1.8031e-01,  1.8487e-01, -1.6871e-01,  5.3942e-01,\n",
       "           3.3819e-01,  1.9990e-01, -4.1667e-01,  6.3733e-05,  2.5826e-01,\n",
       "          -2.8037e-01,  8.7311e-01,  1.6816e-01,  4.6258e-01, -3.0563e-01,\n",
       "           2.2482e-01, -6.0811e-01, -4.3120e-01,  1.4365e-01,  7.4790e-02,\n",
       "          -3.6079e-01,  1.8614e-01, -1.2540e-01, -1.5393e-01,  7.3765e-01,\n",
       "           5.8692e-01, -3.4385e-01, -5.9073e-01, -5.4749e-02,  3.2823e-02,\n",
       "          -2.9436e-02, -2.4520e-01,  7.4517e-02, -2.1959e-02,  1.6957e-01,\n",
       "          -7.0204e-01, -2.6400e-01,  1.2722e-01,  2.0311e-01, -4.2548e-01,\n",
       "           8.2082e-01,  7.0018e-02,  1.7358e-01,  7.3647e-01,  1.3118e-01,\n",
       "           1.7952e-01,  6.3050e-01,  6.9376e-02, -5.5670e-01,  1.2268e-01,\n",
       "           2.8105e-01, -1.1468e-01, -5.5720e-02, -6.3967e-01,  3.2919e-01,\n",
       "           8.0147e-02,  9.4187e-01,  7.0432e-01]], device='mps:0'),\n",
       " tensor([[-0.2744, -0.4852, -0.4880,  ...,  0.2064,  0.2977,  0.6606],\n",
       "         [ 0.0506, -0.6230, -0.6446,  ..., -0.0875,  0.4018,  0.3101],\n",
       "         [-0.3308, -0.4627, -0.2900,  ...,  0.2546,  0.1334,  0.6058]],\n",
       "        device='mps:0'),\n",
       " tensor([[-2.8410e-01, -9.0726e-01, -8.2138e-01, -6.7671e-01, -3.8896e-02,\n",
       "           2.9225e-01, -2.2791e-01,  4.6630e-01, -1.6524e-01,  6.1008e-01,\n",
       "           4.1648e-01, -1.9774e-02, -8.8540e-03, -4.1552e-01, -4.7808e-01,\n",
       "          -2.6873e-01, -8.0084e-02,  4.9857e-01,  3.5622e-01,  1.7044e-01,\n",
       "           6.8039e-01,  3.9424e-01, -1.9472e-01,  3.7961e-01, -5.7647e-01,\n",
       "           1.3869e-01,  2.6591e-01, -2.3492e-02,  2.5411e-01, -7.5615e-01,\n",
       "           1.2406e-02,  6.3200e-01, -2.1181e-02,  9.6303e-01, -7.1360e-02,\n",
       "           2.1652e-02,  1.3258e-01, -2.4279e-01,  2.4313e-01, -1.1724e-01,\n",
       "           6.7979e-01, -1.5143e-01,  3.1783e-01,  3.0120e-01,  3.6170e-01,\n",
       "          -3.7481e-01,  2.5890e-01, -8.8888e-01,  6.0561e-02, -5.6336e-01,\n",
       "          -1.7535e-01,  3.2466e-02,  4.2590e-01,  4.9348e-01, -5.2953e-01,\n",
       "          -3.4813e-01,  1.4112e-01,  3.4483e-01, -3.9610e-01,  1.4425e-01,\n",
       "           5.3351e-03,  3.2589e-01,  6.9159e-02, -3.9174e-01,  4.2056e-03,\n",
       "           1.4789e-02, -2.4017e-01, -3.2790e-01, -3.6851e-01,  3.4575e-02,\n",
       "          -9.9058e-03,  1.0437e-01, -1.6752e-02, -7.5242e-02,  5.8434e-01,\n",
       "          -2.6431e-01,  2.8522e-01,  8.3756e-02, -6.1078e-01,  3.9079e-01,\n",
       "           4.6566e-02,  1.8837e-01,  4.0246e-02, -3.2943e-01,  1.0772e-01,\n",
       "          -4.6080e-02,  4.2466e-01,  3.2280e-01,  3.2453e-01, -3.7554e-01,\n",
       "           2.6490e-01,  1.0716e-01, -4.6677e-01,  2.9562e-02,  3.8778e-01,\n",
       "           6.2538e-02,  4.4825e-01, -2.6339e-01, -1.5771e-01, -4.0255e-01,\n",
       "           3.2256e-02,  1.9701e-01,  1.6448e-01,  2.3657e-01, -4.1179e-01,\n",
       "          -7.7102e-01, -2.0975e-01, -3.0249e-01, -7.6007e-01, -2.3155e-01,\n",
       "           2.4093e-03,  3.0643e-03,  1.9297e-01, -3.5954e-01, -2.4565e-01,\n",
       "           1.4480e-01,  8.5703e-02, -9.8301e-02, -3.2928e-01,  6.7764e-01,\n",
       "           1.8521e-01, -2.0132e-01,  1.4682e-01,  1.3007e-01,  1.6668e-01,\n",
       "          -3.5736e-02,  5.1407e-01, -2.0821e-01,  3.6986e-01,  4.0443e-01,\n",
       "          -2.8872e-01, -8.8650e-02, -2.2858e-01, -1.9203e-01, -7.5065e-02,\n",
       "          -1.6228e-01,  4.1641e-01, -4.7081e-01,  1.1261e-01,  2.2024e-01,\n",
       "          -5.7536e-02, -3.5674e-01, -8.1691e-02, -3.2677e-02,  1.2809e-01,\n",
       "          -6.5025e-02,  1.8242e-02, -2.5743e-01,  1.2635e-01, -6.3615e-01,\n",
       "          -9.5011e-01, -5.9328e-01,  3.4771e-01, -2.2995e-01, -2.1277e-02,\n",
       "           7.0774e-02,  1.2290e-01,  3.2420e-01, -7.2005e-03,  1.0673e-01,\n",
       "           6.3445e-01, -1.5287e-03,  3.6644e-01, -8.3712e-01, -5.1184e-02,\n",
       "          -8.3569e-04,  1.2342e-01, -2.6895e-02,  5.6355e-02,  5.4288e-02,\n",
       "           3.0141e-01, -4.5444e-01,  1.0689e-01, -4.7978e-01,  4.1224e-01,\n",
       "          -6.3805e-01,  5.6762e-01,  1.7935e-01, -2.5734e-01, -1.2358e-01,\n",
       "           1.4936e-01, -1.6593e+00,  5.3200e-01, -4.3747e-01,  9.8774e-02,\n",
       "          -6.7443e-01,  5.3985e-01,  1.5989e-01,  2.7911e-01, -1.1152e-01,\n",
       "          -3.4382e-01, -2.9490e-01, -1.5108e-01,  6.2033e-01,  1.0140e-01,\n",
       "           4.0821e-01,  5.0145e-01,  5.2377e-02, -3.1203e-01, -3.3315e-02,\n",
       "          -5.2458e-01,  1.5478e-01, -2.4363e-02, -7.7202e-02,  2.7055e-01,\n",
       "           5.2355e-01, -3.1375e-01,  1.6361e-01,  3.2687e-01,  8.3109e-02,\n",
       "          -1.6386e-01, -1.7259e-01, -4.9044e-01, -4.5206e-01, -3.4039e-01,\n",
       "           6.7165e-01, -3.9371e-01, -1.6622e-01,  4.6567e-01,  1.5607e-01,\n",
       "          -3.0100e-01, -2.6997e-01,  5.2652e-02, -3.2816e-01,  1.5516e-01,\n",
       "           2.6289e-01, -1.9121e-01, -2.3150e-02, -1.7109e-01, -1.5597e-01,\n",
       "           9.0210e-02, -6.1649e-01,  8.1863e-01,  3.4575e-01, -3.7515e-01,\n",
       "          -2.5985e-01, -2.6903e-01,  1.9054e-01, -3.8720e-01, -2.2831e-01,\n",
       "          -4.6617e-01, -6.5302e-01, -4.1872e-01, -6.2506e-02,  7.2646e-02,\n",
       "           2.6405e-01, -4.3104e-01,  8.5809e-01,  4.6765e-01,  2.8787e-01,\n",
       "          -1.1731e-01, -2.2530e-01,  2.5957e-01, -2.9173e-02, -3.1549e-01,\n",
       "          -5.3183e-01, -4.2127e-02, -7.4157e-02,  4.2566e-01, -3.0446e-01,\n",
       "          -2.2506e-01,  5.4441e-02,  1.6751e-01,  3.8585e-01, -2.4884e-01,\n",
       "           4.5837e-01,  1.2329e-01, -4.5600e-01, -3.4872e-01,  2.0715e-01,\n",
       "          -2.6145e-01,  7.0682e-02, -2.0341e-01, -3.5739e-01, -5.2887e-01,\n",
       "          -4.4800e-01,  1.5137e-01, -1.6499e-01,  3.6054e-01, -5.9794e-01,\n",
       "          -3.3737e-01,  2.2016e-01, -1.9469e-01,  1.1993e-02, -1.3741e-01,\n",
       "           3.9436e-01,  1.8494e-01,  7.4608e-02, -2.4071e-01, -4.5485e-01,\n",
       "          -9.6388e-02,  3.0287e-01, -3.7618e-01,  2.7777e-01,  1.2088e-01,\n",
       "          -4.7114e-02,  6.9037e-03,  3.0233e-01,  1.2851e-01, -4.8270e-01,\n",
       "          -1.3049e-01, -9.8671e-02, -9.6957e-02,  3.8075e-02,  8.0466e-02,\n",
       "          -1.0432e-01,  2.6028e-01,  4.7405e-01, -5.0982e-01, -5.7763e-01,\n",
       "           1.8491e-01,  3.8076e-01, -9.5053e-02, -1.8549e-01, -1.4827e-01,\n",
       "          -1.0550e+00, -2.3885e-01, -4.1791e-01,  8.9087e-03, -3.4341e-01,\n",
       "           5.2151e-01, -2.3990e-02, -7.5561e-02,  2.9047e-01, -4.8588e-01,\n",
       "          -1.0527e-02, -2.5891e-02,  1.0761e-01,  2.4388e-01, -2.5503e-01,\n",
       "          -2.6308e-02, -1.0385e-01, -3.2726e-01, -2.4221e-02,  3.9331e-02,\n",
       "           3.0303e-01,  3.1556e-01, -1.1696e-01, -8.3635e-02, -9.2741e-03,\n",
       "           6.5852e-01, -2.3145e-01, -5.0153e-01,  3.5872e-01, -5.3057e-01,\n",
       "          -1.8119e-01,  1.1570e-01, -3.5868e-01, -6.6811e-02,  4.0729e-01,\n",
       "           1.0192e-01,  1.1587e-02, -1.7193e-01,  7.2431e-03,  3.4062e-01,\n",
       "           2.7961e-01,  3.2652e-01, -4.1415e-01,  1.0437e-01,  7.1963e-01,\n",
       "          -2.1965e-01,  2.1594e-02, -3.8914e-01, -4.9749e-02,  1.8206e-01,\n",
       "           2.9015e-01,  4.0271e-01, -4.8196e-01, -2.2830e-04,  2.3613e-01,\n",
       "          -2.3326e-01, -3.5390e-03, -1.0025e-01,  7.8301e-02,  3.5127e-02,\n",
       "          -2.4748e-01, -4.2859e-01,  3.9256e-01, -2.0757e-02, -4.4708e-01,\n",
       "           2.5400e-01,  2.9843e-01, -1.8380e-01, -2.2817e-01, -3.4703e-01,\n",
       "          -4.9744e-01, -1.3926e-01,  9.5642e-01,  1.4745e-01,  3.9996e-01,\n",
       "           1.9061e-01,  1.1501e-01, -1.1146e-01,  2.4241e-01,  5.1698e-01,\n",
       "          -3.7747e-01, -4.3104e-02, -3.3214e-01,  3.8401e-01,  3.1391e-01,\n",
       "          -2.8255e-01, -1.8663e-01,  2.1814e-01, -2.6956e-01, -1.2567e-01,\n",
       "          -3.3259e-01,  5.9083e-02,  6.1093e-01, -4.9808e-02,  3.1331e-02,\n",
       "           2.3225e-01,  7.6214e-01, -3.7722e-01,  3.0266e-01,  4.7406e-02,\n",
       "           6.8995e-01,  6.6184e-02,  9.4402e-02, -1.4979e-01, -6.3115e-01,\n",
       "           3.7063e-01,  3.9721e-01, -2.8912e-02,  6.7678e-02, -1.1311e+01,\n",
       "           3.4454e-02,  5.6891e-01, -7.5971e-01,  2.2732e-01, -2.5964e-01,\n",
       "           2.9732e-01, -4.4535e-01,  1.6031e-01, -3.9677e-01, -2.6623e-02,\n",
       "          -2.8857e-01,  1.4880e-01, -3.2440e-02, -3.8027e-01, -2.0103e-01,\n",
       "          -5.3780e-02, -7.4392e-01,  1.7976e-01, -6.4940e-02,  5.4619e-01,\n",
       "          -1.7719e-01,  2.0839e-01, -2.0283e-02, -2.3149e-01, -2.0381e-01,\n",
       "          -5.6298e-02, -7.4478e-01,  2.3089e-01,  3.3053e-01,  4.8207e-02,\n",
       "           2.5833e-01,  2.6776e-01,  1.5874e-01, -1.8671e-01, -6.6747e-01,\n",
       "           1.1653e-02, -1.2411e-01,  3.3954e-01, -1.9911e-02, -4.8342e-01,\n",
       "          -3.1314e-01, -4.4473e-02,  6.2822e-01,  1.7275e-01,  8.0393e-01,\n",
       "           4.7681e-01,  7.9388e-01,  1.2465e-02, -2.5039e-01,  1.6468e-01,\n",
       "          -7.3366e-01,  4.6588e-01, -1.4671e+00,  3.6939e-01, -1.2161e-01,\n",
       "           5.2864e-01, -4.8291e-01,  1.5394e-01, -7.2982e-02, -1.8434e-01,\n",
       "           2.5902e-01, -1.4170e-01,  2.9552e-01, -4.0068e-01, -4.3387e-01,\n",
       "           5.8911e-01, -1.1020e-01,  1.3014e-01, -1.8670e-01, -1.4714e-01,\n",
       "          -2.6274e-01,  2.6367e-01,  3.2650e-01, -3.8465e-01,  7.9557e-05,\n",
       "          -2.7393e-01, -3.1154e-01,  6.1599e-01, -2.9421e-02,  1.0948e-01,\n",
       "          -1.9772e-04,  3.9827e-01, -1.9306e-01, -7.8394e-01, -7.2762e-02,\n",
       "          -7.8919e-02,  5.3923e-02,  5.6231e-02,  6.9938e-01, -3.9077e-01,\n",
       "          -3.6566e-01, -4.0675e-02,  3.0838e-01,  1.1084e-01,  8.5880e-01,\n",
       "          -5.9137e-01,  3.1160e-01,  4.2497e-02, -1.4490e-01, -6.4379e-01,\n",
       "           2.5900e-02, -3.4008e-02, -5.2293e-01,  4.0197e-01,  1.6080e-01,\n",
       "          -2.1321e-01,  2.5649e-01,  1.1206e-02,  3.6884e-01, -5.1919e-01,\n",
       "           5.2135e-01,  1.4200e-01,  4.6602e-01, -6.3895e-01,  2.0478e-01,\n",
       "           3.2031e-01,  5.7861e-01, -3.1291e-01, -3.3574e-01, -4.2593e-01,\n",
       "           6.7245e-02,  2.0315e-01, -2.8193e-01, -4.7090e-01, -1.7570e-01,\n",
       "          -5.5198e-01, -1.5751e-01, -2.1259e-01,  1.0660e-01,  4.0589e-01,\n",
       "           1.0012e-01,  8.8547e-03,  2.6150e-01, -1.1498e-01, -7.5667e-01,\n",
       "          -4.4067e-01, -2.2653e-02,  2.8108e-01, -1.7478e-01,  5.3903e-02,\n",
       "           9.3133e-02,  3.3264e-02, -2.6656e-01,  3.5361e-01, -5.4893e-02,\n",
       "           3.6914e-01,  3.3844e-01, -2.7639e-01, -1.1875e-01, -2.2805e-01,\n",
       "          -4.0032e-01, -3.0262e-02, -1.1470e-01, -3.9639e-02, -1.8235e-01,\n",
       "          -1.5471e-01,  1.9659e-01, -2.6834e-01, -1.2059e-01,  1.6358e-01,\n",
       "           3.3242e-01,  1.2926e-01,  3.9192e-02,  2.0116e-01,  7.1898e-02,\n",
       "           3.8818e-02, -1.9482e-01, -6.5967e-01, -8.0501e-02, -6.2730e-02,\n",
       "           2.2119e-02, -4.3437e-01, -4.2454e-01, -4.4626e-01,  1.9494e-01,\n",
       "           1.6663e-01,  1.4173e+00, -2.5614e-01, -6.1296e-01,  5.2611e-01,\n",
       "           3.8131e-01, -1.3115e-01, -6.5846e-02, -1.6166e-01,  3.9772e-01,\n",
       "          -2.5299e-01, -3.3747e-01,  4.0694e-01, -1.0105e-01,  4.3676e-01,\n",
       "          -1.5157e-02,  1.9952e-01, -2.4585e-01,  8.8356e-01, -3.0930e-01,\n",
       "          -1.9445e-01, -1.7599e-01, -5.3702e-01, -1.4873e-01,  3.7056e-01,\n",
       "          -1.0973e-01,  1.9020e-01,  6.8739e-02,  7.3135e-02,  4.1898e-01,\n",
       "           4.3927e-01,  1.0343e+00, -2.6665e-02,  3.2384e-01,  3.2034e-01,\n",
       "           1.1636e-01, -2.5208e-01,  4.9564e-02,  6.3171e-02,  2.5078e-01,\n",
       "           1.1230e-01, -5.2400e-02, -2.3595e-01,  1.3156e-01,  1.6624e-01,\n",
       "           3.1721e-01, -5.1819e-01, -4.2520e-01, -1.1695e-01,  7.3663e-01,\n",
       "           2.3994e-01,  3.3000e-01, -3.0413e-01,  6.1087e-01,  6.5632e-01,\n",
       "          -1.0485e+00, -1.7473e-01, -7.8810e-02,  2.8640e-01, -1.7922e-01,\n",
       "          -5.7621e-01, -4.8697e-02, -1.7951e-01, -3.0363e-01,  2.6832e-01,\n",
       "           2.2400e-01, -9.5763e-01,  3.9214e-01,  2.5776e-01, -6.7744e-01,\n",
       "           3.3738e-02,  4.1360e-02,  1.8309e-01,  1.5037e-01, -2.5656e-02,\n",
       "          -2.7186e-01,  1.6489e-01, -1.1918e-01, -7.1201e-01,  3.3854e-01,\n",
       "           5.4615e-01, -5.6500e-04,  9.7112e-02,  2.9567e-01, -4.5991e-01,\n",
       "          -1.8836e-01, -1.5111e-01, -2.9050e-01,  1.4082e-01,  3.9382e-02,\n",
       "           3.6783e-02, -6.3018e-01,  1.0080e-01,  2.6791e-01, -4.0123e-02,\n",
       "           3.0297e-01,  7.7859e-01,  5.0896e-01, -1.9271e-01,  8.5166e-02,\n",
       "           3.5889e-01,  3.8624e-02, -5.8162e-02,  5.9186e-01, -4.4133e-02,\n",
       "          -4.3764e-01, -9.1292e-02,  3.9013e-01, -3.3665e-01,  1.7664e-01,\n",
       "          -3.4319e-01,  2.8052e-01,  1.3908e-01, -2.1611e-01,  4.9480e-01,\n",
       "           1.3763e-01,  4.6564e-01, -5.8685e-01,  3.3144e-01,  2.2217e-01,\n",
       "           2.0924e-02,  8.0724e-01,  2.7080e-01,  3.1194e-01, -1.8469e-01,\n",
       "          -1.9706e-02, -4.4113e-01, -7.0538e-01,  2.1557e-01,  1.8118e-01,\n",
       "          -2.5109e-01,  8.7990e-02,  7.6911e-02,  1.8638e-01,  8.8096e-01,\n",
       "           7.9640e-01, -1.9490e-01, -5.7815e-01, -4.3192e-01,  1.0734e-01,\n",
       "           2.9062e-02, -4.5282e-01,  3.7426e-01, -1.6249e-01, -8.1733e-02,\n",
       "          -7.8608e-01, -6.3649e-01,  8.5454e-02, -1.5677e-01, -4.8105e-01,\n",
       "           3.3863e-01, -2.0109e-02,  2.9668e-01,  2.0777e-01,  2.3451e-01,\n",
       "           2.0012e-01,  5.3570e-01,  1.5344e-01, -2.4333e-01,  4.3486e-02,\n",
       "           3.2455e-01, -1.6314e-01, -5.4736e-02, -4.1345e-01,  3.7442e-01,\n",
       "           1.8882e-01,  6.9329e-01,  5.1870e-01]], device='mps:0'),\n",
       " tensor([[-0.1672, -0.5249, -0.4540,  ...,  0.3771,  0.2016,  0.5340],\n",
       "         [-0.3292, -0.2590, -0.5016,  ..., -0.0148,  0.3509,  0.1607]],\n",
       "        device='mps:0'),\n",
       " tensor([[-0.1769, -0.4627, -0.4013,  ...,  0.3726,  0.2137,  0.5559],\n",
       "         [-0.1561, -0.5489, -0.7124,  ..., -0.0489, -0.2392,  0.1528]],\n",
       "        device='mps:0'),\n",
       " tensor([[-0.3803, -0.4407, -0.4684,  ...,  0.2509,  0.6254,  0.5274],\n",
       "         [-0.1130, -0.3008, -0.2506,  ...,  0.5065,  0.6999,  0.6288]],\n",
       "        device='mps:0'),\n",
       " tensor([[-0.0580, -0.6728, -0.2770,  ...,  0.1850,  0.1445,  0.4702],\n",
       "         [-0.5135, -0.2107, -0.7904,  ..., -0.1355, -0.2544, -0.0175],\n",
       "         [-0.6806, -0.9010, -0.7204,  ...,  0.1958,  0.5878,  0.5598],\n",
       "         [-0.0943, -0.8092, -0.5285,  ..., -0.0221,  0.1577,  0.5263],\n",
       "         [-0.3315, -0.6398, -0.3819,  ...,  0.1300,  0.2541,  0.3427]],\n",
       "        device='mps:0'),\n",
       " tensor([[-5.8529e-01, -3.0544e-01, -6.0699e-01, -9.3997e-01, -3.9841e-01,\n",
       "          -4.1917e-01, -8.5330e-01,  7.9848e-01,  6.9558e-02,  4.5752e-01,\n",
       "           1.5023e-01,  5.7637e-02,  4.1002e-01, -2.1822e-01, -7.2275e-01,\n",
       "          -4.5044e-01, -3.5296e-02,  3.7504e-01,  3.2675e-01,  1.8490e-01,\n",
       "           5.8154e-01,  8.7251e-02, -9.1417e-02,  7.7611e-02, -5.2690e-01,\n",
       "           3.5945e-01,  2.7204e-01,  1.7083e-02,  2.1532e-01, -8.0033e-01,\n",
       "           2.1181e-01,  3.8386e-01,  2.0063e-01,  8.7402e-01, -2.6759e-01,\n",
       "          -1.2413e-01,  8.5498e-02, -4.1408e-01,  7.0147e-01,  3.5154e-01,\n",
       "           3.3415e-01, -3.4621e-01,  1.2698e-01,  1.0524e-01, -1.9460e-01,\n",
       "           4.5736e-02, -5.2745e-02, -7.0988e-01,  2.8630e-01, -4.4259e-01,\n",
       "          -2.7204e-01, -2.2403e-01,  6.2493e-01,  5.1560e-01, -1.0911e+00,\n",
       "          -5.6968e-01,  4.0561e-01,  3.4125e-01, -7.2408e-01,  2.7212e-01,\n",
       "          -9.1105e-02,  8.3264e-01,  4.0415e-01,  1.2090e-01,  4.9880e-01,\n",
       "           3.5658e-01, -4.2548e-01, -6.0912e-01, -2.1860e-01, -1.1226e-01,\n",
       "           1.0399e-02, -8.9963e-02, -4.9391e-01, -1.4124e-01,  5.1227e-01,\n",
       "           2.8277e-01,  2.0039e-01,  4.9248e-01, -1.6874e-01, -8.0080e-02,\n",
       "          -5.1838e-03, -5.8370e-03, -3.7513e-02,  9.9626e-02,  2.1515e-01,\n",
       "           2.6516e-01,  4.6337e-01,  1.9812e-01,  2.5128e-02, -9.6905e-02,\n",
       "           3.0256e-01, -1.2322e-01, -2.6097e-01, -9.4027e-01,  3.8160e-01,\n",
       "          -3.1446e-01,  3.2456e-01, -4.6281e-01,  1.5717e-01, -3.1183e-01,\n",
       "           1.6291e-01,  5.3941e-01, -2.6366e-01, -7.1920e-02, -3.1881e-01,\n",
       "          -6.3727e-01,  2.7604e-01,  1.1205e-01, -4.0850e-01,  1.9039e-01,\n",
       "           1.4783e-01, -3.4813e-01,  3.2431e-02, -4.0610e-01, -4.2302e-01,\n",
       "          -1.6718e-01,  2.6549e-01, -2.9914e-01, -6.8951e-01,  5.1356e-01,\n",
       "           1.7541e-01, -3.3286e-01, -4.1319e-01, -1.6572e-01,  1.7560e-01,\n",
       "          -3.5712e-01,  2.8851e-01,  1.6759e-01,  1.9363e-01,  6.8241e-01,\n",
       "           5.9573e-02,  1.4806e-01, -2.1903e-01, -1.6041e-01, -2.4485e-01,\n",
       "           7.7898e-02,  5.7879e-02, -2.2716e-01,  5.4378e-02,  8.0136e-02,\n",
       "          -2.3362e-01, -2.9914e-01, -7.6831e-01, -4.2709e-01, -2.2036e-01,\n",
       "          -2.9283e-01,  1.3774e-01,  7.2870e-02, -4.7997e-02, -5.0631e-01,\n",
       "          -7.8639e-01, -4.2786e-01,  8.1361e-01, -8.2078e-02,  8.5291e-02,\n",
       "          -3.6717e-01,  1.6349e-01,  2.6734e-01, -9.5083e-02,  4.6152e-01,\n",
       "           4.0421e-01,  2.2388e-01,  4.7225e-02, -6.1988e-01, -1.0309e-01,\n",
       "           8.5073e-02, -1.8385e-01, -4.4173e-01,  2.0486e-01,  3.1343e-01,\n",
       "           1.4982e-01, -1.3332e-01,  2.1507e-02, -5.9584e-01, -1.3552e-01,\n",
       "          -5.1141e-01,  2.9669e-01,  1.6764e-01, -5.3097e-01, -5.3403e-02,\n",
       "          -8.7282e-02, -1.6922e+00,  4.1328e-01, -5.0001e-01, -2.6912e-01,\n",
       "          -7.3911e-01,  1.4178e-02,  2.8132e-01,  2.8829e-01,  2.6834e-01,\n",
       "           1.2215e-01,  1.4613e-01,  2.7164e-03,  3.7640e-01,  2.3088e-01,\n",
       "           2.8382e-01,  4.1714e-01, -2.8625e-02, -3.3016e-01,  1.8691e-01,\n",
       "          -2.8478e-01,  2.8446e-01, -1.3802e-01,  2.0265e-01,  2.7933e-01,\n",
       "           4.1236e-01, -2.8050e-01,  2.1496e-01,  2.3627e-01, -1.6339e-01,\n",
       "           5.9209e-02, -1.7156e-01, -3.6037e-01,  1.9586e-02, -1.5751e-01,\n",
       "           1.1789e-01, -4.0103e-01, -7.0299e-03,  2.4622e-01,  1.0738e-01,\n",
       "          -7.6959e-02,  2.0045e-01, -1.9158e-01, -4.5727e-01,  1.2565e-01,\n",
       "           4.3581e-01, -1.3514e-01,  3.7668e-01, -3.6600e-01,  1.4422e-01,\n",
       "          -3.7546e-01, -5.4459e-01,  4.6722e-01,  6.3708e-01,  2.2799e-01,\n",
       "          -1.7324e-01, -2.8396e-01,  3.8873e-01, -2.4340e-01, -3.8154e-01,\n",
       "          -4.6933e-01, -2.8438e-01, -1.4970e-01, -7.9224e-02,  6.9374e-01,\n",
       "           1.8211e-01,  4.9121e-02,  1.1290e+00,  7.7529e-01,  2.8013e-01,\n",
       "          -3.8341e-02, -2.0156e-01,  4.4525e-01,  1.9279e-01, -3.9822e-01,\n",
       "           4.8039e-02, -7.2238e-02, -4.6835e-01,  5.6465e-01, -4.0990e-01,\n",
       "           3.5614e-01,  2.5178e-01,  1.7436e-01,  7.7993e-01,  3.5041e-01,\n",
       "          -6.9469e-01, -1.9245e-01, -4.2697e-02, -3.2415e-01, -8.8182e-03,\n",
       "           5.7035e-02,  1.8888e-01, -3.9717e-01, -2.3813e-01, -8.0990e-01,\n",
       "          -2.7504e-01, -1.9103e-01,  2.2059e-01,  3.0005e-01, -6.2333e-01,\n",
       "          -2.2623e-01,  4.2444e-01, -2.2570e-01,  1.7540e-01,  2.1682e-01,\n",
       "           5.6431e-01,  4.8953e-02, -6.5849e-02, -2.0825e-01,  3.7474e-02,\n",
       "          -1.9513e-01,  1.3377e-01, -3.0990e-01, -3.2959e-02,  7.5167e-02,\n",
       "          -3.2047e-03,  5.1146e-01,  1.5852e-01,  4.5032e-02, -5.4633e-01,\n",
       "           5.0991e-03, -3.4575e-01, -1.8532e-01, -3.1348e-01, -2.3796e-01,\n",
       "          -2.3140e-01,  2.1523e-01,  9.2191e-01, -3.9901e-02, -3.2745e-01,\n",
       "           2.0235e-01,  4.3616e-01, -2.2154e-01,  1.2902e-01, -1.0705e-01,\n",
       "          -8.7493e-01, -2.0283e-01, -3.1608e-01, -9.2073e-02, -5.0724e-01,\n",
       "           8.1716e-02,  5.3157e-01, -2.2567e-01, -3.5532e-01, -7.1859e-01,\n",
       "           6.4811e-01,  2.9076e-03,  3.0714e-01,  7.2434e-02, -2.5062e-01,\n",
       "           9.1221e-03, -6.7846e-01,  6.4055e-02, -4.1870e-01, -1.8144e-01,\n",
       "           1.9941e-01,  3.8849e-01,  6.8047e-02,  3.8520e-01, -3.7297e-01,\n",
       "           7.4747e-01, -2.6325e-01, -2.1898e-01,  4.7474e-01, -3.2286e-01,\n",
       "          -2.3083e-01,  3.9582e-01,  5.6485e-02,  4.7441e-01,  1.4216e-01,\n",
       "           5.4268e-01, -8.9980e-02, -4.8985e-01, -9.5028e-02,  3.3060e-01,\n",
       "           1.7727e-01,  3.4890e-01, -3.5007e-01,  1.6903e-01,  2.4325e-01,\n",
       "          -3.8360e-01,  4.3129e-01, -1.5457e-01, -5.6331e-01,  7.7888e-02,\n",
       "           3.5232e-01,  1.5139e-01, -2.5115e-01, -4.3980e-01,  4.2802e-01,\n",
       "           3.3042e-01, -2.8265e-02,  4.7104e-01,  4.0668e-01,  5.7504e-02,\n",
       "          -4.2294e-01, -5.3418e-01,  1.7543e-01, -3.8194e-01, -1.2081e-01,\n",
       "           4.0802e-01, -2.3486e-01,  3.2215e-01, -4.1620e-01, -3.9781e-01,\n",
       "          -4.0553e-01,  1.8830e-01,  7.0715e-01,  3.9212e-01,  2.6673e-02,\n",
       "           1.5473e-01, -1.1818e-01, -2.6865e-01,  6.3599e-01, -2.0134e-01,\n",
       "          -4.4462e-01,  3.3700e-01, -6.9939e-02,  1.1419e-02,  1.6519e-01,\n",
       "          -3.1021e-01,  1.4483e-01,  2.9772e-01,  5.3289e-01, -5.9763e-01,\n",
       "          -3.0273e-01, -7.1125e-02, -3.4974e-01, -2.3673e-01,  1.9744e-01,\n",
       "           6.0802e-02,  6.2468e-01, -2.9033e-01, -1.1427e-01,  5.1646e-01,\n",
       "           3.9408e-02, -3.2582e-01,  5.2736e-02, -6.8153e-01, -5.8121e-01,\n",
       "           2.2218e-02,  1.5617e-01,  1.4613e-01,  4.6585e-01, -1.0673e+01,\n",
       "          -1.9734e-01,  7.2856e-01, -4.3891e-01, -1.3803e-01, -3.1176e-02,\n",
       "           1.6795e-01, -1.3011e-01,  2.4977e-01, -3.3043e-02, -6.1006e-01,\n",
       "          -3.5913e-01, -2.2300e-01,  9.9408e-02,  9.5989e-02, -2.8588e-01,\n",
       "          -8.1194e-02, -7.9884e-01,  1.3113e-02, -5.6701e-01,  2.1474e-01,\n",
       "          -2.4682e-01,  8.1533e-02,  3.1255e-01, -3.0808e-01, -2.1892e-01,\n",
       "          -1.1129e-01, -2.2297e-01,  3.2727e-02,  2.7289e-01, -2.7498e-01,\n",
       "          -6.8322e-02,  5.0010e-02,  9.7761e-02, -1.2109e-01, -8.3639e-01,\n",
       "           4.3223e-02,  8.7456e-02,  5.1261e-01, -1.1128e-01, -7.4569e-01,\n",
       "          -5.7318e-01, -7.7829e-01,  4.3140e-01,  4.9790e-01,  8.6354e-01,\n",
       "           1.0071e+00,  5.7072e-01,  2.7995e-01,  1.0659e-01, -8.8878e-02,\n",
       "          -6.0000e-01,  4.4897e-02, -7.5548e-01, -2.3466e-01, -1.4439e-01,\n",
       "           4.9711e-01, -6.3412e-01,  4.3218e-01, -4.2238e-01, -2.4815e-01,\n",
       "           1.9476e-01,  8.4054e-02,  3.1906e-01, -2.0361e-01, -5.9621e-01,\n",
       "           7.3386e-01, -8.9460e-02,  1.5777e-01, -3.8438e-01,  1.9053e-01,\n",
       "          -1.5129e-01,  5.8248e-01,  3.7510e-01, -3.6062e-01,  4.5076e-01,\n",
       "          -2.6379e-02, -2.1468e-01,  1.6421e+00, -2.2173e-01,  2.8413e-01,\n",
       "          -5.3991e-02,  2.6585e-01, -4.5817e-01, -9.2274e-01,  2.5168e-01,\n",
       "           4.5895e-02, -8.7339e-02,  4.9363e-01,  3.2698e-01, -3.2111e-01,\n",
       "          -1.0401e+00,  1.8524e-01,  1.6447e-01,  4.2083e-01,  9.5653e-01,\n",
       "          -4.4297e-01, -4.2426e-02,  7.7218e-02,  3.6375e-02, -9.9477e-01,\n",
       "          -6.4858e-02, -3.1193e-01, -3.9964e-01,  5.0519e-01,  2.1779e-01,\n",
       "           3.7408e-01,  3.0562e-01, -9.0196e-02,  4.8770e-01, -4.3756e-01,\n",
       "           2.0554e-01, -1.8605e-01,  7.2248e-02, -4.0879e-01,  8.7831e-02,\n",
       "          -1.0684e-01,  2.1824e-01, -3.1494e-01, -4.8838e-01,  9.1534e-03,\n",
       "           2.4548e-02,  1.1405e-01,  4.7704e-01, -6.4296e-01,  6.3246e-01,\n",
       "          -5.5827e-01, -4.6545e-02,  1.6313e-01, -2.3473e-01,  7.0262e-01,\n",
       "          -3.6482e-01, -5.0331e-01, -2.6270e-01, -6.1320e-01, -6.0827e-01,\n",
       "          -4.9465e-01, -1.1648e-01,  2.3293e-01, -8.4324e-01,  2.4662e-02,\n",
       "           1.5833e-01,  3.3180e-01, -2.0775e-01,  1.5904e-01,  3.0296e-01,\n",
       "          -5.1821e-02,  6.7867e-01, -5.1878e-02,  2.2547e-01, -1.7628e-01,\n",
       "          -6.1413e-01, -4.0571e-03,  2.4586e-01, -7.5563e-02, -9.4646e-01,\n",
       "          -1.7237e-01, -9.2544e-02, -6.1398e-01,  2.9202e-01,  4.1018e-02,\n",
       "           1.2396e-02, -1.7625e-01, -4.5151e-01,  3.1504e-01,  6.1181e-01,\n",
       "           2.6428e-01,  3.5205e-02, -3.4921e-01,  1.5642e-01,  4.8614e-01,\n",
       "          -1.4670e-01, -1.3748e-01, -4.2939e-01, -3.7291e-01,  2.5533e-01,\n",
       "           3.4914e-01,  5.1085e-01, -3.4816e-01, -8.5053e-01,  2.3787e-01,\n",
       "           6.5368e-01,  1.1494e-01, -8.7006e-02, -1.6915e-01, -9.7423e-02,\n",
       "          -2.8194e-01, -2.7946e-01,  5.5062e-01, -2.3163e-01,  1.6869e-01,\n",
       "           1.2926e-01,  4.3292e-01,  2.5555e-01,  4.7256e-01, -9.0803e-01,\n",
       "          -2.4659e-01,  1.1982e-02, -9.4261e-01,  3.4586e-01, -7.1104e-02,\n",
       "           1.5347e-01,  3.5674e-01, -1.8531e-01, -1.0088e-02,  3.1475e-01,\n",
       "           5.3564e-01,  7.7270e-01,  1.2693e-01,  3.1631e-01,  9.6411e-01,\n",
       "           5.3028e-01, -4.7644e-02,  1.2383e-01,  4.0571e-01, -6.4512e-02,\n",
       "           3.1255e-01, -1.1711e-01, -2.6947e-01,  8.3976e-02,  2.2205e-01,\n",
       "           5.3073e-01, -1.3487e+00, -2.9501e-01, -1.8982e-01,  2.4601e-01,\n",
       "          -3.1420e-01,  6.4652e-02, -1.5225e-01,  3.2995e-01,  8.1327e-01,\n",
       "          -1.2164e+00, -4.9275e-01,  5.2144e-01,  3.0654e-01, -1.1219e-02,\n",
       "          -3.5008e-01,  2.2770e-01, -4.8373e-01, -9.7997e-02,  5.7713e-01,\n",
       "           3.3090e-01, -8.2741e-01,  2.0685e-01,  4.8348e-01, -4.3296e-01,\n",
       "           1.3549e-01, -9.1803e-03,  1.2389e-01,  6.2907e-02,  9.1241e-02,\n",
       "          -1.1688e-01,  2.9084e-01, -4.7971e-01, -4.3270e-01, -1.2697e-01,\n",
       "           5.6923e-01,  3.2594e-02,  2.4708e-01, -2.1493e-01,  3.4278e-01,\n",
       "          -2.8664e-01, -1.9235e-01, -2.7945e-01, -4.0010e-02, -2.2063e-01,\n",
       "           2.9688e-01, -2.3626e-01, -3.9713e-02,  3.0681e-01, -2.4436e-01,\n",
       "           4.2609e-01,  4.0903e-01,  7.4140e-01,  1.6704e-01, -9.6687e-02,\n",
       "           1.6022e-01,  6.3798e-02, -3.6927e-01,  6.3068e-01, -9.5398e-02,\n",
       "          -3.6084e-01,  5.6818e-02,  1.0051e-01,  1.2848e-01, -8.5435e-02,\n",
       "          -2.9302e-01,  2.4281e-01,  4.6508e-02, -3.4722e-01,  2.0644e-01,\n",
       "           3.7436e-01,  2.0495e-01, -5.2351e-01,  2.1821e-01,  3.3161e-01,\n",
       "          -6.1497e-02,  1.7849e-01, -2.1486e-01,  2.6979e-01, -4.2086e-01,\n",
       "           3.8906e-01, -7.3082e-01, -2.3349e-01,  3.5943e-01, -1.1577e-01,\n",
       "          -5.7683e-01, -2.6655e-02, -2.1028e-01,  4.9877e-02,  9.4740e-01,\n",
       "           9.8394e-01, -7.9699e-02, -5.1125e-01, -1.6238e-01,  2.2794e-01,\n",
       "          -1.2084e-01, -2.6361e-01,  5.1132e-01, -7.5037e-02, -2.3106e-02,\n",
       "          -1.2072e+00, -3.3951e-01,  1.2041e-01,  9.5972e-02, -2.1621e-01,\n",
       "           9.3096e-01, -1.7158e-01,  2.4132e-01,  7.3350e-01,  1.2953e-01,\n",
       "           4.0601e-01,  5.2136e-01, -2.9974e-02, -3.3793e-01, -1.1615e-01,\n",
       "           8.7239e-02, -6.4601e-02, -1.3874e-01, -5.7425e-01,  2.7846e-01,\n",
       "           1.8668e-01,  4.4243e-01,  6.2478e-01]], device='mps:0'),\n",
       " tensor([[-0.0111, -0.5057, -0.4670,  ...,  0.2365, -0.0712,  0.4388],\n",
       "         [-0.3292, -0.2590, -0.5016,  ..., -0.0148,  0.3509,  0.1607],\n",
       "         [-0.2350, -0.6088, -0.4940,  ...,  0.4151,  0.8705,  0.4670],\n",
       "         [-0.4197, -0.4211, -0.1311,  ...,  0.1104,  0.3812,  0.5167],\n",
       "         [-0.2913, -0.5296, -0.2854,  ..., -0.1966,  0.5443,  0.3917]],\n",
       "        device='mps:0'),\n",
       " tensor([[-1.7731e-01, -3.8604e-01, -5.6476e-01, -2.2650e-01, -1.3208e-01,\n",
       "          -3.1625e-01, -5.9292e-01,  4.5825e-01,  2.2151e-01,  5.2940e-01,\n",
       "           3.8502e-01,  1.4343e-01,  6.5615e-02, -2.5563e-01, -4.4222e-01,\n",
       "          -1.6916e-01,  1.6884e-01,  5.0107e-01,  3.6126e-01, -7.6046e-03,\n",
       "           5.9085e-01,  3.7327e-01, -2.0317e-01,  2.5562e-01, -7.5220e-01,\n",
       "           2.6747e-02, -1.6815e-01, -2.5680e-02,  9.6832e-02, -3.0143e-01,\n",
       "          -1.1712e-01,  2.7595e-01, -3.7936e-01,  7.2144e-01, -1.4269e-01,\n",
       "          -2.3945e-01,  1.6259e-01, -1.0450e-01,  5.3649e-01, -1.3024e-01,\n",
       "           5.7769e-01, -2.9111e-01,  4.9321e-01,  3.4162e-02,  2.2711e-01,\n",
       "           1.3206e-01,  2.9819e-01, -4.0790e-01,  1.3721e-01, -5.3671e-01,\n",
       "          -1.6362e-02, -7.3465e-02,  4.1546e-01,  1.0724e-01, -6.3685e-01,\n",
       "          -2.7220e-01, -7.3533e-02,  4.2370e-01, -3.4661e-01,  1.4839e-01,\n",
       "           1.4452e-01,  4.6875e-01,  1.2890e-01,  1.1326e-01, -3.3348e-01,\n",
       "           3.0093e-01, -7.0323e-01, -8.0455e-01, -5.7621e-01, -1.8468e-01,\n",
       "           2.6210e-01,  2.4685e-01,  3.2711e-01, -6.3178e-01,  2.8882e-01,\n",
       "          -5.5500e-01,  4.7778e-01, -3.7641e-02, -3.8107e-01,  2.4976e-02,\n",
       "           4.6662e-01,  1.6673e-01, -2.3893e-01,  2.6411e-01,  2.9965e-01,\n",
       "          -6.2089e-02,  4.4818e-01, -2.9706e-02,  5.6407e-02,  3.8863e-01,\n",
       "           5.0392e-01,  2.6142e-01, -5.4248e-01, -2.1306e-01,  2.3546e-01,\n",
       "          -3.7940e-01,  2.9621e-01, -1.0776e-01,  2.0922e-02, -8.6342e-02,\n",
       "          -3.6881e-02,  3.4334e-01,  3.1658e-01, -6.2528e-02, -1.0724e+00,\n",
       "          -4.3983e-01, -2.4311e-02, -2.7814e-01, -3.4665e-01, -1.0267e-02,\n",
       "           7.0862e-03, -2.1454e-01,  4.5842e-01, -4.4114e-01, -3.9926e-01,\n",
       "           1.3807e-01, -3.8646e-02, -1.5082e-02, -1.4568e-01,  6.0048e-01,\n",
       "           1.6661e-01, -5.3057e-02,  1.5341e-01,  8.8933e-02,  4.9861e-01,\n",
       "          -3.9381e-01,  3.7161e-01, -4.2227e-01,  3.0903e-01,  6.8681e-02,\n",
       "          -2.0183e-02, -7.2510e-02, -2.5207e-01, -2.0895e-01, -2.4617e-02,\n",
       "           2.1165e-02,  2.8627e-01, -4.8458e-01,  5.5727e-02,  5.6803e-02,\n",
       "           2.5004e-01, -1.2498e-01, -4.2667e-01, -1.9318e-01, -6.4563e-02,\n",
       "          -4.0302e-02,  1.9246e-01, -1.1367e-01, -2.7591e-01, -2.6032e-01,\n",
       "          -7.7253e-01, -4.4007e-01,  3.9722e-01,  1.0952e-01,  9.2693e-02,\n",
       "           1.1479e-01,  2.1754e-01, -1.9775e-01, -1.8512e-01, -1.5673e-01,\n",
       "           5.8319e-01, -9.5195e-02,  6.6991e-02, -3.6372e-01, -3.0891e-01,\n",
       "           7.3167e-02,  1.7092e-01, -1.7339e-01,  4.3863e-02,  1.4590e-02,\n",
       "           2.2225e-01, -8.8855e-02, -1.0134e-03, -7.5909e-01, -3.0093e-02,\n",
       "          -2.8356e-01,  3.3814e-01,  2.7840e-01,  1.8768e-01, -2.4537e-01,\n",
       "           5.0899e-02, -1.7294e+00,  7.8671e-01, -5.9166e-01,  2.4877e-01,\n",
       "          -1.8150e-01,  3.1718e-01,  9.1189e-02,  1.8509e-01, -2.9267e-01,\n",
       "          -1.2026e-01, -3.9309e-02,  4.1623e-02,  2.9227e-01, -8.4757e-02,\n",
       "           4.8559e-01,  5.5627e-01,  3.3170e-01,  2.0436e-01, -9.9033e-02,\n",
       "          -6.0977e-01,  3.3235e-01,  1.6388e-01, -2.7455e-01,  2.9857e-02,\n",
       "           2.4361e-01, -3.2791e-01, -5.7703e-02,  3.4800e-01,  3.3556e-02,\n",
       "          -8.5587e-02, -3.0852e-01, -4.0912e-01, -1.8006e-01, -1.6990e-01,\n",
       "           3.0904e-01, -2.3500e-01, -2.0646e-01,  7.4699e-01,  9.5515e-02,\n",
       "          -2.2466e-02, -7.9463e-02, -2.8277e-01, -4.4039e-01, -2.0940e-01,\n",
       "           2.9576e-01, -5.3435e-01, -1.7453e-01, -2.8854e-01, -4.6020e-01,\n",
       "           4.6218e-01, -9.4786e-01,  6.3073e-01,  2.5225e-01,  1.6028e-01,\n",
       "           6.5808e-02, -8.2977e-02, -1.3264e-01, -3.6344e-01, -1.9702e-02,\n",
       "          -2.1377e-02, -5.7642e-01, -9.0486e-01,  1.5472e-01, -7.9254e-02,\n",
       "           1.2900e-01, -4.1901e-02,  6.2316e-01,  4.2972e-01,  5.9118e-01,\n",
       "           2.3140e-01,  5.5355e-03,  4.6999e-01,  4.3441e-01, -4.1017e-01,\n",
       "           4.6498e-01,  1.8871e-01, -4.7838e-02,  6.9646e-02, -3.2186e-01,\n",
       "          -9.1404e-01,  2.0856e-02,  3.2947e-01,  4.9930e-01, -1.7114e-01,\n",
       "           4.2813e-01,  2.6662e-01, -1.3998e-01, -1.7031e-01,  3.1347e-01,\n",
       "          -4.8944e-01, -1.5127e-01,  9.5415e-02, -1.4029e-01, -6.5634e-01,\n",
       "          -4.8841e-01,  2.7357e-02, -2.9199e-01,  2.0484e-01, -9.9702e-01,\n",
       "          -2.6273e-01,  5.6883e-01, -2.8623e-02,  9.3732e-02, -2.3345e-01,\n",
       "           7.0240e-02, -1.0750e-01,  1.0928e-01, -1.5161e-01, -8.9822e-01,\n",
       "          -1.4258e-01,  9.6331e-02, -2.2318e-01,  3.2530e-01,  3.7809e-01,\n",
       "          -2.0083e-01, -2.9617e-01,  2.8364e-01, -9.0550e-02, -2.5039e-01,\n",
       "          -1.3545e-01, -3.7063e-01, -2.4607e-02, -1.3576e-01, -1.1638e-01,\n",
       "           6.9800e-02,  4.9085e-01,  4.8805e-01, -7.7597e-01, -5.5824e-01,\n",
       "          -2.2992e-02,  5.0756e-01, -1.6302e-01,  3.7266e-02, -2.2967e-01,\n",
       "          -1.0825e+00, -2.0374e-01, -1.1835e-01, -3.9640e-01, -3.8528e-01,\n",
       "           5.0161e-01, -1.9830e-01, -2.0671e-01,  9.6594e-02, -3.4539e-01,\n",
       "           3.4160e-01,  8.4200e-01, -4.9032e-02, -4.7416e-02, -2.9725e-01,\n",
       "           1.7477e-02, -1.8692e-01,  4.8213e-02, -1.9462e-01, -2.5485e-02,\n",
       "           3.1570e-01,  1.2741e-01, -1.8351e-01, -8.2635e-02, -1.7372e-01,\n",
       "           7.9347e-01, -5.6947e-01,  1.3088e-01,  3.8738e-01, -2.4876e-01,\n",
       "          -1.3259e-01,  4.8066e-01,  1.4439e-01,  1.9491e-01, -1.0024e-01,\n",
       "          -2.6493e-02, -2.8608e-01, -1.8423e-01, -3.1248e-01,  9.6616e-01,\n",
       "           3.5980e-01,  2.2368e-01, -3.0496e-01,  5.7904e-01,  5.6810e-01,\n",
       "          -3.8525e-01,  7.6086e-02,  8.8516e-02,  3.4151e-03, -3.1432e-01,\n",
       "           3.2810e-01,  2.4729e-01, -4.2766e-01, -2.5123e-02,  4.5374e-03,\n",
       "           1.4485e-01, -5.5797e-01,  1.6191e-01,  1.0782e-01,  8.5195e-02,\n",
       "          -1.4724e-03, -2.6111e-01,  5.2888e-01, -2.4531e-01, -2.8912e-01,\n",
       "          -1.0092e-01, -1.2182e-01,  4.1012e-02, -2.4548e-01, -3.6887e-01,\n",
       "          -2.7428e-01,  2.4193e-02,  3.5074e-01,  1.4285e-02,  7.4821e-02,\n",
       "           2.5544e-01, -2.6796e-01, -3.4638e-01,  4.5500e-01, -1.7660e-02,\n",
       "          -1.6810e-01,  4.0768e-01, -2.4890e-01, -1.1639e-01,  1.3881e-01,\n",
       "           6.2926e-02,  1.0677e-01, -5.8094e-02,  3.7998e-01,  1.7175e-01,\n",
       "          -2.8992e-01,  3.6344e-02,  2.1408e-01, -8.0245e-02,  3.1665e-01,\n",
       "           5.1861e-02,  8.2278e-01, -3.7109e-01, -2.7164e-02, -3.8441e-01,\n",
       "          -1.2821e-01, -1.6499e-01,  7.5490e-03,  1.5024e-01, -7.5773e-01,\n",
       "           3.0131e-01,  2.8842e-01,  3.4034e-01,  1.1331e-01, -1.1458e+01,\n",
       "           1.6305e-01,  6.4218e-01, -5.1374e-01,  1.8438e-01,  1.7594e-01,\n",
       "          -1.4040e-01, -3.9794e-01,  4.0875e-02, -3.4785e-01, -2.6394e-01,\n",
       "          -6.5590e-02,  1.9075e-01, -1.2756e-02, -5.5478e-01,  3.3680e-02,\n",
       "          -2.5398e-01, -6.3192e-01,  2.6160e-01, -9.0323e-02,  4.1335e-01,\n",
       "          -3.0876e-01,  1.8731e-01,  2.3832e-01, -3.6034e-01, -2.7614e-02,\n",
       "          -5.1547e-01, -5.5807e-02,  3.7042e-01,  3.2770e-01, -3.5254e-01,\n",
       "           2.2243e-01, -1.8729e-01,  3.8908e-01, -2.6449e-01, -5.2363e-01,\n",
       "           1.3092e-01, -9.6993e-02,  3.7952e-01,  1.1884e-02, -2.4878e-01,\n",
       "          -2.5432e-02, -5.2584e-01,  5.9952e-02,  2.0270e-01,  3.4279e-01,\n",
       "           1.2541e+00,  9.4924e-01,  1.3760e-01, -1.3198e-01,  1.6614e-01,\n",
       "          -7.3513e-01,  2.0677e-01, -6.2811e-01, -8.1585e-02, -6.4376e-02,\n",
       "           8.3128e-01, -4.3261e-01, -3.3743e-01,  7.3012e-02, -8.0493e-02,\n",
       "           3.1300e-01, -7.9471e-02,  2.6202e-01, -3.1186e-01,  2.9341e-03,\n",
       "           6.1375e-01, -1.8722e-02,  3.8063e-01, -4.3721e-01, -6.2672e-02,\n",
       "          -4.2457e-02,  2.9525e-01,  3.7279e-01,  6.2687e-02, -2.5474e-01,\n",
       "          -4.1450e-01,  4.2406e-02,  5.2243e-01,  1.3990e-01,  1.4541e-01,\n",
       "           5.8211e-01, -2.3165e-01, -4.1432e-01, -6.0104e-01, -3.9034e-01,\n",
       "          -7.6601e-03,  1.3684e-01, -3.9867e-01,  6.4750e-01, -4.2253e-01,\n",
       "          -2.7567e-01, -1.6427e-01, -8.8628e-02,  3.8882e-01,  5.0639e-01,\n",
       "          -3.3627e-01,  8.0837e-02, -1.8585e-01, -4.4707e-03, -6.7852e-01,\n",
       "          -3.5458e-01, -8.4538e-02, -2.9987e-01,  2.3231e-01,  1.7305e-01,\n",
       "          -1.0020e-01,  9.0697e-03,  1.2051e-01,  7.3101e-02, -5.4356e-01,\n",
       "           3.0479e-01, -1.0749e-01,  6.8380e-02, -3.9586e-01, -5.1756e-01,\n",
       "           4.0833e-01,  7.0232e-01, -4.2415e-01, -3.3873e-01,  9.8320e-02,\n",
       "           1.3073e-01,  1.8252e-01,  1.2615e-01, -6.4414e-01,  1.1900e-01,\n",
       "          -4.1931e-01, -1.3812e-01,  1.3512e-01,  7.2335e-01,  6.6950e-01,\n",
       "          -5.3703e-02, -3.2049e-01,  2.4632e-01, -5.0640e-02, -2.5765e-01,\n",
       "          -7.0535e-01, -2.9464e-02,  2.1372e-01, -5.3812e-01,  9.7345e-02,\n",
       "          -3.3445e-01,  1.9755e-01, -1.7248e-01,  1.0825e-01, -7.4644e-02,\n",
       "           4.1662e-01,  4.1987e-01,  7.6810e-02,  4.3515e-01, -1.7908e-01,\n",
       "          -2.1990e-01,  1.9709e-01, -1.7846e-01, -1.1703e-01, -1.7569e-01,\n",
       "          -1.7366e-01,  9.7459e-02, -4.0722e-01, -3.6435e-02,  1.7903e-01,\n",
       "          -1.2723e-01,  6.7948e-02, -3.4229e-01,  2.2739e-01,  3.3538e-01,\n",
       "          -1.0467e-01, -3.7966e-01, -2.3884e-01,  4.7521e-02,  7.1481e-02,\n",
       "           3.8731e-01, -2.6195e-01, -1.0102e-01, -7.3811e-01, -2.0636e-01,\n",
       "           3.6787e-02,  1.0410e+00, -3.7965e-02, -3.0778e-01,  3.5156e-01,\n",
       "           3.9051e-01, -2.4726e-01, -3.4583e-01, -1.8699e-01,  7.4840e-01,\n",
       "          -2.3598e-01, -1.5465e-01,  2.2099e-01,  2.5510e-02,  6.2240e-01,\n",
       "           1.3154e-01,  2.2300e-01, -1.8512e-01,  4.7019e-01,  2.1848e-01,\n",
       "          -4.4102e-01, -1.5073e-01, -3.9303e-01, -3.2472e-01,  4.3754e-01,\n",
       "          -6.9776e-02,  2.7454e-01, -1.1790e-01,  8.1553e-02,  1.4253e-02,\n",
       "           5.4206e-02,  1.1645e+00, -2.0185e-01, -2.3960e-01,  4.7684e-01,\n",
       "           2.7079e-01,  3.4262e-01,  3.7917e-02, -1.3477e-01,  9.5921e-02,\n",
       "           4.9705e-01, -1.6706e-01, -4.9624e-01,  1.5664e-01,  3.6272e-01,\n",
       "           5.9108e-01, -8.7613e-01, -3.6622e-01,  2.0509e-01,  2.2296e-01,\n",
       "          -3.2121e-01,  1.9818e-01, -7.6228e-02,  3.2510e-01,  3.6266e-01,\n",
       "          -1.2771e+00,  1.7642e-01,  7.1262e-02,  1.5943e-01, -1.5269e-01,\n",
       "          -6.7715e-01, -2.5385e-01,  1.0595e-01, -6.2916e-02,  4.2076e-01,\n",
       "           4.4789e-01, -7.1509e-01,  1.2452e-01,  3.1511e-01, -3.7691e-01,\n",
       "           1.2320e-01, -2.7571e-01,  1.6643e-01,  5.2170e-01,  1.6129e-01,\n",
       "          -4.2505e-01,  2.5049e-01, -5.0826e-01, -6.1173e-01,  1.4561e-01,\n",
       "           6.9575e-01,  2.1324e-02, -1.1322e-01, -1.1804e-01, -2.9289e-01,\n",
       "          -1.9107e-01, -3.1790e-01, -3.4967e-01, -2.7542e-02,  1.4802e-01,\n",
       "           2.4431e-02, -3.6606e-02,  2.4884e-01,  1.1143e-01, -2.2300e-01,\n",
       "           5.8287e-01,  4.8747e-01,  7.4334e-01, -1.5062e-02,  1.1963e-02,\n",
       "          -2.2207e-01,  2.9970e-01, -4.7196e-02,  4.7552e-01, -3.4433e-01,\n",
       "          -2.0911e-01,  2.2006e-01,  3.6212e-02, -6.2294e-01, -1.0381e-01,\n",
       "          -3.5492e-01,  3.0044e-01,  1.1075e-02, -2.7609e-01,  4.8880e-01,\n",
       "           2.0553e-02,  1.8280e-01, -5.5884e-01,  4.6756e-01,  1.6596e-01,\n",
       "           2.2073e-02,  9.8054e-01,  1.1685e-01,  5.3145e-01,  5.8221e-01,\n",
       "           1.1828e-01, -8.1138e-01, -1.8763e-01, -7.2256e-02,  1.5035e-01,\n",
       "           1.1544e-02,  1.5872e-03,  1.5741e-01,  2.8459e-02,  9.3799e-01,\n",
       "           4.5776e-01, -2.2136e-01, -3.9686e-01, -3.7782e-01,  1.8491e-01,\n",
       "          -8.4548e-02, -3.3508e-01,  2.0442e-01, -2.2187e-01,  1.7388e-02,\n",
       "          -5.8935e-01, -5.5885e-01, -1.2302e-01,  2.7758e-01, -2.6880e-01,\n",
       "           5.4517e-01,  4.6664e-02,  5.2447e-03,  3.4141e-01, -2.9763e-02,\n",
       "           2.1521e-01,  6.7382e-01,  2.2993e-01, -2.9428e-01,  1.8031e-01,\n",
       "           2.1968e-01, -1.3524e-01,  2.0995e-01, -8.6762e-01,  3.6296e-01,\n",
       "          -5.1872e-02,  5.4038e-01,  4.7469e-01]], device='mps:0'),\n",
       " tensor([[-2.8441e-01, -5.5394e-01, -5.3487e-01, -8.6137e-01, -1.6500e-01,\n",
       "          -1.5627e-01, -6.9154e-01,  6.2842e-01, -3.2176e-01,  7.4840e-01,\n",
       "           2.7524e-01,  6.2658e-01,  4.8185e-02,  8.1655e-02, -5.4587e-01,\n",
       "          -1.8390e-01, -8.5546e-02,  3.8717e-01,  2.1180e-01,  6.1308e-01,\n",
       "           8.1107e-01,  6.8047e-01, -1.9768e-01,  2.1503e-01, -5.0498e-01,\n",
       "           3.5649e-01, -6.9646e-02,  6.8591e-01,  2.3685e-01, -4.8309e-01,\n",
       "           4.5488e-02,  5.1755e-01, -7.6515e-02,  1.6821e+00, -1.0095e-01,\n",
       "          -2.5680e-01,  2.3473e-01, -5.4682e-01, -4.1870e-01, -4.8584e-01,\n",
       "           6.0923e-01, -3.5607e-01,  1.1066e-02,  1.0019e-01, -1.2748e-01,\n",
       "           4.3426e-01,  3.3175e-01, -4.5028e-01,  2.1806e-01, -8.6036e-01,\n",
       "          -6.0268e-01, -1.1943e-01,  1.3884e-01,  7.5491e-01, -7.6410e-01,\n",
       "          -7.1493e-01,  8.1095e-02, -1.2005e-01, -1.3680e-01,  3.9091e-02,\n",
       "           2.6227e-01,  2.0679e-01, -8.1044e-02,  6.1526e-02,  1.7546e-01,\n",
       "           1.7241e-01, -1.9290e-01, -7.4059e-01, -4.7855e-01,  7.6078e-02,\n",
       "           9.5219e-02,  1.1330e-01,  1.1899e-01, -1.9241e-01,  3.2532e-01,\n",
       "          -2.3076e-01,  2.7391e-01, -3.3521e-01, -1.3154e-01, -7.0176e-02,\n",
       "          -4.4051e-02,  8.9766e-02,  1.9541e-01, -1.9095e-01, -1.4401e-01,\n",
       "           1.1033e-02,  5.0526e-01,  5.0223e-02, -8.2932e-02, -4.1653e-02,\n",
       "           1.5756e-02,  1.8359e-02, -6.3429e-01, -4.0049e-01,  1.9855e-01,\n",
       "           7.6685e-03,  3.6200e-01, -3.5243e-01,  1.8487e-01, -2.3329e-01,\n",
       "           2.0495e-01, -1.3106e-01, -1.0602e-01,  5.0891e-02, -6.4400e-01,\n",
       "          -5.0268e-01,  1.7121e-01,  1.3393e-05, -5.4945e-01,  1.4619e-02,\n",
       "          -4.0735e-01, -6.1942e-01,  3.2930e-02, -2.3743e-01, -5.5993e-02,\n",
       "           1.5373e-01,  6.8113e-02, -4.6111e-04,  3.8857e-03, -7.4266e-02,\n",
       "           5.8578e-01, -4.1646e-02,  1.6548e-01,  9.9639e-02,  2.6984e-01,\n",
       "          -3.2575e-01,  6.6620e-01, -3.1006e-01,  2.6532e-01,  4.3677e-01,\n",
       "          -1.2551e-01,  2.2294e-03, -3.9518e-01, -9.1430e-02,  3.8994e-01,\n",
       "           3.5504e-01, -2.5205e-01, -1.6596e-01, -3.2194e-01,  1.2660e-01,\n",
       "          -1.1306e-01, -1.9384e-01, -1.2373e-01, -4.7651e-01, -9.7027e-02,\n",
       "          -2.5137e-01, -1.8891e-01,  9.8524e-03, -1.3107e-02, -5.6441e-01,\n",
       "          -3.7041e-01, -1.0094e+00,  8.9048e-01, -7.7338e-02,  1.0971e-01,\n",
       "          -5.2245e-02,  1.3917e-01,  1.3157e-01,  2.4190e-01,  3.4615e-01,\n",
       "           4.0161e-01, -3.0681e-01,  1.7669e-01, -6.3478e-01, -2.1620e-01,\n",
       "          -1.6021e-02,  5.5562e-02, -4.2343e-01,  4.4272e-01,  1.4258e-01,\n",
       "           6.6604e-02, -3.1665e-01,  9.0541e-02, -2.5514e-01,  2.3988e-01,\n",
       "          -5.2097e-01,  1.3542e-01,  8.6542e-03,  9.9876e-02, -2.3345e-01,\n",
       "           1.6611e-01, -1.2452e+00,  8.5164e-02, -1.3329e-01, -3.9477e-02,\n",
       "          -3.4149e-01,  3.9552e-01,  3.1089e-01,  3.2796e-01, -1.5308e-01,\n",
       "          -8.1520e-02, -7.1932e-02,  9.9920e-02,  2.7144e-02, -2.8755e-01,\n",
       "           4.6967e-01,  6.6779e-01,  4.0208e-01, -1.6045e-01,  9.8117e-02,\n",
       "           2.1663e-02,  2.2509e-01, -6.3057e-02,  1.4737e-01,  3.4920e-01,\n",
       "           1.6982e-01, -1.9862e-01, -5.2368e-02,  3.0330e-01, -4.0755e-02,\n",
       "          -2.1991e-01, -2.3430e-02, -4.2824e-01, -1.0580e-01, -3.9187e-01,\n",
       "           1.2879e-01, -3.9608e-01, -1.2370e-01,  2.9763e-01, -1.3954e-01,\n",
       "          -1.9156e-01,  4.9995e-02,  1.7278e-01, -1.4398e-01,  1.1770e-01,\n",
       "           1.3569e-01, -3.2259e-01,  9.6180e-02, -3.1440e-01, -1.8272e-01,\n",
       "           5.7590e-02, -4.0861e-01,  7.0641e-01,  2.7090e-01, -2.1076e-01,\n",
       "           2.2706e-01, -3.7797e-01,  7.5726e-02, -1.9180e-02,  1.7069e-01,\n",
       "          -5.9430e-03, -3.4191e-01, -9.1396e-01,  2.0747e-01,  6.7531e-02,\n",
       "           2.8930e-01, -2.4971e-02,  4.7363e-01,  8.3110e-01,  1.9939e-01,\n",
       "          -3.0489e-02, -1.6327e-01,  4.5326e-01,  3.1228e-01, -2.1691e-01,\n",
       "          -2.6165e-02, -9.7939e-02, -8.4730e-01,  3.6313e-02, -4.6692e-01,\n",
       "          -2.0889e-02,  2.2189e-01, -3.3192e-03,  2.0753e-01, -3.7387e-01,\n",
       "           6.3320e-01,  4.4671e-01,  9.7346e-02, -3.8640e-01,  6.3860e-01,\n",
       "           1.0362e-01,  3.3258e-01, -3.3726e-01, -1.6538e-01, -5.7175e-01,\n",
       "          -6.9858e-01, -1.9525e-01,  4.4452e-01,  7.1309e-02, -4.3032e-01,\n",
       "          -1.7021e-02,  1.0086e-01,  4.6269e-02, -3.2002e-01,  4.1130e-02,\n",
       "           1.1790e-01, -1.4409e-01, -2.3553e-02, -1.8172e-01, -3.0541e-01,\n",
       "          -4.0109e-01,  3.2059e-01, -4.5628e-01, -1.9039e-02,  6.3341e-01,\n",
       "          -5.6343e-02, -2.5236e-01,  1.3631e-01, -7.7882e-03, -1.5985e-01,\n",
       "           7.7204e-02,  4.9073e-01, -1.7577e-01, -3.1271e-01,  3.3961e-01,\n",
       "          -6.8220e-02,  4.1486e-01,  3.8705e-02, -5.9873e-01, -2.8172e-01,\n",
       "           8.1087e-02,  4.2650e-01, -6.3146e-02, -1.9707e-01, -1.0664e-01,\n",
       "          -7.9628e-01, -3.1771e-01, -1.7417e-01, -3.7691e-01, -4.4016e-01,\n",
       "           1.1354e-02,  2.5387e-01, -1.6713e-01,  3.0450e-02, -2.3705e-01,\n",
       "           3.6587e-01,  9.3493e-01, -3.4931e-01,  4.2838e-02, -4.9836e-01,\n",
       "           4.5254e-02, -8.1734e-02, -6.2819e-02, -3.3827e-01, -4.0564e-01,\n",
       "           3.5635e-01,  1.3258e-01,  2.5235e-01,  2.0424e-02, -2.3089e-01,\n",
       "           3.3129e-01, -3.3075e-01, -8.9622e-02,  7.1975e-01, -2.8549e-01,\n",
       "           4.0563e-02,  2.8078e-01, -6.6874e-01, -8.5723e-02,  3.4916e-01,\n",
       "           1.6608e-01, -6.5247e-02, -8.3725e-01, -7.6890e-03,  4.5139e-01,\n",
       "           3.8668e-01,  1.8128e-01,  2.0962e-01,  5.4342e-01,  3.6529e-01,\n",
       "          -4.0580e-01,  3.7468e-01, -1.1954e-01, -1.3160e-01,  4.5812e-02,\n",
       "           1.4795e-01,  1.4958e-01, -4.0168e-01,  3.3391e-01, -1.7011e-01,\n",
       "           2.4076e-02, -4.0941e-01, -1.8527e-01, -9.4000e-03,  3.9568e-01,\n",
       "          -2.4590e-02, -4.3180e-01,  1.3466e-01, -3.7448e-01, -2.4752e-01,\n",
       "           9.1886e-02,  7.0489e-02,  3.2154e-01, -4.9037e-01, -2.5280e-01,\n",
       "          -1.1939e-01, -3.0043e-01,  5.9537e-01,  1.4481e-02, -1.3875e-01,\n",
       "           3.3885e-01, -2.4500e-01, -1.9853e-01,  3.8153e-01,  3.6541e-02,\n",
       "          -3.2929e-01,  1.7709e-01, -2.1721e-01,  2.6912e-02,  7.2715e-01,\n",
       "          -3.6499e-01, -1.2932e-02, -2.0360e-02,  2.0395e-01,  3.4879e-01,\n",
       "          -2.8039e-01, -1.2410e-01,  1.7326e-01, -4.9986e-01, -4.2984e-02,\n",
       "          -1.2823e-01,  3.1628e-01, -2.6186e-01, -1.3695e-01, -4.3339e-01,\n",
       "           1.3805e-02,  3.3696e-02, -1.2055e-01,  1.0068e-01, -2.8598e-01,\n",
       "           6.6533e-01, -2.4381e-01, -2.6331e-01,  1.3605e-01, -1.1418e+01,\n",
       "           1.1377e-01,  8.4308e-01,  1.6280e-01,  2.3584e-01, -2.9657e-01,\n",
       "           1.0418e-01,  1.1050e-01,  7.0321e-02, -3.2792e-01,  1.4004e-01,\n",
       "          -8.6370e-02,  6.6247e-03,  3.9086e-01, -3.6419e-01,  9.2481e-02,\n",
       "          -1.1838e-01, -1.0034e+00,  3.8269e-02,  2.4191e-01,  4.5080e-01,\n",
       "          -2.9394e-01,  2.1662e-02,  3.1407e-01, -3.1812e-01, -6.7698e-02,\n",
       "          -3.1501e-01, -9.0094e-02,  2.0232e-01,  1.2938e-01, -5.3503e-02,\n",
       "           1.2833e-02, -8.4343e-02,  4.0612e-01, -2.2802e-02, -4.8456e-01,\n",
       "          -1.5634e-01,  6.4033e-02,  3.7287e-02, -5.1122e-01, -4.6959e-01,\n",
       "           1.6187e-01, -2.7521e-01,  1.5326e-01,  3.0423e-01,  5.8338e-01,\n",
       "           6.4433e-01,  8.7874e-01,  9.8589e-02,  6.4743e-03, -1.2715e-01,\n",
       "          -8.6394e-01,  1.4528e-01, -6.1263e-01,  3.3415e-01, -2.5418e-01,\n",
       "           6.9162e-01, -5.7891e-01,  3.3339e-01, -4.9050e-01, -2.3375e-01,\n",
       "           4.9885e-01,  2.1698e-01,  2.7781e-02, -5.2102e-01, -9.8122e-02,\n",
       "           3.4905e-01,  8.7095e-03,  2.4822e-01, -4.3775e-01, -6.8448e-02,\n",
       "           2.5354e-02,  7.5355e-01,  6.8922e-02,  1.5600e-01,  4.6976e-02,\n",
       "          -2.9262e-01, -1.1298e-01,  1.9105e+00, -4.5921e-02,  9.8367e-02,\n",
       "           1.5618e-01,  5.6595e-01, -3.6775e-01, -8.2060e-01, -9.5813e-03,\n",
       "          -1.5806e-01, -4.0744e-02,  6.1770e-02,  5.3651e-01, -4.0122e-01,\n",
       "          -2.8241e-01, -4.2715e-02,  2.2891e-01,  8.1120e-02,  9.9514e-01,\n",
       "          -5.0647e-01, -1.3592e-01,  6.5777e-03,  6.3368e-01, -4.4815e-01,\n",
       "           2.5587e-02, -3.4930e-01, -5.6310e-02, -1.7721e-02, -3.8494e-02,\n",
       "          -1.2066e-01, -3.6498e-02,  6.6482e-02,  3.8879e-01, -5.0006e-02,\n",
       "          -1.0725e-01, -1.0326e-01, -4.3712e-02, -3.7134e-01, -1.2494e-01,\n",
       "           1.8692e-01,  3.5026e-01, -5.6838e-01, -5.4588e-01, -3.2157e-01,\n",
       "          -3.7195e-02,  2.8179e-01,  5.1425e-02, -4.7998e-01,  5.6484e-01,\n",
       "          -6.6180e-01, -1.0760e-01,  2.6979e-01,  2.1886e-01,  4.8424e-01,\n",
       "           2.3962e-01, -2.7152e-01,  4.6382e-01, -5.7533e-01, -3.5601e-01,\n",
       "          -7.6686e-01,  7.1204e-03,  3.3090e-01, -3.4110e-01,  7.7304e-02,\n",
       "           1.5510e-01, -4.8632e-02,  1.7306e-01,  1.1810e-01,  1.6733e-01,\n",
       "          -2.1232e-02, -1.6458e-02, -5.1332e-01,  2.7794e-01, -6.9124e-02,\n",
       "          -4.3402e-01,  3.8644e-02, -3.1164e-01, -1.5215e-01, -1.1196e-01,\n",
       "          -2.3435e-02,  7.5629e-02, -4.2071e-01, -1.6949e-01,  1.5603e-01,\n",
       "           1.2218e-01, -1.6835e-01, -3.5913e-01,  4.8729e-01,  9.7509e-02,\n",
       "          -5.5797e-02, -2.0874e-01, -3.8294e-01, -1.4694e-01,  1.5783e-01,\n",
       "           6.2652e-02, -6.3195e-02, -5.2337e-01, -4.0067e-01, -2.3889e-01,\n",
       "          -5.7472e-01,  1.0858e+00, -9.2285e-02, -5.5553e-01, -7.3773e-02,\n",
       "           3.4282e-01, -2.2721e-01,  2.4674e-01,  1.7079e-01,  3.8558e-01,\n",
       "          -3.3978e-01, -3.9431e-01,  6.1823e-03, -2.5758e-02,  3.9101e-01,\n",
       "           3.4840e-01,  5.9696e-01,  4.9011e-01,  2.1357e-01, -1.1559e-01,\n",
       "          -5.7487e-01,  3.2738e-01, -3.6646e-01, -3.9739e-01,  3.8141e-01,\n",
       "          -9.0689e-02,  2.4697e-02,  2.8430e-01, -3.6602e-01,  1.8617e-01,\n",
       "           6.4427e-01,  7.4935e-01, -6.0195e-02, -3.8977e-01,  5.6978e-01,\n",
       "           2.1737e-01, -1.3383e-01,  1.7105e-01,  4.2291e-01,  1.2353e-01,\n",
       "           3.0668e-01,  1.2484e-01, -2.9104e-01, -9.1789e-02,  1.9966e-01,\n",
       "           3.2000e-01, -4.7134e-01, -5.8480e-01,  6.0997e-02,  1.4006e-01,\n",
       "          -1.2669e-01,  4.8179e-01, -5.6792e-02,  3.6236e-01,  3.2851e-01,\n",
       "          -1.1371e+00,  1.4501e-01, -1.7041e-01,  1.8232e-02, -1.7073e-01,\n",
       "          -1.1745e-01, -4.2501e-01, -2.3848e-01, -2.9930e-01,  2.4263e-01,\n",
       "           2.5979e-01, -6.7001e-01, -6.7164e-02,  3.6919e-01, -2.9916e-01,\n",
       "           4.2863e-01, -4.9112e-01,  4.9717e-01,  6.8122e-01, -3.4430e-01,\n",
       "           2.7410e-01, -8.4230e-02, -6.1617e-01, -3.5603e-01,  2.0665e-01,\n",
       "           5.3170e-01, -2.7458e-02,  1.2537e-01, -2.1460e-01, -5.2783e-01,\n",
       "          -5.7901e-02, -2.4859e-01,  1.4412e-01, -3.5078e-01,  3.9901e-02,\n",
       "          -1.7903e-01, -3.2260e-01, -2.7667e-02,  4.1629e-01,  1.4599e-01,\n",
       "           2.2399e-01,  4.8948e-01,  6.3414e-01,  2.3314e-02, -2.6792e-01,\n",
       "          -9.9237e-02,  3.6744e-01, -5.3180e-02,  9.9155e-01, -7.8306e-02,\n",
       "          -1.8306e-01, -2.1731e-02,  8.6113e-03, -1.2811e-01,  3.4870e-01,\n",
       "          -2.2924e-01,  2.5329e-02, -1.6893e-01, -6.6031e-02,  2.5295e-01,\n",
       "           1.3629e-01,  3.7671e-02, -4.3246e-01,  2.0870e-01,  4.7896e-01,\n",
       "          -2.3456e-02,  4.0253e-01, -1.3764e-01,  4.6894e-01, -4.5455e-01,\n",
       "          -2.9003e-01, -5.5458e-01,  2.4767e-01,  2.0208e-01, -8.0991e-02,\n",
       "          -5.6740e-01, -2.9935e-01,  3.7720e-01, -1.4502e-01,  6.8787e-01,\n",
       "           7.6811e-01,  3.6459e-02, -5.9048e-01, -4.3508e-02, -1.0246e-01,\n",
       "          -1.8006e-01,  4.1084e-01,  1.0845e-01, -2.2925e-01, -1.1876e-01,\n",
       "          -3.1507e-01, -9.5207e-02, -3.5142e-01,  3.5755e-01, -9.2964e-02,\n",
       "           6.9756e-01,  1.4460e-01,  7.2438e-02,  3.5404e-01,  3.9432e-01,\n",
       "          -2.9369e-01,  7.8359e-01,  2.2095e-01, -4.6333e-01,  1.8300e-01,\n",
       "          -7.5266e-02, -2.9379e-01,  6.5779e-01, -7.0699e-01,  2.5311e-01,\n",
       "           1.8474e-01,  8.1891e-01,  4.0170e-01]], device='mps:0'),\n",
       " tensor([[-1.5391e-01, -5.6324e-01, -3.7732e-01, -6.0412e-01, -4.4163e-01,\n",
       "          -2.6384e-01, -3.6018e-01,  2.8607e-01,  1.9835e-01,  5.9710e-01,\n",
       "           1.4905e-01,  6.4190e-01,  4.8391e-02, -4.2024e-01, -3.2081e-01,\n",
       "          -4.4196e-01, -5.2961e-02,  7.4481e-01,  2.9407e-01,  1.0016e-01,\n",
       "           3.9940e-01,  1.3809e-01,  1.4023e-01, -7.4686e-03, -6.1345e-01,\n",
       "          -1.2552e-01,  1.8004e-01,  4.9521e-02,  9.2648e-02, -7.4713e-01,\n",
       "          -1.9843e-01,  5.8813e-01, -2.3928e-01,  1.0376e+00,  1.0937e-01,\n",
       "          -1.6150e-02,  8.5108e-02, -3.6333e-01,  5.4706e-01, -2.2748e-01,\n",
       "           5.1286e-01, -2.0872e-01,  4.5345e-01,  5.7033e-02,  5.4419e-01,\n",
       "           9.5103e-02,  2.2398e-01, -5.1216e-01,  6.3567e-01, -6.1505e-01,\n",
       "          -2.2189e-01, -6.7061e-02,  4.3140e-01,  2.7187e-01, -5.2823e-01,\n",
       "          -2.7415e-01,  2.2295e-01,  2.2705e-01, -4.5818e-01,  1.5492e-01,\n",
       "          -6.3327e-02,  7.0979e-01, -1.0913e-01, -1.6659e-01,  9.3394e-02,\n",
       "           3.5458e-01, -6.9387e-01, -5.5965e-01, -3.6935e-01, -1.6925e-01,\n",
       "          -6.9862e-02,  4.3264e-01, -1.3625e-01, -1.3456e-01,  2.6425e-01,\n",
       "          -1.6312e-01,  5.6649e-02,  1.8570e-01, -3.8227e-01,  1.8097e-02,\n",
       "          -1.6169e-01,  3.6342e-01, -2.1195e-01, -4.1766e-02, -2.3041e-02,\n",
       "          -5.3967e-02,  6.3417e-01,  1.4901e-01,  3.7156e-02, -1.7229e-01,\n",
       "           5.1514e-01, -4.1400e-02, -3.4837e-01, -5.9674e-01,  3.2867e-01,\n",
       "          -6.2711e-02,  5.5346e-01, -4.1668e-01,  2.1734e-02, -1.8990e-01,\n",
       "           2.7536e-01,  1.2358e-01, -2.2695e-01,  2.4531e-01, -2.7322e-01,\n",
       "          -9.3442e-01, -5.5481e-03,  4.2421e-03, -2.4500e-01, -6.5065e-02,\n",
       "          -2.0895e-01, -2.6743e-01,  1.2498e-01, -4.0717e-01,  1.1852e-03,\n",
       "          -1.4863e-01,  1.8804e-01, -1.6106e-01, -5.0271e-01,  4.2192e-02,\n",
       "           4.7428e-02, -4.0510e-01,  2.9450e-01,  3.4466e-01,  3.4124e-01,\n",
       "          -3.6916e-01,  5.8528e-01, -5.1579e-02,  1.1478e-01,  4.2025e-01,\n",
       "           7.4543e-02,  3.8799e-02, -2.4115e-01,  2.4223e-01,  6.5928e-02,\n",
       "           3.7181e-02,  1.2650e-01, -2.1010e-01,  2.8560e-01, -1.6686e-01,\n",
       "          -6.9010e-02, -6.9334e-02, -2.8776e-01, -3.3115e-01,  1.9497e-02,\n",
       "          -1.6330e-01,  3.8429e-01, -4.1416e-01, -1.7856e-01, -4.4173e-01,\n",
       "          -5.7870e-01, -5.6955e-01,  4.6120e-01, -2.7256e-02,  1.0172e-01,\n",
       "          -2.0177e-01,  3.2639e-01,  2.0782e-01, -2.3877e-01,  4.5876e-01,\n",
       "           6.1361e-01,  2.1925e-02,  2.6561e-01, -3.3476e-01, -3.9135e-01,\n",
       "           1.4940e-01, -8.6169e-02, -2.1513e-01, -3.7588e-01, -2.9669e-02,\n",
       "           1.7530e-01, -3.6165e-01,  3.3249e-01, -2.1830e-01,  1.0378e-02,\n",
       "          -3.3128e-01,  1.2047e-01,  2.3010e-01, -2.9709e-01, -2.5766e-01,\n",
       "          -1.0680e-01, -1.1763e+00, -1.5182e-02,  5.0723e-01, -1.6429e-01,\n",
       "          -6.7919e-01,  3.1455e-01,  2.3866e-01,  2.7237e-01, -1.0019e-01,\n",
       "           6.1037e-02,  2.1789e-01,  2.2093e-01,  2.1068e-02,  2.1145e-02,\n",
       "           3.2689e-01,  5.2796e-01,  1.9733e-01,  3.9274e-01,  2.2475e-02,\n",
       "          -3.4407e-01,  1.0000e-01, -1.9501e-02,  2.3629e-01,  2.1975e-01,\n",
       "          -2.3916e-02, -1.7868e-01,  1.7863e-01,  1.0210e-03, -1.7212e-01,\n",
       "          -2.8003e-01, -8.3249e-02, -5.1343e-01,  1.2969e-01, -7.0501e-02,\n",
       "           9.0112e-01, -3.3593e-02, -1.8184e-01,  4.9969e-01,  9.4920e-03,\n",
       "          -2.0454e-01,  8.3272e-02, -6.6332e-02, -2.8288e-01, -1.9053e-01,\n",
       "           7.6979e-01, -3.1248e-01,  8.7851e-02, -9.0241e-02, -2.4998e-01,\n",
       "           2.7705e-01, -4.8339e-02,  5.8265e-01,  4.6811e-01, -5.2564e-01,\n",
       "          -3.5134e-01, -2.8350e-01, -1.7661e-01, -1.8048e-01, -5.0656e-02,\n",
       "          -5.6327e-01, -4.7948e-01, -4.6803e-01, -1.4629e-01,  2.7432e-01,\n",
       "           3.1320e-01, -1.9810e-01,  7.0954e-01,  2.2847e-01,  3.0329e-01,\n",
       "          -1.0342e-01, -1.3093e-01,  3.5781e-01, -6.1361e-02, -4.2183e-01,\n",
       "           6.9988e-02,  1.9677e-01, -5.3027e-01,  6.0555e-01, -1.1143e-01,\n",
       "          -1.1016e-01,  2.2909e-03,  1.8390e-01,  4.7889e-01,  1.0053e-02,\n",
       "          -1.1304e-01, -1.8011e-01, -5.3106e-01, -3.6217e-01,  3.6780e-01,\n",
       "          -9.3683e-02,  4.7459e-01, -3.9765e-02, -4.6579e-01, -5.6814e-01,\n",
       "          -2.4659e-01, -2.1708e-01, -4.0061e-01, -2.7283e-02, -5.8714e-01,\n",
       "          -4.6273e-01,  7.0714e-02, -2.1733e-01,  1.5389e-01, -4.8478e-01,\n",
       "           1.8112e-01,  1.2356e-01,  2.0038e-01,  3.2160e-02, -3.9009e-01,\n",
       "          -2.5854e-01,  3.3718e-02, -1.6326e-01, -1.3589e-01, -9.3844e-02,\n",
       "           1.8264e-01,  1.1545e-01, -9.1362e-03,  2.6080e-02, -4.6757e-01,\n",
       "           2.9600e-01,  1.0013e-01,  2.6514e-01, -1.1710e-01, -1.0805e-01,\n",
       "           1.3031e-01,  3.2594e-01,  3.5181e-01, -2.2289e-01, -5.4491e-01,\n",
       "           2.9477e-03,  3.4494e-01, -8.6842e-02, -4.1126e-01, -1.5513e-01,\n",
       "          -8.4829e-01, -1.0413e-01, -5.8860e-03, -1.8332e-01, -3.2296e-01,\n",
       "          -8.7896e-03,  4.7107e-01, -2.9065e-01,  1.4484e-01, -3.4641e-01,\n",
       "           5.2519e-01, -1.8389e-02,  3.0625e-01,  2.4956e-01, -6.6865e-01,\n",
       "           8.0514e-03, -8.1730e-01, -1.4301e-01, -1.7049e-01, -1.8533e-01,\n",
       "           5.4231e-02,  1.8755e-01,  1.1460e-01,  2.0344e-01, -2.4756e-01,\n",
       "           2.9007e-01, -3.6309e-01, -1.9001e-01,  3.1798e-01, -3.4161e-01,\n",
       "           2.0337e-01,  3.4247e-01, -6.0229e-01, -1.0267e-01,  1.4807e-01,\n",
       "           2.6383e-01, -1.5813e-01, -1.7614e-01, -5.2600e-01,  4.8676e-01,\n",
       "           5.8099e-01,  1.4488e-01, -2.9646e-01,  5.2130e-01,  1.8748e-01,\n",
       "          -2.6477e-01,  2.2608e-01, -4.4081e-01, -3.9997e-01, -2.0275e-01,\n",
       "           2.8552e-01,  2.4838e-01, -5.3133e-01, -1.3205e-01, -1.0403e-01,\n",
       "           1.1636e-01, -3.4566e-01, -4.2330e-02,  2.2323e-01,  8.1047e-03,\n",
       "          -1.5004e-01, -4.3803e-01, -3.4816e-02, -1.8150e-01, -2.2156e-01,\n",
       "           4.4328e-02,  4.5324e-02,  2.1674e-01, -4.4854e-01,  3.4056e-02,\n",
       "          -1.3361e-01, -2.1726e-01,  5.8301e-01, -6.6350e-02, -1.0069e-01,\n",
       "           5.7923e-02, -4.6238e-02, -3.0264e-01,  9.4812e-01,  3.7397e-01,\n",
       "          -2.0230e-01,  5.4472e-01, -9.4555e-02, -1.2716e-01,  4.3323e-01,\n",
       "          -3.2200e-01,  2.5926e-01, -6.2457e-02,  3.5332e-01,  6.4934e-02,\n",
       "          -4.4259e-01, -2.3431e-02,  1.4701e-01,  2.5981e-02,  2.7051e-02,\n",
       "          -1.4677e-01,  8.6863e-01, -8.0766e-02,  1.8544e-01, -9.1987e-02,\n",
       "           9.7363e-02,  5.2155e-02,  1.0602e-01, -2.0415e-01, -3.9360e-01,\n",
       "           6.2516e-01,  3.9107e-01,  6.0653e-01,  5.3680e-01, -1.1511e+01,\n",
       "           7.7992e-02,  7.1489e-01, -6.1083e-01,  1.6402e-01, -3.3659e-02,\n",
       "          -3.1341e-02, -3.1982e-01,  2.0966e-01, -4.7392e-01, -2.6289e-01,\n",
       "           6.7152e-02, -5.6274e-02,  2.0710e-01,  1.2620e-02, -1.6994e-02,\n",
       "          -1.0005e-02, -6.1639e-01,  1.2983e-01, -5.5914e-01,  4.3714e-01,\n",
       "          -3.4429e-01, -1.2961e-01,  3.5155e-01,  1.2487e-01, -1.4066e-02,\n",
       "          -3.7046e-01,  1.9062e-01,  5.9339e-01,  4.0100e-01, -2.3506e-01,\n",
       "           9.6098e-03,  1.0110e-01,  3.9414e-01, -2.0544e-01, -6.3426e-01,\n",
       "           2.9874e-01, -3.7352e-01,  2.9096e-01, -3.6038e-01, -2.6612e-01,\n",
       "          -4.5458e-02, -3.2225e-01,  3.1443e-01,  5.3914e-01,  5.1724e-01,\n",
       "           4.5395e-02,  8.5499e-01,  4.1707e-01, -4.4475e-01, -1.1364e-02,\n",
       "          -9.0177e-01,  1.2203e-01, -8.0428e-01, -7.0576e-01,  3.4049e-02,\n",
       "           5.4635e-01, -3.9119e-01,  2.5081e-01, -4.9838e-01, -2.2831e-01,\n",
       "           2.7231e-01, -1.5651e-03, -8.7145e-02, -6.1163e-01, -1.1570e-02,\n",
       "           5.8485e-02, -7.7605e-02,  4.2672e-02, -1.3318e-01, -1.7427e-01,\n",
       "           2.4972e-02,  3.8274e-01,  3.9832e-01,  2.1251e-01,  2.4021e-01,\n",
       "          -1.1066e-01, -6.3627e-02,  1.9424e+00, -1.5230e-01,  1.1252e-01,\n",
       "          -1.5777e-02,  5.2517e-01, -5.5002e-02, -7.9440e-01, -1.4652e-01,\n",
       "          -3.2842e-02,  3.2529e-01,  6.8884e-02,  3.3871e-01, -1.8516e-01,\n",
       "          -6.5418e-01,  3.1763e-02,  3.3009e-01,  1.1205e-01,  9.6985e-01,\n",
       "          -7.4450e-01,  1.2688e-01,  3.7514e-01, -6.2591e-02, -1.7076e-02,\n",
       "           1.6462e-01, -2.9872e-01, -4.6021e-01,  1.0408e-01,  3.6402e-01,\n",
       "           3.1173e-01,  1.1986e-01,  3.2882e-01,  2.7232e-01, -7.1031e-02,\n",
       "           3.2573e-01,  1.8548e-01,  1.6465e-01, -1.0608e-01, -2.9829e-01,\n",
       "           2.1536e-01,  6.0324e-01, -2.6339e-01, -5.3930e-01, -2.1568e-01,\n",
       "          -6.4477e-02,  1.4832e-01,  4.6066e-01, -4.9563e-01,  6.4186e-01,\n",
       "          -5.0235e-01, -2.8503e-01, -1.8063e-01, -2.4233e-01,  4.4652e-01,\n",
       "           4.0112e-01, -1.7974e-01, -1.6846e-01, -6.0675e-01, -3.6937e-01,\n",
       "          -7.1817e-01, -9.9959e-02,  2.0380e-01,  1.5250e-02,  1.0703e-01,\n",
       "          -4.6720e-01, -2.8816e-02, -2.6403e-01,  4.0321e-01,  5.4066e-02,\n",
       "           4.8176e-01,  8.9004e-02,  6.6121e-03,  4.3130e-01, -9.8989e-02,\n",
       "          -6.5410e-01, -1.2891e-01, -3.3207e-01,  1.9129e-01, -5.1980e-01,\n",
       "          -1.4663e-01, -2.1352e-01,  1.3051e-01, -1.4023e-01,  3.1934e-01,\n",
       "          -1.9339e-01,  3.0070e-01, -9.2249e-02,  2.8282e-01, -1.8250e-02,\n",
       "           4.2421e-02, -9.7102e-02, -1.3769e-01, -1.6050e-01,  4.0808e-01,\n",
       "          -9.0000e-02, -1.0846e-01, -2.1197e-01, -2.7076e-01, -5.3243e-02,\n",
       "           2.7002e-01,  6.9710e-01, -2.3497e-01, -5.8148e-01,  3.2781e-01,\n",
       "           6.0521e-01, -2.5688e-01, -2.7098e-01,  1.4384e-01, -3.0880e-01,\n",
       "          -3.2692e-01, -2.4541e-01,  3.4318e-01, -5.8137e-02,  3.9945e-01,\n",
       "           1.9259e-01, -1.6440e-01, -3.4108e-01,  6.8770e-01,  2.5426e-01,\n",
       "          -2.5456e-01, -2.8649e-01, -4.9171e-01, -1.3213e-01,  5.1546e-01,\n",
       "           1.0131e-01, -3.2435e-01, -8.0523e-02, -1.3402e-01, -3.5568e-01,\n",
       "           5.3077e-01,  5.4930e-01, -1.1278e-01, -1.1791e-01,  9.6732e-01,\n",
       "          -1.5271e-02,  1.2624e-01,  3.2910e-01,  5.3795e-01,  1.7068e-01,\n",
       "           4.5975e-01,  2.5678e-02, -5.3631e-01, -2.0868e-01,  3.8032e-02,\n",
       "           2.5219e-01, -6.3267e-01, -3.9436e-01, -5.8214e-02,  9.8655e-02,\n",
       "           9.0194e-02,  5.2273e-01, -3.1369e-01,  2.4135e-01,  3.9440e-01,\n",
       "          -1.0650e+00, -1.2094e-01, -1.7665e-01, -9.9200e-02,  2.9972e-02,\n",
       "          -2.0712e-01, -9.2434e-02, -6.9741e-01, -1.7784e-01,  3.2692e-01,\n",
       "           1.7023e-01, -4.6515e-01,  1.4948e-02,  7.0991e-01, -3.3077e-01,\n",
       "           1.6590e-01, -1.3220e-01,  6.4256e-02,  2.9215e-01,  9.0042e-02,\n",
       "          -5.7318e-02,  3.1150e-01,  9.2682e-02, -2.5196e-01,  2.3690e-01,\n",
       "           5.6464e-01, -6.6798e-02,  7.2193e-02,  6.1595e-02, -4.9488e-02,\n",
       "          -1.1402e-01, -1.1122e-01, -5.0361e-02, -1.1741e-01, -2.5761e-01,\n",
       "           2.6156e-01, -3.7796e-03, -8.7694e-02,  2.5833e-01, -3.9733e-01,\n",
       "           3.3200e-01,  6.6135e-01,  9.4870e-01,  9.7821e-02, -5.8520e-01,\n",
       "           2.9650e-01,  1.9886e-01, -1.6116e-01,  4.7240e-01, -3.4828e-02,\n",
       "          -3.7288e-01, -1.0324e-02,  1.6503e-01, -3.3663e-01, -3.0271e-02,\n",
       "          -7.2232e-03,  8.2881e-02, -3.2100e-01, -3.0955e-01,  3.8207e-01,\n",
       "           1.4494e-01,  1.0612e-01, -6.8629e-01, -2.1621e-02,  6.5568e-01,\n",
       "          -6.3433e-03,  3.8208e-01, -9.7178e-02,  1.2900e-01, -1.2381e-01,\n",
       "          -6.6524e-02, -8.1817e-01, -3.7755e-01,  2.6580e-01, -3.1467e-01,\n",
       "          -6.7525e-01,  4.6074e-02,  1.9892e-01,  2.7829e-02,  6.2348e-01,\n",
       "           5.1798e-01, -1.3201e-01, -9.4134e-01, -2.0990e-01,  2.2073e-01,\n",
       "          -2.9604e-02, -7.8278e-02,  3.5996e-01, -2.5376e-02,  1.9156e-01,\n",
       "          -8.5942e-01, -3.5147e-01, -1.0267e-01,  6.5321e-02, -3.2071e-01,\n",
       "           6.9500e-01, -1.4305e-01,  9.3903e-02,  2.7582e-01,  2.6654e-01,\n",
       "          -8.0814e-02,  6.2217e-01, -2.6992e-01, -1.6321e-01, -2.1457e-01,\n",
       "           4.0422e-01, -7.7656e-02,  7.7886e-02, -8.0758e-01,  4.8865e-01,\n",
       "          -1.9104e-01,  5.6849e-01,  3.5685e-01]], device='mps:0'),\n",
       " tensor([[-1.6720e-01, -5.2486e-01, -4.5395e-01, -6.3577e-01, -4.3561e-01,\n",
       "          -1.5533e-01, -5.8268e-01,  2.5578e-01, -1.8350e-01,  5.0250e-01,\n",
       "           1.6126e-01,  1.3783e-01, -2.1897e-01, -4.6204e-01, -6.9128e-01,\n",
       "          -2.0657e-01, -2.5177e-01,  1.9475e-01,  4.1034e-01,  5.6174e-01,\n",
       "           2.8700e-01,  1.6722e-01, -3.7843e-01,  2.1455e-01, -5.1055e-01,\n",
       "           1.5812e-01, -2.8156e-01, -6.3669e-02,  4.5597e-01,  6.8516e-02,\n",
       "          -2.1018e-02,  4.7623e-01, -2.8390e-02,  1.0709e+00, -7.7172e-02,\n",
       "          -3.0626e-01,  1.0760e-01, -4.1569e-01,  8.9012e-01, -5.6131e-01,\n",
       "           6.8910e-01,  6.2264e-01, -1.5577e-01,  1.5282e-02,  2.3088e-01,\n",
       "           1.5974e-01,  5.1535e-02, -5.3995e-01, -5.6558e-01, -1.5427e-01,\n",
       "           1.6123e-01,  1.9663e-01,  5.4635e-01,  6.3307e-02, -7.5459e-01,\n",
       "          -2.6525e-01, -1.3650e-01,  3.0473e-01,  1.7296e-02,  3.5135e-01,\n",
       "          -7.4312e-02,  5.0134e-01,  5.1710e-02, -6.9956e-02,  1.0589e-01,\n",
       "           2.1145e-01, -6.0512e-01, -2.5610e-01, -5.7211e-01,  2.0313e-01,\n",
       "           2.0375e-01,  5.2893e-01,  1.1646e-01, -4.0534e-01,  1.4546e-01,\n",
       "           7.6205e-02,  9.5666e-02, -2.8535e-02, -5.2510e-01, -1.6272e-01,\n",
       "           2.0181e-01,  3.7399e-01,  2.1510e-01, -2.0973e-01,  1.4960e-01,\n",
       "           2.5140e-01,  5.5226e-01, -5.2582e-02, -1.1641e-01, -4.8629e-02,\n",
       "           2.9244e-01, -2.7514e-01, -7.8764e-01, -3.1798e-01, -1.0430e-01,\n",
       "          -3.4914e-01,  7.9176e-02, -4.2834e-01, -2.7512e-01, -3.3882e-01,\n",
       "           1.5050e-01,  1.0200e-01,  2.5842e-01,  7.5482e-02, -7.5174e-01,\n",
       "          -4.1705e-01,  4.6029e-01,  8.0138e-02, -4.2192e-01, -3.1245e-01,\n",
       "          -1.3741e-01, -2.1173e-01,  9.7620e-02, -4.0036e-01, -8.6561e-02,\n",
       "          -1.3006e-01,  2.5022e-01, -1.4378e-01,  5.4056e-02,  4.9777e-01,\n",
       "           2.5708e-01, -5.8312e-01,  1.0353e-01,  2.2601e-01,  4.0089e-02,\n",
       "           3.0807e-02,  6.0830e-01, -5.3486e-02,  4.9134e-01,  1.2285e-01,\n",
       "           2.6257e-01,  1.5138e-01, -6.2596e-01, -5.3560e-01, -2.0734e-01,\n",
       "           1.3350e-01, -3.3082e-01, -8.4107e-01,  3.7865e-01,  3.1799e-02,\n",
       "           5.1144e-02, -9.6587e-02, -6.7862e-01, -3.0496e-01, -3.9669e-01,\n",
       "          -2.7230e-01,  1.4708e-01, -3.5011e-01, -1.3439e-01, -2.7277e-01,\n",
       "          -6.7571e-01, -8.5650e-01, -1.1642e-03, -1.3369e-01,  4.2598e-01,\n",
       "          -5.3225e-01,  4.3086e-01, -2.3121e-01, -5.0388e-01,  4.1057e-01,\n",
       "           5.9947e-01, -1.7783e-01,  2.8322e-01, -5.0157e-01,  3.3619e-02,\n",
       "           3.4604e-01, -3.8918e-02, -5.9573e-02, -3.8183e-02,  1.3359e-01,\n",
       "           3.4058e-01,  2.0341e-01,  1.0286e-01, -2.3079e-01,  3.2100e-01,\n",
       "          -3.7378e-01, -5.5154e-03, -2.6355e-01,  1.6162e-01,  3.2661e-02,\n",
       "           2.9978e-01, -1.6971e+00, -2.5911e-01,  2.0065e-02, -1.8571e-01,\n",
       "          -1.9740e-01,  3.3240e-01,  1.4453e-01,  2.5148e-01, -2.5533e-01,\n",
       "           2.1988e-01,  9.8764e-02, -1.4769e-01, -3.0128e-01,  5.8231e-02,\n",
       "           3.8842e-01,  6.9484e-01,  3.7541e-01,  7.7099e-02, -3.0723e-01,\n",
       "          -7.5503e-01,  1.8407e-01,  3.3852e-01, -1.1523e-01,  5.9383e-01,\n",
       "           3.0489e-02, -8.6229e-01,  1.9019e-01,  2.9946e-02,  2.6756e-01,\n",
       "           5.7319e-02, -2.9941e-01, -4.1668e-01, -3.5307e-01, -3.9315e-01,\n",
       "           2.3395e-01, -1.8355e-01, -3.1845e-01,  4.8844e-01,  8.1310e-02,\n",
       "          -3.0776e-02,  4.2831e-01,  6.0508e-02, -4.4367e-01,  1.9303e-01,\n",
       "           4.4887e-01,  2.6861e-02,  1.1130e-01, -2.7214e-01, -4.0374e-02,\n",
       "          -6.2142e-02, -3.2118e-01,  3.6896e-01,  7.7114e-01, -7.4506e-02,\n",
       "           3.7060e-01, -3.0406e-01,  5.1180e-01, -1.9654e-01,  5.8690e-02,\n",
       "          -1.1981e-01,  3.4078e-02, -2.6240e-01,  2.4533e-01, -1.1898e-01,\n",
       "           9.4908e-02,  8.5883e-02,  7.2222e-01,  3.7433e-01,  7.9341e-02,\n",
       "           3.1509e-01, -4.4737e-03,  1.1637e-02,  5.8059e-01, -3.3724e-01,\n",
       "           3.9433e-01, -1.0171e-01, -3.7982e-01,  3.7113e-01, -4.7675e-01,\n",
       "           3.1991e-01, -1.8581e-01,  1.0010e-01,  2.5551e-01,  3.0602e-02,\n",
       "          -2.9218e-01,  6.1325e-01,  2.2058e-01, -4.4818e-01,  5.2200e-01,\n",
       "          -2.5795e-01,  7.1632e-01, -2.0896e-01, -1.6155e-01, -7.6790e-02,\n",
       "          -9.9737e-01, -4.3677e-01,  1.9994e-01,  1.8271e-01, -6.8430e-01,\n",
       "          -4.9875e-01,  2.8702e-01,  1.1765e-01,  1.0125e-01,  4.2891e-01,\n",
       "           3.8920e-01,  9.2143e-02,  1.0493e-01, -4.8046e-01, -3.6847e-01,\n",
       "          -2.1420e-01,  9.0059e-02, -8.5629e-04, -7.5374e-02,  2.0220e-01,\n",
       "           5.9414e-01,  9.6554e-02,  1.8048e-01,  3.6746e-02, -2.8779e-01,\n",
       "          -1.0518e-01,  1.6372e-01,  4.2762e-01, -5.4888e-01, -2.0137e-01,\n",
       "           6.8173e-02,  4.3129e-02,  4.5140e-01, -3.3691e-01, -3.2667e-01,\n",
       "          -4.2742e-02,  4.8308e-01, -1.2353e-01, -7.1202e-02,  3.3410e-02,\n",
       "          -7.8725e-01,  2.7660e-01, -4.5062e-02, -2.6552e-02, -3.0430e-01,\n",
       "           2.3290e-01,  3.4297e-02, -2.8492e-01, -1.1149e-01, -2.7156e-01,\n",
       "           5.8752e-01,  5.5235e-01,  3.3698e-01, -5.6671e-01, -3.5461e-01,\n",
       "           4.4066e-02, -7.0730e-01,  9.8556e-02, -4.9156e-01,  2.2488e-01,\n",
       "           1.4708e-01,  3.3198e-01, -1.1607e-01,  1.3552e-01, -1.6474e-01,\n",
       "           4.9723e-01, -6.3908e-01,  1.2761e-01,  5.5720e-01, -5.3885e-01,\n",
       "          -2.3301e-01,  4.1839e-01, -5.5797e-01,  1.2634e-01,  4.1453e-01,\n",
       "           3.9631e-01,  4.6106e-02, -4.1602e-01, -7.3707e-02,  9.1249e-01,\n",
       "           5.2639e-01,  1.6638e-01, -6.9733e-01,  9.7526e-02,  3.4043e-01,\n",
       "          -2.0714e-01,  9.5206e-02, -3.2600e-01, -4.7544e-01, -2.1026e-01,\n",
       "          -8.4302e-02,  3.9153e-01, -2.6238e-01,  3.5886e-02,  2.8572e-01,\n",
       "           3.6819e-01, -4.7808e-01, -2.3588e-01,  3.2227e-01,  6.0455e-01,\n",
       "          -1.8222e-01, -3.6901e-01,  1.6512e-01, -5.0214e-01,  2.6335e-01,\n",
       "          -1.4632e-01, -3.4035e-01,  2.7259e-01, -1.2606e-01, -3.7457e-01,\n",
       "          -1.8250e-01, -4.7967e-01,  2.5449e-01,  9.2148e-02, -1.2652e-01,\n",
       "           2.8609e-01, -1.1966e-01, -3.8868e-01,  8.2449e-01, -5.2187e-01,\n",
       "           4.2853e-03,  5.3276e-01, -2.7521e-01, -1.8249e-01,  2.4158e-01,\n",
       "          -2.6392e-01, -1.9683e-02,  1.8527e-02,  9.4766e-01,  1.2345e-01,\n",
       "          -2.7053e-03, -1.1552e-01,  5.1146e-01, -3.8026e-01,  1.5931e-01,\n",
       "          -3.3320e-01,  3.1171e-01, -1.6158e-01,  1.8223e-02, -1.0254e+00,\n",
       "           4.4946e-01, -3.7111e-01, -2.1751e-01, -5.0362e-01, -3.0765e-01,\n",
       "           9.4840e-02, -6.3165e-02,  2.3746e-01,  6.3366e-01, -1.1037e+01,\n",
       "           8.9129e-02,  6.0403e-01, -7.6489e-02,  1.1652e-01,  4.1141e-01,\n",
       "           1.7540e-01, -4.2006e-01,  5.2776e-01, -3.4203e-01,  1.1525e-01,\n",
       "          -4.7982e-01, -1.3280e-01, -8.5294e-02, -3.7948e-01,  3.6349e-01,\n",
       "          -9.6031e-02, -6.4688e-01,  2.4074e-01, -1.3490e-01, -1.7654e-02,\n",
       "          -2.2095e-01,  1.1407e-01,  4.3684e-01, -4.6076e-01, -4.4456e-01,\n",
       "           7.1777e-02, -2.7274e-01,  5.1195e-01,  2.8186e-01, -1.1255e-01,\n",
       "           1.1615e-01,  1.5217e-01,  3.0053e-01, -2.7216e-01, -3.1454e-01,\n",
       "           1.2732e-01, -3.4997e-01, -1.8961e-01, -1.0421e+00, -1.5631e-01,\n",
       "          -2.0656e-01, -4.6750e-01, -1.7710e-02,  6.5233e-01,  7.8743e-01,\n",
       "           1.6358e+00,  9.2174e-01,  3.2668e-01,  1.9802e-01, -2.2250e-01,\n",
       "          -4.6619e-01,  1.1894e-01, -6.7995e-02, -7.0546e-01, -2.4133e-01,\n",
       "           2.8985e-01, -7.5931e-01,  5.3131e-01, -4.7860e-01, -1.3681e-01,\n",
       "           4.2374e-01,  3.9276e-02, -4.2065e-02, -5.9925e-01, -5.3467e-01,\n",
       "          -4.9082e-02,  1.2925e-01,  4.8245e-02, -5.7979e-01,  1.4884e-01,\n",
       "           5.0590e-02,  7.6896e-01, -3.4175e-01, -5.9383e-01,  5.7234e-02,\n",
       "          -1.2763e-01,  3.6795e-02,  1.6636e+00, -1.7915e-01, -4.1931e-02,\n",
       "           1.2354e-01,  4.9556e-02, -6.9680e-01, -4.7087e-01, -4.3059e-01,\n",
       "          -2.7038e-01,  2.5252e-01,  6.6455e-01,  2.0274e-01,  9.0694e-02,\n",
       "          -4.6503e-01, -1.2589e-01,  9.3002e-02,  3.9245e-01,  1.0908e+00,\n",
       "          -5.4949e-01,  2.6147e-01, -7.5939e-02,  2.6322e-01,  1.2335e-01,\n",
       "          -3.6152e-01, -6.2216e-02,  3.9533e-02,  1.6284e-02,  4.5349e-01,\n",
       "           8.6971e-02,  4.2124e-01,  6.5534e-02,  1.2702e-01, -5.0361e-01,\n",
       "          -3.5058e-01, -2.8066e-01, -1.1609e-01, -4.6047e-01, -4.4892e-02,\n",
       "          -2.8268e-02,  1.6753e-01, -1.7086e-01, -2.9519e-01, -2.9979e-01,\n",
       "           2.9427e-01,  2.0283e-01, -6.9070e-02,  9.5939e-03,  6.0460e-01,\n",
       "          -6.3640e-01, -1.8307e-01, -9.4406e-02,  8.6813e-02,  6.9132e-02,\n",
       "           2.1208e-01, -3.8688e-01, -1.1241e-01, -7.3959e-02, -8.4460e-01,\n",
       "          -9.6512e-01, -9.7748e-02,  5.1010e-01, -2.4332e-01,  1.0866e-01,\n",
       "           6.7349e-02,  1.2474e-02, -3.2385e-01,  2.6710e-01, -1.9218e-01,\n",
       "           1.1645e-01,  3.9439e-01, -5.4670e-01,  1.5963e-01, -3.1155e-01,\n",
       "           1.8497e-02, -2.0876e-02, -6.5711e-02, -3.8026e-01, -5.2860e-01,\n",
       "          -7.6356e-01, -6.8179e-02,  5.1232e-02,  5.6135e-01, -3.2886e-02,\n",
       "           1.0488e-01, -1.0398e-01,  7.7290e-02,  7.2752e-01,  8.5206e-01,\n",
       "           2.6106e-01, -3.5137e-01, -4.9530e-01,  5.0728e-02, -2.6647e-02,\n",
       "           5.7946e-02,  2.1021e-01,  7.9023e-02, -2.2641e-01, -1.4403e-01,\n",
       "          -2.6117e-01,  1.1740e+00, -2.3018e-02, -2.5236e-01,  3.1798e-04,\n",
       "           1.2502e+00, -5.8815e-01, -1.4557e-01,  1.7334e-01,  2.6380e-01,\n",
       "          -5.2810e-01, -3.4023e-01,  5.6863e-01, -2.1703e-01,  1.5201e-01,\n",
       "           3.9180e-01,  1.8677e-01,  1.0221e-01, -4.4054e-02,  8.4521e-02,\n",
       "          -1.5903e-01,  3.3655e-01,  3.0166e-02, -1.4937e-01,  7.3811e-01,\n",
       "           1.1911e-01,  8.0888e-02, -1.4277e-01, -2.4375e-01, -2.6556e-01,\n",
       "           1.9428e-01,  1.2395e+00, -4.2394e-01, -5.8109e-01,  8.7338e-01,\n",
       "          -1.4019e-01,  3.5768e-01,  6.0006e-01, -3.9499e-02,  1.3149e-02,\n",
       "           5.7371e-01,  1.1784e-01, -6.0531e-01,  2.3557e-01,  3.0019e-01,\n",
       "           1.5461e-01, -8.7253e-01, -2.0869e-01, -8.1431e-02, -1.2343e-01,\n",
       "          -8.4444e-02,  5.6766e-01,  6.7880e-02,  3.7625e-01,  3.3336e-01,\n",
       "          -8.6361e-01,  2.9748e-01,  3.9963e-02,  8.7689e-01,  3.6637e-02,\n",
       "          -3.2763e-01, -2.5701e-01, -2.0231e-01, -2.1073e-01,  3.1221e-01,\n",
       "           2.4523e-01, -1.3614e+00, -8.3672e-02,  4.5844e-01, -8.1731e-02,\n",
       "           4.3829e-01, -7.9391e-02,  2.9827e-01,  3.0846e-01,  4.5552e-02,\n",
       "          -5.2861e-02,  7.4971e-02, -2.2854e-01, -2.7639e-01,  4.5622e-01,\n",
       "           5.5347e-01, -5.8603e-02, -7.4470e-02,  2.2723e-01, -2.7374e-01,\n",
       "          -3.2137e-01, -2.2843e-01, -1.8679e-01, -2.2410e-01, -1.2894e-01,\n",
       "          -3.6518e-01, -6.9317e-01, -3.1199e-02, -5.0747e-02, -4.5982e-02,\n",
       "           3.4356e-01,  2.6472e-01,  6.5181e-01, -1.2203e-01,  1.5073e-01,\n",
       "          -1.0406e-01,  4.4593e-01, -2.4264e-01,  7.3864e-01,  3.8785e-01,\n",
       "          -1.7432e-01,  2.7267e-01,  1.0448e-01, -2.3479e-01,  1.3287e-01,\n",
       "          -2.8065e-01,  2.5046e-01, -4.0412e-01, -4.0210e-01, -5.9590e-02,\n",
       "          -1.5547e-01, -1.1097e-01, -2.4685e-01,  2.4733e-01,  3.3605e-01,\n",
       "          -8.1116e-02,  9.9147e-01, -2.5425e-01,  6.5276e-01, -3.0278e-01,\n",
       "          -9.8080e-02, -6.8560e-01,  3.0100e-01,  1.5904e-01, -8.1888e-02,\n",
       "          -3.9633e-01, -2.5967e-01, -3.2959e-01, -4.0276e-01,  6.6570e-01,\n",
       "           7.1379e-01, -5.7936e-01, -7.3617e-01, -2.6704e-01, -1.7923e-01,\n",
       "          -2.0926e-02,  5.7887e-02,  3.7235e-02, -2.4464e-01,  1.1875e-01,\n",
       "          -8.1963e-01,  1.9554e-01,  2.5091e-02,  3.7079e-01, -2.2153e-01,\n",
       "           7.4679e-01, -2.9177e-02, -4.3907e-02,  9.2664e-01,  2.8644e-01,\n",
       "           4.0754e-02,  8.6899e-01, -1.2868e-01, -3.3744e-01,  1.2946e-01,\n",
       "           8.9648e-02, -3.6052e-01, -2.4705e-02, -5.5916e-01,  3.7844e-01,\n",
       "           3.7713e-01,  2.0163e-01,  5.3400e-01]], device='mps:0'),\n",
       " tensor([[-4.1342e-01, -6.6067e-01, -6.8362e-01, -8.6794e-01, -3.4793e-01,\n",
       "          -3.5434e-01, -9.0358e-01,  4.8098e-01,  3.3210e-01,  4.0348e-01,\n",
       "           5.2261e-02,  3.9987e-01,  2.3238e-01, -1.2007e-01, -4.3507e-01,\n",
       "          -3.9983e-01, -1.3336e-01,  4.4034e-01,  3.2879e-01,  2.2539e-01,\n",
       "           5.8120e-01,  1.4432e-01,  2.2306e-01,  9.3005e-02, -1.2598e-01,\n",
       "           2.1011e-01,  1.6098e-01, -4.5356e-01,  8.7764e-02, -4.6022e-01,\n",
       "           6.4635e-02,  3.3013e-01,  2.2419e-02,  1.1946e+00, -1.1761e-01,\n",
       "          -5.7042e-02,  5.4841e-01,  2.7781e-01,  7.1859e-01, -8.6114e-02,\n",
       "           6.8691e-02, -2.4604e-01,  3.6917e-01, -2.3515e-01,  3.0019e-01,\n",
       "           2.6462e-01,  2.7352e-02, -4.9828e-01,  7.5901e-01, -6.9489e-01,\n",
       "          -4.4400e-01, -4.0496e-02,  5.2511e-01,  5.4794e-02, -9.8927e-01,\n",
       "          -7.8140e-01,  2.3497e-02, -1.3572e-01, -4.8046e-01,  1.4552e-01,\n",
       "          -1.2289e-02,  7.1322e-01,  1.9421e-01, -1.6794e-01,  4.5244e-01,\n",
       "           1.9122e-01, -2.0461e-02, -7.6786e-01, -1.7321e-02, -1.6271e-01,\n",
       "           8.1832e-03,  1.6878e-01, -2.5303e-01, -1.3890e-01,  1.9260e-01,\n",
       "          -3.9388e-01,  3.2672e-01, -6.8902e-02, -2.8873e-01, -1.3778e-02,\n",
       "           3.0880e-01,  1.7019e-01,  2.9753e-01, -5.9634e-02, -5.6012e-01,\n",
       "          -1.1613e-01,  5.3788e-01,  2.6581e-01, -3.6239e-02, -3.2811e-01,\n",
       "           7.2045e-01, -1.5922e-01, -1.1489e-01, -5.6825e-01,  1.8525e-01,\n",
       "           2.2013e-02,  1.2655e-01, -7.5555e-05,  2.4000e-01, -3.3750e-01,\n",
       "           2.9920e-01,  2.7135e-01, -3.1478e-01,  3.0372e-01, -2.6181e-01,\n",
       "          -4.3372e-01,  8.7056e-02,  3.3647e-02, -6.3666e-01, -8.7324e-02,\n",
       "          -1.9992e-01, -3.9560e-01,  4.4888e-01, -8.2193e-01,  1.1965e-01,\n",
       "          -1.7510e-01, -2.2902e-01, -1.6169e-01, -2.0964e-01, -1.7375e-01,\n",
       "           2.0579e-01, -5.1365e-01, -3.0264e-01, -1.0539e-01,  1.2132e-01,\n",
       "          -4.1061e-02,  5.2551e-01, -9.1418e-02,  9.4705e-02,  7.5561e-01,\n",
       "          -1.1964e-02,  1.9192e-01, -3.9433e-01, -8.8581e-02, -3.4326e-02,\n",
       "          -4.1612e-01,  1.3300e-01, -6.0103e-01,  3.3476e-01,  1.1112e-02,\n",
       "          -1.4003e-02,  2.0371e-01, -6.0425e-01, -1.0072e-01,  2.5227e-01,\n",
       "          -2.5351e-01, -1.1990e-01, -1.4864e-01, -1.9660e-01, -2.3195e-01,\n",
       "          -3.1366e-01, -6.6187e-01,  4.6708e-01,  9.1282e-02, -2.0657e-02,\n",
       "          -4.7130e-01,  9.8868e-01,  2.8927e-01, -2.6322e-01,  2.4881e-01,\n",
       "           3.7103e-01,  1.5060e-01,  4.5090e-01, -2.7776e-01,  1.0330e-01,\n",
       "           3.9267e-02, -4.0267e-01, -1.9618e-01, -9.4251e-02, -2.1862e-02,\n",
       "           3.6944e-01, -2.7234e-01,  1.1838e-01, -1.6177e-01, -8.8444e-02,\n",
       "          -4.8445e-01,  4.8371e-01,  1.8832e-01, -2.5267e-01,  2.4388e-01,\n",
       "           2.0967e-02, -1.8949e+00,  3.9871e-02, -2.3105e-01, -4.7897e-01,\n",
       "          -3.2275e-01, -8.5602e-02,  3.1137e-01,  9.3711e-02,  2.4510e-01,\n",
       "           1.4219e-01, -7.8792e-02,  8.4044e-02,  4.6136e-01,  1.0223e-01,\n",
       "           4.0591e-01,  3.4849e-01, -2.0902e-01, -2.5852e-01,  4.5930e-02,\n",
       "          -6.1894e-01, -1.1574e-01,  3.9636e-01,  3.6740e-01,  3.2014e-01,\n",
       "           6.4785e-01,  3.8903e-02, -1.4651e-01,  1.0957e-01, -1.4752e-01,\n",
       "           3.7535e-01, -3.7863e-01, -3.7322e-01,  3.2108e-01,  1.2908e-01,\n",
       "           4.2017e-01, -3.2696e-01,  5.7927e-02,  7.2831e-01,  4.6429e-01,\n",
       "          -1.7688e-01, -2.9665e-02, -1.7307e-01, -5.2286e-02, -1.9980e-01,\n",
       "           4.9209e-01, -3.8249e-01,  1.2946e-01, -3.7197e-01, -9.3180e-02,\n",
       "          -9.5788e-02, -4.7215e-01,  3.5718e-01,  5.6297e-01, -1.1727e-01,\n",
       "          -1.6849e-01,  6.8651e-02,  6.5627e-01, -2.7744e-01,  3.0980e-02,\n",
       "          -5.3857e-01, -2.4725e-01, -6.3286e-01, -3.8587e-02,  5.4192e-01,\n",
       "           7.5255e-01, -2.8667e-01,  1.0714e+00,  3.3238e-01,  3.5256e-01,\n",
       "          -3.5165e-01,  1.4744e-01,  6.7039e-01,  2.7966e-01, -5.5822e-01,\n",
       "           5.9385e-02,  3.4565e-01, -2.8076e-01,  4.3847e-01, -3.8783e-01,\n",
       "          -3.4824e-01,  5.3421e-01,  2.3054e-01,  7.0330e-01, -3.4874e-01,\n",
       "           1.8830e-01,  4.9919e-02, -2.9403e-01, -2.1306e-01,  6.4339e-01,\n",
       "          -3.6489e-01,  2.4952e-01,  1.9514e-01, -2.8572e-01, -6.6550e-01,\n",
       "          -9.1354e-01, -3.0223e-01, -6.7007e-02,  3.5431e-01, -7.4593e-01,\n",
       "          -4.0870e-01,  7.6088e-02, -3.9937e-03,  1.7779e-02, -4.0841e-01,\n",
       "           4.0525e-01,  3.8958e-01, -1.8873e-01, -1.9785e-01, -4.8603e-01,\n",
       "          -4.6016e-01,  2.8749e-01, -4.5658e-02,  1.2681e-01, -7.2637e-02,\n",
       "          -1.5451e-01,  3.6418e-01,  2.1319e-01, -1.2446e-01, -6.8697e-01,\n",
       "          -1.8015e-01, -2.9468e-01,  1.8282e-01, -7.1410e-01,  3.3452e-03,\n",
       "           2.8042e-01,  2.4131e-01,  8.6163e-01,  1.8474e-01, -3.5062e-01,\n",
       "           8.1534e-02,  4.5654e-01, -2.1136e-01,  5.7269e-02, -7.4112e-02,\n",
       "          -1.3620e+00,  1.4242e-02, -4.2839e-01, -1.2316e-01, -4.5303e-01,\n",
       "           5.2100e-02,  8.6576e-01, -1.4549e-01, -4.2126e-01, -2.6990e-01,\n",
       "           5.5070e-01,  1.4865e-01,  6.1919e-01,  3.0836e-01, -4.7741e-01,\n",
       "           3.5972e-01, -8.6493e-01, -3.7815e-02, -1.8380e-01,  9.5874e-02,\n",
       "           2.3366e-01,  1.7494e-02,  1.5963e-01,  3.0321e-02, -3.9683e-01,\n",
       "           3.6102e-01, -5.8238e-01, -1.7930e-01,  2.6541e-01, -2.4271e-01,\n",
       "          -1.4080e-01,  1.7898e-01, -1.6885e-01, -3.0328e-01,  3.7200e-01,\n",
       "           4.3588e-01, -2.4531e-01, -3.2017e-01, -2.1781e-02,  3.8503e-01,\n",
       "           1.9535e-01,  1.8733e-01, -2.3835e-01,  4.3329e-02,  4.3905e-01,\n",
       "          -4.1400e-01,  5.2177e-01, -5.4443e-01, -2.2782e-01, -7.3131e-02,\n",
       "           7.4818e-01, -1.4185e-01, -3.8434e-01, -7.7582e-02, -4.7077e-01,\n",
       "           1.8309e-01, -1.1812e-01, -2.9111e-01,  1.4560e-01,  6.3236e-01,\n",
       "          -7.6692e-02, -4.6538e-01,  4.3522e-01, -8.0042e-01, -2.7581e-01,\n",
       "          -4.8435e-01, -1.7320e-01, -1.7237e-01, -7.8364e-01, -7.2089e-02,\n",
       "          -1.4379e-01,  5.9238e-02,  6.0597e-01,  1.1612e-01, -3.4577e-01,\n",
       "           1.9741e-01,  3.9529e-02, -2.0673e-01,  5.1050e-01,  2.8103e-02,\n",
       "          -8.8494e-02,  3.4831e-01, -2.8377e-01, -2.6060e-01,  1.0872e-01,\n",
       "          -2.7402e-01,  3.1875e-01,  5.0037e-01,  3.5796e-01, -1.5040e-01,\n",
       "          -5.7207e-01, -9.9109e-02,  2.0230e-01,  7.8534e-02,  4.0322e-01,\n",
       "          -2.0581e-01,  3.5773e-01, -2.7323e-01,  1.2061e-01, -4.3974e-02,\n",
       "          -6.0172e-02, -3.0542e-01,  1.5666e-01, -2.5317e-01, -1.1535e-01,\n",
       "           5.8964e-01,  3.4222e-01,  3.8321e-01,  4.7842e-01, -1.1085e+01,\n",
       "           3.0251e-01,  6.8555e-01, -5.4612e-01, -3.1448e-02, -1.2600e-01,\n",
       "          -5.9679e-02, -2.5084e-01,  3.3073e-01, -5.0255e-01, -2.2912e-01,\n",
       "          -2.9391e-01, -3.4054e-01,  2.1679e-01, -2.2097e-01,  5.7837e-03,\n",
       "          -1.8127e-01, -6.0889e-01,  1.0984e-01, -1.4238e-01,  2.8300e-01,\n",
       "          -1.8306e-01, -9.1739e-02,  1.9316e-01, -1.0058e-01,  3.0826e-02,\n",
       "           2.0711e-01, -3.5030e-01,  3.9411e-01,  5.0823e-01, -1.3517e-01,\n",
       "          -1.1587e-01,  1.9940e-01,  1.2952e-01,  5.1348e-03, -7.6151e-01,\n",
       "          -8.0559e-04, -8.8254e-02, -2.1275e-03, -1.0683e-01, -6.8006e-01,\n",
       "          -1.8436e-02, -5.4554e-01, -5.1345e-02,  2.8234e-01,  5.3249e-01,\n",
       "           1.2723e+00,  4.0280e-01,  2.5371e-01, -2.1850e-01, -7.9662e-02,\n",
       "          -5.1892e-01,  3.1356e-01, -1.9801e-01, -1.5832e-01, -2.5070e-01,\n",
       "           5.1976e-01, -6.0179e-01, -1.9737e-02, -6.9313e-01, -5.6417e-02,\n",
       "          -1.2692e-01,  1.5217e-01, -1.9518e-01, -3.4074e-01, -3.3550e-01,\n",
       "           3.1466e-01,  9.7235e-02, -9.9853e-03, -4.3587e-01,  2.8838e-01,\n",
       "          -2.7218e-01,  5.0705e-01,  4.9807e-02, -1.9182e-01,  2.6733e-01,\n",
       "          -2.2772e-01, -2.1764e-01,  1.6242e+00, -4.0316e-01,  4.1373e-01,\n",
       "          -8.2694e-02,  5.2155e-02, -4.6117e-01, -6.6253e-01,  2.6928e-01,\n",
       "           4.0960e-01, -1.6146e-01,  4.3272e-01,  6.1734e-01, -1.2100e-01,\n",
       "          -6.4958e-01,  2.2020e-01,  2.0595e-01,  1.4211e-01,  1.1596e+00,\n",
       "          -7.2573e-01, -4.4947e-01,  3.9700e-01,  7.5931e-02, -1.5306e-01,\n",
       "          -8.6352e-02, -1.2141e-01, -5.7009e-01,  2.9124e-01, -1.3977e-01,\n",
       "           2.8943e-01, -5.7520e-02,  1.1576e-01,  4.6346e-01, -7.6198e-02,\n",
       "           1.5340e-01,  3.1276e-01,  1.0489e-01, -3.3523e-01, -1.9235e-01,\n",
       "           7.6171e-02,  4.7695e-01, -2.3249e-01, -4.1623e-01, -5.5574e-01,\n",
       "          -2.4519e-01,  3.2167e-01,  2.5614e-01, -5.3711e-01,  5.2099e-01,\n",
       "          -4.1074e-01,  4.2873e-02,  1.6751e-01, -2.5217e-01,  5.3745e-01,\n",
       "           6.3953e-01, -1.5056e-01,  9.0297e-02, -2.9923e-01, -6.7923e-01,\n",
       "          -5.6140e-01, -1.7572e-01,  7.4895e-02, -1.5943e-01,  1.0151e-01,\n",
       "           4.7385e-01, -8.1666e-02, -1.2955e-01,  2.1158e-02,  6.3508e-01,\n",
       "           2.5362e-01,  5.2557e-01,  5.1299e-02,  4.5153e-01, -3.8320e-01,\n",
       "          -2.2875e-01, -7.6798e-02, -4.7881e-01,  1.6020e-01, -8.3735e-01,\n",
       "          -2.6543e-01, -1.9041e-01, -3.5256e-01, -2.3736e-01,  2.8701e-01,\n",
       "           2.2443e-01, -2.7327e-01,  3.2664e-02,  3.1588e-01,  4.7129e-02,\n",
       "           2.2330e-01, -5.8842e-01, -1.2519e-01, -5.8593e-02,  2.5337e-01,\n",
       "           4.4590e-01,  6.3779e-02, -5.6118e-01, -4.0430e-01, -1.9113e-02,\n",
       "          -5.6737e-02,  3.5016e-01, -5.4110e-01, -7.3716e-01,  1.4054e-01,\n",
       "           2.8165e-01,  2.1009e-01, -4.0804e-01,  1.1060e-01, -3.0973e-01,\n",
       "          -1.5089e-01, -1.3379e-01,  3.3364e-01,  1.1160e-01,  1.4929e-01,\n",
       "           6.7925e-01, -2.6191e-01, -9.8522e-02,  4.6196e-01, -4.0143e-01,\n",
       "           9.8948e-02, -3.8754e-02, -4.9138e-01,  2.8143e-02,  4.3160e-01,\n",
       "          -1.0606e-01,  2.3330e-01, -1.3458e-01, -4.2530e-01, -8.7063e-02,\n",
       "           5.6818e-01,  1.0691e+00,  3.0944e-01, -3.9635e-01,  1.0782e+00,\n",
       "           3.9582e-01, -1.6262e-01,  3.8718e-01,  6.3732e-02, -8.5683e-02,\n",
       "           4.9432e-01, -2.4501e-01, -5.6687e-01, -2.0238e-01,  5.3839e-01,\n",
       "           4.6194e-01, -5.7736e-01, -1.6203e-01, -6.7851e-02, -1.7918e-02,\n",
       "           2.4152e-01,  3.7685e-01, -1.8566e-02,  9.2135e-02,  4.7997e-01,\n",
       "          -1.0340e+00,  1.1500e-01, -1.9690e-02, -9.9785e-02, -5.0082e-02,\n",
       "          -3.0841e-01, -9.4979e-02, -5.1461e-01, -2.8326e-02,  5.5933e-01,\n",
       "           1.9918e-01, -5.1130e-01,  2.9681e-01,  5.5090e-01, -9.4242e-02,\n",
       "           1.2477e-01,  2.5132e-02, -8.6217e-02, -2.9666e-01,  1.2962e-01,\n",
       "          -4.4988e-01,  3.3660e-01, -2.0391e-01, -4.6334e-01,  3.9294e-01,\n",
       "           7.3043e-01, -2.1437e-02,  2.3346e-01,  6.2195e-02, -2.9360e-01,\n",
       "          -2.3226e-01, -5.1754e-01, -4.1486e-02,  6.9694e-02, -1.9653e-01,\n",
       "           1.7196e-01, -6.0791e-02, -3.5082e-01,  3.4488e-01, -1.0973e-01,\n",
       "           3.3972e-01,  4.5738e-01,  5.1878e-01,  1.1300e-01, -4.9208e-01,\n",
       "          -6.8150e-02,  4.0577e-01, -3.2728e-01,  2.9501e-01, -3.8595e-01,\n",
       "          -1.2050e-01,  9.6066e-02,  1.7771e-01, -5.7625e-02,  6.0858e-02,\n",
       "          -3.2997e-01,  5.4933e-01, -2.2984e-01, -1.6993e-01,  3.2819e-01,\n",
       "           2.3632e-01,  1.3125e-01, -6.2989e-01, -3.1544e-01,  3.9604e-01,\n",
       "           8.6938e-02,  4.6765e-01, -1.5740e-01,  5.9194e-01, -2.7891e-01,\n",
       "          -3.3559e-02, -6.8972e-01, -8.9791e-02,  1.1098e-01,  1.3362e-01,\n",
       "          -3.0568e-01, -3.5784e-01, -2.4294e-01,  1.7385e-01,  7.8556e-01,\n",
       "           2.5632e-01,  2.7652e-01, -8.4203e-01, -2.8352e-01,  3.9307e-01,\n",
       "          -1.6648e-01,  4.8807e-02,  1.0673e-01, -3.1301e-02,  3.4003e-01,\n",
       "          -7.8578e-01, -2.3027e-01, -1.8933e-01, -1.8787e-02, -2.7918e-01,\n",
       "           8.4718e-01,  1.4408e-01,  2.6514e-03,  3.2127e-03, -1.4701e-01,\n",
       "           6.5901e-01,  7.7809e-01, -2.1668e-01, -4.1134e-01, -3.5181e-01,\n",
       "           4.1760e-02,  5.5770e-02,  1.9483e-02, -6.7756e-01,  3.9717e-01,\n",
       "           3.6362e-01,  3.4572e-01,  3.7433e-01]], device='mps:0'),\n",
       " tensor([[-0.4654, -0.4115, -0.4370,  ...,  0.1625,  0.4338,  0.5908],\n",
       "         [-0.2072, -0.6105, -0.7214,  ...,  0.2664,  0.0715,  0.6498]],\n",
       "        device='mps:0'),\n",
       " tensor([[-0.2744, -0.4852, -0.4880,  ...,  0.2064,  0.2977,  0.6606],\n",
       "         [-0.2434, -0.3725, -0.3338,  ...,  0.2009,  0.3829,  0.4365],\n",
       "         [ 0.0506, -0.6230, -0.6446,  ..., -0.0875,  0.4018,  0.3101],\n",
       "         [-0.0764, -0.4928, -0.4785,  ...,  0.1521,  0.8034,  0.4097]],\n",
       "        device='mps:0'),\n",
       " tensor([[-7.6538e-02, -5.2230e-01, -5.2310e-01, -7.1511e-01, -4.1609e-01,\n",
       "          -3.7903e-01, -4.2444e-01,  3.5113e-01, -2.3957e-01,  6.7637e-01,\n",
       "           1.1287e-01,  3.5605e-01,  2.0167e-01, -3.2726e-01, -1.6422e-01,\n",
       "          -1.7800e-01, -6.2987e-01,  5.2032e-01,  4.5464e-01,  4.4539e-01,\n",
       "           3.9602e-01,  4.0237e-01,  9.8502e-03,  1.1366e-01, -6.4128e-01,\n",
       "           3.7512e-01,  3.2868e-02,  1.5788e-01,  3.0904e-01, -5.0473e-01,\n",
       "           1.2010e-01,  3.3559e-01,  4.9274e-02,  9.2958e-01, -6.1034e-02,\n",
       "          -1.0119e-01,  2.3279e-01, -8.9318e-01,  8.5851e-01, -3.4185e-02,\n",
       "           4.6250e-01, -2.6565e-01,  4.4386e-01,  1.9944e-02,  1.7167e-01,\n",
       "          -7.8220e-02, -6.7188e-03, -4.1006e-01,  1.4940e-01, -5.3064e-01,\n",
       "          -7.6399e-02, -1.4576e-01,  6.6078e-01,  3.0525e-01, -4.6931e-01,\n",
       "          -3.5667e-01,  2.3855e-01,  2.7895e-01, -5.9737e-01, -6.4238e-02,\n",
       "          -4.3729e-01,  6.0962e-01,  1.9520e-02, -6.1173e-02,  1.2258e-01,\n",
       "          -6.6264e-02, -4.8517e-01, -5.9352e-01, -1.3909e-01, -3.0623e-01,\n",
       "          -2.6159e-02,  3.2147e-01,  2.5025e-01,  1.3900e-01,  6.3168e-01,\n",
       "          -1.5218e-01, -2.4506e-01, -2.5200e-01, -6.0999e-01,  1.9406e-01,\n",
       "           6.0107e-02,  2.2846e-01,  1.8047e-01, -7.0494e-02,  2.5067e-01,\n",
       "          -8.7409e-02,  4.5535e-01,  3.3401e-01,  9.8844e-02, -1.2752e-01,\n",
       "          -3.5864e-02,  3.2038e-01, -1.2830e-01, -1.7122e-01,  1.4353e-01,\n",
       "          -7.2192e-02,  3.2551e-01, -4.1316e-01, -3.6751e-01, -1.5071e-01,\n",
       "           1.4322e-01,  1.6350e-01,  2.4266e-02,  4.1994e-01, -2.0981e-01,\n",
       "          -7.6233e-01,  2.0835e-01,  3.0140e-02, -7.2533e-01,  2.6141e-02,\n",
       "          -1.0264e-01, -1.9242e-02, -6.3089e-02, -7.0092e-01, -4.0305e-01,\n",
       "           3.4293e-01,  7.7723e-02, -2.5382e-01, -3.2828e-01,  2.1131e-01,\n",
       "           3.0475e-01, -4.5307e-01,  1.4207e-01,  3.0463e-02,  1.0274e-01,\n",
       "          -4.3760e-01,  4.1392e-01, -2.4026e-01,  4.4945e-01,  3.0050e-01,\n",
       "          -4.8433e-02,  1.2928e-01, -2.4092e-01,  3.9267e-02, -1.7796e-01,\n",
       "           5.1520e-02,  3.7367e-01, -1.2084e-01,  6.2232e-01,  6.2456e-02,\n",
       "           2.6376e-02, -1.8051e-01, -1.3097e-02, -2.8977e-02,  7.5907e-02,\n",
       "          -1.2580e-01,  2.3891e-01, -2.0155e-01,  7.5296e-03, -4.1514e-02,\n",
       "          -4.0220e-01, -6.2283e-01,  3.3046e-01, -3.6855e-01,  1.4926e-02,\n",
       "          -1.8132e-03,  2.4794e-01,  3.4250e-01,  9.7456e-02,  2.3483e-01,\n",
       "           9.2272e-01, -1.6996e-01,  7.0218e-03, -5.1127e-01, -8.8591e-02,\n",
       "          -6.2433e-01,  2.8445e-01, -3.4782e-01,  2.8941e-01, -6.5882e-02,\n",
       "           4.0388e-02, -3.0835e-01,  3.2564e-01,  2.2537e-01,  3.2563e-01,\n",
       "          -2.2546e-01,  2.7084e-01,  2.1898e-01, -1.8291e-01, -3.3903e-01,\n",
       "           6.9189e-02, -1.1749e+00,  9.5458e-02, -1.7981e-01, -4.1031e-01,\n",
       "          -2.8000e-01,  1.4508e-01,  9.1935e-02,  2.3104e-01,  1.3294e-01,\n",
       "          -4.0936e-01,  8.0636e-02,  2.7722e-01,  2.4576e-01,  1.8119e-01,\n",
       "           3.1662e-01,  6.0960e-01,  1.8196e-01,  4.9766e-02, -6.9276e-02,\n",
       "          -3.1767e-01,  3.2691e-02, -1.6724e-01, -1.0477e-01, -4.4069e-02,\n",
       "           5.1146e-01, -8.6202e-02,  2.4004e-01,  2.0757e-01,  9.8197e-03,\n",
       "          -3.3107e-02, -1.2680e-01, -2.6420e-01, -2.7553e-01, -2.1220e-01,\n",
       "           5.0339e-01, -3.6670e-01, -1.1146e-01,  3.6907e-01,  1.9792e-02,\n",
       "          -1.6359e-01,  7.1550e-02,  2.1957e-01, -2.1881e-01, -9.4622e-02,\n",
       "           6.7104e-01, -3.7565e-01,  1.8520e-01, -3.8422e-01, -2.6694e-01,\n",
       "           2.4959e-01, -2.3136e-01,  7.3762e-01,  2.0836e-01, -1.7197e-01,\n",
       "          -3.3441e-01, -4.5725e-01,  4.1214e-01, -1.1209e-01, -3.2923e-01,\n",
       "          -5.6535e-01, -3.7804e-01, -8.1047e-01, -5.8460e-02,  1.2732e-01,\n",
       "           2.2923e-01, -3.4034e-02,  9.0953e-01,  2.3980e-01,  1.8989e-01,\n",
       "          -1.6950e-01, -2.4832e-01,  7.0532e-03, -7.5116e-02, -2.4260e-01,\n",
       "           7.0917e-02,  1.8916e-01, -4.6444e-01,  4.0026e-01, -2.9642e-01,\n",
       "          -9.2742e-02, -3.0854e-01,  4.2555e-03,  3.8323e-01, -1.0083e-02,\n",
       "           3.0301e-01,  2.5428e-01, -2.5777e-01, -2.4470e-01,  2.7877e-01,\n",
       "          -4.4515e-01,  2.4692e-01,  5.6397e-01, -6.3838e-01, -5.9971e-01,\n",
       "          -6.6812e-01, -2.2791e-01,  6.5276e-02,  1.4597e-01, -8.5148e-01,\n",
       "           4.4431e-02, -3.1650e-01,  7.9318e-02, -1.5638e-02, -1.9317e-02,\n",
       "           1.5718e-01,  3.5012e-01, -1.0667e-01, -4.2446e-01, -2.0605e-01,\n",
       "           6.9806e-02,  3.4277e-01, -2.9438e-01, -5.1027e-02,  1.8855e-01,\n",
       "          -1.1465e-01,  1.5345e-01,  2.8112e-01,  2.5505e-01, -6.8974e-01,\n",
       "          -2.4501e-01, -1.7696e-01,  3.2095e-01, -7.0660e-02,  4.8017e-01,\n",
       "          -6.5356e-02,  3.3493e-01,  6.2961e-01, -1.7968e-01, -6.0922e-01,\n",
       "          -1.5303e-01,  4.0683e-01, -2.2380e-01, -3.4729e-01, -1.9047e-01,\n",
       "          -1.2405e+00, -1.8468e-01, -4.2674e-01,  2.0700e-01, -1.8526e-01,\n",
       "           1.3076e-01,  1.8541e-01, -8.9671e-02,  2.1847e-01, -2.3560e-01,\n",
       "           3.9909e-01,  7.8109e-01,  3.2998e-01, -7.6496e-02, -1.5381e-01,\n",
       "           3.0437e-02, -5.8957e-01, -4.2392e-01, -3.9808e-01,  2.4403e-01,\n",
       "           1.9542e-01,  1.9267e-01, -3.1038e-01, -1.1792e-01, -4.6528e-01,\n",
       "           6.5367e-01, -3.2673e-01, -3.0089e-01,  2.6004e-01, -1.8801e-01,\n",
       "          -1.0396e-01,  3.8376e-01, -7.5697e-01, -5.6936e-02,  5.8193e-01,\n",
       "           2.4022e-01,  3.7060e-01, -2.6969e-01,  2.1719e-02,  5.6201e-01,\n",
       "           3.7877e-01,  3.7736e-01, -2.8480e-01,  5.5038e-01,  4.0508e-01,\n",
       "          -1.5839e-02,  1.7862e-01, -1.6517e-01, -4.4372e-01, -1.3200e-01,\n",
       "           1.5465e-01,  1.8125e-01, -6.4679e-01,  8.1767e-02, -3.9405e-01,\n",
       "           8.1859e-02, -1.9098e-01,  1.8261e-02,  2.1708e-01,  3.9310e-01,\n",
       "          -1.4055e-01, -5.3100e-01,  5.8902e-01, -5.7342e-01, -1.2414e-01,\n",
       "           3.9959e-02,  6.3690e-02, -3.3564e-02, -2.1708e-01, -2.8389e-01,\n",
       "          -1.1408e-01,  1.9287e-01,  1.0669e+00,  9.0277e-02, -2.2046e-01,\n",
       "           1.8371e-01, -2.4628e-01, -1.9485e-01,  5.0244e-01, -1.9322e-01,\n",
       "          -4.0932e-01,  1.0055e-01, -5.3343e-01, -7.7594e-02,  8.4647e-01,\n",
       "          -3.7746e-01,  3.9002e-01,  1.5952e-01,  6.6473e-02, -1.7283e-01,\n",
       "          -3.2170e-01,  2.3321e-01, -5.7596e-02,  6.6399e-03,  2.5674e-01,\n",
       "           2.0088e-01,  2.8771e-01, -4.3517e-01,  7.5755e-02, -8.4660e-02,\n",
       "           3.1255e-01, -2.3702e-01,  1.5409e-01, -9.9623e-02, -8.9824e-01,\n",
       "           6.6719e-01,  3.6099e-01,  2.3143e-01,  2.0422e-01, -1.1215e+01,\n",
       "           5.1316e-01,  5.1932e-01, -4.2886e-01,  2.6631e-01,  2.6028e-02,\n",
       "           6.1176e-01, -5.9413e-01,  2.3480e-01, -2.1771e-01, -3.8520e-02,\n",
       "           5.0489e-02, -1.0635e-02, -8.8785e-02, -1.8890e-01, -1.2370e-01,\n",
       "          -1.2083e-01, -9.3030e-01,  4.4318e-01, -6.3852e-03, -2.5326e-02,\n",
       "          -2.3489e-01, -3.1725e-02,  2.5275e-01, -7.3426e-01, -1.3707e-01,\n",
       "          -1.9303e-01, -3.0534e-01,  2.4674e-01, -9.0922e-02, -5.0987e-02,\n",
       "          -1.2192e-01,  2.9028e-02,  1.1081e-01, -2.2045e-01, -6.7180e-01,\n",
       "          -1.9668e-01, -1.6661e-02,  5.6527e-01, -1.5623e-01, -1.4682e-01,\n",
       "          -4.5599e-01, -5.2721e-02,  5.2318e-01,  2.6479e-01,  5.6273e-01,\n",
       "           8.1341e-01,  6.6841e-01,  1.1681e-01, -4.8063e-01, -1.5460e-01,\n",
       "          -8.4638e-01,  3.6824e-01, -9.8870e-01, -1.1799e-01, -1.9234e-01,\n",
       "           3.5276e-01, -7.6084e-01,  1.1156e-01, -2.9887e-01, -5.0612e-01,\n",
       "           3.4212e-01,  4.2393e-01, -3.9586e-01, -1.9273e-01, -1.0110e-01,\n",
       "           6.8687e-01, -3.3706e-01,  1.9627e-01, -1.2971e-01,  1.2304e-01,\n",
       "          -3.6504e-01,  3.3495e-01,  3.1651e-01, -7.5882e-01, -7.0900e-02,\n",
       "           2.0573e-01, -1.8896e-01,  2.3494e+00, -3.6768e-01, -3.4614e-02,\n",
       "          -5.8158e-02,  3.3181e-01, -4.2279e-01, -3.6931e-01, -1.9593e-01,\n",
       "           4.0568e-03,  3.8884e-01,  2.2649e-01,  8.2707e-01,  1.4389e-02,\n",
       "          -5.9493e-01,  1.9903e-01,  7.7642e-02,  4.3241e-01,  5.9979e-01,\n",
       "          -6.3828e-01, -1.4111e-01,  3.8405e-01,  2.3743e-01, -7.2048e-01,\n",
       "           6.1837e-02, -8.1194e-02, -3.7828e-01,  2.5224e-01,  3.2323e-01,\n",
       "           4.6019e-02, -6.5391e-02,  3.8440e-02,  3.0771e-01, -3.2995e-01,\n",
       "           4.9307e-01, -7.2347e-02,  2.1291e-01, -2.0684e-01,  8.5857e-02,\n",
       "           2.6578e-01,  8.2700e-01, -4.1293e-01, -2.9021e-01, -1.9361e-01,\n",
       "           9.2027e-02,  1.6705e-01, -2.1872e-01, -9.1750e-02,  2.8731e-01,\n",
       "          -3.8828e-01, -1.7857e-02,  4.3734e-01,  8.0714e-02,  2.7903e-01,\n",
       "           3.6327e-01, -4.7073e-01,  9.0849e-02, -2.7895e-01, -5.1207e-01,\n",
       "          -4.9944e-01, -7.2504e-02,  2.6273e-01,  1.0011e-01, -1.5165e-01,\n",
       "          -9.0208e-02,  1.1102e-02, -2.0806e-01,  4.6693e-01, -6.0519e-04,\n",
       "          -9.1172e-02,  2.1360e-01, -3.6497e-01,  9.6571e-02, -4.9995e-02,\n",
       "           1.7239e-01, -1.6304e-01, -4.2385e-01,  4.7151e-03, -6.6158e-01,\n",
       "          -9.0439e-01, -3.4721e-01, -2.5924e-01, -5.5251e-02, -7.1027e-03,\n",
       "          -1.6287e-01,  2.2693e-01, -5.2483e-01, -1.2381e-02,  1.1386e-01,\n",
       "           2.1114e-03, -4.9996e-01, -1.7126e-01, -7.2456e-02,  1.2010e-02,\n",
       "           3.1882e-01,  1.4409e-01, -3.5663e-01, -1.6426e-01, -2.5675e-01,\n",
       "          -4.7476e-01,  1.3047e+00, -1.1282e-01, -6.2187e-01,  4.8029e-01,\n",
       "           5.9831e-01,  5.3625e-02,  5.0506e-01, -2.7395e-01, -6.9018e-02,\n",
       "          -2.5594e-01, -1.7352e-01,  6.6219e-01,  1.2913e-02,  4.5690e-01,\n",
       "           5.7616e-01,  4.0454e-01, -4.6435e-01,  3.2396e-01,  3.7953e-02,\n",
       "          -3.3310e-01, -1.5078e-01, -5.2639e-01,  1.8837e-01,  6.2501e-01,\n",
       "          -2.4701e-01,  1.7477e-02, -9.3451e-02, -2.6365e-02, -3.3839e-01,\n",
       "           3.2170e-01,  1.1091e+00, -1.3116e-01, -2.8816e-01,  2.6398e-01,\n",
       "           9.6696e-02,  3.5907e-01,  2.4068e-01, -5.5929e-02,  1.6401e-01,\n",
       "           3.5789e-01, -2.7361e-01, -1.0023e-01,  1.9765e-01, -3.6409e-02,\n",
       "           3.9309e-01, -9.1731e-01, -5.1293e-01,  9.7005e-02,  1.5422e-01,\n",
       "          -2.1479e-01,  5.2158e-01,  1.6364e-02,  4.2758e-01,  7.0429e-01,\n",
       "          -1.0354e+00, -2.7544e-01, -1.0368e-01,  2.0273e-01,  2.3710e-01,\n",
       "          -1.6998e-01, -2.6933e-01, -9.6214e-02, -2.8217e-01,  1.7544e-01,\n",
       "           8.2703e-01, -8.3849e-01,  1.5997e-01,  5.2477e-01, -3.1486e-01,\n",
       "           8.0247e-02, -9.6342e-02,  7.7852e-01,  1.4205e-01, -4.3671e-01,\n",
       "          -1.0834e-01,  1.6236e-01, -7.7812e-02, -6.5539e-01, -4.0991e-02,\n",
       "           7.3154e-01,  1.6077e-02, -2.3925e-02, -8.9681e-02, -7.0620e-02,\n",
       "          -1.7730e-01, -2.1863e-01, -2.0176e-01,  3.2027e-01,  2.9161e-01,\n",
       "           2.2537e-02, -2.3227e-01, -8.9027e-02,  2.9712e-01, -6.6061e-02,\n",
       "           3.6929e-01,  5.5046e-01,  7.8235e-01,  1.2076e-01, -1.9017e-01,\n",
       "           1.6968e-02,  1.2995e-01,  1.8040e-02,  4.1119e-01, -2.1263e-01,\n",
       "          -6.0704e-01,  2.9532e-01, -4.6319e-02, -5.9909e-01, -3.8876e-02,\n",
       "          -3.4351e-01,  3.3444e-01, -6.0632e-02, -4.2119e-01,  5.8909e-01,\n",
       "          -1.4234e-02,  2.4912e-01, -5.1883e-01,  1.6080e-01, -9.8272e-02,\n",
       "           2.5211e-02,  7.1988e-01,  1.1870e-01,  5.9651e-01, -5.1953e-01,\n",
       "           2.1086e-01, -8.4833e-01, -4.1395e-01,  1.6058e-01, -3.3878e-01,\n",
       "          -2.6224e-01, -5.8043e-02,  2.7094e-01,  1.0931e-02,  3.7057e-01,\n",
       "           6.6784e-01, -9.2784e-02, -1.0892e+00, -5.9448e-01, -3.4336e-01,\n",
       "           1.5056e-01, -4.2372e-01,  3.9250e-01, -5.9808e-01,  4.2502e-02,\n",
       "          -3.6997e-01, -6.5162e-01,  1.4535e-01,  4.9833e-02, -2.3991e-01,\n",
       "           9.3165e-01, -1.7378e-01,  5.5060e-01,  6.0373e-01,  1.7152e-02,\n",
       "           1.6380e-01,  5.1026e-01, -8.4184e-02, -6.5546e-01, -1.2620e-01,\n",
       "           2.9797e-01, -1.5122e-01, -2.6453e-01, -3.6229e-01,  1.4532e-01,\n",
       "           9.2899e-02,  6.3648e-01,  6.5716e-01]], device='mps:0'),\n",
       " tensor([[-5.6844e-01, -2.5393e-01, -5.0750e-01, -8.7244e-01, -3.2844e-01,\n",
       "          -3.6569e-01, -9.0136e-01,  7.7320e-01,  8.1617e-02,  4.9008e-01,\n",
       "           9.4442e-02,  1.7855e-01,  2.8632e-01, -1.8449e-01, -5.2880e-01,\n",
       "          -3.7395e-01,  1.3400e-01,  3.8131e-01,  3.5838e-01,  1.0985e-01,\n",
       "           5.8069e-01,  1.8646e-01, -1.6628e-01,  6.8736e-02, -6.3904e-01,\n",
       "           2.5407e-01,  2.1383e-01,  3.1362e-02,  1.9298e-01, -8.6934e-01,\n",
       "           1.5436e-01,  4.0132e-01,  1.4270e-01,  9.5617e-01, -2.6119e-01,\n",
       "          -1.3691e-01,  7.3087e-02, -4.0375e-01,  5.6589e-01,  4.6931e-01,\n",
       "           2.6016e-01, -4.5020e-01,  6.3035e-03,  1.8017e-01, -1.3233e-01,\n",
       "           7.1244e-02, -6.5158e-02, -6.6997e-01,  4.3002e-01, -6.2951e-01,\n",
       "          -4.0576e-01, -1.7096e-01,  6.4477e-01,  4.3165e-01, -9.5134e-01,\n",
       "          -5.1421e-01,  3.6650e-01,  3.7034e-01, -6.0353e-01,  2.5449e-01,\n",
       "          -5.6246e-02,  7.4130e-01,  3.2109e-01, -3.4573e-03,  4.6800e-01,\n",
       "           2.8329e-01, -2.3211e-01, -6.1568e-01, -1.7770e-01, -1.6986e-01,\n",
       "          -1.4760e-02, -1.5805e-01, -4.9675e-01, -1.8398e-01,  5.6503e-01,\n",
       "           1.4111e-01,  1.8994e-01,  6.2489e-01, -3.7451e-01,  1.5743e-01,\n",
       "           1.1599e-01,  1.8623e-01,  9.5090e-02,  1.9397e-02,  2.6867e-01,\n",
       "           3.0676e-01,  6.6873e-01,  1.7461e-01, -4.1899e-02,  7.6356e-03,\n",
       "           4.0632e-01, -1.3571e-01, -1.9968e-01, -9.0196e-01,  3.4882e-01,\n",
       "          -2.3752e-01,  2.9844e-01, -3.2413e-01,  2.0613e-01, -1.9357e-01,\n",
       "           2.7677e-01,  4.7619e-01, -3.1541e-01, -2.9064e-02, -3.1801e-01,\n",
       "          -7.2487e-01,  2.4777e-01,  1.3619e-01, -4.8521e-01,  2.4988e-01,\n",
       "           6.3011e-02, -2.1547e-01,  4.0180e-02, -1.5635e-01, -2.3985e-01,\n",
       "          -1.3972e-01,  2.8368e-01, -1.5412e-01, -6.8293e-01,  3.4060e-01,\n",
       "           1.7510e-01, -3.9404e-01, -3.4872e-01, -7.5550e-02,  2.1179e-01,\n",
       "          -3.4693e-01,  4.3310e-01,  5.8511e-02,  2.1442e-01,  7.8144e-01,\n",
       "          -2.3282e-03,  1.7204e-01, -1.7537e-01, -2.0757e-01, -3.1992e-01,\n",
       "           1.2913e-01,  1.1688e-01, -9.7004e-02,  7.2642e-02,  8.6313e-02,\n",
       "          -2.5738e-01, -2.4418e-01, -7.5146e-01, -5.0822e-01, -2.4911e-01,\n",
       "          -4.5583e-01,  2.1359e-01, -4.9226e-02, -4.0862e-02, -4.1612e-01,\n",
       "          -7.7786e-01, -5.3849e-01,  7.3452e-01,  2.2917e-02, -5.7738e-02,\n",
       "          -3.4818e-01,  1.8976e-01,  3.5786e-01, -6.8795e-02,  5.2628e-01,\n",
       "           4.0225e-01,  2.8456e-01,  1.1433e-01, -5.6339e-01, -1.9941e-01,\n",
       "           8.1964e-02, -2.7762e-01, -3.5982e-01,  1.1957e-01,  3.0120e-01,\n",
       "           7.2749e-02, -1.9488e-01,  1.1217e-01, -6.4809e-01, -1.6623e-01,\n",
       "          -6.0746e-01,  4.2744e-01,  2.4433e-01, -4.7229e-01, -1.0579e-01,\n",
       "          -3.3341e-01, -1.6541e+00,  2.9410e-01, -7.4445e-01, -2.2061e-01,\n",
       "          -9.1557e-01,  2.5267e-02,  2.7081e-01,  2.3996e-01,  1.7481e-01,\n",
       "           3.0564e-02,  1.9829e-01, -2.6402e-02,  3.0950e-01,  1.3367e-01,\n",
       "           2.1250e-01,  4.2880e-01, -3.5856e-02, -3.1493e-01,  1.7774e-01,\n",
       "          -2.9625e-01,  3.1158e-01, -2.8723e-02,  1.6269e-01,  3.5764e-01,\n",
       "           3.8199e-01, -6.8891e-02,  2.9418e-01,  1.9944e-01, -2.8504e-01,\n",
       "           7.7662e-02, -2.0543e-01, -3.6897e-01,  1.2753e-02, -1.4946e-01,\n",
       "           5.7360e-02, -4.4640e-01,  6.0071e-03,  2.8395e-01,  1.4853e-01,\n",
       "          -1.1382e-01,  2.8209e-01, -2.7212e-01, -4.6728e-01,  5.0678e-02,\n",
       "           5.0747e-01, -2.5028e-01,  4.1382e-01, -4.3106e-01,  1.1839e-01,\n",
       "          -2.8436e-01, -4.3520e-01,  5.3732e-01,  6.0607e-01,  2.4345e-01,\n",
       "          -1.8731e-01, -3.4659e-01,  3.4032e-01, -1.7321e-01, -3.5382e-01,\n",
       "          -2.7126e-01, -3.0751e-01, -6.6050e-02, -1.4941e-01,  6.6002e-01,\n",
       "           1.4670e-01, -3.1667e-02,  1.0202e+00,  6.9238e-01,  2.0488e-01,\n",
       "          -4.7313e-02, -1.3113e-01,  3.3900e-01,  2.8843e-01, -4.0147e-01,\n",
       "          -2.2174e-01, -4.5826e-02, -5.3387e-01,  5.6696e-01, -2.8629e-01,\n",
       "           3.1854e-01,  8.9703e-02,  9.8121e-02,  7.2426e-01,  4.0746e-01,\n",
       "          -5.5773e-01, -1.0385e-02, -3.8020e-02, -2.2596e-01, -8.8121e-02,\n",
       "           5.8970e-04,  2.4156e-01, -4.0000e-01, -2.9996e-01, -7.2638e-01,\n",
       "          -2.3492e-01, -2.1963e-01,  1.6584e-01,  3.0846e-01, -6.2015e-01,\n",
       "          -2.4234e-01,  3.6705e-01, -2.8687e-01,  2.0502e-01,  2.5558e-01,\n",
       "           4.2903e-01, -3.4778e-02, -9.3549e-02, -2.0382e-01, -7.2037e-02,\n",
       "          -7.4568e-02,  1.1676e-01, -2.9530e-01,  2.0986e-02,  5.1684e-03,\n",
       "          -1.1136e-01,  4.3742e-01,  3.4734e-01,  7.8739e-02, -6.1562e-01,\n",
       "           4.4871e-02, -1.5161e-01, -2.5925e-01, -2.2698e-01, -3.9915e-02,\n",
       "          -3.3285e-01,  3.5655e-01,  8.4436e-01, -4.9563e-02, -2.6343e-01,\n",
       "           1.9915e-01,  5.1512e-01, -1.3016e-01,  1.4426e-01, -2.5021e-02,\n",
       "          -9.0834e-01, -2.5524e-01, -1.9994e-01, -1.2356e-01, -4.6987e-01,\n",
       "           1.1757e-01,  6.0037e-01, -2.2855e-01, -3.6448e-01, -6.8986e-01,\n",
       "           4.8317e-01,  1.6688e-01,  3.3535e-01,  1.3286e-01, -2.9904e-01,\n",
       "          -4.7916e-02, -8.2334e-01,  1.2087e-01, -5.1199e-01, -4.3309e-01,\n",
       "           2.3617e-01,  2.8534e-01,  4.3335e-02,  3.0027e-01, -4.8130e-01,\n",
       "           6.8084e-01, -2.2633e-01, -1.7373e-01,  3.8647e-01, -3.8197e-01,\n",
       "          -2.4056e-01,  5.5692e-01, -1.3181e-01,  3.6477e-01,  5.5141e-02,\n",
       "           5.7557e-01, -2.1858e-01, -5.1305e-01, -1.2626e-01,  4.3394e-01,\n",
       "           2.2980e-01,  3.8016e-01, -2.2147e-01,  6.9901e-02,  2.4774e-01,\n",
       "          -3.1906e-01,  5.5489e-01, -3.6133e-01, -4.8930e-01,  1.1730e-01,\n",
       "           3.6468e-01,  1.4321e-01, -3.3744e-01, -1.9847e-01,  5.3661e-01,\n",
       "           6.1608e-02, -2.2447e-02,  3.8687e-01,  4.1707e-01,  7.5413e-02,\n",
       "          -4.1578e-01, -5.8536e-01,  1.2953e-01, -3.9358e-01, -2.0976e-01,\n",
       "           3.5414e-01, -2.8840e-01,  2.8654e-01, -3.9533e-01, -4.0960e-01,\n",
       "          -4.1113e-01,  1.3799e-01,  6.9650e-01,  2.6831e-01,  7.2333e-02,\n",
       "           2.6272e-01, -4.3981e-02, -2.9518e-01,  5.3852e-01, -1.2421e-01,\n",
       "          -3.4740e-01,  3.1586e-01, -2.1424e-01,  5.4590e-02,  2.9407e-01,\n",
       "          -2.2166e-01,  1.6948e-01,  2.2397e-01,  4.9950e-01, -3.4496e-01,\n",
       "          -2.7747e-01, -1.1121e-01, -2.1886e-01, -1.9180e-01,  1.6433e-01,\n",
       "           1.2031e-01,  5.6576e-01, -1.2635e-01, -7.3580e-02,  5.6376e-01,\n",
       "           1.4494e-02, -2.4303e-01,  1.4364e-01, -5.4605e-01, -4.0354e-01,\n",
       "           8.3474e-03,  1.2952e-01,  1.1459e-01,  3.5272e-01, -1.0831e+01,\n",
       "          -1.4639e-01,  7.7977e-01, -4.0763e-01, -9.5451e-03,  6.6180e-02,\n",
       "           1.1330e-01, -1.9354e-02,  2.2326e-01, -2.0652e-02, -6.5077e-01,\n",
       "          -4.5718e-01, -2.8242e-01,  1.1227e-01,  1.4893e-01, -3.7762e-01,\n",
       "           3.6570e-02, -7.7520e-01, -2.6453e-02, -5.8563e-01,  1.1856e-01,\n",
       "          -3.1038e-01,  1.1832e-01,  2.6041e-01, -4.3650e-01, -1.8852e-01,\n",
       "           1.1934e-01, -1.7524e-01,  9.0805e-02,  8.6730e-02, -1.5353e-01,\n",
       "          -1.3314e-01, -8.9663e-02,  1.7702e-01, -1.7022e-01, -7.7946e-01,\n",
       "          -4.5533e-02, -1.0732e-02,  5.5088e-01, -1.8145e-01, -7.5737e-01,\n",
       "          -5.5589e-01, -7.2691e-01,  4.4681e-01,  4.6576e-01,  8.4337e-01,\n",
       "           7.3694e-01,  7.8225e-01,  2.8821e-01,  1.1654e-01, -7.8781e-03,\n",
       "          -4.5251e-01,  7.6623e-02, -5.4234e-01, -2.0190e-01, -1.6733e-01,\n",
       "           3.1524e-01, -7.4778e-01,  5.4153e-01, -3.3615e-01, -2.3183e-01,\n",
       "           2.1240e-01,  3.8045e-02,  2.9980e-01, -1.5400e-01, -5.7548e-01,\n",
       "           6.9551e-01, -8.1846e-02,  8.7726e-02, -3.6636e-01,  1.2998e-01,\n",
       "          -1.1565e-01,  4.2252e-01,  4.9964e-01, -4.2858e-01,  5.0750e-01,\n",
       "          -2.9582e-02, -3.5795e-01,  1.5328e+00, -2.8105e-01,  3.4889e-01,\n",
       "          -3.9615e-02,  1.9863e-01, -5.1251e-01, -7.2759e-01,  5.4114e-02,\n",
       "           7.7376e-03, -5.2419e-02,  5.1108e-01,  3.0878e-01, -3.2255e-01,\n",
       "          -7.7520e-01,  1.9643e-01,  1.9483e-01,  4.1738e-01,  8.3253e-01,\n",
       "          -3.7144e-01, -2.2433e-01,  1.3851e-01,  4.9568e-02, -9.1221e-01,\n",
       "           1.5130e-03, -4.3752e-01, -3.9226e-01,  4.8852e-01,  4.2343e-02,\n",
       "           5.0267e-01,  1.8224e-01, -5.7171e-03,  5.0501e-01, -4.4502e-01,\n",
       "           3.0443e-01, -2.1046e-01,  1.4119e-01, -4.2433e-01,  1.0247e-01,\n",
       "           3.6354e-02,  1.8873e-01, -2.9864e-01, -4.8747e-01,  1.9256e-02,\n",
       "           5.4759e-03,  1.6032e-01,  4.2622e-01, -5.6251e-01,  5.0070e-01,\n",
       "          -4.8306e-01, -6.8047e-03,  1.4265e-01, -2.3568e-01,  7.4557e-01,\n",
       "          -1.8767e-01, -3.2433e-01, -1.5232e-01, -7.4690e-01, -5.0003e-01,\n",
       "          -5.5172e-01, -1.4172e-01,  1.9040e-01, -9.0813e-01, -1.5122e-02,\n",
       "           1.7257e-01,  3.3071e-01, -3.2582e-01,  2.3588e-01,  3.5528e-01,\n",
       "          -2.3571e-02,  5.9669e-01, -4.8350e-03,  2.2136e-01, -1.7667e-01,\n",
       "          -5.2245e-01,  5.5999e-03,  1.7072e-01,  1.7848e-02, -1.0474e+00,\n",
       "          -2.8334e-01, -8.6200e-02, -4.5816e-01,  3.3449e-01,  2.6597e-02,\n",
       "          -7.7005e-02, -1.1522e-01, -4.6150e-01,  4.5910e-01,  5.7866e-01,\n",
       "           2.6352e-01, -1.0573e-02, -2.1847e-01,  6.8300e-02,  4.7001e-01,\n",
       "           3.7492e-02, -2.0095e-01, -3.8934e-01, -3.0347e-01,  2.2958e-01,\n",
       "           3.3929e-01,  5.1629e-01, -4.2074e-01, -7.4347e-01,  2.6759e-01,\n",
       "           5.2226e-01,  1.6757e-01, -1.0236e-01, -2.4258e-01, -2.6495e-01,\n",
       "          -2.6086e-01, -2.0432e-01,  4.2878e-01, -2.3748e-01,  1.3720e-01,\n",
       "           1.2240e-01,  6.0109e-01,  3.4140e-01,  4.1248e-01, -7.8067e-01,\n",
       "          -2.4939e-01, -2.0628e-02, -9.6719e-01,  1.4958e-01,  1.0040e-01,\n",
       "          -9.1728e-02,  4.3231e-01, -1.2503e-01, -1.5137e-01,  2.4529e-01,\n",
       "           5.3670e-01,  6.4405e-01,  1.3705e-01,  2.8725e-01,  8.6653e-01,\n",
       "           4.5630e-01,  7.6329e-02,  5.2303e-02,  3.7178e-01,  5.7313e-02,\n",
       "           2.7836e-01, -8.3007e-02, -3.8735e-01, -1.3449e-01,  1.5554e-01,\n",
       "           5.8080e-01, -1.1659e+00, -4.9298e-01, -2.5056e-01,  3.4759e-01,\n",
       "          -4.3638e-01,  1.2396e-01, -5.5119e-02,  3.2727e-01,  7.0494e-01,\n",
       "          -1.2319e+00, -4.9664e-01,  4.3808e-01,  2.5547e-01,  6.9514e-03,\n",
       "          -3.5297e-01,  1.8096e-01, -5.0733e-01, -2.1436e-01,  6.9160e-01,\n",
       "           3.2753e-01, -8.7155e-01,  3.8160e-02,  5.9753e-01, -5.1593e-01,\n",
       "           1.8231e-01,  1.0520e-02,  1.2073e-01, -3.5920e-02,  2.1339e-01,\n",
       "          -1.0837e-01,  3.6255e-01, -5.6160e-01, -4.2000e-01, -4.6695e-02,\n",
       "           5.9002e-01,  9.8017e-02,  2.4896e-01, -1.9997e-01,  3.5892e-01,\n",
       "          -2.7904e-01, -3.0208e-01, -2.4714e-01,  2.8074e-02, -2.1321e-01,\n",
       "           2.8785e-01, -1.3190e-01, -1.8593e-01,  2.6220e-01, -1.4457e-01,\n",
       "           5.4373e-01,  4.6093e-01,  6.4338e-01,  9.0278e-02, -1.7838e-01,\n",
       "           2.4395e-01,  6.3504e-02, -3.0764e-01,  5.1212e-01,  1.4861e-02,\n",
       "          -4.7605e-01,  1.3018e-01,  6.3130e-02,  5.3125e-02, -1.4310e-01,\n",
       "          -1.8210e-01,  2.8353e-01,  4.7433e-03, -3.4418e-01,  2.0436e-01,\n",
       "           3.2605e-01,  2.2278e-01, -5.5433e-01,  2.7091e-01,  3.4276e-01,\n",
       "          -1.6297e-02,  1.6658e-02, -1.3971e-01,  1.2699e-01, -6.8731e-01,\n",
       "           2.1912e-01, -7.5659e-01, -2.1822e-01,  2.2174e-01, -1.2384e-01,\n",
       "          -4.9953e-01,  3.5551e-02, -1.5131e-01,  3.5066e-03,  8.9862e-01,\n",
       "           8.9985e-01, -1.2377e-01, -4.9015e-01, -1.2054e-01,  2.6347e-01,\n",
       "          -5.6200e-02, -2.3647e-01,  2.6229e-01, -5.1455e-02, -8.3190e-02,\n",
       "          -1.2290e+00, -2.8710e-01,  2.0861e-01,  7.9545e-02, -1.8979e-01,\n",
       "           1.0801e+00, -2.4190e-01,  1.2983e-01,  6.1443e-01,  2.0776e-01,\n",
       "           3.0221e-01,  4.1798e-01,  1.9241e-02, -3.1289e-01, -1.1601e-01,\n",
       "           1.7249e-01, -6.4133e-02, -1.7759e-01, -5.8779e-01,  2.5980e-01,\n",
       "           7.9414e-02,  4.5242e-01,  6.2170e-01]], device='mps:0'),\n",
       " tensor([[ 2.0208e-02, -7.6503e-01, -5.5929e-01, -2.5692e-01, -5.6472e-01,\n",
       "           5.7194e-02, -2.9899e-01,  6.1910e-01,  5.2920e-03,  4.5897e-01,\n",
       "           4.0233e-01,  2.0171e-01,  1.7813e-02, -1.9294e-01, -5.8388e-01,\n",
       "          -3.3605e-01,  3.3281e-01,  4.2457e-01,  3.0132e-01, -6.8317e-02,\n",
       "           7.2122e-01,  4.3789e-01, -3.0936e-01, -1.1078e-02, -6.4128e-01,\n",
       "           3.0362e-01,  8.5775e-02,  1.2002e-01,  3.6188e-01, -4.3962e-01,\n",
       "          -4.3361e-01,  5.5624e-01, -4.7431e-02,  6.8908e-01, -3.6465e-01,\n",
       "          -1.6057e-01,  1.2906e-01, -2.5347e-01,  3.7705e-01,  1.3642e-01,\n",
       "           7.8618e-01, -1.3841e-01,  1.9413e-01,  3.8705e-01,  5.8566e-01,\n",
       "           5.5818e-02,  5.9252e-02, -9.8198e-01,  4.3724e-01, -4.2433e-01,\n",
       "          -8.7132e-02, -1.9352e-01,  2.9384e-01,  4.3466e-01,  4.1407e-02,\n",
       "          -3.8051e-01,  1.3777e-01,  2.5721e-01, -3.6372e-01,  1.6510e-01,\n",
       "          -7.3787e-02,  8.1903e-01, -1.2874e-01, -6.0702e-02, -5.3394e-02,\n",
       "           4.8926e-02, -1.0550e-01, -7.6464e-01, -3.0272e-01, -1.4867e-01,\n",
       "          -2.1655e-01,  1.6044e-01, -3.2349e-01, -1.7165e-01,  4.2270e-01,\n",
       "          -3.6935e-01,  1.4055e-01, -4.9128e-02, -4.9496e-01, -1.3233e-02,\n",
       "           4.0388e-01,  4.8051e-01,  3.8214e-02, -2.0929e-02, -4.3727e-01,\n",
       "          -6.0656e-02,  6.7806e-01,  4.1573e-01,  3.6109e-01, -6.7068e-03,\n",
       "           4.0578e-01,  1.5997e-01, -1.4478e-02, -3.7510e-01,  3.7012e-01,\n",
       "          -2.0713e-01,  3.9778e-01, -1.8583e-01,  6.5082e-02, -9.5916e-02,\n",
       "           7.1551e-02,  1.8106e-01,  1.3193e-01,  1.7613e-01, -5.0597e-01,\n",
       "          -6.4350e-01,  2.0210e-01, -4.2119e-02, -8.4742e-01, -8.1491e-02,\n",
       "          -2.3964e-01,  2.2045e-01,  3.4319e-01, -4.2882e-01, -1.0789e-01,\n",
       "           1.2355e-01,  3.2049e-01, -3.3026e-01, -5.3377e-01,  3.8314e-01,\n",
       "           2.0406e-01, -5.1428e-01,  1.4590e-01,  2.3334e-01,  3.6936e-01,\n",
       "          -5.1731e-02,  2.7681e-01, -3.6767e-01,  4.4541e-01,  4.9211e-01,\n",
       "          -2.3287e-03,  1.9927e-01, -2.3671e-01, -2.1461e-01, -2.8364e-02,\n",
       "          -1.9802e-02,  1.1374e-01, -1.8721e-01,  3.1578e-01,  7.9393e-02,\n",
       "          -1.7306e-01, -1.4675e-01, -4.4787e-01,  6.8730e-02, -8.9550e-02,\n",
       "          -7.4923e-03, -6.0417e-02,  1.3191e-01, -2.0985e-01, -3.4197e-01,\n",
       "          -8.3526e-01, -6.6114e-01,  2.9746e-01,  1.6620e-02, -2.6681e-02,\n",
       "          -8.1852e-03,  4.1030e-01,  1.2823e-01,  2.0095e-01,  4.6233e-01,\n",
       "           3.9786e-01, -1.6751e-01,  2.4818e-01, -5.1207e-01, -6.5876e-02,\n",
       "           3.7497e-01,  2.1087e-01, -2.8988e-01, -8.3998e-02,  2.4633e-01,\n",
       "           2.3304e-01, -4.3286e-01, -1.2071e-01, -7.9654e-01, -3.2950e-02,\n",
       "          -6.0529e-01,  2.8070e-01,  2.2814e-01, -4.3748e-01,  9.8900e-03,\n",
       "          -4.4120e-01, -2.0305e+00,  2.0706e-01, -2.1848e-01,  1.2624e-01,\n",
       "          -6.4365e-01,  3.0940e-01,  3.7443e-01,  3.4252e-01, -1.5505e-01,\n",
       "          -3.8439e-01, -5.8677e-03,  1.1463e-01,  6.6717e-01, -1.3443e-01,\n",
       "           3.3261e-01,  6.4918e-01,  1.0890e-02, -2.5317e-01, -7.8681e-03,\n",
       "          -3.3116e-01,  1.3954e-01, -1.1193e-01, -4.6732e-01,  1.5745e-02,\n",
       "           2.0312e-01,  1.5326e-01,  2.1860e-02,  4.0112e-01,  2.8193e-03,\n",
       "          -9.9428e-02, -3.1422e-01, -5.0077e-03, -4.3295e-01, -8.7508e-02,\n",
       "           3.7966e-01, -1.5067e-01, -1.0608e-01,  3.9688e-01,  1.2532e-01,\n",
       "          -1.3816e-01,  1.1189e-01, -1.4272e-01, -5.7000e-02, -1.6625e-01,\n",
       "           2.8259e-01, -1.1423e-01, -1.0816e-01, -2.9287e-01, -4.0375e-01,\n",
       "          -3.0225e-03, -2.9997e-01,  7.1459e-01, -5.2832e-02, -2.1241e-01,\n",
       "          -4.9738e-01, -2.7527e-01, -9.5959e-02, -4.1868e-02,  1.8233e-01,\n",
       "          -1.9273e-01, -4.2171e-01, -8.5361e-01, -8.9961e-02,  8.8740e-02,\n",
       "          -4.7861e-02, -2.6897e-01,  9.0623e-01,  1.7193e-01,  2.4860e-01,\n",
       "          -3.4133e-02,  3.6571e-02,  4.1085e-01,  2.5951e-02, -1.6019e-01,\n",
       "           1.0218e-01, -2.8330e-02, -8.7010e-02,  4.1328e-01, -2.0672e-01,\n",
       "           2.8802e-02, -1.2390e-01, -4.0113e-02,  5.2599e-01, -1.0086e-01,\n",
       "           2.7946e-01,  4.4659e-01, -4.4787e-01,  1.7640e-01,  3.1108e-01,\n",
       "          -2.8588e-01,  3.4071e-01,  2.1494e-02, -1.5333e-01, -7.8185e-01,\n",
       "          -2.5145e-01,  4.3187e-01, -1.5070e-01,  1.5915e-01, -7.8193e-01,\n",
       "          -2.8628e-02,  2.9829e-01, -9.5702e-02,  2.2307e-01, -3.3996e-01,\n",
       "           1.0336e-01,  2.7664e-01,  5.4592e-02, -1.8774e-01, -6.2797e-01,\n",
       "           2.4925e-02,  2.6298e-01, -3.6244e-01,  3.6625e-01, -2.3536e-02,\n",
       "          -2.7597e-01,  6.5678e-02,  3.4875e-01, -5.6993e-03, -3.0233e-01,\n",
       "          -1.8564e-01, -1.5399e-01, -4.3299e-01, -8.5505e-02, -1.9394e-01,\n",
       "           1.6030e-01, -1.8583e-02,  4.4168e-01, -7.7571e-01, -6.2004e-01,\n",
       "          -8.2860e-02,  6.8730e-01,  1.1407e-01, -2.6381e-01, -3.6458e-01,\n",
       "          -1.4420e+00, -2.3726e-01, -4.1471e-01, -1.6035e-01, -5.0801e-01,\n",
       "           2.4845e-01,  5.6050e-03, -7.0122e-02,  5.1329e-02, -1.9641e-01,\n",
       "           2.2804e-01,  4.2249e-01, -1.4574e-01, -5.9133e-02, -4.7953e-02,\n",
       "          -2.9291e-01,  5.4336e-02, -1.0803e-01, -2.3218e-01, -7.5940e-02,\n",
       "           8.5886e-02,  1.0303e-01, -2.5835e-01, -5.4286e-02, -3.2694e-01,\n",
       "           6.3005e-01, -9.8360e-02,  2.2665e-01,  3.2024e-02, -3.4603e-01,\n",
       "          -1.4111e-01,  5.2638e-01, -2.6100e-01, -9.5505e-02, -3.1647e-02,\n",
       "          -8.1588e-04, -3.7655e-01, -6.9738e-02, -5.0863e-02,  5.1015e-01,\n",
       "           3.5030e-01,  2.1533e-01, -3.5168e-01,  4.2025e-01,  6.7044e-01,\n",
       "          -1.8195e-01,  2.5602e-01,  2.4438e-02, -1.7030e-01,  8.4277e-03,\n",
       "           2.9253e-01,  1.8083e-01, -5.4752e-01,  9.4749e-02,  3.7275e-01,\n",
       "           2.9185e-01, -3.6818e-01, -6.6197e-02,  2.1315e-01,  2.7459e-01,\n",
       "          -3.4370e-01, -3.8059e-01,  3.9311e-01, -3.9890e-01, -3.4777e-01,\n",
       "           4.3958e-02,  9.1178e-02,  1.5579e-01,  7.0696e-02, -5.7060e-01,\n",
       "          -4.3794e-01,  2.4488e-01,  8.7743e-01,  1.2242e-01,  1.5704e-01,\n",
       "           3.1480e-01, -1.1315e-01, -2.4633e-01,  3.3256e-01, -1.6159e-01,\n",
       "          -7.7163e-02,  3.1051e-01, -4.6952e-01,  4.0089e-01,  3.4175e-01,\n",
       "          -1.6735e-01, -2.5179e-01, -1.0563e-01, -3.9040e-01, -1.1559e-01,\n",
       "          -2.1741e-01,  3.9607e-01,  4.9480e-02,  1.2715e-01, -8.5624e-02,\n",
       "           2.8878e-01,  1.4682e-01, -2.8650e-01,  8.1864e-02, -7.2852e-02,\n",
       "           1.0194e-01,  1.1781e-01, -8.8097e-02, -4.7460e-02, -5.2011e-02,\n",
       "           3.8932e-01,  1.0028e-01,  2.4817e-01,  2.0441e-01, -1.1597e+01,\n",
       "           6.8626e-02,  7.5693e-01, -4.7819e-01, -1.7697e-02,  4.0950e-02,\n",
       "           1.6838e-01, -2.1576e-01, -6.7861e-02, -2.9902e-01, -2.2121e-01,\n",
       "           1.6634e-02,  2.9037e-02,  1.5752e-01, -2.8156e-02, -9.0802e-02,\n",
       "          -1.4198e-01, -6.5549e-01,  5.0503e-01, -4.5816e-02,  1.5790e-01,\n",
       "          -2.2957e-01,  3.6382e-01, -5.2319e-02, -1.6315e-01, -1.3850e-01,\n",
       "          -3.3683e-01, -4.1339e-01,  2.7810e-01,  1.4482e-01, -1.1726e-01,\n",
       "          -3.9508e-02,  3.4742e-01,  1.4369e-01, -4.0219e-01, -5.6183e-01,\n",
       "           2.0400e-01, -1.8716e-02,  4.9096e-02,  4.0898e-02, -1.9458e-01,\n",
       "           1.6506e-01, -6.7758e-01,  3.8554e-01,  1.8458e-01,  7.8416e-01,\n",
       "           7.3742e-01,  7.0307e-01,  2.3267e-01, -4.3792e-02,  2.8482e-02,\n",
       "          -8.0305e-01,  1.6911e-01, -8.0406e-01,  1.9478e-01, -8.5476e-02,\n",
       "           4.9424e-01, -6.4510e-01, -6.0906e-02,  8.3130e-02, -3.2829e-01,\n",
       "           2.5847e-01, -7.8931e-02,  2.7519e-01, -2.7523e-01, -1.2579e-01,\n",
       "           3.5699e-01,  1.1202e-01,  2.9189e-01, -1.4989e-02,  1.9941e-01,\n",
       "          -2.1878e-01,  5.6106e-01,  2.6796e-01, -2.9849e-01,  1.1820e-01,\n",
       "           2.0249e-02,  7.7441e-02,  5.5225e-01, -8.4660e-02, -4.6368e-02,\n",
       "           7.2658e-02,  3.4458e-01, -2.6370e-01, -7.9845e-01, -8.8288e-02,\n",
       "           2.4457e-01, -7.1626e-02,  6.4396e-02,  5.0609e-01, -3.0357e-01,\n",
       "          -2.6946e-01, -4.2302e-01,  1.9476e-02,  1.8042e-01,  7.1453e-01,\n",
       "          -5.8034e-01, -1.0122e-01, -1.6808e-02, -8.1182e-02, -7.3097e-01,\n",
       "          -8.3426e-02, -1.9399e-01, -2.7188e-01,  2.1695e-01, -2.7433e-01,\n",
       "          -7.3079e-03,  4.6410e-02,  2.2299e-01,  3.8007e-01, -5.8152e-01,\n",
       "           5.3584e-01,  3.9769e-02,  1.8878e-01, -1.4119e-01,  1.2564e-01,\n",
       "           1.4194e-01,  5.4991e-01, -2.6273e-01, -5.8466e-01, -2.5381e-01,\n",
       "          -1.5310e-01,  3.0415e-01, -1.5312e-01, -3.1279e-01,  1.1880e-01,\n",
       "          -2.5092e-01, -3.4404e-03,  8.2193e-02,  7.3047e-02,  4.6305e-01,\n",
       "           8.4067e-02,  6.7405e-02,  6.4410e-02, -5.4696e-01, -8.2122e-01,\n",
       "          -5.6732e-01,  1.5895e-01,  1.4352e-01, -2.9913e-01, -1.5565e-01,\n",
       "          -8.1726e-02,  7.9281e-02, -4.6112e-01,  2.3830e-01,  3.3623e-02,\n",
       "           7.5235e-01,  4.5941e-01, -1.1610e-01, -1.2501e-01, -2.4152e-01,\n",
       "          -4.7697e-01, -2.4586e-01,  8.6539e-03,  3.4751e-02, -3.2841e-01,\n",
       "          -3.3981e-01,  2.1232e-01, -2.0264e-01,  9.3159e-02,  2.3010e-01,\n",
       "          -2.8576e-01, -3.4762e-01, -2.4335e-01, -1.8679e-02,  3.3081e-01,\n",
       "           1.3124e-01, -5.1549e-02, -3.5166e-01,  9.7943e-02,  3.3827e-01,\n",
       "           9.2638e-02, -2.7596e-01, -3.1185e-01, -1.7647e-01, -3.0932e-02,\n",
       "           2.4132e-01,  8.9103e-01, -2.2223e-01, -2.5462e-01,  2.4756e-01,\n",
       "           1.2559e-01, -1.6163e-01,  2.4328e-01, -2.1901e-01, -8.2403e-02,\n",
       "          -3.1274e-02, -6.0327e-01,  2.3566e-01,  1.2174e-02,  3.7033e-01,\n",
       "           1.0918e-02,  1.0173e-01,  4.5553e-02,  3.5172e-01, -6.4380e-02,\n",
       "          -1.3083e-01, -7.2966e-02, -5.9665e-01, -1.2796e-01, -1.2882e-02,\n",
       "           1.6433e-02,  1.0146e-02, -2.8245e-01, -2.4282e-01,  2.5145e-01,\n",
       "          -4.8975e-02,  1.0930e+00,  3.7531e-01,  4.5117e-02,  1.7778e-01,\n",
       "           1.8941e-01, -4.0156e-01,  2.1312e-01, -3.3692e-02,  2.1484e-01,\n",
       "           3.7423e-01,  2.5027e-01, -2.1613e-01,  8.1767e-02,  1.1973e-01,\n",
       "           3.6036e-01, -7.3182e-01, -6.4988e-01, -6.2273e-02,  4.1248e-01,\n",
       "          -8.2303e-02,  3.2196e-01,  2.8974e-02,  5.7252e-01,  6.7178e-01,\n",
       "          -1.1460e+00, -2.3901e-01, -2.7814e-01,  7.7240e-01, -2.6478e-02,\n",
       "          -4.1822e-01, -1.6920e-01, -5.3889e-01, -4.6541e-01,  3.1793e-01,\n",
       "           6.9987e-01, -1.0613e+00,  1.6571e-01,  1.2003e-01, -1.9229e-01,\n",
       "           5.4059e-02, -5.7781e-01,  3.3046e-01,  1.5885e-02,  2.1187e-01,\n",
       "          -9.0080e-02,  5.4108e-01, -4.1372e-01, -5.0692e-01,  1.5216e-01,\n",
       "           3.9384e-01,  4.3704e-02,  4.0912e-02,  2.8431e-02,  3.3223e-01,\n",
       "          -1.3847e-01, -1.8566e-01, -2.6675e-01,  3.5006e-01, -4.9763e-02,\n",
       "           2.1789e-01, -5.1372e-01, -7.2430e-02,  2.6108e-01, -3.8436e-02,\n",
       "           3.6857e-01,  6.1013e-01,  6.8394e-01, -9.4953e-02, -2.8380e-01,\n",
       "           4.8577e-01,  1.5567e-01, -7.3465e-02,  6.4399e-01, -1.2834e-01,\n",
       "          -4.3102e-01,  6.4595e-01,  2.9300e-02, -2.5687e-01,  4.9953e-02,\n",
       "          -1.7909e-01,  3.3106e-01,  2.3613e-01, -1.9502e-01,  9.3905e-02,\n",
       "           1.3858e-01,  1.2522e-01, -3.8871e-01,  8.9062e-02,  1.8340e-01,\n",
       "          -3.6279e-01,  7.6027e-01, -1.6448e-01,  5.0500e-01, -2.9734e-01,\n",
       "           1.8890e-01, -3.0684e-01, -4.4585e-01, -3.3397e-01, -2.8147e-01,\n",
       "          -2.8851e-01, -1.6761e-01,  1.7526e-01, -9.1573e-02,  7.4284e-01,\n",
       "           3.8549e-01, -4.0562e-01, -3.0646e-01, -2.1099e-01,  1.9009e-01,\n",
       "           1.6331e-01, -1.7484e-01,  7.9365e-02,  2.1902e-01,  1.8608e-01,\n",
       "          -9.0702e-01, -4.3824e-01,  3.0154e-01,  3.2570e-01, -3.4959e-01,\n",
       "           5.0241e-01, -9.3615e-02,  1.9549e-01,  1.3200e-01,  1.4179e-01,\n",
       "           1.4406e-01,  5.4799e-01,  2.3142e-01, -3.4907e-01,  9.9233e-02,\n",
       "           1.8399e-01, -4.6496e-01,  7.9736e-02, -5.1373e-01,  2.3767e-01,\n",
       "           2.7593e-01,  8.8341e-01,  5.3475e-01]], device='mps:0'),\n",
       " tensor([[-0.6229, -0.6194, -0.4528,  ...,  0.1802,  0.5200,  0.3214],\n",
       "         [-0.3636, -0.5764, -0.4146,  ...,  0.1919,  0.2702,  0.4493]],\n",
       "        device='mps:0'),\n",
       " tensor([[-2.0621e-01, -5.7137e-01, -6.2915e-01, -4.3867e-01, -1.5991e-01,\n",
       "          -8.7672e-02, -6.3202e-01,  5.5693e-01,  8.9707e-04,  6.3299e-01,\n",
       "           3.7152e-01,  1.6441e-01,  1.5376e-01,  1.0229e-01, -8.3722e-01,\n",
       "          -4.7253e-01,  5.2355e-01,  3.5765e-01,  4.1541e-01, -7.7940e-02,\n",
       "           8.8901e-01,  3.4034e-01, -2.5075e-01,  2.1937e-01, -2.9100e-01,\n",
       "           1.7211e-01,  1.8892e-01,  1.6456e-01,  4.4567e-01, -5.2141e-01,\n",
       "          -1.1330e-01,  3.5338e-01, -2.1562e-02,  9.9694e-01, -3.6815e-02,\n",
       "          -1.3158e-01,  6.1652e-01, -9.8846e-02,  4.0152e-01,  4.5394e-01,\n",
       "           3.0653e-01, -1.6297e-01,  3.5753e-02, -2.3295e-03,  1.4073e-01,\n",
       "          -9.2243e-02, -7.1635e-02, -8.0155e-01,  4.4906e-01, -2.0740e-01,\n",
       "          -5.0871e-01, -3.5110e-01,  4.4046e-01,  4.2005e-01, -3.1275e-01,\n",
       "          -2.7520e-01,  3.2524e-01,  2.4472e-01, -4.4691e-01,  3.1896e-01,\n",
       "          -1.2489e-02,  6.2054e-01,  2.4347e-01, -2.0812e-01,  4.6700e-02,\n",
       "           2.2054e-01, -5.5409e-01, -6.0809e-01, -5.8930e-02, -9.2606e-03,\n",
       "          -2.1277e-03, -1.4346e-01, -3.8258e-01, -1.4686e-01,  3.0679e-01,\n",
       "           1.0739e-01,  1.6330e-01,  3.6416e-01, -7.6929e-01,  6.0504e-02,\n",
       "           2.9581e-01,  3.5316e-01, -1.1339e-01, -2.8185e-01, -3.0759e-02,\n",
       "           1.9305e-01,  4.7872e-01,  2.8598e-01,  1.4028e-01, -5.1633e-02,\n",
       "           3.5660e-01,  2.5966e-01,  6.4144e-02, -4.3096e-01,  1.3668e-01,\n",
       "          -1.6119e-01,  3.6609e-01, -2.3512e-01, -4.7580e-02,  4.4678e-02,\n",
       "           1.4386e-01,  1.7705e-01,  5.7727e-02, -1.1825e-01,  7.5857e-02,\n",
       "          -7.2013e-01,  1.7687e-01, -1.5925e-01, -5.9358e-01,  8.8880e-02,\n",
       "          -1.1522e-01,  1.4580e-01,  8.2967e-02, -2.8434e-01, -3.1184e-01,\n",
       "           2.5444e-01,  1.4357e-02, -2.7062e-01, -4.4817e-01,  4.4932e-01,\n",
       "           7.4678e-02, -3.7709e-01, -4.8745e-01, -8.9238e-02,  5.3051e-01,\n",
       "          -1.9290e-01,  5.0163e-01, -2.1543e-01,  4.8382e-01,  9.3619e-01,\n",
       "          -7.4738e-02, -7.2443e-02, -1.6712e-01, -3.7573e-01, -2.6686e-01,\n",
       "           7.3128e-02,  2.0849e-01, -4.3482e-01,  2.2669e-01, -3.1137e-02,\n",
       "          -1.3061e-01, -2.7779e-01, -7.3257e-01, -2.9015e-01, -7.5447e-02,\n",
       "          -3.8020e-01, -1.0147e-01,  2.1837e-01, -3.5400e-01, -5.2369e-01,\n",
       "          -8.7632e-01, -6.5779e-01,  6.1005e-01,  1.1153e-01, -1.2201e-01,\n",
       "          -1.9591e-01,  4.4028e-02,  3.4839e-01, -1.6302e-01,  4.7972e-01,\n",
       "           4.7082e-01,  1.5815e-01,  2.1960e-01, -5.3037e-01,  3.1778e-02,\n",
       "           2.8311e-01, -3.1094e-01, -1.5754e-01, -1.6669e-01,  8.3164e-02,\n",
       "           3.2323e-01, -4.4205e-01, -3.6348e-02, -4.3047e-01, -1.6739e-02,\n",
       "          -5.8021e-01,  4.7504e-01,  2.8052e-01, -2.6933e-01, -1.4995e-01,\n",
       "          -3.3598e-01, -2.0309e+00,  3.3554e-01, -7.6949e-01, -1.2395e-01,\n",
       "          -7.5607e-01,  1.4436e-01,  2.0389e-01,  1.4617e-01, -2.4722e-02,\n",
       "          -1.6815e-01, -8.5462e-02, -2.0010e-01,  3.5571e-01, -2.1715e-02,\n",
       "           4.7750e-01,  5.3291e-01, -2.3525e-01, -2.9038e-01,  6.1370e-02,\n",
       "          -2.2946e-01,  2.8008e-01, -3.9044e-02,  2.4983e-02,  4.2526e-01,\n",
       "           2.7171e-01, -1.5105e-01,  4.4653e-02,  4.9498e-01, -2.1246e-01,\n",
       "          -1.7575e-02, -2.4634e-01, -3.0120e-01, -1.2207e-01, -1.4546e-01,\n",
       "           3.8127e-01, -1.9395e-01,  4.0661e-02,  3.1422e-01,  2.8792e-01,\n",
       "          -8.9038e-02,  2.0295e-01, -1.1464e-01, -4.0706e-01, -1.6857e-01,\n",
       "           6.7647e-01, -4.2159e-01,  6.7670e-02, -2.3589e-01, -1.1826e-01,\n",
       "           3.3354e-02, -5.2021e-01,  5.2372e-01,  2.8879e-01, -5.4850e-02,\n",
       "          -2.4921e-01, -1.8944e-01,  2.7648e-01, -7.2881e-02, -7.1204e-03,\n",
       "          -4.8506e-01, -3.9821e-01, -3.9179e-01, -3.9647e-02,  4.5534e-01,\n",
       "           2.2949e-02, -2.2098e-01,  1.0941e+00,  5.0953e-01,  6.0600e-01,\n",
       "           3.7897e-02,  1.1530e-01,  4.9502e-01,  1.5710e-01, -3.9547e-01,\n",
       "          -2.2632e-01,  7.4666e-02, -2.7712e-01,  4.5764e-01, -4.1547e-01,\n",
       "           2.3447e-01,  1.0714e-01,  3.4805e-01,  7.5241e-01,  9.2606e-02,\n",
       "           6.6595e-02, -1.7748e-01, -2.5724e-01, -2.9808e-01,  2.9295e-01,\n",
       "          -3.4289e-01,  1.2981e-01, -8.7935e-02, -1.7353e-01, -6.8799e-01,\n",
       "          -2.1440e-01,  1.8097e-01, -3.4891e-01,  3.9232e-01, -1.0004e+00,\n",
       "          -2.8831e-01,  8.3423e-01, -2.9479e-01,  4.7938e-01,  2.0652e-02,\n",
       "           5.1265e-01,  3.5804e-01, -4.1528e-02, -4.7010e-01, -3.1204e-01,\n",
       "          -1.1946e-01,  4.2331e-01, -2.7721e-01,  1.5325e-01,  9.7147e-02,\n",
       "          -3.0090e-02,  2.8838e-01, -9.6327e-03, -9.6430e-02, -5.4535e-01,\n",
       "          -4.5189e-02, -1.7355e-01, -2.6512e-01, -3.5364e-01, -8.7946e-02,\n",
       "          -1.5084e-01,  1.4336e-01,  6.3269e-01,  6.4180e-02, -2.5521e-01,\n",
       "           5.3850e-02,  6.0500e-01,  8.0084e-02,  4.8309e-01, -2.4657e-01,\n",
       "          -1.3686e+00, -3.2682e-01, -2.0209e-01, -8.6955e-02, -3.0507e-01,\n",
       "           3.3712e-01,  1.8798e-01, -1.7488e-01, -3.2409e-01, -4.3539e-01,\n",
       "           3.7533e-01, -1.3736e-01,  4.6609e-01,  1.1435e-01,  4.1119e-02,\n",
       "          -2.7175e-01, -3.8649e-01, -5.9549e-02, -1.5741e-01,  1.9218e-02,\n",
       "           1.6505e-01,  1.1748e-01,  6.1846e-03,  3.3485e-01, -3.2696e-01,\n",
       "           7.4918e-01, -3.8158e-01, -2.5432e-01,  2.8698e-01, -4.2666e-01,\n",
       "          -3.4226e-01,  5.3952e-01, -2.8773e-02,  1.7192e-01, -7.1106e-03,\n",
       "           3.6173e-01, -4.9629e-01, -3.1239e-01, -8.0784e-02,  4.4670e-01,\n",
       "           1.1824e-01,  3.9974e-01, -3.4930e-01,  2.7421e-01,  5.0808e-01,\n",
       "          -4.1204e-01,  2.6549e-01, -3.2863e-01, -3.9252e-01, -1.0910e-01,\n",
       "           7.3941e-01,  2.0904e-01, -2.5060e-01, -7.1860e-02,  4.6602e-01,\n",
       "           2.9086e-01, -3.6644e-02, -5.5191e-02,  2.5574e-01,  2.5834e-01,\n",
       "          -1.6220e-01, -4.3542e-01,  3.3377e-01, -3.1665e-01, -5.4469e-01,\n",
       "           5.6553e-02, -1.7542e-01,  9.4532e-02, -9.5221e-03, -6.0573e-01,\n",
       "          -2.9037e-01,  1.7650e-01,  5.9473e-01,  8.2324e-02, -1.3067e-02,\n",
       "           5.1580e-03, -7.8051e-02, -4.8386e-01,  2.1473e-02, -6.7544e-02,\n",
       "           1.8360e-01,  2.9636e-01, -3.6820e-01,  2.1798e-01,  1.0514e-01,\n",
       "          -3.1619e-01,  3.0222e-01,  9.1399e-02,  9.6512e-01, -4.7807e-01,\n",
       "          -1.8870e-01,  1.3744e-01,  1.9660e-01, -1.1393e-02,  1.2920e-01,\n",
       "          -3.3354e-01,  5.9087e-01, -6.1155e-01, -1.9941e-01,  4.8241e-01,\n",
       "           3.5514e-02, -3.0046e-01,  1.2219e-01, -3.4168e-01, -3.8422e-01,\n",
       "           2.1955e-01, -3.7783e-02, -1.3261e-01,  2.4311e-01, -1.1180e+01,\n",
       "           3.4021e-02,  6.4462e-01, -5.1892e-01, -1.3317e-02,  2.3164e-02,\n",
       "           4.5880e-02, -6.6913e-02,  2.3181e-01, -3.7454e-01, -3.3775e-01,\n",
       "          -2.0605e-01, -3.4887e-01,  9.1274e-02, -2.4786e-02, -8.1344e-02,\n",
       "          -1.0752e-02, -8.1515e-01,  2.0446e-01, -8.2931e-02,  3.5919e-01,\n",
       "          -5.6347e-01,  3.2134e-01,  4.2873e-02, -1.8498e-01, -3.9438e-01,\n",
       "           6.6445e-02, -5.0624e-01,  4.4843e-02,  1.5745e-01, -1.8973e-01,\n",
       "          -2.9145e-03, -8.7508e-02,  3.8003e-01, -1.8306e-01, -7.7618e-01,\n",
       "           4.1628e-02,  1.6466e-01,  4.6982e-01,  9.0650e-02, -6.9704e-01,\n",
       "           1.8048e-01, -6.1014e-01,  2.4603e-01,  2.4031e-01,  6.7667e-01,\n",
       "           9.4040e-01,  6.8637e-01,  1.3076e-01,  4.5343e-02,  2.7694e-02,\n",
       "          -6.5202e-01,  3.0345e-01, -6.0078e-01, -9.7342e-02, -1.1828e-01,\n",
       "           4.4198e-01, -4.6662e-01,  1.3110e-01,  1.9679e-02, -1.9915e-01,\n",
       "           1.7752e-01,  1.1272e-01,  1.9861e-01, -4.3576e-01, -4.7354e-01,\n",
       "           8.3872e-01,  1.2153e-01,  1.6601e-01, -4.9381e-01,  9.1408e-02,\n",
       "          -2.5924e-01,  2.7746e-01,  7.6592e-02, -6.5639e-01,  9.1543e-02,\n",
       "          -8.0135e-03, -5.0955e-02,  9.1479e-01, -1.1118e-01,  1.8472e-01,\n",
       "           1.3166e-02,  3.6723e-01, -4.0401e-01, -4.7507e-01, -1.4926e-01,\n",
       "           2.9220e-02,  9.3471e-02,  4.6400e-01,  4.7931e-01, -2.6676e-01,\n",
       "          -6.1721e-01,  8.5166e-02,  7.0610e-02,  3.5640e-02,  1.0173e+00,\n",
       "          -5.2474e-01, -3.3819e-01,  6.0234e-02, -1.9682e-01, -4.8811e-01,\n",
       "          -1.1975e-01, -1.4686e-01, -6.7543e-01,  5.1813e-01, -2.2595e-01,\n",
       "           3.1658e-01,  1.4935e-01,  2.2202e-01,  3.7306e-01, -4.8773e-01,\n",
       "           3.7704e-01, -1.7054e-01,  4.7082e-01, -3.3230e-01, -7.2949e-02,\n",
       "          -8.7939e-03,  5.5455e-01, -4.0307e-01, -5.8894e-01, -2.3060e-01,\n",
       "          -1.5029e-01,  2.6884e-01,  1.4510e-01, -6.3337e-01, -1.4781e-01,\n",
       "          -6.4853e-01,  8.9254e-02, -4.9589e-02,  1.1413e-01,  7.6858e-01,\n",
       "          -2.9678e-02, -2.1556e-02, -2.2875e-01, -5.3220e-01, -2.1202e-01,\n",
       "          -3.2837e-01, -7.0020e-02,  9.1106e-02, -7.5131e-01, -6.4266e-02,\n",
       "          -1.2186e-02,  1.2916e-01, -3.1629e-01,  1.3862e-01,  5.6200e-02,\n",
       "           2.9612e-01,  5.5032e-01, -8.9695e-02,  2.8337e-01, -2.5904e-01,\n",
       "          -5.0918e-01, -1.3999e-01, -1.7362e-01,  1.2615e-02, -8.0753e-01,\n",
       "          -3.5573e-01,  2.4046e-01, -3.5908e-01,  2.5691e-01, -2.3134e-01,\n",
       "          -8.9706e-02, -2.5833e-01, -9.6477e-02,  1.0996e-01,  9.6358e-02,\n",
       "           2.9147e-01, -3.1091e-01, -2.9163e-01,  1.8577e-01,  3.5749e-01,\n",
       "           1.1471e-01, -3.2981e-01, -3.4340e-01, -8.1724e-01,  2.4787e-01,\n",
       "           2.1635e-01,  8.7981e-01, -5.7215e-01, -4.1900e-01,  4.2885e-01,\n",
       "           3.8288e-01, -4.1197e-02, -7.1700e-02, -1.5921e-01,  1.3906e-01,\n",
       "          -1.6050e-01,  5.2240e-02,  2.8492e-01, -3.7961e-01,  1.6191e-01,\n",
       "           5.3604e-02,  3.2853e-01,  6.3804e-02,  2.3148e-01, -1.7644e-01,\n",
       "          -4.7095e-02,  4.2494e-02, -8.7276e-01, -1.9307e-01,  2.1445e-01,\n",
       "          -3.4023e-02,  2.0492e-01, -1.7476e-01, -2.8491e-01,  1.4612e-01,\n",
       "           4.9053e-01,  8.9714e-01,  3.9195e-01, -9.6860e-02,  5.1120e-01,\n",
       "           3.3273e-01, -3.1883e-01,  7.1502e-02,  1.3133e-01,  1.7060e-01,\n",
       "           3.7521e-01,  2.2402e-01, -1.7122e-01,  5.2330e-02, -6.5798e-03,\n",
       "           4.3948e-01, -8.9986e-01, -6.5001e-01, -1.1700e-01,  5.1502e-01,\n",
       "          -2.5728e-01,  6.0001e-02, -2.4565e-01,  4.5911e-01,  7.3565e-01,\n",
       "          -1.2499e+00, -2.3211e-01,  2.0592e-01,  1.6234e-01, -7.0392e-02,\n",
       "          -6.8694e-01,  1.9610e-02, -2.0144e-01, -4.1124e-01,  5.5268e-01,\n",
       "           6.0132e-01, -8.8309e-01,  3.1336e-01,  9.3604e-02, -3.5259e-01,\n",
       "          -7.7943e-02,  7.3349e-02,  1.8296e-01,  2.3476e-01, -5.6140e-02,\n",
       "          -2.4453e-01,  5.2788e-01, -4.1738e-01, -1.6970e-01,  1.6242e-01,\n",
       "           4.0994e-01,  3.0723e-02,  4.4637e-01, -4.6596e-02, -3.5829e-02,\n",
       "          -2.0185e-01, -3.0776e-01, -2.3067e-01, -4.9253e-04, -9.5742e-02,\n",
       "           2.6567e-01, -4.3080e-01, -7.0518e-02,  2.8632e-01, -1.3435e-01,\n",
       "           5.2962e-01,  4.4241e-01,  3.1688e-01, -3.2593e-02, -4.3835e-02,\n",
       "          -1.3256e-01,  2.4849e-01, -2.1036e-01,  4.3722e-01, -3.4079e-01,\n",
       "          -3.3039e-01,  4.1727e-01,  6.4416e-02, -8.1840e-03,  8.1987e-02,\n",
       "          -5.3495e-01,  4.1201e-02,  1.9568e-01, -5.5719e-01,  4.0812e-01,\n",
       "           4.5262e-01,  2.4578e-01, -3.9911e-01,  9.2810e-02,  2.4223e-01,\n",
       "          -2.3519e-01,  8.3257e-01, -8.0967e-02,  1.5931e-01, -1.3214e-01,\n",
       "           4.7725e-02, -3.9148e-01, -3.6870e-01,  2.1105e-02,  1.1192e-01,\n",
       "          -2.9995e-01,  1.5045e-02,  1.6713e-01, -1.5363e-01,  9.2632e-01,\n",
       "           9.1355e-01, -1.3218e-01, -3.8884e-01, -2.5753e-01,  4.0019e-02,\n",
       "          -2.1960e-01, -2.8121e-01,  1.2682e-01,  3.7017e-01, -3.9824e-02,\n",
       "          -1.1699e+00, -3.9359e-01,  6.8521e-04,  8.9965e-02, -3.6688e-01,\n",
       "           1.0298e+00,  2.1244e-01,  4.1909e-02,  6.1522e-01,  1.7390e-01,\n",
       "           3.9107e-01,  4.8973e-01,  2.0192e-02, -3.2765e-01,  4.6539e-02,\n",
       "          -2.7104e-02, -9.7088e-03, -8.2825e-02, -5.8713e-01,  3.2536e-01,\n",
       "           9.5732e-02,  7.1136e-01,  6.5662e-01]], device='mps:0'),\n",
       " tensor([[-5.4527e-01, -4.4131e-01, -2.7396e-01, -9.8075e-01, -1.1024e-01,\n",
       "          -2.3092e-01, -7.2428e-01,  8.0590e-01, -2.5409e-01,  2.3278e-01,\n",
       "           2.2290e-01,  3.5197e-02,  2.4386e-01,  2.3296e-01, -9.5424e-01,\n",
       "           1.5731e-01,  3.5056e-01,  3.1415e-01,  4.1496e-01,  1.9795e-01,\n",
       "           9.0002e-01,  3.0432e-01,  2.0148e-01, -6.8150e-02, -1.0125e-01,\n",
       "           5.7718e-01, -1.9963e-01,  1.5112e-01,  3.3345e-01, -5.2790e-01,\n",
       "          -1.0028e-01,  1.2088e-01, -5.1711e-01,  1.0968e+00, -7.3984e-03,\n",
       "          -2.3877e-01,  3.8329e-02, -4.2081e-01,  6.7213e-01,  2.8478e-02,\n",
       "           1.2509e-01, -5.6996e-02, -3.5589e-01,  3.3372e-01, -9.7158e-02,\n",
       "          -4.8111e-02,  2.7652e-02, -5.7987e-01,  1.3247e-01,  9.8887e-02,\n",
       "           1.0824e-01, -1.6810e-01,  1.1544e-01,  4.3622e-01, -9.7286e-01,\n",
       "          -3.3127e-01,  1.8952e-01,  2.4987e-01, -5.4299e-01,  2.1426e-02,\n",
       "          -9.3018e-02,  7.6667e-01,  2.2257e-01, -2.2242e-01,  1.9997e-01,\n",
       "          -3.8862e-01, -7.1487e-01, -5.1690e-01, -2.5910e-01, -1.2801e-01,\n",
       "           2.3244e-02,  1.2235e-01,  1.8504e-01, -3.0720e-02,  3.0596e-01,\n",
       "           2.8386e-01,  1.5245e-01, -2.4618e-01, -4.1074e-02, -2.8896e-01,\n",
       "           2.6617e-01,  2.0004e-03,  8.2636e-02,  2.2227e-01, -3.6064e-01,\n",
       "          -1.9290e-01,  3.3327e-01,  9.0387e-02,  2.7534e-01, -7.4818e-02,\n",
       "           4.4459e-01, -6.8826e-04,  1.5031e-01, -2.8186e-01,  5.9230e-01,\n",
       "          -3.2333e-01,  2.7051e-01, -1.5943e-01,  4.9616e-02, -2.9940e-01,\n",
       "          -1.4591e-01,  2.6603e-01, -9.7072e-02,  2.9476e-01, -3.1643e-01,\n",
       "          -1.0062e-01,  2.2621e-01,  3.1685e-01, -6.7402e-01,  6.1970e-03,\n",
       "          -1.1806e-01, -5.4711e-01,  7.5615e-02, -6.8162e-01, -6.0684e-01,\n",
       "           4.1592e-01, -1.4142e-01, -3.2279e-01, -3.0242e-01,  4.2448e-01,\n",
       "           3.7692e-01, -6.6306e-02,  1.0946e-02, -3.7419e-01,  1.2116e-01,\n",
       "          -2.4125e-01,  1.2217e-01, -1.1078e-01,  2.7710e-01,  5.4802e-01,\n",
       "           2.8902e-01, -5.9938e-02, -1.4882e-01,  2.6104e-01, -3.4667e-02,\n",
       "           3.9374e-02, -1.4188e-01, -1.1138e-01,  1.4104e-02,  7.0713e-02,\n",
       "          -3.5502e-01, -3.0308e-01,  1.8502e-01, -1.0941e-02, -1.1260e-01,\n",
       "          -1.8103e-01,  1.8956e-01,  2.1591e-01, -1.5030e-01, -3.9936e-01,\n",
       "          -7.0449e-01, -2.9819e-01,  6.5892e-02, -1.0343e-02, -2.7287e-02,\n",
       "          -2.6855e-01,  3.5972e-01,  1.6501e-02,  3.7660e-01,  2.4587e-01,\n",
       "           6.4226e-01, -2.6753e-01,  2.0340e-01, -5.1015e-01,  3.8892e-01,\n",
       "          -3.8784e-02, -5.6036e-01, -4.1771e-01,  6.4538e-02,  1.1477e-01,\n",
       "           6.0306e-01, -6.7005e-02, -4.6359e-02, -6.9096e-02, -4.0158e-01,\n",
       "          -3.2941e-03,  7.0877e-02, -1.6819e-01, -1.0370e-01,  5.7900e-01,\n",
       "          -5.0799e-01, -1.9350e+00,  1.1431e-01, -1.3026e-01, -3.8827e-01,\n",
       "          -2.8690e-01,  2.1729e-01,  4.5948e-02,  4.7488e-01,  1.4568e-01,\n",
       "          -5.2361e-02, -1.2069e-01,  1.7196e-01,  4.5285e-01,  3.6641e-03,\n",
       "           4.2472e-01,  5.2883e-01,  1.3924e-01, -3.2582e-01, -6.5960e-02,\n",
       "          -2.8838e-01,  2.4090e-01, -1.4196e-02,  7.2845e-01,  4.1173e-04,\n",
       "           4.9061e-02,  2.4455e-01, -1.8926e-01,  5.3842e-01,  4.2376e-01,\n",
       "          -1.6910e-01, -7.1277e-02, -4.4601e-01,  5.1923e-02,  2.5925e-02,\n",
       "          -1.1426e-01, -3.8588e-01,  2.5589e-01,  5.1378e-01, -2.2272e-03,\n",
       "          -2.0681e-01,  6.5299e-02, -1.8752e-01, -2.9903e-01, -1.0243e-01,\n",
       "           5.9405e-01,  1.6364e-01, -1.4493e-01,  6.2726e-02, -2.7265e-01,\n",
       "          -8.2879e-02, -4.0184e-01,  6.0260e-01,  4.7775e-01, -7.2975e-02,\n",
       "           2.2162e-02, -8.2710e-01,  2.5612e-01, -2.6893e-01, -5.9709e-02,\n",
       "          -1.0372e-01, -2.8025e-01, -4.9081e-01, -1.7191e-02,  1.5653e-01,\n",
       "           3.5157e-01,  2.3852e-01,  4.9062e-01,  1.0658e+00,  3.3123e-01,\n",
       "           8.3917e-02,  5.2645e-01,  4.3535e-01,  4.5524e-01, -3.2655e-01,\n",
       "           4.3609e-01,  2.0719e-02, -6.5487e-01,  2.8590e-01, -3.8681e-01,\n",
       "           2.4557e-03,  3.5150e-02, -1.7021e-01,  3.6767e-01, -1.3971e-01,\n",
       "           2.9123e-01,  2.5629e-01,  8.2587e-02, -1.4736e-02,  5.0110e-01,\n",
       "          -3.5831e-01,  5.8850e-01, -2.0249e-01, -1.2507e-01, -8.7763e-01,\n",
       "          -7.0394e-01, -8.1322e-02,  4.0948e-01, -1.9420e-01, -5.1456e-01,\n",
       "           9.5588e-02, -3.8584e-01, -9.2144e-02, -2.0299e-01,  9.8581e-02,\n",
       "           2.3135e-01,  3.2924e-02, -4.8616e-02, -1.1799e-01, -3.9347e-01,\n",
       "          -3.5990e-01, -5.6514e-03, -3.9555e-01, -8.6412e-02,  4.9383e-01,\n",
       "          -8.3917e-02,  5.1654e-02,  3.6500e-01, -1.1066e-01, -3.2894e-01,\n",
       "          -1.5522e-01, -7.4742e-02,  2.0169e-01, -5.5709e-02,  2.6692e-01,\n",
       "           4.9539e-01, -1.5881e-02,  6.4642e-01,  6.7373e-03, -4.4718e-01,\n",
       "          -1.6672e-01,  5.6296e-01, -1.7586e-01,  1.9800e-02, -1.3797e-01,\n",
       "          -7.6351e-01, -1.3774e-01, -2.1185e-01, -3.9112e-01, -5.7677e-01,\n",
       "           8.2021e-02,  1.0683e-01,  1.3434e-02,  5.8237e-02, -1.9462e-01,\n",
       "           7.3022e-01,  5.0905e-01, -2.5929e-01, -2.2015e-01, -2.9730e-01,\n",
       "           2.4933e-01, -7.8248e-02, -4.3210e-01, -1.3868e-01,  8.6722e-02,\n",
       "          -1.0360e-01,  1.6386e-01,  4.8609e-03,  2.8168e-01, -3.0205e-01,\n",
       "           5.1132e-01,  9.2969e-02,  8.4555e-02,  4.6938e-01,  9.5302e-02,\n",
       "           1.7058e-01,  3.6953e-01, -4.8372e-01,  1.7663e-01,  3.8623e-01,\n",
       "           3.7631e-01, -9.2477e-01, -5.8362e-01, -4.4542e-02,  2.7047e-01,\n",
       "           3.9328e-01,  3.1202e-01, -1.1433e-01,  1.1079e-01,  1.4732e-01,\n",
       "          -2.5455e-01,  4.2448e-01, -1.1975e-01, -4.8425e-01,  4.1937e-01,\n",
       "           1.9924e-01,  3.6625e-01, -3.4080e-01, -2.4640e-01, -1.2274e-01,\n",
       "           4.7740e-01, -4.1974e-01,  1.5585e-01,  4.4053e-01,  5.0509e-01,\n",
       "          -1.5729e-01, -6.6680e-01, -3.7414e-02, -4.1320e-01, -2.9367e-01,\n",
       "           3.4595e-01, -3.0000e-01,  6.0727e-02, -2.2482e-01,  1.7400e-01,\n",
       "           3.0017e-02, -9.2386e-02,  5.4293e-01,  2.1632e-01, -2.2151e-01,\n",
       "          -6.4640e-02, -1.9233e-01, -1.3852e-01,  6.2262e-01, -3.5634e-01,\n",
       "          -4.7393e-01,  2.9345e-02, -1.4016e-01, -2.7269e-01,  2.6042e-01,\n",
       "          -3.6236e-01, -3.0109e-01,  1.4662e-01, -2.7930e-01, -1.2168e-01,\n",
       "          -4.2597e-01,  1.2990e-01, -2.7801e-02, -3.6857e-02, -6.8168e-02,\n",
       "          -3.8283e-01,  5.3688e-02,  7.6406e-02,  3.5976e-02,  2.1058e-01,\n",
       "           4.5781e-01, -2.5251e-01,  1.9638e-01, -2.3050e-01, -2.9992e-01,\n",
       "           5.8389e-01, -1.1811e-02, -4.8183e-01,  5.8605e-01, -1.1479e+01,\n",
       "           1.2233e-01,  6.5389e-01, -4.7776e-01,  5.7646e-03, -2.3036e-01,\n",
       "           1.9781e-01, -3.1888e-01,  2.2916e-01,  4.5645e-02, -2.6217e-01,\n",
       "          -8.3496e-02, -2.4072e-01, -1.8086e-01, -5.1571e-01, -2.1556e-01,\n",
       "          -2.6995e-01, -6.1633e-01,  1.5954e-01, -5.3584e-02,  2.3661e-01,\n",
       "          -5.9898e-01, -1.6195e-02,  3.9019e-01,  6.8714e-02, -3.9913e-01,\n",
       "          -1.4899e-01, -2.1712e-01,  2.4520e-01,  1.2822e-02, -6.1840e-02,\n",
       "           6.1803e-02, -4.8448e-02,  2.8306e-01, -5.9973e-02, -4.7053e-01,\n",
       "          -3.9133e-01,  2.0440e-01, -1.5224e-01, -1.5422e-01, -1.2846e-01,\n",
       "          -4.2505e-02, -2.2987e-01,  2.3518e-01,  4.0399e-01,  4.8863e-01,\n",
       "           8.8350e-01,  4.4354e-01,  7.9816e-02, -6.9014e-02, -9.9880e-02,\n",
       "          -9.9711e-01,  5.9697e-01, -5.3069e-01,  1.7796e-02, -1.9615e-01,\n",
       "           6.9467e-01, -6.2473e-01,  4.7219e-01, -4.1632e-01, -1.5806e-01,\n",
       "           2.7401e-01,  2.2488e-01, -2.4046e-02, -2.6533e-02, -3.3758e-01,\n",
       "           5.3398e-01, -3.5866e-01, -8.9340e-02, -2.2891e-01,  4.3792e-01,\n",
       "          -3.7955e-01,  4.4072e-01,  4.5447e-02, -9.0939e-02,  5.5396e-02,\n",
       "          -4.5427e-02,  1.6694e-01,  1.3374e+00,  1.2713e-01,  4.4878e-01,\n",
       "           2.4111e-01,  3.7471e-01,  6.2103e-02, -1.0140e+00,  2.0556e-01,\n",
       "           1.0007e-01,  1.2559e-01,  4.9109e-02,  5.5537e-01, -1.0167e-01,\n",
       "          -4.6386e-01, -2.4542e-01, -2.5646e-01,  4.1441e-01,  6.5707e-01,\n",
       "          -3.1670e-01, -1.8193e-01, -1.0470e-01, -7.4587e-02, -3.8196e-01,\n",
       "           1.4356e-01, -1.0076e-01, -5.1896e-01,  8.3327e-03, -5.4559e-01,\n",
       "           1.5940e-01, -2.9075e-02,  7.7057e-02,  4.5584e-01, -3.5257e-01,\n",
       "          -9.3890e-02, -1.4863e-01,  3.2208e-02, -4.5811e-01, -3.1350e-01,\n",
       "          -1.1126e-01,  4.2364e-01, -1.5442e-01, -3.2924e-01, -2.5494e-02,\n",
       "          -3.8654e-01,  2.9519e-01, -4.0350e-02, -7.0828e-01,  2.3569e-01,\n",
       "          -7.6239e-01,  2.2593e-01,  1.7590e-01, -2.0827e-01,  4.6386e-01,\n",
       "          -4.4792e-02, -2.8090e-01,  1.5243e-01, -2.9472e-01, -7.2246e-01,\n",
       "          -3.1186e-01,  5.2344e-02,  2.9289e-01, -3.4723e-01,  1.5726e-01,\n",
       "          -9.2018e-02, -5.4410e-02, -1.8265e-02,  2.6903e-01,  1.1005e-01,\n",
       "           4.3774e-01,  3.4899e-01, -5.4586e-01,  1.2468e-01, -3.2757e-01,\n",
       "          -2.5582e-01,  1.8985e-01,  1.3650e-01,  1.5458e-01, -5.8624e-01,\n",
       "          -2.9476e-01, -1.9112e-01, -2.4676e-01, -3.1141e-01, -3.5544e-01,\n",
       "           3.4545e-01, -2.4686e-03,  6.7204e-02, -1.2187e-01,  4.8018e-01,\n",
       "           4.8333e-02, -4.0617e-02, -2.0569e-01,  3.9442e-01,  5.6237e-01,\n",
       "           2.1327e-01,  2.0417e-01, -2.5996e-01, -6.9686e-01,  1.0706e-02,\n",
       "          -3.2313e-01,  9.3112e-01, -6.5780e-01, -5.5038e-01,  6.3165e-02,\n",
       "          -1.7195e-01, -2.2873e-01, -1.2904e-01, -2.4490e-01,  2.2746e-01,\n",
       "          -1.3872e-01, -1.7019e-01, -7.6565e-02,  1.0922e-01,  2.5129e-01,\n",
       "           4.8866e-01,  4.0182e-01,  2.7097e-02, -1.7308e-01, -2.3880e-01,\n",
       "          -4.4705e-01,  7.3885e-02, -3.6802e-01,  1.9196e-01, -3.6380e-01,\n",
       "          -2.5813e-01, -1.8935e-01, -9.5258e-02, -1.3153e-01,  4.5521e-01,\n",
       "           1.9505e-01,  1.4907e+00, -3.9366e-01, -1.2377e-01,  3.1441e-01,\n",
       "           1.0997e-02,  2.8699e-01,  5.5521e-02,  2.2665e-01,  9.3773e-02,\n",
       "           2.3392e-01, -4.5811e-01, -2.1030e-02, -8.4475e-03,  4.4201e-01,\n",
       "           3.2178e-01, -7.8733e-01, -8.1659e-02, -2.3964e-01,  3.3740e-01,\n",
       "           1.9991e-01,  4.5370e-01, -2.1985e-01,  4.0294e-01,  5.3460e-01,\n",
       "          -9.2366e-01,  3.1534e-02, -2.5676e-01,  3.2674e-01, -8.1001e-02,\n",
       "          -1.4970e-01, -1.7423e-01, -8.1340e-02, -2.9901e-01,  3.0970e-01,\n",
       "           6.3303e-01, -6.5249e-01,  6.3190e-01,  4.6047e-01, -6.9825e-02,\n",
       "           5.6352e-01, -6.0580e-02,  2.4533e-01,  5.9444e-02, -1.2265e-01,\n",
       "           1.4922e-01,  1.2411e-01, -4.6809e-01, -4.9988e-01,  2.3552e-01,\n",
       "           4.0712e-01,  3.4024e-02,  1.2081e-01,  8.8821e-02, -5.9235e-02,\n",
       "          -3.2434e-01, -1.4542e-01, -2.3609e-01, -3.5407e-01,  6.8073e-02,\n",
       "           3.8431e-02, -2.2810e-01,  4.9756e-02,  4.8168e-01, -2.9383e-01,\n",
       "           6.2368e-01,  3.1202e-01,  3.8853e-01, -1.1510e-01, -1.3577e-01,\n",
       "           7.2305e-02,  4.1355e-01, -3.1473e-01,  7.5890e-01, -8.8368e-03,\n",
       "          -2.3466e-01, -2.2230e-02,  1.6344e-01,  2.3902e-01,  4.2397e-01,\n",
       "          -4.1160e-01, -1.4710e-01, -3.0202e-01, -3.3292e-01,  2.6945e-01,\n",
       "          -1.2950e-01,  2.6271e-01, -6.9403e-01,  1.4105e-01,  4.5458e-01,\n",
       "          -2.6243e-01,  3.4871e-01, -2.4071e-01,  6.1586e-01, -2.2477e-01,\n",
       "           7.1011e-02, -6.1702e-01, -2.1490e-01, -1.3001e-01,  5.7989e-02,\n",
       "          -9.4422e-01, -2.4210e-01,  2.6460e-01, -6.2211e-02,  2.5519e-01,\n",
       "           6.3087e-01, -1.0958e-01, -2.8306e-01, -2.4169e-01,  2.6983e-02,\n",
       "          -2.2436e-02,  4.2172e-02, -2.0749e-01,  3.4027e-02, -4.1950e-01,\n",
       "          -6.9794e-01, -2.5795e-01,  1.8008e-02,  2.5659e-01, -3.6133e-01,\n",
       "           8.8307e-01,  9.9844e-02, -5.6155e-02,  3.7544e-01,  1.2142e-01,\n",
       "           5.5749e-01,  5.6106e-01, -1.5671e-01, -5.8616e-01, -9.3439e-02,\n",
       "          -1.1826e-01, -1.5429e-01,  9.8416e-02, -2.1225e-01,  5.2235e-01,\n",
       "           5.7019e-01,  7.8374e-02,  6.2823e-01]], device='mps:0'),\n",
       " tensor([[-8.4367e-02, -9.9544e-01, -7.6676e-01, -3.6066e-01, -6.5543e-01,\n",
       "           2.5891e-01, -3.0088e-01,  6.8470e-01,  1.0237e-01,  6.3124e-01,\n",
       "           4.0418e-01, -2.5821e-01,  6.0564e-02, -3.4243e-01, -4.9831e-01,\n",
       "          -4.7098e-01,  4.0723e-01,  4.7123e-01,  4.3827e-01,  4.8314e-01,\n",
       "           5.9683e-01,  2.7857e-01, -1.0076e-01,  3.3544e-02, -6.4546e-01,\n",
       "           2.4330e-01, -4.2110e-02,  1.3205e-01,  4.8342e-01, -4.3177e-01,\n",
       "          -4.6548e-02,  4.0897e-01,  1.0692e-02,  1.0297e+00, -2.5928e-01,\n",
       "          -1.6171e-01,  1.2562e-01, -3.9049e-02,  2.4548e-01, -2.8336e-01,\n",
       "           7.2436e-01, -1.9736e-03,  3.3338e-01,  2.1337e-01,  3.8305e-01,\n",
       "          -4.1797e-02,  1.1012e-01, -5.8827e-01, -2.5325e-01, -1.1852e-01,\n",
       "           9.8159e-02, -2.6861e-01,  5.2367e-01,  3.3658e-01, -1.1884e-01,\n",
       "          -1.3784e-01,  6.1704e-02,  4.8597e-01, -4.4127e-01,  1.2459e-01,\n",
       "           6.3577e-02,  7.6844e-01,  4.8226e-02,  7.8531e-02, -3.4758e-02,\n",
       "           2.2771e-01, -7.6413e-01, -4.7590e-01, -2.8223e-01,  2.3143e-01,\n",
       "          -5.1286e-02,  2.1924e-01, -1.8105e-01,  5.5858e-02,  4.0894e-01,\n",
       "          -6.0636e-01,  1.2790e-01, -1.5663e-01, -7.6339e-01, -2.8983e-01,\n",
       "           3.1167e-01,  8.5711e-02, -2.1309e-01, -2.3854e-01, -1.6624e-01,\n",
       "          -2.1786e-02,  4.7257e-01,  1.3428e-01,  1.3315e-01, -1.4904e-03,\n",
       "           3.2695e-01,  1.2292e-01, -4.4498e-01, -3.7081e-01,  4.3861e-01,\n",
       "          -3.1129e-01,  3.6942e-01, -1.1072e-01, -1.9192e-01, -1.7740e-01,\n",
       "          -7.3041e-02,  3.6625e-01,  1.3063e-01,  1.0233e-02, -4.2244e-01,\n",
       "          -4.4462e-01,  2.1118e-01, -1.0029e-01, -5.8134e-01,  4.1833e-02,\n",
       "          -9.9223e-02, -1.3356e-01,  2.7734e-01, -7.8352e-01, -2.7233e-01,\n",
       "          -7.4877e-02,  2.0114e-01, -2.1265e-01, -3.4069e-01,  4.7558e-01,\n",
       "           2.9456e-01, -5.7180e-01,  1.8192e-01,  1.8892e-01,  1.9922e-01,\n",
       "          -2.6990e-01,  2.7019e-01, -1.4557e-01,  2.3345e-01,  4.2545e-01,\n",
       "          -2.2775e-01,  1.6894e-02, -7.4285e-02, -2.3522e-01, -2.0285e-01,\n",
       "          -1.6799e-01,  3.0267e-01, -6.2164e-01,  2.7428e-01,  1.9522e-01,\n",
       "          -4.6284e-02,  3.6605e-02, -5.0484e-01, -1.1827e-02,  1.7577e-01,\n",
       "          -1.4853e-01,  1.4923e-01, -8.2681e-02, -3.4502e-01, -1.4070e-01,\n",
       "          -7.9769e-01, -4.7416e-01,  3.6416e-01,  1.5173e-02,  1.5206e-02,\n",
       "           7.0536e-02,  1.8764e-01,  1.3009e-01, -1.2432e-01,  9.6329e-02,\n",
       "           5.3422e-01, -5.2765e-02,  2.5112e-01, -3.6691e-01,  4.4943e-03,\n",
       "           1.1869e-01,  2.7898e-01, -2.2534e-01,  2.5080e-01,  1.4270e-01,\n",
       "           8.2778e-02, -5.4946e-01,  6.5663e-02, -2.3727e-01,  7.5774e-02,\n",
       "          -3.8194e-01,  7.1762e-02,  2.8576e-01, -2.1230e-01, -2.8791e-02,\n",
       "          -6.3730e-02, -1.8550e+00,  1.1640e-01, -1.6118e-01,  1.6947e-01,\n",
       "          -3.5909e-01,  4.1989e-01,  4.0444e-02,  3.5049e-01, -2.3511e-01,\n",
       "          -2.0530e-01, -2.9160e-01,  2.0532e-02,  2.8190e-01,  1.1582e-01,\n",
       "           4.0659e-01,  6.2690e-01,  3.4601e-01, -3.3775e-01, -2.3636e-01,\n",
       "          -4.3631e-01,  7.5651e-02, -1.2973e-01,  5.2947e-01,  1.2589e-01,\n",
       "           2.2308e-01, -3.0865e-01, -8.1769e-02,  3.6243e-01, -1.5496e-01,\n",
       "           6.8469e-03, -3.7726e-01, -4.4730e-01, -2.0761e-01, -1.8935e-01,\n",
       "           6.5341e-01, -1.0807e-01, -1.5733e-01,  2.5209e-01,  3.2275e-01,\n",
       "          -1.7643e-01, -2.2036e-01, -1.1321e-01, -4.1263e-01, -2.3336e-01,\n",
       "           1.7229e-01, -2.7326e-01, -3.3327e-02, -5.0663e-01, -1.4497e-01,\n",
       "           1.7484e-01, -4.6704e-01,  7.1167e-01,  2.3259e-01, -2.9028e-01,\n",
       "          -3.6854e-01, -2.8246e-01,  1.6381e-01, -3.8400e-01,  9.8825e-02,\n",
       "          -3.9668e-01, -7.2905e-01, -6.7199e-01,  6.7727e-02,  2.6025e-01,\n",
       "           1.7070e-01, -2.1004e-01,  1.0842e+00,  2.4469e-01,  1.7280e-01,\n",
       "          -2.1477e-01, -1.7300e-01,  2.7311e-01, -3.1000e-02,  2.2551e-03,\n",
       "          -6.6367e-02,  1.5160e-02, -2.3197e-01,  4.9427e-01, -5.4481e-01,\n",
       "           1.4050e-02, -7.2921e-02, -7.7508e-02,  3.8728e-01, -3.1718e-01,\n",
       "           3.8960e-01,  2.6494e-01, -6.7055e-01, -7.7087e-02,  2.8328e-01,\n",
       "          -5.5764e-01,  4.9447e-01,  2.9289e-01, -4.2483e-01, -4.6639e-01,\n",
       "          -7.4040e-01,  1.2654e-01, -4.4153e-01,  9.3556e-03, -8.4792e-01,\n",
       "           8.9031e-02,  6.9095e-03,  2.7660e-01,  1.3241e-01, -3.5555e-02,\n",
       "           1.6525e-01,  4.0771e-01, -2.7235e-02, -5.2918e-02, -5.0872e-01,\n",
       "           4.6902e-02,  3.2414e-01, -3.0145e-01,  4.0338e-01,  1.7120e-01,\n",
       "          -1.0706e-01, -6.0042e-02,  1.0466e-01,  1.2865e-01, -2.3268e-01,\n",
       "          -3.1868e-01, -2.0738e-01, -5.5217e-03, -2.7404e-01, -1.1589e-01,\n",
       "           1.1605e-01,  2.0180e-01,  9.5597e-01, -1.4523e-01, -4.7065e-01,\n",
       "           1.4229e-02,  4.3059e-01, -1.0153e-01,  1.3948e-01, -3.4498e-01,\n",
       "          -1.4564e+00, -1.6000e-01, -1.2551e-01, -1.1716e-02, -2.4961e-01,\n",
       "           4.4710e-01,  1.4258e-01,  2.1673e-01, -6.6838e-02, -3.1554e-01,\n",
       "           4.2186e-01,  1.8045e-01,  1.5301e-01, -1.9014e-01, -7.6215e-02,\n",
       "          -1.8093e-01,  5.9824e-02, -1.8624e-01, -1.2212e-01,  2.5387e-01,\n",
       "           2.1246e-01,  4.2682e-01,  1.3010e-01,  1.4766e-01, -2.8905e-01,\n",
       "           8.0982e-01, -1.5348e-01, -2.3238e-01,  4.7025e-01, -4.1200e-01,\n",
       "          -4.8283e-01,  4.3726e-01,  1.6591e-01, -1.1289e-01,  1.0452e-01,\n",
       "           4.9590e-01,  1.5858e-01, -2.9403e-01,  1.1842e-01,  2.8889e-01,\n",
       "           4.3753e-01,  4.8374e-01, -1.8678e-01,  5.7783e-01,  8.2683e-01,\n",
       "          -1.6468e-01,  1.5626e-01,  9.6651e-02, -1.0880e-01, -2.0768e-01,\n",
       "           2.2006e-01,  3.2807e-01, -5.4607e-01,  7.2044e-02, -8.0584e-02,\n",
       "           8.3925e-02, -5.2893e-01, -2.0362e-01,  1.1441e-01,  2.3850e-01,\n",
       "          -2.9664e-01, -3.9601e-01,  7.0843e-01, -4.8620e-01, -5.9999e-01,\n",
       "           1.3799e-02,  1.2048e-01,  1.0193e-01, -2.2993e-01, -7.3144e-01,\n",
       "          -3.8818e-01,  5.5136e-02,  7.8710e-01,  1.3443e-01,  1.8723e-01,\n",
       "          -1.3401e-01, -2.3594e-02, -2.5439e-01,  3.4287e-01, -2.7282e-02,\n",
       "          -1.5088e-01,  2.4871e-01, -1.6269e-01,  1.5134e-01,  2.0525e-01,\n",
       "          -1.5518e-01, -1.8537e-01,  2.6337e-02,  1.3023e-02, -3.4143e-01,\n",
       "          -3.6412e-01,  7.6576e-02,  4.4562e-01, -1.4358e-01,  4.0201e-02,\n",
       "           1.6336e-01,  2.4095e-01, -1.2428e-01,  1.3924e-01,  1.2610e-01,\n",
       "           3.6558e-01, -2.7816e-01, -3.8846e-02, -1.9736e-01, -7.2080e-01,\n",
       "           4.9454e-01,  3.9282e-01,  5.0177e-01,  1.7274e-01, -1.1325e+01,\n",
       "           4.3340e-01,  6.2296e-01, -5.5520e-01, -2.8520e-02,  1.5383e-01,\n",
       "           1.3363e-01, -5.5452e-01, -6.2163e-02, -3.7226e-01, -3.4584e-01,\n",
       "           8.8002e-02,  8.1150e-03,  1.6451e-01, -2.7741e-01,  9.2481e-02,\n",
       "          -2.6656e-01, -1.0030e+00,  2.5486e-01,  1.4193e-01,  1.7161e-01,\n",
       "           3.0074e-02,  1.5006e-01, -8.5472e-02,  4.3966e-03, -5.8248e-01,\n",
       "           1.6209e-01, -4.4367e-01,  4.3888e-03,  2.3013e-01, -1.6589e-01,\n",
       "           1.7321e-01,  3.0756e-01,  1.9118e-01, -3.4465e-01, -5.2074e-01,\n",
       "           8.2137e-02,  3.2416e-01,  2.1041e-01,  1.4966e-01, -8.6932e-02,\n",
       "          -1.0027e-01, -2.9252e-01,  2.5918e-01, -9.2526e-02,  7.5756e-01,\n",
       "           6.8354e-01,  9.1070e-01,  2.1976e-02,  5.6146e-02, -9.8465e-03,\n",
       "          -8.3987e-01,  3.4823e-01, -1.0675e+00,  2.4378e-01,  1.7914e-01,\n",
       "           6.6129e-01, -7.1310e-01,  1.1627e-01, -1.5075e-01, -4.4335e-01,\n",
       "           1.9736e-01,  1.9250e-02,  4.8331e-01, -2.3359e-01, -2.7635e-01,\n",
       "           3.2073e-01,  3.3983e-01,  3.7253e-01, -1.7779e-01,  3.9588e-01,\n",
       "          -2.7989e-01,  4.7045e-01,  4.2343e-01, -7.4675e-01, -1.4220e-02,\n",
       "           1.9435e-02,  1.9220e-01,  7.3075e-01,  1.1571e-01, -9.6283e-02,\n",
       "           5.9384e-02,  3.9926e-01, -5.3791e-01, -5.9717e-01,  1.0659e-01,\n",
       "           1.0427e-01,  5.1462e-02,  1.7467e-01,  6.6462e-01, -2.5593e-01,\n",
       "          -5.5864e-01, -1.5824e-01, -1.7096e-01,  1.2117e-01,  7.3767e-01,\n",
       "          -4.2521e-01,  1.0810e-02,  2.9120e-01,  4.9301e-02, -6.4409e-01,\n",
       "          -2.8414e-01, -3.9714e-02, -2.6439e-01,  4.3758e-01,  8.6725e-02,\n",
       "           4.8441e-02,  2.2804e-01,  1.0790e-01,  1.3781e-01, -6.2136e-01,\n",
       "           4.0709e-01, -1.4452e-01,  1.2176e-01, -1.8181e-01, -4.2195e-01,\n",
       "           5.2464e-02,  3.3992e-01, -4.2048e-01, -4.2903e-01, -4.0213e-01,\n",
       "           1.3676e-01,  1.4312e-01, -9.5148e-02, -4.2635e-01,  2.5081e-01,\n",
       "          -5.6908e-01, -1.1600e-02, -2.2168e-01,  1.5547e-01,  2.9300e-01,\n",
       "           1.4328e-01,  7.5556e-01,  4.3220e-01, -1.4347e-01, -9.0420e-01,\n",
       "          -6.6719e-01, -4.3798e-02,  5.6063e-02, -3.1811e-01, -1.4041e-01,\n",
       "          -2.3530e-02, -9.1374e-02, -6.8794e-02,  1.6647e-01, -3.5942e-01,\n",
       "           5.6077e-01,  3.8020e-01,  7.2420e-03, -1.6792e-01, -2.5265e-01,\n",
       "          -2.3487e-01, -4.4025e-01, -1.7584e-01, -4.7575e-02, -4.0234e-01,\n",
       "          -2.7873e-01,  4.8451e-02, -3.7576e-01,  3.7169e-01, -7.9503e-02,\n",
       "           2.0456e-02, -2.5830e-01, -1.1124e-01,  2.1625e-01,  2.2133e-01,\n",
       "           1.4021e-01, -2.2102e-01, -7.9863e-01,  5.5749e-02,  1.2462e-01,\n",
       "           7.0927e-02, -1.5994e-01, -1.7651e-01, -5.8042e-01,  1.8801e-01,\n",
       "          -1.4829e-02,  1.1257e+00, -2.6214e-01, -4.0665e-01,  3.0422e-01,\n",
       "           7.1301e-01, -2.3407e-01,  1.5979e-01, -1.4556e-01,  3.7199e-01,\n",
       "          -8.7036e-02, -3.9455e-01,  2.9156e-01,  1.4516e-01,  3.7153e-01,\n",
       "           4.3518e-01,  1.1903e-01, -7.0225e-02,  2.3938e-01, -3.8350e-01,\n",
       "          -2.3578e-01, -9.9270e-02, -1.0127e-01,  1.8560e-02,  2.4009e-01,\n",
       "           5.1393e-02, -7.1133e-03, -6.7573e-02,  1.4515e-01,  1.7819e-01,\n",
       "           2.0262e-01,  1.3163e+00, -1.5479e-01, -1.1003e-01,  3.1455e-01,\n",
       "           1.8460e-01, -4.0461e-01,  1.0888e-01,  8.3768e-02,  2.6697e-01,\n",
       "           4.1623e-01,  9.5291e-02, -4.3354e-01,  1.3466e-01,  9.9492e-02,\n",
       "           1.6937e-01, -6.7153e-01, -4.1693e-01, -9.9818e-02,  5.2303e-01,\n",
       "           1.9350e-01,  1.7095e-01, -2.9902e-01,  4.5855e-01,  5.6058e-01,\n",
       "          -9.1975e-01, -1.6790e-02, -3.4235e-01,  5.4188e-01,  4.3356e-03,\n",
       "          -5.7435e-01, -2.7842e-01, -2.8756e-01, -1.4246e-01,  3.8933e-01,\n",
       "           4.0278e-01, -9.0740e-01,  1.4352e-01,  9.5229e-02, -4.5865e-01,\n",
       "           8.9898e-02, -1.8778e-01,  4.1324e-01,  2.7071e-01, -1.7722e-01,\n",
       "          -1.6717e-01,  3.1949e-01, -2.2758e-01, -4.9559e-01,  1.2165e-02,\n",
       "           5.2458e-01, -6.8425e-02, -8.0264e-02, -5.9458e-02, -5.3224e-01,\n",
       "          -4.4947e-03, -2.4609e-01, -2.1233e-01,  4.1113e-02, -2.5033e-01,\n",
       "          -1.6418e-01, -4.7583e-01,  8.7349e-02,  4.4863e-01, -4.0693e-01,\n",
       "           6.9590e-01,  7.7167e-01,  4.7521e-01, -1.3519e-01,  2.7051e-01,\n",
       "           3.8944e-02,  1.8352e-01, -1.4401e-01,  6.3044e-01, -1.3380e-01,\n",
       "          -5.2214e-01,  4.4726e-01,  5.2590e-02, -1.8820e-01, -1.5596e-01,\n",
       "          -3.6592e-01,  3.3802e-01,  2.2102e-01, -2.7085e-01,  4.2224e-01,\n",
       "           2.5655e-01, -2.7179e-02, -4.9229e-01,  2.4019e-01,  7.9900e-02,\n",
       "          -2.9798e-01,  9.0969e-01, -3.6684e-02,  7.0918e-01, -1.7199e-01,\n",
       "           3.7460e-01, -6.4847e-01, -6.1896e-01, -1.0694e-02,  2.7532e-02,\n",
       "          -5.0267e-01, -2.0316e-02,  3.6203e-01,  4.1920e-02,  6.7945e-01,\n",
       "           6.3578e-01, -3.4671e-01, -6.0193e-01, -1.3692e-01,  7.8214e-02,\n",
       "           1.4914e-01, -3.0498e-01,  3.4273e-01, -3.2149e-01,  2.0122e-01,\n",
       "          -8.2048e-01, -5.0978e-01,  1.7019e-01,  1.9072e-01, -4.4355e-01,\n",
       "           5.2155e-01, -2.3080e-01,  8.5308e-02,  5.8273e-01,  2.3058e-01,\n",
       "           4.1328e-01,  4.0443e-01,  7.2632e-02, -6.0968e-01,  6.8929e-02,\n",
       "           6.6616e-02, -2.7373e-01,  5.3488e-02, -3.7202e-01,  4.2463e-01,\n",
       "           6.7098e-02,  4.2283e-01,  6.8514e-01]], device='mps:0'),\n",
       " tensor([[-0.2258, -0.7099, -0.3778,  ...,  0.3292,  0.7059,  0.7494],\n",
       "         [-0.1802, -0.4695, -0.5270,  ...,  0.0749,  0.3322,  0.4674],\n",
       "         [ 0.1722, -0.2575, -0.2730,  ..., -0.0414,  0.5308,  0.6272],\n",
       "         [-0.0626, -0.6265, -0.7028,  ...,  0.1217,  0.4033,  0.6177]],\n",
       "        device='mps:0'),\n",
       " tensor([[-0.0111, -0.5057, -0.4670,  ...,  0.2365, -0.0712,  0.4388],\n",
       "         [-0.1796, -0.5245, -0.5098,  ...,  0.2721,  0.5034,  0.5637],\n",
       "         [-0.5193, -0.3907, -0.4656,  ...,  0.1974,  0.7151,  0.0541]],\n",
       "        device='mps:0'),\n",
       " tensor([[-0.2823, -0.7148, -0.6013,  ...,  0.2119,  1.0729,  0.5741],\n",
       "         [-0.3587, -0.4049, -0.7857,  ..., -0.2691,  0.2327,  0.1502],\n",
       "         [-0.0773, -0.4703, -0.8527,  ...,  0.0992,  0.7988,  0.3571]],\n",
       "        device='mps:0'),\n",
       " tensor([[-1.5113e-01, -5.0022e-01, -6.9479e-01, -5.9221e-01, -1.7806e-01,\n",
       "          -3.4556e-01, -3.9978e-01, -9.9048e-02, -4.1947e-01,  5.8486e-01,\n",
       "           1.7746e-01,  3.5640e-01,  7.4587e-02, -7.9935e-04, -7.6955e-01,\n",
       "           4.0723e-01,  3.8807e-01,  5.2797e-01,  5.1424e-02,  2.7138e-01,\n",
       "           6.3935e-01,  2.6715e-01,  1.8339e-01,  8.0367e-02,  1.0257e-01,\n",
       "           2.5637e-02, -1.2897e-01, -3.3973e-02,  5.3384e-01, -2.8555e-01,\n",
       "          -1.7652e-01,  5.9027e-01,  2.5990e-01,  8.8491e-01, -2.1543e-01,\n",
       "           8.2986e-02, -1.5400e-02, -4.3908e-01,  1.3583e+00, -3.1760e-01,\n",
       "           8.6077e-01, -1.0862e-01,  2.5055e-01,  1.5723e-01, -1.1715e-01,\n",
       "          -7.1616e-02,  1.0148e-01, -9.4383e-01,  6.0810e-01, -2.2138e-01,\n",
       "          -1.1468e-01,  1.3109e-01,  4.5574e-01,  1.4599e-01, -2.0881e-01,\n",
       "          -1.0863e-01, -3.1783e-02,  4.6357e-01, -2.3633e-01, -6.7098e-03,\n",
       "           1.0318e-01,  3.4095e-01, -3.4857e-02, -3.9012e-01, -1.0918e-01,\n",
       "          -7.8812e-02, -1.8285e-01, -8.2235e-01,  1.0275e-01, -1.0911e-01,\n",
       "          -4.1343e-02,  3.4714e-01, -2.7782e-02, -1.6658e-01,  4.8132e-02,\n",
       "          -3.2286e-01,  4.6493e-02,  9.1702e-02, -1.7960e-01,  1.8282e-01,\n",
       "           4.0545e-01,  2.3369e-01,  3.3702e-03, -1.0224e-01, -5.9613e-02,\n",
       "          -1.5965e-02,  2.6722e-01,  4.5525e-01,  5.0590e-01,  1.4055e-01,\n",
       "           2.2641e-01,  9.5633e-02, -3.0231e-01, -3.2729e-03,  1.4637e-01,\n",
       "          -8.0590e-02, -6.2208e-02, -3.0434e-02,  1.2500e-01, -1.9653e-01,\n",
       "          -4.0191e-02,  3.6920e-02,  1.9415e-01,  2.0930e-01, -4.7780e-01,\n",
       "          -3.2863e-01,  2.9880e-01, -2.0037e-01, -1.0357e+00, -2.3280e-01,\n",
       "           1.7564e-01, -2.6477e-01,  1.6423e-01, -2.5513e-01, -6.8125e-02,\n",
       "           9.5650e-02,  2.3664e-01,  1.8077e-01, -2.9200e-01,  6.6629e-01,\n",
       "           3.3359e-01, -2.6861e-01,  2.0739e-03, -1.7325e-01,  3.6569e-01,\n",
       "           4.3015e-02,  4.0318e-01, -2.3186e-01,  7.5124e-02,  5.4711e-03,\n",
       "           7.2687e-02,  2.9129e-02, -1.0198e-01, -1.9627e-01,  1.1945e-01,\n",
       "           2.3409e-01,  3.7658e-01, -1.9921e-01,  2.2898e-01,  1.2852e-01,\n",
       "           8.6996e-02, -4.9329e-01, -3.7378e-01, -3.0567e-01, -2.5644e-01,\n",
       "          -2.5868e-01, -3.2709e-01, -3.5726e-01, -1.2829e-01, -4.0712e-01,\n",
       "          -1.0644e+00, -7.6238e-01,  4.2477e-01, -3.0155e-01,  3.6744e-01,\n",
       "           1.4883e-01,  3.5545e-01,  1.6877e-01,  1.8916e-02,  1.2201e-01,\n",
       "           2.4502e-01, -3.8819e-01,  8.9963e-02, -5.3575e-01,  3.2112e-01,\n",
       "          -2.2545e-01, -1.2976e-01, -1.8901e-01,  5.0143e-01,  1.4254e-01,\n",
       "           2.7310e-01, -2.0468e-01,  1.7035e-01, -4.3949e-01,  3.3485e-02,\n",
       "          -4.1902e-01,  2.7215e-02,  2.5276e-01, -1.0714e-01, -3.2463e-01,\n",
       "           2.2636e-01, -2.0430e+00,  3.8586e-01, -4.2096e-01, -3.2483e-01,\n",
       "          -4.1440e-01,  4.0775e-01,  1.4154e-01, -2.1910e-02, -1.5668e-01,\n",
       "          -2.4780e-01, -1.3778e-01, -8.5450e-02, -9.7375e-02,  1.1090e-01,\n",
       "          -5.4265e-02,  4.0604e-01,  1.8873e-03, -2.9752e-01, -5.6560e-02,\n",
       "          -2.8847e-01,  3.2373e-01, -2.1793e-01,  1.2113e-01,  1.4707e-02,\n",
       "           1.9475e-01, -5.5166e-02,  1.1550e-01,  4.3031e-01,  1.2465e-01,\n",
       "          -2.6384e-01,  8.6903e-02, -2.5452e-01,  9.3600e-02, -3.3980e-01,\n",
       "           5.0907e-01, -2.3823e-01, -1.3092e-01,  4.8544e-01,  1.5587e-01,\n",
       "           1.6305e-01, -2.2492e-02,  7.8921e-03,  5.1730e-03,  2.5094e-01,\n",
       "           2.9061e-01, -1.1802e-01,  1.3445e-01, -2.3976e-01, -1.7368e-01,\n",
       "          -9.2502e-02, -5.5181e-01,  2.7240e-01, -8.3747e-03, -1.8720e-01,\n",
       "          -4.4313e-01, -1.5302e-01,  6.5400e-01,  4.6064e-02, -2.1836e-01,\n",
       "          -1.1560e-02, -6.1462e-01, -7.2190e-01, -5.5848e-02,  4.1230e-02,\n",
       "           4.5386e-02, -6.0505e-02,  3.8871e-01,  4.3740e-01,  3.6492e-01,\n",
       "           4.5007e-02, -8.4253e-02,  6.6942e-01,  5.1518e-01, -2.5509e-01,\n",
       "           1.8167e-01,  2.8699e-01, -2.7179e-01,  4.5429e-02, -5.3184e-01,\n",
       "          -4.9170e-01,  7.6698e-02, -3.5775e-02,  2.7035e-01, -1.3579e-01,\n",
       "           8.3090e-01,  5.6750e-01, -1.6858e-01, -3.9632e-01,  5.3783e-02,\n",
       "          -3.3701e-01,  9.0566e-02, -9.4327e-02, -3.6616e-01, -6.9871e-01,\n",
       "          -7.7555e-01, -1.6649e-01,  6.3752e-03, -5.8035e-02, -6.4476e-01,\n",
       "          -1.5239e-01, -2.5395e-02, -1.5187e-01, -2.8254e-01,  1.1515e-01,\n",
       "           5.6702e-02,  5.7953e-02,  1.3719e-01, -3.5915e-01, -3.1925e-01,\n",
       "          -1.4989e-01,  3.9474e-01, -2.3695e-01,  7.0402e-02,  2.9770e-01,\n",
       "          -3.9679e-01,  2.6106e-01,  3.9794e-01, -4.1896e-02, -2.4868e-01,\n",
       "          -1.9687e-01,  6.0917e-01, -9.1315e-02, -2.9301e-02,  2.9272e-01,\n",
       "           9.1653e-02,  1.0144e-01,  9.3188e-01, -6.9244e-01, -3.6049e-01,\n",
       "           4.6287e-02,  7.0523e-01, -2.0584e-01,  5.8428e-02, -3.5274e-01,\n",
       "          -5.2935e-01,  4.6315e-02, -5.8019e-01, -3.7610e-02, -3.6070e-01,\n",
       "           2.0644e-01,  1.0203e-01, -2.0208e-01,  1.8745e-01, -2.6192e-01,\n",
       "           3.5625e-01,  4.1147e-01, -1.2628e-01, -1.6510e-02, -3.7780e-01,\n",
       "           7.3989e-02, -1.3515e-01, -3.8149e-01, -2.7437e-01,  1.4763e-01,\n",
       "           4.3826e-01,  1.1568e-01,  1.7035e-01,  7.8935e-02, -3.2559e-01,\n",
       "           6.3809e-01, -1.6165e-02,  2.7309e-01,  1.0199e-01, -1.8930e-01,\n",
       "          -6.3404e-02,  2.5561e-01, -2.1929e-01,  5.6528e-02,  2.6163e-01,\n",
       "           4.7681e-01, -2.2572e-01, -5.9720e-01,  3.6751e-02,  6.7196e-01,\n",
       "           2.7032e-01,  4.4230e-01, -2.0898e-01,  2.9752e-02,  4.7286e-01,\n",
       "          -1.2338e-01, -5.2409e-02,  1.2126e-01, -2.1737e-01, -2.3014e-01,\n",
       "           5.6690e-01,  3.3245e-01, -3.3335e-01,  5.3287e-03,  1.8847e-01,\n",
       "          -6.3307e-02,  7.3966e-02, -3.1078e-01, -5.4661e-02,  4.7652e-02,\n",
       "          -9.9878e-02, -4.1236e-01,  3.6467e-01, -4.3581e-01, -1.6664e-01,\n",
       "           4.0179e-02, -1.0832e-01, -1.4562e-01, -4.5789e-01, -1.3591e-01,\n",
       "          -3.4116e-01, -1.4709e-01,  3.9061e-01,  1.6641e-02,  2.1035e-02,\n",
       "           1.5451e-01,  8.8186e-02, -1.5246e-01,  6.0835e-01,  2.4131e-01,\n",
       "          -4.0112e-01,  1.4016e-01, -3.1191e-01,  6.0518e-02, -2.1827e-01,\n",
       "          -3.0862e-01, -8.7704e-02,  1.6096e-01,  5.8821e-01, -1.3324e-01,\n",
       "          -4.3396e-01, -6.3519e-02,  6.7673e-02,  2.5896e-02,  2.1417e-01,\n",
       "           9.6439e-02,  4.3854e-01, -9.3842e-02,  1.2876e-01, -4.3058e-01,\n",
       "           3.2865e-01, -9.2162e-02, -2.8592e-01,  1.4543e-01, -5.0612e-01,\n",
       "           6.4904e-01,  8.1930e-02,  9.9213e-03,  2.7996e-01, -1.1640e+01,\n",
       "           2.7702e-01,  7.0784e-01, -4.8320e-01,  2.0061e-01, -3.4047e-02,\n",
       "          -9.6377e-03, -2.9130e-01,  1.3411e-01, -2.6076e-01, -1.8730e-01,\n",
       "          -2.4393e-01, -2.0308e-01,  1.2423e-01, -2.1893e-01, -3.2951e-01,\n",
       "           2.8778e-01, -8.9499e-01,  8.1637e-03, -1.7105e-01,  8.6047e-02,\n",
       "          -4.4758e-01,  1.4902e-01,  3.2097e-01, -3.9378e-01, -1.1321e-01,\n",
       "          -3.5469e-01, -4.8099e-01,  3.8539e-02,  2.9720e-01, -1.9211e-01,\n",
       "           1.4509e-01, -1.2812e-01,  5.8183e-01, -1.2608e-02, -5.8989e-01,\n",
       "           6.8500e-02, -2.5204e-01,  6.8370e-01, -8.2964e-02, -7.4918e-02,\n",
       "          -1.0102e-01, -2.6411e-01,  2.9545e-01,  2.5836e-01,  4.9452e-01,\n",
       "           1.4816e+00,  7.6788e-01,  2.7074e-02, -4.6933e-01,  9.5624e-02,\n",
       "          -5.7241e-01,  3.8474e-01, -9.3114e-01,  5.2832e-01, -1.5152e-01,\n",
       "           1.8632e-01, -3.0042e-01,  2.0861e-01, -1.6792e-01, -3.3873e-01,\n",
       "          -1.2614e-01, -4.6331e-02,  1.2148e-01, -4.6563e-01, -3.2774e-01,\n",
       "           3.4938e-01,  7.7732e-02,  1.0259e-01, -1.4314e-01,  3.0055e-01,\n",
       "           3.3694e-01,  1.5804e-01,  1.2470e-01, -2.6151e-01,  6.3505e-01,\n",
       "          -2.2088e-01, -1.5280e-03,  2.9335e-01,  2.6339e-02,  2.6374e-01,\n",
       "           1.9895e-01,  2.2119e-01, -1.8496e-01, -4.3891e-01,  6.2339e-02,\n",
       "           1.8398e-02,  1.4404e-01, -1.5366e-01,  7.8888e-01, -3.7239e-01,\n",
       "          -3.5460e-01, -5.3902e-02, -1.4617e-01,  4.5490e-01,  7.8645e-01,\n",
       "          -4.6313e-01,  4.6800e-01,  2.3713e-01, -2.0030e-02, -6.9800e-01,\n",
       "           7.2190e-03,  2.8715e-01, -5.1954e-01,  8.5169e-02, -1.1997e-01,\n",
       "          -4.0430e-02, -5.0669e-02, -1.4737e-01,  3.9242e-01, -4.3009e-01,\n",
       "           6.7202e-02,  8.5313e-02, -1.0573e-02, -4.6953e-01, -4.8992e-01,\n",
       "           1.1407e-01,  4.7269e-01, -4.0121e-01, -2.5332e-01, -1.2677e-01,\n",
       "          -1.3267e-01,  1.3796e-01, -4.4119e-01, -5.8666e-01,  3.4584e-01,\n",
       "          -3.9222e-01, -5.4777e-02,  2.5201e-01, -1.2298e-01,  3.9054e-01,\n",
       "          -1.6544e-01, -7.1166e-02,  3.1508e-01, -3.9610e-01, -4.1347e-01,\n",
       "          -4.3657e-01, -9.5072e-02,  3.9414e-01,  2.3657e-01,  1.8042e-01,\n",
       "           2.3793e-01, -1.5579e-01, -3.1195e-01,  6.3123e-01,  2.9225e-01,\n",
       "           3.4407e-01,  4.8776e-01, -3.4859e-01, -8.8077e-02, -1.2254e-01,\n",
       "           1.2658e-02, -3.4888e-02, -1.3716e-01, -2.9211e-01, -2.8470e-01,\n",
       "           4.3388e-02, -7.2488e-02, -6.1885e-01, -2.2140e-03,  1.0264e-01,\n",
       "           4.3239e-01, -6.5918e-03, -2.4603e-01,  6.4183e-01,  4.7707e-01,\n",
       "          -3.1236e-02, -1.8618e-01, -4.9314e-01,  1.0159e-01,  8.6484e-03,\n",
       "           2.9419e-01, -1.5600e-01, -4.7699e-01, -6.0614e-01, -5.1787e-01,\n",
       "          -2.9329e-01,  1.1744e+00, -3.1307e-01, -3.8484e-01,  8.4791e-02,\n",
       "          -1.2962e-01,  5.7376e-02, -6.2623e-02, -1.7157e-01,  2.4831e-01,\n",
       "           2.4776e-02, -7.1515e-01,  4.4348e-01, -1.1308e-01,  4.5116e-01,\n",
       "           1.2039e-01, -9.9335e-02, -2.0626e-02,  7.8577e-01,  3.8263e-02,\n",
       "          -4.5185e-01, -2.3191e-02, -3.4677e-01, -4.1790e-02,  5.0397e-01,\n",
       "           1.8661e-01,  5.7403e-01, -2.3683e-01, -4.0207e-01, -9.9302e-02,\n",
       "           2.7775e-01,  1.2983e+00,  1.5866e-01, -2.7200e-02,  3.3321e-01,\n",
       "           3.8934e-02,  9.3786e-02,  1.3525e-01, -2.4628e-01,  3.0340e-01,\n",
       "           4.5278e-01, -3.2228e-01, -6.0485e-01,  3.6995e-01,  3.1552e-01,\n",
       "           7.2027e-01, -4.2496e-01, -3.8571e-01, -1.2434e-01,  1.7381e-01,\n",
       "           8.8095e-02,  3.4662e-01, -1.2837e-01,  2.8415e-01,  3.7142e-01,\n",
       "          -9.5894e-01,  1.8947e-01,  2.2631e-01,  3.4165e-03, -3.4403e-01,\n",
       "          -5.4783e-01, -4.2356e-01,  3.3082e-02, -5.5542e-01,  4.8017e-02,\n",
       "           5.4018e-01, -1.5793e+00,  3.2386e-01,  5.4258e-01, -6.5719e-01,\n",
       "           1.4773e-01,  8.3246e-02,  3.6932e-01,  1.1398e-01, -7.9051e-02,\n",
       "          -4.5363e-01,  1.0276e-01, -4.4448e-01, -3.8057e-01,  5.0353e-02,\n",
       "           7.1175e-01,  8.5899e-02, -1.3709e-01,  1.5976e-01, -2.0956e-01,\n",
       "          -3.2560e-02,  1.7016e-01, -1.0342e-01,  3.9738e-02,  1.3498e-01,\n",
       "           8.0218e-02, -7.8788e-01,  8.4011e-02,  3.9633e-01, -5.4328e-02,\n",
       "           3.7642e-01,  2.4199e-01,  3.8425e-01, -2.9399e-01, -7.0583e-02,\n",
       "          -1.2754e-01,  2.7443e-01,  5.1723e-02,  5.3687e-01, -3.1938e-01,\n",
       "          -3.3598e-01, -2.0580e-01, -1.9786e-02, -7.7423e-01,  1.4531e-01,\n",
       "          -1.4052e-01, -5.8866e-03, -2.3169e-01, -1.3969e-01,  4.6248e-01,\n",
       "          -9.5356e-02,  2.4145e-01, -6.2983e-01,  1.9613e-01, -1.0665e-01,\n",
       "           9.1874e-02,  5.6356e-01,  3.4979e-01,  5.5486e-01, -2.4270e-01,\n",
       "          -1.1552e-01, -6.0618e-01, -2.6722e-01,  1.3762e-01, -2.1115e-01,\n",
       "          -1.0822e-01, -2.2630e-01, -1.4771e-01,  5.2178e-02,  8.0887e-01,\n",
       "           4.7297e-01, -3.5067e-01, -4.2916e-01, -2.5105e-01, -2.1958e-01,\n",
       "           2.0650e-01, -5.0352e-02,  3.6400e-01,  1.7441e-01, -1.1648e-02,\n",
       "          -7.3792e-01,  1.2704e-02,  1.4008e-01,  2.0883e-01, -2.7849e-01,\n",
       "           7.3488e-01, -1.4453e-01, -4.9529e-01,  1.6092e-01,  6.5393e-02,\n",
       "           1.8388e-01,  6.8510e-01, -1.4929e-01, -1.7128e-01, -5.5066e-02,\n",
       "           2.8421e-01, -2.2646e-01,  5.6539e-02, -3.2935e-01,  1.9314e-01,\n",
       "           1.2037e-01,  3.3154e-01,  2.8650e-01]], device='mps:0'),\n",
       " tensor([[-2.6533e-01, -6.8950e-01, -6.6966e-01, -8.9688e-01, -7.6463e-02,\n",
       "           1.8768e-01, -1.5271e-01,  4.3472e-01, -1.3753e-01,  8.1381e-01,\n",
       "           1.6530e-01,  1.7746e-01, -2.3737e-01, -4.2998e-01, -9.2291e-01,\n",
       "          -3.6031e-01,  8.5072e-02,  5.4586e-01,  7.1489e-01,  6.2235e-01,\n",
       "           4.7745e-01,  3.0086e-01, -2.1980e-01,  2.4390e-01, -7.5098e-01,\n",
       "           4.6335e-01, -1.5258e-02, -1.0704e-01, -4.0444e-02, -5.1242e-01,\n",
       "           1.0470e-01,  2.1805e-01, -7.0148e-02,  9.7376e-01, -1.9363e-01,\n",
       "          -4.1951e-01,  2.3380e-01, -4.8197e-01,  3.2651e-01, -2.8721e-02,\n",
       "           8.4452e-01, -5.8773e-02,  3.1492e-01,  2.5022e-01,  2.5155e-01,\n",
       "          -4.8503e-01,  6.2812e-02, -5.8005e-01,  9.8734e-02, -2.4648e-01,\n",
       "          -7.8361e-02,  7.3329e-02,  5.9526e-01,  4.4990e-01, -7.7301e-01,\n",
       "          -5.2083e-01,  1.2951e-01,  2.2795e-01, -2.0868e-01,  1.6361e-01,\n",
       "          -3.3079e-01,  2.4640e-01,  1.8106e-01, -2.0135e-01,  2.3481e-01,\n",
       "          -3.3018e-01, -2.5992e-01, -4.8086e-01, -2.5658e-01,  2.7417e-01,\n",
       "           2.6774e-01,  3.5998e-01, -1.0694e-01, -1.7320e-01,  6.5764e-01,\n",
       "           9.2755e-02,  7.5344e-02,  1.9121e-01, -5.8336e-01,  1.9958e-01,\n",
       "          -2.3092e-01,  1.3120e-01, -7.7619e-03, -2.6296e-01,  1.5581e-01,\n",
       "           2.9674e-02,  7.1206e-01,  1.3823e-01,  2.1545e-01, -3.3205e-01,\n",
       "           2.0822e-01, -3.0166e-01, -5.6490e-01, -4.9148e-02,  2.1347e-01,\n",
       "          -3.7714e-01,  4.6413e-01, -5.3463e-01, -2.9146e-01, -5.9105e-01,\n",
       "          -2.7589e-01,  3.2603e-01,  2.9770e-03,  1.8825e-01, -8.0317e-01,\n",
       "          -4.3728e-01,  7.6837e-02, -6.4012e-03, -5.2578e-01, -1.3833e-01,\n",
       "          -1.5192e-01,  1.6088e-01,  1.9657e-01, -6.2972e-01, -2.7629e-01,\n",
       "           4.6446e-02,  6.2200e-01, -1.1719e-01, -3.3343e-01,  4.3937e-01,\n",
       "           9.6475e-02, -2.1496e-01,  9.5867e-02, -8.8714e-02, -1.0121e-01,\n",
       "          -4.2952e-01,  5.3135e-01, -1.6577e-01,  4.1714e-01,  3.7492e-01,\n",
       "          -6.3977e-02,  1.4844e-01, -1.6011e-01, -7.4081e-02,  5.5938e-02,\n",
       "           4.0804e-02,  2.8802e-01, -6.7368e-01,  1.6554e-01,  4.9322e-01,\n",
       "          -1.3292e-01, -5.2481e-01, -8.4966e-02, -8.1490e-02,  1.4486e-01,\n",
       "          -9.1834e-02, -1.1413e-01, -4.5229e-01,  1.6662e-01, -3.6145e-01,\n",
       "          -8.7480e-01, -7.8901e-01,  4.2474e-01, -3.7328e-01,  3.7917e-01,\n",
       "          -1.8053e-01,  1.4616e-01,  2.8758e-01,  8.8928e-02,  3.8519e-01,\n",
       "           8.2992e-01,  1.4004e-01,  1.1746e-01, -5.0565e-01, -7.9579e-02,\n",
       "           5.1863e-02,  1.7999e-01, -2.2381e-01, -2.2519e-01,  1.0989e-01,\n",
       "           2.9470e-01, -2.6073e-01,  2.9146e-01, -3.4187e-01,  5.7015e-01,\n",
       "          -8.8868e-01,  3.3237e-01, -3.9130e-02, -1.1759e-01,  1.7430e-01,\n",
       "           3.2245e-01, -1.5324e+00,  5.3997e-01,  1.0413e-01, -2.1667e-01,\n",
       "          -6.0621e-01,  6.4598e-01,  1.8993e-01,  1.9192e-01,  1.4357e-01,\n",
       "           1.2519e-01, -2.9669e-01, -3.9203e-02,  4.4611e-01,  5.1744e-02,\n",
       "           5.1792e-01,  5.4287e-01,  5.3899e-02, -1.8471e-01,  1.1595e-01,\n",
       "          -3.6713e-01,  2.2518e-01,  6.6087e-02, -3.5637e-01,  5.2257e-01,\n",
       "           2.6568e-01, -3.7828e-01,  1.5343e-03,  1.7371e-01,  1.8674e-01,\n",
       "           2.6657e-01, -3.2939e-01, -5.4099e-01, -5.2882e-01, -2.3820e-01,\n",
       "           6.3057e-01, -4.3925e-01, -3.3062e-01,  5.7953e-01, -2.8643e-01,\n",
       "          -1.2297e-01,  2.4581e-02,  1.3399e-01, -2.9919e-01, -6.0877e-02,\n",
       "           1.7603e-01, -2.6621e-01, -1.6154e-01,  9.0128e-02, -1.5409e-01,\n",
       "           2.1355e-01, -4.0403e-01,  7.7777e-01,  7.4227e-01, -3.7511e-01,\n",
       "          -3.1494e-01, -3.0294e-02,  1.3689e-01, -4.5084e-01,  2.4116e-02,\n",
       "          -5.2700e-01, -5.7179e-01, -4.2080e-01,  2.2505e-01, -2.9643e-01,\n",
       "           2.6078e-01, -5.4194e-01,  8.3252e-01,  3.0783e-01,  1.8225e-01,\n",
       "          -1.5820e-01, -2.5062e-01,  1.3880e-01, -7.5936e-02, -2.5160e-01,\n",
       "          -2.6830e-01,  1.5823e-01, -5.1868e-01,  5.8609e-01, -4.6048e-01,\n",
       "          -1.1490e-02,  3.0112e-01, -1.3953e-02,  3.4823e-01,  2.4199e-01,\n",
       "           1.8627e-02,  2.6995e-01, -1.0797e-01, -2.7748e-01,  3.5642e-01,\n",
       "          -2.6809e-01,  1.2800e-01,  2.4806e-02, -3.0835e-01, -4.2991e-01,\n",
       "          -7.5316e-01,  1.4316e-01, -1.3159e-01, -1.1923e-01, -7.1915e-01,\n",
       "          -2.6103e-01,  3.3134e-01, -5.6643e-02, -5.1566e-02,  1.4922e-01,\n",
       "           4.0233e-01,  2.8418e-01, -1.7921e-03, -2.3775e-01, -5.2342e-01,\n",
       "          -1.4947e-01,  2.2512e-01, -4.8271e-01,  1.2031e-01,  1.4482e-01,\n",
       "           7.9105e-03, -1.6475e-01,  2.5160e-02,  3.4875e-01, -3.7924e-01,\n",
       "          -3.2897e-01, -2.3839e-01,  8.1408e-02, -2.0520e-01,  2.2484e-01,\n",
       "           4.4548e-02,  3.1886e-01,  4.0875e-01, -3.4111e-01, -6.2556e-01,\n",
       "           1.0420e-01,  1.5395e-01, -1.3154e-01, -4.1489e-01, -4.7588e-02,\n",
       "          -9.7746e-01, -3.6746e-01, -1.2090e-01,  3.8474e-02, -6.4118e-01,\n",
       "           9.0032e-02,  2.5772e-01, -1.7735e-01,  2.3915e-01, -6.2515e-01,\n",
       "           3.1964e-01,  5.1754e-01, -5.6552e-02, -5.0944e-03, -6.3119e-01,\n",
       "           2.7793e-01, -5.7342e-01, -1.3520e-01, -1.8696e-01, -8.1310e-02,\n",
       "           1.7610e-01,  2.2746e-01,  4.6306e-02,  1.4582e-01, -6.0673e-02,\n",
       "           3.7437e-01, -5.0753e-01, -1.7470e-01,  5.1060e-01, -5.3247e-01,\n",
       "           3.3219e-02,  3.6150e-01, -5.8530e-01, -1.9908e-01,  5.1061e-01,\n",
       "           3.4744e-01, -8.5207e-02, -1.7384e-01, -5.8128e-02,  3.4006e-01,\n",
       "           5.9486e-01,  1.8582e-01, -4.3580e-01,  3.1928e-01,  7.3831e-01,\n",
       "          -3.3921e-01,  3.0133e-01, -3.7374e-01, -3.9089e-01, -1.4181e-01,\n",
       "           1.9966e-01,  2.1614e-01, -7.7652e-01, -1.1479e-01,  7.7280e-02,\n",
       "          -7.4940e-02, -7.4573e-02,  6.6848e-02,  6.7553e-02,  4.1051e-01,\n",
       "          -5.4437e-01, -4.0684e-01,  4.0868e-01, -3.0935e-01, -5.2952e-02,\n",
       "           2.8479e-02,  3.3031e-01, -2.0134e-01, -2.5886e-01, -3.0481e-01,\n",
       "          -3.6810e-01, -1.3579e-01,  8.2344e-01,  4.4351e-01, -3.5524e-02,\n",
       "           4.6413e-01,  2.4733e-01, -5.4218e-02,  2.1582e-01, -1.1333e-01,\n",
       "          -2.2590e-01,  1.1273e-01, -3.7315e-02,  1.4614e-01,  3.0554e-01,\n",
       "          -5.2260e-01, -4.5978e-02,  9.8160e-02,  1.2505e-01,  2.1950e-01,\n",
       "          -4.1997e-01,  1.2014e-01,  5.6231e-01, -2.3818e-01,  5.7056e-03,\n",
       "           2.1569e-01,  3.5813e-01, -2.6926e-01,  3.5523e-03, -4.6185e-01,\n",
       "           4.7367e-01, -1.7571e-01, -1.3319e-01, -3.8660e-01, -6.4094e-01,\n",
       "           4.5721e-01,  6.3604e-02,  1.2044e-01,  1.5595e-01, -1.1025e+01,\n",
       "           4.2404e-02,  5.4334e-01, -5.4688e-01,  2.6733e-01, -1.4314e-01,\n",
       "           3.1749e-01, -5.4458e-01,  4.3840e-01, -2.6414e-01,  1.7073e-01,\n",
       "          -4.8513e-01,  1.1142e-01, -1.9697e-01, -6.8884e-01, -1.4465e-01,\n",
       "          -3.1721e-01, -6.0192e-01, -1.0247e-02, -1.8340e-01,  8.6313e-03,\n",
       "          -2.8297e-01,  1.1634e-01,  9.3654e-02, -1.1421e-01, -1.9497e-01,\n",
       "           1.2458e-01, -3.7677e-01,  3.4346e-01,  4.6021e-01,  3.8886e-01,\n",
       "           3.1642e-01,  6.2433e-01,  1.3460e-01, -1.1798e-02, -5.2588e-01,\n",
       "          -1.6370e-01, -3.0303e-01, -8.9269e-03, -1.8950e-01, -6.5345e-01,\n",
       "          -6.3711e-01, -3.5374e-01,  3.4220e-01,  3.7036e-01,  7.0316e-01,\n",
       "           8.6870e-01,  3.6866e-01,  5.7556e-02,  1.3533e-02, -2.3157e-01,\n",
       "          -8.7566e-01,  8.6933e-02, -5.4105e-01, -3.6820e-01, -2.7034e-01,\n",
       "           5.5938e-01, -7.2463e-01,  3.7153e-01, -4.1348e-01, -4.0242e-01,\n",
       "           6.0027e-01,  6.4177e-02, -8.4798e-02, -4.3395e-01, -4.7601e-01,\n",
       "           4.6093e-01, -3.8016e-01,  1.8293e-01, -2.3001e-01,  1.2896e-01,\n",
       "          -3.3483e-01,  6.3219e-01, -2.5432e-03, -1.6484e-01, -3.1644e-01,\n",
       "          -2.0441e-01, -3.6925e-01,  1.6034e+00, -1.6696e-01,  1.2966e-01,\n",
       "          -1.0415e-01,  3.6733e-01, -5.6696e-01, -7.6303e-01, -1.5781e-01,\n",
       "          -6.3085e-02, -1.1040e-01,  2.1474e-01,  6.3052e-01, -1.0645e-01,\n",
       "          -8.1491e-01,  4.9603e-03,  2.1200e-01,  4.3549e-01,  9.9023e-01,\n",
       "          -7.3265e-01, -1.3341e-02,  1.9920e-01,  2.8616e-01, -3.8444e-01,\n",
       "           6.0277e-02,  1.3596e-01, -3.0970e-01,  2.9662e-01, -7.4180e-03,\n",
       "           1.3231e-02,  3.8704e-01,  9.0435e-02,  4.8925e-01, -5.5418e-01,\n",
       "           1.7393e-01,  2.6596e-01,  2.3527e-01, -3.3337e-01,  3.4406e-01,\n",
       "           4.5800e-01,  5.9489e-01, -2.2815e-01, -3.1088e-01, -3.6427e-01,\n",
       "           3.4363e-01,  1.2801e-01, -6.8334e-02, -4.0900e-01,  1.3368e-01,\n",
       "          -5.6946e-01, -1.9001e-01,  1.9331e-01, -9.2655e-04,  1.7294e-01,\n",
       "           1.7772e-02, -3.9117e-01, -2.6555e-02, -2.3760e-01, -1.1802e+00,\n",
       "          -6.8778e-01, -3.1339e-01,  1.9909e-01, -1.9672e-01,  1.0772e-01,\n",
       "           9.7136e-02,  2.2722e-01, -1.2463e-01, -1.1058e-01, -9.1729e-02,\n",
       "          -5.8882e-02,  1.1916e-01,  8.1656e-02,  2.9589e-01, -3.2190e-01,\n",
       "           1.5846e-01, -8.3153e-02, -1.8996e-01, -3.1738e-02, -4.3791e-01,\n",
       "          -4.5530e-01, -2.2321e-01, -4.2563e-01,  7.1215e-02,  2.5183e-01,\n",
       "           3.6423e-01,  2.5293e-01,  2.2170e-01,  8.1565e-02,  5.6236e-01,\n",
       "           5.4606e-02, -2.7873e-01, -4.2496e-01, -3.6977e-01, -1.5839e-01,\n",
       "           2.4782e-01, -7.6044e-02, -1.9658e-01, -4.4643e-02, -1.2407e-01,\n",
       "          -1.9597e-01,  1.2568e+00, -6.6806e-02, -9.0921e-01,  2.9184e-01,\n",
       "           9.7446e-01, -5.5747e-02,  2.6545e-01,  1.8306e-01,  6.0206e-02,\n",
       "          -5.4065e-01, -3.0152e-01,  3.1530e-01, -1.3060e-02,  4.3962e-01,\n",
       "           3.8639e-01, -6.2370e-02,  1.5214e-01,  8.0873e-01,  1.3832e-01,\n",
       "           2.5183e-01, -2.6046e-02, -3.0195e-01,  2.6911e-02,  7.4304e-01,\n",
       "          -2.9239e-02, -8.1211e-02,  2.1284e-01, -9.9567e-02,  9.8525e-02,\n",
       "           3.5999e-01,  1.0024e+00, -2.1963e-01, -1.4962e-02,  1.1887e+00,\n",
       "           1.5389e-01,  4.1295e-01,  1.7349e-01,  4.4003e-01, -6.2632e-02,\n",
       "           1.1571e-01, -6.8442e-02, -9.5588e-02,  1.6814e-01,  2.6967e-01,\n",
       "           4.5108e-01, -7.2120e-01, -3.3220e-01,  6.2739e-02,  5.6529e-01,\n",
       "           1.1488e-01,  6.2271e-01, -4.0409e-01,  5.3385e-01,  1.0697e+00,\n",
       "          -1.1126e+00, -3.6281e-02, -1.8735e-01,  6.1817e-01,  1.2825e-01,\n",
       "          -2.1697e-01, -1.1644e-01,  1.9126e-02, -6.7079e-02,  3.2893e-01,\n",
       "           2.1435e-01, -9.2544e-01, -3.7570e-02,  3.3609e-01, -1.0560e-01,\n",
       "           2.1342e-01,  1.1023e-03,  3.0899e-01, -9.2216e-02, -1.0733e-01,\n",
       "          -1.5486e-01,  1.9122e-01, -2.0638e-01, -5.2654e-01,  2.9919e-02,\n",
       "           5.0135e-01,  1.3106e-01, -2.6650e-02,  2.0596e-01, -4.3434e-01,\n",
       "          -1.6626e-01, -4.1550e-01, -3.1808e-01,  1.0507e-01,  1.4869e-01,\n",
       "           6.6950e-02, -4.6921e-01, -9.5887e-02,  1.9641e-01, -2.6056e-01,\n",
       "           1.9189e-01,  4.9077e-01,  7.3880e-01, -1.4097e-01,  8.1571e-02,\n",
       "           1.2976e-01,  1.8300e-01, -1.2027e-01,  5.5778e-01, -2.3407e-01,\n",
       "          -5.9215e-01,  1.9163e-02,  3.0167e-01, -2.5520e-01,  5.4009e-02,\n",
       "          -3.2741e-01,  3.0002e-01,  8.3123e-02, -2.8867e-01,  3.0344e-01,\n",
       "           2.8359e-02,  3.4753e-01, -8.0032e-01,  5.9301e-01,  6.1707e-01,\n",
       "          -1.4195e-01,  7.7874e-01, -4.8835e-02,  5.1681e-01, -2.0956e-01,\n",
       "           2.7038e-02, -4.7089e-01, -4.2276e-01,  2.3055e-02, -1.4930e-02,\n",
       "          -5.3021e-01,  4.2665e-02, -2.4749e-01,  3.6119e-02,  6.0380e-01,\n",
       "           1.2651e+00, -2.0687e-01, -7.6018e-01, -4.4412e-01, -5.9586e-02,\n",
       "           5.1270e-02, -5.9312e-02,  2.5061e-01, -6.6872e-02,  1.3954e-01,\n",
       "          -7.6590e-01, -4.1477e-01,  1.6290e-01,  2.7919e-01, -3.8371e-01,\n",
       "           5.9459e-01,  7.1536e-02,  1.8082e-01,  2.0026e-01,  3.8244e-01,\n",
       "           2.8742e-01,  5.6264e-01, -8.1415e-02, -3.6389e-01, -4.2790e-03,\n",
       "           1.9430e-01, -5.0613e-02,  1.5830e-01, -6.1740e-01,  4.3809e-01,\n",
       "           5.1412e-02,  4.3279e-01,  4.3527e-01]], device='mps:0'),\n",
       " tensor([[-9.8608e-02, -6.2949e-01, -5.4737e-01, -4.6335e-01, -3.7454e-01,\n",
       "           1.0248e-01, -2.8097e-01,  4.4707e-01, -1.3869e-01,  5.9022e-01,\n",
       "           3.3979e-01,  3.6518e-02, -7.9829e-02, -1.9712e-01, -4.0961e-01,\n",
       "          -5.4994e-01,  1.7904e-01,  2.6107e-01,  2.6766e-01, -1.3636e-01,\n",
       "           9.8026e-01,  3.6199e-01, -8.5372e-02,  1.8939e-01, -6.9872e-01,\n",
       "           3.9986e-01,  4.5490e-02,  8.0399e-02,  4.6963e-01, -6.9729e-01,\n",
       "          -2.0828e-01,  5.4662e-01, -1.2219e-01,  1.1649e+00, -1.8525e-01,\n",
       "          -2.5129e-01,  4.1644e-01, -2.6874e-01,  8.0973e-01,  4.0156e-02,\n",
       "           6.2909e-01, -1.2839e-01,  2.5974e-01,  2.1204e-01,  2.5654e-01,\n",
       "          -1.6577e-02,  1.0736e-01, -6.2638e-01,  5.4556e-01, -5.1867e-01,\n",
       "          -2.1843e-01, -1.5953e-01,  3.6604e-01,  4.8748e-01, -6.7281e-02,\n",
       "           3.5607e-02,  3.0541e-01,  1.6232e-01, -2.5763e-01,  2.0386e-01,\n",
       "           1.3393e-01,  6.9371e-01,  5.7586e-02, -1.9525e-01, -3.0287e-01,\n",
       "           3.0687e-01, -5.2685e-01, -7.0972e-01, -3.4711e-01, -1.5503e-01,\n",
       "          -6.3145e-02,  3.6453e-01, -2.3968e-01, -8.3683e-02,  7.8001e-02,\n",
       "          -8.8895e-02,  3.9835e-02, -1.7901e-01, -2.7637e-01, -9.8806e-03,\n",
       "           6.4246e-02,  3.2661e-01, -1.9744e-01, -4.1654e-01, -2.1876e-01,\n",
       "           3.0475e-01,  7.8369e-01,  6.3690e-02,  2.5669e-01,  8.5968e-02,\n",
       "           3.1750e-01,  9.4549e-02, -3.4083e-01, -4.9471e-01,  3.1401e-01,\n",
       "          -1.3255e-01,  4.9667e-01, -1.4940e-01, -6.5687e-02, -4.7700e-02,\n",
       "           3.7095e-01,  7.2304e-02,  1.7323e-03,  1.1325e-01, -6.6663e-02,\n",
       "          -4.8542e-01,  1.4204e-01, -2.5931e-01, -6.6520e-01, -1.1522e-01,\n",
       "          -7.9637e-02,  1.3277e-01,  1.5561e-02, -3.1430e-01,  2.6189e-01,\n",
       "          -7.3846e-02,  1.9145e-01, -1.3642e-02, -3.3838e-01,  3.7293e-01,\n",
       "           2.5609e-01, -4.0848e-01,  2.2958e-02,  1.6951e-01,  4.6388e-01,\n",
       "          -2.2394e-01,  6.0049e-01, -2.8694e-01,  2.5392e-01,  4.8866e-01,\n",
       "           8.4207e-02, -6.0898e-02, -9.6990e-02, -1.7751e-01, -3.9547e-01,\n",
       "          -4.2915e-02,  7.3082e-02, -3.2798e-01,  1.0024e-01,  2.1027e-02,\n",
       "          -1.5484e-01, -1.5575e-01, -7.1363e-01, -3.2433e-01,  1.6862e-01,\n",
       "          -2.9423e-01,  2.9391e-01, -7.2839e-02, -5.6183e-01, -3.6233e-01,\n",
       "          -8.1793e-01, -4.5420e-01,  5.0598e-01,  3.1144e-02, -1.4616e-01,\n",
       "          -4.4906e-02,  1.9702e-01,  1.8656e-01, -1.9001e-02,  3.6581e-01,\n",
       "           4.5356e-01, -1.7970e-01,  5.5773e-02, -4.7383e-01,  5.1199e-02,\n",
       "           3.6512e-02, -1.1536e-01, -1.4678e-01, -1.8904e-01,  8.2812e-02,\n",
       "           1.8413e-01, -2.0916e-01, -2.0209e-02, -4.3147e-01,  1.4387e-01,\n",
       "          -9.5740e-02,  8.7516e-02,  2.8605e-01, -3.1460e-01, -3.4357e-01,\n",
       "          -4.4457e-01, -2.0415e+00,  1.3037e-01, -1.6708e-01, -1.9640e-01,\n",
       "          -7.8661e-01,  5.6989e-01,  4.0038e-01,  2.3491e-01,  1.2055e-02,\n",
       "          -4.1441e-01, -1.3256e-01, -2.2065e-01,  2.0346e-01, -1.5937e-01,\n",
       "           4.1104e-01,  3.6856e-01,  2.1222e-01, -8.2505e-02, -5.3417e-02,\n",
       "          -3.3993e-01,  1.9027e-01,  6.6698e-02,  4.3789e-02,  1.3752e-02,\n",
       "           8.6470e-02, -3.4516e-01,  1.5865e-01,  6.3413e-01, -3.2035e-01,\n",
       "          -2.6413e-01, -1.5432e-01, -3.1116e-01, -1.1825e-01, -2.1011e-01,\n",
       "           7.5833e-01, -1.0075e-01, -1.9090e-01,  4.1462e-01,  2.9186e-01,\n",
       "          -2.0576e-01,  2.6479e-02, -1.7243e-01, -4.6025e-01, -8.4004e-02,\n",
       "           8.3166e-01, -2.6809e-01, -1.8435e-01, -1.3576e-01, -2.8072e-01,\n",
       "           9.9627e-02, -3.9657e-01,  7.0865e-01,  1.3131e-01, -2.4437e-01,\n",
       "          -1.3250e-01, -4.9085e-01, -3.6081e-02, -6.5221e-02,  2.7934e-01,\n",
       "          -3.2480e-01, -4.4413e-01, -6.2940e-01,  9.8728e-02,  4.2580e-01,\n",
       "           1.0494e-01, -2.2581e-01,  5.8588e-01,  3.7420e-01,  5.5104e-01,\n",
       "          -7.6877e-02,  1.8477e-03,  4.0139e-01,  4.0454e-02, -3.1153e-01,\n",
       "           4.0137e-02,  1.1839e-01, -6.1671e-02,  2.8916e-01, -5.0007e-01,\n",
       "           2.1460e-01,  1.3048e-01,  4.2285e-02,  4.0869e-01, -3.3156e-01,\n",
       "           3.3759e-01, -6.4578e-04, -4.2352e-01, -2.2792e-01,  3.1587e-01,\n",
       "          -3.4318e-01,  4.7882e-01, -1.1497e-01, -1.3533e-01, -6.5793e-01,\n",
       "          -3.0023e-01,  9.9027e-02, -5.2265e-01,  4.0820e-02, -8.5922e-01,\n",
       "          -1.4457e-01,  4.3467e-01, -1.1843e-01,  2.8682e-01, -4.3268e-01,\n",
       "           2.7079e-01,  1.1150e-01,  1.0583e-01, -1.8427e-01, -4.8035e-01,\n",
       "           6.4368e-03,  3.5471e-01, -1.9635e-01,  6.7501e-02,  1.8728e-02,\n",
       "           1.4821e-01,  7.4914e-02, -1.4411e-01,  1.8116e-02, -3.9952e-01,\n",
       "           1.2334e-01,  2.7083e-01, -6.4203e-02, -1.1696e-01, -2.5127e-01,\n",
       "           8.1457e-02,  5.7341e-02,  4.5826e-01, -1.5574e-01, -4.7792e-01,\n",
       "           1.3602e-02,  5.6159e-01,  4.2917e-02,  9.5691e-02, -4.1476e-02,\n",
       "          -1.3511e+00, -3.8201e-01,  1.3646e-01, -2.3684e-01, -1.6250e-01,\n",
       "           1.4728e-01, -4.5766e-02, -1.0990e-01,  1.3434e-01, -3.3727e-01,\n",
       "           4.9913e-01, -7.4794e-03, -9.4830e-02,  2.6318e-01, -2.9595e-01,\n",
       "          -2.6245e-01, -5.5022e-02, -5.5963e-02, -1.9048e-01, -2.1108e-01,\n",
       "           1.0760e-01,  2.4895e-01, -1.6610e-02,  2.9324e-01, -4.2315e-01,\n",
       "           6.9357e-01,  2.0403e-02, -3.6256e-01,  1.8028e-01, -4.6817e-01,\n",
       "          -2.2470e-01,  6.2020e-01, -2.1111e-01,  1.7508e-01,  7.0164e-02,\n",
       "           2.2583e-01, -1.6263e-01, -2.5922e-01, -1.5555e-01,  5.6515e-01,\n",
       "           9.7645e-02,  3.8631e-01, -3.1465e-01,  4.6445e-01,  6.0830e-01,\n",
       "          -4.3876e-01,  2.7575e-01, -3.0324e-01, -3.5509e-01, -1.4900e-01,\n",
       "           3.7937e-01,  2.4503e-01, -2.4568e-01,  1.2754e-01,  2.1491e-01,\n",
       "           2.4189e-01, -2.8123e-01, -1.4375e-01,  1.5623e-01, -1.6800e-02,\n",
       "          -1.6159e-01, -5.9860e-01,  4.8387e-02, -4.0540e-01, -3.7598e-01,\n",
       "          -5.5542e-02, -9.6940e-02,  2.3018e-01, -3.4557e-01, -2.8109e-01,\n",
       "          -4.1082e-01,  1.5328e-01,  7.9392e-01, -8.2566e-02,  1.3355e-02,\n",
       "          -2.4290e-01,  1.3134e-01, -3.9568e-01,  5.6959e-01,  2.4837e-01,\n",
       "          -5.6746e-02,  5.0092e-01, -5.9799e-01,  3.2502e-01,  1.6325e-01,\n",
       "          -2.5296e-01,  7.9720e-02, -1.1265e-01,  4.2524e-01, -1.9241e-01,\n",
       "          -2.6666e-01,  1.8088e-01,  2.9568e-01,  8.0022e-02,  3.3680e-02,\n",
       "           8.9946e-02,  7.7279e-01, -5.6155e-01, -1.1355e-01, -3.2412e-02,\n",
       "           4.6990e-01, -1.4868e-01,  1.7842e-01, -2.5184e-01, -4.5494e-01,\n",
       "           2.8096e-01,  7.6220e-03,  3.0259e-02,  1.5848e-01, -1.1470e+01,\n",
       "           1.7761e-02,  7.9956e-01, -5.9407e-01,  3.9142e-02, -7.9952e-04,\n",
       "           1.4375e-01, -2.8145e-01,  3.6361e-03, -2.5851e-01, -3.7282e-01,\n",
       "           1.6253e-01,  5.8747e-02,  6.8827e-02, -1.9252e-01, -1.2168e-01,\n",
       "          -4.4456e-02, -9.0133e-01,  1.4013e-01, -2.9238e-01,  3.1633e-01,\n",
       "          -4.6944e-01,  3.5652e-01,  1.0775e-01,  4.8818e-02, -3.3732e-01,\n",
       "          -6.3534e-02, -4.1684e-01,  2.3683e-01,  1.0364e-01, -1.6207e-01,\n",
       "          -1.3703e-01, -1.5828e-01,  4.6034e-01, -1.8993e-01, -4.8947e-01,\n",
       "           1.5862e-01, -7.0142e-02,  4.5043e-01, -1.6798e-02, -2.3410e-01,\n",
       "           5.0011e-02, -4.1456e-01,  3.0195e-01,  6.4668e-02,  5.0457e-01,\n",
       "           4.7572e-01,  6.3490e-01,  1.0739e-01, -2.1848e-01,  4.9891e-02,\n",
       "          -8.9429e-01,  3.3171e-01, -9.8824e-01, -2.6344e-01,  3.6779e-02,\n",
       "           7.4622e-01, -4.6007e-01,  9.9323e-03,  8.2971e-02, -3.9312e-01,\n",
       "           2.4107e-01, -5.5413e-02,  1.0686e-01, -6.2270e-01, -1.1555e-01,\n",
       "           3.6365e-01,  1.0164e-01,  1.3567e-01, -3.5856e-01, -1.3223e-02,\n",
       "          -4.3049e-02,  5.9870e-01,  1.5647e-01, -2.5151e-01,  1.0429e-01,\n",
       "          -8.0721e-03,  2.3224e-01,  6.2748e-01,  4.7218e-02,  1.3164e-01,\n",
       "           1.7459e-01,  5.1672e-01, -5.4105e-01, -4.7941e-01, -1.2530e-01,\n",
       "           6.3722e-02,  2.8461e-01,  8.3486e-02,  3.6163e-01, -3.9872e-01,\n",
       "          -3.1281e-01, -7.8561e-02,  3.1281e-02,  5.0728e-02,  1.2098e+00,\n",
       "          -5.8519e-01, -2.2122e-01, -1.0256e-01, -9.1470e-02, -1.3571e-01,\n",
       "          -3.9828e-02, -3.3651e-01, -4.2370e-01,  3.3182e-01,  1.1019e-01,\n",
       "           1.7328e-01,  1.5282e-02,  3.9299e-01,  1.9321e-01, -5.2099e-01,\n",
       "           2.8514e-01,  1.4470e-01,  4.3958e-01, -1.8846e-01, -4.1365e-01,\n",
       "          -1.0991e-01,  3.3157e-01, -3.8281e-01, -3.4973e-01, -1.0983e-01,\n",
       "           3.0412e-02,  5.9583e-02, -5.8809e-02, -3.7567e-01,  2.5850e-01,\n",
       "          -6.2094e-01, -9.3514e-03, -4.2726e-01, -1.1183e-01,  4.7884e-01,\n",
       "           1.2318e-01, -3.6857e-02,  1.1137e-01, -4.2935e-01, -6.7978e-01,\n",
       "          -6.4684e-01, -7.4975e-02,  5.2269e-02, -3.7956e-01, -1.6842e-02,\n",
       "          -1.8671e-01,  6.6050e-02, -3.6548e-01,  1.8633e-01, -8.2184e-02,\n",
       "           7.5332e-01,  2.3533e-01, -1.8486e-01,  1.2410e-01, -1.4424e-01,\n",
       "          -3.6447e-01, -2.1959e-01, -2.4676e-01,  2.2621e-01, -5.0366e-01,\n",
       "          -1.3080e-01,  1.8011e-01,  1.1438e-01,  6.4878e-03, -3.4448e-01,\n",
       "          -1.2932e-01,  2.0570e-01, -1.3707e-01,  3.0997e-01,  1.7373e-01,\n",
       "           2.0329e-01, -6.4335e-02, -4.6287e-01,  8.7786e-02,  4.5179e-01,\n",
       "          -1.1348e-01, -2.7396e-01,  3.8658e-02, -7.0830e-01,  2.1125e-01,\n",
       "           3.8231e-01,  8.2746e-01, -2.5998e-01, -5.3996e-01,  4.6996e-01,\n",
       "           6.2219e-01, -4.6785e-01, -5.7024e-02, -2.1133e-01,  1.2442e-01,\n",
       "          -1.0488e-01, -3.4183e-01,  2.6033e-01, -2.5177e-01,  4.2375e-01,\n",
       "          -9.9348e-02,  2.0811e-01, -1.9413e-01,  4.1807e-01,  9.7647e-02,\n",
       "          -2.0238e-01, -3.5246e-02, -6.2741e-01, -4.3863e-01,  2.3873e-01,\n",
       "           2.6324e-01, -1.5522e-01, -2.3160e-02, -1.2275e-01,  1.1149e-01,\n",
       "           4.8271e-01,  8.5393e-01,  8.1221e-02,  1.4584e-01,  3.8317e-01,\n",
       "           2.3356e-01, -2.7630e-01,  5.9953e-02,  2.9270e-01,  4.0766e-01,\n",
       "           3.5259e-01,  5.3145e-01, -3.4176e-01,  7.2203e-02, -1.6344e-01,\n",
       "           2.0465e-01, -5.7619e-01, -7.6698e-01, -1.3126e-01,  4.3082e-01,\n",
       "           1.5541e-02,  2.9468e-01, -2.7851e-01,  3.4408e-01,  6.3093e-01,\n",
       "          -1.1306e+00, -5.9181e-02, -3.9846e-02, -9.3994e-02, -6.2591e-02,\n",
       "          -5.4897e-01, -2.2413e-01, -3.5234e-01, -2.4418e-01,  5.2852e-01,\n",
       "           5.8811e-01, -8.5286e-01,  1.6085e-01,  2.2953e-01, -1.5005e-01,\n",
       "           9.4328e-02, -9.1644e-02,  3.2863e-02,  5.0134e-01,  1.5175e-01,\n",
       "          -5.7932e-01,  4.5778e-01,  1.5595e-01, -4.3637e-01,  2.6165e-01,\n",
       "           4.1776e-01, -1.8687e-02,  1.5952e-01,  2.2085e-01, -4.9751e-01,\n",
       "          -1.0141e-02, -4.6200e-01, -1.0629e-01,  1.0751e-01, -6.7537e-02,\n",
       "           2.4815e-02, -4.9794e-01, -6.6236e-02,  3.4484e-01,  2.9505e-03,\n",
       "           5.0621e-01,  6.0668e-01,  7.6526e-01, -5.6484e-02, -2.1306e-01,\n",
       "           3.3852e-01,  2.5943e-01, -9.1834e-02,  5.2227e-01, -2.1885e-02,\n",
       "          -1.6835e-01,  3.1658e-01,  2.3124e-02, -2.2045e-01,  1.7294e-01,\n",
       "          -2.5761e-01,  2.6423e-01,  1.0870e-01, -4.9724e-01,  3.6477e-01,\n",
       "           3.7087e-01,  1.5082e-01, -3.0923e-01,  1.8171e-01,  3.9828e-01,\n",
       "          -3.1100e-01,  6.8356e-01,  1.6607e-04,  2.2973e-01,  6.3476e-02,\n",
       "          -1.1506e-01, -5.5647e-01, -3.1374e-01,  2.1509e-01,  9.2105e-02,\n",
       "          -5.0547e-01, -9.5287e-02,  1.5106e-01, -4.9544e-01,  7.8060e-01,\n",
       "           6.2385e-01, -3.7763e-01, -3.5664e-01, -7.1127e-02,  1.1295e-01,\n",
       "           1.2269e-01, -3.3879e-01,  2.9572e-01,  1.1312e-01,  1.0076e-01,\n",
       "          -1.0219e+00, -2.9326e-01,  2.4530e-01,  4.9147e-02, -4.2291e-01,\n",
       "           5.9939e-01,  7.9206e-02,  5.0359e-02,  3.8428e-01,  3.0534e-01,\n",
       "           1.4161e-01,  4.3910e-01,  7.0775e-02, -2.1375e-01,  6.5311e-02,\n",
       "           2.5600e-01, -3.0970e-01, -3.2766e-01, -6.3757e-01,  5.3684e-01,\n",
       "           1.5068e-01,  8.4160e-01,  6.5313e-01]], device='mps:0'),\n",
       " tensor([[-8.4367e-02, -9.9544e-01, -7.6676e-01, -3.6066e-01, -6.5543e-01,\n",
       "           2.5891e-01, -3.0088e-01,  6.8470e-01,  1.0237e-01,  6.3124e-01,\n",
       "           4.0418e-01, -2.5821e-01,  6.0564e-02, -3.4243e-01, -4.9831e-01,\n",
       "          -4.7098e-01,  4.0723e-01,  4.7123e-01,  4.3827e-01,  4.8314e-01,\n",
       "           5.9683e-01,  2.7857e-01, -1.0076e-01,  3.3544e-02, -6.4546e-01,\n",
       "           2.4330e-01, -4.2110e-02,  1.3205e-01,  4.8342e-01, -4.3177e-01,\n",
       "          -4.6548e-02,  4.0897e-01,  1.0692e-02,  1.0297e+00, -2.5928e-01,\n",
       "          -1.6171e-01,  1.2562e-01, -3.9049e-02,  2.4548e-01, -2.8336e-01,\n",
       "           7.2436e-01, -1.9736e-03,  3.3338e-01,  2.1337e-01,  3.8305e-01,\n",
       "          -4.1797e-02,  1.1012e-01, -5.8827e-01, -2.5325e-01, -1.1852e-01,\n",
       "           9.8159e-02, -2.6861e-01,  5.2367e-01,  3.3658e-01, -1.1884e-01,\n",
       "          -1.3784e-01,  6.1704e-02,  4.8597e-01, -4.4127e-01,  1.2459e-01,\n",
       "           6.3577e-02,  7.6844e-01,  4.8226e-02,  7.8531e-02, -3.4758e-02,\n",
       "           2.2771e-01, -7.6413e-01, -4.7590e-01, -2.8223e-01,  2.3143e-01,\n",
       "          -5.1286e-02,  2.1924e-01, -1.8105e-01,  5.5858e-02,  4.0894e-01,\n",
       "          -6.0636e-01,  1.2790e-01, -1.5663e-01, -7.6339e-01, -2.8983e-01,\n",
       "           3.1167e-01,  8.5711e-02, -2.1309e-01, -2.3854e-01, -1.6624e-01,\n",
       "          -2.1786e-02,  4.7257e-01,  1.3428e-01,  1.3315e-01, -1.4904e-03,\n",
       "           3.2695e-01,  1.2292e-01, -4.4498e-01, -3.7081e-01,  4.3861e-01,\n",
       "          -3.1129e-01,  3.6942e-01, -1.1072e-01, -1.9192e-01, -1.7740e-01,\n",
       "          -7.3041e-02,  3.6625e-01,  1.3063e-01,  1.0233e-02, -4.2244e-01,\n",
       "          -4.4462e-01,  2.1118e-01, -1.0029e-01, -5.8134e-01,  4.1833e-02,\n",
       "          -9.9223e-02, -1.3356e-01,  2.7734e-01, -7.8352e-01, -2.7233e-01,\n",
       "          -7.4877e-02,  2.0114e-01, -2.1265e-01, -3.4069e-01,  4.7558e-01,\n",
       "           2.9456e-01, -5.7180e-01,  1.8192e-01,  1.8892e-01,  1.9922e-01,\n",
       "          -2.6990e-01,  2.7019e-01, -1.4557e-01,  2.3345e-01,  4.2545e-01,\n",
       "          -2.2775e-01,  1.6894e-02, -7.4285e-02, -2.3522e-01, -2.0285e-01,\n",
       "          -1.6799e-01,  3.0267e-01, -6.2164e-01,  2.7428e-01,  1.9522e-01,\n",
       "          -4.6284e-02,  3.6605e-02, -5.0484e-01, -1.1827e-02,  1.7577e-01,\n",
       "          -1.4853e-01,  1.4923e-01, -8.2681e-02, -3.4502e-01, -1.4070e-01,\n",
       "          -7.9769e-01, -4.7416e-01,  3.6416e-01,  1.5173e-02,  1.5206e-02,\n",
       "           7.0536e-02,  1.8764e-01,  1.3009e-01, -1.2432e-01,  9.6329e-02,\n",
       "           5.3422e-01, -5.2765e-02,  2.5112e-01, -3.6691e-01,  4.4943e-03,\n",
       "           1.1869e-01,  2.7898e-01, -2.2534e-01,  2.5080e-01,  1.4270e-01,\n",
       "           8.2778e-02, -5.4946e-01,  6.5663e-02, -2.3727e-01,  7.5774e-02,\n",
       "          -3.8194e-01,  7.1762e-02,  2.8576e-01, -2.1230e-01, -2.8791e-02,\n",
       "          -6.3730e-02, -1.8550e+00,  1.1640e-01, -1.6118e-01,  1.6947e-01,\n",
       "          -3.5909e-01,  4.1989e-01,  4.0444e-02,  3.5049e-01, -2.3511e-01,\n",
       "          -2.0530e-01, -2.9160e-01,  2.0532e-02,  2.8190e-01,  1.1582e-01,\n",
       "           4.0659e-01,  6.2690e-01,  3.4601e-01, -3.3775e-01, -2.3636e-01,\n",
       "          -4.3631e-01,  7.5651e-02, -1.2973e-01,  5.2947e-01,  1.2589e-01,\n",
       "           2.2308e-01, -3.0865e-01, -8.1769e-02,  3.6243e-01, -1.5496e-01,\n",
       "           6.8469e-03, -3.7726e-01, -4.4730e-01, -2.0761e-01, -1.8935e-01,\n",
       "           6.5341e-01, -1.0807e-01, -1.5733e-01,  2.5209e-01,  3.2275e-01,\n",
       "          -1.7643e-01, -2.2036e-01, -1.1321e-01, -4.1263e-01, -2.3336e-01,\n",
       "           1.7229e-01, -2.7326e-01, -3.3327e-02, -5.0663e-01, -1.4497e-01,\n",
       "           1.7484e-01, -4.6704e-01,  7.1167e-01,  2.3259e-01, -2.9028e-01,\n",
       "          -3.6854e-01, -2.8246e-01,  1.6381e-01, -3.8400e-01,  9.8825e-02,\n",
       "          -3.9668e-01, -7.2905e-01, -6.7199e-01,  6.7727e-02,  2.6025e-01,\n",
       "           1.7070e-01, -2.1004e-01,  1.0842e+00,  2.4469e-01,  1.7280e-01,\n",
       "          -2.1477e-01, -1.7300e-01,  2.7311e-01, -3.1000e-02,  2.2551e-03,\n",
       "          -6.6367e-02,  1.5160e-02, -2.3197e-01,  4.9427e-01, -5.4481e-01,\n",
       "           1.4050e-02, -7.2921e-02, -7.7508e-02,  3.8728e-01, -3.1718e-01,\n",
       "           3.8960e-01,  2.6494e-01, -6.7055e-01, -7.7087e-02,  2.8328e-01,\n",
       "          -5.5764e-01,  4.9447e-01,  2.9289e-01, -4.2483e-01, -4.6639e-01,\n",
       "          -7.4040e-01,  1.2654e-01, -4.4153e-01,  9.3556e-03, -8.4792e-01,\n",
       "           8.9031e-02,  6.9095e-03,  2.7660e-01,  1.3241e-01, -3.5555e-02,\n",
       "           1.6525e-01,  4.0771e-01, -2.7235e-02, -5.2918e-02, -5.0872e-01,\n",
       "           4.6902e-02,  3.2414e-01, -3.0145e-01,  4.0338e-01,  1.7120e-01,\n",
       "          -1.0706e-01, -6.0042e-02,  1.0466e-01,  1.2865e-01, -2.3268e-01,\n",
       "          -3.1868e-01, -2.0738e-01, -5.5217e-03, -2.7404e-01, -1.1589e-01,\n",
       "           1.1605e-01,  2.0180e-01,  9.5597e-01, -1.4523e-01, -4.7065e-01,\n",
       "           1.4229e-02,  4.3059e-01, -1.0153e-01,  1.3948e-01, -3.4498e-01,\n",
       "          -1.4564e+00, -1.6000e-01, -1.2551e-01, -1.1716e-02, -2.4961e-01,\n",
       "           4.4710e-01,  1.4258e-01,  2.1673e-01, -6.6838e-02, -3.1554e-01,\n",
       "           4.2186e-01,  1.8045e-01,  1.5301e-01, -1.9014e-01, -7.6215e-02,\n",
       "          -1.8093e-01,  5.9824e-02, -1.8624e-01, -1.2212e-01,  2.5387e-01,\n",
       "           2.1246e-01,  4.2682e-01,  1.3010e-01,  1.4766e-01, -2.8905e-01,\n",
       "           8.0982e-01, -1.5348e-01, -2.3238e-01,  4.7025e-01, -4.1200e-01,\n",
       "          -4.8283e-01,  4.3726e-01,  1.6591e-01, -1.1289e-01,  1.0452e-01,\n",
       "           4.9590e-01,  1.5858e-01, -2.9403e-01,  1.1842e-01,  2.8889e-01,\n",
       "           4.3753e-01,  4.8374e-01, -1.8678e-01,  5.7783e-01,  8.2683e-01,\n",
       "          -1.6468e-01,  1.5626e-01,  9.6651e-02, -1.0880e-01, -2.0768e-01,\n",
       "           2.2006e-01,  3.2807e-01, -5.4607e-01,  7.2044e-02, -8.0584e-02,\n",
       "           8.3925e-02, -5.2893e-01, -2.0362e-01,  1.1441e-01,  2.3850e-01,\n",
       "          -2.9664e-01, -3.9601e-01,  7.0843e-01, -4.8620e-01, -5.9999e-01,\n",
       "           1.3799e-02,  1.2048e-01,  1.0193e-01, -2.2993e-01, -7.3144e-01,\n",
       "          -3.8818e-01,  5.5136e-02,  7.8710e-01,  1.3443e-01,  1.8723e-01,\n",
       "          -1.3401e-01, -2.3594e-02, -2.5439e-01,  3.4287e-01, -2.7282e-02,\n",
       "          -1.5088e-01,  2.4871e-01, -1.6269e-01,  1.5134e-01,  2.0525e-01,\n",
       "          -1.5518e-01, -1.8537e-01,  2.6337e-02,  1.3023e-02, -3.4143e-01,\n",
       "          -3.6412e-01,  7.6576e-02,  4.4562e-01, -1.4358e-01,  4.0201e-02,\n",
       "           1.6336e-01,  2.4095e-01, -1.2428e-01,  1.3924e-01,  1.2610e-01,\n",
       "           3.6558e-01, -2.7816e-01, -3.8846e-02, -1.9736e-01, -7.2080e-01,\n",
       "           4.9454e-01,  3.9282e-01,  5.0177e-01,  1.7274e-01, -1.1325e+01,\n",
       "           4.3340e-01,  6.2296e-01, -5.5520e-01, -2.8520e-02,  1.5383e-01,\n",
       "           1.3363e-01, -5.5452e-01, -6.2163e-02, -3.7226e-01, -3.4584e-01,\n",
       "           8.8002e-02,  8.1150e-03,  1.6451e-01, -2.7741e-01,  9.2481e-02,\n",
       "          -2.6656e-01, -1.0030e+00,  2.5486e-01,  1.4193e-01,  1.7161e-01,\n",
       "           3.0074e-02,  1.5006e-01, -8.5472e-02,  4.3966e-03, -5.8248e-01,\n",
       "           1.6209e-01, -4.4367e-01,  4.3888e-03,  2.3013e-01, -1.6589e-01,\n",
       "           1.7321e-01,  3.0756e-01,  1.9118e-01, -3.4465e-01, -5.2074e-01,\n",
       "           8.2137e-02,  3.2416e-01,  2.1041e-01,  1.4966e-01, -8.6932e-02,\n",
       "          -1.0027e-01, -2.9252e-01,  2.5918e-01, -9.2526e-02,  7.5756e-01,\n",
       "           6.8354e-01,  9.1070e-01,  2.1976e-02,  5.6146e-02, -9.8465e-03,\n",
       "          -8.3987e-01,  3.4823e-01, -1.0675e+00,  2.4378e-01,  1.7914e-01,\n",
       "           6.6129e-01, -7.1310e-01,  1.1627e-01, -1.5075e-01, -4.4335e-01,\n",
       "           1.9736e-01,  1.9250e-02,  4.8331e-01, -2.3359e-01, -2.7635e-01,\n",
       "           3.2073e-01,  3.3983e-01,  3.7253e-01, -1.7779e-01,  3.9588e-01,\n",
       "          -2.7989e-01,  4.7045e-01,  4.2343e-01, -7.4675e-01, -1.4220e-02,\n",
       "           1.9435e-02,  1.9220e-01,  7.3075e-01,  1.1571e-01, -9.6283e-02,\n",
       "           5.9384e-02,  3.9926e-01, -5.3791e-01, -5.9717e-01,  1.0659e-01,\n",
       "           1.0427e-01,  5.1462e-02,  1.7467e-01,  6.6462e-01, -2.5593e-01,\n",
       "          -5.5864e-01, -1.5824e-01, -1.7096e-01,  1.2117e-01,  7.3767e-01,\n",
       "          -4.2521e-01,  1.0810e-02,  2.9120e-01,  4.9301e-02, -6.4409e-01,\n",
       "          -2.8414e-01, -3.9714e-02, -2.6439e-01,  4.3758e-01,  8.6725e-02,\n",
       "           4.8441e-02,  2.2804e-01,  1.0790e-01,  1.3781e-01, -6.2136e-01,\n",
       "           4.0709e-01, -1.4452e-01,  1.2176e-01, -1.8181e-01, -4.2195e-01,\n",
       "           5.2464e-02,  3.3992e-01, -4.2048e-01, -4.2903e-01, -4.0213e-01,\n",
       "           1.3676e-01,  1.4312e-01, -9.5148e-02, -4.2635e-01,  2.5081e-01,\n",
       "          -5.6908e-01, -1.1600e-02, -2.2168e-01,  1.5547e-01,  2.9300e-01,\n",
       "           1.4328e-01,  7.5556e-01,  4.3220e-01, -1.4347e-01, -9.0420e-01,\n",
       "          -6.6719e-01, -4.3798e-02,  5.6063e-02, -3.1811e-01, -1.4041e-01,\n",
       "          -2.3530e-02, -9.1374e-02, -6.8794e-02,  1.6647e-01, -3.5942e-01,\n",
       "           5.6077e-01,  3.8020e-01,  7.2420e-03, -1.6792e-01, -2.5265e-01,\n",
       "          -2.3487e-01, -4.4025e-01, -1.7584e-01, -4.7575e-02, -4.0234e-01,\n",
       "          -2.7873e-01,  4.8451e-02, -3.7576e-01,  3.7169e-01, -7.9503e-02,\n",
       "           2.0456e-02, -2.5830e-01, -1.1124e-01,  2.1625e-01,  2.2133e-01,\n",
       "           1.4021e-01, -2.2102e-01, -7.9863e-01,  5.5749e-02,  1.2462e-01,\n",
       "           7.0927e-02, -1.5994e-01, -1.7651e-01, -5.8042e-01,  1.8801e-01,\n",
       "          -1.4829e-02,  1.1257e+00, -2.6214e-01, -4.0665e-01,  3.0422e-01,\n",
       "           7.1301e-01, -2.3407e-01,  1.5979e-01, -1.4556e-01,  3.7199e-01,\n",
       "          -8.7036e-02, -3.9455e-01,  2.9156e-01,  1.4516e-01,  3.7153e-01,\n",
       "           4.3518e-01,  1.1903e-01, -7.0225e-02,  2.3938e-01, -3.8350e-01,\n",
       "          -2.3578e-01, -9.9270e-02, -1.0127e-01,  1.8560e-02,  2.4009e-01,\n",
       "           5.1393e-02, -7.1133e-03, -6.7573e-02,  1.4515e-01,  1.7819e-01,\n",
       "           2.0262e-01,  1.3163e+00, -1.5479e-01, -1.1003e-01,  3.1455e-01,\n",
       "           1.8460e-01, -4.0461e-01,  1.0888e-01,  8.3768e-02,  2.6697e-01,\n",
       "           4.1623e-01,  9.5291e-02, -4.3354e-01,  1.3466e-01,  9.9492e-02,\n",
       "           1.6937e-01, -6.7153e-01, -4.1693e-01, -9.9818e-02,  5.2303e-01,\n",
       "           1.9350e-01,  1.7095e-01, -2.9902e-01,  4.5855e-01,  5.6058e-01,\n",
       "          -9.1975e-01, -1.6790e-02, -3.4235e-01,  5.4188e-01,  4.3356e-03,\n",
       "          -5.7435e-01, -2.7842e-01, -2.8756e-01, -1.4246e-01,  3.8933e-01,\n",
       "           4.0278e-01, -9.0740e-01,  1.4352e-01,  9.5229e-02, -4.5865e-01,\n",
       "           8.9898e-02, -1.8778e-01,  4.1324e-01,  2.7071e-01, -1.7722e-01,\n",
       "          -1.6717e-01,  3.1949e-01, -2.2758e-01, -4.9559e-01,  1.2165e-02,\n",
       "           5.2458e-01, -6.8425e-02, -8.0264e-02, -5.9458e-02, -5.3224e-01,\n",
       "          -4.4947e-03, -2.4609e-01, -2.1233e-01,  4.1113e-02, -2.5033e-01,\n",
       "          -1.6418e-01, -4.7583e-01,  8.7349e-02,  4.4863e-01, -4.0693e-01,\n",
       "           6.9590e-01,  7.7167e-01,  4.7521e-01, -1.3519e-01,  2.7051e-01,\n",
       "           3.8944e-02,  1.8352e-01, -1.4401e-01,  6.3044e-01, -1.3380e-01,\n",
       "          -5.2214e-01,  4.4726e-01,  5.2590e-02, -1.8820e-01, -1.5596e-01,\n",
       "          -3.6592e-01,  3.3802e-01,  2.2102e-01, -2.7085e-01,  4.2224e-01,\n",
       "           2.5655e-01, -2.7179e-02, -4.9229e-01,  2.4019e-01,  7.9900e-02,\n",
       "          -2.9798e-01,  9.0969e-01, -3.6684e-02,  7.0918e-01, -1.7199e-01,\n",
       "           3.7460e-01, -6.4847e-01, -6.1896e-01, -1.0694e-02,  2.7532e-02,\n",
       "          -5.0267e-01, -2.0316e-02,  3.6203e-01,  4.1920e-02,  6.7945e-01,\n",
       "           6.3578e-01, -3.4671e-01, -6.0193e-01, -1.3692e-01,  7.8214e-02,\n",
       "           1.4914e-01, -3.0498e-01,  3.4273e-01, -3.2149e-01,  2.0122e-01,\n",
       "          -8.2048e-01, -5.0978e-01,  1.7019e-01,  1.9072e-01, -4.4355e-01,\n",
       "           5.2155e-01, -2.3080e-01,  8.5308e-02,  5.8273e-01,  2.3058e-01,\n",
       "           4.1328e-01,  4.0443e-01,  7.2632e-02, -6.0968e-01,  6.8929e-02,\n",
       "           6.6616e-02, -2.7373e-01,  5.3488e-02, -3.7202e-01,  4.2463e-01,\n",
       "           6.7098e-02,  4.2283e-01,  6.8514e-01]], device='mps:0'),\n",
       " tensor([[-5.7009e-01, -7.0616e-01, -7.7654e-01, -9.8719e-01, -3.6567e-01,\n",
       "          -2.5630e-01, -5.9328e-01,  7.3188e-01, -1.3917e-01,  1.4983e-01,\n",
       "           3.0276e-01,  3.9456e-02,  2.9582e-01,  9.8687e-02, -3.7405e-01,\n",
       "           1.4522e-01,  3.7094e-01,  1.9387e-01,  2.4392e-01, -3.1031e-02,\n",
       "           1.0322e+00,  1.3952e-01, -1.1627e-01, -8.6595e-02, -3.9327e-01,\n",
       "           1.7716e-01, -4.1484e-01, -9.9995e-02,  3.3992e-01, -6.6918e-01,\n",
       "          -3.7707e-01,  8.9693e-02, -3.6732e-01,  9.2610e-01, -2.2768e-01,\n",
       "          -2.7508e-01,  2.9704e-02, -6.0159e-01,  6.4213e-01, -2.1046e-01,\n",
       "           2.2804e-01,  1.1963e-01, -1.8568e-01,  3.5767e-01,  8.6975e-02,\n",
       "           1.8528e-01, -2.3806e-02, -7.3258e-01,  7.6715e-02, -3.5348e-01,\n",
       "          -9.4658e-03, -8.4049e-02,  3.7771e-01,  3.6464e-01, -7.4922e-01,\n",
       "           2.2046e-02,  1.7758e-01,  3.7807e-02, -4.0473e-01,  2.5494e-01,\n",
       "           7.9635e-02,  7.1502e-01,  4.7596e-01,  2.1112e-01, -2.6314e-02,\n",
       "          -2.8790e-01, -6.1648e-01, -4.8644e-01, -3.2656e-01, -1.9755e-01,\n",
       "          -3.6959e-02,  1.4623e-02,  1.7640e-01, -1.1145e-01, -2.9496e-01,\n",
       "          -1.6810e-01,  2.4455e-01, -2.2076e-01, -5.6330e-01, -1.2318e-01,\n",
       "           3.4677e-01, -8.0865e-02, -4.3615e-02,  6.3321e-02, -1.6216e-01,\n",
       "          -7.3429e-03,  1.7331e-01, -2.7363e-02,  4.1080e-01,  1.5982e-01,\n",
       "           6.7130e-01, -1.5193e-01,  5.6972e-02, -4.4391e-01,  4.4627e-01,\n",
       "          -2.2024e-01,  3.5710e-01, -3.9368e-02,  7.9424e-02, -3.1228e-01,\n",
       "          -7.3658e-02,  3.0712e-01,  5.5042e-02,  9.0599e-02, -2.4678e-01,\n",
       "          -1.6671e-01, -9.8450e-02, -6.3440e-02, -5.5606e-01,  1.3837e-02,\n",
       "          -2.3955e-02, -4.6652e-01,  1.0598e-01, -5.1628e-01, -5.4694e-01,\n",
       "           3.0904e-01, -4.0919e-01, -4.0399e-01, -5.7538e-02,  1.2257e+00,\n",
       "           4.3965e-01, -3.6422e-01, -8.6200e-02, -1.2102e-02,  5.3759e-01,\n",
       "          -2.3719e-01,  1.6814e-01, -2.3635e-01,  2.2341e-01,  1.3253e-01,\n",
       "           5.6146e-02,  2.4904e-01, -4.5169e-01,  6.7593e-02,  1.9881e-01,\n",
       "          -1.4378e-01, -4.6994e-02, -3.6101e-01, -1.5688e-01, -1.6449e-01,\n",
       "           8.1545e-03, -1.2412e-01, -2.9320e-01,  8.9240e-02,  1.1247e-01,\n",
       "          -1.1443e-01, -1.2694e-01,  1.1758e-01,  8.4576e-02, -3.8067e-01,\n",
       "          -7.1266e-01, -1.2814e-01, -9.7195e-03,  7.3999e-02, -1.5344e-01,\n",
       "          -3.2034e-01,  5.3839e-01,  1.4508e-01,  9.2065e-02,  1.7652e-01,\n",
       "           4.2707e-01, -5.0711e-02,  3.9180e-01, -2.2431e-01,  2.9667e-01,\n",
       "           1.2127e-02, -5.7919e-01, -1.6867e-01,  2.3637e-01,  1.2929e-01,\n",
       "           5.1821e-01,  2.6464e-01, -9.2810e-03, -5.3712e-01, -3.8710e-02,\n",
       "           2.5342e-01,  6.1029e-01,  3.9373e-01, -2.4661e-01, -8.3068e-02,\n",
       "          -2.3387e-01, -1.9842e+00,  1.8765e-01, -5.7396e-01, -5.8766e-01,\n",
       "          -1.8809e-01,  2.2427e-01,  3.3606e-01,  4.3544e-01, -3.2097e-02,\n",
       "          -8.1912e-02,  3.7852e-03,  4.4665e-01,  7.6598e-01,  6.3606e-02,\n",
       "           4.9322e-01,  4.7313e-01,  2.1259e-03,  8.1333e-02, -2.6024e-02,\n",
       "          -5.4774e-01,  1.6095e-01,  9.9883e-02,  3.0353e-01, -7.8005e-02,\n",
       "           2.4050e-01, -2.6641e-01,  6.5831e-02,  4.7276e-01,  2.6703e-01,\n",
       "          -1.9159e-01, -3.1873e-01, -2.6361e-01, -1.4752e-01, -4.0831e-01,\n",
       "           1.4471e-01, -5.0283e-01, -8.6340e-02,  3.9664e-01,  1.1666e-01,\n",
       "           1.2062e-01, -2.6169e-01, -2.7931e-01, -2.6284e-01, -1.8121e-01,\n",
       "           4.4782e-01, -1.6591e-01, -1.1408e-01, -1.4700e-01, -2.4898e-01,\n",
       "          -7.5977e-02, -7.6446e-01,  3.9213e-01,  1.6836e-01,  3.2082e-01,\n",
       "          -1.6143e-01, -3.6315e-01,  3.4564e-01, -3.9865e-02, -3.3263e-02,\n",
       "          -1.0186e-01, -3.2834e-01, -5.4156e-01, -2.2352e-01, -2.5976e-03,\n",
       "           1.9511e-01,  5.5767e-02,  5.1024e-01,  7.9949e-01,  2.1341e-01,\n",
       "           1.3877e-02,  1.4214e-01,  4.7758e-01,  3.1735e-01, -5.4252e-01,\n",
       "           8.6942e-02,  3.5649e-02, -5.1474e-01,  3.1354e-01, -5.3173e-01,\n",
       "          -2.1116e-01, -8.2136e-02, -5.6981e-02,  6.2094e-01, -3.5166e-01,\n",
       "           1.9877e-01,  1.1213e-01,  7.3899e-03, -6.8394e-02,  3.8196e-01,\n",
       "          -6.1278e-01,  6.5257e-01, -5.8941e-03, -1.0083e-01, -7.3893e-01,\n",
       "          -5.8777e-01,  3.4423e-01,  2.2066e-01, -1.0786e-01, -4.4334e-01,\n",
       "           1.2171e-01, -2.0711e-01, -2.2283e-01, -2.0115e-02, -2.6458e-01,\n",
       "           3.5551e-01, -8.8462e-02,  4.5402e-02,  1.4040e-01, -2.9907e-01,\n",
       "          -7.8327e-01,  1.2181e-01, -1.1090e-01,  1.7142e-02,  2.1471e-01,\n",
       "           2.0621e-01,  1.7010e-01,  3.9476e-01, -1.0212e-03, -2.6620e-01,\n",
       "          -2.3867e-01, -5.8177e-02, -2.1834e-01, -1.9893e-01,  1.0638e-01,\n",
       "           6.3627e-01, -1.7385e-01,  4.9350e-01, -8.2697e-02, -6.4181e-01,\n",
       "           2.6336e-01,  9.4825e-01, -2.3364e-02,  2.1056e-01, -3.0474e-01,\n",
       "          -1.2817e+00, -1.9741e-01, -3.2547e-01, -2.5819e-01, -6.1732e-01,\n",
       "           3.6977e-01, -1.8343e-01, -2.6915e-03, -2.1333e-02, -1.8511e-01,\n",
       "           8.1074e-02,  5.9297e-01,  3.2062e-01,  1.3399e-01, -2.5531e-01,\n",
       "           2.6919e-03, -2.9727e-01, -2.3054e-01, -2.4627e-01,  8.1521e-02,\n",
       "           1.2796e-01,  2.8664e-01, -6.2438e-02,  1.6109e-01, -3.2997e-01,\n",
       "           6.4002e-01, -1.4559e-01,  4.0607e-02,  3.1353e-01, -2.0117e-01,\n",
       "          -1.5002e-01,  3.9336e-01, -2.3244e-01,  2.5442e-01,  4.1024e-02,\n",
       "           3.1503e-01, -5.2437e-01, -4.2163e-01,  1.7930e-01,  3.7983e-01,\n",
       "           4.1380e-01,  4.6159e-01, -1.8063e-01,  3.2263e-01,  2.2435e-01,\n",
       "          -3.4466e-01,  3.1789e-01, -3.8466e-01, -7.4143e-02,  7.9789e-02,\n",
       "           2.0190e-01,  1.2090e-01, -2.1209e-01, -8.3214e-02, -7.1386e-02,\n",
       "           3.4720e-01, -7.5740e-01,  3.9949e-02,  3.4476e-01, -2.4606e-01,\n",
       "          -1.6581e-01, -5.7513e-01,  1.6394e-01, -8.6867e-01, -4.3655e-01,\n",
       "          -2.5662e-02, -2.1618e-01,  5.8749e-02, -1.4333e-01, -7.8117e-02,\n",
       "          -1.2008e-02, -2.0377e-01,  4.1521e-01,  3.1926e-01,  7.9173e-02,\n",
       "           1.8330e-01, -1.0445e-01, -1.9646e-01,  2.9591e-01, -4.8153e-01,\n",
       "          -2.7900e-01,  7.7318e-02, -4.3749e-01,  1.6298e-02,  1.3116e-01,\n",
       "          -4.9610e-01, -2.0398e-01,  1.9597e-01, -4.4736e-02, -2.4319e-01,\n",
       "          -2.9726e-01,  1.5671e-02,  5.4157e-01, -1.6570e-01,  5.4771e-02,\n",
       "           1.0225e-02,  4.3404e-01, -1.1906e-01,  2.6383e-01,  2.0142e-01,\n",
       "          -1.4425e-01, -2.2780e-01, -1.2713e-01, -1.1441e-01, -3.1085e-01,\n",
       "           4.7883e-01,  3.3386e-02,  1.9083e-02,  3.1080e-01, -1.1331e+01,\n",
       "          -8.5964e-02,  8.3270e-01, -5.4294e-01, -1.3088e-01, -7.6311e-02,\n",
       "           7.2404e-02, -3.2318e-01,  1.1873e-01, -6.8110e-02, -3.2078e-01,\n",
       "          -3.0743e-01,  1.0373e-01, -2.4920e-01, -5.4285e-01, -2.6106e-01,\n",
       "          -3.6598e-01, -3.6439e-01,  1.7656e-01, -1.0455e-01,  2.5356e-01,\n",
       "          -7.1738e-01,  2.9476e-01,  3.6538e-01, -3.8715e-01, -5.8834e-01,\n",
       "          -5.4509e-01, -1.8663e-01,  1.6010e-01, -2.7008e-02, -2.6430e-01,\n",
       "           1.1522e-01, -1.2302e-01,  6.2389e-01, -1.9110e-02, -6.5540e-01,\n",
       "           5.7875e-02, -1.3937e-02,  5.6140e-01, -2.8702e-01,  6.9802e-02,\n",
       "          -2.1433e-01,  6.7109e-02,  3.5213e-01,  4.2667e-01,  7.1800e-01,\n",
       "           1.2492e+00,  5.3842e-01,  2.1806e-01,  9.8474e-02, -2.5419e-01,\n",
       "          -6.4352e-01,  5.1851e-01, -8.4375e-01,  7.5178e-02, -1.6625e-01,\n",
       "           5.2268e-01, -4.6953e-01, -2.0117e-01, -1.6704e-01, -9.1179e-02,\n",
       "           2.3032e-01,  1.8785e-01,  1.5994e-01, -6.2593e-02, -5.4343e-01,\n",
       "           6.4759e-01,  4.2481e-02,  8.4811e-02, -1.2425e-01, -7.4098e-02,\n",
       "          -2.5627e-01,  7.6494e-01,  2.2170e-02,  2.6402e-01,  4.4012e-01,\n",
       "           7.6412e-02,  2.0867e-01,  9.6449e-01, -8.8395e-02,  1.2303e-01,\n",
       "           5.2642e-01, -1.5003e-01, -3.0321e-01, -7.4052e-01,  4.0759e-02,\n",
       "          -3.2565e-02,  3.0725e-01,  2.0922e-01,  6.6175e-01, -2.0000e-01,\n",
       "          -2.2939e-01, -5.0617e-01, -3.1379e-01,  4.7640e-01,  8.9183e-01,\n",
       "          -4.5251e-01,  1.3039e-01, -1.5541e-01, -3.2147e-02, -5.3570e-01,\n",
       "          -6.7508e-02, -9.9119e-02, -5.7476e-01,  2.0181e-01, -2.8577e-01,\n",
       "           3.3467e-02,  1.6321e-01,  2.8508e-01,  5.7019e-01, -1.8868e-01,\n",
       "           8.2621e-02,  9.7819e-02,  1.3603e-01, -3.9973e-01, -7.5787e-02,\n",
       "          -3.6801e-02,  5.9613e-01, -2.1823e-02, -5.9709e-01,  1.5603e-01,\n",
       "          -1.9537e-01,  5.3122e-02,  6.8860e-02, -1.2172e-01,  4.9706e-01,\n",
       "          -7.0828e-01,  2.7900e-01,  3.2827e-01, -5.9399e-02,  7.8449e-01,\n",
       "           1.6671e-01,  1.8468e-01,  7.4671e-01, -3.8967e-01, -6.2083e-01,\n",
       "          -5.0571e-01, -7.6695e-02,  5.1384e-01, -4.2861e-01,  2.3148e-01,\n",
       "          -9.8841e-02,  1.8745e-01, -2.0415e-01,  3.5488e-01,  4.2969e-02,\n",
       "           3.2306e-01,  5.5474e-01, -1.0544e-02,  7.8583e-02, -1.8719e-01,\n",
       "          -3.1561e-01,  2.5043e-02, -4.9659e-02, -6.0277e-02, -3.0451e-01,\n",
       "          -5.3916e-02,  1.4698e-01, -5.0053e-01, -2.0416e-01, -2.3241e-01,\n",
       "           1.4052e-01, -2.4150e-01,  1.0012e-02,  3.3739e-02,  4.0760e-01,\n",
       "           1.1741e-01, -3.1554e-01, -3.7425e-01,  1.8211e-02,  1.0846e-01,\n",
       "           2.3058e-01, -1.6928e-02, -2.4001e-01, -8.9093e-01, -3.0032e-01,\n",
       "           2.8391e-02,  9.4187e-01, -5.3980e-01, -3.8487e-01,  3.6706e-01,\n",
       "           1.3515e-02, -2.5033e-01, -1.8306e-01, -6.2051e-01,  1.0878e-01,\n",
       "          -1.7889e-01, -1.9702e-01,  5.2566e-01,  2.5948e-03,  3.7097e-01,\n",
       "           5.0790e-01,  1.0635e-01, -1.1689e-02, -4.5858e-02, -4.5605e-01,\n",
       "          -6.9939e-01, -1.3259e-02, -3.2999e-01,  1.7426e-01, -3.2910e-01,\n",
       "          -2.4418e-01,  6.4175e-02, -1.5387e-02, -5.6779e-02,  7.2820e-01,\n",
       "           6.1920e-02,  1.2944e+00, -2.9247e-01,  5.6791e-04,  2.0640e-01,\n",
       "           1.7730e-01,  2.9784e-01,  2.7438e-01, -1.7091e-01,  8.2879e-02,\n",
       "           3.8866e-01, -2.1344e-01, -3.9032e-01,  2.1816e-01,  2.2756e-01,\n",
       "           3.9593e-01, -5.9076e-01, -2.7565e-01, -2.6625e-01,  6.4058e-01,\n",
       "           3.5704e-01, -3.3174e-02, -3.0307e-01,  2.8794e-01,  6.3242e-01,\n",
       "          -7.7053e-01,  2.6025e-01,  3.3501e-01,  1.2954e-01, -1.7979e-01,\n",
       "          -1.2486e-01, -3.4262e-02, -2.9787e-01, -3.1994e-01,  6.1839e-01,\n",
       "           7.7700e-01, -8.2809e-01,  5.0716e-01,  4.3278e-01, -8.7074e-02,\n",
       "           4.6662e-01,  7.1283e-02,  3.2832e-01,  3.8450e-01,  3.9792e-01,\n",
       "           3.7343e-01,  1.7883e-01, -2.9576e-01, -5.1922e-01,  2.4912e-01,\n",
       "           5.2804e-01,  1.6244e-01,  3.1933e-02,  8.2444e-02, -6.1699e-02,\n",
       "          -8.0348e-02,  4.0183e-02, -5.1247e-01, -2.8335e-01,  1.0346e-01,\n",
       "          -8.7673e-02, -2.7936e-01,  7.7286e-02,  3.0930e-01, -5.4603e-01,\n",
       "           5.2040e-01,  6.8617e-01,  4.1754e-01,  7.3055e-02, -1.0444e-01,\n",
       "          -2.2831e-01,  3.0040e-01, -2.4528e-01,  7.4886e-01, -1.0915e-01,\n",
       "          -8.1949e-02, -2.7973e-02,  1.1720e-01, -1.3769e-01,  3.4946e-01,\n",
       "          -3.0301e-01,  1.4474e-01, -2.4709e-01, -2.4159e-01,  1.9357e-01,\n",
       "          -3.3037e-02,  1.1210e-01, -6.5210e-01,  3.3298e-02,  2.3861e-01,\n",
       "          -4.1704e-01,  4.2070e-01, -1.8508e-01,  4.2032e-01,  5.0210e-02,\n",
       "           1.6898e-01, -6.4937e-01, -3.5741e-01,  1.6026e-01,  3.3484e-01,\n",
       "          -4.2869e-01, -5.2079e-01,  1.2549e-01,  1.3522e-01,  5.9171e-01,\n",
       "           3.5069e-01, -1.8784e-02, -4.7269e-01, -2.7303e-01,  2.5930e-01,\n",
       "           2.2028e-01, -3.3033e-01,  2.4769e-02, -1.2633e-01, -3.9334e-02,\n",
       "          -8.1783e-01, -1.6359e-01, -1.9666e-02, -4.6624e-04, -3.0209e-01,\n",
       "           8.2335e-01, -2.5908e-01,  9.1975e-02,  2.1194e-01,  1.1198e-01,\n",
       "           5.5523e-01,  7.0247e-01,  1.5578e-01, -6.2660e-01, -3.0862e-01,\n",
       "           7.5352e-03, -1.8532e-01, -2.1019e-01, -5.5806e-01,  4.0257e-01,\n",
       "           4.8804e-01, -7.9040e-02,  7.2806e-01]], device='mps:0'),\n",
       " tensor([[-1.7731e-01, -3.8604e-01, -5.6476e-01, -2.2650e-01, -1.3208e-01,\n",
       "          -3.1625e-01, -5.9292e-01,  4.5825e-01,  2.2151e-01,  5.2940e-01,\n",
       "           3.8502e-01,  1.4343e-01,  6.5615e-02, -2.5563e-01, -4.4222e-01,\n",
       "          -1.6916e-01,  1.6884e-01,  5.0107e-01,  3.6126e-01, -7.6046e-03,\n",
       "           5.9085e-01,  3.7327e-01, -2.0317e-01,  2.5562e-01, -7.5220e-01,\n",
       "           2.6747e-02, -1.6815e-01, -2.5680e-02,  9.6832e-02, -3.0143e-01,\n",
       "          -1.1712e-01,  2.7595e-01, -3.7936e-01,  7.2144e-01, -1.4269e-01,\n",
       "          -2.3945e-01,  1.6259e-01, -1.0450e-01,  5.3649e-01, -1.3024e-01,\n",
       "           5.7769e-01, -2.9111e-01,  4.9321e-01,  3.4162e-02,  2.2711e-01,\n",
       "           1.3206e-01,  2.9819e-01, -4.0790e-01,  1.3721e-01, -5.3671e-01,\n",
       "          -1.6362e-02, -7.3465e-02,  4.1546e-01,  1.0724e-01, -6.3685e-01,\n",
       "          -2.7220e-01, -7.3533e-02,  4.2370e-01, -3.4661e-01,  1.4839e-01,\n",
       "           1.4452e-01,  4.6875e-01,  1.2890e-01,  1.1326e-01, -3.3348e-01,\n",
       "           3.0093e-01, -7.0323e-01, -8.0455e-01, -5.7621e-01, -1.8468e-01,\n",
       "           2.6210e-01,  2.4685e-01,  3.2711e-01, -6.3178e-01,  2.8882e-01,\n",
       "          -5.5500e-01,  4.7778e-01, -3.7641e-02, -3.8107e-01,  2.4976e-02,\n",
       "           4.6662e-01,  1.6673e-01, -2.3893e-01,  2.6411e-01,  2.9965e-01,\n",
       "          -6.2089e-02,  4.4818e-01, -2.9706e-02,  5.6407e-02,  3.8863e-01,\n",
       "           5.0392e-01,  2.6142e-01, -5.4248e-01, -2.1306e-01,  2.3546e-01,\n",
       "          -3.7940e-01,  2.9621e-01, -1.0776e-01,  2.0922e-02, -8.6342e-02,\n",
       "          -3.6881e-02,  3.4334e-01,  3.1658e-01, -6.2528e-02, -1.0724e+00,\n",
       "          -4.3983e-01, -2.4311e-02, -2.7814e-01, -3.4665e-01, -1.0267e-02,\n",
       "           7.0862e-03, -2.1454e-01,  4.5842e-01, -4.4114e-01, -3.9926e-01,\n",
       "           1.3807e-01, -3.8646e-02, -1.5082e-02, -1.4568e-01,  6.0048e-01,\n",
       "           1.6661e-01, -5.3057e-02,  1.5341e-01,  8.8933e-02,  4.9861e-01,\n",
       "          -3.9381e-01,  3.7161e-01, -4.2227e-01,  3.0903e-01,  6.8681e-02,\n",
       "          -2.0183e-02, -7.2510e-02, -2.5207e-01, -2.0895e-01, -2.4617e-02,\n",
       "           2.1165e-02,  2.8627e-01, -4.8458e-01,  5.5727e-02,  5.6803e-02,\n",
       "           2.5004e-01, -1.2498e-01, -4.2667e-01, -1.9318e-01, -6.4563e-02,\n",
       "          -4.0302e-02,  1.9246e-01, -1.1367e-01, -2.7591e-01, -2.6032e-01,\n",
       "          -7.7253e-01, -4.4007e-01,  3.9722e-01,  1.0952e-01,  9.2693e-02,\n",
       "           1.1479e-01,  2.1754e-01, -1.9775e-01, -1.8512e-01, -1.5673e-01,\n",
       "           5.8319e-01, -9.5195e-02,  6.6991e-02, -3.6372e-01, -3.0891e-01,\n",
       "           7.3167e-02,  1.7092e-01, -1.7339e-01,  4.3863e-02,  1.4590e-02,\n",
       "           2.2225e-01, -8.8855e-02, -1.0134e-03, -7.5909e-01, -3.0093e-02,\n",
       "          -2.8356e-01,  3.3814e-01,  2.7840e-01,  1.8768e-01, -2.4537e-01,\n",
       "           5.0899e-02, -1.7294e+00,  7.8671e-01, -5.9166e-01,  2.4877e-01,\n",
       "          -1.8150e-01,  3.1718e-01,  9.1189e-02,  1.8509e-01, -2.9267e-01,\n",
       "          -1.2026e-01, -3.9309e-02,  4.1623e-02,  2.9227e-01, -8.4757e-02,\n",
       "           4.8559e-01,  5.5627e-01,  3.3170e-01,  2.0436e-01, -9.9033e-02,\n",
       "          -6.0977e-01,  3.3235e-01,  1.6388e-01, -2.7455e-01,  2.9857e-02,\n",
       "           2.4361e-01, -3.2791e-01, -5.7703e-02,  3.4800e-01,  3.3556e-02,\n",
       "          -8.5587e-02, -3.0852e-01, -4.0912e-01, -1.8006e-01, -1.6990e-01,\n",
       "           3.0904e-01, -2.3500e-01, -2.0646e-01,  7.4699e-01,  9.5515e-02,\n",
       "          -2.2466e-02, -7.9463e-02, -2.8277e-01, -4.4039e-01, -2.0940e-01,\n",
       "           2.9576e-01, -5.3435e-01, -1.7453e-01, -2.8854e-01, -4.6020e-01,\n",
       "           4.6218e-01, -9.4786e-01,  6.3073e-01,  2.5225e-01,  1.6028e-01,\n",
       "           6.5808e-02, -8.2977e-02, -1.3264e-01, -3.6344e-01, -1.9702e-02,\n",
       "          -2.1377e-02, -5.7642e-01, -9.0486e-01,  1.5472e-01, -7.9254e-02,\n",
       "           1.2900e-01, -4.1901e-02,  6.2316e-01,  4.2972e-01,  5.9118e-01,\n",
       "           2.3140e-01,  5.5355e-03,  4.6999e-01,  4.3441e-01, -4.1017e-01,\n",
       "           4.6498e-01,  1.8871e-01, -4.7838e-02,  6.9646e-02, -3.2186e-01,\n",
       "          -9.1404e-01,  2.0856e-02,  3.2947e-01,  4.9930e-01, -1.7114e-01,\n",
       "           4.2813e-01,  2.6662e-01, -1.3998e-01, -1.7031e-01,  3.1347e-01,\n",
       "          -4.8944e-01, -1.5127e-01,  9.5415e-02, -1.4029e-01, -6.5634e-01,\n",
       "          -4.8841e-01,  2.7357e-02, -2.9199e-01,  2.0484e-01, -9.9702e-01,\n",
       "          -2.6273e-01,  5.6883e-01, -2.8623e-02,  9.3732e-02, -2.3345e-01,\n",
       "           7.0240e-02, -1.0750e-01,  1.0928e-01, -1.5161e-01, -8.9822e-01,\n",
       "          -1.4258e-01,  9.6331e-02, -2.2318e-01,  3.2530e-01,  3.7809e-01,\n",
       "          -2.0083e-01, -2.9617e-01,  2.8364e-01, -9.0550e-02, -2.5039e-01,\n",
       "          -1.3545e-01, -3.7063e-01, -2.4607e-02, -1.3576e-01, -1.1638e-01,\n",
       "           6.9800e-02,  4.9085e-01,  4.8805e-01, -7.7597e-01, -5.5824e-01,\n",
       "          -2.2992e-02,  5.0756e-01, -1.6302e-01,  3.7266e-02, -2.2967e-01,\n",
       "          -1.0825e+00, -2.0374e-01, -1.1835e-01, -3.9640e-01, -3.8528e-01,\n",
       "           5.0161e-01, -1.9830e-01, -2.0671e-01,  9.6594e-02, -3.4539e-01,\n",
       "           3.4160e-01,  8.4200e-01, -4.9032e-02, -4.7416e-02, -2.9725e-01,\n",
       "           1.7477e-02, -1.8692e-01,  4.8213e-02, -1.9462e-01, -2.5485e-02,\n",
       "           3.1570e-01,  1.2741e-01, -1.8351e-01, -8.2635e-02, -1.7372e-01,\n",
       "           7.9347e-01, -5.6947e-01,  1.3088e-01,  3.8738e-01, -2.4876e-01,\n",
       "          -1.3259e-01,  4.8066e-01,  1.4439e-01,  1.9491e-01, -1.0024e-01,\n",
       "          -2.6493e-02, -2.8608e-01, -1.8423e-01, -3.1248e-01,  9.6616e-01,\n",
       "           3.5980e-01,  2.2368e-01, -3.0496e-01,  5.7904e-01,  5.6810e-01,\n",
       "          -3.8525e-01,  7.6086e-02,  8.8516e-02,  3.4151e-03, -3.1432e-01,\n",
       "           3.2810e-01,  2.4729e-01, -4.2766e-01, -2.5123e-02,  4.5374e-03,\n",
       "           1.4485e-01, -5.5797e-01,  1.6191e-01,  1.0782e-01,  8.5195e-02,\n",
       "          -1.4724e-03, -2.6111e-01,  5.2888e-01, -2.4531e-01, -2.8912e-01,\n",
       "          -1.0092e-01, -1.2182e-01,  4.1012e-02, -2.4548e-01, -3.6887e-01,\n",
       "          -2.7428e-01,  2.4193e-02,  3.5074e-01,  1.4285e-02,  7.4821e-02,\n",
       "           2.5544e-01, -2.6796e-01, -3.4638e-01,  4.5500e-01, -1.7660e-02,\n",
       "          -1.6810e-01,  4.0768e-01, -2.4890e-01, -1.1639e-01,  1.3881e-01,\n",
       "           6.2926e-02,  1.0677e-01, -5.8094e-02,  3.7998e-01,  1.7175e-01,\n",
       "          -2.8992e-01,  3.6344e-02,  2.1408e-01, -8.0245e-02,  3.1665e-01,\n",
       "           5.1861e-02,  8.2278e-01, -3.7109e-01, -2.7164e-02, -3.8441e-01,\n",
       "          -1.2821e-01, -1.6499e-01,  7.5490e-03,  1.5024e-01, -7.5773e-01,\n",
       "           3.0131e-01,  2.8842e-01,  3.4034e-01,  1.1331e-01, -1.1458e+01,\n",
       "           1.6305e-01,  6.4218e-01, -5.1374e-01,  1.8438e-01,  1.7594e-01,\n",
       "          -1.4040e-01, -3.9794e-01,  4.0875e-02, -3.4785e-01, -2.6394e-01,\n",
       "          -6.5590e-02,  1.9075e-01, -1.2756e-02, -5.5478e-01,  3.3680e-02,\n",
       "          -2.5398e-01, -6.3192e-01,  2.6160e-01, -9.0323e-02,  4.1335e-01,\n",
       "          -3.0876e-01,  1.8731e-01,  2.3832e-01, -3.6034e-01, -2.7614e-02,\n",
       "          -5.1547e-01, -5.5807e-02,  3.7042e-01,  3.2770e-01, -3.5254e-01,\n",
       "           2.2243e-01, -1.8729e-01,  3.8908e-01, -2.6449e-01, -5.2363e-01,\n",
       "           1.3092e-01, -9.6993e-02,  3.7952e-01,  1.1884e-02, -2.4878e-01,\n",
       "          -2.5432e-02, -5.2584e-01,  5.9952e-02,  2.0270e-01,  3.4279e-01,\n",
       "           1.2541e+00,  9.4924e-01,  1.3760e-01, -1.3198e-01,  1.6614e-01,\n",
       "          -7.3513e-01,  2.0677e-01, -6.2811e-01, -8.1585e-02, -6.4376e-02,\n",
       "           8.3128e-01, -4.3261e-01, -3.3743e-01,  7.3012e-02, -8.0493e-02,\n",
       "           3.1300e-01, -7.9471e-02,  2.6202e-01, -3.1186e-01,  2.9341e-03,\n",
       "           6.1375e-01, -1.8722e-02,  3.8063e-01, -4.3721e-01, -6.2672e-02,\n",
       "          -4.2457e-02,  2.9525e-01,  3.7279e-01,  6.2687e-02, -2.5474e-01,\n",
       "          -4.1450e-01,  4.2406e-02,  5.2243e-01,  1.3990e-01,  1.4541e-01,\n",
       "           5.8211e-01, -2.3165e-01, -4.1432e-01, -6.0104e-01, -3.9034e-01,\n",
       "          -7.6601e-03,  1.3684e-01, -3.9867e-01,  6.4750e-01, -4.2253e-01,\n",
       "          -2.7567e-01, -1.6427e-01, -8.8628e-02,  3.8882e-01,  5.0639e-01,\n",
       "          -3.3627e-01,  8.0837e-02, -1.8585e-01, -4.4707e-03, -6.7852e-01,\n",
       "          -3.5458e-01, -8.4538e-02, -2.9987e-01,  2.3231e-01,  1.7305e-01,\n",
       "          -1.0020e-01,  9.0697e-03,  1.2051e-01,  7.3101e-02, -5.4356e-01,\n",
       "           3.0479e-01, -1.0749e-01,  6.8380e-02, -3.9586e-01, -5.1756e-01,\n",
       "           4.0833e-01,  7.0232e-01, -4.2415e-01, -3.3873e-01,  9.8320e-02,\n",
       "           1.3073e-01,  1.8252e-01,  1.2615e-01, -6.4414e-01,  1.1900e-01,\n",
       "          -4.1931e-01, -1.3812e-01,  1.3512e-01,  7.2335e-01,  6.6950e-01,\n",
       "          -5.3703e-02, -3.2049e-01,  2.4632e-01, -5.0640e-02, -2.5765e-01,\n",
       "          -7.0535e-01, -2.9464e-02,  2.1372e-01, -5.3812e-01,  9.7345e-02,\n",
       "          -3.3445e-01,  1.9755e-01, -1.7248e-01,  1.0825e-01, -7.4644e-02,\n",
       "           4.1662e-01,  4.1987e-01,  7.6810e-02,  4.3515e-01, -1.7908e-01,\n",
       "          -2.1990e-01,  1.9709e-01, -1.7846e-01, -1.1703e-01, -1.7569e-01,\n",
       "          -1.7366e-01,  9.7459e-02, -4.0722e-01, -3.6435e-02,  1.7903e-01,\n",
       "          -1.2723e-01,  6.7948e-02, -3.4229e-01,  2.2739e-01,  3.3538e-01,\n",
       "          -1.0467e-01, -3.7966e-01, -2.3884e-01,  4.7521e-02,  7.1481e-02,\n",
       "           3.8731e-01, -2.6195e-01, -1.0102e-01, -7.3811e-01, -2.0636e-01,\n",
       "           3.6787e-02,  1.0410e+00, -3.7965e-02, -3.0778e-01,  3.5156e-01,\n",
       "           3.9051e-01, -2.4726e-01, -3.4583e-01, -1.8699e-01,  7.4840e-01,\n",
       "          -2.3598e-01, -1.5465e-01,  2.2099e-01,  2.5510e-02,  6.2240e-01,\n",
       "           1.3154e-01,  2.2300e-01, -1.8512e-01,  4.7019e-01,  2.1848e-01,\n",
       "          -4.4102e-01, -1.5073e-01, -3.9303e-01, -3.2472e-01,  4.3754e-01,\n",
       "          -6.9776e-02,  2.7454e-01, -1.1790e-01,  8.1553e-02,  1.4253e-02,\n",
       "           5.4206e-02,  1.1645e+00, -2.0185e-01, -2.3960e-01,  4.7684e-01,\n",
       "           2.7079e-01,  3.4262e-01,  3.7917e-02, -1.3477e-01,  9.5921e-02,\n",
       "           4.9705e-01, -1.6706e-01, -4.9624e-01,  1.5664e-01,  3.6272e-01,\n",
       "           5.9108e-01, -8.7613e-01, -3.6622e-01,  2.0509e-01,  2.2296e-01,\n",
       "          -3.2121e-01,  1.9818e-01, -7.6228e-02,  3.2510e-01,  3.6266e-01,\n",
       "          -1.2771e+00,  1.7642e-01,  7.1262e-02,  1.5943e-01, -1.5269e-01,\n",
       "          -6.7715e-01, -2.5385e-01,  1.0595e-01, -6.2916e-02,  4.2076e-01,\n",
       "           4.4789e-01, -7.1509e-01,  1.2452e-01,  3.1511e-01, -3.7691e-01,\n",
       "           1.2320e-01, -2.7571e-01,  1.6643e-01,  5.2170e-01,  1.6129e-01,\n",
       "          -4.2505e-01,  2.5049e-01, -5.0826e-01, -6.1173e-01,  1.4561e-01,\n",
       "           6.9575e-01,  2.1324e-02, -1.1322e-01, -1.1804e-01, -2.9289e-01,\n",
       "          -1.9107e-01, -3.1790e-01, -3.4967e-01, -2.7542e-02,  1.4802e-01,\n",
       "           2.4431e-02, -3.6606e-02,  2.4884e-01,  1.1143e-01, -2.2300e-01,\n",
       "           5.8287e-01,  4.8747e-01,  7.4334e-01, -1.5062e-02,  1.1963e-02,\n",
       "          -2.2207e-01,  2.9970e-01, -4.7196e-02,  4.7552e-01, -3.4433e-01,\n",
       "          -2.0911e-01,  2.2006e-01,  3.6212e-02, -6.2294e-01, -1.0381e-01,\n",
       "          -3.5492e-01,  3.0044e-01,  1.1075e-02, -2.7609e-01,  4.8880e-01,\n",
       "           2.0553e-02,  1.8280e-01, -5.5884e-01,  4.6756e-01,  1.6596e-01,\n",
       "           2.2073e-02,  9.8054e-01,  1.1685e-01,  5.3145e-01,  5.8221e-01,\n",
       "           1.1828e-01, -8.1138e-01, -1.8763e-01, -7.2256e-02,  1.5035e-01,\n",
       "           1.1544e-02,  1.5872e-03,  1.5741e-01,  2.8459e-02,  9.3799e-01,\n",
       "           4.5776e-01, -2.2136e-01, -3.9686e-01, -3.7782e-01,  1.8491e-01,\n",
       "          -8.4548e-02, -3.3508e-01,  2.0442e-01, -2.2187e-01,  1.7388e-02,\n",
       "          -5.8935e-01, -5.5885e-01, -1.2302e-01,  2.7758e-01, -2.6880e-01,\n",
       "           5.4517e-01,  4.6664e-02,  5.2447e-03,  3.4141e-01, -2.9763e-02,\n",
       "           2.1521e-01,  6.7382e-01,  2.2993e-01, -2.9428e-01,  1.8031e-01,\n",
       "           2.1968e-01, -1.3524e-01,  2.0995e-01, -8.6762e-01,  3.6296e-01,\n",
       "          -5.1872e-02,  5.4038e-01,  4.7469e-01]], device='mps:0'),\n",
       " tensor([[-5.7615e-01, -6.5704e-01, -6.4122e-01, -8.3786e-01, -1.6839e-01,\n",
       "          -1.0109e-01, -5.9211e-01,  5.7557e-01, -1.0128e-01,  6.6005e-01,\n",
       "           1.7243e-01, -2.0208e-01,  1.3153e-01, -5.5265e-01, -1.0830e+00,\n",
       "          -7.8838e-02,  5.1128e-01,  5.6026e-01,  5.3742e-01,  7.5567e-01,\n",
       "           5.5133e-01,  3.0929e-01, -1.5431e-01, -2.2773e-01, -5.4095e-01,\n",
       "           4.9474e-01, -4.3509e-01,  1.4463e-01,  3.9349e-01, -5.9621e-01,\n",
       "          -2.4738e-01,  2.1042e-01, -2.4473e-01,  7.9294e-01, -3.1638e-01,\n",
       "          -3.5587e-01,  1.7182e-02, -1.1024e+00,  7.3679e-01, -5.0652e-01,\n",
       "           5.6936e-01,  2.5745e-01, -2.5789e-01,  2.7734e-01,  8.4882e-02,\n",
       "           2.9247e-02, -6.8426e-02, -5.6406e-01, -1.6648e-01, -1.1473e-01,\n",
       "           1.3418e-01,  2.4161e-01,  4.6961e-01,  3.9061e-01, -5.1676e-01,\n",
       "          -2.9447e-01, -1.6494e-01,  1.1824e-01, -2.3024e-01,  4.6201e-01,\n",
       "          -2.1743e-01,  3.9686e-01,  6.2028e-01, -1.8022e-02, -8.7176e-02,\n",
       "           1.5209e-01, -2.0011e-01, -6.1498e-01, -3.5713e-01,  1.4377e-01,\n",
       "          -9.8013e-02,  4.2755e-01,  2.4404e-01, -2.5933e-01,  2.7614e-01,\n",
       "           9.9895e-02,  3.0235e-01, -3.6816e-01, -6.1006e-01, -3.5969e-01,\n",
       "           1.1271e-01,  1.3292e-01, -1.2369e-01, -2.4772e-01,  6.6684e-03,\n",
       "           2.1860e-01,  8.0076e-01, -1.5277e-01,  2.2836e-01, -2.4269e-01,\n",
       "           6.2432e-01, -3.5953e-01, -6.1881e-01, -3.6978e-01,  3.3505e-01,\n",
       "          -3.0024e-01, -1.1070e-02, -4.3831e-01, -1.7785e-01, -3.7064e-01,\n",
       "          -2.0989e-01,  2.4600e-01,  2.3810e-02,  2.7494e-01, -6.9127e-01,\n",
       "          -4.6908e-01,  2.9318e-01, -7.5832e-02, -4.3782e-01, -4.1522e-02,\n",
       "          -1.9288e-01, -1.7287e-01,  2.1465e-01, -1.1083e-01, -4.0942e-01,\n",
       "          -3.4837e-01,  1.4955e-01, -4.7899e-02, -8.1845e-02,  1.1204e+00,\n",
       "           5.5146e-01, -5.4591e-01, -8.3567e-02,  7.2848e-02,  1.8651e-01,\n",
       "          -2.4783e-01,  2.3738e-01, -1.6851e-02,  5.1040e-01, -5.0894e-02,\n",
       "           9.7176e-02,  1.7125e-01, -8.1079e-01, -2.3601e-01,  2.1013e-01,\n",
       "           1.2443e-01, -4.8936e-02, -9.5445e-01,  2.4091e-01,  3.3325e-01,\n",
       "          -2.0818e-02, -1.9779e-01, -3.2518e-01,  2.9856e-01, -1.6607e-01,\n",
       "          -2.2383e-01,  1.2827e-01, -3.6553e-01,  1.3999e-01, -5.8861e-01,\n",
       "          -6.9766e-01, -6.9892e-01,  6.6896e-02, -2.6701e-01,  2.7005e-02,\n",
       "          -3.3359e-01,  4.4333e-01, -6.8610e-02,  8.4490e-03,  1.1807e-01,\n",
       "           7.9476e-01,  2.5284e-01,  1.6197e-01, -2.5447e-01,  3.8864e-02,\n",
       "           2.0786e-01, -2.0172e-01, -2.3267e-01, -3.9078e-01,  3.9192e-01,\n",
       "           2.5434e-01, -3.3339e-02,  2.2309e-01, -1.6275e-01,  5.4503e-02,\n",
       "          -4.4797e-01,  7.0886e-01, -5.9299e-02,  5.5927e-02,  1.3653e-01,\n",
       "          -2.0212e-01, -2.0600e+00, -4.9376e-02, -1.5985e-01, -2.3330e-01,\n",
       "          -8.4681e-02,  5.7423e-01,  4.3994e-01,  2.7605e-01,  1.7719e-01,\n",
       "           1.2867e-01, -3.1442e-02,  1.0075e-01,  1.5316e-01,  1.6991e-02,\n",
       "           6.9486e-01,  5.4184e-01,  1.5919e-01, -1.1201e-01,  9.0315e-03,\n",
       "          -5.8823e-01, -3.4280e-02,  8.6766e-02, -8.1745e-02,  4.1977e-01,\n",
       "           1.9525e-01, -5.5465e-01, -5.6126e-02, -2.5246e-02,  1.5008e-01,\n",
       "          -1.0792e-01, -2.5055e-01, -2.5001e-01, -2.2571e-01, -2.5128e-01,\n",
       "           3.4031e-01, -4.6408e-01, -3.9434e-01,  3.7707e-01,  1.2871e-01,\n",
       "          -1.2576e-01,  6.5467e-02, -4.2277e-02, -1.9763e-01, -2.4494e-01,\n",
       "           4.0392e-01, -5.5781e-02, -1.6538e-01, -2.7890e-01, -1.1595e-01,\n",
       "          -4.5414e-02, -3.3709e-01,  2.8175e-01,  4.3210e-01, -1.1086e-01,\n",
       "          -2.3182e-01, -2.3402e-01,  4.4457e-01, -6.2331e-02,  2.0975e-01,\n",
       "          -1.7798e-01, -4.0098e-01, -4.4920e-01,  7.1693e-02, -3.0788e-01,\n",
       "           1.1334e-01, -1.6828e-01,  7.0183e-01,  5.4973e-01,  1.5685e-01,\n",
       "           3.0322e-02, -8.3891e-02,  1.4749e-01,  7.0612e-02, -1.6793e-01,\n",
       "           2.9325e-02, -6.3733e-03, -6.5666e-01,  1.3122e-01, -6.6635e-01,\n",
       "           2.7284e-01,  1.6485e-01, -2.3822e-01,  5.1826e-01,  1.3516e-01,\n",
       "          -3.6073e-01,  4.3361e-01,  1.8393e-01, -1.9529e-01,  1.7906e-01,\n",
       "          -3.3156e-01,  3.8000e-01, -6.0466e-01, -3.2592e-01, -3.4168e-01,\n",
       "          -6.6778e-01,  3.5143e-02,  8.5031e-02, -9.7987e-02, -3.0763e-01,\n",
       "          -7.5976e-02, -1.0046e-01, -1.1834e-01, -1.4386e-01,  2.1545e-01,\n",
       "           3.3339e-01,  8.4060e-02,  1.8698e-01,  2.3279e-02, -5.2587e-01,\n",
       "          -7.4386e-01, -6.5750e-02, -2.8033e-01, -1.2280e-01,  2.0882e-01,\n",
       "           3.1935e-01,  2.2275e-01,  1.9615e-01, -5.3309e-02, -3.7498e-01,\n",
       "          -1.1096e-01,  1.5238e-01, -2.5130e-01,  3.2339e-02,  2.0108e-01,\n",
       "           2.5515e-01,  6.0375e-02,  3.3354e-01, -5.5564e-01, -6.3153e-01,\n",
       "           1.5280e-01,  5.6159e-01, -4.2729e-02, -1.7693e-01, -2.8892e-01,\n",
       "          -1.2814e+00, -1.2565e-01,  3.7478e-02, -4.1422e-02, -5.8016e-01,\n",
       "           3.1827e-01,  2.6937e-01, -2.1833e-01, -1.1941e-01, -3.2375e-01,\n",
       "           6.2373e-01,  8.5622e-01,  3.2905e-01, -2.8542e-01, -4.0055e-01,\n",
       "          -1.0734e-01, -4.9206e-01,  1.7480e-01, -4.3786e-01, -1.4793e-01,\n",
       "           1.7485e-01,  1.5977e-01, -2.1867e-01,  3.5596e-01, -2.7099e-01,\n",
       "           6.9053e-01, -3.8841e-01,  1.3362e-01,  3.6476e-01, -1.7036e-01,\n",
       "          -1.9996e-01,  3.2991e-01, -5.8828e-01,  1.9293e-01,  3.7797e-01,\n",
       "           2.4451e-01, -1.1057e-01, -2.8946e-01, -1.9247e-02,  5.6577e-01,\n",
       "           2.5165e-01,  3.8140e-01, -2.7939e-01,  3.7361e-01,  4.2160e-01,\n",
       "          -6.5135e-02,  2.6019e-01, -3.4836e-01, -4.8447e-01, -2.9425e-01,\n",
       "           1.0680e-02,  1.2168e-01, -5.2652e-01, -7.7081e-03, -1.4545e-01,\n",
       "           2.6012e-01, -4.4731e-01,  2.3379e-02,  7.2634e-03, -6.4099e-02,\n",
       "          -4.0063e-01, -2.0265e-01,  3.4762e-01, -1.6843e-01, -7.4751e-02,\n",
       "          -2.4735e-01, -2.1844e-01,  1.0678e-01, -2.0294e-01, -7.4820e-02,\n",
       "          -2.5837e-01, -1.9888e-01,  7.4448e-01,  3.5489e-01, -6.2137e-02,\n",
       "           8.4115e-02, -2.6451e-01, -3.4805e-01,  2.0965e-01, -8.0504e-01,\n",
       "          -1.2338e-01,  3.0838e-01,  5.1087e-02, -3.0091e-01,  3.7146e-01,\n",
       "          -3.9354e-01, -1.3671e-01,  1.1637e-01, -1.7296e-01,  7.2077e-02,\n",
       "          -3.1670e-01,  6.2362e-02,  2.9431e-01, -3.1489e-01,  3.8135e-02,\n",
       "          -8.9811e-02, -4.5331e-03,  1.3430e-02, -5.8521e-02, -5.1188e-01,\n",
       "           1.1807e-01, -3.1555e-01, -7.1900e-02, -5.4522e-02, -9.0899e-01,\n",
       "           4.8143e-01,  3.0749e-02,  5.8595e-01,  3.9523e-01, -1.1169e+01,\n",
       "          -7.9349e-02,  5.0006e-01, -4.4257e-01,  9.6267e-02, -2.2733e-01,\n",
       "           2.1733e-01, -2.8984e-01,  8.3140e-02, -7.6987e-02, -8.6523e-02,\n",
       "          -5.7455e-01,  7.9717e-02, -1.8007e-01, -2.7352e-01, -6.4293e-02,\n",
       "          -3.4820e-01, -4.7338e-01,  4.8866e-02, -1.1345e-01,  7.4849e-02,\n",
       "          -4.0621e-01, -1.9804e-01,  1.2232e-01, -2.4655e-01, -3.8660e-01,\n",
       "           1.5867e-01, -8.4574e-02,  1.8752e-01,  9.8409e-02, -3.5383e-02,\n",
       "           6.4221e-02,  8.0825e-02,  4.6823e-01,  7.1268e-02, -4.5839e-01,\n",
       "          -1.6589e-01, -1.5026e-01, -2.0886e-02, -6.7135e-01, -7.4151e-03,\n",
       "          -5.2667e-01, -2.2162e-01,  3.7003e-01,  5.3474e-01,  8.2316e-01,\n",
       "           1.3248e+00,  4.6899e-01,  5.6135e-01, -1.2529e-01, -2.1447e-01,\n",
       "          -2.3421e-01,  4.4721e-01, -2.1927e-02, -2.5524e-01, -5.0050e-01,\n",
       "           3.0013e-01, -5.6149e-01,  3.7116e-01, -5.3483e-01, -2.5341e-01,\n",
       "           4.0709e-01,  2.5015e-01, -2.2630e-01, -3.0283e-01, -4.1849e-01,\n",
       "           3.9178e-01,  9.5595e-02, -2.0908e-03, -5.4278e-02,  1.0352e-01,\n",
       "          -3.3390e-01,  8.4297e-01, -3.8100e-01, -2.9446e-01, -2.2888e-01,\n",
       "          -6.1770e-03,  1.7670e-02,  1.5041e+00, -1.8615e-01,  2.5345e-01,\n",
       "           2.5991e-01,  2.2106e-01, -6.7766e-01, -8.0494e-01, -3.2484e-02,\n",
       "          -2.9523e-01, -4.4688e-03,  6.0886e-01,  3.5930e-01, -1.0144e-01,\n",
       "          -5.9594e-01, -5.4455e-01, -1.0963e-01,  5.0477e-01,  1.0680e+00,\n",
       "          -5.4433e-01, -6.5442e-02,  3.6367e-02,  2.0108e-01, -4.8167e-01,\n",
       "           1.0361e-01, -1.7373e-03, -9.9270e-02,  3.3640e-01, -1.2214e-01,\n",
       "          -8.0920e-02,  3.9944e-01,  2.3593e-02,  4.7069e-01, -2.8461e-01,\n",
       "           1.5317e-01, -3.2582e-01,  7.9943e-02, -3.2945e-01,  1.7954e-01,\n",
       "           2.5658e-01,  3.7044e-01, -5.5414e-02, -1.0072e-01, -7.1741e-02,\n",
       "           4.0200e-02,  4.1066e-01, -2.5414e-01, -2.4791e-02,  7.1504e-01,\n",
       "          -4.1960e-01,  1.1327e-03,  3.3982e-01, -1.2480e-01,  2.7091e-01,\n",
       "          -2.0780e-02, -3.3579e-01,  2.5290e-01, -4.1091e-01, -7.6521e-01,\n",
       "          -6.2479e-01, -2.4522e-01,  3.1708e-01, -4.9850e-01,  8.5162e-02,\n",
       "          -1.5864e-01,  2.8505e-01, -1.7160e-01,  2.1356e-01,  2.8801e-01,\n",
       "           4.1145e-02,  4.4170e-01, -2.8230e-01,  1.7537e-01, -2.9118e-01,\n",
       "          -1.9488e-01, -4.9489e-02, -2.3738e-01,  8.8838e-02, -3.1343e-01,\n",
       "          -5.9303e-01, -1.4512e-01, -2.0671e-01,  5.6471e-01, -3.9435e-01,\n",
       "           1.2361e-01, -3.9683e-02,  2.0048e-02,  9.7469e-02,  6.7954e-01,\n",
       "           2.4541e-01, -3.0319e-01, -3.7507e-01,  1.5735e-01,  2.0149e-01,\n",
       "          -5.3145e-02,  1.9386e-01,  5.3104e-02, -1.2348e-01, -5.8350e-01,\n",
       "          -1.8470e-01,  1.0511e+00, -1.1725e-01, -1.5351e-01,  4.7313e-02,\n",
       "           8.9770e-01, -3.6354e-01,  1.4036e-01, -3.1986e-01,  3.9074e-01,\n",
       "          -4.2205e-01, -3.3915e-01,  6.3570e-01,  1.2328e-01,  2.3194e-01,\n",
       "           2.5994e-01,  1.1923e-01,  1.4927e-01,  2.9090e-02, -2.3061e-01,\n",
       "          -3.7583e-01,  1.3948e-01, -2.8231e-03,  1.1812e-01,  5.2428e-01,\n",
       "           1.3840e-01,  3.0043e-02,  1.1924e-01, -2.4907e-02,  1.1144e-01,\n",
       "           1.4940e-01,  1.3576e+00, -5.5958e-01, -2.9751e-01,  1.0644e+00,\n",
       "           3.0314e-01,  3.5966e-01,  5.1509e-01,  8.9098e-02,  8.8212e-02,\n",
       "           5.6689e-01, -4.7902e-01, -1.9253e-01,  2.4092e-01,  3.1908e-01,\n",
       "           7.0138e-01, -1.0120e+00, -4.1509e-01,  1.1535e-01,  4.0852e-01,\n",
       "           2.1248e-01,  4.1912e-01, -2.7914e-01,  4.8775e-01,  1.0055e+00,\n",
       "          -7.5752e-01, -5.2715e-02,  1.0214e-01,  6.2567e-01, -1.5406e-01,\n",
       "           1.7587e-01, -3.8165e-02, -2.0438e-01, -3.7839e-01,  4.5265e-01,\n",
       "           5.1356e-01, -1.1727e+00,  1.9918e-01,  5.0429e-01, -3.1171e-01,\n",
       "           4.1139e-01,  1.6181e-01,  2.3655e-01, -3.0796e-02,  4.0675e-01,\n",
       "           2.7488e-01, -2.8259e-03, -6.6368e-01, -2.1342e-01,  1.3310e-01,\n",
       "           7.1485e-01,  2.8893e-01, -1.4506e-01,  2.0232e-01, -1.7055e-01,\n",
       "          -1.9096e-01, -2.8201e-01, -3.6822e-01, -2.0222e-01, -1.4871e-02,\n",
       "           1.4693e-02, -6.4793e-01, -4.7989e-02,  2.0179e-01, -3.9509e-01,\n",
       "           2.9981e-01,  4.7045e-01,  5.8323e-01,  9.6656e-02,  2.6519e-02,\n",
       "           3.7891e-01,  3.8048e-01, -2.7838e-01,  6.2769e-01,  3.8957e-01,\n",
       "          -3.7634e-01,  8.4374e-02, -5.8000e-02, -2.4602e-01,  1.7443e-01,\n",
       "          -3.6038e-01,  1.5176e-01, -7.6855e-02, -5.6543e-01,  1.4390e-01,\n",
       "          -1.8936e-02,  2.5710e-01, -3.1932e-01,  4.1674e-01,  3.5727e-01,\n",
       "          -2.5598e-01,  5.5207e-01, -3.3991e-02,  7.0746e-01, -3.8692e-01,\n",
       "          -1.9701e-02, -4.7276e-01, -1.6539e-01, -1.4793e-01,  1.7404e-01,\n",
       "          -3.8372e-01, -3.4416e-01,  2.5799e-02, -1.4961e-01,  4.7780e-01,\n",
       "           7.9199e-01, -4.1890e-02, -6.2905e-01, -4.6042e-01, -4.7282e-02,\n",
       "           1.1115e-01,  1.5960e-01,  4.5785e-01, -4.9613e-01,  3.0030e-01,\n",
       "          -5.2803e-01,  5.2096e-02,  1.7662e-02,  2.7639e-01, -1.5694e-01,\n",
       "           8.4870e-01, -6.3609e-02,  2.2024e-01,  3.0922e-01,  4.1710e-01,\n",
       "           3.2422e-01,  8.3971e-01,  1.8044e-01, -9.6706e-02, -1.3637e-01,\n",
       "           1.9280e-01, -1.3515e-01, -6.1174e-02, -6.6594e-01,  4.4822e-01,\n",
       "           4.0340e-01,  2.5364e-01,  6.7030e-01]], device='mps:0')]"
      ]
     },
     "execution_count": 17,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "target_vecs_list"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": []
  },
  {
   "cell_type": "code",
   "execution_count": 86,
   "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>JRCT ID</th>\n",
       "      <th>NCT No</th>\n",
       "      <th>JapicCTI No</th>\n",
       "      <th>Title</th>\n",
       "      <th>TargetJ</th>\n",
       "      <th>Target</th>\n",
       "      <th>TargetEnglish</th>\n",
       "      <th>研究・治験の目的</th>\n",
       "      <th>試験等のフェーズ</th>\n",
       "      <th>試験の種類</th>\n",
       "      <th>...</th>\n",
       "      <th>purpose</th>\n",
       "      <th>Inclusion Criteria</th>\n",
       "      <th>Exclusion Criteria</th>\n",
       "      <th>Age Minimum</th>\n",
       "      <th>Age Maximum</th>\n",
       "      <th>Gender</th>\n",
       "      <th>Discontinuation Criteria</th>\n",
       "      <th>Keyword</th>\n",
       "      <th>Intervention(s)</th>\n",
       "      <th>TargetWord</th>\n",
       "    </tr>\n",
       "  </thead>\n",
       "  <tbody>\n",
       "    <tr>\n",
       "      <th>31</th>\n",
       "      <td>jRCT2061240081</td>\n",
       "      <td>NCT06393374</td>\n",
       "      <td>NaN</td>\n",
       "      <td>術前薬物療法後の手術時に病理学的完全奏効を達成していないトリプルネガティブ乳癌患者を対象に、...</td>\n",
       "      <td>トリプルネガティブ乳癌</td>\n",
       "      <td>トリプルネガティブ乳癌</td>\n",
       "      <td>Triple negative breast cancer</td>\n",
       "      <td>術前薬物療法後の手術時に病理学的完全奏効を達成していないトリプルネガティブ乳癌患者を対象に、...</td>\n",
       "      <td>3</td>\n",
       "      <td>NaN</td>\n",
       "      <td>...</td>\n",
       "      <td>treatment purpose</td>\n",
       "      <td>-Has centrally confirmed TNBC, as defined by t...</td>\n",
       "      <td>-Has a known germline breast cancer gene (BRCA...</td>\n",
       "      <td>18age old over</td>\n",
       "      <td>No limit</td>\n",
       "      <td>Both</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>-Arm 1: MK-2870 4mg/kg intravenous (IV) every ...</td>\n",
       "      <td>[Triple Negative, Breast Cancer]</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>157</th>\n",
       "      <td>jRCT2052240059</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>遺伝子HSD17B4高メチル化を有するHER2陽性ER陰性乳癌における非手術療法の有用性を評...</td>\n",
       "      <td>乳がん</td>\n",
       "      <td>乳がん</td>\n",
       "      <td>Breast cancer</td>\n",
       "      <td>HSD17B4高メチル化(HSD17B4 hypermethylation:HH)を有するH...</td>\n",
       "      <td>2</td>\n",
       "      <td>NaN</td>\n",
       "      <td>...</td>\n",
       "      <td>diagnostic purpose</td>\n",
       "      <td>1. Histologically confirmed invasive breast ca...</td>\n",
       "      <td>1. History of other malignancy within the last...</td>\n",
       "      <td>20age old over</td>\n",
       "      <td>No limit</td>\n",
       "      <td>Female</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>Omitting breast surgery after preoperative che...</td>\n",
       "      <td>[Breast Cancer]</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>215</th>\n",
       "      <td>jRCT2031230723</td>\n",
       "      <td>NCT06112379</td>\n",
       "      <td>NaN</td>\n",
       "      <td>未治療のトリプルネガティブ又はホルモン受容体低発現／HER2陰性乳癌の成人患者を対象として、...</td>\n",
       "      <td>乳癌</td>\n",
       "      <td>乳癌</td>\n",
       "      <td>Breast Cancer</td>\n",
       "      <td>未治療のトリプルネガティブ又はホルモン受容体低発現/HER2陰性乳癌の成人患者を対象として、...</td>\n",
       "      <td>3</td>\n",
       "      <td>NaN</td>\n",
       "      <td>...</td>\n",
       "      <td>treatment purpose</td>\n",
       "      <td>1. Participant must be &gt;= 18 years at the time...</td>\n",
       "      <td>1. As judged by the investigator, any evidence...</td>\n",
       "      <td>18age old over</td>\n",
       "      <td>No limit</td>\n",
       "      <td>Both</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>- Experimental arm: Dato-DXd plus durvalumab n...</td>\n",
       "      <td>[Breast Cancer]</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>240</th>\n",
       "      <td>jRCT2061230102</td>\n",
       "      <td>NCT06103864</td>\n",
       "      <td>NaN</td>\n",
       "      <td>Programmed death-ligand（PD-L1）陽性の局所再発手術不能又は転移性...</td>\n",
       "      <td>乳癌</td>\n",
       "      <td>乳癌</td>\n",
       "      <td>Breast Cancer</td>\n",
       "      <td>PD-L1陽性の局所再発手術不能または転移性TNBC患者を対象に、デュルバルマブ併用または非...</td>\n",
       "      <td>3</td>\n",
       "      <td>NaN</td>\n",
       "      <td>...</td>\n",
       "      <td>treatment purpose</td>\n",
       "      <td>Histologically or cytologically documented loc...</td>\n",
       "      <td>As judged by investigator, severe or uncontrol...</td>\n",
       "      <td>18age old over</td>\n",
       "      <td>No limit</td>\n",
       "      <td>Both</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>Arm 1: Dato-DXd + durvalumab Arm 2: Investigat...</td>\n",
       "      <td>[Breast Cancer]</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>307</th>\n",
       "      <td>jRCT2061230074</td>\n",
       "      <td>NCT05952557</td>\n",
       "      <td>NaN</td>\n",
       "      <td>根治的局所治療（化学療法の併用または非併用）を受けて疾患の兆候のない、再発リスクが中間～高リ...</td>\n",
       "      <td>乳がん、早期乳がん</td>\n",
       "      <td>乳がん、早期乳がん</td>\n",
       "      <td>Breast Cancer, Early Breast Cancer</td>\n",
       "      <td>Treatment</td>\n",
       "      <td>3</td>\n",
       "      <td>NaN</td>\n",
       "      <td>...</td>\n",
       "      <td>treatment purpose</td>\n",
       "      <td>- Women and Men; 18 years or more at the time ...</td>\n",
       "      <td>- Inoperable locally advanced or metastatic br...</td>\n",
       "      <td>18age old over</td>\n",
       "      <td>No limit</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>arm A: continue with SoC ET as directed by inv...</td>\n",
       "      <td>[Breast Cancer, Early Breast Cancer]</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>388</th>\n",
       "      <td>jRCT2031230109</td>\n",
       "      <td>NCT05514054</td>\n",
       "      <td>NaN</td>\n",
       "      <td>EMBER-4：2～5年間の術後内分泌療法による前治療歴を有する再発高リスクのER+、HER...</td>\n",
       "      <td>乳癌</td>\n",
       "      <td>乳癌</td>\n",
       "      <td>Breast Neoplasms</td>\n",
       "      <td>早期乳癌患者を対象としたimlunestrantと標準的な内分泌療法の比較試験</td>\n",
       "      <td>3</td>\n",
       "      <td>NaN</td>\n",
       "      <td>...</td>\n",
       "      <td>treatment purpose</td>\n",
       "      <td>-Have a diagnosis of ER+, HER2- early-stage, r...</td>\n",
       "      <td>-Have any evidence of metastatic disease (incl...</td>\n",
       "      <td>18age old over</td>\n",
       "      <td>No limit</td>\n",
       "      <td>Both</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>-Drug: Imlunestrant Administered orally. Other...</td>\n",
       "      <td>[Breast Neoplasms]</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>391</th>\n",
       "      <td>jRCT2031230090</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>MELK阻害剤OTS167POにおける転移性・進行性乳がん患者を対象とした安全性、忍容性およ...</td>\n",
       "      <td>再発・難治性の局所進行性・転移性乳がん及びトリプルネガティブ乳がん</td>\n",
       "      <td>再発・難治性の局所進行性・転移性乳がん及びトリプルネガティブ乳がん</td>\n",
       "      <td>Relapsed/Refractory Locally Advanced or Metast...</td>\n",
       "      <td>再発・難治性の局所進行性または転移性乳がん患者に対して、OTS167を経口カプセルで投与する...</td>\n",
       "      <td>1</td>\n",
       "      <td>NaN</td>\n",
       "      <td>...</td>\n",
       "      <td>treatment purpose</td>\n",
       "      <td>Dose Escalation and Dose Expansion Cohorts 1. ...</td>\n",
       "      <td>Dose Escalation and Expansion Cohorts 1. Women...</td>\n",
       "      <td>18age old over</td>\n",
       "      <td>No limit</td>\n",
       "      <td>Female</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>This is a Phase I dose escalation/expansion mu...</td>\n",
       "      <td>[Relapsed/Refract, Y Locally Advanced, Metasta...</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>392</th>\n",
       "      <td>jRCT2031230096</td>\n",
       "      <td>NCT05774951</td>\n",
       "      <td>NaN</td>\n",
       "      <td>根治的局所領域療法（化学療法の併用または非併用）および標準補助内分泌療法（ET）を少なくとも...</td>\n",
       "      <td>乳がん、早期乳がん</td>\n",
       "      <td>乳がん、早期乳がん</td>\n",
       "      <td>Breast Cancer, Early Breast Cancer</td>\n",
       "      <td>Treatment</td>\n",
       "      <td>3</td>\n",
       "      <td>NaN</td>\n",
       "      <td>...</td>\n",
       "      <td>treatment purpose</td>\n",
       "      <td>- Women and Men, greater than or equal to 18 y...</td>\n",
       "      <td>- Inoperable locally advanced or metastatic br...</td>\n",
       "      <td>18age old over</td>\n",
       "      <td>No limit</td>\n",
       "      <td>Both</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>arm A: continue with SoC ET as directed by inv...</td>\n",
       "      <td>[Breast Cancer, Early Breast Cancer]</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>402</th>\n",
       "      <td>jRCT2061230009</td>\n",
       "      <td>NCT05485766</td>\n",
       "      <td>NaN</td>\n",
       "      <td>gBRCA1/2遺伝子変異を有するトリプルネガティブ原発乳がんに対するプラチナ製剤、PARP...</td>\n",
       "      <td>トリプルネガティブ乳がん</td>\n",
       "      <td>トリプルネガティブ乳がん</td>\n",
       "      <td>Triple Negative Breast Neoplasms</td>\n",
       "      <td>術前療法としてgBRCA変異陽性手術可能または局所進行TNBCに対してペムブロリズマブ+パク...</td>\n",
       "      <td>2</td>\n",
       "      <td>NaN</td>\n",
       "      <td>...</td>\n",
       "      <td>treatment purpose</td>\n",
       "      <td>1)Male/female subjects who are at least 18 yea...</td>\n",
       "      <td>1) Subjects who has a positive urine pregnancy...</td>\n",
       "      <td>18age old exceed</td>\n",
       "      <td>No limit</td>\n",
       "      <td>Both</td>\n",
       "      <td>NaN</td>\n",
       "      <td>Triple Negative Breast Cancer, Breast Neoplasm...</td>\n",
       "      <td>Drug: Pembrolizumab 200 mg fixed dose, IV, eve...</td>\n",
       "      <td>[Triple Negative, Breast Neoplasms]</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>463</th>\n",
       "      <td>jRCT2061220087</td>\n",
       "      <td>NCT05629585</td>\n",
       "      <td>NaN</td>\n",
       "      <td>術前薬物療法後の外科的切除時に乳房及び／又は腋窩リンパ節に浸潤性残存病変を有するステージI～...</td>\n",
       "      <td>乳癌</td>\n",
       "      <td>乳癌</td>\n",
       "      <td>Breast Cancer</td>\n",
       "      <td>術前薬物療法後の外科的切除時に乳房および/または腋窩リンパ節に浸潤性残存病変を有するI~II...</td>\n",
       "      <td>3</td>\n",
       "      <td>NaN</td>\n",
       "      <td>...</td>\n",
       "      <td>treatment purpose</td>\n",
       "      <td>Participant must be &gt;= 18 years at the time of...</td>\n",
       "      <td>Stage IV (metastatic) TNBC. History of prior i...</td>\n",
       "      <td>18age old over</td>\n",
       "      <td>130age old under</td>\n",
       "      <td>Both</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>Arm 1: Dato-DXd 6 mg/kg IV Q3W x 8 cycles + Du...</td>\n",
       "      <td>[Breast Cancer]</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>515</th>\n",
       "      <td>jRCT2031220276</td>\n",
       "      <td>NCT05307705</td>\n",
       "      <td>NaN</td>\n",
       "      <td>PIK3CA H1047R変異を有する進行乳がん患者及びその他の固形がん患者を対象としたLO...</td>\n",
       "      <td>乳がん</td>\n",
       "      <td>乳がん</td>\n",
       "      <td>Breast Cancer</td>\n",
       "      <td>LOXO-783 の単独投与及び他の抗がん剤との併用投与における 第 2 相試験の推奨用量、...</td>\n",
       "      <td>1</td>\n",
       "      <td>NaN</td>\n",
       "      <td>...</td>\n",
       "      <td>treatment purpose</td>\n",
       "      <td>-Have advanced breast cancer or another solid ...</td>\n",
       "      <td>-Medical Conditions -Colorectal cancer -Endome...</td>\n",
       "      <td>18age old over</td>\n",
       "      <td>No limit</td>\n",
       "      <td>Both</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>-Drug: LOXO-783 Oral Other Name: LY3849524 -Dr...</td>\n",
       "      <td>[Breast Cancer]</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>590</th>\n",
       "      <td>jRCT2031210585</td>\n",
       "      <td>NCT05123482</td>\n",
       "      <td>NaN</td>\n",
       "      <td>進行又は転移性固形がん患者を対象としたAZD8205の単独療法及び他の抗がん剤との併用療法に...</td>\n",
       "      <td>乳癌、胆道癌、卵巣癌、子宮内膜癌</td>\n",
       "      <td>乳癌、胆道癌、卵巣癌、子宮内膜癌</td>\n",
       "      <td>Breast Cancer, Biliary Tract Carcinoma, Ovaria...</td>\n",
       "      <td>進行又は転移性固形がんに対する治療法候補としての新規化合物AZD8205の研究</td>\n",
       "      <td>1-2</td>\n",
       "      <td>NaN</td>\n",
       "      <td>...</td>\n",
       "      <td>treatment purpose</td>\n",
       "      <td>- Age18 years or more - Relapsed/metastatic so...</td>\n",
       "      <td>- Treatment with any of the following: 1. Nitr...</td>\n",
       "      <td>18age old over</td>\n",
       "      <td>No limit</td>\n",
       "      <td>Both</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>Sub-study1 AZD8205 Monotherapy AZD8205 is an a...</td>\n",
       "      <td>[Breast Cancer, Biliary Tract Carcinoma, Ovari...</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>629</th>\n",
       "      <td>jRCT2052210099</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>乳がんを有する成人女性及び健康成人女性を対象とした乳房用マイクロ波画像診断装置IGS-000...</td>\n",
       "      <td>乳がん</td>\n",
       "      <td>乳がん</td>\n",
       "      <td>Breast cancer</td>\n",
       "      <td>本治験の目的は,乳がん又はその疑いのある者,並びに乳がん又はその疑いがない者を対象に乳房画像...</td>\n",
       "      <td>2</td>\n",
       "      <td>NaN</td>\n",
       "      <td>...</td>\n",
       "      <td>screening</td>\n",
       "      <td>Breast cancer (including suspected) (1) One of...</td>\n",
       "      <td>(1) There is trauma with bleeding on the breas...</td>\n",
       "      <td>20age old over</td>\n",
       "      <td>No limit</td>\n",
       "      <td>Female</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>The test using IGS-0001 will be performed twic...</td>\n",
       "      <td>[Breast Cancer]</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>729</th>\n",
       "      <td>jRCT2031200057</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>治療抵抗性乳がんを対象としたTDM-812の腫瘍内投与法の安全性評価を目的とした第I相試験</td>\n",
       "      <td>乳がん</td>\n",
       "      <td>乳がん</td>\n",
       "      <td>Breast cancer</td>\n",
       "      <td>TDM-812の腫瘍内投与の安全性および忍容性を評価し、今後の臨床評価に用いる腫瘍内投与にお...</td>\n",
       "      <td>1</td>\n",
       "      <td>NaN</td>\n",
       "      <td>...</td>\n",
       "      <td>treatment purpose</td>\n",
       "      <td>&lt;Disease Characteristics&gt; 1)Patients with inop...</td>\n",
       "      <td>1)Patients correspond to either of the followi...</td>\n",
       "      <td>20age old over</td>\n",
       "      <td>No limit</td>\n",
       "      <td>Female</td>\n",
       "      <td>NaN</td>\n",
       "      <td>Breast cancer</td>\n",
       "      <td>For the target tumor, the dose settled at each...</td>\n",
       "      <td>[Breast Cancer]</td>\n",
       "    </tr>\n",
       "  </tbody>\n",
       "</table>\n",
       "<p>14 rows × 40 columns</p>\n",
       "</div>"
      ],
      "text/plain": [
       "            JRCT ID       NCT No JapicCTI No  \\\n",
       "31   jRCT2061240081  NCT06393374         NaN   \n",
       "157  jRCT2052240059          NaN         NaN   \n",
       "215  jRCT2031230723  NCT06112379         NaN   \n",
       "240  jRCT2061230102  NCT06103864         NaN   \n",
       "307  jRCT2061230074  NCT05952557         NaN   \n",
       "388  jRCT2031230109  NCT05514054         NaN   \n",
       "391  jRCT2031230090          NaN         NaN   \n",
       "392  jRCT2031230096  NCT05774951         NaN   \n",
       "402  jRCT2061230009  NCT05485766         NaN   \n",
       "463  jRCT2061220087  NCT05629585         NaN   \n",
       "515  jRCT2031220276  NCT05307705         NaN   \n",
       "590  jRCT2031210585  NCT05123482         NaN   \n",
       "629  jRCT2052210099          NaN         NaN   \n",
       "729  jRCT2031200057          NaN         NaN   \n",
       "\n",
       "                                                 Title  \\\n",
       "31   術前薬物療法後の手術時に病理学的完全奏効を達成していないトリプルネガティブ乳癌患者を対象に、...   \n",
       "157  遺伝子HSD17B4高メチル化を有するHER2陽性ER陰性乳癌における非手術療法の有用性を評...   \n",
       "215  未治療のトリプルネガティブ又はホルモン受容体低発現／HER2陰性乳癌の成人患者を対象として、...   \n",
       "240  Programmed death-ligand（PD-L1）陽性の局所再発手術不能又は転移性...   \n",
       "307  根治的局所治療（化学療法の併用または非併用）を受けて疾患の兆候のない、再発リスクが中間～高リ...   \n",
       "388  EMBER-4：2～5年間の術後内分泌療法による前治療歴を有する再発高リスクのER+、HER...   \n",
       "391  MELK阻害剤OTS167POにおける転移性・進行性乳がん患者を対象とした安全性、忍容性およ...   \n",
       "392  根治的局所領域療法（化学療法の併用または非併用）および標準補助内分泌療法（ET）を少なくとも...   \n",
       "402  gBRCA1/2遺伝子変異を有するトリプルネガティブ原発乳がんに対するプラチナ製剤、PARP...   \n",
       "463  術前薬物療法後の外科的切除時に乳房及び／又は腋窩リンパ節に浸潤性残存病変を有するステージI～...   \n",
       "515  PIK3CA H1047R変異を有する進行乳がん患者及びその他の固形がん患者を対象としたLO...   \n",
       "590  進行又は転移性固形がん患者を対象としたAZD8205の単独療法及び他の抗がん剤との併用療法に...   \n",
       "629  乳がんを有する成人女性及び健康成人女性を対象とした乳房用マイクロ波画像診断装置IGS-000...   \n",
       "729      治療抵抗性乳がんを対象としたTDM-812の腫瘍内投与法の安全性評価を目的とした第I相試験   \n",
       "\n",
       "                               TargetJ                             Target  \\\n",
       "31                         トリプルネガティブ乳癌                        トリプルネガティブ乳癌   \n",
       "157                                乳がん                                乳がん   \n",
       "215                                 乳癌                                 乳癌   \n",
       "240                                 乳癌                                 乳癌   \n",
       "307                          乳がん、早期乳がん                          乳がん、早期乳がん   \n",
       "388                                 乳癌                                 乳癌   \n",
       "391  再発・難治性の局所進行性・転移性乳がん及びトリプルネガティブ乳がん  再発・難治性の局所進行性・転移性乳がん及びトリプルネガティブ乳がん   \n",
       "392                          乳がん、早期乳がん                          乳がん、早期乳がん   \n",
       "402                       トリプルネガティブ乳がん                       トリプルネガティブ乳がん   \n",
       "463                                 乳癌                                 乳癌   \n",
       "515                                乳がん                                乳がん   \n",
       "590                   乳癌、胆道癌、卵巣癌、子宮内膜癌                   乳癌、胆道癌、卵巣癌、子宮内膜癌   \n",
       "629                                乳がん                                乳がん   \n",
       "729                                乳がん                                乳がん   \n",
       "\n",
       "                                         TargetEnglish  \\\n",
       "31                       Triple negative breast cancer   \n",
       "157                                      Breast cancer   \n",
       "215                                      Breast Cancer   \n",
       "240                                      Breast Cancer   \n",
       "307                 Breast Cancer, Early Breast Cancer   \n",
       "388                                   Breast Neoplasms   \n",
       "391  Relapsed/Refractory Locally Advanced or Metast...   \n",
       "392                 Breast Cancer, Early Breast Cancer   \n",
       "402                   Triple Negative Breast Neoplasms   \n",
       "463                                      Breast Cancer   \n",
       "515                                      Breast Cancer   \n",
       "590  Breast Cancer, Biliary Tract Carcinoma, Ovaria...   \n",
       "629                                      Breast cancer   \n",
       "729                                      Breast cancer   \n",
       "\n",
       "                                              研究・治験の目的 試験等のフェーズ 試験の種類  ...  \\\n",
       "31   術前薬物療法後の手術時に病理学的完全奏効を達成していないトリプルネガティブ乳癌患者を対象に、...        3   NaN  ...   \n",
       "157  HSD17B4高メチル化(HSD17B4 hypermethylation:HH)を有するH...        2   NaN  ...   \n",
       "215  未治療のトリプルネガティブ又はホルモン受容体低発現/HER2陰性乳癌の成人患者を対象として、...        3   NaN  ...   \n",
       "240  PD-L1陽性の局所再発手術不能または転移性TNBC患者を対象に、デュルバルマブ併用または非...        3   NaN  ...   \n",
       "307                                          Treatment        3   NaN  ...   \n",
       "388            早期乳癌患者を対象としたimlunestrantと標準的な内分泌療法の比較試験        3   NaN  ...   \n",
       "391  再発・難治性の局所進行性または転移性乳がん患者に対して、OTS167を経口カプセルで投与する...        1   NaN  ...   \n",
       "392                                          Treatment        3   NaN  ...   \n",
       "402  術前療法としてgBRCA変異陽性手術可能または局所進行TNBCに対してペムブロリズマブ+パク...        2   NaN  ...   \n",
       "463  術前薬物療法後の外科的切除時に乳房および/または腋窩リンパ節に浸潤性残存病変を有するI~II...        3   NaN  ...   \n",
       "515  LOXO-783 の単独投与及び他の抗がん剤との併用投与における 第 2 相試験の推奨用量、...        1   NaN  ...   \n",
       "590            進行又は転移性固形がんに対する治療法候補としての新規化合物AZD8205の研究      1-2   NaN  ...   \n",
       "629  本治験の目的は,乳がん又はその疑いのある者,並びに乳がん又はその疑いがない者を対象に乳房画像...        2   NaN  ...   \n",
       "729  TDM-812の腫瘍内投与の安全性および忍容性を評価し、今後の臨床評価に用いる腫瘍内投与にお...        1   NaN  ...   \n",
       "\n",
       "                purpose                                 Inclusion Criteria  \\\n",
       "31    treatment purpose  -Has centrally confirmed TNBC, as defined by t...   \n",
       "157  diagnostic purpose  1. Histologically confirmed invasive breast ca...   \n",
       "215   treatment purpose  1. Participant must be >= 18 years at the time...   \n",
       "240   treatment purpose  Histologically or cytologically documented loc...   \n",
       "307   treatment purpose  - Women and Men; 18 years or more at the time ...   \n",
       "388   treatment purpose  -Have a diagnosis of ER+, HER2- early-stage, r...   \n",
       "391   treatment purpose  Dose Escalation and Dose Expansion Cohorts 1. ...   \n",
       "392   treatment purpose  - Women and Men, greater than or equal to 18 y...   \n",
       "402   treatment purpose  1)Male/female subjects who are at least 18 yea...   \n",
       "463   treatment purpose  Participant must be >= 18 years at the time of...   \n",
       "515   treatment purpose  -Have advanced breast cancer or another solid ...   \n",
       "590   treatment purpose  - Age18 years or more - Relapsed/metastatic so...   \n",
       "629           screening  Breast cancer (including suspected) (1) One of...   \n",
       "729   treatment purpose  <Disease Characteristics> 1)Patients with inop...   \n",
       "\n",
       "                                    Exclusion Criteria       Age Minimum  \\\n",
       "31   -Has a known germline breast cancer gene (BRCA...    18age old over   \n",
       "157  1. History of other malignancy within the last...    20age old over   \n",
       "215  1. As judged by the investigator, any evidence...    18age old over   \n",
       "240  As judged by investigator, severe or uncontrol...    18age old over   \n",
       "307  - Inoperable locally advanced or metastatic br...    18age old over   \n",
       "388  -Have any evidence of metastatic disease (incl...    18age old over   \n",
       "391  Dose Escalation and Expansion Cohorts 1. Women...    18age old over   \n",
       "392  - Inoperable locally advanced or metastatic br...    18age old over   \n",
       "402  1) Subjects who has a positive urine pregnancy...  18age old exceed   \n",
       "463  Stage IV (metastatic) TNBC. History of prior i...    18age old over   \n",
       "515  -Medical Conditions -Colorectal cancer -Endome...    18age old over   \n",
       "590  - Treatment with any of the following: 1. Nitr...    18age old over   \n",
       "629  (1) There is trauma with bleeding on the breas...    20age old over   \n",
       "729  1)Patients correspond to either of the followi...    20age old over   \n",
       "\n",
       "          Age Maximum  Gender Discontinuation Criteria  \\\n",
       "31           No limit    Both                      NaN   \n",
       "157          No limit  Female                      NaN   \n",
       "215          No limit    Both                      NaN   \n",
       "240          No limit    Both                      NaN   \n",
       "307          No limit     NaN                      NaN   \n",
       "388          No limit    Both                      NaN   \n",
       "391          No limit  Female                      NaN   \n",
       "392          No limit    Both                      NaN   \n",
       "402          No limit    Both                      NaN   \n",
       "463  130age old under    Both                      NaN   \n",
       "515          No limit    Both                      NaN   \n",
       "590          No limit    Both                      NaN   \n",
       "629          No limit  Female                      NaN   \n",
       "729          No limit  Female                      NaN   \n",
       "\n",
       "                                               Keyword  \\\n",
       "31                                                 NaN   \n",
       "157                                                NaN   \n",
       "215                                                NaN   \n",
       "240                                                NaN   \n",
       "307                                                NaN   \n",
       "388                                                NaN   \n",
       "391                                                NaN   \n",
       "392                                                NaN   \n",
       "402  Triple Negative Breast Cancer, Breast Neoplasm...   \n",
       "463                                                NaN   \n",
       "515                                                NaN   \n",
       "590                                                NaN   \n",
       "629                                                NaN   \n",
       "729                                      Breast cancer   \n",
       "\n",
       "                                       Intervention(s)  \\\n",
       "31   -Arm 1: MK-2870 4mg/kg intravenous (IV) every ...   \n",
       "157  Omitting breast surgery after preoperative che...   \n",
       "215  - Experimental arm: Dato-DXd plus durvalumab n...   \n",
       "240  Arm 1: Dato-DXd + durvalumab Arm 2: Investigat...   \n",
       "307  arm A: continue with SoC ET as directed by inv...   \n",
       "388  -Drug: Imlunestrant Administered orally. Other...   \n",
       "391  This is a Phase I dose escalation/expansion mu...   \n",
       "392  arm A: continue with SoC ET as directed by inv...   \n",
       "402  Drug: Pembrolizumab 200 mg fixed dose, IV, eve...   \n",
       "463  Arm 1: Dato-DXd 6 mg/kg IV Q3W x 8 cycles + Du...   \n",
       "515  -Drug: LOXO-783 Oral Other Name: LY3849524 -Dr...   \n",
       "590  Sub-study1 AZD8205 Monotherapy AZD8205 is an a...   \n",
       "629  The test using IGS-0001 will be performed twic...   \n",
       "729  For the target tumor, the dose settled at each...   \n",
       "\n",
       "                                            TargetWord  \n",
       "31                    [Triple Negative, Breast Cancer]  \n",
       "157                                    [Breast Cancer]  \n",
       "215                                    [Breast Cancer]  \n",
       "240                                    [Breast Cancer]  \n",
       "307               [Breast Cancer, Early Breast Cancer]  \n",
       "388                                 [Breast Neoplasms]  \n",
       "391  [Relapsed/Refract, Y Locally Advanced, Metasta...  \n",
       "392               [Breast Cancer, Early Breast Cancer]  \n",
       "402                [Triple Negative, Breast Neoplasms]  \n",
       "463                                    [Breast Cancer]  \n",
       "515                                    [Breast Cancer]  \n",
       "590  [Breast Cancer, Biliary Tract Carcinoma, Ovari...  \n",
       "629                                    [Breast Cancer]  \n",
       "729                                    [Breast Cancer]  \n",
       "\n",
       "[14 rows x 40 columns]"
      ]
     },
     "execution_count": 86,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "# ターゲットリスト全体を処理\n",
    "matched_indices = []\n",
    "target_vecs_list = []\n",
    "cosine_scores_list = []\n",
    "for idx, target_words in enumerate(basedf['TargetWord']):\n",
    "    # ターゲット内の各単語をベクトル化\n",
    "    target_vecs = model.encode(target_words, convert_to_tensor=True)\n",
    "    # コサイン類似度を計算\n",
    "    cosine_scores = util.cos_sim(query_vec, target_vecs).squeeze()\n",
    "    target_vecs_list.append(target_vecs)\n",
    "    cosine_scores_list.append(cosine_scores)\n",
    "    # 閾値を超えるか確認\n",
    "    if (cosine_scores >= threshold).any():  # いずれかが閾値を超えている場合\n",
    "        matched_indices.append(idx)\n",
    "\n",
    "# 抽出\n",
    "matched_df = basedf.iloc[matched_indices]\n",
    "matched_df"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 14,
   "metadata": {},
   "outputs": [],
   "source": [
    "# 全データのターゲット列をベクトル化\n",
    "target_list = basedf['TargetEnglish'].tolist()\n",
    "target_vecs = model.encode(target_list, convert_to_tensor=True)\n",
    "# コサイン類似度を計算\n",
    "cosine_scores = util.cos_sim(query_vec, target_vecs).squeeze()"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 15,
   "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>JRCT ID</th>\n",
       "      <th>NCT No</th>\n",
       "      <th>JapicCTI No</th>\n",
       "      <th>Title</th>\n",
       "      <th>TargetJ</th>\n",
       "      <th>Target</th>\n",
       "      <th>TargetEnglish</th>\n",
       "      <th>研究・治験の目的</th>\n",
       "      <th>試験等のフェーズ</th>\n",
       "      <th>試験の種類</th>\n",
       "      <th>...</th>\n",
       "      <th>purpose</th>\n",
       "      <th>Inclusion Criteria</th>\n",
       "      <th>Exclusion Criteria</th>\n",
       "      <th>Age Minimum</th>\n",
       "      <th>Age Maximum</th>\n",
       "      <th>Gender</th>\n",
       "      <th>Discontinuation Criteria</th>\n",
       "      <th>Keyword</th>\n",
       "      <th>Intervention(s)</th>\n",
       "      <th>TargetWord</th>\n",
       "    </tr>\n",
       "  </thead>\n",
       "  <tbody>\n",
       "    <tr>\n",
       "      <th>157</th>\n",
       "      <td>jRCT2052240059</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>遺伝子HSD17B4高メチル化を有するHER2陽性ER陰性乳癌における非手術療法の有用性を評...</td>\n",
       "      <td>乳がん</td>\n",
       "      <td>乳がん</td>\n",
       "      <td>Breast cancer</td>\n",
       "      <td>HSD17B4高メチル化(HSD17B4 hypermethylation:HH)を有するH...</td>\n",
       "      <td>2</td>\n",
       "      <td>NaN</td>\n",
       "      <td>...</td>\n",
       "      <td>diagnostic purpose</td>\n",
       "      <td>1. Histologically confirmed invasive breast ca...</td>\n",
       "      <td>1. History of other malignancy within the last...</td>\n",
       "      <td>20age old over</td>\n",
       "      <td>No limit</td>\n",
       "      <td>Female</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>Omitting breast surgery after preoperative che...</td>\n",
       "      <td>[Breast cancer]</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>170</th>\n",
       "      <td>jRCT2031240096</td>\n",
       "      <td>NCT06380751</td>\n",
       "      <td>NaN</td>\n",
       "      <td>本治験の主要目的は、BRCA1、BRCA2 又は PALB2 変異を有し、HR 陽性、HER...</td>\n",
       "      <td>進行乳癌</td>\n",
       "      <td>進行乳癌</td>\n",
       "      <td>Advanced Breast Cancer</td>\n",
       "      <td>Treatment</td>\n",
       "      <td>3</td>\n",
       "      <td>NaN</td>\n",
       "      <td>...</td>\n",
       "      <td>treatment purpose</td>\n",
       "      <td>- Adult females, pre/peri-menopausal and/or po...</td>\n",
       "      <td>- Participants with history of MDS/AML or with...</td>\n",
       "      <td>18age old over</td>\n",
       "      <td>No limit</td>\n",
       "      <td>Both</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>Experimental: Arm 1: saruparib (AZD5305) plus ...</td>\n",
       "      <td>[Advanced Breast Cancer]</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>215</th>\n",
       "      <td>jRCT2031230723</td>\n",
       "      <td>NCT06112379</td>\n",
       "      <td>NaN</td>\n",
       "      <td>未治療のトリプルネガティブ又はホルモン受容体低発現／HER2陰性乳癌の成人患者を対象として、...</td>\n",
       "      <td>乳癌</td>\n",
       "      <td>乳癌</td>\n",
       "      <td>Breast Cancer</td>\n",
       "      <td>未治療のトリプルネガティブ又はホルモン受容体低発現/HER2陰性乳癌の成人患者を対象として、...</td>\n",
       "      <td>3</td>\n",
       "      <td>NaN</td>\n",
       "      <td>...</td>\n",
       "      <td>treatment purpose</td>\n",
       "      <td>1. Participant must be &gt;= 18 years at the time...</td>\n",
       "      <td>1. As judged by the investigator, any evidence...</td>\n",
       "      <td>18age old over</td>\n",
       "      <td>No limit</td>\n",
       "      <td>Both</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>- Experimental arm: Dato-DXd plus durvalumab n...</td>\n",
       "      <td>[Breast Cancer]</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>240</th>\n",
       "      <td>jRCT2061230102</td>\n",
       "      <td>NCT06103864</td>\n",
       "      <td>NaN</td>\n",
       "      <td>Programmed death-ligand（PD-L1）陽性の局所再発手術不能又は転移性...</td>\n",
       "      <td>乳癌</td>\n",
       "      <td>乳癌</td>\n",
       "      <td>Breast Cancer</td>\n",
       "      <td>PD-L1陽性の局所再発手術不能または転移性TNBC患者を対象に、デュルバルマブ併用または非...</td>\n",
       "      <td>3</td>\n",
       "      <td>NaN</td>\n",
       "      <td>...</td>\n",
       "      <td>treatment purpose</td>\n",
       "      <td>Histologically or cytologically documented loc...</td>\n",
       "      <td>As judged by investigator, severe or uncontrol...</td>\n",
       "      <td>18age old over</td>\n",
       "      <td>No limit</td>\n",
       "      <td>Both</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>Arm 1: Dato-DXd + durvalumab Arm 2: Investigat...</td>\n",
       "      <td>[Breast Cancer]</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>307</th>\n",
       "      <td>jRCT2061230074</td>\n",
       "      <td>NCT05952557</td>\n",
       "      <td>NaN</td>\n",
       "      <td>根治的局所治療（化学療法の併用または非併用）を受けて疾患の兆候のない、再発リスクが中間～高リ...</td>\n",
       "      <td>乳がん、早期乳がん</td>\n",
       "      <td>乳がん、早期乳がん</td>\n",
       "      <td>Breast Cancer, Early Breast Cancer</td>\n",
       "      <td>Treatment</td>\n",
       "      <td>3</td>\n",
       "      <td>NaN</td>\n",
       "      <td>...</td>\n",
       "      <td>treatment purpose</td>\n",
       "      <td>- Women and Men; 18 years or more at the time ...</td>\n",
       "      <td>- Inoperable locally advanced or metastatic br...</td>\n",
       "      <td>18age old over</td>\n",
       "      <td>No limit</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>arm A: continue with SoC ET as directed by inv...</td>\n",
       "      <td>[Breast Cancer, Early Breast Cancer]</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>342</th>\n",
       "      <td>jRCT2051230094</td>\n",
       "      <td>NCT05794906</td>\n",
       "      <td>NaN</td>\n",
       "      <td>前立腺癌の高リスク生化学的再発（BCR）患者を対象としたアンドロゲン遮断療法（ADT）を併用...</td>\n",
       "      <td>前立腺癌</td>\n",
       "      <td>前立腺癌</td>\n",
       "      <td>prostate cancer</td>\n",
       "      <td>BAY 1841788(darolutamide)とADTの24ヵ月間併用投与によって、プラ...</td>\n",
       "      <td>3</td>\n",
       "      <td>NaN</td>\n",
       "      <td>...</td>\n",
       "      <td>treatment purpose</td>\n",
       "      <td>- Histologically or cytologically confirmed ad...</td>\n",
       "      <td>- Pathological finding consistent with small c...</td>\n",
       "      <td>18age old over</td>\n",
       "      <td>No limit</td>\n",
       "      <td>Male</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>Drug: Darolutamide (BAY1841788, Nubeqa) Coated...</td>\n",
       "      <td>[prostate cancer]</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>351</th>\n",
       "      <td>jRCT2031230279</td>\n",
       "      <td>NCT05862285</td>\n",
       "      <td>NaN</td>\n",
       "      <td>GENENTECH社及び／又はF.HOFFMANN-LA ROCHE LTDが依頼した試験に...</td>\n",
       "      <td>癌</td>\n",
       "      <td>癌</td>\n",
       "      <td>Cancer</td>\n",
       "      <td>本継続投与試験の目的は,親治験から移行する時点でまだ治験治療を受けており,その地域でその治療...</td>\n",
       "      <td>3</td>\n",
       "      <td>NaN</td>\n",
       "      <td>...</td>\n",
       "      <td>treatment purpose</td>\n",
       "      <td>- Eligible for continuing Roche IMP-based ther...</td>\n",
       "      <td>- Meet any of the study treatment discontinuat...</td>\n",
       "      <td>18age old over</td>\n",
       "      <td>No limit</td>\n",
       "      <td>Both</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>Ipatasertib: Ipatasertib will be administered ...</td>\n",
       "      <td>[Cancer]</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>388</th>\n",
       "      <td>jRCT2031230109</td>\n",
       "      <td>NCT05514054</td>\n",
       "      <td>NaN</td>\n",
       "      <td>EMBER-4：2～5年間の術後内分泌療法による前治療歴を有する再発高リスクのER+、HER...</td>\n",
       "      <td>乳癌</td>\n",
       "      <td>乳癌</td>\n",
       "      <td>Breast Neoplasms</td>\n",
       "      <td>早期乳癌患者を対象としたimlunestrantと標準的な内分泌療法の比較試験</td>\n",
       "      <td>3</td>\n",
       "      <td>NaN</td>\n",
       "      <td>...</td>\n",
       "      <td>treatment purpose</td>\n",
       "      <td>-Have a diagnosis of ER+, HER2- early-stage, r...</td>\n",
       "      <td>-Have any evidence of metastatic disease (incl...</td>\n",
       "      <td>18age old over</td>\n",
       "      <td>No limit</td>\n",
       "      <td>Both</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>-Drug: Imlunestrant Administered orally. Other...</td>\n",
       "      <td>[Breast Neoplasms]</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>392</th>\n",
       "      <td>jRCT2031230096</td>\n",
       "      <td>NCT05774951</td>\n",
       "      <td>NaN</td>\n",
       "      <td>根治的局所領域療法（化学療法の併用または非併用）および標準補助内分泌療法（ET）を少なくとも...</td>\n",
       "      <td>乳がん、早期乳がん</td>\n",
       "      <td>乳がん、早期乳がん</td>\n",
       "      <td>Breast Cancer, Early Breast Cancer</td>\n",
       "      <td>Treatment</td>\n",
       "      <td>3</td>\n",
       "      <td>NaN</td>\n",
       "      <td>...</td>\n",
       "      <td>treatment purpose</td>\n",
       "      <td>- Women and Men, greater than or equal to 18 y...</td>\n",
       "      <td>- Inoperable locally advanced or metastatic br...</td>\n",
       "      <td>18age old over</td>\n",
       "      <td>No limit</td>\n",
       "      <td>Both</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>arm A: continue with SoC ET as directed by inv...</td>\n",
       "      <td>[Breast Cancer, Early Breast Cancer]</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>463</th>\n",
       "      <td>jRCT2061220087</td>\n",
       "      <td>NCT05629585</td>\n",
       "      <td>NaN</td>\n",
       "      <td>術前薬物療法後の外科的切除時に乳房及び／又は腋窩リンパ節に浸潤性残存病変を有するステージI～...</td>\n",
       "      <td>乳癌</td>\n",
       "      <td>乳癌</td>\n",
       "      <td>Breast Cancer</td>\n",
       "      <td>術前薬物療法後の外科的切除時に乳房および/または腋窩リンパ節に浸潤性残存病変を有するI~II...</td>\n",
       "      <td>3</td>\n",
       "      <td>NaN</td>\n",
       "      <td>...</td>\n",
       "      <td>treatment purpose</td>\n",
       "      <td>Participant must be &gt;= 18 years at the time of...</td>\n",
       "      <td>Stage IV (metastatic) TNBC. History of prior i...</td>\n",
       "      <td>18age old over</td>\n",
       "      <td>130age old under</td>\n",
       "      <td>Both</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>Arm 1: Dato-DXd 6 mg/kg IV Q3W x 8 cycles + Du...</td>\n",
       "      <td>[Breast Cancer]</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>515</th>\n",
       "      <td>jRCT2031220276</td>\n",
       "      <td>NCT05307705</td>\n",
       "      <td>NaN</td>\n",
       "      <td>PIK3CA H1047R変異を有する進行乳がん患者及びその他の固形がん患者を対象としたLO...</td>\n",
       "      <td>乳がん</td>\n",
       "      <td>乳がん</td>\n",
       "      <td>Breast Cancer</td>\n",
       "      <td>LOXO-783 の単独投与及び他の抗がん剤との併用投与における 第 2 相試験の推奨用量、...</td>\n",
       "      <td>1</td>\n",
       "      <td>NaN</td>\n",
       "      <td>...</td>\n",
       "      <td>treatment purpose</td>\n",
       "      <td>-Have advanced breast cancer or another solid ...</td>\n",
       "      <td>-Medical Conditions -Colorectal cancer -Endome...</td>\n",
       "      <td>18age old over</td>\n",
       "      <td>No limit</td>\n",
       "      <td>Both</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>-Drug: LOXO-783 Oral Other Name: LY3849524 -Dr...</td>\n",
       "      <td>[Breast Cancer]</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>593</th>\n",
       "      <td>jRCT2031210560</td>\n",
       "      <td>NCT05061823</td>\n",
       "      <td>NaN</td>\n",
       "      <td>複数のBintrafusp alfa（M7824）臨床試験の被験者を対象に長期データを収集す...</td>\n",
       "      <td>固形癌</td>\n",
       "      <td>固形癌</td>\n",
       "      <td>Cancer</td>\n",
       "      <td>本治験は、実施中のbintrafusp alfaの親試験( (NCT02517398, NC...</td>\n",
       "      <td>3</td>\n",
       "      <td>NaN</td>\n",
       "      <td>...</td>\n",
       "      <td>treatment purpose</td>\n",
       "      <td>- Participants who are currently enrolled in a...</td>\n",
       "      <td>- Participants who are pregnant or currently i...</td>\n",
       "      <td>18age old over</td>\n",
       "      <td>No limit</td>\n",
       "      <td>Both</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>Participants who are continuing treatment with...</td>\n",
       "      <td>[Cancer]</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>600</th>\n",
       "      <td>jRCT2041210123</td>\n",
       "      <td>NCT05114746</td>\n",
       "      <td>NaN</td>\n",
       "      <td>進行性PSMA陽性転移性去勢抵抗性前立腺がん（mCRPC）を有する患者を対象に177Lu-P...</td>\n",
       "      <td>前立腺癌</td>\n",
       "      <td>前立腺癌</td>\n",
       "      <td>Prostate Cancer</td>\n",
       "      <td>本治験の目的は、進行性PSMA陽性mCRPCの日本人患者に177Lu-PSMA-617を投与...</td>\n",
       "      <td>2</td>\n",
       "      <td>NaN</td>\n",
       "      <td>...</td>\n",
       "      <td>treatment purpose</td>\n",
       "      <td>- ECOG performance status: 1. Post-taxane popu...</td>\n",
       "      <td>- Previous treatment with any of the following...</td>\n",
       "      <td>20age old over</td>\n",
       "      <td>No limit</td>\n",
       "      <td>Male</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>Radiation: 177Lu-PSMA-617 administered intrave...</td>\n",
       "      <td>[Prostate Cancer]</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>629</th>\n",
       "      <td>jRCT2052210099</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>乳がんを有する成人女性及び健康成人女性を対象とした乳房用マイクロ波画像診断装置IGS-000...</td>\n",
       "      <td>乳がん</td>\n",
       "      <td>乳がん</td>\n",
       "      <td>Breast cancer</td>\n",
       "      <td>本治験の目的は,乳がん又はその疑いのある者,並びに乳がん又はその疑いがない者を対象に乳房画像...</td>\n",
       "      <td>2</td>\n",
       "      <td>NaN</td>\n",
       "      <td>...</td>\n",
       "      <td>screening</td>\n",
       "      <td>Breast cancer (including suspected) (1) One of...</td>\n",
       "      <td>(1) There is trauma with bleeding on the breas...</td>\n",
       "      <td>20age old over</td>\n",
       "      <td>No limit</td>\n",
       "      <td>Female</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>The test using IGS-0001 will be performed twic...</td>\n",
       "      <td>[Breast cancer]</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>691</th>\n",
       "      <td>jRCT2031200384</td>\n",
       "      <td>NCT04221542</td>\n",
       "      <td>NaN</td>\n",
       "      <td>転移性去勢抵抗性前立腺がん患者を対象としたAMG 509の安全性、忍容性、薬物動態及び有効性...</td>\n",
       "      <td>前立腺がん</td>\n",
       "      <td>前立腺がん</td>\n",
       "      <td>Prostate Cancer</td>\n",
       "      <td>成人被験者のAMG 509の安全性及び忍容性を評価し、最大耐量(MTD)又は第II 相推奨用...</td>\n",
       "      <td>1</td>\n",
       "      <td>NaN</td>\n",
       "      <td>...</td>\n",
       "      <td>treatment purpose</td>\n",
       "      <td>- Parts 1, 2, and 5: Participants with histolo...</td>\n",
       "      <td>- Pathological finding consistent with pure sm...</td>\n",
       "      <td>18age old over</td>\n",
       "      <td>No limit</td>\n",
       "      <td>Male</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>- Experimental: Part 1: AMG 509 Intravenous (I...</td>\n",
       "      <td>[Prostate Cancer]</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>698</th>\n",
       "      <td>jRCT2071200086</td>\n",
       "      <td>NCT04640623</td>\n",
       "      <td>NaN</td>\n",
       "      <td>膀胱全摘除術に不適格である，又は膀胱全摘除術を選択しなかった，カルメット・ゲラン桿菌（BCG...</td>\n",
       "      <td>膀胱癌</td>\n",
       "      <td>膀胱癌</td>\n",
       "      <td>Bladder Cancer</td>\n",
       "      <td>試験の目的は、高グレードTa又はT1乳頭状癌の併発の有無を問わない上皮内癌(CIS)患者にT...</td>\n",
       "      <td>2</td>\n",
       "      <td>NaN</td>\n",
       "      <td>...</td>\n",
       "      <td>treatment purpose</td>\n",
       "      <td>- Histologically confirmed diagnosis of persis...</td>\n",
       "      <td>- Histologically confirmed, muscle-invasive, l...</td>\n",
       "      <td>18age old over</td>\n",
       "      <td>No limit</td>\n",
       "      <td>Both</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>TAR-200 Cohort 1: TAR-200 and Cetrelimab: TAR-...</td>\n",
       "      <td>[Bladder Cancer]</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>729</th>\n",
       "      <td>jRCT2031200057</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>治療抵抗性乳がんを対象としたTDM-812の腫瘍内投与法の安全性評価を目的とした第I相試験</td>\n",
       "      <td>乳がん</td>\n",
       "      <td>乳がん</td>\n",
       "      <td>Breast cancer</td>\n",
       "      <td>TDM-812の腫瘍内投与の安全性および忍容性を評価し、今後の臨床評価に用いる腫瘍内投与にお...</td>\n",
       "      <td>1</td>\n",
       "      <td>NaN</td>\n",
       "      <td>...</td>\n",
       "      <td>treatment purpose</td>\n",
       "      <td>&lt;Disease Characteristics&gt; 1)Patients with inop...</td>\n",
       "      <td>1)Patients correspond to either of the followi...</td>\n",
       "      <td>20age old over</td>\n",
       "      <td>No limit</td>\n",
       "      <td>Female</td>\n",
       "      <td>NaN</td>\n",
       "      <td>Breast cancer</td>\n",
       "      <td>For the target tumor, the dose settled at each...</td>\n",
       "      <td>[Breast cancer]</td>\n",
       "    </tr>\n",
       "  </tbody>\n",
       "</table>\n",
       "<p>17 rows × 40 columns</p>\n",
       "</div>"
      ],
      "text/plain": [
       "            JRCT ID       NCT No JapicCTI No  \\\n",
       "157  jRCT2052240059          NaN         NaN   \n",
       "170  jRCT2031240096  NCT06380751         NaN   \n",
       "215  jRCT2031230723  NCT06112379         NaN   \n",
       "240  jRCT2061230102  NCT06103864         NaN   \n",
       "307  jRCT2061230074  NCT05952557         NaN   \n",
       "342  jRCT2051230094  NCT05794906         NaN   \n",
       "351  jRCT2031230279  NCT05862285         NaN   \n",
       "388  jRCT2031230109  NCT05514054         NaN   \n",
       "392  jRCT2031230096  NCT05774951         NaN   \n",
       "463  jRCT2061220087  NCT05629585         NaN   \n",
       "515  jRCT2031220276  NCT05307705         NaN   \n",
       "593  jRCT2031210560  NCT05061823         NaN   \n",
       "600  jRCT2041210123  NCT05114746         NaN   \n",
       "629  jRCT2052210099          NaN         NaN   \n",
       "691  jRCT2031200384  NCT04221542         NaN   \n",
       "698  jRCT2071200086  NCT04640623         NaN   \n",
       "729  jRCT2031200057          NaN         NaN   \n",
       "\n",
       "                                                 Title    TargetJ     Target  \\\n",
       "157  遺伝子HSD17B4高メチル化を有するHER2陽性ER陰性乳癌における非手術療法の有用性を評...        乳がん        乳がん   \n",
       "170  本治験の主要目的は、BRCA1、BRCA2 又は PALB2 変異を有し、HR 陽性、HER...       進行乳癌       進行乳癌   \n",
       "215  未治療のトリプルネガティブ又はホルモン受容体低発現／HER2陰性乳癌の成人患者を対象として、...         乳癌         乳癌   \n",
       "240  Programmed death-ligand（PD-L1）陽性の局所再発手術不能又は転移性...         乳癌         乳癌   \n",
       "307  根治的局所治療（化学療法の併用または非併用）を受けて疾患の兆候のない、再発リスクが中間～高リ...  乳がん、早期乳がん  乳がん、早期乳がん   \n",
       "342  前立腺癌の高リスク生化学的再発（BCR）患者を対象としたアンドロゲン遮断療法（ADT）を併用...       前立腺癌       前立腺癌   \n",
       "351  GENENTECH社及び／又はF.HOFFMANN-LA ROCHE LTDが依頼した試験に...          癌          癌   \n",
       "388  EMBER-4：2～5年間の術後内分泌療法による前治療歴を有する再発高リスクのER+、HER...         乳癌         乳癌   \n",
       "392  根治的局所領域療法（化学療法の併用または非併用）および標準補助内分泌療法（ET）を少なくとも...  乳がん、早期乳がん  乳がん、早期乳がん   \n",
       "463  術前薬物療法後の外科的切除時に乳房及び／又は腋窩リンパ節に浸潤性残存病変を有するステージI～...         乳癌         乳癌   \n",
       "515  PIK3CA H1047R変異を有する進行乳がん患者及びその他の固形がん患者を対象としたLO...        乳がん        乳がん   \n",
       "593  複数のBintrafusp alfa（M7824）臨床試験の被験者を対象に長期データを収集す...        固形癌        固形癌   \n",
       "600  進行性PSMA陽性転移性去勢抵抗性前立腺がん（mCRPC）を有する患者を対象に177Lu-P...       前立腺癌       前立腺癌   \n",
       "629  乳がんを有する成人女性及び健康成人女性を対象とした乳房用マイクロ波画像診断装置IGS-000...        乳がん        乳がん   \n",
       "691  転移性去勢抵抗性前立腺がん患者を対象としたAMG 509の安全性、忍容性、薬物動態及び有効性...      前立腺がん      前立腺がん   \n",
       "698  膀胱全摘除術に不適格である，又は膀胱全摘除術を選択しなかった，カルメット・ゲラン桿菌（BCG...        膀胱癌        膀胱癌   \n",
       "729      治療抵抗性乳がんを対象としたTDM-812の腫瘍内投与法の安全性評価を目的とした第I相試験        乳がん        乳がん   \n",
       "\n",
       "                          TargetEnglish  \\\n",
       "157                       Breast cancer   \n",
       "170              Advanced Breast Cancer   \n",
       "215                       Breast Cancer   \n",
       "240                       Breast Cancer   \n",
       "307  Breast Cancer, Early Breast Cancer   \n",
       "342                     prostate cancer   \n",
       "351                              Cancer   \n",
       "388                    Breast Neoplasms   \n",
       "392  Breast Cancer, Early Breast Cancer   \n",
       "463                       Breast Cancer   \n",
       "515                       Breast Cancer   \n",
       "593                              Cancer   \n",
       "600                     Prostate Cancer   \n",
       "629                       Breast cancer   \n",
       "691                     Prostate Cancer   \n",
       "698                      Bladder Cancer   \n",
       "729                       Breast cancer   \n",
       "\n",
       "                                              研究・治験の目的 試験等のフェーズ 試験の種類  ...  \\\n",
       "157  HSD17B4高メチル化(HSD17B4 hypermethylation:HH)を有するH...        2   NaN  ...   \n",
       "170                                          Treatment        3   NaN  ...   \n",
       "215  未治療のトリプルネガティブ又はホルモン受容体低発現/HER2陰性乳癌の成人患者を対象として、...        3   NaN  ...   \n",
       "240  PD-L1陽性の局所再発手術不能または転移性TNBC患者を対象に、デュルバルマブ併用または非...        3   NaN  ...   \n",
       "307                                          Treatment        3   NaN  ...   \n",
       "342  BAY 1841788(darolutamide)とADTの24ヵ月間併用投与によって、プラ...        3   NaN  ...   \n",
       "351  本継続投与試験の目的は,親治験から移行する時点でまだ治験治療を受けており,その地域でその治療...        3   NaN  ...   \n",
       "388            早期乳癌患者を対象としたimlunestrantと標準的な内分泌療法の比較試験        3   NaN  ...   \n",
       "392                                          Treatment        3   NaN  ...   \n",
       "463  術前薬物療法後の外科的切除時に乳房および/または腋窩リンパ節に浸潤性残存病変を有するI~II...        3   NaN  ...   \n",
       "515  LOXO-783 の単独投与及び他の抗がん剤との併用投与における 第 2 相試験の推奨用量、...        1   NaN  ...   \n",
       "593  本治験は、実施中のbintrafusp alfaの親試験( (NCT02517398, NC...        3   NaN  ...   \n",
       "600  本治験の目的は、進行性PSMA陽性mCRPCの日本人患者に177Lu-PSMA-617を投与...        2   NaN  ...   \n",
       "629  本治験の目的は,乳がん又はその疑いのある者,並びに乳がん又はその疑いがない者を対象に乳房画像...        2   NaN  ...   \n",
       "691  成人被験者のAMG 509の安全性及び忍容性を評価し、最大耐量(MTD)又は第II 相推奨用...        1   NaN  ...   \n",
       "698  試験の目的は、高グレードTa又はT1乳頭状癌の併発の有無を問わない上皮内癌(CIS)患者にT...        2   NaN  ...   \n",
       "729  TDM-812の腫瘍内投与の安全性および忍容性を評価し、今後の臨床評価に用いる腫瘍内投与にお...        1   NaN  ...   \n",
       "\n",
       "                purpose                                 Inclusion Criteria  \\\n",
       "157  diagnostic purpose  1. Histologically confirmed invasive breast ca...   \n",
       "170   treatment purpose  - Adult females, pre/peri-menopausal and/or po...   \n",
       "215   treatment purpose  1. Participant must be >= 18 years at the time...   \n",
       "240   treatment purpose  Histologically or cytologically documented loc...   \n",
       "307   treatment purpose  - Women and Men; 18 years or more at the time ...   \n",
       "342   treatment purpose  - Histologically or cytologically confirmed ad...   \n",
       "351   treatment purpose  - Eligible for continuing Roche IMP-based ther...   \n",
       "388   treatment purpose  -Have a diagnosis of ER+, HER2- early-stage, r...   \n",
       "392   treatment purpose  - Women and Men, greater than or equal to 18 y...   \n",
       "463   treatment purpose  Participant must be >= 18 years at the time of...   \n",
       "515   treatment purpose  -Have advanced breast cancer or another solid ...   \n",
       "593   treatment purpose  - Participants who are currently enrolled in a...   \n",
       "600   treatment purpose  - ECOG performance status: 1. Post-taxane popu...   \n",
       "629           screening  Breast cancer (including suspected) (1) One of...   \n",
       "691   treatment purpose  - Parts 1, 2, and 5: Participants with histolo...   \n",
       "698   treatment purpose  - Histologically confirmed diagnosis of persis...   \n",
       "729   treatment purpose  <Disease Characteristics> 1)Patients with inop...   \n",
       "\n",
       "                                    Exclusion Criteria     Age Minimum  \\\n",
       "157  1. History of other malignancy within the last...  20age old over   \n",
       "170  - Participants with history of MDS/AML or with...  18age old over   \n",
       "215  1. As judged by the investigator, any evidence...  18age old over   \n",
       "240  As judged by investigator, severe or uncontrol...  18age old over   \n",
       "307  - Inoperable locally advanced or metastatic br...  18age old over   \n",
       "342  - Pathological finding consistent with small c...  18age old over   \n",
       "351  - Meet any of the study treatment discontinuat...  18age old over   \n",
       "388  -Have any evidence of metastatic disease (incl...  18age old over   \n",
       "392  - Inoperable locally advanced or metastatic br...  18age old over   \n",
       "463  Stage IV (metastatic) TNBC. History of prior i...  18age old over   \n",
       "515  -Medical Conditions -Colorectal cancer -Endome...  18age old over   \n",
       "593  - Participants who are pregnant or currently i...  18age old over   \n",
       "600  - Previous treatment with any of the following...  20age old over   \n",
       "629  (1) There is trauma with bleeding on the breas...  20age old over   \n",
       "691  - Pathological finding consistent with pure sm...  18age old over   \n",
       "698  - Histologically confirmed, muscle-invasive, l...  18age old over   \n",
       "729  1)Patients correspond to either of the followi...  20age old over   \n",
       "\n",
       "          Age Maximum  Gender Discontinuation Criteria        Keyword  \\\n",
       "157          No limit  Female                      NaN            NaN   \n",
       "170          No limit    Both                      NaN            NaN   \n",
       "215          No limit    Both                      NaN            NaN   \n",
       "240          No limit    Both                      NaN            NaN   \n",
       "307          No limit     NaN                      NaN            NaN   \n",
       "342          No limit    Male                      NaN            NaN   \n",
       "351          No limit    Both                      NaN            NaN   \n",
       "388          No limit    Both                      NaN            NaN   \n",
       "392          No limit    Both                      NaN            NaN   \n",
       "463  130age old under    Both                      NaN            NaN   \n",
       "515          No limit    Both                      NaN            NaN   \n",
       "593          No limit    Both                      NaN            NaN   \n",
       "600          No limit    Male                      NaN            NaN   \n",
       "629          No limit  Female                      NaN            NaN   \n",
       "691          No limit    Male                      NaN            NaN   \n",
       "698          No limit    Both                      NaN            NaN   \n",
       "729          No limit  Female                      NaN  Breast cancer   \n",
       "\n",
       "                                       Intervention(s)  \\\n",
       "157  Omitting breast surgery after preoperative che...   \n",
       "170  Experimental: Arm 1: saruparib (AZD5305) plus ...   \n",
       "215  - Experimental arm: Dato-DXd plus durvalumab n...   \n",
       "240  Arm 1: Dato-DXd + durvalumab Arm 2: Investigat...   \n",
       "307  arm A: continue with SoC ET as directed by inv...   \n",
       "342  Drug: Darolutamide (BAY1841788, Nubeqa) Coated...   \n",
       "351  Ipatasertib: Ipatasertib will be administered ...   \n",
       "388  -Drug: Imlunestrant Administered orally. Other...   \n",
       "392  arm A: continue with SoC ET as directed by inv...   \n",
       "463  Arm 1: Dato-DXd 6 mg/kg IV Q3W x 8 cycles + Du...   \n",
       "515  -Drug: LOXO-783 Oral Other Name: LY3849524 -Dr...   \n",
       "593  Participants who are continuing treatment with...   \n",
       "600  Radiation: 177Lu-PSMA-617 administered intrave...   \n",
       "629  The test using IGS-0001 will be performed twic...   \n",
       "691  - Experimental: Part 1: AMG 509 Intravenous (I...   \n",
       "698  TAR-200 Cohort 1: TAR-200 and Cetrelimab: TAR-...   \n",
       "729  For the target tumor, the dose settled at each...   \n",
       "\n",
       "                               TargetWord  \n",
       "157                       [Breast cancer]  \n",
       "170              [Advanced Breast Cancer]  \n",
       "215                       [Breast Cancer]  \n",
       "240                       [Breast Cancer]  \n",
       "307  [Breast Cancer, Early Breast Cancer]  \n",
       "342                     [prostate cancer]  \n",
       "351                              [Cancer]  \n",
       "388                    [Breast Neoplasms]  \n",
       "392  [Breast Cancer, Early Breast Cancer]  \n",
       "463                       [Breast Cancer]  \n",
       "515                       [Breast Cancer]  \n",
       "593                              [Cancer]  \n",
       "600                     [Prostate Cancer]  \n",
       "629                       [Breast cancer]  \n",
       "691                     [Prostate Cancer]  \n",
       "698                      [Bladder Cancer]  \n",
       "729                       [Breast cancer]  \n",
       "\n",
       "[17 rows x 40 columns]"
      ]
     },
     "execution_count": 15,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "matched_indices_d = (cosine_scores >= threshold).nonzero().tolist()\n",
    "# 入れ子リストをフラットなリストに変換\n",
    "flat_indices_d = [idx[0] for idx in matched_indices_d]\n",
    "\n",
    "# 抽出\n",
    "matched_df_d = basedf.iloc[flat_indices_d]\n",
    "matched_df_d\n"
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {},
   "source": []
  },
  {
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "\n"
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {},
   "source": []
  }
 ],
 "metadata": {
  "kernelspec": {
   "display_name": "gradio",
   "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.3"
  }
 },
 "nbformat": 4,
 "nbformat_minor": 2
}