Transformer.ipynb

{
  "nbformat": 4,
  "nbformat_minor": 0,
  "metadata": {
    "colab": {
      "provenance": []
    },
    "kernelspec": {
      "name": "python3",
      "display_name": "Python 3"
    },
    "language_info": {
      "name": "python"
    },
    "widgets": {
      "application/vnd.jupyter.widget-state+json": {
        "f12eff9a34644d2a8ab452262659d481": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "HBoxModel",
          "model_module_version": "1.5.0",
          "state": {
            "_dom_classes": [],
            "_model_module": "@jupyter-widgets/controls",
            "_model_module_version": "1.5.0",
            "_model_name": "HBoxModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/controls",
            "_view_module_version": "1.5.0",
            "_view_name": "HBoxView",
            "box_style": "",
            "children": [
              "IPY_MODEL_6ee8d9fcd9cf41fea26235196dac39f3",
              "IPY_MODEL_bd846b5317604bcb9f1533a2e9628e65",
              "IPY_MODEL_9db7af7d8ebc43c6bce66d9a5f58248d"
            ],
            "layout": "IPY_MODEL_a164d6185e124a51b7cb71f6ecf2ef16"
          }
        },
        "6ee8d9fcd9cf41fea26235196dac39f3": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "HTMLModel",
          "model_module_version": "1.5.0",
          "state": {
            "_dom_classes": [],
            "_model_module": "@jupyter-widgets/controls",
            "_model_module_version": "1.5.0",
            "_model_name": "HTMLModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/controls",
            "_view_module_version": "1.5.0",
            "_view_name": "HTMLView",
            "description": "",
            "description_tooltip": null,
            "layout": "IPY_MODEL_ab3f035970a24784b8fa7c4ea332766a",
            "placeholder": "​",
            "style": "IPY_MODEL_5d2a72fd7bb746a7808b3e3af04964b5",
            "value": "  0%"
          }
        },
        "bd846b5317604bcb9f1533a2e9628e65": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "FloatProgressModel",
          "model_module_version": "1.5.0",
          "state": {
            "_dom_classes": [],
            "_model_module": "@jupyter-widgets/controls",
            "_model_module_version": "1.5.0",
            "_model_name": "FloatProgressModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/controls",
            "_view_module_version": "1.5.0",
            "_view_name": "ProgressView",
            "bar_style": "danger",
            "description": "",
            "description_tooltip": null,
            "layout": "IPY_MODEL_31dae569e4af459fbb5bc8fc8cba8f1e",
            "max": 1563,
            "min": 0,
            "orientation": "horizontal",
            "style": "IPY_MODEL_3d8eb7942fc946cd9712e3b6d2621474",
            "value": 0
          }
        },
        "9db7af7d8ebc43c6bce66d9a5f58248d": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "HTMLModel",
          "model_module_version": "1.5.0",
          "state": {
            "_dom_classes": [],
            "_model_module": "@jupyter-widgets/controls",
            "_model_module_version": "1.5.0",
            "_model_name": "HTMLModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/controls",
            "_view_module_version": "1.5.0",
            "_view_name": "HTMLView",
            "description": "",
            "description_tooltip": null,
            "layout": "IPY_MODEL_4da8c9704f8043078c7905878e81c9e8",
            "placeholder": "​",
            "style": "IPY_MODEL_48435e59b6844a9f93f8297a55576d13",
            "value": " 0/1563 [00:07&lt;?, ?it/s]"
          }
        },
        "a164d6185e124a51b7cb71f6ecf2ef16": {
          "model_module": "@jupyter-widgets/base",
          "model_name": "LayoutModel",
          "model_module_version": "1.2.0",
          "state": {
            "_model_module": "@jupyter-widgets/base",
            "_model_module_version": "1.2.0",
            "_model_name": "LayoutModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/base",
            "_view_module_version": "1.2.0",
            "_view_name": "LayoutView",
            "align_content": null,
            "align_items": null,
            "align_self": null,
            "border": null,
            "bottom": null,
            "display": null,
            "flex": null,
            "flex_flow": null,
            "grid_area": null,
            "grid_auto_columns": null,
            "grid_auto_flow": null,
            "grid_auto_rows": null,
            "grid_column": null,
            "grid_gap": null,
            "grid_row": null,
            "grid_template_areas": null,
            "grid_template_columns": null,
            "grid_template_rows": null,
            "height": null,
            "justify_content": null,
            "justify_items": null,
            "left": null,
            "margin": null,
            "max_height": null,
            "max_width": null,
            "min_height": null,
            "min_width": null,
            "object_fit": null,
            "object_position": null,
            "order": null,
            "overflow": null,
            "overflow_x": null,
            "overflow_y": null,
            "padding": null,
            "right": null,
            "top": null,
            "visibility": null,
            "width": null
          }
        },
        "ab3f035970a24784b8fa7c4ea332766a": {
          "model_module": "@jupyter-widgets/base",
          "model_name": "LayoutModel",
          "model_module_version": "1.2.0",
          "state": {
            "_model_module": "@jupyter-widgets/base",
            "_model_module_version": "1.2.0",
            "_model_name": "LayoutModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/base",
            "_view_module_version": "1.2.0",
            "_view_name": "LayoutView",
            "align_content": null,
            "align_items": null,
            "align_self": null,
            "border": null,
            "bottom": null,
            "display": null,
            "flex": null,
            "flex_flow": null,
            "grid_area": null,
            "grid_auto_columns": null,
            "grid_auto_flow": null,
            "grid_auto_rows": null,
            "grid_column": null,
            "grid_gap": null,
            "grid_row": null,
            "grid_template_areas": null,
            "grid_template_columns": null,
            "grid_template_rows": null,
            "height": null,
            "justify_content": null,
            "justify_items": null,
            "left": null,
            "margin": null,
            "max_height": null,
            "max_width": null,
            "min_height": null,
            "min_width": null,
            "object_fit": null,
            "object_position": null,
            "order": null,
            "overflow": null,
            "overflow_x": null,
            "overflow_y": null,
            "padding": null,
            "right": null,
            "top": null,
            "visibility": null,
            "width": null
          }
        },
        "5d2a72fd7bb746a7808b3e3af04964b5": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "DescriptionStyleModel",
          "model_module_version": "1.5.0",
          "state": {
            "_model_module": "@jupyter-widgets/controls",
            "_model_module_version": "1.5.0",
            "_model_name": "DescriptionStyleModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/base",
            "_view_module_version": "1.2.0",
            "_view_name": "StyleView",
            "description_width": ""
          }
        },
        "31dae569e4af459fbb5bc8fc8cba8f1e": {
          "model_module": "@jupyter-widgets/base",
          "model_name": "LayoutModel",
          "model_module_version": "1.2.0",
          "state": {
            "_model_module": "@jupyter-widgets/base",
            "_model_module_version": "1.2.0",
            "_model_name": "LayoutModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/base",
            "_view_module_version": "1.2.0",
            "_view_name": "LayoutView",
            "align_content": null,
            "align_items": null,
            "align_self": null,
            "border": null,
            "bottom": null,
            "display": null,
            "flex": null,
            "flex_flow": null,
            "grid_area": null,
            "grid_auto_columns": null,
            "grid_auto_flow": null,
            "grid_auto_rows": null,
            "grid_column": null,
            "grid_gap": null,
            "grid_row": null,
            "grid_template_areas": null,
            "grid_template_columns": null,
            "grid_template_rows": null,
            "height": null,
            "justify_content": null,
            "justify_items": null,
            "left": null,
            "margin": null,
            "max_height": null,
            "max_width": null,
            "min_height": null,
            "min_width": null,
            "object_fit": null,
            "object_position": null,
            "order": null,
            "overflow": null,
            "overflow_x": null,
            "overflow_y": null,
            "padding": null,
            "right": null,
            "top": null,
            "visibility": null,
            "width": null
          }
        },
        "3d8eb7942fc946cd9712e3b6d2621474": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "ProgressStyleModel",
          "model_module_version": "1.5.0",
          "state": {
            "_model_module": "@jupyter-widgets/controls",
            "_model_module_version": "1.5.0",
            "_model_name": "ProgressStyleModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/base",
            "_view_module_version": "1.2.0",
            "_view_name": "StyleView",
            "bar_color": null,
            "description_width": ""
          }
        },
        "4da8c9704f8043078c7905878e81c9e8": {
          "model_module": "@jupyter-widgets/base",
          "model_name": "LayoutModel",
          "model_module_version": "1.2.0",
          "state": {
            "_model_module": "@jupyter-widgets/base",
            "_model_module_version": "1.2.0",
            "_model_name": "LayoutModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/base",
            "_view_module_version": "1.2.0",
            "_view_name": "LayoutView",
            "align_content": null,
            "align_items": null,
            "align_self": null,
            "border": null,
            "bottom": null,
            "display": null,
            "flex": null,
            "flex_flow": null,
            "grid_area": null,
            "grid_auto_columns": null,
            "grid_auto_flow": null,
            "grid_auto_rows": null,
            "grid_column": null,
            "grid_gap": null,
            "grid_row": null,
            "grid_template_areas": null,
            "grid_template_columns": null,
            "grid_template_rows": null,
            "height": null,
            "justify_content": null,
            "justify_items": null,
            "left": null,
            "margin": null,
            "max_height": null,
            "max_width": null,
            "min_height": null,
            "min_width": null,
            "object_fit": null,
            "object_position": null,
            "order": null,
            "overflow": null,
            "overflow_x": null,
            "overflow_y": null,
            "padding": null,
            "right": null,
            "top": null,
            "visibility": null,
            "width": null
          }
        },
        "48435e59b6844a9f93f8297a55576d13": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "DescriptionStyleModel",
          "model_module_version": "1.5.0",
          "state": {
            "_model_module": "@jupyter-widgets/controls",
            "_model_module_version": "1.5.0",
            "_model_name": "DescriptionStyleModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/base",
            "_view_module_version": "1.2.0",
            "_view_name": "StyleView",
            "description_width": ""
          }
        }
      }
    }
  },
  "cells": [
    {
      "cell_type": "code",
      "execution_count": 2,
      "metadata": {
        "colab": {
          "base_uri": "https://localhost:8080/"
        },
        "id": "Vok7QiQ-Gr3Z",
        "outputId": "649a87e5-1148-46c0-df79-5ed3037d712d"
      },
      "outputs": [
        {
          "output_type": "stream",
          "name": "stdout",
          "text": [
            "Requirement already satisfied: transformers in /usr/local/lib/python3.10/dist-packages (4.38.2)\n",
            "Requirement already satisfied: filelock in /usr/local/lib/python3.10/dist-packages (from transformers) (3.13.3)\n",
            "Requirement already satisfied: huggingface-hub<1.0,>=0.19.3 in /usr/local/lib/python3.10/dist-packages (from transformers) (0.20.3)\n",
            "Requirement already satisfied: numpy>=1.17 in /usr/local/lib/python3.10/dist-packages (from transformers) (1.25.2)\n",
            "Requirement already satisfied: packaging>=20.0 in /usr/local/lib/python3.10/dist-packages (from transformers) (24.0)\n",
            "Requirement already satisfied: pyyaml>=5.1 in /usr/local/lib/python3.10/dist-packages (from transformers) (6.0.1)\n",
            "Requirement already satisfied: regex!=2019.12.17 in /usr/local/lib/python3.10/dist-packages (from transformers) (2023.12.25)\n",
            "Requirement already satisfied: requests in /usr/local/lib/python3.10/dist-packages (from transformers) (2.31.0)\n",
            "Requirement already satisfied: tokenizers<0.19,>=0.14 in /usr/local/lib/python3.10/dist-packages (from transformers) (0.15.2)\n",
            "Requirement already satisfied: safetensors>=0.4.1 in /usr/local/lib/python3.10/dist-packages (from transformers) (0.4.2)\n",
            "Requirement already satisfied: tqdm>=4.27 in /usr/local/lib/python3.10/dist-packages (from transformers) (4.66.2)\n",
            "Requirement already satisfied: fsspec>=2023.5.0 in /usr/local/lib/python3.10/dist-packages (from huggingface-hub<1.0,>=0.19.3->transformers) (2023.6.0)\n",
            "Requirement already satisfied: typing-extensions>=3.7.4.3 in /usr/local/lib/python3.10/dist-packages (from huggingface-hub<1.0,>=0.19.3->transformers) (4.10.0)\n",
            "Requirement already satisfied: charset-normalizer<4,>=2 in /usr/local/lib/python3.10/dist-packages (from requests->transformers) (3.3.2)\n",
            "Requirement already satisfied: idna<4,>=2.5 in /usr/local/lib/python3.10/dist-packages (from requests->transformers) (3.6)\n",
            "Requirement already satisfied: urllib3<3,>=1.21.1 in /usr/local/lib/python3.10/dist-packages (from requests->transformers) (2.0.7)\n",
            "Requirement already satisfied: certifi>=2017.4.17 in /usr/local/lib/python3.10/dist-packages (from requests->transformers) (2024.2.2)\n",
            "Requirement already satisfied: torchmetrics in /usr/local/lib/python3.10/dist-packages (1.3.2)\n",
            "Requirement already satisfied: numpy>1.20.0 in /usr/local/lib/python3.10/dist-packages (from torchmetrics) (1.25.2)\n",
            "Requirement already satisfied: packaging>17.1 in /usr/local/lib/python3.10/dist-packages (from torchmetrics) (24.0)\n",
            "Requirement already satisfied: torch>=1.10.0 in /usr/local/lib/python3.10/dist-packages (from torchmetrics) (2.2.1+cu121)\n",
            "Requirement already satisfied: lightning-utilities>=0.8.0 in /usr/local/lib/python3.10/dist-packages (from torchmetrics) (0.11.2)\n",
            "Requirement already satisfied: setuptools in /usr/local/lib/python3.10/dist-packages (from lightning-utilities>=0.8.0->torchmetrics) (67.7.2)\n",
            "Requirement already satisfied: typing-extensions in /usr/local/lib/python3.10/dist-packages (from lightning-utilities>=0.8.0->torchmetrics) (4.10.0)\n",
            "Requirement already satisfied: filelock in /usr/local/lib/python3.10/dist-packages (from torch>=1.10.0->torchmetrics) (3.13.3)\n",
            "Requirement already satisfied: sympy in /usr/local/lib/python3.10/dist-packages (from torch>=1.10.0->torchmetrics) (1.12)\n",
            "Requirement already satisfied: networkx in /usr/local/lib/python3.10/dist-packages (from torch>=1.10.0->torchmetrics) (3.2.1)\n",
            "Requirement already satisfied: jinja2 in /usr/local/lib/python3.10/dist-packages (from torch>=1.10.0->torchmetrics) (3.1.3)\n",
            "Requirement already satisfied: fsspec in /usr/local/lib/python3.10/dist-packages (from torch>=1.10.0->torchmetrics) (2023.6.0)\n",
            "Requirement already satisfied: nvidia-cuda-nvrtc-cu12==12.1.105 in /usr/local/lib/python3.10/dist-packages (from torch>=1.10.0->torchmetrics) (12.1.105)\n",
            "Requirement already satisfied: nvidia-cuda-runtime-cu12==12.1.105 in /usr/local/lib/python3.10/dist-packages (from torch>=1.10.0->torchmetrics) (12.1.105)\n",
            "Requirement already satisfied: nvidia-cuda-cupti-cu12==12.1.105 in /usr/local/lib/python3.10/dist-packages (from torch>=1.10.0->torchmetrics) (12.1.105)\n",
            "Requirement already satisfied: nvidia-cudnn-cu12==8.9.2.26 in /usr/local/lib/python3.10/dist-packages (from torch>=1.10.0->torchmetrics) (8.9.2.26)\n",
            "Requirement already satisfied: nvidia-cublas-cu12==12.1.3.1 in /usr/local/lib/python3.10/dist-packages (from torch>=1.10.0->torchmetrics) (12.1.3.1)\n",
            "Requirement already satisfied: nvidia-cufft-cu12==11.0.2.54 in /usr/local/lib/python3.10/dist-packages (from torch>=1.10.0->torchmetrics) (11.0.2.54)\n",
            "Requirement already satisfied: nvidia-curand-cu12==10.3.2.106 in /usr/local/lib/python3.10/dist-packages (from torch>=1.10.0->torchmetrics) (10.3.2.106)\n",
            "Requirement already satisfied: nvidia-cusolver-cu12==11.4.5.107 in /usr/local/lib/python3.10/dist-packages (from torch>=1.10.0->torchmetrics) (11.4.5.107)\n",
            "Requirement already satisfied: nvidia-cusparse-cu12==12.1.0.106 in /usr/local/lib/python3.10/dist-packages (from torch>=1.10.0->torchmetrics) (12.1.0.106)\n",
            "Requirement already satisfied: nvidia-nccl-cu12==2.19.3 in /usr/local/lib/python3.10/dist-packages (from torch>=1.10.0->torchmetrics) (2.19.3)\n",
            "Requirement already satisfied: nvidia-nvtx-cu12==12.1.105 in /usr/local/lib/python3.10/dist-packages (from torch>=1.10.0->torchmetrics) (12.1.105)\n",
            "Requirement already satisfied: triton==2.2.0 in /usr/local/lib/python3.10/dist-packages (from torch>=1.10.0->torchmetrics) (2.2.0)\n",
            "Requirement already satisfied: nvidia-nvjitlink-cu12 in /usr/local/lib/python3.10/dist-packages (from nvidia-cusolver-cu12==11.4.5.107->torch>=1.10.0->torchmetrics) (12.4.99)\n",
            "Requirement already satisfied: MarkupSafe>=2.0 in /usr/local/lib/python3.10/dist-packages (from jinja2->torch>=1.10.0->torchmetrics) (2.1.5)\n",
            "Requirement already satisfied: mpmath>=0.19 in /usr/local/lib/python3.10/dist-packages (from sympy->torch>=1.10.0->torchmetrics) (1.3.0)\n",
            "Requirement already satisfied: evaluate in /usr/local/lib/python3.10/dist-packages (0.4.1)\n",
            "Requirement already satisfied: datasets>=2.0.0 in /usr/local/lib/python3.10/dist-packages (from evaluate) (2.18.0)\n",
            "Requirement already satisfied: numpy>=1.17 in /usr/local/lib/python3.10/dist-packages (from evaluate) (1.25.2)\n",
            "Requirement already satisfied: dill in /usr/local/lib/python3.10/dist-packages (from evaluate) (0.3.8)\n",
            "Requirement already satisfied: pandas in /usr/local/lib/python3.10/dist-packages (from evaluate) (1.5.3)\n",
            "Requirement already satisfied: requests>=2.19.0 in /usr/local/lib/python3.10/dist-packages (from evaluate) (2.31.0)\n",
            "Requirement already satisfied: tqdm>=4.62.1 in /usr/local/lib/python3.10/dist-packages (from evaluate) (4.66.2)\n",
            "Requirement already satisfied: xxhash in /usr/local/lib/python3.10/dist-packages (from evaluate) (3.4.1)\n",
            "Requirement already satisfied: multiprocess in /usr/local/lib/python3.10/dist-packages (from evaluate) (0.70.16)\n",
            "Requirement already satisfied: fsspec[http]>=2021.05.0 in /usr/local/lib/python3.10/dist-packages (from evaluate) (2023.6.0)\n",
            "Requirement already satisfied: huggingface-hub>=0.7.0 in /usr/local/lib/python3.10/dist-packages (from evaluate) (0.20.3)\n",
            "Requirement already satisfied: packaging in /usr/local/lib/python3.10/dist-packages (from evaluate) (24.0)\n",
            "Requirement already satisfied: responses<0.19 in /usr/local/lib/python3.10/dist-packages (from evaluate) (0.18.0)\n",
            "Requirement already satisfied: filelock in /usr/local/lib/python3.10/dist-packages (from datasets>=2.0.0->evaluate) (3.13.3)\n",
            "Requirement already satisfied: pyarrow>=12.0.0 in /usr/local/lib/python3.10/dist-packages (from datasets>=2.0.0->evaluate) (14.0.2)\n",
            "Requirement already satisfied: pyarrow-hotfix in /usr/local/lib/python3.10/dist-packages (from datasets>=2.0.0->evaluate) (0.6)\n",
            "Requirement already satisfied: aiohttp in /usr/local/lib/python3.10/dist-packages (from datasets>=2.0.0->evaluate) (3.9.3)\n",
            "Requirement already satisfied: pyyaml>=5.1 in /usr/local/lib/python3.10/dist-packages (from datasets>=2.0.0->evaluate) (6.0.1)\n",
            "Requirement already satisfied: typing-extensions>=3.7.4.3 in /usr/local/lib/python3.10/dist-packages (from huggingface-hub>=0.7.0->evaluate) (4.10.0)\n",
            "Requirement already satisfied: charset-normalizer<4,>=2 in /usr/local/lib/python3.10/dist-packages (from requests>=2.19.0->evaluate) (3.3.2)\n",
            "Requirement already satisfied: idna<4,>=2.5 in /usr/local/lib/python3.10/dist-packages (from requests>=2.19.0->evaluate) (3.6)\n",
            "Requirement already satisfied: urllib3<3,>=1.21.1 in /usr/local/lib/python3.10/dist-packages (from requests>=2.19.0->evaluate) (2.0.7)\n",
            "Requirement already satisfied: certifi>=2017.4.17 in /usr/local/lib/python3.10/dist-packages (from requests>=2.19.0->evaluate) (2024.2.2)\n",
            "Requirement already satisfied: python-dateutil>=2.8.1 in /usr/local/lib/python3.10/dist-packages (from pandas->evaluate) (2.8.2)\n",
            "Requirement already satisfied: pytz>=2020.1 in /usr/local/lib/python3.10/dist-packages (from pandas->evaluate) (2023.4)\n",
            "Requirement already satisfied: aiosignal>=1.1.2 in /usr/local/lib/python3.10/dist-packages (from aiohttp->datasets>=2.0.0->evaluate) (1.3.1)\n",
            "Requirement already satisfied: attrs>=17.3.0 in /usr/local/lib/python3.10/dist-packages (from aiohttp->datasets>=2.0.0->evaluate) (23.2.0)\n",
            "Requirement already satisfied: frozenlist>=1.1.1 in /usr/local/lib/python3.10/dist-packages (from aiohttp->datasets>=2.0.0->evaluate) (1.4.1)\n",
            "Requirement already satisfied: multidict<7.0,>=4.5 in /usr/local/lib/python3.10/dist-packages (from aiohttp->datasets>=2.0.0->evaluate) (6.0.5)\n",
            "Requirement already satisfied: yarl<2.0,>=1.0 in /usr/local/lib/python3.10/dist-packages (from aiohttp->datasets>=2.0.0->evaluate) (1.9.4)\n",
            "Requirement already satisfied: async-timeout<5.0,>=4.0 in /usr/local/lib/python3.10/dist-packages (from aiohttp->datasets>=2.0.0->evaluate) (4.0.3)\n",
            "Requirement already satisfied: six>=1.5 in /usr/local/lib/python3.10/dist-packages (from python-dateutil>=2.8.1->pandas->evaluate) (1.16.0)\n"
          ]
        }
      ],
      "source": [
        "!pip install transformers\n",
        "!pip install torchmetrics\n",
        "!pip install evaluate"
      ]
    },
    {
      "cell_type": "markdown",
      "source": [
        "##Load up the libraries"
      ],
      "metadata": {
        "id": "yqtZPxvDG0a3"
      }
    },
    {
      "cell_type": "code",
      "source": [
        "import numpy as np\n",
        "from transformers import AutoTokenizer, T5Tokenizer, T5ForConditionalGeneration\n",
        "import torch\n",
        "\n",
        "device = torch.device(\"cuda\" if torch.cuda.is_available() else \"cpu\")"
      ],
      "metadata": {
        "id": "NT6l1jd5Gxeo"
      },
      "execution_count": 41,
      "outputs": []
    },
    {
      "cell_type": "markdown",
      "source": [
        "## Load up the tokenizer and the model"
      ],
      "metadata": {
        "id": "8iKSb5wmR0ae"
      }
    },
    {
      "cell_type": "code",
      "source": [
        "tokenizer = AutoTokenizer.from_pretrained(\"google/t5-efficient-tiny\")\n",
        "model = T5ForConditionalGeneration.from_pretrained(\"google/t5-efficient-tiny\")"
      ],
      "metadata": {
        "id": "ptZOqReuLPrS"
      },
      "execution_count": 32,
      "outputs": []
    },
    {
      "cell_type": "markdown",
      "source": [
        "## A simple sentence"
      ],
      "metadata": {
        "id": "G4nAXMa8SBM2"
      }
    },
    {
      "cell_type": "code",
      "source": [
        "input_sequence = \"Today's class is about Transformers for the Humanities and Social Sciences Department\""
      ],
      "metadata": {
        "id": "gTUrRpT8LVEQ"
      },
      "execution_count": 39,
      "outputs": []
    },
    {
      "cell_type": "markdown",
      "source": [
        "## Tokenize the sentence and convert to IDs"
      ],
      "metadata": {
        "id": "Sx80YlUNSKAs"
      }
    },
    {
      "cell_type": "code",
      "source": [
        "encoding = tokenizer(\n",
        "    [input_sequence],\n",
        "    padding=\"longest\",\n",
        "    max_length=512,\n",
        "    truncation=True,\n",
        "    return_tensors=\"pt\",\n",
        ")\n",
        "print(encoding, \"\\n\\n\\n\")\n",
        "print(tokenizer.convert_ids_to_tokens(encoding.input_ids[0]))\n",
        "input_ids, attention_mask = encoding.input_ids, encoding.attention_mask"
      ],
      "metadata": {
        "colab": {
          "base_uri": "https://localhost:8080/"
        },
        "id": "NPCpAPOPLf9M",
        "outputId": "b83a5ae8-58d3-48cb-bf9e-67d9831a8a32"
      },
      "execution_count": 40,
      "outputs": [
        {
          "output_type": "stream",
          "name": "stdout",
          "text": [
            "{'input_ids': tensor([[ 1960,    31,     7,   853,    19,    81, 31220,     7,    21,     8,\n",
            "          3892,  2197,    11,  2730,  9226,  1775,     1]]), 'attention_mask': tensor([[1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1]])} \n",
            "\n",
            "\n",
            "\n",
            "['▁Today', \"'\", 's', '▁class', '▁is', '▁about', '▁Transformer', 's', '▁for', '▁the', '▁Human', 'ities', '▁and', '▁Social', '▁Sciences', '▁Department', '</s>']\n"
          ]
        }
      ]
    },
    {
      "cell_type": "code",
      "source": [
        "# outputs = model.generate(input_ids, max_length=50)\n",
        "# print(tokenizer.decode(outputs[0], skip_special_tokens=True))"
      ],
      "metadata": {
        "colab": {
          "base_uri": "https://localhost:8080/"
        },
        "id": "2-xOH7LYORDE",
        "outputId": "204f0ffd-7ce0-43dd-821d-d0bcd59f3906"
      },
      "execution_count": 8,
      "outputs": [
        {
          "output_type": "stream",
          "name": "stdout",
          "text": [
            "spectra of samples from 7 plant species were used to explore the influence of preprocessing and feature extraction on efficiency of machine learning algorithms. Wavelet Tensor Train ( WTT ) and Discrete Wavelet Transform\n"
          ]
        }
      ]
    },
    {
      "cell_type": "markdown",
      "source": [
        "## Load up the libraries"
      ],
      "metadata": {
        "id": "pYtNVXqwirlY"
      }
    },
    {
      "cell_type": "code",
      "source": [
        "import pandas as pd\n",
        "import numpy as np\n",
        "import seaborn as sns\n",
        "import re\n",
        "import copy\n",
        "from tqdm.notebook import tqdm\n",
        "import torch\n",
        "import torch.nn as nn\n",
        "import torch.nn.functional as F\n",
        "from torch import optim\n",
        "from torch.utils.data import Dataset, DataLoader\n",
        "\n",
        "from sklearn.metrics import accuracy_score, f1_score, classification_report\n",
        "\n",
        "from transformers import AutoTokenizer, T5Model, T5ForConditionalGeneration\n",
        "from datasets import load_dataset"
      ],
      "metadata": {
        "id": "ynJuV5bWVfqx"
      },
      "execution_count": 42,
      "outputs": []
    },
    {
      "cell_type": "markdown",
      "source": [
        "## A function to clean text"
      ],
      "metadata": {
        "id": "8a0AeAqGitnX"
      }
    },
    {
      "cell_type": "code",
      "source": [
        "def preprocess(x):\n",
        "    x = re.sub('<.*?>', ' ', x)\n",
        "    x = re.sub('http\\S+', ' ', x)\n",
        "    x = re.sub('\\s+', ' ', x)\n",
        "    return x.lower().strip()"
      ],
      "metadata": {
        "id": "fNKdtsLfVt-4"
      },
      "execution_count": 10,
      "outputs": []
    },
    {
      "cell_type": "markdown",
      "source": [
        "## A pre-processing function"
      ],
      "metadata": {
        "id": "YoC4U_Meixcn"
      }
    },
    {
      "cell_type": "code",
      "source": [
        "def pipeline(dataframe):\n",
        "    # Pre-process the sentences\n",
        "    dataframe['text'] = dataframe['text'].apply(lambda x: preprocess(x))\n",
        "\n",
        "    # Pre-pend CLS token to each sentence\n",
        "    sentences = [\"[CLS] \" + s for s in dataframe.text.values]\n",
        "\n",
        "    # Extract labels\n",
        "    labels = dataframe.label.values\n",
        "\n",
        "    # Tokenize each sentence\n",
        "    tokenizer = AutoTokenizer.from_pretrained(\"bert-base-uncased\", do_lower_case=True)\n",
        "    tokenized = [tokenizer.tokenize(s) for s in tqdm(sentences)]\n",
        "\n",
        "    # Append the SEP token and also set a threshold for the number of tokens in a sentence\n",
        "    MAX_LEN_TRAIN, MAX_LEN_TEST = 140, 140\n",
        "    tokenized = [t[:(MAX_LEN_TRAIN-1)]+['SEP'] for t in tokenized]\n",
        "\n",
        "    # Generate IDs of each token and add padding to sentences smaller than given threshold\n",
        "    ids = [tokenizer.convert_tokens_to_ids(t) for t in tqdm(tokenized)]\n",
        "    ids = np.array([np.pad(i, (0, MAX_LEN_TRAIN-len(i)), mode='constant') for i in ids])\n",
        "\n",
        "    # Also generate Attention masks. An attention mask is a binary tensor\n",
        "    # that indicates the position of padded indices so that the model does not attend to them\n",
        "    amasks = np.asarray([[float(i>0) for i in seq] for seq in tqdm(ids)])\n",
        "\n",
        "    return torch.tensor(ids), torch.tensor(labels), torch.tensor(amasks)"
      ],
      "metadata": {
        "id": "Xxy5SN-gVyBe"
      },
      "execution_count": 11,
      "outputs": []
    },
    {
      "cell_type": "markdown",
      "source": [
        "## Load the IMDB Dataset in pandas form"
      ],
      "metadata": {
        "id": "ZfECDYDii5dZ"
      }
    },
    {
      "cell_type": "code",
      "source": [
        "dataset = load_dataset('imdb')\n",
        "df_train = dataset['train'].to_pandas()\n",
        "display(df_train.head())\n",
        "\n",
        "df_val = dataset['test'].to_pandas()\n",
        "display(df_val.head())"
      ],
      "metadata": {
        "colab": {
          "base_uri": "https://localhost:8080/",
          "height": 395
        },
        "id": "oS3yXUzLVz72",
        "outputId": "f331f39b-3a39-41a1-cbce-55279a1fe712"
      },
      "execution_count": 12,
      "outputs": [
        {
          "output_type": "display_data",
          "data": {
            "text/plain": [
              "                                                text  label\n",
              "0  I rented I AM CURIOUS-YELLOW from my video sto...      0\n",
              "1  \"I Am Curious: Yellow\" is a risible and preten...      0\n",
              "2  If only to avoid making this type of film in t...      0\n",
              "3  This film was probably inspired by Godard's Ma...      0\n",
              "4  Oh, brother...after hearing about this ridicul...      0"
            ],
            "text/html": [
              "\n",
              "  <div id=\"df-068b2741-51f5-447e-82ce-ebedf73519c7\" class=\"colab-df-container\">\n",
              "    <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>text</th>\n",
              "      <th>label</th>\n",
              "    </tr>\n",
              "  </thead>\n",
              "  <tbody>\n",
              "    <tr>\n",
              "      <th>0</th>\n",
              "      <td>I rented I AM CURIOUS-YELLOW from my video sto...</td>\n",
              "      <td>0</td>\n",
              "    </tr>\n",
              "    <tr>\n",
              "      <th>1</th>\n",
              "      <td>\"I Am Curious: Yellow\" is a risible and preten...</td>\n",
              "      <td>0</td>\n",
              "    </tr>\n",
              "    <tr>\n",
              "      <th>2</th>\n",
              "      <td>If only to avoid making this type of film in t...</td>\n",
              "      <td>0</td>\n",
              "    </tr>\n",
              "    <tr>\n",
              "      <th>3</th>\n",
              "      <td>This film was probably inspired by Godard's Ma...</td>\n",
              "      <td>0</td>\n",
              "    </tr>\n",
              "    <tr>\n",
              "      <th>4</th>\n",
              "      <td>Oh, brother...after hearing about this ridicul...</td>\n",
              "      <td>0</td>\n",
              "    </tr>\n",
              "  </tbody>\n",
              "</table>\n",
              "</div>\n",
              "    <div class=\"colab-df-buttons\">\n",
              "\n",
              "  <div class=\"colab-df-container\">\n",
              "    <button class=\"colab-df-convert\" onclick=\"convertToInteractive('df-068b2741-51f5-447e-82ce-ebedf73519c7')\"\n",
              "            title=\"Convert this dataframe to an interactive table.\"\n",
              "            style=\"display:none;\">\n",
              "\n",
              "  <svg xmlns=\"http://www.w3.org/2000/svg\" height=\"24px\" viewBox=\"0 -960 960 960\">\n",
              "    <path d=\"M120-120v-720h720v720H120Zm60-500h600v-160H180v160Zm220 220h160v-160H400v160Zm0 220h160v-160H400v160ZM180-400h160v-160H180v160Zm440 0h160v-160H620v160ZM180-180h160v-160H180v160Zm440 0h160v-160H620v160Z\"/>\n",
              "  </svg>\n",
              "    </button>\n",
              "\n",
              "  <style>\n",
              "    .colab-df-container {\n",
              "      display:flex;\n",
              "      gap: 12px;\n",
              "    }\n",
              "\n",
              "    .colab-df-convert {\n",
              "      background-color: #E8F0FE;\n",
              "      border: none;\n",
              "      border-radius: 50%;\n",
              "      cursor: pointer;\n",
              "      display: none;\n",
              "      fill: #1967D2;\n",
              "      height: 32px;\n",
              "      padding: 0 0 0 0;\n",
              "      width: 32px;\n",
              "    }\n",
              "\n",
              "    .colab-df-convert:hover {\n",
              "      background-color: #E2EBFA;\n",
              "      box-shadow: 0px 1px 2px rgba(60, 64, 67, 0.3), 0px 1px 3px 1px rgba(60, 64, 67, 0.15);\n",
              "      fill: #174EA6;\n",
              "    }\n",
              "\n",
              "    .colab-df-buttons div {\n",
              "      margin-bottom: 4px;\n",
              "    }\n",
              "\n",
              "    [theme=dark] .colab-df-convert {\n",
              "      background-color: #3B4455;\n",
              "      fill: #D2E3FC;\n",
              "    }\n",
              "\n",
              "    [theme=dark] .colab-df-convert:hover {\n",
              "      background-color: #434B5C;\n",
              "      box-shadow: 0px 1px 3px 1px rgba(0, 0, 0, 0.15);\n",
              "      filter: drop-shadow(0px 1px 2px rgba(0, 0, 0, 0.3));\n",
              "      fill: #FFFFFF;\n",
              "    }\n",
              "  </style>\n",
              "\n",
              "    <script>\n",
              "      const buttonEl =\n",
              "        document.querySelector('#df-068b2741-51f5-447e-82ce-ebedf73519c7 button.colab-df-convert');\n",
              "      buttonEl.style.display =\n",
              "        google.colab.kernel.accessAllowed ? 'block' : 'none';\n",
              "\n",
              "      async function convertToInteractive(key) {\n",
              "        const element = document.querySelector('#df-068b2741-51f5-447e-82ce-ebedf73519c7');\n",
              "        const dataTable =\n",
              "          await google.colab.kernel.invokeFunction('convertToInteractive',\n",
              "                                                    [key], {});\n",
              "        if (!dataTable) return;\n",
              "\n",
              "        const docLinkHtml = 'Like what you see? Visit the ' +\n",
              "          '<a target=\"_blank\" href=https://colab.research.google.com/notebooks/data_table.ipynb>data table notebook</a>'\n",
              "          + ' to learn more about interactive tables.';\n",
              "        element.innerHTML = '';\n",
              "        dataTable['output_type'] = 'display_data';\n",
              "        await google.colab.output.renderOutput(dataTable, element);\n",
              "        const docLink = document.createElement('div');\n",
              "        docLink.innerHTML = docLinkHtml;\n",
              "        element.appendChild(docLink);\n",
              "      }\n",
              "    </script>\n",
              "  </div>\n",
              "\n",
              "\n",
              "<div id=\"df-1699d9e8-f42a-42e7-814d-b533a93f62fc\">\n",
              "  <button class=\"colab-df-quickchart\" onclick=\"quickchart('df-1699d9e8-f42a-42e7-814d-b533a93f62fc')\"\n",
              "            title=\"Suggest charts\"\n",
              "            style=\"display:none;\">\n",
              "\n",
              "<svg xmlns=\"http://www.w3.org/2000/svg\" height=\"24px\"viewBox=\"0 0 24 24\"\n",
              "     width=\"24px\">\n",
              "    <g>\n",
              "        <path d=\"M19 3H5c-1.1 0-2 .9-2 2v14c0 1.1.9 2 2 2h14c1.1 0 2-.9 2-2V5c0-1.1-.9-2-2-2zM9 17H7v-7h2v7zm4 0h-2V7h2v10zm4 0h-2v-4h2v4z\"/>\n",
              "    </g>\n",
              "</svg>\n",
              "  </button>\n",
              "\n",
              "<style>\n",
              "  .colab-df-quickchart {\n",
              "      --bg-color: #E8F0FE;\n",
              "      --fill-color: #1967D2;\n",
              "      --hover-bg-color: #E2EBFA;\n",
              "      --hover-fill-color: #174EA6;\n",
              "      --disabled-fill-color: #AAA;\n",
              "      --disabled-bg-color: #DDD;\n",
              "  }\n",
              "\n",
              "  [theme=dark] .colab-df-quickchart {\n",
              "      --bg-color: #3B4455;\n",
              "      --fill-color: #D2E3FC;\n",
              "      --hover-bg-color: #434B5C;\n",
              "      --hover-fill-color: #FFFFFF;\n",
              "      --disabled-bg-color: #3B4455;\n",
              "      --disabled-fill-color: #666;\n",
              "  }\n",
              "\n",
              "  .colab-df-quickchart {\n",
              "    background-color: var(--bg-color);\n",
              "    border: none;\n",
              "    border-radius: 50%;\n",
              "    cursor: pointer;\n",
              "    display: none;\n",
              "    fill: var(--fill-color);\n",
              "    height: 32px;\n",
              "    padding: 0;\n",
              "    width: 32px;\n",
              "  }\n",
              "\n",
              "  .colab-df-quickchart:hover {\n",
              "    background-color: var(--hover-bg-color);\n",
              "    box-shadow: 0 1px 2px rgba(60, 64, 67, 0.3), 0 1px 3px 1px rgba(60, 64, 67, 0.15);\n",
              "    fill: var(--button-hover-fill-color);\n",
              "  }\n",
              "\n",
              "  .colab-df-quickchart-complete:disabled,\n",
              "  .colab-df-quickchart-complete:disabled:hover {\n",
              "    background-color: var(--disabled-bg-color);\n",
              "    fill: var(--disabled-fill-color);\n",
              "    box-shadow: none;\n",
              "  }\n",
              "\n",
              "  .colab-df-spinner {\n",
              "    border: 2px solid var(--fill-color);\n",
              "    border-color: transparent;\n",
              "    border-bottom-color: var(--fill-color);\n",
              "    animation:\n",
              "      spin 1s steps(1) infinite;\n",
              "  }\n",
              "\n",
              "  @keyframes spin {\n",
              "    0% {\n",
              "      border-color: transparent;\n",
              "      border-bottom-color: var(--fill-color);\n",
              "      border-left-color: var(--fill-color);\n",
              "    }\n",
              "    20% {\n",
              "      border-color: transparent;\n",
              "      border-left-color: var(--fill-color);\n",
              "      border-top-color: var(--fill-color);\n",
              "    }\n",
              "    30% {\n",
              "      border-color: transparent;\n",
              "      border-left-color: var(--fill-color);\n",
              "      border-top-color: var(--fill-color);\n",
              "      border-right-color: var(--fill-color);\n",
              "    }\n",
              "    40% {\n",
              "      border-color: transparent;\n",
              "      border-right-color: var(--fill-color);\n",
              "      border-top-color: var(--fill-color);\n",
              "    }\n",
              "    60% {\n",
              "      border-color: transparent;\n",
              "      border-right-color: var(--fill-color);\n",
              "    }\n",
              "    80% {\n",
              "      border-color: transparent;\n",
              "      border-right-color: var(--fill-color);\n",
              "      border-bottom-color: var(--fill-color);\n",
              "    }\n",
              "    90% {\n",
              "      border-color: transparent;\n",
              "      border-bottom-color: var(--fill-color);\n",
              "    }\n",
              "  }\n",
              "</style>\n",
              "\n",
              "  <script>\n",
              "    async function quickchart(key) {\n",
              "      const quickchartButtonEl =\n",
              "        document.querySelector('#' + key + ' button');\n",
              "      quickchartButtonEl.disabled = true;  // To prevent multiple clicks.\n",
              "      quickchartButtonEl.classList.add('colab-df-spinner');\n",
              "      try {\n",
              "        const charts = await google.colab.kernel.invokeFunction(\n",
              "            'suggestCharts', [key], {});\n",
              "      } catch (error) {\n",
              "        console.error('Error during call to suggestCharts:', error);\n",
              "      }\n",
              "      quickchartButtonEl.classList.remove('colab-df-spinner');\n",
              "      quickchartButtonEl.classList.add('colab-df-quickchart-complete');\n",
              "    }\n",
              "    (() => {\n",
              "      let quickchartButtonEl =\n",
              "        document.querySelector('#df-1699d9e8-f42a-42e7-814d-b533a93f62fc button');\n",
              "      quickchartButtonEl.style.display =\n",
              "        google.colab.kernel.accessAllowed ? 'block' : 'none';\n",
              "    })();\n",
              "  </script>\n",
              "</div>\n",
              "\n",
              "    </div>\n",
              "  </div>\n"
            ],
            "application/vnd.google.colaboratory.intrinsic+json": {
              "type": "dataframe",
              "summary": "{\n  \"name\": \"display(df_val\",\n  \"rows\": 5,\n  \"fields\": [\n    {\n      \"column\": \"text\",\n      \"properties\": {\n        \"dtype\": \"string\",\n        \"num_unique_values\": 5,\n        \"samples\": [\n          \"\\\"I Am Curious: Yellow\\\" is a risible and pretentious steaming pile. It doesn't matter what one's political views are because this film can hardly be taken seriously on any level. As for the claim that frontal male nudity is an automatic NC-17, that isn't true. I've seen R-rated films with male nudity. Granted, they only offer some fleeting views, but where are the R-rated films with gaping vulvas and flapping labia? Nowhere, because they don't exist. The same goes for those crappy cable shows: schlongs swinging in the breeze but not a clitoris in sight. And those pretentious indie movies like The Brown Bunny, in which we're treated to the site of Vincent Gallo's throbbing johnson, but not a trace of pink visible on Chloe Sevigny. Before crying (or implying) \\\"double-standard\\\" in matters of nudity, the mentally obtuse should take into account one unavoidably obvious anatomical difference between men and women: there are no genitals on display when actresses appears nude, and the same cannot be said for a man. In fact, you generally won't see female genitals in an American film in anything short of porn or explicit erotica. This alleged double-standard is less a double standard than an admittedly depressing ability to come to terms culturally with the insides of women's bodies.\",\n          \"Oh, brother...after hearing about this ridiculous film for umpteen years all I can think of is that old Peggy Lee song..<br /><br />\\\"Is that all there is??\\\" ...I was just an early teen when this smoked fish hit the U.S. I was too young to get in the theater (although I did manage to sneak into \\\"Goodbye Columbus\\\"). Then a screening at a local film museum beckoned - Finally I could see this film, except now I was as old as my parents were when they schlepped to see it!!<br /><br />The ONLY reason this film was not condemned to the anonymous sands of time was because of the obscenity case sparked by its U.S. release. MILLIONS of people flocked to this stinker, thinking they were going to see a sex film...Instead, they got lots of closeups of gnarly, repulsive Swedes, on-street interviews in bland shopping malls, asinie political pretension...and feeble who-cares simulated sex scenes with saggy, pale actors.<br /><br />Cultural icon, holy grail, historic artifact..whatever this thing was, shred it, burn it, then stuff the ashes in a lead box!<br /><br />Elite esthetes still scrape to find value in its boring pseudo revolutionary political spewings..But if it weren't for the censorship scandal, it would have been ignored, then forgotten.<br /><br />Instead, the \\\"I Am Blank, Blank\\\" rhythymed title was repeated endlessly for years as a titilation for porno films (I am Curious, Lavender - for gay films, I Am Curious, Black - for blaxploitation films, etc..) and every ten years or so the thing rises from the dead, to be viewed by a new generation of suckers who want to see that \\\"naughty sex film\\\" that \\\"revolutionized the film industry\\\"...<br /><br />Yeesh, avoid like the plague..Or if you MUST see it - rent the video and fast forward to the \\\"dirty\\\" parts, just to get it over with.<br /><br />\",\n          \"If only to avoid making this type of film in the future. This film is interesting as an experiment but tells no cogent story.<br /><br />One might feel virtuous for sitting thru it because it touches on so many IMPORTANT issues but it does so without any discernable motive. The viewer comes away with no new perspectives (unless one comes up with one while one's mind wanders, as it will invariably do during this pointless film).<br /><br />One might better spend one's time staring out a window at a tree growing.<br /><br />\"\n        ],\n        \"semantic_type\": \"\",\n        \"description\": \"\"\n      }\n    },\n    {\n      \"column\": \"label\",\n      \"properties\": {\n        \"dtype\": \"number\",\n        \"std\": 0,\n        \"min\": 0,\n        \"max\": 0,\n        \"num_unique_values\": 1,\n        \"samples\": [\n          0\n        ],\n        \"semantic_type\": \"\",\n        \"description\": \"\"\n      }\n    }\n  ]\n}"
            }
          },
          "metadata": {}
        },
        {
          "output_type": "display_data",
          "data": {
            "text/plain": [
              "                                                text  label\n",
              "0  I love sci-fi and am willing to put up with a ...      0\n",
              "1  Worth the entertainment value of a rental, esp...      0\n",
              "2  its a totally average film with a few semi-alr...      0\n",
              "3  STAR RATING: ***** Saturday Night **** Friday ...      0\n",
              "4  First off let me say, If you haven't enjoyed a...      0"
            ],
            "text/html": [
              "\n",
              "  <div id=\"df-6280979c-5caf-4979-bbe3-a99e5405b94a\" class=\"colab-df-container\">\n",
              "    <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>text</th>\n",
              "      <th>label</th>\n",
              "    </tr>\n",
              "  </thead>\n",
              "  <tbody>\n",
              "    <tr>\n",
              "      <th>0</th>\n",
              "      <td>I love sci-fi and am willing to put up with a ...</td>\n",
              "      <td>0</td>\n",
              "    </tr>\n",
              "    <tr>\n",
              "      <th>1</th>\n",
              "      <td>Worth the entertainment value of a rental, esp...</td>\n",
              "      <td>0</td>\n",
              "    </tr>\n",
              "    <tr>\n",
              "      <th>2</th>\n",
              "      <td>its a totally average film with a few semi-alr...</td>\n",
              "      <td>0</td>\n",
              "    </tr>\n",
              "    <tr>\n",
              "      <th>3</th>\n",
              "      <td>STAR RATING: ***** Saturday Night **** Friday ...</td>\n",
              "      <td>0</td>\n",
              "    </tr>\n",
              "    <tr>\n",
              "      <th>4</th>\n",
              "      <td>First off let me say, If you haven't enjoyed a...</td>\n",
              "      <td>0</td>\n",
              "    </tr>\n",
              "  </tbody>\n",
              "</table>\n",
              "</div>\n",
              "    <div class=\"colab-df-buttons\">\n",
              "\n",
              "  <div class=\"colab-df-container\">\n",
              "    <button class=\"colab-df-convert\" onclick=\"convertToInteractive('df-6280979c-5caf-4979-bbe3-a99e5405b94a')\"\n",
              "            title=\"Convert this dataframe to an interactive table.\"\n",
              "            style=\"display:none;\">\n",
              "\n",
              "  <svg xmlns=\"http://www.w3.org/2000/svg\" height=\"24px\" viewBox=\"0 -960 960 960\">\n",
              "    <path d=\"M120-120v-720h720v720H120Zm60-500h600v-160H180v160Zm220 220h160v-160H400v160Zm0 220h160v-160H400v160ZM180-400h160v-160H180v160Zm440 0h160v-160H620v160ZM180-180h160v-160H180v160Zm440 0h160v-160H620v160Z\"/>\n",
              "  </svg>\n",
              "    </button>\n",
              "\n",
              "  <style>\n",
              "    .colab-df-container {\n",
              "      display:flex;\n",
              "      gap: 12px;\n",
              "    }\n",
              "\n",
              "    .colab-df-convert {\n",
              "      background-color: #E8F0FE;\n",
              "      border: none;\n",
              "      border-radius: 50%;\n",
              "      cursor: pointer;\n",
              "      display: none;\n",
              "      fill: #1967D2;\n",
              "      height: 32px;\n",
              "      padding: 0 0 0 0;\n",
              "      width: 32px;\n",
              "    }\n",
              "\n",
              "    .colab-df-convert:hover {\n",
              "      background-color: #E2EBFA;\n",
              "      box-shadow: 0px 1px 2px rgba(60, 64, 67, 0.3), 0px 1px 3px 1px rgba(60, 64, 67, 0.15);\n",
              "      fill: #174EA6;\n",
              "    }\n",
              "\n",
              "    .colab-df-buttons div {\n",
              "      margin-bottom: 4px;\n",
              "    }\n",
              "\n",
              "    [theme=dark] .colab-df-convert {\n",
              "      background-color: #3B4455;\n",
              "      fill: #D2E3FC;\n",
              "    }\n",
              "\n",
              "    [theme=dark] .colab-df-convert:hover {\n",
              "      background-color: #434B5C;\n",
              "      box-shadow: 0px 1px 3px 1px rgba(0, 0, 0, 0.15);\n",
              "      filter: drop-shadow(0px 1px 2px rgba(0, 0, 0, 0.3));\n",
              "      fill: #FFFFFF;\n",
              "    }\n",
              "  </style>\n",
              "\n",
              "    <script>\n",
              "      const buttonEl =\n",
              "        document.querySelector('#df-6280979c-5caf-4979-bbe3-a99e5405b94a button.colab-df-convert');\n",
              "      buttonEl.style.display =\n",
              "        google.colab.kernel.accessAllowed ? 'block' : 'none';\n",
              "\n",
              "      async function convertToInteractive(key) {\n",
              "        const element = document.querySelector('#df-6280979c-5caf-4979-bbe3-a99e5405b94a');\n",
              "        const dataTable =\n",
              "          await google.colab.kernel.invokeFunction('convertToInteractive',\n",
              "                                                    [key], {});\n",
              "        if (!dataTable) return;\n",
              "\n",
              "        const docLinkHtml = 'Like what you see? Visit the ' +\n",
              "          '<a target=\"_blank\" href=https://colab.research.google.com/notebooks/data_table.ipynb>data table notebook</a>'\n",
              "          + ' to learn more about interactive tables.';\n",
              "        element.innerHTML = '';\n",
              "        dataTable['output_type'] = 'display_data';\n",
              "        await google.colab.output.renderOutput(dataTable, element);\n",
              "        const docLink = document.createElement('div');\n",
              "        docLink.innerHTML = docLinkHtml;\n",
              "        element.appendChild(docLink);\n",
              "      }\n",
              "    </script>\n",
              "  </div>\n",
              "\n",
              "\n",
              "<div id=\"df-fe215a1a-5cd3-4837-9ec2-85c1dc73d930\">\n",
              "  <button class=\"colab-df-quickchart\" onclick=\"quickchart('df-fe215a1a-5cd3-4837-9ec2-85c1dc73d930')\"\n",
              "            title=\"Suggest charts\"\n",
              "            style=\"display:none;\">\n",
              "\n",
              "<svg xmlns=\"http://www.w3.org/2000/svg\" height=\"24px\"viewBox=\"0 0 24 24\"\n",
              "     width=\"24px\">\n",
              "    <g>\n",
              "        <path d=\"M19 3H5c-1.1 0-2 .9-2 2v14c0 1.1.9 2 2 2h14c1.1 0 2-.9 2-2V5c0-1.1-.9-2-2-2zM9 17H7v-7h2v7zm4 0h-2V7h2v10zm4 0h-2v-4h2v4z\"/>\n",
              "    </g>\n",
              "</svg>\n",
              "  </button>\n",
              "\n",
              "<style>\n",
              "  .colab-df-quickchart {\n",
              "      --bg-color: #E8F0FE;\n",
              "      --fill-color: #1967D2;\n",
              "      --hover-bg-color: #E2EBFA;\n",
              "      --hover-fill-color: #174EA6;\n",
              "      --disabled-fill-color: #AAA;\n",
              "      --disabled-bg-color: #DDD;\n",
              "  }\n",
              "\n",
              "  [theme=dark] .colab-df-quickchart {\n",
              "      --bg-color: #3B4455;\n",
              "      --fill-color: #D2E3FC;\n",
              "      --hover-bg-color: #434B5C;\n",
              "      --hover-fill-color: #FFFFFF;\n",
              "      --disabled-bg-color: #3B4455;\n",
              "      --disabled-fill-color: #666;\n",
              "  }\n",
              "\n",
              "  .colab-df-quickchart {\n",
              "    background-color: var(--bg-color);\n",
              "    border: none;\n",
              "    border-radius: 50%;\n",
              "    cursor: pointer;\n",
              "    display: none;\n",
              "    fill: var(--fill-color);\n",
              "    height: 32px;\n",
              "    padding: 0;\n",
              "    width: 32px;\n",
              "  }\n",
              "\n",
              "  .colab-df-quickchart:hover {\n",
              "    background-color: var(--hover-bg-color);\n",
              "    box-shadow: 0 1px 2px rgba(60, 64, 67, 0.3), 0 1px 3px 1px rgba(60, 64, 67, 0.15);\n",
              "    fill: var(--button-hover-fill-color);\n",
              "  }\n",
              "\n",
              "  .colab-df-quickchart-complete:disabled,\n",
              "  .colab-df-quickchart-complete:disabled:hover {\n",
              "    background-color: var(--disabled-bg-color);\n",
              "    fill: var(--disabled-fill-color);\n",
              "    box-shadow: none;\n",
              "  }\n",
              "\n",
              "  .colab-df-spinner {\n",
              "    border: 2px solid var(--fill-color);\n",
              "    border-color: transparent;\n",
              "    border-bottom-color: var(--fill-color);\n",
              "    animation:\n",
              "      spin 1s steps(1) infinite;\n",
              "  }\n",
              "\n",
              "  @keyframes spin {\n",
              "    0% {\n",
              "      border-color: transparent;\n",
              "      border-bottom-color: var(--fill-color);\n",
              "      border-left-color: var(--fill-color);\n",
              "    }\n",
              "    20% {\n",
              "      border-color: transparent;\n",
              "      border-left-color: var(--fill-color);\n",
              "      border-top-color: var(--fill-color);\n",
              "    }\n",
              "    30% {\n",
              "      border-color: transparent;\n",
              "      border-left-color: var(--fill-color);\n",
              "      border-top-color: var(--fill-color);\n",
              "      border-right-color: var(--fill-color);\n",
              "    }\n",
              "    40% {\n",
              "      border-color: transparent;\n",
              "      border-right-color: var(--fill-color);\n",
              "      border-top-color: var(--fill-color);\n",
              "    }\n",
              "    60% {\n",
              "      border-color: transparent;\n",
              "      border-right-color: var(--fill-color);\n",
              "    }\n",
              "    80% {\n",
              "      border-color: transparent;\n",
              "      border-right-color: var(--fill-color);\n",
              "      border-bottom-color: var(--fill-color);\n",
              "    }\n",
              "    90% {\n",
              "      border-color: transparent;\n",
              "      border-bottom-color: var(--fill-color);\n",
              "    }\n",
              "  }\n",
              "</style>\n",
              "\n",
              "  <script>\n",
              "    async function quickchart(key) {\n",
              "      const quickchartButtonEl =\n",
              "        document.querySelector('#' + key + ' button');\n",
              "      quickchartButtonEl.disabled = true;  // To prevent multiple clicks.\n",
              "      quickchartButtonEl.classList.add('colab-df-spinner');\n",
              "      try {\n",
              "        const charts = await google.colab.kernel.invokeFunction(\n",
              "            'suggestCharts', [key], {});\n",
              "      } catch (error) {\n",
              "        console.error('Error during call to suggestCharts:', error);\n",
              "      }\n",
              "      quickchartButtonEl.classList.remove('colab-df-spinner');\n",
              "      quickchartButtonEl.classList.add('colab-df-quickchart-complete');\n",
              "    }\n",
              "    (() => {\n",
              "      let quickchartButtonEl =\n",
              "        document.querySelector('#df-fe215a1a-5cd3-4837-9ec2-85c1dc73d930 button');\n",
              "      quickchartButtonEl.style.display =\n",
              "        google.colab.kernel.accessAllowed ? 'block' : 'none';\n",
              "    })();\n",
              "  </script>\n",
              "</div>\n",
              "\n",
              "    </div>\n",
              "  </div>\n"
            ],
            "application/vnd.google.colaboratory.intrinsic+json": {
              "type": "dataframe",
              "summary": "{\n  \"name\": \"display(df_val\",\n  \"rows\": 5,\n  \"fields\": [\n    {\n      \"column\": \"text\",\n      \"properties\": {\n        \"dtype\": \"string\",\n        \"num_unique_values\": 5,\n        \"samples\": [\n          \"Worth the entertainment value of a rental, especially if you like action movies. This one features the usual car chases, fights with the great Van Damme kick style, shooting battles with the 40 shell load shotgun, and even terrorist style bombs. All of this is entertaining and competently handled but there is nothing that really blows you away if you've seen your share before.<br /><br />The plot is made interesting by the inclusion of a rabbit, which is clever but hardly profound. Many of the characters are heavily stereotyped -- the angry veterans, the terrified illegal aliens, the crooked cops, the indifferent feds, the bitchy tough lady station head, the crooked politician, the fat federale who looks like he was typecast as the Mexican in a Hollywood movie from the 1940s. All passably acted but again nothing special.<br /><br />I thought the main villains were pretty well done and fairly well acted. By the end of the movie you certainly knew who the good guys were and weren't. There was an emotional lift as the really bad ones got their just deserts. Very simplistic, but then you weren't expecting Hamlet, right? The only thing I found really annoying was the constant cuts to VDs daughter during the last fight scene.<br /><br />Not bad. Not good. Passable 4.\",\n          \"First off let me say, If you haven't enjoyed a Van Damme movie since bloodsport, you probably will not like this movie. Most of these movies may not have the best plots or best actors but I enjoy these kinds of movies for what they are. This movie is much better than any of the movies the other action guys (Segal and Dolph) have thought about putting out the past few years. Van Damme is good in the movie, the movie is only worth watching to Van Damme fans. It is not as good as Wake of Death (which i highly recommend to anyone of likes Van Damme) or In hell but, in my opinion it's worth watching. It has the same type of feel to it as Nowhere to Run. Good fun stuff!\",\n          \"its a totally average film with a few semi-alright action sequences that make the plot seem a little better and remind the viewer of the classic van dam films. parts of the plot don't make sense and seem to be added in to use up time. the end plot is that of a very basic type that doesn't leave the viewer guessing and any twists are obvious from the beginning. the end scene with the flask backs don't make sense as they are added in and seem to have little relevance to the history of van dam's character. not really worth watching again, bit disappointed in the end production, even though it is apparent it was shot on a low budget certain shots and sections in the film are of poor directed quality\"\n        ],\n        \"semantic_type\": \"\",\n        \"description\": \"\"\n      }\n    },\n    {\n      \"column\": \"label\",\n      \"properties\": {\n        \"dtype\": \"number\",\n        \"std\": 0,\n        \"min\": 0,\n        \"max\": 0,\n        \"num_unique_values\": 1,\n        \"samples\": [\n          0\n        ],\n        \"semantic_type\": \"\",\n        \"description\": \"\"\n      }\n    }\n  ]\n}"
            }
          },
          "metadata": {}
        }
      ]
    },
    {
      "cell_type": "markdown",
      "source": [
        "## Create a class that will hold all the basic configuration"
      ],
      "metadata": {
        "id": "ddFa8xKbjMuf"
      }
    },
    {
      "cell_type": "code",
      "source": [
        "class Config:\n",
        "    def __init__(self):\n",
        "        super(Config, self).__init__()\n",
        "\n",
        "        self.SEED = 42\n",
        "        self.MODEL_PATH = 'google/t5-efficient-tiny'\n",
        "\n",
        "        # data\n",
        "        self.TOKENIZER = AutoTokenizer.from_pretrained(self.MODEL_PATH)\n",
        "        self.SRC_MAX_LENGTH = 320\n",
        "        self.TGT_MAX_LENGTH = 20\n",
        "        self.BATCH_SIZE = 16\n",
        "\n",
        "        # model\n",
        "        self.DEVICE = torch.device('cuda' if torch.cuda.is_available() else 'cpu')\n",
        "        self.LR = 3e-5\n",
        "        self.EPOCHS = 1\n",
        "\n",
        "config = Config()"
      ],
      "metadata": {
        "id": "wAm2s44CWC1G"
      },
      "execution_count": 43,
      "outputs": []
    },
    {
      "cell_type": "markdown",
      "source": [
        "## Create a T5 specific dataset"
      ],
      "metadata": {
        "id": "xDMQbi4ejQ1U"
      }
    },
    {
      "cell_type": "code",
      "source": [
        "class T5Dataset(Dataset):\n",
        "    def __init__(self, df, set_type=None):\n",
        "        super(T5Dataset, self).__init__()\n",
        "\n",
        "        self.texts = df[\"text\"].values\n",
        "        self.set_type = set_type\n",
        "        if self.set_type != 'test':\n",
        "            self.labels = np.asarray([\"positive\" if i==1 else \"negative\" for i in df[\"label\"].values])\n",
        "\n",
        "        self.tokenizer = config.TOKENIZER\n",
        "        self.src_max_length = config.SRC_MAX_LENGTH\n",
        "        self.tgt_max_length = config.TGT_MAX_LENGTH\n",
        "\n",
        "    def __len__(self):\n",
        "        return len(self.texts)\n",
        "\n",
        "    def __getitem__(self, index):\n",
        "        src_tokenized = self.tokenizer.encode_plus(\n",
        "            self.texts[index],\n",
        "            max_length=self.src_max_length,\n",
        "            pad_to_max_length=True,\n",
        "            truncation=True,\n",
        "            return_attention_mask=True,\n",
        "            return_token_type_ids=False,\n",
        "            return_tensors='pt'\n",
        "        )\n",
        "        src_input_ids = src_tokenized['input_ids'].squeeze()\n",
        "        src_attention_mask = src_tokenized['attention_mask'].squeeze()\n",
        "\n",
        "        if self.set_type != 'test':\n",
        "            tgt_tokenized = self.tokenizer(\n",
        "                self.labels[index],\n",
        "                max_length=self.tgt_max_length,\n",
        "                pad_to_max_length=True,\n",
        "                truncation=True,\n",
        "                return_attention_mask=True,\n",
        "                return_token_type_ids=False,\n",
        "                return_tensors='pt'\n",
        "            )\n",
        "            tgt_input_ids = tgt_tokenized['input_ids'].squeeze()\n",
        "            tgt_attention_mask = tgt_tokenized['attention_mask'].squeeze()\n",
        "\n",
        "            return {\n",
        "                'src_input_ids': src_input_ids.long(),\n",
        "                'src_attention_mask': src_attention_mask.long(),\n",
        "                'tgt_input_ids': tgt_input_ids.long(),\n",
        "                'tgt_attention_mask': tgt_attention_mask.long()\n",
        "            }\n",
        "\n",
        "        return {\n",
        "            'src_input_ids': src_input_ids.long(),\n",
        "            'src_attention_mask': src_attention_mask.long()\n",
        "        }"
      ],
      "metadata": {
        "id": "CbhNm7KdWIB2"
      },
      "execution_count": 15,
      "outputs": []
    },
    {
      "cell_type": "markdown",
      "source": [
        "## Load the IMDB dataset in terms of the T5 dataset, create a dataloader"
      ],
      "metadata": {
        "id": "fHlp4OX2jcZX"
      }
    },
    {
      "cell_type": "code",
      "source": [
        "train_data = T5Dataset(df_train)\n",
        "val_data = T5Dataset(df_val)\n",
        "\n",
        "train_dataloader = DataLoader(train_data, batch_size=config.BATCH_SIZE)\n",
        "val_dataloader = DataLoader(val_data, batch_size=config.BATCH_SIZE)\n",
        "\n",
        "b = next(iter(train_dataloader))\n",
        "for k, v in b.items():\n",
        "    print(f'{k} shape: {v.shape}')"
      ],
      "metadata": {
        "colab": {
          "base_uri": "https://localhost:8080/"
        },
        "id": "dKbDB2nwWPnW",
        "outputId": "7411e84e-cef7-479f-f7ff-46cc8e80f72d"
      },
      "execution_count": 16,
      "outputs": [
        {
          "output_type": "stream",
          "name": "stdout",
          "text": [
            "src_input_ids shape: torch.Size([16, 320])\n",
            "src_attention_mask shape: torch.Size([16, 320])\n",
            "tgt_input_ids shape: torch.Size([16, 20])\n",
            "tgt_attention_mask shape: torch.Size([16, 20])\n"
          ]
        },
        {
          "output_type": "stream",
          "name": "stderr",
          "text": [
            "/usr/local/lib/python3.10/dist-packages/transformers/tokenization_utils_base.py:2645: FutureWarning: The `pad_to_max_length` argument is deprecated and will be removed in a future version, use `padding=True` or `padding='longest'` to pad to the longest sequence in the batch, or use `padding='max_length'` to pad to a max length. In this case, you can give a specific length with `max_length` (e.g. `max_length=45`) or leave max_length to None to pad to the maximal input size of the model (e.g. 512 for Bert).\n",
            "  warnings.warn(\n"
          ]
        }
      ]
    },
    {
      "cell_type": "markdown",
      "source": [
        "## Create the model"
      ],
      "metadata": {
        "id": "8_q47qx7jjdP"
      }
    },
    {
      "cell_type": "code",
      "source": [
        "class T5Model(nn.Module):\n",
        "    def __init__(self):\n",
        "        super(T5Model, self).__init__()\n",
        "\n",
        "        self.t5_model = T5ForConditionalGeneration.from_pretrained(config.MODEL_PATH)\n",
        "\n",
        "    def forward(\n",
        "        self,\n",
        "        input_ids,\n",
        "        attention_mask=None,\n",
        "        decoder_input_ids=None,\n",
        "        decoder_attention_mask=None,\n",
        "        labels=None\n",
        "        ):\n",
        "\n",
        "        return self.t5_model(\n",
        "            input_ids,\n",
        "            attention_mask=attention_mask,\n",
        "            decoder_input_ids=decoder_input_ids,\n",
        "            decoder_attention_mask=decoder_attention_mask,\n",
        "            labels=labels,\n",
        "        )"
      ],
      "metadata": {
        "id": "uYDTFpASWne5"
      },
      "execution_count": 17,
      "outputs": []
    },
    {
      "cell_type": "markdown",
      "source": [
        "## Set the decvice type"
      ],
      "metadata": {
        "id": "cJpXau97joIz"
      }
    },
    {
      "cell_type": "code",
      "source": [
        "device = config.DEVICE\n",
        "print(device)"
      ],
      "metadata": {
        "colab": {
          "base_uri": "https://localhost:8080/"
        },
        "id": "rb-xy6TKYpL3",
        "outputId": "8c219f09-f51b-4223-da20-ea80c9518c5c"
      },
      "execution_count": 44,
      "outputs": [
        {
          "output_type": "stream",
          "name": "stdout",
          "text": [
            "cpu\n"
          ]
        }
      ]
    },
    {
      "cell_type": "markdown",
      "source": [
        "## Load the model"
      ],
      "metadata": {
        "id": "1dlqsJN3jqF3"
      }
    },
    {
      "cell_type": "code",
      "source": [
        "model = T5Model()\n",
        "model.to(device);"
      ],
      "metadata": {
        "id": "6Ga6bR-oZF0-"
      },
      "execution_count": 26,
      "outputs": []
    },
    {
      "cell_type": "markdown",
      "source": [
        "## We want only two neurons in the output for binary classification"
      ],
      "metadata": {
        "id": "BdbTco_0juts"
      }
    },
    {
      "cell_type": "code",
      "source": [
        "model.t5_model.lm_head.out_features = 2"
      ],
      "metadata": {
        "id": "pnXhrTBLdQuo"
      },
      "execution_count": 27,
      "outputs": []
    },
    {
      "cell_type": "markdown",
      "source": [
        "# Finally start training"
      ],
      "metadata": {
        "id": "6O6Iq7V6jzO6"
      }
    },
    {
      "cell_type": "code",
      "source": [
        "torch.manual_seed(config.SEED)\n",
        "criterion = nn.CrossEntropyLoss()\n",
        "optimizer = optim.AdamW(model.parameters(), lr=config.LR)\n",
        "total_loss = 0\n",
        "\n",
        "for epoch in range(1):\n",
        "    for step, batch in enumerate(tqdm(train_dataloader)):\n",
        "        b_src_input_ids = batch['src_input_ids'].to(device)\n",
        "        b_src_attention_mask = batch['src_attention_mask'].to(device)\n",
        "\n",
        "        lm_labels = batch['tgt_input_ids'].to(device)\n",
        "        lm_labels[lm_labels[:, :] == config.TOKENIZER.pad_token_id] = -100\n",
        "\n",
        "        b_tgt_attention_mask = batch['tgt_attention_mask'].to(device)\n",
        "\n",
        "        # clear accumulated gradients\n",
        "        optimizer.zero_grad()\n",
        "\n",
        "        # forward pass\n",
        "        outputs = model(input_ids=b_src_input_ids, attention_mask=b_src_attention_mask, decoder_attention_mask=b_tgt_attention_mask, labels=lm_labels)\n",
        "        loss = outputs[0]\n",
        "        total_loss+= loss.item()\n",
        "\n",
        "        # backward pass\n",
        "        loss.backward()\n",
        "\n",
        "        # update weights\n",
        "        optimizer.step()\n",
        "        print(\"Iteration: \", step, \"\\tLoss:\", loss.item(),\"\\n\")\n",
        "\n",
        "    print(\"\\n\\nEpoch: \", epoch, \"\\tAverage Loss:\", total_loss/len(train_dataloader), \"\\n\")"
      ],
      "metadata": {
        "colab": {
          "base_uri": "https://localhost:8080/",
          "height": 1000,
          "referenced_widgets": [
            "f12eff9a34644d2a8ab452262659d481",
            "6ee8d9fcd9cf41fea26235196dac39f3",
            "bd846b5317604bcb9f1533a2e9628e65",
            "9db7af7d8ebc43c6bce66d9a5f58248d",
            "a164d6185e124a51b7cb71f6ecf2ef16",
            "ab3f035970a24784b8fa7c4ea332766a",
            "5d2a72fd7bb746a7808b3e3af04964b5",
            "31dae569e4af459fbb5bc8fc8cba8f1e",
            "3d8eb7942fc946cd9712e3b6d2621474",
            "4da8c9704f8043078c7905878e81c9e8",
            "48435e59b6844a9f93f8297a55576d13"
          ]
        },
        "id": "1grMAS46ZHiH",
        "outputId": "dfe84488-b8e8-4d57-b242-5ebc7dd01f47"
      },
      "execution_count": 30,
      "outputs": [
        {
          "output_type": "display_data",
          "data": {
            "text/plain": [
              "  0%|          | 0/1563 [00:00<?, ?it/s]"
            ],
            "application/vnd.jupyter.widget-view+json": {
              "version_major": 2,
              "version_minor": 0,
              "model_id": "f12eff9a34644d2a8ab452262659d481"
            }
          },
          "metadata": {}
        },
        {
          "output_type": "stream",
          "name": "stderr",
          "text": [
            "/usr/local/lib/python3.10/dist-packages/transformers/tokenization_utils_base.py:2645: FutureWarning: The `pad_to_max_length` argument is deprecated and will be removed in a future version, use `padding=True` or `padding='longest'` to pad to the longest sequence in the batch, or use `padding='max_length'` to pad to a max length. In this case, you can give a specific length with `max_length` (e.g. `max_length=45`) or leave max_length to None to pad to the maximal input size of the model (e.g. 512 for Bert).\n",
            "  warnings.warn(\n"
          ]
        },
        {
          "output_type": "stream",
          "name": "stdout",
          "text": [
            "Seq2SeqLMOutput(loss=tensor(11.1987, grad_fn=<NllLossBackward0>), logits=tensor([[[ -20.9962,  -20.4858,  -22.9923,  ...,  -84.1084,  -83.3525,\n",
            "           -83.6444],\n",
            "         [ -45.9004,   -8.2612,  -36.3058,  ..., -103.5070, -102.4597,\n",
            "          -102.6672],\n",
            "         [ -45.2228,   -5.3172,  -34.6345,  ..., -103.2310, -102.2147,\n",
            "          -102.4168],\n",
            "         ...,\n",
            "         [ -21.3417,  -20.3774,  -23.3393,  ...,  -84.6165,  -83.8563,\n",
            "           -84.1425],\n",
            "         [ -21.3417,  -20.3774,  -23.3393,  ...,  -84.6165,  -83.8563,\n",
            "           -84.1425],\n",
            "         [ -21.3479,  -20.3740,  -23.3454,  ...,  -84.6156,  -83.8556,\n",
            "           -84.1412]],\n",
            "\n",
            "        [[ -20.4709,  -20.3874,  -22.8477,  ...,  -83.4097,  -82.6592,\n",
            "           -82.9736],\n",
            "         [ -42.9222,   -8.2675,  -34.5631,  ...,  -99.3144,  -98.3259,\n",
            "           -98.5778],\n",
            "         [ -42.5699,   -5.8535,  -33.2131,  ..., -100.1186,  -99.1231,\n",
            "           -99.3686],\n",
            "         ...,\n",
            "         [ -20.8635,  -20.2808,  -23.2158,  ...,  -84.0317,  -83.2755,\n",
            "           -83.5846],\n",
            "         [ -20.8635,  -20.2808,  -23.2158,  ...,  -84.0317,  -83.2755,\n",
            "           -83.5845],\n",
            "         [ -20.8727,  -20.2758,  -23.2238,  ...,  -84.0388,  -83.2825,\n",
            "           -83.5911]],\n",
            "\n",
            "        [[ -20.8298,  -20.6862,  -23.0408,  ...,  -85.0617,  -84.2852,\n",
            "           -84.5858],\n",
            "         [ -44.8902,   -8.8117,  -36.7231,  ..., -105.7183, -104.6494,\n",
            "          -104.8423],\n",
            "         [ -43.9429,   -6.2727,  -35.1433,  ..., -105.7050, -104.6428,\n",
            "          -104.8573],\n",
            "         ...,\n",
            "         [ -21.2309,  -20.5672,  -23.4452,  ...,  -85.6982,  -84.9154,\n",
            "           -85.2102],\n",
            "         [ -21.2309,  -20.5672,  -23.4452,  ...,  -85.6982,  -84.9154,\n",
            "           -85.2102],\n",
            "         [ -21.2384,  -20.5645,  -23.4532,  ...,  -85.7020,  -84.9192,\n",
            "           -85.2135]],\n",
            "\n",
            "        ...,\n",
            "\n",
            "        [[ -21.3719,  -20.5964,  -23.3705,  ...,  -84.9411,  -84.1563,\n",
            "           -84.4639],\n",
            "         [ -45.9630,   -8.1418,  -36.8969,  ..., -104.1168, -103.0868,\n",
            "          -103.2801],\n",
            "         [ -44.1496,   -4.8250,  -34.6572,  ..., -103.4232, -102.4007,\n",
            "          -102.6151],\n",
            "         ...,\n",
            "         [ -21.7031,  -20.4641,  -23.6751,  ...,  -85.3830,  -84.5962,\n",
            "           -84.8967],\n",
            "         [ -21.7031,  -20.4641,  -23.6751,  ...,  -85.3830,  -84.5962,\n",
            "           -84.8967],\n",
            "         [ -21.7018,  -20.4623,  -23.6742,  ...,  -85.3699,  -84.5835,\n",
            "           -84.8833]],\n",
            "\n",
            "        [[ -21.2946,  -20.3971,  -23.0834,  ...,  -84.6453,  -83.8594,\n",
            "           -84.1718],\n",
            "         [ -46.0601,   -9.0658,  -36.1556,  ..., -104.3921, -103.2752,\n",
            "          -103.5487],\n",
            "         [ -44.7963,   -5.6744,  -34.4781,  ..., -103.7965, -102.7082,\n",
            "          -102.9597],\n",
            "         ...,\n",
            "         [ -21.8057,  -20.2636,  -23.5430,  ...,  -85.4418,  -84.6472,\n",
            "           -84.9552],\n",
            "         [ -21.8057,  -20.2636,  -23.5430,  ...,  -85.4418,  -84.6472,\n",
            "           -84.9552],\n",
            "         [ -21.8100,  -20.2627,  -23.5479,  ...,  -85.4395,  -84.6451,\n",
            "           -84.9525]],\n",
            "\n",
            "        [[ -21.0499,  -20.3888,  -23.4369,  ...,  -84.3763,  -83.6113,\n",
            "           -83.9068],\n",
            "         [ -46.3711,   -8.0923,  -37.9030,  ..., -105.5547, -104.4794,\n",
            "          -104.7025],\n",
            "         [ -45.1366,   -4.5183,  -35.8821,  ..., -104.8517, -103.8212,\n",
            "          -104.0019],\n",
            "         ...,\n",
            "         [ -21.5505,  -20.2352,  -23.9019,  ...,  -85.1717,  -84.3993,\n",
            "           -84.6893],\n",
            "         [ -21.5505,  -20.2352,  -23.9019,  ...,  -85.1717,  -84.3993,\n",
            "           -84.6893],\n",
            "         [ -21.5651,  -20.2290,  -23.9144,  ...,  -85.1865,  -84.4140,\n",
            "           -84.7035]]], grad_fn=<UnsafeViewBackward0>), past_key_values=((tensor([[[[-0.0802, -0.9374,  0.5559,  ..., -0.3421,  0.3717, -0.0673],\n",
            "          [ 0.2761,  2.6057,  0.7023,  ..., -0.3335,  0.3968, -0.1335],\n",
            "          [-0.2980, -1.4155,  0.9510,  ..., -0.1536,  0.4848, -0.0200],\n",
            "          ...,\n",
            "          [-0.0802, -0.9374,  0.5559,  ..., -0.3421,  0.3717, -0.0673],\n",
            "          [-0.0802, -0.9374,  0.5559,  ..., -0.3421,  0.3717, -0.0673],\n",
            "          [-0.0802, -0.9374,  0.5559,  ..., -0.3421,  0.3717, -0.0673]],\n",
            "\n",
            "         [[ 0.2038,  0.7391, -0.2659,  ..., -0.1168, -0.0383,  0.2858],\n",
            "          [-0.2098,  0.6358, -0.0551,  ..., -0.2809,  0.4859, -0.8332],\n",
            "          [ 0.4086, -0.9085,  1.4012,  ..., -0.7964,  1.3842,  2.8988],\n",
            "          ...,\n",
            "          [ 0.2038,  0.7391, -0.2659,  ..., -0.1168, -0.0383,  0.2858],\n",
            "          [ 0.2038,  0.7391, -0.2659,  ..., -0.1168, -0.0383,  0.2858],\n",
            "          [ 0.2038,  0.7391, -0.2659,  ..., -0.1168, -0.0383,  0.2858]],\n",
            "\n",
            "         [[-1.5675, -0.5233,  0.8922,  ..., -0.4573, -0.1494, -0.2017],\n",
            "          [-1.5565,  0.0480, -1.1054,  ..., -1.7199, -0.6439, -1.1924],\n",
            "          [-1.8930, -1.3750,  0.4017,  ...,  1.4386, -0.1092,  0.2197],\n",
            "          ...,\n",
            "          [-1.5675, -0.5233,  0.8922,  ..., -0.4573, -0.1494, -0.2017],\n",
            "          [-1.5675, -0.5233,  0.8922,  ..., -0.4573, -0.1494, -0.2017],\n",
            "          [-1.5675, -0.5233,  0.8922,  ..., -0.4573, -0.1494, -0.2017]],\n",
            "\n",
            "         [[-0.2548, -0.8824,  3.9518,  ...,  0.1693,  0.5856, -0.2281],\n",
            "          [-0.8397, -2.1420, -2.4968,  ..., -0.5699, -0.2402, -0.2235],\n",
            "          [-0.1646, -1.9496,  1.4994,  ...,  1.7625, -1.0961,  1.4015],\n",
            "          ...,\n",
            "          [-0.2548, -0.8824,  3.9518,  ...,  0.1693,  0.5856, -0.2281],\n",
            "          [-0.2548, -0.8824,  3.9518,  ...,  0.1693,  0.5856, -0.2281],\n",
            "          [-0.2548, -0.8824,  3.9518,  ...,  0.1693,  0.5856, -0.2281]]],\n",
            "\n",
            "\n",
            "        [[[-0.0802, -0.9374,  0.5559,  ..., -0.3421,  0.3717, -0.0673],\n",
            "          [ 0.2761,  2.6057,  0.7023,  ..., -0.3335,  0.3968, -0.1335],\n",
            "          [-0.2980, -1.4155,  0.9510,  ..., -0.1536,  0.4848, -0.0200],\n",
            "          ...,\n",
            "          [-0.0802, -0.9374,  0.5559,  ..., -0.3421,  0.3717, -0.0673],\n",
            "          [-0.0802, -0.9374,  0.5559,  ..., -0.3421,  0.3717, -0.0673],\n",
            "          [-0.0802, -0.9374,  0.5559,  ..., -0.3421,  0.3717, -0.0673]],\n",
            "\n",
            "         [[ 0.2038,  0.7391, -0.2659,  ..., -0.1168, -0.0383,  0.2858],\n",
            "          [-0.2098,  0.6358, -0.0551,  ..., -0.2809,  0.4859, -0.8332],\n",
            "          [ 0.4086, -0.9085,  1.4012,  ..., -0.7964,  1.3842,  2.8988],\n",
            "          ...,\n",
            "          [ 0.2038,  0.7391, -0.2659,  ..., -0.1168, -0.0383,  0.2858],\n",
            "          [ 0.2038,  0.7391, -0.2659,  ..., -0.1168, -0.0383,  0.2858],\n",
            "          [ 0.2038,  0.7391, -0.2659,  ..., -0.1168, -0.0383,  0.2858]],\n",
            "\n",
            "         [[-1.5675, -0.5233,  0.8922,  ..., -0.4573, -0.1494, -0.2017],\n",
            "          [-1.5565,  0.0480, -1.1054,  ..., -1.7199, -0.6439, -1.1924],\n",
            "          [-1.8930, -1.3750,  0.4017,  ...,  1.4386, -0.1092,  0.2197],\n",
            "          ...,\n",
            "          [-1.5675, -0.5233,  0.8922,  ..., -0.4573, -0.1494, -0.2017],\n",
            "          [-1.5675, -0.5233,  0.8922,  ..., -0.4573, -0.1494, -0.2017],\n",
            "          [-1.5675, -0.5233,  0.8922,  ..., -0.4573, -0.1494, -0.2017]],\n",
            "\n",
            "         [[-0.2548, -0.8824,  3.9518,  ...,  0.1693,  0.5856, -0.2281],\n",
            "          [-0.8397, -2.1420, -2.4968,  ..., -0.5699, -0.2402, -0.2235],\n",
            "          [-0.1646, -1.9496,  1.4994,  ...,  1.7625, -1.0961,  1.4015],\n",
            "          ...,\n",
            "          [-0.2548, -0.8824,  3.9518,  ...,  0.1693,  0.5856, -0.2281],\n",
            "          [-0.2548, -0.8824,  3.9518,  ...,  0.1693,  0.5856, -0.2281],\n",
            "          [-0.2548, -0.8824,  3.9518,  ...,  0.1693,  0.5856, -0.2281]]],\n",
            "\n",
            "\n",
            "        [[[-0.0802, -0.9374,  0.5559,  ..., -0.3421,  0.3717, -0.0673],\n",
            "          [ 0.2761,  2.6057,  0.7023,  ..., -0.3335,  0.3968, -0.1335],\n",
            "          [-0.2980, -1.4155,  0.9510,  ..., -0.1536,  0.4848, -0.0200],\n",
            "          ...,\n",
            "          [-0.0802, -0.9374,  0.5559,  ..., -0.3421,  0.3717, -0.0673],\n",
            "          [-0.0802, -0.9374,  0.5559,  ..., -0.3421,  0.3717, -0.0673],\n",
            "          [-0.0802, -0.9374,  0.5559,  ..., -0.3421,  0.3717, -0.0673]],\n",
            "\n",
            "         [[ 0.2038,  0.7391, -0.2659,  ..., -0.1168, -0.0383,  0.2858],\n",
            "          [-0.2098,  0.6358, -0.0551,  ..., -0.2809,  0.4859, -0.8332],\n",
            "          [ 0.4086, -0.9085,  1.4012,  ..., -0.7964,  1.3842,  2.8988],\n",
            "          ...,\n",
            "          [ 0.2038,  0.7391, -0.2659,  ..., -0.1168, -0.0383,  0.2858],\n",
            "          [ 0.2038,  0.7391, -0.2659,  ..., -0.1168, -0.0383,  0.2858],\n",
            "          [ 0.2038,  0.7391, -0.2659,  ..., -0.1168, -0.0383,  0.2858]],\n",
            "\n",
            "         [[-1.5675, -0.5233,  0.8922,  ..., -0.4573, -0.1494, -0.2017],\n",
            "          [-1.5565,  0.0480, -1.1054,  ..., -1.7199, -0.6439, -1.1924],\n",
            "          [-1.8930, -1.3750,  0.4017,  ...,  1.4386, -0.1092,  0.2197],\n",
            "          ...,\n",
            "          [-1.5675, -0.5233,  0.8922,  ..., -0.4573, -0.1494, -0.2017],\n",
            "          [-1.5675, -0.5233,  0.8922,  ..., -0.4573, -0.1494, -0.2017],\n",
            "          [-1.5675, -0.5233,  0.8922,  ..., -0.4573, -0.1494, -0.2017]],\n",
            "\n",
            "         [[-0.2548, -0.8824,  3.9518,  ...,  0.1693,  0.5856, -0.2281],\n",
            "          [-0.8397, -2.1420, -2.4968,  ..., -0.5699, -0.2402, -0.2235],\n",
            "          [-0.1646, -1.9496,  1.4994,  ...,  1.7625, -1.0961,  1.4015],\n",
            "          ...,\n",
            "          [-0.2548, -0.8824,  3.9518,  ...,  0.1693,  0.5856, -0.2281],\n",
            "          [-0.2548, -0.8824,  3.9518,  ...,  0.1693,  0.5856, -0.2281],\n",
            "          [-0.2548, -0.8824,  3.9518,  ...,  0.1693,  0.5856, -0.2281]]],\n",
            "\n",
            "\n",
            "        ...,\n",
            "\n",
            "\n",
            "        [[[-0.0802, -0.9374,  0.5559,  ..., -0.3421,  0.3717, -0.0673],\n",
            "          [ 0.2761,  2.6057,  0.7023,  ..., -0.3335,  0.3968, -0.1335],\n",
            "          [-0.2980, -1.4155,  0.9510,  ..., -0.1536,  0.4848, -0.0200],\n",
            "          ...,\n",
            "          [-0.0802, -0.9374,  0.5559,  ..., -0.3421,  0.3717, -0.0673],\n",
            "          [-0.0802, -0.9374,  0.5559,  ..., -0.3421,  0.3717, -0.0673],\n",
            "          [-0.0802, -0.9374,  0.5559,  ..., -0.3421,  0.3717, -0.0673]],\n",
            "\n",
            "         [[ 0.2038,  0.7391, -0.2659,  ..., -0.1168, -0.0383,  0.2858],\n",
            "          [-0.2098,  0.6358, -0.0551,  ..., -0.2809,  0.4859, -0.8332],\n",
            "          [ 0.4086, -0.9085,  1.4012,  ..., -0.7964,  1.3842,  2.8988],\n",
            "          ...,\n",
            "          [ 0.2038,  0.7391, -0.2659,  ..., -0.1168, -0.0383,  0.2858],\n",
            "          [ 0.2038,  0.7391, -0.2659,  ..., -0.1168, -0.0383,  0.2858],\n",
            "          [ 0.2038,  0.7391, -0.2659,  ..., -0.1168, -0.0383,  0.2858]],\n",
            "\n",
            "         [[-1.5675, -0.5233,  0.8922,  ..., -0.4573, -0.1494, -0.2017],\n",
            "          [-1.5565,  0.0480, -1.1054,  ..., -1.7199, -0.6439, -1.1924],\n",
            "          [-1.8930, -1.3750,  0.4017,  ...,  1.4386, -0.1092,  0.2197],\n",
            "          ...,\n",
            "          [-1.5675, -0.5233,  0.8922,  ..., -0.4573, -0.1494, -0.2017],\n",
            "          [-1.5675, -0.5233,  0.8922,  ..., -0.4573, -0.1494, -0.2017],\n",
            "          [-1.5675, -0.5233,  0.8922,  ..., -0.4573, -0.1494, -0.2017]],\n",
            "\n",
            "         [[-0.2548, -0.8824,  3.9518,  ...,  0.1693,  0.5856, -0.2281],\n",
            "          [-0.8397, -2.1420, -2.4968,  ..., -0.5699, -0.2402, -0.2235],\n",
            "          [-0.1646, -1.9496,  1.4994,  ...,  1.7625, -1.0961,  1.4015],\n",
            "          ...,\n",
            "          [-0.2548, -0.8824,  3.9518,  ...,  0.1693,  0.5856, -0.2281],\n",
            "          [-0.2548, -0.8824,  3.9518,  ...,  0.1693,  0.5856, -0.2281],\n",
            "          [-0.2548, -0.8824,  3.9518,  ...,  0.1693,  0.5856, -0.2281]]],\n",
            "\n",
            "\n",
            "        [[[-0.0802, -0.9374,  0.5559,  ..., -0.3421,  0.3717, -0.0673],\n",
            "          [ 0.2761,  2.6057,  0.7023,  ..., -0.3335,  0.3968, -0.1335],\n",
            "          [-0.2980, -1.4155,  0.9510,  ..., -0.1536,  0.4848, -0.0200],\n",
            "          ...,\n",
            "          [-0.0802, -0.9374,  0.5559,  ..., -0.3421,  0.3717, -0.0673],\n",
            "          [-0.0802, -0.9374,  0.5559,  ..., -0.3421,  0.3717, -0.0673],\n",
            "          [-0.0802, -0.9374,  0.5559,  ..., -0.3421,  0.3717, -0.0673]],\n",
            "\n",
            "         [[ 0.2038,  0.7391, -0.2659,  ..., -0.1168, -0.0383,  0.2858],\n",
            "          [-0.2098,  0.6358, -0.0551,  ..., -0.2809,  0.4859, -0.8332],\n",
            "          [ 0.4086, -0.9085,  1.4012,  ..., -0.7964,  1.3842,  2.8988],\n",
            "          ...,\n",
            "          [ 0.2038,  0.7391, -0.2659,  ..., -0.1168, -0.0383,  0.2858],\n",
            "          [ 0.2038,  0.7391, -0.2659,  ..., -0.1168, -0.0383,  0.2858],\n",
            "          [ 0.2038,  0.7391, -0.2659,  ..., -0.1168, -0.0383,  0.2858]],\n",
            "\n",
            "         [[-1.5675, -0.5233,  0.8922,  ..., -0.4573, -0.1494, -0.2017],\n",
            "          [-1.5565,  0.0480, -1.1054,  ..., -1.7199, -0.6439, -1.1924],\n",
            "          [-1.8930, -1.3750,  0.4017,  ...,  1.4386, -0.1092,  0.2197],\n",
            "          ...,\n",
            "          [-1.5675, -0.5233,  0.8922,  ..., -0.4573, -0.1494, -0.2017],\n",
            "          [-1.5675, -0.5233,  0.8922,  ..., -0.4573, -0.1494, -0.2017],\n",
            "          [-1.5675, -0.5233,  0.8922,  ..., -0.4573, -0.1494, -0.2017]],\n",
            "\n",
            "         [[-0.2548, -0.8824,  3.9518,  ...,  0.1693,  0.5856, -0.2281],\n",
            "          [-0.8397, -2.1420, -2.4968,  ..., -0.5699, -0.2402, -0.2235],\n",
            "          [-0.1646, -1.9496,  1.4994,  ...,  1.7625, -1.0961,  1.4015],\n",
            "          ...,\n",
            "          [-0.2548, -0.8824,  3.9518,  ...,  0.1693,  0.5856, -0.2281],\n",
            "          [-0.2548, -0.8824,  3.9518,  ...,  0.1693,  0.5856, -0.2281],\n",
            "          [-0.2548, -0.8824,  3.9518,  ...,  0.1693,  0.5856, -0.2281]]],\n",
            "\n",
            "\n",
            "        [[[-0.0802, -0.9374,  0.5559,  ..., -0.3421,  0.3717, -0.0673],\n",
            "          [ 0.2761,  2.6057,  0.7023,  ..., -0.3335,  0.3968, -0.1335],\n",
            "          [-0.2980, -1.4155,  0.9510,  ..., -0.1536,  0.4848, -0.0200],\n",
            "          ...,\n",
            "          [-0.0802, -0.9374,  0.5559,  ..., -0.3421,  0.3717, -0.0673],\n",
            "          [-0.0802, -0.9374,  0.5559,  ..., -0.3421,  0.3717, -0.0673],\n",
            "          [-0.0802, -0.9374,  0.5559,  ..., -0.3421,  0.3717, -0.0673]],\n",
            "\n",
            "         [[ 0.2038,  0.7391, -0.2659,  ..., -0.1168, -0.0383,  0.2858],\n",
            "          [-0.2098,  0.6358, -0.0551,  ..., -0.2809,  0.4859, -0.8332],\n",
            "          [ 0.4086, -0.9085,  1.4012,  ..., -0.7964,  1.3842,  2.8988],\n",
            "          ...,\n",
            "          [ 0.2038,  0.7391, -0.2659,  ..., -0.1168, -0.0383,  0.2858],\n",
            "          [ 0.2038,  0.7391, -0.2659,  ..., -0.1168, -0.0383,  0.2858],\n",
            "          [ 0.2038,  0.7391, -0.2659,  ..., -0.1168, -0.0383,  0.2858]],\n",
            "\n",
            "         [[-1.5675, -0.5233,  0.8922,  ..., -0.4573, -0.1494, -0.2017],\n",
            "          [-1.5565,  0.0480, -1.1054,  ..., -1.7199, -0.6439, -1.1924],\n",
            "          [-1.8930, -1.3750,  0.4017,  ...,  1.4386, -0.1092,  0.2197],\n",
            "          ...,\n",
            "          [-1.5675, -0.5233,  0.8922,  ..., -0.4573, -0.1494, -0.2017],\n",
            "          [-1.5675, -0.5233,  0.8922,  ..., -0.4573, -0.1494, -0.2017],\n",
            "          [-1.5675, -0.5233,  0.8922,  ..., -0.4573, -0.1494, -0.2017]],\n",
            "\n",
            "         [[-0.2548, -0.8824,  3.9518,  ...,  0.1693,  0.5856, -0.2281],\n",
            "          [-0.8397, -2.1420, -2.4968,  ..., -0.5699, -0.2402, -0.2235],\n",
            "          [-0.1646, -1.9496,  1.4994,  ...,  1.7625, -1.0961,  1.4015],\n",
            "          ...,\n",
            "          [-0.2548, -0.8824,  3.9518,  ...,  0.1693,  0.5856, -0.2281],\n",
            "          [-0.2548, -0.8824,  3.9518,  ...,  0.1693,  0.5856, -0.2281],\n",
            "          [-0.2548, -0.8824,  3.9518,  ...,  0.1693,  0.5856, -0.2281]]]],\n",
            "       grad_fn=<TransposeBackward0>), tensor([[[[ 0.0752, -4.5247, -0.4626,  ..., -1.4075, -2.3589,  0.4374],\n",
            "          [-0.5287, -1.0504, -2.5758,  ...,  1.1287, -0.9756, -1.2399],\n",
            "          [-0.1439, -0.2531, -1.0258,  ..., -0.1197, -2.2890, -1.3296],\n",
            "          ...,\n",
            "          [ 0.0752, -4.5247, -0.4626,  ..., -1.4075, -2.3589,  0.4374],\n",
            "          [ 0.0752, -4.5247, -0.4626,  ..., -1.4075, -2.3589,  0.4374],\n",
            "          [ 0.0752, -4.5247, -0.4626,  ..., -1.4075, -2.3589,  0.4374]],\n",
            "\n",
            "         [[ 0.0716, -1.8676,  1.8165,  ...,  0.6415, -0.1342, -0.6788],\n",
            "          [ 1.6406, -0.2581, -1.6513,  ..., -1.0369, -3.9917,  1.3927],\n",
            "          [-1.6525, -0.3287,  0.2212,  ...,  0.7428, -0.9865,  2.1311],\n",
            "          ...,\n",
            "          [ 0.0716, -1.8676,  1.8165,  ...,  0.6415, -0.1342, -0.6788],\n",
            "          [ 0.0716, -1.8676,  1.8165,  ...,  0.6415, -0.1342, -0.6788],\n",
            "          [ 0.0716, -1.8676,  1.8165,  ...,  0.6415, -0.1342, -0.6788]],\n",
            "\n",
            "         [[-0.0131, -0.2000, -2.3215,  ..., -0.5908, -1.2056, -0.1269],\n",
            "          [ 0.0865,  1.1614, -2.6137,  ..., -0.1147, -3.1317,  3.1876],\n",
            "          [ 1.1040,  2.9440, -0.6779,  ..., -3.5193, -1.5371,  1.2160],\n",
            "          ...,\n",
            "          [-0.0131, -0.2000, -2.3215,  ..., -0.5908, -1.2056, -0.1269],\n",
            "          [-0.0131, -0.2000, -2.3215,  ..., -0.5908, -1.2056, -0.1269],\n",
            "          [-0.0131, -0.2000, -2.3215,  ..., -0.5908, -1.2056, -0.1269]],\n",
            "\n",
            "         [[-1.0818,  0.1445, -0.6434,  ..., -0.7171,  0.6784,  0.1731],\n",
            "          [-0.2313,  3.5466,  2.5665,  ...,  0.2620,  0.2938, -1.5116],\n",
            "          [ 0.9583,  1.4114, -1.5034,  ...,  1.1955, -0.0332, -0.5989],\n",
            "          ...,\n",
            "          [-1.0818,  0.1445, -0.6434,  ..., -0.7171,  0.6784,  0.1731],\n",
            "          [-1.0818,  0.1445, -0.6434,  ..., -0.7171,  0.6784,  0.1731],\n",
            "          [-1.0818,  0.1445, -0.6434,  ..., -0.7171,  0.6784,  0.1731]]],\n",
            "\n",
            "\n",
            "        [[[ 0.0752, -4.5247, -0.4626,  ..., -1.4075, -2.3589,  0.4374],\n",
            "          [-0.5287, -1.0504, -2.5758,  ...,  1.1287, -0.9756, -1.2399],\n",
            "          [-0.1439, -0.2531, -1.0258,  ..., -0.1197, -2.2890, -1.3296],\n",
            "          ...,\n",
            "          [ 0.0752, -4.5247, -0.4626,  ..., -1.4075, -2.3589,  0.4374],\n",
            "          [ 0.0752, -4.5247, -0.4626,  ..., -1.4075, -2.3589,  0.4374],\n",
            "          [ 0.0752, -4.5247, -0.4626,  ..., -1.4075, -2.3589,  0.4374]],\n",
            "\n",
            "         [[ 0.0716, -1.8676,  1.8165,  ...,  0.6415, -0.1342, -0.6788],\n",
            "          [ 1.6406, -0.2581, -1.6513,  ..., -1.0369, -3.9917,  1.3927],\n",
            "          [-1.6525, -0.3287,  0.2212,  ...,  0.7428, -0.9865,  2.1311],\n",
            "          ...,\n",
            "          [ 0.0716, -1.8676,  1.8165,  ...,  0.6415, -0.1342, -0.6788],\n",
            "          [ 0.0716, -1.8676,  1.8165,  ...,  0.6415, -0.1342, -0.6788],\n",
            "          [ 0.0716, -1.8676,  1.8165,  ...,  0.6415, -0.1342, -0.6788]],\n",
            "\n",
            "         [[-0.0131, -0.2000, -2.3215,  ..., -0.5908, -1.2056, -0.1269],\n",
            "          [ 0.0865,  1.1614, -2.6137,  ..., -0.1147, -3.1317,  3.1876],\n",
            "          [ 1.1040,  2.9440, -0.6779,  ..., -3.5193, -1.5371,  1.2160],\n",
            "          ...,\n",
            "          [-0.0131, -0.2000, -2.3215,  ..., -0.5908, -1.2056, -0.1269],\n",
            "          [-0.0131, -0.2000, -2.3215,  ..., -0.5908, -1.2056, -0.1269],\n",
            "          [-0.0131, -0.2000, -2.3215,  ..., -0.5908, -1.2056, -0.1269]],\n",
            "\n",
            "         [[-1.0818,  0.1445, -0.6434,  ..., -0.7171,  0.6784,  0.1731],\n",
            "          [-0.2313,  3.5466,  2.5665,  ...,  0.2620,  0.2938, -1.5116],\n",
            "          [ 0.9583,  1.4114, -1.5034,  ...,  1.1955, -0.0332, -0.5989],\n",
            "          ...,\n",
            "          [-1.0818,  0.1445, -0.6434,  ..., -0.7171,  0.6784,  0.1731],\n",
            "          [-1.0818,  0.1445, -0.6434,  ..., -0.7171,  0.6784,  0.1731],\n",
            "          [-1.0818,  0.1445, -0.6434,  ..., -0.7171,  0.6784,  0.1731]]],\n",
            "\n",
            "\n",
            "        [[[ 0.0752, -4.5247, -0.4626,  ..., -1.4075, -2.3589,  0.4374],\n",
            "          [-0.5287, -1.0504, -2.5758,  ...,  1.1287, -0.9756, -1.2399],\n",
            "          [-0.1439, -0.2531, -1.0258,  ..., -0.1197, -2.2890, -1.3296],\n",
            "          ...,\n",
            "          [ 0.0752, -4.5247, -0.4626,  ..., -1.4075, -2.3589,  0.4374],\n",
            "          [ 0.0752, -4.5247, -0.4626,  ..., -1.4075, -2.3589,  0.4374],\n",
            "          [ 0.0752, -4.5247, -0.4626,  ..., -1.4075, -2.3589,  0.4374]],\n",
            "\n",
            "         [[ 0.0716, -1.8676,  1.8165,  ...,  0.6415, -0.1342, -0.6788],\n",
            "          [ 1.6406, -0.2581, -1.6513,  ..., -1.0369, -3.9917,  1.3927],\n",
            "          [-1.6525, -0.3287,  0.2212,  ...,  0.7428, -0.9865,  2.1311],\n",
            "          ...,\n",
            "          [ 0.0716, -1.8676,  1.8165,  ...,  0.6415, -0.1342, -0.6788],\n",
            "          [ 0.0716, -1.8676,  1.8165,  ...,  0.6415, -0.1342, -0.6788],\n",
            "          [ 0.0716, -1.8676,  1.8165,  ...,  0.6415, -0.1342, -0.6788]],\n",
            "\n",
            "         [[-0.0131, -0.2000, -2.3215,  ..., -0.5908, -1.2056, -0.1269],\n",
            "          [ 0.0865,  1.1614, -2.6137,  ..., -0.1147, -3.1317,  3.1876],\n",
            "          [ 1.1040,  2.9440, -0.6779,  ..., -3.5193, -1.5371,  1.2160],\n",
            "          ...,\n",
            "          [-0.0131, -0.2000, -2.3215,  ..., -0.5908, -1.2056, -0.1269],\n",
            "          [-0.0131, -0.2000, -2.3215,  ..., -0.5908, -1.2056, -0.1269],\n",
            "          [-0.0131, -0.2000, -2.3215,  ..., -0.5908, -1.2056, -0.1269]],\n",
            "\n",
            "         [[-1.0818,  0.1445, -0.6434,  ..., -0.7171,  0.6784,  0.1731],\n",
            "          [-0.2313,  3.5466,  2.5665,  ...,  0.2620,  0.2938, -1.5116],\n",
            "          [ 0.9583,  1.4114, -1.5034,  ...,  1.1955, -0.0332, -0.5989],\n",
            "          ...,\n",
            "          [-1.0818,  0.1445, -0.6434,  ..., -0.7171,  0.6784,  0.1731],\n",
            "          [-1.0818,  0.1445, -0.6434,  ..., -0.7171,  0.6784,  0.1731],\n",
            "          [-1.0818,  0.1445, -0.6434,  ..., -0.7171,  0.6784,  0.1731]]],\n",
            "\n",
            "\n",
            "        ...,\n",
            "\n",
            "\n",
            "        [[[ 0.0752, -4.5247, -0.4626,  ..., -1.4075, -2.3589,  0.4374],\n",
            "          [-0.5287, -1.0504, -2.5758,  ...,  1.1287, -0.9756, -1.2399],\n",
            "          [-0.1439, -0.2531, -1.0258,  ..., -0.1197, -2.2890, -1.3296],\n",
            "          ...,\n",
            "          [ 0.0752, -4.5247, -0.4626,  ..., -1.4075, -2.3589,  0.4374],\n",
            "          [ 0.0752, -4.5247, -0.4626,  ..., -1.4075, -2.3589,  0.4374],\n",
            "          [ 0.0752, -4.5247, -0.4626,  ..., -1.4075, -2.3589,  0.4374]],\n",
            "\n",
            "         [[ 0.0716, -1.8676,  1.8165,  ...,  0.6415, -0.1342, -0.6788],\n",
            "          [ 1.6406, -0.2581, -1.6513,  ..., -1.0369, -3.9917,  1.3927],\n",
            "          [-1.6525, -0.3287,  0.2212,  ...,  0.7428, -0.9865,  2.1311],\n",
            "          ...,\n",
            "          [ 0.0716, -1.8676,  1.8165,  ...,  0.6415, -0.1342, -0.6788],\n",
            "          [ 0.0716, -1.8676,  1.8165,  ...,  0.6415, -0.1342, -0.6788],\n",
            "          [ 0.0716, -1.8676,  1.8165,  ...,  0.6415, -0.1342, -0.6788]],\n",
            "\n",
            "         [[-0.0131, -0.2000, -2.3215,  ..., -0.5908, -1.2056, -0.1269],\n",
            "          [ 0.0865,  1.1614, -2.6137,  ..., -0.1147, -3.1317,  3.1876],\n",
            "          [ 1.1040,  2.9440, -0.6779,  ..., -3.5193, -1.5371,  1.2160],\n",
            "          ...,\n",
            "          [-0.0131, -0.2000, -2.3215,  ..., -0.5908, -1.2056, -0.1269],\n",
            "          [-0.0131, -0.2000, -2.3215,  ..., -0.5908, -1.2056, -0.1269],\n",
            "          [-0.0131, -0.2000, -2.3215,  ..., -0.5908, -1.2056, -0.1269]],\n",
            "\n",
            "         [[-1.0818,  0.1445, -0.6434,  ..., -0.7171,  0.6784,  0.1731],\n",
            "          [-0.2313,  3.5466,  2.5665,  ...,  0.2620,  0.2938, -1.5116],\n",
            "          [ 0.9583,  1.4114, -1.5034,  ...,  1.1955, -0.0332, -0.5989],\n",
            "          ...,\n",
            "          [-1.0818,  0.1445, -0.6434,  ..., -0.7171,  0.6784,  0.1731],\n",
            "          [-1.0818,  0.1445, -0.6434,  ..., -0.7171,  0.6784,  0.1731],\n",
            "          [-1.0818,  0.1445, -0.6434,  ..., -0.7171,  0.6784,  0.1731]]],\n",
            "\n",
            "\n",
            "        [[[ 0.0752, -4.5247, -0.4626,  ..., -1.4075, -2.3589,  0.4374],\n",
            "          [-0.5287, -1.0504, -2.5758,  ...,  1.1287, -0.9756, -1.2399],\n",
            "          [-0.1439, -0.2531, -1.0258,  ..., -0.1197, -2.2890, -1.3296],\n",
            "          ...,\n",
            "          [ 0.0752, -4.5247, -0.4626,  ..., -1.4075, -2.3589,  0.4374],\n",
            "          [ 0.0752, -4.5247, -0.4626,  ..., -1.4075, -2.3589,  0.4374],\n",
            "          [ 0.0752, -4.5247, -0.4626,  ..., -1.4075, -2.3589,  0.4374]],\n",
            "\n",
            "         [[ 0.0716, -1.8676,  1.8165,  ...,  0.6415, -0.1342, -0.6788],\n",
            "          [ 1.6406, -0.2581, -1.6513,  ..., -1.0369, -3.9917,  1.3927],\n",
            "          [-1.6525, -0.3287,  0.2212,  ...,  0.7428, -0.9865,  2.1311],\n",
            "          ...,\n",
            "          [ 0.0716, -1.8676,  1.8165,  ...,  0.6415, -0.1342, -0.6788],\n",
            "          [ 0.0716, -1.8676,  1.8165,  ...,  0.6415, -0.1342, -0.6788],\n",
            "          [ 0.0716, -1.8676,  1.8165,  ...,  0.6415, -0.1342, -0.6788]],\n",
            "\n",
            "         [[-0.0131, -0.2000, -2.3215,  ..., -0.5908, -1.2056, -0.1269],\n",
            "          [ 0.0865,  1.1614, -2.6137,  ..., -0.1147, -3.1317,  3.1876],\n",
            "          [ 1.1040,  2.9440, -0.6779,  ..., -3.5193, -1.5371,  1.2160],\n",
            "          ...,\n",
            "          [-0.0131, -0.2000, -2.3215,  ..., -0.5908, -1.2056, -0.1269],\n",
            "          [-0.0131, -0.2000, -2.3215,  ..., -0.5908, -1.2056, -0.1269],\n",
            "          [-0.0131, -0.2000, -2.3215,  ..., -0.5908, -1.2056, -0.1269]],\n",
            "\n",
            "         [[-1.0818,  0.1445, -0.6434,  ..., -0.7171,  0.6784,  0.1731],\n",
            "          [-0.2313,  3.5466,  2.5665,  ...,  0.2620,  0.2938, -1.5116],\n",
            "          [ 0.9583,  1.4114, -1.5034,  ...,  1.1955, -0.0332, -0.5989],\n",
            "          ...,\n",
            "          [-1.0818,  0.1445, -0.6434,  ..., -0.7171,  0.6784,  0.1731],\n",
            "          [-1.0818,  0.1445, -0.6434,  ..., -0.7171,  0.6784,  0.1731],\n",
            "          [-1.0818,  0.1445, -0.6434,  ..., -0.7171,  0.6784,  0.1731]]],\n",
            "\n",
            "\n",
            "        [[[ 0.0752, -4.5247, -0.4626,  ..., -1.4075, -2.3589,  0.4374],\n",
            "          [-0.5287, -1.0504, -2.5758,  ...,  1.1287, -0.9756, -1.2399],\n",
            "          [-0.1439, -0.2531, -1.0258,  ..., -0.1197, -2.2890, -1.3296],\n",
            "          ...,\n",
            "          [ 0.0752, -4.5247, -0.4626,  ..., -1.4075, -2.3589,  0.4374],\n",
            "          [ 0.0752, -4.5247, -0.4626,  ..., -1.4075, -2.3589,  0.4374],\n",
            "          [ 0.0752, -4.5247, -0.4626,  ..., -1.4075, -2.3589,  0.4374]],\n",
            "\n",
            "         [[ 0.0716, -1.8676,  1.8165,  ...,  0.6415, -0.1342, -0.6788],\n",
            "          [ 1.6406, -0.2581, -1.6513,  ..., -1.0369, -3.9917,  1.3927],\n",
            "          [-1.6525, -0.3287,  0.2212,  ...,  0.7428, -0.9865,  2.1311],\n",
            "          ...,\n",
            "          [ 0.0716, -1.8676,  1.8165,  ...,  0.6415, -0.1342, -0.6788],\n",
            "          [ 0.0716, -1.8676,  1.8165,  ...,  0.6415, -0.1342, -0.6788],\n",
            "          [ 0.0716, -1.8676,  1.8165,  ...,  0.6415, -0.1342, -0.6788]],\n",
            "\n",
            "         [[-0.0131, -0.2000, -2.3215,  ..., -0.5908, -1.2056, -0.1269],\n",
            "          [ 0.0865,  1.1614, -2.6137,  ..., -0.1147, -3.1317,  3.1876],\n",
            "          [ 1.1040,  2.9440, -0.6779,  ..., -3.5193, -1.5371,  1.2160],\n",
            "          ...,\n",
            "          [-0.0131, -0.2000, -2.3215,  ..., -0.5908, -1.2056, -0.1269],\n",
            "          [-0.0131, -0.2000, -2.3215,  ..., -0.5908, -1.2056, -0.1269],\n",
            "          [-0.0131, -0.2000, -2.3215,  ..., -0.5908, -1.2056, -0.1269]],\n",
            "\n",
            "         [[-1.0818,  0.1445, -0.6434,  ..., -0.7171,  0.6784,  0.1731],\n",
            "          [-0.2313,  3.5466,  2.5665,  ...,  0.2620,  0.2938, -1.5116],\n",
            "          [ 0.9583,  1.4114, -1.5034,  ...,  1.1955, -0.0332, -0.5989],\n",
            "          ...,\n",
            "          [-1.0818,  0.1445, -0.6434,  ..., -0.7171,  0.6784,  0.1731],\n",
            "          [-1.0818,  0.1445, -0.6434,  ..., -0.7171,  0.6784,  0.1731],\n",
            "          [-1.0818,  0.1445, -0.6434,  ..., -0.7171,  0.6784,  0.1731]]]],\n",
            "       grad_fn=<TransposeBackward0>), tensor([[[[ 1.7883e+01,  9.7104e+00, -1.0794e+01,  ...,  4.1224e-01,\n",
            "           -3.8736e+00,  3.1205e+00],\n",
            "          [ 7.2938e+00,  3.5316e+00, -7.4774e+00,  ..., -9.1610e+00,\n",
            "            9.3541e-01, -4.8960e-01],\n",
            "          [ 9.0460e+00,  8.6715e+00, -1.4367e+01,  ..., -8.0990e+00,\n",
            "           -9.0391e-01,  9.3435e-01],\n",
            "          ...,\n",
            "          [ 1.4204e+01,  2.3648e+00, -4.7089e+00,  ..., -4.9364e+00,\n",
            "           -2.3889e+00,  5.8010e+00],\n",
            "          [ 1.1382e+01,  1.7409e+00, -5.7018e+00,  ..., -1.2976e+01,\n",
            "            5.1270e+00,  4.8243e+00],\n",
            "          [-5.3005e+00, -7.0944e-01,  5.2407e+00,  ..., -4.5604e-02,\n",
            "            5.3652e+00,  1.0112e+00]],\n",
            "\n",
            "         [[-5.7832e+00, -2.7651e+00,  2.2166e+00,  ..., -2.0498e+00,\n",
            "            7.3916e+00, -1.5302e+00],\n",
            "          [ 1.3926e+00, -9.5683e-01,  4.6288e+00,  ..., -7.0187e+00,\n",
            "           -2.6598e+00, -1.8746e-01],\n",
            "          [-4.0938e+00,  2.5330e+00,  2.8960e+00,  ..., -2.6454e+00,\n",
            "           -2.5205e+00, -6.7703e+00],\n",
            "          ...,\n",
            "          [ 4.4943e+00, -1.0336e+01, -4.0287e+00,  ...,  4.5529e-01,\n",
            "           -6.5655e+00, -1.1912e+01],\n",
            "          [-1.8677e+00, -1.1691e+01,  5.3414e-01,  ..., -6.9382e+00,\n",
            "           -9.5518e-01, -5.6936e+00],\n",
            "          [-1.9665e+00, -1.6512e+00, -2.4545e+00,  ...,  2.4725e+00,\n",
            "            9.8637e-01, -3.3197e+00]],\n",
            "\n",
            "         [[ 4.4023e+00,  2.1003e+00, -7.7009e+00,  ...,  7.4546e+00,\n",
            "           -1.8384e+00, -5.5603e+00],\n",
            "          [ 2.6813e+00,  3.4102e+00, -4.0717e+00,  ..., -2.2873e+00,\n",
            "            4.2530e-02,  1.5187e+00],\n",
            "          [ 8.1482e-01, -5.5143e+00,  5.7520e+00,  ...,  2.4271e+00,\n",
            "           -1.8552e+00,  1.6234e+00],\n",
            "          ...,\n",
            "          [ 1.0463e+01,  1.0449e+00,  5.4735e+00,  ..., -1.0293e+00,\n",
            "           -6.3513e+00, -3.9906e+00],\n",
            "          [ 2.5707e+00, -2.0088e-01, -4.8452e+00,  ..., -2.4760e+00,\n",
            "           -1.9808e+00, -5.1379e+00],\n",
            "          [ 1.1081e+00, -2.5276e+00, -2.8108e+00,  ...,  1.7326e+00,\n",
            "           -1.5855e+00,  8.9169e-01]],\n",
            "\n",
            "         [[ 8.3530e+00, -2.1083e+00,  8.9544e-01,  ...,  3.3010e+00,\n",
            "           -3.5691e+00,  3.2784e+00],\n",
            "          [ 1.2104e+01,  3.0291e+00,  1.6199e+01,  ...,  1.5317e+01,\n",
            "            1.5346e+01, -2.3332e+00],\n",
            "          [ 1.1900e+01, -4.4501e+00,  1.4696e+01,  ...,  2.1653e+01,\n",
            "            6.6946e+00, -1.4502e+00],\n",
            "          ...,\n",
            "          [ 9.5168e+00,  2.5522e+00,  5.2081e+00,  ...,  1.1597e+00,\n",
            "           -8.7688e+00,  1.0725e+01],\n",
            "          [ 1.0736e+01, -1.0483e+00,  1.5957e+01,  ...,  1.1811e+01,\n",
            "            1.0824e+01,  2.5198e+00],\n",
            "          [-3.3311e-01, -2.4659e-01, -8.2272e+00,  ..., -9.7054e-01,\n",
            "           -3.1103e+00,  6.1654e-01]]],\n",
            "\n",
            "\n",
            "        [[[ 1.0909e+01, -2.9599e+00, -1.3409e+01,  ..., -2.5635e+00,\n",
            "           -1.7677e+00,  1.0433e+00],\n",
            "          [ 9.8891e+00,  3.8476e+00, -1.4354e+01,  ..., -3.0487e+00,\n",
            "           -4.5541e-01,  3.2348e+00],\n",
            "          [ 6.7047e+00,  5.5994e+00, -1.5541e+01,  ..., -4.5466e+00,\n",
            "            4.4010e+00,  2.4239e-01],\n",
            "          ...,\n",
            "          [ 4.0856e+00, -1.8599e+00, -6.4894e+00,  ..., -1.1756e+01,\n",
            "           -1.7989e+00,  6.1598e+00],\n",
            "          [ 1.1814e+01,  6.2868e+00, -1.6431e-01,  ..., -4.7061e+00,\n",
            "           -1.3694e+00,  5.5612e+00],\n",
            "          [-5.4819e+00, -1.0245e+00,  5.3221e+00,  ...,  1.3409e-03,\n",
            "            5.4661e+00,  9.3821e-01]],\n",
            "\n",
            "         [[-1.1692e+01,  6.0135e+00,  8.8345e+00,  ..., -2.4748e+00,\n",
            "           -9.7773e+00, -6.4972e+00],\n",
            "          [-8.0818e+00, -3.8738e-01,  4.7712e+00,  ..., -1.5230e-01,\n",
            "           -5.2426e+00,  2.6531e+00],\n",
            "          [-9.6720e+00, -3.2427e+00, -1.9241e+00,  ...,  2.8887e+00,\n",
            "           -7.6808e-01,  1.7337e+00],\n",
            "          ...,\n",
            "          [-4.8699e+00,  3.3863e+00,  2.4340e+00,  ...,  4.3641e+00,\n",
            "            1.1782e+00, -3.4182e-01],\n",
            "          [-4.8089e+00, -2.2630e+00, -9.6283e-01,  ...,  5.7251e+00,\n",
            "            6.5838e+00, -4.7125e+00],\n",
            "          [-1.7705e+00, -1.6052e+00, -2.3808e+00,  ...,  2.5262e+00,\n",
            "            1.3249e+00, -3.3678e+00]],\n",
            "\n",
            "         [[ 8.3552e+00, -6.1486e-02,  1.5142e+00,  ...,  6.2995e+00,\n",
            "           -4.0986e+00,  1.9366e+00],\n",
            "          [ 2.9695e+00,  3.1349e+00, -8.4250e-01,  ...,  7.3170e+00,\n",
            "            2.7147e+00, -2.5869e+00],\n",
            "          [ 1.1789e+00,  1.1825e+00,  5.6672e+00,  ...,  7.0165e+00,\n",
            "            7.3050e-02, -7.6409e-01],\n",
            "          ...,\n",
            "          [ 4.4407e+00, -1.1578e+00,  1.0125e+00,  ..., -8.3862e-01,\n",
            "           -6.8363e-01, -3.4769e+00],\n",
            "          [ 7.0699e+00, -9.8075e-02,  1.3331e+00,  ..., -5.9075e+00,\n",
            "           -6.3748e+00, -4.5132e+00],\n",
            "          [ 1.0775e+00, -2.4786e+00, -2.6668e+00,  ...,  1.7641e+00,\n",
            "           -1.7687e+00,  9.2950e-01]],\n",
            "\n",
            "         [[ 8.1852e+00, -1.6076e+00,  1.9708e+00,  ...,  7.0989e-01,\n",
            "            7.8712e-01,  7.6308e+00],\n",
            "          [ 6.2166e+00, -7.7168e-01,  5.5690e+00,  ...,  4.0398e+00,\n",
            "            5.5161e+00,  1.0342e+01],\n",
            "          [ 1.3156e+01, -3.4749e+00,  4.2552e+00,  ...,  1.2621e+01,\n",
            "           -1.5923e-01, -6.1393e+00],\n",
            "          ...,\n",
            "          [ 4.2427e+00,  1.0808e+01,  3.8207e+00,  ...,  1.8681e-01,\n",
            "           -3.7811e+00, -4.4497e-02],\n",
            "          [ 1.0100e+01,  4.1458e+00,  4.4219e+00,  ...,  2.6877e+00,\n",
            "           -1.1642e+00,  4.1685e-01],\n",
            "          [-4.7376e-01, -3.6071e-01, -8.4817e+00,  ..., -1.0292e+00,\n",
            "           -3.2261e+00,  5.6405e-01]]],\n",
            "\n",
            "\n",
            "        [[[ 1.5376e+01,  5.5738e+00, -9.0499e+00,  ..., -2.9529e+00,\n",
            "            6.2675e+00,  1.2296e+01],\n",
            "          [ 9.6369e+00,  9.1479e+00, -1.0465e+01,  ..., -5.5500e-01,\n",
            "           -1.5781e+00,  8.9577e+00],\n",
            "          [ 1.1349e+01,  9.6194e+00, -1.4882e+01,  ...,  8.9945e-01,\n",
            "            1.8600e+00,  8.8312e+00],\n",
            "          ...,\n",
            "          [ 1.9850e+00,  5.6767e+00, -4.0856e+00,  ..., -1.1402e+01,\n",
            "           -6.3609e+00,  1.6406e+00],\n",
            "          [ 1.9850e+00,  5.6767e+00, -4.0856e+00,  ..., -1.1402e+01,\n",
            "           -6.3609e+00,  1.6406e+00],\n",
            "          [ 1.9850e+00,  5.6767e+00, -4.0856e+00,  ..., -1.1402e+01,\n",
            "           -6.3609e+00,  1.6406e+00]],\n",
            "\n",
            "         [[-5.4424e+00, -6.5248e+00,  2.6632e+00,  ...,  1.0516e+00,\n",
            "           -1.4083e+00,  5.4189e+00],\n",
            "          [-3.5158e+00,  1.5753e+00,  1.6813e+00,  ...,  1.6209e+00,\n",
            "            5.4548e+00, -2.5718e+00],\n",
            "          [-1.0485e+00, -2.3108e+00,  8.1341e+00,  ...,  1.7771e+00,\n",
            "            2.3351e+00,  1.2115e+01],\n",
            "          ...,\n",
            "          [-2.1835e+00, -4.9671e-01,  4.0498e+00,  ..., -4.1232e+00,\n",
            "           -1.5723e+00, -5.9722e+00],\n",
            "          [-2.1835e+00, -4.9671e-01,  4.0498e+00,  ..., -4.1232e+00,\n",
            "           -1.5723e+00, -5.9722e+00],\n",
            "          [-2.1835e+00, -4.9671e-01,  4.0498e+00,  ..., -4.1232e+00,\n",
            "           -1.5723e+00, -5.9722e+00]],\n",
            "\n",
            "         [[ 1.6647e-01,  3.9064e+00,  7.4925e+00,  ..., -1.5413e+00,\n",
            "            1.2747e+00, -2.5222e+00],\n",
            "          [ 2.1123e+00,  1.7764e+00,  1.1591e+01,  ..., -3.3632e+00,\n",
            "           -1.9667e+00, -1.7608e+00],\n",
            "          [ 1.6326e+00,  6.1781e-01,  7.7663e+00,  ...,  1.2345e+00,\n",
            "            2.7564e+00, -7.8285e+00],\n",
            "          ...,\n",
            "          [ 5.0416e+00, -1.5682e-01, -3.0354e+00,  ..., -1.7726e+00,\n",
            "           -8.6584e-01, -1.7233e+00],\n",
            "          [ 5.0416e+00, -1.5682e-01, -3.0354e+00,  ..., -1.7726e+00,\n",
            "           -8.6584e-01, -1.7233e+00],\n",
            "          [ 5.0416e+00, -1.5682e-01, -3.0354e+00,  ..., -1.7726e+00,\n",
            "           -8.6584e-01, -1.7233e+00]],\n",
            "\n",
            "         [[ 5.1640e+00, -1.7549e+00,  2.6862e+00,  ..., -3.9568e+00,\n",
            "           -3.5897e-01, -6.3975e-01],\n",
            "          [ 6.9517e+00, -3.4907e-01,  8.2874e+00,  ...,  8.3022e-02,\n",
            "            1.4864e+01,  2.8956e+00],\n",
            "          [ 9.1752e+00,  2.2730e+00,  3.4761e+00,  ..., -2.7885e+00,\n",
            "            1.3882e+01,  6.2322e+00],\n",
            "          ...,\n",
            "          [ 6.2017e+00,  1.7162e+00,  1.0802e+01,  ..., -3.9302e-01,\n",
            "            3.9034e+00, -1.1563e-01],\n",
            "          [ 6.2017e+00,  1.7162e+00,  1.0802e+01,  ..., -3.9302e-01,\n",
            "            3.9034e+00, -1.1563e-01],\n",
            "          [ 6.2017e+00,  1.7162e+00,  1.0802e+01,  ..., -3.9302e-01,\n",
            "            3.9034e+00, -1.1563e-01]]],\n",
            "\n",
            "\n",
            "        ...,\n",
            "\n",
            "\n",
            "        [[[ 1.7081e+01,  4.5859e+00, -4.6951e+00,  ...,  5.5568e+00,\n",
            "           -3.2127e+00, -2.5696e+00],\n",
            "          [ 1.5350e+01, -2.4859e-01, -2.1237e+00,  ...,  2.0437e+00,\n",
            "           -1.6597e+00,  3.4248e+00],\n",
            "          [ 1.9435e+01,  5.7574e+00, -4.9927e+00,  ...,  7.9703e+00,\n",
            "           -1.9639e+00,  1.1720e+01],\n",
            "          ...,\n",
            "          [ 2.6194e+00,  5.2345e+00, -2.0368e+00,  ..., -9.3669e+00,\n",
            "           -6.3508e+00,  7.2749e-01],\n",
            "          [ 2.6194e+00,  5.2345e+00, -2.0368e+00,  ..., -9.3669e+00,\n",
            "           -6.3508e+00,  7.2749e-01],\n",
            "          [ 2.6194e+00,  5.2345e+00, -2.0368e+00,  ..., -9.3669e+00,\n",
            "           -6.3508e+00,  7.2749e-01]],\n",
            "\n",
            "         [[-4.5291e+00, -2.8401e+00,  7.4725e+00,  ..., -5.8820e+00,\n",
            "           -4.3785e+00, -9.4559e+00],\n",
            "          [ 4.5124e+00,  1.7421e+00, -3.9896e-01,  ...,  3.2959e+00,\n",
            "            3.4024e+00, -6.4227e+00],\n",
            "          [-2.7462e+00,  8.4251e+00, -5.9954e-01,  ...,  3.3712e+00,\n",
            "           -1.0331e+01, -6.8028e+00],\n",
            "          ...,\n",
            "          [ 4.9660e-01,  8.9760e-02,  2.9794e+00,  ..., -5.9578e+00,\n",
            "           -2.1738e+00, -4.6608e+00],\n",
            "          [ 4.9660e-01,  8.9760e-02,  2.9794e+00,  ..., -5.9578e+00,\n",
            "           -2.1738e+00, -4.6608e+00],\n",
            "          [ 4.9660e-01,  8.9760e-02,  2.9794e+00,  ..., -5.9578e+00,\n",
            "           -2.1738e+00, -4.6608e+00]],\n",
            "\n",
            "         [[ 5.2861e+00, -2.1285e+00, -4.1851e+00,  ...,  3.8269e+00,\n",
            "            2.2234e+00, -6.2959e+00],\n",
            "          [ 5.7712e+00, -1.8970e+00,  2.2676e+00,  ...,  3.5887e+00,\n",
            "            2.3240e+00,  3.3692e+00],\n",
            "          [-3.3028e-01, -5.4452e+00,  5.0079e+00,  ...,  2.6529e+00,\n",
            "            2.5045e+00, -6.8029e+00],\n",
            "          ...,\n",
            "          [ 5.4932e+00, -2.4054e-01, -2.4422e+00,  ..., -1.6477e+00,\n",
            "            4.5045e-01, -1.6284e+00],\n",
            "          [ 5.4932e+00, -2.4054e-01, -2.4422e+00,  ..., -1.6477e+00,\n",
            "            4.5045e-01, -1.6284e+00],\n",
            "          [ 5.4932e+00, -2.4054e-01, -2.4422e+00,  ..., -1.6477e+00,\n",
            "            4.5045e-01, -1.6284e+00]],\n",
            "\n",
            "         [[ 8.3296e+00, -2.1787e+00,  2.6323e+00,  ..., -2.8103e+00,\n",
            "            5.1651e+00,  2.5435e+00],\n",
            "          [ 6.6430e+00, -8.6390e+00,  1.1338e+01,  ..., -3.0850e+00,\n",
            "            1.9372e+01, -6.8309e+00],\n",
            "          [ 1.2752e+01, -4.4744e+00, -1.9299e+00,  ...,  6.2944e+00,\n",
            "            2.1078e+01, -1.6487e+01],\n",
            "          ...,\n",
            "          [ 4.3348e+00,  2.7879e+00,  8.0777e+00,  ..., -3.7036e-01,\n",
            "            5.6149e+00, -7.1744e-01],\n",
            "          [ 4.3348e+00,  2.7879e+00,  8.0777e+00,  ..., -3.7036e-01,\n",
            "            5.6149e+00, -7.1744e-01],\n",
            "          [ 4.3348e+00,  2.7879e+00,  8.0777e+00,  ..., -3.7036e-01,\n",
            "            5.6149e+00, -7.1744e-01]]],\n",
            "\n",
            "\n",
            "        [[[ 1.4066e+01,  1.0408e+01, -1.4663e+01,  ..., -3.2001e+00,\n",
            "           -1.3015e+00, -3.9847e+00],\n",
            "          [ 1.7214e+01,  5.5982e+00, -9.5281e+00,  ..., -4.7450e+00,\n",
            "           -4.5169e+00, -3.0231e+00],\n",
            "          [ 1.3140e+01,  6.2782e+00, -1.4323e+01,  ...,  4.6088e-01,\n",
            "           -1.3495e+00, -5.4037e+00],\n",
            "          ...,\n",
            "          [ 1.0645e+01, -1.4446e-01, -7.1962e+00,  ..., -5.5153e+00,\n",
            "           -3.0482e+00,  8.6774e+00],\n",
            "          [ 7.3791e+00,  3.9813e+00, -2.1690e+00,  ..., -1.1663e+01,\n",
            "           -4.1479e-01,  1.3416e+01],\n",
            "          [-5.4913e+00, -8.0471e-01,  5.2803e+00,  ..., -2.1005e-02,\n",
            "            5.1587e+00,  1.1149e+00]],\n",
            "\n",
            "         [[-7.1610e+00,  5.1980e+00, -1.8649e-01,  ..., -5.9539e-01,\n",
            "            2.2862e+00, -9.3996e-01],\n",
            "          [-4.1602e+00, -2.8401e+00,  1.7291e+00,  ...,  3.9470e+00,\n",
            "            5.5387e+00, -1.2860e+00],\n",
            "          [-1.8848e+00, -6.5319e+00, -2.0119e+00,  ..., -4.4190e+00,\n",
            "            9.4763e-01,  3.1849e+00],\n",
            "          ...,\n",
            "          [-2.2657e+00,  9.8087e-01, -3.4770e+00,  ...,  1.9629e+00,\n",
            "           -9.0694e+00, -8.3773e-01],\n",
            "          [ 1.0231e+00, -4.3426e+00, -7.1354e+00,  ..., -2.9490e+00,\n",
            "            3.1112e+00,  4.7614e+00],\n",
            "          [-1.9645e+00, -1.6681e+00, -2.1737e+00,  ...,  2.2847e+00,\n",
            "            1.3581e+00, -3.2574e+00]],\n",
            "\n",
            "         [[ 1.0711e+00, -1.8254e+00,  2.8164e-01,  ..., -4.9918e+00,\n",
            "           -7.3579e+00, -6.8510e+00],\n",
            "          [ 1.4198e+00, -7.5562e-01, -5.0767e+00,  ...,  2.6532e+00,\n",
            "            3.2074e+00, -2.1175e+00],\n",
            "          [-1.4110e+00,  1.3197e+00,  3.7169e-01,  ...,  5.4505e-01,\n",
            "            4.3444e+00, -1.2931e+00],\n",
            "          ...,\n",
            "          [ 4.1958e+00, -1.5709e+00,  1.2051e+00,  ...,  7.9399e-01,\n",
            "            5.3630e+00, -8.1293e-01],\n",
            "          [ 4.2774e+00, -1.3821e+00,  4.1020e+00,  ..., -4.3992e+00,\n",
            "            2.1803e+00, -3.2568e+00],\n",
            "          [ 1.0745e+00, -2.2722e+00, -2.7772e+00,  ...,  1.6738e+00,\n",
            "           -1.6932e+00,  8.8965e-01]],\n",
            "\n",
            "         [[ 1.0175e+01, -5.6043e+00,  7.8504e+00,  ..., -3.5967e+00,\n",
            "           -4.5565e+00,  2.7735e+00],\n",
            "          [ 2.0391e+01, -6.6179e+00,  5.7312e+00,  ..., -8.1938e+00,\n",
            "            4.1238e+00,  6.7992e+00],\n",
            "          [ 6.6240e+00,  2.0872e+00,  1.0988e+01,  ...,  6.5388e+00,\n",
            "            6.8644e+00, -2.5138e+00],\n",
            "          ...,\n",
            "          [ 1.0560e+01,  3.2067e+00, -3.8865e-01,  ...,  4.1534e+00,\n",
            "            2.2846e+00,  1.3107e+01],\n",
            "          [-1.7074e+00,  3.0965e+00,  2.4292e+00,  ...,  8.3514e-01,\n",
            "            8.5502e+00,  1.6769e+00],\n",
            "          [-7.8853e-01, -1.6219e-01, -8.3327e+00,  ..., -1.0203e+00,\n",
            "           -3.1419e+00,  8.7621e-01]]],\n",
            "\n",
            "\n",
            "        [[[ 8.3950e+00,  4.3116e+00,  1.8629e+00,  ..., -4.3426e-01,\n",
            "           -7.6229e-01, -4.4589e+00],\n",
            "          [ 1.0232e+01,  1.3548e+00, -1.3603e+00,  ..., -1.1807e+01,\n",
            "            4.4209e+00, -1.3538e+00],\n",
            "          [ 1.3514e+01,  2.3261e+00, -1.2169e-01,  ..., -8.1927e+00,\n",
            "           -8.9950e+00, -1.9219e+00],\n",
            "          ...,\n",
            "          [ 3.6582e+00,  1.7216e+00, -3.8117e+00,  ..., -7.1892e+00,\n",
            "           -4.5883e+00,  1.8287e+00],\n",
            "          [ 3.6713e+00,  1.7429e+00, -3.8427e+00,  ..., -7.1842e+00,\n",
            "           -4.5429e+00,  1.8191e+00],\n",
            "          [ 3.6715e+00,  1.6955e+00, -3.9912e+00,  ..., -6.8319e+00,\n",
            "           -4.3784e+00,  1.7499e+00]],\n",
            "\n",
            "         [[-7.1970e+00, -4.0607e+00,  1.6872e+00,  ...,  9.3521e-02,\n",
            "            4.6619e+00, -1.0886e+01],\n",
            "          [ 2.2608e-01, -5.5755e+00, -6.2330e+00,  ..., -6.4646e-01,\n",
            "           -1.0962e+00, -5.6000e+00],\n",
            "          [ 5.1988e-01,  2.9721e+00, -7.5572e-01,  ..., -2.1697e-01,\n",
            "            6.1264e+00, -8.6634e+00],\n",
            "          ...,\n",
            "          [-2.3035e+00,  1.7893e-01,  3.6705e+00,  ..., -5.9023e+00,\n",
            "           -1.1362e+00, -7.3326e+00],\n",
            "          [-2.3262e+00,  1.9749e-01,  3.6741e+00,  ..., -5.9117e+00,\n",
            "           -1.1466e+00, -7.3430e+00],\n",
            "          [-2.3580e+00, -4.7432e-02,  3.5964e+00,  ..., -5.8232e+00,\n",
            "           -1.1848e+00, -7.3052e+00]],\n",
            "\n",
            "         [[ 3.4264e+00, -4.2195e+00,  9.1797e-01,  ...,  5.4450e-01,\n",
            "           -1.3760e+00,  3.1420e-01],\n",
            "          [ 3.4057e+00, -3.3540e+00,  6.0788e+00,  ..., -5.0664e-01,\n",
            "            4.9705e+00, -1.2970e-01],\n",
            "          [ 4.3111e+00, -3.5140e+00,  4.3078e+00,  ...,  4.7304e+00,\n",
            "            2.1142e+00,  1.4442e+00],\n",
            "          ...,\n",
            "          [ 6.0068e+00, -3.4323e-01, -3.7180e+00,  ..., -9.2226e-01,\n",
            "            2.6027e-01, -2.2316e+00],\n",
            "          [ 5.9921e+00, -3.5859e-01, -3.7029e+00,  ..., -9.4519e-01,\n",
            "            2.7715e-01, -2.2662e+00],\n",
            "          [ 5.7217e+00, -3.2273e-01, -3.6858e+00,  ..., -8.2673e-01,\n",
            "            2.0266e-01, -2.2970e+00]],\n",
            "\n",
            "         [[ 2.6069e+00, -5.4845e+00,  7.1315e-01,  ..., -3.2968e+00,\n",
            "            1.0868e+00, -5.5159e-01],\n",
            "          [ 7.4885e+00, -1.6708e+00, -1.4709e+00,  ..., -2.8942e+00,\n",
            "           -3.6487e-01, -2.9671e+00],\n",
            "          [ 9.9952e+00, -7.1640e+00,  7.1228e+00,  ..., -1.6881e+00,\n",
            "            6.6840e+00, -7.1667e+00],\n",
            "          ...,\n",
            "          [ 4.2808e+00,  4.3325e+00,  1.0587e+01,  ..., -1.0868e+00,\n",
            "            4.8062e+00, -7.7587e-01],\n",
            "          [ 4.3248e+00,  4.2983e+00,  1.0630e+01,  ..., -1.0974e+00,\n",
            "            4.7445e+00, -7.5153e-01],\n",
            "          [ 4.1280e+00,  4.0364e+00,  1.0499e+01,  ..., -1.4638e+00,\n",
            "            4.7454e+00, -1.1538e+00]]]], grad_fn=<TransposeBackward0>), tensor([[[[-3.9503e+00,  4.2314e+00, -1.9209e+00,  ...,  3.2615e+00,\n",
            "            2.7986e+00,  3.1497e+00],\n",
            "          [-1.3399e+00,  3.5699e-01, -7.8934e+00,  ..., -6.1327e+00,\n",
            "            4.6439e-02,  8.5078e+00],\n",
            "          [-2.4942e+00,  4.0035e+00, -3.8413e+00,  ..., -2.5910e+00,\n",
            "            3.9987e+00,  6.0857e+00],\n",
            "          ...,\n",
            "          [-5.3196e-01, -4.1698e+00,  2.8302e+00,  ...,  3.2336e+00,\n",
            "            4.4554e-02,  3.4295e+00],\n",
            "          [-8.2697e-01, -5.4884e+00, -2.0566e+00,  ..., -2.6416e-01,\n",
            "            7.9758e+00,  7.3138e+00],\n",
            "          [-1.2007e+00, -3.2459e-01, -3.4484e-01,  ..., -2.2429e-01,\n",
            "            1.7131e+00,  3.1077e-01]],\n",
            "\n",
            "         [[-2.6488e+00, -3.3196e-01,  5.6013e-01,  ..., -1.8459e+00,\n",
            "           -8.6554e-01, -2.0520e+00],\n",
            "          [ 1.9630e+00, -1.5959e+00, -1.7767e+00,  ..., -2.1381e+00,\n",
            "           -1.1125e+00,  1.6924e+00],\n",
            "          [ 1.8443e+00, -7.8372e+00, -2.0630e-01,  ...,  3.0443e+00,\n",
            "           -5.1018e+00,  3.8126e-01],\n",
            "          ...,\n",
            "          [ 4.0068e-01,  3.4742e-01,  6.6301e-02,  ..., -2.1126e+00,\n",
            "            5.1761e+00,  6.7982e+00],\n",
            "          [-1.1257e+00,  2.5536e+00,  1.5038e+00,  ..., -9.0972e+00,\n",
            "           -5.2411e+00, -5.2508e+00],\n",
            "          [-2.3954e+00,  2.1101e+00, -7.0023e-01,  ..., -1.5764e+00,\n",
            "            9.3732e-01,  2.3269e+00]],\n",
            "\n",
            "         [[ 7.7884e+00, -4.0449e+00,  7.9068e+00,  ...,  1.5242e-01,\n",
            "            1.5478e+00,  4.5138e+00],\n",
            "          [ 4.5570e+00, -1.0649e+01, -2.8747e-01,  ...,  6.2009e+00,\n",
            "           -9.0407e+00,  4.6513e+00],\n",
            "          [ 6.2509e+00, -7.6278e+00, -2.6172e+00,  ...,  9.1881e+00,\n",
            "           -3.6753e+00,  8.1322e+00],\n",
            "          ...,\n",
            "          [-1.5708e+00, -8.5226e-01, -2.9569e-01,  ...,  8.8544e-01,\n",
            "           -2.4493e+00, -6.3532e+00],\n",
            "          [-5.6368e+00,  9.0939e-01, -1.2697e-01,  ...,  3.6360e+00,\n",
            "           -7.9353e-01, -7.6553e+00],\n",
            "          [-3.9628e-01, -5.2676e-02,  8.4966e-01,  ..., -1.6549e-01,\n",
            "           -1.5626e+00, -2.4617e+00]],\n",
            "\n",
            "         [[ 2.0780e+00, -3.1924e+00, -9.4377e-01,  ...,  4.3130e+00,\n",
            "           -2.3983e+00, -3.3686e+00],\n",
            "          [ 3.1636e+00,  1.6606e+00, -1.6779e+00,  ...,  1.3916e+00,\n",
            "            9.1536e-02,  1.4824e+00],\n",
            "          [ 4.5869e+00,  2.9099e+00,  2.1934e+00,  ...,  1.5559e+00,\n",
            "            2.5059e+00, -7.0130e-01],\n",
            "          ...,\n",
            "          [ 7.8204e-01,  2.2358e+00, -4.6263e+00,  ...,  7.0455e+00,\n",
            "            2.5646e+00, -2.4481e+00],\n",
            "          [ 6.2238e+00,  2.8534e+00,  1.1645e+00,  ...,  3.7627e+00,\n",
            "           -2.6959e+00, -1.1780e+00],\n",
            "          [-3.9359e-01,  1.1263e+00,  1.1199e+00,  ...,  2.3685e+00,\n",
            "            8.2541e-01,  4.0321e-01]]],\n",
            "\n",
            "\n",
            "        [[[ 2.6612e+00,  2.3938e+00, -6.8199e+00,  ...,  1.9412e+00,\n",
            "            8.3310e+00,  6.8338e-01],\n",
            "          [ 3.1996e+00, -5.4274e-01, -3.7218e+00,  ..., -1.0376e+00,\n",
            "            9.7931e+00,  9.6364e-01],\n",
            "          [ 1.0061e+01,  1.4242e+00,  5.6498e+00,  ..., -5.1900e+00,\n",
            "            3.2211e+00,  2.8426e+00],\n",
            "          ...,\n",
            "          [ 6.8142e-01, -5.1077e+00, -3.9567e+00,  ..., -1.1773e+00,\n",
            "           -2.1680e+00,  7.3841e-01],\n",
            "          [-2.8068e-01, -5.3408e+00,  3.1017e+00,  ..., -2.4003e+00,\n",
            "            9.9739e-01,  3.2773e+00],\n",
            "          [-1.2034e+00, -1.5082e-01, -2.2980e-01,  ..., -1.9831e-01,\n",
            "            1.3515e+00,  3.9363e-02]],\n",
            "\n",
            "         [[ 1.8206e+00,  2.5414e+00, -3.6976e+00,  ...,  2.7001e+00,\n",
            "            2.4846e+00,  1.0747e+01],\n",
            "          [ 7.3818e+00,  4.0495e+00, -3.2185e+00,  ..., -1.1897e+00,\n",
            "            4.7546e+00,  8.1888e+00],\n",
            "          [ 4.0154e+00,  4.8340e+00, -1.7087e+00,  ..., -8.9178e-03,\n",
            "            6.4258e+00,  8.6097e+00],\n",
            "          ...,\n",
            "          [ 2.6182e+00,  2.3911e+00,  4.6551e-01,  ..., -2.3451e+00,\n",
            "            1.3952e+00,  7.8777e+00],\n",
            "          [ 1.2875e+00, -2.9860e+00,  1.0044e+01,  ..., -3.6483e+00,\n",
            "            3.8819e+00,  3.9591e+00],\n",
            "          [-2.3435e+00,  2.2145e+00, -8.0716e-01,  ..., -1.2555e+00,\n",
            "            1.2370e+00,  2.5838e+00]],\n",
            "\n",
            "         [[-7.9737e+00,  1.1188e+00, -2.8626e-01,  ..., -1.6768e+00,\n",
            "           -9.2669e-01,  1.5178e+00],\n",
            "          [-1.4589e+00,  3.7824e+00, -4.8823e+00,  ..., -3.8658e+00,\n",
            "           -8.2318e+00, -2.6298e+00],\n",
            "          [ 4.1751e+00, -6.3472e+00, -1.1110e+01,  ..., -2.3206e+00,\n",
            "           -8.7750e+00,  2.1777e+00],\n",
            "          ...,\n",
            "          [-2.2140e+00,  6.1910e-01,  1.4553e+00,  ..., -1.1259e+00,\n",
            "           -2.2509e-01, -7.7193e+00],\n",
            "          [-1.8898e+00,  5.7158e+00, -3.8366e+00,  ..., -2.1616e+00,\n",
            "            5.3646e+00, -7.8163e+00],\n",
            "          [-4.4617e-01, -1.2430e-01,  7.2484e-01,  ..., -2.5304e-01,\n",
            "           -1.5275e+00, -2.2636e+00]],\n",
            "\n",
            "         [[ 2.5531e+00,  3.1221e+00, -3.0360e+00,  ...,  2.9351e+00,\n",
            "           -9.6761e-01, -8.8536e-01],\n",
            "          [ 2.3979e+00, -2.3748e-01,  5.1671e-01,  ...,  6.7165e+00,\n",
            "           -4.6951e+00,  1.4919e-01],\n",
            "          [-3.2376e-01,  2.3634e+00, -1.3919e+00,  ...,  5.7325e+00,\n",
            "           -9.4615e-01,  1.2275e+00],\n",
            "          ...,\n",
            "          [-1.1390e+00,  1.9043e+00, -1.4270e+00,  ...,  2.2895e+00,\n",
            "            4.8539e+00, -2.7541e+00],\n",
            "          [-1.0174e+00, -1.0113e+00,  1.6631e+00,  ...,  1.9996e+00,\n",
            "           -3.4161e+00,  3.7472e+00],\n",
            "          [-3.4927e-01,  1.0170e+00,  1.0833e+00,  ...,  2.4097e+00,\n",
            "            9.2909e-01,  4.3730e-01]]],\n",
            "\n",
            "\n",
            "        [[[ 2.0743e+00, -7.7324e+00,  1.2028e+00,  ..., -4.9245e+00,\n",
            "           -6.7717e+00,  6.8268e+00],\n",
            "          [-2.3483e-01, -1.9265e+00, -5.9001e-01,  ..., -4.5391e+00,\n",
            "           -1.1823e+01, -5.6979e+00],\n",
            "          [-2.9115e+00,  1.4581e+00,  2.1931e+00,  ..., -4.6414e+00,\n",
            "           -8.6883e+00, -4.7059e+00],\n",
            "          ...,\n",
            "          [-1.2854e+00,  2.4260e+00, -1.7872e+00,  ...,  1.8895e+00,\n",
            "            1.6182e+00,  4.0329e+00],\n",
            "          [-1.2854e+00,  2.4260e+00, -1.7872e+00,  ...,  1.8895e+00,\n",
            "            1.6182e+00,  4.0329e+00],\n",
            "          [-1.2854e+00,  2.4260e+00, -1.7872e+00,  ...,  1.8895e+00,\n",
            "            1.6182e+00,  4.0329e+00]],\n",
            "\n",
            "         [[ 5.6104e+00,  1.9350e+00,  2.1168e+00,  ..., -3.8939e-01,\n",
            "            7.0831e-01, -1.8776e+00],\n",
            "          [ 4.4396e+00, -1.1494e+00,  8.9760e-01,  ..., -1.1302e+00,\n",
            "            2.1935e+00,  1.5170e+00],\n",
            "          [ 7.8283e+00,  2.6892e+00,  4.3145e+00,  ..., -4.4277e+00,\n",
            "            4.0783e+00,  2.0054e+00],\n",
            "          ...,\n",
            "          [ 2.3871e+00, -1.3739e+00, -6.7325e-01,  ..., -2.9515e+00,\n",
            "           -6.4931e-01, -3.1307e-01],\n",
            "          [ 2.3871e+00, -1.3739e+00, -6.7325e-01,  ..., -2.9515e+00,\n",
            "           -6.4931e-01, -3.1307e-01],\n",
            "          [ 2.3871e+00, -1.3739e+00, -6.7325e-01,  ..., -2.9515e+00,\n",
            "           -6.4931e-01, -3.1307e-01]],\n",
            "\n",
            "         [[ 3.7315e-01, -1.1018e+00, -3.3884e+00,  ...,  3.2654e+00,\n",
            "           -1.5053e+01, -6.7114e-02],\n",
            "          [-1.6429e-01, -5.6293e+00, -7.0977e+00,  ...,  9.2948e+00,\n",
            "           -1.1937e+01,  5.2805e-01],\n",
            "          [-7.9734e-01, -9.3045e+00, -9.4914e+00,  ...,  3.4082e+00,\n",
            "           -1.7045e+01,  8.9049e-01],\n",
            "          ...,\n",
            "          [-6.5711e+00, -1.9781e+00,  7.1272e+00,  ..., -1.0950e+00,\n",
            "            1.7841e+00,  3.5013e-01],\n",
            "          [-6.5711e+00, -1.9781e+00,  7.1272e+00,  ..., -1.0950e+00,\n",
            "            1.7841e+00,  3.5013e-01],\n",
            "          [-6.5711e+00, -1.9781e+00,  7.1272e+00,  ..., -1.0950e+00,\n",
            "            1.7841e+00,  3.5013e-01]],\n",
            "\n",
            "         [[ 1.5196e-01, -5.2211e+00,  3.5681e+00,  ...,  5.7438e+00,\n",
            "           -3.9660e+00, -2.3333e+00],\n",
            "          [-1.3847e+00, -2.7154e+00,  1.9442e+00,  ...,  2.3338e+00,\n",
            "            3.3023e+00, -2.2884e+00],\n",
            "          [-7.6249e-01, -5.8557e+00,  1.2694e+00,  ..., -2.5610e-01,\n",
            "            4.2333e+00, -2.5077e+00],\n",
            "          ...,\n",
            "          [-9.5545e-01,  1.5858e+00,  2.1442e+00,  ...,  2.5119e+00,\n",
            "            9.0494e-01,  8.0204e-01],\n",
            "          [-9.5545e-01,  1.5858e+00,  2.1442e+00,  ...,  2.5119e+00,\n",
            "            9.0494e-01,  8.0204e-01],\n",
            "          [-9.5545e-01,  1.5858e+00,  2.1442e+00,  ...,  2.5119e+00,\n",
            "            9.0494e-01,  8.0204e-01]]],\n",
            "\n",
            "\n",
            "        ...,\n",
            "\n",
            "\n",
            "        [[[ 3.4964e-01,  2.0324e+00,  7.3684e-01,  ...,  3.2924e+00,\n",
            "            3.7063e+00,  1.8282e+00],\n",
            "          [ 1.8989e+00,  4.0136e+00, -3.7327e+00,  ..., -7.6870e+00,\n",
            "           -5.9348e+00,  5.3792e+00],\n",
            "          [ 8.5107e-01,  4.7283e+00, -4.0715e+00,  ..., -5.4756e+00,\n",
            "           -5.0440e+00,  2.2040e+00],\n",
            "          ...,\n",
            "          [-1.3145e+00,  1.4956e+00, -1.2683e+00,  ...,  1.0484e+00,\n",
            "            3.0285e+00,  3.7203e+00],\n",
            "          [-1.3145e+00,  1.4956e+00, -1.2683e+00,  ...,  1.0484e+00,\n",
            "            3.0285e+00,  3.7203e+00],\n",
            "          [-1.3145e+00,  1.4956e+00, -1.2683e+00,  ...,  1.0484e+00,\n",
            "            3.0285e+00,  3.7203e+00]],\n",
            "\n",
            "         [[-3.1209e+00,  2.7240e+00, -5.4590e-02,  ...,  2.6094e+00,\n",
            "            5.1622e+00,  2.3468e+00],\n",
            "          [-1.5815e-01,  1.6224e+00,  3.9303e+00,  ...,  9.9253e-03,\n",
            "            6.5706e+00, -2.4120e-01],\n",
            "          [-3.9627e-01,  5.6757e+00,  2.6677e+00,  ...,  6.5073e+00,\n",
            "            8.4724e+00,  2.2968e+00],\n",
            "          ...,\n",
            "          [ 4.2234e+00, -9.8216e-01, -1.0513e+00,  ..., -4.5304e-01,\n",
            "           -1.7163e-02,  5.4387e-02],\n",
            "          [ 4.2234e+00, -9.8216e-01, -1.0513e+00,  ..., -4.5304e-01,\n",
            "           -1.7163e-02,  5.4387e-02],\n",
            "          [ 4.2234e+00, -9.8216e-01, -1.0513e+00,  ..., -4.5304e-01,\n",
            "           -1.7163e-02,  5.4387e-02]],\n",
            "\n",
            "         [[-8.4629e-01, -1.5571e+00, -2.2827e+00,  ..., -3.0158e+00,\n",
            "           -1.1395e+01, -8.3660e+00],\n",
            "          [ 1.0846e+00, -3.6672e+00, -1.0243e-01,  ..., -5.4944e-01,\n",
            "           -5.9884e+00, -7.2631e+00],\n",
            "          [-5.1073e-01,  1.6921e+00,  9.0717e+00,  ..., -5.7289e+00,\n",
            "           -1.0618e+01, -7.3528e+00],\n",
            "          ...,\n",
            "          [-3.7067e+00, -2.2342e+00,  5.8150e+00,  ..., -1.5002e+00,\n",
            "           -4.4378e-01, -2.6678e-01],\n",
            "          [-3.7067e+00, -2.2342e+00,  5.8150e+00,  ..., -1.5002e+00,\n",
            "           -4.4378e-01, -2.6678e-01],\n",
            "          [-3.7067e+00, -2.2342e+00,  5.8150e+00,  ..., -1.5002e+00,\n",
            "           -4.4378e-01, -2.6678e-01]],\n",
            "\n",
            "         [[ 3.1181e+00, -2.9550e+00,  3.2565e+00,  ...,  2.1876e-01,\n",
            "           -3.3318e+00,  2.7742e-02],\n",
            "          [ 1.9793e+00,  1.7477e+00,  8.8717e-01,  ...,  3.6107e+00,\n",
            "           -5.3751e+00,  1.1128e+00],\n",
            "          [ 3.1676e-01, -4.7639e+00,  4.6143e+00,  ...,  2.8399e+00,\n",
            "           -8.8663e+00, -1.5787e+00],\n",
            "          ...,\n",
            "          [-1.8442e+00,  1.5623e+00,  3.0034e+00,  ...,  2.7859e+00,\n",
            "           -8.5600e-01,  5.1908e-01],\n",
            "          [-1.8442e+00,  1.5623e+00,  3.0034e+00,  ...,  2.7859e+00,\n",
            "           -8.5600e-01,  5.1908e-01],\n",
            "          [-1.8442e+00,  1.5623e+00,  3.0034e+00,  ...,  2.7859e+00,\n",
            "           -8.5600e-01,  5.1908e-01]]],\n",
            "\n",
            "\n",
            "        [[[-4.7826e+00,  4.4957e+00,  2.7531e+00,  ...,  1.9498e-01,\n",
            "            3.6668e+00,  8.8641e-01],\n",
            "          [-2.1548e+00,  6.1331e+00,  1.1343e+00,  ...,  1.2547e+00,\n",
            "           -4.4211e-01,  4.3589e+00],\n",
            "          [-4.0584e+00,  7.8497e+00, -5.0162e+00,  ..., -3.5173e+00,\n",
            "            2.7601e-01,  2.4124e+00],\n",
            "          ...,\n",
            "          [ 1.0674e+00,  2.0057e+00, -2.5622e+00,  ..., -1.3637e+00,\n",
            "            3.2062e+00,  6.5685e+00],\n",
            "          [ 1.0422e+01,  6.4485e+00,  1.1829e+00,  ..., -2.9328e+00,\n",
            "            3.4265e+00, -4.5219e+00],\n",
            "          [-9.9368e-01, -2.3330e-01, -3.7960e-01,  ...,  1.8117e-01,\n",
            "            1.3474e+00,  8.9592e-02]],\n",
            "\n",
            "         [[-4.3259e+00,  1.6747e+00,  4.1348e+00,  ..., -1.9160e-01,\n",
            "           -5.3964e+00, -3.5968e+00],\n",
            "          [-2.8692e+00,  2.4878e-01,  1.7952e+00,  ..., -2.2012e+00,\n",
            "           -3.4923e+00, -5.2354e+00],\n",
            "          [-5.6175e-01, -3.6834e+00,  8.1834e+00,  ...,  2.6936e+00,\n",
            "            3.4010e+00, -4.0487e+00],\n",
            "          ...,\n",
            "          [ 5.7574e+00,  3.9984e+00,  2.7589e+00,  ..., -4.9577e+00,\n",
            "            2.3883e+00, -4.6692e-01],\n",
            "          [ 2.9546e+00,  2.0276e+00,  1.3080e+00,  ..., -1.6344e+00,\n",
            "           -2.1162e+00,  4.3710e+00],\n",
            "          [-2.4791e+00,  2.2881e+00, -7.6416e-01,  ..., -1.3838e+00,\n",
            "            1.1584e+00,  2.6109e+00]],\n",
            "\n",
            "         [[-7.3609e-01, -5.0099e+00, -3.8645e+00,  ..., -2.4142e+00,\n",
            "            4.4565e-01,  4.4326e+00],\n",
            "          [ 6.4380e-01, -7.8712e+00,  3.1646e+00,  ..., -1.3910e-01,\n",
            "            1.2060e-01,  8.2058e+00],\n",
            "          [ 3.5760e+00, -8.8333e+00, -1.2235e+00,  ...,  6.0914e+00,\n",
            "           -7.5303e-01,  1.7000e+01],\n",
            "          ...,\n",
            "          [-1.1171e+00, -7.8392e+00, -3.6593e+00,  ...,  1.0738e+01,\n",
            "            2.0718e+00,  4.1331e+00],\n",
            "          [-3.8866e+00, -5.5265e-01, -1.0657e+01,  ...,  1.3856e+01,\n",
            "            1.7379e+00, -6.5135e-01],\n",
            "          [-4.1391e-01, -1.2691e-01,  7.3737e-01,  ..., -8.5786e-02,\n",
            "           -1.6000e+00, -2.1438e+00]],\n",
            "\n",
            "         [[-2.1096e+00, -6.0747e-01,  1.8464e+00,  ..., -2.9133e+00,\n",
            "           -6.4915e+00, -6.1645e+00],\n",
            "          [ 2.9426e+00, -1.3208e+00,  4.1714e+00,  ..., -2.8203e+00,\n",
            "           -7.1395e-01, -2.0528e+00],\n",
            "          [ 4.4815e+00,  4.2602e-01,  5.6422e+00,  ...,  6.5104e-01,\n",
            "            9.8596e-01,  4.1148e-01],\n",
            "          ...,\n",
            "          [-9.7602e-01,  7.4659e+00, -5.8812e+00,  ...,  1.2356e-01,\n",
            "            3.6257e+00, -1.1504e+00],\n",
            "          [ 1.2038e+00,  7.3026e-01,  4.4792e-01,  ...,  4.7720e+00,\n",
            "            3.8529e+00,  1.2090e+00],\n",
            "          [-2.5024e-01,  9.3543e-01,  9.9057e-01,  ...,  2.1822e+00,\n",
            "            9.7967e-01,  4.5177e-01]]],\n",
            "\n",
            "\n",
            "        [[[ 2.1212e+00,  1.7113e+00,  4.2441e+00,  ..., -1.2885e+00,\n",
            "            7.2580e+00,  1.5508e+00],\n",
            "          [ 1.9279e+00,  4.0503e+00, -2.4987e+00,  ..., -1.0739e+01,\n",
            "            7.8450e+00, -3.8996e-01],\n",
            "          [ 2.3649e+00,  2.5573e+00,  4.7846e+00,  ..., -9.2196e-01,\n",
            "            2.7716e-01, -4.9716e+00],\n",
            "          ...,\n",
            "          [-1.6017e+00,  9.3781e-01,  8.1939e-01,  ...,  1.7565e+00,\n",
            "            2.0499e+00,  2.2815e+00],\n",
            "          [-1.6038e+00,  9.1990e-01,  8.4844e-01,  ...,  1.7350e+00,\n",
            "            2.0854e+00,  2.3031e+00],\n",
            "          [-1.6171e+00,  7.1361e-01,  9.2167e-01,  ...,  1.6883e+00,\n",
            "            2.1708e+00,  2.3161e+00]],\n",
            "\n",
            "         [[-1.0054e+01,  3.5165e+00, -3.4821e+00,  ...,  4.1017e-01,\n",
            "           -2.3002e+00,  5.4595e+00],\n",
            "          [ 7.7333e+00, -6.4845e+00,  6.5945e+00,  ..., -3.6112e+00,\n",
            "           -3.4128e+00, -9.0208e+00],\n",
            "          [ 3.1326e+00, -5.3969e-01,  3.8246e+00,  ..., -4.4724e+00,\n",
            "            4.1085e+00, -5.5475e-01],\n",
            "          ...,\n",
            "          [ 5.4942e+00, -2.0991e-01, -4.0453e-01,  ..., -1.8610e+00,\n",
            "            5.2030e-01, -1.4748e+00],\n",
            "          [ 5.5368e+00, -1.9170e-01, -4.3023e-01,  ..., -1.8872e+00,\n",
            "            4.8764e-01, -1.4828e+00],\n",
            "          [ 5.6176e+00, -1.5525e-01, -2.7709e-01,  ..., -2.0052e+00,\n",
            "            4.2582e-01, -1.6889e+00]],\n",
            "\n",
            "         [[ 2.1292e+00, -3.4887e-01, -3.2113e+00,  ...,  1.1938e+00,\n",
            "            5.1082e+00, -5.9109e+00],\n",
            "          [ 3.6201e+00,  1.0040e+00, -5.0722e+00,  ...,  4.8044e+00,\n",
            "            2.9829e+00,  3.2537e+00],\n",
            "          [ 4.4888e+00,  2.3419e+00, -6.0322e-01,  ..., -2.0217e+00,\n",
            "            4.1807e+00, -5.0641e+00],\n",
            "          ...,\n",
            "          [-4.2431e+00, -2.6638e+00,  7.1440e+00,  ..., -5.0837e-01,\n",
            "           -2.9283e+00, -1.1064e+00],\n",
            "          [-4.2121e+00, -2.6242e+00,  7.1107e+00,  ..., -5.1317e-01,\n",
            "           -2.9001e+00, -1.0818e+00],\n",
            "          [-4.2161e+00, -2.4471e+00,  7.2058e+00,  ..., -3.1080e-01,\n",
            "           -2.8877e+00, -1.1426e+00]],\n",
            "\n",
            "         [[ 3.9075e-02, -5.7167e-01,  2.7667e+00,  ...,  5.1452e-01,\n",
            "           -5.7518e+00, -3.7961e+00],\n",
            "          [ 2.0945e+00,  5.6556e+00,  4.8095e+00,  ...,  3.8553e+00,\n",
            "           -2.2792e-01, -2.7708e+00],\n",
            "          [-1.2510e+00,  3.1261e+00, -2.3791e+00,  ...,  5.6260e-01,\n",
            "            3.7648e+00, -6.7844e-01],\n",
            "          ...,\n",
            "          [-5.3363e-01,  1.0083e+00,  3.1126e+00,  ...,  4.2116e+00,\n",
            "           -1.5787e+00, -1.5834e+00],\n",
            "          [-5.3154e-01,  1.0518e+00,  3.1477e+00,  ...,  4.2256e+00,\n",
            "           -1.5716e+00, -1.5777e+00],\n",
            "          [-5.1503e-01,  1.0052e+00,  3.2010e+00,  ...,  4.2689e+00,\n",
            "           -1.4123e+00, -1.5717e+00]]]], grad_fn=<TransposeBackward0>)), (tensor([[[[-3.7051e+00, -1.3113e+00,  2.8504e+00,  ...,  2.0055e+00,\n",
            "           -4.5416e-02, -9.9990e-01],\n",
            "          [-3.4449e+00, -1.0583e+00,  1.3462e+00,  ...,  6.0567e-01,\n",
            "            6.7773e-01, -7.7593e-02],\n",
            "          [-2.0348e+00, -1.9895e+00,  1.0504e+00,  ...,  6.8558e-01,\n",
            "            2.8846e-01, -1.2052e-01],\n",
            "          ...,\n",
            "          [-3.7511e+00, -1.0231e+00,  2.4382e+00,  ...,  1.9296e+00,\n",
            "           -6.4360e-01, -1.1726e+00],\n",
            "          [-3.7511e+00, -1.0231e+00,  2.4382e+00,  ...,  1.9296e+00,\n",
            "           -6.4360e-01, -1.1726e+00],\n",
            "          [-3.7382e+00, -1.0134e+00,  2.4234e+00,  ...,  1.9330e+00,\n",
            "           -6.4693e-01, -1.1825e+00]],\n",
            "\n",
            "         [[-1.7061e+00,  5.5449e+00, -9.7705e+00,  ..., -8.9630e-01,\n",
            "            8.0552e-01, -2.3906e+00],\n",
            "          [-2.6457e+00, -1.4172e-01, -2.3107e+00,  ..., -2.2942e+00,\n",
            "           -1.1623e+00, -1.3991e+00],\n",
            "          [-4.8127e+00,  9.1259e-01, -1.9147e+00,  ..., -5.9316e-01,\n",
            "            4.9227e+00, -4.7412e-01],\n",
            "          ...,\n",
            "          [-1.7366e+00,  5.8413e+00, -1.0740e+01,  ..., -1.0447e+00,\n",
            "            1.6481e+00, -1.9025e+00],\n",
            "          [-1.7366e+00,  5.8413e+00, -1.0740e+01,  ..., -1.0447e+00,\n",
            "            1.6481e+00, -1.9025e+00],\n",
            "          [-1.7223e+00,  5.8463e+00, -1.0743e+01,  ..., -1.0408e+00,\n",
            "            1.6321e+00, -1.8976e+00]],\n",
            "\n",
            "         [[-1.9998e+00, -3.6449e+00, -1.5016e+00,  ...,  3.7899e-01,\n",
            "            2.2896e+00, -9.8343e-01],\n",
            "          [-9.7695e-01, -4.9949e+00, -1.6196e+00,  ..., -6.6160e-01,\n",
            "            2.7771e+00, -3.6893e-02],\n",
            "          [-1.5768e+00, -6.2076e+00, -2.0404e+00,  ...,  1.1286e-01,\n",
            "            3.0724e+00,  1.8393e+00],\n",
            "          ...,\n",
            "          [-1.8578e+00, -3.6219e+00, -1.2832e+00,  ...,  2.4658e-01,\n",
            "            2.3326e+00, -1.2907e+00],\n",
            "          [-1.8578e+00, -3.6219e+00, -1.2832e+00,  ...,  2.4658e-01,\n",
            "            2.3326e+00, -1.2907e+00],\n",
            "          [-1.8615e+00, -3.6226e+00, -1.2820e+00,  ...,  2.4326e-01,\n",
            "            2.3321e+00, -1.2891e+00]],\n",
            "\n",
            "         [[ 3.4357e+00, -6.8621e-01,  6.5117e-01,  ..., -1.4388e-01,\n",
            "           -2.9849e+00,  1.0279e+00],\n",
            "          [ 2.6777e+00, -7.8699e-01, -3.2323e+00,  ...,  1.8592e-01,\n",
            "            1.3671e+00, -2.2771e-01],\n",
            "          [ 1.5610e+00, -7.0894e-01, -7.5427e-01,  ...,  1.5306e-01,\n",
            "           -2.3542e-01, -2.5541e-01],\n",
            "          ...,\n",
            "          [ 4.2799e+00, -7.2895e-01,  1.1848e+00,  ..., -4.8768e-02,\n",
            "           -2.7317e+00,  1.2033e+00],\n",
            "          [ 4.2799e+00, -7.2895e-01,  1.1848e+00,  ..., -4.8768e-02,\n",
            "           -2.7317e+00,  1.2033e+00],\n",
            "          [ 4.2590e+00, -7.1667e-01,  1.1795e+00,  ..., -5.4798e-02,\n",
            "           -2.7246e+00,  1.2024e+00]]],\n",
            "\n",
            "\n",
            "        [[[-3.7544e+00, -1.2804e+00,  2.8557e+00,  ...,  2.0754e+00,\n",
            "           -1.3655e-01, -1.0837e+00],\n",
            "          [-3.4544e+00, -1.0971e+00,  1.2665e+00,  ...,  5.3317e-01,\n",
            "            6.2756e-01, -1.2887e-01],\n",
            "          [-2.0258e+00, -1.9383e+00,  9.3630e-01,  ...,  7.4130e-01,\n",
            "            1.9732e-01, -1.1146e-01],\n",
            "          ...,\n",
            "          [-3.8281e+00, -9.7183e-01,  2.4639e+00,  ...,  2.0028e+00,\n",
            "           -7.2298e-01, -1.2556e+00],\n",
            "          [-3.8281e+00, -9.7183e-01,  2.4639e+00,  ...,  2.0028e+00,\n",
            "           -7.2298e-01, -1.2556e+00],\n",
            "          [-3.8165e+00, -9.6451e-01,  2.4514e+00,  ...,  2.0062e+00,\n",
            "           -7.2235e-01, -1.2656e+00]],\n",
            "\n",
            "         [[-1.6298e+00,  5.5237e+00, -9.7008e+00,  ..., -8.6283e-01,\n",
            "            6.5174e-01, -2.4065e+00],\n",
            "          [-2.6163e+00, -3.0059e-01, -2.2672e+00,  ..., -2.2567e+00,\n",
            "           -1.1575e+00, -1.4250e+00],\n",
            "          [-4.8271e+00,  7.9286e-01, -1.6999e+00,  ..., -5.5331e-01,\n",
            "            5.0769e+00, -5.0969e-01],\n",
            "          ...,\n",
            "          [-1.6550e+00,  5.8359e+00, -1.0688e+01,  ..., -1.0287e+00,\n",
            "            1.4922e+00, -1.9243e+00],\n",
            "          [-1.6550e+00,  5.8359e+00, -1.0688e+01,  ..., -1.0287e+00,\n",
            "            1.4922e+00, -1.9243e+00],\n",
            "          [-1.6412e+00,  5.8425e+00, -1.0688e+01,  ..., -1.0250e+00,\n",
            "            1.4771e+00, -1.9204e+00]],\n",
            "\n",
            "         [[-1.9271e+00, -3.6798e+00, -1.5147e+00,  ...,  3.3911e-01,\n",
            "            2.3698e+00, -1.0122e+00],\n",
            "          [-9.8933e-01, -4.9419e+00, -1.6017e+00,  ..., -6.6398e-01,\n",
            "            2.7778e+00, -7.1383e-03],\n",
            "          [-1.4543e+00, -6.1775e+00, -2.0018e+00,  ...,  1.0024e-03,\n",
            "            3.0878e+00,  1.7728e+00],\n",
            "          ...,\n",
            "          [-1.7848e+00, -3.6240e+00, -1.2905e+00,  ...,  2.1375e-01,\n",
            "            2.3743e+00, -1.3368e+00],\n",
            "          [-1.7848e+00, -3.6240e+00, -1.2905e+00,  ...,  2.1375e-01,\n",
            "            2.3743e+00, -1.3368e+00],\n",
            "          [-1.7876e+00, -3.6247e+00, -1.2898e+00,  ...,  2.0942e-01,\n",
            "            2.3735e+00, -1.3357e+00]],\n",
            "\n",
            "         [[ 3.3016e+00, -6.5760e-01,  7.2602e-01,  ..., -2.3261e-01,\n",
            "           -3.0359e+00,  9.3340e-01],\n",
            "          [ 2.5359e+00, -8.3130e-01, -3.1762e+00,  ...,  1.3268e-01,\n",
            "            1.3510e+00, -1.6675e-01],\n",
            "          [ 1.3616e+00, -8.4519e-01, -8.3453e-01,  ...,  9.9635e-02,\n",
            "           -1.8755e-01, -2.2055e-01],\n",
            "          ...,\n",
            "          [ 4.1573e+00, -7.4428e-01,  1.2670e+00,  ..., -1.2242e-01,\n",
            "           -2.7584e+00,  1.1669e+00],\n",
            "          [ 4.1573e+00, -7.4428e-01,  1.2670e+00,  ..., -1.2242e-01,\n",
            "           -2.7584e+00,  1.1669e+00],\n",
            "          [ 4.1366e+00, -7.3418e-01,  1.2608e+00,  ..., -1.2694e-01,\n",
            "           -2.7524e+00,  1.1667e+00]]],\n",
            "\n",
            "\n",
            "        [[[-3.6073e+00, -1.3472e+00,  2.7455e+00,  ...,  1.9952e+00,\n",
            "           -1.1555e-02, -9.2140e-01],\n",
            "          [-3.3375e+00, -1.0499e+00,  1.2774e+00,  ...,  5.9903e-01,\n",
            "            6.5908e-01, -6.3395e-02],\n",
            "          [-1.9761e+00, -2.0027e+00,  9.8868e-01,  ...,  6.5838e-01,\n",
            "            3.8528e-01, -7.7624e-02],\n",
            "          ...,\n",
            "          [-3.6886e+00, -1.0215e+00,  2.3833e+00,  ...,  1.9304e+00,\n",
            "           -5.7788e-01, -1.1008e+00],\n",
            "          [-3.6886e+00, -1.0215e+00,  2.3833e+00,  ...,  1.9304e+00,\n",
            "           -5.7788e-01, -1.1008e+00],\n",
            "          [-3.6762e+00, -1.0140e+00,  2.3720e+00,  ...,  1.9345e+00,\n",
            "           -5.7964e-01, -1.1091e+00]],\n",
            "\n",
            "         [[-1.7839e+00,  5.5052e+00, -9.7093e+00,  ..., -8.1276e-01,\n",
            "            8.1483e-01, -2.3745e+00],\n",
            "          [-2.6551e+00, -1.8977e-01, -2.3781e+00,  ..., -2.2428e+00,\n",
            "           -1.0694e+00, -1.3148e+00],\n",
            "          [-4.8293e+00,  7.1772e-01, -1.6681e+00,  ..., -5.3719e-01,\n",
            "            5.0404e+00, -4.6574e-01],\n",
            "          ...,\n",
            "          [-1.8194e+00,  5.8471e+00, -1.0658e+01,  ..., -9.7395e-01,\n",
            "            1.7228e+00, -1.9260e+00],\n",
            "          [-1.8194e+00,  5.8471e+00, -1.0658e+01,  ..., -9.7395e-01,\n",
            "            1.7228e+00, -1.9260e+00],\n",
            "          [-1.8067e+00,  5.8549e+00, -1.0657e+01,  ..., -9.7170e-01,\n",
            "            1.7090e+00, -1.9218e+00]],\n",
            "\n",
            "         [[-1.8848e+00, -3.8084e+00, -1.4926e+00,  ...,  3.2189e-01,\n",
            "            2.4011e+00, -9.5301e-01],\n",
            "          [-1.0147e+00, -4.9570e+00, -1.6469e+00,  ..., -6.2274e-01,\n",
            "            2.7949e+00, -4.6221e-02],\n",
            "          [-1.5721e+00, -6.2586e+00, -2.0429e+00,  ...,  9.7389e-02,\n",
            "            3.1000e+00,  1.8255e+00],\n",
            "          ...,\n",
            "          [-1.7352e+00, -3.7283e+00, -1.2634e+00,  ...,  1.9610e-01,\n",
            "            2.3970e+00, -1.2892e+00],\n",
            "          [-1.7352e+00, -3.7283e+00, -1.2634e+00,  ...,  1.9610e-01,\n",
            "            2.3970e+00, -1.2892e+00],\n",
            "          [-1.7385e+00, -3.7302e+00, -1.2618e+00,  ...,  1.9227e-01,\n",
            "            2.3972e+00, -1.2880e+00]],\n",
            "\n",
            "         [[ 3.2233e+00, -5.7400e-01,  6.5616e-01,  ..., -2.2888e-01,\n",
            "           -3.0627e+00,  9.4116e-01],\n",
            "          [ 2.6607e+00, -6.9079e-01, -3.2252e+00,  ...,  1.7841e-01,\n",
            "            1.2734e+00, -2.2987e-01],\n",
            "          [ 1.4423e+00, -5.7568e-01, -8.6744e-01,  ...,  1.2947e-01,\n",
            "           -3.4247e-01, -4.0003e-01],\n",
            "          ...,\n",
            "          [ 4.0582e+00, -6.9452e-01,  1.1856e+00,  ..., -1.1628e-01,\n",
            "           -2.7746e+00,  1.1626e+00],\n",
            "          [ 4.0582e+00, -6.9452e-01,  1.1856e+00,  ..., -1.1628e-01,\n",
            "           -2.7746e+00,  1.1626e+00],\n",
            "          [ 4.0372e+00, -6.8378e-01,  1.1786e+00,  ..., -1.2137e-01,\n",
            "           -2.7691e+00,  1.1613e+00]]],\n",
            "\n",
            "\n",
            "        ...,\n",
            "\n",
            "\n",
            "        [[[-3.6939e+00, -1.6095e+00,  2.8876e+00,  ...,  2.2522e+00,\n",
            "           -2.0436e-01, -1.0682e+00],\n",
            "          [-3.3122e+00, -1.0501e+00,  1.3559e+00,  ...,  5.6895e-01,\n",
            "            7.0609e-01, -6.4596e-02],\n",
            "          [-1.9927e+00, -2.0378e+00,  1.0412e+00,  ...,  6.5668e-01,\n",
            "            2.8295e-01,  2.8688e-02],\n",
            "          ...,\n",
            "          [-3.7004e+00, -1.3706e+00,  2.4788e+00,  ...,  2.2142e+00,\n",
            "           -7.3150e-01, -1.2043e+00],\n",
            "          [-3.7004e+00, -1.3706e+00,  2.4788e+00,  ...,  2.2142e+00,\n",
            "           -7.3150e-01, -1.2043e+00],\n",
            "          [-3.6884e+00, -1.3668e+00,  2.4650e+00,  ...,  2.2220e+00,\n",
            "           -7.3294e-01, -1.2160e+00]],\n",
            "\n",
            "         [[-1.8297e+00,  5.2963e+00, -9.9564e+00,  ..., -7.6416e-01,\n",
            "            7.7583e-01, -2.3272e+00],\n",
            "          [-2.7194e+00,  3.2812e-01, -2.6210e+00,  ..., -2.3557e+00,\n",
            "           -1.1730e+00, -1.1935e+00],\n",
            "          [-4.8290e+00,  8.0134e-01, -1.6225e+00,  ..., -4.9087e-01,\n",
            "            5.0939e+00, -4.9901e-01],\n",
            "          ...,\n",
            "          [-1.9283e+00,  5.5027e+00, -1.0679e+01,  ..., -9.7390e-01,\n",
            "            1.6436e+00, -1.9061e+00],\n",
            "          [-1.9283e+00,  5.5027e+00, -1.0679e+01,  ..., -9.7390e-01,\n",
            "            1.6436e+00, -1.9061e+00],\n",
            "          [-1.9133e+00,  5.5058e+00, -1.0681e+01,  ..., -9.7309e-01,\n",
            "            1.6301e+00, -1.9021e+00]],\n",
            "\n",
            "         [[-2.2692e+00, -3.7567e+00, -1.4060e+00,  ...,  4.6532e-01,\n",
            "            2.2427e+00, -9.9160e-01],\n",
            "          [-1.0617e+00, -5.0266e+00, -1.6866e+00,  ..., -6.7242e-01,\n",
            "            2.8046e+00, -2.9986e-02],\n",
            "          [-1.6837e+00, -6.2706e+00, -2.0328e+00,  ...,  4.5242e-02,\n",
            "            3.0901e+00,  1.7277e+00],\n",
            "          ...,\n",
            "          [-2.0578e+00, -3.7707e+00, -1.2480e+00,  ...,  3.0251e-01,\n",
            "            2.3235e+00, -1.2505e+00],\n",
            "          [-2.0578e+00, -3.7707e+00, -1.2480e+00,  ...,  3.0251e-01,\n",
            "            2.3235e+00, -1.2505e+00],\n",
            "          [-2.0626e+00, -3.7703e+00, -1.2450e+00,  ...,  3.0109e-01,\n",
            "            2.3235e+00, -1.2501e+00]],\n",
            "\n",
            "         [[ 3.3257e+00, -6.8948e-01,  5.6004e-01,  ..., -9.2623e-02,\n",
            "           -2.7669e+00,  9.0380e-01],\n",
            "          [ 2.8771e+00, -4.9428e-01, -3.1765e+00,  ...,  1.9950e-01,\n",
            "            1.3138e+00, -4.6061e-01],\n",
            "          [ 1.4413e+00, -6.5639e-01, -1.0549e+00,  ...,  8.9165e-02,\n",
            "           -2.7831e-01, -1.8898e-01],\n",
            "          ...,\n",
            "          [ 4.1767e+00, -6.8307e-01,  1.0783e+00,  ..., -1.4441e-03,\n",
            "           -2.5636e+00,  9.8783e-01],\n",
            "          [ 4.1767e+00, -6.8307e-01,  1.0783e+00,  ..., -1.4447e-03,\n",
            "           -2.5636e+00,  9.8783e-01],\n",
            "          [ 4.1555e+00, -6.7101e-01,  1.0728e+00,  ..., -7.3109e-03,\n",
            "           -2.5563e+00,  9.8751e-01]]],\n",
            "\n",
            "\n",
            "        [[[-3.8282e+00, -1.3184e+00,  2.9390e+00,  ...,  2.0638e+00,\n",
            "            1.2951e-02, -9.4079e-01],\n",
            "          [-3.4827e+00, -1.0520e+00,  1.2296e+00,  ...,  6.2619e-01,\n",
            "            5.8694e-01, -2.4216e-01],\n",
            "          [-2.0652e+00, -1.9545e+00,  9.3841e-01,  ...,  6.4277e-01,\n",
            "            2.5617e-01, -1.0801e-01],\n",
            "          ...,\n",
            "          [-3.8783e+00, -1.0173e+00,  2.5122e+00,  ...,  1.9729e+00,\n",
            "           -6.0561e-01, -1.1489e+00],\n",
            "          [-3.8783e+00, -1.0173e+00,  2.5122e+00,  ...,  1.9729e+00,\n",
            "           -6.0561e-01, -1.1489e+00],\n",
            "          [-3.8656e+00, -1.0104e+00,  2.5001e+00,  ...,  1.9766e+00,\n",
            "           -6.0679e-01, -1.1598e+00]],\n",
            "\n",
            "         [[-1.6722e+00,  5.6529e+00, -9.7209e+00,  ..., -9.1441e-01,\n",
            "            5.5923e-01, -2.4845e+00],\n",
            "          [-2.6905e+00, -2.0158e-01, -2.4619e+00,  ..., -2.1844e+00,\n",
            "           -1.1683e+00, -1.4936e+00],\n",
            "          [-4.7572e+00,  7.4772e-01, -1.8314e+00,  ..., -5.2049e-01,\n",
            "            4.8505e+00, -4.8130e-01],\n",
            "          ...,\n",
            "          [-1.6558e+00,  5.9122e+00, -1.0641e+01,  ..., -1.0477e+00,\n",
            "            1.4723e+00, -1.9965e+00],\n",
            "          [-1.6558e+00,  5.9122e+00, -1.0641e+01,  ..., -1.0477e+00,\n",
            "            1.4723e+00, -1.9965e+00],\n",
            "          [-1.6398e+00,  5.9209e+00, -1.0646e+01,  ..., -1.0434e+00,\n",
            "            1.4553e+00, -1.9906e+00]],\n",
            "\n",
            "         [[-1.9424e+00, -3.6728e+00, -1.5039e+00,  ...,  3.5814e-01,\n",
            "            2.3856e+00, -1.0292e+00],\n",
            "          [-1.0166e+00, -4.8592e+00, -1.5780e+00,  ..., -7.0195e-01,\n",
            "            2.7006e+00,  3.1140e-02],\n",
            "          [-1.4986e+00, -6.1509e+00, -2.0310e+00,  ...,  8.7797e-02,\n",
            "            3.1135e+00,  1.8155e+00],\n",
            "          ...,\n",
            "          [-1.8008e+00, -3.6340e+00, -1.2743e+00,  ...,  2.2643e-01,\n",
            "            2.4092e+00, -1.3472e+00],\n",
            "          [-1.8008e+00, -3.6340e+00, -1.2743e+00,  ...,  2.2643e-01,\n",
            "            2.4092e+00, -1.3472e+00],\n",
            "          [-1.8042e+00, -3.6352e+00, -1.2724e+00,  ...,  2.2371e-01,\n",
            "            2.4090e+00, -1.3466e+00]],\n",
            "\n",
            "         [[ 3.5175e+00, -5.7870e-01,  7.3521e-01,  ..., -1.8448e-01,\n",
            "           -2.9590e+00,  8.8797e-01],\n",
            "          [ 2.4754e+00, -9.7398e-01, -3.0877e+00,  ...,  1.5954e-01,\n",
            "            1.3738e+00, -8.1248e-02],\n",
            "          [ 1.5270e+00, -6.5974e-01, -8.7072e-01,  ...,  8.1243e-02,\n",
            "           -2.9921e-01, -3.2233e-01],\n",
            "          ...,\n",
            "          [ 4.2999e+00, -6.6709e-01,  1.2149e+00,  ..., -1.0960e-01,\n",
            "           -2.6790e+00,  1.1330e+00],\n",
            "          [ 4.2999e+00, -6.6709e-01,  1.2149e+00,  ..., -1.0960e-01,\n",
            "           -2.6790e+00,  1.1330e+00],\n",
            "          [ 4.2799e+00, -6.5372e-01,  1.2109e+00,  ..., -1.1462e-01,\n",
            "           -2.6749e+00,  1.1305e+00]]],\n",
            "\n",
            "\n",
            "        [[[-3.7223e+00, -1.3038e+00,  2.8839e+00,  ...,  2.0505e+00,\n",
            "            4.1279e-03, -9.2763e-01],\n",
            "          [-3.4399e+00, -1.0412e+00,  1.2464e+00,  ...,  5.4474e-01,\n",
            "            5.7813e-01, -1.2153e-01],\n",
            "          [-1.9951e+00, -1.9831e+00,  1.0144e+00,  ...,  7.0427e-01,\n",
            "            3.0669e-01, -8.2187e-02],\n",
            "          ...,\n",
            "          [-3.8206e+00, -9.8810e-01,  2.5031e+00,  ...,  1.9885e+00,\n",
            "           -6.0500e-01, -1.1083e+00],\n",
            "          [-3.8206e+00, -9.8810e-01,  2.5031e+00,  ...,  1.9885e+00,\n",
            "           -6.0500e-01, -1.1083e+00],\n",
            "          [-3.8086e+00, -9.8156e-01,  2.4905e+00,  ...,  1.9920e+00,\n",
            "           -6.0629e-01, -1.1200e+00]],\n",
            "\n",
            "         [[-1.7112e+00,  5.6227e+00, -9.7555e+00,  ..., -8.6483e-01,\n",
            "            6.0260e-01, -2.3802e+00],\n",
            "          [-2.6403e+00, -2.2565e-01, -2.4715e+00,  ..., -2.1847e+00,\n",
            "           -1.2968e+00, -1.4560e+00],\n",
            "          [-4.8559e+00,  8.8259e-01, -1.9244e+00,  ..., -5.8846e-01,\n",
            "            4.9138e+00, -3.8241e-01],\n",
            "          ...,\n",
            "          [-1.7217e+00,  5.8977e+00, -1.0697e+01,  ..., -1.0082e+00,\n",
            "            1.5605e+00, -1.9151e+00],\n",
            "          [-1.7217e+00,  5.8977e+00, -1.0697e+01,  ..., -1.0082e+00,\n",
            "            1.5606e+00, -1.9151e+00],\n",
            "          [-1.7089e+00,  5.9047e+00, -1.0695e+01,  ..., -1.0035e+00,\n",
            "            1.5508e+00, -1.9077e+00]],\n",
            "\n",
            "         [[-1.8917e+00, -3.7720e+00, -1.5335e+00,  ...,  3.1261e-01,\n",
            "            2.4622e+00, -1.0105e+00],\n",
            "          [-9.8866e-01, -4.9352e+00, -1.6071e+00,  ..., -6.7880e-01,\n",
            "            2.7317e+00, -5.2279e-02],\n",
            "          [-1.4383e+00, -6.1822e+00, -2.0413e+00,  ...,  6.5730e-02,\n",
            "            3.1049e+00,  1.7768e+00],\n",
            "          ...,\n",
            "          [-1.7565e+00, -3.6890e+00, -1.2910e+00,  ...,  1.8269e-01,\n",
            "            2.4478e+00, -1.3553e+00],\n",
            "          [-1.7565e+00, -3.6890e+00, -1.2910e+00,  ...,  1.8269e-01,\n",
            "            2.4478e+00, -1.3553e+00],\n",
            "          [-1.7596e+00, -3.6898e+00, -1.2896e+00,  ...,  1.7797e-01,\n",
            "            2.4483e+00, -1.3537e+00]],\n",
            "\n",
            "         [[ 3.4108e+00, -5.5476e-01,  7.3825e-01,  ..., -2.0694e-01,\n",
            "           -3.0597e+00,  9.5545e-01],\n",
            "          [ 2.5763e+00, -8.9114e-01, -3.1608e+00,  ...,  1.2868e-01,\n",
            "            1.3609e+00, -7.9222e-02],\n",
            "          [ 1.5259e+00, -6.1758e-01, -7.7594e-01,  ...,  9.9635e-02,\n",
            "           -3.0331e-01, -2.6292e-01],\n",
            "          ...,\n",
            "          [ 4.2408e+00, -7.0043e-01,  1.2291e+00,  ..., -1.1690e-01,\n",
            "           -2.7639e+00,  1.2050e+00],\n",
            "          [ 4.2408e+00, -7.0043e-01,  1.2291e+00,  ..., -1.1690e-01,\n",
            "           -2.7639e+00,  1.2050e+00],\n",
            "          [ 4.2198e+00, -6.8975e-01,  1.2237e+00,  ..., -1.2162e-01,\n",
            "           -2.7571e+00,  1.2033e+00]]]], grad_fn=<TransposeBackward0>), tensor([[[[-2.3346e+00,  5.5684e+00, -5.8015e-01,  ...,  7.7494e+00,\n",
            "           -1.2534e+00,  5.7055e+00],\n",
            "          [ 3.0624e+00,  2.6882e+00,  1.3645e+00,  ...,  2.0601e+00,\n",
            "           -9.3031e-01, -3.8019e-01],\n",
            "          [ 1.3807e+00,  3.8533e+00,  5.9128e-01,  ...,  1.8682e+00,\n",
            "           -7.8745e-01, -1.9582e+00],\n",
            "          ...,\n",
            "          [-1.7998e+00,  5.1768e+00, -4.2707e-01,  ...,  7.2928e+00,\n",
            "           -3.3069e-01,  5.0868e+00],\n",
            "          [-1.7998e+00,  5.1768e+00, -4.2707e-01,  ...,  7.2928e+00,\n",
            "           -3.3069e-01,  5.0868e+00],\n",
            "          [-1.8087e+00,  5.1697e+00, -4.2999e-01,  ...,  7.2842e+00,\n",
            "           -3.2098e-01,  5.1050e+00]],\n",
            "\n",
            "         [[ 2.0374e+00, -3.3030e-01,  2.4933e-01,  ...,  1.2523e-01,\n",
            "            8.6318e-02, -7.7918e-02],\n",
            "          [ 5.3359e-01, -1.9078e+00, -1.4795e+00,  ..., -7.0582e-02,\n",
            "           -1.8654e+00, -1.3953e-01],\n",
            "          [-1.8525e+00, -9.6022e-01, -3.5430e+00,  ...,  2.8422e+00,\n",
            "            1.6668e+00, -1.9174e+00],\n",
            "          ...,\n",
            "          [ 2.2669e+00, -2.6220e-01,  3.1776e-01,  ...,  4.7973e-01,\n",
            "           -2.7361e-01, -2.5442e-01],\n",
            "          [ 2.2669e+00, -2.6220e-01,  3.1776e-01,  ...,  4.7973e-01,\n",
            "           -2.7361e-01, -2.5442e-01],\n",
            "          [ 2.2785e+00, -2.5735e-01,  3.2031e-01,  ...,  4.8201e-01,\n",
            "           -2.8033e-01, -2.5158e-01]],\n",
            "\n",
            "         [[ 1.2018e+00, -3.6068e+00, -1.0647e+00,  ...,  1.5113e+00,\n",
            "           -4.0114e+00, -3.0860e+00],\n",
            "          [-9.1470e-03,  3.2453e-01,  5.2155e-01,  ...,  1.2354e+00,\n",
            "           -4.0401e+00, -3.5329e+00],\n",
            "          [ 1.8652e+00, -6.6665e-01, -2.6328e-01,  ...,  1.9343e+00,\n",
            "           -2.1859e+00, -2.6965e+00],\n",
            "          ...,\n",
            "          [ 1.3799e+00, -4.3859e+00, -1.1377e+00,  ...,  1.6650e+00,\n",
            "           -3.5840e+00, -3.9382e+00],\n",
            "          [ 1.3799e+00, -4.3859e+00, -1.1377e+00,  ...,  1.6650e+00,\n",
            "           -3.5840e+00, -3.9382e+00],\n",
            "          [ 1.3924e+00, -4.3883e+00, -1.1207e+00,  ...,  1.6577e+00,\n",
            "           -3.5711e+00, -3.9413e+00]],\n",
            "\n",
            "         [[-3.0669e+00, -2.2297e+00, -2.1566e+00,  ..., -1.6139e-01,\n",
            "            2.2117e+00,  9.8915e-01],\n",
            "          [-1.7436e+00, -4.6659e+00,  2.9705e+00,  ..., -1.3573e+00,\n",
            "            4.0520e+00, -9.1418e-01],\n",
            "          [ 1.3834e+00, -4.0236e-01,  1.7750e+00,  ...,  2.4756e+00,\n",
            "            2.0992e+00,  9.9222e-01],\n",
            "          ...,\n",
            "          [-3.4269e+00, -3.2723e+00, -1.8714e+00,  ..., -6.3265e-01,\n",
            "            3.5553e+00,  8.2952e-01],\n",
            "          [-3.4269e+00, -3.2723e+00, -1.8714e+00,  ..., -6.3265e-01,\n",
            "            3.5553e+00,  8.2952e-01],\n",
            "          [-3.4401e+00, -3.2599e+00, -1.8599e+00,  ..., -6.3533e-01,\n",
            "            3.5628e+00,  8.2575e-01]]],\n",
            "\n",
            "\n",
            "        [[[-2.4827e+00,  5.1184e+00, -6.9609e-01,  ...,  7.5047e+00,\n",
            "           -1.1416e+00,  5.8607e+00],\n",
            "          [ 2.8658e+00,  2.6944e+00,  1.3589e+00,  ...,  2.0457e+00,\n",
            "           -8.7614e-01, -4.6143e-01],\n",
            "          [ 1.2004e+00,  3.5506e+00,  6.7634e-01,  ...,  1.9398e+00,\n",
            "           -8.1215e-01, -1.9166e+00],\n",
            "          ...,\n",
            "          [-1.9156e+00,  4.7738e+00, -5.4377e-01,  ...,  7.0593e+00,\n",
            "           -2.1511e-01,  5.2186e+00],\n",
            "          [-1.9156e+00,  4.7738e+00, -5.4377e-01,  ...,  7.0593e+00,\n",
            "           -2.1511e-01,  5.2186e+00],\n",
            "          [-1.9250e+00,  4.7674e+00, -5.4128e-01,  ...,  7.0491e+00,\n",
            "           -2.0538e-01,  5.2320e+00]],\n",
            "\n",
            "         [[ 2.0578e+00, -3.4261e-01,  1.0081e-01,  ...,  1.5242e-01,\n",
            "           -3.5507e-02,  3.3052e-01],\n",
            "          [ 6.0429e-01, -1.8950e+00, -1.6044e+00,  ...,  1.3077e-01,\n",
            "           -1.7694e+00, -2.2935e-01],\n",
            "          [-1.9110e+00, -8.6368e-01, -3.6769e+00,  ...,  3.0239e+00,\n",
            "            1.5856e+00, -1.7565e+00],\n",
            "          ...,\n",
            "          [ 2.3050e+00, -2.6822e-01,  2.3050e-01,  ...,  4.6762e-01,\n",
            "           -4.0269e-01,  1.1525e-01],\n",
            "          [ 2.3050e+00, -2.6822e-01,  2.3050e-01,  ...,  4.6762e-01,\n",
            "           -4.0268e-01,  1.1525e-01],\n",
            "          [ 2.3162e+00, -2.6411e-01,  2.3416e-01,  ...,  4.7139e-01,\n",
            "           -4.0987e-01,  1.1720e-01]],\n",
            "\n",
            "         [[ 8.6095e-01, -3.7584e+00, -1.0340e+00,  ...,  1.5045e+00,\n",
            "           -4.2158e+00, -3.1892e+00],\n",
            "          [ 4.6508e-02,  3.5218e-01,  5.0328e-01,  ...,  1.3428e+00,\n",
            "           -4.0053e+00, -3.3959e+00],\n",
            "          [ 1.7480e+00, -7.5081e-01, -2.6300e-01,  ...,  1.9131e+00,\n",
            "           -2.2851e+00, -2.6278e+00],\n",
            "          ...,\n",
            "          [ 1.0780e+00, -4.4985e+00, -1.1129e+00,  ...,  1.6331e+00,\n",
            "           -3.7206e+00, -4.0533e+00],\n",
            "          [ 1.0780e+00, -4.4985e+00, -1.1129e+00,  ...,  1.6331e+00,\n",
            "           -3.7206e+00, -4.0533e+00],\n",
            "          [ 1.0894e+00, -4.4990e+00, -1.0981e+00,  ...,  1.6242e+00,\n",
            "           -3.7095e+00, -4.0558e+00]],\n",
            "\n",
            "         [[-3.2494e+00, -2.2891e+00, -2.0541e+00,  ..., -3.0391e-01,\n",
            "            2.2885e+00,  8.5539e-01],\n",
            "          [-1.7531e+00, -4.6364e+00,  3.0281e+00,  ..., -1.3326e+00,\n",
            "            4.1371e+00, -8.8526e-01],\n",
            "          [ 1.2838e+00, -4.2666e-01,  1.7717e+00,  ...,  2.3597e+00,\n",
            "            2.1514e+00,  9.0342e-01],\n",
            "          ...,\n",
            "          [-3.6247e+00, -3.3069e+00, -1.7284e+00,  ..., -7.9585e-01,\n",
            "            3.6691e+00,  6.6709e-01],\n",
            "          [-3.6247e+00, -3.3069e+00, -1.7284e+00,  ..., -7.9585e-01,\n",
            "            3.6691e+00,  6.6709e-01],\n",
            "          [-3.6364e+00, -3.2942e+00, -1.7141e+00,  ..., -7.9728e-01,\n",
            "            3.6747e+00,  6.6205e-01]]],\n",
            "\n",
            "\n",
            "        [[[-2.3840e+00,  5.4210e+00, -4.9796e-02,  ...,  7.3203e+00,\n",
            "           -1.0265e+00,  5.7033e+00],\n",
            "          [ 3.0535e+00,  2.6287e+00,  1.3815e+00,  ...,  2.0797e+00,\n",
            "           -9.8215e-01, -2.1116e-01],\n",
            "          [ 1.4746e+00,  3.6845e+00,  7.4686e-01,  ...,  1.8207e+00,\n",
            "           -7.3560e-01, -1.9380e+00],\n",
            "          ...,\n",
            "          [-1.8738e+00,  5.0020e+00,  8.0710e-02,  ...,  6.8885e+00,\n",
            "           -6.8434e-02,  5.0764e+00],\n",
            "          [-1.8738e+00,  5.0020e+00,  8.0709e-02,  ...,  6.8885e+00,\n",
            "           -6.8434e-02,  5.0764e+00],\n",
            "          [-1.8843e+00,  4.9909e+00,  8.1999e-02,  ...,  6.8806e+00,\n",
            "           -5.7157e-02,  5.0922e+00]],\n",
            "\n",
            "         [[ 2.0284e+00, -3.6204e-01,  5.4580e-02,  ...,  5.9759e-01,\n",
            "            1.4780e-01, -2.1468e-01],\n",
            "          [ 6.1181e-01, -1.9732e+00, -1.3868e+00,  ..., -1.1523e-01,\n",
            "           -1.9306e+00, -1.0455e-01],\n",
            "          [-1.8548e+00, -1.1126e+00, -3.4111e+00,  ...,  3.1020e+00,\n",
            "            1.5032e+00, -2.0030e+00],\n",
            "          ...,\n",
            "          [ 2.2847e+00, -2.9392e-01,  1.7481e-01,  ...,  8.8590e-01,\n",
            "           -1.9730e-01, -4.1168e-01],\n",
            "          [ 2.2847e+00, -2.9392e-01,  1.7481e-01,  ...,  8.8590e-01,\n",
            "           -1.9730e-01, -4.1168e-01],\n",
            "          [ 2.2968e+00, -2.8727e-01,  1.7939e-01,  ...,  8.8873e-01,\n",
            "           -2.0792e-01, -4.0809e-01]],\n",
            "\n",
            "         [[ 1.3192e+00, -3.8039e+00, -1.1115e+00,  ...,  1.4341e+00,\n",
            "           -4.0695e+00, -3.0180e+00],\n",
            "          [ 1.1442e-01,  2.8774e-01,  4.8499e-01,  ...,  1.2510e+00,\n",
            "           -4.0386e+00, -3.4944e+00],\n",
            "          [ 1.8849e+00, -8.5323e-01, -3.6700e-01,  ...,  1.8456e+00,\n",
            "           -2.2290e+00, -2.5448e+00],\n",
            "          ...,\n",
            "          [ 1.4641e+00, -4.5313e+00, -1.1742e+00,  ...,  1.5629e+00,\n",
            "           -3.6087e+00, -3.8524e+00],\n",
            "          [ 1.4641e+00, -4.5313e+00, -1.1742e+00,  ...,  1.5629e+00,\n",
            "           -3.6087e+00, -3.8524e+00],\n",
            "          [ 1.4722e+00, -4.5311e+00, -1.1598e+00,  ...,  1.5533e+00,\n",
            "           -3.5968e+00, -3.8514e+00]],\n",
            "\n",
            "         [[-2.7807e+00, -2.1535e+00, -2.2427e+00,  ..., -6.3136e-02,\n",
            "            2.2342e+00,  9.1639e-01],\n",
            "          [-1.7106e+00, -4.4268e+00,  2.9165e+00,  ..., -1.4074e+00,\n",
            "            3.9842e+00, -8.4884e-01],\n",
            "          [ 1.5290e+00, -3.7617e-01,  1.7194e+00,  ...,  2.3996e+00,\n",
            "            2.1066e+00,  9.7689e-01],\n",
            "          ...,\n",
            "          [-3.1263e+00, -3.1769e+00, -1.9013e+00,  ..., -5.6881e-01,\n",
            "            3.5751e+00,  7.0286e-01],\n",
            "          [-3.1263e+00, -3.1769e+00, -1.9013e+00,  ..., -5.6881e-01,\n",
            "            3.5751e+00,  7.0286e-01],\n",
            "          [-3.1367e+00, -3.1656e+00, -1.8879e+00,  ..., -5.7161e-01,\n",
            "            3.5810e+00,  6.9686e-01]]],\n",
            "\n",
            "\n",
            "        ...,\n",
            "\n",
            "\n",
            "        [[[-2.2759e+00,  5.6540e+00, -6.9168e-01,  ...,  7.8327e+00,\n",
            "           -2.0202e+00,  5.5535e+00],\n",
            "          [ 3.3579e+00,  2.8492e+00,  1.5193e+00,  ...,  2.0277e+00,\n",
            "           -1.0813e+00, -1.8389e-01],\n",
            "          [ 1.5571e+00,  3.9960e+00,  6.0456e-01,  ...,  1.7937e+00,\n",
            "           -1.1012e+00, -1.8901e+00],\n",
            "          ...,\n",
            "          [-1.7055e+00,  5.3261e+00, -4.0774e-01,  ...,  7.2820e+00,\n",
            "           -1.0958e+00,  4.8827e+00],\n",
            "          [-1.7055e+00,  5.3261e+00, -4.0773e-01,  ...,  7.2820e+00,\n",
            "           -1.0958e+00,  4.8827e+00],\n",
            "          [-1.7124e+00,  5.3264e+00, -4.0843e-01,  ...,  7.2787e+00,\n",
            "           -1.0929e+00,  4.9043e+00]],\n",
            "\n",
            "         [[ 2.0908e+00, -4.3803e-01,  7.7585e-01,  ...,  2.4625e-01,\n",
            "           -5.6782e-02, -3.2483e-02],\n",
            "          [ 7.4413e-01, -2.1985e+00, -1.2138e+00,  ..., -1.8245e-01,\n",
            "           -1.8407e+00, -2.4508e-01],\n",
            "          [-1.6449e+00, -1.0590e+00, -3.2751e+00,  ...,  2.7026e+00,\n",
            "            1.3448e+00, -1.8511e+00],\n",
            "          ...,\n",
            "          [ 2.2401e+00, -4.3721e-01,  7.0677e-01,  ...,  5.3354e-01,\n",
            "           -2.9561e-01, -2.0748e-01],\n",
            "          [ 2.2401e+00, -4.3721e-01,  7.0677e-01,  ...,  5.3354e-01,\n",
            "           -2.9561e-01, -2.0748e-01],\n",
            "          [ 2.2528e+00, -4.3447e-01,  7.1458e-01,  ...,  5.3231e-01,\n",
            "           -3.0119e-01, -2.0254e-01]],\n",
            "\n",
            "         [[ 1.1717e+00, -3.4847e+00, -8.4527e-01,  ...,  1.3610e+00,\n",
            "           -3.7222e+00, -2.7968e+00],\n",
            "          [ 2.9337e-01,  3.4975e-01,  3.5803e-01,  ...,  1.2782e+00,\n",
            "           -3.9603e+00, -3.5875e+00],\n",
            "          [ 1.9871e+00, -7.0361e-01, -4.6473e-01,  ...,  1.8494e+00,\n",
            "           -2.1950e+00, -2.4773e+00],\n",
            "          ...,\n",
            "          [ 1.3608e+00, -4.2551e+00, -9.8631e-01,  ...,  1.4782e+00,\n",
            "           -3.4389e+00, -3.5702e+00],\n",
            "          [ 1.3608e+00, -4.2551e+00, -9.8631e-01,  ...,  1.4782e+00,\n",
            "           -3.4389e+00, -3.5702e+00],\n",
            "          [ 1.3702e+00, -4.2568e+00, -9.6929e-01,  ...,  1.4686e+00,\n",
            "           -3.4262e+00, -3.5726e+00]],\n",
            "\n",
            "         [[-2.6595e+00, -1.8908e+00, -2.0510e+00,  ...,  2.0345e-01,\n",
            "            2.3102e+00,  8.0289e-01],\n",
            "          [-1.7126e+00, -4.2718e+00,  2.8432e+00,  ..., -1.4925e+00,\n",
            "            3.9458e+00, -8.5349e-01],\n",
            "          [ 1.3941e+00, -2.6367e-01,  1.8428e+00,  ...,  2.6498e+00,\n",
            "            2.2519e+00,  1.0926e+00],\n",
            "          ...,\n",
            "          [-2.9592e+00, -2.8681e+00, -1.7549e+00,  ..., -2.8597e-01,\n",
            "            3.5603e+00,  7.1549e-01],\n",
            "          [-2.9592e+00, -2.8681e+00, -1.7549e+00,  ..., -2.8597e-01,\n",
            "            3.5603e+00,  7.1549e-01],\n",
            "          [-2.9712e+00, -2.8583e+00, -1.7428e+00,  ..., -2.9077e-01,\n",
            "            3.5685e+00,  7.0845e-01]]],\n",
            "\n",
            "\n",
            "        [[[-2.4533e+00,  5.3697e+00, -6.8925e-01,  ...,  7.6478e+00,\n",
            "           -1.1342e+00,  5.6801e+00],\n",
            "          [ 2.9468e+00,  2.6558e+00,  1.5094e+00,  ...,  2.0012e+00,\n",
            "           -9.0876e-01, -3.3427e-01],\n",
            "          [ 1.4093e+00,  3.6866e+00,  5.6498e-01,  ...,  1.7912e+00,\n",
            "           -7.5232e-01, -1.9961e+00],\n",
            "          ...,\n",
            "          [-1.9196e+00,  4.9712e+00, -5.5622e-01,  ...,  7.2249e+00,\n",
            "           -1.9025e-01,  5.0300e+00],\n",
            "          [-1.9196e+00,  4.9712e+00, -5.5622e-01,  ...,  7.2249e+00,\n",
            "           -1.9025e-01,  5.0300e+00],\n",
            "          [-1.9289e+00,  4.9644e+00, -5.5636e-01,  ...,  7.2132e+00,\n",
            "           -1.8008e-01,  5.0480e+00]],\n",
            "\n",
            "         [[ 2.1138e+00, -3.4241e-01,  9.3748e-02,  ...,  1.1428e-01,\n",
            "           -5.7427e-02, -2.0290e-02],\n",
            "          [ 3.5561e-01, -1.8465e+00, -1.7070e+00,  ...,  6.2653e-02,\n",
            "           -1.7193e+00, -1.1792e-01],\n",
            "          [-1.7576e+00, -8.5904e-01, -3.6297e+00,  ...,  2.9615e+00,\n",
            "            1.5306e+00, -1.8325e+00],\n",
            "          ...,\n",
            "          [ 2.3715e+00, -2.9995e-01,  1.8124e-01,  ...,  4.6038e-01,\n",
            "           -4.1657e-01, -2.0817e-01],\n",
            "          [ 2.3715e+00, -2.9995e-01,  1.8124e-01,  ...,  4.6038e-01,\n",
            "           -4.1657e-01, -2.0817e-01],\n",
            "          [ 2.3852e+00, -2.9501e-01,  1.8406e-01,  ...,  4.6296e-01,\n",
            "           -4.2186e-01, -2.0600e-01]],\n",
            "\n",
            "         [[ 1.0140e+00, -3.7110e+00, -1.2399e+00,  ...,  1.5090e+00,\n",
            "           -4.1546e+00, -3.2925e+00],\n",
            "          [ 1.1051e-01,  3.8491e-01,  6.9161e-01,  ...,  1.3790e+00,\n",
            "           -4.2061e+00, -3.5504e+00],\n",
            "          [ 1.7449e+00, -6.6710e-01, -2.2855e-01,  ...,  1.9834e+00,\n",
            "           -2.2427e+00, -2.7806e+00],\n",
            "          ...,\n",
            "          [ 1.2048e+00, -4.4472e+00, -1.3144e+00,  ...,  1.6693e+00,\n",
            "           -3.7033e+00, -4.0891e+00],\n",
            "          [ 1.2048e+00, -4.4472e+00, -1.3144e+00,  ...,  1.6693e+00,\n",
            "           -3.7033e+00, -4.0891e+00],\n",
            "          [ 1.2140e+00, -4.4486e+00, -1.2993e+00,  ...,  1.6597e+00,\n",
            "           -3.6927e+00, -4.0920e+00]],\n",
            "\n",
            "         [[-3.1624e+00, -2.4985e+00, -2.0939e+00,  ..., -6.4563e-02,\n",
            "            2.2594e+00,  9.8849e-01],\n",
            "          [-1.7095e+00, -4.7735e+00,  2.9029e+00,  ..., -1.2598e+00,\n",
            "            4.0307e+00, -1.0874e+00],\n",
            "          [ 1.3280e+00, -4.3083e-01,  1.9314e+00,  ...,  2.5456e+00,\n",
            "            2.2580e+00,  9.1164e-01],\n",
            "          ...,\n",
            "          [-3.5120e+00, -3.5238e+00, -1.7828e+00,  ..., -5.6227e-01,\n",
            "            3.6109e+00,  7.9248e-01],\n",
            "          [-3.5120e+00, -3.5238e+00, -1.7828e+00,  ..., -5.6227e-01,\n",
            "            3.6109e+00,  7.9248e-01],\n",
            "          [-3.5253e+00, -3.5114e+00, -1.7724e+00,  ..., -5.6432e-01,\n",
            "            3.6136e+00,  7.8815e-01]]],\n",
            "\n",
            "\n",
            "        [[[-2.4314e+00,  5.2033e+00, -3.6221e-01,  ...,  7.5188e+00,\n",
            "           -9.6989e-01,  5.6666e+00],\n",
            "          [ 3.0004e+00,  2.6421e+00,  1.3916e+00,  ...,  1.8923e+00,\n",
            "           -8.8781e-01, -3.8070e-01],\n",
            "          [ 1.3552e+00,  3.5392e+00,  7.2133e-01,  ...,  1.7094e+00,\n",
            "           -5.9338e-01, -2.0312e+00],\n",
            "          ...,\n",
            "          [-1.8910e+00,  4.8275e+00, -2.6596e-01,  ...,  7.1140e+00,\n",
            "           -1.2400e-02,  5.0219e+00],\n",
            "          [-1.8910e+00,  4.8275e+00, -2.6596e-01,  ...,  7.1140e+00,\n",
            "           -1.2400e-02,  5.0219e+00],\n",
            "          [-1.8997e+00,  4.8184e+00, -2.6418e-01,  ...,  7.0987e+00,\n",
            "            2.0565e-03,  5.0329e+00]],\n",
            "\n",
            "         [[ 2.0020e+00, -3.8729e-01,  3.1920e-02,  ...,  3.7492e-01,\n",
            "            1.2083e-02,  1.1300e-02],\n",
            "          [ 4.2780e-01, -1.8446e+00, -1.6631e+00,  ...,  1.8613e-03,\n",
            "           -1.8832e+00, -1.9295e-01],\n",
            "          [-1.8957e+00, -8.8382e-01, -3.5140e+00,  ...,  3.2219e+00,\n",
            "            1.5655e+00, -1.9698e+00],\n",
            "          ...,\n",
            "          [ 2.2919e+00, -3.2391e-01,  1.9404e-01,  ...,  6.7086e-01,\n",
            "           -4.2634e-01, -1.9655e-01],\n",
            "          [ 2.2919e+00, -3.2391e-01,  1.9404e-01,  ...,  6.7087e-01,\n",
            "           -4.2634e-01, -1.9655e-01],\n",
            "          [ 2.3026e+00, -3.1877e-01,  1.9450e-01,  ...,  6.7723e-01,\n",
            "           -4.3416e-01, -1.9538e-01]],\n",
            "\n",
            "         [[ 1.0965e+00, -3.6908e+00, -1.2353e+00,  ...,  1.4075e+00,\n",
            "           -4.1460e+00, -3.1936e+00],\n",
            "          [ 9.7370e-02,  3.5058e-01,  6.2727e-01,  ...,  1.3398e+00,\n",
            "           -4.1248e+00, -3.5518e+00],\n",
            "          [ 1.6909e+00, -6.6054e-01, -3.0215e-01,  ...,  1.8208e+00,\n",
            "           -2.2062e+00, -2.7294e+00],\n",
            "          ...,\n",
            "          [ 1.2568e+00, -4.4281e+00, -1.2859e+00,  ...,  1.5727e+00,\n",
            "           -3.7100e+00, -4.0199e+00],\n",
            "          [ 1.2568e+00, -4.4281e+00, -1.2859e+00,  ...,  1.5727e+00,\n",
            "           -3.7100e+00, -4.0199e+00],\n",
            "          [ 1.2634e+00, -4.4287e+00, -1.2677e+00,  ...,  1.5613e+00,\n",
            "           -3.6992e+00, -4.0216e+00]],\n",
            "\n",
            "         [[-3.1184e+00, -2.2816e+00, -2.1437e+00,  ..., -2.0312e-01,\n",
            "            2.1724e+00,  9.4109e-01],\n",
            "          [-1.7105e+00, -4.6698e+00,  2.9712e+00,  ..., -1.3037e+00,\n",
            "            4.0463e+00, -9.5291e-01],\n",
            "          [ 1.3398e+00, -3.8522e-01,  1.8248e+00,  ...,  2.5292e+00,\n",
            "            2.1518e+00,  8.5497e-01],\n",
            "          ...,\n",
            "          [-3.4782e+00, -3.3492e+00, -1.7857e+00,  ..., -7.3040e-01,\n",
            "            3.5838e+00,  6.9232e-01],\n",
            "          [-3.4782e+00, -3.3492e+00, -1.7857e+00,  ..., -7.3040e-01,\n",
            "            3.5838e+00,  6.9232e-01],\n",
            "          [-3.4902e+00, -3.3366e+00, -1.7707e+00,  ..., -7.3401e-01,\n",
            "            3.5900e+00,  6.8592e-01]]]], grad_fn=<TransposeBackward0>), tensor([[[[ 1.1238e+01, -1.2053e+01, -4.9234e+00,  ..., -4.1485e+00,\n",
            "            1.4108e+00,  6.5920e+00],\n",
            "          [ 8.1533e+00, -1.5267e+00, -6.4883e+00,  ...,  1.2121e+00,\n",
            "           -7.7051e-01,  9.9302e+00],\n",
            "          [ 5.1190e-01, -6.4640e+00, -4.1934e+00,  ..., -3.4995e+00,\n",
            "            3.2569e+00,  1.6252e+01],\n",
            "          ...,\n",
            "          [ 2.0060e+01, -2.9812e+00,  1.0478e+01,  ..., -7.4542e+00,\n",
            "           -9.6851e+00, -2.3198e+00],\n",
            "          [ 1.0444e+01, -1.0337e+01,  4.3492e+00,  ..., -3.8003e+00,\n",
            "            2.4101e+00,  4.4207e+00],\n",
            "          [ 1.6308e+00,  1.7920e+00, -9.7925e-01,  ..., -4.2427e-01,\n",
            "            1.1402e+00,  1.4493e+00]],\n",
            "\n",
            "         [[-1.3489e+01,  7.5698e+00,  5.9909e+00,  ..., -1.4690e+01,\n",
            "            4.3463e+00, -1.7130e+00],\n",
            "          [-7.9410e+00,  6.7303e+00, -7.0066e+00,  ..., -1.3593e+01,\n",
            "            1.5362e+00, -1.4087e+01],\n",
            "          [-1.8657e+01,  6.3672e+00, -4.6194e+00,  ..., -2.2806e+01,\n",
            "            8.3247e+00, -9.8005e+00],\n",
            "          ...,\n",
            "          [-4.3381e+00,  1.1903e+01,  9.0899e+00,  ...,  5.3632e+00,\n",
            "            4.0256e+00, -5.2414e-01],\n",
            "          [-1.4862e+01, -4.3245e+00,  9.6542e+00,  ...,  6.3539e+00,\n",
            "           -2.2131e+00,  2.4211e+00],\n",
            "          [ 4.4936e+00, -1.0063e+01,  4.2929e+00,  ..., -1.3343e+00,\n",
            "           -3.1365e+00,  2.4616e+00]],\n",
            "\n",
            "         [[-9.5379e+00, -5.5524e-01,  4.5378e+01,  ...,  1.0773e+00,\n",
            "            1.3639e+01,  4.9371e+00],\n",
            "          [-5.1636e+00,  3.8236e-01,  3.5092e+01,  ...,  1.5446e+00,\n",
            "            6.1996e+00, -3.3596e-01],\n",
            "          [-6.4038e+00,  6.0854e+00,  3.8144e+01,  ...,  3.2803e+00,\n",
            "            6.8835e+00, -3.5564e-01],\n",
            "          ...,\n",
            "          [ 8.6923e+00,  7.1627e+00,  2.8037e+01,  ...,  2.5455e+00,\n",
            "            3.9697e+00,  2.8842e+00],\n",
            "          [ 1.7998e+00, -1.9031e+00,  3.2789e+01,  ...,  1.0809e+01,\n",
            "            6.7011e+00,  5.0194e+00],\n",
            "          [-2.9103e+00, -7.5604e-01, -1.8415e+01,  ...,  5.7147e+00,\n",
            "            3.8449e-02,  2.7421e+00]],\n",
            "\n",
            "         [[-1.6755e+01, -1.0431e+01, -4.6682e+00,  ...,  1.3816e+01,\n",
            "            2.3421e+00,  4.3025e+00],\n",
            "          [-4.9371e-01, -1.1435e+01, -1.3073e+00,  ...,  1.8404e+00,\n",
            "            1.4789e-01,  3.3544e+00],\n",
            "          [-1.2535e+01, -1.3365e+01,  8.6608e+00,  ...,  9.5811e+00,\n",
            "           -1.5369e+01,  1.1058e+01],\n",
            "          ...,\n",
            "          [-4.8901e+00, -8.8853e+00,  6.3435e+00,  ...,  1.3078e+01,\n",
            "           -7.8888e+00,  2.4740e+00],\n",
            "          [-2.1917e+00, -3.2870e+00,  7.6988e+00,  ..., -1.1881e+00,\n",
            "            6.1440e+00,  7.8647e+00],\n",
            "          [ 4.2975e+00,  4.8458e+00,  1.5423e+00,  ..., -4.3032e+00,\n",
            "           -3.3699e+00, -2.3654e+00]]],\n",
            "\n",
            "\n",
            "        [[[ 1.1663e+01, -1.6222e+01, -1.0246e+00,  ..., -2.1438e+00,\n",
            "           -7.4745e+00,  1.4023e+00],\n",
            "          [ 9.6683e+00, -8.3530e+00, -3.5939e+00,  ...,  6.1305e-01,\n",
            "           -7.7473e+00,  6.8372e+00],\n",
            "          [ 1.6130e+01, -5.5023e+00, -5.0602e+00,  ..., -6.6905e+00,\n",
            "           -6.8740e+00,  7.4082e+00],\n",
            "          ...,\n",
            "          [ 1.6275e+01, -8.0523e+00,  3.5943e-01,  ..., -8.1184e+00,\n",
            "           -2.0330e+00,  6.6113e+00],\n",
            "          [ 1.0449e+01, -1.2714e+01,  1.8258e+00,  ..., -8.8320e+00,\n",
            "           -1.5288e+00,  6.2689e+00],\n",
            "          [ 1.3736e+00,  1.7749e+00, -1.1635e+00,  ..., -4.6786e-01,\n",
            "            9.8635e-01,  1.3353e+00]],\n",
            "\n",
            "         [[-1.2573e+01,  6.6994e+00,  6.9754e+00,  ..., -7.8748e+00,\n",
            "            9.9456e+00, -5.6705e+00],\n",
            "          [-1.4459e+01,  9.8497e+00, -3.6283e-01,  ...,  1.4906e+00,\n",
            "           -1.3961e+00, -1.5905e+00],\n",
            "          [-7.3874e+00,  8.3981e+00, -6.8398e+00,  ..., -4.2925e+00,\n",
            "            3.5293e+00, -1.2989e+01],\n",
            "          ...,\n",
            "          [-1.0098e+01,  5.0472e-01,  2.4531e+00,  ..., -6.3767e+00,\n",
            "            3.6539e+00, -6.5514e+00],\n",
            "          [-1.1462e+01,  4.8880e-01,  1.9850e+00,  ...,  5.0889e+00,\n",
            "           -4.5695e-01, -3.2008e+00],\n",
            "          [ 4.8734e+00, -1.0195e+01,  4.4254e+00,  ..., -1.4329e+00,\n",
            "           -3.0075e+00,  2.4895e+00]],\n",
            "\n",
            "         [[ 7.6613e+00,  4.3268e+00,  5.3557e+01,  ...,  1.3966e+00,\n",
            "            5.4809e+00, -3.4259e+00],\n",
            "          [ 1.7788e+00,  6.4788e+00,  4.6085e+01,  ...,  7.6761e+00,\n",
            "            1.0132e+01, -2.0414e+00],\n",
            "          [-4.8980e+00,  5.5453e+00,  3.6978e+01,  ...,  8.0994e+00,\n",
            "            6.5567e+00,  4.2985e+00],\n",
            "          ...,\n",
            "          [ 5.3624e+00, -1.1775e-01,  3.1911e+01,  ...,  9.3795e+00,\n",
            "            7.3623e+00, -4.3671e+00],\n",
            "          [-6.1593e-01,  7.3799e-01,  2.5525e+01,  ...,  1.2706e+01,\n",
            "            4.7994e+00,  4.5334e+00],\n",
            "          [-3.1384e+00, -7.1832e-01, -1.8862e+01,  ...,  5.7600e+00,\n",
            "           -1.9182e-02,  2.9160e+00]],\n",
            "\n",
            "         [[-1.9648e+01, -1.2705e+01,  6.6729e+00,  ...,  5.9707e+00,\n",
            "           -8.0975e+00,  1.9687e+00],\n",
            "          [-1.1957e+01, -6.6918e+00,  7.0023e+00,  ...,  6.3945e+00,\n",
            "           -2.5624e+00,  6.8163e+00],\n",
            "          [-6.9551e+00, -1.9248e+01,  1.3631e+01,  ...,  1.1469e+00,\n",
            "            7.1652e-02,  1.7048e+01],\n",
            "          ...,\n",
            "          [-7.2311e+00, -4.6311e+00, -1.2224e+00,  ...,  3.9162e+00,\n",
            "            2.1762e+00, -1.1775e+00],\n",
            "          [-8.5260e-01,  1.5483e+00,  3.2558e+00,  ...,  1.1394e+01,\n",
            "            2.2993e+00, -8.0999e-01],\n",
            "          [ 4.3803e+00,  4.9406e+00,  1.1399e+00,  ..., -4.4350e+00,\n",
            "           -3.2682e+00, -2.2504e+00]]],\n",
            "\n",
            "\n",
            "        [[[ 1.0962e+01, -7.7532e+00, -3.8337e+00,  ..., -1.5969e+00,\n",
            "           -1.0440e+01,  5.9430e+00],\n",
            "          [ 1.4812e+01, -5.1892e+00,  4.0852e+00,  ...,  4.9246e+00,\n",
            "           -1.0861e+01,  2.3091e+00],\n",
            "          [ 1.8629e+01, -3.6901e+00,  3.4407e+00,  ..., -1.9348e-02,\n",
            "           -1.5172e+01,  9.6978e-01],\n",
            "          ...,\n",
            "          [ 7.0007e+00, -3.4564e+00, -2.9741e+00,  ..., -5.2179e+00,\n",
            "            3.3514e+00,  1.4844e+00],\n",
            "          [ 7.0007e+00, -3.4564e+00, -2.9741e+00,  ..., -5.2179e+00,\n",
            "            3.3514e+00,  1.4844e+00],\n",
            "          [ 7.0007e+00, -3.4564e+00, -2.9741e+00,  ..., -5.2179e+00,\n",
            "            3.3514e+00,  1.4844e+00]],\n",
            "\n",
            "         [[-1.7062e+01,  5.4019e+00,  1.3390e+01,  ..., -4.8165e+00,\n",
            "            6.6586e+00, -1.1398e+01],\n",
            "          [-8.4186e+00,  5.8729e+00,  7.2055e+00,  ..., -2.4890e+00,\n",
            "            1.1885e+01, -2.4480e+00],\n",
            "          [-1.1434e+01,  7.2005e+00,  1.3672e+01,  ..., -9.6205e+00,\n",
            "            1.8502e+01, -2.8334e+00],\n",
            "          ...,\n",
            "          [-2.3089e+00,  4.8678e+00,  1.7285e+00,  ..., -1.3942e+00,\n",
            "            1.8642e+00,  5.8069e+00],\n",
            "          [-2.3089e+00,  4.8678e+00,  1.7285e+00,  ..., -1.3942e+00,\n",
            "            1.8642e+00,  5.8069e+00],\n",
            "          [-2.3089e+00,  4.8678e+00,  1.7285e+00,  ..., -1.3942e+00,\n",
            "            1.8642e+00,  5.8069e+00]],\n",
            "\n",
            "         [[ 4.1102e+00,  7.3445e+00,  5.0084e+01,  ...,  7.4787e+00,\n",
            "            6.0424e+00,  5.2179e+00],\n",
            "          [ 1.4119e+00,  9.1477e+00,  3.1595e+01,  ...,  6.6758e+00,\n",
            "            4.1065e+00,  5.9571e+00],\n",
            "          [ 3.0169e+00,  1.6834e+00,  5.1414e+01,  ...,  4.2049e+00,\n",
            "            7.0184e+00,  7.5474e+00],\n",
            "          ...,\n",
            "          [-6.0597e+00,  2.8833e-01,  1.8932e+01,  ...,  7.6158e+00,\n",
            "            4.2697e-01, -1.8903e+00],\n",
            "          [-6.0597e+00,  2.8833e-01,  1.8932e+01,  ...,  7.6158e+00,\n",
            "            4.2697e-01, -1.8903e+00],\n",
            "          [-6.0597e+00,  2.8833e-01,  1.8932e+01,  ...,  7.6158e+00,\n",
            "            4.2697e-01, -1.8903e+00]],\n",
            "\n",
            "         [[-1.3666e+01, -1.6698e+01, -6.5536e+00,  ...,  2.2492e+00,\n",
            "           -1.6749e+01,  6.2600e-01],\n",
            "          [-7.5153e+00, -1.3472e+01,  2.7277e+00,  ...,  8.0926e+00,\n",
            "           -2.7369e+00,  9.4950e+00],\n",
            "          [-1.1762e+01, -2.0614e+01,  5.4067e+00,  ...,  3.6713e+00,\n",
            "           -6.7724e+00,  1.0586e+01],\n",
            "          ...,\n",
            "          [ 3.9710e+00, -4.3405e+00, -2.1187e+00,  ..., -7.6705e+00,\n",
            "           -3.7663e+00, -9.1344e+00],\n",
            "          [ 3.9710e+00, -4.3405e+00, -2.1187e+00,  ..., -7.6705e+00,\n",
            "           -3.7663e+00, -9.1344e+00],\n",
            "          [ 3.9710e+00, -4.3405e+00, -2.1187e+00,  ..., -7.6705e+00,\n",
            "           -3.7663e+00, -9.1344e+00]]],\n",
            "\n",
            "\n",
            "        ...,\n",
            "\n",
            "\n",
            "        [[[ 7.5069e+00, -1.2759e+01,  3.9459e+00,  ..., -5.5124e+00,\n",
            "           -5.3640e+00, -3.6873e-01],\n",
            "          [ 5.3725e+00, -3.8471e+00,  5.7370e+00,  ..., -4.5132e+00,\n",
            "           -7.8859e+00,  1.3797e-01],\n",
            "          [ 2.9665e+00, -6.0488e+00,  1.2338e+01,  ...,  7.4512e+00,\n",
            "           -5.5651e+00, -3.2174e+00],\n",
            "          ...,\n",
            "          [ 5.8702e+00, -2.4928e+00, -1.0550e+00,  ..., -3.8633e+00,\n",
            "            2.8126e+00,  1.0252e+00],\n",
            "          [ 5.8702e+00, -2.4928e+00, -1.0550e+00,  ..., -3.8633e+00,\n",
            "            2.8126e+00,  1.0252e+00],\n",
            "          [ 5.8702e+00, -2.4928e+00, -1.0550e+00,  ..., -3.8633e+00,\n",
            "            2.8126e+00,  1.0252e+00]],\n",
            "\n",
            "         [[-1.3606e+01,  9.9082e+00,  4.6178e+00,  ..., -8.7840e+00,\n",
            "            6.4145e+00, -5.3909e+00],\n",
            "          [-3.7355e+00,  1.5040e+01,  4.1462e+00,  ...,  3.5804e+00,\n",
            "            1.1396e+00, -1.1624e+01],\n",
            "          [-4.0870e+00,  1.3022e+01, -1.0699e+00,  ...,  3.6482e+00,\n",
            "            3.3842e+00, -5.0313e+00],\n",
            "          ...,\n",
            "          [ 1.0950e-01,  5.1259e+00, -7.4879e-01,  ..., -3.9910e+00,\n",
            "           -1.4614e-01,  5.7668e+00],\n",
            "          [ 1.0950e-01,  5.1259e+00, -7.4879e-01,  ..., -3.9910e+00,\n",
            "           -1.4614e-01,  5.7668e+00],\n",
            "          [ 1.0950e-01,  5.1259e+00, -7.4879e-01,  ..., -3.9910e+00,\n",
            "           -1.4614e-01,  5.7668e+00]],\n",
            "\n",
            "         [[ 8.4492e+00,  1.5906e+00,  5.6002e+01,  ..., -7.7830e+00,\n",
            "            1.3418e+01,  7.9531e+00],\n",
            "          [ 8.1037e-01, -2.0311e+00,  3.6154e+01,  ..., -4.3067e+00,\n",
            "            5.1671e+00,  3.9233e+00],\n",
            "          [ 2.8364e+00,  1.7718e+00,  4.4258e+01,  ..., -6.5614e+00,\n",
            "            5.0687e-01,  5.2922e+00],\n",
            "          ...,\n",
            "          [-6.0606e+00, -5.3729e-01,  2.0001e+01,  ...,  6.1185e+00,\n",
            "            4.8084e-01, -2.1426e+00],\n",
            "          [-6.0606e+00, -5.3729e-01,  2.0001e+01,  ...,  6.1185e+00,\n",
            "            4.8084e-01, -2.1426e+00],\n",
            "          [-6.0606e+00, -5.3729e-01,  2.0001e+01,  ...,  6.1185e+00,\n",
            "            4.8084e-01, -2.1426e+00]],\n",
            "\n",
            "         [[-9.6960e+00, -8.6340e+00, -1.7107e+00,  ...,  7.6305e+00,\n",
            "           -1.1620e+01,  6.7049e+00],\n",
            "          [ 5.7984e-01, -1.8454e+00, -3.1313e+00,  ...,  4.1330e+00,\n",
            "           -3.4930e+00,  1.0934e+01],\n",
            "          [ 3.2683e+00, -9.4026e-01,  5.2964e-01,  ...,  7.5530e+00,\n",
            "           -7.2679e+00,  7.2473e+00],\n",
            "          ...,\n",
            "          [ 1.8882e+00, -3.8396e+00, -3.9242e+00,  ..., -5.3607e+00,\n",
            "           -5.6311e+00, -7.9887e+00],\n",
            "          [ 1.8882e+00, -3.8396e+00, -3.9242e+00,  ..., -5.3607e+00,\n",
            "           -5.6311e+00, -7.9887e+00],\n",
            "          [ 1.8882e+00, -3.8396e+00, -3.9242e+00,  ..., -5.3607e+00,\n",
            "           -5.6311e+00, -7.9887e+00]]],\n",
            "\n",
            "\n",
            "        [[[ 1.5718e+01, -8.6686e+00,  9.9529e+00,  ..., -6.7082e+00,\n",
            "           -5.1796e+00,  5.9798e+00],\n",
            "          [ 1.3883e+01, -8.4127e+00,  1.7909e+00,  ...,  9.4714e-01,\n",
            "           -2.8994e+00,  2.3587e+00],\n",
            "          [ 3.6034e+00, -1.5994e+00,  6.6310e+00,  ...,  5.4347e-01,\n",
            "           -1.1784e+00,  8.9172e+00],\n",
            "          ...,\n",
            "          [ 1.2962e+01, -1.3400e+01, -3.9550e+00,  ..., -6.0885e+00,\n",
            "           -1.1268e+01,  6.1946e+00],\n",
            "          [ 1.2749e+01, -8.7989e+00,  4.3529e+00,  ...,  1.2897e-03,\n",
            "           -1.3112e+01,  4.1569e+00],\n",
            "          [ 1.5078e+00,  1.8180e+00, -1.1593e+00,  ..., -3.4095e-01,\n",
            "            5.5643e-01,  1.3399e+00]],\n",
            "\n",
            "         [[-1.8899e+01,  3.1322e+00,  1.1172e+01,  ..., -1.2671e+01,\n",
            "            1.5551e+01, -4.2845e+00],\n",
            "          [-1.4999e+00,  6.5676e+00,  1.1186e+01,  ...,  9.6558e-01,\n",
            "           -5.5494e-01, -1.0584e+01],\n",
            "          [-1.0413e+01,  3.7452e+00, -6.5990e+00,  ..., -1.0748e+01,\n",
            "            2.3116e+00, -1.4702e+01],\n",
            "          ...,\n",
            "          [-6.6921e+00,  2.2905e+00, -6.5829e-01,  ...,  1.1180e+01,\n",
            "            7.6458e+00, -1.9751e+01],\n",
            "          [-6.9082e+00,  2.1222e+00,  2.2849e+00,  ...,  2.0111e+00,\n",
            "            6.7101e+00, -6.1835e+00],\n",
            "          [ 4.6204e+00, -1.0245e+01,  4.1006e+00,  ..., -1.2916e+00,\n",
            "           -2.3716e+00,  2.7574e+00]],\n",
            "\n",
            "         [[ 3.5292e+00,  8.2128e+00,  4.9813e+01,  ...,  7.1458e+00,\n",
            "            1.0880e+01,  1.3539e+01],\n",
            "          [-1.5037e+00,  1.2505e+00,  4.0668e+01,  ..., -1.5659e-02,\n",
            "            8.4721e+00,  8.8088e+00],\n",
            "          [-2.5802e+00, -8.1895e-01,  4.0186e+01,  ...,  3.0501e+00,\n",
            "            9.1670e+00,  5.2530e+00],\n",
            "          ...,\n",
            "          [-1.0599e+00,  5.1464e+00,  3.3004e+01,  ...,  9.7601e+00,\n",
            "            2.0385e+00, -4.6373e+00],\n",
            "          [-5.0445e+00,  3.6683e+00,  3.2654e+01,  ...,  6.2420e+00,\n",
            "            3.4533e-01, -1.2585e+00],\n",
            "          [-2.9575e+00, -6.9179e-01, -1.8518e+01,  ...,  5.7035e+00,\n",
            "           -7.9744e-02,  2.6208e+00]],\n",
            "\n",
            "         [[-1.5438e+01, -1.1577e+01,  7.6648e+00,  ...,  8.2966e+00,\n",
            "           -7.0649e+00,  4.2062e+00],\n",
            "          [-1.6716e+01, -2.5800e+00, -2.1288e-01,  ...,  2.3862e+00,\n",
            "            5.9969e+00,  4.9213e+00],\n",
            "          [-1.2190e+01, -2.2830e+00,  2.0374e+00,  ...,  5.8593e+00,\n",
            "            3.1359e+00,  1.5572e+01],\n",
            "          ...,\n",
            "          [-1.7822e+01, -5.8514e+00, -5.5813e+00,  ..., -3.8120e+00,\n",
            "            1.6406e+00,  1.9905e+00],\n",
            "          [-1.9307e+01, -2.4623e+00,  4.3790e+00,  ...,  3.1260e+00,\n",
            "           -1.1336e+01,  2.4804e+00],\n",
            "          [ 3.9764e+00,  4.3333e+00,  1.3576e+00,  ..., -4.4774e+00,\n",
            "           -3.8839e+00, -2.4435e+00]]],\n",
            "\n",
            "\n",
            "        [[[ 1.4752e+00, -1.6171e+01,  5.9247e-01,  ..., -7.7286e+00,\n",
            "            1.2798e+00,  3.8463e+00],\n",
            "          [ 6.0885e+00, -1.0327e+01,  7.6137e+00,  ..., -1.2978e+01,\n",
            "            7.6472e-01, -4.2274e+00],\n",
            "          [ 9.9017e+00, -1.1611e+01,  7.4145e-01,  ..., -8.3944e+00,\n",
            "           -1.1813e+00, -5.1615e+00],\n",
            "          ...,\n",
            "          [ 4.3980e+00, -2.5671e+00,  1.7273e+00,  ..., -2.3606e+00,\n",
            "            3.3193e-01,  1.3489e+00],\n",
            "          [ 4.4250e+00, -2.5600e+00,  1.7539e+00,  ..., -2.3568e+00,\n",
            "            3.4078e-01,  1.3251e+00],\n",
            "          [ 4.5449e+00, -2.4780e+00,  2.2542e+00,  ..., -2.1215e+00,\n",
            "            1.1040e-01,  1.3963e+00]],\n",
            "\n",
            "         [[-1.0074e+01, -9.1588e-01,  1.2265e+00,  ..., -9.9983e+00,\n",
            "            7.6286e+00, -2.9164e+00],\n",
            "          [-7.5409e+00,  4.1734e+00, -8.5190e+00,  ..., -6.0026e+00,\n",
            "            5.5841e+00, -3.5309e+00],\n",
            "          [-1.2940e+01,  1.0957e+01, -9.6477e-01,  ..., -5.7062e+00,\n",
            "            6.1218e+00, -5.6498e+00],\n",
            "          ...,\n",
            "          [ 1.4659e+00,  5.4132e+00, -2.3523e+00,  ..., -2.6915e+00,\n",
            "            2.8586e+00,  1.1085e+01],\n",
            "          [ 1.4780e+00,  5.4158e+00, -2.4061e+00,  ..., -2.7246e+00,\n",
            "            2.9232e+00,  1.1093e+01],\n",
            "          [ 1.3408e+00,  5.2338e+00, -2.0902e+00,  ..., -2.8151e+00,\n",
            "            2.7510e+00,  1.1274e+01]],\n",
            "\n",
            "         [[ 4.4357e+00,  7.4714e+00,  2.6540e+01,  ...,  2.5075e+00,\n",
            "            8.5027e+00,  8.6995e+00],\n",
            "          [ 1.7737e+00, -5.8411e+00,  1.3432e+01,  ...,  6.5776e+00,\n",
            "            1.0966e+00,  3.4247e+00],\n",
            "          [-5.0054e-01,  1.2962e+00,  2.4723e+01,  ...,  2.6603e+00,\n",
            "            1.0361e+01,  7.7377e+00],\n",
            "          ...,\n",
            "          [-4.3052e+00, -2.4063e+00,  1.8778e+01,  ...,  1.7797e+00,\n",
            "            2.0635e+00, -2.5499e+00],\n",
            "          [-4.2097e+00, -2.3691e+00,  1.8719e+01,  ...,  1.7946e+00,\n",
            "            2.0332e+00, -2.5154e+00],\n",
            "          [-4.0562e+00, -2.5918e+00,  1.8586e+01,  ...,  1.8044e+00,\n",
            "            1.8547e+00, -2.3637e+00]],\n",
            "\n",
            "         [[-1.1507e+01, -1.1801e+01, -1.0941e+00,  ...,  1.1620e+01,\n",
            "           -4.4240e+00,  4.3978e+00],\n",
            "          [-5.6528e+00,  2.4428e+00, -1.9411e+00,  ...,  1.0405e+01,\n",
            "            2.5577e+00,  1.3280e+01],\n",
            "          [-5.9267e+00, -5.5579e+00, -3.4524e+00,  ...,  1.2126e+01,\n",
            "            6.9094e+00,  1.1583e+01],\n",
            "          ...,\n",
            "          [ 1.7246e+00, -2.8735e+00, -1.9983e+00,  ..., -2.5788e+00,\n",
            "           -4.9378e+00, -8.9946e+00],\n",
            "          [ 1.7379e+00, -2.8587e+00, -1.8847e+00,  ..., -2.5488e+00,\n",
            "           -4.9442e+00, -8.9856e+00],\n",
            "          [ 1.8569e+00, -2.7811e+00, -1.3060e+00,  ..., -2.1294e+00,\n",
            "           -4.8938e+00, -8.9868e+00]]]], grad_fn=<TransposeBackward0>), tensor([[[[-3.8478e+00,  3.4887e+00,  5.8346e+00,  ..., -1.5494e+01,\n",
            "           -5.0105e+00, -9.8668e+00],\n",
            "          [ 7.8352e+00, -3.4453e+00, -6.9625e-01,  ..., -1.0050e+01,\n",
            "            4.5708e-01, -1.0437e+01],\n",
            "          [ 5.5157e+00, -4.7394e+00,  1.0009e+00,  ..., -1.6707e+01,\n",
            "           -1.4343e+01, -7.4668e+00],\n",
            "          ...,\n",
            "          [ 3.4656e+00,  9.3728e-01, -5.6429e+00,  ..., -9.5791e+00,\n",
            "           -1.4194e+00,  4.9818e+00],\n",
            "          [ 9.8247e+00,  2.6439e+00, -2.8794e+00,  ..., -1.4352e+00,\n",
            "            2.8676e+00,  4.2054e+00],\n",
            "          [ 5.8788e+00,  1.7004e+00,  2.8334e+00,  ..., -1.0454e+00,\n",
            "            2.8600e+00,  5.6926e-01]],\n",
            "\n",
            "         [[-7.1466e+00, -8.1738e+00,  2.2540e+00,  ...,  4.6489e+00,\n",
            "            8.0690e+00,  1.7558e+00],\n",
            "          [ 1.3408e+01,  2.7004e+00, -1.8087e+00,  ..., -1.9232e+00,\n",
            "           -5.0131e+00,  2.2841e+01],\n",
            "          [ 1.8218e+01, -3.4725e+00,  5.8495e-01,  ..., -2.4272e+00,\n",
            "            1.0116e+01,  9.9590e+00],\n",
            "          ...,\n",
            "          [ 6.2408e+00, -2.0155e+01,  9.0044e+00,  ..., -6.5325e+00,\n",
            "            7.6515e+00,  1.2900e+00],\n",
            "          [-9.7994e+00, -5.7689e+00,  3.1545e-01,  ..., -2.6077e+00,\n",
            "            6.8552e+00,  8.1429e+00],\n",
            "          [-2.7261e+00, -3.1056e+00,  3.0383e-01,  ...,  1.7098e+00,\n",
            "           -1.1734e+00, -7.6431e-02]],\n",
            "\n",
            "         [[-5.9105e-01, -4.8727e+00,  2.4798e+00,  ...,  4.0163e+00,\n",
            "           -6.1694e+00, -1.3358e+01],\n",
            "          [ 6.7264e+00, -2.5989e+00, -5.1608e+00,  ...,  3.7489e-01,\n",
            "           -2.9307e+00, -7.0847e+00],\n",
            "          [ 7.5791e+00, -9.9938e+00, -1.1597e+01,  ...,  2.2652e+00,\n",
            "           -1.1834e+01, -9.3556e+00],\n",
            "          ...,\n",
            "          [ 3.2338e+00, -1.2498e+01, -6.2069e-01,  ...,  5.4223e+00,\n",
            "           -7.8964e+00,  3.1474e+00],\n",
            "          [ 7.3425e+00, -9.3169e+00, -2.2508e+00,  ...,  6.2748e-01,\n",
            "           -5.7626e+00, -8.8940e+00],\n",
            "          [ 5.2812e+00,  4.1628e+00, -3.3867e+00,  ..., -4.5184e+00,\n",
            "           -5.7714e+00, -1.8871e+00]],\n",
            "\n",
            "         [[-5.8493e+00, -8.9748e+00, -8.4816e+00,  ..., -7.5425e+00,\n",
            "           -1.0753e+01, -8.4886e+00],\n",
            "          [ 5.3788e+00, -8.9729e+00,  1.0337e+01,  ..., -4.5930e+00,\n",
            "            4.3650e+00,  4.8716e+00],\n",
            "          [-2.0555e+00, -5.7546e+00,  2.7167e+00,  ..., -3.8727e+00,\n",
            "            6.4701e+00, -1.7515e+01],\n",
            "          ...,\n",
            "          [-8.2921e+00, -4.2435e+00, -6.4609e+00,  ...,  5.8525e+00,\n",
            "           -6.0609e+00,  4.2688e+00],\n",
            "          [ 2.4177e+00,  3.2036e+00, -3.5412e+00,  ..., -7.4168e+00,\n",
            "            1.6289e+01, -1.5696e+00],\n",
            "          [ 3.6707e-01, -4.2396e+00, -3.9626e-01,  ...,  1.3167e+00,\n",
            "            2.1276e+00,  3.0915e+00]]],\n",
            "\n",
            "\n",
            "        [[[-7.1349e+00, -1.0326e+01,  2.7768e+00,  ..., -3.9308e+00,\n",
            "           -6.5347e+00, -1.7871e+01],\n",
            "          [-7.5099e+00, -1.2783e+00,  2.6060e+00,  ..., -1.2060e+01,\n",
            "            1.1174e+00, -1.4663e+01],\n",
            "          [-5.9866e+00,  5.9811e+00, -1.4946e-01,  ..., -4.3524e+00,\n",
            "           -2.3492e+00, -1.3588e+01],\n",
            "          ...,\n",
            "          [-3.3650e+00, -3.8040e+00, -1.3310e+00,  ..., -2.3068e+00,\n",
            "            5.7652e+00, -3.8967e+00],\n",
            "          [-1.5426e+01, -5.9883e-01,  2.8453e+00,  ..., -7.8066e+00,\n",
            "           -3.2552e+00, -5.4600e+00],\n",
            "          [ 5.7624e+00,  1.6011e+00,  2.6720e+00,  ..., -6.6038e-01,\n",
            "            2.9090e+00,  5.4190e-01]],\n",
            "\n",
            "         [[-3.5237e+00, -5.1076e+00, -6.6068e+00,  ..., -1.1096e+01,\n",
            "           -2.1066e-02, -2.3308e+00],\n",
            "          [-1.0950e+01, -7.4165e+00, -3.6851e+00,  ..., -7.8872e+00,\n",
            "            1.7252e+00,  5.4883e+00],\n",
            "          [-4.2150e+00, -6.7687e+00,  1.8699e+00,  ..., -5.2568e+00,\n",
            "           -8.8528e+00,  5.6229e+00],\n",
            "          ...,\n",
            "          [ 1.7779e+00, -4.4058e+00,  5.8707e+00,  ...,  9.1273e+00,\n",
            "            1.7443e+00, -3.7815e+00],\n",
            "          [-6.0172e+00, -2.3306e+00,  9.2315e+00,  ...,  4.3353e+00,\n",
            "           -2.2602e+00, -3.6239e+00],\n",
            "          [-2.8708e+00, -2.5008e+00,  2.6163e-01,  ...,  1.4494e+00,\n",
            "           -5.6995e-01,  1.2553e-03]],\n",
            "\n",
            "         [[ 1.0312e+01,  5.3964e-01, -1.4162e+01,  ...,  2.3409e+00,\n",
            "           -3.3122e+00,  1.3437e+00],\n",
            "          [ 3.7230e+00,  5.5475e+00, -2.6308e+00,  ...,  3.7367e+00,\n",
            "           -3.1625e+00, -4.4901e-01],\n",
            "          [ 8.6075e+00,  1.4368e+00, -1.2697e+00,  ..., -5.1632e+00,\n",
            "            3.0533e-01, -1.8021e-01],\n",
            "          ...,\n",
            "          [ 1.7790e+00,  6.4392e-01, -1.2449e+01,  ...,  4.8788e+00,\n",
            "           -1.5019e+00,  9.6650e-01],\n",
            "          [ 7.3485e+00, -1.5255e+00, -4.5411e+00,  ...,  1.8550e+00,\n",
            "           -5.6343e+00,  7.9809e+00],\n",
            "          [ 5.3853e+00,  4.3592e+00, -3.4723e+00,  ..., -4.4627e+00,\n",
            "           -6.1487e+00, -1.7927e+00]],\n",
            "\n",
            "         [[-1.8960e+01,  5.6857e+00, -6.4330e+00,  ...,  3.1389e+00,\n",
            "           -2.8457e+00,  4.8824e+00],\n",
            "          [-3.0249e+00,  2.8683e+00, -2.1211e+01,  ...,  1.0353e+00,\n",
            "            4.5293e+00, -7.4096e+00],\n",
            "          [-2.1208e+00,  1.1336e+01,  7.0456e+00,  ...,  6.9149e+00,\n",
            "            9.9246e+00,  1.2788e-01],\n",
            "          ...,\n",
            "          [ 5.3658e+00, -7.8448e+00, -2.5039e+00,  ...,  1.7503e+01,\n",
            "            1.0941e+00,  6.1777e+00],\n",
            "          [-1.3422e+00, -7.1899e+00,  1.0815e+01,  ...,  1.1081e+01,\n",
            "           -4.9052e+00,  3.4271e+00],\n",
            "          [-1.0838e-01, -4.1223e+00, -8.6422e-01,  ...,  1.1119e+00,\n",
            "            1.2702e+00,  3.2339e+00]]],\n",
            "\n",
            "\n",
            "        [[[-1.3787e+00,  5.7546e+00, -3.4008e+00,  ..., -8.7807e+00,\n",
            "           -5.0515e+00, -1.0652e+01],\n",
            "          [-3.2304e+00,  1.4362e+00,  6.1492e-01,  ..., -7.3294e+00,\n",
            "           -7.3667e+00, -1.1617e+01],\n",
            "          [-3.2627e+00,  3.2399e+00,  9.4591e+00,  ..., -1.1770e+01,\n",
            "           -1.0225e+01, -1.5316e+01],\n",
            "          ...,\n",
            "          [ 7.5640e+00, -3.7438e+00,  5.0539e+00,  ..., -8.8856e+00,\n",
            "            3.2796e+00, -1.3514e+00],\n",
            "          [ 7.5640e+00, -3.7438e+00,  5.0539e+00,  ..., -8.8856e+00,\n",
            "            3.2796e+00, -1.3514e+00],\n",
            "          [ 7.5640e+00, -3.7438e+00,  5.0539e+00,  ..., -8.8856e+00,\n",
            "            3.2796e+00, -1.3514e+00]],\n",
            "\n",
            "         [[ 2.9722e+00, -4.4324e+00, -3.2884e+00,  ..., -3.3458e+00,\n",
            "           -2.6168e+00, -2.3233e+00],\n",
            "          [-8.3249e+00, -1.3558e+01,  3.8806e+00,  ..., -2.0484e+00,\n",
            "            8.5217e+00, -1.4897e+00],\n",
            "          [-3.0639e+00, -9.7909e+00, -1.3271e+00,  ..., -4.4735e+00,\n",
            "            1.1679e+01,  2.6218e+00],\n",
            "          ...,\n",
            "          [ 3.9949e+00, -8.4641e+00, -5.2835e+00,  ..., -5.6685e+00,\n",
            "            5.8177e+00, -6.2566e-01],\n",
            "          [ 3.9949e+00, -8.4641e+00, -5.2835e+00,  ..., -5.6685e+00,\n",
            "            5.8177e+00, -6.2566e-01],\n",
            "          [ 3.9949e+00, -8.4641e+00, -5.2835e+00,  ..., -5.6685e+00,\n",
            "            5.8177e+00, -6.2566e-01]],\n",
            "\n",
            "         [[ 1.1109e+01,  3.9454e+00, -2.7441e+00,  ...,  1.0858e+01,\n",
            "            6.0420e+00,  1.9535e+00],\n",
            "          [ 5.1600e+00, -1.8918e+00, -6.0201e+00,  ...,  5.4360e+00,\n",
            "           -1.0563e+00,  5.8945e+00],\n",
            "          [ 1.0453e+01,  3.5986e+00, -1.5757e+01,  ...,  1.2322e+01,\n",
            "            1.0364e+00,  3.6474e+00],\n",
            "          ...,\n",
            "          [ 1.8904e+00, -1.1778e+01, -5.2064e+00,  ..., -1.0047e+00,\n",
            "            2.4508e+00,  2.8458e+00],\n",
            "          [ 1.8904e+00, -1.1778e+01, -5.2064e+00,  ..., -1.0047e+00,\n",
            "            2.4508e+00,  2.8458e+00],\n",
            "          [ 1.8904e+00, -1.1778e+01, -5.2064e+00,  ..., -1.0047e+00,\n",
            "            2.4508e+00,  2.8458e+00]],\n",
            "\n",
            "         [[-3.8592e+00, -2.4710e+00, -8.5691e+00,  ...,  1.3029e+01,\n",
            "           -1.1915e+00, -1.4377e+00],\n",
            "          [ 4.8427e+00, -7.4170e+00,  7.0921e+00,  ...,  8.7348e+00,\n",
            "            4.7486e+00, -7.9342e+00],\n",
            "          [ 1.1063e+01,  8.1711e+00, -3.3758e+00,  ...,  9.7488e+00,\n",
            "           -6.9223e+00, -1.0950e+01],\n",
            "          ...,\n",
            "          [-2.2375e+00,  3.9694e+00,  3.4311e+00,  ..., -3.0671e+00,\n",
            "            2.7979e+00, -4.7249e+00],\n",
            "          [-2.2375e+00,  3.9694e+00,  3.4311e+00,  ..., -3.0671e+00,\n",
            "            2.7979e+00, -4.7249e+00],\n",
            "          [-2.2375e+00,  3.9694e+00,  3.4311e+00,  ..., -3.0671e+00,\n",
            "            2.7979e+00, -4.7249e+00]]],\n",
            "\n",
            "\n",
            "        ...,\n",
            "\n",
            "\n",
            "        [[[-5.1439e+00, -1.6159e-01,  3.8850e+00,  ..., -7.7084e+00,\n",
            "           -8.6721e+00, -1.6245e+01],\n",
            "          [-3.2516e+00,  8.9324e+00,  4.7444e+00,  ..., -8.4932e+00,\n",
            "           -1.3752e+00, -4.2731e+00],\n",
            "          [-7.9847e-01,  1.5619e+00,  4.6799e+00,  ..., -1.6439e+01,\n",
            "            3.3714e+00, -5.8030e+00],\n",
            "          ...,\n",
            "          [ 8.9655e+00, -5.6331e+00,  2.8480e+00,  ..., -1.0369e+01,\n",
            "            1.7252e+00, -3.5539e+00],\n",
            "          [ 8.9655e+00, -5.6331e+00,  2.8480e+00,  ..., -1.0369e+01,\n",
            "            1.7252e+00, -3.5539e+00],\n",
            "          [ 8.9655e+00, -5.6331e+00,  2.8480e+00,  ..., -1.0369e+01,\n",
            "            1.7252e+00, -3.5539e+00]],\n",
            "\n",
            "         [[-8.7326e+00, -1.3530e+01,  4.4084e+00,  ..., -8.6749e+00,\n",
            "           -5.2533e+00,  2.5417e+00],\n",
            "          [-1.3376e+01,  5.4424e+00,  8.3156e+00,  ..., -1.1661e+01,\n",
            "           -8.0717e+00,  6.8724e+00],\n",
            "          [-1.6231e+01,  5.0149e+00,  1.8274e+01,  ..., -1.0858e+01,\n",
            "           -1.2040e+01,  2.3359e+00],\n",
            "          ...,\n",
            "          [ 3.2939e+00, -9.0267e+00, -5.4274e+00,  ..., -6.1410e+00,\n",
            "            1.8447e+00, -2.1993e+00],\n",
            "          [ 3.2939e+00, -9.0267e+00, -5.4274e+00,  ..., -6.1410e+00,\n",
            "            1.8447e+00, -2.1993e+00],\n",
            "          [ 3.2939e+00, -9.0267e+00, -5.4274e+00,  ..., -6.1410e+00,\n",
            "            1.8447e+00, -2.1993e+00]],\n",
            "\n",
            "         [[-8.1547e+00, -1.1144e+01, -4.9879e+00,  ...,  8.0337e+00,\n",
            "           -1.8666e+00, -5.5852e+00],\n",
            "          [-3.2764e+00, -1.3924e+01, -2.5398e+00,  ...,  5.0625e+00,\n",
            "           -1.6843e+00,  6.5021e-02],\n",
            "          [-2.0254e+00, -2.0507e+01,  2.6739e+00,  ...,  6.1729e+00,\n",
            "            4.0245e+00, -3.0401e+00],\n",
            "          ...,\n",
            "          [ 3.3832e-01, -1.4752e+01, -3.8233e+00,  ...,  1.5785e+00,\n",
            "            1.1974e+00,  2.3907e+00],\n",
            "          [ 3.3832e-01, -1.4752e+01, -3.8233e+00,  ...,  1.5785e+00,\n",
            "            1.1974e+00,  2.3907e+00],\n",
            "          [ 3.3832e-01, -1.4752e+01, -3.8233e+00,  ...,  1.5785e+00,\n",
            "            1.1974e+00,  2.3907e+00]],\n",
            "\n",
            "         [[-6.1911e+00, -2.2925e+00, -7.5193e+00,  ...,  8.0455e-01,\n",
            "           -4.8217e+00, -2.9949e+00],\n",
            "          [-1.8835e-01, -2.4605e+01, -9.2383e+00,  ..., -5.7698e+00,\n",
            "            4.4672e+00, -1.8641e+01],\n",
            "          [ 7.4639e+00, -1.0627e+01,  5.6196e+00,  ..., -5.6326e-01,\n",
            "            1.9796e+01, -1.8278e+01],\n",
            "          ...,\n",
            "          [-1.3693e+00,  2.5494e+00,  3.3600e+00,  ..., -4.3429e+00,\n",
            "            1.4279e+00, -7.6449e+00],\n",
            "          [-1.3693e+00,  2.5494e+00,  3.3600e+00,  ..., -4.3429e+00,\n",
            "            1.4279e+00, -7.6449e+00],\n",
            "          [-1.3693e+00,  2.5494e+00,  3.3600e+00,  ..., -4.3429e+00,\n",
            "            1.4279e+00, -7.6449e+00]]],\n",
            "\n",
            "\n",
            "        [[[-2.4745e+00,  1.1166e+01, -3.1464e+00,  ..., -1.9837e+01,\n",
            "           -1.0359e+01, -2.2810e+01],\n",
            "          [-7.0571e+00,  9.8328e-01,  3.3746e+00,  ..., -1.2451e+01,\n",
            "           -7.3598e-01, -2.2298e+01],\n",
            "          [ 4.3883e-01,  1.3041e+01,  7.1481e+00,  ..., -9.4398e+00,\n",
            "           -1.1565e+01, -1.0848e+01],\n",
            "          ...,\n",
            "          [ 8.7646e+00, -9.8589e-01,  2.9103e+00,  ..., -1.9493e+01,\n",
            "           -5.3893e+00,  3.5230e+00],\n",
            "          [-5.4742e+00, -1.1614e-02, -3.9329e-01,  ..., -6.1042e+00,\n",
            "           -4.8504e+00,  5.9411e+00],\n",
            "          [ 5.4490e+00,  1.7690e+00,  2.7763e+00,  ..., -6.4235e-01,\n",
            "            2.7484e+00,  4.2469e-01]],\n",
            "\n",
            "         [[ 3.5225e+00, -7.8729e+00,  2.4360e+00,  ..., -8.5960e+00,\n",
            "            1.4951e+00, -4.3668e+00],\n",
            "          [-8.9660e+00, -5.2958e+00, -1.6146e+00,  ..., -8.5147e+00,\n",
            "            7.8551e-01, -2.9026e+00],\n",
            "          [ 8.3598e+00, -5.0455e+00,  8.8724e+00,  ...,  5.7351e+00,\n",
            "           -1.6786e+00,  1.5771e+01],\n",
            "          ...,\n",
            "          [ 6.7829e+00,  2.9325e+00,  1.0264e+00,  ...,  1.1569e+01,\n",
            "           -7.8204e+00, -6.8916e+00],\n",
            "          [ 3.6193e+00,  7.4379e+00,  2.0370e+01,  ...,  2.7032e+01,\n",
            "           -4.7819e+00,  6.1792e+00],\n",
            "          [-2.4140e+00, -2.7511e+00,  1.0683e-01,  ...,  1.6658e+00,\n",
            "           -4.4223e-01, -6.2219e-01]],\n",
            "\n",
            "         [[ 1.1808e+01, -1.4800e+01, -2.5075e+00,  ...,  2.8243e+00,\n",
            "            1.1690e-01,  1.2431e+01],\n",
            "          [ 4.8605e+00, -1.0365e+01, -3.8634e+00,  ...,  4.1690e+00,\n",
            "           -3.2894e+00, -5.4879e+00],\n",
            "          [-5.0984e+00, -1.0015e+01, -5.8121e+00,  ...,  4.1906e+00,\n",
            "           -2.8036e+00,  2.3287e-01],\n",
            "          ...,\n",
            "          [ 6.5540e+00, -5.0281e+00, -1.4396e+01,  ...,  7.7047e+00,\n",
            "            5.8987e+00,  4.6953e-01],\n",
            "          [ 1.6988e+01, -1.1953e+01, -8.3338e+00,  ...,  3.7345e+00,\n",
            "           -1.9686e+00, -2.7894e+00],\n",
            "          [ 4.9832e+00,  4.3601e+00, -3.6427e+00,  ..., -4.5982e+00,\n",
            "           -5.7917e+00, -2.1571e+00]],\n",
            "\n",
            "         [[-1.9860e+01, -4.0457e+00,  2.7246e+00,  ..., -3.0634e+00,\n",
            "            8.3940e-01, -3.9363e+00],\n",
            "          [-7.4967e+00, -3.4303e-01, -3.0359e+00,  ..., -1.4879e+01,\n",
            "           -1.1697e+01, -3.9917e+00],\n",
            "          [ 1.4665e+00, -1.4737e+00,  1.5708e+01,  ..., -1.3255e+01,\n",
            "            6.3348e+00,  3.5937e+00],\n",
            "          ...,\n",
            "          [ 1.9573e+00,  6.9516e-01,  1.3092e+00,  ...,  7.2246e-01,\n",
            "           -1.0991e+01, -1.5614e+00],\n",
            "          [-6.5710e+00,  9.9511e+00, -7.0695e+00,  ...,  8.3590e+00,\n",
            "           -9.7549e+00, -1.4324e+00],\n",
            "          [ 8.5388e-01, -3.4034e+00, -1.4728e+00,  ...,  9.9392e-01,\n",
            "            1.4820e+00,  2.9974e+00]]],\n",
            "\n",
            "\n",
            "        [[[-1.2067e+01, -6.6793e-01, -9.6494e-01,  ..., -1.1361e+01,\n",
            "           -6.5030e+00, -1.4901e+01],\n",
            "          [-1.4701e+01,  4.9930e+00, -1.2013e+00,  ..., -4.8301e+00,\n",
            "           -7.0464e+00,  4.1967e+00],\n",
            "          [-1.1641e+01,  6.6806e+00, -1.2980e+00,  ..., -9.9349e-01,\n",
            "           -5.4646e+00, -2.7927e+00],\n",
            "          ...,\n",
            "          [ 1.1239e+01, -2.0624e+00,  3.3090e-01,  ..., -1.0212e+01,\n",
            "            3.7611e+00, -4.8373e+00],\n",
            "          [ 1.1323e+01, -2.0039e+00,  3.2172e-01,  ..., -1.0170e+01,\n",
            "            3.7410e+00, -4.8179e+00],\n",
            "          [ 1.1530e+01, -1.5229e+00,  3.2602e-01,  ..., -1.0101e+01,\n",
            "            3.4963e+00, -4.1566e+00]],\n",
            "\n",
            "         [[-3.8419e+00, -1.8472e+01, -1.1648e+00,  ..., -3.3552e+00,\n",
            "           -6.0573e+00,  2.2304e+00],\n",
            "          [-6.2327e+00,  6.0484e+00,  4.4365e+00,  ..., -1.2151e+01,\n",
            "            1.7852e+00, -5.5601e+00],\n",
            "          [-8.3978e+00, -5.3719e+00,  1.2150e+01,  ..., -9.6108e-01,\n",
            "            4.0135e+00, -9.0257e+00],\n",
            "          ...,\n",
            "          [-1.0465e+00, -1.5104e+00, -2.3855e+00,  ..., -4.3870e+00,\n",
            "            5.5384e+00, -2.8141e+00],\n",
            "          [-1.0998e+00, -1.5261e+00, -2.5169e+00,  ..., -4.4461e+00,\n",
            "            5.6132e+00, -2.8966e+00],\n",
            "          [-1.5989e+00, -1.5469e+00, -2.8073e+00,  ..., -4.9555e+00,\n",
            "            6.0462e+00, -2.9968e+00]],\n",
            "\n",
            "         [[ 1.2101e+01, -4.8616e-01, -9.1627e+00,  ...,  1.0712e+01,\n",
            "           -1.0670e+01, -5.3837e+00],\n",
            "          [ 2.7679e+00, -1.6507e+01, -1.2080e+00,  ...,  8.5825e+00,\n",
            "           -1.5287e+01, -9.1235e-01],\n",
            "          [ 1.6395e+00, -9.6316e+00, -9.2031e+00,  ...,  1.4468e+01,\n",
            "           -9.5623e+00,  6.2022e+00],\n",
            "          ...,\n",
            "          [ 3.1680e+00, -1.7686e+01, -5.4022e+00,  ..., -2.1780e+00,\n",
            "           -3.8706e+00,  5.2742e+00],\n",
            "          [ 3.1437e+00, -1.7652e+01, -5.5501e+00,  ..., -2.2180e+00,\n",
            "           -4.0042e+00,  5.3528e+00],\n",
            "          [ 3.4643e+00, -1.7345e+01, -5.5154e+00,  ..., -2.6183e+00,\n",
            "           -3.9391e+00,  5.8815e+00]],\n",
            "\n",
            "         [[-7.0776e+00, -3.0783e+00, -5.4244e+00,  ...,  4.6225e+00,\n",
            "           -1.3841e+01, -3.0950e+00],\n",
            "          [-1.5094e+01, -2.6014e+00,  4.5105e+00,  ..., -9.1105e+00,\n",
            "            9.2055e+00,  1.3905e+00],\n",
            "          [-9.0779e+00, -2.5611e+00,  2.4718e+00,  ...,  4.1110e+00,\n",
            "           -7.2708e+00, -7.8035e+00],\n",
            "          ...,\n",
            "          [ 1.3684e+00,  1.4526e+00,  5.0123e+00,  ..., -7.7338e+00,\n",
            "            4.5798e-01, -7.3749e+00],\n",
            "          [ 1.4347e+00,  1.3952e+00,  5.0642e+00,  ..., -7.7643e+00,\n",
            "            4.1894e-01, -7.3301e+00],\n",
            "          [ 1.1439e+00,  1.2376e+00,  5.2508e+00,  ..., -7.5270e+00,\n",
            "            4.4203e-01, -7.4510e+00]]]], grad_fn=<TransposeBackward0>)), (tensor([[[[ 0.0974,  0.0179, -0.0347,  ...,  0.6352,  0.1318,  0.0765],\n",
            "          [-0.1523, -0.2850, -0.0315,  ...,  0.8041,  0.0594, -0.0178],\n",
            "          [-0.1623,  0.3011, -0.0110,  ...,  0.6471,  0.4298,  0.1116],\n",
            "          ...,\n",
            "          [ 0.1012,  0.0093, -0.0339,  ...,  0.6454,  0.1202,  0.0759],\n",
            "          [ 0.1012,  0.0093, -0.0339,  ...,  0.6454,  0.1202,  0.0759],\n",
            "          [ 0.1008,  0.0093, -0.0340,  ...,  0.6455,  0.1202,  0.0762]],\n",
            "\n",
            "         [[ 1.0214, -0.5083, -0.0683,  ...,  0.4691, -1.2423,  0.4302],\n",
            "          [ 0.3621, -0.2233, -0.1418,  ...,  0.0625, -0.8258, -0.2775],\n",
            "          [ 0.2987, -0.4228,  0.1778,  ...,  0.5790, -0.4060,  0.4492],\n",
            "          ...,\n",
            "          [ 1.0188, -0.5088, -0.0747,  ...,  0.4666, -1.2419,  0.4244],\n",
            "          [ 1.0188, -0.5088, -0.0747,  ...,  0.4666, -1.2419,  0.4244],\n",
            "          [ 1.0188, -0.5089, -0.0744,  ...,  0.4666, -1.2417,  0.4245]],\n",
            "\n",
            "         [[-0.0757, -0.1137,  0.0881,  ...,  2.5366,  0.3077,  0.0741],\n",
            "          [-0.4094, -0.3064,  0.4627,  ...,  1.4702,  0.2712,  0.0978],\n",
            "          [ 0.0525, -0.4354,  0.6043,  ...,  1.9272,  0.8637,  0.0307],\n",
            "          ...,\n",
            "          [-0.0731, -0.1051,  0.0846,  ...,  2.5293,  0.3090,  0.0800],\n",
            "          [-0.0731, -0.1051,  0.0846,  ...,  2.5293,  0.3090,  0.0800],\n",
            "          [-0.0731, -0.1052,  0.0848,  ...,  2.5292,  0.3089,  0.0803]],\n",
            "\n",
            "         [[-0.3054, -0.1994, -0.2349,  ..., -0.2077,  0.1184, -0.1479],\n",
            "          [-0.3176, -0.4307,  0.4030,  ..., -0.1380,  0.0683, -0.2180],\n",
            "          [-0.3616,  0.1763,  0.1159,  ..., -0.3252,  0.1787, -0.6052],\n",
            "          ...,\n",
            "          [-0.3104, -0.2113, -0.2383,  ..., -0.1978,  0.1238, -0.1509],\n",
            "          [-0.3104, -0.2113, -0.2383,  ..., -0.1978,  0.1238, -0.1509],\n",
            "          [-0.3101, -0.2114, -0.2383,  ..., -0.1977,  0.1237, -0.1510]]],\n",
            "\n",
            "\n",
            "        [[[ 0.0985,  0.0229, -0.0384,  ...,  0.6327,  0.1346,  0.0733],\n",
            "          [-0.1372, -0.2656, -0.0329,  ...,  0.7940,  0.0619, -0.0320],\n",
            "          [-0.1846,  0.3318, -0.0267,  ...,  0.6194,  0.4240,  0.1345],\n",
            "          ...,\n",
            "          [ 0.1035,  0.0148, -0.0375,  ...,  0.6426,  0.1234,  0.0714],\n",
            "          [ 0.1035,  0.0148, -0.0375,  ...,  0.6426,  0.1234,  0.0714],\n",
            "          [ 0.1033,  0.0149, -0.0376,  ...,  0.6426,  0.1235,  0.0715]],\n",
            "\n",
            "         [[ 1.0228, -0.5025, -0.0706,  ...,  0.4654, -1.2488,  0.4285],\n",
            "          [ 0.3571, -0.2195, -0.1537,  ...,  0.0418, -0.8622, -0.3010],\n",
            "          [ 0.2919, -0.4232,  0.1867,  ...,  0.5511, -0.4360,  0.4358],\n",
            "          ...,\n",
            "          [ 1.0209, -0.5021, -0.0759,  ...,  0.4635, -1.2495,  0.4211],\n",
            "          [ 1.0209, -0.5021, -0.0759,  ...,  0.4635, -1.2495,  0.4211],\n",
            "          [ 1.0209, -0.5021, -0.0756,  ...,  0.4634, -1.2496,  0.4211]],\n",
            "\n",
            "         [[-0.0756, -0.1135,  0.0851,  ...,  2.5351,  0.3108,  0.0728],\n",
            "          [-0.4189, -0.3048,  0.4361,  ...,  1.4347,  0.2872,  0.1213],\n",
            "          [ 0.0353, -0.4047,  0.6291,  ...,  1.9054,  0.8463,  0.0527],\n",
            "          ...,\n",
            "          [-0.0744, -0.1052,  0.0823,  ...,  2.5279,  0.3122,  0.0768],\n",
            "          [-0.0744, -0.1052,  0.0823,  ...,  2.5279,  0.3122,  0.0768],\n",
            "          [-0.0744, -0.1053,  0.0826,  ...,  2.5278,  0.3122,  0.0769]],\n",
            "\n",
            "         [[-0.3041, -0.1997, -0.2362,  ..., -0.2091,  0.1182, -0.1437],\n",
            "          [-0.3113, -0.4463,  0.4101,  ..., -0.1677,  0.0515, -0.2206],\n",
            "          [-0.3527,  0.2117,  0.1325,  ..., -0.3633,  0.1896, -0.6161],\n",
            "          ...,\n",
            "          [-0.3096, -0.2115, -0.2385,  ..., -0.1995,  0.1238, -0.1463],\n",
            "          [-0.3096, -0.2115, -0.2385,  ..., -0.1995,  0.1238, -0.1463],\n",
            "          [-0.3093, -0.2115, -0.2385,  ..., -0.1993,  0.1237, -0.1462]]],\n",
            "\n",
            "\n",
            "        [[[ 0.1004,  0.0183, -0.0333,  ...,  0.6317,  0.1339,  0.0713],\n",
            "          [-0.1424, -0.2866, -0.0543,  ...,  0.7861,  0.0472, -0.0172],\n",
            "          [-0.1579,  0.2935, -0.0187,  ...,  0.6651,  0.4049,  0.1136],\n",
            "          ...,\n",
            "          [ 0.1052,  0.0106, -0.0323,  ...,  0.6412,  0.1228,  0.0696],\n",
            "          [ 0.1052,  0.0106, -0.0323,  ...,  0.6412,  0.1228,  0.0696],\n",
            "          [ 0.1049,  0.0107, -0.0324,  ...,  0.6412,  0.1229,  0.0696]],\n",
            "\n",
            "         [[ 1.0233, -0.5069, -0.0742,  ...,  0.4657, -1.2481,  0.4323],\n",
            "          [ 0.3852, -0.2165, -0.1506,  ...,  0.0578, -0.8369, -0.2883],\n",
            "          [ 0.3446, -0.4337,  0.1258,  ...,  0.5514, -0.4138,  0.4663],\n",
            "          ...,\n",
            "          [ 1.0208, -0.5062, -0.0799,  ...,  0.4627, -1.2498,  0.4257],\n",
            "          [ 1.0208, -0.5062, -0.0799,  ...,  0.4627, -1.2498,  0.4257],\n",
            "          [ 1.0208, -0.5061, -0.0796,  ...,  0.4626, -1.2499,  0.4257]],\n",
            "\n",
            "         [[-0.0792, -0.1187,  0.0876,  ...,  2.5344,  0.3144,  0.0735],\n",
            "          [-0.4278, -0.3195,  0.4758,  ...,  1.4465,  0.2884,  0.0850],\n",
            "          [ 0.0171, -0.4389,  0.5800,  ...,  1.9331,  0.8572,  0.0417],\n",
            "          ...,\n",
            "          [-0.0772, -0.1103,  0.0850,  ...,  2.5263,  0.3166,  0.0771],\n",
            "          [-0.0772, -0.1103,  0.0850,  ...,  2.5263,  0.3166,  0.0771],\n",
            "          [-0.0772, -0.1104,  0.0853,  ...,  2.5262,  0.3166,  0.0772]],\n",
            "\n",
            "         [[-0.3094, -0.1923, -0.2345,  ..., -0.2070,  0.1213, -0.1454],\n",
            "          [-0.3407, -0.3827,  0.3946,  ..., -0.1070,  0.0634, -0.1953],\n",
            "          [-0.3784,  0.2233,  0.1016,  ..., -0.3108,  0.2079, -0.6280],\n",
            "          ...,\n",
            "          [-0.3140, -0.2042, -0.2372,  ..., -0.1971,  0.1269, -0.1474],\n",
            "          [-0.3140, -0.2042, -0.2372,  ..., -0.1971,  0.1269, -0.1474],\n",
            "          [-0.3138, -0.2042, -0.2371,  ..., -0.1970,  0.1268, -0.1474]]],\n",
            "\n",
            "\n",
            "        ...,\n",
            "\n",
            "\n",
            "        [[[ 0.0670, -0.0053, -0.0399,  ...,  0.6546,  0.1328,  0.1026],\n",
            "          [-0.2364, -0.3334, -0.0450,  ...,  0.8318,  0.0146,  0.0657],\n",
            "          [-0.2437,  0.2978, -0.0302,  ...,  0.6365,  0.3813,  0.1759],\n",
            "          ...,\n",
            "          [ 0.0714, -0.0119, -0.0372,  ...,  0.6621,  0.1215,  0.1017],\n",
            "          [ 0.0714, -0.0119, -0.0372,  ...,  0.6621,  0.1215,  0.1017],\n",
            "          [ 0.0710, -0.0120, -0.0372,  ...,  0.6621,  0.1215,  0.1020]],\n",
            "\n",
            "         [[ 1.0132, -0.5323, -0.0756,  ...,  0.4692, -1.1832,  0.4437],\n",
            "          [ 0.3912, -0.2304, -0.1821,  ...,  0.0693, -0.6756, -0.2218],\n",
            "          [ 0.2963, -0.3905,  0.1485,  ...,  0.6356, -0.1821,  0.5076],\n",
            "          ...,\n",
            "          [ 1.0124, -0.5312, -0.0813,  ...,  0.4661, -1.1871,  0.4381],\n",
            "          [ 1.0124, -0.5312, -0.0813,  ...,  0.4661, -1.1871,  0.4381],\n",
            "          [ 1.0124, -0.5313, -0.0810,  ...,  0.4661, -1.1870,  0.4380]],\n",
            "\n",
            "         [[-0.0613, -0.1134,  0.0796,  ...,  2.5530,  0.2891,  0.1029],\n",
            "          [-0.4213, -0.2741,  0.4844,  ...,  1.4934,  0.2455,  0.1027],\n",
            "          [ 0.0202, -0.4560,  0.6073,  ...,  1.8772,  0.8842,  0.0258],\n",
            "          ...,\n",
            "          [-0.0611, -0.1040,  0.0764,  ...,  2.5440,  0.2917,  0.1086],\n",
            "          [-0.0611, -0.1040,  0.0764,  ...,  2.5440,  0.2917,  0.1086],\n",
            "          [-0.0611, -0.1040,  0.0767,  ...,  2.5438,  0.2915,  0.1089]],\n",
            "\n",
            "         [[-0.3133, -0.1952, -0.2404,  ..., -0.2034,  0.1177, -0.1807],\n",
            "          [-0.3620, -0.3394,  0.3810,  ..., -0.0843,  0.1152, -0.2481],\n",
            "          [-0.3987,  0.2576,  0.1667,  ..., -0.2783,  0.1725, -0.6415],\n",
            "          ...,\n",
            "          [-0.3167, -0.2081, -0.2420,  ..., -0.1944,  0.1246, -0.1838],\n",
            "          [-0.3167, -0.2081, -0.2420,  ..., -0.1944,  0.1246, -0.1838],\n",
            "          [-0.3165, -0.2081, -0.2419,  ..., -0.1944,  0.1245, -0.1839]]],\n",
            "\n",
            "\n",
            "        [[[ 0.0957,  0.0212, -0.0346,  ...,  0.6329,  0.1340,  0.0736],\n",
            "          [-0.1268, -0.2715, -0.0237,  ...,  0.8001,  0.1014, -0.0496],\n",
            "          [-0.1496,  0.3093, -0.0104,  ...,  0.6462,  0.4740,  0.0981],\n",
            "          ...,\n",
            "          [ 0.0999,  0.0130, -0.0338,  ...,  0.6430,  0.1224,  0.0727],\n",
            "          [ 0.0999,  0.0130, -0.0338,  ...,  0.6430,  0.1224,  0.0727],\n",
            "          [ 0.0996,  0.0130, -0.0339,  ...,  0.6430,  0.1224,  0.0730]],\n",
            "\n",
            "         [[ 1.0233, -0.5063, -0.0664,  ...,  0.4633, -1.2483,  0.4295],\n",
            "          [ 0.3164, -0.2101, -0.1705,  ...,  0.0155, -0.8352, -0.2530],\n",
            "          [ 0.3254, -0.4264,  0.1787,  ...,  0.5481, -0.4281,  0.4498],\n",
            "          ...,\n",
            "          [ 1.0206, -0.5065, -0.0716,  ...,  0.4611, -1.2485,  0.4230],\n",
            "          [ 1.0206, -0.5065, -0.0716,  ...,  0.4611, -1.2485,  0.4230],\n",
            "          [ 1.0205, -0.5066, -0.0714,  ...,  0.4612, -1.2483,  0.4230]],\n",
            "\n",
            "         [[-0.0773, -0.1128,  0.0903,  ...,  2.5309,  0.3093,  0.0720],\n",
            "          [-0.3996, -0.2781,  0.4232,  ...,  1.4616,  0.2935,  0.1281],\n",
            "          [ 0.0310, -0.4240,  0.5783,  ...,  1.9187,  0.8692,  0.0211],\n",
            "          ...,\n",
            "          [-0.0750, -0.1040,  0.0872,  ...,  2.5230,  0.3111,  0.0757],\n",
            "          [-0.0750, -0.1040,  0.0872,  ...,  2.5230,  0.3111,  0.0757],\n",
            "          [-0.0749, -0.1040,  0.0875,  ...,  2.5230,  0.3110,  0.0761]],\n",
            "\n",
            "         [[-0.3047, -0.2015, -0.2331,  ..., -0.2103,  0.1178, -0.1469],\n",
            "          [-0.3213, -0.4961,  0.4353,  ..., -0.1216,  0.0205, -0.2111],\n",
            "          [-0.3585,  0.1606,  0.1271,  ..., -0.3376,  0.1531, -0.5901],\n",
            "          ...,\n",
            "          [-0.3100, -0.2138, -0.2368,  ..., -0.2004,  0.1233, -0.1497],\n",
            "          [-0.3100, -0.2138, -0.2368,  ..., -0.2004,  0.1233, -0.1497],\n",
            "          [-0.3098, -0.2138, -0.2368,  ..., -0.2003,  0.1232, -0.1498]]],\n",
            "\n",
            "\n",
            "        [[[ 0.0979,  0.0206, -0.0347,  ...,  0.6319,  0.1342,  0.0722],\n",
            "          [-0.1116, -0.2639, -0.0222,  ...,  0.8028,  0.0978, -0.0318],\n",
            "          [-0.1615,  0.3206, -0.0146,  ...,  0.6416,  0.4462,  0.1216],\n",
            "          ...,\n",
            "          [ 0.1027,  0.0125, -0.0343,  ...,  0.6420,  0.1230,  0.0706],\n",
            "          [ 0.1027,  0.0125, -0.0343,  ...,  0.6420,  0.1230,  0.0706],\n",
            "          [ 0.1025,  0.0126, -0.0344,  ...,  0.6419,  0.1230,  0.0707]],\n",
            "\n",
            "         [[ 1.0230, -0.5068, -0.0706,  ...,  0.4633, -1.2522,  0.4290],\n",
            "          [ 0.3673, -0.2400, -0.1513,  ...,  0.0585, -0.8596, -0.2455],\n",
            "          [ 0.3307, -0.4558,  0.1739,  ...,  0.5635, -0.4590,  0.4652],\n",
            "          ...,\n",
            "          [ 1.0206, -0.5060, -0.0767,  ...,  0.4605, -1.2534,  0.4228],\n",
            "          [ 1.0206, -0.5060, -0.0767,  ...,  0.4605, -1.2534,  0.4228],\n",
            "          [ 1.0205, -0.5060, -0.0765,  ...,  0.4605, -1.2534,  0.4228]],\n",
            "\n",
            "         [[-0.0770, -0.1141,  0.0881,  ...,  2.5316,  0.3113,  0.0734],\n",
            "          [-0.4054, -0.3062,  0.4082,  ...,  1.5039,  0.2782,  0.1156],\n",
            "          [ 0.0478, -0.4179,  0.5837,  ...,  1.9551,  0.8551,  0.0427],\n",
            "          ...,\n",
            "          [-0.0756, -0.1060,  0.0857,  ...,  2.5236,  0.3136,  0.0768],\n",
            "          [-0.0756, -0.1060,  0.0857,  ...,  2.5236,  0.3136,  0.0768],\n",
            "          [-0.0756, -0.1061,  0.0861,  ...,  2.5234,  0.3136,  0.0770]],\n",
            "\n",
            "         [[-0.3040, -0.1993, -0.2335,  ..., -0.2093,  0.1187, -0.1443],\n",
            "          [-0.3185, -0.4534,  0.3858,  ..., -0.1399,  0.0549, -0.1898],\n",
            "          [-0.3457,  0.2067,  0.1092,  ..., -0.3537,  0.1889, -0.5874],\n",
            "          ...,\n",
            "          [-0.3092, -0.2115, -0.2366,  ..., -0.1995,  0.1238, -0.1462],\n",
            "          [-0.3092, -0.2115, -0.2366,  ..., -0.1995,  0.1238, -0.1462],\n",
            "          [-0.3089, -0.2115, -0.2365,  ..., -0.1993,  0.1237, -0.1462]]]],\n",
            "       grad_fn=<TransposeBackward0>), tensor([[[[ 9.7110e-01,  6.7853e-01, -2.2493e+00,  ...,  5.0109e-01,\n",
            "           -1.0604e+00, -1.2138e+00],\n",
            "          [ 1.0450e+00,  8.8626e-01, -1.5958e+00,  ..., -4.0362e-02,\n",
            "           -9.2874e-01, -9.1641e-01],\n",
            "          [ 2.2281e-01,  6.8913e-01, -1.0564e+00,  ...,  1.3365e-01,\n",
            "            2.6695e-02, -4.5311e-01],\n",
            "          ...,\n",
            "          [ 9.7154e-01,  6.5464e-01, -2.2206e+00,  ...,  4.8520e-01,\n",
            "           -1.0577e+00, -1.2088e+00],\n",
            "          [ 9.7154e-01,  6.5464e-01, -2.2206e+00,  ...,  4.8520e-01,\n",
            "           -1.0577e+00, -1.2088e+00],\n",
            "          [ 9.7285e-01,  6.5501e-01, -2.2203e+00,  ...,  4.8504e-01,\n",
            "           -1.0579e+00, -1.2087e+00]],\n",
            "\n",
            "         [[ 5.9497e-02, -3.5691e-01, -9.4735e-01,  ...,  6.9631e-01,\n",
            "           -4.3777e-01, -7.5996e-01],\n",
            "          [ 4.6704e-01,  6.3553e-02, -1.1804e+00,  ...,  1.1480e+00,\n",
            "           -6.7441e-02, -1.2660e+00],\n",
            "          [ 6.6129e-01, -1.7766e+00, -3.3299e-03,  ..., -6.3211e-02,\n",
            "           -9.4574e-01,  3.9522e-01],\n",
            "          ...,\n",
            "          [ 6.5286e-02, -3.4214e-01, -9.6177e-01,  ...,  6.7245e-01,\n",
            "           -4.3867e-01, -7.4451e-01],\n",
            "          [ 6.5286e-02, -3.4214e-01, -9.6177e-01,  ...,  6.7245e-01,\n",
            "           -4.3867e-01, -7.4451e-01],\n",
            "          [ 6.4814e-02, -3.4200e-01, -9.6262e-01,  ...,  6.7247e-01,\n",
            "           -4.3830e-01, -7.4423e-01]],\n",
            "\n",
            "         [[-1.6312e-01, -1.1805e-01, -4.5951e-01,  ...,  9.3996e-01,\n",
            "           -1.3285e-01,  1.2506e-01],\n",
            "          [-1.1582e+00,  5.8915e-01, -5.0313e-01,  ..., -3.7262e+00,\n",
            "           -6.9210e-01,  2.3112e+00],\n",
            "          [-7.9903e-01,  3.2227e-01,  3.4350e-01,  ...,  1.6731e-02,\n",
            "           -3.1691e+00,  1.1768e-01],\n",
            "          ...,\n",
            "          [-1.4722e-01, -1.2309e-01, -4.6630e-01,  ...,  9.3588e-01,\n",
            "           -1.2507e-01,  1.1404e-01],\n",
            "          [-1.4722e-01, -1.2309e-01, -4.6630e-01,  ...,  9.3588e-01,\n",
            "           -1.2507e-01,  1.1404e-01],\n",
            "          [-1.4617e-01, -1.2380e-01, -4.6690e-01,  ...,  9.3577e-01,\n",
            "           -1.2565e-01,  1.1396e-01]],\n",
            "\n",
            "         [[-4.5165e-01,  1.6314e-01,  2.3370e-01,  ..., -2.9042e-02,\n",
            "           -4.5808e-02, -2.3169e-01],\n",
            "          [ 1.7201e+00, -1.3782e+00,  9.1290e-01,  ..., -3.7618e-03,\n",
            "            1.4080e-01,  4.8170e-01],\n",
            "          [ 1.3470e+00, -9.7395e-01,  1.4821e+00,  ..., -3.3398e-02,\n",
            "            1.8256e-02,  3.2220e+00],\n",
            "          ...,\n",
            "          [-4.6904e-01,  1.4536e-01,  2.3641e-01,  ..., -5.3293e-02,\n",
            "           -1.9307e-02, -2.4892e-01],\n",
            "          [-4.6904e-01,  1.4536e-01,  2.3641e-01,  ..., -5.3294e-02,\n",
            "           -1.9308e-02, -2.4891e-01],\n",
            "          [-4.6729e-01,  1.4535e-01,  2.3806e-01,  ..., -5.4326e-02,\n",
            "           -1.8999e-02, -2.4892e-01]]],\n",
            "\n",
            "\n",
            "        [[[ 9.3980e-01,  6.6437e-01, -2.2410e+00,  ...,  4.9219e-01,\n",
            "           -1.0670e+00, -1.1974e+00],\n",
            "          [ 9.6380e-01,  8.4891e-01, -1.5869e+00,  ..., -1.2506e-01,\n",
            "           -1.0600e+00, -9.8284e-01],\n",
            "          [ 2.2849e-01,  6.7448e-01, -1.0394e+00,  ...,  1.4281e-02,\n",
            "            2.4343e-02, -5.0810e-01],\n",
            "          ...,\n",
            "          [ 9.3739e-01,  6.3935e-01, -2.2139e+00,  ...,  4.7762e-01,\n",
            "           -1.0636e+00, -1.1933e+00],\n",
            "          [ 9.3739e-01,  6.3935e-01, -2.2139e+00,  ...,  4.7762e-01,\n",
            "           -1.0636e+00, -1.1933e+00],\n",
            "          [ 9.3788e-01,  6.3953e-01, -2.2137e+00,  ...,  4.7733e-01,\n",
            "           -1.0637e+00, -1.1933e+00]],\n",
            "\n",
            "         [[ 5.8905e-02, -3.4979e-01, -9.4958e-01,  ...,  6.9791e-01,\n",
            "           -4.5155e-01, -7.7278e-01],\n",
            "          [ 4.9785e-01,  3.0230e-02, -1.0878e+00,  ...,  1.0948e+00,\n",
            "           -7.6774e-02, -1.2556e+00],\n",
            "          [ 5.5877e-01, -1.5695e+00,  1.5971e-02,  ..., -4.4755e-02,\n",
            "           -9.3958e-01,  3.5923e-01],\n",
            "          ...,\n",
            "          [ 6.6669e-02, -3.3889e-01, -9.5948e-01,  ...,  6.7684e-01,\n",
            "           -4.5686e-01, -7.6061e-01],\n",
            "          [ 6.6670e-02, -3.3889e-01, -9.5948e-01,  ...,  6.7684e-01,\n",
            "           -4.5686e-01, -7.6061e-01],\n",
            "          [ 6.6511e-02, -3.3874e-01, -9.5982e-01,  ...,  6.7709e-01,\n",
            "           -4.5651e-01, -7.6074e-01]],\n",
            "\n",
            "         [[-1.8080e-01, -1.2148e-01, -4.5334e-01,  ...,  9.1855e-01,\n",
            "           -1.4802e-01,  1.3846e-01],\n",
            "          [-1.4898e+00,  4.4820e-01, -4.4763e-01,  ..., -4.1006e+00,\n",
            "           -4.6737e-01,  2.5257e+00],\n",
            "          [-6.2266e-01,  3.6630e-01,  7.0268e-01,  ..., -1.4643e-01,\n",
            "           -3.1426e+00,  2.2346e-01],\n",
            "          ...,\n",
            "          [-1.7028e-01, -1.2613e-01, -4.6209e-01,  ...,  9.1298e-01,\n",
            "           -1.4321e-01,  1.2954e-01],\n",
            "          [-1.7028e-01, -1.2613e-01, -4.6209e-01,  ...,  9.1298e-01,\n",
            "           -1.4321e-01,  1.2954e-01],\n",
            "          [-1.7005e-01, -1.2676e-01, -4.6301e-01,  ...,  9.1281e-01,\n",
            "           -1.4391e-01,  1.3007e-01]],\n",
            "\n",
            "         [[-4.7516e-01,  1.7909e-01,  2.3165e-01,  ..., -3.1927e-03,\n",
            "           -5.4093e-02, -2.3015e-01],\n",
            "          [ 1.5522e+00, -1.3709e+00,  1.0233e+00,  ..., -9.9356e-02,\n",
            "            1.5272e-01,  3.8105e-01],\n",
            "          [ 1.3786e+00, -1.0483e+00,  1.6172e+00,  ..., -3.0122e-01,\n",
            "           -2.6142e-02,  3.0836e+00],\n",
            "          ...,\n",
            "          [-4.9166e-01,  1.6504e-01,  2.2741e-01,  ..., -2.5723e-02,\n",
            "           -3.0068e-02, -2.4549e-01],\n",
            "          [-4.9166e-01,  1.6504e-01,  2.2741e-01,  ..., -2.5723e-02,\n",
            "           -3.0068e-02, -2.4549e-01],\n",
            "          [-4.9053e-01,  1.6535e-01,  2.2832e-01,  ..., -2.6336e-02,\n",
            "           -3.0171e-02, -2.4511e-01]]],\n",
            "\n",
            "\n",
            "        [[[ 9.5073e-01,  6.4621e-01, -2.2202e+00,  ...,  4.8858e-01,\n",
            "           -1.0726e+00, -1.2119e+00],\n",
            "          [ 9.6497e-01,  7.3273e-01, -1.4593e+00,  ...,  4.3391e-04,\n",
            "           -9.6135e-01, -9.9749e-01],\n",
            "          [ 1.2268e-01,  4.7454e-01, -9.1924e-01,  ...,  1.9958e-01,\n",
            "           -3.0378e-02, -3.7810e-01],\n",
            "          ...,\n",
            "          [ 9.4678e-01,  6.2160e-01, -2.1911e+00,  ...,  4.7229e-01,\n",
            "           -1.0684e+00, -1.2069e+00],\n",
            "          [ 9.4678e-01,  6.2160e-01, -2.1911e+00,  ...,  4.7229e-01,\n",
            "           -1.0684e+00, -1.2069e+00],\n",
            "          [ 9.4708e-01,  6.2115e-01, -2.1902e+00,  ...,  4.7184e-01,\n",
            "           -1.0686e+00, -1.2071e+00]],\n",
            "\n",
            "         [[ 6.5331e-02, -3.7449e-01, -9.4595e-01,  ...,  7.0208e-01,\n",
            "           -4.4275e-01, -7.5977e-01],\n",
            "          [ 5.3433e-01,  6.8340e-02, -1.1669e+00,  ...,  1.1523e+00,\n",
            "            1.2768e-01, -1.3179e+00],\n",
            "          [ 6.2015e-01, -1.7827e+00, -7.3938e-02,  ..., -5.2463e-02,\n",
            "           -7.6904e-01,  4.1585e-01],\n",
            "          ...,\n",
            "          [ 7.0998e-02, -3.5937e-01, -9.5630e-01,  ...,  6.7853e-01,\n",
            "           -4.4334e-01, -7.5080e-01],\n",
            "          [ 7.0998e-02, -3.5937e-01, -9.5630e-01,  ...,  6.7853e-01,\n",
            "           -4.4334e-01, -7.5080e-01],\n",
            "          [ 7.0851e-02, -3.5915e-01, -9.5652e-01,  ...,  6.7909e-01,\n",
            "           -4.4324e-01, -7.5128e-01]],\n",
            "\n",
            "         [[-1.6563e-01, -1.3312e-01, -4.5739e-01,  ...,  9.5232e-01,\n",
            "           -1.4000e-01,  1.4014e-01],\n",
            "          [-1.1295e+00,  6.0170e-01, -5.2753e-01,  ..., -3.6169e+00,\n",
            "           -4.5859e-01,  2.4918e+00],\n",
            "          [-3.5972e-01,  3.7723e-01,  3.2089e-01,  ...,  4.4402e-01,\n",
            "           -3.1789e+00,  2.7630e-01],\n",
            "          ...,\n",
            "          [-1.5302e-01, -1.3579e-01, -4.6316e-01,  ...,  9.4606e-01,\n",
            "           -1.3414e-01,  1.3414e-01],\n",
            "          [-1.5302e-01, -1.3579e-01, -4.6316e-01,  ...,  9.4606e-01,\n",
            "           -1.3414e-01,  1.3414e-01],\n",
            "          [-1.5226e-01, -1.3627e-01, -4.6397e-01,  ...,  9.4586e-01,\n",
            "           -1.3450e-01,  1.3474e-01]],\n",
            "\n",
            "         [[-4.6107e-01,  1.6699e-01,  2.1471e-01,  ..., -4.1695e-02,\n",
            "           -4.0148e-02, -2.4287e-01],\n",
            "          [ 1.6872e+00, -1.3172e+00,  7.5752e-01,  ..., -3.4956e-02,\n",
            "            2.6878e-01,  6.7547e-01],\n",
            "          [ 1.2728e+00, -1.1802e+00,  1.3237e+00,  ..., -2.3446e-01,\n",
            "            7.1745e-02,  3.1552e+00],\n",
            "          ...,\n",
            "          [-4.7850e-01,  1.5179e-01,  2.1808e-01,  ..., -6.5264e-02,\n",
            "           -1.4875e-02, -2.5841e-01],\n",
            "          [-4.7850e-01,  1.5179e-01,  2.1808e-01,  ..., -6.5264e-02,\n",
            "           -1.4875e-02, -2.5841e-01],\n",
            "          [-4.7768e-01,  1.5220e-01,  2.1896e-01,  ..., -6.5916e-02,\n",
            "           -1.4769e-02, -2.5787e-01]]],\n",
            "\n",
            "\n",
            "        ...,\n",
            "\n",
            "\n",
            "        [[[ 1.1687e+00,  7.0848e-01, -2.2521e+00,  ...,  6.4951e-01,\n",
            "           -1.0599e+00, -1.2637e+00],\n",
            "          [ 1.1364e+00,  6.8436e-01, -1.3269e+00,  ...,  1.8876e-01,\n",
            "           -9.2550e-01, -9.4902e-01],\n",
            "          [ 1.3884e-01,  6.3259e-01, -8.2306e-01,  ...,  5.7289e-01,\n",
            "           -5.4686e-02, -4.5949e-01],\n",
            "          ...,\n",
            "          [ 1.1536e+00,  6.2774e-01, -2.1616e+00,  ...,  6.2116e-01,\n",
            "           -1.0831e+00, -1.2404e+00],\n",
            "          [ 1.1536e+00,  6.2774e-01, -2.1616e+00,  ...,  6.2116e-01,\n",
            "           -1.0831e+00, -1.2404e+00],\n",
            "          [ 1.1545e+00,  6.2759e-01, -2.1610e+00,  ...,  6.2111e-01,\n",
            "           -1.0830e+00, -1.2402e+00]],\n",
            "\n",
            "         [[ 5.1458e-02, -3.5724e-01, -9.8894e-01,  ...,  6.9181e-01,\n",
            "           -4.2622e-01, -6.2930e-01],\n",
            "          [ 6.1037e-01, -1.3705e-02, -1.3442e+00,  ...,  1.2000e+00,\n",
            "            4.8491e-01, -1.2041e+00],\n",
            "          [ 7.7633e-01, -1.9153e+00,  9.6591e-02,  ..., -1.2626e-01,\n",
            "           -3.7393e-01,  5.9666e-01],\n",
            "          ...,\n",
            "          [ 5.2728e-02, -3.3715e-01, -1.0002e+00,  ...,  6.7658e-01,\n",
            "           -4.3490e-01, -6.2275e-01],\n",
            "          [ 5.2728e-02, -3.3715e-01, -1.0002e+00,  ...,  6.7658e-01,\n",
            "           -4.3490e-01, -6.2275e-01],\n",
            "          [ 5.2411e-02, -3.3682e-01, -1.0008e+00,  ...,  6.7683e-01,\n",
            "           -4.3483e-01, -6.2270e-01]],\n",
            "\n",
            "         [[-2.6347e-02, -1.4661e-01, -4.1608e-01,  ...,  9.7873e-01,\n",
            "           -1.5356e-01,  5.1446e-02],\n",
            "          [-6.2968e-01,  8.1104e-01, -6.3275e-01,  ..., -3.1809e+00,\n",
            "           -7.6179e-01,  2.4672e+00],\n",
            "          [-9.1410e-02,  3.7418e-01,  8.4766e-02,  ...,  8.4619e-01,\n",
            "           -3.4914e+00,  3.9849e-01],\n",
            "          ...,\n",
            "          [-4.6918e-03, -1.5467e-01, -4.3140e-01,  ...,  9.7708e-01,\n",
            "           -1.3549e-01,  5.0338e-02],\n",
            "          [-4.6920e-03, -1.5467e-01, -4.3140e-01,  ...,  9.7708e-01,\n",
            "           -1.3549e-01,  5.0338e-02],\n",
            "          [-3.2335e-03, -1.5484e-01, -4.3250e-01,  ...,  9.7718e-01,\n",
            "           -1.3491e-01,  5.0739e-02]],\n",
            "\n",
            "         [[-3.4047e-01,  1.1072e-01,  2.7781e-01,  ..., -1.2647e-01,\n",
            "           -5.7254e-02, -3.1381e-01],\n",
            "          [ 1.7930e+00, -1.5510e+00,  6.4405e-01,  ..., -2.3305e-01,\n",
            "            4.2205e-01,  7.5681e-01],\n",
            "          [ 1.8007e+00, -1.2482e+00,  1.0284e+00,  ..., -4.1218e-01,\n",
            "           -2.3250e-01,  3.3870e+00],\n",
            "          ...,\n",
            "          [-3.5396e-01,  1.1109e-01,  2.7751e-01,  ..., -1.3805e-01,\n",
            "           -9.8248e-03, -3.1594e-01],\n",
            "          [-3.5396e-01,  1.1109e-01,  2.7751e-01,  ..., -1.3805e-01,\n",
            "           -9.8251e-03, -3.1594e-01],\n",
            "          [-3.5196e-01,  1.1153e-01,  2.7956e-01,  ..., -1.3896e-01,\n",
            "           -8.9577e-03, -3.1561e-01]]],\n",
            "\n",
            "\n",
            "        [[[ 9.4581e-01,  6.6351e-01, -2.2525e+00,  ...,  4.8425e-01,\n",
            "           -1.0534e+00, -1.1943e+00],\n",
            "          [ 1.0481e+00,  8.8836e-01, -1.6086e+00,  ..., -1.8377e-01,\n",
            "           -8.1788e-01, -8.0018e-01],\n",
            "          [ 2.8817e-01,  6.2393e-01, -1.0670e+00,  ...,  1.8217e-01,\n",
            "            1.3683e-01, -4.2571e-01],\n",
            "          ...,\n",
            "          [ 9.4428e-01,  6.4088e-01, -2.2262e+00,  ...,  4.6872e-01,\n",
            "           -1.0499e+00, -1.1877e+00],\n",
            "          [ 9.4428e-01,  6.4088e-01, -2.2262e+00,  ...,  4.6872e-01,\n",
            "           -1.0499e+00, -1.1877e+00],\n",
            "          [ 9.4555e-01,  6.4123e-01, -2.2259e+00,  ...,  4.6856e-01,\n",
            "           -1.0500e+00, -1.1876e+00]],\n",
            "\n",
            "         [[ 5.6792e-02, -3.6803e-01, -9.4776e-01,  ...,  6.9259e-01,\n",
            "           -4.5114e-01, -7.7339e-01],\n",
            "          [ 6.0754e-01,  2.9294e-01, -1.0056e+00,  ...,  1.1702e+00,\n",
            "           -3.6949e-01, -1.3780e+00],\n",
            "          [ 6.0050e-01, -1.7010e+00,  4.2144e-02,  ..., -1.7356e-02,\n",
            "           -9.6316e-01,  3.4416e-01],\n",
            "          ...,\n",
            "          [ 6.2782e-02, -3.5345e-01, -9.6178e-01,  ...,  6.6828e-01,\n",
            "           -4.5206e-01, -7.5943e-01],\n",
            "          [ 6.2782e-02, -3.5345e-01, -9.6178e-01,  ...,  6.6828e-01,\n",
            "           -4.5206e-01, -7.5943e-01],\n",
            "          [ 6.2469e-02, -3.5328e-01, -9.6259e-01,  ...,  6.6838e-01,\n",
            "           -4.5186e-01, -7.5910e-01]],\n",
            "\n",
            "         [[-1.4410e-01, -1.4189e-01, -4.5534e-01,  ...,  9.0017e-01,\n",
            "           -1.1882e-01,  1.2674e-01],\n",
            "          [-1.6284e+00,  5.3968e-01, -4.2529e-01,  ..., -4.3676e+00,\n",
            "           -4.5817e-01,  2.2640e+00],\n",
            "          [-6.2515e-01,  2.5107e-01,  5.1628e-01,  ..., -1.2942e-01,\n",
            "           -3.0009e+00, -1.6942e-01],\n",
            "          ...,\n",
            "          [-1.2830e-01, -1.4798e-01, -4.6408e-01,  ...,  8.9457e-01,\n",
            "           -1.1191e-01,  1.2227e-01],\n",
            "          [-1.2830e-01, -1.4798e-01, -4.6408e-01,  ...,  8.9457e-01,\n",
            "           -1.1191e-01,  1.2227e-01],\n",
            "          [-1.2770e-01, -1.4843e-01, -4.6504e-01,  ...,  8.9448e-01,\n",
            "           -1.1210e-01,  1.2207e-01]],\n",
            "\n",
            "         [[-4.5365e-01,  1.8094e-01,  2.4305e-01,  ..., -2.5284e-02,\n",
            "           -2.7491e-02, -2.2579e-01],\n",
            "          [ 1.8526e+00, -1.1376e+00,  1.2365e+00,  ...,  1.1162e-01,\n",
            "            3.5002e-02,  3.9688e-01],\n",
            "          [ 1.4737e+00, -8.8614e-01,  1.5182e+00,  ..., -8.5386e-02,\n",
            "            1.1767e-01,  3.2740e+00],\n",
            "          ...,\n",
            "          [-4.7153e-01,  1.6523e-01,  2.4734e-01,  ..., -4.7956e-02,\n",
            "           -1.6544e-03, -2.4358e-01],\n",
            "          [-4.7153e-01,  1.6523e-01,  2.4734e-01,  ..., -4.7956e-02,\n",
            "           -1.6542e-03, -2.4358e-01],\n",
            "          [-4.7007e-01,  1.6538e-01,  2.4890e-01,  ..., -4.8614e-02,\n",
            "           -1.3430e-03, -2.4320e-01]]],\n",
            "\n",
            "\n",
            "        [[[ 9.3684e-01,  6.5057e-01, -2.2307e+00,  ...,  4.7102e-01,\n",
            "           -1.0726e+00, -1.1991e+00],\n",
            "          [ 8.9272e-01,  8.2784e-01, -1.5782e+00,  ..., -1.4623e-01,\n",
            "           -9.3172e-01, -8.0629e-01],\n",
            "          [ 1.8962e-01,  5.2407e-01, -9.6174e-01,  ...,  1.2401e-01,\n",
            "            7.6945e-03, -4.8325e-01],\n",
            "          ...,\n",
            "          [ 9.3197e-01,  6.2668e-01, -2.2044e+00,  ...,  4.5645e-01,\n",
            "           -1.0710e+00, -1.1951e+00],\n",
            "          [ 9.3197e-01,  6.2668e-01, -2.2044e+00,  ...,  4.5645e-01,\n",
            "           -1.0710e+00, -1.1951e+00],\n",
            "          [ 9.3239e-01,  6.2688e-01, -2.2041e+00,  ...,  4.5607e-01,\n",
            "           -1.0710e+00, -1.1951e+00]],\n",
            "\n",
            "         [[ 6.2324e-02, -3.7196e-01, -9.5283e-01,  ...,  6.9212e-01,\n",
            "           -4.4589e-01, -7.7251e-01],\n",
            "          [ 5.5524e-01,  1.5931e-01, -1.0955e+00,  ...,  1.1510e+00,\n",
            "           -3.0085e-01, -1.2345e+00],\n",
            "          [ 6.5651e-01, -1.6732e+00, -4.4912e-02,  ..., -7.5611e-02,\n",
            "           -9.5491e-01,  2.7311e-01],\n",
            "          ...,\n",
            "          [ 6.6690e-02, -3.5659e-01, -9.6533e-01,  ...,  6.6789e-01,\n",
            "           -4.4756e-01, -7.6341e-01],\n",
            "          [ 6.6690e-02, -3.5659e-01, -9.6533e-01,  ...,  6.6789e-01,\n",
            "           -4.4756e-01, -7.6341e-01],\n",
            "          [ 6.6357e-02, -3.5631e-01, -9.6565e-01,  ...,  6.6805e-01,\n",
            "           -4.4744e-01, -7.6368e-01]],\n",
            "\n",
            "         [[-1.6436e-01, -1.1948e-01, -4.7983e-01,  ...,  9.2635e-01,\n",
            "           -1.3019e-01,  1.5742e-01],\n",
            "          [-1.4907e+00,  5.9560e-01, -3.9837e-01,  ..., -3.9267e+00,\n",
            "           -5.1701e-01,  2.2987e+00],\n",
            "          [-7.1164e-01,  3.2620e-01,  2.9602e-01,  ..., -6.0533e-02,\n",
            "           -3.0633e+00,  2.5890e-01],\n",
            "          ...,\n",
            "          [-1.5441e-01, -1.2055e-01, -4.8848e-01,  ...,  9.2009e-01,\n",
            "           -1.2614e-01,  1.5600e-01],\n",
            "          [-1.5441e-01, -1.2055e-01, -4.8848e-01,  ...,  9.2009e-01,\n",
            "           -1.2614e-01,  1.5600e-01],\n",
            "          [-1.5452e-01, -1.2098e-01, -4.8942e-01,  ...,  9.1971e-01,\n",
            "           -1.2685e-01,  1.5664e-01]],\n",
            "\n",
            "         [[-4.6878e-01,  1.7708e-01,  2.3772e-01,  ..., -2.4052e-02,\n",
            "           -3.3220e-02, -2.2713e-01],\n",
            "          [ 1.8152e+00, -1.2869e+00,  1.0198e+00,  ...,  1.2613e-02,\n",
            "            1.2007e-02,  4.4983e-01],\n",
            "          [ 1.2658e+00, -1.0082e+00,  1.4311e+00,  ..., -2.2643e-01,\n",
            "            7.5240e-02,  3.2273e+00],\n",
            "          ...,\n",
            "          [-4.8791e-01,  1.5985e-01,  2.3993e-01,  ..., -4.7220e-02,\n",
            "           -1.0585e-02, -2.4461e-01],\n",
            "          [-4.8791e-01,  1.5985e-01,  2.3993e-01,  ..., -4.7220e-02,\n",
            "           -1.0586e-02, -2.4461e-01],\n",
            "          [-4.8693e-01,  1.6004e-01,  2.4107e-01,  ..., -4.7712e-02,\n",
            "           -1.0610e-02, -2.4414e-01]]]], grad_fn=<TransposeBackward0>), tensor([[[[-1.6782e+01, -6.2213e-01, -1.5400e+01,  ...,  2.2924e+01,\n",
            "           -7.9479e+00, -6.7922e-01],\n",
            "          [-1.0850e+01,  9.6657e+00, -1.4661e+01,  ...,  1.1087e+01,\n",
            "           -5.2120e+00, -1.1893e+01],\n",
            "          [-1.9502e+01,  6.9180e+00, -1.5127e+01,  ...,  1.2046e+01,\n",
            "           -5.3882e+00, -5.5487e+00],\n",
            "          ...,\n",
            "          [-9.9389e+00, -1.0373e+00, -2.2362e+00,  ...,  2.2607e+01,\n",
            "            4.5418e+00,  2.3242e-01],\n",
            "          [-1.7284e+01,  5.7961e+00,  8.5500e+00,  ...,  1.3869e+01,\n",
            "            3.1584e+00, -1.3252e-01],\n",
            "          [ 4.0033e+00, -3.7593e+00,  2.4434e+00,  ..., -2.8510e+00,\n",
            "            7.0706e+00,  2.2762e+00]],\n",
            "\n",
            "         [[ 2.5251e+01,  3.8361e+00,  1.1236e+01,  ...,  9.4050e+00,\n",
            "            1.7500e+01,  2.0629e+01],\n",
            "          [ 1.3538e+01,  1.0370e+00,  1.0169e+01,  ...,  8.5490e+00,\n",
            "            1.6560e+01,  5.7657e+00],\n",
            "          [ 1.0731e+01,  4.6628e+00,  7.6070e+00,  ...,  1.3620e+01,\n",
            "            1.7958e+01,  1.2691e+01],\n",
            "          ...,\n",
            "          [ 1.4413e+01,  7.4630e+00, -6.6744e-01,  ...,  8.7002e+00,\n",
            "            1.1203e+01,  1.2203e+01],\n",
            "          [ 1.5404e+01,  2.4856e+00, -1.6632e+00,  ...,  5.4514e+00,\n",
            "            1.4639e+01,  2.0654e+01],\n",
            "          [-3.9217e+00, -9.5042e+00, -1.9027e+01,  ..., -5.9542e+00,\n",
            "           -2.2935e+00, -4.5651e-01]],\n",
            "\n",
            "         [[-7.9057e+00, -1.4564e+01, -5.7157e+00,  ...,  8.9669e+00,\n",
            "           -3.1791e+00,  2.2312e+00],\n",
            "          [ 8.5746e+00, -1.4189e+01, -9.6805e+00,  ...,  5.1224e+00,\n",
            "           -3.0592e+00, -1.8987e+00],\n",
            "          [ 6.8299e+00, -2.9205e+01, -1.5101e+01,  ...,  7.1969e+00,\n",
            "           -8.4721e+00,  2.1875e+00],\n",
            "          ...,\n",
            "          [ 8.7876e+00, -1.8831e+01, -7.3265e+00,  ..., -1.8153e+00,\n",
            "            1.2677e+01,  3.2486e-01],\n",
            "          [-6.7973e+00, -1.8323e+01, -1.4149e+00,  ..., -9.5981e+00,\n",
            "            3.2305e+00,  6.7930e-01],\n",
            "          [ 1.4700e+00,  3.8055e+00,  4.8553e+00,  ..., -2.2548e+00,\n",
            "            9.0921e-01, -9.0107e-01]],\n",
            "\n",
            "         [[ 2.3785e+01,  1.0244e+01, -1.0572e+01,  ..., -1.3329e+01,\n",
            "            1.9599e+01, -3.0208e-01],\n",
            "          [ 1.0787e+01,  1.3628e+01, -1.0452e+01,  ..., -8.5645e+00,\n",
            "            7.8096e+00, -4.4882e+00],\n",
            "          [ 5.8141e+00,  1.7031e+01, -2.6041e+00,  ..., -7.0934e+00,\n",
            "            1.0106e+01,  5.7603e+00],\n",
            "          ...,\n",
            "          [ 3.1589e+00, -5.6121e-01, -1.7087e+00,  ...,  6.0478e-01,\n",
            "            2.7964e+00,  1.2037e+01],\n",
            "          [ 3.9463e+00,  4.6959e+00, -1.5091e+00,  ...,  1.7027e+00,\n",
            "            1.6800e+01,  8.1026e+00],\n",
            "          [-4.0073e+00,  3.3813e+00, -2.2353e+00,  ...,  1.9043e+00,\n",
            "            4.7660e+00,  4.2244e+00]]],\n",
            "\n",
            "\n",
            "        [[[-1.6189e+01,  1.2929e+01, -1.4461e+01,  ...,  1.5004e+01,\n",
            "           -1.0291e+01,  1.4182e+00],\n",
            "          [-1.2418e+01,  5.0841e+00, -9.7018e+00,  ...,  1.7430e+01,\n",
            "           -2.3001e+00,  2.5146e+00],\n",
            "          [-1.1610e+01,  7.1996e+00, -4.5127e+00,  ...,  1.3492e+01,\n",
            "           -2.0775e+00, -1.0978e+00],\n",
            "          ...,\n",
            "          [-9.3335e+00,  1.1601e+01, -1.4831e-01,  ...,  1.2732e+01,\n",
            "            7.4517e+00, -3.9785e+00],\n",
            "          [-8.4105e+00,  5.4875e+00,  3.3848e+00,  ...,  1.3245e+01,\n",
            "            5.9149e+00, -3.4992e+00],\n",
            "          [ 4.3087e+00, -3.9333e+00,  2.2602e+00,  ..., -3.1129e+00,\n",
            "            6.9512e+00,  2.1713e+00]],\n",
            "\n",
            "         [[ 1.6136e+01,  3.8385e+00,  4.4363e+00,  ...,  1.1012e+01,\n",
            "            8.3726e+00,  1.2484e+01],\n",
            "          [ 2.3710e+01, -4.0036e+00,  5.6091e+00,  ...,  6.1676e+00,\n",
            "            1.1137e+01,  8.4140e+00],\n",
            "          [ 1.8839e+01, -3.7768e+00,  3.2466e+00,  ...,  1.0118e+01,\n",
            "            8.1147e+00,  4.6280e+00],\n",
            "          ...,\n",
            "          [ 7.1353e+00,  5.2078e-01,  7.1393e-01,  ...,  1.0346e+01,\n",
            "            1.8474e+01,  1.0273e+01],\n",
            "          [ 1.4172e+01,  9.2355e+00, -8.8985e-01,  ...,  1.6404e+01,\n",
            "            1.4927e+01,  1.6648e+01],\n",
            "          [-3.9926e+00, -9.4922e+00, -1.9308e+01,  ..., -6.2059e+00,\n",
            "           -2.3276e+00, -6.3702e-01]],\n",
            "\n",
            "         [[-9.4758e-01, -1.4367e+01, -1.2979e+01,  ...,  1.5783e+01,\n",
            "            8.0864e+00, -4.1752e+00],\n",
            "          [-6.0910e+00, -6.3146e+00, -9.3066e+00,  ...,  1.0130e+01,\n",
            "            2.2923e+00, -2.7451e+00],\n",
            "          [ 2.1611e-01, -1.2301e+01, -7.9440e+00,  ...,  1.4367e+01,\n",
            "            5.3927e+00, -2.2772e+00],\n",
            "          ...,\n",
            "          [-9.7919e+00, -1.5739e+01, -7.2728e+00,  ...,  7.5226e-01,\n",
            "            8.6102e+00, -1.7471e+00],\n",
            "          [-2.1600e+01, -1.7411e+01,  4.5593e-01,  ...,  1.2119e+01,\n",
            "            3.2835e+00, -9.0435e+00],\n",
            "          [ 1.4463e+00,  3.9651e+00,  4.8004e+00,  ..., -1.8318e+00,\n",
            "            8.8491e-01, -8.5380e-01]],\n",
            "\n",
            "         [[ 2.9798e+00,  5.3949e+00,  2.0303e-02,  ..., -1.4702e+01,\n",
            "            5.8170e+00,  1.1476e+01],\n",
            "          [ 1.6766e+01,  2.1301e+00, -1.0851e+01,  ..., -3.9203e+00,\n",
            "            4.0323e+00,  8.1095e+00],\n",
            "          [ 1.6454e+01,  2.6410e+00, -1.9386e+01,  ..., -2.2069e+00,\n",
            "            6.3214e+00,  1.5529e+00],\n",
            "          ...,\n",
            "          [ 2.0922e+01,  1.1487e+01, -9.8209e+00,  ..., -7.3109e+00,\n",
            "            1.5245e+01,  1.4585e+01],\n",
            "          [ 1.3601e+01, -3.4745e+00, -5.1007e+00,  ..., -1.1169e+01,\n",
            "            1.6657e+01,  1.3415e+01],\n",
            "          [-3.9629e+00,  3.4698e+00, -2.4153e+00,  ...,  1.8175e+00,\n",
            "            4.5535e+00,  4.1338e+00]]],\n",
            "\n",
            "\n",
            "        [[[-2.0767e+01,  4.3154e+00, -2.3605e+00,  ...,  1.3273e+01,\n",
            "            6.8284e+00,  1.5616e+01],\n",
            "          [-1.3929e+01,  5.7470e+00, -1.8068e+00,  ...,  8.8981e+00,\n",
            "            6.8668e+00,  1.5727e+00],\n",
            "          [-1.6016e+01,  1.0333e+01, -4.5379e+00,  ...,  1.5676e+01,\n",
            "            2.6588e+00, -1.2712e+00],\n",
            "          ...,\n",
            "          [-2.4088e+00,  5.9811e-01, -1.6515e-01,  ...,  9.5717e-01,\n",
            "            1.9641e+00,  5.2813e+00],\n",
            "          [-2.4088e+00,  5.9811e-01, -1.6515e-01,  ...,  9.5717e-01,\n",
            "            1.9641e+00,  5.2813e+00],\n",
            "          [-2.4088e+00,  5.9811e-01, -1.6515e-01,  ...,  9.5717e-01,\n",
            "            1.9641e+00,  5.2813e+00]],\n",
            "\n",
            "         [[ 2.2578e+01,  1.0543e+01,  1.3773e+01,  ...,  2.0441e+01,\n",
            "            1.7004e+01,  1.1724e+01],\n",
            "          [ 2.0141e+01,  4.4041e+00,  9.8090e+00,  ...,  1.7676e+01,\n",
            "            1.0100e+01,  8.6678e+00],\n",
            "          [ 2.2609e+01,  1.3251e+01,  1.5881e+01,  ...,  1.3086e+01,\n",
            "            1.1841e+01,  9.8080e+00],\n",
            "          ...,\n",
            "          [ 7.6332e+00, -3.8793e+00,  1.5285e+00,  ...,  1.2818e-01,\n",
            "            6.6283e+00, -4.4199e+00],\n",
            "          [ 7.6332e+00, -3.8793e+00,  1.5285e+00,  ...,  1.2818e-01,\n",
            "            6.6283e+00, -4.4199e+00],\n",
            "          [ 7.6332e+00, -3.8793e+00,  1.5285e+00,  ...,  1.2818e-01,\n",
            "            6.6283e+00, -4.4199e+00]],\n",
            "\n",
            "         [[-6.0113e+00, -1.4146e+01, -2.5321e+00,  ...,  6.8476e+00,\n",
            "            2.8692e+00,  1.0822e+01],\n",
            "          [-3.1090e+00, -1.9595e+01,  3.9360e+00,  ...,  8.1269e+00,\n",
            "           -1.5033e+00,  1.2343e+01],\n",
            "          [-5.6940e+00, -1.9785e+01, -8.0646e+00,  ...,  1.5434e+01,\n",
            "           -2.6346e+00,  6.7721e+00],\n",
            "          ...,\n",
            "          [-2.2152e+00, -1.0513e+01, -6.5878e+00,  ...,  2.6297e+00,\n",
            "            5.4264e+00, -2.9451e+00],\n",
            "          [-2.2152e+00, -1.0513e+01, -6.5878e+00,  ...,  2.6297e+00,\n",
            "            5.4264e+00, -2.9451e+00],\n",
            "          [-2.2152e+00, -1.0513e+01, -6.5878e+00,  ...,  2.6297e+00,\n",
            "            5.4264e+00, -2.9451e+00]],\n",
            "\n",
            "         [[ 3.2167e+01,  4.9588e+00, -3.2362e+00,  ..., -1.4103e+01,\n",
            "            1.4921e+01,  2.0480e+01],\n",
            "          [ 9.4104e+00,  1.0623e+01, -4.9694e+00,  ..., -1.9655e+01,\n",
            "            1.1379e+01,  1.2381e+01],\n",
            "          [ 2.1557e+01,  1.5693e+01, -9.2735e-01,  ..., -1.7640e+01,\n",
            "            9.6889e+00,  1.6602e+01],\n",
            "          ...,\n",
            "          [ 1.2718e+01,  4.1273e+00, -2.0908e+00,  ...,  1.9188e+00,\n",
            "            6.5206e+00,  1.3227e+00],\n",
            "          [ 1.2718e+01,  4.1273e+00, -2.0908e+00,  ...,  1.9188e+00,\n",
            "            6.5206e+00,  1.3227e+00],\n",
            "          [ 1.2718e+01,  4.1273e+00, -2.0908e+00,  ...,  1.9188e+00,\n",
            "            6.5206e+00,  1.3227e+00]]],\n",
            "\n",
            "\n",
            "        ...,\n",
            "\n",
            "\n",
            "        [[[-1.4342e+01,  5.2721e+00, -8.1096e+00,  ...,  1.2477e+01,\n",
            "           -9.0658e+00,  2.0836e-01],\n",
            "          [-1.3048e+01,  7.7084e+00, -8.3216e-01,  ...,  5.2784e+00,\n",
            "           -6.4730e+00, -6.4105e-01],\n",
            "          [-1.1630e+01,  3.0994e-01, -2.7907e+00,  ...,  1.1218e+01,\n",
            "           -7.5288e+00,  4.3936e+00],\n",
            "          ...,\n",
            "          [-2.4756e-01, -8.1205e-01, -6.9890e-01,  ...,  6.8434e-01,\n",
            "            6.0953e-01,  5.2184e+00],\n",
            "          [-2.4756e-01, -8.1205e-01, -6.9890e-01,  ...,  6.8434e-01,\n",
            "            6.0953e-01,  5.2184e+00],\n",
            "          [-2.4756e-01, -8.1205e-01, -6.9890e-01,  ...,  6.8434e-01,\n",
            "            6.0953e-01,  5.2184e+00]],\n",
            "\n",
            "         [[ 2.5945e+01,  1.3881e+01,  1.6392e+01,  ...,  1.3488e+01,\n",
            "            1.6034e+01,  2.5325e+01],\n",
            "          [ 1.8058e+01,  9.8142e+00,  9.9287e+00,  ...,  1.4108e+01,\n",
            "            8.6950e+00,  1.5824e+01],\n",
            "          [ 1.7023e+01,  8.0748e+00,  1.2124e+01,  ...,  9.8388e+00,\n",
            "            5.8006e+00,  1.8194e+01],\n",
            "          ...,\n",
            "          [ 5.5680e+00, -2.1598e+00,  3.5298e+00,  ..., -9.4590e-01,\n",
            "            6.6637e+00, -1.9557e+00],\n",
            "          [ 5.5680e+00, -2.1598e+00,  3.5298e+00,  ..., -9.4590e-01,\n",
            "            6.6637e+00, -1.9557e+00],\n",
            "          [ 5.5680e+00, -2.1598e+00,  3.5298e+00,  ..., -9.4590e-01,\n",
            "            6.6637e+00, -1.9557e+00]],\n",
            "\n",
            "         [[-1.2725e+01, -7.5233e+00, -7.2401e+00,  ...,  7.9228e+00,\n",
            "           -6.0802e-01,  3.1739e+00],\n",
            "          [-3.6153e+00, -4.6897e+00, -7.0209e+00,  ...,  9.3320e+00,\n",
            "            2.0847e+00,  3.8730e+00],\n",
            "          [-5.2835e+00,  3.0092e+00, -7.4425e+00,  ...,  3.2669e+00,\n",
            "            9.6761e+00,  9.5102e+00],\n",
            "          ...,\n",
            "          [ 3.4849e-01, -8.1600e+00, -5.1050e+00,  ...,  5.6722e+00,\n",
            "            3.0922e+00, -2.7292e+00],\n",
            "          [ 3.4849e-01, -8.1600e+00, -5.1050e+00,  ...,  5.6722e+00,\n",
            "            3.0922e+00, -2.7292e+00],\n",
            "          [ 3.4849e-01, -8.1600e+00, -5.1050e+00,  ...,  5.6722e+00,\n",
            "            3.0922e+00, -2.7292e+00]],\n",
            "\n",
            "         [[ 1.5126e+01,  2.7363e+00, -6.3900e+00,  ..., -1.5805e+01,\n",
            "            9.0625e+00,  1.0788e+01],\n",
            "          [ 7.6644e+00,  3.3714e+00, -2.8336e+00,  ..., -1.0485e+01,\n",
            "           -1.3363e-01,  1.6117e+00],\n",
            "          [ 1.7109e+01, -6.5531e+00, -3.1057e-01,  ..., -1.3120e+01,\n",
            "            4.2585e-01,  3.8231e+00],\n",
            "          ...,\n",
            "          [ 1.1999e+01,  5.1022e+00, -3.7128e+00,  ..., -8.9348e-01,\n",
            "            4.5809e+00, -9.4987e-01],\n",
            "          [ 1.1999e+01,  5.1022e+00, -3.7128e+00,  ..., -8.9348e-01,\n",
            "            4.5809e+00, -9.4987e-01],\n",
            "          [ 1.1999e+01,  5.1022e+00, -3.7128e+00,  ..., -8.9348e-01,\n",
            "            4.5809e+00, -9.4987e-01]]],\n",
            "\n",
            "\n",
            "        [[[-3.2501e+01,  7.5907e+00, -1.8977e+01,  ...,  2.4905e+01,\n",
            "            9.3053e-01, -3.0491e+00],\n",
            "          [-2.1790e+01, -2.1393e+00, -1.0933e+01,  ...,  1.6895e+01,\n",
            "           -3.0913e+00, -5.0581e+00],\n",
            "          [-1.5892e+01,  7.8451e+00, -3.1068e+00,  ...,  1.1809e+01,\n",
            "           -2.3997e-01, -5.8085e+00],\n",
            "          ...,\n",
            "          [ 9.8238e-03,  7.0717e-01,  1.3721e+01,  ...,  5.4644e+00,\n",
            "            2.9644e+00,  3.5035e+00],\n",
            "          [-1.8744e+01,  3.5875e+00,  6.1572e+00,  ...,  8.4786e+00,\n",
            "            6.1532e-02, -5.8008e+00],\n",
            "          [ 4.0165e+00, -3.6791e+00,  2.1509e+00,  ..., -3.0036e+00,\n",
            "            7.0258e+00,  2.4080e+00]],\n",
            "\n",
            "         [[ 2.5522e+01,  4.1601e+00,  3.2352e+00,  ...,  1.9457e+01,\n",
            "            2.3005e+01,  2.2504e+01],\n",
            "          [ 2.4373e+01,  5.2708e+00,  1.0665e+01,  ...,  7.6934e+00,\n",
            "            1.9479e+01,  1.8196e+01],\n",
            "          [ 1.9277e+01,  6.6951e+00,  1.2048e+01,  ...,  5.9612e+00,\n",
            "            1.5184e+01,  1.1260e+01],\n",
            "          ...,\n",
            "          [ 9.0810e+00,  8.4596e-01,  3.7980e+00,  ...,  1.4424e+01,\n",
            "            4.5210e+00,  2.7306e+00],\n",
            "          [ 7.4903e+00,  1.2884e+01,  8.1322e+00,  ...,  1.6214e+01,\n",
            "            7.0250e+00,  1.4861e+01],\n",
            "          [-4.0368e+00, -9.4184e+00, -1.9151e+01,  ..., -5.8707e+00,\n",
            "           -2.1482e+00, -6.2928e-01]],\n",
            "\n",
            "         [[-1.3650e+01, -2.5923e+01,  1.1506e+00,  ...,  1.3690e+01,\n",
            "            6.2218e-01,  5.6637e+00],\n",
            "          [-6.5235e+00, -2.0828e+01, -6.5878e+00,  ...,  4.4440e+00,\n",
            "           -8.4275e-01, -1.0389e+01],\n",
            "          [ 6.1572e+00, -2.7710e+01, -6.7229e+00,  ...,  3.2463e+00,\n",
            "            5.1124e+00,  5.2058e+00],\n",
            "          ...,\n",
            "          [-5.8977e+00, -1.6687e+01, -1.4839e+00,  ..., -5.1820e-01,\n",
            "           -4.4692e+00, -4.0029e+00],\n",
            "          [-3.3796e+00, -2.6515e+01, -3.9375e+00,  ...,  5.1784e+00,\n",
            "           -4.5254e+00,  5.8149e+00],\n",
            "          [ 1.6838e+00,  3.4703e+00,  5.0121e+00,  ..., -2.3270e+00,\n",
            "            7.7511e-01, -7.2083e-01]],\n",
            "\n",
            "         [[ 8.0342e+00,  1.2758e+01, -5.5159e+00,  ..., -1.8544e+01,\n",
            "            2.1117e+01,  1.1297e+01],\n",
            "          [ 7.4745e+00,  2.9192e+00, -1.2660e+01,  ..., -1.0574e+01,\n",
            "            1.8584e+01,  9.8810e+00],\n",
            "          [ 3.6148e+00,  1.0530e+01, -6.0193e+00,  ..., -5.8997e+00,\n",
            "            8.4128e+00,  1.0667e+01],\n",
            "          ...,\n",
            "          [ 5.7010e+00,  4.3339e+00, -7.4963e+00,  ..., -1.0517e+00,\n",
            "            2.3654e+01,  3.1522e+00],\n",
            "          [ 8.6176e+00,  9.4167e+00, -3.4600e+00,  ..., -4.5120e+00,\n",
            "            1.9541e+01,  1.3217e+01],\n",
            "          [-4.4502e+00,  3.3351e+00, -2.2605e+00,  ...,  1.4772e+00,\n",
            "            4.6581e+00,  4.2271e+00]]],\n",
            "\n",
            "\n",
            "        [[[-1.4239e+01, -4.4353e+00, -8.1657e+00,  ...,  1.0512e+01,\n",
            "            5.4905e+00, -1.0145e+01],\n",
            "          [-6.2569e+00,  1.4892e+00, -1.0280e+01,  ...,  6.8695e+00,\n",
            "           -5.1733e+00, -3.1799e+00],\n",
            "          [-5.9709e-01,  4.6643e+00, -1.0912e+01,  ...,  3.7556e+00,\n",
            "           -3.8945e+00, -7.2824e+00],\n",
            "          ...,\n",
            "          [-3.5934e+00,  3.4134e+00, -3.3589e+00,  ...,  3.8705e-01,\n",
            "           -4.1624e-01,  5.5167e+00],\n",
            "          [-3.6014e+00,  3.3926e+00, -3.3077e+00,  ...,  3.8014e-01,\n",
            "           -4.0180e-01,  5.4791e+00],\n",
            "          [-3.4624e+00,  3.4641e+00, -3.3391e+00,  ...,  3.3540e-01,\n",
            "           -8.6455e-02,  5.3513e+00]],\n",
            "\n",
            "         [[ 1.6410e+01,  1.5309e+00,  2.7322e-01,  ...,  8.3377e+00,\n",
            "            1.0043e+01,  1.5036e+01],\n",
            "          [ 1.4262e+01,  5.3308e+00,  3.0611e+00,  ...,  1.1638e+00,\n",
            "            1.4082e+01,  1.4882e+01],\n",
            "          [ 1.5036e+01,  4.9148e+00,  4.0392e+00,  ...,  1.1371e+00,\n",
            "            7.6640e+00,  1.2524e+01],\n",
            "          ...,\n",
            "          [ 4.6685e+00,  1.6372e+00,  4.1081e+00,  ..., -2.2390e-01,\n",
            "            2.8615e+00, -4.5433e+00],\n",
            "          [ 4.6615e+00,  1.6073e+00,  4.0709e+00,  ..., -1.9991e-01,\n",
            "            2.8172e+00, -4.5513e+00],\n",
            "          [ 4.7111e+00,  1.6531e+00,  4.2690e+00,  ...,  1.9377e-01,\n",
            "            2.4852e+00, -4.9149e+00]],\n",
            "\n",
            "         [[-6.0774e+00, -4.9632e+00, -6.5019e+00,  ...,  1.3813e+01,\n",
            "            7.3407e-01, -6.8465e+00],\n",
            "          [-6.4039e+00,  2.9286e-02, -1.2838e+01,  ...,  2.0516e+01,\n",
            "            1.5604e+00,  4.9167e+00],\n",
            "          [-1.7190e+00, -2.1742e+00, -8.6407e+00,  ...,  1.0013e+01,\n",
            "            7.2251e+00,  4.5955e-01],\n",
            "          ...,\n",
            "          [-2.3830e+00, -8.5285e+00, -3.3362e-01,  ...,  4.2638e+00,\n",
            "            2.6369e+00,  1.5867e+00],\n",
            "          [-2.3475e+00, -8.5283e+00, -2.7981e-01,  ...,  4.3683e+00,\n",
            "            2.6266e+00,  1.6493e+00],\n",
            "          [-2.8311e+00, -8.2152e+00, -2.7470e-01,  ...,  4.4552e+00,\n",
            "            2.5527e+00,  1.5535e+00]],\n",
            "\n",
            "         [[ 5.4083e+00, -4.8790e+00, -9.3785e+00,  ..., -4.4426e+00,\n",
            "            4.4892e+00,  1.7773e+00],\n",
            "          [ 4.6755e-01, -8.8691e-01, -7.9540e+00,  ...,  4.6466e+00,\n",
            "            4.9405e+00, -4.9171e+00],\n",
            "          [ 3.0328e+00, -3.3435e+00, -1.5376e+01,  ...,  7.3036e-01,\n",
            "            1.5917e+00,  5.9844e+00],\n",
            "          ...,\n",
            "          [ 1.2034e+01,  6.1611e+00,  2.0695e+00,  ...,  3.3455e+00,\n",
            "            6.3938e+00,  4.0102e-01],\n",
            "          [ 1.2048e+01,  6.1891e+00,  2.0024e+00,  ...,  3.3771e+00,\n",
            "            6.4049e+00,  4.3234e-01],\n",
            "          [ 1.1890e+01,  6.5305e+00,  2.1417e+00,  ...,  3.7434e+00,\n",
            "            6.4797e+00,  1.9801e-01]]]], grad_fn=<TransposeBackward0>), tensor([[[[-1.1315e+00, -1.7513e+01, -7.0328e+00,  ...,  7.5960e+00,\n",
            "            2.0047e+01, -4.7468e+00],\n",
            "          [-1.5271e+01, -5.7900e+00, -2.7050e+00,  ..., -9.4430e+00,\n",
            "            1.5793e+01, -1.0421e+01],\n",
            "          [-9.2056e+00,  8.3452e+00,  6.3378e+00,  ..., -1.2230e+01,\n",
            "            2.4879e+00, -1.4231e+01],\n",
            "          ...,\n",
            "          [-2.8022e+00, -1.0455e+01, -7.2852e+00,  ..., -3.7545e+00,\n",
            "            8.9160e+00, -4.9788e-01],\n",
            "          [-2.6755e+00, -9.1882e+00, -1.4962e+00,  ..., -2.0073e+01,\n",
            "            6.4798e+00, -5.0928e+00],\n",
            "          [ 1.7237e+00, -2.6284e+00,  2.0768e+00,  ..., -4.5055e-01,\n",
            "           -1.4150e+00, -3.8069e-01]],\n",
            "\n",
            "         [[-8.1392e+00, -1.8091e+00, -1.8279e+01,  ...,  3.6831e+00,\n",
            "           -3.6672e+01,  1.2989e+01],\n",
            "          [-6.4652e+00, -8.8820e+00, -1.7871e+01,  ...,  1.2337e+01,\n",
            "           -1.7712e+01,  9.0922e+00],\n",
            "          [-2.7830e+00,  5.9926e+00, -1.9843e+01,  ...,  6.7976e+00,\n",
            "           -1.0437e+00,  3.1816e+00],\n",
            "          ...,\n",
            "          [-1.9632e+01,  3.6783e+00, -1.5327e+01,  ..., -1.9270e+00,\n",
            "           -1.5835e+01, -9.8747e+00],\n",
            "          [-1.2702e+01, -6.7565e-01,  2.3957e+00,  ...,  1.1149e+01,\n",
            "           -2.5614e+01, -1.8405e+01],\n",
            "          [-5.1155e-01, -5.2284e+00, -4.1351e+00,  ...,  5.9909e-01,\n",
            "           -5.7481e+00, -1.0196e+00]],\n",
            "\n",
            "         [[-5.9165e+00, -7.3648e+00, -8.0315e+00,  ...,  5.5070e+00,\n",
            "            2.5269e+00,  1.2472e+01],\n",
            "          [-1.9321e+01, -5.4428e-01, -2.6992e+00,  ...,  2.6453e+00,\n",
            "           -2.1537e+01, -1.0798e+01],\n",
            "          [-1.5400e+01, -7.2687e+00, -4.4377e+00,  ..., -8.6842e+00,\n",
            "           -1.7659e+00, -5.6311e+00],\n",
            "          ...,\n",
            "          [-2.8460e+00, -6.2424e+00, -3.2552e+01,  ..., -8.1233e+00,\n",
            "           -1.3915e+01, -1.5448e+01],\n",
            "          [ 4.7268e+00, -1.8298e+00, -1.3164e+01,  ...,  9.3756e+00,\n",
            "           -7.5587e+00, -1.2058e-01],\n",
            "          [-3.3572e-01,  3.0191e+00, -2.9461e-01,  ...,  2.4015e+00,\n",
            "           -6.7136e+00,  1.2067e+00]],\n",
            "\n",
            "         [[-5.8746e+00, -2.5855e+00,  4.1136e+00,  ...,  1.8792e-01,\n",
            "            9.9983e+00,  7.8545e+00],\n",
            "          [ 1.1551e+01, -1.2613e+01, -2.5130e-01,  ...,  3.1412e+00,\n",
            "            1.3388e+01,  1.4589e+01],\n",
            "          [ 1.8160e+00, -8.5556e+00,  2.6336e+00,  ...,  1.1594e+01,\n",
            "            3.8649e+00,  1.0201e+01],\n",
            "          ...,\n",
            "          [ 3.3047e+00,  1.2293e+01,  1.2600e+01,  ..., -1.2217e+00,\n",
            "           -1.2464e+00, -1.0623e+01],\n",
            "          [-1.2531e+00,  7.1570e+00,  7.0415e+00,  ..., -1.7884e+01,\n",
            "           -2.7390e+00, -1.5576e+00],\n",
            "          [ 2.9633e+00,  3.4202e+00,  1.0340e+00,  ..., -1.5077e-01,\n",
            "            2.6673e-02,  1.6285e+00]]],\n",
            "\n",
            "\n",
            "        [[[ 4.8252e+00, -1.1005e+01,  9.2036e+00,  ..., -6.9294e+00,\n",
            "            6.6972e+00,  8.7973e+00],\n",
            "          [ 7.8897e+00, -3.4223e+01,  1.0880e+01,  ..., -2.7073e+00,\n",
            "            6.4553e+00, -4.0096e+00],\n",
            "          [ 3.9170e+00, -3.3961e+01, -9.4780e-01,  ...,  2.9961e+00,\n",
            "            5.7087e+00,  1.0437e+00],\n",
            "          ...,\n",
            "          [-6.7230e+00, -5.0496e+00,  1.6501e+01,  ..., -1.6882e+01,\n",
            "           -2.4152e+00,  5.3915e+00],\n",
            "          [-1.1331e+00, -3.9883e+00, -2.4497e+00,  ..., -3.5580e+00,\n",
            "           -8.6141e-02,  8.5859e+00],\n",
            "          [ 1.5516e+00, -2.3299e+00,  2.4001e+00,  ...,  4.3451e-02,\n",
            "           -1.2832e+00, -1.6250e-01]],\n",
            "\n",
            "         [[-1.9134e+01,  1.2137e+01, -1.1156e+01,  ..., -2.5810e+00,\n",
            "            5.1468e+00, -1.0522e+01],\n",
            "          [-1.8063e+01,  3.1167e+00, -1.8966e+01,  ..., -1.2105e+01,\n",
            "            7.4964e-01,  8.0307e+00],\n",
            "          [-1.3278e+01,  2.3016e+00, -1.2192e+01,  ..., -1.5715e+01,\n",
            "           -5.1137e+00,  1.3023e+00],\n",
            "          ...,\n",
            "          [-6.8424e+00, -7.2036e+00, -4.2822e+00,  ..., -2.3012e+00,\n",
            "           -1.7503e+01, -1.1848e+01],\n",
            "          [-9.3519e+00, -1.1719e+01, -7.3652e+00,  ..., -1.2288e+01,\n",
            "           -9.9236e+00, -1.4000e+01],\n",
            "          [-1.2282e-01, -5.3310e+00, -4.3245e+00,  ...,  2.9073e-01,\n",
            "           -5.0407e+00, -9.8728e-01]],\n",
            "\n",
            "         [[-2.6868e+01,  3.3392e+00, -3.3864e+00,  ...,  5.3816e+00,\n",
            "            1.8735e+01, -1.9457e+01],\n",
            "          [-2.3587e+01, -1.0251e+01, -3.0689e+00,  ..., -9.9219e-02,\n",
            "           -3.8880e+00,  7.7954e-01],\n",
            "          [-1.1313e+01, -1.5194e+01,  5.7670e+00,  ..., -1.2913e+01,\n",
            "           -1.4367e+00, -1.2798e+00],\n",
            "          ...,\n",
            "          [-1.2543e+01, -6.7016e+00,  1.2660e+00,  ...,  1.4052e+01,\n",
            "           -3.1351e-01, -8.1015e+00],\n",
            "          [-2.3590e+00, -4.8100e-01, -5.4409e+00,  ...,  1.7283e+01,\n",
            "           -4.7933e+00,  4.7790e+00],\n",
            "          [-1.0069e+00,  3.1875e+00, -4.2917e-02,  ...,  2.3625e+00,\n",
            "           -6.5871e+00,  1.2155e+00]],\n",
            "\n",
            "         [[ 7.9149e-01,  1.9495e+01, -2.8743e+00,  ..., -1.1391e+01,\n",
            "           -1.4973e+01,  2.9666e+00],\n",
            "          [ 3.8941e+00,  1.3909e+01, -1.0118e+01,  ..., -1.6562e+01,\n",
            "           -1.0002e+01,  7.8641e+00],\n",
            "          [ 9.7106e+00,  5.6952e+00, -8.6466e+00,  ..., -9.6169e-01,\n",
            "            5.3094e+00,  5.1103e+00],\n",
            "          ...,\n",
            "          [-1.4814e+01, -4.6182e+00,  3.1110e+00,  ..., -5.7417e+00,\n",
            "           -8.8510e+00, -4.3307e+00],\n",
            "          [ 7.7549e+00, -8.8444e-01, -1.9261e+00,  ..., -7.5750e+00,\n",
            "           -4.2974e+00,  1.0512e+01],\n",
            "          [ 3.1170e+00,  3.2679e+00,  4.7644e-01,  ..., -2.0817e-03,\n",
            "            1.8227e-01,  1.4573e+00]]],\n",
            "\n",
            "\n",
            "        [[[-1.7078e+01, -1.1096e+01, -7.0197e+00,  ...,  1.5712e+01,\n",
            "            1.4713e+00,  5.2976e-01],\n",
            "          [-4.1256e+00, -5.4545e+00, -1.9500e+01,  ...,  9.0272e+00,\n",
            "            7.8939e+00,  5.7834e+00],\n",
            "          [-5.9160e+00, -1.1698e+01, -1.5305e+01,  ...,  9.2086e+00,\n",
            "            1.0385e+01,  1.4617e+00],\n",
            "          ...,\n",
            "          [-6.0967e-01, -5.8443e+00,  3.2803e+00,  ...,  2.3049e-02,\n",
            "           -5.8470e+00, -6.4283e+00],\n",
            "          [-6.0967e-01, -5.8443e+00,  3.2803e+00,  ...,  2.3049e-02,\n",
            "           -5.8470e+00, -6.4283e+00],\n",
            "          [-6.0967e-01, -5.8443e+00,  3.2803e+00,  ...,  2.3049e-02,\n",
            "           -5.8470e+00, -6.4283e+00]],\n",
            "\n",
            "         [[-1.4670e+01, -1.6456e+01, -1.1589e+01,  ...,  5.0019e+00,\n",
            "           -2.1093e+01, -1.5963e+01],\n",
            "          [-5.6453e+00, -2.2659e+01, -1.3110e+00,  ..., -2.5338e+00,\n",
            "           -1.3513e+01, -2.0863e+01],\n",
            "          [-1.8921e+01, -1.1200e+01,  4.3815e+00,  ..., -3.7528e+00,\n",
            "           -1.3547e+01, -1.0112e+01],\n",
            "          ...,\n",
            "          [-4.3972e+00, -5.2528e-02, -1.8510e+01,  ...,  2.9065e-01,\n",
            "           -3.7790e+00, -4.2243e+00],\n",
            "          [-4.3972e+00, -5.2528e-02, -1.8510e+01,  ...,  2.9065e-01,\n",
            "           -3.7790e+00, -4.2243e+00],\n",
            "          [-4.3972e+00, -5.2528e-02, -1.8510e+01,  ...,  2.9065e-01,\n",
            "           -3.7790e+00, -4.2243e+00]],\n",
            "\n",
            "         [[-1.0481e+01, -1.5951e+01,  1.2588e+00,  ...,  4.1595e+00,\n",
            "           -1.0198e+01,  4.9900e+00],\n",
            "          [ 1.1413e+01, -1.5662e+01, -1.6261e+01,  ...,  3.4778e+00,\n",
            "           -4.3819e+00, -1.3397e+01],\n",
            "          [-6.9280e-01, -1.7707e+01, -2.1385e+01,  ...,  6.6318e+00,\n",
            "           -1.6979e+00, -5.5047e+00],\n",
            "          ...,\n",
            "          [ 1.1620e+00, -9.2973e+00, -5.6383e+00,  ..., -2.7149e+00,\n",
            "           -1.6048e+00, -3.1866e+00],\n",
            "          [ 1.1620e+00, -9.2973e+00, -5.6383e+00,  ..., -2.7149e+00,\n",
            "           -1.6048e+00, -3.1866e+00],\n",
            "          [ 1.1620e+00, -9.2973e+00, -5.6383e+00,  ..., -2.7149e+00,\n",
            "           -1.6048e+00, -3.1866e+00]],\n",
            "\n",
            "         [[ 1.2934e+01, -2.7903e+00,  7.5184e+00,  ..., -1.3497e+01,\n",
            "            1.3811e+01,  5.5833e+00],\n",
            "          [ 2.8197e+01,  4.7334e-01,  2.7225e+01,  ...,  1.5555e+01,\n",
            "            1.3867e+01, -1.2996e+00],\n",
            "          [ 2.2734e+01, -1.3141e+00,  1.3459e+01,  ...,  8.9962e+00,\n",
            "            1.4382e+01,  4.8022e+00],\n",
            "          ...,\n",
            "          [ 1.8465e+00,  5.1487e+00,  3.9367e+00,  ..., -9.7698e+00,\n",
            "           -6.3236e+00,  8.8869e+00],\n",
            "          [ 1.8465e+00,  5.1487e+00,  3.9367e+00,  ..., -9.7698e+00,\n",
            "           -6.3236e+00,  8.8869e+00],\n",
            "          [ 1.8465e+00,  5.1487e+00,  3.9367e+00,  ..., -9.7698e+00,\n",
            "           -6.3236e+00,  8.8869e+00]]],\n",
            "\n",
            "\n",
            "        ...,\n",
            "\n",
            "\n",
            "        [[[-5.9374e+00, -2.4426e+01, -7.4709e+00,  ...,  6.6223e+00,\n",
            "            1.8171e+01,  4.4933e+00],\n",
            "          [ 8.4372e+00, -1.3649e+01,  9.1732e+00,  ..., -6.9716e+00,\n",
            "            1.7506e+01, -1.6493e+01],\n",
            "          [ 1.1898e+01, -2.8034e+01,  8.3252e+00,  ...,  4.7425e+00,\n",
            "            1.6354e+01,  4.4241e+00],\n",
            "          ...,\n",
            "          [-6.3056e+00, -7.5744e+00, -3.4961e-03,  ..., -3.0463e+00,\n",
            "           -4.2904e-01, -1.7711e+00],\n",
            "          [-6.3056e+00, -7.5744e+00, -3.4961e-03,  ..., -3.0463e+00,\n",
            "           -4.2904e-01, -1.7711e+00],\n",
            "          [-6.3056e+00, -7.5744e+00, -3.4961e-03,  ..., -3.0463e+00,\n",
            "           -4.2904e-01, -1.7711e+00]],\n",
            "\n",
            "         [[-3.2284e+01, -1.9886e+01, -7.4848e-01,  ..., -1.0586e+00,\n",
            "           -7.5556e-01, -4.3538e+00],\n",
            "          [-2.9662e+01, -1.5539e+01,  2.8917e+00,  ..., -6.1048e-01,\n",
            "            4.2467e+00, -2.7580e+00],\n",
            "          [-7.0180e+00, -2.5791e+01, -6.9319e+00,  ..., -5.2555e+00,\n",
            "           -1.4640e+00, -2.7858e+00],\n",
            "          ...,\n",
            "          [-9.4229e+00, -1.7976e+00, -1.5073e+01,  ...,  5.0689e-01,\n",
            "           -1.9087e-01,  2.5022e-02],\n",
            "          [-9.4229e+00, -1.7976e+00, -1.5073e+01,  ...,  5.0689e-01,\n",
            "           -1.9087e-01,  2.5022e-02],\n",
            "          [-9.4229e+00, -1.7976e+00, -1.5073e+01,  ...,  5.0689e-01,\n",
            "           -1.9087e-01,  2.5022e-02]],\n",
            "\n",
            "         [[-5.9433e+00,  4.7100e+00,  7.0548e-01,  ...,  1.6082e+01,\n",
            "            4.6038e+00, -1.5388e+00],\n",
            "          [-3.6901e+00, -6.5119e+00, -3.9971e+00,  ...,  1.8564e+00,\n",
            "           -1.6674e+01, -8.1692e-01],\n",
            "          [ 5.8362e+00, -8.1609e+00, -6.3204e+00,  ..., -4.0704e-01,\n",
            "            2.3297e+00,  1.1432e+01],\n",
            "          ...,\n",
            "          [ 4.2427e+00, -5.2785e+00, -5.8764e+00,  ...,  3.0440e+00,\n",
            "            6.1718e+00, -5.3813e-01],\n",
            "          [ 4.2427e+00, -5.2785e+00, -5.8764e+00,  ...,  3.0440e+00,\n",
            "            6.1718e+00, -5.3813e-01],\n",
            "          [ 4.2427e+00, -5.2785e+00, -5.8764e+00,  ...,  3.0440e+00,\n",
            "            6.1718e+00, -5.3813e-01]],\n",
            "\n",
            "         [[ 8.1738e+00,  1.7972e+01, -5.9821e+00,  ...,  4.3067e+00,\n",
            "           -1.0575e+01,  1.4247e+01],\n",
            "          [ 2.2471e+01,  7.9120e+00,  1.9769e-01,  ...,  1.5487e+01,\n",
            "           -7.8509e-01,  1.3669e+01],\n",
            "          [ 1.8591e+01,  1.1203e+01,  1.5012e+01,  ...,  9.4361e+00,\n",
            "           -3.3656e-01,  1.9322e+01],\n",
            "          ...,\n",
            "          [ 2.5548e+00,  1.1055e+01, -3.0017e+00,  ..., -5.7112e+00,\n",
            "           -1.0542e+01,  1.1493e+01],\n",
            "          [ 2.5548e+00,  1.1055e+01, -3.0017e+00,  ..., -5.7112e+00,\n",
            "           -1.0542e+01,  1.1493e+01],\n",
            "          [ 2.5548e+00,  1.1055e+01, -3.0017e+00,  ..., -5.7112e+00,\n",
            "           -1.0542e+01,  1.1493e+01]]],\n",
            "\n",
            "\n",
            "        [[[ 2.9442e+00, -3.0330e+01, -1.3255e+01,  ...,  1.9143e+01,\n",
            "            3.6091e+01, -1.4154e+00],\n",
            "          [ 1.0965e+01, -9.4954e+00, -4.9039e+00,  ...,  4.0399e+00,\n",
            "            1.5411e+01,  8.9628e+00],\n",
            "          [-6.9322e+00, -1.4975e+00,  5.1343e+00,  ..., -1.1370e+01,\n",
            "            1.7662e+01, -8.2693e+00],\n",
            "          ...,\n",
            "          [-6.2606e+00, -5.6009e-01, -1.9679e+01,  ..., -3.5586e+01,\n",
            "           -6.3772e+00,  9.6193e+00],\n",
            "          [-2.9034e+00,  8.8714e-01, -2.7070e+00,  ..., -2.0999e+01,\n",
            "            1.0701e+01, -5.5322e+00],\n",
            "          [ 1.4692e+00, -2.6387e+00,  2.0839e+00,  ..., -5.3396e-01,\n",
            "           -1.1867e+00,  6.9412e-02]],\n",
            "\n",
            "         [[ 4.5600e-01, -1.1768e+01,  5.6535e+00,  ..., -2.4501e+00,\n",
            "           -1.4464e+01,  4.9462e+00],\n",
            "          [-9.6647e+00, -2.2951e+01, -9.4070e+00,  ...,  4.0608e+00,\n",
            "           -2.1918e+01,  7.8976e+00],\n",
            "          [-9.2934e-01, -5.9222e+00, -5.7542e+00,  ..., -6.6840e-02,\n",
            "           -6.3706e+00, -5.3992e+00],\n",
            "          ...,\n",
            "          [-3.3191e+01,  1.3039e+01, -4.9488e+00,  ..., -4.6869e+00,\n",
            "           -4.1368e+01,  2.7570e+00],\n",
            "          [-5.2815e-01, -8.0492e-02, -3.2138e+00,  ..., -2.6884e+00,\n",
            "           -2.7813e+01, -3.3894e+00],\n",
            "          [-3.0393e-01, -5.0696e+00, -4.2445e+00,  ...,  3.0352e-01,\n",
            "           -5.0874e+00, -1.1873e+00]],\n",
            "\n",
            "         [[ 1.9836e+00, -2.1220e+00, -1.4521e+01,  ..., -8.3301e+00,\n",
            "            1.4046e+01,  1.0423e+01],\n",
            "          [-2.9410e+00, -8.5918e-01, -1.9262e+01,  ...,  2.5471e+00,\n",
            "           -2.1800e+00,  5.9148e+00],\n",
            "          [-2.1438e+00, -1.0033e+01, -3.9920e+00,  ..., -5.2985e+00,\n",
            "           -6.9457e+00, -6.0385e+00],\n",
            "          ...,\n",
            "          [ 4.7321e+00,  9.3524e+00,  1.3582e+01,  ..., -4.6704e+00,\n",
            "           -1.1663e+01, -6.7272e+00],\n",
            "          [-2.9373e+00,  1.0913e+01,  9.3440e+00,  ..., -4.0581e+00,\n",
            "           -4.2432e+00, -1.5028e+01],\n",
            "          [-3.7419e-01,  3.1981e+00, -3.4647e-01,  ...,  2.1118e+00,\n",
            "           -6.3011e+00,  9.5350e-01]],\n",
            "\n",
            "         [[ 9.4553e+00,  1.6147e+01, -8.4927e+00,  ..., -4.8583e-01,\n",
            "            1.9696e+00,  1.4619e+01],\n",
            "          [ 9.3527e+00,  9.2006e+00, -3.3408e+00,  ...,  1.9445e+00,\n",
            "            9.3329e+00,  1.2921e+01],\n",
            "          [ 1.5352e+01,  6.5776e+00,  1.4444e+01,  ...,  5.8898e+00,\n",
            "            1.6920e+01,  1.0345e+01],\n",
            "          ...,\n",
            "          [ 3.1682e+00, -4.1052e+00,  1.8240e+01,  ..., -2.0260e+01,\n",
            "            4.8356e+00,  6.0796e+00],\n",
            "          [-7.6633e+00,  2.6324e+00,  1.6721e+01,  ..., -1.2886e+01,\n",
            "           -1.0134e+01,  3.4255e+00],\n",
            "          [ 2.7639e+00,  3.9258e+00,  1.3848e+00,  ..., -7.5921e-02,\n",
            "           -9.0640e-02,  1.5061e+00]]],\n",
            "\n",
            "\n",
            "        [[[ 7.3495e+00, -2.0900e+01, -3.5020e+00,  ...,  6.1227e+00,\n",
            "            1.0801e+00,  1.0371e+01],\n",
            "          [-9.9146e+00, -5.7096e+00,  8.4331e+00,  ..., -5.7836e-01,\n",
            "            1.1738e+01,  6.7493e-01],\n",
            "          [ 1.5841e+00, -2.2026e+01, -1.9597e+00,  ..., -5.7774e+00,\n",
            "            8.7294e+00,  9.2790e+00],\n",
            "          ...,\n",
            "          [-6.1633e+00, -6.3877e+00,  2.4764e-02,  ...,  6.0704e-01,\n",
            "           -4.0640e+00,  1.8596e+00],\n",
            "          [-6.2377e+00, -6.4685e+00, -2.0103e-01,  ...,  7.3148e-01,\n",
            "           -3.9876e+00,  1.9249e+00],\n",
            "          [-7.2074e+00, -6.0767e+00, -1.3985e+00,  ...,  1.9963e+00,\n",
            "           -3.7963e+00,  2.3369e+00]],\n",
            "\n",
            "         [[ 1.2774e+01, -1.8151e+01, -1.6868e+01,  ..., -1.4214e+01,\n",
            "           -1.1176e-01, -1.4722e+01],\n",
            "          [ 2.3316e+00, -1.3246e+01, -9.3995e+00,  ...,  1.1106e+01,\n",
            "           -8.9286e+00, -1.1082e+01],\n",
            "          [-1.5609e+01, -1.7059e+01,  5.3444e+00,  ..., -2.5460e+00,\n",
            "           -1.4631e+01, -2.3939e+01],\n",
            "          ...,\n",
            "          [-1.3874e+01, -3.2379e+00, -1.5778e+01,  ...,  3.3286e+00,\n",
            "           -7.7868e+00,  5.3326e+00],\n",
            "          [-1.3792e+01, -3.3626e+00, -1.5641e+01,  ...,  3.3347e+00,\n",
            "           -7.8301e+00,  5.2643e+00],\n",
            "          [-1.3720e+01, -3.4606e+00, -1.5022e+01,  ...,  3.6548e+00,\n",
            "           -7.7734e+00,  5.1798e+00]],\n",
            "\n",
            "         [[-5.6782e+00, -4.8190e-01,  4.0365e+00,  ...,  1.0594e+01,\n",
            "            4.9795e+00, -3.7352e+00],\n",
            "          [ 1.6396e+00, -4.7270e+00, -8.2374e+00,  ..., -8.0763e+00,\n",
            "            5.5794e+00, -2.0526e+01],\n",
            "          [ 5.7370e+00, -2.2933e+00, -7.9875e+00,  ..., -6.3644e+00,\n",
            "           -2.6648e+00, -3.8556e+00],\n",
            "          ...,\n",
            "          [ 6.3980e+00, -9.3753e+00, -8.8109e+00,  ..., -7.0490e-01,\n",
            "            1.4798e+00,  2.8588e+00],\n",
            "          [ 6.4534e+00, -9.5212e+00, -8.7348e+00,  ..., -8.1411e-01,\n",
            "            1.5090e+00,  2.8581e+00],\n",
            "          [ 6.6356e+00, -9.8758e+00, -8.9524e+00,  ..., -1.1556e+00,\n",
            "            1.4167e+00,  3.1487e+00]],\n",
            "\n",
            "         [[-4.0470e+00,  2.5984e+00, -6.4253e-01,  ..., -1.9065e+01,\n",
            "           -9.1570e+00,  8.3104e+00],\n",
            "          [ 9.4585e+00,  4.1791e+00,  1.0980e+00,  ..., -1.2070e+01,\n",
            "           -3.0759e+00,  7.2309e+00],\n",
            "          [ 3.3080e+00,  1.4278e+00,  3.3725e+00,  ..., -1.3529e+01,\n",
            "           -1.0107e+01,  8.8379e+00],\n",
            "          ...,\n",
            "          [ 4.7363e+00,  5.6923e+00,  3.7824e+00,  ..., -5.6564e+00,\n",
            "           -9.8841e+00,  9.2391e+00],\n",
            "          [ 4.7764e+00,  5.7161e+00,  3.7732e+00,  ..., -5.6809e+00,\n",
            "           -9.8479e+00,  9.3159e+00],\n",
            "          [ 5.0613e+00,  4.7981e+00,  3.5910e+00,  ..., -5.6077e+00,\n",
            "           -9.5170e+00,  8.4409e+00]]]], grad_fn=<TransposeBackward0>)), (tensor([[[[-0.2865, -0.1378,  0.1104,  ..., -0.0838, -0.1098,  0.0902],\n",
            "          [-0.0031,  0.0704, -0.5665,  ..., -0.2064, -0.3742,  0.0065],\n",
            "          [ 0.1539,  0.1378, -0.7328,  ...,  0.0599, -0.1804,  0.1629],\n",
            "          ...,\n",
            "          [-0.2489, -0.1251,  0.1070,  ..., -0.0500, -0.1355,  0.1089],\n",
            "          [-0.2489, -0.1251,  0.1070,  ..., -0.0500, -0.1355,  0.1089],\n",
            "          [-0.2470, -0.1239,  0.1067,  ..., -0.0488, -0.1371,  0.1090]],\n",
            "\n",
            "         [[ 0.1590,  0.1666, -0.1325,  ...,  0.0507, -0.1787, -0.1811],\n",
            "          [-0.5534, -0.2436, -0.4516,  ...,  0.2850, -0.2588, -0.6132],\n",
            "          [ 0.0090,  0.0052, -1.0089,  ...,  0.3717, -0.3618, -0.0388],\n",
            "          ...,\n",
            "          [ 0.1198,  0.1396, -0.0987,  ...,  0.0426, -0.1698, -0.2016],\n",
            "          [ 0.1198,  0.1396, -0.0987,  ...,  0.0426, -0.1698, -0.2016],\n",
            "          [ 0.1186,  0.1386, -0.0969,  ...,  0.0422, -0.1698, -0.2027]],\n",
            "\n",
            "         [[-0.0408, -0.0895,  0.1493,  ..., -0.0591,  0.0180, -0.7003],\n",
            "          [-0.5487, -0.2760,  0.2283,  ...,  0.2875, -0.2886, -0.1375],\n",
            "          [-0.3814, -0.2688, -0.3045,  ..., -0.2314,  0.1064, -0.2401],\n",
            "          ...,\n",
            "          [-0.0349, -0.0574,  0.1257,  ..., -0.0507, -0.0077, -0.6948],\n",
            "          [-0.0349, -0.0574,  0.1257,  ..., -0.0507, -0.0077, -0.6948],\n",
            "          [-0.0348, -0.0560,  0.1242,  ..., -0.0510, -0.0098, -0.6946]],\n",
            "\n",
            "         [[ 0.0081,  0.2073, -0.1035,  ..., -0.2016, -0.3686,  0.1106],\n",
            "          [-0.0695, -0.3690, -0.0866,  ..., -0.5957,  0.0673, -0.2675],\n",
            "          [ 0.6297, -0.0116, -0.1448,  ..., -0.7584,  0.0149, -0.4604],\n",
            "          ...,\n",
            "          [-0.0253,  0.1665, -0.1183,  ..., -0.2092, -0.2892,  0.0580],\n",
            "          [-0.0253,  0.1665, -0.1183,  ..., -0.2092, -0.2892,  0.0580],\n",
            "          [-0.0268,  0.1645, -0.1189,  ..., -0.2100, -0.2856,  0.0544]]],\n",
            "\n",
            "\n",
            "        [[[-0.2943, -0.1455,  0.1090,  ..., -0.0871, -0.1061,  0.0872],\n",
            "          [-0.0596,  0.0299, -0.5182,  ..., -0.2475, -0.4375, -0.0219],\n",
            "          [ 0.1116,  0.1460, -0.7535,  ...,  0.0193, -0.1663,  0.0848],\n",
            "          ...,\n",
            "          [-0.2600, -0.1319,  0.1044,  ..., -0.0579, -0.1313,  0.1063],\n",
            "          [-0.2600, -0.1319,  0.1044,  ..., -0.0579, -0.1313,  0.1063],\n",
            "          [-0.2583, -0.1307,  0.1041,  ..., -0.0571, -0.1330,  0.1066]],\n",
            "\n",
            "         [[ 0.1626,  0.1666, -0.1223,  ...,  0.0573, -0.1821, -0.1800],\n",
            "          [-0.4588, -0.2622, -0.3111,  ...,  0.3661, -0.3169, -0.5912],\n",
            "          [ 0.0868, -0.0046, -0.8837,  ...,  0.4210, -0.3215, -0.0014],\n",
            "          ...,\n",
            "          [ 0.1281,  0.1368, -0.0852,  ...,  0.0512, -0.1735, -0.2032],\n",
            "          [ 0.1281,  0.1368, -0.0852,  ...,  0.0512, -0.1735, -0.2032],\n",
            "          [ 0.1272,  0.1357, -0.0834,  ...,  0.0510, -0.1737, -0.2044]],\n",
            "\n",
            "         [[-0.0367, -0.0902,  0.1552,  ..., -0.0598,  0.0158, -0.6993],\n",
            "          [-0.5351, -0.3089,  0.2155,  ...,  0.2532, -0.2927, -0.1350],\n",
            "          [-0.3531, -0.2894, -0.2897,  ..., -0.2240,  0.1032, -0.2356],\n",
            "          ...,\n",
            "          [-0.0318, -0.0632,  0.1312,  ..., -0.0508, -0.0087, -0.6949],\n",
            "          [-0.0318, -0.0632,  0.1312,  ..., -0.0508, -0.0087, -0.6949],\n",
            "          [-0.0317, -0.0624,  0.1298,  ..., -0.0511, -0.0107, -0.6950]],\n",
            "\n",
            "         [[ 0.0073,  0.2092, -0.1005,  ..., -0.1981, -0.3670,  0.1144],\n",
            "          [-0.0668, -0.3995, -0.0892,  ..., -0.5984,  0.0437, -0.2681],\n",
            "          [ 0.6149, -0.0739, -0.1715,  ..., -0.7354,  0.0461, -0.4543],\n",
            "          ...,\n",
            "          [-0.0269,  0.1668, -0.1170,  ..., -0.2102, -0.2948,  0.0639],\n",
            "          [-0.0269,  0.1668, -0.1170,  ..., -0.2102, -0.2948,  0.0639],\n",
            "          [-0.0282,  0.1648, -0.1176,  ..., -0.2111, -0.2917,  0.0606]]],\n",
            "\n",
            "\n",
            "        [[[-0.2889, -0.1411,  0.1172,  ..., -0.0849, -0.1177,  0.0910],\n",
            "          [-0.0305,  0.0528, -0.5889,  ..., -0.2042, -0.4317, -0.0097],\n",
            "          [ 0.0951,  0.1878, -0.6867,  ...,  0.0665, -0.2107,  0.1254],\n",
            "          ...,\n",
            "          [-0.2509, -0.1287,  0.1125,  ..., -0.0521, -0.1465,  0.1096],\n",
            "          [-0.2509, -0.1287,  0.1125,  ..., -0.0521, -0.1465,  0.1096],\n",
            "          [-0.2490, -0.1276,  0.1122,  ..., -0.0512, -0.1485,  0.1097]],\n",
            "\n",
            "         [[ 0.1679,  0.1615, -0.1259,  ...,  0.0525, -0.1848, -0.1814],\n",
            "          [-0.5546, -0.2898, -0.4865,  ...,  0.2810, -0.2504, -0.6010],\n",
            "          [ 0.0424,  0.0049, -1.0214,  ...,  0.3659, -0.3304, -0.0172],\n",
            "          ...,\n",
            "          [ 0.1277,  0.1344, -0.0957,  ...,  0.0454, -0.1792, -0.2008],\n",
            "          [ 0.1277,  0.1344, -0.0957,  ...,  0.0454, -0.1792, -0.2008],\n",
            "          [ 0.1266,  0.1334, -0.0942,  ...,  0.0451, -0.1796, -0.2020]],\n",
            "\n",
            "         [[-0.0365, -0.0932,  0.1474,  ..., -0.0596,  0.0223, -0.6994],\n",
            "          [-0.5732, -0.2703,  0.2495,  ...,  0.2634, -0.2910, -0.1168],\n",
            "          [-0.3650, -0.3276, -0.2817,  ..., -0.2232,  0.1455, -0.2368],\n",
            "          ...,\n",
            "          [-0.0300, -0.0623,  0.1235,  ..., -0.0529, -0.0042, -0.6956],\n",
            "          [-0.0300, -0.0623,  0.1235,  ..., -0.0529, -0.0042, -0.6956],\n",
            "          [-0.0298, -0.0613,  0.1220,  ..., -0.0534, -0.0064, -0.6957]],\n",
            "\n",
            "         [[ 0.0084,  0.2050, -0.1004,  ..., -0.2074, -0.3775,  0.1127],\n",
            "          [-0.1133, -0.3574, -0.1103,  ..., -0.6373,  0.0574, -0.2853],\n",
            "          [ 0.5515, -0.0045, -0.1538,  ..., -0.7576, -0.0045, -0.4776],\n",
            "          ...,\n",
            "          [-0.0241,  0.1653, -0.1171,  ..., -0.2149, -0.2963,  0.0574],\n",
            "          [-0.0241,  0.1653, -0.1171,  ..., -0.2149, -0.2963,  0.0574],\n",
            "          [-0.0256,  0.1633, -0.1176,  ..., -0.2157, -0.2927,  0.0539]]],\n",
            "\n",
            "\n",
            "        ...,\n",
            "\n",
            "\n",
            "        [[[-0.2648, -0.1313,  0.1048,  ..., -0.0548, -0.0923,  0.0773],\n",
            "          [ 0.0328,  0.1186, -0.6228,  ..., -0.1620, -0.3498, -0.0505],\n",
            "          [ 0.1925,  0.1963, -0.7464,  ...,  0.1819, -0.2639,  0.2023],\n",
            "          ...,\n",
            "          [-0.2222, -0.1159,  0.0977,  ..., -0.0173, -0.1172,  0.0958],\n",
            "          [-0.2222, -0.1159,  0.0977,  ..., -0.0173, -0.1172,  0.0958],\n",
            "          [-0.2199, -0.1148,  0.0975,  ..., -0.0161, -0.1191,  0.0959]],\n",
            "\n",
            "         [[ 0.1577,  0.1743, -0.1309,  ...,  0.0420, -0.1247, -0.1876],\n",
            "          [-0.7257, -0.2352, -0.5829,  ...,  0.2632, -0.1466, -0.5049],\n",
            "          [-0.0722, -0.0074, -1.1374,  ...,  0.3742, -0.2686,  0.1031],\n",
            "          ...,\n",
            "          [ 0.1115,  0.1471, -0.1080,  ...,  0.0355, -0.1202, -0.1968],\n",
            "          [ 0.1115,  0.1471, -0.1080,  ...,  0.0355, -0.1202, -0.1968],\n",
            "          [ 0.1101,  0.1462, -0.1066,  ...,  0.0350, -0.1202, -0.1976]],\n",
            "\n",
            "         [[-0.0411, -0.0655,  0.1338,  ..., -0.0490,  0.0142, -0.6607],\n",
            "          [-0.5711, -0.2838,  0.1966,  ...,  0.3494, -0.3228,  0.0695],\n",
            "          [-0.3158, -0.2530, -0.3857,  ..., -0.3459,  0.0769, -0.0426],\n",
            "          ...,\n",
            "          [-0.0333, -0.0336,  0.1135,  ..., -0.0446, -0.0159, -0.6523],\n",
            "          [-0.0333, -0.0336,  0.1135,  ..., -0.0446, -0.0159, -0.6523],\n",
            "          [-0.0331, -0.0324,  0.1120,  ..., -0.0451, -0.0184, -0.6522]],\n",
            "\n",
            "         [[ 0.0015,  0.2159, -0.1073,  ..., -0.1931, -0.3366,  0.0831],\n",
            "          [-0.0382, -0.3220, -0.1503,  ..., -0.6380,  0.1502, -0.2478],\n",
            "          [ 0.7198,  0.0624, -0.2394,  ..., -0.8635,  0.0593, -0.4746],\n",
            "          ...,\n",
            "          [-0.0296,  0.1746, -0.1269,  ..., -0.1982, -0.2450,  0.0280],\n",
            "          [-0.0296,  0.1746, -0.1269,  ..., -0.1982, -0.2450,  0.0280],\n",
            "          [-0.0313,  0.1725, -0.1277,  ..., -0.1988, -0.2407,  0.0240]]],\n",
            "\n",
            "\n",
            "        [[[-0.2896, -0.1425,  0.1120,  ..., -0.0842, -0.1109,  0.0849],\n",
            "          [-0.0179,  0.0468, -0.5862,  ..., -0.2036, -0.4658,  0.0202],\n",
            "          [ 0.1179,  0.1629, -0.7339,  ...,  0.0646, -0.2034,  0.1688],\n",
            "          ...,\n",
            "          [-0.2532, -0.1272,  0.1071,  ..., -0.0537, -0.1390,  0.1055],\n",
            "          [-0.2532, -0.1272,  0.1071,  ..., -0.0537, -0.1390,  0.1055],\n",
            "          [-0.2514, -0.1259,  0.1067,  ..., -0.0526, -0.1407,  0.1058]],\n",
            "\n",
            "         [[ 0.1619,  0.1669, -0.1317,  ...,  0.0521, -0.1869, -0.1779],\n",
            "          [-0.4807, -0.3081, -0.2999,  ...,  0.3087, -0.3633, -0.6700],\n",
            "          [ 0.0901, -0.0111, -0.9407,  ...,  0.3764, -0.3605, -0.0460],\n",
            "          ...,\n",
            "          [ 0.1279,  0.1361, -0.0921,  ...,  0.0424, -0.1784, -0.2029],\n",
            "          [ 0.1279,  0.1361, -0.0921,  ...,  0.0424, -0.1784, -0.2029],\n",
            "          [ 0.1270,  0.1348, -0.0899,  ...,  0.0419, -0.1784, -0.2043]],\n",
            "\n",
            "         [[-0.0411, -0.0937,  0.1509,  ..., -0.0553,  0.0162, -0.6992],\n",
            "          [-0.5118, -0.3109,  0.2116,  ...,  0.2343, -0.3009, -0.1800],\n",
            "          [-0.3374, -0.3009, -0.3293,  ..., -0.2232,  0.0834, -0.2880],\n",
            "          ...,\n",
            "          [-0.0354, -0.0628,  0.1230,  ..., -0.0465, -0.0078, -0.6954],\n",
            "          [-0.0354, -0.0628,  0.1230,  ..., -0.0465, -0.0078, -0.6954],\n",
            "          [-0.0353, -0.0615,  0.1213,  ..., -0.0467, -0.0097, -0.6954]],\n",
            "\n",
            "         [[ 0.0084,  0.2081, -0.0949,  ..., -0.1997, -0.3688,  0.1125],\n",
            "          [-0.0885, -0.3643, -0.0680,  ..., -0.6224,  0.1007, -0.2478],\n",
            "          [ 0.6167, -0.0593, -0.0932,  ..., -0.7321,  0.0168, -0.4978],\n",
            "          ...,\n",
            "          [-0.0258,  0.1643, -0.1098,  ..., -0.2110, -0.2947,  0.0573],\n",
            "          [-0.0258,  0.1643, -0.1098,  ..., -0.2110, -0.2947,  0.0573],\n",
            "          [-0.0274,  0.1622, -0.1106,  ..., -0.2119, -0.2912,  0.0537]]],\n",
            "\n",
            "\n",
            "        [[[-0.2927, -0.1490,  0.1207,  ..., -0.0896, -0.1101,  0.0868],\n",
            "          [-0.0115,  0.0313, -0.5254,  ..., -0.2362, -0.4188,  0.0381],\n",
            "          [ 0.1218,  0.1339, -0.6784,  ..., -0.0129, -0.1900,  0.1208],\n",
            "          ...,\n",
            "          [-0.2578, -0.1369,  0.1171,  ..., -0.0626, -0.1361,  0.1060],\n",
            "          [-0.2578, -0.1369,  0.1171,  ..., -0.0626, -0.1361,  0.1060],\n",
            "          [-0.2560, -0.1358,  0.1167,  ..., -0.0617, -0.1379,  0.1063]],\n",
            "\n",
            "         [[ 0.1640,  0.1667, -0.1245,  ...,  0.0524, -0.1909, -0.1761],\n",
            "          [-0.4557, -0.2708, -0.3636,  ...,  0.3017, -0.3358, -0.6475],\n",
            "          [ 0.0807,  0.0252, -0.8996,  ...,  0.3923, -0.3332, -0.0657],\n",
            "          ...,\n",
            "          [ 0.1294,  0.1389, -0.0899,  ...,  0.0443, -0.1840, -0.1985],\n",
            "          [ 0.1294,  0.1389, -0.0899,  ...,  0.0443, -0.1840, -0.1985],\n",
            "          [ 0.1283,  0.1378, -0.0880,  ...,  0.0440, -0.1843, -0.1998]],\n",
            "\n",
            "         [[-0.0359, -0.0924,  0.1515,  ..., -0.0537,  0.0189, -0.7024],\n",
            "          [-0.4767, -0.2911,  0.2218,  ...,  0.2521, -0.2793, -0.2154],\n",
            "          [-0.3232, -0.3287, -0.2860,  ..., -0.2146,  0.0817, -0.3372],\n",
            "          ...,\n",
            "          [-0.0280, -0.0640,  0.1280,  ..., -0.0452, -0.0044, -0.7003],\n",
            "          [-0.0280, -0.0640,  0.1280,  ..., -0.0452, -0.0044, -0.7003],\n",
            "          [-0.0278, -0.0630,  0.1266,  ..., -0.0455, -0.0064, -0.7005]],\n",
            "\n",
            "         [[ 0.0065,  0.2080, -0.0951,  ..., -0.2004, -0.3717,  0.1140],\n",
            "          [-0.0608, -0.3629, -0.0598,  ..., -0.5477,  0.0397, -0.2212],\n",
            "          [ 0.5922, -0.0852, -0.1655,  ..., -0.6964,  0.0152, -0.4268],\n",
            "          ...,\n",
            "          [-0.0233,  0.1668, -0.1098,  ..., -0.2094, -0.3012,  0.0658],\n",
            "          [-0.0233,  0.1668, -0.1098,  ..., -0.2094, -0.3012,  0.0658],\n",
            "          [-0.0247,  0.1647, -0.1103,  ..., -0.2103, -0.2979,  0.0625]]]],\n",
            "       grad_fn=<TransposeBackward0>), tensor([[[[ 6.8801e-01, -1.6015e+00, -2.2098e+00,  ...,  6.3657e-01,\n",
            "            1.9198e-01, -2.5231e+00],\n",
            "          [-9.1364e-01,  2.4846e+00, -3.2122e+00,  ...,  3.0529e-01,\n",
            "           -7.3428e-01, -1.1065e+00],\n",
            "          [ 1.1446e+00,  3.0180e+00,  5.3353e-01,  ..., -2.0362e+00,\n",
            "            3.1774e-01, -1.2319e+00],\n",
            "          ...,\n",
            "          [ 6.3972e-01, -1.5478e+00, -2.2103e+00,  ...,  6.3628e-01,\n",
            "            7.2305e-02, -2.5545e+00],\n",
            "          [ 6.3972e-01, -1.5478e+00, -2.2103e+00,  ...,  6.3628e-01,\n",
            "            7.2305e-02, -2.5545e+00],\n",
            "          [ 6.3660e-01, -1.5443e+00, -2.2142e+00,  ...,  6.3736e-01,\n",
            "            6.7954e-02, -2.5559e+00]],\n",
            "\n",
            "         [[-3.1274e-01, -5.0672e-01,  1.7166e-01,  ...,  2.3498e-01,\n",
            "            2.7472e-01, -4.2383e-02],\n",
            "          [-1.2251e+00, -1.9052e+00, -3.6731e+00,  ...,  2.2070e+00,\n",
            "            2.2957e+00,  3.8726e+00],\n",
            "          [ 2.5868e-01,  3.2685e-01, -3.0624e+00,  ..., -1.6920e+00,\n",
            "            2.1768e-01,  9.9862e-01],\n",
            "          ...,\n",
            "          [-2.8956e-01, -6.3182e-01,  1.2758e-01,  ...,  3.1994e-01,\n",
            "            3.2332e-01,  4.9894e-02],\n",
            "          [-2.8956e-01, -6.3182e-01,  1.2758e-01,  ...,  3.1994e-01,\n",
            "            3.2332e-01,  4.9894e-02],\n",
            "          [-2.8886e-01, -6.3798e-01,  1.2330e-01,  ...,  3.2636e-01,\n",
            "            3.2443e-01,  5.1404e-02]],\n",
            "\n",
            "         [[-5.1771e-01, -1.7759e-01,  6.9728e-01,  ..., -2.9133e-01,\n",
            "           -5.4063e-01, -1.5709e-02],\n",
            "          [ 7.2555e-01, -2.8273e+00, -4.8630e+00,  ...,  2.3051e+00,\n",
            "            5.6596e-01,  1.1475e+00],\n",
            "          [-4.0188e-01,  1.7888e-01, -1.7263e+00,  ...,  3.7291e+00,\n",
            "           -7.9351e-01,  5.7120e-01],\n",
            "          ...,\n",
            "          [-5.1752e-01, -3.1835e-01,  6.0364e-01,  ..., -1.0756e-01,\n",
            "           -5.9010e-01, -1.1191e-01],\n",
            "          [-5.1752e-01, -3.1835e-01,  6.0364e-01,  ..., -1.0756e-01,\n",
            "           -5.9010e-01, -1.1191e-01],\n",
            "          [-5.1413e-01, -3.2616e-01,  6.0022e-01,  ..., -9.9922e-02,\n",
            "           -5.8991e-01, -1.1572e-01]],\n",
            "\n",
            "         [[-8.0394e-01,  7.2716e-03, -2.3720e-01,  ..., -1.4455e-01,\n",
            "           -3.5115e-01,  1.0504e-01],\n",
            "          [-2.4379e+00, -7.0851e-01, -4.9006e+00,  ..., -2.0320e-01,\n",
            "           -1.5688e+00,  6.5046e-01],\n",
            "          [ 1.9495e+00,  9.0743e-01, -3.9209e+00,  ..., -3.4365e+00,\n",
            "           -2.3045e+00,  1.8233e-01],\n",
            "          ...,\n",
            "          [-9.1224e-01,  1.5256e-01, -3.6871e-01,  ..., -2.5752e-01,\n",
            "           -3.4399e-01,  1.0289e-01],\n",
            "          [-9.1224e-01,  1.5256e-01, -3.6871e-01,  ..., -2.5752e-01,\n",
            "           -3.4399e-01,  1.0289e-01],\n",
            "          [-9.1507e-01,  1.5902e-01, -3.7594e-01,  ..., -2.6282e-01,\n",
            "           -3.4264e-01,  1.0008e-01]]],\n",
            "\n",
            "\n",
            "        [[[ 7.0706e-01, -1.6133e+00, -2.2053e+00,  ...,  6.6497e-01,\n",
            "            1.8963e-01, -2.5216e+00],\n",
            "          [-6.1757e-01,  2.4480e+00, -3.0398e+00,  ...,  3.0687e-01,\n",
            "           -9.6206e-01, -8.6471e-01],\n",
            "          [ 1.2608e+00,  2.9423e+00,  3.7942e-01,  ..., -2.0833e+00,\n",
            "            1.4321e-01, -1.0376e+00],\n",
            "          ...,\n",
            "          [ 6.4962e-01, -1.5577e+00, -2.1981e+00,  ...,  6.7089e-01,\n",
            "            7.4392e-02, -2.5606e+00],\n",
            "          [ 6.4962e-01, -1.5577e+00, -2.1981e+00,  ...,  6.7089e-01,\n",
            "            7.4392e-02, -2.5606e+00],\n",
            "          [ 6.4601e-01, -1.5554e+00, -2.2015e+00,  ...,  6.7218e-01,\n",
            "            7.0476e-02, -2.5620e+00]],\n",
            "\n",
            "         [[-3.1998e-01, -4.9799e-01,  2.1173e-01,  ...,  2.1359e-01,\n",
            "            2.9149e-01, -1.9665e-02],\n",
            "          [-1.1078e+00, -2.0075e+00, -3.4867e+00,  ...,  2.1245e+00,\n",
            "            2.3456e+00,  4.1208e+00],\n",
            "          [ 3.5192e-01,  3.2647e-02, -2.6603e+00,  ..., -1.5935e+00,\n",
            "            2.4595e-01,  1.1987e+00],\n",
            "          ...,\n",
            "          [-3.0263e-01, -6.2890e-01,  1.6885e-01,  ...,  2.9340e-01,\n",
            "            3.3711e-01,  7.5588e-02],\n",
            "          [-3.0263e-01, -6.2890e-01,  1.6885e-01,  ...,  2.9340e-01,\n",
            "            3.3711e-01,  7.5588e-02],\n",
            "          [-3.0176e-01, -6.3486e-01,  1.6558e-01,  ...,  2.9853e-01,\n",
            "            3.3817e-01,  7.7416e-02]],\n",
            "\n",
            "         [[-5.4801e-01, -1.5868e-01,  6.7527e-01,  ..., -3.3221e-01,\n",
            "           -5.5835e-01,  2.7187e-03],\n",
            "          [ 3.9360e-01, -2.7669e+00, -4.7568e+00,  ...,  2.1290e+00,\n",
            "            6.5435e-01,  1.3250e+00],\n",
            "          [-6.0158e-01,  3.8039e-01, -1.8921e+00,  ...,  3.2454e+00,\n",
            "           -7.3385e-01,  8.5982e-01],\n",
            "          ...,\n",
            "          [-5.5770e-01, -2.9996e-01,  5.6794e-01,  ..., -1.5365e-01,\n",
            "           -6.1552e-01, -8.6811e-02],\n",
            "          [-5.5770e-01, -2.9996e-01,  5.6794e-01,  ..., -1.5365e-01,\n",
            "           -6.1552e-01, -8.6811e-02],\n",
            "          [-5.5581e-01, -3.0711e-01,  5.6395e-01,  ..., -1.4670e-01,\n",
            "           -6.1533e-01, -8.9507e-02]],\n",
            "\n",
            "         [[-7.9575e-01,  2.2495e-02, -2.3967e-01,  ..., -1.3714e-01,\n",
            "           -3.0855e-01,  1.1488e-01],\n",
            "          [-2.3187e+00, -6.1708e-01, -4.7197e+00,  ..., -3.2384e-02,\n",
            "           -1.4101e+00,  7.6241e-01],\n",
            "          [ 1.9816e+00,  1.3418e+00, -3.3447e+00,  ..., -2.9891e+00,\n",
            "           -2.0735e+00,  1.2133e-01],\n",
            "          ...,\n",
            "          [-9.0119e-01,  1.6219e-01, -3.5393e-01,  ..., -2.4824e-01,\n",
            "           -3.1078e-01,  1.1728e-01],\n",
            "          [-9.0119e-01,  1.6219e-01, -3.5392e-01,  ..., -2.4824e-01,\n",
            "           -3.1078e-01,  1.1727e-01],\n",
            "          [-9.0406e-01,  1.6745e-01, -3.6053e-01,  ..., -2.5257e-01,\n",
            "           -3.0926e-01,  1.1547e-01]]],\n",
            "\n",
            "\n",
            "        [[[ 7.0625e-01, -1.6128e+00, -2.1815e+00,  ...,  6.1959e-01,\n",
            "            1.8581e-01, -2.5305e+00],\n",
            "          [-1.0215e+00,  2.3943e+00, -3.0953e+00,  ...,  2.9450e-01,\n",
            "           -5.7648e-01, -1.0606e+00],\n",
            "          [ 1.0226e+00,  2.6828e+00,  4.7291e-01,  ..., -2.0774e+00,\n",
            "            2.7732e-01, -1.0143e+00],\n",
            "          ...,\n",
            "          [ 6.4907e-01, -1.5729e+00, -2.1963e+00,  ...,  6.1328e-01,\n",
            "            7.6367e-02, -2.5627e+00],\n",
            "          [ 6.4907e-01, -1.5729e+00, -2.1963e+00,  ...,  6.1328e-01,\n",
            "            7.6367e-02, -2.5627e+00],\n",
            "          [ 6.4443e-01, -1.5707e+00, -2.2001e+00,  ...,  6.1425e-01,\n",
            "            7.2789e-02, -2.5651e+00]],\n",
            "\n",
            "         [[-2.9875e-01, -5.0981e-01,  1.7997e-01,  ...,  2.0542e-01,\n",
            "            2.7751e-01, -6.7044e-02],\n",
            "          [-1.2102e+00, -1.9660e+00, -3.8603e+00,  ...,  2.0750e+00,\n",
            "            2.3241e+00,  3.8534e+00],\n",
            "          [ 3.6350e-01,  4.2783e-02, -2.9333e+00,  ..., -1.7251e+00,\n",
            "            1.6747e-01,  9.0362e-01],\n",
            "          ...,\n",
            "          [-2.8176e-01, -6.4399e-01,  1.3209e-01,  ...,  2.8662e-01,\n",
            "            3.2389e-01,  2.7208e-02],\n",
            "          [-2.8176e-01, -6.4399e-01,  1.3209e-01,  ...,  2.8662e-01,\n",
            "            3.2389e-01,  2.7208e-02],\n",
            "          [-2.8120e-01, -6.5050e-01,  1.2757e-01,  ...,  2.9239e-01,\n",
            "            3.2463e-01,  2.8162e-02]],\n",
            "\n",
            "         [[-5.4432e-01, -1.4477e-01,  6.7478e-01,  ..., -2.8226e-01,\n",
            "           -5.1879e-01, -1.7152e-02],\n",
            "          [ 6.1193e-01, -2.8930e+00, -4.9336e+00,  ...,  2.4873e+00,\n",
            "            4.5801e-01,  1.1082e+00],\n",
            "          [-3.8215e-01, -5.9889e-03, -1.7112e+00,  ...,  3.8759e+00,\n",
            "           -7.9876e-01,  7.0875e-01],\n",
            "          ...,\n",
            "          [-5.5437e-01, -2.9562e-01,  5.7813e-01,  ..., -9.8968e-02,\n",
            "           -5.7864e-01, -1.1573e-01],\n",
            "          [-5.5437e-01, -2.9562e-01,  5.7813e-01,  ..., -9.8968e-02,\n",
            "           -5.7864e-01, -1.1573e-01],\n",
            "          [-5.5141e-01, -3.0387e-01,  5.7470e-01,  ..., -9.0710e-02,\n",
            "           -5.7897e-01, -1.1981e-01]],\n",
            "\n",
            "         [[-7.7792e-01,  4.0567e-03, -2.8027e-01,  ..., -1.4416e-01,\n",
            "           -3.5645e-01,  1.2140e-01],\n",
            "          [-2.6872e+00, -7.0686e-01, -4.7257e+00,  ..., -3.7647e-01,\n",
            "           -1.5760e+00,  8.0439e-01],\n",
            "          [ 1.6854e+00,  1.1801e+00, -3.4146e+00,  ..., -3.4911e+00,\n",
            "           -2.5153e+00,  3.0602e-01],\n",
            "          ...,\n",
            "          [-8.8717e-01,  1.4115e-01, -4.0695e-01,  ..., -2.5477e-01,\n",
            "           -3.4123e-01,  1.2529e-01],\n",
            "          [-8.8717e-01,  1.4115e-01, -4.0695e-01,  ..., -2.5477e-01,\n",
            "           -3.4123e-01,  1.2529e-01],\n",
            "          [-8.9075e-01,  1.4578e-01, -4.1355e-01,  ..., -2.6048e-01,\n",
            "           -3.3948e-01,  1.2285e-01]]],\n",
            "\n",
            "\n",
            "        ...,\n",
            "\n",
            "\n",
            "        [[[ 8.0298e-01, -1.3901e+00, -2.2177e+00,  ...,  7.1238e-01,\n",
            "            9.1809e-02, -2.4826e+00],\n",
            "          [-1.3104e+00,  2.7506e+00, -3.5051e+00,  ...,  5.5990e-01,\n",
            "           -6.7890e-01, -1.6876e+00],\n",
            "          [ 5.1154e-01,  2.7752e+00,  4.1188e-01,  ..., -2.2154e+00,\n",
            "            3.6862e-01, -1.4663e+00],\n",
            "          ...,\n",
            "          [ 7.3940e-01, -1.3548e+00, -2.2319e+00,  ...,  6.9934e-01,\n",
            "           -1.5605e-02, -2.5099e+00],\n",
            "          [ 7.3940e-01, -1.3548e+00, -2.2319e+00,  ...,  6.9934e-01,\n",
            "           -1.5605e-02, -2.5099e+00],\n",
            "          [ 7.3530e-01, -1.3522e+00, -2.2359e+00,  ...,  7.0061e-01,\n",
            "           -1.9226e-02, -2.5130e+00]],\n",
            "\n",
            "         [[-3.5390e-01, -5.8135e-01,  8.9637e-02,  ...,  3.0754e-01,\n",
            "            2.4013e-01, -1.2332e-02],\n",
            "          [-1.6577e+00, -1.8226e+00, -4.5972e+00,  ...,  2.1679e+00,\n",
            "            2.6500e+00,  4.0466e+00],\n",
            "          [ 1.9424e-01, -1.9700e-01, -3.6482e+00,  ..., -1.3370e+00,\n",
            "            1.8546e-01,  1.2029e+00],\n",
            "          ...,\n",
            "          [-3.3119e-01, -7.2338e-01,  1.4202e-02,  ...,  3.7803e-01,\n",
            "            3.0172e-01,  7.6379e-02],\n",
            "          [-3.3119e-01, -7.2339e-01,  1.4202e-02,  ...,  3.7803e-01,\n",
            "            3.0172e-01,  7.6379e-02],\n",
            "          [-3.2945e-01, -7.3099e-01,  8.3955e-03,  ...,  3.8472e-01,\n",
            "            3.0351e-01,  7.6205e-02]],\n",
            "\n",
            "         [[-3.9972e-01, -1.7266e-01,  7.1116e-01,  ..., -3.1617e-01,\n",
            "           -5.5701e-01, -1.1882e-01],\n",
            "          [ 7.3076e-01, -2.7893e+00, -5.1262e+00,  ...,  2.6295e+00,\n",
            "            4.7624e-01,  1.0501e+00],\n",
            "          [-1.5058e-01, -3.1701e-01, -2.3147e+00,  ...,  3.1904e+00,\n",
            "           -1.0191e+00,  6.4292e-01],\n",
            "          ...,\n",
            "          [-3.9758e-01, -3.1956e-01,  6.1225e-01,  ..., -1.4045e-01,\n",
            "           -6.2900e-01, -2.3013e-01],\n",
            "          [-3.9758e-01, -3.1956e-01,  6.1225e-01,  ..., -1.4045e-01,\n",
            "           -6.2900e-01, -2.3013e-01],\n",
            "          [-3.9280e-01, -3.2888e-01,  6.0855e-01,  ..., -1.3233e-01,\n",
            "           -6.3045e-01, -2.3636e-01]],\n",
            "\n",
            "         [[-7.9638e-01,  2.0870e-01, -2.1725e-01,  ..., -1.6905e-01,\n",
            "           -3.6606e-01, -4.0081e-02],\n",
            "          [-2.9439e+00, -6.3340e-02, -4.9991e+00,  ..., -1.1324e+00,\n",
            "           -1.7880e+00,  3.4723e-01],\n",
            "          [ 1.4440e+00,  1.7675e+00, -4.0480e+00,  ..., -3.9661e+00,\n",
            "           -2.4218e+00,  5.5883e-02],\n",
            "          ...,\n",
            "          [-9.0654e-01,  3.4971e-01, -3.5943e-01,  ..., -2.8996e-01,\n",
            "           -3.6493e-01, -5.5987e-02],\n",
            "          [-9.0654e-01,  3.4971e-01, -3.5943e-01,  ..., -2.8996e-01,\n",
            "           -3.6493e-01, -5.5987e-02],\n",
            "          [-9.1003e-01,  3.5652e-01, -3.6736e-01,  ..., -2.9707e-01,\n",
            "           -3.6360e-01, -5.9446e-02]]],\n",
            "\n",
            "\n",
            "        [[[ 7.1242e-01, -1.6130e+00, -2.1738e+00,  ...,  6.5314e-01,\n",
            "            1.8931e-01, -2.4964e+00],\n",
            "          [-6.5601e-01,  2.4509e+00, -3.0657e+00,  ...,  1.6584e-01,\n",
            "           -9.5204e-01, -1.0087e+00],\n",
            "          [ 1.1880e+00,  3.0371e+00,  6.1784e-01,  ..., -2.0305e+00,\n",
            "            3.0658e-01, -1.1394e+00],\n",
            "          ...,\n",
            "          [ 6.5462e-01, -1.5580e+00, -2.1691e+00,  ...,  6.6362e-01,\n",
            "            6.3800e-02, -2.5325e+00],\n",
            "          [ 6.5462e-01, -1.5580e+00, -2.1691e+00,  ...,  6.6362e-01,\n",
            "            6.3800e-02, -2.5325e+00],\n",
            "          [ 6.5107e-01, -1.5545e+00, -2.1724e+00,  ...,  6.6491e-01,\n",
            "            5.9043e-02, -2.5347e+00]],\n",
            "\n",
            "         [[-3.3819e-01, -4.7951e-01,  1.9726e-01,  ...,  2.2581e-01,\n",
            "            2.7101e-01, -4.4893e-02],\n",
            "          [-8.0682e-01, -1.8640e+00, -3.3645e+00,  ...,  2.1949e+00,\n",
            "            2.3177e+00,  4.1433e+00],\n",
            "          [ 3.1853e-01,  2.2016e-01, -2.5575e+00,  ..., -1.4965e+00,\n",
            "            3.6152e-02,  1.0319e+00],\n",
            "          ...,\n",
            "          [-3.0496e-01, -6.1761e-01,  1.6779e-01,  ...,  3.1426e-01,\n",
            "            3.1526e-01,  4.7965e-02],\n",
            "          [-3.0496e-01, -6.1761e-01,  1.6779e-01,  ...,  3.1426e-01,\n",
            "            3.1526e-01,  4.7964e-02],\n",
            "          [-3.0384e-01, -6.2437e-01,  1.6434e-01,  ...,  3.2047e-01,\n",
            "            3.1637e-01,  4.9625e-02]],\n",
            "\n",
            "         [[-5.3046e-01, -1.7766e-01,  6.8641e-01,  ..., -3.1534e-01,\n",
            "           -5.3501e-01, -2.6950e-02],\n",
            "          [ 7.9040e-01, -2.9005e+00, -4.9679e+00,  ...,  2.2082e+00,\n",
            "            5.2285e-01,  1.4926e+00],\n",
            "          [-6.9561e-01,  2.9801e-01, -1.7553e+00,  ...,  3.6547e+00,\n",
            "           -7.4168e-01,  6.1780e-01],\n",
            "          ...,\n",
            "          [-5.3075e-01, -3.2387e-01,  5.7414e-01,  ..., -1.1038e-01,\n",
            "           -5.8178e-01, -1.1078e-01],\n",
            "          [-5.3075e-01, -3.2387e-01,  5.7414e-01,  ..., -1.1038e-01,\n",
            "           -5.8178e-01, -1.1078e-01],\n",
            "          [-5.2733e-01, -3.3173e-01,  5.7029e-01,  ..., -1.0156e-01,\n",
            "           -5.8100e-01, -1.1404e-01]],\n",
            "\n",
            "         [[-7.8790e-01,  8.0713e-03, -2.4026e-01,  ..., -1.2653e-01,\n",
            "           -2.9430e-01,  1.1058e-01],\n",
            "          [-2.2476e+00, -9.4329e-01, -4.3936e+00,  ..., -2.1800e-02,\n",
            "           -1.5686e+00,  7.4759e-01],\n",
            "          [ 1.9375e+00,  8.5425e-01, -3.9678e+00,  ..., -3.4738e+00,\n",
            "           -2.1444e+00,  4.9815e-01],\n",
            "          ...,\n",
            "          [-9.0183e-01,  1.5800e-01, -3.6116e-01,  ..., -2.4109e-01,\n",
            "           -2.9518e-01,  1.1004e-01],\n",
            "          [-9.0183e-01,  1.5800e-01, -3.6116e-01,  ..., -2.4109e-01,\n",
            "           -2.9518e-01,  1.1004e-01],\n",
            "          [-9.0514e-01,  1.6374e-01, -3.6823e-01,  ..., -2.4736e-01,\n",
            "           -2.9385e-01,  1.0761e-01]]],\n",
            "\n",
            "\n",
            "        [[[ 7.0085e-01, -1.6288e+00, -2.1854e+00,  ...,  6.4546e-01,\n",
            "            1.7764e-01, -2.5277e+00],\n",
            "          [-8.0616e-01,  2.4775e+00, -3.2097e+00,  ...,  2.3212e-01,\n",
            "           -8.3477e-01, -1.1300e+00],\n",
            "          [ 1.2683e+00,  2.7361e+00,  4.6457e-01,  ..., -2.1358e+00,\n",
            "            1.2742e-01, -1.0305e+00],\n",
            "          ...,\n",
            "          [ 6.4699e-01, -1.5759e+00, -2.1879e+00,  ...,  6.4706e-01,\n",
            "            6.5064e-02, -2.5561e+00],\n",
            "          [ 6.4698e-01, -1.5759e+00, -2.1879e+00,  ...,  6.4706e-01,\n",
            "            6.5065e-02, -2.5561e+00],\n",
            "          [ 6.4291e-01, -1.5730e+00, -2.1917e+00,  ...,  6.4799e-01,\n",
            "            6.1015e-02, -2.5578e+00]],\n",
            "\n",
            "         [[-3.1533e-01, -4.9246e-01,  1.8465e-01,  ...,  2.1337e-01,\n",
            "            2.7297e-01, -3.5379e-02],\n",
            "          [-1.0906e+00, -1.9928e+00, -3.3230e+00,  ...,  2.1881e+00,\n",
            "            2.2034e+00,  3.9757e+00],\n",
            "          [ 3.7579e-01,  1.6632e-01, -2.7315e+00,  ..., -1.5429e+00,\n",
            "            1.6173e-01,  1.0380e+00],\n",
            "          ...,\n",
            "          [-2.9032e-01, -6.2244e-01,  1.6271e-01,  ...,  3.0118e-01,\n",
            "            3.1557e-01,  6.3630e-02],\n",
            "          [-2.9032e-01, -6.2244e-01,  1.6271e-01,  ...,  3.0118e-01,\n",
            "            3.1557e-01,  6.3629e-02],\n",
            "          [-2.8907e-01, -6.2850e-01,  1.5954e-01,  ...,  3.0706e-01,\n",
            "            3.1642e-01,  6.5669e-02]],\n",
            "\n",
            "         [[-5.3383e-01, -1.8245e-01,  6.9105e-01,  ..., -3.0361e-01,\n",
            "           -5.2603e-01, -2.1302e-02],\n",
            "          [ 4.9478e-01, -2.7855e+00, -4.7103e+00,  ...,  2.1910e+00,\n",
            "            5.8680e-01,  1.3830e+00],\n",
            "          [-8.3443e-01,  3.0211e-01, -1.6097e+00,  ...,  3.7126e+00,\n",
            "           -4.6474e-01,  8.7533e-01],\n",
            "          ...,\n",
            "          [-5.4512e-01, -3.2048e-01,  5.9095e-01,  ..., -1.1952e-01,\n",
            "           -5.7026e-01, -9.6071e-02],\n",
            "          [-5.4512e-01, -3.2048e-01,  5.9095e-01,  ..., -1.1952e-01,\n",
            "           -5.7026e-01, -9.6071e-02],\n",
            "          [-5.4272e-01, -3.2831e-01,  5.8704e-01,  ..., -1.1144e-01,\n",
            "           -5.6947e-01, -9.8674e-02]],\n",
            "\n",
            "         [[-8.1172e-01,  8.7327e-03, -2.3972e-01,  ..., -1.2954e-01,\n",
            "           -3.3254e-01,  1.3690e-01],\n",
            "          [-2.4183e+00, -9.3177e-01, -4.6309e+00,  ..., -1.8800e-02,\n",
            "           -1.5315e+00,  7.4390e-01],\n",
            "          [ 1.9594e+00,  1.1291e+00, -3.3792e+00,  ..., -3.1914e+00,\n",
            "           -2.1466e+00,  6.9302e-01],\n",
            "          ...,\n",
            "          [-9.1880e-01,  1.4994e-01, -3.5052e-01,  ..., -2.3068e-01,\n",
            "           -3.3340e-01,  1.4296e-01],\n",
            "          [-9.1880e-01,  1.4994e-01, -3.5052e-01,  ..., -2.3068e-01,\n",
            "           -3.3340e-01,  1.4296e-01],\n",
            "          [-9.2229e-01,  1.5536e-01, -3.5680e-01,  ..., -2.3579e-01,\n",
            "           -3.3208e-01,  1.4112e-01]]]], grad_fn=<TransposeBackward0>), tensor([[[[-5.4771e+00,  2.1474e+00,  1.7942e+01,  ..., -6.1210e+00,\n",
            "            1.4380e+01, -1.3710e+01],\n",
            "          [-1.1145e+01,  1.4189e-01,  1.2607e+01,  ..., -5.8447e+00,\n",
            "           -2.1934e+00, -1.3669e+01],\n",
            "          [-1.3811e+01, -2.3822e-01,  8.0174e+00,  ...,  4.8098e+00,\n",
            "            8.7729e+00,  3.5513e-01],\n",
            "          ...,\n",
            "          [ 7.9545e+00,  4.8636e-01,  9.9571e+00,  ..., -1.8564e+01,\n",
            "            1.2052e+01, -1.7360e+00],\n",
            "          [-3.5352e+00,  2.1371e+00, -5.8829e+00,  ...,  5.3562e+00,\n",
            "           -5.1087e+00, -6.5436e-01],\n",
            "          [ 2.9903e+00, -1.4428e+00, -1.3024e+01,  ...,  6.1241e+00,\n",
            "            6.3326e+00,  5.5260e+00]],\n",
            "\n",
            "         [[-1.0138e+01, -1.9969e+00, -1.7598e+01,  ...,  9.7458e+00,\n",
            "            8.8230e+00,  1.0002e+00],\n",
            "          [ 8.0852e-02,  1.0972e+01, -1.3571e+01,  ...,  7.8779e+00,\n",
            "            6.0988e+00, -1.1435e+01],\n",
            "          [-3.5558e+00,  1.0153e+01, -1.4885e+01,  ...,  1.2246e+01,\n",
            "            1.6466e+01, -2.5008e+00],\n",
            "          ...,\n",
            "          [-3.5063e+00, -5.5781e-02, -8.5928e+00,  ...,  7.8461e+00,\n",
            "            9.2016e+00, -3.0419e+00],\n",
            "          [ 3.9180e+00,  4.1832e+00, -1.5637e+01,  ...,  2.0189e+01,\n",
            "           -1.2389e+00,  2.5129e+00],\n",
            "          [ 4.2960e+00,  2.2048e+00, -2.9113e+00,  ..., -3.8899e+00,\n",
            "            1.0489e+00,  9.9784e+00]],\n",
            "\n",
            "         [[-1.0981e+01, -7.3188e+00,  3.1438e+00,  ..., -3.4261e+00,\n",
            "           -3.9455e+00, -4.7516e+00],\n",
            "          [-1.2293e+01,  1.0973e+01,  1.1465e-01,  ..., -1.2530e+01,\n",
            "            1.4196e+01, -8.1148e+00],\n",
            "          [ 7.3605e+00,  1.3346e+00,  3.6127e+00,  ..., -8.5965e+00,\n",
            "           -3.3787e+00, -6.3915e+00],\n",
            "          ...,\n",
            "          [-3.3070e+00,  1.6554e-01,  2.4780e+00,  ...,  8.7275e+00,\n",
            "           -1.1398e+01,  9.9963e+00],\n",
            "          [-7.4229e+00,  5.1970e+00,  6.6822e+00,  ...,  7.4617e+00,\n",
            "           -1.2728e+01, -3.5436e+00],\n",
            "          [ 2.3126e+00,  6.5107e-01,  2.0486e+00,  ..., -4.4524e+00,\n",
            "           -2.7732e+00, -2.0169e+00]],\n",
            "\n",
            "         [[ 3.5090e+00,  5.0877e+00, -4.5318e+00,  ..., -2.7274e+01,\n",
            "            6.0475e+00, -2.5245e+00],\n",
            "          [ 8.6157e+00,  4.6657e+00, -1.2791e+00,  ..., -3.1270e+01,\n",
            "           -4.1644e+00,  1.0818e+01],\n",
            "          [ 6.5629e+00, -3.2923e+00, -2.3754e+00,  ..., -3.1423e+01,\n",
            "            1.7632e+00,  4.0684e-03],\n",
            "          ...,\n",
            "          [ 1.1752e+01,  4.4037e+00, -1.3146e+01,  ..., -2.8991e+01,\n",
            "           -1.1536e+01, -1.6815e+00],\n",
            "          [-2.2986e+00,  5.5907e+00, -3.2853e+00,  ..., -2.6705e+01,\n",
            "           -9.2000e+00,  8.6869e+00],\n",
            "          [-1.0091e+01, -2.5396e+00,  1.2844e+01,  ...,  2.7345e+01,\n",
            "            9.4178e+00, -8.9534e+00]]],\n",
            "\n",
            "\n",
            "        [[[-1.5631e+00, -4.3171e-01,  9.9437e+00,  ..., -9.7106e-01,\n",
            "            2.7829e+00, -1.3077e+01],\n",
            "          [-4.9472e+00,  5.0982e+00,  1.2221e+01,  ...,  5.2447e+00,\n",
            "           -1.6019e+00, -1.5525e+01],\n",
            "          [-1.7761e+01,  7.4636e-01, -1.3706e+00,  ..., -4.6938e+00,\n",
            "           -4.4322e-01, -1.2476e+01],\n",
            "          ...,\n",
            "          [-1.3959e+01, -2.5982e+00,  2.3960e+00,  ...,  6.3240e+00,\n",
            "           -7.8276e+00, -6.7656e+00],\n",
            "          [-1.1299e+01,  2.3784e+00,  5.5889e+00,  ...,  4.7281e+00,\n",
            "           -7.7928e+00,  5.3020e+00],\n",
            "          [ 2.9691e+00, -1.1320e+00, -1.2811e+01,  ...,  6.0802e+00,\n",
            "            6.0715e+00,  5.8209e+00]],\n",
            "\n",
            "         [[-3.2746e+00, -7.7141e+00, -1.1544e+01,  ...,  1.3098e+01,\n",
            "            1.4532e+01, -1.5160e+01],\n",
            "          [ 1.1095e+00,  1.1265e+01, -1.5558e+01,  ...,  5.8856e+00,\n",
            "            1.5114e+01, -5.3036e+00],\n",
            "          [-1.4834e+00,  8.7896e+00, -1.5772e+01,  ...,  1.0646e+00,\n",
            "            2.0597e+01,  2.6401e+00],\n",
            "          ...,\n",
            "          [-6.6620e+00,  5.9682e+00, -8.4807e-01,  ...,  1.3215e+01,\n",
            "           -9.3541e-01, -1.0931e+01],\n",
            "          [ 1.5268e+00,  6.0039e+00, -1.3883e+01,  ..., -2.7288e-01,\n",
            "            5.2266e+00,  5.3163e+00],\n",
            "          [ 4.3317e+00,  2.2774e+00, -2.4105e+00,  ..., -4.1539e+00,\n",
            "            1.1841e+00,  9.8650e+00]],\n",
            "\n",
            "         [[ 1.4719e+01,  5.0614e+00,  1.4464e+01,  ...,  1.6976e+01,\n",
            "           -1.4333e+01,  6.1380e+00],\n",
            "          [ 5.1398e+00, -1.5680e+00,  1.6182e+01,  ...,  8.7616e+00,\n",
            "           -1.6300e+01, -4.6895e+00],\n",
            "          [ 4.7177e-01,  4.5603e+00,  1.6089e+01,  ...,  1.1431e+01,\n",
            "            5.4656e+00, -9.8714e+00],\n",
            "          ...,\n",
            "          [ 4.2871e+00,  1.4696e+01,  6.1668e+00,  ...,  1.1693e+01,\n",
            "           -1.3238e+01, -2.0783e+00],\n",
            "          [ 1.3583e+01, -5.1014e+00, -7.4819e-01,  ...,  6.2591e+00,\n",
            "           -6.0252e+00,  6.2862e+00],\n",
            "          [ 2.2538e+00,  7.2107e-01,  1.6515e+00,  ..., -4.2582e+00,\n",
            "           -2.6054e+00, -2.0217e+00]],\n",
            "\n",
            "         [[ 1.0049e+01,  1.2094e+01,  6.1902e+00,  ..., -4.5016e+01,\n",
            "           -1.2491e+01,  1.0105e+01],\n",
            "          [ 9.4624e+00,  4.9760e+00,  1.2811e+00,  ..., -3.9111e+01,\n",
            "           -2.7563e+00,  5.6650e+00],\n",
            "          [ 1.0835e+01,  3.7475e+00,  1.0193e+01,  ..., -3.1090e+01,\n",
            "            8.2571e+00,  4.2302e+00],\n",
            "          ...,\n",
            "          [ 1.1764e+01,  6.7435e-01, -6.8212e+00,  ..., -2.6506e+01,\n",
            "           -1.4578e+01, -4.9873e+00],\n",
            "          [ 1.1701e+01,  6.6586e+00,  1.9508e+00,  ..., -2.2807e+01,\n",
            "           -6.7201e+00, -8.7537e+00],\n",
            "          [-1.0178e+01, -2.2733e+00,  1.3501e+01,  ...,  2.7578e+01,\n",
            "            9.3702e+00, -8.8386e+00]]],\n",
            "\n",
            "\n",
            "        [[[-1.4449e+01, -1.7402e+00,  1.2449e+01,  ...,  2.0568e+00,\n",
            "            5.9743e+00, -9.0200e+00],\n",
            "          [-7.5736e+00,  6.9172e+00,  7.2240e+00,  ...,  5.8874e-01,\n",
            "           -1.6381e+00, -1.3194e+01],\n",
            "          [-1.3412e+01,  4.5989e+00,  1.0215e+01,  ...,  1.7763e+00,\n",
            "            2.1812e+00, -1.8133e+01],\n",
            "          ...,\n",
            "          [-1.1451e+01, -6.3266e+00,  7.4570e+00,  ...,  6.2612e+00,\n",
            "            3.2673e+00, -1.2152e+01],\n",
            "          [-1.1451e+01, -6.3266e+00,  7.4570e+00,  ...,  6.2612e+00,\n",
            "            3.2673e+00, -1.2152e+01],\n",
            "          [-1.1451e+01, -6.3266e+00,  7.4570e+00,  ...,  6.2612e+00,\n",
            "            3.2673e+00, -1.2152e+01]],\n",
            "\n",
            "         [[-2.7766e+00,  7.1483e-01, -1.0951e+01,  ...,  6.3796e+00,\n",
            "            5.3248e+00, -1.3199e+01],\n",
            "          [-1.2940e+00, -2.5272e+00, -1.0988e+01,  ..., -1.6810e+00,\n",
            "            5.7300e+00, -1.3142e+01],\n",
            "          [ 2.7382e+00, -3.9989e+00, -1.3436e+01,  ...,  4.3503e+00,\n",
            "            1.0645e+01, -1.2793e+01],\n",
            "          ...,\n",
            "          [-4.3708e+00,  2.3946e+00, -9.6737e+00,  ...,  9.5962e+00,\n",
            "            9.0474e+00, -1.0657e+01],\n",
            "          [-4.3708e+00,  2.3946e+00, -9.6737e+00,  ...,  9.5962e+00,\n",
            "            9.0474e+00, -1.0657e+01],\n",
            "          [-4.3708e+00,  2.3946e+00, -9.6737e+00,  ...,  9.5962e+00,\n",
            "            9.0474e+00, -1.0657e+01]],\n",
            "\n",
            "         [[ 4.2095e+00, -8.8042e+00,  1.7414e+00,  ..., -5.7899e+00,\n",
            "           -1.2362e+01,  1.6323e+00],\n",
            "          [ 9.1917e+00,  4.3311e+00,  9.9380e+00,  ..., -5.1719e+00,\n",
            "           -6.0122e+00, -1.4008e+01],\n",
            "          [ 2.6914e+00,  4.6217e+00,  9.6047e+00,  ..., -1.4697e+00,\n",
            "           -3.6253e+00, -1.6171e+00],\n",
            "          ...,\n",
            "          [-2.6574e+00,  1.1449e+00,  1.5142e-01,  ...,  1.0358e+01,\n",
            "            5.7487e+00,  5.9409e+00],\n",
            "          [-2.6574e+00,  1.1449e+00,  1.5142e-01,  ...,  1.0358e+01,\n",
            "            5.7487e+00,  5.9409e+00],\n",
            "          [-2.6574e+00,  1.1449e+00,  1.5142e-01,  ...,  1.0358e+01,\n",
            "            5.7487e+00,  5.9409e+00]],\n",
            "\n",
            "         [[ 1.2996e+01,  2.7904e+00, -1.6358e+00,  ..., -3.3500e+01,\n",
            "           -1.8342e+01, -7.9742e+00],\n",
            "          [ 5.3212e+00,  9.2770e+00,  7.3858e+00,  ..., -1.7467e+01,\n",
            "           -1.3021e+01, -1.1263e+01],\n",
            "          [ 9.5181e+00,  1.1994e+01,  9.4808e+00,  ..., -2.8126e+01,\n",
            "           -1.7405e+01, -6.0564e+00],\n",
            "          ...,\n",
            "          [ 1.5821e+01,  5.0398e+00, -1.7825e+01,  ..., -2.4683e+01,\n",
            "           -4.8935e+00,  1.1321e+01],\n",
            "          [ 1.5821e+01,  5.0398e+00, -1.7825e+01,  ..., -2.4683e+01,\n",
            "           -4.8935e+00,  1.1321e+01],\n",
            "          [ 1.5821e+01,  5.0398e+00, -1.7825e+01,  ..., -2.4683e+01,\n",
            "           -4.8935e+00,  1.1321e+01]]],\n",
            "\n",
            "\n",
            "        ...,\n",
            "\n",
            "\n",
            "        [[[ 1.9911e+00,  1.0635e+01,  1.2073e+01,  ..., -1.9655e+01,\n",
            "           -1.4938e+00, -1.4062e+01],\n",
            "          [-1.1634e+01,  1.1610e+01, -4.6295e+00,  ..., -1.4895e+00,\n",
            "            1.4567e-01, -1.7482e+01],\n",
            "          [-5.9822e+00,  1.0597e+01,  1.4626e+01,  ..., -7.6396e+00,\n",
            "           -1.9780e+00, -1.6333e+01],\n",
            "          ...,\n",
            "          [-1.1555e+01, -5.8836e+00,  6.0266e+00,  ...,  2.8450e+00,\n",
            "            4.4649e+00, -1.1745e+01],\n",
            "          [-1.1555e+01, -5.8836e+00,  6.0266e+00,  ...,  2.8450e+00,\n",
            "            4.4649e+00, -1.1745e+01],\n",
            "          [-1.1555e+01, -5.8836e+00,  6.0266e+00,  ...,  2.8450e+00,\n",
            "            4.4649e+00, -1.1745e+01]],\n",
            "\n",
            "         [[-3.5661e+00,  3.0201e+00, -1.2222e+01,  ...,  1.9091e+01,\n",
            "            4.6488e+00,  2.1647e+00],\n",
            "          [-6.0640e+00,  3.3302e+00, -1.1144e+01,  ...,  8.0765e+00,\n",
            "            7.0178e+00, -3.8437e+00],\n",
            "          [-4.5659e+00, -5.5649e+00, -3.1115e+00,  ...,  9.8258e+00,\n",
            "            1.7687e+01,  5.7486e+00],\n",
            "          ...,\n",
            "          [-1.5223e+00,  3.1176e+00, -8.1982e+00,  ...,  7.6655e+00,\n",
            "            9.0052e+00, -1.1939e+01],\n",
            "          [-1.5223e+00,  3.1176e+00, -8.1982e+00,  ...,  7.6655e+00,\n",
            "            9.0052e+00, -1.1939e+01],\n",
            "          [-1.5223e+00,  3.1176e+00, -8.1982e+00,  ...,  7.6655e+00,\n",
            "            9.0052e+00, -1.1939e+01]],\n",
            "\n",
            "         [[-1.8181e+00, -1.0391e+01,  2.5061e+00,  ...,  1.2039e+01,\n",
            "           -1.7666e+00,  4.6848e+00],\n",
            "          [-5.6924e-01,  2.9814e-01,  4.0339e+00,  ...,  1.6338e+00,\n",
            "            3.1628e+00, -4.5675e+00],\n",
            "          [-1.0139e+01,  7.6964e+00,  1.0880e+01,  ...,  1.8545e+00,\n",
            "            1.3698e+00,  4.3452e-01],\n",
            "          ...,\n",
            "          [-8.0187e-02,  3.2152e+00, -2.4302e-01,  ...,  9.9321e+00,\n",
            "            8.8265e+00,  9.0523e+00],\n",
            "          [-8.0187e-02,  3.2152e+00, -2.4302e-01,  ...,  9.9321e+00,\n",
            "            8.8265e+00,  9.0523e+00],\n",
            "          [-8.0187e-02,  3.2152e+00, -2.4302e-01,  ...,  9.9321e+00,\n",
            "            8.8265e+00,  9.0523e+00]],\n",
            "\n",
            "         [[ 6.2050e+00, -1.4851e-01,  2.0282e+00,  ..., -4.0965e+01,\n",
            "           -1.3997e+01, -5.6594e+00],\n",
            "          [ 2.9099e+00,  1.1301e+01,  5.7404e+00,  ..., -2.7293e+01,\n",
            "           -6.4011e+00, -2.1477e+00],\n",
            "          [ 3.7944e+00, -2.7123e+00, -9.6815e+00,  ..., -3.1228e+01,\n",
            "           -2.0185e-01, -6.2541e+00],\n",
            "          ...,\n",
            "          [ 1.4299e+01,  3.5628e+00, -1.7142e+01,  ..., -2.1745e+01,\n",
            "           -2.6068e+00,  1.0111e+01],\n",
            "          [ 1.4299e+01,  3.5628e+00, -1.7142e+01,  ..., -2.1745e+01,\n",
            "           -2.6068e+00,  1.0111e+01],\n",
            "          [ 1.4299e+01,  3.5628e+00, -1.7142e+01,  ..., -2.1745e+01,\n",
            "           -2.6068e+00,  1.0111e+01]]],\n",
            "\n",
            "\n",
            "        [[[-9.9425e+00, -1.7572e+00,  2.0044e+01,  ..., -6.2718e+00,\n",
            "            9.7952e+00,  7.0375e+00],\n",
            "          [-1.0840e+01,  6.9062e+00,  1.1971e+01,  ..., -5.6800e+00,\n",
            "            5.9009e+00,  7.0278e+00],\n",
            "          [-1.7435e+01,  1.1353e+00,  1.3402e+01,  ..., -1.6039e+00,\n",
            "            2.9404e+00, -3.3613e+00],\n",
            "          ...,\n",
            "          [-1.8149e+01, -1.0310e+01,  6.7303e+00,  ...,  3.4574e+00,\n",
            "            6.3356e-01, -2.8292e+00],\n",
            "          [-1.1328e+00,  3.3258e+00, -4.7260e+00,  ..., -6.8494e+00,\n",
            "            1.0284e+01,  6.1911e+00],\n",
            "          [ 3.1581e+00, -1.1005e+00, -1.2948e+01,  ...,  6.3123e+00,\n",
            "            6.1227e+00,  5.5723e+00]],\n",
            "\n",
            "         [[-1.3104e+00, -7.0750e+00, -2.1886e+01,  ...,  4.7852e+00,\n",
            "            1.4985e+01, -9.6393e-01],\n",
            "          [ 4.2575e+00,  1.3207e-02, -2.0479e+01,  ...,  1.7063e+01,\n",
            "            1.8746e+00, -5.3458e+00],\n",
            "          [-2.5049e+00,  1.4372e-01, -2.2032e+01,  ...,  9.1536e+00,\n",
            "            1.3019e+01, -4.3226e+00],\n",
            "          ...,\n",
            "          [ 1.3550e+00,  3.0679e+00, -1.8520e+01,  ...,  1.8675e+01,\n",
            "           -1.0213e+01, -1.3967e+01],\n",
            "          [ 4.3378e+00,  2.8706e+00, -1.0586e+01,  ...,  1.0838e+01,\n",
            "           -1.0260e+01,  3.8443e+00],\n",
            "          [ 4.2485e+00,  2.1322e+00, -2.6987e+00,  ..., -3.6450e+00,\n",
            "            9.7289e-01,  9.9296e+00]],\n",
            "\n",
            "         [[ 4.7680e+00, -2.5952e+00,  9.0157e+00,  ..., -6.6448e+00,\n",
            "            5.1015e+00,  5.2801e+00],\n",
            "          [-1.1966e+00, -7.1765e-01,  1.7012e+01,  ...,  2.9988e+00,\n",
            "            9.9712e+00,  7.4080e+00],\n",
            "          [ 8.7162e+00,  9.3250e+00, -1.4878e+00,  ...,  1.0470e+01,\n",
            "            8.7309e+00, -6.9322e+00],\n",
            "          ...,\n",
            "          [ 1.8497e+00,  9.2548e-01,  5.5544e+00,  ...,  8.3855e+00,\n",
            "           -1.5515e+01, -9.6327e+00],\n",
            "          [ 1.6790e+01, -7.2096e-01,  1.0097e+01,  ...,  8.6498e+00,\n",
            "           -1.0204e+01, -6.2141e+00],\n",
            "          [ 2.3696e+00,  3.4741e-01,  2.2509e+00,  ..., -4.3153e+00,\n",
            "           -3.1466e+00, -2.1379e+00]],\n",
            "\n",
            "         [[ 6.3930e+00,  2.8601e+00, -1.9436e+00,  ..., -2.7192e+01,\n",
            "           -1.5099e+01, -7.8452e+00],\n",
            "          [ 4.5900e+00, -7.9718e-01,  3.0977e+00,  ..., -3.1726e+01,\n",
            "           -5.7540e+00, -7.8497e+00],\n",
            "          [ 1.2932e+01,  1.3343e+00,  4.2026e+00,  ..., -2.7205e+01,\n",
            "           -1.7704e+00, -2.4787e+00],\n",
            "          ...,\n",
            "          [-7.6653e+00,  8.7450e+00, -8.3942e+00,  ..., -3.0996e+01,\n",
            "           -7.1771e+00, -1.3750e+00],\n",
            "          [ 1.8514e+00,  5.3872e+00, -2.1368e+00,  ..., -2.4043e+01,\n",
            "           -3.7830e+00, -2.1899e-01],\n",
            "          [-1.0063e+01, -2.2750e+00,  1.2930e+01,  ...,  2.7251e+01,\n",
            "            9.3594e+00, -8.9081e+00]]],\n",
            "\n",
            "\n",
            "        [[[-5.3188e-01,  1.9617e+01,  2.0309e+01,  ..., -5.3261e+00,\n",
            "            6.9111e+00, -7.1849e+00],\n",
            "          [-1.9578e+01, -2.7045e+00,  1.1728e+01,  ...,  1.4227e+01,\n",
            "           -1.6533e+00,  1.3282e+00],\n",
            "          [-1.6616e+01,  1.5767e+01,  1.2987e+01,  ..., -3.4781e+00,\n",
            "            9.0383e+00, -1.4088e+01],\n",
            "          ...,\n",
            "          [-9.1470e+00, -8.5337e+00,  7.3214e+00,  ...,  9.5897e+00,\n",
            "            4.9515e+00, -1.3712e+01],\n",
            "          [-9.1989e+00, -8.5112e+00,  7.2761e+00,  ...,  9.6177e+00,\n",
            "            4.9333e+00, -1.3773e+01],\n",
            "          [-8.9825e+00, -8.8692e+00,  7.3516e+00,  ...,  1.0028e+01,\n",
            "            4.8477e+00, -1.4399e+01]],\n",
            "\n",
            "         [[-1.0757e+01,  3.7529e+00, -3.9050e+00,  ...,  1.1018e+01,\n",
            "            9.1509e+00,  4.9452e+00],\n",
            "          [ 2.8995e+00, -2.5072e-02, -6.7960e-02,  ...,  1.2494e+00,\n",
            "            7.4070e+00, -2.6220e+00],\n",
            "          [-6.1400e+00,  5.6581e+00, -4.2942e+00,  ...,  9.0136e+00,\n",
            "            1.1821e+01, -4.1775e+00],\n",
            "          ...,\n",
            "          [-6.0207e+00,  2.0905e-01, -9.0615e+00,  ...,  4.2824e+00,\n",
            "            1.3838e+01, -1.2466e+01],\n",
            "          [-6.0275e+00,  1.6345e-01, -9.1434e+00,  ...,  4.3025e+00,\n",
            "            1.3845e+01, -1.2432e+01],\n",
            "          [-6.0301e+00,  1.7854e-01, -9.2192e+00,  ...,  4.0585e+00,\n",
            "            1.4020e+01, -1.2449e+01]],\n",
            "\n",
            "         [[-3.1153e+00, -1.7249e+01,  1.4438e+01,  ...,  5.0570e+00,\n",
            "            1.5635e+00,  5.9790e+00],\n",
            "          [-6.2923e+00,  5.4000e-01,  6.8476e+00,  ...,  9.6558e+00,\n",
            "            8.3072e+00, -5.6637e+00],\n",
            "          [-9.5216e-01,  1.2385e+00,  1.5381e+01,  ...,  1.3570e+01,\n",
            "            2.7533e+00,  2.8868e+00],\n",
            "          ...,\n",
            "          [ 5.8833e-01, -2.0014e+00,  1.7265e+00,  ...,  5.3564e+00,\n",
            "            1.1231e+01,  1.2730e+01],\n",
            "          [ 6.2954e-01, -2.0177e+00,  1.8258e+00,  ...,  5.2078e+00,\n",
            "            1.1237e+01,  1.2825e+01],\n",
            "          [ 3.0729e-01, -1.9038e+00,  1.6803e+00,  ...,  4.4674e+00,\n",
            "            1.1531e+01,  1.2461e+01]],\n",
            "\n",
            "         [[ 1.1737e+01, -6.6537e+00,  5.5083e+00,  ..., -2.7730e+01,\n",
            "           -2.5387e+00,  2.7007e+00],\n",
            "          [ 2.4778e+00, -3.5684e+00,  2.2752e+00,  ..., -1.3770e+01,\n",
            "           -6.8241e+00,  3.3264e+00],\n",
            "          [ 6.3973e-01,  4.5940e+00, -1.5641e+00,  ..., -2.5369e+01,\n",
            "           -3.9978e+00,  8.8012e-01],\n",
            "          ...,\n",
            "          [ 9.6718e+00,  5.4786e+00, -1.3563e+01,  ..., -1.9041e+01,\n",
            "           -6.7815e+00,  6.4927e+00],\n",
            "          [ 9.6503e+00,  5.4822e+00, -1.3524e+01,  ..., -1.9010e+01,\n",
            "           -6.7855e+00,  6.4735e+00],\n",
            "          [ 9.1785e+00,  5.6419e+00, -1.3745e+01,  ..., -1.8630e+01,\n",
            "           -7.0843e+00,  6.4588e+00]]]], grad_fn=<TransposeBackward0>), tensor([[[[-1.4943e+01,  1.6212e+01, -6.4865e-01,  ...,  8.3282e-01,\n",
            "            1.5237e+01,  1.3593e+01],\n",
            "          [ 2.0954e+01,  7.8167e+00,  2.7409e+00,  ..., -1.4081e+00,\n",
            "            5.8601e+00,  1.8616e+01],\n",
            "          [ 1.4706e+01, -1.5765e+00,  2.0545e+01,  ...,  1.1894e+01,\n",
            "           -1.0303e+01,  1.7687e+01],\n",
            "          ...,\n",
            "          [-1.9753e+00, -3.7280e+00, -4.7797e-01,  ...,  1.7982e+01,\n",
            "            2.2664e+00, -2.0477e+01],\n",
            "          [-2.0566e+01, -5.1623e+00, -7.0470e+00,  ...,  1.3367e+01,\n",
            "            1.4882e+01, -6.2918e+00],\n",
            "          [ 2.0287e+00, -2.1327e-01, -1.2875e+00,  ..., -1.0931e+00,\n",
            "           -1.6302e+00,  2.4388e+00]],\n",
            "\n",
            "         [[-2.1196e+00, -4.5789e+01, -1.6982e+01,  ..., -2.8990e+01,\n",
            "            4.6401e+01,  3.6670e+01],\n",
            "          [-4.8731e+00, -8.7162e+00,  7.9863e+00,  ..., -1.6878e+00,\n",
            "            8.6773e+00,  2.2310e+01],\n",
            "          [ 1.4092e+01, -1.0313e+01,  4.1481e+01,  ..., -6.7063e+00,\n",
            "            1.3344e+01,  2.0196e+01],\n",
            "          ...,\n",
            "          [ 9.9806e+00,  2.9984e+00,  3.4862e+00,  ..., -6.6430e+00,\n",
            "           -5.5289e-01,  4.6494e+00],\n",
            "          [ 2.6574e+01,  1.3895e+01,  1.7310e+01,  ..., -1.1384e+01,\n",
            "           -2.6099e+01, -1.8751e+01],\n",
            "          [-4.1700e+00,  7.5673e-03,  3.0544e+00,  ...,  8.9865e-01,\n",
            "           -6.7659e+00, -4.6999e+00]],\n",
            "\n",
            "         [[-1.0980e+01, -9.7089e-01,  3.9543e+01,  ...,  3.8592e+01,\n",
            "            2.0363e+01,  1.5375e+01],\n",
            "          [ 1.9993e+01,  1.5561e+01, -1.9569e+00,  ...,  3.6133e+00,\n",
            "            4.0742e+01, -7.7117e+00],\n",
            "          [-1.9761e+00,  3.8576e+00, -7.1862e+00,  ...,  2.0069e+01,\n",
            "            1.3048e+01, -7.4404e+00],\n",
            "          ...,\n",
            "          [ 8.8715e+00, -2.3883e+01,  3.5359e+01,  ...,  2.0541e+01,\n",
            "            7.2948e-01, -1.9356e+01],\n",
            "          [-1.6084e+01, -7.7178e+00,  2.5546e+01,  ...,  1.3385e+01,\n",
            "           -1.2996e+01, -2.3195e+00],\n",
            "          [ 1.7507e+00, -9.6870e-01,  4.9659e+00,  ...,  1.2371e+00,\n",
            "            4.6210e-01,  7.6174e-01]],\n",
            "\n",
            "         [[ 8.1792e+00, -1.7900e+01,  6.1801e+00,  ..., -2.6516e+00,\n",
            "            3.7457e+00, -8.8804e-02],\n",
            "          [ 4.2584e+00,  8.1231e+00, -1.8508e+01,  ...,  7.4988e+00,\n",
            "           -2.8240e-02,  1.8330e+01],\n",
            "          [-4.0807e+00, -8.9144e+00, -1.7159e+01,  ...,  1.5673e+01,\n",
            "           -2.1670e-01,  3.8384e+00],\n",
            "          ...,\n",
            "          [-4.1070e+00,  1.1492e+01, -9.8567e+00,  ..., -2.3682e+01,\n",
            "            7.7063e-01, -1.3964e+01],\n",
            "          [-2.6987e+01, -1.2339e+00, -1.9929e+01,  ...,  2.0867e+01,\n",
            "           -4.8042e+00, -5.8158e+01],\n",
            "          [ 4.4263e+00,  4.4870e+00, -7.6553e+00,  ...,  8.6697e-01,\n",
            "           -4.8828e+00,  9.5942e+00]]],\n",
            "\n",
            "\n",
            "        [[[ 3.7577e+00,  1.6934e+01,  1.6814e+01,  ..., -3.7195e+00,\n",
            "            2.3240e+00, -6.3755e+00],\n",
            "          [-2.1929e+00,  2.1047e+01, -8.9329e+00,  ...,  7.3646e+00,\n",
            "            2.1529e+01, -7.0529e-01],\n",
            "          [ 1.6133e+01, -2.3622e+01,  2.1431e+01,  ..., -2.3280e+01,\n",
            "            1.3528e+01,  1.8637e+01],\n",
            "          ...,\n",
            "          [ 6.9191e+00,  1.1978e+01, -9.7086e-01,  ..., -1.4057e+01,\n",
            "            1.7202e+01, -7.1434e+00],\n",
            "          [-5.8485e+00,  3.9054e+00,  1.5597e+01,  ..., -1.0509e+01,\n",
            "            2.3063e+01,  3.4485e+00],\n",
            "          [ 2.3694e+00, -1.4698e+00, -1.3404e+00,  ...,  2.8148e-01,\n",
            "           -1.3926e+00,  2.0164e+00]],\n",
            "\n",
            "         [[-7.0515e+00,  9.7771e+00,  9.7081e+00,  ...,  3.3806e+00,\n",
            "            1.0583e+01,  7.8754e+00],\n",
            "          [-8.5114e+00, -7.8157e+00, -1.2566e+00,  ..., -1.5509e+01,\n",
            "            1.3348e+01,  8.4639e+00],\n",
            "          [-3.3159e+00,  2.6728e+00,  1.2888e+01,  ...,  6.2679e+00,\n",
            "            1.6711e+01, -4.4014e+00],\n",
            "          ...,\n",
            "          [ 1.9715e+00, -9.1266e-01, -1.0352e+00,  ...,  1.1860e+01,\n",
            "            1.1479e+01, -7.2794e+00],\n",
            "          [-1.7508e+01,  1.3246e+01, -8.8216e+00,  ..., -1.0219e+01,\n",
            "            2.7680e+00, -1.1510e+01],\n",
            "          [-4.2607e+00, -2.3774e-01,  2.7064e+00,  ...,  2.9305e-01,\n",
            "           -6.4563e+00, -5.3450e+00]],\n",
            "\n",
            "         [[-3.0746e+00,  5.9784e+00,  1.6375e+01,  ..., -3.3134e+01,\n",
            "           -1.0196e+01,  6.0529e+00],\n",
            "          [ 7.2421e-01,  4.7186e+00,  4.4561e+01,  ..., -7.2407e-01,\n",
            "            1.7870e-01,  8.0735e-01],\n",
            "          [-5.1867e+00, -7.8791e+00,  2.3545e+00,  ...,  1.1861e+01,\n",
            "           -2.6425e+00, -1.7582e+01],\n",
            "          ...,\n",
            "          [-1.1109e+01, -1.5468e+00,  1.1236e+01,  ..., -4.9871e+00,\n",
            "            5.0823e+00,  6.1635e+00],\n",
            "          [-1.5158e+01, -4.5414e+00,  4.8925e+01,  ...,  3.8873e+01,\n",
            "            2.4130e-01, -2.4995e+01],\n",
            "          [ 1.7786e+00, -1.0512e+00,  4.3796e+00,  ...,  1.2381e+00,\n",
            "            5.2705e-01,  1.0972e+00]],\n",
            "\n",
            "         [[ 4.4195e+00,  5.6562e+00, -1.2120e+01,  ...,  8.9792e-01,\n",
            "            4.7981e+01,  7.1356e+00],\n",
            "          [-1.2728e+01,  1.5064e+01, -5.0368e+00,  ..., -7.0702e+00,\n",
            "            9.7408e+00,  6.1061e+00],\n",
            "          [-1.5837e+01, -7.2643e+00, -1.2091e+01,  ...,  1.5695e+01,\n",
            "            1.1847e+01,  1.3610e+01],\n",
            "          ...,\n",
            "          [ 9.1279e+00, -4.5580e+00, -6.3340e+00,  ...,  7.3478e+00,\n",
            "            4.6055e+00,  6.9093e+00],\n",
            "          [-1.9296e+01,  8.0908e+00, -7.2073e+00,  ..., -1.0945e+01,\n",
            "           -9.3378e+00, -5.2753e+00],\n",
            "          [ 5.1654e+00,  3.3610e+00, -7.8672e+00,  ...,  1.4830e+00,\n",
            "           -5.2060e+00,  1.1047e+01]]],\n",
            "\n",
            "\n",
            "        [[[-3.3976e-01,  8.8467e+00,  1.7446e+01,  ...,  1.2247e+01,\n",
            "            8.9783e+00, -3.1419e+01],\n",
            "          [ 8.9922e+00, -1.1800e+01, -4.3403e+00,  ...,  1.5310e+00,\n",
            "           -1.2112e+01,  1.8173e+01],\n",
            "          [ 2.0842e+01,  2.3147e+01,  9.2462e+00,  ..., -6.6019e+00,\n",
            "           -3.9372e+00,  1.8612e+01],\n",
            "          ...,\n",
            "          [-8.9336e+00, -5.3410e-01, -2.6894e+00,  ..., -1.1557e+00,\n",
            "           -1.8585e+01, -6.1761e-01],\n",
            "          [-8.9336e+00, -5.3410e-01, -2.6894e+00,  ..., -1.1557e+00,\n",
            "           -1.8585e+01, -6.1761e-01],\n",
            "          [-8.9336e+00, -5.3410e-01, -2.6894e+00,  ..., -1.1557e+00,\n",
            "           -1.8585e+01, -6.1761e-01]],\n",
            "\n",
            "         [[ 2.8264e+01, -2.5553e+01,  4.8584e+00,  ..., -2.1365e+01,\n",
            "           -2.5990e+01, -2.0096e+01],\n",
            "          [ 6.6065e+00, -1.2891e+01, -1.8164e+01,  ..., -7.6562e+00,\n",
            "            4.0317e+00, -3.5657e+00],\n",
            "          [ 1.5960e+01, -1.4704e+01, -1.5331e+00,  ..., -9.9649e+00,\n",
            "            5.1988e+00, -1.2267e+01],\n",
            "          ...,\n",
            "          [ 3.5324e+00, -1.1617e+00,  4.7247e+00,  ..., -1.1704e+01,\n",
            "           -1.0520e+01,  1.4011e+01],\n",
            "          [ 3.5324e+00, -1.1617e+00,  4.7247e+00,  ..., -1.1704e+01,\n",
            "           -1.0520e+01,  1.4011e+01],\n",
            "          [ 3.5324e+00, -1.1617e+00,  4.7247e+00,  ..., -1.1704e+01,\n",
            "           -1.0520e+01,  1.4011e+01]],\n",
            "\n",
            "         [[-2.0805e+01,  2.1514e+01,  3.5565e+00,  ...,  1.5711e+00,\n",
            "           -2.7812e+01,  1.0326e+01],\n",
            "          [-3.7539e+00,  2.2836e+01,  7.1676e+00,  ...,  8.0071e+00,\n",
            "            8.2454e+00,  3.7125e+01],\n",
            "          [-8.6644e+00,  1.9461e+00,  1.3468e+01,  ..., -1.9956e+01,\n",
            "            1.4748e+01,  3.4972e+00],\n",
            "          ...,\n",
            "          [ 2.0376e+01, -1.0356e+01,  8.6687e+00,  ..., -8.5990e+00,\n",
            "            7.2120e+00,  5.7453e+00],\n",
            "          [ 2.0376e+01, -1.0356e+01,  8.6687e+00,  ..., -8.5990e+00,\n",
            "            7.2120e+00,  5.7453e+00],\n",
            "          [ 2.0376e+01, -1.0356e+01,  8.6687e+00,  ..., -8.5990e+00,\n",
            "            7.2120e+00,  5.7453e+00]],\n",
            "\n",
            "         [[ 2.4396e+01, -1.2649e+01,  4.5334e+00,  ..., -1.4541e+01,\n",
            "            7.2028e+00,  1.0236e+01],\n",
            "          [-2.0411e+01, -2.0668e+01, -2.5681e+00,  ...,  7.9653e+00,\n",
            "           -1.8070e+01, -7.3518e+00],\n",
            "          [ 2.0831e+00, -3.7067e+00, -5.8004e+00,  ..., -3.5374e+00,\n",
            "           -3.9614e+00,  9.3751e+00],\n",
            "          ...,\n",
            "          [ 3.6587e+00,  4.9808e+00, -9.5954e-01,  ...,  1.6688e+01,\n",
            "            1.0755e+01, -3.1844e+01],\n",
            "          [ 3.6587e+00,  4.9808e+00, -9.5954e-01,  ...,  1.6688e+01,\n",
            "            1.0755e+01, -3.1844e+01],\n",
            "          [ 3.6587e+00,  4.9808e+00, -9.5954e-01,  ...,  1.6688e+01,\n",
            "            1.0755e+01, -3.1844e+01]]],\n",
            "\n",
            "\n",
            "        ...,\n",
            "\n",
            "\n",
            "        [[[ 2.3942e+00, -3.0350e+00, -5.2496e+00,  ...,  2.0397e+01,\n",
            "            2.0525e+01, -1.4521e+00],\n",
            "          [ 7.0624e+00, -1.1871e-01,  1.7617e-02,  ...,  2.8490e+00,\n",
            "            2.2229e+00,  2.7416e+01],\n",
            "          [ 1.2377e+01,  4.0270e-01, -7.6391e+00,  ..., -2.0430e+01,\n",
            "            1.6319e+01,  3.0361e+01],\n",
            "          ...,\n",
            "          [-1.1061e+01, -4.4977e-01, -3.7594e+00,  ..., -4.4906e+00,\n",
            "           -1.3579e+01,  3.8989e-02],\n",
            "          [-1.1061e+01, -4.4977e-01, -3.7594e+00,  ..., -4.4906e+00,\n",
            "           -1.3579e+01,  3.8989e-02],\n",
            "          [-1.1061e+01, -4.4977e-01, -3.7594e+00,  ..., -4.4906e+00,\n",
            "           -1.3579e+01,  3.8989e-02]],\n",
            "\n",
            "         [[ 1.1002e+01,  4.9501e+00, -2.2587e-01,  ..., -9.0321e+00,\n",
            "           -6.7106e+00, -4.8863e+00],\n",
            "          [-2.9850e+01,  2.4679e+00,  5.3980e+00,  ...,  2.5989e+01,\n",
            "            9.3459e+00, -6.6451e+00],\n",
            "          [-1.7733e+01, -4.2034e+00, -9.1535e+00,  ...,  1.6864e+01,\n",
            "           -9.2325e-01,  4.8751e+00],\n",
            "          ...,\n",
            "          [ 3.1327e+00,  8.3720e-01, -3.8924e-01,  ..., -1.3104e+01,\n",
            "           -1.3612e+01,  1.1463e+01],\n",
            "          [ 3.1327e+00,  8.3720e-01, -3.8924e-01,  ..., -1.3104e+01,\n",
            "           -1.3612e+01,  1.1463e+01],\n",
            "          [ 3.1327e+00,  8.3720e-01, -3.8924e-01,  ..., -1.3104e+01,\n",
            "           -1.3612e+01,  1.1463e+01]],\n",
            "\n",
            "         [[ 1.8402e+01,  2.1265e+01,  2.3887e+01,  ...,  9.3586e+00,\n",
            "           -4.4698e+00, -8.8504e+00],\n",
            "          [ 1.2223e+01, -1.4102e+01,  1.6089e+01,  ...,  1.2900e+00,\n",
            "            3.3862e+00,  2.5769e+00],\n",
            "          [ 1.5868e+00,  2.3627e+01,  1.5505e+01,  ...,  3.0200e+01,\n",
            "           -9.1901e+00,  4.0761e+00],\n",
            "          ...,\n",
            "          [ 2.0298e+01, -1.1680e+01,  1.2928e+01,  ..., -9.1170e+00,\n",
            "            2.6301e+00,  2.7667e+00],\n",
            "          [ 2.0298e+01, -1.1680e+01,  1.2928e+01,  ..., -9.1170e+00,\n",
            "            2.6301e+00,  2.7667e+00],\n",
            "          [ 2.0298e+01, -1.1680e+01,  1.2928e+01,  ..., -9.1170e+00,\n",
            "            2.6301e+00,  2.7667e+00]],\n",
            "\n",
            "         [[ 1.0641e+01,  3.6553e-02, -1.7597e+01,  ..., -1.0220e+01,\n",
            "            1.6384e+01, -7.7998e-01],\n",
            "          [-1.2763e+01,  2.9938e+00,  4.9586e+00,  ...,  4.3187e+00,\n",
            "            7.7764e+00, -8.6275e+00],\n",
            "          [-9.6650e+00, -9.8636e+00, -2.0452e+01,  ..., -1.1861e+00,\n",
            "           -6.3153e+00, -2.8129e+01],\n",
            "          ...,\n",
            "          [ 2.8846e+00,  8.1963e+00, -2.0117e+00,  ...,  1.8244e+01,\n",
            "            7.7286e+00, -3.4892e+01],\n",
            "          [ 2.8846e+00,  8.1963e+00, -2.0117e+00,  ...,  1.8244e+01,\n",
            "            7.7286e+00, -3.4892e+01],\n",
            "          [ 2.8846e+00,  8.1963e+00, -2.0117e+00,  ...,  1.8244e+01,\n",
            "            7.7286e+00, -3.4892e+01]]],\n",
            "\n",
            "\n",
            "        [[[ 2.1791e+01,  3.2253e+01,  1.3200e+01,  ..., -2.0227e+00,\n",
            "            3.6887e+01, -1.2846e+01],\n",
            "          [ 6.4555e-01,  1.2933e+01,  4.0690e-01,  ..., -2.7950e+00,\n",
            "            1.8961e+01,  1.6342e+00],\n",
            "          [ 4.9241e+00, -2.2351e-01,  2.1712e+01,  ..., -9.1343e+00,\n",
            "            8.4972e+00,  1.2553e+01],\n",
            "          ...,\n",
            "          [ 2.4181e+01,  1.1597e+01,  5.9369e+00,  ..., -3.1428e+00,\n",
            "            5.8521e+00, -4.1578e+00],\n",
            "          [-1.3401e+01,  2.0228e+00,  2.8194e+01,  ...,  7.0391e+00,\n",
            "            3.0918e-01,  1.0015e+00],\n",
            "          [ 2.3092e+00, -4.4091e-01, -1.6928e+00,  ..., -4.1691e-01,\n",
            "           -6.4947e-01,  3.0128e+00]],\n",
            "\n",
            "         [[-1.2783e+00,  1.4957e+01,  2.3210e+00,  ..., -1.5080e+01,\n",
            "            1.2934e+01,  2.3287e+01],\n",
            "          [-7.1648e+00, -2.2265e+01,  4.6671e+00,  ..., -2.6476e+01,\n",
            "            2.1750e+01,  2.8893e+01],\n",
            "          [ 2.3116e+01, -1.1677e+01,  1.4929e+01,  ..., -1.9014e+01,\n",
            "           -1.0124e+01,  2.0220e+01],\n",
            "          ...,\n",
            "          [ 8.4654e+00,  1.1717e+01, -3.7487e+01,  ..., -4.3473e+01,\n",
            "           -3.8572e+00, -9.7639e+00],\n",
            "          [ 1.9403e+01, -7.3550e+00, -9.2919e+00,  ..., -4.0923e+00,\n",
            "           -2.8746e+01,  6.4668e+00],\n",
            "          [-4.2918e+00,  2.0924e-01,  3.3820e+00,  ...,  2.1801e-01,\n",
            "           -6.3885e+00, -4.6806e+00]],\n",
            "\n",
            "         [[-6.8252e+00,  1.5730e+01, -1.6582e+01,  ...,  3.7423e+01,\n",
            "           -1.6651e+01,  2.5042e+01],\n",
            "          [-1.3089e+01, -5.2790e+00,  3.1870e+01,  ...,  4.9053e+01,\n",
            "           -1.7867e-01,  3.8141e+00],\n",
            "          [ 3.2931e+00,  4.3216e+00,  6.5909e+00,  ...,  1.7628e+01,\n",
            "            3.8784e+00, -7.8885e+00],\n",
            "          ...,\n",
            "          [ 2.7652e+01, -1.2807e+01,  3.2657e+01,  ..., -3.0060e+01,\n",
            "           -1.8707e+01,  3.0339e-02],\n",
            "          [ 1.9134e+00, -1.9146e+01,  3.8211e+01,  ...,  4.9132e+01,\n",
            "           -1.2090e+01,  7.9664e-01],\n",
            "          [ 1.6032e+00, -8.3514e-01,  4.1111e+00,  ...,  1.6369e+00,\n",
            "            5.3536e-01,  4.9881e-01]],\n",
            "\n",
            "         [[-1.8063e+01,  5.9380e+00, -2.3764e+01,  ..., -2.8332e+01,\n",
            "            1.9680e+00, -3.6193e-01],\n",
            "          [-1.8534e+01, -1.5887e+01, -7.6297e+00,  ...,  8.6971e-01,\n",
            "            8.5139e+00, -3.5761e+00],\n",
            "          [-9.3440e-02, -1.4340e+01, -8.0167e+00,  ..., -1.2984e+00,\n",
            "           -1.1604e+01,  5.5332e+00],\n",
            "          ...,\n",
            "          [-1.3736e+01, -1.4976e+01, -2.2822e+00,  ..., -1.1831e+01,\n",
            "            1.1806e+01, -4.4301e+00],\n",
            "          [ 1.2870e+00, -2.6185e+01, -1.4564e+01,  ..., -5.3066e+00,\n",
            "           -1.7202e+01, -7.3902e+00],\n",
            "          [ 4.7449e+00,  4.4217e+00, -7.4623e+00,  ...,  1.2205e+00,\n",
            "           -4.9278e+00,  1.0299e+01]]],\n",
            "\n",
            "\n",
            "        [[[ 3.5287e+01, -1.4079e+01,  5.4448e+00,  ..., -8.4297e+00,\n",
            "            2.4990e+01,  1.9871e+00],\n",
            "          [ 2.4202e+01,  4.0713e+01,  3.4161e+00,  ...,  1.5452e+01,\n",
            "            2.9601e+01,  1.3124e+01],\n",
            "          [ 1.0395e+01, -1.6539e+01,  2.4471e+01,  ...,  2.6014e+01,\n",
            "            1.1156e+01, -1.7851e+01],\n",
            "          ...,\n",
            "          [-6.7101e+00, -5.5168e+00, -1.3232e+00,  ..., -3.6252e+00,\n",
            "           -1.6713e+01,  1.8314e+00],\n",
            "          [-6.5843e+00, -5.7346e+00, -1.2601e+00,  ..., -3.6845e+00,\n",
            "           -1.6695e+01,  1.9125e+00],\n",
            "          [-6.3032e+00, -5.2501e+00, -5.0787e-01,  ..., -3.8217e+00,\n",
            "           -1.6984e+01,  1.4865e+00]],\n",
            "\n",
            "         [[ 3.0114e+00,  3.5771e+01,  1.8771e+01,  ..., -1.4603e+01,\n",
            "            3.1030e+00, -2.4943e+00],\n",
            "          [ 1.9819e+01,  2.0328e+00,  6.6832e+00,  ..., -3.5210e+01,\n",
            "           -4.5353e+01, -3.0851e+00],\n",
            "          [-5.0619e+00,  2.3945e+01,  6.7137e-02,  ..., -1.7100e+00,\n",
            "           -1.8319e+01,  2.1501e+00],\n",
            "          ...,\n",
            "          [ 3.2509e+00,  2.9525e+00,  2.7463e+00,  ..., -1.3547e+01,\n",
            "           -1.1587e+01,  1.0967e+01],\n",
            "          [ 3.3786e+00,  2.9263e+00,  2.7762e+00,  ..., -1.3471e+01,\n",
            "           -1.1614e+01,  1.0954e+01],\n",
            "          [ 3.4561e+00,  2.7708e+00,  3.2686e+00,  ..., -1.3065e+01,\n",
            "           -1.2070e+01,  1.0482e+01]],\n",
            "\n",
            "         [[-1.4489e+01,  3.8916e+01,  4.4077e+01,  ...,  3.3957e+01,\n",
            "           -1.9705e+01, -3.4759e+01],\n",
            "          [-1.9366e+01, -2.0636e+01,  1.9307e+01,  ...,  2.6608e+01,\n",
            "           -3.5512e+01, -1.0063e+01],\n",
            "          [-1.0300e+01, -7.4438e-02,  3.1880e+01,  ...,  2.3986e+01,\n",
            "           -1.5638e+01, -1.8765e+01],\n",
            "          ...,\n",
            "          [ 2.0122e+01, -4.8400e+00,  1.1041e+01,  ..., -1.1410e+01,\n",
            "            7.9484e+00,  5.2019e+00],\n",
            "          [ 2.0165e+01, -4.8748e+00,  1.1078e+01,  ..., -1.1551e+01,\n",
            "            7.8851e+00,  5.2240e+00],\n",
            "          [ 1.9813e+01, -4.8357e+00,  1.1418e+01,  ..., -1.2013e+01,\n",
            "            7.9514e+00,  4.7623e+00]],\n",
            "\n",
            "         [[ 6.8456e-01,  1.4807e+01, -1.9031e+01,  ..., -1.6745e+00,\n",
            "            1.6858e+01, -1.2565e+01],\n",
            "          [ 7.6164e+00, -3.1621e+01, -3.4302e+00,  ..., -1.8160e+00,\n",
            "            1.0226e+01,  6.7674e+00],\n",
            "          [-2.4163e+01, -1.0710e+01,  4.1101e+00,  ...,  1.6145e+01,\n",
            "           -2.6440e+00,  1.3152e+01],\n",
            "          ...,\n",
            "          [ 1.0071e+00,  1.8878e+01, -5.5912e+00,  ...,  1.3419e+01,\n",
            "            3.4591e+00, -2.9719e+01],\n",
            "          [ 1.0510e+00,  1.8916e+01, -5.7107e+00,  ...,  1.3500e+01,\n",
            "            3.4685e+00, -2.9816e+01],\n",
            "          [ 1.4860e+00,  1.9816e+01, -5.9034e+00,  ...,  1.3658e+01,\n",
            "            3.3322e+00, -3.0049e+01]]]], grad_fn=<TransposeBackward0>))), decoder_hidden_states=None, decoder_attentions=None, cross_attentions=None, encoder_last_hidden_state=tensor([[[-0.3085,  0.9929, -0.7906,  ...,  1.6782, -0.1140,  0.5790],\n",
            "         [-0.7930, -0.1418, -1.6175,  ...,  0.4223, -0.6120, -0.7113],\n",
            "         [-0.4827, -1.7006, -2.5418,  ...,  1.1923,  1.2131, -0.2392],\n",
            "         ...,\n",
            "         [-0.2556,  0.3328, -0.5716,  ..., -0.8342,  0.3552, -0.2026],\n",
            "         [-0.5006, -0.1018, -0.4076,  ...,  1.4123,  0.8235, -0.2426],\n",
            "         [-0.1012, -0.1295,  0.3350,  ...,  0.3116, -0.3580, -0.1827]],\n",
            "\n",
            "        [[-0.3545,  1.3980, -1.5512,  ...,  0.1075, -0.2535,  1.1275],\n",
            "         [-0.2501,  2.6587, -0.6083,  ...,  1.8642, -0.8755,  1.2026],\n",
            "         [ 0.1774,  2.0471, -1.1461,  ..., -0.0776,  0.0046,  0.0142],\n",
            "         ...,\n",
            "         [ 1.0752,  1.0581, -0.7135,  ..., -0.7073, -0.1777,  0.4352],\n",
            "         [ 2.0847,  0.8395, -0.2796,  ...,  0.0584,  0.1375, -0.2291],\n",
            "         [-0.1285, -0.0828,  0.3298,  ...,  0.3042, -0.3362, -0.1596]],\n",
            "\n",
            "        [[-0.4752,  0.8746, -2.0202,  ...,  0.1264,  0.7291,  0.1532],\n",
            "         [ 0.1201,  0.9691, -1.7506,  ..., -0.8307,  0.2130,  1.4098],\n",
            "         [-0.7005,  0.7749, -2.5441,  ..., -1.3752,  0.6586,  0.7416],\n",
            "         ...,\n",
            "         [ 0.0513, -0.4454, -0.5006,  ...,  0.3052, -0.5238, -0.3241],\n",
            "         [ 0.0513, -0.4454, -0.5006,  ...,  0.3052, -0.5238, -0.3241],\n",
            "         [ 0.0513, -0.4454, -0.5006,  ...,  0.3052, -0.5238, -0.3241]],\n",
            "\n",
            "        ...,\n",
            "\n",
            "        [[-0.7513,  0.4781, -1.2694,  ..., -0.2517,  0.0124, -1.3655],\n",
            "         [-0.3785,  0.9712,  0.3323,  ...,  1.7581,  0.7794,  0.0838],\n",
            "         [-0.3409,  0.9092, -1.5046,  ...,  0.0233,  0.2400, -0.0687],\n",
            "         ...,\n",
            "         [ 0.1278, -0.5747, -0.2200,  ..., -0.1013, -0.4193, -0.5582],\n",
            "         [ 0.1278, -0.5747, -0.2200,  ..., -0.1013, -0.4193, -0.5582],\n",
            "         [ 0.1278, -0.5747, -0.2200,  ..., -0.1013, -0.4193, -0.5582]],\n",
            "\n",
            "        [[-1.4554, -1.1051, -3.3019,  ...,  0.3302,  0.9782, -0.7686],\n",
            "         [-0.8710, -1.2985, -2.1254,  ...,  0.6183,  0.0333, -0.5122],\n",
            "         [-1.3179, -0.5172, -1.7261,  ...,  0.7049, -0.8633, -2.1870],\n",
            "         ...,\n",
            "         [ 1.0099,  0.3730, -0.0940,  ...,  0.2418, -0.3556, -1.2053],\n",
            "         [ 0.8382,  1.0822, -1.2808,  ..., -0.3015, -0.9759, -2.0632],\n",
            "         [-0.0939, -0.0743,  0.2964,  ...,  0.2617, -0.3617, -0.2523]],\n",
            "\n",
            "        [[ 0.0145, -0.2161, -2.6183,  ..., -0.0663,  0.2290,  0.0408],\n",
            "         [-1.9041,  1.5769, -2.3594,  ..., -1.4891,  0.1466,  1.2123],\n",
            "         [-0.0521,  1.3992, -1.7283,  ...,  1.4621, -0.4758,  1.1732],\n",
            "         ...,\n",
            "         [ 0.2887,  0.1381,  0.0078,  ...,  0.5018, -0.6632,  0.1311],\n",
            "         [ 0.2825,  0.1472,  0.0055,  ...,  0.5053, -0.6692,  0.1584],\n",
            "         [ 0.2466,  0.1718,  0.0943,  ...,  0.5508, -0.6450,  0.2826]]],\n",
            "       grad_fn=<MulBackward0>), encoder_hidden_states=None, encoder_attentions=None)\n"
          ]
        }
      ]
    },
    {
      "cell_type": "code",
      "source": [],
      "metadata": {
        "id": "HNZ-PdAnZVB5"
      },
      "execution_count": null,
      "outputs": []
    }
  ]
}