fruit-classifier.ipynb

{
 "cells": [
  {
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "## <u>Aim: To classifiy fruits images.</u>\n",
    "### Dataset: Fruits360\n",
    "\n",
    "- Then to build an iOS app to use the machine learning model."
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 24,
   "metadata": {},
   "outputs": [],
   "source": [
    "# importing\n",
    "import random\n",
    "import pandas as pd\n",
    "import numpy as np\n",
    "import torchvision\n",
    "import torch\n",
    "from torch import nn\n",
    "from torch.autograd import Variable\n",
    "from sklearn.model_selection import train_test_split\n",
    "import torchvision.transforms as T\n",
    "from torchvision.datasets import ImageFolder\n",
    "import matplotlib.pyplot as plt\n",
    "import cv2\n",
    "from torch.utils.data import DataLoader\n",
    "from sklearn.metrics import accuracy_score"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 25,
   "metadata": {},
   "outputs": [],
   "source": [
    "# Loading data\n",
    "transforms_train = T.Compose([T.ToTensor(),T.Normalize([0.5,0.5,0.5],[0.5,0.5,0.5])])\n",
    "image_data_train = ImageFolder(\"./fruits-360/Training\",transform=transforms_train)\n",
    "image_data_test = ImageFolder(\"./fruits-360/Validation\",transform=transforms_train)\n",
    "\n",
    "# Shuffling data and then collecting all the labels.\n",
    "random.shuffle(image_data_train.samples)\n",
    "random.shuffle(image_data_test.samples)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 26,
   "metadata": {},
   "outputs": [],
   "source": [
    "# Total classes\n",
    "classes_idx = image_data_train.class_to_idx\n",
    "classes = len(image_data_train.classes)\n",
    "len_train_data = len(image_data_train)\n",
    "len_test_data = len(image_data_test)\n",
    "\n",
    "\n",
    "def get_labels():\n",
    "    labels_train = [] # All the labels\n",
    "    labels_test = []\n",
    "    for i in image_data_train.imgs:\n",
    "        labels_train.append(i[1])\n",
    "    \n",
    "    for j in image_data_test.imgs:\n",
    "        labels_test.append(j[1])\n",
    "    \n",
    "    return (labels_train, labels_test)\n",
    "\n",
    "labels_train, labels_test = get_labels()\n"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 27,
   "metadata": {},
   "outputs": [],
   "source": [
    "train_loader = DataLoader(dataset=image_data_train,batch_size=100)\n",
    "test_loader = DataLoader(dataset=image_data_test,batch_size=100)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 37,
   "metadata": {},
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "torch.Size([100, 3, 100, 100])\n"
     ]
    },
    {
     "data": {
      "text/plain": [
       "torch.Size([3, 100, 100])"
      ]
     },
     "execution_count": 37,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "print (iter(train_loader).next()[0].shape)\n",
    "record = iter(train_loader).next()[0]"
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "- We can see that the image is (batch_size, channel, image_height, image_width)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 3,
   "metadata": {},
   "outputs": [],
   "source": [
    "# Flatten Layer\n",
    "class Flatten(nn.Module):\n",
    "    def forward(self, x):\n",
    "        return x.view(x.size(0), -1)\n",
    "# Main Model\n",
    "class Model:\n",
    "    def build_model(self):\n",
    "        model = nn.Sequential(nn.Conv2d(3, 64, kernel_size=5, stride=1), nn.ReLU(),nn.MaxPool2d(2), nn.Conv2d(64, 64, kernel_size=7, stride=1),\n",
    "                             nn.ReLU(), nn.MaxPool2d(3), nn.Conv2d(64, 64, kernel_size=7), nn.ReLU(), nn.MaxPool2d(5), Flatten(), nn.Linear(64, 100), nn.ReLU(), nn.Linear(100, 65))\n",
    "        return model"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 4,
   "metadata": {},
   "outputs": [
    {
     "data": {
      "text/plain": [
       "Sequential(\n",
       "  (0): Conv2d(3, 64, kernel_size=(5, 5), stride=(1, 1))\n",
       "  (1): ReLU()\n",
       "  (2): MaxPool2d(kernel_size=2, stride=2, padding=0, dilation=1, ceil_mode=False)\n",
       "  (3): Conv2d(64, 64, kernel_size=(7, 7), stride=(1, 1))\n",
       "  (4): ReLU()\n",
       "  (5): MaxPool2d(kernel_size=3, stride=3, padding=0, dilation=1, ceil_mode=False)\n",
       "  (6): Conv2d(64, 64, kernel_size=(7, 7), stride=(1, 1))\n",
       "  (7): ReLU()\n",
       "  (8): MaxPool2d(kernel_size=5, stride=5, padding=0, dilation=1, ceil_mode=False)\n",
       "  (9): Flatten()\n",
       "  (10): Linear(in_features=64, out_features=100, bias=True)\n",
       "  (11): ReLU()\n",
       "  (12): Linear(in_features=100, out_features=65, bias=True)\n",
       ")"
      ]
     },
     "execution_count": 4,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "# Building \n",
    "model = Model()\n",
    "model = model.build_model()\n",
    "\n",
    "# Checking for GPU\n",
    "device = torch.device(\"cuda:0\" if torch.cuda.is_available() else \"cpu\")\n",
    "model.to(device)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 109,
   "metadata": {},
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "Sequential(\n",
      "  (0): Conv2d(3, 64, kernel_size=(5, 5), stride=(1, 1))\n",
      "  (1): ReLU()\n",
      "  (2): MaxPool2d(kernel_size=2, stride=2, padding=0, dilation=1, ceil_mode=False)\n",
      "  (3): Conv2d(64, 64, kernel_size=(7, 7), stride=(1, 1))\n",
      "  (4): ReLU()\n",
      "  (5): MaxPool2d(kernel_size=3, stride=3, padding=0, dilation=1, ceil_mode=False)\n",
      "  (6): Conv2d(64, 64, kernel_size=(7, 7), stride=(1, 1))\n",
      "  (7): ReLU()\n",
      "  (8): MaxPool2d(kernel_size=5, stride=5, padding=0, dilation=1, ceil_mode=False)\n",
      "  (9): Flatten()\n",
      "  (10): Linear(in_features=64, out_features=100, bias=True)\n",
      "  (11): ReLU()\n",
      "  (12): Linear(in_features=100, out_features=65, bias=True)\n",
      ")\n",
      "62\n",
      "62\n"
     ]
    }
   ],
   "source": [
    "print (model)\n",
    "print (labels_train[0])\n",
    "print (image_data_train.samples[0][1])"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 151,
   "metadata": {},
   "outputs": [],
   "source": [
    "optimizer = torch.optim.Adagrad(model.parameters(), lr=0.01)\n",
    "criterion = nn.CrossEntropyLoss()"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 152,
   "metadata": {},
   "outputs": [],
   "source": [
    "# Train the model\n",
    "def train(epochs):\n",
    "    model.train()\n",
    "    losses = []\n",
    "    for epoch in range(1, epochs+1):\n",
    "        print (\"epoch #\", epoch)\n",
    "        current_loss = 0.0\n",
    "        for feature, label in train_loader:\n",
    "            x = Variable(feature, requires_grad=False).float().to(device)\n",
    "            y = Variable(label, requires_grad=False).long().to(device)\n",
    "            optimizer.zero_grad() # Zeroing the grads\n",
    "            y_pred = model(x) # Calculating prediction\n",
    "            correct = y_pred.max(1)[1].eq(y).sum()\n",
    "            print (\"no. of correct items classified: \", correct.item())\n",
    "            loss = criterion(y_pred, y) # Calculating loss (log_softmax already included)\n",
    "            print (\"loss: \", loss.item())\n",
    "            current_loss+=loss.item()\n",
    "            loss.backward() # Gradient cal\n",
    "            optimizer.step() # Changing weights\n",
    "        losses.append(current_loss) # Only storing loss after every epoch\n",
    "    return losses\n",
    "# Test the model\n",
    "def test():\n",
    "    model.eval()\n",
    "    with torch.no_grad():\n",
    "        for feature, label in test_loader:\n",
    "            pred = model(feature)\n",
    "            print (\"acc: \", accuracy_score(labels_test, pred.max(1)[1].data.numpy()) * 100)\n",
    "            loss = criterion(pred, label)\n",
    "            print (\"loss: \", loss.item())"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 6,
   "metadata": {},
   "outputs": [
    {
     "data": {
      "text/plain": [
       "Sequential(\n",
       "  (0): Conv2d(3, 64, kernel_size=(5, 5), stride=(1, 1))\n",
       "  (1): ReLU()\n",
       "  (2): MaxPool2d(kernel_size=2, stride=2, padding=0, dilation=1, ceil_mode=False)\n",
       "  (3): Conv2d(64, 64, kernel_size=(7, 7), stride=(1, 1))\n",
       "  (4): ReLU()\n",
       "  (5): MaxPool2d(kernel_size=3, stride=3, padding=0, dilation=1, ceil_mode=False)\n",
       "  (6): Conv2d(64, 64, kernel_size=(7, 7), stride=(1, 1))\n",
       "  (7): ReLU()\n",
       "  (8): MaxPool2d(kernel_size=5, stride=5, padding=0, dilation=1, ceil_mode=False)\n",
       "  (9): Flatten()\n",
       "  (10): Linear(in_features=64, out_features=100, bias=True)\n",
       "  (11): ReLU()\n",
       "  (12): Linear(in_features=100, out_features=65, bias=True)\n",
       ")"
      ]
     },
     "execution_count": 6,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "model"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 15,
   "metadata": {},
   "outputs": [],
   "source": [
    "loaded_state_dict = torch.load(\"model-state-dict.pth\",map_location=\"cpu\")"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 17,
   "metadata": {},
   "outputs": [],
   "source": [
    "model.load_state_dict(loaded_state_dict)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 18,
   "metadata": {},
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "('0.weight', Parameter containing:\n",
      "tensor([[[[-0.0016,  0.0048, -0.0146, -0.1456, -0.1541],\n",
      "          [-0.0630,  0.0308,  0.0080, -0.1073, -0.1134],\n",
      "          [-0.1058,  0.0090, -0.1061,  0.0478,  0.0654],\n",
      "          [ 0.0175,  0.0800,  0.0240,  0.0162,  0.0625],\n",
      "          [ 0.0913, -0.0833, -0.0338,  0.0798, -0.0163]],\n",
      "\n",
      "         [[ 0.0605,  0.0374,  0.0662, -0.1401, -0.0259],\n",
      "          [ 0.0589, -0.1049,  0.0120, -0.0084, -0.1260],\n",
      "          [-0.0768, -0.1128, -0.0596,  0.0139,  0.0571],\n",
      "          [ 0.0931, -0.0645, -0.0827,  0.0600,  0.0205],\n",
      "          [ 0.1170,  0.1265, -0.0124, -0.0439,  0.1113]],\n",
      "\n",
      "         [[-0.0642, -0.0756,  0.0292,  0.0497, -0.1144],\n",
      "          [ 0.0153, -0.0897, -0.0337,  0.0108, -0.1262],\n",
      "          [ 0.0991, -0.0447, -0.0442, -0.0280,  0.0873],\n",
      "          [-0.0269,  0.0825, -0.1120, -0.0491, -0.0136],\n",
      "          [ 0.0866, -0.0420,  0.1009,  0.1126,  0.0893]]],\n",
      "\n",
      "\n",
      "        [[[-0.0209,  0.0852,  0.0590, -0.0332, -0.0586],\n",
      "          [ 0.0518,  0.1475,  0.0091,  0.0288,  0.0954],\n",
      "          [ 0.0341,  0.0415, -0.0412,  0.0043,  0.1360],\n",
      "          [ 0.1538, -0.0249,  0.0003,  0.1572,  0.0010],\n",
      "          [ 0.0510,  0.0192,  0.1027,  0.1332,  0.0991]],\n",
      "\n",
      "         [[ 0.0177, -0.0105, -0.0383,  0.0657,  0.0859],\n",
      "          [-0.0725, -0.1320, -0.0927, -0.1090, -0.0635],\n",
      "          [ 0.0969, -0.0837, -0.1208,  0.0329, -0.0886],\n",
      "          [-0.0370,  0.0143, -0.0048, -0.0736, -0.0601],\n",
      "          [-0.0365, -0.0917, -0.0431, -0.0236, -0.0755]],\n",
      "\n",
      "         [[ 0.0102, -0.0525,  0.0135, -0.0179, -0.0234],\n",
      "          [ 0.0643, -0.0145,  0.1480,  0.0188,  0.0504],\n",
      "          [ 0.0169, -0.0020,  0.0158, -0.0133, -0.0076],\n",
      "          [ 0.1242, -0.0200,  0.0480,  0.1099,  0.0300],\n",
      "          [ 0.0524, -0.0310,  0.0239,  0.0870,  0.1372]]],\n",
      "\n",
      "\n",
      "        [[[-0.0898, -0.0185, -0.0380, -0.0599, -0.0783],\n",
      "          [ 0.0435, -0.0673, -0.0520, -0.0430,  0.0795],\n",
      "          [-0.0218, -0.0063,  0.0079,  0.0160,  0.0481],\n",
      "          [ 0.0058,  0.1550,  0.0125,  0.0755,  0.0940],\n",
      "          [ 0.1355,  0.0955,  0.1428,  0.1401,  0.1735]],\n",
      "\n",
      "         [[-0.1866, -0.1629,  0.0517, -0.0109, -0.1201],\n",
      "          [-0.1588, -0.1707, -0.1438,  0.0682,  0.0742],\n",
      "          [-0.0229, -0.1558,  0.0158, -0.1283,  0.0976],\n",
      "          [-0.1197, -0.0310, -0.0727,  0.0966, -0.0852],\n",
      "          [ 0.0389,  0.0312,  0.0191,  0.0596,  0.0384]],\n",
      "\n",
      "         [[ 0.0068,  0.0797, -0.0105,  0.0056,  0.1451],\n",
      "          [ 0.0565, -0.0742, -0.0368,  0.0914,  0.0443],\n",
      "          [-0.0046,  0.1062,  0.0461,  0.1649,  0.1579],\n",
      "          [ 0.1141, -0.0795,  0.1050, -0.0006,  0.0684],\n",
      "          [-0.0832, -0.0160,  0.0237,  0.1942,  0.1150]]],\n",
      "\n",
      "\n",
      "        ...,\n",
      "\n",
      "\n",
      "        [[[-0.0482, -0.0629, -0.1348, -0.0321, -0.1168],\n",
      "          [-0.0721, -0.0334, -0.0868, -0.0112,  0.0183],\n",
      "          [ 0.0814, -0.0759,  0.0558,  0.0165, -0.1443],\n",
      "          [ 0.0478, -0.1126,  0.0319,  0.0282,  0.0392],\n",
      "          [-0.0818,  0.0456,  0.0469,  0.0682, -0.0889]],\n",
      "\n",
      "         [[-0.0964, -0.0441, -0.0376, -0.1273,  0.0643],\n",
      "          [-0.0678,  0.0213, -0.0184,  0.0526,  0.0937],\n",
      "          [-0.0995, -0.0961,  0.0146,  0.0857,  0.0363],\n",
      "          [-0.0216, -0.1249,  0.0229, -0.0659, -0.0795],\n",
      "          [ 0.0812, -0.0560, -0.1054, -0.0096,  0.0889]],\n",
      "\n",
      "         [[-0.0897, -0.1021,  0.0180,  0.0298, -0.0659],\n",
      "          [ 0.0528, -0.1170, -0.0548,  0.0178, -0.0131],\n",
      "          [ 0.0153,  0.0072,  0.0386,  0.0121, -0.1135],\n",
      "          [-0.1317,  0.0072, -0.0720,  0.0915, -0.1055],\n",
      "          [-0.1206, -0.0260, -0.1168,  0.0285, -0.1064]]],\n",
      "\n",
      "\n",
      "        [[[ 0.0775,  0.0365,  0.0013,  0.0921,  0.1069],\n",
      "          [-0.0504, -0.0125, -0.0210,  0.0183,  0.0713],\n",
      "          [-0.0559, -0.0774, -0.1068, -0.0869, -0.0683],\n",
      "          [-0.0192,  0.0008,  0.0784,  0.0213, -0.0400],\n",
      "          [-0.0524,  0.0806, -0.0096,  0.0329, -0.0262]],\n",
      "\n",
      "         [[ 0.0902,  0.0546,  0.0144,  0.1076, -0.0733],\n",
      "          [ 0.0053, -0.0091, -0.0023, -0.0353, -0.0744],\n",
      "          [ 0.0175, -0.0493, -0.0767, -0.0778, -0.0176],\n",
      "          [ 0.0258,  0.0045,  0.0684,  0.0814,  0.0049],\n",
      "          [-0.0952,  0.0733,  0.0422,  0.0057, -0.0058]],\n",
      "\n",
      "         [[-0.0374,  0.0999,  0.0383,  0.0544,  0.1199],\n",
      "          [-0.1001, -0.0429,  0.1261, -0.0481, -0.0135],\n",
      "          [ 0.0285,  0.0622, -0.0188,  0.0707, -0.0995],\n",
      "          [-0.0672,  0.1120,  0.0732,  0.0780,  0.0502],\n",
      "          [-0.1165,  0.0694,  0.0052, -0.0275,  0.0033]]],\n",
      "\n",
      "\n",
      "        [[[-0.0018,  0.1250, -0.0596,  0.0056, -0.0844],\n",
      "          [ 0.0031, -0.0784, -0.0879,  0.1046,  0.0131],\n",
      "          [ 0.0248,  0.1182,  0.0623,  0.1107,  0.0702],\n",
      "          [-0.0020, -0.0631,  0.0097, -0.0636, -0.0109],\n",
      "          [-0.0106,  0.1198,  0.0468, -0.0776,  0.0931]],\n",
      "\n",
      "         [[ 0.0912, -0.0293, -0.1130, -0.1091, -0.1368],\n",
      "          [-0.0486, -0.0580, -0.0293,  0.0825,  0.0432],\n",
      "          [-0.0488, -0.0165, -0.0907,  0.1175, -0.1161],\n",
      "          [-0.0058,  0.0306, -0.0454,  0.0105,  0.1014],\n",
      "          [ 0.1089,  0.0913,  0.0672,  0.0102,  0.0053]],\n",
      "\n",
      "         [[-0.0369,  0.1566,  0.0700,  0.0545,  0.0613],\n",
      "          [ 0.0877,  0.0052,  0.0940,  0.1503,  0.0434],\n",
      "          [-0.0281,  0.0453,  0.0012,  0.0196, -0.0414],\n",
      "          [ 0.0855, -0.0163,  0.1347,  0.0424,  0.0502],\n",
      "          [ 0.0874,  0.0372,  0.1193,  0.1401,  0.1866]]]]))\n",
      "('0.bias', Parameter containing:\n",
      "tensor([-0.0074, -0.0474,  0.0857,  0.1327,  0.0287,  0.1414,  0.1547,\n",
      "         0.0509, -0.0440, -0.0592,  0.0788, -0.0021,  0.0642,  0.1670,\n",
      "         0.0317,  0.0466,  0.1044, -0.0274, -0.0708,  0.0774,  0.0230,\n",
      "         0.0278,  0.1082,  0.1173,  0.0120, -0.0614,  0.0690,  0.0154,\n",
      "        -0.0326,  0.0599, -0.0667,  0.0317, -0.0404,  0.1326, -0.0984,\n",
      "         0.0518, -0.0816, -0.0412, -0.0766,  0.0903, -0.0249, -0.0253,\n",
      "         0.2320, -0.0652, -0.0272,  0.0391,  0.0159, -0.0892, -0.0242,\n",
      "        -0.0119, -0.0370,  0.1231, -0.0702, -0.0439, -0.0870,  0.0212,\n",
      "         0.1375, -0.1098,  0.0907,  0.0009,  0.0788,  0.0816, -0.1044,\n",
      "         0.0091]))\n",
      "('3.weight', Parameter containing:\n",
      "tensor([[[[-2.6779e-02, -1.2962e-02, -6.7731e-03,  ..., -2.1328e-02,\n",
      "           -1.5803e-02, -7.1797e-03],\n",
      "          [-1.8022e-02, -2.3607e-03, -2.2527e-02,  ..., -3.2269e-02,\n",
      "            1.1309e-02,  7.9132e-03],\n",
      "          [-1.8926e-02, -1.6236e-02, -1.8413e-02,  ...,  9.0630e-03,\n",
      "            9.2606e-03,  1.7090e-02],\n",
      "          ...,\n",
      "          [-3.7086e-03, -7.2559e-04,  1.0614e-02,  ...,  6.1826e-04,\n",
      "           -1.2080e-02,  4.3929e-04],\n",
      "          [-2.3554e-02, -1.1957e-02, -6.5553e-03,  ..., -1.1912e-02,\n",
      "           -1.3815e-02, -2.2450e-03],\n",
      "          [-2.0905e-02, -2.5623e-02, -3.3727e-02,  ..., -8.8794e-03,\n",
      "            2.5970e-03,  8.4659e-04]],\n",
      "\n",
      "         [[-1.0403e-02,  9.8815e-03,  1.3379e-02,  ...,  1.6973e-02,\n",
      "           -7.1956e-03,  4.0096e-03],\n",
      "          [ 1.1212e-02, -1.4063e-02,  1.7008e-02,  ..., -6.0592e-03,\n",
      "           -1.2660e-02, -1.1657e-02],\n",
      "          [-3.2382e-02, -4.0698e-03,  6.5229e-03,  ..., -3.1508e-03,\n",
      "           -3.7202e-02, -1.1686e-02],\n",
      "          ...,\n",
      "          [-9.1540e-03,  1.0237e-02,  1.6859e-02,  ...,  8.6157e-03,\n",
      "           -3.7181e-03,  9.6950e-03],\n",
      "          [-1.7268e-02,  1.4214e-02, -1.1315e-02,  ..., -8.8661e-03,\n",
      "           -1.2371e-02,  6.7527e-03],\n",
      "          [ 9.9682e-03, -1.3675e-02, -6.5624e-03,  ...,  3.1749e-03,\n",
      "            9.4326e-03, -2.9477e-03]],\n",
      "\n",
      "         [[-7.0483e-03, -3.1922e-02, -2.1498e-02,  ..., -3.1507e-03,\n",
      "           -3.0696e-02, -2.8737e-03],\n",
      "          [-6.8231e-03, -1.3644e-02, -1.6670e-03,  ..., -2.8231e-02,\n",
      "           -3.1186e-02, -2.6410e-02],\n",
      "          [-9.8460e-03, -3.6508e-02, -1.1149e-02,  ..., -1.7134e-02,\n",
      "           -3.2647e-02, -2.3700e-02],\n",
      "          ...,\n",
      "          [-5.7745e-04,  9.0411e-04, -1.4712e-02,  ...,  1.3643e-02,\n",
      "           -1.8622e-03, -1.3857e-02],\n",
      "          [-3.7543e-02,  1.7850e-02, -2.0641e-03,  ...,  5.8658e-03,\n",
      "            4.6741e-03, -8.6243e-03],\n",
      "          [-3.3316e-02, -4.9936e-03, -1.4988e-02,  ..., -8.6911e-05,\n",
      "           -1.7218e-02, -2.7561e-03]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[ 1.5939e-02, -1.4361e-02,  1.4564e-02,  ..., -8.7929e-03,\n",
      "           -1.2084e-02,  1.2491e-02],\n",
      "          [ 1.7160e-02, -6.9554e-03,  1.0544e-02,  ..., -1.0389e-02,\n",
      "           -4.2872e-03, -4.1053e-03],\n",
      "          [-2.0027e-03,  1.0370e-02,  1.7050e-02,  ..., -6.4324e-03,\n",
      "            1.6785e-02,  3.4043e-03],\n",
      "          ...,\n",
      "          [ 1.3323e-02, -3.4984e-03, -1.5882e-02,  ..., -2.4456e-03,\n",
      "            2.4521e-03,  1.6473e-02],\n",
      "          [-1.7732e-02,  1.5460e-02,  6.0378e-03,  ..., -1.2608e-02,\n",
      "            1.7068e-02,  1.6477e-02],\n",
      "          [ 7.6953e-03,  9.7408e-03, -5.5379e-03,  ..., -1.6686e-03,\n",
      "           -1.5905e-03,  1.2670e-02]],\n",
      "\n",
      "         [[-2.7136e-02, -3.4884e-02, -3.1536e-02,  ...,  8.7532e-03,\n",
      "            1.7536e-02, -8.5740e-03],\n",
      "          [-3.2289e-02, -2.1074e-02, -2.7792e-03,  ...,  1.3592e-03,\n",
      "            6.7092e-03,  8.8825e-03],\n",
      "          [-2.4370e-03, -2.1794e-02, -1.4332e-02,  ..., -2.6290e-03,\n",
      "           -9.7182e-04, -1.3434e-02],\n",
      "          ...,\n",
      "          [-2.7251e-02, -5.9368e-03, -1.7157e-02,  ...,  4.8391e-03,\n",
      "            1.3846e-02, -2.5880e-02],\n",
      "          [-8.4309e-03, -9.7086e-03, -8.2329e-03,  ..., -1.1860e-02,\n",
      "           -1.2285e-02,  6.6090e-03],\n",
      "          [-2.8973e-03,  9.6497e-03,  2.2892e-03,  ...,  1.2426e-02,\n",
      "           -9.8844e-03, -2.6804e-02]],\n",
      "\n",
      "         [[-1.7853e-03,  1.4545e-02,  1.3259e-02,  ...,  6.5922e-03,\n",
      "           -5.1442e-03, -1.0759e-02],\n",
      "          [ 1.4357e-02,  4.0669e-03,  1.0755e-02,  ..., -1.3151e-02,\n",
      "            1.4049e-02,  1.1277e-03],\n",
      "          [ 9.0401e-03,  1.0939e-02, -1.3673e-02,  ..., -8.7622e-03,\n",
      "            5.9570e-04, -1.2833e-02],\n",
      "          ...,\n",
      "          [ 2.0013e-03,  6.9983e-03,  4.6257e-03,  ..., -8.0778e-03,\n",
      "            1.2425e-02, -9.2927e-03],\n",
      "          [ 1.3665e-02, -1.7009e-02,  3.3687e-03,  ..., -4.1017e-03,\n",
      "           -1.0301e-02,  1.4668e-02],\n",
      "          [ 5.8163e-03, -2.8335e-06,  9.8780e-04,  ...,  1.0928e-02,\n",
      "           -1.1426e-02, -1.3808e-02]]],\n",
      "\n",
      "\n",
      "        [[[-2.8145e-02, -1.8742e-02, -5.5234e-03,  ...,  4.0994e-03,\n",
      "            2.8125e-04, -3.5711e-02],\n",
      "          [-1.9029e-02, -3.9364e-03, -6.7715e-03,  ..., -9.1552e-03,\n",
      "            1.5949e-02,  4.7563e-03],\n",
      "          [-7.7093e-03, -3.4516e-02, -2.2431e-02,  ..., -3.2317e-02,\n",
      "            9.8823e-03,  1.2922e-03],\n",
      "          ...,\n",
      "          [-2.9283e-02, -2.6186e-02, -1.6897e-02,  ..., -4.7393e-03,\n",
      "           -3.1830e-02, -1.0085e-02],\n",
      "          [-3.5755e-02, -3.4435e-02, -1.2168e-02,  ..., -2.0105e-02,\n",
      "           -4.2665e-03, -2.0468e-02],\n",
      "          [-3.3734e-02, -1.6996e-02, -9.8357e-03,  ..., -2.9043e-02,\n",
      "           -3.3291e-02, -1.3980e-02]],\n",
      "\n",
      "         [[-6.3367e-03, -4.0852e-03, -1.3327e-02,  ..., -6.2933e-03,\n",
      "            1.3628e-02,  8.6691e-03],\n",
      "          [ 1.3802e-02, -7.2884e-03, -1.4474e-02,  ...,  1.3953e-02,\n",
      "           -5.3896e-03,  9.0456e-03],\n",
      "          [-6.9317e-03,  1.6150e-02,  8.1835e-03,  ..., -3.0702e-04,\n",
      "            1.3830e-02, -1.6633e-03],\n",
      "          ...,\n",
      "          [-8.3356e-04, -9.3114e-03,  7.4694e-03,  ..., -1.2985e-02,\n",
      "           -1.4350e-02,  4.2311e-03],\n",
      "          [-5.0927e-03,  1.0877e-02,  4.0688e-03,  ...,  9.2379e-03,\n",
      "           -1.0290e-02,  1.0049e-02],\n",
      "          [-9.4879e-03,  1.0646e-02,  1.3236e-02,  ..., -9.2430e-03,\n",
      "           -1.0380e-03, -1.4155e-02]],\n",
      "\n",
      "         [[-3.6744e-02, -2.1535e-02, -3.1799e-02,  ..., -1.4108e-02,\n",
      "            2.2022e-03,  1.5080e-02],\n",
      "          [-2.9610e-02, -4.6011e-03, -1.4337e-02,  ..., -8.4109e-03,\n",
      "            4.4273e-03, -2.2268e-02],\n",
      "          [-1.1714e-02, -5.3899e-03, -1.1769e-02,  ..., -2.0215e-02,\n",
      "           -4.4180e-03, -9.1483e-03],\n",
      "          ...,\n",
      "          [-3.1818e-02, -2.7536e-02, -1.3271e-02,  ..., -6.5540e-03,\n",
      "           -7.9542e-03, -1.0677e-02],\n",
      "          [-9.0505e-03, -3.1940e-02, -1.5119e-02,  ..., -2.1554e-02,\n",
      "           -1.0521e-02, -2.7976e-02],\n",
      "          [-2.0902e-02, -3.2283e-03, -1.3045e-02,  ..., -3.6176e-02,\n",
      "           -2.9619e-02, -1.6289e-02]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[-1.4943e-02,  1.2392e-03,  1.6166e-02,  ..., -1.1477e-02,\n",
      "           -1.5734e-02, -1.2199e-02],\n",
      "          [ 1.1283e-02,  1.1453e-03,  1.2160e-02,  ...,  6.1564e-03,\n",
      "            3.5162e-03, -8.8078e-03],\n",
      "          [ 1.5972e-02,  1.8743e-03, -1.4592e-02,  ...,  7.7799e-03,\n",
      "           -4.6349e-03,  3.2799e-03],\n",
      "          ...,\n",
      "          [ 8.4355e-03, -4.7946e-04, -1.7618e-02,  ..., -1.6736e-02,\n",
      "           -6.3067e-03,  6.9063e-03],\n",
      "          [ 1.0057e-02,  5.4642e-03,  1.1619e-02,  ...,  1.4251e-02,\n",
      "           -8.2950e-03,  1.3677e-02],\n",
      "          [ 3.1885e-03, -1.5601e-05,  1.7460e-02,  ..., -1.1777e-02,\n",
      "           -6.7834e-03,  3.3414e-03]],\n",
      "\n",
      "         [[-3.0452e-02, -1.2080e-02,  1.0256e-02,  ..., -3.0990e-02,\n",
      "            8.6174e-03,  1.6501e-03],\n",
      "          [-5.2719e-04, -3.2968e-02, -2.6674e-02,  ..., -1.6440e-02,\n",
      "           -3.6365e-03, -3.2362e-02],\n",
      "          [-2.7233e-02, -2.7331e-02, -3.5724e-02,  ..., -9.7884e-03,\n",
      "           -3.2167e-02, -9.0837e-03],\n",
      "          ...,\n",
      "          [-2.0292e-02, -3.3341e-03, -7.6577e-03,  ...,  1.4283e-02,\n",
      "           -1.4460e-03, -1.3133e-02],\n",
      "          [-3.3671e-02, -1.4031e-02, -7.2689e-03,  ...,  1.6328e-02,\n",
      "           -7.3200e-03, -2.9146e-02],\n",
      "          [-2.9474e-02, -2.9688e-02, -1.0572e-02,  ..., -3.2480e-02,\n",
      "           -1.3261e-02, -1.5741e-02]],\n",
      "\n",
      "         [[-2.3804e-03,  5.5173e-03, -1.2740e-02,  ..., -4.8662e-03,\n",
      "           -7.2586e-03,  1.0922e-02],\n",
      "          [-6.3082e-03, -9.7869e-03, -1.6668e-02,  ..., -1.0346e-02,\n",
      "           -7.1375e-04, -1.4136e-02],\n",
      "          [-9.3555e-03,  5.5218e-03,  1.6871e-02,  ..., -7.9774e-03,\n",
      "            1.0834e-02,  6.2287e-03],\n",
      "          ...,\n",
      "          [-1.0119e-02, -1.4812e-03, -2.6835e-03,  ...,  5.7517e-03,\n",
      "            1.2835e-02,  8.3581e-03],\n",
      "          [-1.7669e-02, -4.3534e-04, -1.3433e-02,  ...,  1.3524e-02,\n",
      "            5.1122e-04,  1.2765e-02],\n",
      "          [-1.2379e-02, -4.0298e-03,  1.3297e-02,  ..., -9.6577e-03,\n",
      "           -8.4847e-03, -1.0439e-02]]],\n",
      "\n",
      "\n",
      "        [[[-2.2538e-03, -4.1233e-03, -1.0215e-02,  ..., -2.4550e-03,\n",
      "           -2.0032e-02, -2.6995e-02],\n",
      "          [-3.4619e-02, -1.8392e-02, -2.5360e-02,  ..., -1.4965e-02,\n",
      "           -1.0830e-02, -3.1260e-02],\n",
      "          [-1.0557e-02, -2.3744e-02,  1.6909e-03,  ..., -6.7882e-03,\n",
      "            1.5083e-03, -1.6190e-02],\n",
      "          ...,\n",
      "          [-1.7339e-03,  6.6879e-03, -9.7566e-03,  ...,  1.1324e-02,\n",
      "           -2.9404e-02,  1.9570e-02],\n",
      "          [ 1.0242e-02, -9.2871e-03, -9.3168e-03,  ..., -1.3156e-02,\n",
      "           -1.4865e-02, -1.1270e-02],\n",
      "          [ 1.9390e-02, -8.4842e-03,  9.8267e-03,  ...,  4.0320e-03,\n",
      "           -2.6696e-02, -1.6334e-02]],\n",
      "\n",
      "         [[-9.2508e-03, -1.1350e-02, -1.7665e-02,  ...,  4.3904e-03,\n",
      "            2.7492e-03, -5.2415e-03],\n",
      "          [-8.4670e-03, -7.3156e-03, -1.9307e-02,  ..., -2.3375e-02,\n",
      "           -1.0639e-02, -1.7652e-02],\n",
      "          [-2.3184e-02, -1.7577e-02, -6.0105e-03,  ...,  4.4489e-03,\n",
      "           -2.6661e-02,  3.8427e-03],\n",
      "          ...,\n",
      "          [-7.8044e-03, -1.7909e-02,  2.7993e-03,  ..., -1.1152e-03,\n",
      "            7.6839e-03, -1.1714e-02],\n",
      "          [-9.9899e-03,  2.4783e-03, -3.1628e-03,  ..., -1.6662e-02,\n",
      "           -1.4662e-02,  6.5154e-03],\n",
      "          [-1.4905e-02, -1.5384e-02, -1.1265e-02,  ..., -4.1862e-03,\n",
      "           -1.9127e-03, -2.4848e-02]],\n",
      "\n",
      "         [[ 3.4846e-03,  8.0098e-03,  1.7443e-02,  ...,  1.2480e-02,\n",
      "           -1.2735e-02,  8.6934e-03],\n",
      "          [-8.2303e-03, -1.7087e-02, -1.0395e-02,  ..., -1.6124e-02,\n",
      "            7.0822e-03, -3.8555e-04],\n",
      "          [-5.2301e-03,  4.9656e-03,  9.1322e-03,  ..., -3.9297e-03,\n",
      "           -1.1945e-02, -3.1185e-03],\n",
      "          ...,\n",
      "          [ 1.0450e-02,  1.8954e-02,  2.9704e-03,  ...,  1.4643e-02,\n",
      "            1.6012e-02, -3.2929e-03],\n",
      "          [ 1.0485e-02,  7.6941e-03,  7.7856e-03,  ...,  6.2871e-03,\n",
      "            7.3308e-03,  1.3855e-02],\n",
      "          [-1.3529e-02, -2.3717e-03, -1.4512e-02,  ...,  1.0026e-02,\n",
      "           -1.6163e-02, -5.7107e-03]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[-1.4811e-02, -2.7844e-02, -2.0868e-02,  ..., -2.6010e-02,\n",
      "           -2.7611e-02,  6.3969e-03],\n",
      "          [-2.2012e-03, -1.9145e-02, -1.9199e-02,  ..., -2.0287e-02,\n",
      "           -2.2382e-02, -2.2633e-03],\n",
      "          [-1.0251e-02, -1.5751e-02, -2.7219e-02,  ...,  2.9438e-03,\n",
      "           -2.6035e-02, -2.6711e-02],\n",
      "          ...,\n",
      "          [-3.0947e-04,  5.9995e-03,  2.1770e-05,  ..., -1.0780e-02,\n",
      "            4.2132e-03,  3.1212e-03],\n",
      "          [-1.9642e-02, -1.5439e-02, -1.3787e-02,  ..., -2.8270e-02,\n",
      "           -8.3740e-03, -6.7641e-03],\n",
      "          [-6.7095e-03, -1.8616e-02, -2.7250e-02,  ..., -2.8718e-02,\n",
      "           -2.0472e-02,  4.0441e-04]],\n",
      "\n",
      "         [[-2.8262e-02, -7.3320e-03, -1.9765e-02,  ..., -8.0278e-03,\n",
      "            1.7557e-02,  1.8543e-02],\n",
      "          [-1.7225e-02, -1.3504e-02, -2.6834e-02,  ..., -2.0847e-02,\n",
      "           -3.4311e-03,  8.1041e-03],\n",
      "          [-2.3213e-02,  8.8160e-03,  1.5753e-02,  ..., -1.5599e-02,\n",
      "           -2.2783e-02,  1.7687e-03],\n",
      "          ...,\n",
      "          [-1.2956e-02, -3.4006e-02, -1.7315e-02,  ..., -6.5962e-03,\n",
      "           -3.2383e-02,  1.5012e-02],\n",
      "          [ 1.8599e-02, -1.7383e-02, -1.5422e-02,  ..., -9.3824e-03,\n",
      "           -1.5369e-02, -2.1481e-02],\n",
      "          [-3.9189e-04, -1.3674e-02, -2.4964e-02,  ..., -9.4626e-03,\n",
      "           -1.0716e-02, -1.9111e-03]],\n",
      "\n",
      "         [[-2.4925e-02, -1.9391e-03,  1.8677e-03,  ..., -7.9364e-03,\n",
      "           -1.1269e-02, -1.0375e-02],\n",
      "          [-2.2959e-02,  1.3838e-03, -5.1879e-03,  ..., -2.0389e-02,\n",
      "            1.8722e-04, -2.2228e-02],\n",
      "          [-2.5972e-02, -1.9240e-02, -1.4870e-02,  ..., -1.1329e-02,\n",
      "           -1.0629e-02, -1.2778e-03],\n",
      "          ...,\n",
      "          [ 4.6231e-03, -9.8229e-03, -1.6067e-02,  ..., -1.4338e-02,\n",
      "            1.0771e-03, -1.4400e-02],\n",
      "          [-1.5673e-02, -4.5755e-03, -1.2823e-02,  ..., -1.7801e-02,\n",
      "            2.3182e-03, -1.8283e-02],\n",
      "          [ 2.1563e-03, -5.6639e-03,  6.3025e-03,  ..., -1.9270e-02,\n",
      "           -2.7177e-02,  1.7473e-04]]],\n",
      "\n",
      "\n",
      "        ...,\n",
      "\n",
      "\n",
      "        [[[ 7.6536e-03, -1.8783e-03,  1.8841e-02,  ..., -8.4884e-03,\n",
      "           -2.2051e-03, -5.5705e-03],\n",
      "          [-1.1814e-02, -1.4657e-02,  1.1615e-02,  ..., -3.1463e-03,\n",
      "           -3.4147e-02, -2.1097e-02],\n",
      "          [ 1.7014e-02, -1.1913e-03, -1.3618e-02,  ..., -2.0940e-02,\n",
      "           -1.9440e-02, -3.0029e-02],\n",
      "          ...,\n",
      "          [ 1.0215e-02,  1.5527e-02, -6.3834e-03,  ...,  6.0160e-03,\n",
      "           -1.6559e-02, -2.9295e-03],\n",
      "          [ 1.0632e-02, -1.1926e-02, -1.3686e-03,  ...,  1.6902e-02,\n",
      "           -1.0786e-02, -2.5942e-02],\n",
      "          [ 3.9939e-03, -3.3778e-02, -1.2943e-02,  ...,  1.9375e-03,\n",
      "           -2.5962e-02, -2.6246e-02]],\n",
      "\n",
      "         [[-1.8836e-02, -1.2895e-02, -2.8608e-02,  ..., -1.1196e-02,\n",
      "           -1.9619e-02, -3.0789e-02],\n",
      "          [-3.1237e-02, -3.7592e-02, -8.0640e-03,  ..., -2.0608e-02,\n",
      "           -7.0762e-03, -3.3432e-02],\n",
      "          [-3.4544e-02, -3.1376e-02, -1.8247e-02,  ..., -4.2953e-03,\n",
      "           -2.5268e-02, -3.7524e-02],\n",
      "          ...,\n",
      "          [-2.0865e-02, -2.8106e-02, -4.8804e-03,  ...,  4.9353e-03,\n",
      "           -9.5544e-03, -3.9235e-04],\n",
      "          [-2.0468e-02, -4.8933e-03, -2.7534e-02,  ..., -1.2462e-02,\n",
      "           -2.6990e-02, -7.4737e-03],\n",
      "          [-2.3287e-02, -2.4754e-02, -2.8761e-02,  ..., -1.5842e-02,\n",
      "           -2.1052e-02, -3.2656e-02]],\n",
      "\n",
      "         [[ 8.0048e-04,  2.2151e-03,  1.5473e-02,  ..., -1.2530e-02,\n",
      "           -1.6299e-02, -3.9971e-03],\n",
      "          [-1.7649e-02,  6.1190e-03,  5.3151e-03,  ..., -2.9324e-02,\n",
      "           -1.8867e-02, -3.1613e-02],\n",
      "          [-3.7128e-02, -3.5173e-02, -3.1409e-02,  ..., -2.0030e-02,\n",
      "           -3.2676e-03, -3.1105e-02],\n",
      "          ...,\n",
      "          [-2.1572e-02, -3.5458e-02, -2.1744e-02,  ..., -4.4794e-03,\n",
      "           -4.9605e-03, -2.9795e-02],\n",
      "          [-1.9600e-02, -1.8060e-02, -2.9290e-02,  ..., -1.0334e-02,\n",
      "           -2.3278e-02, -2.6473e-02],\n",
      "          [-3.4211e-02, -5.7403e-03, -3.1517e-02,  ..., -2.5251e-02,\n",
      "           -2.8706e-02, -3.5840e-02]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[ 1.1339e-02, -1.7595e-02,  3.8017e-03,  ...,  2.5522e-03,\n",
      "           -1.7517e-02,  4.3034e-03],\n",
      "          [ 9.7192e-03,  1.5814e-02,  5.0818e-03,  ...,  1.2481e-02,\n",
      "           -3.0841e-02,  3.4227e-03],\n",
      "          [-1.1800e-02,  4.6344e-03,  2.4301e-03,  ..., -3.6808e-02,\n",
      "           -3.3497e-02, -2.8690e-02],\n",
      "          ...,\n",
      "          [ 3.6129e-03, -1.1561e-02, -3.0395e-02,  ..., -1.2388e-02,\n",
      "           -9.4928e-03, -1.5173e-03],\n",
      "          [-1.1360e-02, -4.0666e-03, -2.0922e-02,  ...,  8.6673e-04,\n",
      "           -1.3469e-02, -4.8741e-03],\n",
      "          [ 5.7896e-03, -1.7382e-02, -7.8392e-03,  ..., -2.3734e-03,\n",
      "           -1.1630e-02, -7.0134e-03]],\n",
      "\n",
      "         [[-3.2164e-02, -6.2788e-03,  1.3968e-02,  ...,  3.6180e-03,\n",
      "           -1.0840e-02, -2.7565e-02],\n",
      "          [-2.6980e-02, -2.4073e-02,  1.0033e-03,  ..., -2.5264e-04,\n",
      "           -5.5285e-03, -1.2939e-02],\n",
      "          [ 2.1923e-04, -1.4998e-03, -2.6163e-02,  ..., -4.7604e-03,\n",
      "            9.5060e-03,  2.7361e-03],\n",
      "          ...,\n",
      "          [-1.5853e-02,  1.8252e-02, -1.6496e-02,  ..., -3.9139e-03,\n",
      "           -2.4966e-02, -3.1966e-03],\n",
      "          [-2.7687e-02, -2.8252e-02,  1.8075e-03,  ..., -2.1508e-02,\n",
      "            3.0193e-03, -1.8301e-02],\n",
      "          [ 3.9693e-03,  1.4852e-02,  6.8361e-03,  ..., -2.5844e-02,\n",
      "           -1.3109e-02, -2.7742e-02]],\n",
      "\n",
      "         [[ 1.9207e-03, -1.1992e-02,  1.2561e-02,  ..., -2.8050e-02,\n",
      "           -2.8156e-02, -3.3904e-02],\n",
      "          [ 1.7325e-02,  1.1727e-02,  4.4550e-03,  ..., -2.3177e-02,\n",
      "           -3.0512e-02, -1.1348e-02],\n",
      "          [ 1.2288e-02,  3.4642e-04, -2.0631e-02,  ..., -9.7264e-03,\n",
      "           -1.9462e-02, -9.1786e-03],\n",
      "          ...,\n",
      "          [-1.4261e-02, -1.8814e-02, -9.2540e-03,  ..., -6.0748e-03,\n",
      "           -1.2992e-02,  4.8125e-04],\n",
      "          [-1.3841e-02, -2.1739e-02, -1.6609e-02,  ..., -3.4722e-03,\n",
      "           -3.2457e-03,  1.4795e-02],\n",
      "          [-4.3364e-03, -8.1214e-03, -1.2953e-02,  ..., -9.9520e-03,\n",
      "            3.6117e-03, -7.1621e-03]]],\n",
      "\n",
      "\n",
      "        [[[-2.1822e-02, -1.6658e-02, -7.6389e-03,  ..., -2.2257e-02,\n",
      "            7.1451e-03, -1.5845e-03],\n",
      "          [ 1.1925e-03, -3.5487e-03, -1.8371e-02,  ...,  1.0235e-02,\n",
      "           -2.0721e-02,  7.1239e-03],\n",
      "          [-1.4366e-02,  7.3110e-03, -9.1615e-04,  ..., -5.8821e-03,\n",
      "            1.5872e-02,  2.1394e-02],\n",
      "          ...,\n",
      "          [-3.5594e-03,  6.8701e-03, -1.0854e-02,  ..., -2.3067e-02,\n",
      "           -3.2157e-03, -1.1290e-02],\n",
      "          [ 2.6806e-02,  3.3598e-04, -1.0921e-02,  ..., -1.1420e-02,\n",
      "            4.6863e-03,  6.0898e-03],\n",
      "          [ 3.0626e-03, -1.7101e-02, -5.4114e-03,  ..., -1.0365e-02,\n",
      "           -1.0822e-02, -2.6236e-02]],\n",
      "\n",
      "         [[-1.3101e-02, -1.3059e-02,  7.6095e-03,  ..., -1.1628e-02,\n",
      "           -2.6642e-02, -6.9455e-03],\n",
      "          [-2.5564e-02, -1.5327e-02,  4.0728e-03,  ..., -2.5864e-02,\n",
      "           -2.2147e-02,  4.5925e-03],\n",
      "          [-8.2005e-03,  5.9101e-03, -6.4722e-03,  ..., -1.4526e-02,\n",
      "           -4.1891e-03, -1.5764e-02],\n",
      "          ...,\n",
      "          [-2.6733e-02, -2.7073e-02, -1.7734e-02,  ..., -1.5771e-02,\n",
      "           -8.3647e-04,  5.2966e-03],\n",
      "          [ 7.4729e-03, -2.3610e-02, -1.9502e-02,  ..., -1.5676e-02,\n",
      "            1.1866e-03,  4.3461e-04],\n",
      "          [-2.7914e-03, -3.7813e-03, -2.0831e-02,  ..., -1.0264e-02,\n",
      "            8.9520e-04, -2.4958e-02]],\n",
      "\n",
      "         [[-6.4742e-03, -1.8906e-02, -5.6844e-03,  ...,  7.3853e-03,\n",
      "            2.2777e-03, -6.9292e-03],\n",
      "          [ 2.8477e-03,  2.0978e-03,  3.7272e-03,  ..., -7.4928e-03,\n",
      "            1.6647e-02,  1.2275e-02],\n",
      "          [ 1.1717e-02, -4.8451e-04,  2.7634e-02,  ..., -6.2747e-03,\n",
      "           -7.3154e-03,  4.8498e-04],\n",
      "          ...,\n",
      "          [ 1.0360e-02,  2.0931e-02,  1.1289e-02,  ...,  3.9727e-03,\n",
      "           -1.4086e-02,  1.7404e-02],\n",
      "          [-2.5335e-03, -2.7018e-03,  1.6044e-02,  ..., -3.9768e-03,\n",
      "           -1.3957e-02, -1.4771e-02],\n",
      "          [ 1.6544e-02,  2.1240e-02,  1.7475e-02,  ..., -2.5158e-02,\n",
      "           -1.9777e-02, -1.7363e-04]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[ 1.4401e-02,  1.2646e-03, -1.3304e-02,  ...,  7.7538e-03,\n",
      "           -1.9440e-02, -2.1017e-02],\n",
      "          [ 1.6771e-02, -2.3978e-04, -5.0331e-03,  ..., -7.6139e-03,\n",
      "           -2.1629e-04,  6.4135e-03],\n",
      "          [ 6.8014e-03,  9.8460e-03, -7.2331e-03,  ..., -7.9818e-03,\n",
      "            3.1137e-03, -2.2293e-02],\n",
      "          ...,\n",
      "          [-1.4249e-02,  2.2670e-03, -2.0135e-02,  ..., -1.9543e-04,\n",
      "            9.2510e-04, -3.7952e-03],\n",
      "          [ 2.5527e-02,  3.9357e-03, -2.0599e-02,  ..., -1.2822e-02,\n",
      "           -2.0320e-02,  5.3710e-03],\n",
      "          [ 2.4788e-02, -5.9360e-03, -1.9608e-02,  ..., -8.5284e-03,\n",
      "           -5.3702e-03,  2.7787e-03]],\n",
      "\n",
      "         [[-1.3842e-02,  9.3919e-04,  1.0927e-02,  ..., -2.3249e-02,\n",
      "            2.1997e-02, -2.7843e-03],\n",
      "          [-1.8800e-02,  3.3210e-03, -5.7883e-03,  ..., -3.2548e-02,\n",
      "            1.0355e-02, -7.5103e-05],\n",
      "          [ 8.8634e-04, -2.7359e-02,  9.0961e-04,  ..., -8.1291e-03,\n",
      "           -3.1408e-04, -5.1685e-03],\n",
      "          ...,\n",
      "          [-2.1556e-02,  7.1239e-04, -2.7186e-02,  ..., -2.6574e-02,\n",
      "           -1.7427e-02, -7.3701e-03],\n",
      "          [ 1.7685e-02,  1.9347e-03, -2.5782e-03,  ..., -1.3726e-02,\n",
      "           -1.9573e-02, -3.1529e-05],\n",
      "          [ 1.6412e-02, -2.5740e-02,  1.9379e-03,  ..., -2.6090e-02,\n",
      "           -7.1123e-04,  1.2818e-02]],\n",
      "\n",
      "         [[-2.5872e-02, -1.1990e-02, -1.4029e-02,  ...,  4.0431e-03,\n",
      "            9.0229e-04, -1.4076e-02],\n",
      "          [-1.9221e-02, -2.4752e-02, -1.5499e-02,  ..., -7.9520e-03,\n",
      "           -7.1294e-03, -4.3111e-03],\n",
      "          [ 3.4378e-03, -1.3511e-02, -1.3479e-02,  ..., -1.0406e-02,\n",
      "           -9.2696e-03, -5.8095e-03],\n",
      "          ...,\n",
      "          [ 5.9763e-03, -6.7900e-04, -2.3468e-02,  ...,  7.2678e-03,\n",
      "           -2.4124e-02,  5.2814e-03],\n",
      "          [-2.6482e-02, -1.1136e-02, -2.3210e-02,  ..., -2.3877e-03,\n",
      "           -9.3244e-03, -1.3438e-02],\n",
      "          [-1.4087e-02,  8.4801e-04,  4.2950e-03,  ..., -2.6069e-02,\n",
      "            5.7832e-03, -1.9748e-02]]],\n",
      "\n",
      "\n",
      "        [[[-1.2947e-02, -1.3831e-02, -6.6514e-03,  ...,  1.1251e-02,\n",
      "           -2.7273e-03, -2.9318e-04],\n",
      "          [ 1.6319e-02, -4.2163e-03,  9.5971e-04,  ...,  8.7225e-03,\n",
      "            3.4388e-03, -4.2100e-03],\n",
      "          [-8.9748e-03,  6.1063e-03, -1.0801e-03,  ...,  5.1986e-03,\n",
      "            8.0231e-03, -2.8249e-02],\n",
      "          ...,\n",
      "          [-3.6449e-03, -2.3086e-03, -2.5649e-02,  ..., -1.1459e-02,\n",
      "           -5.9115e-03,  5.8053e-03],\n",
      "          [-3.4264e-02, -3.3125e-02, -3.9721e-03,  ..., -1.2224e-02,\n",
      "            1.9460e-02, -5.8133e-03],\n",
      "          [-3.4569e-02, -3.0598e-02, -3.6949e-02,  ...,  5.6422e-03,\n",
      "           -3.6649e-03,  8.9027e-05]],\n",
      "\n",
      "         [[-2.5274e-02, -2.2699e-02, -2.4185e-02,  ..., -1.9850e-02,\n",
      "           -1.6500e-02, -2.5491e-02],\n",
      "          [-2.0654e-03, -1.8466e-02, -1.2426e-02,  ...,  1.5545e-03,\n",
      "           -1.6919e-02, -6.4482e-03],\n",
      "          [-2.1573e-03,  3.2694e-03, -8.2869e-04,  ..., -1.9123e-02,\n",
      "           -3.0052e-02, -6.3218e-03],\n",
      "          ...,\n",
      "          [-2.0544e-02, -1.8392e-02, -1.7850e-02,  ..., -1.2365e-02,\n",
      "           -1.7570e-02, -2.4573e-02],\n",
      "          [-3.4599e-02, -1.4640e-02, -3.5621e-02,  ..., -2.3353e-02,\n",
      "           -1.7889e-02, -3.3007e-02],\n",
      "          [-2.4938e-02, -5.0069e-03, -2.1860e-02,  ..., -2.3937e-02,\n",
      "           -1.6531e-02, -2.4050e-02]],\n",
      "\n",
      "         [[ 1.8155e-02, -3.6595e-04,  2.9583e-03,  ..., -1.4410e-02,\n",
      "            1.0625e-02,  1.4375e-03],\n",
      "          [-4.8929e-03, -8.0628e-03,  1.1450e-02,  ..., -1.5520e-02,\n",
      "            1.3821e-02, -1.5309e-02],\n",
      "          [ 1.1759e-02, -3.2002e-03, -1.5472e-02,  ..., -9.7236e-03,\n",
      "            4.4048e-04,  1.5701e-02],\n",
      "          ...,\n",
      "          [ 3.1347e-03, -1.0985e-02, -3.2569e-02,  ..., -9.3643e-03,\n",
      "            4.0229e-03, -4.7593e-03],\n",
      "          [-1.4315e-02, -1.1992e-02,  1.4437e-02,  ..., -9.9650e-03,\n",
      "            1.5973e-03,  1.9719e-03],\n",
      "          [-1.6013e-02,  2.7376e-03,  1.3703e-02,  ...,  4.5032e-03,\n",
      "           -8.3267e-03, -1.5758e-02]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[-1.6352e-02, -7.9971e-03, -3.6600e-04,  ..., -2.4343e-02,\n",
      "            4.1341e-03, -1.2726e-02],\n",
      "          [-7.5895e-03, -2.5708e-02,  6.5365e-05,  ..., -1.3495e-03,\n",
      "           -5.5134e-03, -1.8683e-02],\n",
      "          [-1.0314e-02, -9.2459e-03, -5.5137e-03,  ..., -1.4481e-02,\n",
      "           -9.4491e-03, -1.8040e-02],\n",
      "          ...,\n",
      "          [-3.3748e-02, -2.9064e-02, -5.2415e-03,  ..., -8.1776e-03,\n",
      "           -1.4581e-02, -2.4246e-02],\n",
      "          [-6.7054e-03, -6.5064e-03, -5.5634e-03,  ..., -1.8302e-02,\n",
      "           -1.3915e-02, -1.9616e-02],\n",
      "          [-1.2851e-02, -3.7664e-02, -7.9373e-03,  ..., -1.1542e-02,\n",
      "           -2.7742e-02, -1.2360e-02]],\n",
      "\n",
      "         [[ 5.9764e-03,  1.1124e-02,  3.3182e-03,  ...,  1.0216e-02,\n",
      "           -1.5117e-02,  1.3690e-02],\n",
      "          [ 4.9395e-04, -7.5153e-03,  8.6296e-03,  ...,  8.1994e-03,\n",
      "            7.1977e-03,  2.5272e-03],\n",
      "          [-3.7261e-03, -2.5792e-03, -3.5059e-02,  ...,  1.6435e-02,\n",
      "            1.1533e-02, -5.8290e-03],\n",
      "          ...,\n",
      "          [-1.1111e-02, -1.8693e-02, -3.3552e-02,  ..., -1.2267e-02,\n",
      "            8.4279e-04, -3.2727e-02],\n",
      "          [-3.3329e-02, -2.4417e-02, -3.4563e-03,  ..., -3.7300e-03,\n",
      "           -4.3485e-03, -2.6191e-03],\n",
      "          [-2.4551e-02, -2.3301e-02, -5.5629e-03,  ..., -2.4263e-02,\n",
      "           -7.4279e-03, -5.4921e-03]],\n",
      "\n",
      "         [[-1.8330e-02, -3.5020e-02, -4.5647e-03,  ..., -1.6847e-02,\n",
      "           -1.8493e-02, -2.4594e-02],\n",
      "          [-4.1776e-03, -2.8195e-02, -1.9245e-02,  ..., -2.0432e-02,\n",
      "           -4.7981e-03, -1.4396e-02],\n",
      "          [-1.4868e-02, -1.7694e-02, -2.6225e-02,  ..., -3.2991e-03,\n",
      "           -3.6838e-03, -4.8610e-03],\n",
      "          ...,\n",
      "          [-2.5331e-03, -4.7969e-03, -2.0132e-02,  ..., -4.7765e-03,\n",
      "           -5.7052e-03, -4.8429e-03],\n",
      "          [-3.4539e-03, -3.3050e-02, -1.4698e-02,  ..., -1.2370e-02,\n",
      "           -2.3638e-02, -2.4548e-02],\n",
      "          [-1.5635e-02, -3.4954e-02, -1.7280e-02,  ..., -1.2131e-02,\n",
      "           -2.3583e-02, -1.5356e-02]]]]))\n",
      "('3.bias', Parameter containing:\n",
      "tensor(1.00000e-02 *\n",
      "       [-1.0348, -0.6432, -2.8054,  0.4010,  0.0355, -1.6032, -2.2993,\n",
      "        -1.0429, -0.9652,  0.4127, -1.8207, -1.2974, -0.3693,  0.1035,\n",
      "        -3.5085,  0.2993, -1.0232, -0.8651,  0.9668, -2.0374, -0.7152,\n",
      "         0.3597, -1.8070,  0.8926, -3.2788, -0.0156, -1.5564, -1.5726,\n",
      "        -0.7210,  4.4880, -0.0121, -2.9784, -1.1779, -1.4455,  1.3124,\n",
      "         1.3883, -2.0414, -0.6788, -0.2914,  4.7307,  0.1653, -0.8118,\n",
      "        -0.6951, -1.2998,  1.0289,  1.4498, -0.9608, -2.2262, -1.0067,\n",
      "         0.1164,  1.2346, -1.9317, -0.8354, -0.7387, -0.8948, -0.9721,\n",
      "        -0.1986, -0.8877, -0.2728, -1.8956, -1.8667, -3.1181, -1.0578,\n",
      "        -0.4842]))\n",
      "('6.weight', Parameter containing:\n",
      "tensor([[[[-6.6457e-03, -1.4171e-02, -2.9045e-03,  ..., -1.8137e-02,\n",
      "           -2.4039e-02,  6.8482e-03],\n",
      "          [ 7.1761e-03, -4.3415e-04, -3.1281e-03,  ..., -2.2076e-02,\n",
      "            3.7009e-03, -2.6765e-02],\n",
      "          [ 1.5630e-03, -1.9424e-03,  9.1876e-04,  ..., -1.7380e-02,\n",
      "            7.2866e-03, -1.5511e-02],\n",
      "          ...,\n",
      "          [-6.0648e-03, -7.3269e-03, -1.2549e-02,  ..., -5.2128e-03,\n",
      "           -8.3172e-03, -1.7570e-02],\n",
      "          [-7.9316e-03, -2.4722e-02,  2.5008e-03,  ..., -2.3667e-02,\n",
      "           -2.3427e-02, -1.3126e-02],\n",
      "          [-6.7987e-03, -1.5110e-02, -8.4143e-03,  ..., -1.4240e-04,\n",
      "           -1.7958e-02,  8.3597e-04]],\n",
      "\n",
      "         [[-7.3525e-03,  7.4036e-03,  4.3098e-04,  ..., -1.5146e-02,\n",
      "            7.1143e-03,  6.2732e-04],\n",
      "          [ 1.6696e-02, -6.1884e-03, -1.7679e-02,  ...,  3.8439e-03,\n",
      "           -1.2341e-02, -1.3297e-02],\n",
      "          [ 2.2285e-02, -7.4451e-03,  9.5357e-03,  ...,  5.9170e-03,\n",
      "           -1.4165e-02,  5.7909e-03],\n",
      "          ...,\n",
      "          [ 8.5605e-03,  1.5439e-03,  3.1369e-03,  ...,  2.6004e-02,\n",
      "            1.6715e-02, -6.2791e-03],\n",
      "          [-2.9113e-03, -5.9755e-03,  2.4794e-02,  ...,  1.4971e-02,\n",
      "            2.4299e-02,  4.9134e-03],\n",
      "          [ 8.1300e-03,  2.7531e-02,  2.3008e-02,  ...,  2.6274e-02,\n",
      "            2.3459e-02,  8.4176e-03]],\n",
      "\n",
      "         [[ 1.6915e-03,  1.5662e-02, -2.1010e-02,  ..., -2.5390e-02,\n",
      "           -1.8454e-02, -2.5313e-03],\n",
      "          [-2.4235e-03, -1.7186e-02, -1.6349e-02,  ..., -1.0117e-02,\n",
      "           -2.0963e-03,  4.7621e-03],\n",
      "          [ 5.7097e-03, -5.0673e-03, -1.0786e-02,  ..., -2.2081e-02,\n",
      "           -1.2243e-02, -1.6443e-02],\n",
      "          ...,\n",
      "          [ 1.8973e-04,  5.9129e-03,  2.4763e-02,  ..., -1.0624e-02,\n",
      "            2.7799e-02,  1.7528e-02],\n",
      "          [ 1.5302e-02,  1.1425e-02,  1.0583e-03,  ..., -2.5907e-02,\n",
      "           -6.2298e-03,  2.6912e-02],\n",
      "          [ 1.3923e-03,  3.1232e-03,  2.4442e-02,  ...,  4.6041e-03,\n",
      "            1.3593e-03,  3.2995e-04]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[ 2.5818e-02, -2.5142e-03,  1.4200e-03,  ..., -2.1599e-03,\n",
      "           -2.2257e-03, -1.5570e-02],\n",
      "          [ 1.1808e-02,  5.1975e-03, -1.2687e-02,  ..., -9.0135e-03,\n",
      "           -1.0521e-02, -7.2020e-03],\n",
      "          [ 2.6967e-02,  1.1292e-02,  4.3858e-03,  ..., -1.5487e-02,\n",
      "           -7.1060e-03, -2.4728e-02],\n",
      "          ...,\n",
      "          [ 6.9493e-03, -7.3873e-03, -1.4296e-03,  ..., -1.5632e-03,\n",
      "           -9.1091e-03, -1.7611e-02],\n",
      "          [-8.2599e-03, -1.2565e-02, -9.7295e-04,  ..., -1.3228e-02,\n",
      "           -2.3057e-02,  2.0873e-03],\n",
      "          [ 2.5067e-02,  2.2780e-02, -2.1365e-02,  ..., -2.5433e-02,\n",
      "           -1.2683e-02, -1.1745e-02]],\n",
      "\n",
      "         [[ 5.5638e-04, -1.4176e-02, -2.1675e-02,  ...,  6.9348e-03,\n",
      "           -2.5002e-02, -7.3456e-03],\n",
      "          [-2.0352e-02,  6.1517e-03, -1.1137e-02,  ..., -1.0299e-02,\n",
      "           -7.1797e-03, -1.7937e-02],\n",
      "          [-2.4783e-02,  7.8138e-03, -8.3013e-03,  ..., -7.0645e-03,\n",
      "            5.7574e-03, -6.3606e-03],\n",
      "          ...,\n",
      "          [-8.1102e-03,  5.2446e-04,  1.3026e-02,  ...,  3.2308e-04,\n",
      "           -6.0556e-03, -2.3312e-02],\n",
      "          [-1.9393e-02, -1.0644e-02, -1.3462e-02,  ..., -2.9526e-03,\n",
      "           -1.1095e-02, -4.4025e-03],\n",
      "          [-2.0711e-02, -1.2897e-02, -2.3291e-02,  ..., -2.2207e-02,\n",
      "           -2.4636e-02, -9.8060e-03]],\n",
      "\n",
      "         [[-2.7536e-02,  2.2127e-04, -2.0344e-02,  ..., -2.6237e-02,\n",
      "           -1.6757e-02, -1.0473e-02],\n",
      "          [ 2.9774e-03, -3.4460e-03,  4.1607e-04,  ..., -1.8696e-02,\n",
      "           -1.7803e-02, -1.5318e-02],\n",
      "          [-6.8567e-03, -5.4140e-03, -2.2211e-03,  ..., -1.5153e-02,\n",
      "           -9.4086e-03, -2.4556e-02],\n",
      "          ...,\n",
      "          [-2.0466e-02, -1.5368e-02, -1.3362e-03,  ..., -1.2661e-02,\n",
      "           -2.7657e-02, -2.6702e-02],\n",
      "          [-1.6134e-04, -5.4864e-03, -6.9648e-03,  ..., -8.6749e-03,\n",
      "           -1.6099e-02, -1.1674e-02],\n",
      "          [-3.9682e-03, -2.1888e-04, -1.8551e-02,  ...,  3.1419e-03,\n",
      "           -1.6960e-02, -3.9024e-03]]],\n",
      "\n",
      "\n",
      "        [[[ 2.6815e-03,  2.7600e-03, -2.3110e-02,  ..., -6.3854e-03,\n",
      "            6.4987e-03,  1.1546e-03],\n",
      "          [-1.9149e-02,  2.1954e-03, -2.2608e-02,  ...,  1.9906e-03,\n",
      "           -9.6423e-03, -1.6311e-02],\n",
      "          [-4.5714e-04, -6.5635e-03, -1.8411e-02,  ...,  2.6722e-02,\n",
      "           -1.0920e-02, -2.1461e-02],\n",
      "          ...,\n",
      "          [ 6.3454e-03, -7.8911e-03, -2.8480e-03,  ..., -1.6411e-02,\n",
      "            6.4979e-03, -6.4111e-03],\n",
      "          [-2.7195e-02, -2.6426e-02,  6.3926e-03,  ...,  7.5039e-03,\n",
      "            1.6723e-02, -2.7925e-03],\n",
      "          [-9.6588e-03, -1.8814e-03, -3.7692e-03,  ..., -2.3274e-02,\n",
      "           -3.0475e-04, -2.4586e-02]],\n",
      "\n",
      "         [[-1.9053e-02, -1.4415e-02, -3.9138e-04,  ..., -2.2224e-02,\n",
      "           -2.5606e-02, -2.7464e-03],\n",
      "          [ 3.4185e-03,  6.5735e-04, -6.7643e-03,  ...,  3.8953e-03,\n",
      "           -2.2330e-02,  3.5497e-04],\n",
      "          [-1.7709e-03, -2.3377e-02, -6.9347e-03,  ..., -1.8656e-02,\n",
      "           -1.9381e-02, -6.5882e-03],\n",
      "          ...,\n",
      "          [ 7.6387e-03, -2.6929e-02, -7.6107e-04,  ..., -2.7788e-02,\n",
      "           -1.9311e-02, -2.3376e-02],\n",
      "          [-1.8804e-02, -1.5826e-02, -2.3658e-02,  ..., -4.7362e-03,\n",
      "           -3.9745e-03, -2.0802e-02],\n",
      "          [-2.3835e-02, -2.0257e-03,  6.2832e-03,  ..., -1.6132e-02,\n",
      "           -2.4613e-02, -1.9445e-02]],\n",
      "\n",
      "         [[ 3.7441e-03, -2.8301e-04, -2.1012e-02,  ..., -2.1116e-02,\n",
      "           -1.9632e-02,  4.1838e-03],\n",
      "          [ 5.3022e-03, -1.0277e-02,  4.0184e-03,  ...,  1.8427e-03,\n",
      "           -1.7806e-02, -5.5921e-03],\n",
      "          [-2.4765e-02, -1.2122e-02, -3.3981e-03,  ..., -2.7065e-03,\n",
      "            7.5494e-03, -8.5881e-03],\n",
      "          ...,\n",
      "          [-2.5949e-03, -1.2332e-02, -9.9704e-04,  ...,  6.1100e-03,\n",
      "           -2.1041e-02,  1.5915e-03],\n",
      "          [-2.9155e-03, -1.2471e-03,  2.3624e-03,  ..., -4.0809e-04,\n",
      "            2.2107e-03,  5.9018e-03],\n",
      "          [-2.7314e-02, -8.8494e-03,  3.4153e-03,  ...,  1.9356e-03,\n",
      "           -2.2871e-02, -1.3754e-02]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[-1.4542e-02, -1.1539e-02,  8.9263e-05,  ..., -7.3279e-03,\n",
      "           -2.6264e-02, -1.3329e-02],\n",
      "          [-2.7454e-02,  3.3421e-03, -2.5765e-02,  ..., -1.0423e-02,\n",
      "           -2.6844e-02,  7.3982e-03],\n",
      "          [-2.3613e-02, -1.3673e-02, -1.8357e-02,  ..., -1.7721e-02,\n",
      "           -8.5604e-03, -2.2132e-02],\n",
      "          ...,\n",
      "          [-2.6013e-02, -3.6969e-03,  2.5747e-03,  ..., -3.1840e-03,\n",
      "           -1.5127e-02, -5.5397e-03],\n",
      "          [-1.4254e-02, -2.5712e-02, -1.9869e-02,  ..., -1.7686e-02,\n",
      "            4.4819e-03, -9.8499e-03],\n",
      "          [-8.2287e-03, -2.6694e-02, -2.0531e-02,  ..., -1.7120e-03,\n",
      "           -2.0798e-02, -1.7615e-02]],\n",
      "\n",
      "         [[-1.1641e-02,  1.9956e-03,  1.1898e-02,  ...,  3.6846e-03,\n",
      "            1.1974e-02, -4.5329e-04],\n",
      "          [-2.5668e-02,  1.7122e-02,  1.7507e-02,  ...,  9.9175e-03,\n",
      "            2.5714e-02, -7.6476e-04],\n",
      "          [-1.7509e-02,  4.3558e-03,  7.5123e-04,  ..., -1.4198e-03,\n",
      "            2.5469e-03,  2.4452e-03],\n",
      "          ...,\n",
      "          [-4.9957e-03,  2.1021e-02,  1.2628e-02,  ...,  5.9583e-03,\n",
      "           -8.1602e-04,  5.6060e-03],\n",
      "          [ 9.3684e-03, -3.8993e-03, -7.4301e-03,  ...,  2.5262e-02,\n",
      "           -1.1567e-02, -2.2611e-02],\n",
      "          [ 1.5262e-02,  8.9769e-03,  9.7152e-03,  ..., -7.4974e-03,\n",
      "           -4.8129e-03,  5.6780e-03]],\n",
      "\n",
      "         [[ 2.3000e-03, -2.4523e-02,  6.6512e-03,  ..., -2.7391e-02,\n",
      "            6.8425e-03, -2.7274e-03],\n",
      "          [-7.4881e-03, -2.6908e-02, -1.0024e-02,  ...,  6.7545e-03,\n",
      "           -1.3386e-02, -2.3434e-02],\n",
      "          [-1.5010e-03, -1.7844e-03, -1.2954e-02,  ..., -1.4116e-02,\n",
      "           -5.1217e-03, -8.3684e-03],\n",
      "          ...,\n",
      "          [-2.3182e-02, -1.3679e-02, -2.6907e-02,  ...,  7.7890e-03,\n",
      "            5.6772e-03, -2.3604e-03],\n",
      "          [-1.3363e-02,  3.2209e-03, -2.5748e-02,  ..., -2.0797e-02,\n",
      "           -1.4555e-02, -1.4172e-02],\n",
      "          [-1.8342e-03, -1.0991e-02, -1.9722e-02,  ..., -1.5551e-02,\n",
      "           -1.0138e-02, -1.0672e-02]]],\n",
      "\n",
      "\n",
      "        [[[ 1.0803e-02,  4.7048e-03,  1.6983e-02,  ...,  4.1527e-03,\n",
      "            2.2373e-02,  5.4560e-04],\n",
      "          [ 6.3237e-03,  2.1544e-02,  1.7252e-02,  ..., -2.2766e-03,\n",
      "            1.3347e-02, -4.3080e-03],\n",
      "          [ 2.6734e-02,  2.6813e-02,  6.2577e-03,  ...,  2.1421e-02,\n",
      "           -3.0671e-03,  2.6876e-02],\n",
      "          ...,\n",
      "          [ 3.8466e-03,  1.4918e-02,  9.1837e-04,  ...,  1.4482e-02,\n",
      "           -2.3947e-02,  2.3661e-02],\n",
      "          [-5.0735e-03,  1.9258e-02,  7.1542e-03,  ...,  2.7190e-02,\n",
      "           -2.4411e-02,  2.2396e-02],\n",
      "          [ 5.5550e-03,  2.6562e-03,  2.5110e-02,  ...,  1.3546e-02,\n",
      "            6.1975e-03,  1.0409e-02]],\n",
      "\n",
      "         [[-1.1136e-02,  6.9883e-03,  3.0030e-03,  ..., -5.5232e-04,\n",
      "           -1.7540e-02, -2.3302e-02],\n",
      "          [-1.4507e-02,  3.2926e-03, -2.3579e-02,  ..., -1.3034e-02,\n",
      "           -1.0693e-02, -1.6482e-02],\n",
      "          [-1.4580e-02, -9.5563e-03, -6.7839e-03,  ..., -1.4063e-02,\n",
      "           -1.8502e-02, -2.5409e-02],\n",
      "          ...,\n",
      "          [-1.0305e-02,  6.6265e-03, -6.3571e-03,  ..., -1.4248e-02,\n",
      "           -1.3378e-03, -2.7316e-02],\n",
      "          [-2.7698e-02,  5.9108e-03, -2.4755e-02,  ..., -1.2901e-02,\n",
      "           -1.3346e-02, -1.8909e-02],\n",
      "          [-7.2682e-03,  2.3862e-03,  7.2970e-03,  ..., -1.9567e-03,\n",
      "            2.5399e-03, -1.2041e-02]],\n",
      "\n",
      "         [[ 9.3208e-04, -2.4150e-02, -8.2703e-03,  ..., -9.9188e-03,\n",
      "           -2.7008e-02, -6.4440e-03],\n",
      "          [-2.8552e-03, -2.5445e-02, -1.7352e-03,  ..., -5.3413e-03,\n",
      "            3.9580e-03, -1.5058e-02],\n",
      "          [-2.2481e-02, -7.4268e-03, -1.6228e-02,  ...,  6.2900e-03,\n",
      "           -4.6813e-03, -2.2208e-02],\n",
      "          ...,\n",
      "          [-2.2341e-02,  1.4652e-03, -6.4600e-03,  ..., -1.5374e-02,\n",
      "           -2.3160e-02, -3.9129e-03],\n",
      "          [-1.1384e-02, -2.0828e-02,  5.7455e-03,  ..., -1.1452e-02,\n",
      "           -7.6033e-03,  7.8521e-03],\n",
      "          [-1.1966e-03, -1.0845e-02, -9.7271e-04,  ..., -9.2989e-03,\n",
      "           -8.6757e-03, -2.6281e-02]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[-1.3458e-02, -7.7781e-03, -2.3052e-02,  ...,  2.8025e-03,\n",
      "           -7.8977e-03, -3.7936e-03],\n",
      "          [-1.1218e-02, -1.6381e-02,  6.8840e-03,  ...,  3.5669e-03,\n",
      "           -2.7674e-02, -3.5688e-03],\n",
      "          [-8.9585e-03, -1.2310e-02, -2.2705e-02,  ..., -9.7054e-03,\n",
      "           -2.3737e-03, -6.4271e-04],\n",
      "          ...,\n",
      "          [ 3.4600e-03, -9.8721e-04,  2.1120e-02,  ..., -2.6087e-02,\n",
      "           -4.4846e-03, -5.2615e-03],\n",
      "          [ 8.7108e-04, -8.7200e-03,  1.5654e-02,  ..., -9.8368e-03,\n",
      "           -1.5175e-02,  1.6077e-03],\n",
      "          [ 3.6899e-03, -1.1976e-02, -3.6526e-03,  ...,  1.2364e-04,\n",
      "           -1.8276e-02, -2.1552e-02]],\n",
      "\n",
      "         [[ 2.5739e-03,  3.8341e-03,  4.8284e-03,  ...,  1.5254e-02,\n",
      "           -1.8352e-02, -1.8539e-02],\n",
      "          [-1.0978e-03,  3.0958e-03, -2.1609e-02,  ..., -1.6518e-02,\n",
      "            1.1658e-02, -1.6776e-02],\n",
      "          [ 2.3345e-02,  1.4186e-02,  7.7396e-03,  ...,  2.4179e-02,\n",
      "            7.5861e-03,  1.4574e-03],\n",
      "          ...,\n",
      "          [ 9.2825e-03,  3.2912e-03,  1.9568e-03,  ..., -1.0460e-03,\n",
      "           -4.2239e-04,  2.6380e-03],\n",
      "          [ 2.2806e-02,  2.7072e-02,  1.1315e-02,  ...,  5.2665e-03,\n",
      "            3.7033e-04,  2.7828e-02],\n",
      "          [ 2.7164e-02,  1.1026e-03,  2.7055e-03,  ...,  1.7518e-02,\n",
      "            2.3056e-02, -6.8682e-03]],\n",
      "\n",
      "         [[-1.4881e-02, -5.9023e-03, -1.9649e-02,  ..., -4.1369e-04,\n",
      "           -1.3280e-02, -2.6200e-02],\n",
      "          [-2.6755e-02, -2.1669e-02,  4.5764e-03,  ..., -7.8507e-03,\n",
      "            6.4255e-03,  9.9325e-03],\n",
      "          [-8.6054e-03, -7.6855e-03,  5.8127e-03,  ...,  2.5860e-02,\n",
      "            6.4223e-03,  3.5202e-03],\n",
      "          ...,\n",
      "          [-6.3107e-03,  4.4912e-03, -7.6356e-03,  ...,  2.1135e-02,\n",
      "           -1.6345e-03,  5.7075e-03],\n",
      "          [ 3.9341e-03,  1.1165e-02, -7.7545e-03,  ...,  1.0315e-02,\n",
      "           -4.8345e-03,  1.4341e-02],\n",
      "          [ 1.8492e-02,  1.3883e-02,  1.0993e-02,  ...,  3.8087e-03,\n",
      "            2.1333e-02,  5.9169e-03]]],\n",
      "\n",
      "\n",
      "        ...,\n",
      "\n",
      "\n",
      "        [[[-1.7079e-02, -2.3652e-02,  1.7911e-02,  ..., -1.2363e-02,\n",
      "            2.8492e-03,  7.0113e-03],\n",
      "          [ 1.6031e-03, -2.7487e-02, -1.1339e-03,  ..., -2.4735e-02,\n",
      "            1.1502e-02,  1.2130e-02],\n",
      "          [-5.5296e-04, -2.5571e-02, -6.1638e-04,  ...,  1.6285e-02,\n",
      "            1.5937e-02,  6.5888e-03],\n",
      "          ...,\n",
      "          [-5.1860e-04, -5.0372e-03,  8.0611e-03,  ..., -1.0326e-03,\n",
      "           -1.7013e-02,  6.6016e-03],\n",
      "          [-1.2407e-02, -7.1872e-03,  2.0173e-02,  ...,  2.6709e-02,\n",
      "            2.6810e-02, -2.2917e-02],\n",
      "          [-4.1939e-03,  1.3234e-02,  7.8543e-03,  ...,  5.6746e-03,\n",
      "           -1.7797e-02,  4.2977e-03]],\n",
      "\n",
      "         [[-5.5146e-03, -1.6946e-02, -6.5365e-03,  ..., -2.4159e-02,\n",
      "           -2.6040e-02, -9.4051e-03],\n",
      "          [ 7.1273e-03, -4.6073e-03, -8.6110e-03,  ...,  1.7259e-03,\n",
      "           -3.6051e-03,  2.7651e-03],\n",
      "          [-1.9700e-02, -2.6362e-02, -9.3747e-03,  ..., -2.6163e-03,\n",
      "           -1.9860e-02, -4.4046e-04],\n",
      "          ...,\n",
      "          [-1.1889e-02, -9.4752e-03, -2.4356e-02,  ..., -2.3420e-02,\n",
      "           -2.0885e-02, -1.9287e-02],\n",
      "          [-2.6928e-02, -2.4612e-02,  4.7622e-03,  ..., -1.5503e-02,\n",
      "           -2.4363e-02,  7.5888e-03],\n",
      "          [-1.5518e-02, -2.3157e-02, -1.0659e-03,  ..., -2.4205e-02,\n",
      "           -9.6639e-03, -2.3377e-02]],\n",
      "\n",
      "         [[-2.6864e-02, -2.4899e-02, -2.2055e-03,  ..., -1.4565e-02,\n",
      "           -8.2756e-03, -2.1349e-02],\n",
      "          [-1.9350e-02, -2.6102e-02, -7.5557e-03,  ..., -2.6522e-02,\n",
      "           -2.1705e-02, -2.3488e-02],\n",
      "          [ 6.7886e-03, -2.0861e-02, -7.1895e-03,  ..., -2.0731e-02,\n",
      "           -1.6870e-02, -1.9063e-02],\n",
      "          ...,\n",
      "          [-8.5981e-03, -1.5810e-02, -2.1077e-02,  ...,  1.8970e-02,\n",
      "           -7.3123e-03,  6.5245e-03],\n",
      "          [-1.9654e-02, -2.7379e-02, -1.4654e-02,  ...,  1.7281e-02,\n",
      "           -2.6117e-02, -9.2734e-03],\n",
      "          [-1.7851e-02, -7.7316e-03,  4.3901e-03,  ..., -3.3982e-03,\n",
      "            5.3714e-03,  4.5850e-03]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[ 7.6055e-03, -9.5157e-03, -2.1007e-02,  ...,  2.3747e-03,\n",
      "            5.1110e-03, -2.7625e-02],\n",
      "          [-1.3811e-02,  5.6590e-03,  1.1003e-04,  ..., -1.8449e-02,\n",
      "           -8.6538e-04,  1.8602e-03],\n",
      "          [ 1.5298e-03,  5.9226e-03, -1.5929e-02,  ..., -7.0295e-03,\n",
      "           -6.9297e-03, -3.3335e-04],\n",
      "          ...,\n",
      "          [-8.5364e-03,  5.2309e-04, -2.4532e-02,  ..., -2.7381e-02,\n",
      "           -2.3954e-02, -7.8353e-04],\n",
      "          [-1.2966e-02, -9.3232e-04, -1.1649e-02,  ...,  7.2251e-03,\n",
      "           -2.4828e-02, -7.1678e-03],\n",
      "          [-2.7845e-02, -7.3668e-03, -1.3037e-02,  ..., -2.5463e-03,\n",
      "           -1.1670e-02,  6.9228e-04]],\n",
      "\n",
      "         [[ 4.2987e-03, -1.3866e-02, -3.0098e-04,  ..., -2.0312e-02,\n",
      "           -2.6673e-02, -7.5054e-03],\n",
      "          [-2.5288e-02, -2.7234e-02, -3.2072e-03,  ..., -1.6922e-03,\n",
      "            5.1159e-03,  1.8809e-03],\n",
      "          [-7.8404e-03, -2.4190e-02, -5.6545e-03,  ..., -9.9718e-03,\n",
      "            2.2418e-03, -1.8270e-03],\n",
      "          ...,\n",
      "          [-2.1183e-02, -1.7935e-02, -1.2816e-02,  ..., -4.2227e-03,\n",
      "            3.4619e-03, -5.6836e-03],\n",
      "          [ 5.1959e-03,  1.4082e-03, -1.8438e-02,  ...,  5.0958e-03,\n",
      "           -4.6172e-03, -1.0442e-02],\n",
      "          [ 3.1867e-03,  1.6476e-02, -2.3770e-03,  ..., -1.1497e-03,\n",
      "           -5.9099e-03, -1.2312e-05]],\n",
      "\n",
      "         [[-4.6278e-03, -1.7988e-02, -1.4015e-02,  ..., -7.2963e-03,\n",
      "           -1.5966e-02, -2.2753e-02],\n",
      "          [-2.2051e-02, -1.8920e-02, -2.3424e-02,  ..., -1.3496e-02,\n",
      "           -4.3788e-03, -1.8417e-02],\n",
      "          [ 6.6881e-03, -2.0990e-03, -1.6915e-02,  ..., -1.0629e-02,\n",
      "           -1.8865e-02, -2.7226e-02],\n",
      "          ...,\n",
      "          [-2.1795e-02, -9.9422e-03,  1.1856e-03,  ..., -1.4233e-02,\n",
      "           -5.5066e-03, -1.3657e-03],\n",
      "          [-9.7043e-03, -1.9177e-02, -1.1190e-02,  ...,  6.3276e-03,\n",
      "            3.6512e-03, -4.3879e-04],\n",
      "          [-3.5802e-03, -1.2609e-02,  3.3233e-03,  ...,  6.7619e-04,\n",
      "            4.9258e-03,  3.0377e-03]]],\n",
      "\n",
      "\n",
      "        [[[-7.1877e-03, -1.5681e-02, -1.6549e-02,  ...,  9.0503e-03,\n",
      "            2.6306e-02, -2.6235e-03],\n",
      "          [-2.3821e-02, -8.0970e-03, -7.9879e-03,  ..., -2.6150e-02,\n",
      "           -1.9651e-02, -2.1417e-02],\n",
      "          [ 1.0182e-02,  2.1408e-02, -1.7475e-02,  ..., -3.8979e-03,\n",
      "           -8.5626e-03, -2.2207e-02],\n",
      "          ...,\n",
      "          [ 1.2065e-02, -3.3887e-03, -2.4597e-02,  ..., -7.3119e-03,\n",
      "            5.5528e-04, -2.4736e-02],\n",
      "          [ 6.9064e-03, -1.4728e-03, -7.0641e-03,  ..., -5.9020e-03,\n",
      "           -1.9837e-03,  9.0277e-04],\n",
      "          [ 3.5109e-03,  8.2354e-03,  1.1213e-02,  ...,  2.2563e-03,\n",
      "           -2.5394e-03, -4.8740e-04]],\n",
      "\n",
      "         [[ 1.4663e-02, -2.5992e-02,  6.7058e-03,  ...,  1.3753e-03,\n",
      "           -1.5902e-02, -3.0446e-04],\n",
      "          [-1.5684e-02,  1.5662e-03,  1.4345e-04,  ...,  3.9895e-03,\n",
      "           -1.3194e-02, -1.8605e-02],\n",
      "          [-1.2445e-02, -1.7827e-02, -3.9375e-03,  ...,  5.2872e-03,\n",
      "           -1.1453e-03,  6.0658e-03],\n",
      "          ...,\n",
      "          [ 1.9661e-02, -2.8060e-03,  5.9395e-03,  ..., -1.0668e-02,\n",
      "           -1.9702e-02, -1.1407e-02],\n",
      "          [-5.6370e-03, -8.0981e-03, -9.2554e-03,  ...,  4.9647e-03,\n",
      "            6.1075e-03, -2.8991e-03],\n",
      "          [-3.0469e-04,  9.4089e-03, -3.9601e-03,  ..., -1.7853e-02,\n",
      "           -2.6795e-02, -1.4895e-03]],\n",
      "\n",
      "         [[-1.0753e-02,  1.3153e-02, -7.8495e-03,  ..., -6.2965e-03,\n",
      "           -6.8527e-03, -4.3924e-03],\n",
      "          [ 2.5400e-03, -2.6458e-02,  3.5085e-03,  ..., -2.7219e-02,\n",
      "           -1.2093e-02,  4.6735e-03],\n",
      "          [-1.0301e-02, -2.7772e-02, -2.2750e-02,  ..., -7.2343e-03,\n",
      "           -5.1331e-04,  2.2712e-03],\n",
      "          ...,\n",
      "          [-1.7714e-02, -3.8750e-03,  5.8806e-03,  ...,  1.6281e-02,\n",
      "           -2.3514e-03,  1.7316e-02],\n",
      "          [-1.8493e-02, -1.9961e-02, -1.1608e-02,  ...,  2.9879e-03,\n",
      "           -1.2182e-02,  1.6411e-03],\n",
      "          [-3.9716e-03, -2.7665e-02, -1.9828e-02,  ..., -6.4639e-03,\n",
      "           -9.6503e-03,  5.3992e-04]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[ 1.6866e-02,  2.1281e-02, -1.9741e-02,  ..., -1.8323e-02,\n",
      "           -2.5315e-02, -2.2827e-02],\n",
      "          [ 1.2587e-02, -2.1247e-02, -1.3705e-02,  ...,  4.2168e-04,\n",
      "            4.2513e-03, -1.8466e-02],\n",
      "          [-2.4485e-02, -1.5450e-02, -1.6921e-02,  ...,  3.0821e-03,\n",
      "           -1.4335e-02,  5.8570e-03],\n",
      "          ...,\n",
      "          [-7.2738e-03, -4.1662e-03, -1.5849e-02,  ...,  5.0289e-04,\n",
      "           -7.0477e-03, -1.7925e-02],\n",
      "          [-3.5463e-03, -1.9288e-02, -1.6590e-02,  ..., -4.3331e-03,\n",
      "            4.0621e-03, -1.3745e-02],\n",
      "          [ 2.4187e-02,  1.1818e-02, -1.6819e-02,  ..., -2.5773e-02,\n",
      "           -7.9688e-03, -1.2778e-02]],\n",
      "\n",
      "         [[-2.6345e-02, -2.2053e-02, -1.0640e-02,  ..., -1.8684e-02,\n",
      "           -1.4214e-02, -6.4522e-03],\n",
      "          [-1.3135e-02, -1.2603e-02,  2.7860e-03,  ..., -1.1153e-02,\n",
      "           -2.6949e-02, -1.9304e-02],\n",
      "          [ 6.7585e-03,  2.1437e-03, -2.2104e-02,  ..., -2.6568e-02,\n",
      "           -1.2645e-02, -4.9750e-03],\n",
      "          ...,\n",
      "          [ 9.7202e-05,  2.2102e-02, -1.1010e-02,  ..., -2.9561e-03,\n",
      "           -1.6349e-02, -8.2710e-03],\n",
      "          [ 1.4986e-02, -2.2819e-03,  1.5725e-02,  ..., -2.8215e-03,\n",
      "           -8.3108e-04,  7.6056e-03],\n",
      "          [ 1.5228e-02,  1.1487e-02, -2.5822e-02,  ...,  9.8976e-03,\n",
      "           -2.6789e-02, -4.1409e-03]],\n",
      "\n",
      "         [[-2.9711e-03, -1.4094e-02,  3.8414e-03,  ..., -2.0743e-02,\n",
      "           -9.5010e-03, -2.3399e-02],\n",
      "          [-5.2331e-04, -6.4809e-03, -2.0062e-02,  ...,  7.0671e-03,\n",
      "           -1.0242e-02, -1.8892e-02],\n",
      "          [-1.8418e-02, -1.2636e-02,  2.0545e-03,  ..., -1.7045e-03,\n",
      "           -1.2336e-02, -2.4090e-02],\n",
      "          ...,\n",
      "          [-7.5012e-03,  3.8512e-03, -2.6576e-02,  ..., -2.0135e-02,\n",
      "           -1.1112e-02, -2.2840e-02],\n",
      "          [-9.6134e-03, -2.5728e-02, -1.8202e-02,  ...,  6.0040e-04,\n",
      "            7.6143e-03, -7.5557e-03],\n",
      "          [ 5.8229e-03, -3.4572e-03, -1.0796e-02,  ..., -1.7034e-04,\n",
      "           -2.3383e-02, -8.9818e-03]]],\n",
      "\n",
      "\n",
      "        [[[-2.3425e-02, -7.5719e-03,  9.8183e-03,  ..., -7.3464e-03,\n",
      "            1.0208e-02,  2.6022e-02],\n",
      "          [ 1.9817e-02,  2.0486e-03,  7.5612e-03,  ...,  2.3252e-02,\n",
      "            3.6919e-03,  2.4617e-02],\n",
      "          [ 1.5216e-02,  7.6164e-03,  1.2121e-02,  ..., -2.5427e-02,\n",
      "            5.9616e-03, -1.5313e-02],\n",
      "          ...,\n",
      "          [-1.8953e-02, -2.7664e-02, -8.3372e-04,  ..., -1.1394e-02,\n",
      "           -3.9599e-03, -1.0342e-03],\n",
      "          [-1.6751e-02, -2.0885e-02, -1.7630e-02,  ..., -1.6645e-03,\n",
      "           -2.3710e-02, -1.7802e-02],\n",
      "          [-6.0117e-03,  3.8652e-03, -2.6424e-02,  ..., -2.2235e-02,\n",
      "           -1.6443e-03, -2.7785e-02]],\n",
      "\n",
      "         [[-4.1271e-03,  4.0829e-03,  3.4600e-03,  ..., -2.2573e-02,\n",
      "           -1.6629e-02, -3.0176e-03],\n",
      "          [ 2.0742e-02, -8.8202e-03, -2.5944e-02,  ...,  2.3755e-02,\n",
      "           -6.2048e-03,  1.5848e-02],\n",
      "          [ 6.0385e-03,  9.0078e-03,  1.1106e-02,  ..., -2.3429e-02,\n",
      "            6.0346e-03,  2.3764e-02],\n",
      "          ...,\n",
      "          [-3.2518e-03,  1.1081e-02, -2.9847e-03,  ...,  2.1810e-03,\n",
      "            2.1432e-02,  1.4721e-02],\n",
      "          [ 2.4611e-03,  2.6107e-02, -1.9289e-03,  ...,  1.3872e-02,\n",
      "            2.3297e-03, -1.8788e-03],\n",
      "          [ 2.1329e-02,  1.6712e-03,  2.4485e-02,  ...,  1.1636e-02,\n",
      "            9.4464e-03,  2.6383e-02]],\n",
      "\n",
      "         [[-2.0996e-02,  6.0790e-03, -8.0994e-03,  ...,  3.8938e-03,\n",
      "           -1.3891e-03,  7.1402e-03],\n",
      "          [-7.0939e-03,  6.2268e-03, -1.4198e-02,  ..., -5.2488e-03,\n",
      "            1.4681e-02,  1.2842e-02],\n",
      "          [ 4.9848e-03, -2.7536e-03, -2.2892e-02,  ...,  3.7270e-03,\n",
      "            2.0247e-02,  1.3076e-02],\n",
      "          ...,\n",
      "          [-1.8175e-03,  1.1675e-02,  1.6029e-02,  ...,  3.0613e-03,\n",
      "            1.4126e-03, -1.0311e-03],\n",
      "          [ 1.1964e-02, -2.5340e-03,  2.3853e-02,  ...,  6.0123e-03,\n",
      "            1.8182e-02,  4.9782e-03],\n",
      "          [ 2.6864e-02, -6.5097e-03,  1.3913e-02,  ..., -5.3833e-03,\n",
      "            1.0168e-03,  3.6364e-03]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[-2.2790e-02, -2.2155e-02,  2.9486e-03,  ..., -4.1600e-04,\n",
      "           -9.5766e-05, -9.7980e-03],\n",
      "          [ 2.8710e-04, -8.9594e-03, -2.1569e-03,  ...,  3.8048e-03,\n",
      "           -9.6475e-03, -1.3285e-02],\n",
      "          [-1.6881e-03, -2.4389e-02, -2.6588e-02,  ..., -1.8146e-02,\n",
      "           -2.5774e-02, -1.9062e-02],\n",
      "          ...,\n",
      "          [ 6.1846e-03, -2.6508e-02, -2.3966e-02,  ..., -8.0941e-03,\n",
      "           -1.7214e-03, -1.8662e-02],\n",
      "          [-2.4548e-02,  2.0505e-02, -1.8847e-02,  ..., -1.8596e-02,\n",
      "           -2.2775e-02, -1.5670e-02],\n",
      "          [-6.7244e-03,  1.5245e-02,  2.6767e-03,  ..., -9.2259e-04,\n",
      "           -1.3007e-02,  3.1601e-03]],\n",
      "\n",
      "         [[-2.3199e-02, -8.7317e-03, -3.8718e-03,  ...,  7.0642e-03,\n",
      "           -2.6819e-02, -2.3488e-02],\n",
      "          [-2.7719e-02,  1.8092e-03, -4.2290e-03,  ..., -3.7541e-03,\n",
      "            2.3110e-02, -5.9336e-03],\n",
      "          [-1.3882e-02, -9.4410e-03,  7.1976e-03,  ...,  6.1983e-03,\n",
      "            7.5404e-03,  2.6457e-02],\n",
      "          ...,\n",
      "          [-3.6166e-03, -1.9016e-02,  7.3500e-03,  ..., -7.7176e-03,\n",
      "           -4.5644e-03, -7.5307e-03],\n",
      "          [-1.6041e-02, -1.3890e-03, -8.0973e-03,  ..., -2.1373e-02,\n",
      "           -1.3475e-02, -1.5834e-02],\n",
      "          [-1.5544e-02, -5.1768e-04,  5.4445e-03,  ..., -1.8930e-02,\n",
      "           -2.3949e-02, -5.7021e-03]],\n",
      "\n",
      "         [[ 1.0378e-03,  8.1250e-04, -1.7659e-02,  ...,  4.4009e-03,\n",
      "           -5.7019e-04,  5.7071e-03],\n",
      "          [-8.5712e-04,  5.3408e-03, -2.6760e-02,  ..., -2.3124e-02,\n",
      "           -5.0353e-03, -1.1309e-02],\n",
      "          [ 2.5485e-02, -1.7972e-02,  3.4282e-03,  ..., -4.4639e-03,\n",
      "            1.3660e-03, -8.8214e-04],\n",
      "          ...,\n",
      "          [-9.1250e-03, -1.2143e-02, -1.7638e-03,  ..., -2.3895e-02,\n",
      "           -2.5546e-02,  5.2568e-03],\n",
      "          [-1.0582e-02,  3.3731e-03, -2.1635e-02,  ..., -1.5461e-02,\n",
      "           -8.5758e-03, -7.9359e-03],\n",
      "          [-1.5926e-02, -1.1710e-02, -1.3718e-02,  ..., -1.8985e-02,\n",
      "            1.9257e-03, -3.7265e-03]]]]))\n",
      "('6.bias', Parameter containing:\n",
      "tensor(1.00000e-02 *\n",
      "       [-0.6255,  4.9319, -3.6779, -2.9463, -2.5801,  0.8595, -1.0889,\n",
      "         4.8599,  0.5028,  1.3037, -0.7622,  3.5637,  1.6952, -2.5448,\n",
      "        -0.1713,  1.7510, -3.0964, -0.9239,  2.0107, -1.5601, -4.1642,\n",
      "        -1.7541, -0.9685,  2.0369, -0.7735,  4.4888,  1.0942, -0.2250,\n",
      "         1.5792,  1.1349, -1.6582, -1.1896, -1.3102,  4.1395, -0.1528,\n",
      "        -4.3832, -3.0477,  2.4790, -0.6793, -0.5944, -2.2985,  0.9787,\n",
      "        -1.1641,  0.8887, -2.2574, -0.9951, -1.2451,  2.0503, -7.5850,\n",
      "        -0.5772, -1.6409,  0.9793, -1.5066,  4.9639, -2.9812, -1.3580,\n",
      "         0.0373,  1.6401, -0.9795, -2.6780, -0.8956,  3.6520, -1.1645,\n",
      "         2.0801]))\n",
      "('10.weight', Parameter containing:\n",
      "tensor([[-7.5301e-02, -1.7809e-02, -3.9599e-02,  ...,  7.6142e-02,\n",
      "         -1.0008e-01, -7.2872e-02],\n",
      "        [ 5.1972e-02, -1.6626e-01,  8.4465e-02,  ..., -7.6805e-02,\n",
      "         -9.8366e-02,  6.6758e-02],\n",
      "        [-3.3718e-02,  1.3370e-01, -5.7192e-02,  ..., -9.0067e-02,\n",
      "          7.1618e-02, -2.5465e-02],\n",
      "        ...,\n",
      "        [ 6.2247e-02,  6.4090e-02,  8.5833e-02,  ..., -1.2668e-02,\n",
      "          1.3935e-02, -5.4030e-02],\n",
      "        [-2.3584e-01,  1.3796e-01,  8.0857e-02,  ...,  1.2481e-01,\n",
      "         -4.7108e-02, -1.1746e-01],\n",
      "        [ 1.1030e-01, -4.2672e-02,  2.4807e-02,  ...,  3.9567e-02,\n",
      "         -5.2033e-02,  5.8760e-02]]))\n",
      "('10.bias', Parameter containing:\n",
      "tensor([-0.0170, -0.0177,  0.0905,  0.0724,  0.0496,  0.0758,  0.0863,\n",
      "        -0.0121, -0.1114, -0.1325,  0.0691,  0.0418,  0.0570,  0.0775,\n",
      "         0.1363, -0.0270,  0.0533, -0.0383, -0.0989,  0.0183,  0.0397,\n",
      "         0.1140,  0.0707,  0.0802,  0.0219, -0.0462, -0.0971, -0.0339,\n",
      "         0.0596,  0.1036, -0.0411, -0.0024,  0.0568,  0.0786, -0.0063,\n",
      "        -0.1060,  0.0394, -0.1205,  0.0189, -0.0318,  0.0170, -0.1569,\n",
      "         0.1222, -0.0652, -0.0800, -0.1372, -0.0851, -0.1614, -0.0519,\n",
      "        -0.1048,  0.1035, -0.0287, -0.0404, -0.1073,  0.0188,  0.1039,\n",
      "        -0.0469,  0.0518,  0.0717,  0.0868,  0.0811,  0.0636,  0.0492,\n",
      "        -0.0230, -0.0766, -0.0199,  0.0445, -0.0969,  0.0910, -0.0910,\n",
      "         0.0038, -0.0456,  0.0315, -0.0613, -0.0293, -0.0596,  0.0228,\n",
      "        -0.0958,  0.1407, -0.0356, -0.0213, -0.0660,  0.0504,  0.0503,\n",
      "        -0.0463,  0.0843,  0.0917,  0.0356, -0.0272, -0.1018, -0.0236,\n",
      "        -0.1267, -0.0688,  0.0875,  0.0440, -0.0899,  0.0881,  0.0511,\n",
      "         0.1372, -0.0741]))\n",
      "('12.weight', Parameter containing:\n",
      "tensor([[-0.0017, -0.0766,  0.0924,  ...,  0.0853, -0.0931, -0.1333],\n",
      "        [-0.1053,  0.0104,  0.0136,  ..., -0.0441, -0.0201,  0.0947],\n",
      "        [ 0.0697,  0.0490, -0.0532,  ..., -0.0319,  0.1249, -0.0447],\n",
      "        ...,\n",
      "        [-0.0374, -0.0541,  0.0537,  ...,  0.0963,  0.0487,  0.0269],\n",
      "        [ 0.0725,  0.0446, -0.0674,  ..., -0.0942,  0.0055, -0.0619],\n",
      "        [-0.0330,  0.0317, -0.0500,  ..., -0.0020, -0.0695, -0.1809]]))\n",
      "('12.bias', Parameter containing:\n",
      "tensor([-0.0489,  0.1297, -0.0611,  0.0806, -0.0072, -0.0580, -0.0646,\n",
      "         0.0661, -0.0694, -0.0294, -0.0115, -0.0775,  0.0346, -0.0762,\n",
      "        -0.1076,  0.0488, -0.1113, -0.0511, -0.0543,  0.0004, -0.0727,\n",
      "        -0.0120, -0.1211,  0.1344, -0.0039,  0.0428,  0.0673,  0.1046,\n",
      "         0.0122, -0.0327, -0.0499,  0.0923, -0.1074, -0.0446,  0.0439,\n",
      "        -0.1016, -0.0942, -0.1311, -0.0221,  0.1263,  0.0584, -0.0257,\n",
      "        -0.0426,  0.0923,  0.0596,  0.0151,  0.0068, -0.0514, -0.0368,\n",
      "        -0.0243,  0.0778,  0.0376,  0.0819, -0.0541,  0.0811,  0.0558,\n",
      "        -0.0639,  0.0004,  0.0526, -0.0735,  0.0347, -0.0209, -0.0446,\n",
      "        -0.1415, -0.0052]))\n"
     ]
    }
   ],
   "source": [
    "for name in model.named_parameters():\n",
    "    print (name)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 31,
   "metadata": {},
   "outputs": [],
   "source": [
    "first = record[0].view(1, 3,100,100)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 32,
   "metadata": {},
   "outputs": [
    {
     "data": {
      "text/plain": [
       "torch.Size([1, 3, 100, 100])"
      ]
     },
     "execution_count": 32,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "first.shape"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 34,
   "metadata": {},
   "outputs": [
    {
     "data": {
      "text/plain": [
       "tensor([ 50])"
      ]
     },
     "execution_count": 34,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "model(first).max(1)[1]"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 38,
   "metadata": {},
   "outputs": [
    {
     "data": {
      "text/plain": [
       "50"
      ]
     },
     "execution_count": 38,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "labels_train[0]"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 39,
   "metadata": {},
   "outputs": [],
   "source": [
    "record = iter(test_loader).next()[0][0]"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 41,
   "metadata": {},
   "outputs": [],
   "source": [
    "first = record.view(1, 3,100,100)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 42,
   "metadata": {},
   "outputs": [
    {
     "data": {
      "text/plain": [
       "tensor([ 26])"
      ]
     },
     "execution_count": 42,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "model(first).max(1)[1]"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 43,
   "metadata": {},
   "outputs": [
    {
     "data": {
      "text/plain": [
       "44"
      ]
     },
     "execution_count": 43,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "labels_test[0]"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 52,
   "metadata": {},
   "outputs": [
    {
     "data": {
      "text/plain": [
       "OrderedDict([('0.weight',\n",
       "              tensor([[[[-0.0016,  0.0048, -0.0146, -0.1456, -0.1541],\n",
       "                        [-0.0630,  0.0308,  0.0080, -0.1073, -0.1134],\n",
       "                        [-0.1058,  0.0090, -0.1061,  0.0478,  0.0654],\n",
       "                        [ 0.0175,  0.0800,  0.0240,  0.0162,  0.0625],\n",
       "                        [ 0.0913, -0.0833, -0.0338,  0.0798, -0.0163]],\n",
       "              \n",
       "                       [[ 0.0605,  0.0374,  0.0662, -0.1401, -0.0259],\n",
       "                        [ 0.0589, -0.1049,  0.0120, -0.0084, -0.1260],\n",
       "                        [-0.0768, -0.1128, -0.0596,  0.0139,  0.0571],\n",
       "                        [ 0.0931, -0.0645, -0.0827,  0.0600,  0.0205],\n",
       "                        [ 0.1170,  0.1265, -0.0124, -0.0439,  0.1113]],\n",
       "              \n",
       "                       [[-0.0642, -0.0756,  0.0292,  0.0497, -0.1144],\n",
       "                        [ 0.0153, -0.0897, -0.0337,  0.0108, -0.1262],\n",
       "                        [ 0.0991, -0.0447, -0.0442, -0.0280,  0.0873],\n",
       "                        [-0.0269,  0.0825, -0.1120, -0.0491, -0.0136],\n",
       "                        [ 0.0866, -0.0420,  0.1009,  0.1126,  0.0893]]],\n",
       "              \n",
       "              \n",
       "                      [[[-0.0209,  0.0852,  0.0590, -0.0332, -0.0586],\n",
       "                        [ 0.0518,  0.1475,  0.0091,  0.0288,  0.0954],\n",
       "                        [ 0.0341,  0.0415, -0.0412,  0.0043,  0.1360],\n",
       "                        [ 0.1538, -0.0249,  0.0003,  0.1572,  0.0010],\n",
       "                        [ 0.0510,  0.0192,  0.1027,  0.1332,  0.0991]],\n",
       "              \n",
       "                       [[ 0.0177, -0.0105, -0.0383,  0.0657,  0.0859],\n",
       "                        [-0.0725, -0.1320, -0.0927, -0.1090, -0.0635],\n",
       "                        [ 0.0969, -0.0837, -0.1208,  0.0329, -0.0886],\n",
       "                        [-0.0370,  0.0143, -0.0048, -0.0736, -0.0601],\n",
       "                        [-0.0365, -0.0917, -0.0431, -0.0236, -0.0755]],\n",
       "              \n",
       "                       [[ 0.0102, -0.0525,  0.0135, -0.0179, -0.0234],\n",
       "                        [ 0.0643, -0.0145,  0.1480,  0.0188,  0.0504],\n",
       "                        [ 0.0169, -0.0020,  0.0158, -0.0133, -0.0076],\n",
       "                        [ 0.1242, -0.0200,  0.0480,  0.1099,  0.0300],\n",
       "                        [ 0.0524, -0.0310,  0.0239,  0.0870,  0.1372]]],\n",
       "              \n",
       "              \n",
       "                      [[[-0.0898, -0.0185, -0.0380, -0.0599, -0.0783],\n",
       "                        [ 0.0435, -0.0673, -0.0520, -0.0430,  0.0795],\n",
       "                        [-0.0218, -0.0063,  0.0079,  0.0160,  0.0481],\n",
       "                        [ 0.0058,  0.1550,  0.0125,  0.0755,  0.0940],\n",
       "                        [ 0.1355,  0.0955,  0.1428,  0.1401,  0.1735]],\n",
       "              \n",
       "                       [[-0.1866, -0.1629,  0.0517, -0.0109, -0.1201],\n",
       "                        [-0.1588, -0.1707, -0.1438,  0.0682,  0.0742],\n",
       "                        [-0.0229, -0.1558,  0.0158, -0.1283,  0.0976],\n",
       "                        [-0.1197, -0.0310, -0.0727,  0.0966, -0.0852],\n",
       "                        [ 0.0389,  0.0312,  0.0191,  0.0596,  0.0384]],\n",
       "              \n",
       "                       [[ 0.0068,  0.0797, -0.0105,  0.0056,  0.1451],\n",
       "                        [ 0.0565, -0.0742, -0.0368,  0.0914,  0.0443],\n",
       "                        [-0.0046,  0.1062,  0.0461,  0.1649,  0.1579],\n",
       "                        [ 0.1141, -0.0795,  0.1050, -0.0006,  0.0684],\n",
       "                        [-0.0832, -0.0160,  0.0237,  0.1942,  0.1150]]],\n",
       "              \n",
       "              \n",
       "                      ...,\n",
       "              \n",
       "              \n",
       "                      [[[-0.0482, -0.0629, -0.1348, -0.0321, -0.1168],\n",
       "                        [-0.0721, -0.0334, -0.0868, -0.0112,  0.0183],\n",
       "                        [ 0.0814, -0.0759,  0.0558,  0.0165, -0.1443],\n",
       "                        [ 0.0478, -0.1126,  0.0319,  0.0282,  0.0392],\n",
       "                        [-0.0818,  0.0456,  0.0469,  0.0682, -0.0889]],\n",
       "              \n",
       "                       [[-0.0964, -0.0441, -0.0376, -0.1273,  0.0643],\n",
       "                        [-0.0678,  0.0213, -0.0184,  0.0526,  0.0937],\n",
       "                        [-0.0995, -0.0961,  0.0146,  0.0857,  0.0363],\n",
       "                        [-0.0216, -0.1249,  0.0229, -0.0659, -0.0795],\n",
       "                        [ 0.0812, -0.0560, -0.1054, -0.0096,  0.0889]],\n",
       "              \n",
       "                       [[-0.0897, -0.1021,  0.0180,  0.0298, -0.0659],\n",
       "                        [ 0.0528, -0.1170, -0.0548,  0.0178, -0.0131],\n",
       "                        [ 0.0153,  0.0072,  0.0386,  0.0121, -0.1135],\n",
       "                        [-0.1317,  0.0072, -0.0720,  0.0915, -0.1055],\n",
       "                        [-0.1206, -0.0260, -0.1168,  0.0285, -0.1064]]],\n",
       "              \n",
       "              \n",
       "                      [[[ 0.0775,  0.0365,  0.0013,  0.0921,  0.1069],\n",
       "                        [-0.0504, -0.0125, -0.0210,  0.0183,  0.0713],\n",
       "                        [-0.0559, -0.0774, -0.1068, -0.0869, -0.0683],\n",
       "                        [-0.0192,  0.0008,  0.0784,  0.0213, -0.0400],\n",
       "                        [-0.0524,  0.0806, -0.0096,  0.0329, -0.0262]],\n",
       "              \n",
       "                       [[ 0.0902,  0.0546,  0.0144,  0.1076, -0.0733],\n",
       "                        [ 0.0053, -0.0091, -0.0023, -0.0353, -0.0744],\n",
       "                        [ 0.0175, -0.0493, -0.0767, -0.0778, -0.0176],\n",
       "                        [ 0.0258,  0.0045,  0.0684,  0.0814,  0.0049],\n",
       "                        [-0.0952,  0.0733,  0.0422,  0.0057, -0.0058]],\n",
       "              \n",
       "                       [[-0.0374,  0.0999,  0.0383,  0.0544,  0.1199],\n",
       "                        [-0.1001, -0.0429,  0.1261, -0.0481, -0.0135],\n",
       "                        [ 0.0285,  0.0622, -0.0188,  0.0707, -0.0995],\n",
       "                        [-0.0672,  0.1120,  0.0732,  0.0780,  0.0502],\n",
       "                        [-0.1165,  0.0694,  0.0052, -0.0275,  0.0033]]],\n",
       "              \n",
       "              \n",
       "                      [[[-0.0018,  0.1250, -0.0596,  0.0056, -0.0844],\n",
       "                        [ 0.0031, -0.0784, -0.0879,  0.1046,  0.0131],\n",
       "                        [ 0.0248,  0.1182,  0.0623,  0.1107,  0.0702],\n",
       "                        [-0.0020, -0.0631,  0.0097, -0.0636, -0.0109],\n",
       "                        [-0.0106,  0.1198,  0.0468, -0.0776,  0.0931]],\n",
       "              \n",
       "                       [[ 0.0912, -0.0293, -0.1130, -0.1091, -0.1368],\n",
       "                        [-0.0486, -0.0580, -0.0293,  0.0825,  0.0432],\n",
       "                        [-0.0488, -0.0165, -0.0907,  0.1175, -0.1161],\n",
       "                        [-0.0058,  0.0306, -0.0454,  0.0105,  0.1014],\n",
       "                        [ 0.1089,  0.0913,  0.0672,  0.0102,  0.0053]],\n",
       "              \n",
       "                       [[-0.0369,  0.1566,  0.0700,  0.0545,  0.0613],\n",
       "                        [ 0.0877,  0.0052,  0.0940,  0.1503,  0.0434],\n",
       "                        [-0.0281,  0.0453,  0.0012,  0.0196, -0.0414],\n",
       "                        [ 0.0855, -0.0163,  0.1347,  0.0424,  0.0502],\n",
       "                        [ 0.0874,  0.0372,  0.1193,  0.1401,  0.1866]]]])),\n",
       "             ('0.bias',\n",
       "              tensor([-0.0074, -0.0474,  0.0857,  0.1327,  0.0287,  0.1414,  0.1547,\n",
       "                       0.0509, -0.0440, -0.0592,  0.0788, -0.0021,  0.0642,  0.1670,\n",
       "                       0.0317,  0.0466,  0.1044, -0.0274, -0.0708,  0.0774,  0.0230,\n",
       "                       0.0278,  0.1082,  0.1173,  0.0120, -0.0614,  0.0690,  0.0154,\n",
       "                      -0.0326,  0.0599, -0.0667,  0.0317, -0.0404,  0.1326, -0.0984,\n",
       "                       0.0518, -0.0816, -0.0412, -0.0766,  0.0903, -0.0249, -0.0253,\n",
       "                       0.2320, -0.0652, -0.0272,  0.0391,  0.0159, -0.0892, -0.0242,\n",
       "                      -0.0119, -0.0370,  0.1231, -0.0702, -0.0439, -0.0870,  0.0212,\n",
       "                       0.1375, -0.1098,  0.0907,  0.0009,  0.0788,  0.0816, -0.1044,\n",
       "                       0.0091])),\n",
       "             ('3.weight',\n",
       "              tensor([[[[-2.6779e-02, -1.2962e-02, -6.7731e-03,  ..., -2.1328e-02,\n",
       "                         -1.5803e-02, -7.1797e-03],\n",
       "                        [-1.8022e-02, -2.3607e-03, -2.2527e-02,  ..., -3.2269e-02,\n",
       "                          1.1309e-02,  7.9132e-03],\n",
       "                        [-1.8926e-02, -1.6236e-02, -1.8413e-02,  ...,  9.0630e-03,\n",
       "                          9.2606e-03,  1.7090e-02],\n",
       "                        ...,\n",
       "                        [-3.7086e-03, -7.2559e-04,  1.0614e-02,  ...,  6.1826e-04,\n",
       "                         -1.2080e-02,  4.3929e-04],\n",
       "                        [-2.3554e-02, -1.1957e-02, -6.5553e-03,  ..., -1.1912e-02,\n",
       "                         -1.3815e-02, -2.2450e-03],\n",
       "                        [-2.0905e-02, -2.5623e-02, -3.3727e-02,  ..., -8.8794e-03,\n",
       "                          2.5970e-03,  8.4659e-04]],\n",
       "              \n",
       "                       [[-1.0403e-02,  9.8815e-03,  1.3379e-02,  ...,  1.6973e-02,\n",
       "                         -7.1956e-03,  4.0096e-03],\n",
       "                        [ 1.1212e-02, -1.4063e-02,  1.7008e-02,  ..., -6.0592e-03,\n",
       "                         -1.2660e-02, -1.1657e-02],\n",
       "                        [-3.2382e-02, -4.0698e-03,  6.5229e-03,  ..., -3.1508e-03,\n",
       "                         -3.7202e-02, -1.1686e-02],\n",
       "                        ...,\n",
       "                        [-9.1540e-03,  1.0237e-02,  1.6859e-02,  ...,  8.6157e-03,\n",
       "                         -3.7181e-03,  9.6950e-03],\n",
       "                        [-1.7268e-02,  1.4214e-02, -1.1315e-02,  ..., -8.8661e-03,\n",
       "                         -1.2371e-02,  6.7527e-03],\n",
       "                        [ 9.9682e-03, -1.3675e-02, -6.5624e-03,  ...,  3.1749e-03,\n",
       "                          9.4326e-03, -2.9477e-03]],\n",
       "              \n",
       "                       [[-7.0483e-03, -3.1922e-02, -2.1498e-02,  ..., -3.1507e-03,\n",
       "                         -3.0696e-02, -2.8737e-03],\n",
       "                        [-6.8231e-03, -1.3644e-02, -1.6670e-03,  ..., -2.8231e-02,\n",
       "                         -3.1186e-02, -2.6410e-02],\n",
       "                        [-9.8460e-03, -3.6508e-02, -1.1149e-02,  ..., -1.7134e-02,\n",
       "                         -3.2647e-02, -2.3700e-02],\n",
       "                        ...,\n",
       "                        [-5.7745e-04,  9.0411e-04, -1.4712e-02,  ...,  1.3643e-02,\n",
       "                         -1.8622e-03, -1.3857e-02],\n",
       "                        [-3.7543e-02,  1.7850e-02, -2.0641e-03,  ...,  5.8658e-03,\n",
       "                          4.6741e-03, -8.6243e-03],\n",
       "                        [-3.3316e-02, -4.9936e-03, -1.4988e-02,  ..., -8.6911e-05,\n",
       "                         -1.7218e-02, -2.7561e-03]],\n",
       "              \n",
       "                       ...,\n",
       "              \n",
       "                       [[ 1.5939e-02, -1.4361e-02,  1.4564e-02,  ..., -8.7929e-03,\n",
       "                         -1.2084e-02,  1.2491e-02],\n",
       "                        [ 1.7160e-02, -6.9554e-03,  1.0544e-02,  ..., -1.0389e-02,\n",
       "                         -4.2872e-03, -4.1053e-03],\n",
       "                        [-2.0027e-03,  1.0370e-02,  1.7050e-02,  ..., -6.4324e-03,\n",
       "                          1.6785e-02,  3.4043e-03],\n",
       "                        ...,\n",
       "                        [ 1.3323e-02, -3.4984e-03, -1.5882e-02,  ..., -2.4456e-03,\n",
       "                          2.4521e-03,  1.6473e-02],\n",
       "                        [-1.7732e-02,  1.5460e-02,  6.0378e-03,  ..., -1.2608e-02,\n",
       "                          1.7068e-02,  1.6477e-02],\n",
       "                        [ 7.6953e-03,  9.7408e-03, -5.5379e-03,  ..., -1.6686e-03,\n",
       "                         -1.5905e-03,  1.2670e-02]],\n",
       "              \n",
       "                       [[-2.7136e-02, -3.4884e-02, -3.1536e-02,  ...,  8.7532e-03,\n",
       "                          1.7536e-02, -8.5740e-03],\n",
       "                        [-3.2289e-02, -2.1074e-02, -2.7792e-03,  ...,  1.3592e-03,\n",
       "                          6.7092e-03,  8.8825e-03],\n",
       "                        [-2.4370e-03, -2.1794e-02, -1.4332e-02,  ..., -2.6290e-03,\n",
       "                         -9.7182e-04, -1.3434e-02],\n",
       "                        ...,\n",
       "                        [-2.7251e-02, -5.9368e-03, -1.7157e-02,  ...,  4.8391e-03,\n",
       "                          1.3846e-02, -2.5880e-02],\n",
       "                        [-8.4309e-03, -9.7086e-03, -8.2329e-03,  ..., -1.1860e-02,\n",
       "                         -1.2285e-02,  6.6090e-03],\n",
       "                        [-2.8973e-03,  9.6497e-03,  2.2892e-03,  ...,  1.2426e-02,\n",
       "                         -9.8844e-03, -2.6804e-02]],\n",
       "              \n",
       "                       [[-1.7853e-03,  1.4545e-02,  1.3259e-02,  ...,  6.5922e-03,\n",
       "                         -5.1442e-03, -1.0759e-02],\n",
       "                        [ 1.4357e-02,  4.0669e-03,  1.0755e-02,  ..., -1.3151e-02,\n",
       "                          1.4049e-02,  1.1277e-03],\n",
       "                        [ 9.0401e-03,  1.0939e-02, -1.3673e-02,  ..., -8.7622e-03,\n",
       "                          5.9570e-04, -1.2833e-02],\n",
       "                        ...,\n",
       "                        [ 2.0013e-03,  6.9983e-03,  4.6257e-03,  ..., -8.0778e-03,\n",
       "                          1.2425e-02, -9.2927e-03],\n",
       "                        [ 1.3665e-02, -1.7009e-02,  3.3687e-03,  ..., -4.1017e-03,\n",
       "                         -1.0301e-02,  1.4668e-02],\n",
       "                        [ 5.8163e-03, -2.8335e-06,  9.8780e-04,  ...,  1.0928e-02,\n",
       "                         -1.1426e-02, -1.3808e-02]]],\n",
       "              \n",
       "              \n",
       "                      [[[-2.8145e-02, -1.8742e-02, -5.5234e-03,  ...,  4.0994e-03,\n",
       "                          2.8125e-04, -3.5711e-02],\n",
       "                        [-1.9029e-02, -3.9364e-03, -6.7715e-03,  ..., -9.1552e-03,\n",
       "                          1.5949e-02,  4.7563e-03],\n",
       "                        [-7.7093e-03, -3.4516e-02, -2.2431e-02,  ..., -3.2317e-02,\n",
       "                          9.8823e-03,  1.2922e-03],\n",
       "                        ...,\n",
       "                        [-2.9283e-02, -2.6186e-02, -1.6897e-02,  ..., -4.7393e-03,\n",
       "                         -3.1830e-02, -1.0085e-02],\n",
       "                        [-3.5755e-02, -3.4435e-02, -1.2168e-02,  ..., -2.0105e-02,\n",
       "                         -4.2665e-03, -2.0468e-02],\n",
       "                        [-3.3734e-02, -1.6996e-02, -9.8357e-03,  ..., -2.9043e-02,\n",
       "                         -3.3291e-02, -1.3980e-02]],\n",
       "              \n",
       "                       [[-6.3367e-03, -4.0852e-03, -1.3327e-02,  ..., -6.2933e-03,\n",
       "                          1.3628e-02,  8.6691e-03],\n",
       "                        [ 1.3802e-02, -7.2884e-03, -1.4474e-02,  ...,  1.3953e-02,\n",
       "                         -5.3896e-03,  9.0456e-03],\n",
       "                        [-6.9317e-03,  1.6150e-02,  8.1835e-03,  ..., -3.0702e-04,\n",
       "                          1.3830e-02, -1.6633e-03],\n",
       "                        ...,\n",
       "                        [-8.3356e-04, -9.3114e-03,  7.4694e-03,  ..., -1.2985e-02,\n",
       "                         -1.4350e-02,  4.2311e-03],\n",
       "                        [-5.0927e-03,  1.0877e-02,  4.0688e-03,  ...,  9.2379e-03,\n",
       "                         -1.0290e-02,  1.0049e-02],\n",
       "                        [-9.4879e-03,  1.0646e-02,  1.3236e-02,  ..., -9.2430e-03,\n",
       "                         -1.0380e-03, -1.4155e-02]],\n",
       "              \n",
       "                       [[-3.6744e-02, -2.1535e-02, -3.1799e-02,  ..., -1.4108e-02,\n",
       "                          2.2022e-03,  1.5080e-02],\n",
       "                        [-2.9610e-02, -4.6011e-03, -1.4337e-02,  ..., -8.4109e-03,\n",
       "                          4.4273e-03, -2.2268e-02],\n",
       "                        [-1.1714e-02, -5.3899e-03, -1.1769e-02,  ..., -2.0215e-02,\n",
       "                         -4.4180e-03, -9.1483e-03],\n",
       "                        ...,\n",
       "                        [-3.1818e-02, -2.7536e-02, -1.3271e-02,  ..., -6.5540e-03,\n",
       "                         -7.9542e-03, -1.0677e-02],\n",
       "                        [-9.0505e-03, -3.1940e-02, -1.5119e-02,  ..., -2.1554e-02,\n",
       "                         -1.0521e-02, -2.7976e-02],\n",
       "                        [-2.0902e-02, -3.2283e-03, -1.3045e-02,  ..., -3.6176e-02,\n",
       "                         -2.9619e-02, -1.6289e-02]],\n",
       "              \n",
       "                       ...,\n",
       "              \n",
       "                       [[-1.4943e-02,  1.2392e-03,  1.6166e-02,  ..., -1.1477e-02,\n",
       "                         -1.5734e-02, -1.2199e-02],\n",
       "                        [ 1.1283e-02,  1.1453e-03,  1.2160e-02,  ...,  6.1564e-03,\n",
       "                          3.5162e-03, -8.8078e-03],\n",
       "                        [ 1.5972e-02,  1.8743e-03, -1.4592e-02,  ...,  7.7799e-03,\n",
       "                         -4.6349e-03,  3.2799e-03],\n",
       "                        ...,\n",
       "                        [ 8.4355e-03, -4.7946e-04, -1.7618e-02,  ..., -1.6736e-02,\n",
       "                         -6.3067e-03,  6.9063e-03],\n",
       "                        [ 1.0057e-02,  5.4642e-03,  1.1619e-02,  ...,  1.4251e-02,\n",
       "                         -8.2950e-03,  1.3677e-02],\n",
       "                        [ 3.1885e-03, -1.5601e-05,  1.7460e-02,  ..., -1.1777e-02,\n",
       "                         -6.7834e-03,  3.3414e-03]],\n",
       "              \n",
       "                       [[-3.0452e-02, -1.2080e-02,  1.0256e-02,  ..., -3.0990e-02,\n",
       "                          8.6174e-03,  1.6501e-03],\n",
       "                        [-5.2719e-04, -3.2968e-02, -2.6674e-02,  ..., -1.6440e-02,\n",
       "                         -3.6365e-03, -3.2362e-02],\n",
       "                        [-2.7233e-02, -2.7331e-02, -3.5724e-02,  ..., -9.7884e-03,\n",
       "                         -3.2167e-02, -9.0837e-03],\n",
       "                        ...,\n",
       "                        [-2.0292e-02, -3.3341e-03, -7.6577e-03,  ...,  1.4283e-02,\n",
       "                         -1.4460e-03, -1.3133e-02],\n",
       "                        [-3.3671e-02, -1.4031e-02, -7.2689e-03,  ...,  1.6328e-02,\n",
       "                         -7.3200e-03, -2.9146e-02],\n",
       "                        [-2.9474e-02, -2.9688e-02, -1.0572e-02,  ..., -3.2480e-02,\n",
       "                         -1.3261e-02, -1.5741e-02]],\n",
       "              \n",
       "                       [[-2.3804e-03,  5.5173e-03, -1.2740e-02,  ..., -4.8662e-03,\n",
       "                         -7.2586e-03,  1.0922e-02],\n",
       "                        [-6.3082e-03, -9.7869e-03, -1.6668e-02,  ..., -1.0346e-02,\n",
       "                         -7.1375e-04, -1.4136e-02],\n",
       "                        [-9.3555e-03,  5.5218e-03,  1.6871e-02,  ..., -7.9774e-03,\n",
       "                          1.0834e-02,  6.2287e-03],\n",
       "                        ...,\n",
       "                        [-1.0119e-02, -1.4812e-03, -2.6835e-03,  ...,  5.7517e-03,\n",
       "                          1.2835e-02,  8.3581e-03],\n",
       "                        [-1.7669e-02, -4.3534e-04, -1.3433e-02,  ...,  1.3524e-02,\n",
       "                          5.1122e-04,  1.2765e-02],\n",
       "                        [-1.2379e-02, -4.0298e-03,  1.3297e-02,  ..., -9.6577e-03,\n",
       "                         -8.4847e-03, -1.0439e-02]]],\n",
       "              \n",
       "              \n",
       "                      [[[-2.2538e-03, -4.1233e-03, -1.0215e-02,  ..., -2.4550e-03,\n",
       "                         -2.0032e-02, -2.6995e-02],\n",
       "                        [-3.4619e-02, -1.8392e-02, -2.5360e-02,  ..., -1.4965e-02,\n",
       "                         -1.0830e-02, -3.1260e-02],\n",
       "                        [-1.0557e-02, -2.3744e-02,  1.6909e-03,  ..., -6.7882e-03,\n",
       "                          1.5083e-03, -1.6190e-02],\n",
       "                        ...,\n",
       "                        [-1.7339e-03,  6.6879e-03, -9.7566e-03,  ...,  1.1324e-02,\n",
       "                         -2.9404e-02,  1.9570e-02],\n",
       "                        [ 1.0242e-02, -9.2871e-03, -9.3168e-03,  ..., -1.3156e-02,\n",
       "                         -1.4865e-02, -1.1270e-02],\n",
       "                        [ 1.9390e-02, -8.4842e-03,  9.8267e-03,  ...,  4.0320e-03,\n",
       "                         -2.6696e-02, -1.6334e-02]],\n",
       "              \n",
       "                       [[-9.2508e-03, -1.1350e-02, -1.7665e-02,  ...,  4.3904e-03,\n",
       "                          2.7492e-03, -5.2415e-03],\n",
       "                        [-8.4670e-03, -7.3156e-03, -1.9307e-02,  ..., -2.3375e-02,\n",
       "                         -1.0639e-02, -1.7652e-02],\n",
       "                        [-2.3184e-02, -1.7577e-02, -6.0105e-03,  ...,  4.4489e-03,\n",
       "                         -2.6661e-02,  3.8427e-03],\n",
       "                        ...,\n",
       "                        [-7.8044e-03, -1.7909e-02,  2.7993e-03,  ..., -1.1152e-03,\n",
       "                          7.6839e-03, -1.1714e-02],\n",
       "                        [-9.9899e-03,  2.4783e-03, -3.1628e-03,  ..., -1.6662e-02,\n",
       "                         -1.4662e-02,  6.5154e-03],\n",
       "                        [-1.4905e-02, -1.5384e-02, -1.1265e-02,  ..., -4.1862e-03,\n",
       "                         -1.9127e-03, -2.4848e-02]],\n",
       "              \n",
       "                       [[ 3.4846e-03,  8.0098e-03,  1.7443e-02,  ...,  1.2480e-02,\n",
       "                         -1.2735e-02,  8.6934e-03],\n",
       "                        [-8.2303e-03, -1.7087e-02, -1.0395e-02,  ..., -1.6124e-02,\n",
       "                          7.0822e-03, -3.8555e-04],\n",
       "                        [-5.2301e-03,  4.9656e-03,  9.1322e-03,  ..., -3.9297e-03,\n",
       "                         -1.1945e-02, -3.1185e-03],\n",
       "                        ...,\n",
       "                        [ 1.0450e-02,  1.8954e-02,  2.9704e-03,  ...,  1.4643e-02,\n",
       "                          1.6012e-02, -3.2929e-03],\n",
       "                        [ 1.0485e-02,  7.6941e-03,  7.7856e-03,  ...,  6.2871e-03,\n",
       "                          7.3308e-03,  1.3855e-02],\n",
       "                        [-1.3529e-02, -2.3717e-03, -1.4512e-02,  ...,  1.0026e-02,\n",
       "                         -1.6163e-02, -5.7107e-03]],\n",
       "              \n",
       "                       ...,\n",
       "              \n",
       "                       [[-1.4811e-02, -2.7844e-02, -2.0868e-02,  ..., -2.6010e-02,\n",
       "                         -2.7611e-02,  6.3969e-03],\n",
       "                        [-2.2012e-03, -1.9145e-02, -1.9199e-02,  ..., -2.0287e-02,\n",
       "                         -2.2382e-02, -2.2633e-03],\n",
       "                        [-1.0251e-02, -1.5751e-02, -2.7219e-02,  ...,  2.9438e-03,\n",
       "                         -2.6035e-02, -2.6711e-02],\n",
       "                        ...,\n",
       "                        [-3.0947e-04,  5.9995e-03,  2.1770e-05,  ..., -1.0780e-02,\n",
       "                          4.2132e-03,  3.1212e-03],\n",
       "                        [-1.9642e-02, -1.5439e-02, -1.3787e-02,  ..., -2.8270e-02,\n",
       "                         -8.3740e-03, -6.7641e-03],\n",
       "                        [-6.7095e-03, -1.8616e-02, -2.7250e-02,  ..., -2.8718e-02,\n",
       "                         -2.0472e-02,  4.0441e-04]],\n",
       "              \n",
       "                       [[-2.8262e-02, -7.3320e-03, -1.9765e-02,  ..., -8.0278e-03,\n",
       "                          1.7557e-02,  1.8543e-02],\n",
       "                        [-1.7225e-02, -1.3504e-02, -2.6834e-02,  ..., -2.0847e-02,\n",
       "                         -3.4311e-03,  8.1041e-03],\n",
       "                        [-2.3213e-02,  8.8160e-03,  1.5753e-02,  ..., -1.5599e-02,\n",
       "                         -2.2783e-02,  1.7687e-03],\n",
       "                        ...,\n",
       "                        [-1.2956e-02, -3.4006e-02, -1.7315e-02,  ..., -6.5962e-03,\n",
       "                         -3.2383e-02,  1.5012e-02],\n",
       "                        [ 1.8599e-02, -1.7383e-02, -1.5422e-02,  ..., -9.3824e-03,\n",
       "                         -1.5369e-02, -2.1481e-02],\n",
       "                        [-3.9189e-04, -1.3674e-02, -2.4964e-02,  ..., -9.4626e-03,\n",
       "                         -1.0716e-02, -1.9111e-03]],\n",
       "              \n",
       "                       [[-2.4925e-02, -1.9391e-03,  1.8677e-03,  ..., -7.9364e-03,\n",
       "                         -1.1269e-02, -1.0375e-02],\n",
       "                        [-2.2959e-02,  1.3838e-03, -5.1879e-03,  ..., -2.0389e-02,\n",
       "                          1.8722e-04, -2.2228e-02],\n",
       "                        [-2.5972e-02, -1.9240e-02, -1.4870e-02,  ..., -1.1329e-02,\n",
       "                         -1.0629e-02, -1.2778e-03],\n",
       "                        ...,\n",
       "                        [ 4.6231e-03, -9.8229e-03, -1.6067e-02,  ..., -1.4338e-02,\n",
       "                          1.0771e-03, -1.4400e-02],\n",
       "                        [-1.5673e-02, -4.5755e-03, -1.2823e-02,  ..., -1.7801e-02,\n",
       "                          2.3182e-03, -1.8283e-02],\n",
       "                        [ 2.1563e-03, -5.6639e-03,  6.3025e-03,  ..., -1.9270e-02,\n",
       "                         -2.7177e-02,  1.7473e-04]]],\n",
       "              \n",
       "              \n",
       "                      ...,\n",
       "              \n",
       "              \n",
       "                      [[[ 7.6536e-03, -1.8783e-03,  1.8841e-02,  ..., -8.4884e-03,\n",
       "                         -2.2051e-03, -5.5705e-03],\n",
       "                        [-1.1814e-02, -1.4657e-02,  1.1615e-02,  ..., -3.1463e-03,\n",
       "                         -3.4147e-02, -2.1097e-02],\n",
       "                        [ 1.7014e-02, -1.1913e-03, -1.3618e-02,  ..., -2.0940e-02,\n",
       "                         -1.9440e-02, -3.0029e-02],\n",
       "                        ...,\n",
       "                        [ 1.0215e-02,  1.5527e-02, -6.3834e-03,  ...,  6.0160e-03,\n",
       "                         -1.6559e-02, -2.9295e-03],\n",
       "                        [ 1.0632e-02, -1.1926e-02, -1.3686e-03,  ...,  1.6902e-02,\n",
       "                         -1.0786e-02, -2.5942e-02],\n",
       "                        [ 3.9939e-03, -3.3778e-02, -1.2943e-02,  ...,  1.9375e-03,\n",
       "                         -2.5962e-02, -2.6246e-02]],\n",
       "              \n",
       "                       [[-1.8836e-02, -1.2895e-02, -2.8608e-02,  ..., -1.1196e-02,\n",
       "                         -1.9619e-02, -3.0789e-02],\n",
       "                        [-3.1237e-02, -3.7592e-02, -8.0640e-03,  ..., -2.0608e-02,\n",
       "                         -7.0762e-03, -3.3432e-02],\n",
       "                        [-3.4544e-02, -3.1376e-02, -1.8247e-02,  ..., -4.2953e-03,\n",
       "                         -2.5268e-02, -3.7524e-02],\n",
       "                        ...,\n",
       "                        [-2.0865e-02, -2.8106e-02, -4.8804e-03,  ...,  4.9353e-03,\n",
       "                         -9.5544e-03, -3.9235e-04],\n",
       "                        [-2.0468e-02, -4.8933e-03, -2.7534e-02,  ..., -1.2462e-02,\n",
       "                         -2.6990e-02, -7.4737e-03],\n",
       "                        [-2.3287e-02, -2.4754e-02, -2.8761e-02,  ..., -1.5842e-02,\n",
       "                         -2.1052e-02, -3.2656e-02]],\n",
       "              \n",
       "                       [[ 8.0048e-04,  2.2151e-03,  1.5473e-02,  ..., -1.2530e-02,\n",
       "                         -1.6299e-02, -3.9971e-03],\n",
       "                        [-1.7649e-02,  6.1190e-03,  5.3151e-03,  ..., -2.9324e-02,\n",
       "                         -1.8867e-02, -3.1613e-02],\n",
       "                        [-3.7128e-02, -3.5173e-02, -3.1409e-02,  ..., -2.0030e-02,\n",
       "                         -3.2676e-03, -3.1105e-02],\n",
       "                        ...,\n",
       "                        [-2.1572e-02, -3.5458e-02, -2.1744e-02,  ..., -4.4794e-03,\n",
       "                         -4.9605e-03, -2.9795e-02],\n",
       "                        [-1.9600e-02, -1.8060e-02, -2.9290e-02,  ..., -1.0334e-02,\n",
       "                         -2.3278e-02, -2.6473e-02],\n",
       "                        [-3.4211e-02, -5.7403e-03, -3.1517e-02,  ..., -2.5251e-02,\n",
       "                         -2.8706e-02, -3.5840e-02]],\n",
       "              \n",
       "                       ...,\n",
       "              \n",
       "                       [[ 1.1339e-02, -1.7595e-02,  3.8017e-03,  ...,  2.5522e-03,\n",
       "                         -1.7517e-02,  4.3034e-03],\n",
       "                        [ 9.7192e-03,  1.5814e-02,  5.0818e-03,  ...,  1.2481e-02,\n",
       "                         -3.0841e-02,  3.4227e-03],\n",
       "                        [-1.1800e-02,  4.6344e-03,  2.4301e-03,  ..., -3.6808e-02,\n",
       "                         -3.3497e-02, -2.8690e-02],\n",
       "                        ...,\n",
       "                        [ 3.6129e-03, -1.1561e-02, -3.0395e-02,  ..., -1.2388e-02,\n",
       "                         -9.4928e-03, -1.5173e-03],\n",
       "                        [-1.1360e-02, -4.0666e-03, -2.0922e-02,  ...,  8.6673e-04,\n",
       "                         -1.3469e-02, -4.8741e-03],\n",
       "                        [ 5.7896e-03, -1.7382e-02, -7.8392e-03,  ..., -2.3734e-03,\n",
       "                         -1.1630e-02, -7.0134e-03]],\n",
       "              \n",
       "                       [[-3.2164e-02, -6.2788e-03,  1.3968e-02,  ...,  3.6180e-03,\n",
       "                         -1.0840e-02, -2.7565e-02],\n",
       "                        [-2.6980e-02, -2.4073e-02,  1.0033e-03,  ..., -2.5264e-04,\n",
       "                         -5.5285e-03, -1.2939e-02],\n",
       "                        [ 2.1923e-04, -1.4998e-03, -2.6163e-02,  ..., -4.7604e-03,\n",
       "                          9.5060e-03,  2.7361e-03],\n",
       "                        ...,\n",
       "                        [-1.5853e-02,  1.8252e-02, -1.6496e-02,  ..., -3.9139e-03,\n",
       "                         -2.4966e-02, -3.1966e-03],\n",
       "                        [-2.7687e-02, -2.8252e-02,  1.8075e-03,  ..., -2.1508e-02,\n",
       "                          3.0193e-03, -1.8301e-02],\n",
       "                        [ 3.9693e-03,  1.4852e-02,  6.8361e-03,  ..., -2.5844e-02,\n",
       "                         -1.3109e-02, -2.7742e-02]],\n",
       "              \n",
       "                       [[ 1.9207e-03, -1.1992e-02,  1.2561e-02,  ..., -2.8050e-02,\n",
       "                         -2.8156e-02, -3.3904e-02],\n",
       "                        [ 1.7325e-02,  1.1727e-02,  4.4550e-03,  ..., -2.3177e-02,\n",
       "                         -3.0512e-02, -1.1348e-02],\n",
       "                        [ 1.2288e-02,  3.4642e-04, -2.0631e-02,  ..., -9.7264e-03,\n",
       "                         -1.9462e-02, -9.1786e-03],\n",
       "                        ...,\n",
       "                        [-1.4261e-02, -1.8814e-02, -9.2540e-03,  ..., -6.0748e-03,\n",
       "                         -1.2992e-02,  4.8125e-04],\n",
       "                        [-1.3841e-02, -2.1739e-02, -1.6609e-02,  ..., -3.4722e-03,\n",
       "                         -3.2457e-03,  1.4795e-02],\n",
       "                        [-4.3364e-03, -8.1214e-03, -1.2953e-02,  ..., -9.9520e-03,\n",
       "                          3.6117e-03, -7.1621e-03]]],\n",
       "              \n",
       "              \n",
       "                      [[[-2.1822e-02, -1.6658e-02, -7.6389e-03,  ..., -2.2257e-02,\n",
       "                          7.1451e-03, -1.5845e-03],\n",
       "                        [ 1.1925e-03, -3.5487e-03, -1.8371e-02,  ...,  1.0235e-02,\n",
       "                         -2.0721e-02,  7.1239e-03],\n",
       "                        [-1.4366e-02,  7.3110e-03, -9.1615e-04,  ..., -5.8821e-03,\n",
       "                          1.5872e-02,  2.1394e-02],\n",
       "                        ...,\n",
       "                        [-3.5594e-03,  6.8701e-03, -1.0854e-02,  ..., -2.3067e-02,\n",
       "                         -3.2157e-03, -1.1290e-02],\n",
       "                        [ 2.6806e-02,  3.3598e-04, -1.0921e-02,  ..., -1.1420e-02,\n",
       "                          4.6863e-03,  6.0898e-03],\n",
       "                        [ 3.0626e-03, -1.7101e-02, -5.4114e-03,  ..., -1.0365e-02,\n",
       "                         -1.0822e-02, -2.6236e-02]],\n",
       "              \n",
       "                       [[-1.3101e-02, -1.3059e-02,  7.6095e-03,  ..., -1.1628e-02,\n",
       "                         -2.6642e-02, -6.9455e-03],\n",
       "                        [-2.5564e-02, -1.5327e-02,  4.0728e-03,  ..., -2.5864e-02,\n",
       "                         -2.2147e-02,  4.5925e-03],\n",
       "                        [-8.2005e-03,  5.9101e-03, -6.4722e-03,  ..., -1.4526e-02,\n",
       "                         -4.1891e-03, -1.5764e-02],\n",
       "                        ...,\n",
       "                        [-2.6733e-02, -2.7073e-02, -1.7734e-02,  ..., -1.5771e-02,\n",
       "                         -8.3647e-04,  5.2966e-03],\n",
       "                        [ 7.4729e-03, -2.3610e-02, -1.9502e-02,  ..., -1.5676e-02,\n",
       "                          1.1866e-03,  4.3461e-04],\n",
       "                        [-2.7914e-03, -3.7813e-03, -2.0831e-02,  ..., -1.0264e-02,\n",
       "                          8.9520e-04, -2.4958e-02]],\n",
       "              \n",
       "                       [[-6.4742e-03, -1.8906e-02, -5.6844e-03,  ...,  7.3853e-03,\n",
       "                          2.2777e-03, -6.9292e-03],\n",
       "                        [ 2.8477e-03,  2.0978e-03,  3.7272e-03,  ..., -7.4928e-03,\n",
       "                          1.6647e-02,  1.2275e-02],\n",
       "                        [ 1.1717e-02, -4.8451e-04,  2.7634e-02,  ..., -6.2747e-03,\n",
       "                         -7.3154e-03,  4.8498e-04],\n",
       "                        ...,\n",
       "                        [ 1.0360e-02,  2.0931e-02,  1.1289e-02,  ...,  3.9727e-03,\n",
       "                         -1.4086e-02,  1.7404e-02],\n",
       "                        [-2.5335e-03, -2.7018e-03,  1.6044e-02,  ..., -3.9768e-03,\n",
       "                         -1.3957e-02, -1.4771e-02],\n",
       "                        [ 1.6544e-02,  2.1240e-02,  1.7475e-02,  ..., -2.5158e-02,\n",
       "                         -1.9777e-02, -1.7363e-04]],\n",
       "              \n",
       "                       ...,\n",
       "              \n",
       "                       [[ 1.4401e-02,  1.2646e-03, -1.3304e-02,  ...,  7.7538e-03,\n",
       "                         -1.9440e-02, -2.1017e-02],\n",
       "                        [ 1.6771e-02, -2.3978e-04, -5.0331e-03,  ..., -7.6139e-03,\n",
       "                         -2.1629e-04,  6.4135e-03],\n",
       "                        [ 6.8014e-03,  9.8460e-03, -7.2331e-03,  ..., -7.9818e-03,\n",
       "                          3.1137e-03, -2.2293e-02],\n",
       "                        ...,\n",
       "                        [-1.4249e-02,  2.2670e-03, -2.0135e-02,  ..., -1.9543e-04,\n",
       "                          9.2510e-04, -3.7952e-03],\n",
       "                        [ 2.5527e-02,  3.9357e-03, -2.0599e-02,  ..., -1.2822e-02,\n",
       "                         -2.0320e-02,  5.3710e-03],\n",
       "                        [ 2.4788e-02, -5.9360e-03, -1.9608e-02,  ..., -8.5284e-03,\n",
       "                         -5.3702e-03,  2.7787e-03]],\n",
       "              \n",
       "                       [[-1.3842e-02,  9.3919e-04,  1.0927e-02,  ..., -2.3249e-02,\n",
       "                          2.1997e-02, -2.7843e-03],\n",
       "                        [-1.8800e-02,  3.3210e-03, -5.7883e-03,  ..., -3.2548e-02,\n",
       "                          1.0355e-02, -7.5103e-05],\n",
       "                        [ 8.8634e-04, -2.7359e-02,  9.0961e-04,  ..., -8.1291e-03,\n",
       "                         -3.1408e-04, -5.1685e-03],\n",
       "                        ...,\n",
       "                        [-2.1556e-02,  7.1239e-04, -2.7186e-02,  ..., -2.6574e-02,\n",
       "                         -1.7427e-02, -7.3701e-03],\n",
       "                        [ 1.7685e-02,  1.9347e-03, -2.5782e-03,  ..., -1.3726e-02,\n",
       "                         -1.9573e-02, -3.1529e-05],\n",
       "                        [ 1.6412e-02, -2.5740e-02,  1.9379e-03,  ..., -2.6090e-02,\n",
       "                         -7.1123e-04,  1.2818e-02]],\n",
       "              \n",
       "                       [[-2.5872e-02, -1.1990e-02, -1.4029e-02,  ...,  4.0431e-03,\n",
       "                          9.0229e-04, -1.4076e-02],\n",
       "                        [-1.9221e-02, -2.4752e-02, -1.5499e-02,  ..., -7.9520e-03,\n",
       "                         -7.1294e-03, -4.3111e-03],\n",
       "                        [ 3.4378e-03, -1.3511e-02, -1.3479e-02,  ..., -1.0406e-02,\n",
       "                         -9.2696e-03, -5.8095e-03],\n",
       "                        ...,\n",
       "                        [ 5.9763e-03, -6.7900e-04, -2.3468e-02,  ...,  7.2678e-03,\n",
       "                         -2.4124e-02,  5.2814e-03],\n",
       "                        [-2.6482e-02, -1.1136e-02, -2.3210e-02,  ..., -2.3877e-03,\n",
       "                         -9.3244e-03, -1.3438e-02],\n",
       "                        [-1.4087e-02,  8.4801e-04,  4.2950e-03,  ..., -2.6069e-02,\n",
       "                          5.7832e-03, -1.9748e-02]]],\n",
       "              \n",
       "              \n",
       "                      [[[-1.2947e-02, -1.3831e-02, -6.6514e-03,  ...,  1.1251e-02,\n",
       "                         -2.7273e-03, -2.9318e-04],\n",
       "                        [ 1.6319e-02, -4.2163e-03,  9.5971e-04,  ...,  8.7225e-03,\n",
       "                          3.4388e-03, -4.2100e-03],\n",
       "                        [-8.9748e-03,  6.1063e-03, -1.0801e-03,  ...,  5.1986e-03,\n",
       "                          8.0231e-03, -2.8249e-02],\n",
       "                        ...,\n",
       "                        [-3.6449e-03, -2.3086e-03, -2.5649e-02,  ..., -1.1459e-02,\n",
       "                         -5.9115e-03,  5.8053e-03],\n",
       "                        [-3.4264e-02, -3.3125e-02, -3.9721e-03,  ..., -1.2224e-02,\n",
       "                          1.9460e-02, -5.8133e-03],\n",
       "                        [-3.4569e-02, -3.0598e-02, -3.6949e-02,  ...,  5.6422e-03,\n",
       "                         -3.6649e-03,  8.9027e-05]],\n",
       "              \n",
       "                       [[-2.5274e-02, -2.2699e-02, -2.4185e-02,  ..., -1.9850e-02,\n",
       "                         -1.6500e-02, -2.5491e-02],\n",
       "                        [-2.0654e-03, -1.8466e-02, -1.2426e-02,  ...,  1.5545e-03,\n",
       "                         -1.6919e-02, -6.4482e-03],\n",
       "                        [-2.1573e-03,  3.2694e-03, -8.2869e-04,  ..., -1.9123e-02,\n",
       "                         -3.0052e-02, -6.3218e-03],\n",
       "                        ...,\n",
       "                        [-2.0544e-02, -1.8392e-02, -1.7850e-02,  ..., -1.2365e-02,\n",
       "                         -1.7570e-02, -2.4573e-02],\n",
       "                        [-3.4599e-02, -1.4640e-02, -3.5621e-02,  ..., -2.3353e-02,\n",
       "                         -1.7889e-02, -3.3007e-02],\n",
       "                        [-2.4938e-02, -5.0069e-03, -2.1860e-02,  ..., -2.3937e-02,\n",
       "                         -1.6531e-02, -2.4050e-02]],\n",
       "              \n",
       "                       [[ 1.8155e-02, -3.6595e-04,  2.9583e-03,  ..., -1.4410e-02,\n",
       "                          1.0625e-02,  1.4375e-03],\n",
       "                        [-4.8929e-03, -8.0628e-03,  1.1450e-02,  ..., -1.5520e-02,\n",
       "                          1.3821e-02, -1.5309e-02],\n",
       "                        [ 1.1759e-02, -3.2002e-03, -1.5472e-02,  ..., -9.7236e-03,\n",
       "                          4.4048e-04,  1.5701e-02],\n",
       "                        ...,\n",
       "                        [ 3.1347e-03, -1.0985e-02, -3.2569e-02,  ..., -9.3643e-03,\n",
       "                          4.0229e-03, -4.7593e-03],\n",
       "                        [-1.4315e-02, -1.1992e-02,  1.4437e-02,  ..., -9.9650e-03,\n",
       "                          1.5973e-03,  1.9719e-03],\n",
       "                        [-1.6013e-02,  2.7376e-03,  1.3703e-02,  ...,  4.5032e-03,\n",
       "                         -8.3267e-03, -1.5758e-02]],\n",
       "              \n",
       "                       ...,\n",
       "              \n",
       "                       [[-1.6352e-02, -7.9971e-03, -3.6600e-04,  ..., -2.4343e-02,\n",
       "                          4.1341e-03, -1.2726e-02],\n",
       "                        [-7.5895e-03, -2.5708e-02,  6.5365e-05,  ..., -1.3495e-03,\n",
       "                         -5.5134e-03, -1.8683e-02],\n",
       "                        [-1.0314e-02, -9.2459e-03, -5.5137e-03,  ..., -1.4481e-02,\n",
       "                         -9.4491e-03, -1.8040e-02],\n",
       "                        ...,\n",
       "                        [-3.3748e-02, -2.9064e-02, -5.2415e-03,  ..., -8.1776e-03,\n",
       "                         -1.4581e-02, -2.4246e-02],\n",
       "                        [-6.7054e-03, -6.5064e-03, -5.5634e-03,  ..., -1.8302e-02,\n",
       "                         -1.3915e-02, -1.9616e-02],\n",
       "                        [-1.2851e-02, -3.7664e-02, -7.9373e-03,  ..., -1.1542e-02,\n",
       "                         -2.7742e-02, -1.2360e-02]],\n",
       "              \n",
       "                       [[ 5.9764e-03,  1.1124e-02,  3.3182e-03,  ...,  1.0216e-02,\n",
       "                         -1.5117e-02,  1.3690e-02],\n",
       "                        [ 4.9395e-04, -7.5153e-03,  8.6296e-03,  ...,  8.1994e-03,\n",
       "                          7.1977e-03,  2.5272e-03],\n",
       "                        [-3.7261e-03, -2.5792e-03, -3.5059e-02,  ...,  1.6435e-02,\n",
       "                          1.1533e-02, -5.8290e-03],\n",
       "                        ...,\n",
       "                        [-1.1111e-02, -1.8693e-02, -3.3552e-02,  ..., -1.2267e-02,\n",
       "                          8.4279e-04, -3.2727e-02],\n",
       "                        [-3.3329e-02, -2.4417e-02, -3.4563e-03,  ..., -3.7300e-03,\n",
       "                         -4.3485e-03, -2.6191e-03],\n",
       "                        [-2.4551e-02, -2.3301e-02, -5.5629e-03,  ..., -2.4263e-02,\n",
       "                         -7.4279e-03, -5.4921e-03]],\n",
       "              \n",
       "                       [[-1.8330e-02, -3.5020e-02, -4.5647e-03,  ..., -1.6847e-02,\n",
       "                         -1.8493e-02, -2.4594e-02],\n",
       "                        [-4.1776e-03, -2.8195e-02, -1.9245e-02,  ..., -2.0432e-02,\n",
       "                         -4.7981e-03, -1.4396e-02],\n",
       "                        [-1.4868e-02, -1.7694e-02, -2.6225e-02,  ..., -3.2991e-03,\n",
       "                         -3.6838e-03, -4.8610e-03],\n",
       "                        ...,\n",
       "                        [-2.5331e-03, -4.7969e-03, -2.0132e-02,  ..., -4.7765e-03,\n",
       "                         -5.7052e-03, -4.8429e-03],\n",
       "                        [-3.4539e-03, -3.3050e-02, -1.4698e-02,  ..., -1.2370e-02,\n",
       "                         -2.3638e-02, -2.4548e-02],\n",
       "                        [-1.5635e-02, -3.4954e-02, -1.7280e-02,  ..., -1.2131e-02,\n",
       "                         -2.3583e-02, -1.5356e-02]]]])),\n",
       "             ('3.bias', tensor(1.00000e-02 *\n",
       "                     [-1.0348, -0.6432, -2.8054,  0.4010,  0.0355, -1.6032, -2.2993,\n",
       "                      -1.0429, -0.9652,  0.4127, -1.8207, -1.2974, -0.3693,  0.1035,\n",
       "                      -3.5085,  0.2993, -1.0232, -0.8651,  0.9668, -2.0374, -0.7152,\n",
       "                       0.3597, -1.8070,  0.8926, -3.2788, -0.0156, -1.5564, -1.5726,\n",
       "                      -0.7210,  4.4880, -0.0121, -2.9784, -1.1779, -1.4455,  1.3124,\n",
       "                       1.3883, -2.0414, -0.6788, -0.2914,  4.7307,  0.1653, -0.8118,\n",
       "                      -0.6951, -1.2998,  1.0289,  1.4498, -0.9608, -2.2262, -1.0067,\n",
       "                       0.1164,  1.2346, -1.9317, -0.8354, -0.7387, -0.8948, -0.9721,\n",
       "                      -0.1986, -0.8877, -0.2728, -1.8956, -1.8667, -3.1181, -1.0578,\n",
       "                      -0.4842])),\n",
       "             ('6.weight',\n",
       "              tensor([[[[-6.6457e-03, -1.4171e-02, -2.9045e-03,  ..., -1.8137e-02,\n",
       "                         -2.4039e-02,  6.8482e-03],\n",
       "                        [ 7.1761e-03, -4.3415e-04, -3.1281e-03,  ..., -2.2076e-02,\n",
       "                          3.7009e-03, -2.6765e-02],\n",
       "                        [ 1.5630e-03, -1.9424e-03,  9.1876e-04,  ..., -1.7380e-02,\n",
       "                          7.2866e-03, -1.5511e-02],\n",
       "                        ...,\n",
       "                        [-6.0648e-03, -7.3269e-03, -1.2549e-02,  ..., -5.2128e-03,\n",
       "                         -8.3172e-03, -1.7570e-02],\n",
       "                        [-7.9316e-03, -2.4722e-02,  2.5008e-03,  ..., -2.3667e-02,\n",
       "                         -2.3427e-02, -1.3126e-02],\n",
       "                        [-6.7987e-03, -1.5110e-02, -8.4143e-03,  ..., -1.4240e-04,\n",
       "                         -1.7958e-02,  8.3597e-04]],\n",
       "              \n",
       "                       [[-7.3525e-03,  7.4036e-03,  4.3098e-04,  ..., -1.5146e-02,\n",
       "                          7.1143e-03,  6.2732e-04],\n",
       "                        [ 1.6696e-02, -6.1884e-03, -1.7679e-02,  ...,  3.8439e-03,\n",
       "                         -1.2341e-02, -1.3297e-02],\n",
       "                        [ 2.2285e-02, -7.4451e-03,  9.5357e-03,  ...,  5.9170e-03,\n",
       "                         -1.4165e-02,  5.7909e-03],\n",
       "                        ...,\n",
       "                        [ 8.5605e-03,  1.5439e-03,  3.1369e-03,  ...,  2.6004e-02,\n",
       "                          1.6715e-02, -6.2791e-03],\n",
       "                        [-2.9113e-03, -5.9755e-03,  2.4794e-02,  ...,  1.4971e-02,\n",
       "                          2.4299e-02,  4.9134e-03],\n",
       "                        [ 8.1300e-03,  2.7531e-02,  2.3008e-02,  ...,  2.6274e-02,\n",
       "                          2.3459e-02,  8.4176e-03]],\n",
       "              \n",
       "                       [[ 1.6915e-03,  1.5662e-02, -2.1010e-02,  ..., -2.5390e-02,\n",
       "                         -1.8454e-02, -2.5313e-03],\n",
       "                        [-2.4235e-03, -1.7186e-02, -1.6349e-02,  ..., -1.0117e-02,\n",
       "                         -2.0963e-03,  4.7621e-03],\n",
       "                        [ 5.7097e-03, -5.0673e-03, -1.0786e-02,  ..., -2.2081e-02,\n",
       "                         -1.2243e-02, -1.6443e-02],\n",
       "                        ...,\n",
       "                        [ 1.8973e-04,  5.9129e-03,  2.4763e-02,  ..., -1.0624e-02,\n",
       "                          2.7799e-02,  1.7528e-02],\n",
       "                        [ 1.5302e-02,  1.1425e-02,  1.0583e-03,  ..., -2.5907e-02,\n",
       "                         -6.2298e-03,  2.6912e-02],\n",
       "                        [ 1.3923e-03,  3.1232e-03,  2.4442e-02,  ...,  4.6041e-03,\n",
       "                          1.3593e-03,  3.2995e-04]],\n",
       "              \n",
       "                       ...,\n",
       "              \n",
       "                       [[ 2.5818e-02, -2.5142e-03,  1.4200e-03,  ..., -2.1599e-03,\n",
       "                         -2.2257e-03, -1.5570e-02],\n",
       "                        [ 1.1808e-02,  5.1975e-03, -1.2687e-02,  ..., -9.0135e-03,\n",
       "                         -1.0521e-02, -7.2020e-03],\n",
       "                        [ 2.6967e-02,  1.1292e-02,  4.3858e-03,  ..., -1.5487e-02,\n",
       "                         -7.1060e-03, -2.4728e-02],\n",
       "                        ...,\n",
       "                        [ 6.9493e-03, -7.3873e-03, -1.4296e-03,  ..., -1.5632e-03,\n",
       "                         -9.1091e-03, -1.7611e-02],\n",
       "                        [-8.2599e-03, -1.2565e-02, -9.7295e-04,  ..., -1.3228e-02,\n",
       "                         -2.3057e-02,  2.0873e-03],\n",
       "                        [ 2.5067e-02,  2.2780e-02, -2.1365e-02,  ..., -2.5433e-02,\n",
       "                         -1.2683e-02, -1.1745e-02]],\n",
       "              \n",
       "                       [[ 5.5638e-04, -1.4176e-02, -2.1675e-02,  ...,  6.9348e-03,\n",
       "                         -2.5002e-02, -7.3456e-03],\n",
       "                        [-2.0352e-02,  6.1517e-03, -1.1137e-02,  ..., -1.0299e-02,\n",
       "                         -7.1797e-03, -1.7937e-02],\n",
       "                        [-2.4783e-02,  7.8138e-03, -8.3013e-03,  ..., -7.0645e-03,\n",
       "                          5.7574e-03, -6.3606e-03],\n",
       "                        ...,\n",
       "                        [-8.1102e-03,  5.2446e-04,  1.3026e-02,  ...,  3.2308e-04,\n",
       "                         -6.0556e-03, -2.3312e-02],\n",
       "                        [-1.9393e-02, -1.0644e-02, -1.3462e-02,  ..., -2.9526e-03,\n",
       "                         -1.1095e-02, -4.4025e-03],\n",
       "                        [-2.0711e-02, -1.2897e-02, -2.3291e-02,  ..., -2.2207e-02,\n",
       "                         -2.4636e-02, -9.8060e-03]],\n",
       "              \n",
       "                       [[-2.7536e-02,  2.2127e-04, -2.0344e-02,  ..., -2.6237e-02,\n",
       "                         -1.6757e-02, -1.0473e-02],\n",
       "                        [ 2.9774e-03, -3.4460e-03,  4.1607e-04,  ..., -1.8696e-02,\n",
       "                         -1.7803e-02, -1.5318e-02],\n",
       "                        [-6.8567e-03, -5.4140e-03, -2.2211e-03,  ..., -1.5153e-02,\n",
       "                         -9.4086e-03, -2.4556e-02],\n",
       "                        ...,\n",
       "                        [-2.0466e-02, -1.5368e-02, -1.3362e-03,  ..., -1.2661e-02,\n",
       "                         -2.7657e-02, -2.6702e-02],\n",
       "                        [-1.6134e-04, -5.4864e-03, -6.9648e-03,  ..., -8.6749e-03,\n",
       "                         -1.6099e-02, -1.1674e-02],\n",
       "                        [-3.9682e-03, -2.1888e-04, -1.8551e-02,  ...,  3.1419e-03,\n",
       "                         -1.6960e-02, -3.9024e-03]]],\n",
       "              \n",
       "              \n",
       "                      [[[ 2.6815e-03,  2.7600e-03, -2.3110e-02,  ..., -6.3854e-03,\n",
       "                          6.4987e-03,  1.1546e-03],\n",
       "                        [-1.9149e-02,  2.1954e-03, -2.2608e-02,  ...,  1.9906e-03,\n",
       "                         -9.6423e-03, -1.6311e-02],\n",
       "                        [-4.5714e-04, -6.5635e-03, -1.8411e-02,  ...,  2.6722e-02,\n",
       "                         -1.0920e-02, -2.1461e-02],\n",
       "                        ...,\n",
       "                        [ 6.3454e-03, -7.8911e-03, -2.8480e-03,  ..., -1.6411e-02,\n",
       "                          6.4979e-03, -6.4111e-03],\n",
       "                        [-2.7195e-02, -2.6426e-02,  6.3926e-03,  ...,  7.5039e-03,\n",
       "                          1.6723e-02, -2.7925e-03],\n",
       "                        [-9.6588e-03, -1.8814e-03, -3.7692e-03,  ..., -2.3274e-02,\n",
       "                         -3.0475e-04, -2.4586e-02]],\n",
       "              \n",
       "                       [[-1.9053e-02, -1.4415e-02, -3.9138e-04,  ..., -2.2224e-02,\n",
       "                         -2.5606e-02, -2.7464e-03],\n",
       "                        [ 3.4185e-03,  6.5735e-04, -6.7643e-03,  ...,  3.8953e-03,\n",
       "                         -2.2330e-02,  3.5497e-04],\n",
       "                        [-1.7709e-03, -2.3377e-02, -6.9347e-03,  ..., -1.8656e-02,\n",
       "                         -1.9381e-02, -6.5882e-03],\n",
       "                        ...,\n",
       "                        [ 7.6387e-03, -2.6929e-02, -7.6107e-04,  ..., -2.7788e-02,\n",
       "                         -1.9311e-02, -2.3376e-02],\n",
       "                        [-1.8804e-02, -1.5826e-02, -2.3658e-02,  ..., -4.7362e-03,\n",
       "                         -3.9745e-03, -2.0802e-02],\n",
       "                        [-2.3835e-02, -2.0257e-03,  6.2832e-03,  ..., -1.6132e-02,\n",
       "                         -2.4613e-02, -1.9445e-02]],\n",
       "              \n",
       "                       [[ 3.7441e-03, -2.8301e-04, -2.1012e-02,  ..., -2.1116e-02,\n",
       "                         -1.9632e-02,  4.1838e-03],\n",
       "                        [ 5.3022e-03, -1.0277e-02,  4.0184e-03,  ...,  1.8427e-03,\n",
       "                         -1.7806e-02, -5.5921e-03],\n",
       "                        [-2.4765e-02, -1.2122e-02, -3.3981e-03,  ..., -2.7065e-03,\n",
       "                          7.5494e-03, -8.5881e-03],\n",
       "                        ...,\n",
       "                        [-2.5949e-03, -1.2332e-02, -9.9704e-04,  ...,  6.1100e-03,\n",
       "                         -2.1041e-02,  1.5915e-03],\n",
       "                        [-2.9155e-03, -1.2471e-03,  2.3624e-03,  ..., -4.0809e-04,\n",
       "                          2.2107e-03,  5.9018e-03],\n",
       "                        [-2.7314e-02, -8.8494e-03,  3.4153e-03,  ...,  1.9356e-03,\n",
       "                         -2.2871e-02, -1.3754e-02]],\n",
       "              \n",
       "                       ...,\n",
       "              \n",
       "                       [[-1.4542e-02, -1.1539e-02,  8.9263e-05,  ..., -7.3279e-03,\n",
       "                         -2.6264e-02, -1.3329e-02],\n",
       "                        [-2.7454e-02,  3.3421e-03, -2.5765e-02,  ..., -1.0423e-02,\n",
       "                         -2.6844e-02,  7.3982e-03],\n",
       "                        [-2.3613e-02, -1.3673e-02, -1.8357e-02,  ..., -1.7721e-02,\n",
       "                         -8.5604e-03, -2.2132e-02],\n",
       "                        ...,\n",
       "                        [-2.6013e-02, -3.6969e-03,  2.5747e-03,  ..., -3.1840e-03,\n",
       "                         -1.5127e-02, -5.5397e-03],\n",
       "                        [-1.4254e-02, -2.5712e-02, -1.9869e-02,  ..., -1.7686e-02,\n",
       "                          4.4819e-03, -9.8499e-03],\n",
       "                        [-8.2287e-03, -2.6694e-02, -2.0531e-02,  ..., -1.7120e-03,\n",
       "                         -2.0798e-02, -1.7615e-02]],\n",
       "              \n",
       "                       [[-1.1641e-02,  1.9956e-03,  1.1898e-02,  ...,  3.6846e-03,\n",
       "                          1.1974e-02, -4.5329e-04],\n",
       "                        [-2.5668e-02,  1.7122e-02,  1.7507e-02,  ...,  9.9175e-03,\n",
       "                          2.5714e-02, -7.6476e-04],\n",
       "                        [-1.7509e-02,  4.3558e-03,  7.5123e-04,  ..., -1.4198e-03,\n",
       "                          2.5469e-03,  2.4452e-03],\n",
       "                        ...,\n",
       "                        [-4.9957e-03,  2.1021e-02,  1.2628e-02,  ...,  5.9583e-03,\n",
       "                         -8.1602e-04,  5.6060e-03],\n",
       "                        [ 9.3684e-03, -3.8993e-03, -7.4301e-03,  ...,  2.5262e-02,\n",
       "                         -1.1567e-02, -2.2611e-02],\n",
       "                        [ 1.5262e-02,  8.9769e-03,  9.7152e-03,  ..., -7.4974e-03,\n",
       "                         -4.8129e-03,  5.6780e-03]],\n",
       "              \n",
       "                       [[ 2.3000e-03, -2.4523e-02,  6.6512e-03,  ..., -2.7391e-02,\n",
       "                          6.8425e-03, -2.7274e-03],\n",
       "                        [-7.4881e-03, -2.6908e-02, -1.0024e-02,  ...,  6.7545e-03,\n",
       "                         -1.3386e-02, -2.3434e-02],\n",
       "                        [-1.5010e-03, -1.7844e-03, -1.2954e-02,  ..., -1.4116e-02,\n",
       "                         -5.1217e-03, -8.3684e-03],\n",
       "                        ...,\n",
       "                        [-2.3182e-02, -1.3679e-02, -2.6907e-02,  ...,  7.7890e-03,\n",
       "                          5.6772e-03, -2.3604e-03],\n",
       "                        [-1.3363e-02,  3.2209e-03, -2.5748e-02,  ..., -2.0797e-02,\n",
       "                         -1.4555e-02, -1.4172e-02],\n",
       "                        [-1.8342e-03, -1.0991e-02, -1.9722e-02,  ..., -1.5551e-02,\n",
       "                         -1.0138e-02, -1.0672e-02]]],\n",
       "              \n",
       "              \n",
       "                      [[[ 1.0803e-02,  4.7048e-03,  1.6983e-02,  ...,  4.1527e-03,\n",
       "                          2.2373e-02,  5.4560e-04],\n",
       "                        [ 6.3237e-03,  2.1544e-02,  1.7252e-02,  ..., -2.2766e-03,\n",
       "                          1.3347e-02, -4.3080e-03],\n",
       "                        [ 2.6734e-02,  2.6813e-02,  6.2577e-03,  ...,  2.1421e-02,\n",
       "                         -3.0671e-03,  2.6876e-02],\n",
       "                        ...,\n",
       "                        [ 3.8466e-03,  1.4918e-02,  9.1837e-04,  ...,  1.4482e-02,\n",
       "                         -2.3947e-02,  2.3661e-02],\n",
       "                        [-5.0735e-03,  1.9258e-02,  7.1542e-03,  ...,  2.7190e-02,\n",
       "                         -2.4411e-02,  2.2396e-02],\n",
       "                        [ 5.5550e-03,  2.6562e-03,  2.5110e-02,  ...,  1.3546e-02,\n",
       "                          6.1975e-03,  1.0409e-02]],\n",
       "              \n",
       "                       [[-1.1136e-02,  6.9883e-03,  3.0030e-03,  ..., -5.5232e-04,\n",
       "                         -1.7540e-02, -2.3302e-02],\n",
       "                        [-1.4507e-02,  3.2926e-03, -2.3579e-02,  ..., -1.3034e-02,\n",
       "                         -1.0693e-02, -1.6482e-02],\n",
       "                        [-1.4580e-02, -9.5563e-03, -6.7839e-03,  ..., -1.4063e-02,\n",
       "                         -1.8502e-02, -2.5409e-02],\n",
       "                        ...,\n",
       "                        [-1.0305e-02,  6.6265e-03, -6.3571e-03,  ..., -1.4248e-02,\n",
       "                         -1.3378e-03, -2.7316e-02],\n",
       "                        [-2.7698e-02,  5.9108e-03, -2.4755e-02,  ..., -1.2901e-02,\n",
       "                         -1.3346e-02, -1.8909e-02],\n",
       "                        [-7.2682e-03,  2.3862e-03,  7.2970e-03,  ..., -1.9567e-03,\n",
       "                          2.5399e-03, -1.2041e-02]],\n",
       "              \n",
       "                       [[ 9.3208e-04, -2.4150e-02, -8.2703e-03,  ..., -9.9188e-03,\n",
       "                         -2.7008e-02, -6.4440e-03],\n",
       "                        [-2.8552e-03, -2.5445e-02, -1.7352e-03,  ..., -5.3413e-03,\n",
       "                          3.9580e-03, -1.5058e-02],\n",
       "                        [-2.2481e-02, -7.4268e-03, -1.6228e-02,  ...,  6.2900e-03,\n",
       "                         -4.6813e-03, -2.2208e-02],\n",
       "                        ...,\n",
       "                        [-2.2341e-02,  1.4652e-03, -6.4600e-03,  ..., -1.5374e-02,\n",
       "                         -2.3160e-02, -3.9129e-03],\n",
       "                        [-1.1384e-02, -2.0828e-02,  5.7455e-03,  ..., -1.1452e-02,\n",
       "                         -7.6033e-03,  7.8521e-03],\n",
       "                        [-1.1966e-03, -1.0845e-02, -9.7271e-04,  ..., -9.2989e-03,\n",
       "                         -8.6757e-03, -2.6281e-02]],\n",
       "              \n",
       "                       ...,\n",
       "              \n",
       "                       [[-1.3458e-02, -7.7781e-03, -2.3052e-02,  ...,  2.8025e-03,\n",
       "                         -7.8977e-03, -3.7936e-03],\n",
       "                        [-1.1218e-02, -1.6381e-02,  6.8840e-03,  ...,  3.5669e-03,\n",
       "                         -2.7674e-02, -3.5688e-03],\n",
       "                        [-8.9585e-03, -1.2310e-02, -2.2705e-02,  ..., -9.7054e-03,\n",
       "                         -2.3737e-03, -6.4271e-04],\n",
       "                        ...,\n",
       "                        [ 3.4600e-03, -9.8721e-04,  2.1120e-02,  ..., -2.6087e-02,\n",
       "                         -4.4846e-03, -5.2615e-03],\n",
       "                        [ 8.7108e-04, -8.7200e-03,  1.5654e-02,  ..., -9.8368e-03,\n",
       "                         -1.5175e-02,  1.6077e-03],\n",
       "                        [ 3.6899e-03, -1.1976e-02, -3.6526e-03,  ...,  1.2364e-04,\n",
       "                         -1.8276e-02, -2.1552e-02]],\n",
       "              \n",
       "                       [[ 2.5739e-03,  3.8341e-03,  4.8284e-03,  ...,  1.5254e-02,\n",
       "                         -1.8352e-02, -1.8539e-02],\n",
       "                        [-1.0978e-03,  3.0958e-03, -2.1609e-02,  ..., -1.6518e-02,\n",
       "                          1.1658e-02, -1.6776e-02],\n",
       "                        [ 2.3345e-02,  1.4186e-02,  7.7396e-03,  ...,  2.4179e-02,\n",
       "                          7.5861e-03,  1.4574e-03],\n",
       "                        ...,\n",
       "                        [ 9.2825e-03,  3.2912e-03,  1.9568e-03,  ..., -1.0460e-03,\n",
       "                         -4.2239e-04,  2.6380e-03],\n",
       "                        [ 2.2806e-02,  2.7072e-02,  1.1315e-02,  ...,  5.2665e-03,\n",
       "                          3.7033e-04,  2.7828e-02],\n",
       "                        [ 2.7164e-02,  1.1026e-03,  2.7055e-03,  ...,  1.7518e-02,\n",
       "                          2.3056e-02, -6.8682e-03]],\n",
       "              \n",
       "                       [[-1.4881e-02, -5.9023e-03, -1.9649e-02,  ..., -4.1369e-04,\n",
       "                         -1.3280e-02, -2.6200e-02],\n",
       "                        [-2.6755e-02, -2.1669e-02,  4.5764e-03,  ..., -7.8507e-03,\n",
       "                          6.4255e-03,  9.9325e-03],\n",
       "                        [-8.6054e-03, -7.6855e-03,  5.8127e-03,  ...,  2.5860e-02,\n",
       "                          6.4223e-03,  3.5202e-03],\n",
       "                        ...,\n",
       "                        [-6.3107e-03,  4.4912e-03, -7.6356e-03,  ...,  2.1135e-02,\n",
       "                         -1.6345e-03,  5.7075e-03],\n",
       "                        [ 3.9341e-03,  1.1165e-02, -7.7545e-03,  ...,  1.0315e-02,\n",
       "                         -4.8345e-03,  1.4341e-02],\n",
       "                        [ 1.8492e-02,  1.3883e-02,  1.0993e-02,  ...,  3.8087e-03,\n",
       "                          2.1333e-02,  5.9169e-03]]],\n",
       "              \n",
       "              \n",
       "                      ...,\n",
       "              \n",
       "              \n",
       "                      [[[-1.7079e-02, -2.3652e-02,  1.7911e-02,  ..., -1.2363e-02,\n",
       "                          2.8492e-03,  7.0113e-03],\n",
       "                        [ 1.6031e-03, -2.7487e-02, -1.1339e-03,  ..., -2.4735e-02,\n",
       "                          1.1502e-02,  1.2130e-02],\n",
       "                        [-5.5296e-04, -2.5571e-02, -6.1638e-04,  ...,  1.6285e-02,\n",
       "                          1.5937e-02,  6.5888e-03],\n",
       "                        ...,\n",
       "                        [-5.1860e-04, -5.0372e-03,  8.0611e-03,  ..., -1.0326e-03,\n",
       "                         -1.7013e-02,  6.6016e-03],\n",
       "                        [-1.2407e-02, -7.1872e-03,  2.0173e-02,  ...,  2.6709e-02,\n",
       "                          2.6810e-02, -2.2917e-02],\n",
       "                        [-4.1939e-03,  1.3234e-02,  7.8543e-03,  ...,  5.6746e-03,\n",
       "                         -1.7797e-02,  4.2977e-03]],\n",
       "              \n",
       "                       [[-5.5146e-03, -1.6946e-02, -6.5365e-03,  ..., -2.4159e-02,\n",
       "                         -2.6040e-02, -9.4051e-03],\n",
       "                        [ 7.1273e-03, -4.6073e-03, -8.6110e-03,  ...,  1.7259e-03,\n",
       "                         -3.6051e-03,  2.7651e-03],\n",
       "                        [-1.9700e-02, -2.6362e-02, -9.3747e-03,  ..., -2.6163e-03,\n",
       "                         -1.9860e-02, -4.4046e-04],\n",
       "                        ...,\n",
       "                        [-1.1889e-02, -9.4752e-03, -2.4356e-02,  ..., -2.3420e-02,\n",
       "                         -2.0885e-02, -1.9287e-02],\n",
       "                        [-2.6928e-02, -2.4612e-02,  4.7622e-03,  ..., -1.5503e-02,\n",
       "                         -2.4363e-02,  7.5888e-03],\n",
       "                        [-1.5518e-02, -2.3157e-02, -1.0659e-03,  ..., -2.4205e-02,\n",
       "                         -9.6639e-03, -2.3377e-02]],\n",
       "              \n",
       "                       [[-2.6864e-02, -2.4899e-02, -2.2055e-03,  ..., -1.4565e-02,\n",
       "                         -8.2756e-03, -2.1349e-02],\n",
       "                        [-1.9350e-02, -2.6102e-02, -7.5557e-03,  ..., -2.6522e-02,\n",
       "                         -2.1705e-02, -2.3488e-02],\n",
       "                        [ 6.7886e-03, -2.0861e-02, -7.1895e-03,  ..., -2.0731e-02,\n",
       "                         -1.6870e-02, -1.9063e-02],\n",
       "                        ...,\n",
       "                        [-8.5981e-03, -1.5810e-02, -2.1077e-02,  ...,  1.8970e-02,\n",
       "                         -7.3123e-03,  6.5245e-03],\n",
       "                        [-1.9654e-02, -2.7379e-02, -1.4654e-02,  ...,  1.7281e-02,\n",
       "                         -2.6117e-02, -9.2734e-03],\n",
       "                        [-1.7851e-02, -7.7316e-03,  4.3901e-03,  ..., -3.3982e-03,\n",
       "                          5.3714e-03,  4.5850e-03]],\n",
       "              \n",
       "                       ...,\n",
       "              \n",
       "                       [[ 7.6055e-03, -9.5157e-03, -2.1007e-02,  ...,  2.3747e-03,\n",
       "                          5.1110e-03, -2.7625e-02],\n",
       "                        [-1.3811e-02,  5.6590e-03,  1.1003e-04,  ..., -1.8449e-02,\n",
       "                         -8.6538e-04,  1.8602e-03],\n",
       "                        [ 1.5298e-03,  5.9226e-03, -1.5929e-02,  ..., -7.0295e-03,\n",
       "                         -6.9297e-03, -3.3335e-04],\n",
       "                        ...,\n",
       "                        [-8.5364e-03,  5.2309e-04, -2.4532e-02,  ..., -2.7381e-02,\n",
       "                         -2.3954e-02, -7.8353e-04],\n",
       "                        [-1.2966e-02, -9.3232e-04, -1.1649e-02,  ...,  7.2251e-03,\n",
       "                         -2.4828e-02, -7.1678e-03],\n",
       "                        [-2.7845e-02, -7.3668e-03, -1.3037e-02,  ..., -2.5463e-03,\n",
       "                         -1.1670e-02,  6.9228e-04]],\n",
       "              \n",
       "                       [[ 4.2987e-03, -1.3866e-02, -3.0098e-04,  ..., -2.0312e-02,\n",
       "                         -2.6673e-02, -7.5054e-03],\n",
       "                        [-2.5288e-02, -2.7234e-02, -3.2072e-03,  ..., -1.6922e-03,\n",
       "                          5.1159e-03,  1.8809e-03],\n",
       "                        [-7.8404e-03, -2.4190e-02, -5.6545e-03,  ..., -9.9718e-03,\n",
       "                          2.2418e-03, -1.8270e-03],\n",
       "                        ...,\n",
       "                        [-2.1183e-02, -1.7935e-02, -1.2816e-02,  ..., -4.2227e-03,\n",
       "                          3.4619e-03, -5.6836e-03],\n",
       "                        [ 5.1959e-03,  1.4082e-03, -1.8438e-02,  ...,  5.0958e-03,\n",
       "                         -4.6172e-03, -1.0442e-02],\n",
       "                        [ 3.1867e-03,  1.6476e-02, -2.3770e-03,  ..., -1.1497e-03,\n",
       "                         -5.9099e-03, -1.2312e-05]],\n",
       "              \n",
       "                       [[-4.6278e-03, -1.7988e-02, -1.4015e-02,  ..., -7.2963e-03,\n",
       "                         -1.5966e-02, -2.2753e-02],\n",
       "                        [-2.2051e-02, -1.8920e-02, -2.3424e-02,  ..., -1.3496e-02,\n",
       "                         -4.3788e-03, -1.8417e-02],\n",
       "                        [ 6.6881e-03, -2.0990e-03, -1.6915e-02,  ..., -1.0629e-02,\n",
       "                         -1.8865e-02, -2.7226e-02],\n",
       "                        ...,\n",
       "                        [-2.1795e-02, -9.9422e-03,  1.1856e-03,  ..., -1.4233e-02,\n",
       "                         -5.5066e-03, -1.3657e-03],\n",
       "                        [-9.7043e-03, -1.9177e-02, -1.1190e-02,  ...,  6.3276e-03,\n",
       "                          3.6512e-03, -4.3879e-04],\n",
       "                        [-3.5802e-03, -1.2609e-02,  3.3233e-03,  ...,  6.7619e-04,\n",
       "                          4.9258e-03,  3.0377e-03]]],\n",
       "              \n",
       "              \n",
       "                      [[[-7.1877e-03, -1.5681e-02, -1.6549e-02,  ...,  9.0503e-03,\n",
       "                          2.6306e-02, -2.6235e-03],\n",
       "                        [-2.3821e-02, -8.0970e-03, -7.9879e-03,  ..., -2.6150e-02,\n",
       "                         -1.9651e-02, -2.1417e-02],\n",
       "                        [ 1.0182e-02,  2.1408e-02, -1.7475e-02,  ..., -3.8979e-03,\n",
       "                         -8.5626e-03, -2.2207e-02],\n",
       "                        ...,\n",
       "                        [ 1.2065e-02, -3.3887e-03, -2.4597e-02,  ..., -7.3119e-03,\n",
       "                          5.5528e-04, -2.4736e-02],\n",
       "                        [ 6.9064e-03, -1.4728e-03, -7.0641e-03,  ..., -5.9020e-03,\n",
       "                         -1.9837e-03,  9.0277e-04],\n",
       "                        [ 3.5109e-03,  8.2354e-03,  1.1213e-02,  ...,  2.2563e-03,\n",
       "                         -2.5394e-03, -4.8740e-04]],\n",
       "              \n",
       "                       [[ 1.4663e-02, -2.5992e-02,  6.7058e-03,  ...,  1.3753e-03,\n",
       "                         -1.5902e-02, -3.0446e-04],\n",
       "                        [-1.5684e-02,  1.5662e-03,  1.4345e-04,  ...,  3.9895e-03,\n",
       "                         -1.3194e-02, -1.8605e-02],\n",
       "                        [-1.2445e-02, -1.7827e-02, -3.9375e-03,  ...,  5.2872e-03,\n",
       "                         -1.1453e-03,  6.0658e-03],\n",
       "                        ...,\n",
       "                        [ 1.9661e-02, -2.8060e-03,  5.9395e-03,  ..., -1.0668e-02,\n",
       "                         -1.9702e-02, -1.1407e-02],\n",
       "                        [-5.6370e-03, -8.0981e-03, -9.2554e-03,  ...,  4.9647e-03,\n",
       "                          6.1075e-03, -2.8991e-03],\n",
       "                        [-3.0469e-04,  9.4089e-03, -3.9601e-03,  ..., -1.7853e-02,\n",
       "                         -2.6795e-02, -1.4895e-03]],\n",
       "              \n",
       "                       [[-1.0753e-02,  1.3153e-02, -7.8495e-03,  ..., -6.2965e-03,\n",
       "                         -6.8527e-03, -4.3924e-03],\n",
       "                        [ 2.5400e-03, -2.6458e-02,  3.5085e-03,  ..., -2.7219e-02,\n",
       "                         -1.2093e-02,  4.6735e-03],\n",
       "                        [-1.0301e-02, -2.7772e-02, -2.2750e-02,  ..., -7.2343e-03,\n",
       "                         -5.1331e-04,  2.2712e-03],\n",
       "                        ...,\n",
       "                        [-1.7714e-02, -3.8750e-03,  5.8806e-03,  ...,  1.6281e-02,\n",
       "                         -2.3514e-03,  1.7316e-02],\n",
       "                        [-1.8493e-02, -1.9961e-02, -1.1608e-02,  ...,  2.9879e-03,\n",
       "                         -1.2182e-02,  1.6411e-03],\n",
       "                        [-3.9716e-03, -2.7665e-02, -1.9828e-02,  ..., -6.4639e-03,\n",
       "                         -9.6503e-03,  5.3992e-04]],\n",
       "              \n",
       "                       ...,\n",
       "              \n",
       "                       [[ 1.6866e-02,  2.1281e-02, -1.9741e-02,  ..., -1.8323e-02,\n",
       "                         -2.5315e-02, -2.2827e-02],\n",
       "                        [ 1.2587e-02, -2.1247e-02, -1.3705e-02,  ...,  4.2168e-04,\n",
       "                          4.2513e-03, -1.8466e-02],\n",
       "                        [-2.4485e-02, -1.5450e-02, -1.6921e-02,  ...,  3.0821e-03,\n",
       "                         -1.4335e-02,  5.8570e-03],\n",
       "                        ...,\n",
       "                        [-7.2738e-03, -4.1662e-03, -1.5849e-02,  ...,  5.0289e-04,\n",
       "                         -7.0477e-03, -1.7925e-02],\n",
       "                        [-3.5463e-03, -1.9288e-02, -1.6590e-02,  ..., -4.3331e-03,\n",
       "                          4.0621e-03, -1.3745e-02],\n",
       "                        [ 2.4187e-02,  1.1818e-02, -1.6819e-02,  ..., -2.5773e-02,\n",
       "                         -7.9688e-03, -1.2778e-02]],\n",
       "              \n",
       "                       [[-2.6345e-02, -2.2053e-02, -1.0640e-02,  ..., -1.8684e-02,\n",
       "                         -1.4214e-02, -6.4522e-03],\n",
       "                        [-1.3135e-02, -1.2603e-02,  2.7860e-03,  ..., -1.1153e-02,\n",
       "                         -2.6949e-02, -1.9304e-02],\n",
       "                        [ 6.7585e-03,  2.1437e-03, -2.2104e-02,  ..., -2.6568e-02,\n",
       "                         -1.2645e-02, -4.9750e-03],\n",
       "                        ...,\n",
       "                        [ 9.7202e-05,  2.2102e-02, -1.1010e-02,  ..., -2.9561e-03,\n",
       "                         -1.6349e-02, -8.2710e-03],\n",
       "                        [ 1.4986e-02, -2.2819e-03,  1.5725e-02,  ..., -2.8215e-03,\n",
       "                         -8.3108e-04,  7.6056e-03],\n",
       "                        [ 1.5228e-02,  1.1487e-02, -2.5822e-02,  ...,  9.8976e-03,\n",
       "                         -2.6789e-02, -4.1409e-03]],\n",
       "              \n",
       "                       [[-2.9711e-03, -1.4094e-02,  3.8414e-03,  ..., -2.0743e-02,\n",
       "                         -9.5010e-03, -2.3399e-02],\n",
       "                        [-5.2331e-04, -6.4809e-03, -2.0062e-02,  ...,  7.0671e-03,\n",
       "                         -1.0242e-02, -1.8892e-02],\n",
       "                        [-1.8418e-02, -1.2636e-02,  2.0545e-03,  ..., -1.7045e-03,\n",
       "                         -1.2336e-02, -2.4090e-02],\n",
       "                        ...,\n",
       "                        [-7.5012e-03,  3.8512e-03, -2.6576e-02,  ..., -2.0135e-02,\n",
       "                         -1.1112e-02, -2.2840e-02],\n",
       "                        [-9.6134e-03, -2.5728e-02, -1.8202e-02,  ...,  6.0040e-04,\n",
       "                          7.6143e-03, -7.5557e-03],\n",
       "                        [ 5.8229e-03, -3.4572e-03, -1.0796e-02,  ..., -1.7034e-04,\n",
       "                         -2.3383e-02, -8.9818e-03]]],\n",
       "              \n",
       "              \n",
       "                      [[[-2.3425e-02, -7.5719e-03,  9.8183e-03,  ..., -7.3464e-03,\n",
       "                          1.0208e-02,  2.6022e-02],\n",
       "                        [ 1.9817e-02,  2.0486e-03,  7.5612e-03,  ...,  2.3252e-02,\n",
       "                          3.6919e-03,  2.4617e-02],\n",
       "                        [ 1.5216e-02,  7.6164e-03,  1.2121e-02,  ..., -2.5427e-02,\n",
       "                          5.9616e-03, -1.5313e-02],\n",
       "                        ...,\n",
       "                        [-1.8953e-02, -2.7664e-02, -8.3372e-04,  ..., -1.1394e-02,\n",
       "                         -3.9599e-03, -1.0342e-03],\n",
       "                        [-1.6751e-02, -2.0885e-02, -1.7630e-02,  ..., -1.6645e-03,\n",
       "                         -2.3710e-02, -1.7802e-02],\n",
       "                        [-6.0117e-03,  3.8652e-03, -2.6424e-02,  ..., -2.2235e-02,\n",
       "                         -1.6443e-03, -2.7785e-02]],\n",
       "              \n",
       "                       [[-4.1271e-03,  4.0829e-03,  3.4600e-03,  ..., -2.2573e-02,\n",
       "                         -1.6629e-02, -3.0176e-03],\n",
       "                        [ 2.0742e-02, -8.8202e-03, -2.5944e-02,  ...,  2.3755e-02,\n",
       "                         -6.2048e-03,  1.5848e-02],\n",
       "                        [ 6.0385e-03,  9.0078e-03,  1.1106e-02,  ..., -2.3429e-02,\n",
       "                          6.0346e-03,  2.3764e-02],\n",
       "                        ...,\n",
       "                        [-3.2518e-03,  1.1081e-02, -2.9847e-03,  ...,  2.1810e-03,\n",
       "                          2.1432e-02,  1.4721e-02],\n",
       "                        [ 2.4611e-03,  2.6107e-02, -1.9289e-03,  ...,  1.3872e-02,\n",
       "                          2.3297e-03, -1.8788e-03],\n",
       "                        [ 2.1329e-02,  1.6712e-03,  2.4485e-02,  ...,  1.1636e-02,\n",
       "                          9.4464e-03,  2.6383e-02]],\n",
       "              \n",
       "                       [[-2.0996e-02,  6.0790e-03, -8.0994e-03,  ...,  3.8938e-03,\n",
       "                         -1.3891e-03,  7.1402e-03],\n",
       "                        [-7.0939e-03,  6.2268e-03, -1.4198e-02,  ..., -5.2488e-03,\n",
       "                          1.4681e-02,  1.2842e-02],\n",
       "                        [ 4.9848e-03, -2.7536e-03, -2.2892e-02,  ...,  3.7270e-03,\n",
       "                          2.0247e-02,  1.3076e-02],\n",
       "                        ...,\n",
       "                        [-1.8175e-03,  1.1675e-02,  1.6029e-02,  ...,  3.0613e-03,\n",
       "                          1.4126e-03, -1.0311e-03],\n",
       "                        [ 1.1964e-02, -2.5340e-03,  2.3853e-02,  ...,  6.0123e-03,\n",
       "                          1.8182e-02,  4.9782e-03],\n",
       "                        [ 2.6864e-02, -6.5097e-03,  1.3913e-02,  ..., -5.3833e-03,\n",
       "                          1.0168e-03,  3.6364e-03]],\n",
       "              \n",
       "                       ...,\n",
       "              \n",
       "                       [[-2.2790e-02, -2.2155e-02,  2.9486e-03,  ..., -4.1600e-04,\n",
       "                         -9.5766e-05, -9.7980e-03],\n",
       "                        [ 2.8710e-04, -8.9594e-03, -2.1569e-03,  ...,  3.8048e-03,\n",
       "                         -9.6475e-03, -1.3285e-02],\n",
       "                        [-1.6881e-03, -2.4389e-02, -2.6588e-02,  ..., -1.8146e-02,\n",
       "                         -2.5774e-02, -1.9062e-02],\n",
       "                        ...,\n",
       "                        [ 6.1846e-03, -2.6508e-02, -2.3966e-02,  ..., -8.0941e-03,\n",
       "                         -1.7214e-03, -1.8662e-02],\n",
       "                        [-2.4548e-02,  2.0505e-02, -1.8847e-02,  ..., -1.8596e-02,\n",
       "                         -2.2775e-02, -1.5670e-02],\n",
       "                        [-6.7244e-03,  1.5245e-02,  2.6767e-03,  ..., -9.2259e-04,\n",
       "                         -1.3007e-02,  3.1601e-03]],\n",
       "              \n",
       "                       [[-2.3199e-02, -8.7317e-03, -3.8718e-03,  ...,  7.0642e-03,\n",
       "                         -2.6819e-02, -2.3488e-02],\n",
       "                        [-2.7719e-02,  1.8092e-03, -4.2290e-03,  ..., -3.7541e-03,\n",
       "                          2.3110e-02, -5.9336e-03],\n",
       "                        [-1.3882e-02, -9.4410e-03,  7.1976e-03,  ...,  6.1983e-03,\n",
       "                          7.5404e-03,  2.6457e-02],\n",
       "                        ...,\n",
       "                        [-3.6166e-03, -1.9016e-02,  7.3500e-03,  ..., -7.7176e-03,\n",
       "                         -4.5644e-03, -7.5307e-03],\n",
       "                        [-1.6041e-02, -1.3890e-03, -8.0973e-03,  ..., -2.1373e-02,\n",
       "                         -1.3475e-02, -1.5834e-02],\n",
       "                        [-1.5544e-02, -5.1768e-04,  5.4445e-03,  ..., -1.8930e-02,\n",
       "                         -2.3949e-02, -5.7021e-03]],\n",
       "              \n",
       "                       [[ 1.0378e-03,  8.1250e-04, -1.7659e-02,  ...,  4.4009e-03,\n",
       "                         -5.7019e-04,  5.7071e-03],\n",
       "                        [-8.5712e-04,  5.3408e-03, -2.6760e-02,  ..., -2.3124e-02,\n",
       "                         -5.0353e-03, -1.1309e-02],\n",
       "                        [ 2.5485e-02, -1.7972e-02,  3.4282e-03,  ..., -4.4639e-03,\n",
       "                          1.3660e-03, -8.8214e-04],\n",
       "                        ...,\n",
       "                        [-9.1250e-03, -1.2143e-02, -1.7638e-03,  ..., -2.3895e-02,\n",
       "                         -2.5546e-02,  5.2568e-03],\n",
       "                        [-1.0582e-02,  3.3731e-03, -2.1635e-02,  ..., -1.5461e-02,\n",
       "                         -8.5758e-03, -7.9359e-03],\n",
       "                        [-1.5926e-02, -1.1710e-02, -1.3718e-02,  ..., -1.8985e-02,\n",
       "                          1.9257e-03, -3.7265e-03]]]])),\n",
       "             ('6.bias', tensor(1.00000e-02 *\n",
       "                     [-0.6255,  4.9319, -3.6779, -2.9463, -2.5801,  0.8595, -1.0889,\n",
       "                       4.8599,  0.5028,  1.3037, -0.7622,  3.5637,  1.6952, -2.5448,\n",
       "                      -0.1713,  1.7510, -3.0964, -0.9239,  2.0107, -1.5601, -4.1642,\n",
       "                      -1.7541, -0.9685,  2.0369, -0.7735,  4.4888,  1.0942, -0.2250,\n",
       "                       1.5792,  1.1349, -1.6582, -1.1896, -1.3102,  4.1395, -0.1528,\n",
       "                      -4.3832, -3.0477,  2.4790, -0.6793, -0.5944, -2.2985,  0.9787,\n",
       "                      -1.1641,  0.8887, -2.2574, -0.9951, -1.2451,  2.0503, -7.5850,\n",
       "                      -0.5772, -1.6409,  0.9793, -1.5066,  4.9639, -2.9812, -1.3580,\n",
       "                       0.0373,  1.6401, -0.9795, -2.6780, -0.8956,  3.6520, -1.1645,\n",
       "                       2.0801])),\n",
       "             ('10.weight',\n",
       "              tensor([[-7.5301e-02, -1.7809e-02, -3.9599e-02,  ...,  7.6142e-02,\n",
       "                       -1.0008e-01, -7.2872e-02],\n",
       "                      [ 5.1972e-02, -1.6626e-01,  8.4465e-02,  ..., -7.6805e-02,\n",
       "                       -9.8366e-02,  6.6758e-02],\n",
       "                      [-3.3718e-02,  1.3370e-01, -5.7192e-02,  ..., -9.0067e-02,\n",
       "                        7.1618e-02, -2.5465e-02],\n",
       "                      ...,\n",
       "                      [ 6.2247e-02,  6.4090e-02,  8.5833e-02,  ..., -1.2668e-02,\n",
       "                        1.3935e-02, -5.4030e-02],\n",
       "                      [-2.3584e-01,  1.3796e-01,  8.0857e-02,  ...,  1.2481e-01,\n",
       "                       -4.7108e-02, -1.1746e-01],\n",
       "                      [ 1.1030e-01, -4.2672e-02,  2.4807e-02,  ...,  3.9567e-02,\n",
       "                       -5.2033e-02,  5.8760e-02]])),\n",
       "             ('10.bias',\n",
       "              tensor([-0.0170, -0.0177,  0.0905,  0.0724,  0.0496,  0.0758,  0.0863,\n",
       "                      -0.0121, -0.1114, -0.1325,  0.0691,  0.0418,  0.0570,  0.0775,\n",
       "                       0.1363, -0.0270,  0.0533, -0.0383, -0.0989,  0.0183,  0.0397,\n",
       "                       0.1140,  0.0707,  0.0802,  0.0219, -0.0462, -0.0971, -0.0339,\n",
       "                       0.0596,  0.1036, -0.0411, -0.0024,  0.0568,  0.0786, -0.0063,\n",
       "                      -0.1060,  0.0394, -0.1205,  0.0189, -0.0318,  0.0170, -0.1569,\n",
       "                       0.1222, -0.0652, -0.0800, -0.1372, -0.0851, -0.1614, -0.0519,\n",
       "                      -0.1048,  0.1035, -0.0287, -0.0404, -0.1073,  0.0188,  0.1039,\n",
       "                      -0.0469,  0.0518,  0.0717,  0.0868,  0.0811,  0.0636,  0.0492,\n",
       "                      -0.0230, -0.0766, -0.0199,  0.0445, -0.0969,  0.0910, -0.0910,\n",
       "                       0.0038, -0.0456,  0.0315, -0.0613, -0.0293, -0.0596,  0.0228,\n",
       "                      -0.0958,  0.1407, -0.0356, -0.0213, -0.0660,  0.0504,  0.0503,\n",
       "                      -0.0463,  0.0843,  0.0917,  0.0356, -0.0272, -0.1018, -0.0236,\n",
       "                      -0.1267, -0.0688,  0.0875,  0.0440, -0.0899,  0.0881,  0.0511,\n",
       "                       0.1372, -0.0741])),\n",
       "             ('12.weight',\n",
       "              tensor([[-0.0017, -0.0766,  0.0924,  ...,  0.0853, -0.0931, -0.1333],\n",
       "                      [-0.1053,  0.0104,  0.0136,  ..., -0.0441, -0.0201,  0.0947],\n",
       "                      [ 0.0697,  0.0490, -0.0532,  ..., -0.0319,  0.1249, -0.0447],\n",
       "                      ...,\n",
       "                      [-0.0374, -0.0541,  0.0537,  ...,  0.0963,  0.0487,  0.0269],\n",
       "                      [ 0.0725,  0.0446, -0.0674,  ..., -0.0942,  0.0055, -0.0619],\n",
       "                      [-0.0330,  0.0317, -0.0500,  ..., -0.0020, -0.0695, -0.1809]])),\n",
       "             ('12.bias',\n",
       "              tensor([-0.0489,  0.1297, -0.0611,  0.0806, -0.0072, -0.0580, -0.0646,\n",
       "                       0.0661, -0.0694, -0.0294, -0.0115, -0.0775,  0.0346, -0.0762,\n",
       "                      -0.1076,  0.0488, -0.1113, -0.0511, -0.0543,  0.0004, -0.0727,\n",
       "                      -0.0120, -0.1211,  0.1344, -0.0039,  0.0428,  0.0673,  0.1046,\n",
       "                       0.0122, -0.0327, -0.0499,  0.0923, -0.1074, -0.0446,  0.0439,\n",
       "                      -0.1016, -0.0942, -0.1311, -0.0221,  0.1263,  0.0584, -0.0257,\n",
       "                      -0.0426,  0.0923,  0.0596,  0.0151,  0.0068, -0.0514, -0.0368,\n",
       "                      -0.0243,  0.0778,  0.0376,  0.0819, -0.0541,  0.0811,  0.0558,\n",
       "                      -0.0639,  0.0004,  0.0526, -0.0735,  0.0347, -0.0209, -0.0446,\n",
       "                      -0.1415, -0.0052]))])"
      ]
     },
     "execution_count": 52,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "model.state_dict()"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": []
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": []
  }
 ],
 "metadata": {
  "kernelspec": {
   "display_name": "Python 3",
   "language": "python",
   "name": "python3"
  },
  "language_info": {
   "codemirror_mode": {
    "name": "ipython",
    "version": 3
   },
   "file_extension": ".py",
   "mimetype": "text/x-python",
   "name": "python",
   "nbconvert_exporter": "python",
   "pygments_lexer": "ipython3",
   "version": "3.6.5"
  }
 },
 "nbformat": 4,
 "nbformat_minor": 2
}