TransferLearning.ipynb

{
 "cells": [
  {
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "# Transfer Learning Pytorch\n",
    "\n",
    "#### References\n",
    "* https://pytorch.org/tutorials/beginner/finetuning_torchvision_models_tutorial.html\n",
    "* https://pytorch.org/tutorials/beginner/transfer_learning_tutorial.html\n",
    "* https://github.com/mortezamg63/Accessing-and-modifying-different-layers-of-a-pretrained-model-in-pytorch\n",
    "* https://discuss.pytorch.org/t/insert-new-layer-in-the-middle-of-a-pre-trained-model/12414/4\n",
    "* https://forums.fast.ai/t/pytorch-best-way-to-get-at-intermediate-layers-in-vgg-and-resnet/5707/2\n",
    "* https://discuss.pytorch.org/t/extracting-and-using-features-from-a-pretrained-model/20723/6"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 2,
   "metadata": {},
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "PyTorch Version:  1.1.0\n",
      "Torchvision Version:  0.3.0\n"
     ]
    }
   ],
   "source": [
    "import torch\n",
    "import torch.nn as nn\n",
    "import torch.optim as optim\n",
    "import numpy as np\n",
    "import torchvision\n",
    "from torchvision import datasets, models, transforms\n",
    "import matplotlib.pyplot as plt\n",
    "from torch.hub import load_state_dict_from_url\n",
    "import time\n",
    "import os\n",
    "import copy\n",
    "import utils_resnet_TL as utils_resnet\n",
    "print(\"PyTorch Version: \",torch.__version__)\n",
    "print(\"Torchvision Version: \",torchvision.__version__)\n",
    "input_size = 224\n",
    "num_classes = 2"
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "#### Get definition of some module\n",
    "What we're interested here is to inspect how resnet18 is implemented.\n",
    "``` python\n",
    "def forward(self, x):\n",
    "        x = self.conv1(x)\n",
    "        x = self.bn1(x)\n",
    "        x = self.relu(x)\n",
    "        x = self.maxpool(x)\n",
    "\n",
    "        x = self.layer1(x)\n",
    "        x = self.layer2(x)\n",
    "        x = self.layer3(x)\n",
    "        x = self.layer4(x)\n",
    "\n",
    "        x = self.avgpool(x)\n",
    "        x = x.reshape(x.size(0), -1)\n",
    "        x = self.fc(x)\n",
    "\n",
    "        return x\n",
    "```"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 3,
   "metadata": {},
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "Resnet Source: /mnt/anaconda3/lib/python3.7/site-packages/torchvision/models/resnet.py\n"
     ]
    }
   ],
   "source": [
    "import inspect\n",
    "resnet_source = inspect.getfile(models.resnet18)\n",
    "print('Resnet Source:', resnet_source)\n",
    "#with open(resnet_source, 'r') as f:\n",
    "#    print(f.read())"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 4,
   "metadata": {},
   "outputs": [],
   "source": [
    "def set_parameter_requires_grad(model, feature_extracting=True):\n",
    "    if feature_extracting:\n",
    "        # Mark parameters to be freezed\n",
    "        for param in model.parameters():\n",
    "            param.requires_grad = False"
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "#### Load Pretrained Resnet18"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 5,
   "metadata": {},
   "outputs": [],
   "source": [
    "model_ft = models.resnet18(pretrained=True)\n",
    "set_parameter_requires_grad(model_ft, feature_extracting=True)\n",
    "# Print model structure\n",
    "#print(model_ft)"
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "#### Make changes on the final layer\n",
    "The vanilla way of doing transfer learning consist on substituting the final layer of a model and retrain."
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 6,
   "metadata": {},
   "outputs": [],
   "source": [
    "# Get original number of features\n",
    "num_ftrs = model_ft.fc.in_features\n",
    "model_ft.fc = nn.Linear(num_ftrs, num_classes)"
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "### More advanced way of changing base model\n",
    "Imagine that you need change something on the middle of the model, or if you want to make the model output more things (change the forward definition). The way we described before can't be used for that.\n",
    "\n",
    "The following example will prepare the Resnet model to be used as feature extraction for RetinaNet, or ChangeNet"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 7,
   "metadata": {},
   "outputs": [],
   "source": [
    "class ResnetFeatures(models.ResNet):\n",
    "    def __init__(self, block, layers, num_classes=1000):\n",
    "        super(ResnetFeatures, self).__init__(block, layers, num_classes)        \n",
    "        \n",
    "    def forward(self, x):\n",
    "        x = self.conv1(x)\n",
    "        x = self.bn1(x)\n",
    "        x = self.relu(x)\n",
    "        x = self.maxpool(x)\n",
    "\n",
    "        layer1 = self.layer1(x)\n",
    "        layer2 = self.layer2(layer1)\n",
    "        layer3 = self.layer3(layer2)\n",
    "        layer4 = self.layer4(layer3)\n",
    "\n",
    "        #x = self.avgpool(x)\n",
    "        #x = x.reshape(x.size(0), -1)\n",
    "        #x = self.fc(x)\n",
    "        return layer1, layer2, layer3, layer4\n",
    "\n"
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "#### Redefinition of some helper function\n",
    "Instantiate custom Resnet with pre-trained weights."
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 8,
   "metadata": {},
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "Parameter containing:\n",
      "tensor([[[[-7.9648e-03, -3.7614e-02,  6.2047e-03,  ..., -1.2469e-02,\n",
      "            3.7047e-02,  6.4989e-04],\n",
      "          [ 3.5415e-02,  3.5940e-02, -3.2289e-02,  ...,  3.4159e-03,\n",
      "            1.3469e-02, -1.0689e-02],\n",
      "          [ 9.8453e-05,  2.0491e-02, -5.7877e-03,  ...,  1.6177e-02,\n",
      "           -2.8275e-03, -7.0718e-03],\n",
      "          ...,\n",
      "          [ 1.3436e-02, -5.3551e-02,  3.7207e-03,  ..., -1.1829e-02,\n",
      "           -2.5833e-02,  3.9328e-02],\n",
      "          [-3.8391e-03,  2.4145e-02, -1.8424e-02,  ...,  5.2373e-02,\n",
      "           -4.0217e-02,  2.4476e-02],\n",
      "          [-7.8571e-03,  1.9448e-02, -3.0415e-02,  ..., -3.6407e-02,\n",
      "            1.2486e-02,  7.0944e-04]],\n",
      "\n",
      "         [[ 6.4988e-03,  1.0799e-02,  1.0879e-02,  ..., -1.1941e-02,\n",
      "            2.8047e-02,  1.3347e-02],\n",
      "          [-3.4459e-02, -1.8909e-02, -3.3859e-03,  ...,  5.7913e-05,\n",
      "           -5.4104e-04, -1.8716e-02],\n",
      "          [ 1.5679e-02, -1.0635e-02,  5.9927e-02,  ..., -1.9105e-02,\n",
      "            1.5371e-02, -1.5712e-02],\n",
      "          ...,\n",
      "          [ 5.0537e-02, -4.7143e-03, -4.5992e-03,  ..., -9.5208e-03,\n",
      "            2.0365e-02,  2.3031e-02],\n",
      "          [ 2.5149e-02, -1.8859e-02,  8.3098e-03,  ..., -2.4504e-03,\n",
      "           -1.4681e-02,  5.4466e-02],\n",
      "          [ 2.0917e-02,  1.2838e-02,  4.3636e-03,  ...,  2.0287e-02,\n",
      "           -1.8275e-02, -8.3729e-03]],\n",
      "\n",
      "         [[ 5.4992e-02, -2.6693e-02, -9.0010e-03,  ..., -2.1648e-02,\n",
      "           -4.7102e-03,  1.6682e-02],\n",
      "          [ 8.5835e-03, -2.0958e-02, -1.4349e-02,  ..., -8.5303e-03,\n",
      "           -4.7648e-02,  1.7649e-02],\n",
      "          [ 4.8970e-04,  3.3518e-02,  8.5677e-03,  ...,  5.2430e-04,\n",
      "           -2.1838e-02, -2.8262e-02],\n",
      "          ...,\n",
      "          [ 4.5454e-03,  2.3573e-02, -3.6237e-02,  ...,  1.0651e-02,\n",
      "            2.8090e-02, -2.5246e-02],\n",
      "          [ 1.0932e-02, -3.2793e-02, -2.9905e-02,  ...,  6.9732e-03,\n",
      "           -2.6187e-02,  4.4746e-02],\n",
      "          [ 2.7373e-02,  2.8613e-02, -9.5367e-03,  ...,  6.4723e-02,\n",
      "           -2.7575e-02, -1.4732e-03]]],\n",
      "\n",
      "\n",
      "        [[[ 5.9812e-02,  1.6921e-02, -2.6616e-02,  ..., -1.3912e-02,\n",
      "            3.4703e-02, -1.9128e-02],\n",
      "          [ 2.2046e-02, -1.5744e-02,  3.7807e-02,  ...,  1.9308e-02,\n",
      "            1.7275e-02,  2.8881e-02],\n",
      "          [ 3.1952e-03,  1.4652e-02,  5.7451e-03,  ...,  5.5569e-02,\n",
      "           -2.7984e-02, -1.7476e-02],\n",
      "          ...,\n",
      "          [ 1.8048e-02,  7.0648e-02, -1.3732e-03,  ..., -4.1519e-02,\n",
      "            4.8481e-02, -1.6235e-02],\n",
      "          [ 6.2487e-02, -3.2362e-02,  1.0703e-02,  ..., -1.7769e-02,\n",
      "           -1.2319e-02,  1.9953e-02],\n",
      "          [ 2.7304e-02, -1.9475e-02, -3.3497e-03,  ...,  6.9612e-04,\n",
      "           -3.6034e-02, -2.9815e-02]],\n",
      "\n",
      "         [[ 1.0442e-02,  1.2001e-04, -3.5088e-02,  ...,  1.1578e-02,\n",
      "            2.7381e-02,  5.0817e-03],\n",
      "          [-2.7511e-02, -6.2387e-03,  3.3535e-03,  ..., -1.0102e-02,\n",
      "            4.7344e-02,  4.3532e-02],\n",
      "          [-5.9906e-03, -5.0136e-03, -2.0227e-02,  ..., -2.9089e-02,\n",
      "           -3.0439e-02, -1.4970e-02],\n",
      "          ...,\n",
      "          [ 4.4704e-03, -2.0631e-02,  1.4307e-02,  ..., -6.0400e-02,\n",
      "           -3.5311e-02, -3.9305e-04],\n",
      "          [-1.8036e-02,  1.2293e-02,  1.8986e-03,  ..., -1.6006e-03,\n",
      "           -2.0624e-02, -8.2808e-03],\n",
      "          [ 4.4686e-02,  4.4385e-03, -1.8716e-02,  ..., -1.3353e-02,\n",
      "           -4.6972e-03,  2.1696e-02]],\n",
      "\n",
      "         [[-3.1387e-02,  7.0625e-04,  1.2056e-02,  ...,  1.8504e-02,\n",
      "           -1.7700e-02,  2.2973e-02],\n",
      "          [-1.4069e-02,  1.1633e-02,  1.8648e-02,  ...,  1.1540e-02,\n",
      "           -1.9325e-02,  4.6433e-03],\n",
      "          [ 4.1837e-02, -3.0270e-03, -2.6315e-04,  ...,  1.8291e-03,\n",
      "            7.1453e-03, -1.0708e-02],\n",
      "          ...,\n",
      "          [-1.5115e-02,  9.8550e-03,  2.4965e-02,  ..., -7.4680e-03,\n",
      "            5.1836e-02,  6.2399e-03],\n",
      "          [-2.1219e-02, -1.0298e-02,  1.1634e-02,  ..., -5.4755e-03,\n",
      "            5.3244e-03, -2.1287e-02],\n",
      "          [ 3.8863e-02,  2.9559e-02,  3.5532e-03,  ..., -4.1919e-02,\n",
      "            2.0114e-02, -1.4483e-02]]],\n",
      "\n",
      "\n",
      "        [[[-1.7628e-02, -5.2464e-02,  5.3522e-03,  ...,  4.7917e-03,\n",
      "            2.5364e-02,  1.0860e-02],\n",
      "          [-5.1288e-03,  4.3336e-02,  3.5180e-02,  ..., -5.2237e-02,\n",
      "            7.4615e-03, -1.5305e-02],\n",
      "          [ 1.0829e-02, -2.2180e-02,  3.1028e-05,  ...,  9.5323e-03,\n",
      "            5.9119e-02, -6.9751e-03],\n",
      "          ...,\n",
      "          [ 2.8560e-03,  1.7154e-02, -2.0939e-02,  ...,  3.8602e-02,\n",
      "            7.5591e-03, -2.9892e-02],\n",
      "          [-2.1606e-02,  4.3798e-03,  1.2109e-02,  ...,  2.1571e-02,\n",
      "           -1.1822e-02,  2.4272e-02],\n",
      "          [ 5.8794e-03, -1.7325e-02, -3.1880e-02,  ..., -3.2708e-02,\n",
      "           -3.3655e-02,  3.7985e-02]],\n",
      "\n",
      "         [[-4.9489e-03,  6.3622e-02, -1.1280e-02,  ...,  4.1508e-02,\n",
      "            2.3917e-02, -1.7020e-03],\n",
      "          [ 2.6305e-02,  2.6882e-02, -1.5519e-03,  ..., -4.5774e-03,\n",
      "            2.2647e-02,  1.9694e-02],\n",
      "          [-1.2303e-02,  3.0093e-03,  2.2145e-02,  ...,  3.9984e-02,\n",
      "           -6.0514e-02,  3.2177e-02],\n",
      "          ...,\n",
      "          [-1.7058e-02,  1.0432e-02,  3.2214e-02,  ..., -3.9750e-02,\n",
      "           -9.1049e-03, -2.7948e-02],\n",
      "          [-4.6640e-03, -4.1501e-02,  2.9080e-02,  ...,  9.3570e-03,\n",
      "           -3.2368e-03, -1.6109e-02],\n",
      "          [-3.1777e-02,  3.3439e-02, -1.1375e-02,  ...,  5.7008e-03,\n",
      "            3.2225e-02,  4.1636e-03]],\n",
      "\n",
      "         [[ 3.6966e-02, -1.8077e-02, -6.8812e-03,  ..., -5.3315e-03,\n",
      "            9.6360e-03,  2.2102e-03],\n",
      "          [-2.5517e-02, -4.9137e-02,  3.0111e-02,  ...,  1.3106e-02,\n",
      "           -8.2223e-03,  1.9681e-02],\n",
      "          [-1.5278e-02, -3.5571e-02,  1.8607e-02,  ..., -1.8711e-02,\n",
      "           -1.0931e-02, -3.9332e-02],\n",
      "          ...,\n",
      "          [-2.5590e-02,  1.3377e-03, -3.2702e-02,  ..., -1.9486e-02,\n",
      "            1.2353e-02,  6.4878e-02],\n",
      "          [ 9.1922e-03, -2.2909e-02,  1.4666e-02,  ...,  8.2756e-03,\n",
      "           -1.0988e-02,  1.7784e-02],\n",
      "          [ 4.6347e-02, -1.6017e-02, -7.7536e-04,  ...,  9.1562e-03,\n",
      "           -7.6071e-03, -9.4692e-03]]],\n",
      "\n",
      "\n",
      "        ...,\n",
      "\n",
      "\n",
      "        [[[ 7.1510e-03,  3.8168e-02,  1.1055e-02,  ...,  2.8943e-02,\n",
      "           -5.2489e-03,  4.9101e-05],\n",
      "          [ 1.0034e-03, -5.4912e-02, -2.3308e-02,  ...,  1.1134e-02,\n",
      "           -6.4790e-03,  1.6548e-02],\n",
      "          [ 6.5783e-02, -2.3643e-03, -1.8400e-02,  ..., -3.2866e-02,\n",
      "            2.1341e-02, -5.5078e-02],\n",
      "          ...,\n",
      "          [-2.4905e-02,  1.1082e-02,  1.0689e-02,  ..., -1.8378e-02,\n",
      "            2.6466e-02, -2.4105e-03],\n",
      "          [-6.4471e-03, -1.8564e-02,  1.1335e-02,  ..., -3.5597e-03,\n",
      "           -1.5381e-02, -2.5174e-02],\n",
      "          [-4.9659e-03, -9.9362e-03,  1.4201e-02,  ...,  1.6477e-02,\n",
      "            9.9971e-03, -5.9402e-03]],\n",
      "\n",
      "         [[ 1.7435e-02,  6.7346e-03,  2.1948e-02,  ...,  1.8533e-02,\n",
      "           -4.8223e-02,  2.4784e-02],\n",
      "          [-5.4974e-02,  6.9341e-03, -2.5813e-02,  ..., -1.5388e-02,\n",
      "           -2.5896e-02, -3.4920e-02],\n",
      "          [ 3.8608e-03,  1.2108e-02, -3.4767e-02,  ...,  3.2806e-02,\n",
      "           -4.4831e-03, -1.7234e-02],\n",
      "          ...,\n",
      "          [ 2.8524e-03,  1.5135e-02,  2.1356e-02,  ..., -2.4440e-03,\n",
      "           -2.8433e-02,  3.6056e-02],\n",
      "          [ 1.1886e-02,  3.5404e-02, -1.6468e-02,  ..., -1.2719e-02,\n",
      "            3.8019e-02,  1.2513e-02],\n",
      "          [ 3.3371e-02,  5.0568e-02, -4.2000e-02,  ...,  1.5048e-02,\n",
      "           -5.1184e-02, -1.1113e-03]],\n",
      "\n",
      "         [[ 5.0881e-03,  4.5819e-02,  2.9627e-02,  ...,  1.8866e-02,\n",
      "           -3.6867e-02,  3.1512e-03],\n",
      "          [ 1.2646e-02, -1.6950e-02,  5.5358e-02,  ...,  4.2712e-02,\n",
      "           -4.0193e-02, -1.1375e-02],\n",
      "          [-2.8075e-03, -1.0040e-02, -1.2647e-02,  ..., -2.4508e-03,\n",
      "            2.9859e-02,  1.3643e-02],\n",
      "          ...,\n",
      "          [ 2.3938e-02,  7.5011e-03, -3.7979e-02,  ...,  4.1182e-03,\n",
      "           -1.3060e-02, -3.2860e-02],\n",
      "          [ 7.5823e-03,  3.0192e-02,  6.9570e-03,  ...,  3.0306e-02,\n",
      "           -3.4184e-02,  4.2617e-02],\n",
      "          [-8.8789e-02,  2.8951e-02,  5.0286e-02,  ..., -2.7228e-02,\n",
      "           -2.0076e-02,  3.7554e-02]]],\n",
      "\n",
      "\n",
      "        [[[-1.2722e-02, -2.8895e-02, -1.8584e-02,  ...,  3.2978e-02,\n",
      "           -9.5038e-03, -1.2311e-02],\n",
      "          [ 1.9387e-02,  5.7726e-02,  1.2626e-02,  ...,  1.3754e-02,\n",
      "           -2.0298e-02,  2.3756e-02],\n",
      "          [ 8.8959e-03,  2.7244e-02,  2.0145e-02,  ..., -3.1792e-02,\n",
      "           -2.4044e-02,  1.3646e-02],\n",
      "          ...,\n",
      "          [-1.1861e-02, -2.8148e-02, -3.9370e-03,  ..., -6.1751e-03,\n",
      "           -2.8097e-03, -1.9735e-02],\n",
      "          [-2.8179e-03,  1.4111e-02,  6.2236e-03,  ...,  1.9156e-03,\n",
      "           -1.9871e-02,  6.6981e-03],\n",
      "          [-5.0744e-02,  1.1460e-02, -1.6991e-02,  ..., -1.8085e-02,\n",
      "            7.5382e-03, -1.9976e-02]],\n",
      "\n",
      "         [[ 6.5160e-03,  2.0982e-02, -1.9488e-02,  ..., -1.6098e-02,\n",
      "           -5.0343e-02, -2.5016e-02],\n",
      "          [-1.7582e-02,  6.3516e-03,  7.2353e-02,  ..., -3.9651e-03,\n",
      "            2.4148e-02, -2.7136e-02],\n",
      "          [ 3.1852e-02, -1.2614e-02, -3.6649e-04,  ..., -1.2881e-02,\n",
      "            1.0070e-02,  2.6623e-02],\n",
      "          ...,\n",
      "          [ 4.9581e-02, -1.2984e-02,  1.8160e-02,  ..., -2.7444e-04,\n",
      "            1.1117e-02, -2.5766e-02],\n",
      "          [-4.2114e-02,  4.5808e-03,  2.1952e-02,  ..., -1.3091e-02,\n",
      "            3.0999e-03, -2.1878e-02],\n",
      "          [-2.0369e-02,  2.7848e-02, -3.9850e-02,  ...,  4.0849e-03,\n",
      "            1.4565e-02,  1.5025e-02]],\n",
      "\n",
      "         [[ 5.5908e-02, -4.5792e-05, -4.5013e-02,  ..., -7.2470e-03,\n",
      "           -3.6922e-02,  4.0957e-02],\n",
      "          [-3.3049e-02,  2.1341e-02,  3.1676e-02,  ..., -8.9434e-03,\n",
      "            1.5454e-03,  2.9546e-02],\n",
      "          [-1.5693e-02, -4.3326e-02, -1.9368e-02,  ...,  3.1356e-02,\n",
      "            3.5639e-02,  4.0891e-02],\n",
      "          ...,\n",
      "          [-2.1505e-02,  2.0663e-02,  2.4295e-02,  ..., -2.6329e-02,\n",
      "            6.3726e-02, -2.1297e-02],\n",
      "          [ 1.2989e-03, -1.8642e-02,  3.0918e-02,  ..., -1.0730e-02,\n",
      "            8.0865e-03, -2.0327e-02],\n",
      "          [ 5.2017e-03, -1.4429e-02,  1.8307e-02,  ..., -9.0144e-03,\n",
      "            8.9511e-04, -3.2883e-02]]],\n",
      "\n",
      "\n",
      "        [[[-1.1005e-03, -2.1919e-02, -4.1794e-03,  ..., -6.0815e-03,\n",
      "           -2.7126e-02,  1.7449e-02],\n",
      "          [ 5.9162e-04, -2.0027e-03, -2.2932e-02,  ...,  9.1049e-03,\n",
      "            1.9663e-02, -4.5874e-03],\n",
      "          [ 3.2194e-02, -3.4209e-02,  1.5035e-02,  ..., -2.1612e-02,\n",
      "            2.7444e-02, -4.0208e-02],\n",
      "          ...,\n",
      "          [ 4.6060e-02, -3.0842e-02,  8.5162e-03,  ..., -2.4987e-02,\n",
      "            8.7555e-03,  6.4839e-02],\n",
      "          [-2.0692e-02,  6.1245e-02, -3.7719e-02,  ..., -9.5907e-05,\n",
      "            1.2927e-02, -1.7490e-02],\n",
      "          [ 1.2028e-02,  1.8903e-02, -2.6772e-02,  ...,  2.1545e-02,\n",
      "           -5.2447e-03, -5.1495e-03]],\n",
      "\n",
      "         [[-1.8416e-02,  3.1315e-02, -4.5917e-05,  ..., -1.4202e-02,\n",
      "            5.5175e-03,  5.5052e-02],\n",
      "          [ 1.7641e-02, -2.1564e-02,  2.6993e-02,  ..., -1.9484e-02,\n",
      "           -1.2073e-02, -2.8904e-02],\n",
      "          [-3.2444e-02, -3.6667e-02, -2.1511e-02,  ..., -1.9921e-02,\n",
      "           -7.1856e-03, -1.0874e-02],\n",
      "          ...,\n",
      "          [ 2.3107e-02, -1.4039e-02,  5.9218e-03,  ...,  1.9959e-02,\n",
      "           -2.3728e-02,  6.2114e-02],\n",
      "          [-5.8760e-03,  3.1254e-02,  2.6505e-02,  ..., -1.5126e-02,\n",
      "            3.1594e-03, -1.8121e-02],\n",
      "          [ 2.4820e-02,  3.3954e-02, -4.7607e-02,  ..., -1.3488e-02,\n",
      "            1.9602e-02,  2.2174e-02]],\n",
      "\n",
      "         [[-1.0557e-03, -1.2848e-02,  3.4459e-03,  ...,  2.4252e-02,\n",
      "           -3.8901e-02,  6.3792e-03],\n",
      "          [-3.2665e-02, -1.3563e-02, -1.2358e-02,  ...,  2.3612e-02,\n",
      "            2.2203e-02, -5.2919e-02],\n",
      "          [ 2.3218e-02, -7.1900e-03,  2.9341e-02,  ..., -4.0726e-02,\n",
      "            1.8863e-02, -9.2652e-03],\n",
      "          ...,\n",
      "          [-2.4267e-02,  3.7345e-02, -1.2338e-02,  ..., -1.4021e-02,\n",
      "           -1.2689e-02, -4.0719e-02],\n",
      "          [-1.3577e-02, -5.3801e-02, -2.5948e-02,  ..., -1.3937e-02,\n",
      "           -1.5591e-03, -1.3886e-02],\n",
      "          [-7.6093e-03,  1.4550e-02, -2.1311e-02,  ..., -2.7092e-02,\n",
      "            1.9961e-03, -5.2630e-02]]]], requires_grad=True)\n",
      "Parameter containing:\n",
      "tensor([1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1.], requires_grad=True)\n",
      "Parameter containing:\n",
      "tensor([0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.],\n",
      "       requires_grad=True)\n",
      "Parameter containing:\n",
      "tensor([[[[-0.2000]],\n",
      "\n",
      "         [[ 0.3182]],\n",
      "\n",
      "         [[-0.0041]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[ 0.1972]],\n",
      "\n",
      "         [[-0.3193]],\n",
      "\n",
      "         [[-0.1043]]],\n",
      "\n",
      "\n",
      "        [[[ 0.0152]],\n",
      "\n",
      "         [[-0.2128]],\n",
      "\n",
      "         [[-0.1687]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[-0.2020]],\n",
      "\n",
      "         [[ 0.2002]],\n",
      "\n",
      "         [[ 0.1069]]],\n",
      "\n",
      "\n",
      "        [[[ 0.3902]],\n",
      "\n",
      "         [[-0.0591]],\n",
      "\n",
      "         [[-0.2174]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[ 0.0810]],\n",
      "\n",
      "         [[-0.0532]],\n",
      "\n",
      "         [[ 0.2270]]],\n",
      "\n",
      "\n",
      "        ...,\n",
      "\n",
      "\n",
      "        [[[-0.0959]],\n",
      "\n",
      "         [[ 0.0035]],\n",
      "\n",
      "         [[ 0.0915]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[ 0.2471]],\n",
      "\n",
      "         [[ 0.0177]],\n",
      "\n",
      "         [[-0.0816]]],\n",
      "\n",
      "\n",
      "        [[[ 0.0879]],\n",
      "\n",
      "         [[ 0.1961]],\n",
      "\n",
      "         [[ 0.1702]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[ 0.1110]],\n",
      "\n",
      "         [[-0.0487]],\n",
      "\n",
      "         [[ 0.1846]]],\n",
      "\n",
      "\n",
      "        [[[ 0.0341]],\n",
      "\n",
      "         [[-0.1401]],\n",
      "\n",
      "         [[ 0.0982]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[ 0.1276]],\n",
      "\n",
      "         [[ 0.0659]],\n",
      "\n",
      "         [[ 0.1298]]]], requires_grad=True)\n",
      "Parameter containing:\n",
      "tensor([1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1.], requires_grad=True)\n",
      "Parameter containing:\n",
      "tensor([0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.],\n",
      "       requires_grad=True)\n",
      "Parameter containing:\n",
      "tensor([[[[-9.5835e-02,  9.9635e-03,  7.3890e-02],\n",
      "          [-3.0258e-02,  8.5647e-02, -1.4049e-02],\n",
      "          [ 3.8670e-02, -4.4908e-02,  8.9096e-03]],\n",
      "\n",
      "         [[ 2.6582e-02,  5.7142e-02,  4.5542e-02],\n",
      "          [ 1.2345e-01,  3.3254e-02,  3.8932e-02],\n",
      "          [ 8.9207e-03, -3.9369e-02,  7.8257e-02]],\n",
      "\n",
      "         [[-4.5919e-02, -3.0326e-02, -1.2204e-01],\n",
      "          [-7.2990e-03,  4.6049e-02,  2.5494e-02],\n",
      "          [ 8.2734e-02,  5.3028e-02,  9.8413e-03]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[-1.2991e-02,  3.7277e-02,  8.3062e-02],\n",
      "          [ 2.9690e-02,  3.7584e-02, -9.6329e-02],\n",
      "          [-9.6619e-02,  6.1001e-02, -7.1393e-02]],\n",
      "\n",
      "         [[-2.8800e-03,  6.5227e-03, -3.8877e-02],\n",
      "          [ 4.9568e-03,  2.7448e-02, -2.1928e-02],\n",
      "          [ 5.8134e-02, -5.8741e-03, -7.4373e-02]],\n",
      "\n",
      "         [[ 7.5723e-02,  5.6869e-02,  2.1160e-02],\n",
      "          [ 2.3710e-02,  9.4855e-02, -6.2296e-02],\n",
      "          [-5.9774e-02, -1.4600e-01,  4.6929e-02]]],\n",
      "\n",
      "\n",
      "        [[[-1.4527e-02,  4.9059e-02, -4.5914e-02],\n",
      "          [-6.1507e-02,  4.9850e-02,  2.4573e-02],\n",
      "          [ 2.0438e-02, -4.2267e-02,  6.0894e-02]],\n",
      "\n",
      "         [[ 4.9455e-02, -4.6220e-02,  4.5090e-03],\n",
      "          [ 5.8309e-02, -2.1945e-02,  1.6677e-03],\n",
      "          [ 6.1844e-02,  7.1626e-02,  2.4038e-02]],\n",
      "\n",
      "         [[-5.3141e-02, -1.4970e-02,  2.9355e-02],\n",
      "          [ 8.4002e-02, -4.4516e-02, -2.6761e-02],\n",
      "          [ 1.2515e-01,  2.6423e-02, -9.9878e-02]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[ 1.3413e-02, -6.4756e-02, -4.9423e-02],\n",
      "          [ 1.1252e-02, -9.9862e-02,  5.7871e-02],\n",
      "          [-3.5847e-02, -2.4899e-02,  1.7058e-01]],\n",
      "\n",
      "         [[ 3.0114e-03,  2.5660e-02,  3.0714e-02],\n",
      "          [-8.5461e-02,  7.6477e-02, -2.9758e-02],\n",
      "          [-1.6092e-01, -3.5872e-02,  5.3497e-02]],\n",
      "\n",
      "         [[ 3.7328e-02, -7.4056e-02, -5.2029e-02],\n",
      "          [ 9.9815e-02,  2.7093e-02, -8.1842e-02],\n",
      "          [ 2.3817e-02,  1.3344e-01,  2.6443e-02]]],\n",
      "\n",
      "\n",
      "        [[[-3.2228e-02,  1.2006e-03,  4.0581e-02],\n",
      "          [-1.8791e-02,  1.1237e-01, -4.8216e-02],\n",
      "          [-1.5052e-04,  3.8977e-02, -7.8137e-02]],\n",
      "\n",
      "         [[-8.4345e-02,  7.6923e-03, -2.7787e-02],\n",
      "          [-2.3975e-02, -1.3853e-02,  5.7689e-02],\n",
      "          [ 3.0627e-02,  3.8792e-03,  1.5669e-02]],\n",
      "\n",
      "         [[ 9.4957e-02, -3.6912e-02,  8.0383e-02],\n",
      "          [-2.4382e-02,  1.7449e-03, -4.0589e-03],\n",
      "          [-7.8550e-02, -2.8050e-03,  2.1495e-02]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[-4.5986e-02,  2.5817e-02, -5.2503e-03],\n",
      "          [-2.3621e-02, -7.9289e-02, -4.4981e-02],\n",
      "          [-4.1530e-02,  2.8085e-02,  2.2476e-02]],\n",
      "\n",
      "         [[-2.0422e-02,  7.2098e-03,  2.5515e-02],\n",
      "          [-4.5726e-02,  8.0452e-03,  4.3381e-03],\n",
      "          [-8.4003e-02, -3.7465e-02,  8.0011e-02]],\n",
      "\n",
      "         [[-8.0341e-03,  4.8564e-02,  8.1257e-02],\n",
      "          [-6.9912e-02,  5.4145e-02, -6.3826e-02],\n",
      "          [ 8.0963e-02,  1.6393e-02,  5.9501e-02]]],\n",
      "\n",
      "\n",
      "        ...,\n",
      "\n",
      "\n",
      "        [[[-4.8176e-02, -3.5637e-02,  6.4530e-02],\n",
      "          [ 4.6929e-02, -2.5460e-02, -8.4750e-02],\n",
      "          [-7.0606e-04,  1.2916e-01, -5.9801e-02]],\n",
      "\n",
      "         [[ 1.1793e-01,  4.1632e-02, -8.0660e-02],\n",
      "          [-5.3037e-02,  1.1371e-01,  5.9293e-02],\n",
      "          [-1.1536e-01,  3.1821e-02,  2.8931e-02]],\n",
      "\n",
      "         [[-3.2045e-03,  7.6157e-02, -1.0862e-02],\n",
      "          [ 8.1284e-02, -4.1742e-02,  1.5917e-02],\n",
      "          [ 9.0610e-02,  1.0547e-01, -4.8894e-02]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[-4.1708e-02, -6.2574e-02,  8.9517e-02],\n",
      "          [ 1.9624e-02,  1.7678e-02, -1.4893e-01],\n",
      "          [-1.3343e-01, -1.9481e-02,  1.0159e-01]],\n",
      "\n",
      "         [[-1.2728e-01, -1.9433e-02, -1.7325e-02],\n",
      "          [-4.2121e-03,  3.1627e-03, -7.9826e-02],\n",
      "          [-6.0785e-02,  8.6995e-02, -8.7874e-02]],\n",
      "\n",
      "         [[-7.8675e-02,  4.6700e-02,  4.4709e-02],\n",
      "          [-3.1053e-02, -1.1177e-01, -5.2996e-02],\n",
      "          [-1.3255e-02,  2.6127e-02, -5.1746e-02]]],\n",
      "\n",
      "\n",
      "        [[[ 8.9126e-02,  1.2408e-02,  4.8397e-02],\n",
      "          [-1.2001e-05,  1.5181e-02, -2.0410e-02],\n",
      "          [ 5.1655e-03, -6.3538e-02,  5.8836e-02]],\n",
      "\n",
      "         [[ 1.9529e-02, -3.7229e-02,  1.0189e-02],\n",
      "          [ 9.6955e-02, -2.9802e-02, -3.0398e-02],\n",
      "          [-2.7188e-02, -4.2715e-02,  6.2053e-02]],\n",
      "\n",
      "         [[ 4.1692e-02,  3.7528e-02,  5.2612e-02],\n",
      "          [-2.0067e-02,  6.4347e-02,  1.2223e-01],\n",
      "          [ 1.1012e-02, -2.5184e-02, -5.6820e-02]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[-6.0234e-02,  2.9414e-02, -3.7588e-02],\n",
      "          [-4.4550e-03,  4.3318e-02, -3.8012e-03],\n",
      "          [-9.0359e-02,  3.5624e-02, -4.6465e-02]],\n",
      "\n",
      "         [[ 3.3380e-03,  2.5325e-02, -8.9240e-02],\n",
      "          [ 1.6828e-02,  3.2827e-02,  2.1873e-02],\n",
      "          [ 2.2606e-02,  8.9593e-02,  8.8867e-02]],\n",
      "\n",
      "         [[-9.4841e-02, -1.3972e-02, -1.0667e-02],\n",
      "          [ 4.1940e-02, -4.3675e-03,  5.3756e-02],\n",
      "          [ 7.9053e-02, -9.4415e-02,  6.7883e-02]]],\n",
      "\n",
      "\n",
      "        [[[-2.6645e-02,  1.5754e-01,  1.0064e-01],\n",
      "          [-3.6659e-02,  2.7878e-02,  1.5444e-01],\n",
      "          [ 4.0766e-02, -5.1992e-02, -2.1494e-03]],\n",
      "\n",
      "         [[ 3.5434e-02, -1.3601e-03,  1.2199e-02],\n",
      "          [ 9.2648e-02, -1.0583e-01,  4.9369e-02],\n",
      "          [-4.3113e-02, -4.3787e-02, -3.3358e-02]],\n",
      "\n",
      "         [[ 3.2975e-02, -1.4803e-02, -3.5186e-03],\n",
      "          [ 1.1315e-01,  2.9893e-02, -3.0536e-02],\n",
      "          [-3.5316e-02,  8.7248e-02,  7.4209e-02]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[ 1.2845e-01,  9.5357e-03, -3.6589e-03],\n",
      "          [-1.5698e-01,  1.9756e-02,  3.2336e-02],\n",
      "          [ 6.3583e-02,  7.1584e-04, -2.6468e-02]],\n",
      "\n",
      "         [[ 4.9318e-02, -6.2678e-02, -3.3335e-02],\n",
      "          [-4.9411e-02,  6.2809e-02,  9.2306e-02],\n",
      "          [-2.2539e-02,  1.0111e-02, -3.2429e-02]],\n",
      "\n",
      "         [[ 4.5021e-02, -7.1210e-02,  1.4613e-02],\n",
      "          [ 1.3734e-01,  6.2341e-02, -5.9748e-02],\n",
      "          [ 8.3545e-03, -4.4082e-02, -8.9899e-02]]]], requires_grad=True)\n",
      "Parameter containing:\n",
      "tensor([1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1.], requires_grad=True)\n",
      "Parameter containing:\n",
      "tensor([0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.],\n",
      "       requires_grad=True)\n",
      "Parameter containing:\n",
      "tensor([[[[ 0.0547]],\n",
      "\n",
      "         [[ 0.0199]],\n",
      "\n",
      "         [[ 0.0057]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[ 0.0485]],\n",
      "\n",
      "         [[-0.0548]],\n",
      "\n",
      "         [[ 0.0481]]],\n",
      "\n",
      "\n",
      "        [[[ 0.0267]],\n",
      "\n",
      "         [[ 0.0097]],\n",
      "\n",
      "         [[ 0.1070]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[ 0.0078]],\n",
      "\n",
      "         [[-0.0597]],\n",
      "\n",
      "         [[-0.0181]]],\n",
      "\n",
      "\n",
      "        [[[ 0.0732]],\n",
      "\n",
      "         [[-0.0529]],\n",
      "\n",
      "         [[-0.0274]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[ 0.1066]],\n",
      "\n",
      "         [[ 0.0630]],\n",
      "\n",
      "         [[ 0.0623]]],\n",
      "\n",
      "\n",
      "        ...,\n",
      "\n",
      "\n",
      "        [[[-0.1077]],\n",
      "\n",
      "         [[ 0.0072]],\n",
      "\n",
      "         [[-0.1885]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[-0.0658]],\n",
      "\n",
      "         [[-0.0112]],\n",
      "\n",
      "         [[ 0.0773]]],\n",
      "\n",
      "\n",
      "        [[[ 0.0798]],\n",
      "\n",
      "         [[-0.0196]],\n",
      "\n",
      "         [[ 0.0018]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[ 0.0016]],\n",
      "\n",
      "         [[ 0.0574]],\n",
      "\n",
      "         [[ 0.0555]]],\n",
      "\n",
      "\n",
      "        [[[-0.0345]],\n",
      "\n",
      "         [[-0.0601]],\n",
      "\n",
      "         [[-0.0180]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[-0.0724]],\n",
      "\n",
      "         [[-0.0065]],\n",
      "\n",
      "         [[-0.0526]]]], requires_grad=True)\n",
      "Parameter containing:\n",
      "tensor([1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1.], requires_grad=True)\n",
      "Parameter containing:\n",
      "tensor([0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.],\n",
      "       requires_grad=True)\n",
      "Parameter containing:\n",
      "tensor([[[[-0.1251]],\n",
      "\n",
      "         [[-0.1008]],\n",
      "\n",
      "         [[-0.0103]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[-0.0342]],\n",
      "\n",
      "         [[-0.0519]],\n",
      "\n",
      "         [[-0.0779]]],\n",
      "\n",
      "\n",
      "        [[[ 0.0463]],\n",
      "\n",
      "         [[-0.0245]],\n",
      "\n",
      "         [[ 0.0373]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[-0.0365]],\n",
      "\n",
      "         [[ 0.1634]],\n",
      "\n",
      "         [[-0.0117]]],\n",
      "\n",
      "\n",
      "        [[[ 0.0738]],\n",
      "\n",
      "         [[ 0.0751]],\n",
      "\n",
      "         [[-0.0163]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[-0.0585]],\n",
      "\n",
      "         [[-0.0146]],\n",
      "\n",
      "         [[ 0.0576]]],\n",
      "\n",
      "\n",
      "        ...,\n",
      "\n",
      "\n",
      "        [[[ 0.0441]],\n",
      "\n",
      "         [[ 0.0026]],\n",
      "\n",
      "         [[ 0.0284]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[-0.0234]],\n",
      "\n",
      "         [[ 0.0740]],\n",
      "\n",
      "         [[ 0.0295]]],\n",
      "\n",
      "\n",
      "        [[[-0.0785]],\n",
      "\n",
      "         [[ 0.2005]],\n",
      "\n",
      "         [[ 0.0598]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[-0.1087]],\n",
      "\n",
      "         [[ 0.0080]],\n",
      "\n",
      "         [[-0.0351]]],\n",
      "\n",
      "\n",
      "        [[[-0.0207]],\n",
      "\n",
      "         [[ 0.1042]],\n",
      "\n",
      "         [[-0.1296]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[-0.1050]],\n",
      "\n",
      "         [[ 0.0332]],\n",
      "\n",
      "         [[ 0.1581]]]], requires_grad=True)\n",
      "Parameter containing:\n",
      "tensor([1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1.], requires_grad=True)\n",
      "Parameter containing:\n",
      "tensor([0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.],\n",
      "       requires_grad=True)\n",
      "Parameter containing:\n",
      "tensor([[[[ 0.1156]],\n",
      "\n",
      "         [[ 0.1530]],\n",
      "\n",
      "         [[ 0.1951]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[ 0.1991]],\n",
      "\n",
      "         [[-0.0083]],\n",
      "\n",
      "         [[ 0.0396]]],\n",
      "\n",
      "\n",
      "        [[[-0.0660]],\n",
      "\n",
      "         [[-0.0383]],\n",
      "\n",
      "         [[-0.2203]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[ 0.0520]],\n",
      "\n",
      "         [[ 0.0741]],\n",
      "\n",
      "         [[ 0.1945]]],\n",
      "\n",
      "\n",
      "        [[[ 0.1721]],\n",
      "\n",
      "         [[-0.0558]],\n",
      "\n",
      "         [[-0.0063]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[ 0.1748]],\n",
      "\n",
      "         [[-0.2080]],\n",
      "\n",
      "         [[ 0.1668]]],\n",
      "\n",
      "\n",
      "        ...,\n",
      "\n",
      "\n",
      "        [[[ 0.1126]],\n",
      "\n",
      "         [[ 0.0423]],\n",
      "\n",
      "         [[ 0.1812]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[ 0.0896]],\n",
      "\n",
      "         [[-0.0361]],\n",
      "\n",
      "         [[ 0.3315]]],\n",
      "\n",
      "\n",
      "        [[[ 0.1050]],\n",
      "\n",
      "         [[ 0.0899]],\n",
      "\n",
      "         [[ 0.0004]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[-0.2715]],\n",
      "\n",
      "         [[ 0.2481]],\n",
      "\n",
      "         [[-0.0724]]],\n",
      "\n",
      "\n",
      "        [[[ 0.3508]],\n",
      "\n",
      "         [[-0.1286]],\n",
      "\n",
      "         [[-0.0478]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[ 0.1885]],\n",
      "\n",
      "         [[-0.1570]],\n",
      "\n",
      "         [[-0.0111]]]], requires_grad=True)\n",
      "Parameter containing:\n",
      "tensor([1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1.], requires_grad=True)\n",
      "Parameter containing:\n",
      "tensor([0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.],\n",
      "       requires_grad=True)\n",
      "Parameter containing:\n",
      "tensor([[[[-2.6242e-02, -2.6786e-02,  3.8564e-02],\n",
      "          [ 6.3045e-02, -3.0038e-02,  1.1220e-01],\n",
      "          [-1.4988e-01,  5.3056e-03,  1.9122e-02]],\n",
      "\n",
      "         [[-8.3901e-02, -7.0862e-02,  4.1771e-02],\n",
      "          [-7.2050e-02, -3.3059e-02, -3.7408e-02],\n",
      "          [-2.7291e-02, -8.6397e-02, -8.1468e-02]],\n",
      "\n",
      "         [[ 4.2466e-02,  2.8381e-02, -6.6077e-02],\n",
      "          [ 9.8062e-03, -4.0428e-02,  6.6281e-02],\n",
      "          [-2.0713e-02, -5.6058e-02,  3.1704e-02]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[-1.3614e-02, -3.4245e-02,  2.0246e-02],\n",
      "          [-6.4003e-02, -6.6491e-03,  3.2117e-02],\n",
      "          [ 7.4151e-02,  1.0729e-02, -5.0233e-02]],\n",
      "\n",
      "         [[ 5.6823e-02,  2.3370e-02, -4.0073e-02],\n",
      "          [-1.8118e-02, -9.9994e-02,  5.3214e-02],\n",
      "          [-7.4275e-02,  3.8916e-02,  1.1461e-02]],\n",
      "\n",
      "         [[ 2.2437e-02, -9.5771e-02, -8.0145e-02],\n",
      "          [-5.3902e-02,  5.3158e-02,  6.9605e-02],\n",
      "          [-2.5184e-02,  1.9262e-02, -3.3717e-02]]],\n",
      "\n",
      "\n",
      "        [[[-2.6709e-05,  4.4546e-02, -8.8228e-03],\n",
      "          [ 1.7916e-01,  2.3173e-02, -8.4523e-02],\n",
      "          [-5.6956e-02,  3.5549e-02, -8.9912e-02]],\n",
      "\n",
      "         [[-1.4848e-01,  4.2615e-02, -1.0622e-01],\n",
      "          [ 2.4738e-02,  6.5996e-02, -4.2736e-02],\n",
      "          [ 4.0404e-02,  2.7600e-02, -1.2907e-02]],\n",
      "\n",
      "         [[ 3.3378e-02,  2.8212e-02, -2.1699e-02],\n",
      "          [-5.4006e-02,  8.2356e-02,  3.8896e-02],\n",
      "          [-1.8055e-02, -8.1275e-02, -1.0676e-01]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[ 4.0008e-02,  1.7975e-02,  9.4202e-02],\n",
      "          [-9.7492e-02, -1.4051e-02, -4.2236e-02],\n",
      "          [ 1.4389e-02, -2.0754e-02, -2.3581e-02]],\n",
      "\n",
      "         [[ 4.6663e-03,  9.3178e-03,  6.6848e-02],\n",
      "          [ 3.3601e-02,  8.8346e-02, -5.0096e-02],\n",
      "          [ 1.4613e-02, -7.3510e-02, -8.6144e-02]],\n",
      "\n",
      "         [[-4.4409e-02,  1.0204e-01, -6.1084e-02],\n",
      "          [ 3.0879e-02,  8.4140e-02, -2.3826e-02],\n",
      "          [-2.3988e-02,  2.9018e-02,  1.0183e-01]]],\n",
      "\n",
      "\n",
      "        [[[ 3.8681e-02,  4.8797e-02,  1.9144e-02],\n",
      "          [ 6.1072e-02, -4.2021e-02, -7.3111e-02],\n",
      "          [ 7.1703e-02, -8.6962e-03,  9.3619e-02]],\n",
      "\n",
      "         [[-1.6312e-02,  3.7568e-02,  2.0392e-02],\n",
      "          [ 1.4367e-01, -1.3357e-02, -8.1782e-02],\n",
      "          [ 5.5127e-02,  2.0501e-02, -1.0363e-02]],\n",
      "\n",
      "         [[-2.4699e-02, -8.0246e-02, -2.4971e-02],\n",
      "          [-2.4959e-02, -6.6561e-02, -5.2956e-02],\n",
      "          [ 3.2936e-02, -2.0221e-02,  7.8024e-02]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[-6.3264e-02, -7.2656e-02, -1.2858e-01],\n",
      "          [ 1.2953e-01, -1.4529e-01, -4.1715e-02],\n",
      "          [-2.8907e-02,  3.1277e-02, -9.3153e-02]],\n",
      "\n",
      "         [[ 3.1688e-02, -7.6803e-02, -5.2817e-02],\n",
      "          [-1.3861e-02,  8.1881e-02,  8.9780e-03],\n",
      "          [ 7.9950e-02,  2.0682e-02, -8.5629e-03]],\n",
      "\n",
      "         [[ 2.7552e-02,  3.4069e-02,  1.0660e-02],\n",
      "          [-1.7012e-02,  2.0226e-02, -8.2904e-02],\n",
      "          [-3.3868e-02,  4.2948e-02, -2.5390e-02]]],\n",
      "\n",
      "\n",
      "        ...,\n",
      "\n",
      "\n",
      "        [[[-7.7384e-02,  4.1843e-04,  7.2949e-03],\n",
      "          [ 7.0635e-03,  3.7058e-02, -7.8655e-02],\n",
      "          [-4.4192e-02, -6.5971e-02,  1.0473e-01]],\n",
      "\n",
      "         [[-2.9104e-02, -3.9702e-02,  4.9176e-02],\n",
      "          [ 5.6683e-02,  3.9799e-02,  7.1621e-02],\n",
      "          [-1.1927e-02, -3.8574e-02,  9.0237e-02]],\n",
      "\n",
      "         [[ 8.1243e-02,  3.2838e-02, -6.4411e-02],\n",
      "          [ 1.5662e-02, -1.9643e-02, -1.5034e-02],\n",
      "          [-2.0877e-02, -4.9389e-02, -1.0846e-02]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[ 4.4938e-02,  7.2788e-02,  5.9610e-02],\n",
      "          [ 6.4901e-02,  6.6192e-04, -1.0470e-01],\n",
      "          [ 3.5129e-02,  6.5934e-03, -8.4909e-03]],\n",
      "\n",
      "         [[ 8.8505e-02, -2.5206e-03, -2.8599e-02],\n",
      "          [-5.3464e-02,  5.5908e-02,  2.0662e-02],\n",
      "          [ 7.7018e-03, -9.2283e-02, -3.9502e-02]],\n",
      "\n",
      "         [[-1.1157e-02,  3.8024e-02,  1.6694e-01],\n",
      "          [-3.7686e-02,  4.5556e-02, -5.3145e-02],\n",
      "          [ 8.7216e-02,  2.9434e-02,  2.4400e-02]]],\n",
      "\n",
      "\n",
      "        [[[ 4.8416e-02,  1.9539e-02, -2.8279e-02],\n",
      "          [ 1.1927e-03, -2.3604e-02, -7.6474e-02],\n",
      "          [-1.5735e-02, -1.7926e-02, -2.2728e-03]],\n",
      "\n",
      "         [[-3.4117e-02,  5.8497e-03, -2.3069e-02],\n",
      "          [-2.8063e-02, -9.1988e-02,  1.8739e-02],\n",
      "          [-1.0850e-02,  9.1107e-02,  1.2208e-01]],\n",
      "\n",
      "         [[-4.0958e-02,  1.1642e-01, -7.7995e-03],\n",
      "          [-2.5184e-02, -9.8270e-03, -1.4638e-02],\n",
      "          [-6.3137e-02,  5.4550e-02, -3.6849e-02]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[-1.1690e-02, -5.0369e-03,  1.6600e-02],\n",
      "          [-3.3884e-02,  8.0050e-02,  7.5592e-03],\n",
      "          [-5.0673e-02, -1.7057e-02, -7.9866e-02]],\n",
      "\n",
      "         [[ 4.7483e-02, -4.0707e-02, -8.3661e-02],\n",
      "          [-3.5521e-02,  1.0964e-02,  1.0167e-01],\n",
      "          [ 9.9412e-03,  6.6924e-02,  5.8669e-02]],\n",
      "\n",
      "         [[ 1.2975e-01,  8.9784e-03,  3.1838e-02],\n",
      "          [-4.7427e-02,  1.8761e-02, -2.2756e-02],\n",
      "          [ 3.5195e-03,  1.8830e-04, -4.8425e-02]]],\n",
      "\n",
      "\n",
      "        [[[-5.2785e-02,  6.3493e-03,  3.2436e-02],\n",
      "          [ 3.2510e-02, -3.1820e-02, -3.4129e-04],\n",
      "          [-2.0416e-02,  1.1173e-01,  6.5569e-02]],\n",
      "\n",
      "         [[ 6.2160e-03,  3.3443e-02,  9.7128e-02],\n",
      "          [-4.8528e-02, -6.3659e-02, -6.2906e-03],\n",
      "          [-1.0469e-01,  4.7164e-02,  5.8472e-03]],\n",
      "\n",
      "         [[ 6.2278e-02, -6.0180e-02, -1.9048e-02],\n",
      "          [ 5.8553e-03,  7.1288e-02,  5.0775e-02],\n",
      "          [ 1.9762e-02, -7.3844e-02,  5.5006e-04]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[-6.3804e-02,  5.3513e-02, -5.6058e-02],\n",
      "          [-2.8146e-02,  1.5581e-01,  9.6026e-02],\n",
      "          [-9.5402e-03, -4.5547e-02, -4.0405e-02]],\n",
      "\n",
      "         [[ 9.0815e-02, -2.2517e-02,  5.6349e-02],\n",
      "          [-5.2628e-02,  2.7353e-03,  4.2939e-02],\n",
      "          [ 3.4351e-02,  1.2934e-01,  1.1968e-01]],\n",
      "\n",
      "         [[ 3.6872e-02, -3.1800e-02,  4.7972e-02],\n",
      "          [ 4.0792e-02, -8.0777e-02, -6.1106e-02],\n",
      "          [ 4.7484e-02, -7.3866e-02, -1.0916e-01]]]], requires_grad=True)\n",
      "Parameter containing:\n",
      "tensor([1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1.], requires_grad=True)\n",
      "Parameter containing:\n",
      "tensor([0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.],\n",
      "       requires_grad=True)\n",
      "Parameter containing:\n",
      "tensor([[[[ 0.0866]],\n",
      "\n",
      "         [[ 0.0844]],\n",
      "\n",
      "         [[ 0.1941]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[-0.0979]],\n",
      "\n",
      "         [[-0.1513]],\n",
      "\n",
      "         [[ 0.1286]]],\n",
      "\n",
      "\n",
      "        [[[-0.0364]],\n",
      "\n",
      "         [[ 0.0285]],\n",
      "\n",
      "         [[ 0.1006]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[ 0.0282]],\n",
      "\n",
      "         [[-0.0439]],\n",
      "\n",
      "         [[-0.1096]]],\n",
      "\n",
      "\n",
      "        [[[ 0.1218]],\n",
      "\n",
      "         [[-0.0689]],\n",
      "\n",
      "         [[-0.0015]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[ 0.1322]],\n",
      "\n",
      "         [[-0.0785]],\n",
      "\n",
      "         [[ 0.1473]]],\n",
      "\n",
      "\n",
      "        ...,\n",
      "\n",
      "\n",
      "        [[[ 0.0383]],\n",
      "\n",
      "         [[-0.0528]],\n",
      "\n",
      "         [[ 0.1068]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[ 0.0970]],\n",
      "\n",
      "         [[-0.0492]],\n",
      "\n",
      "         [[-0.0727]]],\n",
      "\n",
      "\n",
      "        [[[ 0.0244]],\n",
      "\n",
      "         [[ 0.1021]],\n",
      "\n",
      "         [[-0.0171]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[-0.0151]],\n",
      "\n",
      "         [[ 0.0150]],\n",
      "\n",
      "         [[-0.0995]]],\n",
      "\n",
      "\n",
      "        [[[ 0.0823]],\n",
      "\n",
      "         [[ 0.1111]],\n",
      "\n",
      "         [[ 0.0809]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[-0.0502]],\n",
      "\n",
      "         [[ 0.0538]],\n",
      "\n",
      "         [[ 0.0627]]]], requires_grad=True)\n",
      "Parameter containing:\n",
      "tensor([1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1.], requires_grad=True)\n",
      "Parameter containing:\n",
      "tensor([0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.],\n",
      "       requires_grad=True)\n",
      "Parameter containing:\n",
      "tensor([[[[ 0.0042]],\n",
      "\n",
      "         [[ 0.0045]],\n",
      "\n",
      "         [[-0.3186]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[-0.1706]],\n",
      "\n",
      "         [[ 0.0019]],\n",
      "\n",
      "         [[-0.0676]]],\n",
      "\n",
      "\n",
      "        [[[-0.0689]],\n",
      "\n",
      "         [[-0.0934]],\n",
      "\n",
      "         [[ 0.2230]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[ 0.1652]],\n",
      "\n",
      "         [[ 0.1408]],\n",
      "\n",
      "         [[ 0.1800]]],\n",
      "\n",
      "\n",
      "        [[[-0.1140]],\n",
      "\n",
      "         [[ 0.0556]],\n",
      "\n",
      "         [[-0.2333]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[-0.1057]],\n",
      "\n",
      "         [[ 0.0774]],\n",
      "\n",
      "         [[-0.1957]]],\n",
      "\n",
      "\n",
      "        ...,\n",
      "\n",
      "\n",
      "        [[[ 0.0515]],\n",
      "\n",
      "         [[-0.0551]],\n",
      "\n",
      "         [[-0.0633]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[ 0.1714]],\n",
      "\n",
      "         [[-0.0661]],\n",
      "\n",
      "         [[-0.0776]]],\n",
      "\n",
      "\n",
      "        [[[-0.0665]],\n",
      "\n",
      "         [[ 0.0381]],\n",
      "\n",
      "         [[-0.2280]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[ 0.0492]],\n",
      "\n",
      "         [[ 0.1127]],\n",
      "\n",
      "         [[-0.0440]]],\n",
      "\n",
      "\n",
      "        [[[-0.1887]],\n",
      "\n",
      "         [[-0.1200]],\n",
      "\n",
      "         [[ 0.0712]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[ 0.2389]],\n",
      "\n",
      "         [[-0.1100]],\n",
      "\n",
      "         [[ 0.2772]]]], requires_grad=True)\n",
      "Parameter containing:\n",
      "tensor([1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1.], requires_grad=True)\n",
      "Parameter containing:\n",
      "tensor([0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.],\n",
      "       requires_grad=True)\n",
      "Parameter containing:\n",
      "tensor([[[[-0.0446,  0.0569, -0.0139],\n",
      "          [ 0.0049, -0.0482, -0.0346],\n",
      "          [ 0.0663, -0.0146, -0.0176]],\n",
      "\n",
      "         [[-0.0716, -0.0125, -0.0901],\n",
      "          [ 0.0273, -0.1227, -0.0436],\n",
      "          [-0.0374, -0.1046, -0.0673]],\n",
      "\n",
      "         [[-0.0137, -0.0641,  0.0925],\n",
      "          [ 0.0761, -0.0237, -0.0594],\n",
      "          [-0.0220,  0.0502,  0.0898]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[ 0.0353,  0.0034,  0.0103],\n",
      "          [-0.0133,  0.0651,  0.0202],\n",
      "          [-0.0643,  0.0021, -0.0249]],\n",
      "\n",
      "         [[-0.0017,  0.0612, -0.0318],\n",
      "          [ 0.0785, -0.0204, -0.0042],\n",
      "          [ 0.0759,  0.0350,  0.0444]],\n",
      "\n",
      "         [[ 0.0407, -0.0200, -0.0004],\n",
      "          [ 0.0527,  0.1015,  0.1283],\n",
      "          [ 0.0184,  0.0359,  0.1414]]],\n",
      "\n",
      "\n",
      "        [[[-0.0217, -0.0126,  0.0009],\n",
      "          [ 0.0026, -0.0572,  0.0318],\n",
      "          [-0.0952,  0.0533,  0.0131]],\n",
      "\n",
      "         [[-0.0108, -0.0365, -0.0094],\n",
      "          [ 0.0850,  0.1126,  0.0422],\n",
      "          [-0.0056, -0.0199,  0.0806]],\n",
      "\n",
      "         [[-0.0194,  0.0918, -0.0265],\n",
      "          [ 0.0028,  0.0393, -0.0203],\n",
      "          [ 0.0707, -0.0226,  0.0196]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[-0.0598, -0.1078,  0.1062],\n",
      "          [ 0.0154,  0.0343, -0.0161],\n",
      "          [-0.0584,  0.0429, -0.1093]],\n",
      "\n",
      "         [[ 0.0216, -0.0062,  0.1194],\n",
      "          [ 0.0145, -0.1338, -0.0356],\n",
      "          [ 0.0406, -0.0178,  0.0090]],\n",
      "\n",
      "         [[-0.0189, -0.0414,  0.0782],\n",
      "          [ 0.0521,  0.0117,  0.0484],\n",
      "          [-0.0038,  0.0400, -0.0430]]],\n",
      "\n",
      "\n",
      "        [[[ 0.0436, -0.0465,  0.0130],\n",
      "          [ 0.0589, -0.0226, -0.0178],\n",
      "          [ 0.0508, -0.0240,  0.0368]],\n",
      "\n",
      "         [[-0.1041, -0.0094,  0.0390],\n",
      "          [-0.0355,  0.0502,  0.0451],\n",
      "          [ 0.0247,  0.1731, -0.0432]],\n",
      "\n",
      "         [[-0.0363,  0.0394, -0.0306],\n",
      "          [ 0.0277,  0.0344,  0.0356],\n",
      "          [-0.1107, -0.0195, -0.0766]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[-0.0241, -0.0073,  0.0677],\n",
      "          [-0.1121,  0.0686,  0.0655],\n",
      "          [-0.0026,  0.0253, -0.0196]],\n",
      "\n",
      "         [[-0.0251,  0.0218,  0.0152],\n",
      "          [-0.0172,  0.0739, -0.0149],\n",
      "          [-0.0795,  0.0511, -0.0193]],\n",
      "\n",
      "         [[ 0.0644, -0.0091, -0.0775],\n",
      "          [ 0.0733, -0.0453,  0.0315],\n",
      "          [-0.0268,  0.0346,  0.0535]]],\n",
      "\n",
      "\n",
      "        ...,\n",
      "\n",
      "\n",
      "        [[[-0.0759,  0.1257, -0.0223],\n",
      "          [ 0.1643, -0.0096,  0.0398],\n",
      "          [ 0.0024, -0.0137, -0.0076]],\n",
      "\n",
      "         [[-0.0450,  0.0023,  0.0593],\n",
      "          [-0.0576,  0.0164,  0.0467],\n",
      "          [-0.0861, -0.0733, -0.0169]],\n",
      "\n",
      "         [[ 0.0033, -0.0092,  0.0091],\n",
      "          [ 0.0747,  0.0984, -0.0256],\n",
      "          [ 0.0572,  0.0594,  0.0879]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[-0.0511, -0.0256, -0.0262],\n",
      "          [-0.0453, -0.0690,  0.0077],\n",
      "          [ 0.0881, -0.0555,  0.0274]],\n",
      "\n",
      "         [[-0.0733,  0.0479, -0.0210],\n",
      "          [-0.0441, -0.0620,  0.0773],\n",
      "          [ 0.0016, -0.0554,  0.0590]],\n",
      "\n",
      "         [[-0.0190,  0.0711, -0.0112],\n",
      "          [-0.0486,  0.0549,  0.0368],\n",
      "          [ 0.0243,  0.0449,  0.0126]]],\n",
      "\n",
      "\n",
      "        [[[-0.0643, -0.0311, -0.0383],\n",
      "          [ 0.0029,  0.0055, -0.1254],\n",
      "          [-0.0816, -0.0385,  0.0176]],\n",
      "\n",
      "         [[-0.1068, -0.0588,  0.0024],\n",
      "          [ 0.0482, -0.0230,  0.0559],\n",
      "          [ 0.0112,  0.0019, -0.0380]],\n",
      "\n",
      "         [[-0.0656,  0.0838,  0.0311],\n",
      "          [-0.1105, -0.0248,  0.0865],\n",
      "          [-0.0152, -0.0322,  0.0291]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[-0.0278,  0.0316,  0.0174],\n",
      "          [ 0.0249, -0.0713,  0.0473],\n",
      "          [ 0.0276, -0.0412, -0.0234]],\n",
      "\n",
      "         [[ 0.0007,  0.0768,  0.0609],\n",
      "          [ 0.0931,  0.0943, -0.0189],\n",
      "          [ 0.0072,  0.0354, -0.0748]],\n",
      "\n",
      "         [[-0.0373, -0.0184, -0.0867],\n",
      "          [ 0.0387, -0.0662, -0.0300],\n",
      "          [-0.0582, -0.1010, -0.0428]]],\n",
      "\n",
      "\n",
      "        [[[-0.0154, -0.0292, -0.0238],\n",
      "          [-0.0030, -0.0305, -0.0122],\n",
      "          [-0.0614,  0.0709, -0.0243]],\n",
      "\n",
      "         [[ 0.0194,  0.0208,  0.0378],\n",
      "          [-0.1388, -0.0446,  0.0048],\n",
      "          [-0.0445,  0.0511,  0.0116]],\n",
      "\n",
      "         [[-0.0228,  0.0029,  0.0074],\n",
      "          [-0.0378,  0.0065,  0.0356],\n",
      "          [-0.0138, -0.0605,  0.0320]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[-0.0481, -0.0324, -0.0091],\n",
      "          [-0.0448,  0.1323, -0.0723],\n",
      "          [ 0.0337, -0.1272,  0.0165]],\n",
      "\n",
      "         [[ 0.1009, -0.0343, -0.0186],\n",
      "          [-0.0091, -0.0031, -0.0428],\n",
      "          [-0.0636,  0.0586,  0.0049]],\n",
      "\n",
      "         [[-0.0754, -0.0008, -0.0768],\n",
      "          [-0.0382, -0.0458,  0.0319],\n",
      "          [-0.0602,  0.0259, -0.0656]]]], requires_grad=True)\n",
      "Parameter containing:\n",
      "tensor([1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1.], requires_grad=True)\n",
      "Parameter containing:\n",
      "tensor([0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.],\n",
      "       requires_grad=True)\n",
      "Parameter containing:\n",
      "tensor([[[[-0.0308]],\n",
      "\n",
      "         [[ 0.0378]],\n",
      "\n",
      "         [[ 0.0056]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[ 0.0301]],\n",
      "\n",
      "         [[-0.0614]],\n",
      "\n",
      "         [[ 0.1125]]],\n",
      "\n",
      "\n",
      "        [[[ 0.0162]],\n",
      "\n",
      "         [[-0.0247]],\n",
      "\n",
      "         [[-0.0559]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[ 0.0130]],\n",
      "\n",
      "         [[ 0.1204]],\n",
      "\n",
      "         [[ 0.0852]]],\n",
      "\n",
      "\n",
      "        [[[-0.0680]],\n",
      "\n",
      "         [[-0.1057]],\n",
      "\n",
      "         [[ 0.1188]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[ 0.0378]],\n",
      "\n",
      "         [[ 0.1222]],\n",
      "\n",
      "         [[-0.1134]]],\n",
      "\n",
      "\n",
      "        ...,\n",
      "\n",
      "\n",
      "        [[[-0.0731]],\n",
      "\n",
      "         [[-0.0839]],\n",
      "\n",
      "         [[-0.0363]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[ 0.0290]],\n",
      "\n",
      "         [[ 0.0901]],\n",
      "\n",
      "         [[ 0.1396]]],\n",
      "\n",
      "\n",
      "        [[[-0.0304]],\n",
      "\n",
      "         [[-0.1109]],\n",
      "\n",
      "         [[ 0.1040]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[ 0.0885]],\n",
      "\n",
      "         [[ 0.0270]],\n",
      "\n",
      "         [[-0.1523]]],\n",
      "\n",
      "\n",
      "        [[[-0.0177]],\n",
      "\n",
      "         [[ 0.0152]],\n",
      "\n",
      "         [[ 0.0072]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[-0.0421]],\n",
      "\n",
      "         [[ 0.0148]],\n",
      "\n",
      "         [[-0.0430]]]], requires_grad=True)\n",
      "Parameter containing:\n",
      "tensor([1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1.], requires_grad=True)\n",
      "Parameter containing:\n",
      "tensor([0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.],\n",
      "       requires_grad=True)\n",
      "Parameter containing:\n",
      "tensor([[[[ 0.2673]],\n",
      "\n",
      "         [[-0.0649]],\n",
      "\n",
      "         [[-0.0852]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[ 0.0686]],\n",
      "\n",
      "         [[ 0.0806]],\n",
      "\n",
      "         [[ 0.0415]]],\n",
      "\n",
      "\n",
      "        [[[-0.0522]],\n",
      "\n",
      "         [[-0.0933]],\n",
      "\n",
      "         [[-0.0591]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[-0.2029]],\n",
      "\n",
      "         [[ 0.0603]],\n",
      "\n",
      "         [[-0.0747]]],\n",
      "\n",
      "\n",
      "        [[[ 0.0058]],\n",
      "\n",
      "         [[-0.2049]],\n",
      "\n",
      "         [[ 0.1978]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[-0.2535]],\n",
      "\n",
      "         [[ 0.0266]],\n",
      "\n",
      "         [[-0.1549]]],\n",
      "\n",
      "\n",
      "        ...,\n",
      "\n",
      "\n",
      "        [[[ 0.0509]],\n",
      "\n",
      "         [[-0.0084]],\n",
      "\n",
      "         [[-0.0110]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[ 0.1429]],\n",
      "\n",
      "         [[ 0.1060]],\n",
      "\n",
      "         [[ 0.0645]]],\n",
      "\n",
      "\n",
      "        [[[ 0.1584]],\n",
      "\n",
      "         [[-0.0487]],\n",
      "\n",
      "         [[-0.0675]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[-0.0506]],\n",
      "\n",
      "         [[ 0.0733]],\n",
      "\n",
      "         [[-0.1615]]],\n",
      "\n",
      "\n",
      "        [[[-0.0541]],\n",
      "\n",
      "         [[ 0.0737]],\n",
      "\n",
      "         [[ 0.0779]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[ 0.1265]],\n",
      "\n",
      "         [[ 0.0660]],\n",
      "\n",
      "         [[ 0.2057]]]], requires_grad=True)\n",
      "Parameter containing:\n",
      "tensor([1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1.], requires_grad=True)\n",
      "Parameter containing:\n",
      "tensor([0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0.], requires_grad=True)\n",
      "Parameter containing:\n",
      "tensor([[[[ 0.0105,  0.0372, -0.0116],\n",
      "          [-0.0401,  0.0441, -0.0137],\n",
      "          [-0.0325, -0.0738, -0.0164]],\n",
      "\n",
      "         [[-0.0229,  0.0046, -0.0501],\n",
      "          [-0.0329, -0.0945,  0.0353],\n",
      "          [ 0.0285, -0.0033,  0.0567]],\n",
      "\n",
      "         [[-0.0004,  0.0223, -0.0142],\n",
      "          [-0.0225, -0.0018,  0.0299],\n",
      "          [-0.0098,  0.0754, -0.0255]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[-0.0369,  0.0044, -0.0429],\n",
      "          [-0.0076, -0.0501,  0.0120],\n",
      "          [ 0.0562,  0.0446,  0.0304]],\n",
      "\n",
      "         [[-0.0302, -0.0191,  0.0555],\n",
      "          [ 0.0162,  0.0860, -0.0147],\n",
      "          [ 0.0043,  0.0340, -0.0241]],\n",
      "\n",
      "         [[ 0.0579,  0.0801, -0.0414],\n",
      "          [-0.0020,  0.0431,  0.0941],\n",
      "          [-0.0527, -0.0281,  0.0401]]],\n",
      "\n",
      "\n",
      "        [[[ 0.0666,  0.0535, -0.0420],\n",
      "          [-0.0304,  0.0505, -0.0863],\n",
      "          [ 0.0306, -0.0371,  0.0123]],\n",
      "\n",
      "         [[-0.0004, -0.0493,  0.0678],\n",
      "          [ 0.0053, -0.0297, -0.0429],\n",
      "          [-0.0471, -0.0169, -0.0641]],\n",
      "\n",
      "         [[ 0.0216,  0.0048,  0.0868],\n",
      "          [-0.0167, -0.0427,  0.0108],\n",
      "          [ 0.0490,  0.0348, -0.0456]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[-0.0264,  0.0581, -0.0745],\n",
      "          [ 0.0209,  0.0166, -0.0297],\n",
      "          [-0.0320,  0.0246, -0.0191]],\n",
      "\n",
      "         [[ 0.1022, -0.0340,  0.0075],\n",
      "          [-0.0195, -0.0534, -0.0280],\n",
      "          [-0.0098,  0.0112, -0.0159]],\n",
      "\n",
      "         [[-0.0676, -0.0698,  0.0161],\n",
      "          [ 0.0513, -0.0031,  0.0042],\n",
      "          [-0.0100,  0.0437,  0.0311]]],\n",
      "\n",
      "\n",
      "        [[[-0.0072, -0.0246, -0.0090],\n",
      "          [ 0.0192, -0.0035, -0.0300],\n",
      "          [ 0.0048, -0.0023, -0.0399]],\n",
      "\n",
      "         [[ 0.0716,  0.0252,  0.0386],\n",
      "          [-0.0319, -0.0062, -0.0585],\n",
      "          [ 0.0471, -0.0540,  0.0443]],\n",
      "\n",
      "         [[ 0.0345,  0.0118,  0.1007],\n",
      "          [ 0.0963, -0.0290,  0.0385],\n",
      "          [-0.0437, -0.0403, -0.0294]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[ 0.0230,  0.0122,  0.0696],\n",
      "          [ 0.0622,  0.0990,  0.0414],\n",
      "          [ 0.0077, -0.0222,  0.0491]],\n",
      "\n",
      "         [[ 0.0252, -0.0174, -0.0136],\n",
      "          [-0.0282,  0.0120,  0.0291],\n",
      "          [ 0.0127,  0.0357,  0.0774]],\n",
      "\n",
      "         [[-0.0875, -0.1262,  0.0415],\n",
      "          [-0.0093,  0.0350,  0.0139],\n",
      "          [ 0.0307, -0.0083, -0.0467]]],\n",
      "\n",
      "\n",
      "        ...,\n",
      "\n",
      "\n",
      "        [[[ 0.0433,  0.0234,  0.0310],\n",
      "          [-0.0141, -0.0932, -0.0270],\n",
      "          [ 0.0272, -0.0175, -0.0116]],\n",
      "\n",
      "         [[-0.0489, -0.0541,  0.0098],\n",
      "          [-0.0524, -0.0746, -0.0218],\n",
      "          [ 0.0395, -0.0125, -0.0010]],\n",
      "\n",
      "         [[-0.0101,  0.0190, -0.0274],\n",
      "          [-0.0022, -0.0006, -0.0385],\n",
      "          [ 0.0012, -0.0139,  0.0026]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[ 0.0006, -0.0148,  0.0256],\n",
      "          [ 0.0513,  0.0769, -0.0530],\n",
      "          [ 0.0539,  0.0097, -0.0699]],\n",
      "\n",
      "         [[-0.0374,  0.0282, -0.0915],\n",
      "          [-0.0013,  0.0572,  0.0179],\n",
      "          [-0.0960, -0.0316,  0.0284]],\n",
      "\n",
      "         [[ 0.0356,  0.0509, -0.0064],\n",
      "          [ 0.0067,  0.0636, -0.0060],\n",
      "          [-0.0558, -0.0363, -0.0324]]],\n",
      "\n",
      "\n",
      "        [[[ 0.0524,  0.0192,  0.0284],\n",
      "          [-0.0415,  0.0091, -0.0306],\n",
      "          [ 0.0141,  0.0358,  0.0177]],\n",
      "\n",
      "         [[ 0.0757, -0.0410,  0.0340],\n",
      "          [-0.0468,  0.0680,  0.0465],\n",
      "          [ 0.0082, -0.0001,  0.0057]],\n",
      "\n",
      "         [[ 0.1047, -0.0072, -0.0121],\n",
      "          [-0.0230, -0.0166,  0.0160],\n",
      "          [-0.0603, -0.0516, -0.0418]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[-0.0602, -0.0136, -0.0343],\n",
      "          [-0.0881,  0.0204,  0.0130],\n",
      "          [ 0.0480, -0.0185, -0.0232]],\n",
      "\n",
      "         [[ 0.0357,  0.0362,  0.0135],\n",
      "          [ 0.1189,  0.0214, -0.0535],\n",
      "          [ 0.0719, -0.0131, -0.0207]],\n",
      "\n",
      "         [[ 0.0379,  0.0344, -0.0463],\n",
      "          [-0.0175,  0.0454,  0.0611],\n",
      "          [-0.0146,  0.0634,  0.0157]]],\n",
      "\n",
      "\n",
      "        [[[ 0.0558, -0.0327, -0.0026],\n",
      "          [-0.0146,  0.0060,  0.0797],\n",
      "          [-0.0061, -0.0388,  0.0645]],\n",
      "\n",
      "         [[ 0.0166,  0.0357, -0.0720],\n",
      "          [ 0.0191,  0.0516, -0.0212],\n",
      "          [-0.0005, -0.0792, -0.0530]],\n",
      "\n",
      "         [[ 0.0204,  0.0247,  0.0837],\n",
      "          [-0.0375,  0.0020, -0.0007],\n",
      "          [-0.0472, -0.0130, -0.0213]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[ 0.0023, -0.0238, -0.0225],\n",
      "          [ 0.0906,  0.0537, -0.0443],\n",
      "          [ 0.0024,  0.0214,  0.0118]],\n",
      "\n",
      "         [[ 0.0251,  0.0060,  0.0286],\n",
      "          [-0.0072,  0.0381, -0.0043],\n",
      "          [ 0.0055, -0.0242, -0.0063]],\n",
      "\n",
      "         [[ 0.0050,  0.0086, -0.0369],\n",
      "          [ 0.0642,  0.0218, -0.0079],\n",
      "          [ 0.0248,  0.0052,  0.0792]]]], requires_grad=True)\n",
      "Parameter containing:\n",
      "tensor([1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1.], requires_grad=True)\n",
      "Parameter containing:\n",
      "tensor([0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0.], requires_grad=True)\n",
      "Parameter containing:\n",
      "tensor([[[[ 0.0097]],\n",
      "\n",
      "         [[ 0.0389]],\n",
      "\n",
      "         [[-0.1123]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[-0.0444]],\n",
      "\n",
      "         [[-0.0728]],\n",
      "\n",
      "         [[-0.0139]]],\n",
      "\n",
      "\n",
      "        [[[-0.0304]],\n",
      "\n",
      "         [[-0.1065]],\n",
      "\n",
      "         [[-0.0130]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[ 0.0192]],\n",
      "\n",
      "         [[ 0.0218]],\n",
      "\n",
      "         [[-0.0759]]],\n",
      "\n",
      "\n",
      "        [[[ 0.0735]],\n",
      "\n",
      "         [[ 0.0240]],\n",
      "\n",
      "         [[ 0.0757]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[-0.0805]],\n",
      "\n",
      "         [[-0.0455]],\n",
      "\n",
      "         [[-0.0880]]],\n",
      "\n",
      "\n",
      "        ...,\n",
      "\n",
      "\n",
      "        [[[-0.0136]],\n",
      "\n",
      "         [[ 0.0988]],\n",
      "\n",
      "         [[-0.0184]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[-0.0006]],\n",
      "\n",
      "         [[ 0.0145]],\n",
      "\n",
      "         [[-0.1270]]],\n",
      "\n",
      "\n",
      "        [[[ 0.0182]],\n",
      "\n",
      "         [[-0.0143]],\n",
      "\n",
      "         [[ 0.0430]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[-0.0667]],\n",
      "\n",
      "         [[-0.0549]],\n",
      "\n",
      "         [[-0.0736]]],\n",
      "\n",
      "\n",
      "        [[[-0.0087]],\n",
      "\n",
      "         [[-0.0598]],\n",
      "\n",
      "         [[ 0.1390]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[ 0.1231]],\n",
      "\n",
      "         [[ 0.0051]],\n",
      "\n",
      "         [[ 0.0453]]]], requires_grad=True)\n",
      "Parameter containing:\n",
      "tensor([1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1.], requires_grad=True)\n",
      "Parameter containing:\n",
      "tensor([0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0.], requires_grad=True)\n",
      "Parameter containing:\n",
      "tensor([[[[-0.0504]],\n",
      "\n",
      "         [[-0.0875]],\n",
      "\n",
      "         [[ 0.0996]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[-0.1070]],\n",
      "\n",
      "         [[-0.0036]],\n",
      "\n",
      "         [[-0.0806]]],\n",
      "\n",
      "\n",
      "        [[[-0.0400]],\n",
      "\n",
      "         [[ 0.0318]],\n",
      "\n",
      "         [[-0.0768]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[ 0.0236]],\n",
      "\n",
      "         [[ 0.0194]],\n",
      "\n",
      "         [[ 0.0042]]],\n",
      "\n",
      "\n",
      "        [[[-0.0925]],\n",
      "\n",
      "         [[-0.0872]],\n",
      "\n",
      "         [[ 0.0800]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[-0.0239]],\n",
      "\n",
      "         [[ 0.0402]],\n",
      "\n",
      "         [[-0.0318]]],\n",
      "\n",
      "\n",
      "        ...,\n",
      "\n",
      "\n",
      "        [[[-0.0007]],\n",
      "\n",
      "         [[-0.0338]],\n",
      "\n",
      "         [[-0.0366]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[-0.0194]],\n",
      "\n",
      "         [[-0.0304]],\n",
      "\n",
      "         [[-0.0420]]],\n",
      "\n",
      "\n",
      "        [[[-0.0658]],\n",
      "\n",
      "         [[-0.0307]],\n",
      "\n",
      "         [[ 0.0717]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[-0.0703]],\n",
      "\n",
      "         [[ 0.0154]],\n",
      "\n",
      "         [[ 0.0344]]],\n",
      "\n",
      "\n",
      "        [[[ 0.0261]],\n",
      "\n",
      "         [[-0.0407]],\n",
      "\n",
      "         [[-0.0914]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[ 0.0145]],\n",
      "\n",
      "         [[-0.0836]],\n",
      "\n",
      "         [[-0.0471]]]], requires_grad=True)\n",
      "Parameter containing:\n",
      "tensor([1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1.], requires_grad=True)\n",
      "Parameter containing:\n",
      "tensor([0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0.], requires_grad=True)\n",
      "Parameter containing:\n",
      "tensor([[[[ 0.0832]],\n",
      "\n",
      "         [[ 0.1645]],\n",
      "\n",
      "         [[ 0.0077]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[ 0.0670]],\n",
      "\n",
      "         [[ 0.0444]],\n",
      "\n",
      "         [[ 0.1141]]],\n",
      "\n",
      "\n",
      "        [[[ 0.0346]],\n",
      "\n",
      "         [[-0.0234]],\n",
      "\n",
      "         [[ 0.1851]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[ 0.2686]],\n",
      "\n",
      "         [[ 0.1430]],\n",
      "\n",
      "         [[ 0.0499]]],\n",
      "\n",
      "\n",
      "        [[[-0.0073]],\n",
      "\n",
      "         [[ 0.0402]],\n",
      "\n",
      "         [[ 0.3529]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[ 0.2508]],\n",
      "\n",
      "         [[-0.2596]],\n",
      "\n",
      "         [[ 0.2326]]],\n",
      "\n",
      "\n",
      "        ...,\n",
      "\n",
      "\n",
      "        [[[-0.1109]],\n",
      "\n",
      "         [[-0.0952]],\n",
      "\n",
      "         [[ 0.0239]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[ 0.3361]],\n",
      "\n",
      "         [[ 0.2558]],\n",
      "\n",
      "         [[ 0.2142]]],\n",
      "\n",
      "\n",
      "        [[[-0.0771]],\n",
      "\n",
      "         [[ 0.1711]],\n",
      "\n",
      "         [[ 0.1857]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[-0.2719]],\n",
      "\n",
      "         [[-0.1373]],\n",
      "\n",
      "         [[ 0.0381]]],\n",
      "\n",
      "\n",
      "        [[[ 0.0399]],\n",
      "\n",
      "         [[ 0.1049]],\n",
      "\n",
      "         [[ 0.2122]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[-0.0609]],\n",
      "\n",
      "         [[ 0.2443]],\n",
      "\n",
      "         [[-0.2131]]]], requires_grad=True)\n",
      "Parameter containing:\n",
      "tensor([1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1.], requires_grad=True)\n",
      "Parameter containing:\n",
      "tensor([0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0.], requires_grad=True)\n",
      "Parameter containing:\n",
      "tensor([[[[-3.1322e-02,  7.5456e-03,  1.2008e-02],\n",
      "          [ 6.8335e-03, -3.2110e-02, -1.6310e-02],\n",
      "          [ 2.0985e-02,  9.9531e-03, -2.3577e-02]],\n",
      "\n",
      "         [[ 1.6897e-02, -3.2265e-02, -1.9844e-02],\n",
      "          [-6.0005e-03,  5.2465e-03, -8.9149e-02],\n",
      "          [ 4.2633e-02, -1.0908e-02,  1.9390e-02]],\n",
      "\n",
      "         [[ 1.0122e-02, -5.1070e-02,  7.0938e-02],\n",
      "          [-9.3463e-03,  4.9196e-03,  3.2809e-02],\n",
      "          [-4.3885e-02,  9.5049e-02,  3.2235e-02]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[ 1.1704e-02, -4.4370e-02,  4.5663e-02],\n",
      "          [-2.2732e-02,  5.4156e-02, -1.6402e-03],\n",
      "          [ 1.4117e-01,  5.0246e-02, -4.9885e-03]],\n",
      "\n",
      "         [[ 1.3380e-02, -6.4376e-02, -4.9463e-02],\n",
      "          [ 2.1459e-02,  7.7134e-04, -3.9908e-03],\n",
      "          [ 5.0712e-02, -5.3063e-02, -3.3210e-02]],\n",
      "\n",
      "         [[-1.7745e-03, -6.1415e-02,  2.1104e-02],\n",
      "          [-6.6098e-02, -3.6296e-02, -2.9688e-02],\n",
      "          [-5.2142e-03,  3.8909e-02, -2.9687e-02]]],\n",
      "\n",
      "\n",
      "        [[[-4.5623e-02,  3.7365e-02, -3.6468e-02],\n",
      "          [ 3.7770e-02,  1.6597e-02, -2.9239e-02],\n",
      "          [-4.7104e-03,  6.1064e-02,  8.4781e-02]],\n",
      "\n",
      "         [[-3.3501e-02,  1.6740e-02, -3.3265e-03],\n",
      "          [ 1.1280e-02,  4.7021e-02,  3.3429e-02],\n",
      "          [-2.7157e-03, -6.3530e-02,  3.1113e-02]],\n",
      "\n",
      "         [[ 1.9494e-03, -4.8883e-03, -1.1162e-02],\n",
      "          [-6.8001e-02,  9.0601e-02,  1.9549e-02],\n",
      "          [ 5.0988e-04,  4.8897e-03, -9.0519e-03]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[ 7.1491e-02,  4.1569e-03, -3.7434e-02],\n",
      "          [-1.7346e-02,  4.3213e-02,  2.0487e-02],\n",
      "          [ 1.8854e-02, -2.5963e-02,  2.8329e-02]],\n",
      "\n",
      "         [[-6.2476e-02,  2.0647e-02, -5.8429e-02],\n",
      "          [-6.1096e-02, -3.6874e-02, -3.7335e-02],\n",
      "          [ 1.3405e-02,  1.4682e-02, -1.4809e-02]],\n",
      "\n",
      "         [[-5.7954e-03, -5.1455e-02,  5.8445e-02],\n",
      "          [ 4.5439e-02, -2.2036e-02,  5.0512e-02],\n",
      "          [-4.1749e-02, -2.6314e-02,  1.1953e-02]]],\n",
      "\n",
      "\n",
      "        [[[ 1.4809e-02, -5.9791e-02,  7.2033e-02],\n",
      "          [ 7.5819e-02, -2.0869e-02,  4.0679e-02],\n",
      "          [-1.1027e-02, -3.6778e-02,  1.9954e-03]],\n",
      "\n",
      "         [[ 3.2150e-02, -4.2845e-02, -5.3214e-02],\n",
      "          [-1.9517e-02,  5.5802e-03,  6.1517e-02],\n",
      "          [-2.1544e-02,  5.4857e-02,  5.4078e-02]],\n",
      "\n",
      "         [[ 2.0634e-03, -2.0480e-04,  1.1265e-02],\n",
      "          [ 2.4808e-02,  4.8632e-03, -1.1378e-04],\n",
      "          [ 6.5699e-02, -4.6201e-02, -1.5249e-02]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[-3.0703e-02,  1.0780e-02,  6.9150e-04],\n",
      "          [ 2.0071e-02, -7.5378e-02,  9.8802e-02],\n",
      "          [-1.6674e-02, -7.4994e-02,  1.4678e-01]],\n",
      "\n",
      "         [[ 9.6909e-03, -1.5048e-02,  9.0717e-03],\n",
      "          [ 2.3380e-02,  1.0734e-02, -6.9288e-02],\n",
      "          [-9.3027e-02,  3.4724e-02,  8.6023e-03]],\n",
      "\n",
      "         [[-1.7558e-02,  1.3119e-03, -1.3507e-02],\n",
      "          [ 8.3281e-03, -5.4776e-02, -2.4171e-02],\n",
      "          [ 4.1616e-02,  1.8918e-02, -3.7918e-02]]],\n",
      "\n",
      "\n",
      "        ...,\n",
      "\n",
      "\n",
      "        [[[-3.2755e-02, -5.7663e-02, -2.7584e-02],\n",
      "          [-2.7271e-02, -4.8749e-02, -3.2615e-03],\n",
      "          [ 2.1862e-03,  6.9651e-02,  4.3663e-02]],\n",
      "\n",
      "         [[-3.4075e-02, -1.4986e-02,  8.8633e-03],\n",
      "          [ 2.8942e-02,  3.2460e-03, -3.1328e-02],\n",
      "          [-4.0235e-02, -2.9262e-02,  1.8045e-02]],\n",
      "\n",
      "         [[ 5.2407e-02, -1.3696e-02,  7.7005e-02],\n",
      "          [-4.0164e-02,  9.7321e-02,  1.6488e-02],\n",
      "          [ 1.2236e-02, -1.3359e-02,  1.5114e-02]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[-3.2707e-02, -3.0324e-02, -1.4987e-02],\n",
      "          [ 4.1291e-02, -4.1622e-02,  1.1319e-02],\n",
      "          [-1.0496e-01, -9.0679e-03,  2.7914e-02]],\n",
      "\n",
      "         [[ 2.6100e-02,  4.8061e-03, -7.9213e-02],\n",
      "          [-3.4281e-02,  4.2271e-02, -2.0823e-02],\n",
      "          [ 8.2796e-02, -4.0459e-02,  4.0608e-02]],\n",
      "\n",
      "         [[ 1.0410e-02,  2.2172e-02,  7.1662e-02],\n",
      "          [ 5.7285e-02,  2.0639e-02, -1.2206e-02],\n",
      "          [-2.2300e-02, -1.0054e-01, -6.8577e-02]]],\n",
      "\n",
      "\n",
      "        [[[-1.6776e-02, -4.5917e-02, -6.7179e-02],\n",
      "          [ 1.5004e-02, -1.8192e-02,  3.3902e-02],\n",
      "          [ 2.4229e-02,  3.4149e-03,  2.8665e-02]],\n",
      "\n",
      "         [[ 4.4369e-02,  7.6341e-02,  3.9116e-02],\n",
      "          [ 4.3553e-02,  1.2497e-02, -1.2328e-01],\n",
      "          [ 8.0837e-03, -1.0208e-01,  3.1088e-02]],\n",
      "\n",
      "         [[-3.2805e-03, -8.6097e-03, -3.0355e-02],\n",
      "          [-2.2860e-02,  1.4164e-02, -4.7270e-02],\n",
      "          [ 2.7269e-02,  1.6365e-03, -8.0083e-03]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[ 1.4714e-02, -2.8122e-02,  7.4853e-02],\n",
      "          [ 8.8952e-03,  6.5473e-02,  2.6427e-02],\n",
      "          [-9.9812e-02, -6.3443e-03, -3.1630e-02]],\n",
      "\n",
      "         [[ 7.4005e-02,  1.0338e-02, -1.3882e-02],\n",
      "          [ 6.0364e-02,  1.6110e-02, -1.1548e-02],\n",
      "          [-4.5889e-02, -5.8536e-02,  1.8829e-02]],\n",
      "\n",
      "         [[-2.1125e-02,  8.1121e-02,  4.8907e-02],\n",
      "          [-4.4347e-02, -1.2831e-01, -1.9589e-02],\n",
      "          [ 1.0100e-01,  3.6779e-02,  5.1223e-02]]],\n",
      "\n",
      "\n",
      "        [[[-4.9521e-02,  1.6608e-02, -3.3348e-02],\n",
      "          [-6.9224e-02, -2.4022e-02,  2.3517e-02],\n",
      "          [-1.5628e-02,  1.8875e-02, -6.5525e-02]],\n",
      "\n",
      "         [[ 3.4363e-02, -3.5449e-03, -6.9979e-02],\n",
      "          [-3.1744e-03, -3.0869e-02, -2.1059e-02],\n",
      "          [ 5.1357e-02, -5.6064e-02,  3.5149e-03]],\n",
      "\n",
      "         [[-4.7649e-02, -5.2253e-03, -3.1354e-02],\n",
      "          [ 5.2746e-03,  1.3817e-02,  5.9400e-02],\n",
      "          [ 1.8122e-02, -5.1058e-02,  1.6830e-03]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[-3.2525e-02,  3.2382e-02, -2.1297e-04],\n",
      "          [ 4.9404e-02,  4.0983e-03, -6.7457e-02],\n",
      "          [ 2.0963e-02, -3.8154e-03,  2.2071e-03]],\n",
      "\n",
      "         [[ 1.7463e-02,  1.0243e-02,  3.6197e-02],\n",
      "          [-3.3671e-02, -1.8778e-02, -1.7692e-02],\n",
      "          [-1.8202e-02,  8.8475e-03,  6.5215e-03]],\n",
      "\n",
      "         [[ 2.3606e-02, -2.3462e-02,  2.5093e-02],\n",
      "          [-4.7121e-02, -5.8315e-02, -2.8864e-02],\n",
      "          [-3.6719e-02,  5.7658e-04, -1.2124e-02]]]], requires_grad=True)\n",
      "Parameter containing:\n",
      "tensor([1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1.], requires_grad=True)\n",
      "Parameter containing:\n",
      "tensor([0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0.], requires_grad=True)\n",
      "Parameter containing:\n",
      "tensor([[[[ 0.0465]],\n",
      "\n",
      "         [[-0.0145]],\n",
      "\n",
      "         [[-0.0395]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[-0.0366]],\n",
      "\n",
      "         [[-0.0440]],\n",
      "\n",
      "         [[-0.1014]]],\n",
      "\n",
      "\n",
      "        [[[-0.0182]],\n",
      "\n",
      "         [[-0.0217]],\n",
      "\n",
      "         [[ 0.0414]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[-0.0372]],\n",
      "\n",
      "         [[ 0.1290]],\n",
      "\n",
      "         [[-0.1768]]],\n",
      "\n",
      "\n",
      "        [[[-0.0125]],\n",
      "\n",
      "         [[ 0.1215]],\n",
      "\n",
      "         [[-0.0483]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[-0.1082]],\n",
      "\n",
      "         [[-0.0669]],\n",
      "\n",
      "         [[ 0.0515]]],\n",
      "\n",
      "\n",
      "        ...,\n",
      "\n",
      "\n",
      "        [[[ 0.0154]],\n",
      "\n",
      "         [[-0.0057]],\n",
      "\n",
      "         [[-0.0124]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[-0.0102]],\n",
      "\n",
      "         [[ 0.0743]],\n",
      "\n",
      "         [[ 0.0594]]],\n",
      "\n",
      "\n",
      "        [[[ 0.0675]],\n",
      "\n",
      "         [[ 0.0337]],\n",
      "\n",
      "         [[ 0.1193]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[ 0.0122]],\n",
      "\n",
      "         [[ 0.0716]],\n",
      "\n",
      "         [[ 0.0063]]],\n",
      "\n",
      "\n",
      "        [[[-0.0191]],\n",
      "\n",
      "         [[-0.0398]],\n",
      "\n",
      "         [[ 0.0625]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[ 0.0258]],\n",
      "\n",
      "         [[ 0.1239]],\n",
      "\n",
      "         [[-0.0016]]]], requires_grad=True)\n",
      "Parameter containing:\n",
      "tensor([1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1.], requires_grad=True)\n",
      "Parameter containing:\n",
      "tensor([0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0.], requires_grad=True)\n",
      "Parameter containing:\n",
      "tensor([[[[ 0.1343]],\n",
      "\n",
      "         [[ 0.0898]],\n",
      "\n",
      "         [[ 0.0435]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[ 0.2013]],\n",
      "\n",
      "         [[ 0.0789]],\n",
      "\n",
      "         [[-0.0042]]],\n",
      "\n",
      "\n",
      "        [[[ 0.0881]],\n",
      "\n",
      "         [[-0.2149]],\n",
      "\n",
      "         [[ 0.1163]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[-0.0151]],\n",
      "\n",
      "         [[-0.0581]],\n",
      "\n",
      "         [[-0.0546]]],\n",
      "\n",
      "\n",
      "        [[[ 0.1572]],\n",
      "\n",
      "         [[-0.1505]],\n",
      "\n",
      "         [[-0.0390]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[ 0.1179]],\n",
      "\n",
      "         [[-0.1111]],\n",
      "\n",
      "         [[ 0.1188]]],\n",
      "\n",
      "\n",
      "        ...,\n",
      "\n",
      "\n",
      "        [[[-0.2373]],\n",
      "\n",
      "         [[-0.0555]],\n",
      "\n",
      "         [[-0.3375]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[-0.1341]],\n",
      "\n",
      "         [[ 0.1158]],\n",
      "\n",
      "         [[ 0.1263]]],\n",
      "\n",
      "\n",
      "        [[[ 0.0172]],\n",
      "\n",
      "         [[ 0.0888]],\n",
      "\n",
      "         [[ 0.2742]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[ 0.0525]],\n",
      "\n",
      "         [[ 0.0301]],\n",
      "\n",
      "         [[ 0.0382]]],\n",
      "\n",
      "\n",
      "        [[[-0.0049]],\n",
      "\n",
      "         [[ 0.2507]],\n",
      "\n",
      "         [[ 0.0291]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[-0.1557]],\n",
      "\n",
      "         [[ 0.0006]],\n",
      "\n",
      "         [[ 0.0697]]]], requires_grad=True)\n",
      "Parameter containing:\n",
      "tensor([1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1.], requires_grad=True)\n",
      "Parameter containing:\n",
      "tensor([0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0.], requires_grad=True)\n",
      "Parameter containing:\n",
      "tensor([[[[ 1.4230e-02, -1.6262e-02, -9.8670e-03],\n",
      "          [-4.7071e-02, -4.2044e-02,  2.0133e-02],\n",
      "          [-3.8051e-02, -1.7790e-02,  3.4678e-02]],\n",
      "\n",
      "         [[-5.8986e-02, -5.8298e-02, -8.8446e-02],\n",
      "          [ 3.6696e-02,  1.5907e-02,  4.1427e-02],\n",
      "          [-1.7327e-02, -1.9547e-02, -3.7981e-02]],\n",
      "\n",
      "         [[ 9.4457e-03, -4.9682e-02, -3.9658e-02],\n",
      "          [-3.2335e-02, -4.0916e-02, -7.3290e-02],\n",
      "          [ 1.1831e-02, -6.7732e-02,  2.0678e-02]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[-6.6629e-02, -3.5581e-03, -3.3333e-02],\n",
      "          [ 5.7738e-03,  2.7353e-02,  3.2769e-02],\n",
      "          [ 3.3778e-02, -3.7047e-02, -4.0655e-03]],\n",
      "\n",
      "         [[-6.1542e-02, -1.0896e-01, -1.4856e-03],\n",
      "          [ 2.3158e-02,  3.9176e-02,  5.4783e-02],\n",
      "          [ 4.0668e-02,  1.8781e-03, -8.6878e-04]],\n",
      "\n",
      "         [[-2.0485e-02, -3.3598e-02, -7.6615e-02],\n",
      "          [-2.6591e-02, -1.9955e-02, -1.8824e-02],\n",
      "          [ 3.3488e-02, -3.2505e-02, -3.1384e-02]]],\n",
      "\n",
      "\n",
      "        [[[ 1.0728e-02, -3.4216e-02, -9.0374e-03],\n",
      "          [ 4.6442e-02,  8.7427e-02, -8.9690e-02],\n",
      "          [ 2.6218e-02, -1.1750e-02, -3.8962e-02]],\n",
      "\n",
      "         [[-1.1802e-02,  5.7067e-02, -3.0758e-02],\n",
      "          [-3.4854e-03,  4.9164e-02,  4.9795e-02],\n",
      "          [-7.7024e-02, -1.2996e-01,  3.3894e-02]],\n",
      "\n",
      "         [[-1.6436e-02, -1.8543e-03,  7.7549e-02],\n",
      "          [-2.0113e-02, -1.7565e-03,  2.7964e-02],\n",
      "          [ 7.0411e-02,  6.3166e-02, -1.6809e-02]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[-2.2178e-02,  1.2902e-02,  2.4596e-02],\n",
      "          [ 3.1261e-02,  2.1534e-02, -7.9096e-03],\n",
      "          [ 3.5151e-03,  1.4091e-02,  2.1040e-02]],\n",
      "\n",
      "         [[-2.3054e-02,  7.3445e-02, -6.7163e-02],\n",
      "          [ 3.5311e-02, -3.2933e-02, -2.5916e-02],\n",
      "          [-1.8168e-02,  1.3978e-02, -7.4959e-02]],\n",
      "\n",
      "         [[-6.8173e-02,  6.8950e-02, -7.1140e-02],\n",
      "          [-4.7628e-02,  3.4812e-02, -5.1197e-02],\n",
      "          [ 1.2995e-03,  3.4389e-02, -3.6788e-02]]],\n",
      "\n",
      "\n",
      "        [[[-3.6860e-02, -4.4157e-02, -7.9047e-03],\n",
      "          [ 4.6779e-02,  2.4077e-03,  1.2802e-02],\n",
      "          [-1.8763e-02, -2.7117e-02, -3.2018e-02]],\n",
      "\n",
      "         [[ 8.7294e-02, -2.3098e-02,  7.4089e-03],\n",
      "          [-5.7251e-03, -4.6426e-02,  3.9831e-02],\n",
      "          [-3.4091e-03,  2.2912e-02, -9.9324e-03]],\n",
      "\n",
      "         [[ 1.2535e-02,  2.3883e-02, -2.5485e-02],\n",
      "          [-5.3254e-03,  4.4118e-02,  7.2715e-04],\n",
      "          [-8.1162e-03,  5.3585e-02,  6.7971e-03]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[ 6.0000e-03, -7.2373e-03,  2.8093e-02],\n",
      "          [ 7.0155e-02,  5.4153e-02,  5.0598e-02],\n",
      "          [-3.4358e-02, -7.3474e-03,  8.7726e-02]],\n",
      "\n",
      "         [[-2.5859e-02,  2.0038e-02, -2.1992e-03],\n",
      "          [-5.7693e-03,  3.3738e-02,  7.8738e-02],\n",
      "          [-8.8143e-02, -3.4032e-02,  9.0436e-02]],\n",
      "\n",
      "         [[ 5.1744e-03, -3.8332e-02, -1.0935e-02],\n",
      "          [-4.8562e-03,  6.3561e-02, -4.5486e-02],\n",
      "          [-1.1811e-02,  7.3982e-02, -5.8767e-02]]],\n",
      "\n",
      "\n",
      "        ...,\n",
      "\n",
      "\n",
      "        [[[-1.1164e-02,  2.6027e-03,  2.9682e-02],\n",
      "          [ 4.1756e-02,  7.0055e-02,  2.7609e-04],\n",
      "          [-2.2328e-02,  8.8449e-04,  1.2146e-02]],\n",
      "\n",
      "         [[-1.3002e-02, -1.6046e-02,  1.9886e-02],\n",
      "          [-1.7821e-02, -2.7131e-03, -2.9207e-03],\n",
      "          [-9.0343e-02,  6.1709e-02,  5.5717e-02]],\n",
      "\n",
      "         [[-1.3823e-02,  3.0893e-02,  2.2618e-02],\n",
      "          [ 5.7307e-03, -1.6386e-02, -1.0285e-01],\n",
      "          [-1.4308e-02, -4.0911e-02,  1.5149e-02]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[-7.1708e-02,  8.8669e-02,  5.2367e-02],\n",
      "          [ 7.1382e-02,  3.2148e-02, -2.3576e-02],\n",
      "          [-4.2101e-02, -2.8694e-02,  3.0295e-02]],\n",
      "\n",
      "         [[-9.1265e-03,  5.5715e-02,  6.8566e-02],\n",
      "          [-4.3296e-02, -3.2892e-03,  1.2069e-02],\n",
      "          [ 1.5018e-02, -5.2703e-02, -3.5714e-02]],\n",
      "\n",
      "         [[-7.3518e-02, -9.1491e-04,  3.2475e-02],\n",
      "          [ 4.1988e-02,  6.5480e-02,  1.8939e-02],\n",
      "          [ 1.5632e-02,  5.5462e-02,  5.7567e-02]]],\n",
      "\n",
      "\n",
      "        [[[ 3.2616e-02,  1.7002e-02,  3.5989e-02],\n",
      "          [ 1.8749e-03, -2.8836e-02,  4.1014e-03],\n",
      "          [ 3.7029e-02, -6.6206e-03,  3.0281e-02]],\n",
      "\n",
      "         [[-3.1081e-02, -3.8454e-02, -4.0953e-02],\n",
      "          [ 3.0135e-02, -7.0341e-02,  1.6869e-02],\n",
      "          [ 8.1822e-02,  1.1819e-02, -9.1819e-02]],\n",
      "\n",
      "         [[ 5.0415e-02, -5.3984e-02, -6.4108e-02],\n",
      "          [-2.5277e-02, -2.8272e-02, -3.3609e-02],\n",
      "          [ 6.3568e-02,  1.0317e-01, -2.4500e-02]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[ 9.4985e-03,  4.1295e-02,  1.7545e-02],\n",
      "          [ 1.0691e-02, -6.5428e-03, -1.0836e-02],\n",
      "          [ 2.3848e-02, -2.0222e-02, -5.8549e-02]],\n",
      "\n",
      "         [[ 6.1963e-02,  2.1396e-02, -2.3755e-02],\n",
      "          [ 6.6101e-02,  9.6514e-02,  1.9071e-02],\n",
      "          [-6.9309e-03, -4.3069e-02, -2.7376e-02]],\n",
      "\n",
      "         [[-1.3630e-02, -3.9433e-02, -7.1782e-02],\n",
      "          [-9.8561e-05, -3.3200e-02, -1.7379e-02],\n",
      "          [-9.3339e-03, -1.4585e-02, -2.4099e-02]]],\n",
      "\n",
      "\n",
      "        [[[ 3.3953e-03,  2.4244e-02, -6.5307e-03],\n",
      "          [ 3.0785e-02,  3.9771e-02, -2.0924e-02],\n",
      "          [ 1.0242e-02,  1.1872e-02,  3.9285e-02]],\n",
      "\n",
      "         [[-3.6242e-02,  1.5963e-02, -2.4778e-02],\n",
      "          [ 7.1276e-03, -5.6474e-03,  1.4339e-02],\n",
      "          [-3.5630e-02,  5.1343e-02,  1.4447e-02]],\n",
      "\n",
      "         [[ 5.3722e-02, -8.5525e-03, -6.0526e-03],\n",
      "          [ 9.9780e-02,  1.7064e-02,  3.9691e-02],\n",
      "          [ 2.7636e-02, -2.1833e-02, -3.2825e-02]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[-6.8698e-02, -4.9055e-02, -1.1811e-02],\n",
      "          [ 4.8362e-02,  5.1872e-02,  1.1304e-02],\n",
      "          [ 2.3429e-02,  3.1187e-02, -1.0451e-02]],\n",
      "\n",
      "         [[-4.6436e-03, -3.1160e-02, -9.3229e-02],\n",
      "          [-3.8935e-02, -1.0535e-02,  1.9997e-02],\n",
      "          [ 5.7535e-02, -6.2687e-02, -3.2334e-02]],\n",
      "\n",
      "         [[ 1.2236e-02, -1.0312e-02, -6.5422e-03],\n",
      "          [-2.2979e-02, -5.8267e-03,  2.7316e-02],\n",
      "          [ 3.9143e-02, -2.5978e-02,  1.6585e-02]]]], requires_grad=True)\n",
      "Parameter containing:\n",
      "tensor([1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1.], requires_grad=True)\n",
      "Parameter containing:\n",
      "tensor([0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0.], requires_grad=True)\n",
      "Parameter containing:\n",
      "tensor([[[[ 0.0172]],\n",
      "\n",
      "         [[-0.0007]],\n",
      "\n",
      "         [[ 0.0516]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[ 0.1361]],\n",
      "\n",
      "         [[-0.1496]],\n",
      "\n",
      "         [[ 0.0220]]],\n",
      "\n",
      "\n",
      "        [[[ 0.1176]],\n",
      "\n",
      "         [[-0.0196]],\n",
      "\n",
      "         [[ 0.0617]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[-0.0719]],\n",
      "\n",
      "         [[-0.0425]],\n",
      "\n",
      "         [[ 0.0485]]],\n",
      "\n",
      "\n",
      "        [[[ 0.0754]],\n",
      "\n",
      "         [[-0.0213]],\n",
      "\n",
      "         [[-0.0065]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[-0.0081]],\n",
      "\n",
      "         [[-0.0619]],\n",
      "\n",
      "         [[ 0.0359]]],\n",
      "\n",
      "\n",
      "        ...,\n",
      "\n",
      "\n",
      "        [[[-0.0475]],\n",
      "\n",
      "         [[-0.0985]],\n",
      "\n",
      "         [[-0.0035]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[ 0.0306]],\n",
      "\n",
      "         [[ 0.0529]],\n",
      "\n",
      "         [[ 0.0425]]],\n",
      "\n",
      "\n",
      "        [[[-0.0819]],\n",
      "\n",
      "         [[-0.1046]],\n",
      "\n",
      "         [[ 0.0639]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[ 0.0752]],\n",
      "\n",
      "         [[ 0.1026]],\n",
      "\n",
      "         [[-0.0756]]],\n",
      "\n",
      "\n",
      "        [[[ 0.1135]],\n",
      "\n",
      "         [[ 0.0166]],\n",
      "\n",
      "         [[-0.0762]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[-0.0099]],\n",
      "\n",
      "         [[-0.0572]],\n",
      "\n",
      "         [[ 0.0531]]]], requires_grad=True)\n",
      "Parameter containing:\n",
      "tensor([1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1.], requires_grad=True)\n",
      "Parameter containing:\n",
      "tensor([0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0.], requires_grad=True)\n",
      "Parameter containing:\n",
      "tensor([[[[-0.0255]],\n",
      "\n",
      "         [[ 0.0005]],\n",
      "\n",
      "         [[-0.1702]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[-0.1520]],\n",
      "\n",
      "         [[-0.0756]],\n",
      "\n",
      "         [[-0.0619]]],\n",
      "\n",
      "\n",
      "        [[[-0.0726]],\n",
      "\n",
      "         [[-0.0502]],\n",
      "\n",
      "         [[ 0.0296]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[ 0.1183]],\n",
      "\n",
      "         [[ 0.0276]],\n",
      "\n",
      "         [[ 0.2231]]],\n",
      "\n",
      "\n",
      "        [[[ 0.1395]],\n",
      "\n",
      "         [[ 0.1309]],\n",
      "\n",
      "         [[ 0.1555]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[ 0.1307]],\n",
      "\n",
      "         [[-0.0528]],\n",
      "\n",
      "         [[ 0.1421]]],\n",
      "\n",
      "\n",
      "        ...,\n",
      "\n",
      "\n",
      "        [[[-0.1946]],\n",
      "\n",
      "         [[ 0.0784]],\n",
      "\n",
      "         [[-0.1331]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[-0.0281]],\n",
      "\n",
      "         [[ 0.1030]],\n",
      "\n",
      "         [[-0.0521]]],\n",
      "\n",
      "\n",
      "        [[[-0.1999]],\n",
      "\n",
      "         [[ 0.1648]],\n",
      "\n",
      "         [[-0.1222]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[-0.1410]],\n",
      "\n",
      "         [[ 0.0054]],\n",
      "\n",
      "         [[ 0.1477]]],\n",
      "\n",
      "\n",
      "        [[[-0.0753]],\n",
      "\n",
      "         [[ 0.0407]],\n",
      "\n",
      "         [[-0.0757]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[ 0.0551]],\n",
      "\n",
      "         [[ 0.0409]],\n",
      "\n",
      "         [[ 0.0224]]]], requires_grad=True)\n",
      "Parameter containing:\n",
      "tensor([1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1.], requires_grad=True)\n",
      "Parameter containing:\n",
      "tensor([0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0.], requires_grad=True)\n",
      "Parameter containing:\n",
      "tensor([[[[-0.1048,  0.0150,  0.0107],\n",
      "          [ 0.0385,  0.0153,  0.0531],\n",
      "          [-0.0152, -0.0314, -0.0022]],\n",
      "\n",
      "         [[ 0.0869,  0.0297,  0.0206],\n",
      "          [ 0.0785,  0.0605,  0.0153],\n",
      "          [-0.0064,  0.0351, -0.0651]],\n",
      "\n",
      "         [[ 0.0226,  0.0337,  0.0244],\n",
      "          [ 0.0068,  0.0849, -0.0758],\n",
      "          [ 0.0376, -0.0302, -0.0466]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[ 0.0107,  0.0136, -0.0228],\n",
      "          [-0.0338,  0.0245,  0.0138],\n",
      "          [ 0.0314, -0.0379, -0.0245]],\n",
      "\n",
      "         [[-0.0086, -0.0115,  0.0079],\n",
      "          [-0.0803, -0.0328, -0.0092],\n",
      "          [ 0.0305,  0.0314,  0.0047]],\n",
      "\n",
      "         [[ 0.0346, -0.0160, -0.0030],\n",
      "          [ 0.0166, -0.0070,  0.0045],\n",
      "          [-0.0548, -0.0460, -0.0707]]],\n",
      "\n",
      "\n",
      "        [[[-0.0146,  0.1036, -0.0010],\n",
      "          [-0.0523,  0.0721,  0.0205],\n",
      "          [-0.0040,  0.0197,  0.0756]],\n",
      "\n",
      "         [[ 0.0508, -0.0400,  0.0444],\n",
      "          [-0.0072, -0.0354, -0.0261],\n",
      "          [-0.0081, -0.0379, -0.0490]],\n",
      "\n",
      "         [[-0.0460,  0.0257, -0.0107],\n",
      "          [ 0.0521, -0.0089, -0.0752],\n",
      "          [-0.0279,  0.0263,  0.0759]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[-0.0474, -0.0461, -0.1031],\n",
      "          [ 0.0140, -0.0123,  0.0011],\n",
      "          [-0.0376,  0.0448, -0.0307]],\n",
      "\n",
      "         [[ 0.0222, -0.0107, -0.0581],\n",
      "          [ 0.0226, -0.0428, -0.0050],\n",
      "          [-0.0062, -0.0022, -0.0413]],\n",
      "\n",
      "         [[ 0.0414,  0.0129, -0.1072],\n",
      "          [-0.0258, -0.0504, -0.0138],\n",
      "          [ 0.0346,  0.0103, -0.0232]]],\n",
      "\n",
      "\n",
      "        [[[ 0.0447,  0.0313,  0.0075],\n",
      "          [-0.0105,  0.0199,  0.0298],\n",
      "          [ 0.0050,  0.0303, -0.0039]],\n",
      "\n",
      "         [[ 0.0831,  0.0227,  0.0085],\n",
      "          [-0.0340,  0.0263,  0.0034],\n",
      "          [ 0.0098, -0.0330,  0.0496]],\n",
      "\n",
      "         [[-0.0595, -0.0077,  0.0263],\n",
      "          [ 0.0725, -0.0152,  0.0619],\n",
      "          [-0.0701, -0.0166, -0.0877]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[ 0.0419, -0.0027, -0.0431],\n",
      "          [ 0.0702,  0.0279,  0.0534],\n",
      "          [-0.0139,  0.0166, -0.0382]],\n",
      "\n",
      "         [[ 0.0351,  0.0389,  0.0103],\n",
      "          [ 0.0354, -0.0990, -0.0047],\n",
      "          [ 0.0098,  0.0349,  0.0441]],\n",
      "\n",
      "         [[ 0.0680,  0.0084, -0.0427],\n",
      "          [-0.0344, -0.0394,  0.0673],\n",
      "          [ 0.0176,  0.0334,  0.0837]]],\n",
      "\n",
      "\n",
      "        ...,\n",
      "\n",
      "\n",
      "        [[[ 0.0225,  0.0302, -0.0151],\n",
      "          [ 0.0352, -0.0524,  0.0254],\n",
      "          [ 0.0342,  0.0264, -0.0635]],\n",
      "\n",
      "         [[ 0.0503,  0.0868,  0.0127],\n",
      "          [-0.0073,  0.0599,  0.0492],\n",
      "          [-0.0456,  0.0555, -0.0476]],\n",
      "\n",
      "         [[ 0.0491,  0.0215,  0.0198],\n",
      "          [-0.0508, -0.0023,  0.0622],\n",
      "          [-0.0093, -0.0328,  0.0281]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[ 0.0062, -0.0193,  0.0121],\n",
      "          [-0.0112, -0.0102,  0.0266],\n",
      "          [-0.0772,  0.0552, -0.0165]],\n",
      "\n",
      "         [[ 0.0177,  0.0938,  0.0119],\n",
      "          [-0.0241, -0.0018, -0.1290],\n",
      "          [ 0.0412, -0.0207, -0.0027]],\n",
      "\n",
      "         [[-0.0089, -0.0101,  0.0594],\n",
      "          [ 0.0227,  0.0626,  0.0713],\n",
      "          [ 0.0702, -0.0061,  0.0216]]],\n",
      "\n",
      "\n",
      "        [[[-0.0302, -0.0281, -0.0057],\n",
      "          [ 0.0181, -0.0913, -0.0985],\n",
      "          [-0.0492,  0.0633, -0.0117]],\n",
      "\n",
      "         [[-0.0592,  0.0301,  0.0464],\n",
      "          [ 0.0860, -0.0232,  0.0485],\n",
      "          [ 0.0059, -0.0078,  0.0373]],\n",
      "\n",
      "         [[-0.0370,  0.0191,  0.0156],\n",
      "          [ 0.0273, -0.0322,  0.0220],\n",
      "          [-0.0237, -0.0481,  0.0257]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[ 0.0177, -0.0583, -0.0686],\n",
      "          [ 0.0006,  0.0457,  0.0317],\n",
      "          [-0.0217,  0.0189, -0.0576]],\n",
      "\n",
      "         [[-0.0080,  0.0490, -0.0297],\n",
      "          [-0.0750,  0.0230, -0.0083],\n",
      "          [-0.0090, -0.0664,  0.0483]],\n",
      "\n",
      "         [[-0.0159,  0.0380, -0.0063],\n",
      "          [-0.0629,  0.0250, -0.0146],\n",
      "          [-0.0264,  0.0108,  0.0052]]],\n",
      "\n",
      "\n",
      "        [[[-0.0362, -0.0436,  0.0299],\n",
      "          [ 0.0748, -0.0167, -0.0431],\n",
      "          [ 0.0717,  0.0600,  0.0306]],\n",
      "\n",
      "         [[ 0.0287,  0.0095, -0.0565],\n",
      "          [-0.0772, -0.0180, -0.0371],\n",
      "          [ 0.0591,  0.0179, -0.0178]],\n",
      "\n",
      "         [[ 0.0306, -0.0982, -0.0440],\n",
      "          [-0.0082, -0.0294,  0.0291],\n",
      "          [ 0.0337,  0.0011,  0.0108]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[ 0.0922, -0.0090, -0.0602],\n",
      "          [ 0.0222, -0.0420,  0.0259],\n",
      "          [-0.0027,  0.0476, -0.0181]],\n",
      "\n",
      "         [[ 0.0083,  0.0029,  0.0250],\n",
      "          [-0.0426,  0.0162, -0.0007],\n",
      "          [-0.0372, -0.0804, -0.0025]],\n",
      "\n",
      "         [[ 0.0175, -0.0122, -0.0268],\n",
      "          [ 0.0605, -0.0234, -0.0242],\n",
      "          [ 0.0113, -0.0202,  0.0194]]]], requires_grad=True)\n",
      "Parameter containing:\n",
      "tensor([1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1.], requires_grad=True)\n",
      "Parameter containing:\n",
      "tensor([0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0.], requires_grad=True)\n",
      "Parameter containing:\n",
      "tensor([[[[ 0.0493]],\n",
      "\n",
      "         [[ 0.0169]],\n",
      "\n",
      "         [[ 0.0214]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[ 0.0185]],\n",
      "\n",
      "         [[ 0.0110]],\n",
      "\n",
      "         [[-0.0338]]],\n",
      "\n",
      "\n",
      "        [[[-0.0231]],\n",
      "\n",
      "         [[ 0.0467]],\n",
      "\n",
      "         [[ 0.0413]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[ 0.0380]],\n",
      "\n",
      "         [[-0.0544]],\n",
      "\n",
      "         [[ 0.1462]]],\n",
      "\n",
      "\n",
      "        [[[ 0.0147]],\n",
      "\n",
      "         [[-0.0382]],\n",
      "\n",
      "         [[ 0.0268]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[-0.0047]],\n",
      "\n",
      "         [[-0.0204]],\n",
      "\n",
      "         [[ 0.0133]]],\n",
      "\n",
      "\n",
      "        ...,\n",
      "\n",
      "\n",
      "        [[[-0.0406]],\n",
      "\n",
      "         [[-0.0450]],\n",
      "\n",
      "         [[-0.0214]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[-0.0456]],\n",
      "\n",
      "         [[-0.0301]],\n",
      "\n",
      "         [[ 0.0506]]],\n",
      "\n",
      "\n",
      "        [[[-0.0750]],\n",
      "\n",
      "         [[ 0.0266]],\n",
      "\n",
      "         [[-0.0373]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[-0.0791]],\n",
      "\n",
      "         [[-0.0195]],\n",
      "\n",
      "         [[ 0.0143]]],\n",
      "\n",
      "\n",
      "        [[[-0.0173]],\n",
      "\n",
      "         [[ 0.0247]],\n",
      "\n",
      "         [[-0.0055]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[ 0.0189]],\n",
      "\n",
      "         [[-0.1061]],\n",
      "\n",
      "         [[ 0.1004]]]], requires_grad=True)\n",
      "Parameter containing:\n",
      "tensor([1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1.], requires_grad=True)\n",
      "Parameter containing:\n",
      "tensor([0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0.], requires_grad=True)\n",
      "Parameter containing:\n",
      "tensor([[[[-0.1288]],\n",
      "\n",
      "         [[-0.0930]],\n",
      "\n",
      "         [[-0.1033]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[-0.0060]],\n",
      "\n",
      "         [[ 0.0351]],\n",
      "\n",
      "         [[-0.0418]]],\n",
      "\n",
      "\n",
      "        [[[-0.0418]],\n",
      "\n",
      "         [[-0.0522]],\n",
      "\n",
      "         [[ 0.0445]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[ 0.3478]],\n",
      "\n",
      "         [[ 0.1112]],\n",
      "\n",
      "         [[-0.0633]]],\n",
      "\n",
      "\n",
      "        [[[-0.0835]],\n",
      "\n",
      "         [[-0.0037]],\n",
      "\n",
      "         [[-0.0509]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[-0.0330]],\n",
      "\n",
      "         [[ 0.0865]],\n",
      "\n",
      "         [[-0.0239]]],\n",
      "\n",
      "\n",
      "        ...,\n",
      "\n",
      "\n",
      "        [[[ 0.1619]],\n",
      "\n",
      "         [[-0.0043]],\n",
      "\n",
      "         [[-0.0249]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[ 0.0509]],\n",
      "\n",
      "         [[-0.0083]],\n",
      "\n",
      "         [[ 0.0033]]],\n",
      "\n",
      "\n",
      "        [[[ 0.0640]],\n",
      "\n",
      "         [[-0.1110]],\n",
      "\n",
      "         [[ 0.1458]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[ 0.0077]],\n",
      "\n",
      "         [[ 0.0347]],\n",
      "\n",
      "         [[-0.1393]]],\n",
      "\n",
      "\n",
      "        [[[-0.1392]],\n",
      "\n",
      "         [[ 0.0456]],\n",
      "\n",
      "         [[ 0.0537]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[ 0.0102]],\n",
      "\n",
      "         [[ 0.0243]],\n",
      "\n",
      "         [[-0.0019]]]], requires_grad=True)\n",
      "Parameter containing:\n",
      "tensor([1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1.], requires_grad=True)\n",
      "Parameter containing:\n",
      "tensor([0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.],\n",
      "       requires_grad=True)\n",
      "Parameter containing:\n",
      "tensor([[[[ 8.6450e-03, -3.6220e-02, -1.2370e-02],\n",
      "          [-3.6236e-03,  3.1731e-02, -6.7624e-03],\n",
      "          [-1.5471e-02, -3.7503e-02, -5.7250e-02]],\n",
      "\n",
      "         [[-1.0017e-02, -1.0074e-02,  5.4915e-03],\n",
      "          [ 3.1515e-02, -1.5750e-03, -5.8689e-02],\n",
      "          [-3.6270e-02, -1.5889e-02,  9.5614e-03]],\n",
      "\n",
      "         [[-5.0546e-04,  3.3067e-02, -2.7436e-02],\n",
      "          [-6.9619e-02, -1.1787e-02, -1.7915e-02],\n",
      "          [ 6.3659e-02, -4.7417e-02, -3.0805e-03]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[ 2.5794e-02, -2.3910e-02, -1.5953e-02],\n",
      "          [-3.5737e-04,  1.0125e-03, -6.1502e-03],\n",
      "          [ 3.5020e-02, -2.8112e-02,  3.7381e-02]],\n",
      "\n",
      "         [[ 3.1662e-03, -2.0708e-02,  1.6886e-02],\n",
      "          [ 2.2564e-02, -1.3523e-02,  1.0247e-02],\n",
      "          [-3.6253e-02, -2.5158e-02,  3.2220e-02]],\n",
      "\n",
      "         [[ 2.0494e-03,  1.1031e-02,  5.4340e-02],\n",
      "          [ 7.3318e-03, -1.8108e-02,  1.6223e-02],\n",
      "          [ 2.2228e-02, -4.3517e-02, -1.7127e-02]]],\n",
      "\n",
      "\n",
      "        [[[-3.8536e-02,  3.4226e-02, -1.9786e-02],\n",
      "          [ 1.8396e-02, -2.6733e-02,  3.6561e-02],\n",
      "          [ 1.8441e-02,  4.9259e-02,  3.5324e-02]],\n",
      "\n",
      "         [[ 1.4272e-02,  3.0279e-02,  2.2425e-02],\n",
      "          [-7.2963e-03,  7.4193e-03, -1.6857e-02],\n",
      "          [-3.6338e-02, -2.3628e-02, -2.8804e-02]],\n",
      "\n",
      "         [[-7.3541e-03,  3.8323e-03,  5.2550e-03],\n",
      "          [ 2.6682e-02, -2.2740e-02, -1.9216e-02],\n",
      "          [ 4.2924e-02, -2.2448e-02, -4.7410e-02]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[-1.4102e-02, -8.4964e-03, -3.2271e-02],\n",
      "          [ 4.6531e-02, -2.5178e-02,  4.4408e-02],\n",
      "          [ 2.6202e-03,  2.4007e-02,  3.9943e-02]],\n",
      "\n",
      "         [[-7.1578e-02, -2.3453e-02,  2.8639e-02],\n",
      "          [-4.5490e-02, -5.1045e-02, -6.9599e-02],\n",
      "          [-8.6851e-03,  4.4051e-02, -1.1111e-02]],\n",
      "\n",
      "         [[ 6.0129e-03,  3.4856e-02,  5.5759e-03],\n",
      "          [ 5.4938e-02, -2.3354e-03,  1.7759e-02],\n",
      "          [-1.4026e-02, -1.6329e-02, -6.7204e-03]]],\n",
      "\n",
      "\n",
      "        [[[-4.9266e-03,  1.8196e-03,  4.2342e-02],\n",
      "          [-3.4199e-02, -2.1663e-02,  6.2849e-03],\n",
      "          [ 2.5063e-02, -6.1477e-03, -5.1457e-02]],\n",
      "\n",
      "         [[-2.1594e-02,  4.6923e-02, -1.2558e-02],\n",
      "          [ 2.2300e-02,  3.9030e-03,  7.0939e-04],\n",
      "          [ 1.1768e-02,  3.9038e-02,  7.6763e-02]],\n",
      "\n",
      "         [[ 1.9731e-03, -2.9982e-02,  4.3085e-02],\n",
      "          [ 1.8253e-02, -1.9898e-02, -4.7471e-03],\n",
      "          [ 2.5349e-02,  3.0437e-02,  1.2006e-02]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[ 1.3391e-04, -1.2220e-02,  5.7744e-04],\n",
      "          [-1.9945e-02, -1.8891e-03, -4.0860e-02],\n",
      "          [-3.8708e-02, -1.6232e-02, -6.2687e-03]],\n",
      "\n",
      "         [[ 3.3511e-02, -3.6786e-02, -1.1674e-02],\n",
      "          [-5.3478e-02,  3.7815e-02, -2.3119e-02],\n",
      "          [ 1.4151e-02,  3.6356e-02, -2.2273e-02]],\n",
      "\n",
      "         [[-1.5529e-02, -1.9329e-03,  8.5645e-02],\n",
      "          [ 1.2331e-02,  2.8454e-02, -6.7287e-02],\n",
      "          [ 3.5371e-02,  8.2864e-02,  3.9460e-02]]],\n",
      "\n",
      "\n",
      "        ...,\n",
      "\n",
      "\n",
      "        [[[ 2.2075e-02,  1.1078e-02,  2.8810e-02],\n",
      "          [ 2.8054e-02, -5.9727e-02,  3.2843e-02],\n",
      "          [-1.2434e-02,  4.7407e-02,  6.2827e-02]],\n",
      "\n",
      "         [[ 5.7659e-02, -2.8401e-02, -3.7880e-02],\n",
      "          [-2.9512e-02, -1.0028e-02, -9.4585e-03],\n",
      "          [ 4.1132e-02,  2.9286e-04, -1.5831e-02]],\n",
      "\n",
      "         [[-1.4087e-02,  1.5805e-02, -3.7179e-04],\n",
      "          [-1.8611e-02, -3.3807e-02,  9.8982e-03],\n",
      "          [-4.6268e-02, -2.0114e-02, -1.0755e-02]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[ 4.5369e-02, -5.5088e-03, -4.1013e-02],\n",
      "          [-2.7242e-02, -2.4957e-03, -1.9771e-02],\n",
      "          [ 2.0025e-02, -3.3236e-02,  4.6695e-02]],\n",
      "\n",
      "         [[-3.6731e-03, -6.6800e-03,  1.6770e-02],\n",
      "          [-7.0417e-02, -5.7572e-03,  9.5455e-03],\n",
      "          [-5.5460e-02,  1.3488e-03,  2.8914e-02]],\n",
      "\n",
      "         [[-2.4099e-02, -7.8005e-03,  3.0240e-02],\n",
      "          [-2.1466e-02, -1.3633e-02,  3.1374e-02],\n",
      "          [ 2.5701e-02,  9.8833e-03,  2.6736e-02]]],\n",
      "\n",
      "\n",
      "        [[[-1.7732e-02,  3.6289e-03,  5.5313e-02],\n",
      "          [-2.3405e-02,  1.9414e-02, -1.7504e-02],\n",
      "          [-4.2984e-02,  1.7238e-02,  5.4131e-03]],\n",
      "\n",
      "         [[-9.2196e-04, -1.6275e-02,  1.5663e-02],\n",
      "          [ 5.4654e-02,  1.0567e-02,  2.7551e-02],\n",
      "          [ 9.2423e-04, -1.0343e-03,  2.0372e-02]],\n",
      "\n",
      "         [[-1.8219e-02, -1.4261e-02, -2.2790e-02],\n",
      "          [ 3.5550e-02, -2.2293e-02,  2.8467e-02],\n",
      "          [ 2.0403e-02,  5.3843e-02,  3.6060e-02]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[-5.9613e-03,  1.8261e-02,  2.3899e-02],\n",
      "          [ 4.7187e-02, -5.6193e-02, -1.7249e-02],\n",
      "          [-2.5599e-03, -3.5927e-02, -1.7552e-02]],\n",
      "\n",
      "         [[ 3.0460e-03, -7.5494e-02, -3.1038e-02],\n",
      "          [ 1.5843e-03, -9.5869e-02,  6.0737e-02],\n",
      "          [-1.8542e-03,  5.9428e-02,  2.2577e-02]],\n",
      "\n",
      "         [[-1.0706e-02, -1.7088e-02,  1.3423e-02],\n",
      "          [ 1.9165e-02, -1.0075e-02,  1.0145e-03],\n",
      "          [ 1.7173e-02, -1.7788e-02,  2.1995e-02]]],\n",
      "\n",
      "\n",
      "        [[[-2.0757e-02, -5.4126e-03, -4.3861e-02],\n",
      "          [ 1.9929e-03,  2.0243e-02,  3.2892e-02],\n",
      "          [ 3.2314e-05,  5.3802e-03, -4.7783e-03]],\n",
      "\n",
      "         [[ 3.0538e-03, -1.7069e-02, -1.0062e-02],\n",
      "          [-3.5603e-02,  8.2916e-03,  8.4039e-03],\n",
      "          [ 2.2530e-03,  7.7052e-03, -4.0716e-04]],\n",
      "\n",
      "         [[ 1.0601e-02, -1.1875e-02, -3.3064e-02],\n",
      "          [ 1.2496e-02,  6.6994e-03, -3.8271e-02],\n",
      "          [ 2.2973e-02, -8.9931e-04, -1.4169e-02]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[ 1.3068e-02, -6.8434e-03, -3.8263e-02],\n",
      "          [ 9.1860e-03,  4.2982e-02,  7.4634e-02],\n",
      "          [-5.7035e-03, -7.4133e-03,  2.2298e-02]],\n",
      "\n",
      "         [[ 7.5734e-03,  4.7200e-02, -9.7152e-03],\n",
      "          [-4.8154e-02,  3.7818e-02, -1.5118e-02],\n",
      "          [ 1.9427e-02, -2.1329e-02,  4.4302e-02]],\n",
      "\n",
      "         [[ 2.9608e-02,  8.9050e-03, -1.6396e-02],\n",
      "          [-4.0602e-02,  2.6723e-03,  2.2734e-02],\n",
      "          [ 2.6886e-02, -1.4196e-02, -2.2184e-02]]]], requires_grad=True)\n",
      "Parameter containing:\n",
      "tensor([1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1.], requires_grad=True)\n",
      "Parameter containing:\n",
      "tensor([0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.],\n",
      "       requires_grad=True)\n",
      "Parameter containing:\n",
      "tensor([[[[-0.0025]],\n",
      "\n",
      "         [[ 0.0735]],\n",
      "\n",
      "         [[ 0.0374]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[ 0.0791]],\n",
      "\n",
      "         [[-0.0293]],\n",
      "\n",
      "         [[-0.0522]]],\n",
      "\n",
      "\n",
      "        [[[ 0.0145]],\n",
      "\n",
      "         [[-0.0181]],\n",
      "\n",
      "         [[-0.0281]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[-0.0848]],\n",
      "\n",
      "         [[ 0.0608]],\n",
      "\n",
      "         [[ 0.0843]]],\n",
      "\n",
      "\n",
      "        [[[ 0.0637]],\n",
      "\n",
      "         [[ 0.0271]],\n",
      "\n",
      "         [[-0.0880]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[-0.0137]],\n",
      "\n",
      "         [[-0.0072]],\n",
      "\n",
      "         [[ 0.0499]]],\n",
      "\n",
      "\n",
      "        ...,\n",
      "\n",
      "\n",
      "        [[[ 0.0350]],\n",
      "\n",
      "         [[ 0.0914]],\n",
      "\n",
      "         [[ 0.0641]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[ 0.0553]],\n",
      "\n",
      "         [[-0.0551]],\n",
      "\n",
      "         [[ 0.0573]]],\n",
      "\n",
      "\n",
      "        [[[ 0.0819]],\n",
      "\n",
      "         [[-0.0105]],\n",
      "\n",
      "         [[ 0.0200]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[ 0.0234]],\n",
      "\n",
      "         [[-0.0431]],\n",
      "\n",
      "         [[-0.0652]]],\n",
      "\n",
      "\n",
      "        [[[ 0.0085]],\n",
      "\n",
      "         [[ 0.0632]],\n",
      "\n",
      "         [[-0.0107]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[-0.0944]],\n",
      "\n",
      "         [[-0.0035]],\n",
      "\n",
      "         [[-0.0952]]]], requires_grad=True)\n",
      "Parameter containing:\n",
      "tensor([1., 1., 1.,  ..., 1., 1., 1.], requires_grad=True)\n",
      "Parameter containing:\n",
      "tensor([0., 0., 0.,  ..., 0., 0., 0.], requires_grad=True)\n",
      "Parameter containing:\n",
      "tensor([[[[ 0.0472]],\n",
      "\n",
      "         [[-0.0144]],\n",
      "\n",
      "         [[-0.0449]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[-0.0332]],\n",
      "\n",
      "         [[ 0.0462]],\n",
      "\n",
      "         [[ 0.0423]]],\n",
      "\n",
      "\n",
      "        [[[-0.0045]],\n",
      "\n",
      "         [[ 0.0178]],\n",
      "\n",
      "         [[ 0.0128]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[ 0.0071]],\n",
      "\n",
      "         [[-0.0774]],\n",
      "\n",
      "         [[ 0.0758]]],\n",
      "\n",
      "\n",
      "        [[[ 0.1121]],\n",
      "\n",
      "         [[ 0.0192]],\n",
      "\n",
      "         [[ 0.0416]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[ 0.0159]],\n",
      "\n",
      "         [[-0.0056]],\n",
      "\n",
      "         [[ 0.0085]]],\n",
      "\n",
      "\n",
      "        ...,\n",
      "\n",
      "\n",
      "        [[[-0.0084]],\n",
      "\n",
      "         [[ 0.0019]],\n",
      "\n",
      "         [[ 0.0430]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[-0.0642]],\n",
      "\n",
      "         [[-0.0102]],\n",
      "\n",
      "         [[-0.0295]]],\n",
      "\n",
      "\n",
      "        [[[ 0.0160]],\n",
      "\n",
      "         [[-0.0782]],\n",
      "\n",
      "         [[ 0.0714]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[ 0.0361]],\n",
      "\n",
      "         [[-0.0215]],\n",
      "\n",
      "         [[-0.0382]]],\n",
      "\n",
      "\n",
      "        [[[ 0.0252]],\n",
      "\n",
      "         [[ 0.0207]],\n",
      "\n",
      "         [[ 0.0033]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[ 0.0389]],\n",
      "\n",
      "         [[-0.0721]],\n",
      "\n",
      "         [[ 0.0252]]]], requires_grad=True)\n",
      "Parameter containing:\n",
      "tensor([1., 1., 1.,  ..., 1., 1., 1.], requires_grad=True)\n",
      "Parameter containing:\n",
      "tensor([0., 0., 0.,  ..., 0., 0., 0.], requires_grad=True)\n",
      "Parameter containing:\n",
      "tensor([[[[ 0.0273]],\n",
      "\n",
      "         [[ 0.0630]],\n",
      "\n",
      "         [[-0.1712]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[-0.0040]],\n",
      "\n",
      "         [[-0.1404]],\n",
      "\n",
      "         [[ 0.0373]]],\n",
      "\n",
      "\n",
      "        [[[-0.0194]],\n",
      "\n",
      "         [[ 0.1403]],\n",
      "\n",
      "         [[ 0.1534]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[ 0.0872]],\n",
      "\n",
      "         [[ 0.0768]],\n",
      "\n",
      "         [[-0.0250]]],\n",
      "\n",
      "\n",
      "        [[[-0.2410]],\n",
      "\n",
      "         [[ 0.0080]],\n",
      "\n",
      "         [[-0.0181]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[ 0.0861]],\n",
      "\n",
      "         [[ 0.0097]],\n",
      "\n",
      "         [[ 0.1372]]],\n",
      "\n",
      "\n",
      "        ...,\n",
      "\n",
      "\n",
      "        [[[-0.0770]],\n",
      "\n",
      "         [[-0.0737]],\n",
      "\n",
      "         [[ 0.1083]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[-0.0812]],\n",
      "\n",
      "         [[ 0.0639]],\n",
      "\n",
      "         [[-0.0801]]],\n",
      "\n",
      "\n",
      "        [[[-0.0719]],\n",
      "\n",
      "         [[-0.0112]],\n",
      "\n",
      "         [[-0.0593]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[-0.0597]],\n",
      "\n",
      "         [[ 0.0956]],\n",
      "\n",
      "         [[ 0.1059]]],\n",
      "\n",
      "\n",
      "        [[[ 0.1759]],\n",
      "\n",
      "         [[ 0.0409]],\n",
      "\n",
      "         [[ 0.0356]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[ 0.0090]],\n",
      "\n",
      "         [[ 0.0524]],\n",
      "\n",
      "         [[-0.0655]]]], requires_grad=True)\n",
      "Parameter containing:\n",
      "tensor([1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1.], requires_grad=True)\n",
      "Parameter containing:\n",
      "tensor([0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.],\n",
      "       requires_grad=True)\n",
      "Parameter containing:\n",
      "tensor([[[[-2.1633e-02, -6.2556e-03, -1.7565e-02],\n",
      "          [-2.7176e-02, -1.0414e-02, -9.3964e-03],\n",
      "          [ 2.9437e-02,  9.9999e-03,  3.9103e-02]],\n",
      "\n",
      "         [[ 4.1530e-02, -4.8456e-02, -9.8153e-03],\n",
      "          [ 5.2492e-02,  2.8286e-03, -1.5742e-02],\n",
      "          [-5.7288e-03,  3.4527e-02,  6.0712e-02]],\n",
      "\n",
      "         [[-3.4483e-02,  1.9734e-02, -4.1369e-02],\n",
      "          [-2.6665e-02, -3.5534e-02,  4.9004e-03],\n",
      "          [-7.6747e-03,  2.6262e-02, -3.9190e-02]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[-3.1917e-03, -1.0694e-02, -6.3750e-03],\n",
      "          [-6.1946e-02,  4.4390e-02,  2.4201e-02],\n",
      "          [-4.2262e-02,  2.6302e-02,  2.7604e-02]],\n",
      "\n",
      "         [[ 4.5122e-02, -1.3205e-02,  4.3689e-02],\n",
      "          [-2.1562e-02, -1.6623e-02,  3.2843e-02],\n",
      "          [ 2.8159e-02,  1.3300e-02, -3.4047e-03]],\n",
      "\n",
      "         [[ 5.2710e-03, -2.5382e-02, -1.3950e-02],\n",
      "          [ 2.4368e-02, -4.9448e-02, -1.8197e-02],\n",
      "          [-6.3151e-02, -1.2301e-02,  1.8641e-02]]],\n",
      "\n",
      "\n",
      "        [[[ 4.7220e-02,  2.8493e-02,  2.0551e-02],\n",
      "          [ 3.2258e-02,  5.6500e-03,  1.1850e-02],\n",
      "          [-8.7728e-03, -1.6998e-02,  4.4779e-02]],\n",
      "\n",
      "         [[ 1.1388e-02, -4.5892e-02, -4.3710e-02],\n",
      "          [-3.6562e-02, -1.5562e-02, -5.0290e-03],\n",
      "          [ 5.8536e-03, -7.6249e-03, -2.7781e-03]],\n",
      "\n",
      "         [[ 1.8222e-02,  3.4243e-02,  4.1114e-02],\n",
      "          [-1.3391e-03, -2.1108e-03,  1.6720e-02],\n",
      "          [ 2.0153e-02,  2.6922e-02, -3.6700e-02]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[-6.2760e-02,  3.8914e-02, -5.1967e-02],\n",
      "          [-1.3257e-02,  4.5878e-03, -5.1419e-03],\n",
      "          [-1.3159e-02, -1.6067e-03,  1.6388e-02]],\n",
      "\n",
      "         [[ 2.8184e-02, -6.3088e-02, -1.5102e-02],\n",
      "          [ 2.1941e-02,  3.2683e-02,  6.1578e-02],\n",
      "          [ 1.5930e-02,  3.1701e-02, -8.2280e-03]],\n",
      "\n",
      "         [[-5.6700e-02,  1.6665e-02, -9.8362e-03],\n",
      "          [ 2.6427e-03,  1.3511e-02, -1.5494e-02],\n",
      "          [ 2.9573e-02,  3.3305e-02, -1.9605e-04]]],\n",
      "\n",
      "\n",
      "        [[[ 1.8311e-02, -4.3148e-03, -1.3897e-02],\n",
      "          [-1.0801e-02, -5.9802e-02, -4.3849e-02],\n",
      "          [ 3.7835e-03, -1.6016e-02, -4.5165e-02]],\n",
      "\n",
      "         [[ 3.6621e-02, -2.2555e-02,  8.2920e-03],\n",
      "          [-4.7936e-02, -3.1506e-02, -2.4844e-02],\n",
      "          [-4.1356e-03,  1.7744e-02,  1.6921e-02]],\n",
      "\n",
      "         [[-1.3992e-02, -8.5304e-02, -1.2118e-02],\n",
      "          [-2.6336e-02, -6.6468e-02,  7.0981e-02],\n",
      "          [-4.5086e-03, -2.0590e-02,  8.7054e-02]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[-8.9704e-03,  1.4459e-02, -6.0695e-03],\n",
      "          [ 4.9926e-02, -5.9740e-03, -1.7945e-04],\n",
      "          [ 6.1411e-02, -8.9193e-03, -3.0004e-02]],\n",
      "\n",
      "         [[ 3.3124e-02, -7.4831e-03, -3.7899e-02],\n",
      "          [ 1.1621e-02, -2.0374e-02,  1.1097e-02],\n",
      "          [ 2.7172e-02,  3.2579e-03,  2.1230e-02]],\n",
      "\n",
      "         [[-1.0596e-02, -2.4040e-02,  2.3210e-02],\n",
      "          [-2.5958e-02,  3.1329e-03, -4.9191e-03],\n",
      "          [ 6.2984e-04,  2.2089e-02,  4.9437e-03]]],\n",
      "\n",
      "\n",
      "        ...,\n",
      "\n",
      "\n",
      "        [[[-3.2986e-02,  5.9220e-02,  1.6920e-02],\n",
      "          [ 2.1099e-02,  1.5517e-03, -3.4709e-03],\n",
      "          [ 2.5518e-02, -8.5471e-03,  2.3098e-02]],\n",
      "\n",
      "         [[-6.5014e-03, -2.9501e-02,  3.9018e-02],\n",
      "          [-3.0604e-03,  9.8948e-03,  2.9425e-03],\n",
      "          [-1.9892e-02,  8.6301e-03,  2.7451e-03]],\n",
      "\n",
      "         [[ 3.1796e-02,  7.9100e-03, -1.1980e-02],\n",
      "          [ 4.4959e-02, -6.7314e-02,  2.6241e-02],\n",
      "          [ 4.0287e-02,  2.1216e-02,  4.2142e-05]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[-3.9620e-02,  2.7072e-03, -5.4759e-03],\n",
      "          [ 1.0174e-03, -6.3164e-02,  1.7876e-02],\n",
      "          [ 1.7386e-02, -6.6101e-02, -5.0446e-03]],\n",
      "\n",
      "         [[ 3.6486e-02,  2.5797e-02,  4.7008e-03],\n",
      "          [ 2.1431e-02, -1.9652e-02,  2.8356e-02],\n",
      "          [-5.0331e-04,  3.5619e-02, -5.9216e-02]],\n",
      "\n",
      "         [[-3.9632e-02,  3.0057e-02, -1.3800e-02],\n",
      "          [ 2.0620e-03,  1.8637e-03, -3.5909e-02],\n",
      "          [ 3.2994e-03,  5.4513e-02,  4.6552e-03]]],\n",
      "\n",
      "\n",
      "        [[[ 3.2702e-02, -6.4298e-02,  1.3628e-02],\n",
      "          [ 2.8340e-02, -1.3280e-02, -2.0823e-02],\n",
      "          [-2.0422e-02,  4.8771e-02, -3.0548e-02]],\n",
      "\n",
      "         [[ 2.3599e-02,  2.4966e-02,  3.7863e-02],\n",
      "          [-1.3936e-02, -3.3259e-02, -1.6177e-02],\n",
      "          [-2.8920e-02,  1.8759e-03,  2.0166e-02]],\n",
      "\n",
      "         [[ 3.1059e-03,  1.3037e-02,  2.0411e-02],\n",
      "          [-2.5129e-02,  1.5331e-02, -4.0612e-02],\n",
      "          [ 2.1184e-04,  5.8345e-02,  2.7618e-02]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[ 3.2227e-02, -2.0048e-02, -1.2190e-03],\n",
      "          [-3.4710e-02,  2.8494e-02,  7.3413e-03],\n",
      "          [ 3.1746e-02, -8.7655e-03, -3.1200e-02]],\n",
      "\n",
      "         [[-1.3525e-02, -6.9152e-03,  2.5218e-02],\n",
      "          [ 1.1186e-02, -3.5562e-03,  1.9937e-03],\n",
      "          [-3.3482e-02, -4.6315e-02,  1.4220e-02]],\n",
      "\n",
      "         [[-2.4250e-02, -1.3781e-02,  2.5308e-04],\n",
      "          [-1.5785e-02, -1.7483e-02,  2.2761e-02],\n",
      "          [ 1.1646e-02, -3.9696e-02, -7.0195e-02]]],\n",
      "\n",
      "\n",
      "        [[[ 2.1802e-02, -2.6051e-02, -3.8277e-03],\n",
      "          [ 2.8843e-02,  4.7176e-02,  3.0422e-02],\n",
      "          [ 8.9039e-03,  1.6011e-02,  4.5221e-02]],\n",
      "\n",
      "         [[ 8.9427e-03,  1.6229e-02, -5.0926e-02],\n",
      "          [ 4.3477e-02, -4.9209e-02,  3.3306e-02],\n",
      "          [-2.5741e-02, -2.1584e-03, -3.8832e-03]],\n",
      "\n",
      "         [[ 1.7636e-03,  5.6759e-03, -3.7416e-03],\n",
      "          [ 6.6438e-02,  2.1708e-02,  4.4017e-03],\n",
      "          [ 2.7677e-02, -3.7403e-02, -1.1365e-02]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[-2.2883e-02,  1.1991e-02, -3.1722e-02],\n",
      "          [ 2.3152e-03, -3.1073e-02,  1.9906e-02],\n",
      "          [-3.0671e-03, -1.0319e-01, -2.1137e-03]],\n",
      "\n",
      "         [[ 3.3572e-02, -5.9525e-04, -9.8765e-03],\n",
      "          [-5.8230e-03,  2.4302e-02, -3.4925e-02],\n",
      "          [-5.7017e-03, -3.0090e-02, -6.5260e-05]],\n",
      "\n",
      "         [[-4.0854e-02, -1.8535e-02, -3.9794e-02],\n",
      "          [ 5.4147e-03, -2.5603e-02, -1.4352e-02],\n",
      "          [ 1.7410e-02, -3.5337e-02, -5.0565e-02]]]], requires_grad=True)\n",
      "Parameter containing:\n",
      "tensor([1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1.], requires_grad=True)\n",
      "Parameter containing:\n",
      "tensor([0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.],\n",
      "       requires_grad=True)\n",
      "Parameter containing:\n",
      "tensor([[[[-0.0633]],\n",
      "\n",
      "         [[ 0.0064]],\n",
      "\n",
      "         [[-0.0541]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[ 0.0011]],\n",
      "\n",
      "         [[-0.0144]],\n",
      "\n",
      "         [[-0.0055]]],\n",
      "\n",
      "\n",
      "        [[[ 0.0676]],\n",
      "\n",
      "         [[-0.0076]],\n",
      "\n",
      "         [[ 0.0479]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[-0.0561]],\n",
      "\n",
      "         [[ 0.0211]],\n",
      "\n",
      "         [[ 0.0904]]],\n",
      "\n",
      "\n",
      "        [[[ 0.0002]],\n",
      "\n",
      "         [[-0.0301]],\n",
      "\n",
      "         [[ 0.0631]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[-0.0389]],\n",
      "\n",
      "         [[ 0.0079]],\n",
      "\n",
      "         [[-0.0248]]],\n",
      "\n",
      "\n",
      "        ...,\n",
      "\n",
      "\n",
      "        [[[-0.0176]],\n",
      "\n",
      "         [[-0.0271]],\n",
      "\n",
      "         [[-0.0199]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[ 0.0247]],\n",
      "\n",
      "         [[-0.0590]],\n",
      "\n",
      "         [[ 0.0040]]],\n",
      "\n",
      "\n",
      "        [[[-0.0171]],\n",
      "\n",
      "         [[ 0.0219]],\n",
      "\n",
      "         [[-0.0632]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[-0.0657]],\n",
      "\n",
      "         [[ 0.0735]],\n",
      "\n",
      "         [[-0.0494]]],\n",
      "\n",
      "\n",
      "        [[[ 0.0192]],\n",
      "\n",
      "         [[-0.0177]],\n",
      "\n",
      "         [[ 0.0172]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[ 0.0301]],\n",
      "\n",
      "         [[-0.0041]],\n",
      "\n",
      "         [[-0.0271]]]], requires_grad=True)\n",
      "Parameter containing:\n",
      "tensor([1., 1., 1.,  ..., 1., 1., 1.], requires_grad=True)\n",
      "Parameter containing:\n",
      "tensor([0., 0., 0.,  ..., 0., 0., 0.], requires_grad=True)\n",
      "Parameter containing:\n",
      "tensor([[[[ 0.0387]],\n",
      "\n",
      "         [[ 0.1858]],\n",
      "\n",
      "         [[-0.0205]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[-0.1549]],\n",
      "\n",
      "         [[ 0.0185]],\n",
      "\n",
      "         [[-0.0088]]],\n",
      "\n",
      "\n",
      "        [[[-0.0387]],\n",
      "\n",
      "         [[ 0.0877]],\n",
      "\n",
      "         [[-0.0346]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[ 0.0723]],\n",
      "\n",
      "         [[ 0.0562]],\n",
      "\n",
      "         [[ 0.0762]]],\n",
      "\n",
      "\n",
      "        [[[-0.0363]],\n",
      "\n",
      "         [[-0.0916]],\n",
      "\n",
      "         [[-0.0322]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[ 0.1011]],\n",
      "\n",
      "         [[-0.1409]],\n",
      "\n",
      "         [[ 0.1207]]],\n",
      "\n",
      "\n",
      "        ...,\n",
      "\n",
      "\n",
      "        [[[ 0.1065]],\n",
      "\n",
      "         [[-0.1158]],\n",
      "\n",
      "         [[-0.0166]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[ 0.0119]],\n",
      "\n",
      "         [[ 0.2378]],\n",
      "\n",
      "         [[-0.0723]]],\n",
      "\n",
      "\n",
      "        [[[-0.0519]],\n",
      "\n",
      "         [[-0.0209]],\n",
      "\n",
      "         [[ 0.2203]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[-0.0363]],\n",
      "\n",
      "         [[-0.1383]],\n",
      "\n",
      "         [[ 0.0129]]],\n",
      "\n",
      "\n",
      "        [[[-0.0277]],\n",
      "\n",
      "         [[ 0.0888]],\n",
      "\n",
      "         [[-0.0996]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[-0.0756]],\n",
      "\n",
      "         [[-0.1841]],\n",
      "\n",
      "         [[ 0.0491]]]], requires_grad=True)\n",
      "Parameter containing:\n",
      "tensor([1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1.], requires_grad=True)\n",
      "Parameter containing:\n",
      "tensor([0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.],\n",
      "       requires_grad=True)\n",
      "Parameter containing:\n",
      "tensor([[[[-1.2538e-02, -1.2491e-03,  1.9862e-02],\n",
      "          [ 9.6418e-03,  3.2837e-02, -1.0114e-02],\n",
      "          [-4.0021e-02,  2.0841e-02,  1.5479e-02]],\n",
      "\n",
      "         [[-2.6892e-03,  4.4154e-02, -3.9614e-02],\n",
      "          [ 3.9538e-02,  3.1559e-02, -1.1889e-02],\n",
      "          [ 1.7015e-02,  3.1766e-02,  2.7985e-03]],\n",
      "\n",
      "         [[ 2.5367e-02, -2.0648e-02, -1.4971e-02],\n",
      "          [ 2.4029e-02,  4.0393e-02,  3.2597e-05],\n",
      "          [ 1.4851e-02,  3.1054e-02, -2.0684e-02]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[ 6.1315e-03, -3.3316e-02,  1.2067e-02],\n",
      "          [ 1.8067e-02, -8.1441e-03, -3.4957e-02],\n",
      "          [-7.5747e-02,  2.4396e-02,  6.7319e-02]],\n",
      "\n",
      "         [[ 3.5480e-02,  5.3239e-03, -3.4178e-03],\n",
      "          [-1.0962e-02, -2.9270e-02,  6.4376e-03],\n",
      "          [-1.1102e-02,  3.9663e-02, -1.3887e-02]],\n",
      "\n",
      "         [[ 1.2791e-02, -1.3587e-02,  2.8428e-02],\n",
      "          [ 4.1935e-02,  2.8283e-02, -3.9836e-03],\n",
      "          [ 3.6684e-02, -3.3586e-02, -5.0523e-02]]],\n",
      "\n",
      "\n",
      "        [[[-2.9560e-02, -1.2355e-02,  2.5626e-02],\n",
      "          [-8.5089e-02,  1.2820e-02, -1.2871e-02],\n",
      "          [-4.0237e-02,  9.5408e-03,  4.8999e-02]],\n",
      "\n",
      "         [[-1.7742e-03, -2.5095e-02,  3.0925e-02],\n",
      "          [-1.5568e-02, -1.7270e-03, -6.0209e-03],\n",
      "          [-1.2972e-02, -1.4721e-02, -1.7687e-02]],\n",
      "\n",
      "         [[ 3.2964e-02,  6.6691e-02, -2.6126e-02],\n",
      "          [-3.7178e-02,  3.1924e-02,  6.6067e-02],\n",
      "          [-2.6409e-02,  2.0674e-02,  2.8401e-02]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[-3.1226e-03,  2.4785e-02,  1.8434e-03],\n",
      "          [ 5.8213e-02,  1.6818e-02, -4.1871e-02],\n",
      "          [-7.8831e-03, -1.9349e-02, -5.1468e-02]],\n",
      "\n",
      "         [[-5.1162e-02, -1.8593e-02,  1.7993e-02],\n",
      "          [ 3.3960e-02, -3.3237e-02, -2.0055e-03],\n",
      "          [ 1.9652e-02, -5.6526e-03, -2.4328e-02]],\n",
      "\n",
      "         [[ 1.9455e-02, -5.8356e-03,  6.6340e-02],\n",
      "          [ 8.4906e-03,  1.7928e-02, -7.8184e-03],\n",
      "          [-3.7120e-02,  3.1860e-02,  1.1120e-03]]],\n",
      "\n",
      "\n",
      "        [[[-1.1755e-03,  3.3303e-03, -1.1175e-02],\n",
      "          [ 6.1783e-03,  2.0851e-02,  4.8620e-02],\n",
      "          [ 1.1369e-02,  4.9440e-02, -1.1486e-02]],\n",
      "\n",
      "         [[ 4.2271e-02, -2.8632e-03,  1.0149e-02],\n",
      "          [ 1.9648e-02, -6.8255e-03,  3.6946e-02],\n",
      "          [ 4.2992e-02, -1.6996e-02, -5.6512e-02]],\n",
      "\n",
      "         [[ 3.5154e-03, -1.1589e-02, -3.0525e-02],\n",
      "          [-2.6996e-02,  2.4046e-02, -7.2063e-02],\n",
      "          [ 3.4527e-02,  6.5651e-03,  4.6969e-02]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[-2.6787e-02, -6.9887e-03, -5.3458e-04],\n",
      "          [ 3.9635e-03, -2.1424e-02, -3.4647e-02],\n",
      "          [-1.0625e-02,  8.4828e-03,  3.5090e-02]],\n",
      "\n",
      "         [[ 2.5554e-02, -1.9092e-02, -4.2964e-03],\n",
      "          [-3.9753e-03,  6.3134e-04, -1.0042e-03],\n",
      "          [-4.5226e-03, -6.7261e-02, -6.9722e-03]],\n",
      "\n",
      "         [[ 9.6707e-04, -2.6268e-02,  2.0421e-02],\n",
      "          [ 3.1357e-02, -4.9349e-03,  4.7038e-03],\n",
      "          [-1.9814e-02, -4.4514e-02, -2.7878e-03]]],\n",
      "\n",
      "\n",
      "        ...,\n",
      "\n",
      "\n",
      "        [[[-5.7016e-03, -1.0024e-02, -3.0865e-02],\n",
      "          [ 7.1743e-03,  3.3725e-03, -2.6791e-02],\n",
      "          [-4.5346e-02, -7.4096e-02, -6.1102e-02]],\n",
      "\n",
      "         [[ 3.5054e-02, -2.5327e-02,  1.8057e-02],\n",
      "          [-1.6627e-02, -4.4302e-02, -7.6995e-03],\n",
      "          [ 4.4458e-02, -2.1951e-02,  2.5266e-03]],\n",
      "\n",
      "         [[-1.4606e-02,  5.0053e-02,  2.3601e-02],\n",
      "          [ 5.3266e-02, -3.8150e-02, -1.6193e-02],\n",
      "          [ 6.2072e-03, -7.4181e-03, -8.7824e-02]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[ 1.8299e-02, -1.4417e-02, -9.2338e-03],\n",
      "          [ 2.6756e-02,  1.5428e-03, -2.9092e-02],\n",
      "          [-4.0638e-02, -5.7148e-02,  1.2168e-02]],\n",
      "\n",
      "         [[-4.2233e-02, -1.5242e-02,  5.3666e-02],\n",
      "          [-3.7176e-02,  6.3818e-03,  1.4499e-02],\n",
      "          [ 3.1722e-02, -1.4641e-02,  4.8281e-02]],\n",
      "\n",
      "         [[-2.0991e-02, -1.2628e-02,  1.5928e-02],\n",
      "          [-1.5809e-02, -5.3705e-03, -4.5449e-03],\n",
      "          [ 6.2169e-03, -1.6663e-02, -4.6904e-03]]],\n",
      "\n",
      "\n",
      "        [[[-2.8790e-02,  4.7984e-02,  5.3953e-02],\n",
      "          [-5.9179e-03,  5.0935e-04, -4.4104e-02],\n",
      "          [ 1.8085e-03,  2.7696e-02,  4.7109e-03]],\n",
      "\n",
      "         [[ 2.0602e-02, -1.6338e-02,  1.4050e-03],\n",
      "          [-2.0060e-02,  3.0551e-02, -4.6878e-02],\n",
      "          [ 7.0503e-03,  3.5271e-02,  4.6735e-02]],\n",
      "\n",
      "         [[ 2.2374e-02,  4.3981e-03, -3.4599e-04],\n",
      "          [ 1.2414e-04, -1.6944e-04, -2.2346e-02],\n",
      "          [-4.3945e-02,  3.1856e-03,  3.7408e-02]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[ 3.7559e-02,  1.6463e-02,  1.1845e-03],\n",
      "          [-2.4918e-02, -9.2356e-03,  1.1169e-03],\n",
      "          [ 3.2455e-02,  3.3842e-02,  3.4772e-02]],\n",
      "\n",
      "         [[-8.6709e-03,  2.1854e-02, -5.0379e-03],\n",
      "          [ 1.8958e-02, -6.6191e-02,  1.8466e-02],\n",
      "          [-2.3984e-02, -2.0534e-02, -2.3669e-02]],\n",
      "\n",
      "         [[-1.8693e-02, -2.7298e-02, -3.1339e-02],\n",
      "          [ 8.4375e-03,  1.0751e-02,  4.9196e-02],\n",
      "          [ 3.5648e-02,  5.2104e-03, -1.4531e-02]]],\n",
      "\n",
      "\n",
      "        [[[ 4.0070e-02,  2.8240e-02,  1.8204e-02],\n",
      "          [ 3.9746e-02, -8.7449e-03,  1.2643e-02],\n",
      "          [ 2.3827e-02, -4.7841e-02, -7.1128e-03]],\n",
      "\n",
      "         [[ 5.8574e-02,  5.3519e-02,  1.7564e-02],\n",
      "          [ 2.7670e-03, -7.2087e-03, -6.0735e-02],\n",
      "          [-2.1270e-02, -2.3024e-02,  2.8618e-03]],\n",
      "\n",
      "         [[-4.3464e-02,  1.4467e-02,  4.2954e-02],\n",
      "          [ 3.2558e-02, -4.6429e-03, -4.1515e-03],\n",
      "          [-7.9827e-03,  3.2641e-02, -2.7388e-03]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[ 3.1216e-02, -3.0592e-02,  1.3357e-02],\n",
      "          [-7.4518e-03, -2.9667e-03, -4.1802e-02],\n",
      "          [-1.0368e-02, -1.8845e-02, -4.7453e-02]],\n",
      "\n",
      "         [[ 8.7053e-03,  2.1853e-02,  1.1272e-03],\n",
      "          [ 1.0699e-02, -5.9382e-02,  2.0675e-02],\n",
      "          [ 2.0219e-02, -3.0639e-02, -2.4966e-02]],\n",
      "\n",
      "         [[-1.0548e-02,  6.5719e-03, -3.0514e-02],\n",
      "          [-1.3737e-02, -3.7613e-03,  9.9114e-03],\n",
      "          [-7.5374e-03, -2.1278e-02,  3.7387e-02]]]], requires_grad=True)\n",
      "Parameter containing:\n",
      "tensor([1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1.], requires_grad=True)\n",
      "Parameter containing:\n",
      "tensor([0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.],\n",
      "       requires_grad=True)\n",
      "Parameter containing:\n",
      "tensor([[[[ 0.0114]],\n",
      "\n",
      "         [[ 0.0127]],\n",
      "\n",
      "         [[-0.0250]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[ 0.0106]],\n",
      "\n",
      "         [[ 0.0200]],\n",
      "\n",
      "         [[ 0.0081]]],\n",
      "\n",
      "\n",
      "        [[[ 0.0338]],\n",
      "\n",
      "         [[-0.0884]],\n",
      "\n",
      "         [[ 0.0345]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[ 0.0130]],\n",
      "\n",
      "         [[ 0.0557]],\n",
      "\n",
      "         [[-0.0118]]],\n",
      "\n",
      "\n",
      "        [[[-0.0545]],\n",
      "\n",
      "         [[ 0.0833]],\n",
      "\n",
      "         [[-0.0342]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[-0.0129]],\n",
      "\n",
      "         [[-0.0309]],\n",
      "\n",
      "         [[-0.0083]]],\n",
      "\n",
      "\n",
      "        ...,\n",
      "\n",
      "\n",
      "        [[[-0.0195]],\n",
      "\n",
      "         [[-0.0462]],\n",
      "\n",
      "         [[-0.0298]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[ 0.0101]],\n",
      "\n",
      "         [[ 0.0332]],\n",
      "\n",
      "         [[-0.0390]]],\n",
      "\n",
      "\n",
      "        [[[-0.0101]],\n",
      "\n",
      "         [[ 0.0359]],\n",
      "\n",
      "         [[-0.0072]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[ 0.0112]],\n",
      "\n",
      "         [[-0.0904]],\n",
      "\n",
      "         [[-0.0012]]],\n",
      "\n",
      "\n",
      "        [[[ 0.0945]],\n",
      "\n",
      "         [[-0.0071]],\n",
      "\n",
      "         [[ 0.0476]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[-0.0436]],\n",
      "\n",
      "         [[-0.0582]],\n",
      "\n",
      "         [[ 0.0182]]]], requires_grad=True)\n",
      "Parameter containing:\n",
      "tensor([1., 1., 1.,  ..., 1., 1., 1.], requires_grad=True)\n",
      "Parameter containing:\n",
      "tensor([0., 0., 0.,  ..., 0., 0., 0.], requires_grad=True)\n",
      "Parameter containing:\n",
      "tensor([[[[ 0.0101]],\n",
      "\n",
      "         [[ 0.0926]],\n",
      "\n",
      "         [[-0.0762]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[-0.0636]],\n",
      "\n",
      "         [[-0.0537]],\n",
      "\n",
      "         [[ 0.0376]]],\n",
      "\n",
      "\n",
      "        [[[-0.0825]],\n",
      "\n",
      "         [[-0.0570]],\n",
      "\n",
      "         [[-0.0695]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[-0.0414]],\n",
      "\n",
      "         [[ 0.0142]],\n",
      "\n",
      "         [[ 0.0723]]],\n",
      "\n",
      "\n",
      "        [[[ 0.0872]],\n",
      "\n",
      "         [[ 0.0058]],\n",
      "\n",
      "         [[-0.0541]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[-0.0293]],\n",
      "\n",
      "         [[ 0.0996]],\n",
      "\n",
      "         [[-0.1061]]],\n",
      "\n",
      "\n",
      "        ...,\n",
      "\n",
      "\n",
      "        [[[-0.1076]],\n",
      "\n",
      "         [[ 0.1782]],\n",
      "\n",
      "         [[-0.1443]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[-0.0440]],\n",
      "\n",
      "         [[-0.0170]],\n",
      "\n",
      "         [[ 0.0398]]],\n",
      "\n",
      "\n",
      "        [[[ 0.0099]],\n",
      "\n",
      "         [[ 0.0947]],\n",
      "\n",
      "         [[-0.0005]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[ 0.0320]],\n",
      "\n",
      "         [[-0.0793]],\n",
      "\n",
      "         [[ 0.1774]]],\n",
      "\n",
      "\n",
      "        [[[ 0.0256]],\n",
      "\n",
      "         [[-0.0775]],\n",
      "\n",
      "         [[ 0.0499]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[ 0.0205]],\n",
      "\n",
      "         [[-0.1277]],\n",
      "\n",
      "         [[ 0.0255]]]], requires_grad=True)\n",
      "Parameter containing:\n",
      "tensor([1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1.], requires_grad=True)\n",
      "Parameter containing:\n",
      "tensor([0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.],\n",
      "       requires_grad=True)\n",
      "Parameter containing:\n",
      "tensor([[[[ 0.0127, -0.0314,  0.0635],\n",
      "          [ 0.0490, -0.0304,  0.0584],\n",
      "          [ 0.0236, -0.0015, -0.0085]],\n",
      "\n",
      "         [[ 0.0060, -0.0022, -0.0363],\n",
      "          [ 0.0032,  0.0273, -0.0016],\n",
      "          [ 0.0451, -0.0107,  0.0227]],\n",
      "\n",
      "         [[ 0.0331,  0.0046, -0.0080],\n",
      "          [ 0.0236, -0.0173,  0.0311],\n",
      "          [ 0.0114, -0.0202,  0.0278]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[ 0.0277,  0.0313, -0.0039],\n",
      "          [-0.0143, -0.0141, -0.0349],\n",
      "          [-0.0563, -0.0116, -0.0074]],\n",
      "\n",
      "         [[ 0.0373,  0.0403,  0.0319],\n",
      "          [-0.0347,  0.0087, -0.0344],\n",
      "          [ 0.0122, -0.0351,  0.0153]],\n",
      "\n",
      "         [[ 0.0718,  0.0444, -0.0023],\n",
      "          [ 0.0436,  0.0056,  0.0179],\n",
      "          [-0.0098, -0.0429,  0.0195]]],\n",
      "\n",
      "\n",
      "        [[[-0.0454, -0.0076, -0.0340],\n",
      "          [-0.0133,  0.0157,  0.0458],\n",
      "          [-0.0068,  0.0060,  0.0225]],\n",
      "\n",
      "         [[ 0.0206, -0.0036, -0.0553],\n",
      "          [-0.0028, -0.0080, -0.0102],\n",
      "          [-0.0019, -0.0041, -0.0403]],\n",
      "\n",
      "         [[ 0.0407, -0.0735, -0.0039],\n",
      "          [ 0.0070,  0.0411, -0.0105],\n",
      "          [-0.0090,  0.0680,  0.0030]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[-0.0087, -0.0153, -0.0548],\n",
      "          [-0.0140,  0.0415,  0.0452],\n",
      "          [-0.0173,  0.0079,  0.0104]],\n",
      "\n",
      "         [[-0.0493,  0.0303,  0.0406],\n",
      "          [ 0.0117,  0.0253, -0.0027],\n",
      "          [-0.0251, -0.0255,  0.0449]],\n",
      "\n",
      "         [[ 0.0351,  0.0491, -0.0548],\n",
      "          [ 0.0056, -0.0129,  0.0134],\n",
      "          [ 0.0149,  0.0409, -0.0022]]],\n",
      "\n",
      "\n",
      "        [[[-0.0164, -0.0034, -0.0296],\n",
      "          [ 0.0105,  0.0720,  0.0092],\n",
      "          [ 0.0019,  0.0139,  0.0953]],\n",
      "\n",
      "         [[ 0.0257, -0.0036,  0.0152],\n",
      "          [ 0.0232,  0.0162,  0.0181],\n",
      "          [-0.0431, -0.0608, -0.0379]],\n",
      "\n",
      "         [[-0.0055, -0.0088,  0.0496],\n",
      "          [ 0.0154, -0.0074,  0.0046],\n",
      "          [ 0.0650,  0.0020,  0.0422]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[ 0.0220,  0.0512,  0.0030],\n",
      "          [ 0.0215,  0.0760,  0.0354],\n",
      "          [-0.0274, -0.0036,  0.0373]],\n",
      "\n",
      "         [[-0.0041, -0.0179,  0.0041],\n",
      "          [ 0.0085, -0.0054, -0.0263],\n",
      "          [ 0.0129,  0.0322,  0.0078]],\n",
      "\n",
      "         [[-0.0286, -0.0281,  0.0001],\n",
      "          [-0.0227,  0.0082, -0.0229],\n",
      "          [-0.0053, -0.0026,  0.0402]]],\n",
      "\n",
      "\n",
      "        ...,\n",
      "\n",
      "\n",
      "        [[[ 0.0454,  0.0593, -0.0201],\n",
      "          [ 0.0098, -0.0059, -0.0321],\n",
      "          [ 0.0027, -0.0422, -0.0043]],\n",
      "\n",
      "         [[-0.0256,  0.0265, -0.0229],\n",
      "          [-0.0049,  0.0417,  0.0065],\n",
      "          [-0.0353,  0.0165,  0.0020]],\n",
      "\n",
      "         [[ 0.0501,  0.0081,  0.0099],\n",
      "          [ 0.0287, -0.0463,  0.0330],\n",
      "          [-0.0126,  0.0156,  0.0343]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[-0.0041, -0.0342, -0.0016],\n",
      "          [-0.0323, -0.0067,  0.0309],\n",
      "          [ 0.0099, -0.0445, -0.0384]],\n",
      "\n",
      "         [[ 0.0044, -0.0139,  0.0456],\n",
      "          [-0.0121, -0.0286, -0.0197],\n",
      "          [ 0.0206,  0.0084,  0.0135]],\n",
      "\n",
      "         [[-0.0066, -0.0042, -0.0169],\n",
      "          [ 0.0274,  0.0360, -0.0097],\n",
      "          [-0.0197, -0.0030, -0.0243]]],\n",
      "\n",
      "\n",
      "        [[[-0.0498,  0.0015, -0.0151],\n",
      "          [-0.0063,  0.0287, -0.0470],\n",
      "          [-0.0201, -0.0043,  0.0029]],\n",
      "\n",
      "         [[-0.0178,  0.0653,  0.0202],\n",
      "          [-0.0301,  0.0590, -0.0013],\n",
      "          [-0.0370, -0.0203, -0.0167]],\n",
      "\n",
      "         [[-0.0211, -0.0621,  0.0381],\n",
      "          [-0.0263, -0.0162, -0.0009],\n",
      "          [-0.0562, -0.0390,  0.0178]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[-0.0236,  0.0157, -0.0332],\n",
      "          [-0.0189,  0.0022, -0.0442],\n",
      "          [ 0.0035,  0.0427, -0.0272]],\n",
      "\n",
      "         [[ 0.0382,  0.0341, -0.0184],\n",
      "          [-0.0529,  0.0717,  0.0023],\n",
      "          [-0.0305, -0.0013, -0.0732]],\n",
      "\n",
      "         [[ 0.0126, -0.0120,  0.0142],\n",
      "          [ 0.0065,  0.0153,  0.0386],\n",
      "          [-0.0069,  0.0267,  0.0389]]],\n",
      "\n",
      "\n",
      "        [[[-0.0413, -0.0124, -0.0451],\n",
      "          [-0.0200, -0.0624,  0.0343],\n",
      "          [ 0.0021,  0.0212, -0.0062]],\n",
      "\n",
      "         [[ 0.0147, -0.0093,  0.0032],\n",
      "          [-0.0305, -0.0015,  0.0721],\n",
      "          [ 0.0389, -0.0028, -0.0009]],\n",
      "\n",
      "         [[ 0.0040, -0.0106, -0.0065],\n",
      "          [-0.0631,  0.0183,  0.0150],\n",
      "          [ 0.0219, -0.0256, -0.0667]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[-0.0009, -0.0096, -0.0195],\n",
      "          [ 0.0503, -0.0603, -0.0156],\n",
      "          [ 0.0122,  0.0260, -0.0192]],\n",
      "\n",
      "         [[ 0.0114,  0.0132,  0.0175],\n",
      "          [-0.0064, -0.0145, -0.0178],\n",
      "          [-0.0104, -0.0113,  0.0255]],\n",
      "\n",
      "         [[ 0.0054,  0.0060, -0.0355],\n",
      "          [ 0.0253,  0.0381,  0.0133],\n",
      "          [-0.0381,  0.0004,  0.0106]]]], requires_grad=True)\n",
      "Parameter containing:\n",
      "tensor([1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1.], requires_grad=True)\n",
      "Parameter containing:\n",
      "tensor([0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.],\n",
      "       requires_grad=True)\n",
      "Parameter containing:\n",
      "tensor([[[[ 0.0446]],\n",
      "\n",
      "         [[-0.0252]],\n",
      "\n",
      "         [[ 0.0222]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[ 0.0387]],\n",
      "\n",
      "         [[-0.0377]],\n",
      "\n",
      "         [[-0.0120]]],\n",
      "\n",
      "\n",
      "        [[[-0.0829]],\n",
      "\n",
      "         [[ 0.0126]],\n",
      "\n",
      "         [[ 0.0050]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[ 0.0317]],\n",
      "\n",
      "         [[-0.0233]],\n",
      "\n",
      "         [[-0.0513]]],\n",
      "\n",
      "\n",
      "        [[[-0.0102]],\n",
      "\n",
      "         [[-0.0701]],\n",
      "\n",
      "         [[-0.0239]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[ 0.0634]],\n",
      "\n",
      "         [[ 0.0649]],\n",
      "\n",
      "         [[-0.0087]]],\n",
      "\n",
      "\n",
      "        ...,\n",
      "\n",
      "\n",
      "        [[[-0.0177]],\n",
      "\n",
      "         [[ 0.0146]],\n",
      "\n",
      "         [[ 0.0660]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[-0.0297]],\n",
      "\n",
      "         [[-0.0197]],\n",
      "\n",
      "         [[-0.0456]]],\n",
      "\n",
      "\n",
      "        [[[-0.0250]],\n",
      "\n",
      "         [[ 0.0133]],\n",
      "\n",
      "         [[-0.0006]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[ 0.0054]],\n",
      "\n",
      "         [[ 0.0150]],\n",
      "\n",
      "         [[-0.0294]]],\n",
      "\n",
      "\n",
      "        [[[-0.0260]],\n",
      "\n",
      "         [[ 0.0285]],\n",
      "\n",
      "         [[ 0.0116]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[-0.0766]],\n",
      "\n",
      "         [[-0.0171]],\n",
      "\n",
      "         [[-0.0482]]]], requires_grad=True)\n",
      "Parameter containing:\n",
      "tensor([1., 1., 1.,  ..., 1., 1., 1.], requires_grad=True)\n",
      "Parameter containing:\n",
      "tensor([0., 0., 0.,  ..., 0., 0., 0.], requires_grad=True)\n",
      "Parameter containing:\n",
      "tensor([[[[ 1.3915e-02]],\n",
      "\n",
      "         [[ 4.8186e-02]],\n",
      "\n",
      "         [[-6.1901e-05]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[-9.9178e-02]],\n",
      "\n",
      "         [[ 7.1702e-02]],\n",
      "\n",
      "         [[-2.5747e-02]]],\n",
      "\n",
      "\n",
      "        [[[-4.1799e-02]],\n",
      "\n",
      "         [[ 2.1064e-01]],\n",
      "\n",
      "         [[-5.2455e-02]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[ 3.1093e-02]],\n",
      "\n",
      "         [[-2.5428e-02]],\n",
      "\n",
      "         [[ 8.1152e-02]]],\n",
      "\n",
      "\n",
      "        [[[-1.6662e-02]],\n",
      "\n",
      "         [[ 2.2606e-03]],\n",
      "\n",
      "         [[-1.7032e-02]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[ 1.4777e-02]],\n",
      "\n",
      "         [[ 3.7585e-02]],\n",
      "\n",
      "         [[-7.6076e-02]]],\n",
      "\n",
      "\n",
      "        ...,\n",
      "\n",
      "\n",
      "        [[[-7.6972e-02]],\n",
      "\n",
      "         [[-3.7497e-02]],\n",
      "\n",
      "         [[-9.0635e-02]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[-8.2110e-02]],\n",
      "\n",
      "         [[-1.0463e-01]],\n",
      "\n",
      "         [[ 8.7292e-02]]],\n",
      "\n",
      "\n",
      "        [[[-8.2730e-02]],\n",
      "\n",
      "         [[-3.7287e-02]],\n",
      "\n",
      "         [[ 1.0541e-01]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[ 9.0350e-02]],\n",
      "\n",
      "         [[ 1.4535e-02]],\n",
      "\n",
      "         [[-9.4805e-02]]],\n",
      "\n",
      "\n",
      "        [[[ 3.1996e-02]],\n",
      "\n",
      "         [[ 1.4568e-01]],\n",
      "\n",
      "         [[ 2.8843e-02]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[ 1.4213e-01]],\n",
      "\n",
      "         [[-8.4905e-02]],\n",
      "\n",
      "         [[-4.6513e-02]]]], requires_grad=True)\n",
      "Parameter containing:\n",
      "tensor([1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1.], requires_grad=True)\n",
      "Parameter containing:\n",
      "tensor([0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.],\n",
      "       requires_grad=True)\n",
      "Parameter containing:\n",
      "tensor([[[[ 0.0499,  0.0636,  0.0057],\n",
      "          [-0.0050,  0.0134, -0.0440],\n",
      "          [ 0.0249, -0.0279,  0.0004]],\n",
      "\n",
      "         [[ 0.0151, -0.0206,  0.0194],\n",
      "          [ 0.0033,  0.0754, -0.0174],\n",
      "          [-0.0177,  0.0290, -0.0224]],\n",
      "\n",
      "         [[ 0.0389, -0.0134, -0.0348],\n",
      "          [-0.0191, -0.0344,  0.0124],\n",
      "          [ 0.0389,  0.0004,  0.0432]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[-0.0050, -0.0066, -0.0077],\n",
      "          [ 0.0454, -0.0431, -0.0433],\n",
      "          [-0.0215,  0.0159,  0.0142]],\n",
      "\n",
      "         [[ 0.0346,  0.0341, -0.0419],\n",
      "          [-0.0567, -0.0141, -0.0335],\n",
      "          [ 0.0228, -0.0759,  0.0216]],\n",
      "\n",
      "         [[-0.0184,  0.0032,  0.0063],\n",
      "          [ 0.0252, -0.0095, -0.0087],\n",
      "          [ 0.0150, -0.0143,  0.0132]]],\n",
      "\n",
      "\n",
      "        [[[-0.0057,  0.0335, -0.0138],\n",
      "          [ 0.0347, -0.0230,  0.0404],\n",
      "          [-0.0130, -0.0585,  0.0275]],\n",
      "\n",
      "         [[ 0.0018,  0.0015, -0.0814],\n",
      "          [-0.0168,  0.0576,  0.0438],\n",
      "          [-0.0259, -0.0152,  0.0014]],\n",
      "\n",
      "         [[ 0.0237, -0.0477, -0.0713],\n",
      "          [-0.0303,  0.0038, -0.0210],\n",
      "          [-0.0102, -0.0533,  0.0163]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[-0.0147,  0.0281,  0.0280],\n",
      "          [ 0.0377, -0.0113, -0.0150],\n",
      "          [-0.0079,  0.0509,  0.0182]],\n",
      "\n",
      "         [[-0.0292,  0.0535, -0.0066],\n",
      "          [-0.0029,  0.0106,  0.0160],\n",
      "          [ 0.0425,  0.0581,  0.0318]],\n",
      "\n",
      "         [[-0.0013, -0.0008, -0.0465],\n",
      "          [-0.0003, -0.0039, -0.0543],\n",
      "          [-0.0233, -0.0403, -0.0474]]],\n",
      "\n",
      "\n",
      "        [[[-0.0373,  0.0189, -0.0425],\n",
      "          [ 0.0640, -0.0364, -0.0096],\n",
      "          [ 0.0130,  0.0412,  0.0043]],\n",
      "\n",
      "         [[-0.0364, -0.0400,  0.0424],\n",
      "          [-0.0141,  0.0411, -0.0223],\n",
      "          [-0.0440,  0.0241, -0.0038]],\n",
      "\n",
      "         [[ 0.0116,  0.0281, -0.0158],\n",
      "          [ 0.0316, -0.0131,  0.0283],\n",
      "          [ 0.0152,  0.0096, -0.0743]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[-0.0363, -0.0227,  0.0254],\n",
      "          [ 0.0294,  0.0031,  0.0113],\n",
      "          [ 0.0240,  0.0013, -0.0160]],\n",
      "\n",
      "         [[-0.0051, -0.0117,  0.0382],\n",
      "          [ 0.0217,  0.0397,  0.0350],\n",
      "          [-0.0151, -0.0123,  0.0007]],\n",
      "\n",
      "         [[ 0.0138, -0.0242,  0.0172],\n",
      "          [-0.0114, -0.0060, -0.0031],\n",
      "          [ 0.0311,  0.0091,  0.0080]]],\n",
      "\n",
      "\n",
      "        ...,\n",
      "\n",
      "\n",
      "        [[[-0.0153, -0.0379,  0.0088],\n",
      "          [-0.0072,  0.0243,  0.0302],\n",
      "          [ 0.0436,  0.0080,  0.0001]],\n",
      "\n",
      "         [[-0.0293,  0.0053, -0.0215],\n",
      "          [-0.0473, -0.0306, -0.0483],\n",
      "          [ 0.0326, -0.0146,  0.0006]],\n",
      "\n",
      "         [[-0.0622,  0.0091, -0.0028],\n",
      "          [ 0.0457,  0.0067, -0.0392],\n",
      "          [ 0.0401,  0.0252,  0.0530]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[-0.0133,  0.0541, -0.0520],\n",
      "          [-0.0016,  0.0322, -0.0455],\n",
      "          [ 0.0378, -0.0172,  0.0288]],\n",
      "\n",
      "         [[ 0.0206,  0.0113, -0.0237],\n",
      "          [ 0.0079, -0.0524,  0.0009],\n",
      "          [ 0.0066,  0.0064,  0.0208]],\n",
      "\n",
      "         [[-0.0187,  0.0407,  0.0429],\n",
      "          [ 0.0391, -0.0583, -0.0274],\n",
      "          [-0.0753, -0.0022,  0.0399]]],\n",
      "\n",
      "\n",
      "        [[[ 0.0025, -0.0280,  0.0028],\n",
      "          [-0.0421,  0.0177,  0.0303],\n",
      "          [-0.0203, -0.0389,  0.0165]],\n",
      "\n",
      "         [[ 0.0087, -0.0154,  0.0171],\n",
      "          [-0.0192,  0.0107,  0.0201],\n",
      "          [-0.0243,  0.0167,  0.0289]],\n",
      "\n",
      "         [[ 0.0221, -0.0194, -0.0387],\n",
      "          [-0.0796,  0.0011, -0.0108],\n",
      "          [ 0.0231, -0.0028, -0.0106]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[-0.0194, -0.0315,  0.0007],\n",
      "          [ 0.0200,  0.0257,  0.0054],\n",
      "          [ 0.0150, -0.0386, -0.0087]],\n",
      "\n",
      "         [[-0.0437, -0.0082, -0.0082],\n",
      "          [-0.0264,  0.0006,  0.0474],\n",
      "          [ 0.0264,  0.0120,  0.0489]],\n",
      "\n",
      "         [[ 0.0167, -0.0165,  0.0287],\n",
      "          [ 0.0276, -0.0202,  0.0280],\n",
      "          [-0.0368, -0.0034, -0.0263]]],\n",
      "\n",
      "\n",
      "        [[[ 0.0216, -0.0239,  0.0362],\n",
      "          [ 0.0234,  0.0525,  0.0320],\n",
      "          [ 0.0058,  0.0312, -0.0779]],\n",
      "\n",
      "         [[ 0.0072, -0.0043,  0.0658],\n",
      "          [ 0.0077, -0.0070, -0.0323],\n",
      "          [ 0.0181,  0.0121,  0.0625]],\n",
      "\n",
      "         [[ 0.0123,  0.0197, -0.0487],\n",
      "          [ 0.0678,  0.0439,  0.0223],\n",
      "          [-0.0133, -0.0706, -0.0399]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[ 0.0070,  0.0195,  0.0209],\n",
      "          [-0.0455, -0.0092,  0.0184],\n",
      "          [ 0.0082, -0.0265, -0.0188]],\n",
      "\n",
      "         [[-0.0083,  0.0466, -0.0401],\n",
      "          [ 0.0046,  0.0230,  0.0095],\n",
      "          [-0.0010, -0.0234,  0.0165]],\n",
      "\n",
      "         [[ 0.0024, -0.0063, -0.0233],\n",
      "          [ 0.0281, -0.0131, -0.0390],\n",
      "          [ 0.0036, -0.0115, -0.0549]]]], requires_grad=True)\n",
      "Parameter containing:\n",
      "tensor([1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1.], requires_grad=True)\n",
      "Parameter containing:\n",
      "tensor([0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.],\n",
      "       requires_grad=True)\n",
      "Parameter containing:\n",
      "tensor([[[[-0.0054]],\n",
      "\n",
      "         [[-0.0655]],\n",
      "\n",
      "         [[-0.0593]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[ 0.0101]],\n",
      "\n",
      "         [[ 0.0029]],\n",
      "\n",
      "         [[ 0.0086]]],\n",
      "\n",
      "\n",
      "        [[[-0.0134]],\n",
      "\n",
      "         [[ 0.0421]],\n",
      "\n",
      "         [[-0.0480]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[ 0.0526]],\n",
      "\n",
      "         [[-0.0210]],\n",
      "\n",
      "         [[ 0.0003]]],\n",
      "\n",
      "\n",
      "        [[[-0.0283]],\n",
      "\n",
      "         [[-0.0012]],\n",
      "\n",
      "         [[-0.0756]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[ 0.0400]],\n",
      "\n",
      "         [[-0.0454]],\n",
      "\n",
      "         [[-0.1236]]],\n",
      "\n",
      "\n",
      "        ...,\n",
      "\n",
      "\n",
      "        [[[-0.0064]],\n",
      "\n",
      "         [[-0.0306]],\n",
      "\n",
      "         [[ 0.0347]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[ 0.0481]],\n",
      "\n",
      "         [[ 0.0741]],\n",
      "\n",
      "         [[-0.0228]]],\n",
      "\n",
      "\n",
      "        [[[-0.0329]],\n",
      "\n",
      "         [[ 0.0039]],\n",
      "\n",
      "         [[ 0.0491]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[ 0.0480]],\n",
      "\n",
      "         [[-0.0252]],\n",
      "\n",
      "         [[ 0.0314]]],\n",
      "\n",
      "\n",
      "        [[[-0.0357]],\n",
      "\n",
      "         [[ 0.0280]],\n",
      "\n",
      "         [[ 0.0171]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[-0.0467]],\n",
      "\n",
      "         [[-0.0369]],\n",
      "\n",
      "         [[-0.0675]]]], requires_grad=True)\n",
      "Parameter containing:\n",
      "tensor([1., 1., 1.,  ..., 1., 1., 1.], requires_grad=True)\n",
      "Parameter containing:\n",
      "tensor([0., 0., 0.,  ..., 0., 0., 0.], requires_grad=True)\n",
      "Parameter containing:\n",
      "tensor([[[[-0.0635]],\n",
      "\n",
      "         [[-0.0257]],\n",
      "\n",
      "         [[ 0.0053]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[ 0.0085]],\n",
      "\n",
      "         [[-0.0596]],\n",
      "\n",
      "         [[ 0.0647]]],\n",
      "\n",
      "\n",
      "        [[[-0.0441]],\n",
      "\n",
      "         [[-0.0433]],\n",
      "\n",
      "         [[-0.1286]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[-0.0642]],\n",
      "\n",
      "         [[ 0.0200]],\n",
      "\n",
      "         [[-0.0750]]],\n",
      "\n",
      "\n",
      "        [[[ 0.0004]],\n",
      "\n",
      "         [[ 0.1617]],\n",
      "\n",
      "         [[-0.0363]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[-0.0805]],\n",
      "\n",
      "         [[ 0.0520]],\n",
      "\n",
      "         [[-0.0356]]],\n",
      "\n",
      "\n",
      "        ...,\n",
      "\n",
      "\n",
      "        [[[ 0.0303]],\n",
      "\n",
      "         [[ 0.1702]],\n",
      "\n",
      "         [[ 0.0085]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[ 0.0806]],\n",
      "\n",
      "         [[-0.1225]],\n",
      "\n",
      "         [[-0.0080]]],\n",
      "\n",
      "\n",
      "        [[[-0.0693]],\n",
      "\n",
      "         [[ 0.0577]],\n",
      "\n",
      "         [[-0.0521]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[-0.0528]],\n",
      "\n",
      "         [[-0.0674]],\n",
      "\n",
      "         [[-0.0820]]],\n",
      "\n",
      "\n",
      "        [[[ 0.1878]],\n",
      "\n",
      "         [[ 0.0657]],\n",
      "\n",
      "         [[-0.0537]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[-0.0447]],\n",
      "\n",
      "         [[ 0.1442]],\n",
      "\n",
      "         [[-0.0918]]]], requires_grad=True)\n",
      "Parameter containing:\n",
      "tensor([1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1.], requires_grad=True)\n",
      "Parameter containing:\n",
      "tensor([0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.],\n",
      "       requires_grad=True)\n",
      "Parameter containing:\n",
      "tensor([[[[-0.0006, -0.0046,  0.0023],\n",
      "          [-0.0461,  0.0254,  0.0501],\n",
      "          [-0.0467,  0.0047, -0.0071]],\n",
      "\n",
      "         [[ 0.0166, -0.0419,  0.0264],\n",
      "          [-0.0489,  0.0314, -0.0273],\n",
      "          [-0.0529,  0.0699, -0.0429]],\n",
      "\n",
      "         [[ 0.0222, -0.0573, -0.0475],\n",
      "          [-0.0455, -0.0130, -0.0433],\n",
      "          [-0.0505,  0.0283,  0.0113]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[-0.0125,  0.0146, -0.0131],\n",
      "          [ 0.0199, -0.0038, -0.0044],\n",
      "          [-0.0339, -0.0258, -0.0423]],\n",
      "\n",
      "         [[-0.0015, -0.0048,  0.0149],\n",
      "          [ 0.0226,  0.0392, -0.0299],\n",
      "          [-0.0291,  0.0120, -0.0153]],\n",
      "\n",
      "         [[-0.0238, -0.0332, -0.0052],\n",
      "          [-0.0206, -0.0217, -0.0169],\n",
      "          [-0.0109,  0.0340, -0.0275]]],\n",
      "\n",
      "\n",
      "        [[[-0.0353,  0.0307, -0.0097],\n",
      "          [-0.0263,  0.0236,  0.0102],\n",
      "          [-0.0415,  0.0277, -0.0756]],\n",
      "\n",
      "         [[ 0.0362,  0.0159,  0.0038],\n",
      "          [-0.0007, -0.0589,  0.0066],\n",
      "          [-0.0801,  0.0419, -0.0512]],\n",
      "\n",
      "         [[-0.0108,  0.0141, -0.0183],\n",
      "          [-0.0171, -0.0354,  0.0068],\n",
      "          [ 0.0202, -0.0338, -0.0806]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[-0.0021,  0.0144, -0.0025],\n",
      "          [ 0.0225, -0.0030,  0.0460],\n",
      "          [-0.0283,  0.0084,  0.0115]],\n",
      "\n",
      "         [[-0.0269, -0.0048,  0.0293],\n",
      "          [-0.0077, -0.0095, -0.0014],\n",
      "          [-0.0091,  0.0140, -0.0122]],\n",
      "\n",
      "         [[-0.0464, -0.0003, -0.0571],\n",
      "          [-0.0233,  0.0129,  0.0446],\n",
      "          [ 0.0213,  0.0004,  0.0100]]],\n",
      "\n",
      "\n",
      "        [[[ 0.0053,  0.0222, -0.0343],\n",
      "          [-0.0250,  0.0046, -0.0228],\n",
      "          [ 0.0272, -0.0026,  0.0095]],\n",
      "\n",
      "         [[-0.0724,  0.0236,  0.0249],\n",
      "          [-0.0071, -0.0185,  0.0059],\n",
      "          [-0.0361, -0.0313,  0.0371]],\n",
      "\n",
      "         [[-0.0352,  0.0074,  0.0102],\n",
      "          [ 0.0052,  0.0066, -0.0025],\n",
      "          [ 0.0180, -0.0234,  0.0499]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[ 0.0504,  0.0186,  0.0600],\n",
      "          [ 0.0603,  0.0221,  0.0052],\n",
      "          [-0.0079, -0.0354, -0.0534]],\n",
      "\n",
      "         [[ 0.0294,  0.0182,  0.0352],\n",
      "          [ 0.0415, -0.0371,  0.0167],\n",
      "          [ 0.0391, -0.0369, -0.0288]],\n",
      "\n",
      "         [[ 0.0024, -0.0352, -0.0192],\n",
      "          [ 0.0884, -0.0257, -0.0273],\n",
      "          [-0.0138,  0.0241, -0.0261]]],\n",
      "\n",
      "\n",
      "        ...,\n",
      "\n",
      "\n",
      "        [[[ 0.0012, -0.0131, -0.0632],\n",
      "          [-0.0176, -0.0328,  0.0315],\n",
      "          [ 0.0516,  0.0025, -0.0050]],\n",
      "\n",
      "         [[ 0.0611,  0.0562,  0.0988],\n",
      "          [-0.0127,  0.0357, -0.0665],\n",
      "          [-0.0041,  0.0357, -0.0147]],\n",
      "\n",
      "         [[-0.0646, -0.0891,  0.0371],\n",
      "          [ 0.0069,  0.0211, -0.0134],\n",
      "          [-0.0716, -0.0590, -0.0514]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[ 0.0432,  0.0541,  0.0286],\n",
      "          [-0.0571, -0.0031, -0.0399],\n",
      "          [-0.0261,  0.0067, -0.0121]],\n",
      "\n",
      "         [[-0.0295,  0.0357,  0.0403],\n",
      "          [ 0.0062,  0.0138, -0.0195],\n",
      "          [ 0.0475, -0.0090, -0.0048]],\n",
      "\n",
      "         [[-0.0110,  0.0223,  0.0311],\n",
      "          [ 0.0067,  0.0009,  0.0268],\n",
      "          [ 0.0253, -0.0054, -0.0648]]],\n",
      "\n",
      "\n",
      "        [[[ 0.0155,  0.0163, -0.0573],\n",
      "          [-0.0084, -0.0034, -0.0296],\n",
      "          [ 0.0234, -0.0334,  0.0099]],\n",
      "\n",
      "         [[ 0.0489,  0.0150, -0.0299],\n",
      "          [ 0.0282,  0.0277, -0.0233],\n",
      "          [-0.0139,  0.0343,  0.0071]],\n",
      "\n",
      "         [[ 0.0525,  0.0084,  0.0160],\n",
      "          [-0.0580, -0.0475, -0.0506],\n",
      "          [ 0.0026, -0.0650,  0.0342]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[-0.0388, -0.0006, -0.0282],\n",
      "          [ 0.0241, -0.0028,  0.0114],\n",
      "          [ 0.0007, -0.0095, -0.0111]],\n",
      "\n",
      "         [[ 0.0139,  0.0051, -0.0075],\n",
      "          [-0.0277,  0.0163,  0.0148],\n",
      "          [ 0.0296, -0.0404,  0.0194]],\n",
      "\n",
      "         [[ 0.0115,  0.0159,  0.0154],\n",
      "          [ 0.0709,  0.0420,  0.0044],\n",
      "          [ 0.0027,  0.0146,  0.0015]]],\n",
      "\n",
      "\n",
      "        [[[ 0.0479, -0.0052, -0.0491],\n",
      "          [ 0.0460,  0.0387,  0.0240],\n",
      "          [ 0.0223,  0.0445,  0.0156]],\n",
      "\n",
      "         [[-0.0584,  0.0148, -0.0295],\n",
      "          [-0.0113,  0.0144, -0.0300],\n",
      "          [ 0.0160, -0.0263,  0.0468]],\n",
      "\n",
      "         [[ 0.0170, -0.0541, -0.0259],\n",
      "          [ 0.0138,  0.0282, -0.0908],\n",
      "          [-0.0029,  0.0273, -0.0634]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[-0.0057,  0.0178, -0.0079],\n",
      "          [ 0.0124,  0.0421, -0.0012],\n",
      "          [ 0.0039,  0.0107, -0.0178]],\n",
      "\n",
      "         [[-0.0185, -0.0203,  0.0414],\n",
      "          [-0.0013, -0.0205, -0.0083],\n",
      "          [-0.0299, -0.0675, -0.0189]],\n",
      "\n",
      "         [[ 0.0385, -0.0064, -0.0476],\n",
      "          [ 0.0003,  0.0074, -0.0165],\n",
      "          [-0.0397,  0.0278, -0.0394]]]], requires_grad=True)\n",
      "Parameter containing:\n",
      "tensor([1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1.], requires_grad=True)\n",
      "Parameter containing:\n",
      "tensor([0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.],\n",
      "       requires_grad=True)\n",
      "Parameter containing:\n",
      "tensor([[[[-0.1159]],\n",
      "\n",
      "         [[ 0.0418]],\n",
      "\n",
      "         [[-0.0034]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[-0.0123]],\n",
      "\n",
      "         [[-0.0119]],\n",
      "\n",
      "         [[ 0.0892]]],\n",
      "\n",
      "\n",
      "        [[[ 0.0571]],\n",
      "\n",
      "         [[ 0.0193]],\n",
      "\n",
      "         [[ 0.0165]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[ 0.0885]],\n",
      "\n",
      "         [[-0.0438]],\n",
      "\n",
      "         [[-0.0041]]],\n",
      "\n",
      "\n",
      "        [[[-0.0675]],\n",
      "\n",
      "         [[-0.0214]],\n",
      "\n",
      "         [[ 0.0018]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[ 0.0512]],\n",
      "\n",
      "         [[ 0.0084]],\n",
      "\n",
      "         [[-0.0161]]],\n",
      "\n",
      "\n",
      "        ...,\n",
      "\n",
      "\n",
      "        [[[-0.0438]],\n",
      "\n",
      "         [[-0.0062]],\n",
      "\n",
      "         [[ 0.0253]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[ 0.0174]],\n",
      "\n",
      "         [[ 0.0122]],\n",
      "\n",
      "         [[-0.0420]]],\n",
      "\n",
      "\n",
      "        [[[-0.0084]],\n",
      "\n",
      "         [[ 0.0158]],\n",
      "\n",
      "         [[-0.0657]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[ 0.0183]],\n",
      "\n",
      "         [[ 0.0816]],\n",
      "\n",
      "         [[ 0.0152]]],\n",
      "\n",
      "\n",
      "        [[[ 0.0053]],\n",
      "\n",
      "         [[-0.0403]],\n",
      "\n",
      "         [[ 0.0002]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[ 0.0570]],\n",
      "\n",
      "         [[ 0.0100]],\n",
      "\n",
      "         [[ 0.0252]]]], requires_grad=True)\n",
      "Parameter containing:\n",
      "tensor([1., 1., 1.,  ..., 1., 1., 1.], requires_grad=True)\n",
      "Parameter containing:\n",
      "tensor([0., 0., 0.,  ..., 0., 0., 0.], requires_grad=True)\n",
      "Parameter containing:\n",
      "tensor([[[[ 0.0407]],\n",
      "\n",
      "         [[ 0.0549]],\n",
      "\n",
      "         [[-0.0531]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[ 0.0420]],\n",
      "\n",
      "         [[-0.1109]],\n",
      "\n",
      "         [[-0.0231]]],\n",
      "\n",
      "\n",
      "        [[[-0.0575]],\n",
      "\n",
      "         [[-0.0815]],\n",
      "\n",
      "         [[ 0.1006]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[-0.0781]],\n",
      "\n",
      "         [[ 0.1166]],\n",
      "\n",
      "         [[-0.0250]]],\n",
      "\n",
      "\n",
      "        [[[-0.1686]],\n",
      "\n",
      "         [[ 0.0179]],\n",
      "\n",
      "         [[ 0.0206]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[ 0.0268]],\n",
      "\n",
      "         [[ 0.0770]],\n",
      "\n",
      "         [[-0.0270]]],\n",
      "\n",
      "\n",
      "        ...,\n",
      "\n",
      "\n",
      "        [[[-0.0374]],\n",
      "\n",
      "         [[ 0.0012]],\n",
      "\n",
      "         [[ 0.1230]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[-0.0277]],\n",
      "\n",
      "         [[-0.0846]],\n",
      "\n",
      "         [[-0.0347]]],\n",
      "\n",
      "\n",
      "        [[[ 0.0581]],\n",
      "\n",
      "         [[ 0.0261]],\n",
      "\n",
      "         [[-0.0645]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[ 0.0361]],\n",
      "\n",
      "         [[ 0.0288]],\n",
      "\n",
      "         [[ 0.0482]]],\n",
      "\n",
      "\n",
      "        [[[ 0.0065]],\n",
      "\n",
      "         [[ 0.0494]],\n",
      "\n",
      "         [[-0.0605]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[-0.0493]],\n",
      "\n",
      "         [[-0.0799]],\n",
      "\n",
      "         [[-0.0798]]]], requires_grad=True)\n",
      "Parameter containing:\n",
      "tensor([1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1.], requires_grad=True)\n",
      "Parameter containing:\n",
      "tensor([0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0.], requires_grad=True)\n",
      "Parameter containing:\n",
      "tensor([[[[-2.6982e-02,  1.4270e-02,  2.4922e-02],\n",
      "          [ 1.0436e-03,  7.3310e-03, -2.0533e-02],\n",
      "          [ 3.0630e-02,  5.9167e-03,  1.0117e-02]],\n",
      "\n",
      "         [[-1.3664e-02,  2.7720e-02,  1.2861e-02],\n",
      "          [-4.6483e-03,  2.7390e-02, -1.5463e-02],\n",
      "          [ 1.0378e-02, -1.2770e-02, -1.4012e-03]],\n",
      "\n",
      "         [[-1.6554e-02, -4.4300e-04, -1.5985e-02],\n",
      "          [ 2.0889e-02, -1.5297e-02,  2.2720e-03],\n",
      "          [-1.2916e-02,  8.8228e-04, -8.9750e-03]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[-2.7464e-03, -1.7754e-03,  2.0854e-02],\n",
      "          [-1.5691e-02,  2.9171e-03,  5.9086e-03],\n",
      "          [-1.8856e-02,  4.5786e-02,  2.3218e-03]],\n",
      "\n",
      "         [[-2.3035e-02, -1.9989e-02, -3.5553e-02],\n",
      "          [-1.8612e-02,  2.6263e-02,  3.9197e-02],\n",
      "          [ 6.0234e-03, -1.0416e-02, -2.9752e-02]],\n",
      "\n",
      "         [[-2.0947e-02, -1.3456e-02,  2.0420e-03],\n",
      "          [ 2.9126e-02, -1.1516e-03, -5.9499e-03],\n",
      "          [-2.8312e-03, -3.7552e-02, -6.0927e-03]]],\n",
      "\n",
      "\n",
      "        [[[ 2.8998e-02,  1.5860e-02,  2.6454e-02],\n",
      "          [ 1.0869e-03,  1.7863e-02, -4.6923e-03],\n",
      "          [-9.7959e-05, -1.0595e-02,  2.7951e-02]],\n",
      "\n",
      "         [[ 2.4390e-02,  3.7527e-02,  1.9862e-02],\n",
      "          [ 9.9602e-03,  6.6221e-03, -1.7601e-02],\n",
      "          [ 5.2182e-04, -1.1259e-02,  2.0617e-02]],\n",
      "\n",
      "         [[-3.8269e-02, -7.1198e-03,  1.2310e-02],\n",
      "          [-2.6586e-02,  1.0677e-02,  8.2741e-03],\n",
      "          [ 3.6866e-03,  1.1039e-02,  8.2098e-03]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[ 8.2525e-03, -2.0904e-02,  1.1392e-02],\n",
      "          [-1.1848e-02,  1.2441e-02,  1.7575e-02],\n",
      "          [-6.8085e-03,  1.7992e-02,  7.7995e-03]],\n",
      "\n",
      "         [[-1.8269e-02, -4.2542e-03, -1.6578e-02],\n",
      "          [ 2.1576e-02,  2.1793e-02,  4.1871e-02],\n",
      "          [-4.3252e-02,  1.2604e-02, -2.0865e-02]],\n",
      "\n",
      "         [[ 3.4212e-04, -1.6203e-02,  4.4753e-02],\n",
      "          [-3.0584e-02,  3.7214e-02,  9.0801e-04],\n",
      "          [ 1.4785e-02,  3.2295e-02,  4.4819e-02]]],\n",
      "\n",
      "\n",
      "        [[[ 2.1322e-02, -4.0906e-03,  1.8544e-02],\n",
      "          [ 1.6995e-03,  1.2371e-02,  1.8033e-02],\n",
      "          [ 3.8180e-02, -2.1112e-03,  6.9813e-03]],\n",
      "\n",
      "         [[ 9.9592e-03,  1.4085e-04, -1.9600e-03],\n",
      "          [-5.0280e-03,  2.9557e-02, -1.5787e-03],\n",
      "          [ 5.1541e-02, -1.1172e-02, -9.7371e-03]],\n",
      "\n",
      "         [[-1.2264e-02,  3.1971e-03,  5.2440e-03],\n",
      "          [-3.3682e-02,  4.3952e-02, -5.4633e-03],\n",
      "          [-1.2222e-02, -6.7004e-02, -1.1021e-03]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[ 2.9407e-02,  1.5797e-02,  1.2621e-02],\n",
      "          [-5.5734e-03,  1.1843e-02,  1.1211e-02],\n",
      "          [ 2.0752e-02,  1.2972e-03, -9.4396e-03]],\n",
      "\n",
      "         [[-1.4209e-02,  4.1008e-03, -1.0348e-02],\n",
      "          [-2.9894e-02, -5.7587e-03, -1.8731e-02],\n",
      "          [-4.1924e-03, -2.0011e-02,  6.6098e-02]],\n",
      "\n",
      "         [[-7.7652e-03,  3.6595e-02, -2.9373e-02],\n",
      "          [ 1.6721e-02, -3.5756e-02,  9.8369e-03],\n",
      "          [ 3.8369e-02, -2.5919e-02,  4.5546e-02]]],\n",
      "\n",
      "\n",
      "        ...,\n",
      "\n",
      "\n",
      "        [[[ 2.6789e-02,  3.9687e-02, -3.4896e-02],\n",
      "          [ 6.5210e-03, -1.5825e-02,  4.5894e-03],\n",
      "          [-1.7786e-02,  9.2954e-03, -3.3478e-03]],\n",
      "\n",
      "         [[-1.6326e-02,  1.6819e-02,  2.5296e-02],\n",
      "          [-1.9177e-02, -9.6165e-03, -5.3898e-03],\n",
      "          [ 2.8426e-02,  2.0093e-03, -8.0695e-03]],\n",
      "\n",
      "         [[-4.1930e-02, -5.4860e-02, -2.1843e-02],\n",
      "          [ 1.3167e-02, -6.0640e-03,  2.2907e-02],\n",
      "          [ 1.2363e-02,  4.0292e-03, -1.8874e-03]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[ 1.6629e-03, -1.8274e-02, -3.6568e-02],\n",
      "          [ 1.3097e-02, -2.2318e-02,  7.9740e-03],\n",
      "          [-2.1669e-02,  2.7981e-02,  1.6448e-04]],\n",
      "\n",
      "         [[-7.9189e-03, -1.1191e-02, -3.5689e-02],\n",
      "          [ 1.7935e-02, -2.6850e-03,  2.2724e-02],\n",
      "          [-2.5622e-02,  1.5494e-02,  1.6009e-03]],\n",
      "\n",
      "         [[-2.6373e-02, -2.4847e-03, -4.5960e-02],\n",
      "          [ 1.8680e-02,  1.7853e-02,  1.5235e-02],\n",
      "          [ 4.3105e-03, -6.5225e-03,  3.1987e-02]]],\n",
      "\n",
      "\n",
      "        [[[ 6.9631e-03,  1.9535e-02, -1.8558e-02],\n",
      "          [-6.3476e-03, -7.8952e-03, -4.9716e-03],\n",
      "          [-1.1908e-02,  1.8502e-02, -1.3209e-02]],\n",
      "\n",
      "         [[ 7.8293e-03, -2.0601e-02, -1.8570e-02],\n",
      "          [-1.6385e-03,  2.1136e-02,  2.7687e-02],\n",
      "          [ 3.6032e-02,  9.4445e-04, -2.1977e-02]],\n",
      "\n",
      "         [[-4.5723e-03, -6.2532e-03, -2.6111e-02],\n",
      "          [-1.4578e-02, -7.8708e-04,  7.3163e-03],\n",
      "          [-2.2712e-02, -8.7004e-03,  2.0867e-02]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[ 3.0654e-02, -5.4246e-03, -1.5859e-02],\n",
      "          [ 3.2323e-02,  3.6563e-02, -1.4114e-02],\n",
      "          [ 2.0439e-02, -7.5813e-03, -8.3500e-03]],\n",
      "\n",
      "         [[-4.7323e-04, -5.1440e-03, -3.4305e-03],\n",
      "          [-1.3498e-02,  1.5552e-02,  2.5491e-02],\n",
      "          [-1.4870e-02, -5.8159e-03,  2.0357e-02]],\n",
      "\n",
      "         [[-1.6097e-03,  9.3074e-03, -3.1624e-02],\n",
      "          [ 3.5176e-03,  3.5418e-02, -9.3193e-03],\n",
      "          [-2.3285e-02, -6.6319e-03, -7.6597e-03]]],\n",
      "\n",
      "\n",
      "        [[[ 1.4591e-02,  1.9932e-02, -9.4803e-03],\n",
      "          [ 1.3187e-02,  5.6773e-03, -3.3027e-03],\n",
      "          [-3.7447e-02,  1.0461e-02,  3.3384e-02]],\n",
      "\n",
      "         [[ 1.3599e-02,  1.3906e-03, -3.2672e-03],\n",
      "          [ 1.7547e-02,  9.5809e-03, -7.9815e-03],\n",
      "          [ 2.4098e-02,  2.5341e-02, -2.5127e-02]],\n",
      "\n",
      "         [[ 4.0306e-02,  4.1039e-02,  2.1363e-02],\n",
      "          [-9.5621e-03, -2.5184e-03,  2.6348e-02],\n",
      "          [-2.0169e-02, -2.4316e-03,  1.9893e-02]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[ 1.7585e-02,  1.7464e-02, -6.6566e-03],\n",
      "          [-5.7387e-03, -3.4996e-02, -3.1708e-02],\n",
      "          [-4.4592e-04,  3.7834e-02,  2.5028e-02]],\n",
      "\n",
      "         [[ 3.3249e-02,  8.1972e-03,  3.7148e-03],\n",
      "          [ 7.8644e-03, -2.2705e-02, -2.1323e-02],\n",
      "          [ 2.8436e-03, -2.8238e-03, -2.6557e-02]],\n",
      "\n",
      "         [[ 9.4224e-03, -1.3421e-02,  2.0971e-02],\n",
      "          [-2.1639e-03,  1.6529e-02, -3.1472e-02],\n",
      "          [ 3.9229e-02,  5.1921e-02,  2.3319e-03]]]], requires_grad=True)\n",
      "Parameter containing:\n",
      "tensor([1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1.], requires_grad=True)\n",
      "Parameter containing:\n",
      "tensor([0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0.], requires_grad=True)\n",
      "Parameter containing:\n",
      "tensor([[[[-0.0109]],\n",
      "\n",
      "         [[-0.0012]],\n",
      "\n",
      "         [[ 0.0126]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[ 0.0036]],\n",
      "\n",
      "         [[-0.0121]],\n",
      "\n",
      "         [[-0.0169]]],\n",
      "\n",
      "\n",
      "        [[[-0.0025]],\n",
      "\n",
      "         [[-0.0072]],\n",
      "\n",
      "         [[-0.0403]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[ 0.0315]],\n",
      "\n",
      "         [[-0.0494]],\n",
      "\n",
      "         [[ 0.0268]]],\n",
      "\n",
      "\n",
      "        [[[ 0.0069]],\n",
      "\n",
      "         [[ 0.0037]],\n",
      "\n",
      "         [[ 0.0472]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[ 0.0150]],\n",
      "\n",
      "         [[ 0.0238]],\n",
      "\n",
      "         [[ 0.0309]]],\n",
      "\n",
      "\n",
      "        ...,\n",
      "\n",
      "\n",
      "        [[[ 0.0132]],\n",
      "\n",
      "         [[ 0.0099]],\n",
      "\n",
      "         [[-0.0088]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[ 0.0394]],\n",
      "\n",
      "         [[-0.0194]],\n",
      "\n",
      "         [[-0.0223]]],\n",
      "\n",
      "\n",
      "        [[[ 0.0534]],\n",
      "\n",
      "         [[-0.0053]],\n",
      "\n",
      "         [[-0.0225]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[-0.0032]],\n",
      "\n",
      "         [[-0.0266]],\n",
      "\n",
      "         [[ 0.0235]]],\n",
      "\n",
      "\n",
      "        [[[-0.0076]],\n",
      "\n",
      "         [[ 0.0250]],\n",
      "\n",
      "         [[ 0.0230]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[ 0.0434]],\n",
      "\n",
      "         [[-0.0723]],\n",
      "\n",
      "         [[-0.0160]]]], requires_grad=True)\n",
      "Parameter containing:\n",
      "tensor([1., 1., 1.,  ..., 1., 1., 1.], requires_grad=True)\n",
      "Parameter containing:\n",
      "tensor([0., 0., 0.,  ..., 0., 0., 0.], requires_grad=True)\n",
      "Parameter containing:\n",
      "tensor([[[[-0.0358]],\n",
      "\n",
      "         [[ 0.0076]],\n",
      "\n",
      "         [[ 0.0331]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[ 0.0023]],\n",
      "\n",
      "         [[ 0.0301]],\n",
      "\n",
      "         [[-0.0201]]],\n",
      "\n",
      "\n",
      "        [[[-0.0185]],\n",
      "\n",
      "         [[ 0.0386]],\n",
      "\n",
      "         [[ 0.0033]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[-0.0131]],\n",
      "\n",
      "         [[ 0.0089]],\n",
      "\n",
      "         [[-0.0418]]],\n",
      "\n",
      "\n",
      "        [[[-0.0325]],\n",
      "\n",
      "         [[-0.0045]],\n",
      "\n",
      "         [[ 0.0328]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[-0.0302]],\n",
      "\n",
      "         [[ 0.0001]],\n",
      "\n",
      "         [[-0.0017]]],\n",
      "\n",
      "\n",
      "        ...,\n",
      "\n",
      "\n",
      "        [[[-0.0139]],\n",
      "\n",
      "         [[-0.0596]],\n",
      "\n",
      "         [[-0.0258]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[ 0.0096]],\n",
      "\n",
      "         [[-0.0071]],\n",
      "\n",
      "         [[ 0.0182]]],\n",
      "\n",
      "\n",
      "        [[[ 0.0233]],\n",
      "\n",
      "         [[ 0.0382]],\n",
      "\n",
      "         [[-0.0141]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[ 0.0211]],\n",
      "\n",
      "         [[-0.0022]],\n",
      "\n",
      "         [[ 0.0053]]],\n",
      "\n",
      "\n",
      "        [[[-0.0169]],\n",
      "\n",
      "         [[-0.0023]],\n",
      "\n",
      "         [[ 0.0247]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[ 0.0168]],\n",
      "\n",
      "         [[-0.0336]],\n",
      "\n",
      "         [[ 0.0132]]]], requires_grad=True)\n",
      "Parameter containing:\n",
      "tensor([1., 1., 1.,  ..., 1., 1., 1.], requires_grad=True)\n",
      "Parameter containing:\n",
      "tensor([0., 0., 0.,  ..., 0., 0., 0.], requires_grad=True)\n",
      "Parameter containing:\n",
      "tensor([[[[-0.0623]],\n",
      "\n",
      "         [[ 0.0013]],\n",
      "\n",
      "         [[-0.0509]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[-0.0050]],\n",
      "\n",
      "         [[-0.0937]],\n",
      "\n",
      "         [[-0.0028]]],\n",
      "\n",
      "\n",
      "        [[[ 0.1286]],\n",
      "\n",
      "         [[ 0.0446]],\n",
      "\n",
      "         [[-0.0011]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[-0.0396]],\n",
      "\n",
      "         [[-0.0106]],\n",
      "\n",
      "         [[ 0.0113]]],\n",
      "\n",
      "\n",
      "        [[[ 0.0002]],\n",
      "\n",
      "         [[ 0.1506]],\n",
      "\n",
      "         [[-0.0815]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[-0.0306]],\n",
      "\n",
      "         [[-0.0080]],\n",
      "\n",
      "         [[-0.0903]]],\n",
      "\n",
      "\n",
      "        ...,\n",
      "\n",
      "\n",
      "        [[[-0.0538]],\n",
      "\n",
      "         [[-0.0226]],\n",
      "\n",
      "         [[-0.1216]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[-0.0036]],\n",
      "\n",
      "         [[ 0.0698]],\n",
      "\n",
      "         [[-0.0591]]],\n",
      "\n",
      "\n",
      "        [[[-0.0241]],\n",
      "\n",
      "         [[-0.0004]],\n",
      "\n",
      "         [[ 0.0508]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[-0.0510]],\n",
      "\n",
      "         [[ 0.0991]],\n",
      "\n",
      "         [[ 0.0485]]],\n",
      "\n",
      "\n",
      "        [[[ 0.1194]],\n",
      "\n",
      "         [[-0.0428]],\n",
      "\n",
      "         [[ 0.0627]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[ 0.0194]],\n",
      "\n",
      "         [[-0.1170]],\n",
      "\n",
      "         [[-0.0713]]]], requires_grad=True)\n",
      "Parameter containing:\n",
      "tensor([1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1.], requires_grad=True)\n",
      "Parameter containing:\n",
      "tensor([0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0.], requires_grad=True)\n",
      "Parameter containing:\n",
      "tensor([[[[-0.0077,  0.0013,  0.0063],\n",
      "          [ 0.0201, -0.0332,  0.0115],\n",
      "          [-0.0343, -0.0284, -0.0005]],\n",
      "\n",
      "         [[-0.0015,  0.0096, -0.0068],\n",
      "          [ 0.0272,  0.0009,  0.0088],\n",
      "          [ 0.0200, -0.0167,  0.0102]],\n",
      "\n",
      "         [[ 0.0086,  0.0409,  0.0169],\n",
      "          [-0.0431, -0.0167,  0.0146],\n",
      "          [ 0.0052,  0.0155,  0.0318]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[ 0.0089,  0.0142,  0.0183],\n",
      "          [ 0.0303,  0.0203,  0.0060],\n",
      "          [ 0.0269, -0.0275,  0.0216]],\n",
      "\n",
      "         [[-0.0244, -0.0266, -0.0414],\n",
      "          [-0.0280,  0.0225, -0.0013],\n",
      "          [ 0.0316,  0.0311, -0.0301]],\n",
      "\n",
      "         [[ 0.0029,  0.0026,  0.0211],\n",
      "          [ 0.0212, -0.0500, -0.0073],\n",
      "          [-0.0099, -0.0375, -0.0057]]],\n",
      "\n",
      "\n",
      "        [[[ 0.0233, -0.0372,  0.0060],\n",
      "          [-0.0041,  0.0170,  0.0043],\n",
      "          [-0.0257, -0.0535, -0.0058]],\n",
      "\n",
      "         [[-0.0010,  0.0140, -0.0044],\n",
      "          [-0.0176,  0.0057, -0.0359],\n",
      "          [ 0.0148,  0.0013,  0.0221]],\n",
      "\n",
      "         [[-0.0294, -0.0112,  0.0031],\n",
      "          [-0.0206,  0.0167, -0.0127],\n",
      "          [-0.0127, -0.0289, -0.0592]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[-0.0138,  0.0129, -0.0165],\n",
      "          [-0.0024, -0.0275, -0.0080],\n",
      "          [-0.0325, -0.0219, -0.0206]],\n",
      "\n",
      "         [[-0.0083,  0.0258,  0.0104],\n",
      "          [ 0.0088, -0.0202, -0.0069],\n",
      "          [-0.0084, -0.0496, -0.0141]],\n",
      "\n",
      "         [[-0.0214,  0.0222,  0.0021],\n",
      "          [-0.0158,  0.0261,  0.0054],\n",
      "          [ 0.0049,  0.0153,  0.0139]]],\n",
      "\n",
      "\n",
      "        [[[ 0.0040,  0.0083, -0.0143],\n",
      "          [ 0.0236,  0.0341, -0.0129],\n",
      "          [ 0.0116,  0.0318, -0.0062]],\n",
      "\n",
      "         [[ 0.0145,  0.0366,  0.0143],\n",
      "          [ 0.0275,  0.0074, -0.0269],\n",
      "          [ 0.0011, -0.0153,  0.0231]],\n",
      "\n",
      "         [[ 0.0091,  0.0046,  0.0284],\n",
      "          [ 0.0161, -0.0517, -0.0100],\n",
      "          [-0.0023, -0.0329,  0.0274]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[ 0.0197, -0.0151,  0.0174],\n",
      "          [ 0.0065,  0.0234, -0.0404],\n",
      "          [-0.0112,  0.0247, -0.0077]],\n",
      "\n",
      "         [[ 0.0044, -0.0031, -0.0298],\n",
      "          [ 0.0101, -0.0193, -0.0024],\n",
      "          [-0.0095,  0.0094, -0.0285]],\n",
      "\n",
      "         [[-0.0022, -0.0367,  0.0254],\n",
      "          [-0.0018, -0.0098, -0.0236],\n",
      "          [-0.0289, -0.0181, -0.0043]]],\n",
      "\n",
      "\n",
      "        ...,\n",
      "\n",
      "\n",
      "        [[[-0.0071, -0.0084, -0.0120],\n",
      "          [-0.0371,  0.0180,  0.0465],\n",
      "          [-0.0251, -0.0244,  0.0335]],\n",
      "\n",
      "         [[-0.0078, -0.0255, -0.0174],\n",
      "          [ 0.0012, -0.0342,  0.0208],\n",
      "          [ 0.0039, -0.0028, -0.0167]],\n",
      "\n",
      "         [[-0.0334, -0.0012,  0.0316],\n",
      "          [-0.0044,  0.0282, -0.0333],\n",
      "          [ 0.0454,  0.0249, -0.0017]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[ 0.0026,  0.0333, -0.0107],\n",
      "          [-0.0369, -0.0190,  0.0282],\n",
      "          [-0.0235, -0.0271, -0.0189]],\n",
      "\n",
      "         [[-0.0098,  0.0073,  0.0291],\n",
      "          [-0.0178,  0.0291, -0.0086],\n",
      "          [-0.0342, -0.0099, -0.0122]],\n",
      "\n",
      "         [[-0.0088, -0.0072, -0.0423],\n",
      "          [-0.0032,  0.0146,  0.0112],\n",
      "          [ 0.0225, -0.0086,  0.0038]]],\n",
      "\n",
      "\n",
      "        [[[ 0.0058, -0.0195, -0.0086],\n",
      "          [ 0.0054, -0.0153,  0.0206],\n",
      "          [ 0.0103, -0.0215,  0.0055]],\n",
      "\n",
      "         [[-0.0434, -0.0122, -0.0285],\n",
      "          [ 0.0109, -0.0193,  0.0225],\n",
      "          [ 0.0004,  0.0206, -0.0134]],\n",
      "\n",
      "         [[-0.0097, -0.0202,  0.0170],\n",
      "          [-0.0111, -0.0299,  0.0313],\n",
      "          [-0.0102,  0.0002, -0.0245]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[-0.0119,  0.0135,  0.0302],\n",
      "          [ 0.0215,  0.0085, -0.0280],\n",
      "          [-0.0682, -0.0459, -0.0197]],\n",
      "\n",
      "         [[-0.0110,  0.0128, -0.0368],\n",
      "          [-0.0299, -0.0190, -0.0076],\n",
      "          [-0.0119,  0.0100,  0.0142]],\n",
      "\n",
      "         [[-0.0124,  0.0064,  0.0216],\n",
      "          [ 0.0144, -0.0034,  0.0096],\n",
      "          [-0.0116,  0.0061,  0.0059]]],\n",
      "\n",
      "\n",
      "        [[[ 0.0098,  0.0018, -0.0207],\n",
      "          [-0.0126, -0.0114, -0.0295],\n",
      "          [-0.0185, -0.0137,  0.0021]],\n",
      "\n",
      "         [[ 0.0030, -0.0152, -0.0028],\n",
      "          [-0.0029, -0.0221,  0.0248],\n",
      "          [-0.0282,  0.0132, -0.0150]],\n",
      "\n",
      "         [[-0.0125,  0.0210,  0.0120],\n",
      "          [ 0.0027, -0.0030, -0.0191],\n",
      "          [ 0.0091,  0.0183, -0.0047]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[-0.0366, -0.0115, -0.0144],\n",
      "          [ 0.0029, -0.0067,  0.0081],\n",
      "          [-0.0125, -0.0272, -0.0416]],\n",
      "\n",
      "         [[ 0.0119,  0.0436, -0.0016],\n",
      "          [-0.0006, -0.0090, -0.0537],\n",
      "          [ 0.0069,  0.0170, -0.0272]],\n",
      "\n",
      "         [[ 0.0275,  0.0139,  0.0059],\n",
      "          [ 0.0394,  0.0079,  0.0322],\n",
      "          [ 0.0034, -0.0389,  0.0239]]]], requires_grad=True)\n",
      "Parameter containing:\n",
      "tensor([1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1.], requires_grad=True)\n",
      "Parameter containing:\n",
      "tensor([0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0.], requires_grad=True)\n",
      "Parameter containing:\n",
      "tensor([[[[-0.0156]],\n",
      "\n",
      "         [[ 0.0622]],\n",
      "\n",
      "         [[ 0.0243]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[-0.0774]],\n",
      "\n",
      "         [[-0.0525]],\n",
      "\n",
      "         [[-0.0294]]],\n",
      "\n",
      "\n",
      "        [[[ 0.0314]],\n",
      "\n",
      "         [[ 0.0059]],\n",
      "\n",
      "         [[-0.0123]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[ 0.0009]],\n",
      "\n",
      "         [[ 0.0040]],\n",
      "\n",
      "         [[ 0.0016]]],\n",
      "\n",
      "\n",
      "        [[[-0.0008]],\n",
      "\n",
      "         [[ 0.0190]],\n",
      "\n",
      "         [[ 0.0080]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[ 0.0361]],\n",
      "\n",
      "         [[ 0.0234]],\n",
      "\n",
      "         [[-0.0310]]],\n",
      "\n",
      "\n",
      "        ...,\n",
      "\n",
      "\n",
      "        [[[-0.0431]],\n",
      "\n",
      "         [[-0.0153]],\n",
      "\n",
      "         [[-0.0101]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[ 0.0028]],\n",
      "\n",
      "         [[ 0.0098]],\n",
      "\n",
      "         [[-0.0019]]],\n",
      "\n",
      "\n",
      "        [[[-0.0258]],\n",
      "\n",
      "         [[-0.0155]],\n",
      "\n",
      "         [[ 0.0403]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[ 0.0189]],\n",
      "\n",
      "         [[-0.0128]],\n",
      "\n",
      "         [[-0.0189]]],\n",
      "\n",
      "\n",
      "        [[[ 0.0221]],\n",
      "\n",
      "         [[-0.0092]],\n",
      "\n",
      "         [[-0.0287]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[-0.0515]],\n",
      "\n",
      "         [[ 0.0424]],\n",
      "\n",
      "         [[-0.0177]]]], requires_grad=True)\n",
      "Parameter containing:\n",
      "tensor([1., 1., 1.,  ..., 1., 1., 1.], requires_grad=True)\n",
      "Parameter containing:\n",
      "tensor([0., 0., 0.,  ..., 0., 0., 0.], requires_grad=True)\n",
      "Parameter containing:\n",
      "tensor([[[[ 0.0168]],\n",
      "\n",
      "         [[ 0.0318]],\n",
      "\n",
      "         [[ 0.0107]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[ 0.1440]],\n",
      "\n",
      "         [[-0.1364]],\n",
      "\n",
      "         [[ 0.0041]]],\n",
      "\n",
      "\n",
      "        [[[ 0.1488]],\n",
      "\n",
      "         [[-0.1068]],\n",
      "\n",
      "         [[ 0.0874]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[-0.0274]],\n",
      "\n",
      "         [[ 0.0488]],\n",
      "\n",
      "         [[ 0.0852]]],\n",
      "\n",
      "\n",
      "        [[[-0.0081]],\n",
      "\n",
      "         [[-0.0069]],\n",
      "\n",
      "         [[-0.0457]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[ 0.0139]],\n",
      "\n",
      "         [[-0.1019]],\n",
      "\n",
      "         [[-0.0427]]],\n",
      "\n",
      "\n",
      "        ...,\n",
      "\n",
      "\n",
      "        [[[-0.0055]],\n",
      "\n",
      "         [[ 0.1249]],\n",
      "\n",
      "         [[-0.0015]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[ 0.0297]],\n",
      "\n",
      "         [[ 0.0582]],\n",
      "\n",
      "         [[ 0.0655]]],\n",
      "\n",
      "\n",
      "        [[[ 0.0355]],\n",
      "\n",
      "         [[-0.0213]],\n",
      "\n",
      "         [[-0.0180]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[ 0.0228]],\n",
      "\n",
      "         [[-0.0861]],\n",
      "\n",
      "         [[ 0.0191]]],\n",
      "\n",
      "\n",
      "        [[[ 0.0387]],\n",
      "\n",
      "         [[-0.0127]],\n",
      "\n",
      "         [[ 0.0583]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[-0.0824]],\n",
      "\n",
      "         [[ 0.1040]],\n",
      "\n",
      "         [[-0.0205]]]], requires_grad=True)\n",
      "Parameter containing:\n",
      "tensor([1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1.], requires_grad=True)\n",
      "Parameter containing:\n",
      "tensor([0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0.], requires_grad=True)\n",
      "Parameter containing:\n",
      "tensor([[[[ 0.0271,  0.0157, -0.0070],\n",
      "          [-0.0109,  0.0373,  0.0030],\n",
      "          [-0.0076,  0.0142,  0.0141]],\n",
      "\n",
      "         [[ 0.0209, -0.0128, -0.0257],\n",
      "          [ 0.0008,  0.0005,  0.0182],\n",
      "          [ 0.0129, -0.0117,  0.0184]],\n",
      "\n",
      "         [[ 0.0174, -0.0114, -0.0176],\n",
      "          [ 0.0355, -0.0366,  0.0067],\n",
      "          [ 0.0065, -0.0126, -0.0037]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[ 0.0084,  0.0394, -0.0065],\n",
      "          [ 0.0001, -0.0001, -0.0015],\n",
      "          [-0.0079, -0.0042, -0.0430]],\n",
      "\n",
      "         [[ 0.0337,  0.0284, -0.0128],\n",
      "          [-0.0107, -0.0288,  0.0394],\n",
      "          [-0.0130, -0.0104,  0.0517]],\n",
      "\n",
      "         [[ 0.0012,  0.0058,  0.0225],\n",
      "          [-0.0330, -0.0006, -0.0210],\n",
      "          [-0.0375,  0.0224, -0.0120]]],\n",
      "\n",
      "\n",
      "        [[[-0.0282,  0.0177, -0.0306],\n",
      "          [ 0.0223, -0.0091, -0.0029],\n",
      "          [-0.0340, -0.0135,  0.0091]],\n",
      "\n",
      "         [[ 0.0100, -0.0092, -0.0454],\n",
      "          [ 0.0096, -0.0121, -0.0109],\n",
      "          [ 0.0161, -0.0185,  0.0100]],\n",
      "\n",
      "         [[-0.0064, -0.0177,  0.0087],\n",
      "          [ 0.0085,  0.0192,  0.0072],\n",
      "          [ 0.0101, -0.0373, -0.0029]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[-0.0064, -0.0062,  0.0036],\n",
      "          [ 0.0002,  0.0123,  0.0267],\n",
      "          [ 0.0241, -0.0107,  0.0024]],\n",
      "\n",
      "         [[ 0.0028,  0.0148, -0.0161],\n",
      "          [-0.0096,  0.0193,  0.0119],\n",
      "          [ 0.0019, -0.0110, -0.0103]],\n",
      "\n",
      "         [[ 0.0063,  0.0171, -0.0065],\n",
      "          [ 0.0101, -0.0059,  0.0166],\n",
      "          [ 0.0135, -0.0307, -0.0089]]],\n",
      "\n",
      "\n",
      "        [[[ 0.0001,  0.0307, -0.0305],\n",
      "          [-0.0102,  0.0031,  0.0141],\n",
      "          [-0.0471, -0.0080,  0.0379]],\n",
      "\n",
      "         [[-0.0114,  0.0037, -0.0310],\n",
      "          [-0.0240,  0.0219,  0.0166],\n",
      "          [-0.0364,  0.0209, -0.0015]],\n",
      "\n",
      "         [[ 0.0013, -0.0342, -0.0053],\n",
      "          [-0.0036, -0.0039, -0.0314],\n",
      "          [-0.0017,  0.0252,  0.0198]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[ 0.0031, -0.0150, -0.0108],\n",
      "          [-0.0078,  0.0062,  0.0174],\n",
      "          [ 0.0202,  0.0006,  0.0252]],\n",
      "\n",
      "         [[-0.0148, -0.0335,  0.0039],\n",
      "          [ 0.0086,  0.0027, -0.0007],\n",
      "          [ 0.0082,  0.0118,  0.0211]],\n",
      "\n",
      "         [[ 0.0139, -0.0223,  0.0374],\n",
      "          [ 0.0038,  0.0040, -0.0064],\n",
      "          [ 0.0258, -0.0020,  0.0120]]],\n",
      "\n",
      "\n",
      "        ...,\n",
      "\n",
      "\n",
      "        [[[ 0.0133, -0.0278, -0.0203],\n",
      "          [-0.0173, -0.0056,  0.0070],\n",
      "          [ 0.0141, -0.0075,  0.0008]],\n",
      "\n",
      "         [[ 0.0269, -0.0125,  0.0227],\n",
      "          [ 0.0158, -0.0241, -0.0335],\n",
      "          [ 0.0001,  0.0163,  0.0089]],\n",
      "\n",
      "         [[ 0.0093, -0.0158,  0.0009],\n",
      "          [ 0.0063, -0.0081,  0.0057],\n",
      "          [-0.0006,  0.0119,  0.0291]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[-0.0259, -0.0032,  0.0024],\n",
      "          [-0.0019, -0.0063,  0.0041],\n",
      "          [-0.0270,  0.0231, -0.0019]],\n",
      "\n",
      "         [[ 0.0041, -0.0083,  0.0088],\n",
      "          [-0.0069, -0.0248, -0.0352],\n",
      "          [ 0.0031, -0.0298,  0.0496]],\n",
      "\n",
      "         [[-0.0199,  0.0053,  0.0012],\n",
      "          [-0.0252,  0.0189, -0.0119],\n",
      "          [ 0.0087,  0.0136,  0.0133]]],\n",
      "\n",
      "\n",
      "        [[[ 0.0005, -0.0030,  0.0124],\n",
      "          [-0.0007,  0.0055,  0.0347],\n",
      "          [-0.0056, -0.0116,  0.0154]],\n",
      "\n",
      "         [[ 0.0049,  0.0446, -0.0115],\n",
      "          [ 0.0106,  0.0004,  0.0344],\n",
      "          [-0.0162, -0.0236,  0.0123]],\n",
      "\n",
      "         [[-0.0019,  0.0310,  0.0079],\n",
      "          [-0.0128,  0.0154, -0.0308],\n",
      "          [ 0.0108,  0.0125, -0.0180]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[-0.0392, -0.0409,  0.0366],\n",
      "          [-0.0018,  0.0071, -0.0033],\n",
      "          [-0.0063,  0.0300,  0.0300]],\n",
      "\n",
      "         [[-0.0156,  0.0015, -0.0346],\n",
      "          [-0.0337,  0.0111,  0.0214],\n",
      "          [-0.0275,  0.0348, -0.0332]],\n",
      "\n",
      "         [[ 0.0121, -0.0135,  0.0171],\n",
      "          [-0.0215, -0.0140, -0.0052],\n",
      "          [-0.0071, -0.0578,  0.0034]]],\n",
      "\n",
      "\n",
      "        [[[ 0.0173, -0.0073, -0.0174],\n",
      "          [ 0.0020, -0.0195, -0.0008],\n",
      "          [ 0.0068,  0.0028,  0.0119]],\n",
      "\n",
      "         [[-0.0228,  0.0201,  0.0088],\n",
      "          [-0.0074,  0.0178, -0.0062],\n",
      "          [-0.0117,  0.0510, -0.0141]],\n",
      "\n",
      "         [[-0.0059,  0.0086, -0.0193],\n",
      "          [-0.0363, -0.0198,  0.0011],\n",
      "          [ 0.0082,  0.0083,  0.0046]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[ 0.0116,  0.0322,  0.0234],\n",
      "          [-0.0285, -0.0183, -0.0086],\n",
      "          [-0.0560,  0.0312, -0.0134]],\n",
      "\n",
      "         [[-0.0314,  0.0081,  0.0202],\n",
      "          [ 0.0192,  0.0432, -0.0157],\n",
      "          [-0.0042, -0.0042, -0.0080]],\n",
      "\n",
      "         [[ 0.0181, -0.0039,  0.0076],\n",
      "          [-0.0104, -0.0520, -0.0319],\n",
      "          [-0.0225,  0.0242,  0.0214]]]], requires_grad=True)\n",
      "Parameter containing:\n",
      "tensor([1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,\n",
      "        1., 1., 1., 1., 1., 1., 1., 1.], requires_grad=True)\n",
      "Parameter containing:\n",
      "tensor([0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
      "        0., 0., 0., 0., 0., 0., 0., 0.], requires_grad=True)\n",
      "Parameter containing:\n",
      "tensor([[[[-0.0066]],\n",
      "\n",
      "         [[-0.0501]],\n",
      "\n",
      "         [[ 0.0516]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[-0.0539]],\n",
      "\n",
      "         [[ 0.0321]],\n",
      "\n",
      "         [[ 0.0433]]],\n",
      "\n",
      "\n",
      "        [[[ 0.0408]],\n",
      "\n",
      "         [[ 0.0470]],\n",
      "\n",
      "         [[ 0.0677]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[-0.0262]],\n",
      "\n",
      "         [[-0.0482]],\n",
      "\n",
      "         [[-0.0375]]],\n",
      "\n",
      "\n",
      "        [[[-0.0152]],\n",
      "\n",
      "         [[ 0.0069]],\n",
      "\n",
      "         [[ 0.1056]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[-0.0052]],\n",
      "\n",
      "         [[-0.0024]],\n",
      "\n",
      "         [[ 0.0064]]],\n",
      "\n",
      "\n",
      "        ...,\n",
      "\n",
      "\n",
      "        [[[ 0.0354]],\n",
      "\n",
      "         [[-0.0005]],\n",
      "\n",
      "         [[-0.0256]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[ 0.0022]],\n",
      "\n",
      "         [[-0.0020]],\n",
      "\n",
      "         [[-0.0063]]],\n",
      "\n",
      "\n",
      "        [[[-0.0506]],\n",
      "\n",
      "         [[-0.0207]],\n",
      "\n",
      "         [[-0.0297]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[ 0.0878]],\n",
      "\n",
      "         [[ 0.0030]],\n",
      "\n",
      "         [[-0.0123]]],\n",
      "\n",
      "\n",
      "        [[[ 0.0022]],\n",
      "\n",
      "         [[-0.0261]],\n",
      "\n",
      "         [[ 0.0371]],\n",
      "\n",
      "         ...,\n",
      "\n",
      "         [[-0.0630]],\n",
      "\n",
      "         [[ 0.0264]],\n",
      "\n",
      "         [[ 0.0051]]]], requires_grad=True)\n",
      "Parameter containing:\n",
      "tensor([1., 1., 1.,  ..., 1., 1., 1.], requires_grad=True)\n",
      "Parameter containing:\n",
      "tensor([0., 0., 0.,  ..., 0., 0., 0.], requires_grad=True)\n",
      "Parameter containing:\n",
      "tensor([[ 1.6682e-03,  7.0758e-03,  9.5269e-03,  ..., -7.6639e-03,\n",
      "         -1.0844e-02,  1.5807e-02],\n",
      "        [ 1.9008e-02, -1.9891e-02,  2.1546e-02,  ...,  6.8192e-03,\n",
      "          1.4673e-02,  5.2995e-03],\n",
      "        [-3.6234e-05, -4.0254e-03,  2.1461e-02,  ..., -1.0849e-02,\n",
      "         -4.5506e-03,  5.9968e-03],\n",
      "        ...,\n",
      "        [ 2.0796e-02, -1.3401e-02, -3.8263e-03,  ..., -9.6361e-03,\n",
      "         -3.5342e-03,  1.8948e-02],\n",
      "        [-1.9292e-02, -1.8049e-03,  1.8755e-02,  ..., -8.0715e-03,\n",
      "          1.9901e-02, -7.7262e-03],\n",
      "        [-2.0524e-02,  1.3747e-02, -3.1916e-03,  ...,  2.1428e-02,\n",
      "          4.6511e-03,  1.6255e-02]], requires_grad=True)\n",
      "Parameter containing:\n",
      "tensor([-1.2728e-02, -1.5014e-02, -3.8270e-03, -7.6295e-04, -1.0053e-02,\n",
      "         2.8012e-03,  5.2336e-03, -1.3199e-02,  1.5968e-02,  8.3886e-03,\n",
      "        -1.4685e-02,  1.7516e-02,  3.3843e-03, -2.0427e-02,  2.3952e-03,\n",
      "        -1.8397e-02, -1.7721e-02,  1.1945e-02, -2.0311e-02, -1.3888e-02,\n",
      "        -1.6689e-02, -1.3609e-02, -1.1256e-02, -4.5608e-04, -1.0755e-02,\n",
      "         2.0609e-02, -8.8968e-03, -1.0347e-02,  8.5742e-03, -1.3574e-02,\n",
      "        -1.4434e-02,  1.7547e-02, -4.9570e-04,  1.9457e-02,  1.0624e-02,\n",
      "        -1.9817e-02,  8.4063e-03,  1.7485e-02, -4.4546e-03,  8.4330e-03,\n",
      "        -8.9778e-03, -2.4051e-03,  3.7525e-03, -3.9556e-04,  5.8806e-03,\n",
      "        -1.4597e-02, -1.4185e-02, -1.8686e-02,  1.3237e-02, -1.7272e-02,\n",
      "        -1.6586e-03, -1.1694e-02,  9.9074e-03, -1.7834e-02,  2.0484e-02,\n",
      "         6.0217e-03, -1.7323e-02,  5.4788e-03, -1.3369e-02,  9.3343e-04,\n",
      "        -8.5988e-03,  1.9214e-02,  1.5769e-02, -1.1018e-02, -1.0891e-02,\n",
      "        -9.8356e-03,  1.4217e-02,  2.0580e-02, -1.1829e-02,  1.9479e-02,\n",
      "        -4.0232e-03, -2.2050e-02,  4.1784e-03,  2.0341e-02, -5.3097e-03,\n",
      "        -1.7391e-03, -2.0167e-02,  1.3365e-02, -1.1277e-03, -1.8563e-02,\n",
      "         1.6233e-02,  7.9407e-03, -1.6138e-02,  1.8062e-04,  5.7154e-03,\n",
      "         1.4531e-02,  6.1608e-03, -5.4625e-03, -6.2725e-03,  7.1506e-03,\n",
      "         1.7337e-02,  6.8510e-03, -1.1802e-02, -7.7270e-03, -8.0540e-03,\n",
      "         2.1269e-02, -1.6507e-02,  2.0191e-02,  1.9858e-02,  1.0595e-04,\n",
      "         6.2419e-03,  1.8899e-02, -1.2087e-02,  9.6008e-03,  2.0787e-02,\n",
      "         1.2004e-02, -1.8380e-02,  2.0753e-02, -3.1723e-03, -9.1817e-03,\n",
      "        -3.5055e-03,  6.1607e-03, -1.1365e-02,  1.5394e-02, -1.2974e-02,\n",
      "        -1.5052e-02,  2.1546e-02, -7.1659e-03, -6.1171e-03,  1.7113e-02,\n",
      "        -2.6889e-05,  1.0666e-02,  1.4387e-02, -1.8154e-02,  3.6079e-03,\n",
      "         1.2670e-02, -1.6380e-02, -1.9855e-02, -7.5371e-03,  9.7148e-04,\n",
      "         9.7445e-03,  2.1942e-02, -1.0595e-02, -7.7757e-03,  1.5711e-03,\n",
      "        -9.4213e-03,  1.0688e-02,  1.7976e-02,  1.3752e-02,  8.8434e-03,\n",
      "        -2.0741e-02, -1.8840e-02, -1.5854e-02, -3.1613e-03, -1.9010e-02,\n",
      "        -1.5090e-02, -5.9325e-03, -1.1180e-02, -1.3933e-02,  1.0200e-02,\n",
      "         7.7584e-03,  8.8431e-03,  2.0585e-02, -1.8089e-02,  7.4825e-03,\n",
      "         1.1336e-02,  7.3158e-03, -1.6990e-02,  2.1093e-02,  1.5846e-02,\n",
      "         5.5414e-03,  2.0059e-02,  2.8740e-03, -4.2276e-03, -9.7020e-03,\n",
      "         1.2028e-02, -1.5253e-02,  1.2346e-02,  3.0940e-03,  9.1402e-03,\n",
      "        -1.6670e-02, -1.9810e-02, -7.2631e-03, -2.4350e-03,  9.3434e-03,\n",
      "         1.2210e-02,  2.0918e-02,  1.5098e-02,  1.5842e-02, -1.8647e-02,\n",
      "         1.1455e-02,  1.3157e-02,  1.1463e-02,  2.0714e-02, -1.6704e-02,\n",
      "        -1.6090e-02, -7.0128e-03,  1.3023e-02,  1.0840e-02,  1.9664e-02,\n",
      "         1.1692e-02,  1.9067e-03, -1.3109e-02, -8.8425e-03,  1.2959e-02,\n",
      "        -6.4751e-03, -1.8620e-02, -1.9059e-02,  1.3111e-03, -1.0580e-02,\n",
      "         7.0063e-03,  9.8280e-03,  1.1720e-02, -6.8668e-03, -2.8211e-03,\n",
      "         3.7160e-03, -1.2005e-02, -2.1167e-02,  1.5986e-02,  7.9158e-03,\n",
      "        -1.0328e-02,  1.8113e-02, -1.5935e-02, -8.9706e-04, -3.4443e-03,\n",
      "         3.0653e-03,  1.5055e-02,  1.5910e-02, -5.5295e-03, -1.7225e-02,\n",
      "        -1.9023e-02, -1.8896e-02,  3.1749e-03,  6.2529e-03,  6.7038e-03,\n",
      "        -1.2703e-02, -1.1856e-02, -1.8660e-02,  7.3165e-03, -1.8528e-04,\n",
      "         7.5708e-03,  9.2874e-03, -1.3512e-02, -1.4405e-02, -5.4810e-03,\n",
      "         1.4237e-02,  9.1352e-03, -6.4879e-03,  1.7574e-02,  1.7890e-02,\n",
      "        -1.3804e-02, -1.1025e-02, -4.0870e-05, -1.7573e-02, -1.5454e-02,\n",
      "         2.1777e-02,  5.5525e-03,  1.0643e-02, -1.3794e-02,  1.7958e-02,\n",
      "        -7.6434e-03, -1.0707e-02,  1.2712e-02,  1.9897e-02, -1.7874e-02,\n",
      "         1.6919e-02, -1.7816e-02,  6.1088e-03, -7.3977e-03,  8.6979e-03,\n",
      "        -8.1350e-03, -4.0316e-03,  1.9202e-02, -1.2060e-02,  1.2513e-02,\n",
      "        -1.3381e-02, -1.4334e-02, -2.0316e-02,  1.1359e-02, -6.6061e-03,\n",
      "         4.4457e-03,  1.1792e-02, -3.0397e-03, -8.8598e-03, -2.1438e-03,\n",
      "        -1.3900e-02,  1.1319e-02,  5.5874e-03, -1.1410e-02,  1.9205e-02,\n",
      "         2.4158e-03, -1.0077e-02,  1.1785e-02,  6.2850e-03,  3.7506e-04,\n",
      "        -1.1925e-05, -5.8881e-03,  9.0487e-04,  8.5811e-03, -1.7796e-02,\n",
      "         2.4812e-03, -1.9181e-02, -8.6259e-03, -1.1539e-02, -1.7014e-03,\n",
      "        -1.6416e-02, -5.9479e-03,  1.6581e-02, -1.5289e-02,  1.1876e-02,\n",
      "        -8.1665e-03,  1.3057e-02, -1.1610e-02,  9.7474e-03, -5.0975e-04,\n",
      "        -5.6734e-03, -1.8260e-03,  3.6348e-03,  2.0168e-03, -1.4871e-02,\n",
      "        -8.8852e-04, -9.6178e-03, -1.1315e-02,  1.3694e-02,  4.0152e-03,\n",
      "        -1.4960e-02,  6.0569e-03,  2.1935e-02, -1.1188e-02, -1.3225e-02,\n",
      "        -4.6094e-03,  6.1245e-03,  1.5811e-02,  1.8871e-02,  1.4213e-03,\n",
      "        -9.2272e-03, -8.5876e-03, -4.0706e-03,  1.3667e-02,  1.8313e-02,\n",
      "         1.7321e-02, -1.6022e-02, -1.6677e-02,  1.8061e-02,  1.6810e-03,\n",
      "        -3.0227e-03, -2.2056e-03, -2.1343e-02,  1.9701e-03,  4.1769e-03,\n",
      "         8.7099e-03,  2.1666e-02, -1.7327e-02,  1.0170e-02,  7.3971e-03,\n",
      "         2.1879e-02, -2.1905e-03,  5.8471e-04,  4.9106e-03,  9.8066e-03,\n",
      "         1.4257e-03,  2.8510e-03,  1.4511e-02, -1.9549e-02,  1.4540e-02,\n",
      "        -1.0289e-02,  1.6750e-02, -5.4532e-03, -1.8908e-02,  1.3604e-02,\n",
      "        -1.0940e-02,  6.0827e-03,  1.5969e-02, -8.6737e-03,  1.7284e-02,\n",
      "        -1.2496e-02, -1.2859e-02, -2.5459e-03,  1.2423e-02,  1.9338e-02,\n",
      "         2.4947e-03,  1.6574e-03,  1.2994e-02, -1.5933e-02,  6.1111e-03,\n",
      "         2.1765e-02, -3.0208e-03,  1.2183e-02, -1.9044e-02, -2.0375e-03,\n",
      "         4.0060e-03,  2.1563e-02,  1.7950e-02, -1.0020e-02, -1.0165e-02,\n",
      "         5.2515e-03, -1.2034e-02,  1.6373e-02, -4.0091e-03, -3.9905e-03,\n",
      "         1.3988e-02, -8.4857e-03,  1.1013e-03,  1.3096e-02, -1.8805e-02,\n",
      "        -2.1950e-02,  2.2924e-03,  3.5778e-03,  2.0328e-02,  3.8545e-03,\n",
      "         1.6430e-02, -1.9544e-02, -2.2902e-04,  6.7925e-03, -1.0941e-02,\n",
      "        -9.7605e-03,  2.0497e-02,  6.8392e-03, -1.6344e-02, -9.5020e-03,\n",
      "         9.5272e-03,  7.7117e-04,  1.4450e-02,  6.4724e-03,  1.9404e-02,\n",
      "        -4.9091e-03, -1.7229e-02, -2.0215e-02,  5.8156e-03,  2.1589e-02,\n",
      "        -1.7389e-02, -2.1943e-02,  1.8552e-02, -5.7087e-04, -8.7777e-03,\n",
      "         2.1698e-02,  2.1385e-02, -5.5235e-03, -1.5792e-02,  1.2601e-03,\n",
      "        -1.6026e-02,  8.9496e-03,  5.8059e-03, -2.1159e-02, -2.1613e-02,\n",
      "        -1.7342e-02, -1.2745e-02,  5.0485e-03,  1.3178e-02, -1.2068e-02,\n",
      "         4.2402e-03,  1.8640e-02,  9.3766e-05,  4.3545e-04,  1.2751e-02,\n",
      "         1.6465e-02, -9.7760e-03,  9.0490e-03,  3.8315e-03, -1.0375e-02,\n",
      "        -1.4810e-02,  1.9545e-02,  9.4098e-03,  2.1092e-02, -1.2929e-05,\n",
      "        -6.9694e-04,  1.8334e-03, -6.9148e-03,  1.9975e-03, -1.6467e-02,\n",
      "         1.1392e-02,  1.0429e-02,  6.0460e-03,  1.3611e-02, -8.7531e-03,\n",
      "         1.3703e-02, -8.1410e-04, -1.2023e-02, -2.1128e-02, -9.3933e-03,\n",
      "         8.0559e-03, -1.5565e-03, -2.0209e-02,  1.2813e-02, -9.0839e-03,\n",
      "        -1.3384e-02,  1.7522e-02, -6.8227e-03, -2.0788e-03, -2.1547e-02,\n",
      "        -1.0894e-03, -9.9869e-03,  2.1670e-02,  1.5041e-02,  1.8838e-02,\n",
      "        -6.3834e-03, -1.5626e-02, -2.1104e-02, -2.0066e-02,  5.6055e-03,\n",
      "        -1.6191e-02, -1.5552e-02,  1.2187e-02, -1.3070e-02, -1.9038e-02,\n",
      "         4.3012e-03,  1.4206e-02, -1.8238e-02, -1.8129e-02, -1.6225e-02,\n",
      "        -2.0379e-02, -1.4553e-02,  4.9369e-03,  1.0726e-02,  2.6023e-03,\n",
      "         2.1531e-02,  2.1887e-03,  1.1617e-02,  3.8486e-03, -1.6102e-02,\n",
      "         1.6447e-03,  1.1150e-02,  9.1607e-04,  1.0330e-02,  1.5561e-02,\n",
      "        -9.1084e-04,  1.3866e-02,  1.6800e-02,  1.8746e-02,  1.8518e-02,\n",
      "         1.2840e-02, -1.0760e-02, -1.4236e-02, -1.9006e-02, -9.1484e-03,\n",
      "         1.5583e-02,  1.3227e-02, -1.6388e-02, -1.4225e-02, -8.6714e-03,\n",
      "        -1.7446e-02, -2.0156e-02, -1.0629e-02, -1.6107e-02,  7.9489e-04,\n",
      "        -1.3605e-02, -6.9941e-03,  1.5764e-02,  1.2754e-02, -4.6783e-03,\n",
      "        -2.7070e-03,  6.7684e-03, -2.0674e-03,  5.7603e-03,  1.3768e-02,\n",
      "        -1.3815e-02, -1.3603e-02,  9.9677e-03, -5.7729e-04, -1.2694e-02,\n",
      "         7.1768e-03,  1.1537e-02,  1.0199e-02,  7.0154e-03, -9.3405e-03,\n",
      "         4.1787e-03,  9.0126e-03,  4.6344e-03, -6.2404e-04,  7.0906e-03,\n",
      "         1.1477e-03,  1.7248e-02, -1.8673e-02,  2.3636e-04,  6.8046e-03,\n",
      "        -4.3892e-03,  5.0427e-03,  9.1086e-03, -1.1176e-02, -1.9031e-02,\n",
      "        -1.2295e-02, -1.7129e-02, -3.5194e-03, -2.2016e-02, -1.5903e-02,\n",
      "         1.9759e-02,  2.8362e-03, -1.7653e-02,  1.6587e-02,  1.2919e-02,\n",
      "        -2.0966e-02, -7.9758e-03, -2.2079e-02,  1.1416e-02, -8.1355e-03,\n",
      "        -8.3043e-03,  2.0359e-02, -1.3625e-02, -1.0251e-02,  6.6159e-03,\n",
      "        -7.7268e-04,  8.5245e-03,  8.0464e-03, -1.0087e-02,  1.6093e-03,\n",
      "        -1.4680e-02,  3.7856e-03,  1.2652e-03,  3.0416e-03,  1.9996e-02,\n",
      "        -1.5358e-02,  1.3160e-02,  1.4193e-02,  7.6226e-03, -1.4581e-02,\n",
      "        -1.9174e-04, -2.5164e-03,  1.7540e-02, -1.4396e-02, -1.5368e-02,\n",
      "         2.9614e-03, -9.5711e-03,  1.6125e-02, -1.2739e-02, -1.8259e-02,\n",
      "         2.1220e-02,  8.3511e-03,  1.2825e-02, -1.5802e-02,  1.4361e-03,\n",
      "        -1.1969e-02, -8.6187e-03, -8.6493e-03,  9.1584e-03,  2.1676e-02,\n",
      "        -3.2075e-04,  2.4576e-03,  2.1287e-02, -3.8942e-03,  1.8283e-02,\n",
      "         7.2624e-03,  1.4787e-02, -1.5438e-02, -1.7911e-02, -9.8249e-03,\n",
      "        -2.1542e-02,  1.5791e-02, -2.0820e-02, -1.4635e-02, -1.1710e-02,\n",
      "        -4.4667e-03, -2.9704e-03,  6.8136e-03,  1.4855e-02,  2.2959e-03,\n",
      "         1.8827e-02, -1.2818e-02,  2.1064e-03, -1.0102e-02, -1.9816e-02,\n",
      "        -5.2724e-03, -1.7818e-02, -2.1560e-02,  3.8777e-03, -1.4036e-02,\n",
      "         1.8258e-02, -7.1175e-03,  9.1119e-03, -2.5835e-03,  1.8887e-02,\n",
      "         1.4433e-02,  1.4698e-02,  1.8233e-02, -8.2320e-03,  5.6497e-03,\n",
      "         2.1704e-02, -9.4287e-05, -7.8503e-03,  7.8087e-04,  1.4980e-02,\n",
      "        -1.0309e-02,  1.7588e-02,  1.7262e-02,  2.0067e-02, -2.2025e-02,\n",
      "        -2.1140e-02, -1.1037e-02,  1.3123e-03, -2.1785e-02, -1.5209e-02,\n",
      "         3.4623e-05,  1.8419e-02,  2.1943e-02,  7.5834e-03, -1.4977e-02,\n",
      "         1.3991e-02, -9.7478e-03,  2.8207e-03,  7.6422e-03,  1.7638e-02,\n",
      "        -8.7829e-03,  8.9411e-03,  1.8467e-02,  1.0733e-02,  1.6348e-02,\n",
      "         1.9779e-02, -2.4714e-03,  3.6438e-03,  6.2349e-03,  1.6117e-02,\n",
      "        -1.1976e-02,  4.3738e-03,  3.1230e-03,  1.8560e-02, -1.9707e-02,\n",
      "         2.3232e-03,  2.1707e-02,  1.1722e-02, -1.1092e-02,  1.2927e-02,\n",
      "         1.2334e-02,  1.2984e-03, -2.0979e-02, -7.0643e-03, -7.4460e-03,\n",
      "         4.0243e-03, -1.4965e-02,  4.9570e-03, -1.6694e-03,  4.2841e-03,\n",
      "         1.7376e-03,  3.5128e-03,  1.3782e-02,  1.1273e-02, -9.0975e-03,\n",
      "        -2.7900e-03, -4.8319e-03, -1.8336e-02,  1.5680e-02,  9.0714e-03,\n",
      "        -1.8078e-02,  1.0122e-03, -1.6991e-02, -8.9245e-03, -1.2185e-02,\n",
      "        -1.0991e-02, -1.2587e-02, -1.6874e-02, -3.7612e-03, -1.9403e-02,\n",
      "         5.3854e-03,  1.7764e-02, -1.6663e-02,  4.8159e-04,  1.3891e-03,\n",
      "         2.0551e-02, -6.1506e-03,  1.5413e-02,  1.8718e-02, -3.3166e-03,\n",
      "        -2.8417e-03, -8.9551e-03, -2.8043e-03, -1.5523e-02,  3.3351e-03,\n",
      "        -1.9034e-02,  1.7156e-02, -2.3859e-03, -2.1884e-02,  1.8556e-02,\n",
      "         1.1798e-02, -1.4705e-02,  5.8677e-03,  5.3092e-03,  2.0251e-02,\n",
      "         1.1459e-02, -1.9141e-02,  1.9367e-02,  7.3569e-03, -8.5437e-03,\n",
      "         1.1623e-02, -1.5253e-02, -6.5228e-03, -8.3603e-04, -1.4242e-04,\n",
      "        -1.3674e-02, -1.4417e-02,  1.5294e-02,  2.1673e-02, -3.4535e-04,\n",
      "        -3.1118e-03, -1.1222e-02,  2.0159e-02,  1.6798e-02,  1.4202e-02,\n",
      "         2.1201e-02,  6.2446e-03,  2.5231e-03,  1.3835e-02,  1.3955e-02,\n",
      "        -7.3504e-05,  8.9882e-03, -1.0689e-02,  5.9578e-03,  5.7384e-03,\n",
      "         1.8860e-03,  1.3563e-02,  1.7437e-02, -1.2780e-02,  1.7979e-03,\n",
      "         2.4600e-04,  7.0522e-03, -7.4707e-03, -1.7987e-02, -1.1452e-02,\n",
      "         1.3305e-02,  1.2799e-02,  1.0206e-02, -2.4049e-03, -1.5902e-02,\n",
      "        -6.5735e-03, -1.2940e-02,  1.2201e-02,  8.2721e-03, -2.2037e-02,\n",
      "        -3.7438e-03,  6.3246e-04,  7.9168e-03, -5.4365e-03, -1.7492e-02,\n",
      "        -3.6588e-03,  7.6538e-03, -1.7329e-02,  9.5813e-03, -1.6436e-02,\n",
      "        -5.4025e-03, -1.9979e-02, -5.9576e-03,  1.3734e-02,  1.0761e-02,\n",
      "        -9.9895e-03, -1.3242e-02,  1.3827e-02,  1.8506e-02, -9.4789e-03,\n",
      "        -1.0928e-02, -1.4130e-02, -1.8114e-02, -1.3163e-02, -1.8296e-02,\n",
      "         1.4646e-02, -1.6488e-02,  2.0983e-02, -2.0126e-02,  4.1154e-04,\n",
      "         2.1656e-02,  8.9585e-04, -2.7872e-04,  1.7058e-02, -6.5169e-03,\n",
      "         8.9925e-03,  5.2577e-03, -1.3292e-02,  1.4099e-02, -8.7254e-03,\n",
      "        -2.1917e-02,  3.4592e-03,  6.9896e-03,  8.3830e-03,  8.7235e-03,\n",
      "        -6.6657e-03, -2.1187e-02, -2.1704e-02, -1.3193e-02,  2.0874e-02,\n",
      "         2.1993e-02,  1.2290e-02,  1.9280e-02, -2.6996e-03,  1.5626e-03,\n",
      "        -4.9233e-03, -1.7989e-02,  6.9684e-03, -1.8971e-02,  9.6533e-03,\n",
      "         3.3366e-03, -2.1424e-02,  3.5363e-03, -1.3006e-02,  2.0192e-02,\n",
      "        -1.6050e-02,  4.8594e-03, -2.7238e-03, -1.5753e-02,  8.0817e-03,\n",
      "        -1.9486e-02,  1.5159e-02,  1.6769e-03,  2.0220e-02,  1.8776e-03,\n",
      "         1.3967e-02, -7.4913e-03, -2.6560e-03, -6.6407e-03, -2.3490e-03,\n",
      "         6.5025e-03, -1.1953e-02,  2.7722e-03, -1.8621e-02,  1.1810e-02,\n",
      "        -8.8484e-03,  1.4931e-02, -3.3105e-03,  1.5878e-02, -2.0149e-02,\n",
      "         7.8326e-03,  1.6922e-02, -1.8906e-02,  4.8889e-03, -2.0069e-02,\n",
      "        -1.4512e-02,  7.1097e-03, -1.2168e-03, -5.7079e-04, -6.0768e-03,\n",
      "        -1.9242e-02, -7.0323e-03,  2.1230e-02,  2.4710e-03, -1.1076e-02,\n",
      "         1.7512e-02, -2.0999e-02,  1.2450e-02,  4.7847e-03, -4.6867e-04,\n",
      "         1.1343e-02, -1.7019e-02, -2.0373e-02,  1.6813e-02, -2.1253e-02,\n",
      "         1.7710e-02, -2.1920e-02,  1.2745e-02, -7.0813e-03, -1.2843e-02,\n",
      "         2.0586e-02, -1.2079e-02, -6.5914e-03, -8.3847e-03,  8.1358e-03,\n",
      "         6.8677e-03,  1.1888e-02, -1.4422e-03,  6.8105e-03,  7.1985e-03,\n",
      "        -2.9619e-03,  6.5091e-03, -1.9466e-02, -1.7549e-02,  1.5208e-02,\n",
      "         3.9493e-03,  4.0399e-03,  1.9668e-02, -6.5193e-03,  1.7998e-02,\n",
      "        -8.3127e-03, -7.9823e-03, -1.4086e-02,  1.0833e-02, -2.1830e-02,\n",
      "         6.8849e-04, -1.4897e-02,  1.9434e-02, -1.9211e-02,  2.0403e-02,\n",
      "        -4.3845e-03,  1.5612e-03,  5.9587e-03,  9.6959e-03, -7.9811e-03,\n",
      "         2.1021e-02, -1.5913e-02,  2.2550e-03,  1.9456e-02, -1.4606e-02,\n",
      "         2.0363e-02, -1.9856e-02, -9.4712e-03,  1.0587e-02,  8.5173e-03,\n",
      "        -1.6744e-02, -3.3463e-03,  8.1643e-03,  2.1927e-02,  1.7899e-02,\n",
      "         1.5718e-02,  1.7863e-02,  8.1589e-03,  1.2522e-02, -1.3487e-02,\n",
      "         1.4744e-02,  1.5162e-02, -6.4584e-03,  1.9373e-02, -1.6509e-02,\n",
      "         7.2542e-03,  5.7215e-03,  1.1252e-02,  1.1012e-02, -4.5516e-03],\n",
      "       requires_grad=True)\n",
      "Layer 1: torch.Size([1, 256, 56, 56])\n",
      "Layer 2: torch.Size([1, 512, 28, 28])\n",
      "Layer 3: torch.Size([1, 1024, 14, 14])\n",
      "Layer 4: torch.Size([1, 2048, 7, 7])\n"
     ]
    }
   ],
   "source": [
    "model_pretrained = utils_resnet.resnet50(ResnetFeatures, pretrained=True)\n",
    "x = torch.randn(1, 3, 224, 224)\n",
    "output = model_pretrained(x)\n",
    "print('Layer 1:', output[0].shape)\n",
    "print('Layer 2:', output[1].shape)\n",
    "print('Layer 3:', output[2].shape)\n",
    "print('Layer 4:', output[3].shape)"
   ]
  },
  {
   "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.7.3"
  }
 },
 "nbformat": 4,
 "nbformat_minor": 4
}